49
51
if sys.version_info.major == 2:
54
# Show warnings by default
55
if not sys.warnoptions:
56
warnings.simplefilter("default")
58
log = logging.getLogger(os.path.basename(sys.argv[0]))
59
logging.basicConfig(level="NOTSET", # Show all messages
60
format="%(message)s") # Show basic log messages
62
logging.captureWarnings(True) # Show warnings via the logging system
64
locale.setlocale(locale.LC_ALL, "")
66
logging.getLogger("dbus.proxies").setLevel(logging.CRITICAL)
67
logging.getLogger("urwid").setLevel(logging.INFO)
54
locale.setlocale(locale.LC_ALL, '')
56
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
69
58
# Some useful constants
70
domain = "se.recompile"
71
server_interface = domain + ".Mandos"
72
client_interface = domain + ".Mandos.Client"
59
domain = 'se.recompile'
60
server_interface = domain + '.Mandos'
61
client_interface = domain + '.Mandos.Client'
76
65
dbus.OBJECT_MANAGER_IFACE
135
124
def __init__(self, server_proxy_object=None, update_hook=None,
136
delete_hook=None, **kwargs):
125
delete_hook=None, logger=None, **kwargs):
137
126
# Called on update
138
127
self.update_hook = update_hook
139
128
# Called on delete
140
129
self.delete_hook = delete_hook
141
130
# Mandos Server proxy object
142
131
self.server_proxy_object = server_proxy_object
144
135
self._update_timer_callback_tag = None
181
173
if flag and self._update_timer_callback_tag is None:
182
174
# Will update the shown timer value every second
183
self._update_timer_callback_tag = (
184
GLib.timeout_add(1000,
185
glib_safely(self.update_timer)))
175
self._update_timer_callback_tag = (GLib.timeout_add
186
178
elif not (flag or self._update_timer_callback_tag is None):
187
179
GLib.source_remove(self._update_timer_callback_tag)
188
180
self._update_timer_callback_tag = None
190
182
def checker_completed(self, exitstatus, condition, command):
191
183
if exitstatus == 0:
192
log.debug('Checker for client %s (command "%s")'
193
" succeeded", self.properties["Name"], command)
184
self.logger('Checker for client {} (command "{}")'
185
' succeeded'.format(self.properties["Name"],
197
190
if os.WIFEXITED(condition):
198
log.info('Checker for client %s (command "%s") failed'
199
" with exit code %d", self.properties["Name"],
200
command, os.WEXITSTATUS(condition))
191
self.logger('Checker for client {} (command "{}") failed'
193
.format(self.properties["Name"], command,
194
os.WEXITSTATUS(condition)))
201
195
elif os.WIFSIGNALED(condition):
202
log.info('Checker for client %s (command "%s") was'
203
" killed by signal %d", self.properties["Name"],
204
command, os.WTERMSIG(condition))
196
self.logger('Checker for client {} (command "{}") was'
197
' killed by signal {}'
198
.format(self.properties["Name"], command,
199
os.WTERMSIG(condition)))
207
202
def checker_started(self, command):
208
203
"""Server signals that a checker started."""
209
log.debug('Client %s started checker "%s"',
210
self.properties["Name"], command)
204
self.logger('Client {} started checker "{}"'
205
.format(self.properties["Name"],
212
208
def got_secret(self):
213
log.info("Client %s received its secret",
214
self.properties["Name"])
209
self.logger('Client {} received its secret'
210
.format(self.properties["Name"]))
216
212
def need_approval(self, timeout, default):
218
message = "Client %s needs approval within %f seconds"
214
message = 'Client {} needs approval within {} seconds'
220
message = "Client %s will get its secret in %f seconds"
221
log.info(message, self.properties["Name"], timeout/1000)
216
message = 'Client {} will get its secret in {} seconds'
217
self.logger(message.format(self.properties["Name"],
223
220
def rejected(self, reason):
224
log.info("Client %s was rejected; reason: %s",
225
self.properties["Name"], reason)
221
self.logger('Client {} was rejected; reason: {}'
222
.format(self.properties["Name"], reason))
227
224
def selectable(self):
228
225
"""Make this a "selectable" widget.
282
279
timer = datetime.timedelta(0)
284
281
expires = (datetime.datetime.strptime
285
(expires, "%Y-%m-%dT%H:%M:%S.%f"))
282
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
286
283
timer = max(expires - datetime.datetime.utcnow(),
287
284
datetime.timedelta())
288
message = ("A checker has failed! Time until client"
285
message = ('A checker has failed! Time until client'
290
287
.format(str(timer).rsplit(".", 1)[0]))
291
288
self.using_timer(True)
403
class UserInterface(object):
417
404
"""This is the entire user interface - the whole screen
418
405
with boxes, lists of client widgets, etc.
420
def __init__(self, max_log_length=1000):
407
def __init__(self, max_log_length=1000, log_level=1):
421
408
DBusGMainLoop(set_as_default=True)
423
410
self.screen = urwid.curses_display.Screen()
469
458
self.log_visible = True
470
459
self.log_wrap = "any"
472
self.loghandler = UILogHandler(self)
475
self.add_log_line(("bold",
476
"Mandos Monitor version " + version))
477
self.add_log_line(("bold", "q: Quit ?: Help"))
462
self.log_message_raw(("bold",
463
"Mandos Monitor version " + version))
464
self.log_message_raw(("bold",
479
self.busname = domain + ".Mandos"
467
self.busname = domain + '.Mandos'
480
468
self.main_loop = GLib.MainLoop()
482
def client_not_found(self, key_id, address):
483
log.info("Client with address %s and key ID %s could"
484
" not be found", address, key_id)
470
def client_not_found(self, fingerprint, address):
471
self.log_message("Client with address {} and fingerprint {}"
472
" could not be found"
473
.format(address, fingerprint))
486
475
def rebuild(self):
487
476
"""This rebuilds the User Interface.
498
487
self.uilist.append(self.logbox)
499
488
self.topwidget = urwid.Pile(self.uilist)
501
def add_log_line(self, markup):
490
def log_message(self, message, level=1):
491
"""Log message formatted with timestamp"""
492
if level < self.log_level:
494
timestamp = datetime.datetime.now().isoformat()
495
self.log_message_raw("{}: {}".format(timestamp, message),
498
def log_message_raw(self, markup, level=1):
499
"""Add a log message to the log buffer."""
500
if level < self.log_level:
502
502
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
503
503
if self.max_log_length:
504
504
if len(self.log) > self.max_log_length:
505
del self.log[0:(len(self.log) - self.max_log_length)]
505
del self.log[0:len(self.log)-self.max_log_length-1]
506
506
self.logbox.set_focus(len(self.logbox.body.contents)-1,
507
507
coming_from="above")
594
593
mandos_clients = (self.mandos_serv
595
594
.GetAllClientsWithProperties())
596
595
if not mandos_clients:
597
log.warning("Note: Server has no clients.")
596
self.log_message_raw(("bold",
597
"Note: Server has no clients."))
598
598
except dbus.exceptions.DBusException:
599
log.warning("Note: No Mandos server running.")
599
self.log_message_raw(("bold",
600
"Note: No Mandos server running."))
600
601
mandos_clients = dbus.Dictionary()
602
603
(self.mandos_serv
622
623
proxy_object=client_proxy_object,
623
624
properties=client,
624
625
update_hook=self.refresh,
625
delete_hook=self.remove_client),
626
delete_hook=self.remove_client,
627
logger=self.log_message),
629
self._input_callback_tag = (
631
GLib.IOChannel.unix_new(sys.stdin.fileno()),
632
GLib.PRIORITY_DEFAULT, GLib.IO_IN,
633
glib_safely(self.process_input)))
631
self._input_callback_tag = (GLib.io_add_watch
634
635
self.main_loop.run()
635
636
# Main loop has finished, we should close everything now
636
637
GLib.source_remove(self._input_callback_tag)
637
with warnings.catch_warnings():
638
warnings.simplefilter("ignore", BytesWarning)
642
641
self.main_loop.quit()
643
log.removeHandler(self.loghandler)
644
log.propagate = self.orig_log_propagate
646
643
def process_input(self, source, condition):
647
644
keys = self.screen.get_input()
707
705
self.topwidget.set_focus(self.logbox)
710
if log.level < logging.INFO:
711
log.setLevel(logging.INFO)
712
log.info("Verbose mode: Off")
708
if self.log_level == 0:
710
self.log_message("Verbose mode: Off")
714
log.setLevel(logging.NOTSET)
715
log.info("Verbose mode: On")
713
self.log_message("Verbose mode: On")
716
714
# elif (key == "end" or key == "meta >" or key == "G"
717
715
# or key == ">"):
718
716
# pass # xxx end-of-buffer
740
class UILogHandler(logging.Handler):
741
def __init__(self, ui, *args, **kwargs):
743
super(UILogHandler, self).__init__(*args, **kwargs)
745
logging.Formatter("%(asctime)s: %(message)s"))
746
def emit(self, record):
747
msg = self.format(record)
748
if record.levelno > logging.INFO:
750
self.ui.add_log_line(msg)
753
738
ui = UserInterface()
756
741
except KeyboardInterrupt:
757
with warnings.catch_warnings():
758
warnings.filterwarnings("ignore", "", BytesWarning)
761
with warnings.catch_warnings():
762
warnings.filterwarnings("ignore", "", BytesWarning)
743
except Exception as e:
744
ui.log_message(str(e))