40
30
urwid.curses_display.curses.A_UNDERLINE |= (
41
31
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
59
33
class MandosClientPropertyCache(object):
60
34
"""This wraps a Mandos Client D-Bus proxy object, caches the
61
35
properties and calls a hook function when any of them are
64
def __init__(self, proxy_object=None, *args, **kwargs):
38
def __init__(self, proxy_object=None, properties=None, *args,
65
40
self.proxy = proxy_object # Mandos Client proxy object
67
self.properties = dict()
68
self.proxy.connect_to_signal(u"PropertyChanged",
42
if properties is None:
43
self.properties = dict()
45
self.properties = properties
46
self.proxy.connect_to_signal("PropertyChanged",
69
47
self.property_changed,
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__(
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)
81
59
def property_changed(self, property=None, value=None):
82
60
"""This is called whenever we get a PropertyChanged signal
93
71
def __init__(self, server_proxy_object=None, update_hook=None,
94
delete_hook=None, logger=None, *args, **kwargs):
72
delete_hook=None, *args, **kwargs):
96
74
self.update_hook = update_hook
98
76
self.delete_hook = delete_hook
99
77
# Mandos Server proxy object
100
78
self.server_proxy_object = server_proxy_object
104
self._update_timer_callback_tag = None
105
self.last_checker_failed = False
107
80
# The widget shown normally
108
self._text_widget = urwid.Text(u"")
81
self._text_widget = urwid.Text("")
109
82
# The widget shown when we have focus
110
self._focus_text_widget = urwid.Text(u"")
83
self._focus_text_widget = urwid.Text("")
111
84
super(MandosClientWidget, self).__init__(
112
85
update_hook=update_hook, delete_hook=delete_hook,
115
88
self.opened = False
116
self.proxy.connect_to_signal(u"CheckerCompleted",
117
self.checker_completed,
120
self.proxy.connect_to_signal(u"CheckerStarted",
121
self.checker_started,
124
self.proxy.connect_to_signal(u"GotSecret",
128
self.proxy.connect_to_signal(u"NeedApproval",
132
self.proxy.connect_to_signal(u"Rejected",
136
last_checked_ok = isoformat_to_datetime(self.properties
138
if last_checked_ok is None:
139
self.last_checker_failed = True
141
self.last_checker_failed = ((datetime.datetime.utcnow()
145
self.properties["interval"]))
146
if self.last_checker_failed:
147
self._update_timer_callback_tag = (gobject.timeout_add
151
def checker_completed(self, exitstatus, condition, command):
153
if self.last_checker_failed:
154
self.last_checker_failed = False
155
gobject.source_remove(self._update_timer_callback_tag)
156
self._update_timer_callback_tag = None
157
self.logger(u'Checker for client %s (command "%s")'
159
% (self.properties[u"name"], command))
163
if not self.last_checker_failed:
164
self.last_checker_failed = True
165
self._update_timer_callback_tag = (gobject.timeout_add
168
if os.WIFEXITED(condition):
169
self.logger(u'Checker for client %s (command "%s")'
170
u' failed with exit code %s'
171
% (self.properties[u"name"], command,
172
os.WEXITSTATUS(condition)))
173
elif os.WIFSIGNALED(condition):
174
self.logger(u'Checker for client %s (command "%s")'
175
u' was killed by signal %s'
176
% (self.properties[u"name"], command,
177
os.WTERMSIG(condition)))
178
elif os.WCOREDUMP(condition):
179
self.logger(u'Checker for client %s (command "%s")'
181
% (self.properties[u"name"], command))
183
self.logger(u'Checker for client %s completed mysteriously')
186
def checker_started(self, command):
187
#self.logger(u'Client %s started checker "%s"'
188
# % (self.properties[u"name"], unicode(command)))
191
def got_secret(self):
192
self.last_checker_failed = False
193
self.logger(u'Client %s received its secret'
194
% self.properties[u"name"])
196
def need_approval(self, timeout, default):
198
message = u'Client %s needs approval within %s seconds'
200
message = u'Client %s will get its secret in %s seconds'
202
% (self.properties[u"name"], timeout/1000))
204
def rejected(self, reason):
205
self.logger(u'Client %s was rejected; reason: %s'
206
% (self.properties[u"name"], reason))
208
90
def selectable(self):
209
91
"""Make this a "selectable" widget.
231
113
u"bold-underline-blink":
232
114
u"bold-underline-blink-standout",
235
117
# Rebuild focus and non-focus widgets using current properties
237
# Base part of a client. Name!
238
base = (u'%(name)s: '
239
% {u"name": self.properties[u"name"]})
240
if not self.properties[u"enabled"]:
241
message = u"DISABLED"
242
elif self.properties[u"approved_pending"]:
243
if self.properties[u"approved_by_default"]:
244
message = u"Connection established to client. (d)eny?"
246
message = u"Seeks approval to send secret. (a)pprove?"
247
elif self.last_checker_failed:
248
timeout = datetime.timedelta(milliseconds
249
= self.properties[u"timeout"])
250
last_ok = isoformat_to_datetime(
251
max((self.properties["last_checked_ok"]
252
or self.properties["created"]),
253
self.properties[u"last_enabled"]))
254
timer = timeout - (datetime.datetime.utcnow() - last_ok)
255
message = (u'A checker has failed! Time until client gets diabled: %s'
256
% unicode(timer).rsplit(".", 1)[0])
259
self._text = "%s%s" % (base, message)
118
self._text = (u'name="%(name)s", enabled=%(enabled)s'
261
120
if not urwid.supports_unicode():
262
121
self._text = self._text.encode("ascii", "replace")
263
textlist = [(u"normal", self._text)]
122
textlist = [(u"normal", u"BLARGH: "), (u"bold", self._text)]
264
123
self._text_widget.set_text(textlist)
265
124
self._focus_text_widget.set_text([(with_standout[text[0]],
295
146
def keypress(self, (maxcol,), key):
297
148
This overrides the method from urwid.FlowWidget"""
299
self.proxy.Enable(dbus_interface = client_interface)
301
self.proxy.Disable(dbus_interface = client_interface)
303
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
304
dbus_interface = client_interface)
306
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
307
dbus_interface = client_interface)
308
elif key == u"r" or key == u"_" or key == u"ctrl k":
149
if key == u"e" or key == u"+":
151
elif key == u"d" or key == u"-":
153
elif key == u"r" or key == u"_":
309
154
self.server_proxy_object.RemoveClient(self.proxy
311
156
elif key == u"s":
312
self.proxy.StartChecker(dbus_interface = client_interface)
157
self.proxy.StartChecker()
159
self.proxy.StopChecker()
313
160
elif key == u"S":
314
self.proxy.StopChecker(dbus_interface = client_interface)
316
self.proxy.CheckedOK(dbus_interface = client_interface)
161
self.proxy.CheckedOK()
318
163
# elif key == u"p" or key == "=":
319
164
# self.proxy.pause()
429
270
mandos_clients = dbus.Dictionary()
431
272
(self.mandos_serv
432
.connect_to_signal(u"ClientRemoved",
273
.connect_to_signal("ClientRemoved",
433
274
self.find_and_remove_client,
434
275
dbus_interface=server_interface,
435
276
byte_arrays=True))
436
277
(self.mandos_serv
437
.connect_to_signal(u"ClientAdded",
278
.connect_to_signal("ClientAdded",
438
279
self.add_new_client,
439
280
dbus_interface=server_interface,
440
281
byte_arrays=True))
442
.connect_to_signal(u"ClientNotFound",
443
self.client_not_found,
444
dbus_interface=server_interface,
446
282
for path, client in mandos_clients.iteritems():
447
283
client_proxy_object = self.bus.get_object(self.busname,
479
308
self.topwidget = urwid.Pile(self.uilist)
481
def log_message(self, message):
482
timestamp = datetime.datetime.now().isoformat()
483
self.log_message_raw(timestamp + u": " + message)
485
def log_message_raw(self, markup):
310
def log_message(self, markup):
486
311
"""Add a log message to the log buffer."""
487
312
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
488
313
if (self.max_log_length
489
314
and len(self.log) > self.max_log_length):
490
315
del self.log[0:len(self.log)-self.max_log_length-1]
491
self.logbox.set_focus(len(self.logbox.body.contents),
492
coming_from=u"above")
495
317
def toggle_log_display(self):
496
318
"""Toggle visibility of the log buffer."""
638
439
self.topwidget.set_focus(self.logbox)
640
#elif (key == u"end" or key == u"meta >" or key == u"G"
642
# pass # xxx end-of-buffer
643
#elif (key == u"home" or key == u"meta <" or key == u"g"
645
# pass # xxx beginning-of-buffer
646
#elif key == u"ctrl e" or key == u"$":
647
# pass # xxx move-end-of-line
648
#elif key == u"ctrl a" or key == u"^":
649
# pass # xxx move-beginning-of-line
650
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
652
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
655
# pass # scroll up log
657
# pass # scroll down log
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
658
459
elif self.topwidget.selectable():
659
460
self.topwidget.keypress(self.size, key)