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