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@recompile.se>.
4
from __future__ import division, absolute_import, with_statement
25
from __future__ import (division, absolute_import, print_function,
44
66
"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)
69
d, t = iso.split("T", 1)
70
year, month, day = d.split("-", 2)
71
hour, minute, second = t.split(":", 2)
50
72
second, fraction = divmod(float(second), 1)
51
73
return datetime.datetime(int(year),
65
87
self.proxy = proxy_object # Mandos Client proxy object
67
89
self.properties = dict()
68
self.proxy.connect_to_signal(u"PropertyChanged",
69
self.property_changed,
90
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
73
96
self.properties.update(
74
97
self.proxy.GetAll(client_interface,
75
98
dbus_interface = dbus.PROPERTIES_IFACE))
77
#XXX This break good super behaviour!
100
#XXX This breaks good super behaviour
78
101
# super(MandosClientPropertyCache, self).__init__(
79
102
# *args, **kwargs)
133
161
if self.need_approval:
134
162
self.using_timer(True)
136
self.proxy.connect_to_signal(u"CheckerCompleted",
137
self.checker_completed,
140
self.proxy.connect_to_signal(u"CheckerStarted",
141
self.checker_started,
144
self.proxy.connect_to_signal(u"GotSecret",
148
self.proxy.connect_to_signal(u"NeedApproval",
152
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"]))
157
187
def property_changed(self, property=None, value=None):
158
188
super(self, MandosClientWidget).property_changed(property,
160
if property == u"ApprovalPending":
190
if property == "ApprovalPending":
161
191
using_timer(bool(value))
163
193
def using_timer(self, flag):
192
223
self.last_checker_failed = True
193
224
self.using_timer(True)
194
225
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,
226
self.logger('Checker for client %s (command "%s")'
227
' failed with exit code %s'
228
% (self.properties["Name"], command,
198
229
os.WEXITSTATUS(condition)))
199
230
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,
231
self.logger('Checker for client %s (command "%s")'
232
' was killed by signal %s'
233
% (self.properties["Name"], command,
203
234
os.WTERMSIG(condition)))
204
235
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
236
self.logger('Checker for client %s (command "%s")'
238
% (self.properties["Name"], command))
209
self.logger(u'Checker for client %s completed'
240
self.logger('Checker for client %s completed'
213
244
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (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)))
218
251
def got_secret(self):
219
252
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
253
self.logger('Client %s received its secret'
254
% self.properties["Name"])
223
256
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
258
message = 'Client %s needs approval within %s seconds'
227
message = u'Client %s will get its secret in %s seconds'
260
message = 'Client %s will get its secret in %s seconds'
228
261
self.logger(message
229
% (self.properties[u"Name"], timeout/1000))
262
% (self.properties["Name"], timeout/1000))
230
263
self.using_timer(True)
232
265
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
266
self.logger('Client %s was rejected; reason: %s'
267
% (self.properties["Name"], reason))
236
269
def selectable(self):
237
270
"""Make this a "selectable" widget.
238
271
This overrides the method from urwid.FlowWidget."""
241
def rows(self, (maxcol,), focus=False):
274
def rows(self, maxcolrow, focus=False):
242
275
"""How many rows this widget will occupy might depend on
243
276
whether we have focus or not.
244
277
This overrides the method from urwid.FlowWidget"""
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
278
return self.current_widget(focus).rows(maxcolrow, focus=focus)
247
280
def current_widget(self, focus=False):
248
281
if focus or self.opened:
252
285
def update(self):
253
286
"Called when what is visible on the screen should be updated."
254
287
# 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",
288
with_standout = { "normal": "standout",
289
"bold": "bold-standout",
291
"underline-blink-standout",
292
"bold-underline-blink":
293
"bold-underline-blink-standout",
263
296
# Rebuild focus and non-focus widgets using current properties
265
298
# 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"]:
300
% {"name": self.properties["Name"]})
301
if not self.properties["Enabled"]:
303
elif self.properties["ApprovalPending"]:
271
304
timeout = datetime.timedelta(milliseconds
272
305
= self.properties
274
307
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
308
self.properties["LastApprovalRequest"])
276
309
if last_approval_request is not None:
277
310
timer = timeout - (datetime.datetime.utcnow()
278
311
- last_approval_request)
280
313
timer = datetime.timedelta()
281
if self.properties[u"ApprovedByDefault"]:
282
message = u"Approval in %s. (d)eny?"
314
if self.properties["ApprovedByDefault"]:
315
message = "Approval in %s. (d)eny?"
284
message = u"Denial in %s. (a)pprove?"
317
message = "Denial in %s. (a)pprove?"
285
318
message = message % unicode(timer).rsplit(".", 1)[0]
286
319
elif self.last_checker_failed:
287
timeout = datetime.timedelta(milliseconds
290
last_ok = isoformat_to_datetime(
291
max((self.properties[u"LastCheckedOK"]
292
or self.properties[u"Created"]),
293
self.properties[u"LastEnabled"]))
294
timer = timeout - (datetime.datetime.utcnow() - last_ok)
295
message = (u'A checker has failed! Time until client'
296
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'
297
330
% unicode(timer).rsplit(".", 1)[0])
300
333
self._text = "%s%s" % (base, message)
302
335
if not urwid.supports_unicode():
303
336
self._text = self._text.encode("ascii", "replace")
304
textlist = [(u"normal", self._text)]
337
textlist = [("normal", self._text)]
305
338
self._text_widget.set_text(textlist)
306
339
self._focus_text_widget.set_text([(with_standout[text[0]],
316
349
self.update_hook()
318
351
def update_timer(self):
352
"""called by gobject. Will indefinitely loop until
353
gobject.source_remove() on tag is called"""
321
355
return True # Keep calling this
357
def delete(self, *args, **kwargs):
324
358
if self._update_timer_callback_tag is not None:
325
359
gobject.source_remove(self._update_timer_callback_tag)
326
360
self._update_timer_callback_tag = None
361
for match in self.match_objects:
363
self.match_objects = ()
327
364
if self.delete_hook is not None:
328
365
self.delete_hook(self)
366
return super(MandosClientWidget, self).delete(*args, **kwargs)
330
def render(self, (maxcol,), focus=False):
368
def render(self, maxcolrow, focus=False):
331
369
"""Render differently if we have focus.
332
370
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render((maxcol,),
371
return self.current_widget(focus).render(maxcolrow,
336
def keypress(self, (maxcol,), key):
374
def keypress(self, maxcolrow, key):
338
376
This overrides the method from urwid.FlowWidget"""
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
378
self.proxy.Enable(dbus_interface = client_interface,
381
self.proxy.Disable(dbus_interface = client_interface,
344
384
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
385
dbus_interface = client_interface,
347
388
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":
389
dbus_interface = client_interface,
391
elif key == "R" or key == "_" or key == "ctrl k":
350
392
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)
396
self.proxy.StartChecker(dbus_interface = client_interface,
399
self.proxy.StopChecker(dbus_interface = client_interface,
402
self.proxy.CheckedOK(dbus_interface = client_interface,
359
# elif key == u"p" or key == "=":
405
# elif key == "p" or key == "=":
360
406
# self.proxy.pause()
361
# elif key == u"u" or key == ":":
407
# elif key == "u" or key == ":":
362
408
# self.proxy.unpause()
363
# elif key == u"RET":
382
428
"down" key presses, thus not allowing any containing widgets to
383
429
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"):
431
def keypress(self, maxcolrow, key):
432
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
433
if ret in ("up", "down"):
400
445
self.screen = urwid.curses_display.Screen()
402
447
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",
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",
422
467
if urwid.supports_unicode():
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
468
self.divider = "─" # \u2500
469
#self.divider = "━" # \u2501
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
471
#self.divider = "-" # \u002d
472
self.divider = "_" # \u005f
429
474
self.screen.start()
444
489
# This keeps track of whether self.uilist currently has
445
490
# self.logbox in it or not
446
491
self.log_visible = True
447
self.log_wrap = u"any"
492
self.log_wrap = "any"
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
495
self.log_message_raw(("bold",
496
"Mandos Monitor version " + version))
497
self.log_message_raw(("bold",
455
500
self.busname = domain + '.Mandos'
456
501
self.main_loop = gobject.MainLoop()
457
self.bus = dbus.SystemBus()
458
mandos_dbus_objc = self.bus.get_object(
459
self.busname, u"/", follow_name_owner_changes=True)
460
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
464
mandos_clients = (self.mandos_serv
465
.GetAllClientsWithProperties())
466
except dbus.exceptions.DBusException:
467
mandos_clients = dbus.Dictionary()
470
.connect_to_signal(u"ClientRemoved",
471
self.find_and_remove_client,
472
dbus_interface=server_interface,
475
.connect_to_signal(u"ClientAdded",
477
dbus_interface=server_interface,
480
.connect_to_signal(u"ClientNotFound",
481
self.client_not_found,
482
dbus_interface=server_interface,
484
for path, client in mandos_clients.iteritems():
485
client_proxy_object = self.bus.get_object(self.busname,
487
self.add_client(MandosClientWidget(server_proxy_object
490
=client_proxy_object,
500
503
def client_not_found(self, fingerprint, address):
501
self.log_message((u"Client with address %s and fingerprint %s"
502
u" could not be found" % (address,
504
self.log_message(("Client with address %s and fingerprint %s"
505
" could not be found" % (address,
505
508
def rebuild(self):
530
532
and len(self.log) > self.max_log_length):
531
533
del self.log[0:len(self.log)-self.max_log_length-1]
532
534
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
536
538
def toggle_log_display(self):
537
539
"""Toggle visibility of the log buffer."""
538
540
self.log_visible = not self.log_visible
540
#self.log_message(u"Log visibility changed to: "
542
#self.log_message("Log visibility changed to: "
541
543
# + unicode(self.log_visible))
543
545
def change_log_display(self):
544
546
"""Change type of log display.
545
547
Currently, this toggles wrapping of text lines."""
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
548
if self.log_wrap == "clip":
549
self.log_wrap = "any"
549
self.log_wrap = u"clip"
551
self.log_wrap = "clip"
550
552
for textwidget in self.log:
551
553
textwidget.set_wrap_mode(self.log_wrap)
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
554
#self.log_message("Wrap mode: " + self.log_wrap)
554
556
def find_and_remove_client(self, path, name):
555
"""Find an client from its object path and remove it.
557
"""Find a client by its object path and remove it.
557
559
This is connected to the ClientRemoved signal from the
558
560
Mandos server object."""
606
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,
608
655
self._input_callback_tag = (gobject.io_add_watch
609
656
(sys.stdin.fileno(),
620
667
def process_input(self, source, condition):
621
668
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
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
635
682
except KeyError: # :-)
638
if key == u"q" or key == u"Q":
685
if key == "q" or key == "Q":
641
elif key == u"window resize":
688
elif key == "window resize":
642
689
self.size = self.screen.get_cols_rows()
644
elif key == u"\f": # Ctrl-L
691
elif key == "\f": # Ctrl-L
646
elif key == u"l" or key == u"D":
693
elif key == "l" or key == "D":
647
694
self.toggle_log_display()
649
elif key == u"w" or key == u"i":
696
elif key == "w" or key == "i":
650
697
self.change_log_display()
652
elif key == u"?" or key == u"f1" or key == u"esc":
699
elif key == "?" or key == "f1" or key == "esc":
653
700
if not self.log_visible:
654
701
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",
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",
676
723
if self.topwidget.get_focus() is self.logbox:
677
724
self.topwidget.set_focus(0)
679
726
self.topwidget.set_focus(self.logbox)
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
728
#elif (key == "end" or key == "meta >" or key == "G"
683
730
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
731
#elif (key == "home" or key == "meta <" or key == "g"
686
733
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
734
#elif key == "ctrl e" or key == "$":
688
735
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
736
#elif key == "ctrl a" or key == "^":
690
737
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
738
#elif key == "ctrl b" or key == "meta (" or key == "h":
692
739
# pass # xxx left
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
740
#elif key == "ctrl f" or key == "meta )" or key == "l":
694
741
# pass # xxx right
696
743
# pass # scroll up log
698
745
# pass # scroll down log
699
746
elif self.topwidget.selectable():
700
747
self.topwidget.keypress(self.size, key)