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