4
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2011 Teddy Hogeborn
7
# Copyright © 2009-2011 Björn Påhlsson
6
# Copyright © 2009-2012 Teddy Hogeborn
7
# Copyright © 2009-2012 Björn Påhlsson
9
9
# This program is free software: you can redistribute it and/or modify
10
10
# it under the terms of the GNU General Public License as published by
19
19
# You should have received a copy of the GNU General Public License
20
20
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
# Contact the authors at <mandos@fukt.bsnet.se>.
22
# Contact the authors at <mandos@recompile.se>.
25
from __future__ import division, absolute_import, print_function, unicode_literals
25
from __future__ import (division, absolute_import, print_function,
48
49
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
50
51
# Some useful constants
51
domain = 'se.bsnet.fukt'
52
domain = 'se.recompile'
52
53
server_interface = domain + '.Mandos'
53
54
client_interface = domain + '.Mandos.Client'
56
57
# Always run in monochrome mode
57
58
urwid.curses_display.curses.has_colors = lambda : False
86
87
self.proxy = proxy_object # Mandos Client proxy object
88
89
self.properties = dict()
89
self.proxy.connect_to_signal("PropertyChanged",
90
self.property_changed,
90
self.property_changed_match = (
91
self.proxy.connect_to_signal("PropertyChanged",
92
self.property_changed,
94
96
self.properties.update(
95
97
self.proxy.GetAll(client_interface,
96
98
dbus_interface = dbus.PROPERTIES_IFACE))
98
#XXX This break good super behaviour!
100
#XXX This breaks good super behaviour
99
101
# super(MandosClientPropertyCache, self).__init__(
100
102
# *args, **kwargs)
106
108
# Update properties dict with new value
107
109
self.properties[property] = value
111
def delete(self, *args, **kwargs):
112
self.property_changed_match.remove()
113
super(MandosClientPropertyCache, self).__init__(
110
117
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
154
161
if self.need_approval:
155
162
self.using_timer(True)
157
self.proxy.connect_to_signal("CheckerCompleted",
158
self.checker_completed,
161
self.proxy.connect_to_signal("CheckerStarted",
162
self.checker_started,
165
self.proxy.connect_to_signal("GotSecret",
169
self.proxy.connect_to_signal("NeedApproval",
173
self.proxy.connect_to_signal("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"]))
178
187
def property_changed(self, property=None, value=None):
179
188
super(self, MandosClientWidget).property_changed(property,
234
244
def checker_started(self, command):
245
"""Server signals that a checker started. This could be useful
246
to log in the future. """
235
247
#self.logger('Client %s started checker "%s"'
236
248
# % (self.properties["Name"], unicode(command)))
305
317
message = "Denial in %s. (a)pprove?"
306
318
message = message % unicode(timer).rsplit(".", 1)[0]
307
319
elif self.last_checker_failed:
308
timeout = datetime.timedelta(milliseconds
311
last_ok = isoformat_to_datetime(
312
max((self.properties["LastCheckedOK"]
313
or self.properties["Created"]),
314
self.properties["LastEnabled"]))
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
320
# When checker has failed, print a timer until client expires
321
expires = self.properties["Expires"]
323
timer = datetime.timedelta(0)
325
expires = datetime.datetime.strptime(expires,
326
'%Y-%m-%dT%H:%M:%S.%f')
327
timer = expires - datetime.datetime.utcnow()
316
328
message = ('A checker has failed! Time until client'
317
329
' gets disabled: %s'
318
330
% unicode(timer).rsplit(".", 1)[0])
337
349
self.update_hook()
339
351
def update_timer(self):
352
"""called by gobject. Will indefinitely loop until
353
gobject.source_remove() on tag is called"""
342
355
return True # Keep calling this
357
def delete(self, *args, **kwargs):
345
358
if self._update_timer_callback_tag is not None:
346
359
gobject.source_remove(self._update_timer_callback_tag)
347
360
self._update_timer_callback_tag = None
361
for match in self.match_objects:
363
self.match_objects = ()
348
364
if self.delete_hook is not None:
349
365
self.delete_hook(self)
366
return super(MandosClientWidget, self).delete(*args, **kwargs)
351
368
def render(self, maxcolrow, focus=False):
352
369
"""Render differently if we have focus.
483
500
self.busname = domain + '.Mandos'
484
501
self.main_loop = gobject.MainLoop()
485
self.bus = dbus.SystemBus()
486
mandos_dbus_objc = self.bus.get_object(
487
self.busname, "/", follow_name_owner_changes=True)
488
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
492
mandos_clients = (self.mandos_serv
493
.GetAllClientsWithProperties())
494
except dbus.exceptions.DBusException:
495
mandos_clients = dbus.Dictionary()
498
.connect_to_signal("ClientRemoved",
499
self.find_and_remove_client,
500
dbus_interface=server_interface,
503
.connect_to_signal("ClientAdded",
505
dbus_interface=server_interface,
508
.connect_to_signal("ClientNotFound",
509
self.client_not_found,
510
dbus_interface=server_interface,
512
for path, client in mandos_clients.iteritems():
513
client_proxy_object = self.bus.get_object(self.busname,
515
self.add_client(MandosClientWidget(server_proxy_object
518
=client_proxy_object,
528
503
def client_not_found(self, fingerprint, address):
529
504
self.log_message(("Client with address %s and fingerprint %s"
580
554
#self.log_message("Wrap mode: " + self.log_wrap)
582
556
def find_and_remove_client(self, path, name):
583
"""Find an client from its object path and remove it.
557
"""Find a client by its object path and remove it.
585
559
This is connected to the ClientRemoved signal from the
586
560
Mandos server object."""
588
562
client = self.clients_dict[path]
565
self.log_message("Unknown client %r (%r) removed", name,
592
self.remove_client(client, path)
594
570
def add_new_client(self, path):
595
571
client_proxy_object = self.bus.get_object(self.busname, path)
634
610
"""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,
618
mandos_clients = (self.mandos_serv
619
.GetAllClientsWithProperties())
620
except dbus.exceptions.DBusException:
621
mandos_clients = dbus.Dictionary()
624
.connect_to_signal("ClientRemoved",
625
self.find_and_remove_client,
626
dbus_interface=server_interface,
629
.connect_to_signal("ClientAdded",
631
dbus_interface=server_interface,
634
.connect_to_signal("ClientNotFound",
635
self.client_not_found,
636
dbus_interface=server_interface,
638
for path, client in mandos_clients.iteritems():
639
client_proxy_object = self.bus.get_object(self.busname,
641
self.add_client(MandosClientWidget(server_proxy_object
644
=client_proxy_object,
636
655
self._input_callback_tag = (gobject.io_add_watch
637
656
(sys.stdin.fileno(),