30
40
urwid.curses_display.curses.A_UNDERLINE |= (
31
41
urwid.curses_display.curses.A_BLINK)
43
def isoformat_to_datetime(iso):
44
"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)
50
second, fraction = divmod(float(second), 1)
51
return datetime.datetime(int(year),
56
int(second), # Whole seconds
57
int(fraction*1000000)) # Microseconds
33
59
class MandosClientPropertyCache(object):
34
60
"""This wraps a Mandos Client D-Bus proxy object, caches the
35
61
properties and calls a hook function when any of them are
38
def __init__(self, proxy_object=None, properties=None, *args,
64
def __init__(self, proxy_object=None, *args, **kwargs):
40
65
self.proxy = proxy_object # Mandos Client proxy object
42
if properties is None:
43
self.properties = dict()
45
self.properties = properties
46
self.proxy.connect_to_signal("PropertyChanged",
67
self.properties = dict()
68
self.proxy.connect_to_signal(u"PropertyChanged",
47
69
self.property_changed,
51
if properties is None:
52
self.properties.update(self.proxy.GetAll(client_interface,
54
dbus.PROPERTIES_IFACE))
55
super(MandosClientPropertyCache, self).__init__(
56
proxy_object=proxy_object,
57
properties=properties, *args, **kwargs)
73
self.properties.update(
74
self.proxy.GetAll(client_interface,
75
dbus_interface = dbus.PROPERTIES_IFACE))
77
#XXX This break good super behaviour!
78
# super(MandosClientPropertyCache, self).__init__(
59
81
def property_changed(self, property=None, value=None):
60
82
"""This is called whenever we get a PropertyChanged signal
71
93
def __init__(self, server_proxy_object=None, update_hook=None,
72
delete_hook=None, *args, **kwargs):
94
delete_hook=None, logger=None, *args, **kwargs):
74
96
self.update_hook = update_hook
76
98
self.delete_hook = delete_hook
77
99
# Mandos Server proxy object
78
100
self.server_proxy_object = server_proxy_object
104
self._update_timer_callback_tag = None
105
self._update_timer_callback_lock = 0
106
self.last_checker_failed = False
80
108
# The widget shown normally
81
self._text_widget = urwid.Text("")
109
self._text_widget = urwid.Text(u"")
82
110
# The widget shown when we have focus
83
self._focus_text_widget = urwid.Text("")
111
self._focus_text_widget = urwid.Text(u"")
84
112
super(MandosClientWidget, self).__init__(
85
113
update_hook=update_hook, delete_hook=delete_hook,
88
116
self.opened = False
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()
130
if self.last_checker_failed:
131
self.using_timer(True)
133
if self.need_approval:
134
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
def property_changed(self, property=None, value=None):
158
super(self, MandosClientWidget).property_changed(property,
160
if property == u"ApprovalPending":
161
using_timer(bool(value))
163
def using_timer(self, flag):
164
"""Call this method with True or False when timer should be
165
activated or deactivated.
167
old = self._update_timer_callback_lock
169
self._update_timer_callback_lock += 1
171
self._update_timer_callback_lock -= 1
172
if old == 0 and self._update_timer_callback_lock:
173
self._update_timer_callback_tag = (gobject.timeout_add
176
elif old and self._update_timer_callback_lock == 0:
177
gobject.source_remove(self._update_timer_callback_tag)
178
self._update_timer_callback_tag = None
180
def checker_completed(self, exitstatus, condition, command):
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
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)))
199
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)))
204
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
209
self.logger(u'Checker for client %s completed'
213
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
218
def got_secret(self):
219
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
223
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
227
message = u'Client %s will get its secret in %s seconds'
229
% (self.properties[u"Name"], timeout/1000))
230
self.using_timer(True)
232
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
90
236
def selectable(self):
91
237
"""Make this a "selectable" widget.
113
259
u"bold-underline-blink":
114
260
u"bold-underline-blink-standout",
117
263
# Rebuild focus and non-focus widgets using current properties
118
self._text = (u'name="%(name)s", enabled=%(enabled)s'
265
# 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"]:
271
timeout = datetime.timedelta(milliseconds
274
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
276
if last_approval_request is not None:
277
timer = timeout - (datetime.datetime.utcnow()
278
- last_approval_request)
280
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])
300
self._text = "%s%s" % (base, message)
120
302
if not urwid.supports_unicode():
121
303
self._text = self._text.encode("ascii", "replace")
122
textlist = [(u"normal", u"BLARGH: "), (u"bold", self._text)]
304
textlist = [(u"normal", self._text)]
123
305
self._text_widget.set_text(textlist)
124
306
self._focus_text_widget.set_text([(with_standout[text[0]],
146
336
def keypress(self, (maxcol,), key):
148
338
This overrides the method from urwid.FlowWidget"""
149
if key == u"e" or key == u"+":
151
elif key == u"d" or key == u"-":
153
elif key == u"r" or key == u"_":
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
344
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
347
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":
154
350
self.server_proxy_object.RemoveClient(self.proxy
156
352
elif key == u"s":
157
self.proxy.StartChecker()
159
self.proxy.StopChecker()
353
self.proxy.StartChecker(dbus_interface = client_interface)
160
354
elif key == u"S":
161
self.proxy.CheckedOK()
355
self.proxy.StopChecker(dbus_interface = client_interface)
357
self.proxy.CheckedOK(dbus_interface = client_interface)
163
359
# elif key == u"p" or key == "=":
164
360
# self.proxy.pause()
270
467
mandos_clients = dbus.Dictionary()
272
469
(self.mandos_serv
273
.connect_to_signal("ClientRemoved",
470
.connect_to_signal(u"ClientRemoved",
274
471
self.find_and_remove_client,
275
472
dbus_interface=server_interface,
276
473
byte_arrays=True))
277
474
(self.mandos_serv
278
.connect_to_signal("ClientAdded",
475
.connect_to_signal(u"ClientAdded",
279
476
self.add_new_client,
280
477
dbus_interface=server_interface,
281
478
byte_arrays=True))
480
.connect_to_signal(u"ClientNotFound",
481
self.client_not_found,
482
dbus_interface=server_interface,
282
484
for path, client in mandos_clients.iteritems():
283
485
client_proxy_object = self.bus.get_object(self.busname,
293
=self.remove_client),
500
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,
296
505
def rebuild(self):
297
506
"""This rebuilds the User Interface.
298
507
Call this when the widget layout needs to change"""
300
509
#self.uilist.append(urwid.ListBox(self.clients))
301
self.uilist.append(urwid.Frame(ConstrainedListBox(self.clients),
510
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
302
512
#header=urwid.Divider(),
304
footer=urwid.Divider(div_char=self.divider)))
515
urwid.Divider(div_char=
305
517
if self.log_visible:
306
518
self.uilist.append(self.logbox)
308
520
self.topwidget = urwid.Pile(self.uilist)
310
def log_message(self, markup):
522
def log_message(self, message):
523
timestamp = datetime.datetime.now().isoformat()
524
self.log_message_raw(timestamp + u": " + message)
526
def log_message_raw(self, markup):
311
527
"""Add a log message to the log buffer."""
312
528
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
313
529
if (self.max_log_length
314
530
and len(self.log) > self.max_log_length):
315
531
del self.log[0:len(self.log)-self.max_log_length-1]
532
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
317
536
def toggle_log_display(self):
318
537
"""Toggle visibility of the log buffer."""
319
538
self.log_visible = not self.log_visible
321
self.log_message(u"Log visibility changed to: "
322
+ unicode(self.log_visible))
540
#self.log_message(u"Log visibility changed to: "
541
# + unicode(self.log_visible))
324
543
def change_log_display(self):
325
544
"""Change type of log display.
439
679
self.topwidget.set_focus(self.logbox)
441
elif (key == u"end" or key == u"meta >" or key == u"G"
443
pass # xxx end-of-buffer
444
elif (key == u"home" or key == u"meta <" or key == u"g"
446
pass # xxx beginning-of-buffer
447
elif key == u"ctrl e" or key == u"$":
448
pass # xxx move-end-of-line
449
elif key == u"ctrl a" or key == u"^":
450
pass # xxx move-beginning-of-line
451
elif key == u"ctrl b" or key == u"meta (" or key == u"h":
453
elif key == u"ctrl f" or key == u"meta )" or key == u"l":
458
pass # scroll down log
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
683
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
686
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
688
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
690
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
696
# pass # scroll up log
698
# pass # scroll down log
459
699
elif self.topwidget.selectable():
460
700
self.topwidget.keypress(self.size, key)