33
40
urwid.curses_display.curses.A_UNDERLINE |= (
34
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
36
59
class MandosClientPropertyCache(object):
37
60
"""This wraps a Mandos Client D-Bus proxy object, caches the
38
61
properties and calls a hook function when any of them are
41
def __init__(self, proxy_object=None, properties=None, *args,
64
def __init__(self, proxy_object=None, *args, **kwargs):
43
65
self.proxy = proxy_object # Mandos Client proxy object
45
if properties is None:
46
self.properties = dict()
48
self.properties = properties
67
self.properties = dict()
49
68
self.proxy.connect_to_signal(u"PropertyChanged",
50
69
self.property_changed,
54
if properties is None:
55
self.properties.update(self.proxy.GetAll(client_interface,
57
dbus.PROPERTIES_IFACE))
58
super(MandosClientPropertyCache, self).__init__(
59
proxy_object=proxy_object,
60
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__(
62
81
def property_changed(self, property=None, value=None):
63
82
"""This is called whenever we get a PropertyChanged signal
104
126
client_interface,
105
127
byte_arrays=True)
128
self.proxy.connect_to_signal(u"NeedApproval",
106
132
self.proxy.connect_to_signal(u"Rejected",
108
134
client_interface,
109
135
byte_arrays=True)
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()
147
if self.last_checker_failed:
148
self._update_timer_callback_tag = (gobject.timeout_add
111
152
def checker_completed(self, exitstatus, condition, command):
112
153
if exitstatus == 0:
113
self.logger(u'Checker for client %s (command "%s")'
115
% (self.properties[u"name"], command))
154
if self.last_checker_failed:
155
self.last_checker_failed = False
156
gobject.source_remove(self._update_timer_callback_tag)
157
self._update_timer_callback_tag = None
158
#self.logger(u'Checker for client %s (command "%s")'
160
# % (self.properties[u"Name"], command))
164
if not self.last_checker_failed:
165
self.last_checker_failed = True
166
self._update_timer_callback_tag = (gobject.timeout_add
117
169
if os.WIFEXITED(condition):
118
170
self.logger(u'Checker for client %s (command "%s")'
119
171
u' failed with exit code %s'
120
% (self.properties[u"name"], command,
172
% (self.properties[u"Name"], command,
121
173
os.WEXITSTATUS(condition)))
123
if os.WIFSIGNALED(condition):
174
elif os.WIFSIGNALED(condition):
124
175
self.logger(u'Checker for client %s (command "%s")'
125
176
u' was killed by signal %s'
126
% (self.properties[u"name"], command,
177
% (self.properties[u"Name"], command,
127
178
os.WTERMSIG(condition)))
129
if os.WCOREDUMP(condition):
179
elif os.WCOREDUMP(condition):
130
180
self.logger(u'Checker for client %s (command "%s")'
132
% (self.properties[u"name"], command))
133
self.logger(u'Checker for client %s completed mysteriously')
182
% (self.properties[u"Name"], command))
184
self.logger(u'Checker for client %s completed'
135
188
def checker_started(self, command):
136
self.logger(u'Client %s started checker "%s"'
137
% (self.properties[u"name"], unicode(command)))
189
#self.logger(u'Client %s started checker "%s"'
190
# % (self.properties[u"Name"], unicode(command)))
139
193
def got_secret(self):
194
self.last_checker_failed = False
140
195
self.logger(u'Client %s received its secret'
141
% self.properties[u"name"])
144
self.logger(u'Client %s was rejected'
145
% self.properties[u"name"])
196
% self.properties[u"Name"])
198
def need_approval(self, timeout, default):
200
message = u'Client %s needs approval within %s seconds'
202
message = u'Client %s will get its secret in %s seconds'
204
% (self.properties[u"Name"], timeout/1000))
206
def rejected(self, reason):
207
self.logger(u'Client %s was rejected; reason: %s'
208
% (self.properties[u"Name"], reason))
147
210
def selectable(self):
148
211
"""Make this a "selectable" widget.
170
233
u"bold-underline-blink":
171
234
u"bold-underline-blink-standout",
174
237
# Rebuild focus and non-focus widgets using current properties
175
self._text = (u'%(name)s: %(enabled)s'
176
% { u"name": self.properties[u"name"],
179
if self.properties[u"enabled"]
239
# Base part of a client. Name!
240
base = (u'%(name)s: '
241
% {u"name": self.properties[u"Name"]})
242
if not self.properties[u"Enabled"]:
243
message = u"DISABLED"
244
elif self.properties[u"ApprovalPending"]:
245
if self.properties[u"ApprovedByDefault"]:
246
message = u"Connection established to client. (d)eny?"
248
message = u"Seeks approval to send secret. (a)pprove?"
249
elif self.last_checker_failed:
250
timeout = datetime.timedelta(milliseconds
253
last_ok = isoformat_to_datetime(
254
max((self.properties[u"LastCheckedOK"]
255
or self.properties[u"Created"]),
256
self.properties[u"LastEnabled"]))
257
timer = timeout - (datetime.datetime.utcnow() - last_ok)
258
message = (u'A checker has failed! Time until client'
260
% unicode(timer).rsplit(".", 1)[0])
263
self._text = "%s%s" % (base, message)
181
265
if not urwid.supports_unicode():
182
266
self._text = self._text.encode("ascii", "replace")
183
267
textlist = [(u"normal", self._text)]
207
299
def keypress(self, (maxcol,), key):
209
301
This overrides the method from urwid.FlowWidget"""
210
if key == u"e" or key == u"+":
212
elif key == u"d" or key == u"-":
303
self.proxy.Enable(dbus_interface = client_interface)
305
self.proxy.Disable(dbus_interface = client_interface)
307
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
308
dbus_interface = client_interface)
310
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
311
dbus_interface = client_interface)
214
312
elif key == u"r" or key == u"_" or key == u"ctrl k":
215
313
self.server_proxy_object.RemoveClient(self.proxy
217
315
elif key == u"s":
218
self.proxy.StartChecker()
316
self.proxy.StartChecker(dbus_interface = client_interface)
219
317
elif key == u"S":
220
self.proxy.StopChecker()
318
self.proxy.StopChecker(dbus_interface = client_interface)
221
319
elif key == u"C":
222
self.proxy.CheckedOK()
320
self.proxy.CheckedOK(dbus_interface = client_interface)
224
322
# elif key == u"p" or key == "=":
225
323
# self.proxy.pause()