To get this branch, use:
bzr branch
http://bzr.recompile.se/loggerhead/mandos/release
« back to all changes in this revision
Viewing changes to mandos-monitor
- browse files at revision 237.2.167
- compare with another revision
- download diff
- download tarball
- view history from revision 237.2.167
- Committer: Teddy Hogeborn
- Date: 2009-11-05 19:16:46 UTC
- mto: (24.1.146 mandos) (237.12.2 mandos-persistent) (299.1.1 release) (300.1.1 release) (301.1.1 release)
- mto: This revision was merged to the branch mainline in revision 270.
- 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.1"
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
self.last_checker_failed = False
135
120
136
121
# The widget shown normally
137
122
self._text_widget = urwid.Text("")
142
127
*args, **kwargs)
143
128
self.update()
144
129
self.opened = False
145
146
last_checked_ok = isoformat_to_datetime(self.properties
147
["LastCheckedOK"])
148
if last_checked_ok is None:
149
self.last_checker_failed = True
150
else:
151
self.last_checker_failed = ((datetime.datetime.utcnow()
152
- last_checked_ok)
153
> datetime.timedelta
154
(milliseconds=
155
self.properties
156
["Interval"]))
157
158
if self.last_checker_failed:
159
self.using_timer(True)
160
161
if self.need_approval:
162
self.using_timer(True)
163
164
self.match_objects = (
165
self.proxy.connect_to_signal("CheckerCompleted",
166
self.checker_completed,
167
client_interface,
168
byte_arrays=True),
169
self.proxy.connect_to_signal("CheckerStarted",
170
self.checker_started,
171
client_interface,
172
byte_arrays=True),
173
self.proxy.connect_to_signal("GotSecret",
174
self.got_secret,
175
client_interface,
176
byte_arrays=True),
177
self.proxy.connect_to_signal("NeedApproval",
178
self.need_approval,
179
client_interface,
180
byte_arrays=True),
181
self.proxy.connect_to_signal("Rejected",
182
self.rejected,
183
client_interface,
184
byte_arrays=True))
185
#self.logger('Created client %s' % (self.properties["Name"]))
186
187
def property_changed(self, property=None, value=None):
188
super(self, MandosClientWidget).property_changed(property,
189
value)
190
if property == "ApprovalPending":
191
using_timer(bool(value))
192
193
def using_timer(self, flag):
194
"""Call this method with True or False when timer should be
195
activated or deactivated.
196
"""
197
old = self._update_timer_callback_lock
198
if flag:
199
self._update_timer_callback_lock += 1
200
else:
201
self._update_timer_callback_lock -= 1
202
if old == 0 and self._update_timer_callback_lock:
203
# Will update the shown timer value every second
204
self._update_timer_callback_tag = (gobject.timeout_add
205
(1000,
206
self.update_timer))
207
elif old and self._update_timer_callback_lock == 0:
208
gobject.source_remove(self._update_timer_callback_tag)
209
self._update_timer_callback_tag = None
210
211
def checker_completed(self, exitstatus, condition, command):
212
if exitstatus == 0:
213
if self.last_checker_failed:
214
self.last_checker_failed = False
215
self.using_timer(False)
216
#self.logger('Checker for client %s (command "%s")'
217
# ' was successful'
218
# % (self.properties["Name"], command))
219
self.update()
220
return
221
# Checker failed
222
if not self.last_checker_failed:
223
self.last_checker_failed = True
224
self.using_timer(True)
225
if os.WIFEXITED(condition):
226
self.logger('Checker for client %s (command "%s")'
227
' failed with exit code %s'
228
% (self.properties["Name"], command,
229
os.WEXITSTATUS(condition)))
230
elif os.WIFSIGNALED(condition):
231
self.logger('Checker for client %s (command "%s")'
232
' was killed by signal %s'
233
% (self.properties["Name"], command,
234
os.WTERMSIG(condition)))
235
elif os.WCOREDUMP(condition):
236
self.logger('Checker for client %s (command "%s")'
237
' dumped core'
238
% (self.properties["Name"], command))
239
else:
240
self.logger('Checker for client %s completed'
241
' mysteriously')
242
self.update()
243
244
def checker_started(self, command):
245
"""Server signals that a checker started. This could be useful
246
to log in the future. """
247
#self.logger('Client %s started checker "%s"'
248
# % (self.properties["Name"], unicode(command)))
249
pass
250
251
def got_secret(self):
252
self.last_checker_failed = False
253
self.logger('Client %s received its secret'
254
% self.properties["Name"])
255
256
def need_approval(self, timeout, default):
257
if not default:
258
message = 'Client %s needs approval within %s seconds'
259
else:
260
message = 'Client %s will get its secret in %s seconds'
261
self.logger(message
262
% (self.properties["Name"], timeout/1000))
263
self.using_timer(True)
264
265
def rejected(self, reason):
266
self.logger('Client %s was rejected; reason: %s'
267
% (self.properties["Name"], reason))
268
130
269
131
def selectable(self):
270
132
"""Make this a "selectable" widget.
271
133
This overrides the method from urwid.FlowWidget."""
272
134
return True
273
135
274
def rows(self, maxcolrow, focus=False):
136
def rows(self, (maxcol,), focus=False):
275
137
"""How many rows this widget will occupy might depend on
276
138
whether we have focus or not.
277
139
This overrides the method from urwid.FlowWidget"""
278
return self.current_widget(focus).rows(maxcolrow, focus=focus)
140
return self.current_widget(focus).rows((maxcol,), focus=focus)
279
141
280
142
def current_widget(self, focus=False):
281
143
if focus or self.opened:
285
147
def update(self):
286
148
"Called when what is visible on the screen should be updated."
287
149
# How to add standout mode to a style
288
with_standout = { "normal": "standout",
289
"bold": "bold-standout",
290
"underline-blink":
291
"underline-blink-standout",
292
"bold-underline-blink":
293
"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",
294
156
}
295
157
296
158
# Rebuild focus and non-focus widgets using current properties
297
298
# Base part of a client. Name!
299
base = ('%(name)s: '
300
% {"name": self.properties["Name"]})
301
if not self.properties["Enabled"]:
302
message = "DISABLED"
303
elif self.properties["ApprovalPending"]:
304
timeout = datetime.timedelta(milliseconds
305
= self.properties
306
["ApprovalDelay"])
307
last_approval_request = isoformat_to_datetime(
308
self.properties["LastApprovalRequest"])
309
if last_approval_request is not None:
310
timer = timeout - (datetime.datetime.utcnow()
311
- last_approval_request)
312
else:
313
timer = datetime.timedelta()
314
if self.properties["ApprovedByDefault"]:
315
message = "Approval in %s. (d)eny?"
316
else:
317
message = "Denial in %s. (a)pprove?"
318
message = message % unicode(timer).rsplit(".", 1)[0]
319
elif self.last_checker_failed:
320
# When checker has failed, print a timer until client expires
321
expires = self.properties["Expires"]
322
if expires == "":
323
timer = datetime.timedelta(0)
324
else:
325
expires = datetime.datetime.strptime(expires,
326
'%Y-%m-%dT%H:%M:%S.%f')
327
timer = expires - datetime.datetime.utcnow()
328
message = ('A checker has failed! Time until client'
329
' gets disabled: %s'
330
% unicode(timer).rsplit(".", 1)[0])
331
else:
332
message = "enabled"
333
self._text = "%s%s" % (base, message)
334
159
self._text = (u'name="%(name)s", enabled=%(enabled)s'
160
% self.properties)
335
161
if not urwid.supports_unicode():
336
162
self._text = self._text.encode("ascii", "replace")
337
textlist = [("normal", self._text)]
163
textlist = [(u"normal", u"BLÄRGH: "), (u"bold", self._text)]
338
164
self._text_widget.set_text(textlist)
339
165
self._focus_text_widget.set_text([(with_standout[text[0]],
340
166
text[1])
348
174
if self.update_hook is not None:
349
175
self.update_hook()
350
176
351
def update_timer(self):
352
"""called by gobject. Will indefinitely loop until
353
gobject.source_remove() on tag is called"""
354
self.update()
355
return True # Keep calling this
356
357
def delete(self, *args, **kwargs):
358
if self._update_timer_callback_tag is not None:
359
gobject.source_remove(self._update_timer_callback_tag)
360
self._update_timer_callback_tag = None
361
for match in self.match_objects:
362
match.remove()
363
self.match_objects = ()
177
def delete(self):
364
178
if self.delete_hook is not None:
365
179
self.delete_hook(self)
366
return super(MandosClientWidget, self).delete(*args, **kwargs)
367
180
368
def render(self, maxcolrow, focus=False):
181
def render(self, (maxcol,), focus=False):
369
182
"""Render differently if we have focus.
370
183
This overrides the method from urwid.FlowWidget"""
371
return self.current_widget(focus).render(maxcolrow,
184
return self.current_widget(focus).render((maxcol,),
372
185
focus=focus)
373
186
374
def keypress(self, maxcolrow, key):
187
def keypress(self, (maxcol,), key):
375
188
"""Handle keys.
376
189
This overrides the method from urwid.FlowWidget"""
377
if key == "+":
378
self.proxy.Enable(dbus_interface = client_interface,
379
ignore_reply=True)
380
elif key == "-":
381
self.proxy.Disable(dbus_interface = client_interface,
382
ignore_reply=True)
383
elif key == "a":
384
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
385
dbus_interface = client_interface,
386
ignore_reply=True)
387
elif key == "d":
388
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
389
dbus_interface = client_interface,
390
ignore_reply=True)
391
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"_":
392
195
self.server_proxy_object.RemoveClient(self.proxy
393
.object_path,
394
ignore_reply=True)
395
elif key == "s":
396
self.proxy.StartChecker(dbus_interface = client_interface,
397
ignore_reply=True)
398
elif key == "S":
399
self.proxy.StopChecker(dbus_interface = client_interface,
400
ignore_reply=True)
401
elif key == "C":
402
self.proxy.CheckedOK(dbus_interface = client_interface,
403
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()
404
203
# xxx
405
# elif key == "p" or key == "=":
204
# elif key == u"p" or key == "=":
406
205
# self.proxy.pause()
407
# elif key == "u" or key == ":":
206
# elif key == u"u" or key == ":":
408
207
# self.proxy.unpause()
409
# elif key == "RET":
208
# elif key == u"RET":
410
209
# self.open()
411
210
else:
412
211
return key
423
222
self.update()
424
223
425
224
426
class ConstrainedListBox(urwid.ListBox):
427
"""Like a normal urwid.ListBox, but will consume all "up" or
428
"down" key presses, thus not allowing any containing widgets to
429
use them as an excuse to shift focus away from this widget.
430
"""
431
def keypress(self, maxcolrow, key):
432
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
433
if ret in ("up", "down"):
434
return
435
return ret
436
437
438
225
class UserInterface(object):
439
226
"""This is the entire user interface - the whole screen
440
227
with boxes, lists of client widgets, etc.
441
228
"""
442
def __init__(self, max_log_length=1000):
443
DBusGMainLoop(set_as_default=True)
229
def __init__(self):
230
DBusGMainLoop(set_as_default=True )
444
231
445
232
self.screen = urwid.curses_display.Screen()
446
233
447
234
self.screen.register_palette((
448
("normal",
449
"default", "default", None),
450
("bold",
451
"default", "default", "bold"),
452
("underline-blink",
453
"default", "default", "underline"),
454
("standout",
455
"default", "default", "standout"),
456
("bold-underline-blink",
457
"default", "default", ("bold", "underline")),
458
("bold-standout",
459
"default", "default", ("bold", "standout")),
460
("underline-blink-standout",
461
"default", "default", ("underline", "standout")),
462
("bold-underline-blink-standout",
463
"default", "default", ("bold", "underline",
464
"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")),
465
252
))
466
253
467
if urwid.supports_unicode():
468
self.divider = "─" # \u2500
469
#self.divider = "━" # \u2501
470
else:
471
#self.divider = "-" # \u002d
472
self.divider = "_" # \u005f
473
474
254
self.screen.start()
475
255
476
256
self.size = self.screen.get_cols_rows()
477
257
478
258
self.clients = urwid.SimpleListWalker([])
479
259
self.clients_dict = {}
480
481
# We will add Text widgets to this list
482
self.log = []
483
self.max_log_length = max_log_length
484
485
# We keep a reference to the log widget so we can remove it
486
# from the ListWalker without it getting destroyed
487
self.logbox = ConstrainedListBox(self.log)
488
489
# This keeps track of whether self.uilist currently has
490
# self.logbox in it or not
491
self.log_visible = True
492
self.log_wrap = "any"
493
494
self.rebuild()
495
self.log_message_raw(("bold",
496
"Mandos Monitor version " + version))
497
self.log_message_raw(("bold",
498
"q: Quit ?: Help"))
260
self.topwidget = urwid.LineBox(urwid.ListBox(self.clients))
261
#self.topwidget = urwid.ListBox(clients)
499
262
500
263
self.busname = domain + '.Mandos'
501
264
self.main_loop = gobject.MainLoop()
502
503
def client_not_found(self, fingerprint, address):
504
self.log_message(("Client with address %s and fingerprint %s"
505
" could not be found" % (address,
506
fingerprint)))
507
508
def rebuild(self):
509
"""This rebuilds the User Interface.
510
Call this when the widget layout needs to change"""
511
self.uilist = []
512
#self.uilist.append(urwid.ListBox(self.clients))
513
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
514
clients),
515
#header=urwid.Divider(),
516
header=None,
517
footer=
518
urwid.Divider(div_char=
519
self.divider)))
520
if self.log_visible:
521
self.uilist.append(self.logbox)
522
self.topwidget = urwid.Pile(self.uilist)
523
524
def log_message(self, message):
525
timestamp = datetime.datetime.now().isoformat()
526
self.log_message_raw(timestamp + ": " + message)
527
528
def log_message_raw(self, markup):
529
"""Add a log message to the log buffer."""
530
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
531
if (self.max_log_length
532
and len(self.log) > self.max_log_length):
533
del self.log[0:len(self.log)-self.max_log_length-1]
534
self.logbox.set_focus(len(self.logbox.body.contents),
535
coming_from="above")
536
self.refresh()
537
538
def toggle_log_display(self):
539
"""Toggle visibility of the log buffer."""
540
self.log_visible = not self.log_visible
541
self.rebuild()
542
#self.log_message("Log visibility changed to: "
543
# + unicode(self.log_visible))
544
545
def change_log_display(self):
546
"""Change type of log display.
547
Currently, this toggles wrapping of text lines."""
548
if self.log_wrap == "clip":
549
self.log_wrap = "any"
550
else:
551
self.log_wrap = "clip"
552
for textwidget in self.log:
553
textwidget.set_wrap_mode(self.log_wrap)
554
#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)
555
300
556
301
def find_and_remove_client(self, path, name):
557
"""Find a client by its object path and remove it.
302
"""Find an client from its object path and remove it.
558
303
559
304
This is connected to the ClientRemoved signal from the
560
305
Mandos server object."""
562
307
client = self.clients_dict[path]
563
308
except KeyError:
564
309
# not found?
565
self.log_message("Unknown client %r (%r) removed", name,
566
path)
567
310
return
568
client.delete()
311
self.remove_client(client, path)
569
312
570
def add_new_client(self, path):
313
def add_new_client(self, path, properties):
571
314
client_proxy_object = self.bus.get_object(self.busname, path)
572
315
self.add_client(MandosClientWidget(server_proxy_object
573
316
=self.mandos_serv,
574
317
proxy_object
575
318
=client_proxy_object,
319
properties=properties,
576
320
update_hook
577
321
=self.refresh,
578
322
delete_hook
579
=self.remove_client,
580
logger
581
=self.log_message),
323
=self.remove_client),
582
324
path=path)
583
325
584
326
def add_client(self, client, path=None):
586
328
if path is None:
587
329
path = client.proxy.object_path
588
330
self.clients_dict[path] = client
589
self.clients.sort(None, lambda c: c.properties["Name"])
331
self.clients.sort(None, lambda c: c.properties[u"name"])
590
332
self.refresh()
591
333
592
334
def remove_client(self, client, path=None):
594
336
if path is None:
595
337
path = client.proxy.object_path
596
338
del self.clients_dict[path]
597
if not self.clients_dict:
598
# Work around bug in Urwid 0.9.8.3 - if a SimpleListWalker
599
# is completely emptied, we need to recreate it.
600
self.clients = urwid.SimpleListWalker([])
601
self.rebuild()
602
339
self.refresh()
603
340
604
341
def refresh(self):
608
345
609
346
def run(self):
610
347
"""Start the main loop and exit when it's done."""
611
self.bus = dbus.SystemBus()
612
mandos_dbus_objc = self.bus.get_object(
613
self.busname, "/", follow_name_owner_changes=True)
614
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
615
dbus_interface
616
= server_interface)
617
try:
618
mandos_clients = (self.mandos_serv
619
.GetAllClientsWithProperties())
620
except dbus.exceptions.DBusException:
621
mandos_clients = dbus.Dictionary()
622
623
(self.mandos_serv
624
.connect_to_signal("ClientRemoved",
625
self.find_and_remove_client,
626
dbus_interface=server_interface,
627
byte_arrays=True))
628
(self.mandos_serv
629
.connect_to_signal("ClientAdded",
630
self.add_new_client,
631
dbus_interface=server_interface,
632
byte_arrays=True))
633
(self.mandos_serv
634
.connect_to_signal("ClientNotFound",
635
self.client_not_found,
636
dbus_interface=server_interface,
637
byte_arrays=True))
638
for path, client in mandos_clients.iteritems():
639
client_proxy_object = self.bus.get_object(self.busname,
640
path)
641
self.add_client(MandosClientWidget(server_proxy_object
642
=self.mandos_serv,
643
proxy_object
644
=client_proxy_object,
645
properties=client,
646
update_hook
647
=self.refresh,
648
delete_hook
649
=self.remove_client,
650
logger
651
=self.log_message),
652
path=path)
653
654
348
self.refresh()
655
349
self._input_callback_tag = (gobject.io_add_watch
656
350
(sys.stdin.fileno(),
666
360
667
361
def process_input(self, source, condition):
668
362
keys = self.screen.get_input()
669
translations = { "ctrl n": "down", # Emacs
670
"ctrl p": "up", # Emacs
671
"ctrl v": "page down", # Emacs
672
"meta v": "page up", # Emacs
673
" ": "page down", # less
674
"f": "page down", # less
675
"b": "page up", # less
676
"j": "down", # vi
677
"k": "up", # vi
363
translations = { u"j": u"down",
364
u"k": u"up",
678
365
}
679
366
for key in keys:
680
367
try:
682
369
except KeyError: # :-)
683
370
pass
684
371
685
if key == "q" or key == "Q":
372
if key == u"q" or key == u"Q":
686
373
self.stop()
687
374
break
688
elif key == "window resize":
375
elif key == u"window resize":
689
376
self.size = self.screen.get_cols_rows()
690
377
self.refresh()
691
elif key == "\f": # Ctrl-L
692
self.refresh()
693
elif key == "l" or key == "D":
694
self.toggle_log_display()
695
self.refresh()
696
elif key == "w" or key == "i":
697
self.change_log_display()
698
self.refresh()
699
elif key == "?" or key == "f1" or key == "esc":
700
if not self.log_visible:
701
self.log_visible = True
702
self.rebuild()
703
self.log_message_raw(("bold",
704
" ".
705
join(("q: Quit",
706
"?: Help",
707
"l: Log window toggle",
708
"TAB: Switch window",
709
"w: Wrap (log)"))))
710
self.log_message_raw(("bold",
711
" "
712
.join(("Clients:",
713
"+: Enable",
714
"-: Disable",
715
"R: Remove",
716
"s: Start new checker",
717
"S: Stop checker",
718
"C: Checker OK",
719
"a: Approve",
720
"d: Deny"))))
721
self.refresh()
722
elif key == "tab":
723
if self.topwidget.get_focus() is self.logbox:
724
self.topwidget.set_focus(0)
725
else:
726
self.topwidget.set_focus(self.logbox)
727
self.refresh()
728
#elif (key == "end" or key == "meta >" or key == "G"
729
# or key == ">"):
730
# pass # xxx end-of-buffer
731
#elif (key == "home" or key == "meta <" or key == "g"
732
# or key == "<"):
733
# pass # xxx beginning-of-buffer
734
#elif key == "ctrl e" or key == "$":
735
# pass # xxx move-end-of-line
736
#elif key == "ctrl a" or key == "^":
737
# pass # xxx move-beginning-of-line
738
#elif key == "ctrl b" or key == "meta (" or key == "h":
739
# pass # xxx left
740
#elif key == "ctrl f" or key == "meta )" or key == "l":
741
# pass # xxx right
742
#elif key == "a":
743
# pass # scroll up log
744
#elif key == "z":
745
# pass # scroll down log
378
elif key == " ":
379
self.refresh()
746
380
elif self.topwidget.selectable():
747
381
self.topwidget.keypress(self.size, key)
748
382
self.refresh()
751
385
ui = UserInterface()
752
386
try:
753
387
ui.run()
754
except KeyboardInterrupt:
755
ui.screen.stop()
756
except Exception, e:
757
ui.log_message(unicode(e))
388
except:
758
389
ui.screen.stop()
759
390
raise
Loggerhead is a web-based interface for Breezy
Version: 2.0.2.dev0