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>.
25
4
from __future__ import division, absolute_import, with_statement
61
33
urwid.curses_display.curses.A_UNDERLINE |= (
62
34
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
80
36
class MandosClientPropertyCache(object):
81
37
"""This wraps a Mandos Client D-Bus proxy object, caches the
82
38
properties and calls a hook function when any of them are
85
def __init__(self, proxy_object=None, *args, **kwargs):
41
def __init__(self, proxy_object=None, properties=None, *args,
86
43
self.proxy = proxy_object # Mandos Client proxy object
88
self.properties = dict()
45
if properties is None:
46
self.properties = dict()
48
self.properties = properties
89
49
self.proxy.connect_to_signal(u"PropertyChanged",
90
50
self.property_changed,
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__(
54
if properties is None:
55
self.properties.update(self.proxy.GetAll(client_interface,
57
dbus.PROPERTIES_IFACE))
58
super(MandosClientPropertyCache, self).__init__(
59
proxy_object=proxy_object,
60
properties=properties, *args, **kwargs)
102
62
def property_changed(self, property=None, value=None):
103
63
"""This is called whenever we get a PropertyChanged signal
167
104
client_interface,
168
105
byte_arrays=True)
169
self.proxy.connect_to_signal(u"NeedApproval",
173
106
self.proxy.connect_to_signal(u"Rejected",
175
108
client_interface,
176
109
byte_arrays=True)
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
111
def checker_completed(self, exitstatus, condition, command):
202
112
if exitstatus == 0:
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))
113
self.logger(u'Checker for client %s (command "%s")'
115
% (self.properties[u"name"], command))
212
if not self.last_checker_failed:
213
self.last_checker_failed = True
214
self.using_timer(True)
215
117
if os.WIFEXITED(condition):
216
118
self.logger(u'Checker for client %s (command "%s")'
217
119
u' failed with exit code %s'
218
% (self.properties[u"Name"], command,
120
% (self.properties[u"name"], command,
219
121
os.WEXITSTATUS(condition)))
220
elif os.WIFSIGNALED(condition):
123
if os.WIFSIGNALED(condition):
221
124
self.logger(u'Checker for client %s (command "%s")'
222
125
u' was killed by signal %s'
223
% (self.properties[u"Name"], command,
126
% (self.properties[u"name"], command,
224
127
os.WTERMSIG(condition)))
225
elif os.WCOREDUMP(condition):
129
if os.WCOREDUMP(condition):
226
130
self.logger(u'Checker for client %s (command "%s")'
228
% (self.properties[u"Name"], command))
230
self.logger(u'Checker for client %s completed'
132
% (self.properties[u"name"], command))
133
self.logger(u'Checker for client %s completed mysteriously')
234
135
def checker_started(self, command):
235
#self.logger(u'Client %s started checker "%s"'
236
# % (self.properties[u"Name"], unicode(command)))
136
self.logger(u'Client %s started checker "%s"'
137
% (self.properties[u"name"], unicode(command)))
239
139
def got_secret(self):
240
self.last_checker_failed = False
241
140
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))
141
% self.properties[u"name"])
144
self.logger(u'Client %s was rejected'
145
% self.properties[u"name"])
257
147
def selectable(self):
258
148
"""Make this a "selectable" widget.
259
149
This overrides the method from urwid.FlowWidget."""
262
def rows(self, maxcolrow, focus=False):
152
def rows(self, (maxcol,), focus=False):
263
153
"""How many rows this widget will occupy might depend on
264
154
whether we have focus or not.
265
155
This overrides the method from urwid.FlowWidget"""
266
return self.current_widget(focus).rows(maxcolrow, focus=focus)
156
return self.current_widget(focus).rows((maxcol,), focus=focus)
268
158
def current_widget(self, focus=False):
269
159
if focus or self.opened:
280
170
u"bold-underline-blink":
281
171
u"bold-underline-blink-standout",
284
174
# Rebuild focus and non-focus widgets using current properties
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)
175
self._text = (u'%(name)s: %(enabled)s'
176
% { u"name": self.properties[u"name"],
179
if self.properties[u"enabled"]
323
181
if not urwid.supports_unicode():
324
self._text = self._text.encode(u"ascii", u"replace")
182
self._text = self._text.encode("ascii", "replace")
325
183
textlist = [(u"normal", self._text)]
326
184
self._text_widget.set_text(textlist)
327
185
self._focus_text_widget.set_text([(with_standout[text[0]],
331
189
for text in textlist])
332
190
self._widget = self._text_widget
333
191
self._focus_widget = urwid.AttrWrap(self._focus_text_widget,
335
193
# Run update hook, if any
336
194
if self.update_hook is not None:
337
195
self.update_hook()
339
def update_timer(self):
342
return True # Keep calling this
344
197
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
348
198
if self.delete_hook is not None:
349
199
self.delete_hook(self)
351
def render(self, maxcolrow, focus=False):
201
def render(self, (maxcol,), focus=False):
352
202
"""Render differently if we have focus.
353
203
This overrides the method from urwid.FlowWidget"""
354
return self.current_widget(focus).render(maxcolrow,
204
return self.current_widget(focus).render((maxcol,),
357
def keypress(self, maxcolrow, key):
207
def keypress(self, (maxcol,), key):
359
209
This overrides the method from urwid.FlowWidget"""
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":
210
if key == u"e" or key == u"+":
212
elif key == u"d" or key == u"-":
214
elif key == u"r" or key == u"_" or key == u"ctrl k":
371
215
self.server_proxy_object.RemoveClient(self.proxy
373
217
elif key == u"s":
374
self.proxy.StartChecker(dbus_interface = client_interface)
218
self.proxy.StartChecker()
375
219
elif key == u"S":
376
self.proxy.StopChecker(dbus_interface = client_interface)
220
self.proxy.StopChecker()
377
221
elif key == u"C":
378
self.proxy.CheckedOK(dbus_interface = client_interface)
222
self.proxy.CheckedOK()
380
224
# elif key == u"p" or key == "=":
381
225
# self.proxy.pause()