65
30
urwid.curses_display.curses.A_UNDERLINE |= (
66
31
urwid.curses_display.curses.A_BLINK)
68
def isoformat_to_datetime(iso):
69
"Parse an ISO 8601 date string to a datetime.datetime()"
72
d, t = iso.split("T", 1)
73
year, month, day = d.split("-", 2)
74
hour, minute, second = t.split(":", 2)
75
second, fraction = divmod(float(second), 1)
76
return datetime.datetime(int(year),
81
int(second), # Whole seconds
82
int(fraction*1000000)) # Microseconds
84
33
class MandosClientPropertyCache(object):
85
34
"""This wraps a Mandos Client D-Bus proxy object, caches the
86
35
properties and calls a hook function when any of them are
89
def __init__(self, proxy_object=None, *args, **kwargs):
38
def __init__(self, proxy_object=None, properties=None, *args,
40
# Type conversion mapping
42
dbus.ObjectPath: unicode,
44
dbus.Signature: unicode,
52
dbus.Dictionary: dict,
90
59
self.proxy = proxy_object # Mandos Client proxy object
92
self.properties = dict()
93
self.property_changed_match = (
94
self.proxy.connect_to_signal("PropertyChanged",
95
self.property_changed,
61
if properties is None:
62
self.properties = dict()
64
self.properties = dict(self.convert_property(prop, val)
66
properties.iteritems())
67
self.proxy.connect_to_signal("PropertyChanged",
68
self.property_changed,
99
self.properties.update(
100
self.proxy.GetAll(client_interface,
101
dbus_interface = dbus.PROPERTIES_IFACE))
103
#XXX This breaks good super behaviour
104
# super(MandosClientPropertyCache, self).__init__(
72
if properties is None:
73
self.properties.update(
74
self.convert_property(prop, val)
76
self.proxy.GetAll(client_interface,
78
dbus.PROPERTIES_IFACE).iteritems())
79
super(MandosClientPropertyCache, self).__init__(
80
proxy_object=proxy_object,
81
properties=properties, *args, **kwargs)
83
def convert_property(self, property, value):
84
"""This converts the arguments from a D-Bus signal, which are
85
D-Bus types, into normal Python types, using a conversion
86
function from "self.type_map".
88
property_name = unicode(property) # Always a dbus.String
89
if isinstance(value, dbus.UTF8String):
90
# Should not happen, but prepare for it anyway
91
value = dbus.String(str(value).decode("utf-8"))
93
convfunc = self.type_map[type(value)]
95
# Unknown type, return unmodified
96
return property_name, value
97
return property_name, convfunc(value)
107
98
def property_changed(self, property=None, value=None):
108
99
"""This is called whenever we get a PropertyChanged signal
109
100
It updates the changed property in the "properties" dict.
102
# Convert name and value
103
property_name, cvalue = self.convert_property(property, value)
111
104
# Update properties dict with new value
112
self.properties[property] = value
114
def delete(self, *args, **kwargs):
115
self.property_changed_match.remove()
116
super(MandosClientPropertyCache, self).__init__(
105
self.properties[property_name] = cvalue
120
108
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
146
129
self.opened = False
148
last_checked_ok = isoformat_to_datetime(self.properties
151
if self.properties ["LastCheckerStatus"] != 0:
152
self.using_timer(True)
154
if self.need_approval:
155
self.using_timer(True)
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"]))
181
def property_changed(self, property=None, value=None):
182
super(self, MandosClientWidget).property_changed(property,
184
if property == "ApprovalPending":
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"],
192
def using_timer(self, flag):
193
"""Call this method with True or False when timer should be
194
activated or deactivated.
196
old = self._update_timer_callback_lock
198
self._update_timer_callback_lock += 1
200
self._update_timer_callback_lock -= 1
201
if old == 0 and self._update_timer_callback_lock:
202
# Will update the shown timer value every second
203
self._update_timer_callback_tag = (gobject.timeout_add
206
elif old and self._update_timer_callback_lock == 0:
207
gobject.source_remove(self._update_timer_callback_tag)
208
self._update_timer_callback_tag = None
210
def checker_completed(self, exitstatus, condition, command):
215
if os.WIFEXITED(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)))
220
elif os.WIFSIGNALED(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)))
225
elif os.WCOREDUMP(condition):
226
self.logger('Checker for client {0} (command "{1}")'
228
.format(self.properties["Name"], command))
230
self.logger('Checker for client {0} completed'
232
.format(self.properties["Name"]))
235
def checker_started(self, 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"],
243
def got_secret(self):
244
self.logger('Client {0} received its secret'
245
.format(self.properties["Name"]))
247
def need_approval(self, timeout, default):
249
message = 'Client {0} needs approval within {1} seconds'
251
message = 'Client {0} will get its secret in {1} seconds'
252
self.logger(message.format(self.properties["Name"],
254
self.using_timer(True)
256
def rejected(self, reason):
257
self.logger('Client {0} was rejected; reason: {1}'
258
.format(self.properties["Name"], reason))
260
131
def selectable(self):
261
132
"""Make this a "selectable" widget.
262
133
This overrides the method from urwid.FlowWidget."""
265
def rows(self, maxcolrow, focus=False):
136
def rows(self, (maxcol,), focus=False):
266
137
"""How many rows this widget will occupy might depend on
267
138
whether we have focus or not.
268
139
This overrides the method from urwid.FlowWidget"""
269
return self.current_widget(focus).rows(maxcolrow, focus=focus)
140
return self.current_widget(focus).rows((maxcol,), focus=focus)
271
142
def current_widget(self, focus=False):
272
143
if focus or self.opened:
276
147
def update(self):
277
148
"Called when what is visible on the screen should be updated."
278
149
# How to add standout mode to a style
279
with_standout = { "normal": "standout",
280
"bold": "bold-standout",
282
"underline-blink-standout",
283
"bold-underline-blink":
284
"bold-underline-blink-standout",
150
with_standout = { u"normal": u"standout",
151
u"bold": u"bold-standout",
153
u"underline-blink-standout",
154
u"bold-underline-blink":
155
u"bold-underline-blink-standout",
287
158
# Rebuild focus and non-focus widgets using current properties
289
# Base part of a client. Name!
290
base = '{name}: '.format(name=self.properties["Name"])
291
if not self.properties["Enabled"]:
293
elif self.properties["ApprovalPending"]:
294
timeout = datetime.timedelta(milliseconds
297
last_approval_request = isoformat_to_datetime(
298
self.properties["LastApprovalRequest"])
299
if last_approval_request is not None:
300
timer = timeout - (datetime.datetime.utcnow()
301
- last_approval_request)
303
timer = datetime.timedelta()
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]))
323
self._text = "{0}{1}".format(base, message)
159
self._text = (u'name="%(name)s", enabled=%(enabled)s'
325
161
if not urwid.supports_unicode():
326
162
self._text = self._text.encode("ascii", "replace")
327
textlist = [("normal", self._text)]
163
textlist = [(u"normal", u"BLÄRGH: "), (u"bold", self._text)]
328
164
self._text_widget.set_text(textlist)
329
165
self._focus_text_widget.set_text([(with_standout[text[0]],
338
174
if self.update_hook is not None:
339
175
self.update_hook()
341
def update_timer(self):
342
"""called by gobject. Will indefinitely loop until
343
gobject.source_remove() on tag is called"""
345
return True # Keep calling this
347
def delete(self, *args, **kwargs):
348
if self._update_timer_callback_tag is not None:
349
gobject.source_remove(self._update_timer_callback_tag)
350
self._update_timer_callback_tag = None
351
for match in self.match_objects:
353
self.match_objects = ()
354
178
if self.delete_hook is not None:
355
179
self.delete_hook(self)
356
return super(MandosClientWidget, self).delete(*args, **kwargs)
358
def render(self, maxcolrow, focus=False):
181
def render(self, (maxcol,), focus=False):
359
182
"""Render differently if we have focus.
360
183
This overrides the method from urwid.FlowWidget"""
361
return self.current_widget(focus).render(maxcolrow,
184
return self.current_widget(focus).render((maxcol,),
364
def keypress(self, maxcolrow, key):
187
def keypress(self, (maxcol,), key):
366
189
This overrides the method from urwid.FlowWidget"""
368
self.proxy.Enable(dbus_interface = client_interface,
371
self.proxy.Disable(dbus_interface = client_interface,
374
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
375
dbus_interface = client_interface,
378
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
379
dbus_interface = client_interface,
381
elif key == "R" or key == "_" or key == "ctrl k":
190
if key == u"e" or key == u"+":
192
elif key == u"d" or key == u"-":
194
elif key == u"r" or key == u"_":
382
195
self.server_proxy_object.RemoveClient(self.proxy
386
self.proxy.StartChecker(dbus_interface = client_interface,
389
self.proxy.StopChecker(dbus_interface = client_interface,
392
self.proxy.CheckedOK(dbus_interface = client_interface,
198
self.proxy.StartChecker()
200
self.proxy.StopChecker()
202
self.proxy.CheckedOK()
395
# elif key == "p" or key == "=":
204
# elif key == u"p" or key == "=":
396
205
# self.proxy.pause()
397
# elif key == "u" or key == ":":
206
# elif key == u"u" or key == ":":
398
207
# self.proxy.unpause()
208
# elif key == u"RET":
416
class ConstrainedListBox(urwid.ListBox):
417
"""Like a normal urwid.ListBox, but will consume all "up" or
418
"down" key presses, thus not allowing any containing widgets to
419
use them as an excuse to shift focus away from this widget.
421
def keypress(self, maxcolrow, key):
422
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
423
if ret in ("up", "down"):
428
225
class UserInterface(object):
429
226
"""This is the entire user interface - the whole screen
430
227
with boxes, lists of client widgets, etc.
432
def __init__(self, max_log_length=1000):
433
DBusGMainLoop(set_as_default=True)
230
DBusGMainLoop(set_as_default=True )
435
232
self.screen = urwid.curses_display.Screen()
437
234
self.screen.register_palette((
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",
236
u"default", u"default", None),
238
u"default", u"default", u"bold"),
240
u"default", u"default", u"underline"),
242
u"default", u"default", u"standout"),
243
(u"bold-underline-blink",
244
u"default", u"default", (u"bold", u"underline")),
246
u"default", u"default", (u"bold", u"standout")),
247
(u"underline-blink-standout",
248
u"default", u"default", (u"underline", u"standout")),
249
(u"bold-underline-blink-standout",
250
u"default", u"default", (u"bold", u"underline",
457
if urwid.supports_unicode():
458
self.divider = "─" # \u2500
459
#self.divider = "━" # \u2501
461
#self.divider = "-" # \u002d
462
self.divider = "_" # \u005f
464
254
self.screen.start()
466
256
self.size = self.screen.get_cols_rows()
468
258
self.clients = urwid.SimpleListWalker([])
469
259
self.clients_dict = {}
471
# We will add Text widgets to this list
473
self.max_log_length = max_log_length
475
# We keep a reference to the log widget so we can remove it
476
# from the ListWalker without it getting destroyed
477
self.logbox = ConstrainedListBox(self.log)
479
# This keeps track of whether self.uilist currently has
480
# self.logbox in it or not
481
self.log_visible = True
482
self.log_wrap = "any"
485
self.log_message_raw(("bold",
486
"Mandos Monitor version " + version))
487
self.log_message_raw(("bold",
260
self.topwidget = urwid.LineBox(urwid.ListBox(self.clients))
261
#self.topwidget = urwid.ListBox(clients)
490
263
self.busname = domain + '.Mandos'
491
264
self.main_loop = gobject.MainLoop()
493
def client_not_found(self, fingerprint, address):
494
self.log_message("Client with address {0} and fingerprint"
495
" {1} could not be found"
496
.format(address, fingerprint))
499
"""This rebuilds the User Interface.
500
Call this when the widget layout needs to change"""
502
#self.uilist.append(urwid.ListBox(self.clients))
503
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
505
#header=urwid.Divider(),
508
urwid.Divider(div_char=
511
self.uilist.append(self.logbox)
512
self.topwidget = urwid.Pile(self.uilist)
514
def log_message(self, message):
515
timestamp = datetime.datetime.now().isoformat()
516
self.log_message_raw(timestamp + ": " + message)
518
def log_message_raw(self, markup):
519
"""Add a log message to the log buffer."""
520
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
521
if (self.max_log_length
522
and len(self.log) > self.max_log_length):
523
del self.log[0:len(self.log)-self.max_log_length-1]
524
self.logbox.set_focus(len(self.logbox.body.contents),
528
def toggle_log_display(self):
529
"""Toggle visibility of the log buffer."""
530
self.log_visible = not self.log_visible
532
#self.log_message("Log visibility changed to: "
533
# + unicode(self.log_visible))
535
def change_log_display(self):
536
"""Change type of log display.
537
Currently, this toggles wrapping of text lines."""
538
if self.log_wrap == "clip":
539
self.log_wrap = "any"
541
self.log_wrap = "clip"
542
for textwidget in self.log:
543
textwidget.set_wrap_mode(self.log_wrap)
544
#self.log_message("Wrap mode: " + self.log_wrap)
265
self.bus = dbus.SystemBus()
266
mandos_dbus_objc = self.bus.get_object(
267
self.busname, u"/", follow_name_owner_changes=True)
268
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
272
mandos_clients = (self.mandos_serv
273
.GetAllClientsWithProperties())
274
except dbus.exceptions.DBusException:
275
mandos_clients = dbus.Dictionary()
278
.connect_to_signal("ClientRemoved",
279
self.find_and_remove_client,
280
dbus_interface=server_interface,
283
.connect_to_signal("ClientAdded",
285
dbus_interface=server_interface,
287
for path, client in (mandos_clients.iteritems()):
288
client_proxy_object = self.bus.get_object(self.busname,
290
self.add_client(MandosClientWidget(server_proxy_object
293
=client_proxy_object,
298
=self.remove_client),
546
301
def find_and_remove_client(self, path, name):
547
"""Find a client by its object path and remove it.
302
"""Find an client from its object path and remove it.
549
304
This is connected to the ClientRemoved signal from the
550
305
Mandos server object."""
672
369
except KeyError: # :-)
675
if key == "q" or key == "Q":
372
if key == u"q" or key == u"Q":
678
elif key == "window resize":
375
elif key == u"window resize":
679
376
self.size = self.screen.get_cols_rows()
681
elif key == "\f": # Ctrl-L
683
elif key == "l" or key == "D":
684
self.toggle_log_display()
686
elif key == "w" or key == "i":
687
self.change_log_display()
689
elif key == "?" or key == "f1" or key == "esc":
690
if not self.log_visible:
691
self.log_visible = True
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",
713
if self.topwidget.get_focus() is self.logbox:
714
self.topwidget.set_focus(0)
716
self.topwidget.set_focus(self.logbox)
718
#elif (key == "end" or key == "meta >" or key == "G"
720
# pass # xxx end-of-buffer
721
#elif (key == "home" or key == "meta <" or key == "g"
723
# pass # xxx beginning-of-buffer
724
#elif key == "ctrl e" or key == "$":
725
# pass # xxx move-end-of-line
726
#elif key == "ctrl a" or key == "^":
727
# pass # xxx move-beginning-of-line
728
#elif key == "ctrl b" or key == "meta (" or key == "h":
730
#elif key == "ctrl f" or key == "meta )" or key == "l":
733
# pass # scroll up log
735
# pass # scroll down log
736
380
elif self.topwidget.selectable():
737
381
self.topwidget.keypress(self.size, key)