42
15
from dbus.mainloop.glib import DBusGMainLoop
43
from gi.repository import GLib
24
locale.setlocale(locale.LC_ALL, u'')
51
if sys.version_info.major == 2:
54
locale.setlocale(locale.LC_ALL, '')
56
27
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
58
29
# Some useful constants
59
domain = 'se.recompile'
30
domain = 'se.bsnet.fukt'
60
31
server_interface = domain + '.Mandos'
61
32
client_interface = domain + '.Mandos.Client'
65
dbus.OBJECT_MANAGER_IFACE
66
except AttributeError:
67
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
35
# Always run in monochrome mode
36
urwid.curses_display.curses.has_colors = lambda : False
38
# Urwid doesn't support blinking, but we want it. Since we have no
39
# use for underline on its own, we make underline also always blink.
40
urwid.curses_display.curses.A_UNDERLINE |= (
41
urwid.curses_display.curses.A_BLINK)
70
43
def isoformat_to_datetime(iso):
71
44
"Parse an ISO 8601 date string to a datetime.datetime()"
74
d, t = iso.split("T", 1)
75
year, month, day = d.split("-", 2)
76
hour, minute, second = t.split(":", 2)
47
d, t = iso.split(u"T", 1)
48
year, month, day = d.split(u"-", 2)
49
hour, minute, second = t.split(u":", 2)
77
50
second, fraction = divmod(float(second), 1)
78
51
return datetime.datetime(int(year),
83
int(second), # Whole seconds
84
int(fraction*1000000)) # Microseconds
56
int(second), # Whole seconds
57
int(fraction*1000000)) # Microseconds
87
59
class MandosClientPropertyCache(object):
88
60
"""This wraps a Mandos Client D-Bus proxy object, caches the
89
61
properties and calls a hook function when any of them are
92
def __init__(self, proxy_object=None, properties=None, **kwargs):
93
self.proxy = proxy_object # Mandos Client proxy object
94
self.properties = dict() if properties is None else properties
95
self.property_changed_match = (
96
self.proxy.connect_to_signal("PropertiesChanged",
97
self.properties_changed,
98
dbus.PROPERTIES_IFACE,
101
if properties is None:
102
self.properties.update(self.proxy.GetAll(
104
dbus_interface=dbus.PROPERTIES_IFACE))
106
super(MandosClientPropertyCache, self).__init__(**kwargs)
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.
64
def __init__(self, proxy_object=None, *args, **kwargs):
65
self.proxy = proxy_object # Mandos Client proxy object
67
self.properties = dict()
68
self.proxy.connect_to_signal(u"PropertyChanged",
69
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__(
81
def property_changed(self, property=None, value=None):
82
"""This is called whenever we get a PropertyChanged signal
83
It updates the changed property in the "properties" dict.
112
85
# Update properties dict with new value
113
if interface == client_interface:
114
self.properties.update(properties)
117
self.property_changed_match.remove()
86
self.properties[property] = value
120
89
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
121
90
"""A Mandos Client which is visible on the screen.
124
93
def __init__(self, server_proxy_object=None, update_hook=None,
125
delete_hook=None, logger=None, **kwargs):
94
delete_hook=None, logger=None, *args, **kwargs):
126
95
# Called on update
127
96
self.update_hook = update_hook
128
97
# Called on delete
131
100
self.server_proxy_object = server_proxy_object
133
102
self.logger = logger
135
104
self._update_timer_callback_tag = None
105
self._update_timer_callback_lock = 0
106
self.last_checker_failed = False
137
108
# The widget shown normally
138
self._text_widget = urwid.Text("")
109
self._text_widget = urwid.Text(u"")
139
110
# The widget shown when we have focus
140
self._focus_text_widget = urwid.Text("")
141
super(MandosClientWidget, self).__init__(**kwargs)
111
self._focus_text_widget = urwid.Text(u"")
112
super(MandosClientWidget, self).__init__(
113
update_hook=update_hook, delete_hook=delete_hook,
143
116
self.opened = False
145
self.match_objects = (
146
self.proxy.connect_to_signal("CheckerCompleted",
147
self.checker_completed,
150
self.proxy.connect_to_signal("CheckerStarted",
151
self.checker_started,
154
self.proxy.connect_to_signal("GotSecret",
158
self.proxy.connect_to_signal("NeedApproval",
162
self.proxy.connect_to_signal("Rejected",
166
self.logger('Created client {}'
167
.format(self.properties["Name"]), level=0)
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))
169
163
def using_timer(self, flag):
170
164
"""Call this method with True or False when timer should be
171
165
activated or deactivated.
173
if flag and self._update_timer_callback_tag is None:
174
# Will update the shown timer value every second
175
self._update_timer_callback_tag = (GLib.timeout_add
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
177
175
self.update_timer))
178
elif not (flag or self._update_timer_callback_tag is None):
179
GLib.source_remove(self._update_timer_callback_tag)
176
elif old and self._update_timer_callback_lock == 0:
177
gobject.source_remove(self._update_timer_callback_tag)
180
178
self._update_timer_callback_tag = None
182
180
def checker_completed(self, exitstatus, condition, command):
183
181
if exitstatus == 0:
184
self.logger('Checker for client {} (command "{}")'
185
' succeeded'.format(self.properties["Name"],
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)
190
194
if os.WIFEXITED(condition):
191
self.logger('Checker for client {} (command "{}") failed'
193
.format(self.properties["Name"], command,
194
os.WEXITSTATUS(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)))
195
199
elif os.WIFSIGNALED(condition):
196
self.logger('Checker for client {} (command "{}") was'
197
' killed by signal {}'
198
.format(self.properties["Name"], command,
199
os.WTERMSIG(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'
202
213
def checker_started(self, command):
203
"""Server signals that a checker started."""
204
self.logger('Client {} started checker "{}"'
205
.format(self.properties["Name"],
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
208
218
def got_secret(self):
209
self.logger('Client {} received its secret'
210
.format(self.properties["Name"]))
219
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
212
223
def need_approval(self, timeout, default):
214
message = 'Client {} needs approval within {} seconds'
225
message = u'Client %s needs approval within %s seconds'
216
message = 'Client {} will get its secret in {} seconds'
217
self.logger(message.format(self.properties["Name"],
227
message = u'Client %s will get its secret in %s seconds'
229
% (self.properties[u"Name"], timeout/1000))
230
self.using_timer(True)
220
232
def rejected(self, reason):
221
self.logger('Client {} was rejected; reason: {}'
222
.format(self.properties["Name"], reason))
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
224
236
def selectable(self):
225
237
"""Make this a "selectable" widget.
226
238
This overrides the method from urwid.FlowWidget."""
229
def rows(self, maxcolrow, focus=False):
241
def rows(self, (maxcol,), focus=False):
230
242
"""How many rows this widget will occupy might depend on
231
243
whether we have focus or not.
232
244
This overrides the method from urwid.FlowWidget"""
233
return self.current_widget(focus).rows(maxcolrow, focus=focus)
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
235
247
def current_widget(self, focus=False):
236
248
if focus or self.opened:
237
249
return self._focus_widget
238
250
return self._widget
240
252
def update(self):
241
253
"Called when what is visible on the screen should be updated."
242
254
# How to add standout mode to a style
243
with_standout = {"normal": "standout",
244
"bold": "bold-standout",
246
"underline-blink-standout",
247
"bold-underline-blink":
248
"bold-underline-blink-standout",
255
with_standout = { u"normal": u"standout",
256
u"bold": u"bold-standout",
258
u"underline-blink-standout",
259
u"bold-underline-blink":
260
u"bold-underline-blink-standout",
251
263
# Rebuild focus and non-focus widgets using current properties
253
265
# Base part of a client. Name!
254
base = '{name}: '.format(name=self.properties["Name"])
255
if not self.properties["Enabled"]:
257
self.using_timer(False)
258
elif self.properties["ApprovalPending"]:
259
timeout = datetime.timedelta(
260
milliseconds=self.properties["ApprovalDelay"])
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
261
274
last_approval_request = isoformat_to_datetime(
262
self.properties["LastApprovalRequest"])
275
self.properties[u"LastApprovalRequest"])
263
276
if last_approval_request is not None:
264
timer = max(timeout - (datetime.datetime.utcnow()
265
- last_approval_request),
266
datetime.timedelta())
277
timer = timeout - (datetime.datetime.utcnow()
278
- last_approval_request)
268
280
timer = datetime.timedelta()
269
if self.properties["ApprovedByDefault"]:
270
message = "Approval in {}. (d)eny?"
272
message = "Denial in {}. (a)pprove?"
273
message = message.format(str(timer).rsplit(".", 1)[0])
274
self.using_timer(True)
275
elif self.properties["LastCheckerStatus"] != 0:
276
# When checker has failed, show timer until client expires
277
expires = self.properties["Expires"]
279
timer = datetime.timedelta(0)
281
expires = (datetime.datetime.strptime
282
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
283
timer = max(expires - datetime.datetime.utcnow(),
284
datetime.timedelta())
285
message = ('A checker has failed! Time until client'
287
.format(str(timer).rsplit(".", 1)[0]))
288
self.using_timer(True)
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])
291
self.using_timer(False)
292
self._text = "{}{}".format(base, message)
300
self._text = "%s%s" % (base, message)
294
302
if not urwid.supports_unicode():
295
303
self._text = self._text.encode("ascii", "replace")
296
textlist = [("normal", self._text)]
304
textlist = [(u"normal", self._text)]
297
305
self._text_widget.set_text(textlist)
298
306
self._focus_text_widget.set_text([(with_standout[text[0]],
306
314
# Run update hook, if any
307
315
if self.update_hook is not None:
308
316
self.update_hook()
310
318
def update_timer(self):
311
"""called by GLib. Will indefinitely loop until
312
GLib.source_remove() on tag is called
315
321
return True # Keep calling this
317
def delete(self, **kwargs):
318
324
if self._update_timer_callback_tag is not None:
319
GLib.source_remove(self._update_timer_callback_tag)
325
gobject.source_remove(self._update_timer_callback_tag)
320
326
self._update_timer_callback_tag = None
321
for match in self.match_objects:
323
self.match_objects = ()
324
327
if self.delete_hook is not None:
325
328
self.delete_hook(self)
326
return super(MandosClientWidget, self).delete(**kwargs)
328
def render(self, maxcolrow, focus=False):
330
def render(self, (maxcol,), focus=False):
329
331
"""Render differently if we have focus.
330
332
This overrides the method from urwid.FlowWidget"""
331
return self.current_widget(focus).render(maxcolrow,
333
return self.current_widget(focus).render((maxcol,),
334
def keypress(self, maxcolrow, key):
336
def keypress(self, (maxcol,), key):
336
338
This overrides the method from urwid.FlowWidget"""
338
self.proxy.Set(client_interface, "Enabled",
339
dbus.Boolean(True), ignore_reply=True,
340
dbus_interface=dbus.PROPERTIES_IFACE)
342
self.proxy.Set(client_interface, "Enabled", False,
344
dbus_interface=dbus.PROPERTIES_IFACE)
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
346
344
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
347
dbus_interface=client_interface,
345
dbus_interface = client_interface)
350
347
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
351
dbus_interface=client_interface,
353
elif key == "R" or key == "_" or key == "ctrl k":
348
dbus_interface = client_interface)
349
elif key == u"r" or key == u"_" or key == u"ctrl k":
354
350
self.server_proxy_object.RemoveClient(self.proxy
358
self.proxy.Set(client_interface, "CheckerRunning",
359
dbus.Boolean(True), ignore_reply=True,
360
dbus_interface=dbus.PROPERTIES_IFACE)
362
self.proxy.Set(client_interface, "CheckerRunning",
363
dbus.Boolean(False), ignore_reply=True,
364
dbus_interface=dbus.PROPERTIES_IFACE)
366
self.proxy.CheckedOK(dbus_interface=client_interface,
353
self.proxy.StartChecker(dbus_interface = client_interface)
355
self.proxy.StopChecker(dbus_interface = client_interface)
357
self.proxy.CheckedOK(dbus_interface = client_interface)
369
# elif key == "p" or key == "=":
359
# elif key == u"p" or key == "=":
370
360
# self.proxy.pause()
371
# elif key == "u" or key == ":":
361
# elif key == u"u" or key == ":":
372
362
# self.proxy.unpause()
363
# elif key == u"RET":
378
def properties_changed(self, interface, properties, invalidated):
379
"""Call self.update() if any properties changed.
368
def property_changed(self, property=None, value=None,
370
"""Call self.update() if old value is not new value.
380
371
This overrides the method from MandosClientPropertyCache"""
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):
372
property_name = unicode(property)
373
old_value = self.properties.get(property_name)
374
super(MandosClientWidget, self).property_changed(
375
property=property, value=value, *args, **kwargs)
376
if self.properties.get(property_name) != old_value:
404
394
"""This is the entire user interface - the whole screen
405
395
with boxes, lists of client widgets, etc.
407
def __init__(self, max_log_length=1000, log_level=1):
397
def __init__(self, max_log_length=1000):
408
398
DBusGMainLoop(set_as_default=True)
410
400
self.screen = urwid.curses_display.Screen()
412
402
self.screen.register_palette((
414
"default", "default", None),
416
"bold", "default", "bold"),
418
"underline,blink", "default", "underline,blink"),
420
"standout", "default", "standout"),
421
("bold-underline-blink",
422
"bold,underline,blink", "default",
423
"bold,underline,blink"),
425
"bold,standout", "default", "bold,standout"),
426
("underline-blink-standout",
427
"underline,blink,standout", "default",
428
"underline,blink,standout"),
429
("bold-underline-blink-standout",
430
"bold,underline,blink,standout", "default",
431
"bold,underline,blink,standout"),
404
u"default", u"default", None),
406
u"default", u"default", u"bold"),
408
u"default", u"default", u"underline"),
410
u"default", u"default", u"standout"),
411
(u"bold-underline-blink",
412
u"default", u"default", (u"bold", u"underline")),
414
u"default", u"default", (u"bold", u"standout")),
415
(u"underline-blink-standout",
416
u"default", u"default", (u"underline", u"standout")),
417
(u"bold-underline-blink-standout",
418
u"default", u"default", (u"bold", u"underline",
434
422
if urwid.supports_unicode():
435
self.divider = "─" # \u2500
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
437
self.divider = "_" # \u005f
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
439
429
self.screen.start()
441
431
self.size = self.screen.get_cols_rows()
443
433
self.clients = urwid.SimpleListWalker([])
444
434
self.clients_dict = {}
446
436
# We will add Text widgets to this list
448
438
self.max_log_length = max_log_length
450
self.log_level = log_level
452
440
# We keep a reference to the log widget so we can remove it
453
441
# from the ListWalker without it getting destroyed
454
442
self.logbox = ConstrainedListBox(self.log)
456
444
# This keeps track of whether self.uilist currently has
457
445
# self.logbox in it or not
458
446
self.log_visible = True
459
self.log_wrap = "any"
447
self.log_wrap = u"any"
462
self.log_message_raw(("bold",
463
"Mandos Monitor version " + version))
464
self.log_message_raw(("bold",
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
467
455
self.busname = domain + '.Mandos'
468
self.main_loop = GLib.MainLoop()
456
self.main_loop = gobject.MainLoop()
457
self.bus = dbus.SystemBus()
458
mandos_dbus_objc = self.bus.get_object(
459
self.busname, u"/", follow_name_owner_changes=True)
460
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
464
mandos_clients = (self.mandos_serv
465
.GetAllClientsWithProperties())
466
except dbus.exceptions.DBusException:
467
mandos_clients = dbus.Dictionary()
470
.connect_to_signal(u"ClientRemoved",
471
self.find_and_remove_client,
472
dbus_interface=server_interface,
475
.connect_to_signal(u"ClientAdded",
477
dbus_interface=server_interface,
480
.connect_to_signal(u"ClientNotFound",
481
self.client_not_found,
482
dbus_interface=server_interface,
484
for path, client in mandos_clients.iteritems():
485
client_proxy_object = self.bus.get_object(self.busname,
487
self.add_client(MandosClientWidget(server_proxy_object
490
=client_proxy_object,
470
500
def client_not_found(self, fingerprint, address):
471
self.log_message("Client with address {} and fingerprint {}"
472
" could not be found"
473
.format(address, fingerprint))
501
self.log_message((u"Client with address %s and fingerprint %s"
502
u" could not be found" % (address,
475
505
def rebuild(self):
476
506
"""This rebuilds the User Interface.
477
507
Call this when the widget layout needs to change"""
479
# self.uilist.append(urwid.ListBox(self.clients))
509
#self.uilist.append(urwid.ListBox(self.clients))
480
510
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
482
# header=urwid.Divider(),
512
#header=urwid.Divider(),
484
footer=urwid.Divider(
485
div_char=self.divider)))
515
urwid.Divider(div_char=
486
517
if self.log_visible:
487
518
self.uilist.append(self.logbox)
488
520
self.topwidget = urwid.Pile(self.uilist)
490
def log_message(self, message, level=1):
491
"""Log message formatted with timestamp"""
492
if level < self.log_level:
522
def log_message(self, message):
494
523
timestamp = datetime.datetime.now().isoformat()
495
self.log_message_raw("{}: {}".format(timestamp, message),
498
def log_message_raw(self, markup, level=1):
524
self.log_message_raw(timestamp + u": " + message)
526
def log_message_raw(self, markup):
499
527
"""Add a log message to the log buffer."""
500
if level < self.log_level:
502
528
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
503
if self.max_log_length:
504
if len(self.log) > self.max_log_length:
505
del self.log[0:len(self.log)-self.max_log_length-1]
529
if (self.max_log_length
530
and len(self.log) > self.max_log_length):
531
del self.log[0:len(self.log)-self.max_log_length-1]
506
532
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
510
536
def toggle_log_display(self):
511
537
"""Toggle visibility of the log buffer."""
512
538
self.log_visible = not self.log_visible
514
self.log_message("Log visibility changed to: {}"
515
.format(self.log_visible), level=0)
540
#self.log_message(u"Log visibility changed to: "
541
# + unicode(self.log_visible))
517
543
def change_log_display(self):
518
544
"""Change type of log display.
519
545
Currently, this toggles wrapping of text lines."""
520
if self.log_wrap == "clip":
521
self.log_wrap = "any"
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
523
self.log_wrap = "clip"
549
self.log_wrap = u"clip"
524
550
for textwidget in self.log:
525
551
textwidget.set_wrap_mode(self.log_wrap)
526
self.log_message("Wrap mode: {}".format(self.log_wrap),
529
def find_and_remove_client(self, path, interfaces):
530
"""Find a client by its object path and remove it.
532
This is connected to the InterfacesRemoved signal from the
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
554
def find_and_remove_client(self, path, name):
555
"""Find an client from its object path and remove it.
557
This is connected to the ClientRemoved signal from the
533
558
Mandos server object."""
534
if client_interface not in interfaces:
535
# Not a Mandos client object; ignore
538
560
client = self.clients_dict[path]
541
self.log_message("Unknown client {!r} removed"
546
def add_new_client(self, path, ifs_and_props):
547
"""Find a client by its object path and remove it.
549
This is connected to the InterfacesAdded signal from the
550
Mandos server object.
552
if client_interface not in ifs_and_props:
553
# Not a Mandos client object; ignore
564
self.remove_client(client, path)
566
def add_new_client(self, path):
555
567
client_proxy_object = self.bus.get_object(self.busname, path)
556
self.add_client(MandosClientWidget(
557
server_proxy_object=self.mandos_serv,
558
proxy_object=client_proxy_object,
559
update_hook=self.refresh,
560
delete_hook=self.remove_client,
561
logger=self.log_message,
562
properties=dict(ifs_and_props[client_interface])),
568
self.add_client(MandosClientWidget(server_proxy_object
571
=client_proxy_object,
565
580
def add_client(self, client, path=None):
566
581
self.clients.append(client)
568
583
path = client.proxy.object_path
569
584
self.clients_dict[path] = client
570
self.clients.sort(key=lambda c: c.properties["Name"])
585
self.clients.sort(None, lambda c: c.properties[u"Name"])
573
588
def remove_client(self, client, path=None):
574
589
self.clients.remove(client)
576
591
path = client.proxy.object_path
577
592
del self.clients_dict[path]
593
if not self.clients_dict:
594
# Work around bug in Urwid 0.9.8.3 - if a SimpleListWalker
595
# is completely emptied, we need to recreate it.
596
self.clients = urwid.SimpleListWalker([])
580
600
def refresh(self):
581
601
"""Redraw the screen"""
582
602
canvas = self.topwidget.render(self.size, focus=True)
583
603
self.screen.draw_screen(self.size, canvas)
586
606
"""Start the main loop and exit when it's done."""
587
self.bus = dbus.SystemBus()
588
mandos_dbus_objc = self.bus.get_object(
589
self.busname, "/", follow_name_owner_changes=True)
590
self.mandos_serv = dbus.Interface(
591
mandos_dbus_objc, dbus_interface=server_interface)
593
mandos_clients = (self.mandos_serv
594
.GetAllClientsWithProperties())
595
if not mandos_clients:
596
self.log_message_raw(("bold",
597
"Note: Server has no clients."))
598
except dbus.exceptions.DBusException:
599
self.log_message_raw(("bold",
600
"Note: No Mandos server running."))
601
mandos_clients = dbus.Dictionary()
604
.connect_to_signal("InterfacesRemoved",
605
self.find_and_remove_client,
606
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
609
.connect_to_signal("InterfacesAdded",
611
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
614
.connect_to_signal("ClientNotFound",
615
self.client_not_found,
616
dbus_interface=server_interface,
618
for path, client in mandos_clients.items():
619
client_proxy_object = self.bus.get_object(self.busname,
621
self.add_client(MandosClientWidget(
622
server_proxy_object=self.mandos_serv,
623
proxy_object=client_proxy_object,
625
update_hook=self.refresh,
626
delete_hook=self.remove_client,
627
logger=self.log_message),
631
self._input_callback_tag = (GLib.io_add_watch
608
self._input_callback_tag = (gobject.io_add_watch
632
609
(sys.stdin.fileno(),
634
611
self.process_input))
635
612
self.main_loop.run()
636
613
# Main loop has finished, we should close everything now
637
GLib.source_remove(self._input_callback_tag)
614
gobject.source_remove(self._input_callback_tag)
638
615
self.screen.stop()
641
618
self.main_loop.quit()
643
620
def process_input(self, source, condition):
644
621
keys = self.screen.get_input()
645
translations = {"ctrl n": "down", # Emacs
646
"ctrl p": "up", # Emacs
647
"ctrl v": "page down", # Emacs
648
"meta v": "page up", # Emacs
649
" ": "page down", # less
650
"f": "page down", # less
651
"b": "page up", # less
622
translations = { u"ctrl n": u"down", # Emacs
623
u"ctrl p": u"up", # Emacs
624
u"ctrl v": u"page down", # Emacs
625
u"meta v": u"page up", # Emacs
626
u" ": u"page down", # less
627
u"f": u"page down", # less
628
u"b": u"page up", # less
657
634
key = translations[key]
658
635
except KeyError: # :-)
661
if key == "q" or key == "Q":
638
if key == u"q" or key == u"Q":
664
elif key == "window resize":
641
elif key == u"window resize":
665
642
self.size = self.screen.get_cols_rows()
667
elif key == "ctrl l":
644
elif key == u"\f": # Ctrl-L
670
elif key == "l" or key == "D":
646
elif key == u"l" or key == u"D":
671
647
self.toggle_log_display()
673
elif key == "w" or key == "i":
649
elif key == u"w" or key == u"i":
674
650
self.change_log_display()
676
elif key == "?" or key == "f1" or key == "esc":
652
elif key == u"?" or key == u"f1" or key == u"esc":
677
653
if not self.log_visible:
678
654
self.log_visible = True
680
self.log_message_raw(("bold",
684
"l: Log window toggle",
685
"TAB: Switch window",
686
"w: Wrap (log lines)",
687
"v: Toggle verbose log",
689
self.log_message_raw(("bold",
695
"s: Start new checker",
656
self.log_message_raw((u"bold",
660
u"l: Log window toggle",
661
u"TAB: Switch window",
663
self.log_message_raw((u"bold",
669
u"s: Start new checker",
702
676
if self.topwidget.get_focus() is self.logbox:
703
677
self.topwidget.set_focus(0)
705
679
self.topwidget.set_focus(self.logbox)
708
if self.log_level == 0:
710
self.log_message("Verbose mode: Off")
713
self.log_message("Verbose mode: On")
714
# elif (key == "end" or key == "meta >" or key == "G"
716
# pass # xxx end-of-buffer
717
# elif (key == "home" or key == "meta <" or key == "g"
719
# pass # xxx beginning-of-buffer
720
# elif key == "ctrl e" or key == "$":
721
# pass # xxx move-end-of-line
722
# elif key == "ctrl a" or key == "^":
723
# pass # xxx move-beginning-of-line
724
# elif key == "ctrl b" or key == "meta (" or key == "h":
726
# elif key == "ctrl f" or key == "meta )" or key == "l":
729
# pass # scroll up log
731
# 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
732
699
elif self.topwidget.selectable():
733
700
self.topwidget.keypress(self.size, key)
738
704
ui = UserInterface()
741
707
except KeyboardInterrupt:
743
except Exception as e:
744
ui.log_message(str(e))
710
ui.log_message(unicode(e))