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 405
- compare with another revision
- download diff
- download tarball
- view history from revision 405
- Committer: Teddy Hogeborn
- Date: 2009-11-09 05:10:35 UTC
- Revision ID: teddy@fukt.bsnet.se-20091109051035-zcmvuhioo9gjs0cb
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
D-Bus types to native
Python types.
(MandosClientPropertyCache.type_map): Removed attribute.
(MandosClientPropertyCache.conversion): Removed method.
(ConstrainedListBox): New; a ListBox which will keep the focus.
(UserInterface.__init__): Take new argument "max_log_length".
(UserInterface.divider, UserInterface.log,
UserInterface.max_log_length, UserInterface.logbox,
UserInterface.log_visible, UserInterface.log_wrap): New attributes.
(UserInterface.rebuild, UserInterface.log_message,
UserInterface.toggle_log_display, UserInterface.change_log_display):
New methods.
(UserInterface.remove_client): Bug fix: Eliminate double client
listing when an empty list gets a
ClientAdded signal.
(UserInterface.process_input): Add more key translations.
D-Bus types to native
Python types.
(MandosClientPropertyCache.type_map): Removed attribute.
(MandosClientPropertyCache.conversion): Removed method.
(ConstrainedListBox): New; a ListBox which will keep the focus.
(UserInterface.__init__): Take new argument "max_log_length".
(UserInterface.divider, UserInterface.log,
UserInterface.max_log_length, UserInterface.logbox,
UserInterface.log_visible, UserInterface.log_wrap): New attributes.
(UserInterface.rebuild, UserInterface.log_message,
UserInterface.toggle_log_display, UserInterface.change_log_display):
New methods.
(UserInterface.remove_client): Bug fix: Eliminate double client
listing when an empty list gets a
ClientAdded signal.
(UserInterface.process_input): Add more key translations.
- files removed:
- DBUS-API
- debian/source
- debian/upstream
- network-hooks.d
- files modified:
- .bzrignore
added
removed
mandos-monitor
1
1
#!/usr/bin/python
2
2
# -*- mode: python; coding: utf-8 -*-
3
#
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
11
# the Free Software Foundation, either version 3 of the License, or
12
# (at your option) any later version.
13
#
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
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
# GNU General Public License for more details.
18
#
19
# 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
#
23
# Contact the authors at <mandos@recompile.se>.
24
#
25
3
26
from __future__ import (division, absolute_import, print_function,
27
unicode_literals)
28
try:
29
from future_builtins import *
30
except ImportError:
31
pass
4
from __future__ import division, absolute_import, with_statement
32
5
33
6
import sys
34
import os
35
36
import datetime
7
import signal
37
8
38
9
import urwid.curses_display
39
10
import urwid
40
11
41
12
from dbus.mainloop.glib import DBusGMainLoop
42
try:
43
import gobject
44
except ImportError:
45
from gi.repository import GObject as gobject
13
import gobject
46
14
47
15
import dbus
48
16
49
import locale
50
51
if sys.version_info.major == 2:
52
str = unicode
53
54
locale.setlocale(locale.LC_ALL, '')
55
56
import logging
57
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
17
import UserList
58
18
59
19
# Some useful constants
60
domain = 'se.recompile'
20
domain = 'se.bsnet.fukt'
61
21
server_interface = domain + '.Mandos'
62
22
client_interface = domain + '.Mandos.Client'
63
version = "1.6.9"
64
65
def isoformat_to_datetime(iso):
66
"Parse an ISO 8601 date string to a datetime.datetime()"
67
if not iso:
68
return None
69
d, t = iso.split("T", 1)
70
year, month, day = d.split("-", 2)
71
hour, minute, second = t.split(":", 2)
72
second, fraction = divmod(float(second), 1)
73
return datetime.datetime(int(year),
74
int(month),
75
int(day),
76
int(hour),
77
int(minute),
78
int(second), # Whole seconds
79
int(fraction*1000000)) # Microseconds
23
version = "1.0.14"
24
25
# Always run in monochrome mode
26
urwid.curses_display.curses.has_colors = lambda : False
27
28
# Urwid doesn't support blinking, but we want it. Since we have no
29
# use for underline on its own, we make underline also always blink.
30
urwid.curses_display.curses.A_UNDERLINE |= (
31
urwid.curses_display.curses.A_BLINK)
80
32
81
33
class MandosClientPropertyCache(object):
82
34
"""This wraps a Mandos Client D-Bus proxy object, caches the
83
35
properties and calls a hook function when any of them are
84
36
changed.
85
37
"""
86
def __init__(self, proxy_object=None, properties=None, **kwargs):
38
def __init__(self, proxy_object=None, properties=None, *args,
39
**kwargs):
87
40
self.proxy = proxy_object # Mandos Client proxy object
88
self.properties = dict() if properties is None else properties
89
self.property_changed_match = (
90
self.proxy.connect_to_signal("PropertiesChanged",
91
self.properties_changed,
92
dbus.PROPERTIES_IFACE,
93
byte_arrays=True))
94
95
if properties is None:
96
self.properties.update(
97
self.proxy.GetAll(client_interface,
98
dbus_interface
99
= dbus.PROPERTIES_IFACE))
100
101
super(MandosClientPropertyCache, self).__init__(**kwargs)
41
42
if properties is None:
43
self.properties = dict()
44
else:
45
self.properties = properties
46
self.proxy.connect_to_signal("PropertyChanged",
47
self.property_changed,
48
client_interface,
49
byte_arrays=True)
50
51
if properties is None:
52
self.properties.update(self.proxy.GetAll(client_interface,
53
dbus_interface =
54
dbus.PROPERTIES_IFACE))
55
super(MandosClientPropertyCache, self).__init__(
56
proxy_object=proxy_object,
57
properties=properties, *args, **kwargs)
102
58
103
def properties_changed(self, interface, properties, invalidated):
104
"""This is called whenever we get a PropertiesChanged signal
105
It updates the changed properties in the "properties" dict.
59
def property_changed(self, property=None, value=None):
60
"""This is called whenever we get a PropertyChanged signal
61
It updates the changed property in the "properties" dict.
106
62
"""
107
63
# Update properties dict with new value
108
self.properties.update(properties)
109
110
def delete(self):
111
self.property_changed_match.remove()
64
self.properties[property] = value
112
65
113
66
114
67
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
116
69
"""
117
70
118
71
def __init__(self, server_proxy_object=None, update_hook=None,
119
delete_hook=None, logger=None, **kwargs):
72
delete_hook=None, *args, **kwargs):
120
73
# Called on update
121
74
self.update_hook = update_hook
122
75
# Called on delete
123
76
self.delete_hook = delete_hook
124
77
# Mandos Server proxy object
125
78
self.server_proxy_object = server_proxy_object
126
# Logger
127
self.logger = logger
128
129
self._update_timer_callback_tag = None
130
79
131
80
# The widget shown normally
132
81
self._text_widget = urwid.Text("")
133
82
# The widget shown when we have focus
134
83
self._focus_text_widget = urwid.Text("")
135
super(MandosClientWidget, self).__init__(**kwargs)
84
super(MandosClientWidget, self).__init__(
85
update_hook=update_hook, delete_hook=delete_hook,
86
*args, **kwargs)
136
87
self.update()
137
88
self.opened = False
138
139
self.match_objects = (
140
self.proxy.connect_to_signal("CheckerCompleted",
141
self.checker_completed,
142
client_interface,
143
byte_arrays=True),
144
self.proxy.connect_to_signal("CheckerStarted",
145
self.checker_started,
146
client_interface,
147
byte_arrays=True),
148
self.proxy.connect_to_signal("GotSecret",
149
self.got_secret,
150
client_interface,
151
byte_arrays=True),
152
self.proxy.connect_to_signal("NeedApproval",
153
self.need_approval,
154
client_interface,
155
byte_arrays=True),
156
self.proxy.connect_to_signal("Rejected",
157
self.rejected,
158
client_interface,
159
byte_arrays=True))
160
self.logger('Created client {}'
161
.format(self.properties["Name"]), level=0)
162
163
def using_timer(self, flag):
164
"""Call this method with True or False when timer should be
165
activated or deactivated.
166
"""
167
if flag and self._update_timer_callback_tag is None:
168
# Will update the shown timer value every second
169
self._update_timer_callback_tag = (gobject.timeout_add
170
(1000,
171
self.update_timer))
172
elif not (flag or self._update_timer_callback_tag is None):
173
gobject.source_remove(self._update_timer_callback_tag)
174
self._update_timer_callback_tag = None
175
176
def checker_completed(self, exitstatus, condition, command):
177
if exitstatus == 0:
178
self.logger('Checker for client {} (command "{}")'
179
' succeeded'.format(self.properties["Name"],
180
command), level=0)
181
self.update()
182
return
183
# Checker failed
184
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)))
189
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"]))
202
self.update()
203
204
def checker_started(self, command):
205
"""Server signals that a checker started."""
206
self.logger('Client {} started checker "{}"'
207
.format(self.properties["Name"],
208
command), level=0)
209
210
def got_secret(self):
211
self.logger('Client {} received its secret'
212
.format(self.properties["Name"]))
213
214
def need_approval(self, timeout, default):
215
if not default:
216
message = 'Client {} needs approval within {} seconds'
217
else:
218
message = 'Client {} will get its secret in {} seconds'
219
self.logger(message.format(self.properties["Name"],
220
timeout/1000))
221
222
def rejected(self, reason):
223
self.logger('Client {} was rejected; reason: {}'
224
.format(self.properties["Name"], reason))
225
89
226
90
def selectable(self):
227
91
"""Make this a "selectable" widget.
228
92
This overrides the method from urwid.FlowWidget."""
229
93
return True
230
94
231
def rows(self, maxcolrow, focus=False):
95
def rows(self, (maxcol,), focus=False):
232
96
"""How many rows this widget will occupy might depend on
233
97
whether we have focus or not.
234
98
This overrides the method from urwid.FlowWidget"""
235
return self.current_widget(focus).rows(maxcolrow, focus=focus)
99
return self.current_widget(focus).rows((maxcol,), focus=focus)
236
100
237
101
def current_widget(self, focus=False):
238
102
if focus or self.opened:
242
106
def update(self):
243
107
"Called when what is visible on the screen should be updated."
244
108
# 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",
109
with_standout = { u"normal": u"standout",
110
u"bold": u"bold-standout",
111
u"underline-blink":
112
u"underline-blink-standout",
113
u"bold-underline-blink":
114
u"bold-underline-blink-standout",
251
115
}
252
116
253
117
# Rebuild focus and non-focus widgets using current properties
254
255
# Base part of a client. Name!
256
base = '{name}: '.format(name=self.properties["Name"])
257
if not self.properties["Enabled"]:
258
message = "DISABLED"
259
self.using_timer(False)
260
elif self.properties["ApprovalPending"]:
261
timeout = datetime.timedelta(milliseconds
262
= self.properties
263
["ApprovalDelay"])
264
last_approval_request = isoformat_to_datetime(
265
self.properties["LastApprovalRequest"])
266
if last_approval_request is not None:
267
timer = max(timeout - (datetime.datetime.utcnow()
268
- last_approval_request),
269
datetime.timedelta())
270
else:
271
timer = datetime.timedelta()
272
if self.properties["ApprovedByDefault"]:
273
message = "Approval in {}. (d)eny?"
274
else:
275
message = "Denial in {}. (a)pprove?"
276
message = message.format(str(timer).rsplit(".", 1)[0])
277
self.using_timer(True)
278
elif self.properties["LastCheckerStatus"] != 0:
279
# When checker has failed, show timer until client expires
280
expires = self.properties["Expires"]
281
if expires == "":
282
timer = datetime.timedelta(0)
283
else:
284
expires = (datetime.datetime.strptime
285
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
286
timer = max(expires - datetime.datetime.utcnow(),
287
datetime.timedelta())
288
message = ('A checker has failed! Time until client'
289
' gets disabled: {}'
290
.format(str(timer).rsplit(".", 1)[0]))
291
self.using_timer(True)
292
else:
293
message = "enabled"
294
self.using_timer(False)
295
self._text = "{}{}".format(base, message)
296
118
self._text = (u'name="%(name)s", enabled=%(enabled)s'
119
% self.properties)
297
120
if not urwid.supports_unicode():
298
121
self._text = self._text.encode("ascii", "replace")
299
textlist = [("normal", self._text)]
122
textlist = [(u"normal", u"BLARGH: "), (u"bold", self._text)]
300
123
self._text_widget.set_text(textlist)
301
124
self._focus_text_widget.set_text([(with_standout[text[0]],
302
125
text[1])
310
133
if self.update_hook is not None:
311
134
self.update_hook()
312
135
313
def update_timer(self):
314
"""called by gobject. Will indefinitely loop until
315
gobject.source_remove() on tag is called"""
316
self.update()
317
return True # Keep calling this
318
319
def delete(self, **kwargs):
320
if self._update_timer_callback_tag is not None:
321
gobject.source_remove(self._update_timer_callback_tag)
322
self._update_timer_callback_tag = None
323
for match in self.match_objects:
324
match.remove()
325
self.match_objects = ()
136
def delete(self):
326
137
if self.delete_hook is not None:
327
138
self.delete_hook(self)
328
return super(MandosClientWidget, self).delete(**kwargs)
329
139
330
def render(self, maxcolrow, focus=False):
140
def render(self, (maxcol,), focus=False):
331
141
"""Render differently if we have focus.
332
142
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render(maxcolrow,
143
return self.current_widget(focus).render((maxcol,),
334
144
focus=focus)
335
145
336
def keypress(self, maxcolrow, key):
146
def keypress(self, (maxcol,), key):
337
147
"""Handle keys.
338
148
This overrides the method from urwid.FlowWidget"""
339
if key == "+":
340
self.proxy.Enable(dbus_interface = client_interface,
341
ignore_reply=True)
342
elif key == "-":
343
self.proxy.Disable(dbus_interface = client_interface,
344
ignore_reply=True)
345
elif key == "a":
346
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
347
dbus_interface = client_interface,
348
ignore_reply=True)
349
elif key == "d":
350
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
351
dbus_interface = client_interface,
352
ignore_reply=True)
353
elif key == "R" or key == "_" or key == "ctrl k":
149
if key == u"e" or key == u"+":
150
self.proxy.Enable()
151
elif key == u"d" or key == u"-":
152
self.proxy.Disable()
153
elif key == u"r" or key == u"_":
354
154
self.server_proxy_object.RemoveClient(self.proxy
355
.object_path,
356
ignore_reply=True)
357
elif key == "s":
358
self.proxy.StartChecker(dbus_interface = client_interface,
359
ignore_reply=True)
360
elif key == "S":
361
self.proxy.StopChecker(dbus_interface = client_interface,
362
ignore_reply=True)
363
elif key == "C":
364
self.proxy.CheckedOK(dbus_interface = client_interface,
365
ignore_reply=True)
155
.object_path)
156
elif key == u"s":
157
self.proxy.StartChecker()
158
elif key == u"c":
159
self.proxy.StopChecker()
160
elif key == u"S":
161
self.proxy.CheckedOK()
366
162
# xxx
367
# elif key == "p" or key == "=":
163
# elif key == u"p" or key == "=":
368
164
# self.proxy.pause()
369
# elif key == "u" or key == ":":
165
# elif key == u"u" or key == ":":
370
166
# self.proxy.unpause()
371
# elif key == "RET":
167
# elif key == u"RET":
372
168
# self.open()
373
169
else:
374
170
return key
375
171
376
def properties_changed(self, interface, properties, invalidated):
377
"""Call self.update() if any properties changed.
172
def property_changed(self, property=None, value=None,
173
*args, **kwargs):
174
"""Call self.update() if old value is not new value.
378
175
This overrides the method from MandosClientPropertyCache"""
379
old_values = { key: self.properties.get(key)
380
for key in properties.keys() }
381
super(MandosClientWidget, self).properties_changed(
382
interface, properties, invalidated)
383
if any(old_values[key] != self.properties.get(key)
384
for key in old_values):
176
property_name = unicode(property)
177
old_value = self.properties.get(property_name)
178
super(MandosClientWidget, self).property_changed(
179
property=property, value=value, *args, **kwargs)
180
if self.properties.get(property_name) != old_value:
385
181
self.update()
386
182
387
183
390
186
"down" key presses, thus not allowing any containing widgets to
391
187
use them as an excuse to shift focus away from this widget.
392
188
"""
393
def keypress(self, *args, **kwargs):
394
ret = super(ConstrainedListBox, self).keypress(*args, **kwargs)
395
if ret in ("up", "down"):
189
def keypress(self, (maxcol, maxrow), key):
190
ret = super(ConstrainedListBox, self).keypress((maxcol, maxrow), key)
191
if ret in (u"up", u"down"):
396
192
return
397
193
return ret
398
194
401
197
"""This is the entire user interface - the whole screen
402
198
with boxes, lists of client widgets, etc.
403
199
"""
404
def __init__(self, max_log_length=1000, log_level=1):
200
def __init__(self, max_log_length=1000):
405
201
DBusGMainLoop(set_as_default=True)
406
202
407
203
self.screen = urwid.curses_display.Screen()
408
204
409
205
self.screen.register_palette((
410
("normal",
411
"default", "default", None),
412
("bold",
413
"bold", "default", "bold"),
414
("underline-blink",
415
"underline,blink", "default", "underline,blink"),
416
("standout",
417
"standout", "default", "standout"),
418
("bold-underline-blink",
419
"bold,underline,blink", "default", "bold,underline,blink"),
420
("bold-standout",
421
"bold,standout", "default", "bold,standout"),
422
("underline-blink-standout",
423
"underline,blink,standout", "default",
424
"underline,blink,standout"),
425
("bold-underline-blink-standout",
426
"bold,underline,blink,standout", "default",
427
"bold,underline,blink,standout"),
206
(u"normal",
207
u"default", u"default", None),
208
(u"bold",
209
u"default", u"default", u"bold"),
210
(u"underline-blink",
211
u"default", u"default", u"underline"),
212
(u"standout",
213
u"default", u"default", u"standout"),
214
(u"bold-underline-blink",
215
u"default", u"default", (u"bold", u"underline")),
216
(u"bold-standout",
217
u"default", u"default", (u"bold", u"standout")),
218
(u"underline-blink-standout",
219
u"default", u"default", (u"underline", u"standout")),
220
(u"bold-underline-blink-standout",
221
u"default", u"default", (u"bold", u"underline",
222
u"standout")),
428
223
))
429
224
430
225
if urwid.supports_unicode():
431
self.divider = "─" # \u2500
432
#self.divider = "━" # \u2501
226
#self.divider = u"─" # \u2500
227
self.divider = u"━" # \u2501
433
228
else:
434
#self.divider = "-" # \u002d
435
self.divider = "_" # \u005f
229
#self.divider = u"-" # \u002d
230
self.divider = u"_" # \u005f
436
231
437
232
self.screen.start()
438
233
445
240
self.log = []
446
241
self.max_log_length = max_log_length
447
242
448
self.log_level = log_level
449
450
243
# We keep a reference to the log widget so we can remove it
451
244
# from the ListWalker without it getting destroyed
452
245
self.logbox = ConstrainedListBox(self.log)
454
247
# This keeps track of whether self.uilist currently has
455
248
# self.logbox in it or not
456
249
self.log_visible = True
457
self.log_wrap = "any"
250
self.log_wrap = u"any"
458
251
459
252
self.rebuild()
460
self.log_message_raw(("bold",
461
"Mandos Monitor version " + version))
462
self.log_message_raw(("bold",
463
"q: Quit ?: Help"))
253
self.log_message(u"Message")
254
self.log_message(u"Message0 Message1 Message2 Message3 Message4 Message5 Message6 Message7 Message8 Message9")
255
self.log_message(u"Message10 Message11 Message12 Message13 Message14 Message15 Message16 Message17 Message18 Message19")
256
self.log_message(u"Message20 Message21 Message22 Message23 Message24 Message25 Message26 Message27 Message28 Message29")
464
257
465
258
self.busname = domain + '.Mandos'
466
259
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))
260
self.bus = dbus.SystemBus()
261
mandos_dbus_objc = self.bus.get_object(
262
self.busname, u"/", follow_name_owner_changes=True)
263
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
264
dbus_interface
265
= server_interface)
266
try:
267
mandos_clients = (self.mandos_serv
268
.GetAllClientsWithProperties())
269
except dbus.exceptions.DBusException:
270
mandos_clients = dbus.Dictionary()
271
272
(self.mandos_serv
273
.connect_to_signal("ClientRemoved",
274
self.find_and_remove_client,
275
dbus_interface=server_interface,
276
byte_arrays=True))
277
(self.mandos_serv
278
.connect_to_signal("ClientAdded",
279
self.add_new_client,
280
dbus_interface=server_interface,
281
byte_arrays=True))
282
for path, client in mandos_clients.iteritems():
283
client_proxy_object = self.bus.get_object(self.busname,
284
path)
285
self.add_client(MandosClientWidget(server_proxy_object
286
=self.mandos_serv,
287
proxy_object
288
=client_proxy_object,
289
properties=client,
290
update_hook
291
=self.refresh,
292
delete_hook
293
=self.remove_client),
294
path=path)
472
295
473
296
def rebuild(self):
474
297
"""This rebuilds the User Interface.
475
298
Call this when the widget layout needs to change"""
476
299
self.uilist = []
477
300
#self.uilist.append(urwid.ListBox(self.clients))
478
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
479
clients),
301
self.uilist.append(urwid.Frame(ConstrainedListBox(self.clients),
480
302
#header=urwid.Divider(),
481
303
header=None,
482
footer=
483
urwid.Divider(div_char=
484
self.divider)))
304
footer=urwid.Divider(div_char=self.divider)))
485
305
if self.log_visible:
486
306
self.uilist.append(self.logbox)
307
pass
487
308
self.topwidget = urwid.Pile(self.uilist)
488
309
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):
310
def log_message(self, markup):
498
311
"""Add a log message to the log buffer."""
499
if level < self.log_level:
500
return
501
312
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
313
if (self.max_log_length
503
314
and len(self.log) > self.max_log_length):
504
315
del self.log[0:len(self.log)-self.max_log_length-1]
505
self.logbox.set_focus(len(self.logbox.body.contents),
506
coming_from="above")
507
self.refresh()
508
316
509
317
def toggle_log_display(self):
510
318
"""Toggle visibility of the log buffer."""
511
319
self.log_visible = not self.log_visible
512
320
self.rebuild()
513
self.log_message("Log visibility changed to: {}"
514
.format(self.log_visible), level=0)
321
self.log_message(u"Log visibility changed to: "
322
+ unicode(self.log_visible))
515
323
516
324
def change_log_display(self):
517
325
"""Change type of log display.
518
326
Currently, this toggles wrapping of text lines."""
519
if self.log_wrap == "clip":
520
self.log_wrap = "any"
327
if self.log_wrap == u"clip":
328
self.log_wrap = u"any"
521
329
else:
522
self.log_wrap = "clip"
330
self.log_wrap = u"clip"
523
331
for textwidget in self.log:
524
332
textwidget.set_wrap_mode(self.log_wrap)
525
self.log_message("Wrap mode: {}".format(self.log_wrap),
526
level=0)
333
self.log_message(u"Wrap mode: " + self.log_wrap)
527
334
528
335
def find_and_remove_client(self, path, name):
529
"""Find a client by its object path and remove it.
336
"""Find an client from its object path and remove it.
530
337
531
338
This is connected to the ClientRemoved signal from the
532
339
Mandos server object."""
534
341
client = self.clients_dict[path]
535
342
except KeyError:
536
343
# not found?
537
self.log_message("Unknown client {!r} ({!r}) removed"
538
.format(name, path))
539
344
return
540
client.delete()
345
self.remove_client(client, path)
541
346
542
def add_new_client(self, path):
347
def add_new_client(self, path, properties):
543
348
client_proxy_object = self.bus.get_object(self.busname, path)
544
349
self.add_client(MandosClientWidget(server_proxy_object
545
350
=self.mandos_serv,
546
351
proxy_object
547
352
=client_proxy_object,
353
properties=properties,
548
354
update_hook
549
355
=self.refresh,
550
356
delete_hook
551
=self.remove_client,
552
logger
553
=self.log_message),
357
=self.remove_client),
554
358
path=path)
555
359
556
360
def add_client(self, client, path=None):
558
362
if path is None:
559
363
path = client.proxy.object_path
560
364
self.clients_dict[path] = client
561
self.clients.sort(key=lambda c: c.properties["Name"])
365
self.clients.sort(None, lambda c: c.properties[u"name"])
562
366
self.refresh()
563
367
564
368
def remove_client(self, client, path=None):
566
370
if path is None:
567
371
path = client.proxy.object_path
568
372
del self.clients_dict[path]
373
if not self.clients_dict:
374
# Work around bug in Urwid 0.9.8.3 - if a SimpleListWalker
375
# is completely emptied, we need to recreate it.
376
self.clients = urwid.SimpleListWalker([])
377
self.rebuild()
569
378
self.refresh()
570
379
571
380
def refresh(self):
575
384
576
385
def run(self):
577
386
"""Start the main loop and exit when it's done."""
578
self.bus = dbus.SystemBus()
579
mandos_dbus_objc = self.bus.get_object(
580
self.busname, "/", follow_name_owner_changes=True)
581
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
582
dbus_interface
583
= server_interface)
584
try:
585
mandos_clients = (self.mandos_serv
586
.GetAllClientsWithProperties())
587
if not mandos_clients:
588
self.log_message_raw(("bold", "Note: Server has no clients."))
589
except dbus.exceptions.DBusException:
590
self.log_message_raw(("bold", "Note: No Mandos server running."))
591
mandos_clients = dbus.Dictionary()
592
593
(self.mandos_serv
594
.connect_to_signal("ClientRemoved",
595
self.find_and_remove_client,
596
dbus_interface=server_interface,
597
byte_arrays=True))
598
(self.mandos_serv
599
.connect_to_signal("ClientAdded",
600
self.add_new_client,
601
dbus_interface=server_interface,
602
byte_arrays=True))
603
(self.mandos_serv
604
.connect_to_signal("ClientNotFound",
605
self.client_not_found,
606
dbus_interface=server_interface,
607
byte_arrays=True))
608
for path, client in mandos_clients.items():
609
client_proxy_object = self.bus.get_object(self.busname,
610
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
path=path)
623
624
387
self.refresh()
625
388
self._input_callback_tag = (gobject.io_add_watch
626
389
(sys.stdin.fileno(),
636
399
637
400
def process_input(self, source, condition):
638
401
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
402
translations = { u"ctrl n": u"down", # Emacs
403
u"ctrl p": u"up", # Emacs
404
u"ctrl v": u"page down", # Emacs
405
u"meta v": u"page up", # Emacs
406
u" ": u"page down", # less
407
u"f": u"page down", # less
408
u"b": u"page up", # less
409
u"j": u"down", # vi
410
u"k": u"up", # vi
648
411
}
649
412
for key in keys:
650
413
try:
652
415
except KeyError: # :-)
653
416
pass
654
417
655
if key == "q" or key == "Q":
418
if key == u"q" or key == u"Q":
656
419
self.stop()
657
420
break
658
elif key == "window resize":
421
elif key == u"window resize":
659
422
self.size = self.screen.get_cols_rows()
660
423
self.refresh()
661
elif key == "ctrl l":
662
self.screen.clear()
424
elif key == u"\f": # Ctrl-L
663
425
self.refresh()
664
elif key == "l" or key == "D":
426
elif key == u"l" or key == u"D":
665
427
self.toggle_log_display()
666
428
self.refresh()
667
elif key == "w" or key == "i":
429
elif key == u"w" or key == u"i":
668
430
self.change_log_display()
669
431
self.refresh()
670
elif key == "?" or key == "f1" or key == "esc":
671
if not self.log_visible:
672
self.log_visible = True
673
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"))))
432
elif key == u"?" or key == u"f1":
433
self.log_message(u"Help!")
694
434
self.refresh()
695
elif key == "tab":
435
elif key == u"tab":
696
436
if self.topwidget.get_focus() is self.logbox:
697
437
self.topwidget.set_focus(0)
698
438
else:
699
439
self.topwidget.set_focus(self.logbox)
700
440
self.refresh()
701
elif key == "v":
702
if self.log_level == 0:
703
self.log_level = 1
704
self.log_message("Verbose mode: Off")
705
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
441
elif (key == u"end" or key == u"meta >" or key == u"G"
442
or key == u">"):
443
pass # xxx end-of-buffer
444
elif (key == u"home" or key == u"meta <" or key == u"g"
445
or key == u"<"):
446
pass # xxx beginning-of-buffer
447
elif key == u"ctrl e" or key == u"$":
448
pass # xxx move-end-of-line
449
elif key == u"ctrl a" or key == u"^":
450
pass # xxx move-beginning-of-line
451
elif key == u"ctrl b" or key == u"meta (" or key == u"h":
452
pass # xxx left
453
elif key == u"ctrl f" or key == u"meta )" or key == u"l":
454
pass # xxx right
455
elif key == u"a":
456
pass # scroll up log
457
elif key == u"z":
458
pass # scroll down log
726
459
elif self.topwidget.selectable():
727
460
self.topwidget.keypress(self.size, key)
728
461
self.refresh()
731
464
ui = UserInterface()
732
465
try:
733
466
ui.run()
734
except KeyboardInterrupt:
735
ui.screen.stop()
736
except Exception as e:
737
ui.log_message(str(e))
467
except:
738
468
ui.screen.stop()
739
469
raise
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0