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