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
361
self.proxy.Enable(dbus_interface = client_interface)
363
363
self.proxy.Disable(dbus_interface = client_interface)
365
365
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
366
366
dbus_interface = client_interface)
368
368
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
369
369
dbus_interface = client_interface)
370
elif key == "R" or key == "_" or key == "ctrl k":
370
elif key == u"R" or key == u"_" or key == u"ctrl k":
371
371
self.server_proxy_object.RemoveClient(self.proxy
374
374
self.proxy.StartChecker(dbus_interface = client_interface)
376
376
self.proxy.StopChecker(dbus_interface = client_interface)
378
378
self.proxy.CheckedOK(dbus_interface = client_interface)
380
# elif key == "p" or key == "=":
380
# elif key == u"p" or key == "=":
381
381
# self.proxy.pause()
382
# elif key == "u" or key == ":":
382
# elif key == u"u" or key == ":":
383
383
# self.proxy.unpause()
384
# elif key == u"RET":
420
420
self.screen = urwid.curses_display.Screen()
422
422
self.screen.register_palette((
424
"default", "default", None),
426
"default", "default", "bold"),
428
"default", "default", "underline"),
430
"default", "default", "standout"),
431
("bold-underline-blink",
432
"default", "default", ("bold", "underline")),
434
"default", "default", ("bold", "standout")),
435
("underline-blink-standout",
436
"default", "default", ("underline", "standout")),
437
("bold-underline-blink-standout",
438
"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",
442
442
if urwid.supports_unicode():
443
self.divider = "─" # \u2500
444
#self.divider = "━" # \u2501
443
self.divider = u"─" # \u2500
444
#self.divider = u"━" # \u2501
446
#self.divider = "-" # \u002d
447
self.divider = "_" # \u005f
446
#self.divider = u"-" # \u002d
447
self.divider = u"_" # \u005f
449
449
self.screen.start()
464
464
# This keeps track of whether self.uilist currently has
465
465
# self.logbox in it or not
466
466
self.log_visible = True
467
self.log_wrap = "any"
467
self.log_wrap = u"any"
470
self.log_message_raw(("bold",
471
"Mandos Monitor version " + version))
472
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",
475
475
self.busname = domain + '.Mandos'
476
476
self.main_loop = gobject.MainLoop()
477
477
self.bus = dbus.SystemBus()
478
478
mandos_dbus_objc = self.bus.get_object(
479
self.busname, "/", follow_name_owner_changes=True)
479
self.busname, u"/", follow_name_owner_changes=True)
480
480
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
482
482
= server_interface)
487
487
mandos_clients = dbus.Dictionary()
489
489
(self.mandos_serv
490
.connect_to_signal("ClientRemoved",
490
.connect_to_signal(u"ClientRemoved",
491
491
self.find_and_remove_client,
492
492
dbus_interface=server_interface,
493
493
byte_arrays=True))
494
494
(self.mandos_serv
495
.connect_to_signal("ClientAdded",
495
.connect_to_signal(u"ClientAdded",
496
496
self.add_new_client,
497
497
dbus_interface=server_interface,
498
498
byte_arrays=True))
499
499
(self.mandos_serv
500
.connect_to_signal("ClientNotFound",
500
.connect_to_signal(u"ClientNotFound",
501
501
self.client_not_found,
502
502
dbus_interface=server_interface,
503
503
byte_arrays=True))
550
550
and len(self.log) > self.max_log_length):
551
551
del self.log[0:len(self.log)-self.max_log_length-1]
552
552
self.logbox.set_focus(len(self.logbox.body.contents),
553
coming_from=u"above")
556
556
def toggle_log_display(self):
557
557
"""Toggle visibility of the log buffer."""
558
558
self.log_visible = not self.log_visible
560
#self.log_message("Log visibility changed to: "
560
#self.log_message(u"Log visibility changed to: "
561
561
# + unicode(self.log_visible))
563
563
def change_log_display(self):
564
564
"""Change type of log display.
565
565
Currently, this toggles wrapping of text lines."""
566
if self.log_wrap == "clip":
567
self.log_wrap = "any"
566
if self.log_wrap == u"clip":
567
self.log_wrap = u"any"
569
self.log_wrap = "clip"
569
self.log_wrap = u"clip"
570
570
for textwidget in self.log:
571
571
textwidget.set_wrap_mode(self.log_wrap)
572
#self.log_message("Wrap mode: " + self.log_wrap)
572
#self.log_message(u"Wrap mode: " + self.log_wrap)
574
574
def find_and_remove_client(self, path, name):
575
575
"""Find an client from its object path and remove it.
640
640
def process_input(self, source, condition):
641
641
keys = self.screen.get_input()
642
translations = { "ctrl n": "down", # Emacs
643
"ctrl p": "up", # Emacs
644
"ctrl v": "page down", # Emacs
645
"meta v": "page up", # Emacs
646
" ": "page down", # less
647
"f": "page down", # less
648
"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
655
655
except KeyError: # :-)
658
if key == "q" or key == "Q":
658
if key == u"q" or key == u"Q":
661
elif key == "window resize":
661
elif key == u"window resize":
662
662
self.size = self.screen.get_cols_rows()
664
elif key == "\f": # Ctrl-L
664
elif key == u"\f": # Ctrl-L
666
elif key == "l" or key == "D":
666
elif key == u"l" or key == u"D":
667
667
self.toggle_log_display()
669
elif key == "w" or key == "i":
669
elif key == u"w" or key == u"i":
670
670
self.change_log_display()
672
elif key == "?" or key == "f1" or key == "esc":
672
elif key == u"?" or key == u"f1" or key == u"esc":
673
673
if not self.log_visible:
674
674
self.log_visible = True
676
self.log_message_raw(("bold",
680
"l: Log window toggle",
681
"TAB: Switch window",
683
self.log_message_raw(("bold",
689
"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",
696
696
if self.topwidget.get_focus() is self.logbox:
697
697
self.topwidget.set_focus(0)
699
699
self.topwidget.set_focus(self.logbox)
701
#elif (key == "end" or key == "meta >" or key == "G"
701
#elif (key == u"end" or key == u"meta >" or key == u"G"
703
703
# pass # xxx end-of-buffer
704
#elif (key == "home" or key == "meta <" or key == "g"
704
#elif (key == u"home" or key == u"meta <" or key == u"g"
706
706
# pass # xxx beginning-of-buffer
707
#elif key == "ctrl e" or key == "$":
707
#elif key == u"ctrl e" or key == u"$":
708
708
# pass # xxx move-end-of-line
709
#elif key == "ctrl a" or key == "^":
709
#elif key == u"ctrl a" or key == u"^":
710
710
# pass # xxx move-beginning-of-line
711
#elif key == "ctrl b" or key == "meta (" or key == "h":
711
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
712
712
# pass # xxx left
713
#elif key == "ctrl f" or key == "meta )" or key == "l":
713
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
714
714
# pass # xxx right
716
716
# pass # scroll up log
718
718
# pass # scroll down log
719
719
elif self.topwidget.selectable():
720
720
self.topwidget.keypress(self.size, key)