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>.
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),
104
132
self._update_timer_callback_tag = None
105
133
self._update_timer_callback_lock = 0
106
self.last_checker_failed = False
108
135
# The widget shown normally
109
self._text_widget = urwid.Text(u"")
136
self._text_widget = urwid.Text("")
110
137
# The widget shown when we have focus
111
self._focus_text_widget = urwid.Text(u"")
138
self._focus_text_widget = urwid.Text("")
112
139
super(MandosClientWidget, self).__init__(
113
140
update_hook=update_hook, delete_hook=delete_hook,
116
143
self.opened = False
118
145
last_checked_ok = isoformat_to_datetime(self.properties
120
if last_checked_ok is None:
121
self.last_checker_failed = True
123
self.last_checker_failed = ((datetime.datetime.utcnow()
130
if self.last_checker_failed:
148
if self.properties ["LastCheckerStatus"] != 0:
131
149
self.using_timer(True)
133
151
if self.need_approval:
134
152
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",
154
self.match_objects = (
155
self.proxy.connect_to_signal("CheckerCompleted",
156
self.checker_completed,
159
self.proxy.connect_to_signal("CheckerStarted",
160
self.checker_started,
163
self.proxy.connect_to_signal("GotSecret",
167
self.proxy.connect_to_signal("NeedApproval",
171
self.proxy.connect_to_signal("Rejected",
175
#self.logger('Created client %s' % (self.properties["Name"]))
157
177
def property_changed(self, property=None, value=None):
158
178
super(self, MandosClientWidget).property_changed(property,
160
if property == u"ApprovalPending":
180
if property == "ApprovalPending":
161
181
using_timer(bool(value))
182
if property == "LastCheckerStatus":
183
using_timer(value != 0)
184
#self.logger('Checker for client %s (command "%s")'
186
# % (self.properties["Name"], command))
163
188
def using_timer(self, flag):
164
189
"""Call this method with True or False when timer should be
165
190
activated or deactivated.
180
206
def checker_completed(self, exitstatus, condition, command):
181
207
if exitstatus == 0:
182
if self.last_checker_failed:
183
self.last_checker_failed = False
184
self.using_timer(False)
185
#self.logger(u'Checker for client %s (command "%s")'
187
# % (self.properties[u"Name"], command))
191
if not self.last_checker_failed:
192
self.last_checker_failed = True
193
self.using_timer(True)
194
211
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,
212
self.logger('Checker for client %s (command "%s")'
213
' failed with exit code %s'
214
% (self.properties["Name"], command,
198
215
os.WEXITSTATUS(condition)))
199
216
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,
217
self.logger('Checker for client %s (command "%s")'
218
' was killed by signal %s'
219
% (self.properties["Name"], command,
203
220
os.WTERMSIG(condition)))
204
221
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
222
self.logger('Checker for client %s (command "%s")'
224
% (self.properties["Name"], command))
209
self.logger(u'Checker for client %s completed'
226
self.logger('Checker for client %s completed'
213
230
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
231
"""Server signals that a checker started. This could be useful
232
to log in the future. """
233
#self.logger('Client %s started checker "%s"'
234
# % (self.properties["Name"], unicode(command)))
218
237
def got_secret(self):
219
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
238
self.logger('Client %s received its secret'
239
% self.properties["Name"])
223
241
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
243
message = 'Client %s needs approval within %s seconds'
227
message = u'Client %s will get its secret in %s seconds'
245
message = 'Client %s will get its secret in %s seconds'
228
246
self.logger(message
229
% (self.properties[u"Name"], timeout/1000))
247
% (self.properties["Name"], timeout/1000))
230
248
self.using_timer(True)
232
250
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
251
self.logger('Client %s was rejected; reason: %s'
252
% (self.properties["Name"], reason))
236
254
def selectable(self):
237
255
"""Make this a "selectable" widget.
238
256
This overrides the method from urwid.FlowWidget."""
241
def rows(self, (maxcol,), focus=False):
259
def rows(self, maxcolrow, focus=False):
242
260
"""How many rows this widget will occupy might depend on
243
261
whether we have focus or not.
244
262
This overrides the method from urwid.FlowWidget"""
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
263
return self.current_widget(focus).rows(maxcolrow, focus=focus)
247
265
def current_widget(self, focus=False):
248
266
if focus or self.opened:
252
270
def update(self):
253
271
"Called when what is visible on the screen should be updated."
254
272
# 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",
273
with_standout = { "normal": "standout",
274
"bold": "bold-standout",
276
"underline-blink-standout",
277
"bold-underline-blink":
278
"bold-underline-blink-standout",
263
281
# Rebuild focus and non-focus widgets using current properties
265
283
# 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"]:
285
% {"name": self.properties["Name"]})
286
if not self.properties["Enabled"]:
288
elif self.properties["ApprovalPending"]:
271
289
timeout = datetime.timedelta(milliseconds
272
290
= self.properties
274
292
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
293
self.properties["LastApprovalRequest"])
276
294
if last_approval_request is not None:
277
295
timer = timeout - (datetime.datetime.utcnow()
278
296
- last_approval_request)
280
298
timer = datetime.timedelta()
281
if self.properties[u"ApprovedByDefault"]:
282
message = u"Approval in %s. (d)eny?"
299
if self.properties["ApprovedByDefault"]:
300
message = "Approval in %s. (d)eny?"
284
message = u"Denial in %s. (a)pprove?"
302
message = "Denial in %s. (a)pprove?"
285
303
message = message % unicode(timer).rsplit(".", 1)[0]
286
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'
304
elif self.properties["LastCheckerStatus"] != 0:
305
# When checker has failed, print a timer until client expires
306
expires = self.properties["Expires"]
308
timer = datetime.timedelta(0)
310
expires = datetime.datetime.strptime(expires,
311
'%Y-%m-%dT%H:%M:%S.%f')
312
timer = expires - datetime.datetime.utcnow()
313
message = ('A checker has failed! Time until client'
297
315
% unicode(timer).rsplit(".", 1)[0])
300
318
self._text = "%s%s" % (base, message)
302
320
if not urwid.supports_unicode():
303
321
self._text = self._text.encode("ascii", "replace")
304
textlist = [(u"normal", self._text)]
322
textlist = [("normal", self._text)]
305
323
self._text_widget.set_text(textlist)
306
324
self._focus_text_widget.set_text([(with_standout[text[0]],
316
334
self.update_hook()
318
336
def update_timer(self):
337
"""called by gobject. Will indefinitely loop until
338
gobject.source_remove() on tag is called"""
321
340
return True # Keep calling this
342
def delete(self, *args, **kwargs):
324
343
if self._update_timer_callback_tag is not None:
325
344
gobject.source_remove(self._update_timer_callback_tag)
326
345
self._update_timer_callback_tag = None
346
for match in self.match_objects:
348
self.match_objects = ()
327
349
if self.delete_hook is not None:
328
350
self.delete_hook(self)
351
return super(MandosClientWidget, self).delete(*args, **kwargs)
330
def render(self, (maxcol,), focus=False):
353
def render(self, maxcolrow, focus=False):
331
354
"""Render differently if we have focus.
332
355
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render((maxcol,),
356
return self.current_widget(focus).render(maxcolrow,
336
def keypress(self, (maxcol,), key):
359
def keypress(self, maxcolrow, key):
338
361
This overrides the method from urwid.FlowWidget"""
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
363
self.proxy.Enable(dbus_interface = client_interface,
366
self.proxy.Disable(dbus_interface = client_interface,
344
369
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
370
dbus_interface = client_interface,
347
373
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":
374
dbus_interface = client_interface,
376
elif key == "R" or key == "_" or key == "ctrl k":
350
377
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)
381
self.proxy.StartChecker(dbus_interface = client_interface,
384
self.proxy.StopChecker(dbus_interface = client_interface,
387
self.proxy.CheckedOK(dbus_interface = client_interface,
359
# elif key == u"p" or key == "=":
390
# elif key == "p" or key == "=":
360
391
# self.proxy.pause()
361
# elif key == u"u" or key == ":":
392
# elif key == "u" or key == ":":
362
393
# self.proxy.unpause()
363
# elif key == u"RET":
382
413
"down" key presses, thus not allowing any containing widgets to
383
414
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"):
416
def keypress(self, maxcolrow, key):
417
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
418
if ret in ("up", "down"):
400
430
self.screen = urwid.curses_display.Screen()
402
432
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",
434
"default", "default", None),
436
"default", "default", "bold"),
438
"default", "default", "underline"),
440
"default", "default", "standout"),
441
("bold-underline-blink",
442
"default", "default", ("bold", "underline")),
444
"default", "default", ("bold", "standout")),
445
("underline-blink-standout",
446
"default", "default", ("underline", "standout")),
447
("bold-underline-blink-standout",
448
"default", "default", ("bold", "underline",
422
452
if urwid.supports_unicode():
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
453
self.divider = "─" # \u2500
454
#self.divider = "━" # \u2501
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
456
#self.divider = "-" # \u002d
457
self.divider = "_" # \u005f
429
459
self.screen.start()
444
474
# This keeps track of whether self.uilist currently has
445
475
# self.logbox in it or not
446
476
self.log_visible = True
447
self.log_wrap = u"any"
477
self.log_wrap = "any"
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
480
self.log_message_raw(("bold",
481
"Mandos Monitor version " + version))
482
self.log_message_raw(("bold",
455
485
self.busname = domain + '.Mandos'
456
486
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
488
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,
489
self.log_message(("Client with address %s and fingerprint %s"
490
" could not be found" % (address,
505
493
def rebuild(self):
530
517
and len(self.log) > self.max_log_length):
531
518
del self.log[0:len(self.log)-self.max_log_length-1]
532
519
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
536
523
def toggle_log_display(self):
537
524
"""Toggle visibility of the log buffer."""
538
525
self.log_visible = not self.log_visible
540
#self.log_message(u"Log visibility changed to: "
527
#self.log_message("Log visibility changed to: "
541
528
# + unicode(self.log_visible))
543
530
def change_log_display(self):
544
531
"""Change type of log display.
545
532
Currently, this toggles wrapping of text lines."""
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
533
if self.log_wrap == "clip":
534
self.log_wrap = "any"
549
self.log_wrap = u"clip"
536
self.log_wrap = "clip"
550
537
for textwidget in self.log:
551
538
textwidget.set_wrap_mode(self.log_wrap)
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
539
#self.log_message("Wrap mode: " + self.log_wrap)
554
541
def find_and_remove_client(self, path, name):
555
"""Find an client from its object path and remove it.
542
"""Find a client by its object path and remove it.
557
544
This is connected to the ClientRemoved signal from the
558
545
Mandos server object."""
606
595
"""Start the main loop and exit when it's done."""
596
self.bus = dbus.SystemBus()
597
mandos_dbus_objc = self.bus.get_object(
598
self.busname, "/", follow_name_owner_changes=True)
599
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
603
mandos_clients = (self.mandos_serv
604
.GetAllClientsWithProperties())
605
except dbus.exceptions.DBusException:
606
mandos_clients = dbus.Dictionary()
609
.connect_to_signal("ClientRemoved",
610
self.find_and_remove_client,
611
dbus_interface=server_interface,
614
.connect_to_signal("ClientAdded",
616
dbus_interface=server_interface,
619
.connect_to_signal("ClientNotFound",
620
self.client_not_found,
621
dbus_interface=server_interface,
623
for path, client in mandos_clients.iteritems():
624
client_proxy_object = self.bus.get_object(self.busname,
626
self.add_client(MandosClientWidget(server_proxy_object
629
=client_proxy_object,
608
640
self._input_callback_tag = (gobject.io_add_watch
609
641
(sys.stdin.fileno(),
620
652
def process_input(self, source, condition):
621
653
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
654
translations = { "ctrl n": "down", # Emacs
655
"ctrl p": "up", # Emacs
656
"ctrl v": "page down", # Emacs
657
"meta v": "page up", # Emacs
658
" ": "page down", # less
659
"f": "page down", # less
660
"b": "page up", # less
635
667
except KeyError: # :-)
638
if key == u"q" or key == u"Q":
670
if key == "q" or key == "Q":
641
elif key == u"window resize":
673
elif key == "window resize":
642
674
self.size = self.screen.get_cols_rows()
644
elif key == u"\f": # Ctrl-L
676
elif key == "\f": # Ctrl-L
646
elif key == u"l" or key == u"D":
678
elif key == "l" or key == "D":
647
679
self.toggle_log_display()
649
elif key == u"w" or key == u"i":
681
elif key == "w" or key == "i":
650
682
self.change_log_display()
652
elif key == u"?" or key == u"f1" or key == u"esc":
684
elif key == "?" or key == "f1" or key == "esc":
653
685
if not self.log_visible:
654
686
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",
688
self.log_message_raw(("bold",
692
"l: Log window toggle",
693
"TAB: Switch window",
695
self.log_message_raw(("bold",
701
"s: Start new checker",
676
708
if self.topwidget.get_focus() is self.logbox:
677
709
self.topwidget.set_focus(0)
679
711
self.topwidget.set_focus(self.logbox)
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
713
#elif (key == "end" or key == "meta >" or key == "G"
683
715
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
716
#elif (key == "home" or key == "meta <" or key == "g"
686
718
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
719
#elif key == "ctrl e" or key == "$":
688
720
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
721
#elif key == "ctrl a" or key == "^":
690
722
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
723
#elif key == "ctrl b" or key == "meta (" or key == "h":
692
724
# pass # xxx left
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
725
#elif key == "ctrl f" or key == "meta )" or key == "l":
694
726
# pass # xxx right
696
728
# pass # scroll up log
698
730
# pass # scroll down log
699
731
elif self.topwidget.selectable():
700
732
self.topwidget.keypress(self.size, key)