55
60
domain = 'se.recompile'
56
61
server_interface = domain + '.Mandos'
57
62
client_interface = domain + '.Mandos.Client'
60
# Always run in monochrome mode
61
urwid.curses_display.curses.has_colors = lambda : False
63
# Urwid doesn't support blinking, but we want it. Since we have no
64
# use for underline on its own, we make underline also always blink.
65
urwid.curses_display.curses.A_UNDERLINE |= (
66
urwid.curses_display.curses.A_BLINK)
66
dbus.OBJECT_MANAGER_IFACE
67
except AttributeError:
68
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
68
70
def isoformat_to_datetime(iso):
69
71
"Parse an ISO 8601 date string to a datetime.datetime()"
86
88
properties and calls a hook function when any of them are
89
def __init__(self, proxy_object=None, *args, **kwargs):
91
def __init__(self, proxy_object=None, properties=None, **kwargs):
90
92
self.proxy = proxy_object # Mandos Client proxy object
92
self.properties = dict()
93
self.properties = dict() if properties is None else properties
93
94
self.property_changed_match = (
94
self.proxy.connect_to_signal("PropertyChanged",
95
self.property_changed,
95
self.proxy.connect_to_signal("PropertiesChanged",
96
self.properties_changed,
97
dbus.PROPERTIES_IFACE,
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__(
100
if properties is None:
101
self.properties.update(
102
self.proxy.GetAll(client_interface,
104
= dbus.PROPERTIES_IFACE))
106
super(MandosClientPropertyCache, self).__init__(**kwargs)
107
def property_changed(self, property=None, value=None):
108
"""This is called whenever we get a PropertyChanged signal
109
It updates the changed property in the "properties" dict.
108
def properties_changed(self, interface, properties, invalidated):
109
"""This is called whenever we get a PropertiesChanged signal
110
It updates the changed properties in the "properties" dict.
111
112
# Update properties dict with new value
112
self.properties[property] = value
113
if interface == client_interface:
114
self.properties.update(properties)
114
def delete(self, *args, **kwargs):
115
117
self.property_changed_match.remove()
116
super(MandosClientPropertyCache, self).__init__(
120
120
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
133
133
self.logger = logger
135
135
self._update_timer_callback_tag = None
136
self._update_timer_callback_lock = 0
138
137
# The widget shown normally
139
138
self._text_widget = urwid.Text("")
140
139
# The widget shown when we have focus
141
140
self._focus_text_widget = urwid.Text("")
142
super(MandosClientWidget, self).__init__(
143
update_hook=update_hook, delete_hook=delete_hook,
141
super(MandosClientWidget, self).__init__(**kwargs)
146
143
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
145
self.match_objects = (
158
146
self.proxy.connect_to_signal("CheckerCompleted",
159
147
self.checker_completed,
176
164
client_interface,
177
165
byte_arrays=True))
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"],
166
self.logger('Created client {}'
167
.format(self.properties["Name"]), level=0)
192
169
def using_timer(self, flag):
193
170
"""Call this method with True or False when timer should be
194
171
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:
173
if flag and self._update_timer_callback_tag is None:
202
174
# Will update the shown timer value every second
203
175
self._update_timer_callback_tag = (gobject.timeout_add
205
177
self.update_timer))
206
elif old and self._update_timer_callback_lock == 0:
178
elif not (flag or self._update_timer_callback_tag is None):
207
179
gobject.source_remove(self._update_timer_callback_tag)
208
180
self._update_timer_callback_tag = None
210
182
def checker_completed(self, exitstatus, condition, command):
211
183
if exitstatus == 0:
184
self.logger('Checker for client {} (command "{}")'
185
' succeeded'.format(self.properties["Name"],
215
190
if os.WIFEXITED(condition):
216
self.logger('Checker for client {0} (command "{1}")'
217
' failed with exit code {2}'
191
self.logger('Checker for client {} (command "{}") failed'
218
193
.format(self.properties["Name"], command,
219
194
os.WEXITSTATUS(condition)))
220
195
elif os.WIFSIGNALED(condition):
221
self.logger('Checker for client {0} (command "{1}") was'
222
' killed by signal {2}'
196
self.logger('Checker for client {} (command "{}") was'
197
' killed by signal {}'
223
198
.format(self.properties["Name"], command,
224
199
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
202
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"],
203
"""Server signals that a checker started."""
204
self.logger('Client {} started checker "{}"'
205
.format(self.properties["Name"],
243
208
def got_secret(self):
244
self.logger('Client {0} received its secret'
209
self.logger('Client {} received its secret'
245
210
.format(self.properties["Name"]))
247
212
def need_approval(self, timeout, default):
249
message = 'Client {0} needs approval within {1} seconds'
214
message = 'Client {} needs approval within {} seconds'
251
message = 'Client {0} will get its secret in {1} seconds'
216
message = 'Client {} will get its secret in {} seconds'
252
217
self.logger(message.format(self.properties["Name"],
254
self.using_timer(True)
256
220
def rejected(self, reason):
257
self.logger('Client {0} was rejected; reason: {1}'
221
self.logger('Client {} was rejected; reason: {}'
258
222
.format(self.properties["Name"], reason))
260
224
def selectable(self):
297
262
last_approval_request = isoformat_to_datetime(
298
263
self.properties["LastApprovalRequest"])
299
264
if last_approval_request is not None:
300
timer = timeout - (datetime.datetime.utcnow()
301
- last_approval_request)
265
timer = max(timeout - (datetime.datetime.utcnow()
266
- last_approval_request),
267
datetime.timedelta())
303
269
timer = datetime.timedelta()
304
270
if self.properties["ApprovedByDefault"]:
305
message = "Approval in {0}. (d)eny?"
271
message = "Approval in {}. (d)eny?"
307
message = "Denial in {0}. (a)pprove?"
308
message = message.format(unicode(timer).rsplit(".", 1)[0])
273
message = "Denial in {}. (a)pprove?"
274
message = message.format(str(timer).rsplit(".", 1)[0])
275
self.using_timer(True)
309
276
elif self.properties["LastCheckerStatus"] != 0:
310
277
# When checker has failed, show timer until client expires
311
278
expires = self.properties["Expires"]
315
282
expires = (datetime.datetime.strptime
316
283
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
317
timer = expires - datetime.datetime.utcnow()
284
timer = max(expires - datetime.datetime.utcnow(),
285
datetime.timedelta())
318
286
message = ('A checker has failed! Time until client'
319
' gets disabled: {0}'
320
.format(unicode(timer).rsplit(".", 1)[0]))
288
.format(str(timer).rsplit(".", 1)[0]))
289
self.using_timer(True)
322
291
message = "enabled"
323
self._text = "{0}{1}".format(base, message)
292
self.using_timer(False)
293
self._text = "{}{}".format(base, message)
325
295
if not urwid.supports_unicode():
326
296
self._text = self._text.encode("ascii", "replace")
327
297
textlist = [("normal", self._text)]
366
336
This overrides the method from urwid.FlowWidget"""
368
self.proxy.Enable(dbus_interface = client_interface,
338
self.proxy.Set(client_interface, "Enabled",
339
dbus.Boolean(True), ignore_reply = True,
340
dbus_interface = dbus.PROPERTIES_IFACE)
371
self.proxy.Disable(dbus_interface = client_interface,
342
self.proxy.Set(client_interface, "Enabled", False,
344
dbus_interface = dbus.PROPERTIES_IFACE)
374
346
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
375
347
dbus_interface = client_interface,
384
356
ignore_reply=True)
386
self.proxy.StartChecker(dbus_interface = client_interface,
358
self.proxy.Set(client_interface, "CheckerRunning",
359
dbus.Boolean(True), ignore_reply = True,
360
dbus_interface = dbus.PROPERTIES_IFACE)
389
self.proxy.StopChecker(dbus_interface = client_interface,
362
self.proxy.Set(client_interface, "CheckerRunning",
363
dbus.Boolean(False), ignore_reply = True,
364
dbus_interface = dbus.PROPERTIES_IFACE)
392
366
self.proxy.CheckedOK(dbus_interface = client_interface,
393
367
ignore_reply=True)
404
def property_changed(self, property=None, value=None,
406
"""Call self.update() if old value is not new value.
378
def properties_changed(self, interface, properties, invalidated):
379
"""Call self.update() if any properties changed.
407
380
This overrides the method from MandosClientPropertyCache"""
408
property_name = unicode(property)
409
old_value = self.properties.get(property_name)
410
super(MandosClientWidget, self).property_changed(
411
property=property, value=value, *args, **kwargs)
412
if self.properties.get(property_name) != old_value:
381
old_values = { key: self.properties.get(key)
382
for key in properties.keys() }
383
super(MandosClientWidget, self).properties_changed(
384
interface, properties, invalidated)
385
if any(old_values[key] != self.properties.get(key)
386
for key in old_values):
439
413
"default", "default", None),
441
"default", "default", "bold"),
415
"bold", "default", "bold"),
442
416
("underline-blink",
443
"default", "default", "underline"),
417
"underline,blink", "default", "underline,blink"),
445
"default", "default", "standout"),
419
"standout", "default", "standout"),
446
420
("bold-underline-blink",
447
"default", "default", ("bold", "underline")),
421
"bold,underline,blink", "default", "bold,underline,blink"),
448
422
("bold-standout",
449
"default", "default", ("bold", "standout")),
423
"bold,standout", "default", "bold,standout"),
450
424
("underline-blink-standout",
451
"default", "default", ("underline", "standout")),
425
"underline,blink,standout", "default",
426
"underline,blink,standout"),
452
427
("bold-underline-blink-standout",
453
"default", "default", ("bold", "underline",
428
"bold,underline,blink,standout", "default",
429
"bold,underline,blink,standout"),
457
432
if urwid.supports_unicode():
511
488
self.uilist.append(self.logbox)
512
489
self.topwidget = urwid.Pile(self.uilist)
514
def log_message(self, message):
491
def log_message(self, message, level=1):
492
"""Log message formatted with timestamp"""
493
if level < self.log_level:
515
495
timestamp = datetime.datetime.now().isoformat()
516
self.log_message_raw(timestamp + ": " + message)
496
self.log_message_raw("{}: {}".format(timestamp, message),
518
def log_message_raw(self, markup):
499
def log_message_raw(self, markup, level=1):
519
500
"""Add a log message to the log buffer."""
501
if level < self.log_level:
520
503
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
521
504
if (self.max_log_length
522
505
and len(self.log) > self.max_log_length):
541
524
self.log_wrap = "clip"
542
525
for textwidget in self.log:
543
526
textwidget.set_wrap_mode(self.log_wrap)
544
#self.log_message("Wrap mode: " + self.log_wrap)
527
self.log_message("Wrap mode: {}".format(self.log_wrap),
546
def find_and_remove_client(self, path, name):
530
def find_and_remove_client(self, path, interfaces):
547
531
"""Find a client by its object path and remove it.
549
This is connected to the ClientRemoved signal from the
533
This is connected to the InterfacesRemoved signal from the
550
534
Mandos server object."""
535
if client_interface not in interfaces:
536
# Not a Mandos client object; ignore
552
539
client = self.clients_dict[path]
555
self.log_message("Unknown client {0!r} ({1!r}) removed"
542
self.log_message("Unknown client {!r} removed"
560
def add_new_client(self, path):
547
def add_new_client(self, path, ifs_and_props):
548
"""Find a client by its object path and remove it.
550
This is connected to the InterfacesAdded signal from the
551
Mandos server object.
553
if client_interface not in ifs_and_props:
554
# Not a Mandos client object; ignore
561
556
client_proxy_object = self.bus.get_object(self.busname, path)
562
557
self.add_client(MandosClientWidget(server_proxy_object
563
558
=self.mandos_serv,
608
601
mandos_clients = (self.mandos_serv
609
602
.GetAllClientsWithProperties())
603
if not mandos_clients:
604
self.log_message_raw(("bold", "Note: Server has no clients."))
610
605
except dbus.exceptions.DBusException:
606
self.log_message_raw(("bold", "Note: No Mandos server running."))
611
607
mandos_clients = dbus.Dictionary()
613
609
(self.mandos_serv
614
.connect_to_signal("ClientRemoved",
610
.connect_to_signal("InterfacesRemoved",
615
611
self.find_and_remove_client,
616
dbus_interface=server_interface,
613
= dbus.OBJECT_MANAGER_IFACE,
617
614
byte_arrays=True))
618
615
(self.mandos_serv
619
.connect_to_signal("ClientAdded",
616
.connect_to_signal("InterfacesAdded",
620
617
self.add_new_client,
621
dbus_interface=server_interface,
619
= dbus.OBJECT_MANAGER_IFACE,
622
620
byte_arrays=True))
623
621
(self.mandos_serv
624
622
.connect_to_signal("ClientNotFound",
625
623
self.client_not_found,
626
624
dbus_interface=server_interface,
627
625
byte_arrays=True))
628
for path, client in mandos_clients.iteritems():
626
for path, client in mandos_clients.items():
629
627
client_proxy_object = self.bus.get_object(self.busname,
631
629
self.add_client(MandosClientWidget(server_proxy_object