83
85
properties and calls a hook function when any of them are
86
def __init__(self, proxy_object=None, *args, **kwargs):
88
def __init__(self, proxy_object=None, properties=None, **kwargs):
87
89
self.proxy = proxy_object # Mandos Client proxy object
89
self.properties = dict()
90
self.properties = dict() if properties is None else properties
90
91
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
92
self.proxy.connect_to_signal("PropertiesChanged",
93
self.properties_changed,
94
dbus.PROPERTIES_IFACE,
96
self.properties.update(
97
self.proxy.GetAll(client_interface,
98
dbus_interface = dbus.PROPERTIES_IFACE))
100
#XXX This breaks good super behaviour
101
# super(MandosClientPropertyCache, self).__init__(
97
if properties is None:
98
self.properties.update(
99
self.proxy.GetAll(client_interface,
101
= dbus.PROPERTIES_IFACE))
103
super(MandosClientPropertyCache, self).__init__(**kwargs)
104
def property_changed(self, property=None, value=None):
105
"""This is called whenever we get a PropertyChanged signal
106
It updates the changed property in the "properties" dict.
105
def properties_changed(self, interface, properties, invalidated):
106
"""This is called whenever we get a PropertiesChanged signal
107
It updates the changed properties in the "properties" dict.
108
109
# Update properties dict with new value
109
self.properties[property] = value
110
if interface == client_interface:
111
self.properties.update(properties)
111
def delete(self, *args, **kwargs):
112
114
self.property_changed_match.remove()
113
super(MandosClientPropertyCache, self).__init__(
117
117
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
130
130
self.logger = logger
132
132
self._update_timer_callback_tag = None
133
self._update_timer_callback_lock = 0
134
self.last_checker_failed = False
136
134
# The widget shown normally
137
135
self._text_widget = urwid.Text("")
138
136
# The widget shown when we have focus
139
137
self._focus_text_widget = urwid.Text("")
140
super(MandosClientWidget, self).__init__(
141
update_hook=update_hook, delete_hook=delete_hook,
138
super(MandosClientWidget, self).__init__(**kwargs)
144
140
self.opened = False
146
last_checked_ok = isoformat_to_datetime(self.properties
148
if last_checked_ok is None:
149
self.last_checker_failed = True
151
self.last_checker_failed = ((datetime.datetime.utcnow()
158
if self.last_checker_failed:
159
self.using_timer(True)
161
if self.need_approval:
162
self.using_timer(True)
164
142
self.match_objects = (
165
143
self.proxy.connect_to_signal("CheckerCompleted",
166
144
self.checker_completed,
183
161
client_interface,
184
162
byte_arrays=True))
185
#self.logger('Created client %s' % (self.properties["Name"]))
163
self.logger('Created client {}'
164
.format(self.properties["Name"]), level=0)
187
def property_changed(self, property=None, value=None):
188
super(self, MandosClientWidget).property_changed(property,
190
if property == "ApprovalPending":
191
using_timer(bool(value))
193
166
def using_timer(self, flag):
194
167
"""Call this method with True or False when timer should be
195
168
activated or deactivated.
197
old = self._update_timer_callback_lock
199
self._update_timer_callback_lock += 1
201
self._update_timer_callback_lock -= 1
202
if old == 0 and self._update_timer_callback_lock:
170
if flag and self._update_timer_callback_tag is None:
203
171
# Will update the shown timer value every second
204
self._update_timer_callback_tag = (gobject.timeout_add
172
self._update_timer_callback_tag = (GLib.timeout_add
206
174
self.update_timer))
207
elif old and self._update_timer_callback_lock == 0:
208
gobject.source_remove(self._update_timer_callback_tag)
175
elif not (flag or self._update_timer_callback_tag is None):
176
GLib.source_remove(self._update_timer_callback_tag)
209
177
self._update_timer_callback_tag = None
211
179
def checker_completed(self, exitstatus, condition, command):
212
180
if exitstatus == 0:
213
if self.last_checker_failed:
214
self.last_checker_failed = False
215
self.using_timer(False)
216
#self.logger('Checker for client %s (command "%s")'
218
# % (self.properties["Name"], command))
181
self.logger('Checker for client {} (command "{}")'
182
' succeeded'.format(self.properties["Name"],
222
if not self.last_checker_failed:
223
self.last_checker_failed = True
224
self.using_timer(True)
225
187
if os.WIFEXITED(condition):
226
self.logger('Checker for client %s (command "%s")'
227
' failed with exit code %s'
228
% (self.properties["Name"], command,
229
os.WEXITSTATUS(condition)))
188
self.logger('Checker for client {} (command "{}") failed'
190
.format(self.properties["Name"], command,
191
os.WEXITSTATUS(condition)))
230
192
elif os.WIFSIGNALED(condition):
231
self.logger('Checker for client %s (command "%s")'
232
' was killed by signal %s'
233
% (self.properties["Name"], command,
234
os.WTERMSIG(condition)))
235
elif os.WCOREDUMP(condition):
236
self.logger('Checker for client %s (command "%s")'
238
% (self.properties["Name"], command))
240
self.logger('Checker for client %s completed'
193
self.logger('Checker for client {} (command "{}") was'
194
' killed by signal {}'
195
.format(self.properties["Name"], command,
196
os.WTERMSIG(condition)))
244
199
def checker_started(self, command):
245
"""Server signals that a checker started. This could be useful
246
to log in the future. """
247
#self.logger('Client %s started checker "%s"'
248
# % (self.properties["Name"], unicode(command)))
200
"""Server signals that a checker started."""
201
self.logger('Client {} started checker "{}"'
202
.format(self.properties["Name"],
251
205
def got_secret(self):
252
self.last_checker_failed = False
253
self.logger('Client %s received its secret'
254
% self.properties["Name"])
206
self.logger('Client {} received its secret'
207
.format(self.properties["Name"]))
256
209
def need_approval(self, timeout, default):
258
message = 'Client %s needs approval within %s seconds'
211
message = 'Client {} needs approval within {} seconds'
260
message = 'Client %s will get its secret in %s seconds'
262
% (self.properties["Name"], timeout/1000))
263
self.using_timer(True)
213
message = 'Client {} will get its secret in {} seconds'
214
self.logger(message.format(self.properties["Name"],
265
217
def rejected(self, reason):
266
self.logger('Client %s was rejected; reason: %s'
267
% (self.properties["Name"], reason))
218
self.logger('Client {} was rejected; reason: {}'
219
.format(self.properties["Name"], reason))
269
221
def selectable(self):
270
222
"""Make this a "selectable" widget.
307
259
last_approval_request = isoformat_to_datetime(
308
260
self.properties["LastApprovalRequest"])
309
261
if last_approval_request is not None:
310
timer = timeout - (datetime.datetime.utcnow()
311
- last_approval_request)
262
timer = max(timeout - (datetime.datetime.utcnow()
263
- last_approval_request),
264
datetime.timedelta())
313
266
timer = datetime.timedelta()
314
267
if self.properties["ApprovedByDefault"]:
315
message = "Approval in %s. (d)eny?"
268
message = "Approval in {}. (d)eny?"
317
message = "Denial in %s. (a)pprove?"
318
message = message % unicode(timer).rsplit(".", 1)[0]
319
elif self.last_checker_failed:
320
# When checker has failed, print a timer until client expires
270
message = "Denial in {}. (a)pprove?"
271
message = message.format(str(timer).rsplit(".", 1)[0])
272
self.using_timer(True)
273
elif self.properties["LastCheckerStatus"] != 0:
274
# When checker has failed, show timer until client expires
321
275
expires = self.properties["Expires"]
322
276
if expires == "":
323
277
timer = datetime.timedelta(0)
325
expires = datetime.datetime.strptime(expires,
326
'%Y-%m-%dT%H:%M:%S.%f')
327
timer = expires - datetime.datetime.utcnow()
279
expires = (datetime.datetime.strptime
280
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
281
timer = max(expires - datetime.datetime.utcnow(),
282
datetime.timedelta())
328
283
message = ('A checker has failed! Time until client'
330
% unicode(timer).rsplit(".", 1)[0])
285
.format(str(timer).rsplit(".", 1)[0]))
286
self.using_timer(True)
332
288
message = "enabled"
333
self._text = "%s%s" % (base, message)
289
self.using_timer(False)
290
self._text = "{}{}".format(base, message)
335
292
if not urwid.supports_unicode():
336
293
self._text = self._text.encode("ascii", "replace")
337
294
textlist = [("normal", self._text)]
349
306
self.update_hook()
351
308
def update_timer(self):
352
"""called by gobject. Will indefinitely loop until
353
gobject.source_remove() on tag is called"""
309
"""called by GLib. Will indefinitely loop until
310
GLib.source_remove() on tag is called
355
313
return True # Keep calling this
357
def delete(self, *args, **kwargs):
315
def delete(self, **kwargs):
358
316
if self._update_timer_callback_tag is not None:
359
gobject.source_remove(self._update_timer_callback_tag)
317
GLib.source_remove(self._update_timer_callback_tag)
360
318
self._update_timer_callback_tag = None
361
319
for match in self.match_objects:
363
321
self.match_objects = ()
364
322
if self.delete_hook is not None:
365
323
self.delete_hook(self)
366
return super(MandosClientWidget, self).delete(*args, **kwargs)
324
return super(MandosClientWidget, self).delete(**kwargs)
368
326
def render(self, maxcolrow, focus=False):
369
327
"""Render differently if we have focus.
376
334
This overrides the method from urwid.FlowWidget"""
378
self.proxy.Enable(dbus_interface = client_interface,
336
self.proxy.Set(client_interface, "Enabled",
337
dbus.Boolean(True), ignore_reply = True,
338
dbus_interface = dbus.PROPERTIES_IFACE)
381
self.proxy.Disable(dbus_interface = client_interface,
340
self.proxy.Set(client_interface, "Enabled", False,
342
dbus_interface = dbus.PROPERTIES_IFACE)
384
344
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
385
345
dbus_interface = client_interface,
414
def property_changed(self, property=None, value=None,
416
"""Call self.update() if old value is not new value.
376
def properties_changed(self, interface, properties, invalidated):
377
"""Call self.update() if any properties changed.
417
378
This overrides the method from MandosClientPropertyCache"""
418
property_name = unicode(property)
419
old_value = self.properties.get(property_name)
420
super(MandosClientWidget, self).property_changed(
421
property=property, value=value, *args, **kwargs)
422
if self.properties.get(property_name) != old_value:
379
old_values = { key: self.properties.get(key)
380
for key in properties.keys() }
381
super(MandosClientWidget, self).properties_changed(
382
interface, properties, invalidated)
383
if any(old_values[key] != self.properties.get(key)
384
for key in old_values):
449
411
"default", "default", None),
451
"default", "default", "bold"),
413
"bold", "default", "bold"),
452
414
("underline-blink",
453
"default", "default", "underline"),
415
"underline,blink", "default", "underline,blink"),
455
"default", "default", "standout"),
417
"standout", "default", "standout"),
456
418
("bold-underline-blink",
457
"default", "default", ("bold", "underline")),
419
"bold,underline,blink", "default", "bold,underline,blink"),
458
420
("bold-standout",
459
"default", "default", ("bold", "standout")),
421
"bold,standout", "default", "bold,standout"),
460
422
("underline-blink-standout",
461
"default", "default", ("underline", "standout")),
423
"underline,blink,standout", "default",
424
"underline,blink,standout"),
462
425
("bold-underline-blink-standout",
463
"default", "default", ("bold", "underline",
426
"bold,underline,blink,standout", "default",
427
"bold,underline,blink,standout"),
467
430
if urwid.supports_unicode():
521
486
self.uilist.append(self.logbox)
522
487
self.topwidget = urwid.Pile(self.uilist)
524
def log_message(self, message):
489
def log_message(self, message, level=1):
490
"""Log message formatted with timestamp"""
491
if level < self.log_level:
525
493
timestamp = datetime.datetime.now().isoformat()
526
self.log_message_raw(timestamp + ": " + message)
494
self.log_message_raw("{}: {}".format(timestamp, message),
528
def log_message_raw(self, markup):
497
def log_message_raw(self, markup, level=1):
529
498
"""Add a log message to the log buffer."""
499
if level < self.log_level:
530
501
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
531
502
if (self.max_log_length
532
503
and len(self.log) > self.max_log_length):
551
522
self.log_wrap = "clip"
552
523
for textwidget in self.log:
553
524
textwidget.set_wrap_mode(self.log_wrap)
554
#self.log_message("Wrap mode: " + self.log_wrap)
525
self.log_message("Wrap mode: {}".format(self.log_wrap),
556
def find_and_remove_client(self, path, name):
528
def find_and_remove_client(self, path, interfaces):
557
529
"""Find a client by its object path and remove it.
559
This is connected to the ClientRemoved signal from the
531
This is connected to the InterfacesRemoved signal from the
560
532
Mandos server object."""
533
if client_interface not in interfaces:
534
# Not a Mandos client object; ignore
562
537
client = self.clients_dict[path]
565
self.log_message("Unknown client %r (%r) removed", name,
540
self.log_message("Unknown client {!r} removed"
570
def add_new_client(self, path):
545
def add_new_client(self, path, ifs_and_props):
546
"""Find a client by its object path and remove it.
548
This is connected to the InterfacesAdded signal from the
549
Mandos server object.
551
if client_interface not in ifs_and_props:
552
# Not a Mandos client object; ignore
571
554
client_proxy_object = self.bus.get_object(self.busname, path)
572
555
self.add_client(MandosClientWidget(server_proxy_object
573
556
=self.mandos_serv,
618
599
mandos_clients = (self.mandos_serv
619
600
.GetAllClientsWithProperties())
601
if not mandos_clients:
602
self.log_message_raw(("bold", "Note: Server has no clients."))
620
603
except dbus.exceptions.DBusException:
604
self.log_message_raw(("bold", "Note: No Mandos server running."))
621
605
mandos_clients = dbus.Dictionary()
623
607
(self.mandos_serv
624
.connect_to_signal("ClientRemoved",
608
.connect_to_signal("InterfacesRemoved",
625
609
self.find_and_remove_client,
626
dbus_interface=server_interface,
611
= dbus.OBJECT_MANAGER_IFACE,
627
612
byte_arrays=True))
628
613
(self.mandos_serv
629
.connect_to_signal("ClientAdded",
614
.connect_to_signal("InterfacesAdded",
630
615
self.add_new_client,
631
dbus_interface=server_interface,
617
= dbus.OBJECT_MANAGER_IFACE,
632
618
byte_arrays=True))
633
619
(self.mandos_serv
634
620
.connect_to_signal("ClientNotFound",
635
621
self.client_not_found,
636
622
dbus_interface=server_interface,
637
623
byte_arrays=True))
638
for path, client in mandos_clients.iteritems():
624
for path, client in mandos_clients.items():
639
625
client_proxy_object = self.bus.get_object(self.busname,
641
627
self.add_client(MandosClientWidget(server_proxy_object