2
2
# -*- mode: python; coding: utf-8 -*-
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009,2010 Teddy Hogeborn
7
# Copyright © 2009,2010 Björn Påhlsson
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.
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.
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/>.
22
# Contact the authors at <mandos@fukt.bsnet.se>.
4
25
from __future__ import division, absolute_import, with_statement
9
33
import urwid.curses_display
30
61
urwid.curses_display.curses.A_UNDERLINE |= (
31
62
urwid.curses_display.curses.A_BLINK)
64
def isoformat_to_datetime(iso):
65
"Parse an ISO 8601 date string to a datetime.datetime()"
68
d, t = iso.split(u"T", 1)
69
year, month, day = d.split(u"-", 2)
70
hour, minute, second = t.split(u":", 2)
71
second, fraction = divmod(float(second), 1)
72
return datetime.datetime(int(year),
77
int(second), # Whole seconds
78
int(fraction*1000000)) # Microseconds
33
80
class MandosClientPropertyCache(object):
34
81
"""This wraps a Mandos Client D-Bus proxy object, caches the
35
82
properties and calls a hook function when any of them are
38
def __init__(self, proxy_object=None, properties=None, *args,
85
def __init__(self, proxy_object=None, *args, **kwargs):
40
86
self.proxy = proxy_object # Mandos Client proxy object
42
if properties is None:
43
self.properties = dict()
45
self.properties = properties
46
self.proxy.connect_to_signal("PropertyChanged",
88
self.properties = dict()
89
self.proxy.connect_to_signal(u"PropertyChanged",
47
90
self.property_changed,
51
if properties is None:
52
self.properties.update(self.proxy.GetAll(client_interface,
54
dbus.PROPERTIES_IFACE))
55
super(MandosClientPropertyCache, self).__init__(
56
proxy_object=proxy_object,
57
properties=properties, *args, **kwargs)
94
self.properties.update(
95
self.proxy.GetAll(client_interface,
96
dbus_interface = dbus.PROPERTIES_IFACE))
98
#XXX This break good super behaviour!
99
# super(MandosClientPropertyCache, self).__init__(
59
102
def property_changed(self, property=None, value=None):
60
103
"""This is called whenever we get a PropertyChanged signal
71
114
def __init__(self, server_proxy_object=None, update_hook=None,
72
delete_hook=None, *args, **kwargs):
115
delete_hook=None, logger=None, *args, **kwargs):
73
116
# Called on update
74
117
self.update_hook = update_hook
75
118
# Called on delete
76
119
self.delete_hook = delete_hook
77
120
# Mandos Server proxy object
78
121
self.server_proxy_object = server_proxy_object
125
self._update_timer_callback_tag = None
126
self._update_timer_callback_lock = 0
127
self.last_checker_failed = False
80
129
# The widget shown normally
81
self._text_widget = urwid.Text("")
130
self._text_widget = urwid.Text(u"")
82
131
# The widget shown when we have focus
83
self._focus_text_widget = urwid.Text("")
132
self._focus_text_widget = urwid.Text(u"")
84
133
super(MandosClientWidget, self).__init__(
85
134
update_hook=update_hook, delete_hook=delete_hook,
88
137
self.opened = False
139
last_checked_ok = isoformat_to_datetime(self.properties
141
if last_checked_ok is None:
142
self.last_checker_failed = True
144
self.last_checker_failed = ((datetime.datetime.utcnow()
151
if self.last_checker_failed:
152
self.using_timer(True)
154
if self.need_approval:
155
self.using_timer(True)
157
self.proxy.connect_to_signal(u"CheckerCompleted",
158
self.checker_completed,
161
self.proxy.connect_to_signal(u"CheckerStarted",
162
self.checker_started,
165
self.proxy.connect_to_signal(u"GotSecret",
169
self.proxy.connect_to_signal(u"NeedApproval",
173
self.proxy.connect_to_signal(u"Rejected",
178
def property_changed(self, property=None, value=None):
179
super(self, MandosClientWidget).property_changed(property,
181
if property == u"ApprovalPending":
182
using_timer(bool(value))
184
def using_timer(self, flag):
185
"""Call this method with True or False when timer should be
186
activated or deactivated.
188
old = self._update_timer_callback_lock
190
self._update_timer_callback_lock += 1
192
self._update_timer_callback_lock -= 1
193
if old == 0 and self._update_timer_callback_lock:
194
self._update_timer_callback_tag = (gobject.timeout_add
197
elif old and self._update_timer_callback_lock == 0:
198
gobject.source_remove(self._update_timer_callback_tag)
199
self._update_timer_callback_tag = None
201
def checker_completed(self, exitstatus, condition, command):
203
if self.last_checker_failed:
204
self.last_checker_failed = False
205
self.using_timer(False)
206
#self.logger(u'Checker for client %s (command "%s")'
208
# % (self.properties[u"Name"], command))
212
if not self.last_checker_failed:
213
self.last_checker_failed = True
214
self.using_timer(True)
215
if os.WIFEXITED(condition):
216
self.logger(u'Checker for client %s (command "%s")'
217
u' failed with exit code %s'
218
% (self.properties[u"Name"], command,
219
os.WEXITSTATUS(condition)))
220
elif os.WIFSIGNALED(condition):
221
self.logger(u'Checker for client %s (command "%s")'
222
u' was killed by signal %s'
223
% (self.properties[u"Name"], command,
224
os.WTERMSIG(condition)))
225
elif os.WCOREDUMP(condition):
226
self.logger(u'Checker for client %s (command "%s")'
228
% (self.properties[u"Name"], command))
230
self.logger(u'Checker for client %s completed'
234
def checker_started(self, command):
235
#self.logger(u'Client %s started checker "%s"'
236
# % (self.properties[u"Name"], unicode(command)))
239
def got_secret(self):
240
self.last_checker_failed = False
241
self.logger(u'Client %s received its secret'
242
% self.properties[u"Name"])
244
def need_approval(self, timeout, default):
246
message = u'Client %s needs approval within %s seconds'
248
message = u'Client %s will get its secret in %s seconds'
250
% (self.properties[u"Name"], timeout/1000))
251
self.using_timer(True)
253
def rejected(self, reason):
254
self.logger(u'Client %s was rejected; reason: %s'
255
% (self.properties[u"Name"], reason))
90
257
def selectable(self):
91
258
"""Make this a "selectable" widget.
92
259
This overrides the method from urwid.FlowWidget."""
95
def rows(self, (maxcol,), focus=False):
262
def rows(self, maxcolrow, focus=False):
96
263
"""How many rows this widget will occupy might depend on
97
264
whether we have focus or not.
98
265
This overrides the method from urwid.FlowWidget"""
99
return self.current_widget(focus).rows((maxcol,), focus=focus)
266
return self.current_widget(focus).rows(maxcolrow, focus=focus)
101
268
def current_widget(self, focus=False):
102
269
if focus or self.opened:
113
280
u"bold-underline-blink":
114
281
u"bold-underline-blink-standout",
117
284
# Rebuild focus and non-focus widgets using current properties
118
self._text = (u'name="%(name)s", enabled=%(enabled)s'
286
# Base part of a client. Name!
287
base = (u'%(name)s: '
288
% {u"name": self.properties[u"Name"]})
289
if not self.properties[u"Enabled"]:
290
message = u"DISABLED"
291
elif self.properties[u"ApprovalPending"]:
292
timeout = datetime.timedelta(milliseconds
295
last_approval_request = isoformat_to_datetime(
296
self.properties[u"LastApprovalRequest"])
297
if last_approval_request is not None:
298
timer = timeout - (datetime.datetime.utcnow()
299
- last_approval_request)
301
timer = datetime.timedelta()
302
if self.properties[u"ApprovedByDefault"]:
303
message = u"Approval in %s. (d)eny?"
305
message = u"Denial in %s. (a)pprove?"
306
message = message % unicode(timer).rsplit(".", 1)[0]
307
elif self.last_checker_failed:
308
timeout = datetime.timedelta(milliseconds
311
last_ok = isoformat_to_datetime(
312
max((self.properties[u"LastCheckedOK"]
313
or self.properties[u"Created"]),
314
self.properties[u"LastEnabled"]))
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
316
message = (u'A checker has failed! Time until client'
317
u' gets disabled: %s'
318
% unicode(timer).rsplit(".", 1)[0])
321
self._text = u"%s%s" % (base, message)
120
323
if not urwid.supports_unicode():
121
self._text = self._text.encode("ascii", "replace")
122
textlist = [(u"normal", u"BLARGH: "), (u"bold", self._text)]
324
self._text = self._text.encode(u"ascii", u"replace")
325
textlist = [(u"normal", self._text)]
123
326
self._text_widget.set_text(textlist)
124
327
self._focus_text_widget.set_text([(with_standout[text[0]],
128
331
for text in textlist])
129
332
self._widget = self._text_widget
130
333
self._focus_widget = urwid.AttrWrap(self._focus_text_widget,
132
335
# Run update hook, if any
133
336
if self.update_hook is not None:
134
337
self.update_hook()
339
def update_timer(self):
342
return True # Keep calling this
136
344
def delete(self):
345
if self._update_timer_callback_tag is not None:
346
gobject.source_remove(self._update_timer_callback_tag)
347
self._update_timer_callback_tag = None
137
348
if self.delete_hook is not None:
138
349
self.delete_hook(self)
140
def render(self, (maxcol,), focus=False):
351
def render(self, maxcolrow, focus=False):
141
352
"""Render differently if we have focus.
142
353
This overrides the method from urwid.FlowWidget"""
143
return self.current_widget(focus).render((maxcol,),
354
return self.current_widget(focus).render(maxcolrow,
146
def keypress(self, (maxcol,), key):
357
def keypress(self, maxcolrow, key):
148
359
This overrides the method from urwid.FlowWidget"""
149
if key == u"e" or key == u"+":
151
elif key == u"d" or key == u"-":
153
elif key == u"r" or key == u"_":
361
self.proxy.Enable(dbus_interface = client_interface)
363
self.proxy.Disable(dbus_interface = client_interface)
365
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
366
dbus_interface = client_interface)
368
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
369
dbus_interface = client_interface)
370
elif key == u"R" or key == u"_" or key == u"ctrl k":
154
371
self.server_proxy_object.RemoveClient(self.proxy
156
373
elif key == u"s":
157
self.proxy.StartChecker()
374
self.proxy.StartChecker(dbus_interface = client_interface)
158
375
elif key == u"S":
159
self.proxy.StopChecker()
376
self.proxy.StopChecker(dbus_interface = client_interface)
160
377
elif key == u"C":
161
self.proxy.CheckedOK()
378
self.proxy.CheckedOK(dbus_interface = client_interface)
163
380
# elif key == u"p" or key == "=":
164
381
# self.proxy.pause()
293
=self.remove_client),
520
def client_not_found(self, fingerprint, address):
521
self.log_message((u"Client with address %s and fingerprint %s"
522
u" could not be found" % (address,
296
525
def rebuild(self):
297
526
"""This rebuilds the User Interface.
298
527
Call this when the widget layout needs to change"""
300
529
#self.uilist.append(urwid.ListBox(self.clients))
301
self.uilist.append(urwid.Frame(ConstrainedListBox(self.clients),
530
self.uilist.append(urwid.Frame(ConstrainedListBox(self.
302
532
#header=urwid.Divider(),
304
footer=urwid.Divider(div_char=self.divider)))
535
urwid.Divider(div_char=
305
537
if self.log_visible:
306
538
self.uilist.append(self.logbox)
308
540
self.topwidget = urwid.Pile(self.uilist)
310
def log_message(self, markup):
542
def log_message(self, message):
543
timestamp = datetime.datetime.now().isoformat()
544
self.log_message_raw(timestamp + u": " + message)
546
def log_message_raw(self, markup):
311
547
"""Add a log message to the log buffer."""
312
548
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
313
549
if (self.max_log_length
314
550
and len(self.log) > self.max_log_length):
315
551
del self.log[0:len(self.log)-self.max_log_length-1]
552
self.logbox.set_focus(len(self.logbox.body.contents),
553
coming_from=u"above")
317
556
def toggle_log_display(self):
318
557
"""Toggle visibility of the log buffer."""
319
558
self.log_visible = not self.log_visible
321
self.log_message(u"Log visibility changed to: "
322
+ unicode(self.log_visible))
560
#self.log_message(u"Log visibility changed to: "
561
# + unicode(self.log_visible))
324
563
def change_log_display(self):
325
564
"""Change type of log display.