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>.
123
130
def __init__(self, server_proxy_object=None, update_hook=None,
124
delete_hook=None, logger=None, **kwargs):
131
delete_hook=None, **kwargs):
125
132
# Called on update
126
133
self.update_hook = update_hook
127
134
# Called on delete
128
135
self.delete_hook = delete_hook
129
136
# Mandos Server proxy object
130
137
self.server_proxy_object = server_proxy_object
134
139
self._update_timer_callback_tag = None
172
176
if flag and self._update_timer_callback_tag is None:
173
177
# Will update the shown timer value every second
174
self._update_timer_callback_tag = (GLib.timeout_add
178
self._update_timer_callback_tag = (
179
GLib.timeout_add(1000,
180
glib_safely(self.update_timer)))
177
181
elif not (flag or self._update_timer_callback_tag is None):
178
182
GLib.source_remove(self._update_timer_callback_tag)
179
183
self._update_timer_callback_tag = None
181
185
def checker_completed(self, exitstatus, condition, command):
182
186
if exitstatus == 0:
183
self.logger('Checker for client {} (command "{}")'
184
' succeeded'.format(self.properties["Name"],
187
log.debug('Checker for client %s (command "%s")'
188
' succeeded', self.properties["Name"], command)
189
192
if os.WIFEXITED(condition):
190
self.logger('Checker for client {} (command "{}") failed'
192
.format(self.properties["Name"], command,
193
os.WEXITSTATUS(condition)))
193
log.info('Checker for client %s (command "%s") failed'
194
' with exit code %d', self.properties["Name"],
195
command, os.WEXITSTATUS(condition))
194
196
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)))
197
log.info('Checker for client %s (command "%s") was'
198
' killed by signal %d', self.properties["Name"],
199
command, os.WTERMSIG(condition))
201
202
def checker_started(self, command):
202
203
"""Server signals that a checker started."""
203
self.logger('Client {} started checker "{}"'
204
.format(self.properties["Name"],
204
log.debug('Client %s started checker "%s"',
205
self.properties["Name"], command)
207
207
def got_secret(self):
208
self.logger('Client {} received its secret'
209
.format(self.properties["Name"]))
208
log.info("Client %s received its secret",
209
self.properties["Name"])
211
211
def need_approval(self, timeout, default):
213
message = 'Client {} needs approval within {} seconds'
213
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"],
215
message = "Client %s will get its secret in %f seconds"
216
log.info(message, self.properties["Name"], timeout/1000)
219
218
def rejected(self, reason):
220
self.logger('Client {} was rejected; reason: {}'
221
.format(self.properties["Name"], reason))
219
log.info("Client %s was rejected; reason: %s",
220
self.properties["Name"], reason)
223
222
def selectable(self):
224
223
"""Make this a "selectable" widget.
457
464
self.log_visible = True
458
465
self.log_wrap = "any"
467
self.loghandler = UILogHandler(self)
461
self.log_message_raw(("bold",
462
"Mandos Monitor version " + version))
463
self.log_message_raw(("bold",
470
self.add_log_line(("bold",
471
"Mandos Monitor version " + version))
472
self.add_log_line(("bold", "q: Quit ?: Help"))
466
474
self.busname = domain + '.Mandos'
467
475
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))
477
def client_not_found(self, key_id, address):
478
log.info("Client with address %s and key ID %s could"
479
" not be found", address, key_id)
474
481
def rebuild(self):
475
482
"""This rebuilds the User Interface.
486
493
self.uilist.append(self.logbox)
487
494
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:
496
def add_log_line(self, markup):
501
497
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
502
498
if self.max_log_length:
503
499
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),
500
del self.log[0:(len(self.log) - self.max_log_length)]
501
self.logbox.set_focus(len(self.logbox.body.contents)-1,
506
502
coming_from="above")
592
589
mandos_clients = (self.mandos_serv
593
590
.GetAllClientsWithProperties())
594
591
if not mandos_clients:
595
self.log_message_raw(("bold",
596
"Note: Server has no clients."))
592
log.warning("Note: Server has no clients.")
597
593
except dbus.exceptions.DBusException:
598
self.log_message_raw(("bold",
599
"Note: No Mandos server running."))
594
log.warning("Note: No Mandos server running.")
600
595
mandos_clients = dbus.Dictionary()
602
597
(self.mandos_serv
622
617
proxy_object=client_proxy_object,
623
618
properties=client,
624
619
update_hook=self.refresh,
625
delete_hook=self.remove_client,
626
logger=self.log_message),
620
delete_hook=self.remove_client),
630
self._input_callback_tag = (GLib.io_add_watch
624
self._input_callback_tag = (
626
GLib.IOChannel.unix_new(sys.stdin.fileno()),
627
GLib.PRIORITY_DEFAULT, GLib.IO_IN,
628
glib_safely(self.process_input)))
634
629
self.main_loop.run()
635
630
# Main loop has finished, we should close everything now
636
631
GLib.source_remove(self._input_callback_tag)
632
with warnings.catch_warnings():
633
warnings.simplefilter("ignore", BytesWarning)
640
637
self.main_loop.quit()
638
log.removeHandler(self.loghandler)
639
log.propagate = self.orig_log_propagate
642
641
def process_input(self, source, condition):
643
642
keys = self.screen.get_input()
704
702
self.topwidget.set_focus(self.logbox)
707
if self.log_level == 0:
709
self.log_message("Verbose mode: Off")
705
if log.level < logging.INFO:
706
log.setLevel(logging.INFO)
707
log.info("Verbose mode: Off")
712
self.log_message("Verbose mode: On")
709
log.setLevel(logging.NOTSET)
710
log.info("Verbose mode: On")
713
711
# elif (key == "end" or key == "meta >" or key == "G"
714
712
# or key == ">"):
715
713
# pass # xxx end-of-buffer
735
class UILogHandler(logging.Handler):
736
def __init__(self, ui, *args, **kwargs):
738
super(UILogHandler, self).__init__(*args, **kwargs)
740
logging.Formatter("%(asctime)s: %(message)s"))
741
def emit(self, record):
742
msg = self.format(record)
743
if record.levelno > logging.INFO:
745
self.ui.add_log_line(msg)
736
748
ui = UserInterface()
739
751
except KeyboardInterrupt:
741
except Exception as e:
742
ui.log_message(str(e))
752
with warnings.catch_warnings():
753
warnings.filterwarnings("ignore", "", BytesWarning)
756
with warnings.catch_warnings():
757
warnings.filterwarnings("ignore", "", BytesWarning)