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