4
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2015 Teddy Hogeborn
7
# Copyright © 2009-2015 Björn Påhlsson
6
# Copyright © 2009-2014 Teddy Hogeborn
7
# Copyright © 2009-2014 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
87
87
self.proxy = proxy_object # Mandos Client proxy object
88
88
self.properties = dict() if properties is None else properties
89
89
self.property_changed_match = (
90
self.proxy.connect_to_signal("PropertiesChanged",
91
self.properties_changed,
92
dbus.PROPERTIES_IFACE,
90
self.proxy.connect_to_signal("PropertyChanged",
91
self._property_changed,
95
95
if properties is None:
101
101
super(MandosClientPropertyCache, self).__init__(**kwargs)
103
def properties_changed(self, interface, properties, invalidated):
104
"""This is called whenever we get a PropertiesChanged signal
105
It updates the changed properties in the "properties" dict.
103
def _property_changed(self, property, value):
104
"""Helper which takes positional arguments"""
105
return self.property_changed(property=property, value=value)
107
def property_changed(self, property=None, value=None):
108
"""This is called whenever we get a PropertyChanged signal
109
It updates the changed property in the "properties" dict.
107
111
# Update properties dict with new value
108
self.properties.update(properties)
112
self.properties[property] = value
110
114
def delete(self):
111
115
self.property_changed_match.remove()
158
162
client_interface,
159
163
byte_arrays=True))
160
self.logger('Created client {}'
161
.format(self.properties["Name"]), level=0)
164
#self.logger('Created client {0}'
165
# .format(self.properties["Name"]))
163
167
def using_timer(self, flag):
164
168
"""Call this method with True or False when timer should be
176
180
def checker_completed(self, exitstatus, condition, command):
177
181
if exitstatus == 0:
178
self.logger('Checker for client {} (command "{}")'
179
' succeeded'.format(self.properties["Name"],
184
185
if os.WIFEXITED(condition):
185
self.logger('Checker for client {} (command "{}") failed'
186
self.logger('Checker for client {0} (command "{1}")'
187
' failed with exit code {2}'
187
188
.format(self.properties["Name"], command,
188
189
os.WEXITSTATUS(condition)))
189
190
elif os.WIFSIGNALED(condition):
190
self.logger('Checker for client {} (command "{}") was'
191
' killed by signal {}'
191
self.logger('Checker for client {0} (command "{1}") was'
192
' killed by signal {2}'
192
193
.format(self.properties["Name"], command,
193
194
os.WTERMSIG(condition)))
195
elif os.WCOREDUMP(condition):
196
self.logger('Checker for client {0} (command "{1}")'
198
.format(self.properties["Name"], command))
200
self.logger('Checker for client {0} completed'
202
.format(self.properties["Name"]))
196
205
def checker_started(self, command):
197
"""Server signals that a checker started."""
198
self.logger('Client {} started checker "{}"'
199
.format(self.properties["Name"],
206
"""Server signals that a checker started. This could be useful
207
to log in the future. """
208
#self.logger('Client {0} started checker "{1}"'
209
# .format(self.properties["Name"],
202
213
def got_secret(self):
203
self.logger('Client {} received its secret'
214
self.logger('Client {0} received its secret'
204
215
.format(self.properties["Name"]))
206
217
def need_approval(self, timeout, default):
208
message = 'Client {} needs approval within {} seconds'
219
message = 'Client {0} needs approval within {1} seconds'
210
message = 'Client {} will get its secret in {} seconds'
221
message = 'Client {0} will get its secret in {1} seconds'
211
222
self.logger(message.format(self.properties["Name"],
214
225
def rejected(self, reason):
215
self.logger('Client {} was rejected; reason: {}'
226
self.logger('Client {0} was rejected; reason: {1}'
216
227
.format(self.properties["Name"], reason))
218
229
def selectable(self):
263
274
timer = datetime.timedelta()
264
275
if self.properties["ApprovedByDefault"]:
265
message = "Approval in {}. (d)eny?"
276
message = "Approval in {0}. (d)eny?"
267
message = "Denial in {}. (a)pprove?"
278
message = "Denial in {0}. (a)pprove?"
268
279
message = message.format(str(timer).rsplit(".", 1)[0])
269
280
self.using_timer(True)
270
281
elif self.properties["LastCheckerStatus"] != 0:
278
289
timer = max(expires - datetime.datetime.utcnow(),
279
290
datetime.timedelta())
280
291
message = ('A checker has failed! Time until client'
292
' gets disabled: {0}'
282
293
.format(str(timer).rsplit(".", 1)[0]))
283
294
self.using_timer(True)
285
296
message = "enabled"
286
297
self.using_timer(False)
287
self._text = "{}{}".format(base, message)
298
self._text = "{0}{1}".format(base, message)
289
300
if not urwid.supports_unicode():
290
301
self._text = self._text.encode("ascii", "replace")
330
341
This overrides the method from urwid.FlowWidget"""
332
self.proxy.Set(client_interface, "Enabled",
333
dbus.Boolean(True), ignore_reply = True,
334
dbus_interface = dbus.PROPERTIES_IFACE)
343
self.proxy.Enable(dbus_interface = client_interface,
336
self.proxy.Set(client_interface, "Enabled", False,
338
dbus_interface = dbus.PROPERTIES_IFACE)
346
self.proxy.Disable(dbus_interface = client_interface,
340
349
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
341
350
dbus_interface = client_interface,
350
359
ignore_reply=True)
352
self.proxy.Set(client_interface, "CheckerRunning",
353
dbus.Boolean(True), ignore_reply = True,
354
dbus_interface = dbus.PROPERTIES_IFACE)
361
self.proxy.StartChecker(dbus_interface = client_interface,
356
self.proxy.Set(client_interface, "CheckerRunning",
357
dbus.Boolean(False), ignore_reply = True,
358
dbus_interface = dbus.PROPERTIES_IFACE)
364
self.proxy.StopChecker(dbus_interface = client_interface,
360
367
self.proxy.CheckedOK(dbus_interface = client_interface,
361
368
ignore_reply=True)
372
def properties_changed(self, interface, properties, invalidated):
373
"""Call self.update() if any properties changed.
379
def property_changed(self, property=None, **kwargs):
380
"""Call self.update() if old value is not new value.
374
381
This overrides the method from MandosClientPropertyCache"""
375
old_values = { key: self.properties.get(key)
376
for key in properties.keys() }
377
super(MandosClientWidget, self).properties_changed(
378
interface, properties, invalidated)
379
if any(old_values[key] != self.properties.get(key)
380
for key in old_values):
382
property_name = str(property)
383
old_value = self.properties.get(property_name)
384
super(MandosClientWidget, self).property_changed(
385
property=property, **kwargs)
386
if self.properties.get(property_name) != old_value:
397
403
"""This is the entire user interface - the whole screen
398
404
with boxes, lists of client widgets, etc.
400
def __init__(self, max_log_length=1000, log_level=1):
406
def __init__(self, max_log_length=1000):
401
407
DBusGMainLoop(set_as_default=True)
403
409
self.screen = urwid.curses_display.Screen()
462
466
self.main_loop = gobject.MainLoop()
464
468
def client_not_found(self, fingerprint, address):
465
self.log_message("Client with address {} and fingerprint {}"
466
" could not be found"
469
self.log_message("Client with address {0} and fingerprint"
470
" {1} could not be found"
467
471
.format(address, fingerprint))
469
473
def rebuild(self):
482
486
self.uilist.append(self.logbox)
483
487
self.topwidget = urwid.Pile(self.uilist)
485
def log_message(self, message, level=1):
489
def log_message(self, message):
486
490
"""Log message formatted with timestamp"""
487
if level < self.log_level:
489
491
timestamp = datetime.datetime.now().isoformat()
490
self.log_message_raw("{}: {}".format(timestamp, message),
492
self.log_message_raw(timestamp + ": " + message)
493
def log_message_raw(self, markup, level=1):
494
def log_message_raw(self, markup):
494
495
"""Add a log message to the log buffer."""
495
if level < self.log_level:
497
496
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
498
497
if (self.max_log_length
499
498
and len(self.log) > self.max_log_length):
506
505
"""Toggle visibility of the log buffer."""
507
506
self.log_visible = not self.log_visible
509
self.log_message("Log visibility changed to: {}"
510
.format(self.log_visible), level=0)
508
#self.log_message("Log visibility changed to: "
509
# + str(self.log_visible))
512
511
def change_log_display(self):
513
512
"""Change type of log display.
518
517
self.log_wrap = "clip"
519
518
for textwidget in self.log:
520
519
textwidget.set_wrap_mode(self.log_wrap)
521
self.log_message("Wrap mode: {}".format(self.log_wrap),
520
#self.log_message("Wrap mode: " + self.log_wrap)
524
522
def find_and_remove_client(self, path, name):
525
523
"""Find a client by its object path and remove it.