2
2
# -*- mode: python; coding: utf-8 -*-
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2011 Teddy Hogeborn
7
# Copyright © 2009-2011 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
from __future__ import division, absolute_import, with_statement
25
from __future__ import (division, absolute_import, print_function,
44
66
"Parse an ISO 8601 date string to a datetime.datetime()"
47
d, t = iso.split(u"T", 1)
48
year, month, day = d.split(u"-", 2)
49
hour, minute, second = t.split(u":", 2)
69
d, t = iso.split("T", 1)
70
year, month, day = d.split("-", 2)
71
hour, minute, second = t.split(":", 2)
50
72
second, fraction = divmod(float(second), 1)
51
73
return datetime.datetime(int(year),
65
87
self.proxy = proxy_object # Mandos Client proxy object
67
89
self.properties = dict()
68
self.proxy.connect_to_signal(u"PropertyChanged",
69
self.property_changed,
90
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
73
96
self.properties.update(
74
97
self.proxy.GetAll(client_interface,
75
98
dbus_interface = dbus.PROPERTIES_IFACE))
77
#XXX This break good super behaviour!
100
#XXX This breaks good super behaviour
78
101
# super(MandosClientPropertyCache, self).__init__(
79
102
# *args, **kwargs)
106
134
self.last_checker_failed = False
108
136
# The widget shown normally
109
self._text_widget = urwid.Text(u"")
137
self._text_widget = urwid.Text("")
110
138
# The widget shown when we have focus
111
self._focus_text_widget = urwid.Text(u"")
139
self._focus_text_widget = urwid.Text("")
112
140
super(MandosClientWidget, self).__init__(
113
141
update_hook=update_hook, delete_hook=delete_hook,
133
161
if self.need_approval:
134
162
self.using_timer(True)
136
self.proxy.connect_to_signal(u"CheckerCompleted",
137
self.checker_completed,
140
self.proxy.connect_to_signal(u"CheckerStarted",
141
self.checker_started,
144
self.proxy.connect_to_signal(u"GotSecret",
148
self.proxy.connect_to_signal(u"NeedApproval",
152
self.proxy.connect_to_signal(u"Rejected",
164
self.match_objects = (
165
self.proxy.connect_to_signal("CheckerCompleted",
166
self.checker_completed,
169
self.proxy.connect_to_signal("CheckerStarted",
170
self.checker_started,
173
self.proxy.connect_to_signal("GotSecret",
177
self.proxy.connect_to_signal("NeedApproval",
181
self.proxy.connect_to_signal("Rejected",
185
#self.logger('Created client %s' % (self.properties["Name"]))
157
187
def property_changed(self, property=None, value=None):
158
188
super(self, MandosClientWidget).property_changed(property,
160
if property == u"ApprovalPending":
190
if property == "ApprovalPending":
161
191
using_timer(bool(value))
163
193
def using_timer(self, flag):
192
223
self.last_checker_failed = True
193
224
self.using_timer(True)
194
225
if os.WIFEXITED(condition):
195
self.logger(u'Checker for client %s (command "%s")'
196
u' failed with exit code %s'
197
% (self.properties[u"Name"], command,
226
self.logger('Checker for client %s (command "%s")'
227
' failed with exit code %s'
228
% (self.properties["Name"], command,
198
229
os.WEXITSTATUS(condition)))
199
230
elif os.WIFSIGNALED(condition):
200
self.logger(u'Checker for client %s (command "%s")'
201
u' was killed by signal %s'
202
% (self.properties[u"Name"], command,
231
self.logger('Checker for client %s (command "%s")'
232
' was killed by signal %s'
233
% (self.properties["Name"], command,
203
234
os.WTERMSIG(condition)))
204
235
elif os.WCOREDUMP(condition):
205
self.logger(u'Checker for client %s (command "%s")'
207
% (self.properties[u"Name"], command))
236
self.logger('Checker for client %s (command "%s")'
238
% (self.properties["Name"], command))
209
self.logger(u'Checker for client %s completed'
240
self.logger('Checker for client %s completed'
213
244
def checker_started(self, command):
214
#self.logger(u'Client %s started checker "%s"'
215
# % (self.properties[u"Name"], unicode(command)))
245
#self.logger('Client %s started checker "%s"'
246
# % (self.properties["Name"], unicode(command)))
218
249
def got_secret(self):
219
250
self.last_checker_failed = False
220
self.logger(u'Client %s received its secret'
221
% self.properties[u"Name"])
251
self.logger('Client %s received its secret'
252
% self.properties["Name"])
223
254
def need_approval(self, timeout, default):
225
message = u'Client %s needs approval within %s seconds'
256
message = 'Client %s needs approval within %s seconds'
227
message = u'Client %s will get its secret in %s seconds'
258
message = 'Client %s will get its secret in %s seconds'
228
259
self.logger(message
229
% (self.properties[u"Name"], timeout/1000))
260
% (self.properties["Name"], timeout/1000))
230
261
self.using_timer(True)
232
263
def rejected(self, reason):
233
self.logger(u'Client %s was rejected; reason: %s'
234
% (self.properties[u"Name"], reason))
264
self.logger('Client %s was rejected; reason: %s'
265
% (self.properties["Name"], reason))
236
267
def selectable(self):
237
268
"""Make this a "selectable" widget.
238
269
This overrides the method from urwid.FlowWidget."""
241
def rows(self, (maxcol,), focus=False):
272
def rows(self, maxcolrow, focus=False):
242
273
"""How many rows this widget will occupy might depend on
243
274
whether we have focus or not.
244
275
This overrides the method from urwid.FlowWidget"""
245
return self.current_widget(focus).rows((maxcol,), focus=focus)
276
return self.current_widget(focus).rows(maxcolrow, focus=focus)
247
278
def current_widget(self, focus=False):
248
279
if focus or self.opened:
252
283
def update(self):
253
284
"Called when what is visible on the screen should be updated."
254
285
# How to add standout mode to a style
255
with_standout = { u"normal": u"standout",
256
u"bold": u"bold-standout",
258
u"underline-blink-standout",
259
u"bold-underline-blink":
260
u"bold-underline-blink-standout",
286
with_standout = { "normal": "standout",
287
"bold": "bold-standout",
289
"underline-blink-standout",
290
"bold-underline-blink":
291
"bold-underline-blink-standout",
263
294
# Rebuild focus and non-focus widgets using current properties
265
296
# Base part of a client. Name!
266
base = (u'%(name)s: '
267
% {u"name": self.properties[u"Name"]})
268
if not self.properties[u"Enabled"]:
269
message = u"DISABLED"
270
elif self.properties[u"ApprovalPending"]:
298
% {"name": self.properties["Name"]})
299
if not self.properties["Enabled"]:
301
elif self.properties["ApprovalPending"]:
271
302
timeout = datetime.timedelta(milliseconds
272
303
= self.properties
274
305
last_approval_request = isoformat_to_datetime(
275
self.properties[u"LastApprovalRequest"])
306
self.properties["LastApprovalRequest"])
276
307
if last_approval_request is not None:
277
308
timer = timeout - (datetime.datetime.utcnow()
278
309
- last_approval_request)
280
311
timer = datetime.timedelta()
281
if self.properties[u"ApprovedByDefault"]:
282
message = u"Approval in %s. (d)eny?"
312
if self.properties["ApprovedByDefault"]:
313
message = "Approval in %s. (d)eny?"
284
message = u"Denial in %s. (a)pprove?"
315
message = "Denial in %s. (a)pprove?"
285
316
message = message % unicode(timer).rsplit(".", 1)[0]
286
317
elif self.last_checker_failed:
287
timeout = datetime.timedelta(milliseconds
290
last_ok = isoformat_to_datetime(
291
max((self.properties[u"LastCheckedOK"]
292
or self.properties[u"Created"]),
293
self.properties[u"LastEnabled"]))
294
timer = timeout - (datetime.datetime.utcnow() - last_ok)
295
message = (u'A checker has failed! Time until client'
296
u' gets disabled: %s'
318
# When checker has failed, print a timer until client expires
319
expires = self.properties["Expires"]
321
timer = datetime.timedelta(0)
323
expires = datetime.datetime.strptime(expires,
324
'%Y-%m-%dT%H:%M:%S.%f')
325
timer = expires - datetime.datetime.utcnow()
326
message = ('A checker has failed! Time until client'
297
328
% unicode(timer).rsplit(".", 1)[0])
300
331
self._text = "%s%s" % (base, message)
302
333
if not urwid.supports_unicode():
303
334
self._text = self._text.encode("ascii", "replace")
304
textlist = [(u"normal", self._text)]
335
textlist = [("normal", self._text)]
305
336
self._text_widget.set_text(textlist)
306
337
self._focus_text_widget.set_text([(with_standout[text[0]],
316
347
self.update_hook()
318
349
def update_timer(self):
350
"""called by gobject. Will indefinitely loop until
351
gobject.source_remove() on tag is called"""
321
353
return True # Keep calling this
355
def delete(self, *args, **kwargs):
324
356
if self._update_timer_callback_tag is not None:
325
357
gobject.source_remove(self._update_timer_callback_tag)
326
358
self._update_timer_callback_tag = None
359
for match in self.match_objects:
361
self.match_objects = ()
327
362
if self.delete_hook is not None:
328
363
self.delete_hook(self)
364
return super(MandosClientWidget, self).delete(*args, **kwargs)
330
def render(self, (maxcol,), focus=False):
366
def render(self, maxcolrow, focus=False):
331
367
"""Render differently if we have focus.
332
368
This overrides the method from urwid.FlowWidget"""
333
return self.current_widget(focus).render((maxcol,),
369
return self.current_widget(focus).render(maxcolrow,
336
def keypress(self, (maxcol,), key):
372
def keypress(self, maxcolrow, key):
338
374
This overrides the method from urwid.FlowWidget"""
340
self.proxy.Enable(dbus_interface = client_interface)
342
self.proxy.Disable(dbus_interface = client_interface)
376
self.proxy.Enable(dbus_interface = client_interface,
379
self.proxy.Disable(dbus_interface = client_interface,
344
382
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
345
dbus_interface = client_interface)
383
dbus_interface = client_interface,
347
386
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
348
dbus_interface = client_interface)
349
elif key == u"r" or key == u"_" or key == u"ctrl k":
387
dbus_interface = client_interface,
389
elif key == "R" or key == "_" or key == "ctrl k":
350
390
self.server_proxy_object.RemoveClient(self.proxy
353
self.proxy.StartChecker(dbus_interface = client_interface)
355
self.proxy.StopChecker(dbus_interface = client_interface)
357
self.proxy.CheckedOK(dbus_interface = client_interface)
394
self.proxy.StartChecker(dbus_interface = client_interface,
397
self.proxy.StopChecker(dbus_interface = client_interface,
400
self.proxy.CheckedOK(dbus_interface = client_interface,
359
# elif key == u"p" or key == "=":
403
# elif key == "p" or key == "=":
360
404
# self.proxy.pause()
361
# elif key == u"u" or key == ":":
405
# elif key == "u" or key == ":":
362
406
# self.proxy.unpause()
363
# elif key == u"RET":
382
426
"down" key presses, thus not allowing any containing widgets to
383
427
use them as an excuse to shift focus away from this widget.
385
def keypress(self, (maxcol, maxrow), key):
386
ret = super(ConstrainedListBox, self).keypress((maxcol,
388
if ret in (u"up", u"down"):
429
def keypress(self, maxcolrow, key):
430
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
431
if ret in ("up", "down"):
400
443
self.screen = urwid.curses_display.Screen()
402
445
self.screen.register_palette((
404
u"default", u"default", None),
406
u"default", u"default", u"bold"),
408
u"default", u"default", u"underline"),
410
u"default", u"default", u"standout"),
411
(u"bold-underline-blink",
412
u"default", u"default", (u"bold", u"underline")),
414
u"default", u"default", (u"bold", u"standout")),
415
(u"underline-blink-standout",
416
u"default", u"default", (u"underline", u"standout")),
417
(u"bold-underline-blink-standout",
418
u"default", u"default", (u"bold", u"underline",
447
"default", "default", None),
449
"default", "default", "bold"),
451
"default", "default", "underline"),
453
"default", "default", "standout"),
454
("bold-underline-blink",
455
"default", "default", ("bold", "underline")),
457
"default", "default", ("bold", "standout")),
458
("underline-blink-standout",
459
"default", "default", ("underline", "standout")),
460
("bold-underline-blink-standout",
461
"default", "default", ("bold", "underline",
422
465
if urwid.supports_unicode():
423
self.divider = u"─" # \u2500
424
#self.divider = u"━" # \u2501
466
self.divider = "─" # \u2500
467
#self.divider = "━" # \u2501
426
#self.divider = u"-" # \u002d
427
self.divider = u"_" # \u005f
469
#self.divider = "-" # \u002d
470
self.divider = "_" # \u005f
429
472
self.screen.start()
444
487
# This keeps track of whether self.uilist currently has
445
488
# self.logbox in it or not
446
489
self.log_visible = True
447
self.log_wrap = u"any"
490
self.log_wrap = "any"
450
self.log_message_raw((u"bold",
451
u"Mandos Monitor version " + version))
452
self.log_message_raw((u"bold",
493
self.log_message_raw(("bold",
494
"Mandos Monitor version " + version))
495
self.log_message_raw(("bold",
455
498
self.busname = domain + '.Mandos'
456
499
self.main_loop = gobject.MainLoop()
457
500
self.bus = dbus.SystemBus()
458
501
mandos_dbus_objc = self.bus.get_object(
459
self.busname, u"/", follow_name_owner_changes=True)
502
self.busname, "/", follow_name_owner_changes=True)
460
503
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
462
505
= server_interface)
467
510
mandos_clients = dbus.Dictionary()
469
512
(self.mandos_serv
470
.connect_to_signal(u"ClientRemoved",
513
.connect_to_signal("ClientRemoved",
471
514
self.find_and_remove_client,
472
515
dbus_interface=server_interface,
473
516
byte_arrays=True))
474
517
(self.mandos_serv
475
.connect_to_signal(u"ClientAdded",
518
.connect_to_signal("ClientAdded",
476
519
self.add_new_client,
477
520
dbus_interface=server_interface,
478
521
byte_arrays=True))
479
522
(self.mandos_serv
480
.connect_to_signal(u"ClientNotFound",
523
.connect_to_signal("ClientNotFound",
481
524
self.client_not_found,
482
525
dbus_interface=server_interface,
483
526
byte_arrays=True))
530
573
and len(self.log) > self.max_log_length):
531
574
del self.log[0:len(self.log)-self.max_log_length-1]
532
575
self.logbox.set_focus(len(self.logbox.body.contents),
533
coming_from=u"above")
536
579
def toggle_log_display(self):
537
580
"""Toggle visibility of the log buffer."""
538
581
self.log_visible = not self.log_visible
540
#self.log_message(u"Log visibility changed to: "
583
#self.log_message("Log visibility changed to: "
541
584
# + unicode(self.log_visible))
543
586
def change_log_display(self):
544
587
"""Change type of log display.
545
588
Currently, this toggles wrapping of text lines."""
546
if self.log_wrap == u"clip":
547
self.log_wrap = u"any"
589
if self.log_wrap == "clip":
590
self.log_wrap = "any"
549
self.log_wrap = u"clip"
592
self.log_wrap = "clip"
550
593
for textwidget in self.log:
551
594
textwidget.set_wrap_mode(self.log_wrap)
552
#self.log_message(u"Wrap mode: " + self.log_wrap)
595
#self.log_message("Wrap mode: " + self.log_wrap)
554
597
def find_and_remove_client(self, path, name):
555
"""Find an client from its object path and remove it.
598
"""Find a client by its object path and remove it.
557
600
This is connected to the ClientRemoved signal from the
558
601
Mandos server object."""
620
665
def process_input(self, source, condition):
621
666
keys = self.screen.get_input()
622
translations = { u"ctrl n": u"down", # Emacs
623
u"ctrl p": u"up", # Emacs
624
u"ctrl v": u"page down", # Emacs
625
u"meta v": u"page up", # Emacs
626
u" ": u"page down", # less
627
u"f": u"page down", # less
628
u"b": u"page up", # less
667
translations = { "ctrl n": "down", # Emacs
668
"ctrl p": "up", # Emacs
669
"ctrl v": "page down", # Emacs
670
"meta v": "page up", # Emacs
671
" ": "page down", # less
672
"f": "page down", # less
673
"b": "page up", # less
635
680
except KeyError: # :-)
638
if key == u"q" or key == u"Q":
683
if key == "q" or key == "Q":
641
elif key == u"window resize":
686
elif key == "window resize":
642
687
self.size = self.screen.get_cols_rows()
644
elif key == u"\f": # Ctrl-L
689
elif key == "\f": # Ctrl-L
646
elif key == u"l" or key == u"D":
691
elif key == "l" or key == "D":
647
692
self.toggle_log_display()
649
elif key == u"w" or key == u"i":
694
elif key == "w" or key == "i":
650
695
self.change_log_display()
652
elif key == u"?" or key == u"f1" or key == u"esc":
697
elif key == "?" or key == "f1" or key == "esc":
653
698
if not self.log_visible:
654
699
self.log_visible = True
656
self.log_message_raw((u"bold",
660
u"l: Log window toggle",
661
u"TAB: Switch window",
663
self.log_message_raw((u"bold",
669
u"s: Start new checker",
701
self.log_message_raw(("bold",
705
"l: Log window toggle",
706
"TAB: Switch window",
708
self.log_message_raw(("bold",
714
"s: Start new checker",
676
721
if self.topwidget.get_focus() is self.logbox:
677
722
self.topwidget.set_focus(0)
679
724
self.topwidget.set_focus(self.logbox)
681
#elif (key == u"end" or key == u"meta >" or key == u"G"
726
#elif (key == "end" or key == "meta >" or key == "G"
683
728
# pass # xxx end-of-buffer
684
#elif (key == u"home" or key == u"meta <" or key == u"g"
729
#elif (key == "home" or key == "meta <" or key == "g"
686
731
# pass # xxx beginning-of-buffer
687
#elif key == u"ctrl e" or key == u"$":
732
#elif key == "ctrl e" or key == "$":
688
733
# pass # xxx move-end-of-line
689
#elif key == u"ctrl a" or key == u"^":
734
#elif key == "ctrl a" or key == "^":
690
735
# pass # xxx move-beginning-of-line
691
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
736
#elif key == "ctrl b" or key == "meta (" or key == "h":
692
737
# pass # xxx left
693
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
738
#elif key == "ctrl f" or key == "meta )" or key == "l":
694
739
# pass # xxx right
696
741
# pass # scroll up log
698
743
# pass # scroll down log
699
744
elif self.topwidget.selectable():
700
745
self.topwidget.keypress(self.size, key)