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