2
2
# -*- mode: python; coding: utf-8 -*-
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2011 Teddy Hogeborn
7
# Copyright © 2009-2011 Björn Påhlsson
9
# This program is free software: you can redistribute it and/or modify
10
# it under the terms of the GNU General Public License as published by
11
# the Free Software Foundation, either version 3 of the License, or
12
# (at your option) any later version.
14
# This program is distributed in the hope that it will be useful,
15
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
# GNU General Public License for more details.
19
# You should have received a copy of the GNU General Public License
20
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
# Contact the authors at <mandos@fukt.bsnet.se>.
4
from __future__ import division, absolute_import, with_statement
25
from __future__ import division, absolute_import, print_function, unicode_literals
44
65
"Parse an ISO 8601 date string to a datetime.datetime()"
47
d, t = iso.split(u"T", 1)
48
year, month, day = d.split(u"-", 2)
49
hour, minute, second = t.split(u":", 2)
68
d, t = iso.split("T", 1)
69
year, month, day = d.split("-", 2)
70
hour, minute, second = t.split(":", 2)
50
71
second, fraction = divmod(float(second), 1)
51
72
return datetime.datetime(int(year),
106
127
self.last_checker_failed = False
108
129
# The widget shown normally
109
self._text_widget = urwid.Text(u"")
130
self._text_widget = urwid.Text("")
110
131
# The widget shown when we have focus
111
self._focus_text_widget = urwid.Text(u"")
132
self._focus_text_widget = urwid.Text("")
112
133
super(MandosClientWidget, self).__init__(
113
134
update_hook=update_hook, delete_hook=delete_hook,
133
154
if self.need_approval:
134
155
self.using_timer(True)
136
self.proxy.connect_to_signal(u"CheckerCompleted",
157
self.proxy.connect_to_signal("CheckerCompleted",
137
158
self.checker_completed,
138
159
client_interface,
139
160
byte_arrays=True)
140
self.proxy.connect_to_signal(u"CheckerStarted",
161
self.proxy.connect_to_signal("CheckerStarted",
141
162
self.checker_started,
142
163
client_interface,
143
164
byte_arrays=True)
144
self.proxy.connect_to_signal(u"GotSecret",
165
self.proxy.connect_to_signal("GotSecret",
146
167
client_interface,
147
168
byte_arrays=True)
148
self.proxy.connect_to_signal(u"NeedApproval",
169
self.proxy.connect_to_signal("NeedApproval",
149
170
self.need_approval,
150
171
client_interface,
151
172
byte_arrays=True)
152
self.proxy.connect_to_signal(u"Rejected",
173
self.proxy.connect_to_signal("Rejected",
154
175
client_interface,
155
176
byte_arrays=True)
192
213
self.last_checker_failed = True
193
214
self.using_timer(True)
194
215
if os.WIFEXITED(condition):
195
self.logger(u'Checker for client %s (command "%s")'
196
u' failed with exit code %s'
197
% (self.properties[u"Name"], command,
216
self.logger('Checker for client %s (command "%s")'
217
' failed with exit code %s'
218
% (self.properties["Name"], command,
198
219
os.WEXITSTATUS(condition)))
199
220
elif os.WIFSIGNALED(condition):
200
self.logger(u'Checker for client %s (command "%s")'
201
u' was killed by signal %s'
202
% (self.properties[u"Name"], command,
221
self.logger('Checker for client %s (command "%s")'
222
' was killed by signal %s'
223
% (self.properties["Name"], command,
203
224
os.WTERMSIG(condition)))
204
225
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
226
self.logger('Checker for client %s (command "%s")'
228
% (self.properties["Name"], command))
209
self.logger(u'Checker for client %s completed'
230
self.logger('Checker for client %s completed'
213
234
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
235
#self.logger('Client %s started checker "%s"'
236
# % (self.properties["Name"], unicode(command)))
218
239
def got_secret(self):
219
240
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
241
self.logger('Client %s received its secret'
242
% self.properties["Name"])
223
244
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
246
message = 'Client %s needs approval within %s seconds'
227
message = u'Client %s will get its secret in %s seconds'
248
message = 'Client %s will get its secret in %s seconds'
228
249
self.logger(message
229
% (self.properties[u"Name"], timeout/1000))
250
% (self.properties["Name"], timeout/1000))
230
251
self.using_timer(True)
232
253
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
254
self.logger('Client %s was rejected; reason: %s'
255
% (self.properties["Name"], reason))
236
257
def selectable(self):
237
258
"""Make this a "selectable" widget.
238
259
This overrides the method from urwid.FlowWidget."""
241
def rows(self, (maxcol,), focus=False):
262
def rows(self, maxcolrow, focus=False):
242
263
"""How many rows this widget will occupy might depend on
243
264
whether we have focus or not.
244
265
This overrides the method from urwid.FlowWidget"""
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
266
return self.current_widget(focus).rows(maxcolrow, focus=focus)
247
268
def current_widget(self, focus=False):
248
269
if focus or self.opened:
252
273
def update(self):
253
274
"Called when what is visible on the screen should be updated."
254
275
# How to add standout mode to a style
255
with_standout = { u"normal": u"standout",
256
u"bold": u"bold-standout",
258
u"underline-blink-standout",
259
u"bold-underline-blink":
260
u"bold-underline-blink-standout",
276
with_standout = { "normal": "standout",
277
"bold": "bold-standout",
279
"underline-blink-standout",
280
"bold-underline-blink":
281
"bold-underline-blink-standout",
263
284
# Rebuild focus and non-focus widgets using current properties
265
286
# Base part of a client. Name!
266
base = (u'%(name)s: '
267
% {u"name": self.properties[u"Name"]})
268
if not self.properties[u"Enabled"]:
269
message = u"DISABLED"
270
elif self.properties[u"ApprovalPending"]:
288
% {"name": self.properties["Name"]})
289
if not self.properties["Enabled"]:
291
elif self.properties["ApprovalPending"]:
271
292
timeout = datetime.timedelta(milliseconds
272
293
= self.properties
274
295
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
296
self.properties["LastApprovalRequest"])
276
297
if last_approval_request is not None:
277
298
timer = timeout - (datetime.datetime.utcnow()
278
299
- last_approval_request)
280
301
timer = datetime.timedelta()
281
if self.properties[u"ApprovedByDefault"]:
282
message = u"Approval in %s. (d)eny?"
302
if self.properties["ApprovedByDefault"]:
303
message = "Approval in %s. (d)eny?"
284
message = u"Denial in %s. (a)pprove?"
305
message = "Denial in %s. (a)pprove?"
285
306
message = message % unicode(timer).rsplit(".", 1)[0]
286
307
elif self.last_checker_failed:
287
308
timeout = datetime.timedelta(milliseconds
288
309
= self.properties
290
311
last_ok = isoformat_to_datetime(
291
max((self.properties[u"LastCheckedOK"]
292
or self.properties[u"Created"]),
293
self.properties[u"LastEnabled"]))
312
max((self.properties["LastCheckedOK"]
313
or self.properties["Created"]),
314
self.properties["LastEnabled"]))
294
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
295
message = (u'A checker has failed! Time until client'
296
u' gets disabled: %s'
316
message = ('A checker has failed! Time until client'
297
318
% unicode(timer).rsplit(".", 1)[0])
300
321
self._text = "%s%s" % (base, message)
302
323
if not urwid.supports_unicode():
303
324
self._text = self._text.encode("ascii", "replace")
304
textlist = [(u"normal", self._text)]
325
textlist = [("normal", self._text)]
305
326
self._text_widget.set_text(textlist)
306
327
self._focus_text_widget.set_text([(with_standout[text[0]],
327
348
if self.delete_hook is not None:
328
349
self.delete_hook(self)
330
def render(self, (maxcol,), focus=False):
351
def render(self, maxcolrow, focus=False):
331
352
"""Render differently if we have focus.
332
353
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render((maxcol,),
354
return self.current_widget(focus).render(maxcolrow,
336
def keypress(self, (maxcol,), key):
357
def keypress(self, maxcolrow, key):
338
359
This overrides the method from urwid.FlowWidget"""
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
361
self.proxy.Enable(dbus_interface = client_interface,
364
self.proxy.Disable(dbus_interface = client_interface,
344
367
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
368
dbus_interface = client_interface,
347
371
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
348
dbus_interface = client_interface)
349
elif key == u"r" or key == u"_" or key == u"ctrl k":
372
dbus_interface = client_interface,
374
elif key == "R" or key == "_" or key == "ctrl k":
350
375
self.server_proxy_object.RemoveClient(self.proxy
353
self.proxy.StartChecker(dbus_interface = client_interface)
355
self.proxy.StopChecker(dbus_interface = client_interface)
357
self.proxy.CheckedOK(dbus_interface = client_interface)
379
self.proxy.StartChecker(dbus_interface = client_interface,
382
self.proxy.StopChecker(dbus_interface = client_interface,
385
self.proxy.CheckedOK(dbus_interface = client_interface,
359
# elif key == u"p" or key == "=":
388
# elif key == "p" or key == "=":
360
389
# self.proxy.pause()
361
# elif key == u"u" or key == ":":
390
# elif key == "u" or key == ":":
362
391
# self.proxy.unpause()
363
# elif key == u"RET":
382
411
"down" key presses, thus not allowing any containing widgets to
383
412
use them as an excuse to shift focus away from this widget.
385
def keypress(self, (maxcol, maxrow), key):
386
ret = super(ConstrainedListBox, self).keypress((maxcol,
388
if ret in (u"up", u"down"):
414
def keypress(self, maxcolrow, key):
415
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
416
if ret in ("up", "down"):
400
428
self.screen = urwid.curses_display.Screen()
402
430
self.screen.register_palette((
404
u"default", u"default", None),
406
u"default", u"default", u"bold"),
408
u"default", u"default", u"underline"),
410
u"default", u"default", u"standout"),
411
(u"bold-underline-blink",
412
u"default", u"default", (u"bold", u"underline")),
414
u"default", u"default", (u"bold", u"standout")),
415
(u"underline-blink-standout",
416
u"default", u"default", (u"underline", u"standout")),
417
(u"bold-underline-blink-standout",
418
u"default", u"default", (u"bold", u"underline",
432
"default", "default", None),
434
"default", "default", "bold"),
436
"default", "default", "underline"),
438
"default", "default", "standout"),
439
("bold-underline-blink",
440
"default", "default", ("bold", "underline")),
442
"default", "default", ("bold", "standout")),
443
("underline-blink-standout",
444
"default", "default", ("underline", "standout")),
445
("bold-underline-blink-standout",
446
"default", "default", ("bold", "underline",
422
450
if urwid.supports_unicode():
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
451
self.divider = "─" # \u2500
452
#self.divider = "━" # \u2501
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
454
#self.divider = "-" # \u002d
455
self.divider = "_" # \u005f
429
457
self.screen.start()
444
472
# This keeps track of whether self.uilist currently has
445
473
# self.logbox in it or not
446
474
self.log_visible = True
447
self.log_wrap = u"any"
475
self.log_wrap = "any"
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
478
self.log_message_raw(("bold",
479
"Mandos Monitor version " + version))
480
self.log_message_raw(("bold",
455
483
self.busname = domain + '.Mandos'
456
484
self.main_loop = gobject.MainLoop()
457
485
self.bus = dbus.SystemBus()
458
486
mandos_dbus_objc = self.bus.get_object(
459
self.busname, u"/", follow_name_owner_changes=True)
487
self.busname, "/", follow_name_owner_changes=True)
460
488
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
462
490
= server_interface)
467
495
mandos_clients = dbus.Dictionary()
469
497
(self.mandos_serv
470
.connect_to_signal(u"ClientRemoved",
498
.connect_to_signal("ClientRemoved",
471
499
self.find_and_remove_client,
472
500
dbus_interface=server_interface,
473
501
byte_arrays=True))
474
502
(self.mandos_serv
475
.connect_to_signal(u"ClientAdded",
503
.connect_to_signal("ClientAdded",
476
504
self.add_new_client,
477
505
dbus_interface=server_interface,
478
506
byte_arrays=True))
479
507
(self.mandos_serv
480
.connect_to_signal(u"ClientNotFound",
508
.connect_to_signal("ClientNotFound",
481
509
self.client_not_found,
482
510
dbus_interface=server_interface,
483
511
byte_arrays=True))
530
558
and len(self.log) > self.max_log_length):
531
559
del self.log[0:len(self.log)-self.max_log_length-1]
532
560
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
536
564
def toggle_log_display(self):
537
565
"""Toggle visibility of the log buffer."""
538
566
self.log_visible = not self.log_visible
540
#self.log_message(u"Log visibility changed to: "
568
#self.log_message("Log visibility changed to: "
541
569
# + unicode(self.log_visible))
543
571
def change_log_display(self):
544
572
"""Change type of log display.
545
573
Currently, this toggles wrapping of text lines."""
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
574
if self.log_wrap == "clip":
575
self.log_wrap = "any"
549
self.log_wrap = u"clip"
577
self.log_wrap = "clip"
550
578
for textwidget in self.log:
551
579
textwidget.set_wrap_mode(self.log_wrap)
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
580
#self.log_message("Wrap mode: " + self.log_wrap)
554
582
def find_and_remove_client(self, path, name):
555
583
"""Find an client from its object path and remove it.
620
648
def process_input(self, source, condition):
621
649
keys = self.screen.get_input()
622
translations = { u"ctrl n": u"down", # Emacs
623
u"ctrl p": u"up", # Emacs
624
u"ctrl v": u"page down", # Emacs
625
u"meta v": u"page up", # Emacs
626
u" ": u"page down", # less
627
u"f": u"page down", # less
628
u"b": u"page up", # less
650
translations = { "ctrl n": "down", # Emacs
651
"ctrl p": "up", # Emacs
652
"ctrl v": "page down", # Emacs
653
"meta v": "page up", # Emacs
654
" ": "page down", # less
655
"f": "page down", # less
656
"b": "page up", # less
635
663
except KeyError: # :-)
638
if key == u"q" or key == u"Q":
666
if key == "q" or key == "Q":
641
elif key == u"window resize":
669
elif key == "window resize":
642
670
self.size = self.screen.get_cols_rows()
644
elif key == u"\f": # Ctrl-L
672
elif key == "\f": # Ctrl-L
646
elif key == u"l" or key == u"D":
674
elif key == "l" or key == "D":
647
675
self.toggle_log_display()
649
elif key == u"w" or key == u"i":
677
elif key == "w" or key == "i":
650
678
self.change_log_display()
652
elif key == u"?" or key == u"f1" or key == u"esc":
680
elif key == "?" or key == "f1" or key == "esc":
653
681
if not self.log_visible:
654
682
self.log_visible = True
656
self.log_message_raw((u"bold",
660
u"l: Log window toggle",
661
u"TAB: Switch window",
663
self.log_message_raw((u"bold",
669
u"s: Start new checker",
684
self.log_message_raw(("bold",
688
"l: Log window toggle",
689
"TAB: Switch window",
691
self.log_message_raw(("bold",
697
"s: Start new checker",
676
704
if self.topwidget.get_focus() is self.logbox:
677
705
self.topwidget.set_focus(0)
679
707
self.topwidget.set_focus(self.logbox)
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
709
#elif (key == "end" or key == "meta >" or key == "G"
683
711
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
712
#elif (key == "home" or key == "meta <" or key == "g"
686
714
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
715
#elif key == "ctrl e" or key == "$":
688
716
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
717
#elif key == "ctrl a" or key == "^":
690
718
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
719
#elif key == "ctrl b" or key == "meta (" or key == "h":
692
720
# pass # xxx left
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
721
#elif key == "ctrl f" or key == "meta )" or key == "l":
694
722
# pass # xxx right
696
724
# pass # scroll up log
698
726
# pass # scroll down log
699
727
elif self.topwidget.selectable():
700
728
self.topwidget.keypress(self.size, key)