1
#!/usr/bin/python3 -bbI
2
2
# -*- mode: python; coding: utf-8 -*-
4
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2016 Teddy Hogeborn
7
# Copyright © 2009-2016 Björn Påhlsson
9
# This program is free software: you can redistribute it and/or modify
10
# it under the terms of the GNU General Public License as published by
6
# Copyright © 2009-2019 Teddy Hogeborn
7
# Copyright © 2009-2019 Björn Påhlsson
9
# This file is part of Mandos.
11
# Mandos is free software: you can redistribute it and/or modify it
12
# under the terms of the GNU General Public License as published by
11
13
# the Free Software Foundation, either version 3 of the License, or
12
14
# (at your option) any later version.
14
# This program is distributed in the hope that it will be useful,
15
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16
# Mandos is distributed in the hope that it will be useful, but
17
# WITHOUT ANY WARRANTY; without even the implied warranty of
16
18
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
19
# GNU General Public License for more details.
19
21
# You should have received a copy of the GNU General Public License
20
# along with this program. If not, see
21
# <http://www.gnu.org/licenses/>.
22
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
23
24
# Contact the authors at <mandos@recompile.se>.
26
26
from __future__ import (division, absolute_import, print_function,
29
30
from future_builtins import *
30
31
except ImportError:
38
41
import urwid.curses_display
50
49
if sys.version_info.major == 2:
53
locale.setlocale(locale.LC_ALL, '')
55
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)
67
logging.getLogger("urwid").setLevel(logging.INFO)
57
69
# Some useful constants
58
domain = 'se.recompile'
59
server_interface = domain + '.Mandos'
60
client_interface = domain + '.Mandos.Client'
70
domain = "se.recompile"
71
server_interface = domain + ".Mandos"
72
client_interface = domain + ".Mandos.Client"
64
76
dbus.OBJECT_MANAGER_IFACE
116
128
self.property_changed_match.remove()
119
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
131
class MandosClientWidget(MandosClientPropertyCache, urwid.Widget):
120
132
"""A Mandos Client which is visible on the screen.
135
_sizing = frozenset(["flow"])
123
137
def __init__(self, server_proxy_object=None, update_hook=None,
124
delete_hook=None, logger=None, **kwargs):
138
delete_hook=None, **kwargs):
125
139
# Called on update
126
140
self.update_hook = update_hook
127
141
# Called on delete
128
142
self.delete_hook = delete_hook
129
143
# Mandos Server proxy object
130
144
self.server_proxy_object = server_proxy_object
134
146
self._update_timer_callback_tag = None
172
183
if flag and self._update_timer_callback_tag is None:
173
184
# Will update the shown timer value every second
174
self._update_timer_callback_tag = (GLib.timeout_add
185
self._update_timer_callback_tag = (
186
GLib.timeout_add(1000,
187
glib_safely(self.update_timer)))
177
188
elif not (flag or self._update_timer_callback_tag is None):
178
189
GLib.source_remove(self._update_timer_callback_tag)
179
190
self._update_timer_callback_tag = None
181
192
def checker_completed(self, exitstatus, condition, command):
182
193
if exitstatus == 0:
183
self.logger('Checker for client {} (command "{}")'
184
' succeeded'.format(self.properties["Name"],
194
log.debug('Checker for client %s (command "%s")'
195
" succeeded", self.properties["Name"], command)
189
199
if os.WIFEXITED(condition):
190
self.logger('Checker for client {} (command "{}") failed'
192
.format(self.properties["Name"], command,
193
os.WEXITSTATUS(condition)))
200
log.info('Checker for client %s (command "%s") failed'
201
" with exit code %d", self.properties["Name"],
202
command, os.WEXITSTATUS(condition))
194
203
elif os.WIFSIGNALED(condition):
195
self.logger('Checker for client {} (command "{}") was'
196
' killed by signal {}'
197
.format(self.properties["Name"], command,
198
os.WTERMSIG(condition)))
204
log.info('Checker for client %s (command "%s") was'
205
" killed by signal %d", self.properties["Name"],
206
command, os.WTERMSIG(condition))
201
209
def checker_started(self, command):
202
210
"""Server signals that a checker started."""
203
self.logger('Client {} started checker "{}"'
204
.format(self.properties["Name"],
211
log.debug('Client %s started checker "%s"',
212
self.properties["Name"], command)
207
214
def got_secret(self):
208
self.logger('Client {} received its secret'
209
.format(self.properties["Name"]))
215
log.info("Client %s received its secret",
216
self.properties["Name"])
211
218
def need_approval(self, timeout, default):
213
message = 'Client {} needs approval within {} seconds'
220
message = "Client %s needs approval within %f seconds"
215
message = 'Client {} will get its secret in {} seconds'
216
self.logger(message.format(self.properties["Name"],
222
message = "Client %s will get its secret in %f seconds"
223
log.info(message, self.properties["Name"], timeout/1000)
219
225
def rejected(self, reason):
220
self.logger('Client {} was rejected; reason: {}'
221
.format(self.properties["Name"], reason))
226
log.info("Client %s was rejected; reason: %s",
227
self.properties["Name"], reason)
223
229
def selectable(self):
224
230
"""Make this a "selectable" widget.
225
This overrides the method from urwid.FlowWidget."""
231
This overrides the method from urwid.Widget."""
228
234
def rows(self, maxcolrow, focus=False):
229
235
"""How many rows this widget will occupy might depend on
230
236
whether we have focus or not.
231
This overrides the method from urwid.FlowWidget"""
237
This overrides the method from urwid.Widget"""
232
238
return self.current_widget(focus).rows(maxcolrow, focus=focus)
234
240
def current_widget(self, focus=False):
327
333
def render(self, maxcolrow, focus=False):
328
334
"""Render differently if we have focus.
329
This overrides the method from urwid.FlowWidget"""
335
This overrides the method from urwid.Widget"""
330
336
return self.current_widget(focus).render(maxcolrow,
333
339
def keypress(self, maxcolrow, key):
335
This overrides the method from urwid.FlowWidget"""
341
This overrides the method from urwid.Widget"""
337
343
self.proxy.Set(client_interface, "Enabled",
338
344
dbus.Boolean(True), ignore_reply=True,
457
471
self.log_visible = True
458
472
self.log_wrap = "any"
474
self.loghandler = UILogHandler(self)
461
self.log_message_raw(("bold",
462
"Mandos Monitor version " + version))
463
self.log_message_raw(("bold",
477
self.add_log_line(("bold",
478
"Mandos Monitor version " + version))
479
self.add_log_line(("bold", "q: Quit ?: Help"))
466
self.busname = domain + '.Mandos'
481
self.busname = domain + ".Mandos"
467
482
self.main_loop = GLib.MainLoop()
469
def client_not_found(self, fingerprint, address):
470
self.log_message("Client with address {} and fingerprint {}"
471
" could not be found"
472
.format(address, fingerprint))
484
def client_not_found(self, key_id, address):
485
log.info("Client with address %s and key ID %s could"
486
" not be found", address, key_id)
474
488
def rebuild(self):
475
489
"""This rebuilds the User Interface.
486
500
self.uilist.append(self.logbox)
487
501
self.topwidget = urwid.Pile(self.uilist)
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:
503
def add_log_line(self, markup):
501
504
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
505
if self.max_log_length:
503
506
if len(self.log) > self.max_log_length:
504
del self.log[0:len(self.log)-self.max_log_length-1]
505
self.logbox.set_focus(len(self.logbox.body.contents),
507
del self.log[0:(len(self.log) - self.max_log_length)]
508
self.logbox.set_focus(len(self.logbox.body.contents)-1,
506
509
coming_from="above")
622
624
proxy_object=client_proxy_object,
623
625
properties=client,
624
626
update_hook=self.refresh,
625
delete_hook=self.remove_client,
626
logger=self.log_message),
627
delete_hook=self.remove_client),
630
self._input_callback_tag = (GLib.io_add_watch
631
self._input_callback_tag = (
633
GLib.IOChannel.unix_new(sys.stdin.fileno()),
634
GLib.PRIORITY_DEFAULT, GLib.IO_IN,
635
glib_safely(self.process_input)))
634
636
self.main_loop.run()
635
637
# Main loop has finished, we should close everything now
636
638
GLib.source_remove(self._input_callback_tag)
639
with warnings.catch_warnings():
640
warnings.simplefilter("ignore", BytesWarning)
640
644
self.main_loop.quit()
645
log.removeHandler(self.loghandler)
646
log.propagate = self.orig_log_propagate
642
648
def process_input(self, source, condition):
643
649
keys = self.screen.get_input()
742
class UILogHandler(logging.Handler):
743
def __init__(self, ui, *args, **kwargs):
745
super(UILogHandler, self).__init__(*args, **kwargs)
747
logging.Formatter("%(asctime)s: %(message)s"))
748
def emit(self, record):
749
msg = self.format(record)
750
if record.levelno > logging.INFO:
752
self.ui.add_log_line(msg)
737
755
ui = UserInterface()
740
758
except KeyboardInterrupt:
742
except Exception as e:
743
ui.log_message(str(e))
759
with warnings.catch_warnings():
760
warnings.filterwarnings("ignore", "", BytesWarning)
763
with warnings.catch_warnings():
764
warnings.filterwarnings("ignore", "", BytesWarning)