133
102
self.logger = logger
135
104
self._update_timer_callback_tag = None
136
self._update_timer_callback_lock = 0
105
self.last_checker_failed = False
138
107
# The widget shown normally
139
self._text_widget = urwid.Text("")
108
self._text_widget = urwid.Text(u"")
140
109
# The widget shown when we have focus
141
self._focus_text_widget = urwid.Text("")
110
self._focus_text_widget = urwid.Text(u"")
142
111
super(MandosClientWidget, self).__init__(
143
112
update_hook=update_hook, delete_hook=delete_hook,
146
115
self.opened = False
116
self.proxy.connect_to_signal(u"CheckerCompleted",
117
self.checker_completed,
120
self.proxy.connect_to_signal(u"CheckerStarted",
121
self.checker_started,
124
self.proxy.connect_to_signal(u"GotSecret",
128
self.proxy.connect_to_signal(u"NeedApproval",
132
self.proxy.connect_to_signal(u"Rejected",
148
136
last_checked_ok = isoformat_to_datetime(self.properties
151
if self.properties ["LastCheckerStatus"] != 0:
152
self.using_timer(True)
154
if self.need_approval:
155
self.using_timer(True)
157
self.match_objects = (
158
self.proxy.connect_to_signal("CheckerCompleted",
159
self.checker_completed,
162
self.proxy.connect_to_signal("CheckerStarted",
163
self.checker_started,
166
self.proxy.connect_to_signal("GotSecret",
170
self.proxy.connect_to_signal("NeedApproval",
174
self.proxy.connect_to_signal("Rejected",
178
#self.logger('Created client {0}'
179
# .format(self.properties["Name"]))
181
def property_changed(self, property=None, value=None):
182
super(self, MandosClientWidget).property_changed(property,
184
if property == "ApprovalPending":
185
using_timer(bool(value))
186
if property == "LastCheckerStatus":
187
using_timer(value != 0)
188
#self.logger('Checker for client {0} (command "{1}") was '
189
# ' successful'.format(self.properties["Name"],
192
def using_timer(self, flag):
193
"""Call this method with True or False when timer should be
194
activated or deactivated.
196
old = self._update_timer_callback_lock
198
self._update_timer_callback_lock += 1
138
if last_checked_ok is None:
139
self.last_checker_failed = True
200
self._update_timer_callback_lock -= 1
201
if old == 0 and self._update_timer_callback_lock:
202
# Will update the shown timer value every second
141
self.last_checker_failed = ((datetime.datetime.utcnow()
147
if self.last_checker_failed:
203
148
self._update_timer_callback_tag = (gobject.timeout_add
205
150
self.update_timer))
206
elif old and self._update_timer_callback_lock == 0:
207
gobject.source_remove(self._update_timer_callback_tag)
208
self._update_timer_callback_tag = None
210
152
def checker_completed(self, exitstatus, condition, command):
211
153
if exitstatus == 0:
154
if self.last_checker_failed:
155
self.last_checker_failed = False
156
gobject.source_remove(self._update_timer_callback_tag)
157
self._update_timer_callback_tag = None
158
#self.logger(u'Checker for client %s (command "%s")'
160
# % (self.properties[u"Name"], command))
164
if not self.last_checker_failed:
165
self.last_checker_failed = True
166
self._update_timer_callback_tag = (gobject.timeout_add
215
169
if os.WIFEXITED(condition):
216
self.logger('Checker for client {0} (command "{1}")'
217
' failed with exit code {2}'
218
.format(self.properties["Name"], command,
219
os.WEXITSTATUS(condition)))
170
self.logger(u'Checker for client %s (command "%s")'
171
u' failed with exit code %s'
172
% (self.properties[u"Name"], command,
173
os.WEXITSTATUS(condition)))
220
174
elif os.WIFSIGNALED(condition):
221
self.logger('Checker for client {0} (command "{1}") was'
222
' killed by signal {2}'
223
.format(self.properties["Name"], command,
224
os.WTERMSIG(condition)))
175
self.logger(u'Checker for client %s (command "%s")'
176
u' was killed by signal %s'
177
% (self.properties[u"Name"], command,
178
os.WTERMSIG(condition)))
225
179
elif os.WCOREDUMP(condition):
226
self.logger('Checker for client {0} (command "{1}")'
228
.format(self.properties["Name"], command))
180
self.logger(u'Checker for client %s (command "%s")'
182
% (self.properties[u"Name"], command))
230
self.logger('Checker for client {0} completed'
232
.format(self.properties["Name"]))
184
self.logger(u'Checker for client %s completed'
235
188
def checker_started(self, command):
236
"""Server signals that a checker started. This could be useful
237
to log in the future. """
238
#self.logger('Client {0} started checker "{1}"'
239
# .format(self.properties["Name"],
189
#self.logger(u'Client %s started checker "%s"'
190
# % (self.properties[u"Name"], unicode(command)))
243
193
def got_secret(self):
244
self.logger('Client {0} received its secret'
245
.format(self.properties["Name"]))
194
self.last_checker_failed = False
195
self.logger(u'Client %s received its secret'
196
% self.properties[u"Name"])
247
198
def need_approval(self, timeout, default):
249
message = 'Client {0} needs approval within {1} seconds'
200
message = u'Client %s needs approval within %s seconds'
251
message = 'Client {0} will get its secret in {1} seconds'
252
self.logger(message.format(self.properties["Name"],
254
self.using_timer(True)
202
message = u'Client %s will get its secret in %s seconds'
204
% (self.properties[u"Name"], timeout/1000))
256
206
def rejected(self, reason):
257
self.logger('Client {0} was rejected; reason: {1}'
258
.format(self.properties["Name"], reason))
207
self.logger(u'Client %s was rejected; reason: %s'
208
% (self.properties[u"Name"], reason))
260
210
def selectable(self):
261
211
"""Make this a "selectable" widget.
262
212
This overrides the method from urwid.FlowWidget."""
265
def rows(self, maxcolrow, focus=False):
215
def rows(self, (maxcol,), focus=False):
266
216
"""How many rows this widget will occupy might depend on
267
217
whether we have focus or not.
268
218
This overrides the method from urwid.FlowWidget"""
269
return self.current_widget(focus).rows(maxcolrow, focus=focus)
219
return self.current_widget(focus).rows((maxcol,), focus=focus)
271
221
def current_widget(self, focus=False):
272
222
if focus or self.opened:
276
226
def update(self):
277
227
"Called when what is visible on the screen should be updated."
278
228
# How to add standout mode to a style
279
with_standout = { "normal": "standout",
280
"bold": "bold-standout",
282
"underline-blink-standout",
283
"bold-underline-blink":
284
"bold-underline-blink-standout",
229
with_standout = { u"normal": u"standout",
230
u"bold": u"bold-standout",
232
u"underline-blink-standout",
233
u"bold-underline-blink":
234
u"bold-underline-blink-standout",
287
237
# Rebuild focus and non-focus widgets using current properties
289
239
# Base part of a client. Name!
290
base = '{name}: '.format(name=self.properties["Name"])
291
if not self.properties["Enabled"]:
293
elif self.properties["ApprovalPending"]:
240
base = (u'%(name)s: '
241
% {u"name": self.properties[u"Name"]})
242
if not self.properties[u"Enabled"]:
243
message = u"DISABLED"
244
elif self.properties[u"ApprovalPending"]:
245
if self.properties[u"ApprovedByDefault"]:
246
message = u"Connection established to client. (d)eny?"
248
message = u"Seeks approval to send secret. (a)pprove?"
249
elif self.last_checker_failed:
294
250
timeout = datetime.timedelta(milliseconds
295
251
= self.properties
297
last_approval_request = isoformat_to_datetime(
298
self.properties["LastApprovalRequest"])
299
if last_approval_request is not None:
300
timer = timeout - (datetime.datetime.utcnow()
301
- last_approval_request)
303
timer = datetime.timedelta()
304
if self.properties["ApprovedByDefault"]:
305
message = "Approval in {0}. (d)eny?"
307
message = "Denial in {0}. (a)pprove?"
308
message = message.format(unicode(timer).rsplit(".", 1)[0])
309
elif self.properties["LastCheckerStatus"] != 0:
310
# When checker has failed, show timer until client expires
311
expires = self.properties["Expires"]
313
timer = datetime.timedelta(0)
315
expires = (datetime.datetime.strptime
316
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
317
timer = expires - datetime.datetime.utcnow()
318
message = ('A checker has failed! Time until client'
319
' gets disabled: {0}'
320
.format(unicode(timer).rsplit(".", 1)[0]))
253
last_ok = isoformat_to_datetime(
254
max((self.properties[u"LastCheckedOK"]
255
or self.properties[u"Created"]),
256
self.properties[u"LastEnabled"]))
257
timer = timeout - (datetime.datetime.utcnow() - last_ok)
258
message = (u'A checker has failed! Time until client'
260
% unicode(timer).rsplit(".", 1)[0])
323
self._text = "{0}{1}".format(base, message)
263
self._text = "%s%s" % (base, message)
325
265
if not urwid.supports_unicode():
326
266
self._text = self._text.encode("ascii", "replace")
327
textlist = [("normal", self._text)]
267
textlist = [(u"normal", self._text)]
328
268
self._text_widget.set_text(textlist)
329
269
self._focus_text_widget.set_text([(with_standout[text[0]],
339
279
self.update_hook()
341
281
def update_timer(self):
342
"""called by gobject. Will indefinitely loop until
343
gobject.source_remove() on tag is called"""
345
284
return True # Keep calling this
347
def delete(self, *args, **kwargs):
348
287
if self._update_timer_callback_tag is not None:
349
288
gobject.source_remove(self._update_timer_callback_tag)
350
289
self._update_timer_callback_tag = None
351
for match in self.match_objects:
353
self.match_objects = ()
354
290
if self.delete_hook is not None:
355
291
self.delete_hook(self)
356
return super(MandosClientWidget, self).delete(*args, **kwargs)
358
def render(self, maxcolrow, focus=False):
293
def render(self, (maxcol,), focus=False):
359
294
"""Render differently if we have focus.
360
295
This overrides the method from urwid.FlowWidget"""
361
return self.current_widget(focus).render(maxcolrow,
296
return self.current_widget(focus).render((maxcol,),
364
def keypress(self, maxcolrow, key):
299
def keypress(self, (maxcol,), key):
366
301
This overrides the method from urwid.FlowWidget"""
368
self.proxy.Enable(dbus_interface = client_interface,
371
self.proxy.Disable(dbus_interface = client_interface,
303
self.proxy.Enable(dbus_interface = client_interface)
305
self.proxy.Disable(dbus_interface = client_interface)
374
307
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
375
dbus_interface = client_interface,
308
dbus_interface = client_interface)
378
310
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
379
dbus_interface = client_interface,
381
elif key == "R" or key == "_" or key == "ctrl k":
311
dbus_interface = client_interface)
312
elif key == u"r" or key == u"_" or key == u"ctrl k":
382
313
self.server_proxy_object.RemoveClient(self.proxy
386
self.proxy.StartChecker(dbus_interface = client_interface,
389
self.proxy.StopChecker(dbus_interface = client_interface,
392
self.proxy.CheckedOK(dbus_interface = client_interface,
316
self.proxy.StartChecker(dbus_interface = client_interface)
318
self.proxy.StopChecker(dbus_interface = client_interface)
320
self.proxy.CheckedOK(dbus_interface = client_interface)
395
# elif key == "p" or key == "=":
322
# elif key == u"p" or key == "=":
396
323
# self.proxy.pause()
397
# elif key == "u" or key == ":":
324
# elif key == u"u" or key == ":":
398
325
# self.proxy.unpause()
326
# elif key == u"RET":
329
# self.proxy.Approve(True)
331
# self.proxy.Approve(False)
435
367
self.screen = urwid.curses_display.Screen()
437
369
self.screen.register_palette((
439
"default", "default", None),
441
"default", "default", "bold"),
443
"default", "default", "underline"),
445
"default", "default", "standout"),
446
("bold-underline-blink",
447
"default", "default", ("bold", "underline")),
449
"default", "default", ("bold", "standout")),
450
("underline-blink-standout",
451
"default", "default", ("underline", "standout")),
452
("bold-underline-blink-standout",
453
"default", "default", ("bold", "underline",
371
u"default", u"default", None),
373
u"default", u"default", u"bold"),
375
u"default", u"default", u"underline"),
377
u"default", u"default", u"standout"),
378
(u"bold-underline-blink",
379
u"default", u"default", (u"bold", u"underline")),
381
u"default", u"default", (u"bold", u"standout")),
382
(u"underline-blink-standout",
383
u"default", u"default", (u"underline", u"standout")),
384
(u"bold-underline-blink-standout",
385
u"default", u"default", (u"bold", u"underline",
457
389
if urwid.supports_unicode():
458
self.divider = "─" # \u2500
459
#self.divider = "━" # \u2501
390
self.divider = u"─" # \u2500
391
#self.divider = u"━" # \u2501
461
#self.divider = "-" # \u002d
462
self.divider = "_" # \u005f
393
#self.divider = u"-" # \u002d
394
self.divider = u"_" # \u005f
464
396
self.screen.start()
479
411
# This keeps track of whether self.uilist currently has
480
412
# self.logbox in it or not
481
413
self.log_visible = True
482
self.log_wrap = "any"
414
self.log_wrap = u"any"
485
self.log_message_raw(("bold",
486
"Mandos Monitor version " + version))
487
self.log_message_raw(("bold",
417
self.log_message_raw((u"bold",
418
u"Mandos Monitor version " + version))
419
self.log_message_raw((u"bold",
490
422
self.busname = domain + '.Mandos'
491
423
self.main_loop = gobject.MainLoop()
424
self.bus = dbus.SystemBus()
425
mandos_dbus_objc = self.bus.get_object(
426
self.busname, u"/", follow_name_owner_changes=True)
427
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
431
mandos_clients = (self.mandos_serv
432
.GetAllClientsWithProperties())
433
except dbus.exceptions.DBusException:
434
mandos_clients = dbus.Dictionary()
437
.connect_to_signal(u"ClientRemoved",
438
self.find_and_remove_client,
439
dbus_interface=server_interface,
442
.connect_to_signal(u"ClientAdded",
444
dbus_interface=server_interface,
447
.connect_to_signal(u"ClientNotFound",
448
self.client_not_found,
449
dbus_interface=server_interface,
451
for path, client in mandos_clients.iteritems():
452
client_proxy_object = self.bus.get_object(self.busname,
454
self.add_client(MandosClientWidget(server_proxy_object
457
=client_proxy_object,
493
467
def client_not_found(self, fingerprint, address):
494
self.log_message("Client with address {0} and fingerprint"
495
" {1} could not be found"
496
.format(address, fingerprint))
468
self.log_message((u"Client with address %s and fingerprint %s"
469
u" could not be found" % (address,
498
472
def rebuild(self):
499
473
"""This rebuilds the User Interface.
522
497
and len(self.log) > self.max_log_length):
523
498
del self.log[0:len(self.log)-self.max_log_length-1]
524
499
self.logbox.set_focus(len(self.logbox.body.contents),
500
coming_from=u"above")
528
503
def toggle_log_display(self):
529
504
"""Toggle visibility of the log buffer."""
530
505
self.log_visible = not self.log_visible
532
#self.log_message("Log visibility changed to: "
533
# + unicode(self.log_visible))
507
self.log_message(u"Log visibility changed to: "
508
+ unicode(self.log_visible))
535
510
def change_log_display(self):
536
511
"""Change type of log display.
537
512
Currently, this toggles wrapping of text lines."""
538
if self.log_wrap == "clip":
539
self.log_wrap = "any"
513
if self.log_wrap == u"clip":
514
self.log_wrap = u"any"
541
self.log_wrap = "clip"
516
self.log_wrap = u"clip"
542
517
for textwidget in self.log:
543
518
textwidget.set_wrap_mode(self.log_wrap)
544
#self.log_message("Wrap mode: " + self.log_wrap)
519
self.log_message(u"Wrap mode: " + self.log_wrap)
546
521
def find_and_remove_client(self, path, name):
547
"""Find a client by its object path and remove it.
522
"""Find an client from its object path and remove it.
549
524
This is connected to the ClientRemoved signal from the
550
525
Mandos server object."""
600
573
"""Start the main loop and exit when it's done."""
601
self.bus = dbus.SystemBus()
602
mandos_dbus_objc = self.bus.get_object(
603
self.busname, "/", follow_name_owner_changes=True)
604
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
608
mandos_clients = (self.mandos_serv
609
.GetAllClientsWithProperties())
610
except dbus.exceptions.DBusException:
611
mandos_clients = dbus.Dictionary()
614
.connect_to_signal("ClientRemoved",
615
self.find_and_remove_client,
616
dbus_interface=server_interface,
619
.connect_to_signal("ClientAdded",
621
dbus_interface=server_interface,
624
.connect_to_signal("ClientNotFound",
625
self.client_not_found,
626
dbus_interface=server_interface,
628
for path, client in mandos_clients.iteritems():
629
client_proxy_object = self.bus.get_object(self.busname,
631
self.add_client(MandosClientWidget(server_proxy_object
634
=client_proxy_object,
645
575
self._input_callback_tag = (gobject.io_add_watch
646
576
(sys.stdin.fileno(),
672
602
except KeyError: # :-)
675
if key == "q" or key == "Q":
605
if key == u"q" or key == u"Q":
678
elif key == "window resize":
608
elif key == u"window resize":
679
609
self.size = self.screen.get_cols_rows()
681
elif key == "\f": # Ctrl-L
611
elif key == u"\f": # Ctrl-L
683
elif key == "l" or key == "D":
613
elif key == u"l" or key == u"D":
684
614
self.toggle_log_display()
686
elif key == "w" or key == "i":
616
elif key == u"w" or key == u"i":
687
617
self.change_log_display()
689
elif key == "?" or key == "f1" or key == "esc":
619
elif key == u"?" or key == u"f1" or key == u"esc":
690
620
if not self.log_visible:
691
621
self.log_visible = True
693
self.log_message_raw(("bold",
697
"l: Log window toggle",
698
"TAB: Switch window",
700
self.log_message_raw(("bold",
706
"s: Start new checker",
623
self.log_message_raw((u"bold",
627
u"l: Log window toggle",
628
u"TAB: Switch window",
630
self.log_message_raw((u"bold",
636
u"s: Start new checker",
713
643
if self.topwidget.get_focus() is self.logbox:
714
644
self.topwidget.set_focus(0)
716
646
self.topwidget.set_focus(self.logbox)
718
#elif (key == "end" or key == "meta >" or key == "G"
648
#elif (key == u"end" or key == u"meta >" or key == u"G"
720
650
# pass # xxx end-of-buffer
721
#elif (key == "home" or key == "meta <" or key == "g"
651
#elif (key == u"home" or key == u"meta <" or key == u"g"
723
653
# pass # xxx beginning-of-buffer
724
#elif key == "ctrl e" or key == "$":
654
#elif key == u"ctrl e" or key == u"$":
725
655
# pass # xxx move-end-of-line
726
#elif key == "ctrl a" or key == "^":
656
#elif key == u"ctrl a" or key == u"^":
727
657
# pass # xxx move-beginning-of-line
728
#elif key == "ctrl b" or key == "meta (" or key == "h":
658
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
729
659
# pass # xxx left
730
#elif key == "ctrl f" or key == "meta )" or key == "l":
660
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
731
661
# pass # xxx right
733
663
# pass # scroll up log
735
665
# pass # scroll down log
736
666
elif self.topwidget.selectable():
737
667
self.topwidget.keypress(self.size, key)