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