65
66
"Parse an ISO 8601 date string to a datetime.datetime()"
68
d, t = iso.split(u"T", 1)
69
year, month, day = d.split(u"-", 2)
70
hour, minute, second = t.split(u":", 2)
69
d, t = iso.split("T", 1)
70
year, month, day = d.split("-", 2)
71
hour, minute, second = t.split(":", 2)
71
72
second, fraction = divmod(float(second), 1)
72
73
return datetime.datetime(int(year),
86
87
self.proxy = proxy_object # Mandos Client proxy object
88
89
self.properties = dict()
89
self.proxy.connect_to_signal(u"PropertyChanged",
90
self.property_changed,
90
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
94
96
self.properties.update(
95
97
self.proxy.GetAll(client_interface,
96
98
dbus_interface = dbus.PROPERTIES_IFACE))
98
#XXX This break good super behaviour!
100
#XXX This breaks good super behaviour
99
101
# super(MandosClientPropertyCache, self).__init__(
100
102
# *args, **kwargs)
154
161
if self.need_approval:
155
162
self.using_timer(True)
157
self.proxy.connect_to_signal(u"CheckerCompleted",
158
self.checker_completed,
161
self.proxy.connect_to_signal(u"CheckerStarted",
162
self.checker_started,
165
self.proxy.connect_to_signal(u"GotSecret",
169
self.proxy.connect_to_signal(u"NeedApproval",
173
self.proxy.connect_to_signal(u"Rejected",
164
self.match_objects = (
165
self.proxy.connect_to_signal("CheckerCompleted",
166
self.checker_completed,
169
self.proxy.connect_to_signal("CheckerStarted",
170
self.checker_started,
173
self.proxy.connect_to_signal("GotSecret",
177
self.proxy.connect_to_signal("NeedApproval",
181
self.proxy.connect_to_signal("Rejected",
185
#self.logger('Created client %s' % (self.properties["Name"]))
178
187
def property_changed(self, property=None, value=None):
179
188
super(self, MandosClientWidget).property_changed(property,
181
if property == u"ApprovalPending":
190
if property == "ApprovalPending":
182
191
using_timer(bool(value))
184
193
def using_timer(self, flag):
213
223
self.last_checker_failed = True
214
224
self.using_timer(True)
215
225
if os.WIFEXITED(condition):
216
self.logger(u'Checker for client %s (command "%s")'
217
u' failed with exit code %s'
218
% (self.properties[u"Name"], command,
226
self.logger('Checker for client %s (command "%s")'
227
' failed with exit code %s'
228
% (self.properties["Name"], command,
219
229
os.WEXITSTATUS(condition)))
220
230
elif os.WIFSIGNALED(condition):
221
self.logger(u'Checker for client %s (command "%s")'
222
u' was killed by signal %s'
223
% (self.properties[u"Name"], command,
231
self.logger('Checker for client %s (command "%s")'
232
' was killed by signal %s'
233
% (self.properties["Name"], command,
224
234
os.WTERMSIG(condition)))
225
235
elif os.WCOREDUMP(condition):
226
self.logger(u'Checker for client %s (command "%s")'
228
% (self.properties[u"Name"], command))
236
self.logger('Checker for client %s (command "%s")'
238
% (self.properties["Name"], command))
230
self.logger(u'Checker for client %s completed'
240
self.logger('Checker for client %s completed'
234
244
def checker_started(self, command):
235
#self.logger(u'Client %s started checker "%s"'
236
# % (self.properties[u"Name"], unicode(command)))
245
"""Server signals that a checker started. This could be useful
246
to log in the future. """
247
#self.logger('Client %s started checker "%s"'
248
# % (self.properties["Name"], unicode(command)))
239
251
def got_secret(self):
240
252
self.last_checker_failed = False
241
self.logger(u'Client %s received its secret'
242
% self.properties[u"Name"])
253
self.logger('Client %s received its secret'
254
% self.properties["Name"])
244
256
def need_approval(self, timeout, default):
246
message = u'Client %s needs approval within %s seconds'
258
message = 'Client %s needs approval within %s seconds'
248
message = u'Client %s will get its secret in %s seconds'
260
message = 'Client %s will get its secret in %s seconds'
249
261
self.logger(message
250
% (self.properties[u"Name"], timeout/1000))
262
% (self.properties["Name"], timeout/1000))
251
263
self.using_timer(True)
253
265
def rejected(self, reason):
254
self.logger(u'Client %s was rejected; reason: %s'
255
% (self.properties[u"Name"], reason))
266
self.logger('Client %s was rejected; reason: %s'
267
% (self.properties["Name"], reason))
257
269
def selectable(self):
258
270
"""Make this a "selectable" widget.
259
271
This overrides the method from urwid.FlowWidget."""
262
def rows(self, (maxcol,), focus=False):
274
def rows(self, maxcolrow, focus=False):
263
275
"""How many rows this widget will occupy might depend on
264
276
whether we have focus or not.
265
277
This overrides the method from urwid.FlowWidget"""
266
return self.current_widget(focus).rows((maxcol,), focus=focus)
278
return self.current_widget(focus).rows(maxcolrow, focus=focus)
268
280
def current_widget(self, focus=False):
269
281
if focus or self.opened:
273
285
def update(self):
274
286
"Called when what is visible on the screen should be updated."
275
287
# How to add standout mode to a style
276
with_standout = { u"normal": u"standout",
277
u"bold": u"bold-standout",
279
u"underline-blink-standout",
280
u"bold-underline-blink":
281
u"bold-underline-blink-standout",
288
with_standout = { "normal": "standout",
289
"bold": "bold-standout",
291
"underline-blink-standout",
292
"bold-underline-blink":
293
"bold-underline-blink-standout",
284
296
# Rebuild focus and non-focus widgets using current properties
286
298
# Base part of a client. Name!
287
base = (u'%(name)s: '
288
% {u"name": self.properties[u"Name"]})
289
if not self.properties[u"Enabled"]:
290
message = u"DISABLED"
291
elif self.properties[u"ApprovalPending"]:
300
% {"name": self.properties["Name"]})
301
if not self.properties["Enabled"]:
303
elif self.properties["ApprovalPending"]:
292
304
timeout = datetime.timedelta(milliseconds
293
305
= self.properties
295
307
last_approval_request = isoformat_to_datetime(
296
self.properties[u"LastApprovalRequest"])
308
self.properties["LastApprovalRequest"])
297
309
if last_approval_request is not None:
298
310
timer = timeout - (datetime.datetime.utcnow()
299
311
- last_approval_request)
301
313
timer = datetime.timedelta()
302
if self.properties[u"ApprovedByDefault"]:
303
message = u"Approval in %s. (d)eny?"
314
if self.properties["ApprovedByDefault"]:
315
message = "Approval in %s. (d)eny?"
305
message = u"Denial in %s. (a)pprove?"
317
message = "Denial in %s. (a)pprove?"
306
318
message = message % unicode(timer).rsplit(".", 1)[0]
307
319
elif self.last_checker_failed:
308
timeout = datetime.timedelta(milliseconds
311
last_ok = isoformat_to_datetime(
312
max((self.properties[u"LastCheckedOK"]
313
or self.properties[u"Created"]),
314
self.properties[u"LastEnabled"]))
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
316
message = (u'A checker has failed! Time until client'
317
u' gets disabled: %s'
320
# When checker has failed, print a timer until client expires
321
expires = self.properties["Expires"]
323
timer = datetime.timedelta(0)
325
expires = datetime.datetime.strptime(expires,
326
'%Y-%m-%dT%H:%M:%S.%f')
327
timer = expires - datetime.datetime.utcnow()
328
message = ('A checker has failed! Time until client'
318
330
% unicode(timer).rsplit(".", 1)[0])
321
333
self._text = "%s%s" % (base, message)
323
335
if not urwid.supports_unicode():
324
336
self._text = self._text.encode("ascii", "replace")
325
textlist = [(u"normal", self._text)]
337
textlist = [("normal", self._text)]
326
338
self._text_widget.set_text(textlist)
327
339
self._focus_text_widget.set_text([(with_standout[text[0]],
337
349
self.update_hook()
339
351
def update_timer(self):
352
"""called by gobject. Will indefinitely loop until
353
gobject.source_remove() on tag is called"""
342
355
return True # Keep calling this
357
def delete(self, *args, **kwargs):
345
358
if self._update_timer_callback_tag is not None:
346
359
gobject.source_remove(self._update_timer_callback_tag)
347
360
self._update_timer_callback_tag = None
361
for match in self.match_objects:
363
self.match_objects = ()
348
364
if self.delete_hook is not None:
349
365
self.delete_hook(self)
366
return super(MandosClientWidget, self).delete(*args, **kwargs)
351
def render(self, (maxcol,), focus=False):
368
def render(self, maxcolrow, focus=False):
352
369
"""Render differently if we have focus.
353
370
This overrides the method from urwid.FlowWidget"""
354
return self.current_widget(focus).render((maxcol,),
371
return self.current_widget(focus).render(maxcolrow,
357
def keypress(self, (maxcol,), key):
374
def keypress(self, maxcolrow, key):
359
376
This overrides the method from urwid.FlowWidget"""
361
self.proxy.Enable(dbus_interface = client_interface)
363
self.proxy.Disable(dbus_interface = client_interface)
378
self.proxy.Enable(dbus_interface = client_interface,
381
self.proxy.Disable(dbus_interface = client_interface,
365
384
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
366
dbus_interface = client_interface)
385
dbus_interface = client_interface,
368
388
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
369
dbus_interface = client_interface)
370
elif key == u"R" or key == u"_" or key == u"ctrl k":
389
dbus_interface = client_interface,
391
elif key == "R" or key == "_" or key == "ctrl k":
371
392
self.server_proxy_object.RemoveClient(self.proxy
374
self.proxy.StartChecker(dbus_interface = client_interface)
376
self.proxy.StopChecker(dbus_interface = client_interface)
378
self.proxy.CheckedOK(dbus_interface = client_interface)
396
self.proxy.StartChecker(dbus_interface = client_interface,
399
self.proxy.StopChecker(dbus_interface = client_interface,
402
self.proxy.CheckedOK(dbus_interface = client_interface,
380
# elif key == u"p" or key == "=":
405
# elif key == "p" or key == "=":
381
406
# self.proxy.pause()
382
# elif key == u"u" or key == ":":
407
# elif key == "u" or key == ":":
383
408
# self.proxy.unpause()
384
# elif key == u"RET":
403
428
"down" key presses, thus not allowing any containing widgets to
404
429
use them as an excuse to shift focus away from this widget.
406
def keypress(self, (maxcol, maxrow), key):
407
ret = super(ConstrainedListBox, self).keypress((maxcol,
409
if ret in (u"up", u"down"):
431
def keypress(self, maxcolrow, key):
432
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
433
if ret in ("up", "down"):
421
445
self.screen = urwid.curses_display.Screen()
423
447
self.screen.register_palette((
425
u"default", u"default", None),
427
u"default", u"default", u"bold"),
429
u"default", u"default", u"underline"),
431
u"default", u"default", u"standout"),
432
(u"bold-underline-blink",
433
u"default", u"default", (u"bold", u"underline")),
435
u"default", u"default", (u"bold", u"standout")),
436
(u"underline-blink-standout",
437
u"default", u"default", (u"underline", u"standout")),
438
(u"bold-underline-blink-standout",
439
u"default", u"default", (u"bold", u"underline",
449
"default", "default", None),
451
"default", "default", "bold"),
453
"default", "default", "underline"),
455
"default", "default", "standout"),
456
("bold-underline-blink",
457
"default", "default", ("bold", "underline")),
459
"default", "default", ("bold", "standout")),
460
("underline-blink-standout",
461
"default", "default", ("underline", "standout")),
462
("bold-underline-blink-standout",
463
"default", "default", ("bold", "underline",
443
467
if urwid.supports_unicode():
444
self.divider = u"─" # \u2500
445
#self.divider = u"━" # \u2501
468
self.divider = "─" # \u2500
469
#self.divider = "━" # \u2501
447
#self.divider = u"-" # \u002d
448
self.divider = u"_" # \u005f
471
#self.divider = "-" # \u002d
472
self.divider = "_" # \u005f
450
474
self.screen.start()
465
489
# This keeps track of whether self.uilist currently has
466
490
# self.logbox in it or not
467
491
self.log_visible = True
468
self.log_wrap = u"any"
492
self.log_wrap = "any"
471
self.log_message_raw((u"bold",
472
u"Mandos Monitor version " + version))
473
self.log_message_raw((u"bold",
495
self.log_message_raw(("bold",
496
"Mandos Monitor version " + version))
497
self.log_message_raw(("bold",
476
500
self.busname = domain + '.Mandos'
477
501
self.main_loop = gobject.MainLoop()
478
self.bus = dbus.SystemBus()
479
mandos_dbus_objc = self.bus.get_object(
480
self.busname, u"/", follow_name_owner_changes=True)
481
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
485
mandos_clients = (self.mandos_serv
486
.GetAllClientsWithProperties())
487
except dbus.exceptions.DBusException:
488
mandos_clients = dbus.Dictionary()
491
.connect_to_signal(u"ClientRemoved",
492
self.find_and_remove_client,
493
dbus_interface=server_interface,
496
.connect_to_signal(u"ClientAdded",
498
dbus_interface=server_interface,
501
.connect_to_signal(u"ClientNotFound",
502
self.client_not_found,
503
dbus_interface=server_interface,
505
for path, client in mandos_clients.iteritems():
506
client_proxy_object = self.bus.get_object(self.busname,
508
self.add_client(MandosClientWidget(server_proxy_object
511
=client_proxy_object,
521
503
def client_not_found(self, fingerprint, address):
522
self.log_message((u"Client with address %s and fingerprint %s"
523
u" could not be found" % (address,
504
self.log_message(("Client with address %s and fingerprint %s"
505
" could not be found" % (address,
526
508
def rebuild(self):
551
532
and len(self.log) > self.max_log_length):
552
533
del self.log[0:len(self.log)-self.max_log_length-1]
553
534
self.logbox.set_focus(len(self.logbox.body.contents),
554
coming_from=u"above")
557
538
def toggle_log_display(self):
558
539
"""Toggle visibility of the log buffer."""
559
540
self.log_visible = not self.log_visible
561
#self.log_message(u"Log visibility changed to: "
542
#self.log_message("Log visibility changed to: "
562
543
# + unicode(self.log_visible))
564
545
def change_log_display(self):
565
546
"""Change type of log display.
566
547
Currently, this toggles wrapping of text lines."""
567
if self.log_wrap == u"clip":
568
self.log_wrap = u"any"
548
if self.log_wrap == "clip":
549
self.log_wrap = "any"
570
self.log_wrap = u"clip"
551
self.log_wrap = "clip"
571
552
for textwidget in self.log:
572
553
textwidget.set_wrap_mode(self.log_wrap)
573
#self.log_message(u"Wrap mode: " + self.log_wrap)
554
#self.log_message("Wrap mode: " + self.log_wrap)
575
556
def find_and_remove_client(self, path, name):
576
"""Find an client from its object path and remove it.
557
"""Find a client by its object path and remove it.
578
559
This is connected to the ClientRemoved signal from the
579
560
Mandos server object."""
627
610
"""Start the main loop and exit when it's done."""
611
self.bus = dbus.SystemBus()
612
mandos_dbus_objc = self.bus.get_object(
613
self.busname, "/", follow_name_owner_changes=True)
614
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
618
mandos_clients = (self.mandos_serv
619
.GetAllClientsWithProperties())
620
except dbus.exceptions.DBusException:
621
mandos_clients = dbus.Dictionary()
624
.connect_to_signal("ClientRemoved",
625
self.find_and_remove_client,
626
dbus_interface=server_interface,
629
.connect_to_signal("ClientAdded",
631
dbus_interface=server_interface,
634
.connect_to_signal("ClientNotFound",
635
self.client_not_found,
636
dbus_interface=server_interface,
638
for path, client in mandos_clients.iteritems():
639
client_proxy_object = self.bus.get_object(self.busname,
641
self.add_client(MandosClientWidget(server_proxy_object
644
=client_proxy_object,
629
655
self._input_callback_tag = (gobject.io_add_watch
630
656
(sys.stdin.fileno(),
641
667
def process_input(self, source, condition):
642
668
keys = self.screen.get_input()
643
translations = { u"ctrl n": u"down", # Emacs
644
u"ctrl p": u"up", # Emacs
645
u"ctrl v": u"page down", # Emacs
646
u"meta v": u"page up", # Emacs
647
u" ": u"page down", # less
648
u"f": u"page down", # less
649
u"b": u"page up", # less
669
translations = { "ctrl n": "down", # Emacs
670
"ctrl p": "up", # Emacs
671
"ctrl v": "page down", # Emacs
672
"meta v": "page up", # Emacs
673
" ": "page down", # less
674
"f": "page down", # less
675
"b": "page up", # less
656
682
except KeyError: # :-)
659
if key == u"q" or key == u"Q":
685
if key == "q" or key == "Q":
662
elif key == u"window resize":
688
elif key == "window resize":
663
689
self.size = self.screen.get_cols_rows()
665
elif key == u"\f": # Ctrl-L
691
elif key == "\f": # Ctrl-L
667
elif key == u"l" or key == u"D":
693
elif key == "l" or key == "D":
668
694
self.toggle_log_display()
670
elif key == u"w" or key == u"i":
696
elif key == "w" or key == "i":
671
697
self.change_log_display()
673
elif key == u"?" or key == u"f1" or key == u"esc":
699
elif key == "?" or key == "f1" or key == "esc":
674
700
if not self.log_visible:
675
701
self.log_visible = True
677
self.log_message_raw((u"bold",
681
u"l: Log window toggle",
682
u"TAB: Switch window",
684
self.log_message_raw((u"bold",
690
u"s: Start new checker",
703
self.log_message_raw(("bold",
707
"l: Log window toggle",
708
"TAB: Switch window",
710
self.log_message_raw(("bold",
716
"s: Start new checker",
697
723
if self.topwidget.get_focus() is self.logbox:
698
724
self.topwidget.set_focus(0)
700
726
self.topwidget.set_focus(self.logbox)
702
#elif (key == u"end" or key == u"meta >" or key == u"G"
728
#elif (key == "end" or key == "meta >" or key == "G"
704
730
# pass # xxx end-of-buffer
705
#elif (key == u"home" or key == u"meta <" or key == u"g"
731
#elif (key == "home" or key == "meta <" or key == "g"
707
733
# pass # xxx beginning-of-buffer
708
#elif key == u"ctrl e" or key == u"$":
734
#elif key == "ctrl e" or key == "$":
709
735
# pass # xxx move-end-of-line
710
#elif key == u"ctrl a" or key == u"^":
736
#elif key == "ctrl a" or key == "^":
711
737
# pass # xxx move-beginning-of-line
712
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
738
#elif key == "ctrl b" or key == "meta (" or key == "h":
713
739
# pass # xxx left
714
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
740
#elif key == "ctrl f" or key == "meta )" or key == "l":
715
741
# pass # xxx right
717
743
# pass # scroll up log
719
745
# pass # scroll down log
720
746
elif self.topwidget.selectable():
721
747
self.topwidget.keypress(self.size, key)