45
locale.setlocale(locale.LC_ALL, '')
45
locale.setlocale(locale.LC_ALL, u'')
48
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
48
logging.getLogger(u'dbus.proxies').setLevel(logging.CRITICAL)
50
50
# Some useful constants
51
domain = 'se.bsnet.fukt'
52
server_interface = domain + '.Mandos'
53
client_interface = domain + '.Mandos.Client'
51
domain = u'se.bsnet.fukt'
52
server_interface = domain + u'.Mandos'
53
client_interface = domain + u'.Mandos.Client'
56
56
# Always run in monochrome mode
57
57
urwid.curses_display.curses.has_colors = lambda : False
65
65
"Parse an ISO 8601 date string to a datetime.datetime()"
68
d, t = iso.split("T", 1)
69
year, month, day = d.split("-", 2)
70
hour, minute, second = t.split(":", 2)
68
d, t = iso.split(u"T", 1)
69
year, month, day = d.split(u"-", 2)
70
hour, minute, second = t.split(u":", 2)
71
71
second, fraction = divmod(float(second), 1)
72
72
return datetime.datetime(int(year),
127
127
self.last_checker_failed = False
129
129
# The widget shown normally
130
self._text_widget = urwid.Text("")
130
self._text_widget = urwid.Text(u"")
131
131
# The widget shown when we have focus
132
self._focus_text_widget = urwid.Text("")
132
self._focus_text_widget = urwid.Text(u"")
133
133
super(MandosClientWidget, self).__init__(
134
134
update_hook=update_hook, delete_hook=delete_hook,
154
154
if self.need_approval:
155
155
self.using_timer(True)
157
self.proxy.connect_to_signal("CheckerCompleted",
157
self.proxy.connect_to_signal(u"CheckerCompleted",
158
158
self.checker_completed,
159
159
client_interface,
160
160
byte_arrays=True)
161
self.proxy.connect_to_signal("CheckerStarted",
161
self.proxy.connect_to_signal(u"CheckerStarted",
162
162
self.checker_started,
163
163
client_interface,
164
164
byte_arrays=True)
165
self.proxy.connect_to_signal("GotSecret",
165
self.proxy.connect_to_signal(u"GotSecret",
167
167
client_interface,
168
168
byte_arrays=True)
169
self.proxy.connect_to_signal("NeedApproval",
169
self.proxy.connect_to_signal(u"NeedApproval",
170
170
self.need_approval,
171
171
client_interface,
172
172
byte_arrays=True)
173
self.proxy.connect_to_signal("Rejected",
173
self.proxy.connect_to_signal(u"Rejected",
175
175
client_interface,
176
176
byte_arrays=True)
213
213
self.last_checker_failed = True
214
214
self.using_timer(True)
215
215
if os.WIFEXITED(condition):
216
self.logger('Checker for client %s (command "%s")'
217
' failed with exit code %s'
218
% (self.properties["Name"], command,
216
self.logger(u'Checker for client %s (command "%s")'
217
u' failed with exit code %s'
218
% (self.properties[u"Name"], command,
219
219
os.WEXITSTATUS(condition)))
220
220
elif os.WIFSIGNALED(condition):
221
self.logger('Checker for client %s (command "%s")'
222
' was killed by signal %s'
223
% (self.properties["Name"], command,
221
self.logger(u'Checker for client %s (command "%s")'
222
u' was killed by signal %s'
223
% (self.properties[u"Name"], command,
224
224
os.WTERMSIG(condition)))
225
225
elif os.WCOREDUMP(condition):
226
self.logger('Checker for client %s (command "%s")'
228
% (self.properties["Name"], command))
226
self.logger(u'Checker for client %s (command "%s")'
228
% (self.properties[u"Name"], command))
230
self.logger('Checker for client %s completed'
230
self.logger(u'Checker for client %s completed'
234
234
def checker_started(self, command):
235
#self.logger('Client %s started checker "%s"'
236
# % (self.properties["Name"], unicode(command)))
235
#self.logger(u'Client %s started checker "%s"'
236
# % (self.properties[u"Name"], unicode(command)))
239
239
def got_secret(self):
240
240
self.last_checker_failed = False
241
self.logger('Client %s received its secret'
242
% self.properties["Name"])
241
self.logger(u'Client %s received its secret'
242
% self.properties[u"Name"])
244
244
def need_approval(self, timeout, default):
246
message = 'Client %s needs approval within %s seconds'
246
message = u'Client %s needs approval within %s seconds'
248
message = 'Client %s will get its secret in %s seconds'
248
message = u'Client %s will get its secret in %s seconds'
249
249
self.logger(message
250
% (self.properties["Name"], timeout/1000))
250
% (self.properties[u"Name"], timeout/1000))
251
251
self.using_timer(True)
253
253
def rejected(self, reason):
254
self.logger('Client %s was rejected; reason: %s'
255
% (self.properties["Name"], reason))
254
self.logger(u'Client %s was rejected; reason: %s'
255
% (self.properties[u"Name"], reason))
257
257
def selectable(self):
258
258
"""Make this a "selectable" widget.
273
273
def update(self):
274
274
"Called when what is visible on the screen should be updated."
275
275
# How to add standout mode to a style
276
with_standout = { "normal": "standout",
277
"bold": "bold-standout",
279
"underline-blink-standout",
280
"bold-underline-blink":
281
"bold-underline-blink-standout",
276
with_standout = { u"normal": u"standout",
277
u"bold": u"bold-standout",
279
u"underline-blink-standout",
280
u"bold-underline-blink":
281
u"bold-underline-blink-standout",
284
284
# Rebuild focus and non-focus widgets using current properties
286
286
# Base part of a client. Name!
288
% {"name": self.properties["Name"]})
289
if not self.properties["Enabled"]:
291
elif self.properties["ApprovalPending"]:
287
base = (u'%(name)s: '
288
% {u"name": self.properties[u"Name"]})
289
if not self.properties[u"Enabled"]:
290
message = u"DISABLED"
291
elif self.properties[u"ApprovalPending"]:
292
292
timeout = datetime.timedelta(milliseconds
293
293
= self.properties
295
295
last_approval_request = isoformat_to_datetime(
296
self.properties["LastApprovalRequest"])
296
self.properties[u"LastApprovalRequest"])
297
297
if last_approval_request is not None:
298
298
timer = timeout - (datetime.datetime.utcnow()
299
299
- last_approval_request)
301
301
timer = datetime.timedelta()
302
if self.properties["ApprovedByDefault"]:
303
message = "Approval in %s. (d)eny?"
302
if self.properties[u"ApprovedByDefault"]:
303
message = u"Approval in %s. (d)eny?"
305
message = "Denial in %s. (a)pprove?"
305
message = u"Denial in %s. (a)pprove?"
306
306
message = message % unicode(timer).rsplit(".", 1)[0]
307
307
elif self.last_checker_failed:
308
308
timeout = datetime.timedelta(milliseconds
309
309
= self.properties
311
311
last_ok = isoformat_to_datetime(
312
max((self.properties["LastCheckedOK"]
313
or self.properties["Created"]),
314
self.properties["LastEnabled"]))
312
max((self.properties[u"LastCheckedOK"]
313
or self.properties[u"Created"]),
314
self.properties[u"LastEnabled"]))
315
315
timer = timeout - (datetime.datetime.utcnow() - last_ok)
316
message = ('A checker has failed! Time until client'
316
message = (u'A checker has failed! Time until client'
317
u' gets disabled: %s'
318
318
% unicode(timer).rsplit(".", 1)[0])
321
self._text = "%s%s" % (base, message)
321
self._text = u"%s%s" % (base, message)
323
323
if not urwid.supports_unicode():
324
self._text = self._text.encode("ascii", "replace")
325
textlist = [("normal", self._text)]
324
self._text = self._text.encode(u"ascii", u"replace")
325
textlist = [(u"normal", self._text)]
326
326
self._text_widget.set_text(textlist)
327
327
self._focus_text_widget.set_text([(with_standout[text[0]],
357
357
def keypress(self, maxcolrow, key):
359
359
This overrides the method from urwid.FlowWidget"""
361
self.proxy.Enable(dbus_interface = client_interface,
364
self.proxy.Disable(dbus_interface = client_interface,
361
self.proxy.Enable(dbus_interface = client_interface)
363
self.proxy.Disable(dbus_interface = client_interface)
367
365
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
368
dbus_interface = client_interface,
366
dbus_interface = client_interface)
371
368
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
372
dbus_interface = client_interface,
374
elif key == "R" or key == "_" or key == "ctrl k":
369
dbus_interface = client_interface)
370
elif key == u"R" or key == u"_" or key == u"ctrl k":
375
371
self.server_proxy_object.RemoveClient(self.proxy
379
self.proxy.StartChecker(dbus_interface = client_interface,
382
self.proxy.StopChecker(dbus_interface = client_interface,
385
self.proxy.CheckedOK(dbus_interface = client_interface,
374
self.proxy.StartChecker(dbus_interface = client_interface)
376
self.proxy.StopChecker(dbus_interface = client_interface)
378
self.proxy.CheckedOK(dbus_interface = client_interface)
388
# elif key == "p" or key == "=":
380
# elif key == u"p" or key == "=":
389
381
# self.proxy.pause()
390
# elif key == "u" or key == ":":
382
# elif key == u"u" or key == ":":
391
383
# self.proxy.unpause()
384
# elif key == u"RET":
428
420
self.screen = urwid.curses_display.Screen()
430
422
self.screen.register_palette((
432
"default", "default", None),
434
"default", "default", "bold"),
436
"default", "default", "underline"),
438
"default", "default", "standout"),
439
("bold-underline-blink",
440
"default", "default", ("bold", "underline")),
442
"default", "default", ("bold", "standout")),
443
("underline-blink-standout",
444
"default", "default", ("underline", "standout")),
445
("bold-underline-blink-standout",
446
"default", "default", ("bold", "underline",
424
u"default", u"default", None),
426
u"default", u"default", u"bold"),
428
u"default", u"default", u"underline"),
430
u"default", u"default", u"standout"),
431
(u"bold-underline-blink",
432
u"default", u"default", (u"bold", u"underline")),
434
u"default", u"default", (u"bold", u"standout")),
435
(u"underline-blink-standout",
436
u"default", u"default", (u"underline", u"standout")),
437
(u"bold-underline-blink-standout",
438
u"default", u"default", (u"bold", u"underline",
450
442
if urwid.supports_unicode():
451
self.divider = "─" # \u2500
452
#self.divider = "━" # \u2501
443
self.divider = u"─" # \u2500
444
#self.divider = u"━" # \u2501
454
#self.divider = "-" # \u002d
455
self.divider = "_" # \u005f
446
#self.divider = u"-" # \u002d
447
self.divider = u"_" # \u005f
457
449
self.screen.start()
472
464
# This keeps track of whether self.uilist currently has
473
465
# self.logbox in it or not
474
466
self.log_visible = True
475
self.log_wrap = "any"
467
self.log_wrap = u"any"
478
self.log_message_raw(("bold",
479
"Mandos Monitor version " + version))
480
self.log_message_raw(("bold",
470
self.log_message_raw((u"bold",
471
u"Mandos Monitor version " + version))
472
self.log_message_raw((u"bold",
483
475
self.busname = domain + '.Mandos'
484
476
self.main_loop = gobject.MainLoop()
485
477
self.bus = dbus.SystemBus()
486
478
mandos_dbus_objc = self.bus.get_object(
487
self.busname, "/", follow_name_owner_changes=True)
479
self.busname, u"/", follow_name_owner_changes=True)
488
480
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
490
482
= server_interface)
495
487
mandos_clients = dbus.Dictionary()
497
489
(self.mandos_serv
498
.connect_to_signal("ClientRemoved",
490
.connect_to_signal(u"ClientRemoved",
499
491
self.find_and_remove_client,
500
492
dbus_interface=server_interface,
501
493
byte_arrays=True))
502
494
(self.mandos_serv
503
.connect_to_signal("ClientAdded",
495
.connect_to_signal(u"ClientAdded",
504
496
self.add_new_client,
505
497
dbus_interface=server_interface,
506
498
byte_arrays=True))
507
499
(self.mandos_serv
508
.connect_to_signal("ClientNotFound",
500
.connect_to_signal(u"ClientNotFound",
509
501
self.client_not_found,
510
502
dbus_interface=server_interface,
511
503
byte_arrays=True))
558
550
and len(self.log) > self.max_log_length):
559
551
del self.log[0:len(self.log)-self.max_log_length-1]
560
552
self.logbox.set_focus(len(self.logbox.body.contents),
553
coming_from=u"above")
564
556
def toggle_log_display(self):
565
557
"""Toggle visibility of the log buffer."""
566
558
self.log_visible = not self.log_visible
568
#self.log_message("Log visibility changed to: "
560
#self.log_message(u"Log visibility changed to: "
569
561
# + unicode(self.log_visible))
571
563
def change_log_display(self):
572
564
"""Change type of log display.
573
565
Currently, this toggles wrapping of text lines."""
574
if self.log_wrap == "clip":
575
self.log_wrap = "any"
566
if self.log_wrap == u"clip":
567
self.log_wrap = u"any"
577
self.log_wrap = "clip"
569
self.log_wrap = u"clip"
578
570
for textwidget in self.log:
579
571
textwidget.set_wrap_mode(self.log_wrap)
580
#self.log_message("Wrap mode: " + self.log_wrap)
572
#self.log_message(u"Wrap mode: " + self.log_wrap)
582
574
def find_and_remove_client(self, path, name):
583
575
"""Find an client from its object path and remove it.
648
640
def process_input(self, source, condition):
649
641
keys = self.screen.get_input()
650
translations = { "ctrl n": "down", # Emacs
651
"ctrl p": "up", # Emacs
652
"ctrl v": "page down", # Emacs
653
"meta v": "page up", # Emacs
654
" ": "page down", # less
655
"f": "page down", # less
656
"b": "page up", # less
642
translations = { u"ctrl n": u"down", # Emacs
643
u"ctrl p": u"up", # Emacs
644
u"ctrl v": u"page down", # Emacs
645
u"meta v": u"page up", # Emacs
646
u" ": u"page down", # less
647
u"f": u"page down", # less
648
u"b": u"page up", # less
663
655
except KeyError: # :-)
666
if key == "q" or key == "Q":
658
if key == u"q" or key == u"Q":
669
elif key == "window resize":
661
elif key == u"window resize":
670
662
self.size = self.screen.get_cols_rows()
672
elif key == "\f": # Ctrl-L
664
elif key == u"\f": # Ctrl-L
674
elif key == "l" or key == "D":
666
elif key == u"l" or key == u"D":
675
667
self.toggle_log_display()
677
elif key == "w" or key == "i":
669
elif key == u"w" or key == u"i":
678
670
self.change_log_display()
680
elif key == "?" or key == "f1" or key == "esc":
672
elif key == u"?" or key == u"f1" or key == u"esc":
681
673
if not self.log_visible:
682
674
self.log_visible = True
684
self.log_message_raw(("bold",
688
"l: Log window toggle",
689
"TAB: Switch window",
691
self.log_message_raw(("bold",
697
"s: Start new checker",
676
self.log_message_raw((u"bold",
680
u"l: Log window toggle",
681
u"TAB: Switch window",
683
self.log_message_raw((u"bold",
689
u"s: Start new checker",
704
696
if self.topwidget.get_focus() is self.logbox:
705
697
self.topwidget.set_focus(0)
707
699
self.topwidget.set_focus(self.logbox)
709
#elif (key == "end" or key == "meta >" or key == "G"
701
#elif (key == u"end" or key == u"meta >" or key == u"G"
711
703
# pass # xxx end-of-buffer
712
#elif (key == "home" or key == "meta <" or key == "g"
704
#elif (key == u"home" or key == u"meta <" or key == u"g"
714
706
# pass # xxx beginning-of-buffer
715
#elif key == "ctrl e" or key == "$":
707
#elif key == u"ctrl e" or key == u"$":
716
708
# pass # xxx move-end-of-line
717
#elif key == "ctrl a" or key == "^":
709
#elif key == u"ctrl a" or key == u"^":
718
710
# pass # xxx move-beginning-of-line
719
#elif key == "ctrl b" or key == "meta (" or key == "h":
711
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
720
712
# pass # xxx left
721
#elif key == "ctrl f" or key == "meta )" or key == "l":
713
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
722
714
# pass # xxx right
724
716
# pass # scroll up log
726
718
# pass # scroll down log
727
719
elif self.topwidget.selectable():
728
720
self.topwidget.keypress(self.size, key)