To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/trunk
« back to all changes in this revision
Viewing changes to mandos-monitor
- browse files at revision 1316
- compare with another revision
- download diff
- download tarball
- view history from revision 1316
- Committer: Teddy Hogeborn
- Date: 2024-11-24 22:19:53 UTC
- mfrom: (237.4.144 release)
- Revision ID: teddy@recompile.se-20241124221953-qu9unqqbgi456wrx
Merge from release branch
- files added:
- bugs.xml
- debian/tests
- dracut-module
- plugin-helpers
- files removed:
- initramfs-tools-hook-conf
- files modified:
- .bzrignore
added
removed
mandos-monitor
1
#!/usr/bin/python
1
#!/usr/bin/python3 -bbI
2
2
# -*- mode: python; coding: utf-8 -*-
3
#
3
#
4
4
# Mandos Monitor - Control and monitor the Mandos server
5
#
6
# Copyright © 2009-2014 Teddy Hogeborn
7
# Copyright © 2009-2014 Björn Påhlsson
8
#
9
# This program is free software: you can redistribute it and/or modify
10
# it under the terms of the GNU General Public License as published by
5
#
6
# Copyright © 2009-2019 Teddy Hogeborn
7
# Copyright © 2009-2019 Björn Påhlsson
8
#
9
# This file is part of Mandos.
10
#
11
# Mandos is free software: you can redistribute it and/or modify it
12
# under the terms of the GNU General Public License as published by
11
13
# the Free Software Foundation, either version 3 of the License, or
12
14
# (at your option) any later version.
13
15
#
14
# This program is distributed in the hope that it will be useful,
15
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16
# Mandos is distributed in the hope that it will be useful, but
17
# WITHOUT ANY WARRANTY; without even the implied warranty of
16
18
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
19
# GNU General Public License for more details.
18
#
20
#
19
21
# You should have received a copy of the GNU General Public License
20
# along with this program. If not, see
21
# <http://www.gnu.org/licenses/>.
22
#
22
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
23
#
23
24
# Contact the authors at <mandos@recompile.se>.
24
#
25
25
#
26
26
from __future__ import (division, absolute_import, print_function,
27
27
unicode_literals)
28
28
29
try:
29
30
from future_builtins import *
30
31
except ImportError:
31
32
pass
32
33
33
34
import sys
35
import logging
34
36
import os
35
37
import warnings
36
38
import datetime
39
import locale
37
40
38
41
import urwid.curses_display
39
42
import urwid
40
43
41
44
from dbus.mainloop.glib import DBusGMainLoop
42
try:
43
import gobject
44
except ImportError:
45
from gi.repository import GObject as gobject
45
from gi.repository import GLib
46
46
47
47
import dbus
48
48
49
import locale
50
51
49
if sys.version_info.major == 2:
50
__metaclass__ = type
52
51
str = unicode
53
54
locale.setlocale(locale.LC_ALL, '')
55
56
import logging
57
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
52
input = raw_input
53
54
# Show warnings by default
55
if not sys.warnoptions:
56
warnings.simplefilter("default")
57
58
log = logging.getLogger(os.path.basename(sys.argv[0]))
59
logging.basicConfig(level="NOTSET", # Show all messages
60
format="%(message)s") # Show basic log messages
61
62
logging.captureWarnings(True) # Show warnings via the logging system
63
64
locale.setlocale(locale.LC_ALL, "")
65
66
logging.getLogger("dbus.proxies").setLevel(logging.CRITICAL)
67
logging.getLogger("urwid").setLevel(logging.INFO)
58
68
59
69
# Some useful constants
60
domain = 'se.recompile'
61
server_interface = domain + '.Mandos'
62
client_interface = domain + '.Mandos.Client'
63
version = "1.6.9"
70
domain = "se.recompile"
71
server_interface = domain + ".Mandos"
72
client_interface = domain + ".Mandos.Client"
73
version = "1.8.18"
74
75
try:
76
dbus.OBJECT_MANAGER_IFACE
77
except AttributeError:
78
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
79
64
80
65
81
def isoformat_to_datetime(iso):
66
82
"Parse an ISO 8601 date string to a datetime.datetime()"
75
91
int(day),
76
92
int(hour),
77
93
int(minute),
78
int(second), # Whole seconds
79
int(fraction*1000000)) # Microseconds
80
81
class MandosClientPropertyCache(object):
94
int(second), # Whole seconds
95
int(fraction*1000000)) # Microseconds
96
97
98
class MandosClientPropertyCache:
82
99
"""This wraps a Mandos Client D-Bus proxy object, caches the
83
100
properties and calls a hook function when any of them are
84
101
changed.
85
102
"""
86
103
def __init__(self, proxy_object=None, properties=None, **kwargs):
87
self.proxy = proxy_object # Mandos Client proxy object
104
self.proxy = proxy_object # Mandos Client proxy object
88
105
self.properties = dict() if properties is None else properties
89
106
self.property_changed_match = (
90
107
self.proxy.connect_to_signal("PropertiesChanged",
91
108
self.properties_changed,
92
109
dbus.PROPERTIES_IFACE,
93
110
byte_arrays=True))
94
111
95
112
if properties is None:
96
self.properties.update(
97
self.proxy.GetAll(client_interface,
98
dbus_interface
99
= dbus.PROPERTIES_IFACE))
100
113
self.properties.update(self.proxy.GetAll(
114
client_interface,
115
dbus_interface=dbus.PROPERTIES_IFACE))
116
101
117
super(MandosClientPropertyCache, self).__init__(**kwargs)
102
118
103
119
def properties_changed(self, interface, properties, invalidated):
104
120
"""This is called whenever we get a PropertiesChanged signal
105
121
It updates the changed properties in the "properties" dict.
106
122
"""
107
123
# Update properties dict with new value
108
self.properties.update(properties)
109
124
if interface == client_interface:
125
self.properties.update(properties)
126
110
127
def delete(self):
111
128
self.property_changed_match.remove()
112
129
113
130
114
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
131
class MandosClientWidget(MandosClientPropertyCache, urwid.Widget):
115
132
"""A Mandos Client which is visible on the screen.
116
133
"""
117
134
135
_sizing = frozenset(["flow"])
136
118
137
def __init__(self, server_proxy_object=None, update_hook=None,
119
delete_hook=None, logger=None, **kwargs):
138
delete_hook=None, **kwargs):
120
139
# Called on update
121
140
self.update_hook = update_hook
122
141
# Called on delete
123
142
self.delete_hook = delete_hook
124
143
# Mandos Server proxy object
125
144
self.server_proxy_object = server_proxy_object
126
# Logger
127
self.logger = logger
128
145
129
146
self._update_timer_callback_tag = None
130
147
131
148
# The widget shown normally
132
149
self._text_widget = urwid.Text("")
133
150
# The widget shown when we have focus
135
152
super(MandosClientWidget, self).__init__(**kwargs)
136
153
self.update()
137
154
self.opened = False
138
155
139
156
self.match_objects = (
140
157
self.proxy.connect_to_signal("CheckerCompleted",
141
158
self.checker_completed,
157
174
self.rejected,
158
175
client_interface,
159
176
byte_arrays=True))
160
self.logger('Created client {}'
161
.format(self.properties["Name"]), level=0)
162
177
log.debug("Created client %s", self.properties["Name"])
178
163
179
def using_timer(self, flag):
164
180
"""Call this method with True or False when timer should be
165
181
activated or deactivated.
166
182
"""
167
183
if flag and self._update_timer_callback_tag is None:
168
184
# Will update the shown timer value every second
169
self._update_timer_callback_tag = (gobject.timeout_add
170
(1000,
171
self.update_timer))
185
self._update_timer_callback_tag = (
186
GLib.timeout_add(1000,
187
glib_safely(self.update_timer)))
172
188
elif not (flag or self._update_timer_callback_tag is None):
173
gobject.source_remove(self._update_timer_callback_tag)
189
GLib.source_remove(self._update_timer_callback_tag)
174
190
self._update_timer_callback_tag = None
175
191
176
192
def checker_completed(self, exitstatus, condition, command):
177
193
if exitstatus == 0:
178
self.logger('Checker for client {} (command "{}")'
179
' succeeded'.format(self.properties["Name"],
180
command), level=0)
194
log.debug('Checker for client %s (command "%s")'
195
" succeeded", self.properties["Name"], command)
181
196
self.update()
182
197
return
183
198
# Checker failed
184
199
if os.WIFEXITED(condition):
185
self.logger('Checker for client {} (command "{}") failed'
186
' with exit code {}'
187
.format(self.properties["Name"], command,
188
os.WEXITSTATUS(condition)))
200
log.info('Checker for client %s (command "%s") failed'
201
" with exit code %d", self.properties["Name"],
202
command, os.WEXITSTATUS(condition))
189
203
elif os.WIFSIGNALED(condition):
190
self.logger('Checker for client {} (command "{}") was'
191
' killed by signal {}'
192
.format(self.properties["Name"], command,
193
os.WTERMSIG(condition)))
194
elif os.WCOREDUMP(condition):
195
self.logger('Checker for client {} (command "{}") dumped'
196
' core'.format(self.properties["Name"],
197
command))
198
else:
199
self.logger('Checker for client {} completed'
200
' mysteriously'
201
.format(self.properties["Name"]))
204
log.info('Checker for client %s (command "%s") was'
205
" killed by signal %d", self.properties["Name"],
206
command, os.WTERMSIG(condition))
202
207
self.update()
203
208
204
209
def checker_started(self, command):
205
210
"""Server signals that a checker started."""
206
self.logger('Client {} started checker "{}"'
207
.format(self.properties["Name"],
208
command), level=0)
209
211
log.debug('Client %s started checker "%s"',
212
self.properties["Name"], command)
213
210
214
def got_secret(self):
211
self.logger('Client {} received its secret'
212
.format(self.properties["Name"]))
213
215
log.info("Client %s received its secret",
216
self.properties["Name"])
217
214
218
def need_approval(self, timeout, default):
215
219
if not default:
216
message = 'Client {} needs approval within {} seconds'
220
message = "Client %s needs approval within %f seconds"
217
221
else:
218
message = 'Client {} will get its secret in {} seconds'
219
self.logger(message.format(self.properties["Name"],
220
timeout/1000))
221
222
message = "Client %s will get its secret in %f seconds"
223
log.info(message, self.properties["Name"], timeout/1000)
224
222
225
def rejected(self, reason):
223
self.logger('Client {} was rejected; reason: {}'
224
.format(self.properties["Name"], reason))
225
226
log.info("Client %s was rejected; reason: %s",
227
self.properties["Name"], reason)
228
226
229
def selectable(self):
227
230
"""Make this a "selectable" widget.
228
This overrides the method from urwid.FlowWidget."""
231
This overrides the method from urwid.Widget."""
229
232
return True
230
233
231
234
def rows(self, maxcolrow, focus=False):
232
235
"""How many rows this widget will occupy might depend on
233
236
whether we have focus or not.
234
This overrides the method from urwid.FlowWidget"""
237
This overrides the method from urwid.Widget"""
235
238
return self.current_widget(focus).rows(maxcolrow, focus=focus)
236
239
237
240
def current_widget(self, focus=False):
238
241
if focus or self.opened:
239
242
return self._focus_widget
240
243
return self._widget
241
244
242
245
def update(self):
243
246
"Called when what is visible on the screen should be updated."
244
247
# How to add standout mode to a style
245
with_standout = { "normal": "standout",
246
"bold": "bold-standout",
247
"underline-blink":
248
"underline-blink-standout",
249
"bold-underline-blink":
250
"bold-underline-blink-standout",
251
}
252
248
with_standout = {"normal": "standout",
249
"bold": "bold-standout",
250
"underline-blink":
251
"underline-blink-standout",
252
"bold-underline-blink":
253
"bold-underline-blink-standout",
254
}
255
253
256
# Rebuild focus and non-focus widgets using current properties
254
257
255
258
# Base part of a client. Name!
256
base = '{name}: '.format(name=self.properties["Name"])
259
base = "{name}: ".format(name=self.properties["Name"])
257
260
if not self.properties["Enabled"]:
258
261
message = "DISABLED"
259
262
self.using_timer(False)
260
263
elif self.properties["ApprovalPending"]:
261
timeout = datetime.timedelta(milliseconds
262
= self.properties
263
["ApprovalDelay"])
264
timeout = datetime.timedelta(
265
milliseconds=self.properties["ApprovalDelay"])
264
266
last_approval_request = isoformat_to_datetime(
265
267
self.properties["LastApprovalRequest"])
266
268
if last_approval_request is not None:
282
284
timer = datetime.timedelta(0)
283
285
else:
284
286
expires = (datetime.datetime.strptime
285
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
287
(expires, "%Y-%m-%dT%H:%M:%S.%f"))
286
288
timer = max(expires - datetime.datetime.utcnow(),
287
289
datetime.timedelta())
288
message = ('A checker has failed! Time until client'
289
' gets disabled: {}'
290
message = ("A checker has failed! Time until client"
291
" gets disabled: {}"
290
292
.format(str(timer).rsplit(".", 1)[0]))
291
293
self.using_timer(True)
292
294
else:
293
295
message = "enabled"
294
296
self.using_timer(False)
295
297
self._text = "{}{}".format(base, message)
296
298
297
299
if not urwid.supports_unicode():
298
300
self._text = self._text.encode("ascii", "replace")
299
301
textlist = [("normal", self._text)]
309
311
# Run update hook, if any
310
312
if self.update_hook is not None:
311
313
self.update_hook()
312
314
313
315
def update_timer(self):
314
"""called by gobject. Will indefinitely loop until
315
gobject.source_remove() on tag is called"""
316
"""called by GLib. Will indefinitely loop until
317
GLib.source_remove() on tag is called
318
"""
316
319
self.update()
317
320
return True # Keep calling this
318
321
319
322
def delete(self, **kwargs):
320
323
if self._update_timer_callback_tag is not None:
321
gobject.source_remove(self._update_timer_callback_tag)
324
GLib.source_remove(self._update_timer_callback_tag)
322
325
self._update_timer_callback_tag = None
323
326
for match in self.match_objects:
324
327
match.remove()
326
329
if self.delete_hook is not None:
327
330
self.delete_hook(self)
328
331
return super(MandosClientWidget, self).delete(**kwargs)
329
332
330
333
def render(self, maxcolrow, focus=False):
331
334
"""Render differently if we have focus.
332
This overrides the method from urwid.FlowWidget"""
335
This overrides the method from urwid.Widget"""
333
336
return self.current_widget(focus).render(maxcolrow,
334
337
focus=focus)
335
338
336
339
def keypress(self, maxcolrow, key):
337
340
"""Handle keys.
338
This overrides the method from urwid.FlowWidget"""
341
This overrides the method from urwid.Widget"""
339
342
if key == "+":
340
self.proxy.Enable(dbus_interface = client_interface,
341
ignore_reply=True)
343
self.proxy.Set(client_interface, "Enabled",
344
dbus.Boolean(True), ignore_reply=True,
345
dbus_interface=dbus.PROPERTIES_IFACE)
342
346
elif key == "-":
343
self.proxy.Disable(dbus_interface = client_interface,
344
ignore_reply=True)
347
self.proxy.Set(client_interface, "Enabled", False,
348
ignore_reply=True,
349
dbus_interface=dbus.PROPERTIES_IFACE)
345
350
elif key == "a":
346
351
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
347
dbus_interface = client_interface,
352
dbus_interface=client_interface,
348
353
ignore_reply=True)
349
354
elif key == "d":
350
355
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
351
dbus_interface = client_interface,
356
dbus_interface=client_interface,
352
357
ignore_reply=True)
353
358
elif key == "R" or key == "_" or key == "ctrl k":
354
359
self.server_proxy_object.RemoveClient(self.proxy
355
360
.object_path,
356
361
ignore_reply=True)
357
362
elif key == "s":
358
self.proxy.StartChecker(dbus_interface = client_interface,
359
ignore_reply=True)
363
self.proxy.Set(client_interface, "CheckerRunning",
364
dbus.Boolean(True), ignore_reply=True,
365
dbus_interface=dbus.PROPERTIES_IFACE)
360
366
elif key == "S":
361
self.proxy.StopChecker(dbus_interface = client_interface,
362
ignore_reply=True)
367
self.proxy.Set(client_interface, "CheckerRunning",
368
dbus.Boolean(False), ignore_reply=True,
369
dbus_interface=dbus.PROPERTIES_IFACE)
363
370
elif key == "C":
364
self.proxy.CheckedOK(dbus_interface = client_interface,
371
self.proxy.CheckedOK(dbus_interface=client_interface,
365
372
ignore_reply=True)
366
373
# xxx
367
374
# elif key == "p" or key == "=":
372
379
# self.open()
373
380
else:
374
381
return key
375
382
376
383
def properties_changed(self, interface, properties, invalidated):
377
384
"""Call self.update() if any properties changed.
378
385
This overrides the method from MandosClientPropertyCache"""
379
old_values = { key: self.properties.get(key)
380
for key in properties.keys() }
386
old_values = {key: self.properties.get(key)
387
for key in properties.keys()}
381
388
super(MandosClientWidget, self).properties_changed(
382
389
interface, properties, invalidated)
383
390
if any(old_values[key] != self.properties.get(key)
385
392
self.update()
386
393
387
394
395
def glib_safely(func, retval=True):
396
def safe_func(*args, **kwargs):
397
try:
398
return func(*args, **kwargs)
399
except Exception:
400
log.exception("")
401
return retval
402
return safe_func
403
404
388
405
class ConstrainedListBox(urwid.ListBox):
389
406
"""Like a normal urwid.ListBox, but will consume all "up" or
390
407
"down" key presses, thus not allowing any containing widgets to
391
408
use them as an excuse to shift focus away from this widget.
392
409
"""
393
410
def keypress(self, *args, **kwargs):
394
ret = super(ConstrainedListBox, self).keypress(*args, **kwargs)
411
ret = (super(ConstrainedListBox, self)
412
.keypress(*args, **kwargs))
395
413
if ret in ("up", "down"):
396
414
return
397
415
return ret
398
416
399
417
400
class UserInterface(object):
418
class UserInterface:
401
419
"""This is the entire user interface - the whole screen
402
420
with boxes, lists of client widgets, etc.
403
421
"""
404
def __init__(self, max_log_length=1000, log_level=1):
422
def __init__(self, max_log_length=1000):
405
423
DBusGMainLoop(set_as_default=True)
406
424
407
425
self.screen = urwid.curses_display.Screen()
408
426
409
427
self.screen.register_palette((
410
428
("normal",
411
429
"default", "default", None),
416
434
("standout",
417
435
"standout", "default", "standout"),
418
436
("bold-underline-blink",
419
"bold,underline,blink", "default", "bold,underline,blink"),
437
"bold,underline,blink", "default",
438
"bold,underline,blink"),
420
439
("bold-standout",
421
440
"bold,standout", "default", "bold,standout"),
422
441
("underline-blink-standout",
426
445
"bold,underline,blink,standout", "default",
427
446
"bold,underline,blink,standout"),
428
447
))
429
448
430
449
if urwid.supports_unicode():
431
self.divider = "─" # \u2500
432
#self.divider = "━" # \u2501
450
self.divider = "─" # \u2500
433
451
else:
434
#self.divider = "-" # \u002d
435
self.divider = "_" # \u005f
436
452
self.divider = "_" # \u005f
453
437
454
self.screen.start()
438
455
439
456
self.size = self.screen.get_cols_rows()
440
457
441
458
self.clients = urwid.SimpleListWalker([])
442
459
self.clients_dict = {}
443
460
444
461
# We will add Text widgets to this list
445
self.log = []
462
self.log = urwid.SimpleListWalker([])
446
463
self.max_log_length = max_log_length
447
448
self.log_level = log_level
449
464
450
465
# We keep a reference to the log widget so we can remove it
451
466
# from the ListWalker without it getting destroyed
452
467
self.logbox = ConstrainedListBox(self.log)
453
468
454
469
# This keeps track of whether self.uilist currently has
455
470
# self.logbox in it or not
456
471
self.log_visible = True
457
472
self.log_wrap = "any"
458
473
474
self.loghandler = UILogHandler(self)
475
459
476
self.rebuild()
460
self.log_message_raw(("bold",
461
"Mandos Monitor version " + version))
462
self.log_message_raw(("bold",
463
"q: Quit ?: Help"))
464
465
self.busname = domain + '.Mandos'
466
self.main_loop = gobject.MainLoop()
467
468
def client_not_found(self, fingerprint, address):
469
self.log_message("Client with address {} and fingerprint {}"
470
" could not be found"
471
.format(address, fingerprint))
472
477
self.add_log_line(("bold",
478
"Mandos Monitor version " + version))
479
self.add_log_line(("bold", "q: Quit ?: Help"))
480
481
self.busname = domain + ".Mandos"
482
self.main_loop = GLib.MainLoop()
483
484
def client_not_found(self, key_id, address):
485
log.info("Client with address %s and key ID %s could"
486
" not be found", address, key_id)
487
473
488
def rebuild(self):
474
489
"""This rebuilds the User Interface.
475
490
Call this when the widget layout needs to change"""
476
491
self.uilist = []
477
#self.uilist.append(urwid.ListBox(self.clients))
492
# self.uilist.append(urwid.ListBox(self.clients))
478
493
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
479
494
clients),
480
#header=urwid.Divider(),
495
# header=urwid.Divider(),
481
496
header=None,
482
footer=
483
urwid.Divider(div_char=
484
self.divider)))
497
footer=urwid.Divider(
498
div_char=self.divider)))
485
499
if self.log_visible:
486
500
self.uilist.append(self.logbox)
487
501
self.topwidget = urwid.Pile(self.uilist)
488
489
def log_message(self, message, level=1):
490
"""Log message formatted with timestamp"""
491
if level < self.log_level:
492
return
493
timestamp = datetime.datetime.now().isoformat()
494
self.log_message_raw("{}: {}".format(timestamp, message),
495
level=level)
496
497
def log_message_raw(self, markup, level=1):
498
"""Add a log message to the log buffer."""
499
if level < self.log_level:
500
return
502
503
def add_log_line(self, markup):
501
504
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
if (self.max_log_length
503
and len(self.log) > self.max_log_length):
504
del self.log[0:len(self.log)-self.max_log_length-1]
505
self.logbox.set_focus(len(self.logbox.body.contents),
505
if self.max_log_length:
506
if len(self.log) > self.max_log_length:
507
del self.log[0:(len(self.log) - self.max_log_length)]
508
self.logbox.set_focus(len(self.logbox.body.contents)-1,
506
509
coming_from="above")
507
510
self.refresh()
508
511
509
512
def toggle_log_display(self):
510
513
"""Toggle visibility of the log buffer."""
511
514
self.log_visible = not self.log_visible
512
515
self.rebuild()
513
self.log_message("Log visibility changed to: {}"
514
.format(self.log_visible), level=0)
515
516
log.debug("Log visibility changed to: %s", self.log_visible)
517
516
518
def change_log_display(self):
517
519
"""Change type of log display.
518
520
Currently, this toggles wrapping of text lines."""
522
524
self.log_wrap = "clip"
523
525
for textwidget in self.log:
524
526
textwidget.set_wrap_mode(self.log_wrap)
525
self.log_message("Wrap mode: {}".format(self.log_wrap),
526
level=0)
527
528
def find_and_remove_client(self, path, name):
527
log.debug("Wrap mode: %s", self.log_wrap)
528
529
def find_and_remove_client(self, path, interfaces):
529
530
"""Find a client by its object path and remove it.
530
531
This is connected to the ClientRemoved signal from the
531
532
This is connected to the InterfacesRemoved signal from the
532
533
Mandos server object."""
534
if client_interface not in interfaces:
535
# Not a Mandos client object; ignore
536
return
533
537
try:
534
538
client = self.clients_dict[path]
535
539
except KeyError:
536
540
# not found?
537
self.log_message("Unknown client {!r} ({!r}) removed"
538
.format(name, path))
541
log.warning("Unknown client %s removed", path)
539
542
return
540
543
client.delete()
541
542
def add_new_client(self, path):
544
545
def add_new_client(self, path, ifs_and_props):
546
"""Find a client by its object path and remove it.
547
548
This is connected to the InterfacesAdded signal from the
549
Mandos server object.
550
"""
551
if client_interface not in ifs_and_props:
552
# Not a Mandos client object; ignore
553
return
543
554
client_proxy_object = self.bus.get_object(self.busname, path)
544
self.add_client(MandosClientWidget(server_proxy_object
545
=self.mandos_serv,
546
proxy_object
547
=client_proxy_object,
548
update_hook
549
=self.refresh,
550
delete_hook
551
=self.remove_client,
552
logger
553
=self.log_message),
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
properties=dict(ifs_and_props[client_interface])),
554
561
path=path)
555
562
556
563
def add_client(self, client, path=None):
557
564
self.clients.append(client)
558
565
if path is None:
560
567
self.clients_dict[path] = client
561
568
self.clients.sort(key=lambda c: c.properties["Name"])
562
569
self.refresh()
563
570
564
571
def remove_client(self, client, path=None):
565
572
self.clients.remove(client)
566
573
if path is None:
567
574
path = client.proxy.object_path
568
575
del self.clients_dict[path]
569
576
self.refresh()
570
577
571
578
def refresh(self):
572
579
"""Redraw the screen"""
573
580
canvas = self.topwidget.render(self.size, focus=True)
574
581
self.screen.draw_screen(self.size, canvas)
575
582
576
583
def run(self):
577
584
"""Start the main loop and exit when it's done."""
585
log.addHandler(self.loghandler)
586
self.orig_log_propagate = log.propagate
587
log.propagate = False
588
self.orig_log_level = log.level
589
log.setLevel("INFO")
578
590
self.bus = dbus.SystemBus()
579
591
mandos_dbus_objc = self.bus.get_object(
580
592
self.busname, "/", follow_name_owner_changes=True)
581
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
582
dbus_interface
583
= server_interface)
593
self.mandos_serv = dbus.Interface(
594
mandos_dbus_objc, dbus_interface=server_interface)
584
595
try:
585
596
mandos_clients = (self.mandos_serv
586
597
.GetAllClientsWithProperties())
587
598
if not mandos_clients:
588
self.log_message_raw(("bold", "Note: Server has no clients."))
599
log.warning("Note: Server has no clients.")
589
600
except dbus.exceptions.DBusException:
590
self.log_message_raw(("bold", "Note: No Mandos server running."))
601
log.warning("Note: No Mandos server running.")
591
602
mandos_clients = dbus.Dictionary()
592
603
593
604
(self.mandos_serv
594
.connect_to_signal("ClientRemoved",
605
.connect_to_signal("InterfacesRemoved",
595
606
self.find_and_remove_client,
596
dbus_interface=server_interface,
607
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
597
608
byte_arrays=True))
598
609
(self.mandos_serv
599
.connect_to_signal("ClientAdded",
610
.connect_to_signal("InterfacesAdded",
600
611
self.add_new_client,
601
dbus_interface=server_interface,
612
dbus_interface=dbus.OBJECT_MANAGER_IFACE,
602
613
byte_arrays=True))
603
614
(self.mandos_serv
604
615
.connect_to_signal("ClientNotFound",
608
619
for path, client in mandos_clients.items():
609
620
client_proxy_object = self.bus.get_object(self.busname,
610
621
path)
611
self.add_client(MandosClientWidget(server_proxy_object
612
=self.mandos_serv,
613
proxy_object
614
=client_proxy_object,
615
properties=client,
616
update_hook
617
=self.refresh,
618
delete_hook
619
=self.remove_client,
620
logger
621
=self.log_message),
622
self.add_client(MandosClientWidget(
623
server_proxy_object=self.mandos_serv,
624
proxy_object=client_proxy_object,
625
properties=client,
626
update_hook=self.refresh,
627
delete_hook=self.remove_client),
622
628
path=path)
623
629
624
630
self.refresh()
625
self._input_callback_tag = (gobject.io_add_watch
626
(sys.stdin.fileno(),
627
gobject.IO_IN,
628
self.process_input))
631
self._input_callback_tag = (
632
GLib.io_add_watch(
633
GLib.IOChannel.unix_new(sys.stdin.fileno()),
634
GLib.PRIORITY_DEFAULT, GLib.IO_IN,
635
glib_safely(self.process_input)))
629
636
self.main_loop.run()
630
637
# Main loop has finished, we should close everything now
631
gobject.source_remove(self._input_callback_tag)
632
self.screen.stop()
633
638
GLib.source_remove(self._input_callback_tag)
639
with warnings.catch_warnings():
640
warnings.simplefilter("ignore", BytesWarning)
641
self.screen.stop()
642
634
643
def stop(self):
635
644
self.main_loop.quit()
636
645
log.removeHandler(self.loghandler)
646
log.propagate = self.orig_log_propagate
647
637
648
def process_input(self, source, condition):
638
649
keys = self.screen.get_input()
639
translations = { "ctrl n": "down", # Emacs
640
"ctrl p": "up", # Emacs
641
"ctrl v": "page down", # Emacs
642
"meta v": "page up", # Emacs
643
" ": "page down", # less
644
"f": "page down", # less
645
"b": "page up", # less
646
"j": "down", # vi
647
"k": "up", # vi
648
}
650
translations = {"ctrl n": "down", # Emacs
651
"ctrl p": "up", # Emacs
652
"ctrl v": "page down", # Emacs
653
"meta v": "page up", # Emacs
654
" ": "page down", # less
655
"f": "page down", # less
656
"b": "page up", # less
657
"j": "down", # vi
658
"k": "up", # vi
659
}
649
660
for key in keys:
650
661
try:
651
662
key = translations[key]
652
663
except KeyError: # :-)
653
664
pass
654
665
655
666
if key == "q" or key == "Q":
656
667
self.stop()
657
668
break
671
682
if not self.log_visible:
672
683
self.log_visible = True
673
684
self.rebuild()
674
self.log_message_raw(("bold",
675
" ".
676
join(("q: Quit",
677
"?: Help",
678
"l: Log window toggle",
679
"TAB: Switch window",
680
"w: Wrap (log lines)",
681
"v: Toggle verbose log",
682
))))
683
self.log_message_raw(("bold",
684
" "
685
.join(("Clients:",
686
"+: Enable",
687
"-: Disable",
688
"R: Remove",
689
"s: Start new checker",
690
"S: Stop checker",
691
"C: Checker OK",
692
"a: Approve",
693
"d: Deny"))))
685
self.add_log_line(("bold",
686
" ".join(("q: Quit",
687
"?: Help",
688
"l: Log window toggle",
689
"TAB: Switch window",
690
"w: Wrap (log lines)",
691
"v: Toggle verbose log",
692
))))
693
self.add_log_line(("bold",
694
" ".join(("Clients:",
695
"+: Enable",
696
"-: Disable",
697
"R: Remove",
698
"s: Start new checker",
699
"S: Stop checker",
700
"C: Checker OK",
701
"a: Approve",
702
"d: Deny",
703
))))
694
704
self.refresh()
695
705
elif key == "tab":
696
706
if self.topwidget.get_focus() is self.logbox:
699
709
self.topwidget.set_focus(self.logbox)
700
710
self.refresh()
701
711
elif key == "v":
702
if self.log_level == 0:
703
self.log_level = 1
704
self.log_message("Verbose mode: Off")
712
if log.level < logging.INFO:
713
log.setLevel(logging.INFO)
714
log.info("Verbose mode: Off")
705
715
else:
706
self.log_level = 0
707
self.log_message("Verbose mode: On")
708
#elif (key == "end" or key == "meta >" or key == "G"
709
# or key == ">"):
710
# pass # xxx end-of-buffer
711
#elif (key == "home" or key == "meta <" or key == "g"
712
# or key == "<"):
713
# pass # xxx beginning-of-buffer
714
#elif key == "ctrl e" or key == "$":
715
# pass # xxx move-end-of-line
716
#elif key == "ctrl a" or key == "^":
717
# pass # xxx move-beginning-of-line
718
#elif key == "ctrl b" or key == "meta (" or key == "h":
719
# pass # xxx left
720
#elif key == "ctrl f" or key == "meta )" or key == "l":
721
# pass # xxx right
722
#elif key == "a":
723
# pass # scroll up log
724
#elif key == "z":
725
# pass # scroll down log
716
log.setLevel(logging.NOTSET)
717
log.info("Verbose mode: On")
718
# elif (key == "end" or key == "meta >" or key == "G"
719
# or key == ">"):
720
# pass # xxx end-of-buffer
721
# elif (key == "home" or key == "meta <" or key == "g"
722
# or key == "<"):
723
# pass # xxx beginning-of-buffer
724
# elif key == "ctrl e" or key == "$":
725
# pass # xxx move-end-of-line
726
# elif key == "ctrl a" or key == "^":
727
# pass # xxx move-beginning-of-line
728
# elif key == "ctrl b" or key == "meta (" or key == "h":
729
# pass # xxx left
730
# elif key == "ctrl f" or key == "meta )" or key == "l":
731
# pass # xxx right
732
# elif key == "a":
733
# pass # scroll up log
734
# elif key == "z":
735
# pass # scroll down log
726
736
elif self.topwidget.selectable():
727
737
self.topwidget.keypress(self.size, key)
728
738
self.refresh()
729
739
return True
730
740
741
742
class UILogHandler(logging.Handler):
743
def __init__(self, ui, *args, **kwargs):
744
self.ui = ui
745
super(UILogHandler, self).__init__(*args, **kwargs)
746
self.setFormatter(
747
logging.Formatter("%(asctime)s: %(message)s"))
748
def emit(self, record):
749
msg = self.format(record)
750
if record.levelno > logging.INFO:
751
msg = ("bold", msg)
752
self.ui.add_log_line(msg)
753
754
731
755
ui = UserInterface()
732
756
try:
733
757
ui.run()
734
758
except KeyboardInterrupt:
735
ui.screen.stop()
736
except Exception as e:
737
ui.log_message(str(e))
738
ui.screen.stop()
759
with warnings.catch_warnings():
760
warnings.filterwarnings("ignore", "", BytesWarning)
761
ui.screen.stop()
762
except Exception:
763
with warnings.catch_warnings():
764
warnings.filterwarnings("ignore", "", BytesWarning)
765
ui.screen.stop()
739
766
raise
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0