17
17
# GNU General Public License for more details.
19
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
from __future__ import division, absolute_import, print_function, unicode_literals
20
# along with this program. If not, see
21
# <http://www.gnu.org/licenses/>.
23
# Contact the authors at <mandos@recompile.se>.
26
from __future__ import (division, absolute_import, print_function,
29
from future_builtins import *
82
86
properties and calls a hook function when any of them are
85
def __init__(self, proxy_object=None, *args, **kwargs):
89
def __init__(self, proxy_object=None, properties=None, **kwargs):
86
90
self.proxy = proxy_object # Mandos Client proxy object
88
self.properties = dict()
89
self.proxy.connect_to_signal("PropertyChanged",
90
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__(
91
self.properties = dict() if properties is None else properties
92
self.property_changed_match = (
93
self.proxy.connect_to_signal("PropertyChanged",
94
self._property_changed,
98
if properties is None:
99
self.properties.update(
100
self.proxy.GetAll(client_interface,
102
= dbus.PROPERTIES_IFACE))
104
super(MandosClientPropertyCache, self).__init__(**kwargs)
106
def _property_changed(self, property, value):
107
"""Helper which takes positional arguments"""
108
return self.property_changed(property=property, value=value)
102
110
def property_changed(self, property=None, value=None):
103
111
"""This is called whenever we get a PropertyChanged signal
125
136
self._update_timer_callback_tag = None
126
137
self._update_timer_callback_lock = 0
127
self.last_checker_failed = False
129
139
# The widget shown normally
130
140
self._text_widget = urwid.Text("")
131
141
# The widget shown when we have focus
132
142
self._focus_text_widget = urwid.Text("")
133
super(MandosClientWidget, self).__init__(
134
update_hook=update_hook, delete_hook=delete_hook,
143
super(MandosClientWidget, self).__init__(**kwargs)
137
145
self.opened = False
139
147
last_checked_ok = isoformat_to_datetime(self.properties
140
148
["LastCheckedOK"])
141
if last_checked_ok is None:
142
self.last_checker_failed = True
144
self.last_checker_failed = ((datetime.datetime.utcnow()
151
if self.last_checker_failed:
150
if self.properties ["LastCheckerStatus"] != 0:
152
151
self.using_timer(True)
154
153
if self.need_approval:
155
154
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",
156
self.match_objects = (
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",
177
#self.logger('Created client {0}'
178
# .format(self.properties["Name"]))
178
180
def property_changed(self, property=None, value=None):
179
181
super(self, MandosClientWidget).property_changed(property,
181
183
if property == "ApprovalPending":
182
184
using_timer(bool(value))
185
if property == "LastCheckerStatus":
186
using_timer(value != 0)
187
#self.logger('Checker for client {0} (command "{1}") was '
188
# ' successful'.format(self.properties["Name"],
184
191
def using_timer(self, flag):
185
192
"""Call this method with True or False when timer should be
186
193
activated or deactivated.
201
209
def checker_completed(self, exitstatus, condition, command):
202
210
if exitstatus == 0:
203
if self.last_checker_failed:
204
self.last_checker_failed = False
205
self.using_timer(False)
206
#self.logger('Checker for client %s (command "%s")'
208
# % (self.properties["Name"], command))
212
if not self.last_checker_failed:
213
self.last_checker_failed = True
214
self.using_timer(True)
215
214
if os.WIFEXITED(condition):
216
self.logger('Checker for client %s (command "%s")'
217
' failed with exit code %s'
218
% (self.properties["Name"], command,
219
os.WEXITSTATUS(condition)))
215
self.logger('Checker for client {0} (command "{1}")'
216
' failed with exit code {2}'
217
.format(self.properties["Name"], command,
218
os.WEXITSTATUS(condition)))
220
219
elif os.WIFSIGNALED(condition):
221
self.logger('Checker for client %s (command "%s")'
222
' was killed by signal %s'
223
% (self.properties["Name"], command,
224
os.WTERMSIG(condition)))
220
self.logger('Checker for client {0} (command "{1}") was'
221
' killed by signal {2}'
222
.format(self.properties["Name"], command,
223
os.WTERMSIG(condition)))
225
224
elif os.WCOREDUMP(condition):
226
self.logger('Checker for client %s (command "%s")'
225
self.logger('Checker for client {0} (command "{1}")'
228
% (self.properties["Name"], command))
227
.format(self.properties["Name"], command))
230
self.logger('Checker for client %s completed'
229
self.logger('Checker for client {0} completed'
231
.format(self.properties["Name"]))
234
234
def checker_started(self, command):
235
#self.logger('Client %s started checker "%s"'
236
# % (self.properties["Name"], unicode(command)))
235
"""Server signals that a checker started. This could be useful
236
to log in the future. """
237
#self.logger('Client {0} started checker "{1}"'
238
# .format(self.properties["Name"],
239
242
def got_secret(self):
240
self.last_checker_failed = False
241
self.logger('Client %s received its secret'
242
% self.properties["Name"])
243
self.logger('Client {0} received its secret'
244
.format(self.properties["Name"]))
244
246
def need_approval(self, timeout, default):
246
message = 'Client %s needs approval within %s seconds'
248
message = 'Client {0} needs approval within {1} seconds'
248
message = 'Client %s will get its secret in %s seconds'
250
% (self.properties["Name"], timeout/1000))
250
message = 'Client {0} will get its secret in {1} seconds'
251
self.logger(message.format(self.properties["Name"],
251
253
self.using_timer(True)
253
255
def rejected(self, reason):
254
self.logger('Client %s was rejected; reason: %s'
255
% (self.properties["Name"], reason))
256
self.logger('Client {0} was rejected; reason: {1}'
257
.format(self.properties["Name"], reason))
257
259
def selectable(self):
258
260
"""Make this a "selectable" widget.
301
302
timer = datetime.timedelta()
302
303
if self.properties["ApprovedByDefault"]:
303
message = "Approval in %s. (d)eny?"
305
message = "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["LastCheckedOK"]
313
or self.properties["Created"]),
314
self.properties["LastEnabled"]))
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
304
message = "Approval in {0}. (d)eny?"
306
message = "Denial in {0}. (a)pprove?"
307
message = message.format(unicode(timer).rsplit(".", 1)[0])
308
elif self.properties["LastCheckerStatus"] != 0:
309
# When checker has failed, show timer until client expires
310
expires = self.properties["Expires"]
312
timer = datetime.timedelta(0)
314
expires = (datetime.datetime.strptime
315
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
316
timer = expires - datetime.datetime.utcnow()
316
317
message = ('A checker has failed! Time until client'
318
% unicode(timer).rsplit(".", 1)[0])
318
' gets disabled: {0}'
319
.format(unicode(timer).rsplit(".", 1)[0]))
320
321
message = "enabled"
321
self._text = "%s%s" % (base, message)
322
self._text = "{0}{1}".format(base, message)
323
324
if not urwid.supports_unicode():
324
325
self._text = self._text.encode("ascii", "replace")
337
338
self.update_hook()
339
340
def update_timer(self):
341
"""called by gobject. Will indefinitely loop until
342
gobject.source_remove() on tag is called"""
342
344
return True # Keep calling this
346
def delete(self, **kwargs):
345
347
if self._update_timer_callback_tag is not None:
346
348
gobject.source_remove(self._update_timer_callback_tag)
347
349
self._update_timer_callback_tag = None
350
for match in self.match_objects:
352
self.match_objects = ()
348
353
if self.delete_hook is not None:
349
354
self.delete_hook(self)
355
return super(MandosClientWidget, self).delete(**kwargs)
351
357
def render(self, maxcolrow, focus=False):
352
358
"""Render differently if we have focus.
359
365
This overrides the method from urwid.FlowWidget"""
361
self.proxy.Enable(dbus_interface = client_interface)
367
self.proxy.Enable(dbus_interface = client_interface,
363
self.proxy.Disable(dbus_interface = client_interface)
370
self.proxy.Disable(dbus_interface = client_interface,
365
373
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
366
dbus_interface = client_interface)
374
dbus_interface = client_interface,
368
377
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
369
dbus_interface = client_interface)
378
dbus_interface = client_interface,
370
380
elif key == "R" or key == "_" or key == "ctrl k":
371
381
self.server_proxy_object.RemoveClient(self.proxy
374
self.proxy.StartChecker(dbus_interface = client_interface)
385
self.proxy.StartChecker(dbus_interface = client_interface,
376
self.proxy.StopChecker(dbus_interface = client_interface)
388
self.proxy.StopChecker(dbus_interface = client_interface,
378
self.proxy.CheckedOK(dbus_interface = client_interface)
391
self.proxy.CheckedOK(dbus_interface = client_interface,
380
394
# elif key == "p" or key == "=":
381
395
# self.proxy.pause()
389
def property_changed(self, property=None, value=None,
403
def property_changed(self, property=None, **kwargs):
391
404
"""Call self.update() if old value is not new value.
392
405
This overrides the method from MandosClientPropertyCache"""
393
406
property_name = unicode(property)
394
407
old_value = self.properties.get(property_name)
395
408
super(MandosClientWidget, self).property_changed(
396
property=property, value=value, *args, **kwargs)
409
property=property, **kwargs)
397
410
if self.properties.get(property_name) != old_value:
403
416
"down" key presses, thus not allowing any containing widgets to
404
417
use them as an excuse to shift focus away from this widget.
406
def keypress(self, maxcolrow, key):
407
ret = super(ConstrainedListBox, self).keypress(maxcolrow, key)
419
def keypress(self, *args, **kwargs):
420
ret = super(ConstrainedListBox, self).keypress(*args, **kwargs)
408
421
if ret in ("up", "down"):
475
488
self.busname = domain + '.Mandos'
476
489
self.main_loop = gobject.MainLoop()
477
self.bus = dbus.SystemBus()
478
mandos_dbus_objc = self.bus.get_object(
479
self.busname, "/", follow_name_owner_changes=True)
480
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
484
mandos_clients = (self.mandos_serv
485
.GetAllClientsWithProperties())
486
except dbus.exceptions.DBusException:
487
mandos_clients = dbus.Dictionary()
490
.connect_to_signal("ClientRemoved",
491
self.find_and_remove_client,
492
dbus_interface=server_interface,
495
.connect_to_signal("ClientAdded",
497
dbus_interface=server_interface,
500
.connect_to_signal("ClientNotFound",
501
self.client_not_found,
502
dbus_interface=server_interface,
504
for path, client in mandos_clients.iteritems():
505
client_proxy_object = self.bus.get_object(self.busname,
507
self.add_client(MandosClientWidget(server_proxy_object
510
=client_proxy_object,
520
491
def client_not_found(self, fingerprint, address):
521
self.log_message(("Client with address %s and fingerprint %s"
522
" could not be found" % (address,
492
self.log_message("Client with address {0} and fingerprint"
493
" {1} could not be found"
494
.format(address, fingerprint))
525
496
def rebuild(self):
526
497
"""This rebuilds the User Interface.
626
598
"""Start the main loop and exit when it's done."""
599
self.bus = dbus.SystemBus()
600
mandos_dbus_objc = self.bus.get_object(
601
self.busname, "/", follow_name_owner_changes=True)
602
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
606
mandos_clients = (self.mandos_serv
607
.GetAllClientsWithProperties())
608
except dbus.exceptions.DBusException:
609
mandos_clients = dbus.Dictionary()
612
.connect_to_signal("ClientRemoved",
613
self.find_and_remove_client,
614
dbus_interface=server_interface,
617
.connect_to_signal("ClientAdded",
619
dbus_interface=server_interface,
622
.connect_to_signal("ClientNotFound",
623
self.client_not_found,
624
dbus_interface=server_interface,
626
for path, client in mandos_clients.iteritems():
627
client_proxy_object = self.bus.get_object(self.busname,
629
self.add_client(MandosClientWidget(server_proxy_object
632
=client_proxy_object,
628
643
self._input_callback_tag = (gobject.io_add_watch
629
644
(sys.stdin.fileno(),