91
86
def __init__(self, proxy_object=None, properties=None, **kwargs):
92
self.proxy = proxy_object # Mandos Client proxy object
87
self.proxy = proxy_object # Mandos Client proxy object
93
88
self.properties = dict() if properties is None else properties
94
89
self.property_changed_match = (
95
self.proxy.connect_to_signal("PropertiesChanged",
96
self.properties_changed,
97
dbus.PROPERTIES_IFACE,
90
self.proxy.connect_to_signal("PropertyChanged",
91
self._property_changed,
100
95
if properties is None:
101
self.properties.update(self.proxy.GetAll(
103
dbus_interface=dbus.PROPERTIES_IFACE))
96
self.properties.update(
97
self.proxy.GetAll(client_interface,
99
= dbus.PROPERTIES_IFACE))
105
101
super(MandosClientPropertyCache, self).__init__(**kwargs)
107
def properties_changed(self, interface, properties, invalidated):
108
"""This is called whenever we get a PropertiesChanged signal
109
It updates the changed properties in the "properties" dict.
103
def _property_changed(self, property, value):
104
"""Helper which takes positional arguments"""
105
return self.property_changed(property=property, value=value)
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.
111
111
# Update properties dict with new value
112
if interface == client_interface:
113
self.properties.update(properties)
112
self.properties[property] = value
115
114
def delete(self):
116
115
self.property_changed_match.remove()
163
162
client_interface,
164
163
byte_arrays=True))
165
self.logger('Created client {}'
166
.format(self.properties["Name"]), level=0)
164
#self.logger('Created client {0}'
165
# .format(self.properties["Name"]))
168
167
def using_timer(self, flag):
169
168
"""Call this method with True or False when timer should be
170
169
activated or deactivated.
172
171
if flag and self._update_timer_callback_tag is None:
173
172
# Will update the shown timer value every second
174
self._update_timer_callback_tag = (GLib.timeout_add
173
self._update_timer_callback_tag = (gobject.timeout_add
176
175
self.update_timer))
177
176
elif not (flag or self._update_timer_callback_tag is None):
178
GLib.source_remove(self._update_timer_callback_tag)
177
gobject.source_remove(self._update_timer_callback_tag)
179
178
self._update_timer_callback_tag = None
181
180
def checker_completed(self, exitstatus, condition, command):
182
181
if exitstatus == 0:
183
self.logger('Checker for client {} (command "{}")'
184
' succeeded'.format(self.properties["Name"],
189
185
if os.WIFEXITED(condition):
190
self.logger('Checker for client {} (command "{}") failed'
186
self.logger('Checker for client {0} (command "{1}")'
187
' failed with exit code {2}'
192
188
.format(self.properties["Name"], command,
193
189
os.WEXITSTATUS(condition)))
194
190
elif os.WIFSIGNALED(condition):
195
self.logger('Checker for client {} (command "{}") was'
196
' killed by signal {}'
191
self.logger('Checker for client {0} (command "{1}") was'
192
' killed by signal {2}'
197
193
.format(self.properties["Name"], command,
198
194
os.WTERMSIG(condition)))
195
elif os.WCOREDUMP(condition):
196
self.logger('Checker for client {0} (command "{1}")'
198
.format(self.properties["Name"], command))
200
self.logger('Checker for client {0} completed'
202
.format(self.properties["Name"]))
201
205
def checker_started(self, command):
202
"""Server signals that a checker started."""
203
self.logger('Client {} started checker "{}"'
204
.format(self.properties["Name"],
206
"""Server signals that a checker started. This could be useful
207
to log in the future. """
208
#self.logger('Client {0} started checker "{1}"'
209
# .format(self.properties["Name"],
207
213
def got_secret(self):
208
self.logger('Client {} received its secret'
214
self.logger('Client {0} received its secret'
209
215
.format(self.properties["Name"]))
211
217
def need_approval(self, timeout, default):
213
message = 'Client {} needs approval within {} seconds'
219
message = 'Client {0} needs approval within {1} seconds'
215
message = 'Client {} will get its secret in {} seconds'
221
message = 'Client {0} will get its secret in {1} seconds'
216
222
self.logger(message.format(self.properties["Name"],
219
225
def rejected(self, reason):
220
self.logger('Client {} was rejected; reason: {}'
226
self.logger('Client {0} was rejected; reason: {1}'
221
227
.format(self.properties["Name"], reason))
223
229
def selectable(self):
224
230
"""Make this a "selectable" widget.
225
231
This overrides the method from urwid.FlowWidget."""
228
234
def rows(self, maxcolrow, focus=False):
229
235
"""How many rows this widget will occupy might depend on
230
236
whether we have focus or not.
231
237
This overrides the method from urwid.FlowWidget"""
232
238
return self.current_widget(focus).rows(maxcolrow, focus=focus)
234
240
def current_widget(self, focus=False):
235
241
if focus or self.opened:
236
242
return self._focus_widget
237
243
return self._widget
239
245
def update(self):
240
246
"Called when what is visible on the screen should be updated."
241
247
# How to add standout mode to a style
242
with_standout = {"normal": "standout",
243
"bold": "bold-standout",
245
"underline-blink-standout",
246
"bold-underline-blink":
247
"bold-underline-blink-standout",
248
with_standout = { "normal": "standout",
249
"bold": "bold-standout",
251
"underline-blink-standout",
252
"bold-underline-blink":
253
"bold-underline-blink-standout",
250
256
# Rebuild focus and non-focus widgets using current properties
252
258
# Base part of a client. Name!
253
259
base = '{name}: '.format(name=self.properties["Name"])
254
260
if not self.properties["Enabled"]:
255
261
message = "DISABLED"
256
262
self.using_timer(False)
257
263
elif self.properties["ApprovalPending"]:
258
timeout = datetime.timedelta(
259
milliseconds=self.properties["ApprovalDelay"])
264
timeout = datetime.timedelta(milliseconds
260
267
last_approval_request = isoformat_to_datetime(
261
268
self.properties["LastApprovalRequest"])
262
269
if last_approval_request is not None:
323
329
if self.delete_hook is not None:
324
330
self.delete_hook(self)
325
331
return super(MandosClientWidget, self).delete(**kwargs)
327
333
def render(self, maxcolrow, focus=False):
328
334
"""Render differently if we have focus.
329
335
This overrides the method from urwid.FlowWidget"""
330
336
return self.current_widget(focus).render(maxcolrow,
333
339
def keypress(self, maxcolrow, key):
335
341
This overrides the method from urwid.FlowWidget"""
337
self.proxy.Set(client_interface, "Enabled",
338
dbus.Boolean(True), ignore_reply=True,
339
dbus_interface=dbus.PROPERTIES_IFACE)
343
self.proxy.Enable(dbus_interface = client_interface,
341
self.proxy.Set(client_interface, "Enabled", False,
343
dbus_interface=dbus.PROPERTIES_IFACE)
346
self.proxy.Disable(dbus_interface = client_interface,
345
349
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
346
dbus_interface=client_interface,
350
dbus_interface = client_interface,
347
351
ignore_reply=True)
349
353
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
350
dbus_interface=client_interface,
354
dbus_interface = client_interface,
351
355
ignore_reply=True)
352
356
elif key == "R" or key == "_" or key == "ctrl k":
353
357
self.server_proxy_object.RemoveClient(self.proxy
355
359
ignore_reply=True)
357
self.proxy.Set(client_interface, "CheckerRunning",
358
dbus.Boolean(True), ignore_reply=True,
359
dbus_interface=dbus.PROPERTIES_IFACE)
361
self.proxy.StartChecker(dbus_interface = client_interface,
361
self.proxy.Set(client_interface, "CheckerRunning",
362
dbus.Boolean(False), ignore_reply=True,
363
dbus_interface=dbus.PROPERTIES_IFACE)
364
self.proxy.StopChecker(dbus_interface = client_interface,
365
self.proxy.CheckedOK(dbus_interface=client_interface,
367
self.proxy.CheckedOK(dbus_interface = client_interface,
366
368
ignore_reply=True)
368
370
# elif key == "p" or key == "=":
377
def properties_changed(self, interface, properties, invalidated):
378
"""Call self.update() if any properties changed.
379
def property_changed(self, property=None, **kwargs):
380
"""Call self.update() if old value is not new value.
379
381
This overrides the method from MandosClientPropertyCache"""
380
old_values = {key: self.properties.get(key)
381
for key in properties.keys()}
382
super(MandosClientWidget, self).properties_changed(
383
interface, properties, invalidated)
384
if any(old_values[key] != self.properties.get(key)
385
for key in old_values):
382
property_name = str(property)
383
old_value = self.properties.get(property_name)
384
super(MandosClientWidget, self).property_changed(
385
property=property, **kwargs)
386
if self.properties.get(property_name) != old_value:
429
428
"bold,underline,blink,standout", "default",
430
429
"bold,underline,blink,standout"),
433
432
if urwid.supports_unicode():
434
self.divider = "─" # \u2500
433
self.divider = "─" # \u2500
434
#self.divider = "━" # \u2501
436
self.divider = "_" # \u005f
436
#self.divider = "-" # \u002d
437
self.divider = "_" # \u005f
438
439
self.screen.start()
440
441
self.size = self.screen.get_cols_rows()
442
443
self.clients = urwid.SimpleListWalker([])
443
444
self.clients_dict = {}
445
446
# We will add Text widgets to this list
447
448
self.max_log_length = max_log_length
449
self.log_level = log_level
451
450
# We keep a reference to the log widget so we can remove it
452
451
# from the ListWalker without it getting destroyed
453
452
self.logbox = ConstrainedListBox(self.log)
455
454
# This keeps track of whether self.uilist currently has
456
455
# self.logbox in it or not
457
456
self.log_visible = True
458
457
self.log_wrap = "any"
461
460
self.log_message_raw(("bold",
462
461
"Mandos Monitor version " + version))
463
462
self.log_message_raw(("bold",
464
463
"q: Quit ?: Help"))
466
465
self.busname = domain + '.Mandos'
467
self.main_loop = GLib.MainLoop()
466
self.main_loop = gobject.MainLoop()
469
468
def client_not_found(self, fingerprint, address):
470
self.log_message("Client with address {} and fingerprint {}"
471
" could not be found"
469
self.log_message("Client with address {0} and fingerprint"
470
" {1} could not be found"
472
471
.format(address, fingerprint))
474
473
def rebuild(self):
475
474
"""This rebuilds the User Interface.
476
475
Call this when the widget layout needs to change"""
478
# self.uilist.append(urwid.ListBox(self.clients))
477
#self.uilist.append(urwid.ListBox(self.clients))
479
478
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
481
# header=urwid.Divider(),
480
#header=urwid.Divider(),
483
footer=urwid.Divider(
484
div_char=self.divider)))
483
urwid.Divider(div_char=
485
485
if self.log_visible:
486
486
self.uilist.append(self.logbox)
487
487
self.topwidget = urwid.Pile(self.uilist)
489
def log_message(self, message, level=1):
489
def log_message(self, message):
490
490
"""Log message formatted with timestamp"""
491
if level < self.log_level:
493
491
timestamp = datetime.datetime.now().isoformat()
494
self.log_message_raw("{}: {}".format(timestamp, message),
497
def log_message_raw(self, markup, level=1):
492
self.log_message_raw(timestamp + ": " + message)
494
def log_message_raw(self, markup):
498
495
"""Add a log message to the log buffer."""
499
if level < self.log_level:
501
496
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
if self.max_log_length:
503
if len(self.log) > self.max_log_length:
504
del self.log[0:len(self.log)-self.max_log_length-1]
497
if (self.max_log_length
498
and len(self.log) > self.max_log_length):
499
del self.log[0:len(self.log)-self.max_log_length-1]
505
500
self.logbox.set_focus(len(self.logbox.body.contents),
506
501
coming_from="above")
509
504
def toggle_log_display(self):
510
505
"""Toggle visibility of the log buffer."""
511
506
self.log_visible = not self.log_visible
513
self.log_message("Log visibility changed to: {}"
514
.format(self.log_visible), level=0)
508
#self.log_message("Log visibility changed to: "
509
# + str(self.log_visible))
516
511
def change_log_display(self):
517
512
"""Change type of log display.
518
513
Currently, this toggles wrapping of text lines."""
522
517
self.log_wrap = "clip"
523
518
for textwidget in self.log:
524
519
textwidget.set_wrap_mode(self.log_wrap)
525
self.log_message("Wrap mode: {}".format(self.log_wrap),
528
def find_and_remove_client(self, path, interfaces):
520
#self.log_message("Wrap mode: " + self.log_wrap)
522
def find_and_remove_client(self, path, name):
529
523
"""Find a client by its object path and remove it.
531
This is connected to the InterfacesRemoved signal from the
525
This is connected to the ClientRemoved signal from the
532
526
Mandos server object."""
533
if client_interface not in interfaces:
534
# Not a Mandos client object; ignore
537
528
client = self.clients_dict[path]
540
self.log_message("Unknown client {!r} removed"
531
self.log_message("Unknown client {0!r} ({1!r}) removed"
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
536
def add_new_client(self, path):
554
537
client_proxy_object = self.bus.get_object(self.busname, path)
555
self.add_client(MandosClientWidget(
556
server_proxy_object=self.mandos_serv,
557
proxy_object=client_proxy_object,
558
update_hook=self.refresh,
559
delete_hook=self.remove_client,
560
logger=self.log_message,
561
properties=dict(ifs_and_props[client_interface])),
538
self.add_client(MandosClientWidget(server_proxy_object
541
=client_proxy_object,
564
550
def add_client(self, client, path=None):
565
551
self.clients.append(client)
568
554
self.clients_dict[path] = client
569
555
self.clients.sort(key=lambda c: c.properties["Name"])
572
558
def remove_client(self, client, path=None):
573
559
self.clients.remove(client)
575
561
path = client.proxy.object_path
576
562
del self.clients_dict[path]
579
565
def refresh(self):
580
566
"""Redraw the screen"""
581
567
canvas = self.topwidget.render(self.size, focus=True)
582
568
self.screen.draw_screen(self.size, canvas)
585
571
"""Start the main loop and exit when it's done."""
586
572
self.bus = dbus.SystemBus()
587
573
mandos_dbus_objc = self.bus.get_object(
588
574
self.busname, "/", follow_name_owner_changes=True)
589
self.mandos_serv = dbus.Interface(
590
mandos_dbus_objc, dbus_interface=server_interface)
575
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
592
579
mandos_clients = (self.mandos_serv
593
580
.GetAllClientsWithProperties())
594
581
if not mandos_clients:
595
self.log_message_raw(("bold",
596
"Note: Server has no clients."))
582
self.log_message_raw(("bold", "Note: Server has no clients."))
597
583
except dbus.exceptions.DBusException:
598
self.log_message_raw(("bold",
599
"Note: No Mandos server running."))
584
self.log_message_raw(("bold", "Note: No Mandos server running."))
600
585
mandos_clients = dbus.Dictionary()
602
587
(self.mandos_serv
603
.connect_to_signal("InterfacesRemoved",
588
.connect_to_signal("ClientRemoved",
604
589
self.find_and_remove_client,
605
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
590
dbus_interface=server_interface,
606
591
byte_arrays=True))
607
592
(self.mandos_serv
608
.connect_to_signal("InterfacesAdded",
593
.connect_to_signal("ClientAdded",
609
594
self.add_new_client,
610
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
595
dbus_interface=server_interface,
611
596
byte_arrays=True))
612
597
(self.mandos_serv
613
598
.connect_to_signal("ClientNotFound",
617
602
for path, client in mandos_clients.items():
618
603
client_proxy_object = self.bus.get_object(self.busname,
620
self.add_client(MandosClientWidget(
621
server_proxy_object=self.mandos_serv,
622
proxy_object=client_proxy_object,
624
update_hook=self.refresh,
625
delete_hook=self.remove_client,
626
logger=self.log_message),
605
self.add_client(MandosClientWidget(server_proxy_object
608
=client_proxy_object,
630
self._input_callback_tag = (GLib.io_add_watch
619
self._input_callback_tag = (gobject.io_add_watch
631
620
(sys.stdin.fileno(),
633
622
self.process_input))
634
623
self.main_loop.run()
635
624
# Main loop has finished, we should close everything now
636
GLib.source_remove(self._input_callback_tag)
625
gobject.source_remove(self._input_callback_tag)
637
626
self.screen.stop()
640
629
self.main_loop.quit()
642
631
def process_input(self, source, condition):
643
632
keys = self.screen.get_input()
644
translations = {"ctrl n": "down", # Emacs
645
"ctrl p": "up", # Emacs
646
"ctrl v": "page down", # Emacs
647
"meta v": "page up", # Emacs
648
" ": "page down", # less
649
"f": "page down", # less
650
"b": "page up", # less
633
translations = { "ctrl n": "down", # Emacs
634
"ctrl p": "up", # Emacs
635
"ctrl v": "page down", # Emacs
636
"meta v": "page up", # Emacs
637
" ": "page down", # less
638
"f": "page down", # less
639
"b": "page up", # less
656
645
key = translations[key]
657
646
except KeyError: # :-)
660
649
if key == "q" or key == "Q":
663
652
elif key == "window resize":
664
653
self.size = self.screen.get_cols_rows()
666
elif key == "ctrl l":
655
elif key == "\f": # Ctrl-L
669
657
elif key == "l" or key == "D":
670
658
self.toggle_log_display()
704
690
self.topwidget.set_focus(self.logbox)
707
if self.log_level == 0:
709
self.log_message("Verbose mode: Off")
712
self.log_message("Verbose mode: On")
713
# elif (key == "end" or key == "meta >" or key == "G"
715
# pass # xxx end-of-buffer
716
# elif (key == "home" or key == "meta <" or key == "g"
718
# pass # xxx beginning-of-buffer
719
# elif key == "ctrl e" or key == "$":
720
# pass # xxx move-end-of-line
721
# elif key == "ctrl a" or key == "^":
722
# pass # xxx move-beginning-of-line
723
# elif key == "ctrl b" or key == "meta (" or key == "h":
725
# elif key == "ctrl f" or key == "meta )" or key == "l":
728
# pass # scroll up log
730
# pass # scroll down log
692
#elif (key == "end" or key == "meta >" or key == "G"
694
# pass # xxx end-of-buffer
695
#elif (key == "home" or key == "meta <" or key == "g"
697
# pass # xxx beginning-of-buffer
698
#elif key == "ctrl e" or key == "$":
699
# pass # xxx move-end-of-line
700
#elif key == "ctrl a" or key == "^":
701
# pass # xxx move-beginning-of-line
702
#elif key == "ctrl b" or key == "meta (" or key == "h":
704
#elif key == "ctrl f" or key == "meta )" or key == "l":
707
# pass # scroll up log
709
# pass # scroll down log
731
710
elif self.topwidget.selectable():
732
711
self.topwidget.keypress(self.size, key)
737
715
ui = UserInterface()