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)
54
locale.setlocale(locale.LC_ALL, '')
56
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
68
58
# Some useful constants
69
domain = "se.recompile"
70
server_interface = domain + ".Mandos"
71
client_interface = domain + ".Mandos.Client"
59
domain = 'se.recompile'
60
server_interface = domain + '.Mandos'
61
client_interface = domain + '.Mandos.Client'
75
65
dbus.OBJECT_MANAGER_IFACE
134
124
def __init__(self, server_proxy_object=None, update_hook=None,
135
delete_hook=None, **kwargs):
125
delete_hook=None, logger=None, **kwargs):
136
126
# Called on update
137
127
self.update_hook = update_hook
138
128
# Called on delete
139
129
self.delete_hook = delete_hook
140
130
# Mandos Server proxy object
141
131
self.server_proxy_object = server_proxy_object
143
135
self._update_timer_callback_tag = None
180
173
if flag and self._update_timer_callback_tag is None:
181
174
# Will update the shown timer value every second
182
self._update_timer_callback_tag = (
183
GLib.timeout_add(1000,
184
glib_safely(self.update_timer)))
175
self._update_timer_callback_tag = (GLib.timeout_add
185
178
elif not (flag or self._update_timer_callback_tag is None):
186
179
GLib.source_remove(self._update_timer_callback_tag)
187
180
self._update_timer_callback_tag = None
189
182
def checker_completed(self, exitstatus, condition, command):
190
183
if exitstatus == 0:
191
log.debug('Checker for client %s (command "%s")'
192
" succeeded", self.properties["Name"], command)
184
self.logger('Checker for client {} (command "{}")'
185
' succeeded'.format(self.properties["Name"],
196
190
if os.WIFEXITED(condition):
197
log.info('Checker for client %s (command "%s") failed'
198
" with exit code %d", self.properties["Name"],
199
command, os.WEXITSTATUS(condition))
191
self.logger('Checker for client {} (command "{}") failed'
193
.format(self.properties["Name"], command,
194
os.WEXITSTATUS(condition)))
200
195
elif os.WIFSIGNALED(condition):
201
log.info('Checker for client %s (command "%s") was'
202
" killed by signal %d", self.properties["Name"],
203
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)))
206
202
def checker_started(self, command):
207
203
"""Server signals that a checker started."""
208
log.debug('Client %s started checker "%s"',
209
self.properties["Name"], command)
204
self.logger('Client {} started checker "{}"'
205
.format(self.properties["Name"],
211
208
def got_secret(self):
212
log.info("Client %s received its secret",
213
self.properties["Name"])
209
self.logger('Client {} received its secret'
210
.format(self.properties["Name"]))
215
212
def need_approval(self, timeout, default):
217
message = "Client %s needs approval within %f seconds"
214
message = 'Client {} needs approval within {} seconds'
219
message = "Client %s will get its secret in %f seconds"
220
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"],
222
220
def rejected(self, reason):
223
log.info("Client %s was rejected; reason: %s",
224
self.properties["Name"], reason)
221
self.logger('Client {} was rejected; reason: {}'
222
.format(self.properties["Name"], reason))
226
224
def selectable(self):
227
225
"""Make this a "selectable" widget.
281
279
timer = datetime.timedelta(0)
283
281
expires = (datetime.datetime.strptime
284
(expires, "%Y-%m-%dT%H:%M:%S.%f"))
282
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
285
283
timer = max(expires - datetime.datetime.utcnow(),
286
284
datetime.timedelta())
287
message = ("A checker has failed! Time until client"
285
message = ('A checker has failed! Time until client'
289
287
.format(str(timer).rsplit(".", 1)[0]))
290
288
self.using_timer(True)
403
class UserInterface(object):
416
404
"""This is the entire user interface - the whole screen
417
405
with boxes, lists of client widgets, etc.
419
def __init__(self, max_log_length=1000):
407
def __init__(self, max_log_length=1000, log_level=1):
420
408
DBusGMainLoop(set_as_default=True)
422
410
self.screen = urwid.curses_display.Screen()
468
458
self.log_visible = True
469
459
self.log_wrap = "any"
471
self.loghandler = UILogHandler(self)
474
self.add_log_line(("bold",
475
"Mandos Monitor version " + version))
476
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",
478
self.busname = domain + ".Mandos"
467
self.busname = domain + '.Mandos'
479
468
self.main_loop = GLib.MainLoop()
481
470
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)
471
self.log_message("Client with address {} and key ID {} could"
472
" not be found".format(address, key_id))
485
474
def rebuild(self):
486
475
"""This rebuilds the User Interface.
497
486
self.uilist.append(self.logbox)
498
487
self.topwidget = urwid.Pile(self.uilist)
500
def add_log_line(self, markup):
489
def log_message(self, message, level=1):
490
"""Log message formatted with timestamp"""
491
if level < self.log_level:
493
timestamp = datetime.datetime.now().isoformat()
494
self.log_message_raw("{}: {}".format(timestamp, message),
497
def log_message_raw(self, markup, level=1):
498
"""Add a log message to the log buffer."""
499
if level < self.log_level:
501
501
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
502
if self.max_log_length:
503
503
if len(self.log) > self.max_log_length:
504
del self.log[0:(len(self.log) - self.max_log_length)]
504
del self.log[0:len(self.log)-self.max_log_length-1]
505
505
self.logbox.set_focus(len(self.logbox.body.contents)-1,
506
506
coming_from="above")
593
592
mandos_clients = (self.mandos_serv
594
593
.GetAllClientsWithProperties())
595
594
if not mandos_clients:
596
log.warning("Note: Server has no clients.")
595
self.log_message_raw(("bold",
596
"Note: Server has no clients."))
597
597
except dbus.exceptions.DBusException:
598
log.warning("Note: No Mandos server running.")
598
self.log_message_raw(("bold",
599
"Note: No Mandos server running."))
599
600
mandos_clients = dbus.Dictionary()
601
602
(self.mandos_serv
706
707
self.topwidget.set_focus(self.logbox)
709
if log.level < logging.INFO:
710
log.setLevel(logging.INFO)
711
log.info("Verbose mode: Off")
710
if self.log_level == 0:
712
self.log_message("Verbose mode: Off")
713
log.setLevel(logging.NOTSET)
714
log.info("Verbose mode: On")
715
self.log_message("Verbose mode: On")
715
716
# elif (key == "end" or key == "meta >" or key == "G"
716
717
# or key == ">"):
717
718
# 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)
752
740
ui = UserInterface()
755
743
except KeyboardInterrupt:
756
with warnings.catch_warnings():
757
warnings.filterwarnings("ignore", "", BytesWarning)
760
with warnings.catch_warnings():
761
warnings.filterwarnings("ignore", "", BytesWarning)
745
except Exception as e:
746
ui.log_message(str(e))