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
40
urwid.curses_display.curses.A_UNDERLINE |= (
62
41
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
43
class MandosClientPropertyCache(object):
81
44
"""This wraps a Mandos Client D-Bus proxy object, caches the
82
45
properties and calls a hook function when any of them are
94
57
self.properties.update(
95
58
self.proxy.GetAll(client_interface,
96
59
dbus_interface = dbus.PROPERTIES_IFACE))
98
#XXX This break good super behaviour!
99
# super(MandosClientPropertyCache, self).__init__(
60
super(MandosClientPropertyCache, self).__init__(
102
63
def property_changed(self, property=None, value=None):
103
64
"""This is called whenever we get a PropertyChanged signal
175
113
client_interface,
176
114
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
116
def checker_completed(self, exitstatus, condition, command):
202
117
if exitstatus == 0:
203
if self.last_checker_failed:
204
self.last_checker_failed = False
205
self.using_timer(False)
206
118
#self.logger(u'Checker for client %s (command "%s")'
207
119
# u' was successful'
208
# % (self.properties[u"Name"], command))
120
# % (self.properties[u"name"], command))
212
if not self.last_checker_failed:
213
self.last_checker_failed = True
214
self.using_timer(True)
215
122
if os.WIFEXITED(condition):
216
123
self.logger(u'Checker for client %s (command "%s")'
217
124
u' failed with exit code %s'
218
% (self.properties[u"Name"], command,
125
% (self.properties[u"name"], command,
219
126
os.WEXITSTATUS(condition)))
220
elif os.WIFSIGNALED(condition):
128
if os.WIFSIGNALED(condition):
221
129
self.logger(u'Checker for client %s (command "%s")'
222
130
u' was killed by signal %s'
223
% (self.properties[u"Name"], command,
131
% (self.properties[u"name"], command,
224
132
os.WTERMSIG(condition)))
225
elif os.WCOREDUMP(condition):
134
if os.WCOREDUMP(condition):
226
135
self.logger(u'Checker for client %s (command "%s")'
228
% (self.properties[u"Name"], command))
230
self.logger(u'Checker for client %s completed'
137
% (self.properties[u"name"], command))
138
self.logger(u'Checker for client %s completed mysteriously')
234
140
def checker_started(self, command):
235
141
#self.logger(u'Client %s started checker "%s"'
236
# % (self.properties[u"Name"], unicode(command)))
142
# % (self.properties[u"name"], unicode(command)))
239
145
def got_secret(self):
240
self.last_checker_failed = False
241
146
self.logger(u'Client %s received its secret'
242
% self.properties[u"Name"])
147
% self.properties[u"name"])
244
149
def need_approval(self, timeout, default):
248
153
message = u'Client %s will get its secret in %s seconds'
249
154
self.logger(message
250
% (self.properties[u"Name"], timeout/1000))
251
self.using_timer(True)
155
% (self.properties[u"name"], timeout/1000))
253
157
def rejected(self, reason):
254
158
self.logger(u'Client %s was rejected; reason: %s'
255
% (self.properties[u"Name"], reason))
159
% (self.properties[u"name"], reason))
257
161
def selectable(self):
258
162
"""Make this a "selectable" widget.
280
184
u"bold-underline-blink":
281
185
u"bold-underline-blink-standout",
284
188
# 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 = "%s%s" % (base, message)
189
self._text = (u'%(name)s: %(enabled)s'
190
% { u"name": self.properties[u"name"],
193
if self.properties[u"enabled"]
323
195
if not urwid.supports_unicode():
324
196
self._text = self._text.encode("ascii", "replace")
325
197
textlist = [(u"normal", self._text)]
357
221
def keypress(self, (maxcol,), key):
359
223
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":
224
if key == u"e" or key == u"+":
226
elif key == u"d" or key == u"-":
228
elif key == u"r" or key == u"_" or key == u"ctrl k":
371
229
self.server_proxy_object.RemoveClient(self.proxy
373
231
elif key == u"s":
374
self.proxy.StartChecker(dbus_interface = client_interface)
232
self.proxy.StartChecker()
375
233
elif key == u"S":
376
self.proxy.StopChecker(dbus_interface = client_interface)
234
self.proxy.StopChecker()
377
235
elif key == u"C":
378
self.proxy.CheckedOK(dbus_interface = client_interface)
236
self.proxy.CheckedOK()
380
238
# elif key == u"p" or key == "=":
381
239
# self.proxy.pause()
404
266
use them as an excuse to shift focus away from this widget.
406
268
def keypress(self, (maxcol, maxrow), key):
407
ret = super(ConstrainedListBox, self).keypress((maxcol,
269
ret = super(ConstrainedListBox, self).keypress((maxcol, maxrow), key)
409
270
if ret in (u"up", u"down"):