63
30
urwid.curses_display.curses.A_UNDERLINE |= (
64
31
urwid.curses_display.curses.A_BLINK)
66
def isoformat_to_datetime(iso):
67
"Parse an ISO 8601 date string to a datetime.datetime()"
70
d, t = iso.split("T", 1)
71
year, month, day = d.split("-", 2)
72
hour, minute, second = t.split(":", 2)
73
second, fraction = divmod(float(second), 1)
74
return datetime.datetime(int(year),
79
int(second), # Whole seconds
80
int(fraction*1000000)) # Microseconds
82
33
class MandosClientPropertyCache(object):
83
34
"""This wraps a Mandos Client D-Bus proxy object, caches the
84
35
properties and calls a hook function when any of them are
87
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,
88
59
self.proxy = proxy_object # Mandos Client proxy object
90
self.properties = dict()
91
self.property_changed_match = (
92
self.proxy.connect_to_signal("PropertyChanged",
93
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,
97
self.properties.update(
98
self.proxy.GetAll(client_interface,
99
dbus_interface = dbus.PROPERTIES_IFACE))
101
#XXX This breaks good super behaviour
102
# 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)
105
98
def property_changed(self, property=None, value=None):
106
99
"""This is called whenever we get a PropertyChanged signal
107
100
It updates the changed property in the "properties" dict.
102
# Convert name and value
103
property_name, cvalue = self.convert_property(property, value)
109
104
# Update properties dict with new value
110
self.properties[property] = value
112
def delete(self, *args, **kwargs):
113
self.property_changed_match.remove()
114
super(MandosClientPropertyCache, self).__init__(
105
self.properties[property_name] = cvalue
118
108
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
144
129
self.opened = False
146
last_checked_ok = isoformat_to_datetime(self.properties
149
if self.properties ["LastCheckerStatus"] != 0:
150
self.using_timer(True)
152
if self.need_approval:
153
self.using_timer(True)
155
self.match_objects = (
156
self.proxy.connect_to_signal("CheckerCompleted",
157
self.checker_completed,
160
self.proxy.connect_to_signal("CheckerStarted",
161
self.checker_started,
164
self.proxy.connect_to_signal("GotSecret",
168
self.proxy.connect_to_signal("NeedApproval",
172
self.proxy.connect_to_signal("Rejected",
176
#self.logger('Created client {0}'
177
# .format(self.properties["Name"]))
179
def property_changed(self, property=None, value=None):
180
super(self, MandosClientWidget).property_changed(property,
182
if property == "ApprovalPending":
183
using_timer(bool(value))
184
if property == "LastCheckerStatus":
185
using_timer(value != 0)
186
#self.logger('Checker for client {0} (command "{1}") was '
187
# ' successful'.format(self.properties["Name"],
190
def using_timer(self, flag):
191
"""Call this method with True or False when timer should be
192
activated or deactivated.
194
old = self._update_timer_callback_lock
196
self._update_timer_callback_lock += 1
198
self._update_timer_callback_lock -= 1
199
if old == 0 and self._update_timer_callback_lock:
200
# Will update the shown timer value every second
201
self._update_timer_callback_tag = (gobject.timeout_add
204
elif old and self._update_timer_callback_lock == 0:
205
gobject.source_remove(self._update_timer_callback_tag)
206
self._update_timer_callback_tag = None
208
def checker_completed(self, exitstatus, condition, command):
213
if os.WIFEXITED(condition):
214
self.logger('Checker for client {0} (command "{1}")'
215
' failed with exit code {2}'
216
.format(self.properties["Name"], command,
217
os.WEXITSTATUS(condition)))
218
elif os.WIFSIGNALED(condition):
219
self.logger('Checker for client {0} (command "{1}") was'
220
' killed by signal {2}'
221
.format(self.properties["Name"], command,
222
os.WTERMSIG(condition)))
223
elif os.WCOREDUMP(condition):
224
self.logger('Checker for client {0} (command "{1}")'
226
.format(self.properties["Name"], command))
228
self.logger('Checker for client {0} completed'
230
.format(self.properties["Name"]))
233
def checker_started(self, command):
234
"""Server signals that a checker started. This could be useful
235
to log in the future. """
236
#self.logger('Client {0} started checker "{1}"'
237
# .format(self.properties["Name"],
241
def got_secret(self):
242
self.logger('Client {0} received its secret'
243
.format(self.properties["Name"]))
245
def need_approval(self, timeout, default):
247
message = 'Client {0} needs approval within {1} seconds'
249
message = 'Client {0} will get its secret in {1} seconds'
250
self.logger(message.format(self.properties["Name"],
252
self.using_timer(True)
254
def rejected(self, reason):
255
self.logger('Client {0} was rejected; reason: {1}'
256
.format(self.properties["Name"], reason))
258
131
def selectable(self):
259
132
"""Make this a "selectable" widget.
260
133
This overrides the method from urwid.FlowWidget."""
263
def rows(self, maxcolrow, focus=False):
136
def rows(self, (maxcol,), focus=False):
264
137
"""How many rows this widget will occupy might depend on
265
138
whether we have focus or not.
266
139
This overrides the method from urwid.FlowWidget"""
267
return self.current_widget(focus).rows(maxcolrow, focus=focus)
140
return self.current_widget(focus).rows((maxcol,), focus=focus)
269
142
def current_widget(self, focus=False):
270
143
if focus or self.opened:
274
147
def update(self):
275
148
"Called when what is visible on the screen should be updated."
276
149
# How to add standout mode to a style
277
with_standout = { "normal": "standout",
278
"bold": "bold-standout",
280
"underline-blink-standout",
281
"bold-underline-blink":
282
"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",
285
158
# Rebuild focus and non-focus widgets using current properties
287
# Base part of a client. Name!
288
base = '{name}: '.format(name=self.properties["Name"])
289
if not self.properties["Enabled"]:
291
elif self.properties["ApprovalPending"]:
292
timeout = datetime.timedelta(milliseconds
295
last_approval_request = isoformat_to_datetime(
296
self.properties["LastApprovalRequest"])
297
if last_approval_request is not None:
298
timer = timeout - (datetime.datetime.utcnow()
299
- last_approval_request)
301
timer = datetime.timedelta()
302
if self.properties["ApprovedByDefault"]:
303
message = "Approval in {0}. (d)eny?"
305
message = "Denial in {0}. (a)pprove?"
306
message = message.format(unicode(timer).rsplit(".", 1)[0])
307
elif self.properties["LastCheckerStatus"] != 0:
308
# When checker has failed, show timer until client expires
309
expires = self.properties["Expires"]
311
timer = datetime.timedelta(0)
313
expires = (datetime.datetime.strptime
314
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
315
timer = expires - datetime.datetime.utcnow()
316
message = ('A checker has failed! Time until client'
317
' gets disabled: {0}'
318
.format(unicode(timer).rsplit(".", 1)[0]))
321
self._text = "{0}{1}".format(base, message)
159
self._text = (u'name="%(name)s", enabled=%(enabled)s'
323
161
if not urwid.supports_unicode():
324
162
self._text = self._text.encode("ascii", "replace")
325
textlist = [("normal", self._text)]
163
textlist = [(u"normal", u"BLÄRGH: "), (u"bold", self._text)]
326
164
self._text_widget.set_text(textlist)
327
165
self._focus_text_widget.set_text([(with_standout[text[0]],
336
174
if self.update_hook is not None:
337
175
self.update_hook()
339
def update_timer(self):
340
"""called by gobject. Will indefinitely loop until
341
gobject.source_remove() on tag is called"""
343
return True # Keep calling this
345
def delete(self, *args, **kwargs):
346
if self._update_timer_callback_tag is not None:
347
gobject.source_remove(self._update_timer_callback_tag)
348
self._update_timer_callback_tag = None
349
for match in self.match_objects:
351
self.match_objects = ()
352
178
if self.delete_hook is not None:
353
179
self.delete_hook(self)
354
return super(MandosClientWidget, self).delete(*args, **kwargs)
356
def render(self, maxcolrow, focus=False):
181
def render(self, (maxcol,), focus=False):
357
182
"""Render differently if we have focus.
358
183
This overrides the method from urwid.FlowWidget"""
359
return self.current_widget(focus).render(maxcolrow,
184
return self.current_widget(focus).render((maxcol,),
362
def keypress(self, maxcolrow, key):
187
def keypress(self, (maxcol,), key):
364
189
This overrides the method from urwid.FlowWidget"""
366
self.proxy.Enable(dbus_interface = client_interface,
369
self.proxy.Disable(dbus_interface = client_interface,
372
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
373
dbus_interface = client_interface,
376
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
377
dbus_interface = client_interface,
379
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"_":
380
195
self.server_proxy_object.RemoveClient(self.proxy
384
self.proxy.StartChecker(dbus_interface = client_interface,
387
self.proxy.StopChecker(dbus_interface = client_interface,
390
self.proxy.CheckedOK(dbus_interface = client_interface,
198
self.proxy.StartChecker()
200
self.proxy.StopChecker()
202
self.proxy.CheckedOK()
393
# elif key == "p" or key == "=":
204
# elif key == u"p" or key == "=":
394
205
# self.proxy.pause()
395
# elif key == "u" or key == ":":
206
# elif key == u"u" or key == ":":
396
207
# self.proxy.unpause()
208
# elif key == u"RET":
414
class ConstrainedListBox(urwid.ListBox):
415
"""Like a normal urwid.ListBox, but will consume all "up" or
416
"down" key presses, thus not allowing any containing widgets to
417
use them as an excuse to shift focus away from this widget.
419
def keypress(self, maxcolrow, key):
420
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
421
if ret in ("up", "down"):
426
225
class UserInterface(object):
427
226
"""This is the entire user interface - the whole screen
428
227
with boxes, lists of client widgets, etc.
430
def __init__(self, max_log_length=1000):
431
DBusGMainLoop(set_as_default=True)
230
DBusGMainLoop(set_as_default=True )
433
232
self.screen = urwid.curses_display.Screen()
435
234
self.screen.register_palette((
437
"default", "default", None),
439
"default", "default", "bold"),
441
"default", "default", "underline"),
443
"default", "default", "standout"),
444
("bold-underline-blink",
445
"default", "default", ("bold", "underline")),
447
"default", "default", ("bold", "standout")),
448
("underline-blink-standout",
449
"default", "default", ("underline", "standout")),
450
("bold-underline-blink-standout",
451
"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",
455
if urwid.supports_unicode():
456
self.divider = "─" # \u2500
457
#self.divider = "━" # \u2501
459
#self.divider = "-" # \u002d
460
self.divider = "_" # \u005f
462
254
self.screen.start()
464
256
self.size = self.screen.get_cols_rows()
466
258
self.clients = urwid.SimpleListWalker([])
467
259
self.clients_dict = {}
469
# We will add Text widgets to this list
471
self.max_log_length = max_log_length
473
# We keep a reference to the log widget so we can remove it
474
# from the ListWalker without it getting destroyed
475
self.logbox = ConstrainedListBox(self.log)
477
# This keeps track of whether self.uilist currently has
478
# self.logbox in it or not
479
self.log_visible = True
480
self.log_wrap = "any"
483
self.log_message_raw(("bold",
484
"Mandos Monitor version " + version))
485
self.log_message_raw(("bold",
260
self.topwidget = urwid.LineBox(urwid.ListBox(self.clients))
261
#self.topwidget = urwid.ListBox(clients)
488
263
self.busname = domain + '.Mandos'
489
264
self.main_loop = gobject.MainLoop()
491
def client_not_found(self, fingerprint, address):
492
self.log_message("Client with address {0} and fingerprint"
493
" {1} could not be found"
494
.format(address, fingerprint))
497
"""This rebuilds the User Interface.
498
Call this when the widget layout needs to change"""
500
#self.uilist.append(urwid.ListBox(self.clients))
501
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
503
#header=urwid.Divider(),
506
urwid.Divider(div_char=
509
self.uilist.append(self.logbox)
510
self.topwidget = urwid.Pile(self.uilist)
512
def log_message(self, message):
513
timestamp = datetime.datetime.now().isoformat()
514
self.log_message_raw(timestamp + ": " + message)
516
def log_message_raw(self, markup):
517
"""Add a log message to the log buffer."""
518
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
519
if (self.max_log_length
520
and len(self.log) > self.max_log_length):
521
del self.log[0:len(self.log)-self.max_log_length-1]
522
self.logbox.set_focus(len(self.logbox.body.contents),
526
def toggle_log_display(self):
527
"""Toggle visibility of the log buffer."""
528
self.log_visible = not self.log_visible
530
#self.log_message("Log visibility changed to: "
531
# + unicode(self.log_visible))
533
def change_log_display(self):
534
"""Change type of log display.
535
Currently, this toggles wrapping of text lines."""
536
if self.log_wrap == "clip":
537
self.log_wrap = "any"
539
self.log_wrap = "clip"
540
for textwidget in self.log:
541
textwidget.set_wrap_mode(self.log_wrap)
542
#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),
544
301
def find_and_remove_client(self, path, name):
545
"""Find a client by its object path and remove it.
302
"""Find an client from its object path and remove it.
547
304
This is connected to the ClientRemoved signal from the
548
305
Mandos server object."""
670
369
except KeyError: # :-)
673
if key == "q" or key == "Q":
372
if key == u"q" or key == u"Q":
676
elif key == "window resize":
375
elif key == u"window resize":
677
376
self.size = self.screen.get_cols_rows()
679
elif key == "\f": # Ctrl-L
681
elif key == "l" or key == "D":
682
self.toggle_log_display()
684
elif key == "w" or key == "i":
685
self.change_log_display()
687
elif key == "?" or key == "f1" or key == "esc":
688
if not self.log_visible:
689
self.log_visible = True
691
self.log_message_raw(("bold",
695
"l: Log window toggle",
696
"TAB: Switch window",
698
self.log_message_raw(("bold",
704
"s: Start new checker",
711
if self.topwidget.get_focus() is self.logbox:
712
self.topwidget.set_focus(0)
714
self.topwidget.set_focus(self.logbox)
716
#elif (key == "end" or key == "meta >" or key == "G"
718
# pass # xxx end-of-buffer
719
#elif (key == "home" or key == "meta <" or key == "g"
721
# pass # xxx beginning-of-buffer
722
#elif key == "ctrl e" or key == "$":
723
# pass # xxx move-end-of-line
724
#elif key == "ctrl a" or key == "^":
725
# pass # xxx move-beginning-of-line
726
#elif key == "ctrl b" or key == "meta (" or key == "h":
728
#elif key == "ctrl f" or key == "meta )" or key == "l":
731
# pass # scroll up log
733
# pass # scroll down log
734
380
elif self.topwidget.selectable():
735
381
self.topwidget.keypress(self.size, key)