2
2
# -*- mode: python; coding: utf-8 -*-
4
from __future__ import division, absolute_import, with_statement
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
21
# <http://www.gnu.org/licenses/>.
23
# Contact the authors at <mandos@recompile.se>.
26
from __future__ import (division, absolute_import, print_function,
29
from future_builtins import *
44
69
"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)
72
d, t = iso.split("T", 1)
73
year, month, day = d.split("-", 2)
74
hour, minute, second = t.split(":", 2)
50
75
second, fraction = divmod(float(second), 1)
51
76
return datetime.datetime(int(year),
116
146
self.opened = False
118
148
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:
151
if self.properties ["LastCheckerStatus"] != 0:
131
152
self.using_timer(True)
133
154
if self.need_approval:
134
155
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",
157
self.match_objects = (
158
self.proxy.connect_to_signal("CheckerCompleted",
159
self.checker_completed,
162
self.proxy.connect_to_signal("CheckerStarted",
163
self.checker_started,
166
self.proxy.connect_to_signal("GotSecret",
170
self.proxy.connect_to_signal("NeedApproval",
174
self.proxy.connect_to_signal("Rejected",
178
#self.logger('Created client {0}'
179
# .format(self.properties["Name"]))
157
181
def property_changed(self, property=None, value=None):
158
182
super(self, MandosClientWidget).property_changed(property,
160
if property == u"ApprovalPending":
184
if property == "ApprovalPending":
161
185
using_timer(bool(value))
186
if property == "LastCheckerStatus":
187
using_timer(value != 0)
188
#self.logger('Checker for client {0} (command "{1}") was '
189
# ' successful'.format(self.properties["Name"],
163
192
def using_timer(self, flag):
164
193
"""Call this method with True or False when timer should be
165
194
activated or deactivated.
180
210
def checker_completed(self, exitstatus, condition, command):
181
211
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
215
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,
198
os.WEXITSTATUS(condition)))
216
self.logger('Checker for client {0} (command "{1}")'
217
' failed with exit code {2}'
218
.format(self.properties["Name"], command,
219
os.WEXITSTATUS(condition)))
199
220
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,
203
os.WTERMSIG(condition)))
221
self.logger('Checker for client {0} (command "{1}") was'
222
' killed by signal {2}'
223
.format(self.properties["Name"], command,
224
os.WTERMSIG(condition)))
204
225
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
226
self.logger('Checker for client {0} (command "{1}")'
228
.format(self.properties["Name"], command))
209
self.logger(u'Checker for client %s completed'
230
self.logger('Checker for client {0} completed'
232
.format(self.properties["Name"]))
213
235
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
236
"""Server signals that a checker started. This could be useful
237
to log in the future. """
238
#self.logger('Client {0} started checker "{1}"'
239
# .format(self.properties["Name"],
218
243
def got_secret(self):
219
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
244
self.logger('Client {0} received its secret'
245
.format(self.properties["Name"]))
223
247
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
249
message = 'Client {0} needs approval within {1} seconds'
227
message = u'Client %s will get its secret in %s seconds'
229
% (self.properties[u"Name"], timeout/1000))
251
message = 'Client {0} will get its secret in {1} seconds'
252
self.logger(message.format(self.properties["Name"],
230
254
self.using_timer(True)
232
256
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
257
self.logger('Client {0} was rejected; reason: {1}'
258
.format(self.properties["Name"], reason))
236
260
def selectable(self):
237
261
"""Make this a "selectable" widget.
238
262
This overrides the method from urwid.FlowWidget."""
241
def rows(self, (maxcol,), focus=False):
265
def rows(self, maxcolrow, focus=False):
242
266
"""How many rows this widget will occupy might depend on
243
267
whether we have focus or not.
244
268
This overrides the method from urwid.FlowWidget"""
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
269
return self.current_widget(focus).rows(maxcolrow, focus=focus)
247
271
def current_widget(self, focus=False):
248
272
if focus or self.opened:
252
276
def update(self):
253
277
"Called when what is visible on the screen should be updated."
254
278
# 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",
279
with_standout = { "normal": "standout",
280
"bold": "bold-standout",
282
"underline-blink-standout",
283
"bold-underline-blink":
284
"bold-underline-blink-standout",
263
287
# Rebuild focus and non-focus widgets using current properties
265
289
# 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"]:
290
base = '{name}: '.format(name=self.properties["Name"])
291
if not self.properties["Enabled"]:
293
elif self.properties["ApprovalPending"]:
271
294
timeout = datetime.timedelta(milliseconds
272
295
= self.properties
274
297
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
298
self.properties["LastApprovalRequest"])
276
299
if last_approval_request is not None:
277
300
timer = timeout - (datetime.datetime.utcnow()
278
301
- last_approval_request)
280
303
timer = datetime.timedelta()
281
if self.properties[u"ApprovedByDefault"]:
282
message = u"Approval in %s. (d)eny?"
284
message = u"Denial in %s. (a)pprove?"
285
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'
297
% unicode(timer).rsplit(".", 1)[0])
304
if self.properties["ApprovedByDefault"]:
305
message = "Approval in {0}. (d)eny?"
307
message = "Denial in {0}. (a)pprove?"
308
message = message.format(unicode(timer).rsplit(".", 1)[0])
309
elif self.properties["LastCheckerStatus"] != 0:
310
# When checker has failed, show timer until client expires
311
expires = self.properties["Expires"]
313
timer = datetime.timedelta(0)
315
expires = (datetime.datetime.strptime
316
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
317
timer = expires - datetime.datetime.utcnow()
318
message = ('A checker has failed! Time until client'
319
' gets disabled: {0}'
320
.format(unicode(timer).rsplit(".", 1)[0]))
300
self._text = "%s%s" % (base, message)
323
self._text = "{0}{1}".format(base, message)
302
325
if not urwid.supports_unicode():
303
326
self._text = self._text.encode("ascii", "replace")
304
textlist = [(u"normal", self._text)]
327
textlist = [("normal", self._text)]
305
328
self._text_widget.set_text(textlist)
306
329
self._focus_text_widget.set_text([(with_standout[text[0]],
316
339
self.update_hook()
318
341
def update_timer(self):
342
"""called by gobject. Will indefinitely loop until
343
gobject.source_remove() on tag is called"""
321
345
return True # Keep calling this
347
def delete(self, *args, **kwargs):
324
348
if self._update_timer_callback_tag is not None:
325
349
gobject.source_remove(self._update_timer_callback_tag)
326
350
self._update_timer_callback_tag = None
351
for match in self.match_objects:
353
self.match_objects = ()
327
354
if self.delete_hook is not None:
328
355
self.delete_hook(self)
356
return super(MandosClientWidget, self).delete(*args, **kwargs)
330
def render(self, (maxcol,), focus=False):
358
def render(self, maxcolrow, focus=False):
331
359
"""Render differently if we have focus.
332
360
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render((maxcol,),
361
return self.current_widget(focus).render(maxcolrow,
336
def keypress(self, (maxcol,), key):
364
def keypress(self, maxcolrow, key):
338
366
This overrides the method from urwid.FlowWidget"""
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
368
self.proxy.Enable(dbus_interface = client_interface,
371
self.proxy.Disable(dbus_interface = client_interface,
344
374
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
375
dbus_interface = client_interface,
347
378
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":
379
dbus_interface = client_interface,
381
elif key == "R" or key == "_" or key == "ctrl k":
350
382
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)
386
self.proxy.StartChecker(dbus_interface = client_interface,
389
self.proxy.StopChecker(dbus_interface = client_interface,
392
self.proxy.CheckedOK(dbus_interface = client_interface,
359
# elif key == u"p" or key == "=":
395
# elif key == "p" or key == "=":
360
396
# self.proxy.pause()
361
# elif key == u"u" or key == ":":
397
# elif key == "u" or key == ":":
362
398
# self.proxy.unpause()
363
# elif key == u"RET":
382
418
"down" key presses, thus not allowing any containing widgets to
383
419
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"):
421
def keypress(self, maxcolrow, key):
422
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
423
if ret in ("up", "down"):
400
435
self.screen = urwid.curses_display.Screen()
402
437
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",
439
"default", "default", None),
441
"default", "default", "bold"),
443
"default", "default", "underline"),
445
"default", "default", "standout"),
446
("bold-underline-blink",
447
"default", "default", ("bold", "underline")),
449
"default", "default", ("bold", "standout")),
450
("underline-blink-standout",
451
"default", "default", ("underline", "standout")),
452
("bold-underline-blink-standout",
453
"default", "default", ("bold", "underline",
422
457
if urwid.supports_unicode():
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
458
self.divider = "─" # \u2500
459
#self.divider = "━" # \u2501
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
461
#self.divider = "-" # \u002d
462
self.divider = "_" # \u005f
429
464
self.screen.start()
444
479
# This keeps track of whether self.uilist currently has
445
480
# self.logbox in it or not
446
481
self.log_visible = True
447
self.log_wrap = u"any"
482
self.log_wrap = "any"
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
485
self.log_message_raw(("bold",
486
"Mandos Monitor version " + version))
487
self.log_message_raw(("bold",
455
490
self.busname = domain + '.Mandos'
456
491
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
493
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,
494
self.log_message("Client with address {0} and fingerprint"
495
" {1} could not be found"
496
.format(address, fingerprint))
505
498
def rebuild(self):
506
499
"""This rebuilds the User Interface.
530
522
and len(self.log) > self.max_log_length):
531
523
del self.log[0:len(self.log)-self.max_log_length-1]
532
524
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
536
528
def toggle_log_display(self):
537
529
"""Toggle visibility of the log buffer."""
538
530
self.log_visible = not self.log_visible
540
#self.log_message(u"Log visibility changed to: "
532
#self.log_message("Log visibility changed to: "
541
533
# + unicode(self.log_visible))
543
535
def change_log_display(self):
544
536
"""Change type of log display.
545
537
Currently, this toggles wrapping of text lines."""
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
538
if self.log_wrap == "clip":
539
self.log_wrap = "any"
549
self.log_wrap = u"clip"
541
self.log_wrap = "clip"
550
542
for textwidget in self.log:
551
543
textwidget.set_wrap_mode(self.log_wrap)
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
544
#self.log_message("Wrap mode: " + self.log_wrap)
554
546
def find_and_remove_client(self, path, name):
555
"""Find an client from its object path and remove it.
547
"""Find a client by its object path and remove it.
557
549
This is connected to the ClientRemoved signal from the
558
550
Mandos server object."""
606
600
"""Start the main loop and exit when it's done."""
601
self.bus = dbus.SystemBus()
602
mandos_dbus_objc = self.bus.get_object(
603
self.busname, "/", follow_name_owner_changes=True)
604
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
608
mandos_clients = (self.mandos_serv
609
.GetAllClientsWithProperties())
610
except dbus.exceptions.DBusException:
611
mandos_clients = dbus.Dictionary()
614
.connect_to_signal("ClientRemoved",
615
self.find_and_remove_client,
616
dbus_interface=server_interface,
619
.connect_to_signal("ClientAdded",
621
dbus_interface=server_interface,
624
.connect_to_signal("ClientNotFound",
625
self.client_not_found,
626
dbus_interface=server_interface,
628
for path, client in mandos_clients.iteritems():
629
client_proxy_object = self.bus.get_object(self.busname,
631
self.add_client(MandosClientWidget(server_proxy_object
634
=client_proxy_object,
608
645
self._input_callback_tag = (gobject.io_add_watch
609
646
(sys.stdin.fileno(),
620
657
def process_input(self, source, condition):
621
658
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
659
translations = { "ctrl n": "down", # Emacs
660
"ctrl p": "up", # Emacs
661
"ctrl v": "page down", # Emacs
662
"meta v": "page up", # Emacs
663
" ": "page down", # less
664
"f": "page down", # less
665
"b": "page up", # less
635
672
except KeyError: # :-)
638
if key == u"q" or key == u"Q":
675
if key == "q" or key == "Q":
641
elif key == u"window resize":
678
elif key == "window resize":
642
679
self.size = self.screen.get_cols_rows()
644
elif key == u"\f": # Ctrl-L
681
elif key == "\f": # Ctrl-L
646
elif key == u"l" or key == u"D":
683
elif key == "l" or key == "D":
647
684
self.toggle_log_display()
649
elif key == u"w" or key == u"i":
686
elif key == "w" or key == "i":
650
687
self.change_log_display()
652
elif key == u"?" or key == u"f1" or key == u"esc":
689
elif key == "?" or key == "f1" or key == "esc":
653
690
if not self.log_visible:
654
691
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",
693
self.log_message_raw(("bold",
697
"l: Log window toggle",
698
"TAB: Switch window",
700
self.log_message_raw(("bold",
706
"s: Start new checker",
676
713
if self.topwidget.get_focus() is self.logbox:
677
714
self.topwidget.set_focus(0)
679
716
self.topwidget.set_focus(self.logbox)
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
718
#elif (key == "end" or key == "meta >" or key == "G"
683
720
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
721
#elif (key == "home" or key == "meta <" or key == "g"
686
723
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
724
#elif key == "ctrl e" or key == "$":
688
725
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
726
#elif key == "ctrl a" or key == "^":
690
727
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
728
#elif key == "ctrl b" or key == "meta (" or key == "h":
692
729
# pass # xxx left
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
730
#elif key == "ctrl f" or key == "meta )" or key == "l":
694
731
# pass # xxx right
696
733
# pass # scroll up log
698
735
# pass # scroll down log
699
736
elif self.topwidget.selectable():
700
737
self.topwidget.keypress(self.size, key)