/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk
1185 by Teddy Hogeborn
Add -I option to python 3 on the interpreter script ("shebang") line
1
#!/usr/bin/python3 -bbI
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
2
# -*- mode: python; coding: utf-8 -*-
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
3
#
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
4
# Mandos Monitor - Control and monitor the Mandos server
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
5
#
969 by Teddy Hogeborn
Update copyright year to 2019
6
# Copyright © 2009-2019 Teddy Hogeborn
7
# Copyright © 2009-2019 Björn Påhlsson
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
8
#
907 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
9
# This file is part of Mandos.
10
#
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
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
13
# the Free Software Foundation, either version 3 of the License, or
14
# (at your option) any later version.
15
#
907 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
16
#     Mandos is distributed in the hope that it will be useful, but
17
#     WITHOUT ANY WARRANTY; without even the implied warranty of
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
18
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19
#     GNU General Public License for more details.
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
20
#
444 by Teddy Hogeborn
Update copyright year to "2010" wherever appropriate.
21
# You should have received a copy of the GNU General Public License
907 by Teddy Hogeborn
Alter copyright notices slightly. Actual license is unchanged!
22
# along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
23
#
505.1.2 by Teddy Hogeborn
Change "fukt.bsnet.se" to "recompile.se" throughout.
24
# Contact the authors at <mandos@recompile.se>.
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
25
#
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
26
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
27
from __future__ import (division, absolute_import, print_function,
28
                        unicode_literals)
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
29
try:
30
    from future_builtins import *
31
except ImportError:
32
    pass
579 by Teddy Hogeborn
* mandos: Use all new builtins.
33
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
34
import sys
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
35
import os
1174 by Teddy Hogeborn
Address Python 3 bytes/str warnings
36
import warnings
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
37
import datetime
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
38
import locale
39
import logging
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
40
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
41
import urwid.curses_display
42
import urwid
43
44
from dbus.mainloop.glib import DBusGMainLoop
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
45
from gi.repository import GLib
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
46
47
import dbus
48
723.1.7 by Teddy Hogeborn
Use the .major attribute on sys.version_info instead of using "[0]".
49
if sys.version_info.major == 2:
1192 by teddy at recompile
Simplification of Python 3 compatibility code
50
    __metaclass__ = type
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
51
    str = unicode
52
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
53
log = logging.getLogger(os.path.basename(sys.argv[0]))
54
logging.basicConfig(level="NOTSET", # Show all messages
55
                    format="%(message)s") # Show basic log messages
56
57
logging.captureWarnings(True)   # Show warnings via the logging system
58
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
59
locale.setlocale(locale.LC_ALL, "")
411 by Teddy Hogeborn
More consistent terminology: Clients are no longer "invalid" - they
60
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
61
logging.getLogger("dbus.proxies").setLevel(logging.CRITICAL)
24.1.153 by Björn Påhlsson
early commit to ease todays coding
62
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
63
# Some useful constants
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
64
domain = "se.recompile"
65
server_interface = domain + ".Mandos"
66
client_interface = domain + ".Mandos.Client"
237.4.124 by Teddy Hogeborn
* Makefile (version): Change to 1.8.9.
67
version = "1.8.9"
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
68
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
69
try:
70
    dbus.OBJECT_MANAGER_IFACE
71
except AttributeError:
72
    dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
73
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
74
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
75
def isoformat_to_datetime(iso):
76
    "Parse an ISO 8601 date string to a datetime.datetime()"
77
    if not iso:
78
        return None
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
79
    d, t = iso.split("T", 1)
80
    year, month, day = d.split("-", 2)
81
    hour, minute, second = t.split(":", 2)
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
82
    second, fraction = divmod(float(second), 1)
83
    return datetime.datetime(int(year),
84
                             int(month),
85
                             int(day),
86
                             int(hour),
87
                             int(minute),
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
88
                             int(second),            # Whole seconds
89
                             int(fraction*1000000))  # Microseconds
90
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
91
1192 by teddy at recompile
Simplification of Python 3 compatibility code
92
class MandosClientPropertyCache:
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
93
    """This wraps a Mandos Client D-Bus proxy object, caches the
94
    properties and calls a hook function when any of them are
95
    changed.
96
    """
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
97
    def __init__(self, proxy_object=None, properties=None, **kwargs):
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
98
        self.proxy = proxy_object  # Mandos Client proxy object
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
99
        self.properties = dict() if properties is None else properties
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
100
        self.property_changed_match = (
732 by Teddy Hogeborn
Emit D-Bus "org.freedesktop.DBus.Properties.PropertiesChanged" signal.
101
            self.proxy.connect_to_signal("PropertiesChanged",
102
                                         self.properties_changed,
103
                                         dbus.PROPERTIES_IFACE,
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
104
                                         byte_arrays=True))
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
105
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
106
        if properties is None:
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
107
            self.properties.update(self.proxy.GetAll(
108
                client_interface,
109
                dbus_interface=dbus.PROPERTIES_IFACE))
110
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
111
        super(MandosClientPropertyCache, self).__init__(**kwargs)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
112
732 by Teddy Hogeborn
Emit D-Bus "org.freedesktop.DBus.Properties.PropertiesChanged" signal.
113
    def properties_changed(self, interface, properties, invalidated):
114
        """This is called whenever we get a PropertiesChanged signal
115
        It updates the changed properties in the "properties" dict.
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
116
        """
117
        # Update properties dict with new value
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
118
        if interface == client_interface:
119
            self.properties.update(properties)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
120
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
121
    def delete(self):
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
122
        self.property_changed_match.remove()
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
123
124
125
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
126
    """A Mandos Client which is visible on the screen.
127
    """
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
128
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
129
    def __init__(self, server_proxy_object=None, update_hook=None,
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
130
                 delete_hook=None, **kwargs):
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
131
        # Called on update
132
        self.update_hook = update_hook
133
        # Called on delete
134
        self.delete_hook = delete_hook
135
        # Mandos Server proxy object
136
        self.server_proxy_object = server_proxy_object
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
137
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
138
        self._update_timer_callback_tag = None
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
139
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
140
        # The widget shown normally
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
141
        self._text_widget = urwid.Text("")
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
142
        # The widget shown when we have focus
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
143
        self._focus_text_widget = urwid.Text("")
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
144
        super(MandosClientWidget, self).__init__(**kwargs)
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
145
        self.update()
146
        self.opened = False
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
147
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
148
        self.match_objects = (
149
            self.proxy.connect_to_signal("CheckerCompleted",
150
                                         self.checker_completed,
151
                                         client_interface,
152
                                         byte_arrays=True),
153
            self.proxy.connect_to_signal("CheckerStarted",
154
                                         self.checker_started,
155
                                         client_interface,
156
                                         byte_arrays=True),
157
            self.proxy.connect_to_signal("GotSecret",
158
                                         self.got_secret,
159
                                         client_interface,
160
                                         byte_arrays=True),
161
            self.proxy.connect_to_signal("NeedApproval",
162
                                         self.need_approval,
163
                                         client_interface,
164
                                         byte_arrays=True),
165
            self.proxy.connect_to_signal("Rejected",
166
                                         self.rejected,
167
                                         client_interface,
168
                                         byte_arrays=True))
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
169
        log.debug("Created client %s", self.properties["Name"])
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
170
442 by Teddy Hogeborn
* DBUS-API: Document new "LastApprovalRequest" client property.
171
    def using_timer(self, flag):
172
        """Call this method with True or False when timer should be
173
        activated or deactivated.
174
        """
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
175
        if flag and self._update_timer_callback_tag is None:
24.1.179 by Björn Påhlsson
New feature:
176
            # Will update the shown timer value every second
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
177
            self._update_timer_callback_tag = (
178
                GLib.timeout_add(1000,
179
                                 glib_safely(self.update_timer)))
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
180
        elif not (flag or self._update_timer_callback_tag is None):
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
181
            GLib.source_remove(self._update_timer_callback_tag)
442 by Teddy Hogeborn
* DBUS-API: Document new "LastApprovalRequest" client property.
182
            self._update_timer_callback_tag = None
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
183
783 by Teddy Hogeborn
Revert change to D-Bus API.
184
    def checker_completed(self, exitstatus, condition, command):
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
185
        if exitstatus == 0:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
186
            log.debug('Checker for client %s (command "%s")'
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
187
                      " succeeded", self.properties["Name"], command)
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
188
            self.update()
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
189
            return
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
190
        # Checker failed
783 by Teddy Hogeborn
Revert change to D-Bus API.
191
        if os.WIFEXITED(condition):
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
192
            log.info('Checker for client %s (command "%s") failed'
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
193
                     " with exit code %d", self.properties["Name"],
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
194
                     command, os.WEXITSTATUS(condition))
783 by Teddy Hogeborn
Revert change to D-Bus API.
195
        elif os.WIFSIGNALED(condition):
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
196
            log.info('Checker for client %s (command "%s") was'
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
197
                     " killed by signal %d", self.properties["Name"],
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
198
                     command, os.WTERMSIG(condition))
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
199
        self.update()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
200
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
201
    def checker_started(self, command):
713 by Teddy Hogeborn
mandos-monitor: New "verbose" mode to see less important log messages.
202
        """Server signals that a checker started."""
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
203
        log.debug('Client %s started checker "%s"',
204
                  self.properties["Name"], command)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
205
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
206
    def got_secret(self):
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
207
        log.info("Client %s received its secret",
208
                 self.properties["Name"])
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
209
24.1.153 by Björn Påhlsson
early commit to ease todays coding
210
    def need_approval(self, timeout, default):
211
        if not default:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
212
            message = "Client %s needs approval within %f seconds"
24.1.153 by Björn Påhlsson
early commit to ease todays coding
213
        else:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
214
            message = "Client %s will get its secret in %f seconds"
215
        log.info(message, self.properties["Name"], timeout/1000)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
216
24.1.153 by Björn Påhlsson
early commit to ease todays coding
217
    def rejected(self, reason):
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
218
        log.info("Client %s was rejected; reason: %s",
219
                 self.properties["Name"], reason)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
220
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
221
    def selectable(self):
222
        """Make this a "selectable" widget.
223
        This overrides the method from urwid.FlowWidget."""
224
        return True
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
225
463.1.1 by teddy at bsnet
* mandos-monitor: Use only unicode string literals.
226
    def rows(self, maxcolrow, focus=False):
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
227
        """How many rows this widget will occupy might depend on
228
        whether we have focus or not.
229
        This overrides the method from urwid.FlowWidget"""
463.1.1 by teddy at bsnet
* mandos-monitor: Use only unicode string literals.
230
        return self.current_widget(focus).rows(maxcolrow, focus=focus)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
231
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
232
    def current_widget(self, focus=False):
233
        if focus or self.opened:
234
            return self._focus_widget
235
        return self._widget
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
236
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
237
    def update(self):
238
        "Called when what is visible on the screen should be updated."
239
        # How to add standout mode to a style
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
240
        with_standout = {"normal": "standout",
241
                         "bold": "bold-standout",
242
                         "underline-blink":
243
                         "underline-blink-standout",
244
                         "bold-underline-blink":
245
                         "bold-underline-blink-standout",
246
                         }
247
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
248
        # Rebuild focus and non-focus widgets using current properties
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
249
24.1.154 by Björn Påhlsson
merge
250
        # Base part of a client. Name!
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
251
        base = "{name}: ".format(name=self.properties["Name"])
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
252
        if not self.properties["Enabled"]:
253
            message = "DISABLED"
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
254
            self.using_timer(False)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
255
        elif self.properties["ApprovalPending"]:
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
256
            timeout = datetime.timedelta(
257
                milliseconds=self.properties["ApprovalDelay"])
442 by Teddy Hogeborn
* DBUS-API: Document new "LastApprovalRequest" client property.
258
            last_approval_request = isoformat_to_datetime(
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
259
                self.properties["LastApprovalRequest"])
442 by Teddy Hogeborn
* DBUS-API: Document new "LastApprovalRequest" client property.
260
            if last_approval_request is not None:
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
261
                timer = max(timeout - (datetime.datetime.utcnow()
262
                                       - last_approval_request),
263
                            datetime.timedelta())
442 by Teddy Hogeborn
* DBUS-API: Document new "LastApprovalRequest" client property.
264
            else:
265
                timer = datetime.timedelta()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
266
            if self.properties["ApprovedByDefault"]:
723.1.5 by Teddy Hogeborn
Use the new auto-numbered "{}" syntax for the .format() string method.
267
                message = "Approval in {}. (d)eny?"
24.1.159 by Björn Påhlsson
added approval to mandos-ctl
268
            else:
723.1.5 by Teddy Hogeborn
Use the new auto-numbered "{}" syntax for the .format() string method.
269
                message = "Denial in {}. (a)pprove?"
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
270
            message = message.format(str(timer).rsplit(".", 1)[0])
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
271
            self.using_timer(True)
557 by Teddy Hogeborn
Use the new Client.LastCheckerStatus property.
272
        elif self.properties["LastCheckerStatus"] != 0:
572 by Teddy Hogeborn
* mandos-ctl: Break long lines.
273
            # When checker has failed, show timer until client expires
24.1.179 by Björn Påhlsson
New feature:
274
            expires = self.properties["Expires"]
275
            if expires == "":
276
                timer = datetime.timedelta(0)
277
            else:
572 by Teddy Hogeborn
* mandos-ctl: Break long lines.
278
                expires = (datetime.datetime.strptime
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
279
                           (expires, "%Y-%m-%dT%H:%M:%S.%f"))
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
280
                timer = max(expires - datetime.datetime.utcnow(),
281
                            datetime.timedelta())
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
282
            message = ("A checker has failed! Time until client"
283
                       " gets disabled: {}"
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
284
                       .format(str(timer).rsplit(".", 1)[0]))
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
285
            self.using_timer(True)
24.1.154 by Björn Påhlsson
merge
286
        else:
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
287
            message = "enabled"
581 by Teddy Hogeborn
* mandos (Client.enable, Client.disable, ClientDBus.approve): Call
288
            self.using_timer(False)
723.1.5 by Teddy Hogeborn
Use the new auto-numbered "{}" syntax for the .format() string method.
289
        self._text = "{}{}".format(base, message)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
290
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
291
        if not urwid.supports_unicode():
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
292
            self._text = self._text.encode("ascii", "replace")
293
        textlist = [("normal", self._text)]
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
294
        self._text_widget.set_text(textlist)
295
        self._focus_text_widget.set_text([(with_standout[text[0]],
296
                                           text[1])
297
                                          if isinstance(text, tuple)
298
                                          else text
299
                                          for text in textlist])
300
        self._widget = self._text_widget
301
        self._focus_widget = urwid.AttrWrap(self._focus_text_widget,
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
302
                                            "standout")
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
303
        # Run update hook, if any
304
        if self.update_hook is not None:
305
            self.update_hook()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
306
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
307
    def update_timer(self):
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
308
        """called by GLib. Will indefinitely loop until
309
        GLib.source_remove() on tag is called
310
        """
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
311
        self.update()
312
        return True             # Keep calling this
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
313
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
314
    def delete(self, **kwargs):
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
315
        if self._update_timer_callback_tag is not None:
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
316
            GLib.source_remove(self._update_timer_callback_tag)
417 by Teddy Hogeborn
* mandos (AvahiService.entry_group_state_changed): Better debug log
317
            self._update_timer_callback_tag = None
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
318
        for match in self.match_objects:
319
            match.remove()
320
        self.match_objects = ()
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
321
        if self.delete_hook is not None:
322
            self.delete_hook(self)
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
323
        return super(MandosClientWidget, self).delete(**kwargs)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
324
463.1.1 by teddy at bsnet
* mandos-monitor: Use only unicode string literals.
325
    def render(self, maxcolrow, focus=False):
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
326
        """Render differently if we have focus.
327
        This overrides the method from urwid.FlowWidget"""
463.1.1 by teddy at bsnet
* mandos-monitor: Use only unicode string literals.
328
        return self.current_widget(focus).render(maxcolrow,
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
329
                                                 focus=focus)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
330
463.1.1 by teddy at bsnet
* mandos-monitor: Use only unicode string literals.
331
    def keypress(self, maxcolrow, key):
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
332
        """Handle keys.
333
        This overrides the method from urwid.FlowWidget"""
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
334
        if key == "+":
781 by Teddy Hogeborn
Deprecate some D-Bus methods in favor of D-Bus properties.
335
            self.proxy.Set(client_interface, "Enabled",
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
336
                           dbus.Boolean(True), ignore_reply=True,
337
                           dbus_interface=dbus.PROPERTIES_IFACE)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
338
        elif key == "-":
781 by Teddy Hogeborn
Deprecate some D-Bus methods in favor of D-Bus properties.
339
            self.proxy.Set(client_interface, "Enabled", False,
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
340
                           ignore_reply=True,
341
                           dbus_interface=dbus.PROPERTIES_IFACE)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
342
        elif key == "a":
24.1.154 by Björn Påhlsson
merge
343
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
344
                               dbus_interface=client_interface,
482 by Teddy Hogeborn
* mandos: Tolerate restarting Avahi servers. Also Changed to new
345
                               ignore_reply=True)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
346
        elif key == "d":
24.1.154 by Björn Påhlsson
merge
347
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
348
                               dbus_interface=client_interface,
482 by Teddy Hogeborn
* mandos: Tolerate restarting Avahi servers. Also Changed to new
349
                               ignore_reply=True)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
350
        elif key == "R" or key == "_" or key == "ctrl k":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
351
            self.server_proxy_object.RemoveClient(self.proxy
482 by Teddy Hogeborn
* mandos: Tolerate restarting Avahi servers. Also Changed to new
352
                                                  .object_path,
353
                                                  ignore_reply=True)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
354
        elif key == "s":
781 by Teddy Hogeborn
Deprecate some D-Bus methods in favor of D-Bus properties.
355
            self.proxy.Set(client_interface, "CheckerRunning",
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
356
                           dbus.Boolean(True), ignore_reply=True,
357
                           dbus_interface=dbus.PROPERTIES_IFACE)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
358
        elif key == "S":
781 by Teddy Hogeborn
Deprecate some D-Bus methods in favor of D-Bus properties.
359
            self.proxy.Set(client_interface, "CheckerRunning",
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
360
                           dbus.Boolean(False), ignore_reply=True,
361
                           dbus_interface=dbus.PROPERTIES_IFACE)
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
362
        elif key == "C":
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
363
            self.proxy.CheckedOK(dbus_interface=client_interface,
482 by Teddy Hogeborn
* mandos: Tolerate restarting Avahi servers. Also Changed to new
364
                                 ignore_reply=True)
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
365
        # xxx
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
366
#         elif key == "p" or key == "=":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
367
#             self.proxy.pause()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
368
#         elif key == "u" or key == ":":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
369
#             self.proxy.unpause()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
370
#         elif key == "RET":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
371
#             self.open()
372
        else:
373
            return key
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
374
732 by Teddy Hogeborn
Emit D-Bus "org.freedesktop.DBus.Properties.PropertiesChanged" signal.
375
    def properties_changed(self, interface, properties, invalidated):
376
        """Call self.update() if any properties changed.
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
377
        This overrides the method from MandosClientPropertyCache"""
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
378
        old_values = {key: self.properties.get(key)
379
                      for key in properties.keys()}
732 by Teddy Hogeborn
Emit D-Bus "org.freedesktop.DBus.Properties.PropertiesChanged" signal.
380
        super(MandosClientWidget, self).properties_changed(
381
            interface, properties, invalidated)
382
        if any(old_values[key] != self.properties.get(key)
383
               for key in old_values):
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
384
            self.update()
385
386
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
387
def glib_safely(func, retval=True):
388
    def safe_func(*args, **kwargs):
389
        try:
390
            return func(*args, **kwargs)
391
        except Exception:
392
            log.exception("")
393
            return retval
394
    return safe_func
395
396
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
397
class ConstrainedListBox(urwid.ListBox):
398
    """Like a normal urwid.ListBox, but will consume all "up" or
399
    "down" key presses, thus not allowing any containing widgets to
400
    use them as an excuse to shift focus away from this widget.
401
    """
580 by Teddy Hogeborn
* mandos-monitor: Speedup: Use properties from D-Bus
402
    def keypress(self, *args, **kwargs):
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
403
        ret = (super(ConstrainedListBox, self)
404
               .keypress(*args, **kwargs))
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
405
        if ret in ("up", "down"):
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
406
            return
407
        return ret
408
409
1192 by teddy at recompile
Simplification of Python 3 compatibility code
410
class UserInterface:
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
411
    """This is the entire user interface - the whole screen
412
    with boxes, lists of client widgets, etc.
413
    """
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
414
    def __init__(self, max_log_length=1000):
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
415
        DBusGMainLoop(set_as_default=True)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
416
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
417
        self.screen = urwid.curses_display.Screen()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
418
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
419
        self.screen.register_palette((
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
420
                ("normal",
421
                 "default", "default", None),
422
                ("bold",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
423
                 "bold", "default", "bold"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
424
                ("underline-blink",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
425
                 "underline,blink", "default", "underline,blink"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
426
                ("standout",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
427
                 "standout", "default", "standout"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
428
                ("bold-underline-blink",
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
429
                 "bold,underline,blink", "default",
430
                 "bold,underline,blink"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
431
                ("bold-standout",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
432
                 "bold,standout", "default", "bold,standout"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
433
                ("underline-blink-standout",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
434
                 "underline,blink,standout", "default",
435
                 "underline,blink,standout"),
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
436
                ("bold-underline-blink-standout",
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
437
                 "bold,underline,blink,standout", "default",
438
                 "bold,underline,blink,standout"),
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
439
                ))
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
440
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
441
        if urwid.supports_unicode():
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
442
            self.divider = "─"  # \u2500
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
443
        else:
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
444
            self.divider = "_"  # \u005f
445
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
446
        self.screen.start()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
447
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
448
        self.size = self.screen.get_cols_rows()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
449
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
450
        self.clients = urwid.SimpleListWalker([])
451
        self.clients_dict = {}
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
452
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
453
        # We will add Text widgets to this list
968 by Teddy Hogeborn
Do minor fix to make mandos-monitor work with new python-urwid
454
        self.log = urwid.SimpleListWalker([])
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
455
        self.max_log_length = max_log_length
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
456
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
457
        # We keep a reference to the log widget so we can remove it
458
        # from the ListWalker without it getting destroyed
459
        self.logbox = ConstrainedListBox(self.log)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
460
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
461
        # This keeps track of whether self.uilist currently has
462
        # self.logbox in it or not
463
        self.log_visible = True
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
464
        self.log_wrap = "any"
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
465
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
466
        self.loghandler = UILogHandler(self)
467
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
468
        self.rebuild()
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
469
        self.add_log_line(("bold",
470
                           "Mandos Monitor version " + version))
471
        self.add_log_line(("bold", "q: Quit  ?: Help"))
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
472
1184 by Teddy Hogeborn
mandos-monitor: Formatting changes only
473
        self.busname = domain + ".Mandos"
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
474
        self.main_loop = GLib.MainLoop()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
475
1175 by Teddy Hogeborn
mandos-monitor: Update message from "fingerprint" to "key ID"
476
    def client_not_found(self, key_id, address):
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
477
        log.info("Client with address %s and key ID %s could"
478
                 " not be found", address, key_id)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
479
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
480
    def rebuild(self):
481
        """This rebuilds the User Interface.
482
        Call this when the widget layout needs to change"""
483
        self.uilist = []
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
484
        # self.uilist.append(urwid.ListBox(self.clients))
422 by Teddy Hogeborn
Rename all D-Bus properties to conform to D-Bus naming conventions;
485
        self.uilist.append(urwid.Frame(ConstrainedListBox(self.
486
                                                          clients),
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
487
                                       # header=urwid.Divider(),
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
488
                                       header=None,
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
489
                                       footer=urwid.Divider(
490
                                           div_char=self.divider)))
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
491
        if self.log_visible:
492
            self.uilist.append(self.logbox)
493
        self.topwidget = urwid.Pile(self.uilist)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
494
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
495
    def add_log_line(self, markup):
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
496
        self.log.append(urwid.Text(markup, wrap=self.log_wrap))
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
497
        if self.max_log_length:
498
            if len(self.log) > self.max_log_length:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
499
                del self.log[0:(len(self.log) - self.max_log_length)]
968 by Teddy Hogeborn
Do minor fix to make mandos-monitor work with new python-urwid
500
        self.logbox.set_focus(len(self.logbox.body.contents)-1,
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
501
                              coming_from="above")
409 by Teddy Hogeborn
* mandos (MandosServer.handle_ipc): Better log message.
502
        self.refresh()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
503
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
504
    def toggle_log_display(self):
505
        """Toggle visibility of the log buffer."""
506
        self.log_visible = not self.log_visible
507
        self.rebuild()
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
508
        log.debug("Log visibility changed to: %s", self.log_visible)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
509
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
510
    def change_log_display(self):
511
        """Change type of log display.
512
        Currently, this toggles wrapping of text lines."""
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
513
        if self.log_wrap == "clip":
514
            self.log_wrap = "any"
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
515
        else:
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
516
            self.log_wrap = "clip"
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
517
        for textwidget in self.log:
518
            textwidget.set_wrap_mode(self.log_wrap)
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
519
        log.debug("Wrap mode: %s", self.log_wrap)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
520
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
521
    def find_and_remove_client(self, path, interfaces):
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
522
        """Find a client by its object path and remove it.
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
523
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
524
        This is connected to the InterfacesRemoved signal from the
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
525
        Mandos server object."""
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
526
        if client_interface not in interfaces:
527
            # Not a Mandos client object; ignore
528
            return
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
529
        try:
530
            client = self.clients_dict[path]
531
        except KeyError:
532
            # not found?
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
533
            log.warning("Unknown client %s removed", path)
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
534
            return
488 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Save match tag for
535
        client.delete()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
536
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
537
    def add_new_client(self, path, ifs_and_props):
538
        """Find a client by its object path and remove it.
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
539
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
540
        This is connected to the InterfacesAdded signal from the
541
        Mandos server object.
542
        """
543
        if client_interface not in ifs_and_props:
544
            # Not a Mandos client object; ignore
545
            return
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
546
        client_proxy_object = self.bus.get_object(self.busname, path)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
547
        self.add_client(MandosClientWidget(
548
            server_proxy_object=self.mandos_serv,
549
            proxy_object=client_proxy_object,
550
            update_hook=self.refresh,
551
            delete_hook=self.remove_client,
552
            properties=dict(ifs_and_props[client_interface])),
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
553
                        path=path)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
554
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
555
    def add_client(self, client, path=None):
556
        self.clients.append(client)
557
        if path is None:
558
            path = client.proxy.object_path
559
        self.clients_dict[path] = client
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
560
        self.clients.sort(key=lambda c: c.properties["Name"])
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
561
        self.refresh()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
562
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
563
    def remove_client(self, client, path=None):
564
        self.clients.remove(client)
565
        if path is None:
566
            path = client.proxy.object_path
567
        del self.clients_dict[path]
568
        self.refresh()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
569
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
570
    def refresh(self):
571
        """Redraw the screen"""
572
        canvas = self.topwidget.render(self.size, focus=True)
573
        self.screen.draw_screen(self.size, canvas)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
574
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
575
    def run(self):
576
        """Start the main loop and exit when it's done."""
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
577
        log.addHandler(self.loghandler)
578
        self.orig_log_propagate = log.propagate
579
        log.propagate = False
580
        self.orig_log_level = log.level
581
        log.setLevel("INFO")
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
582
        self.bus = dbus.SystemBus()
583
        mandos_dbus_objc = self.bus.get_object(
584
            self.busname, "/", follow_name_owner_changes=True)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
585
        self.mandos_serv = dbus.Interface(
586
            mandos_dbus_objc, dbus_interface=server_interface)
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
587
        try:
588
            mandos_clients = (self.mandos_serv
589
                              .GetAllClientsWithProperties())
612 by Björn Påhlsson
added note messages when mandos-monitor starts without running server,
590
            if not mandos_clients:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
591
                log.warning("Note: Server has no clients.")
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
592
        except dbus.exceptions.DBusException:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
593
            log.warning("Note: No Mandos server running.")
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
594
            mandos_clients = dbus.Dictionary()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
595
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
596
        (self.mandos_serv
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
597
         .connect_to_signal("InterfacesRemoved",
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
598
                            self.find_and_remove_client,
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
599
                            dbus_interface=dbus.OBJECT_MANAGER_IFACE,
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
600
                            byte_arrays=True))
601
        (self.mandos_serv
785 by Teddy Hogeborn
Support the standard org.freedesktop.DBus.ObjectManager interface.
602
         .connect_to_signal("InterfacesAdded",
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
603
                            self.add_new_client,
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
604
                            dbus_interface=dbus.OBJECT_MANAGER_IFACE,
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
605
                            byte_arrays=True))
606
        (self.mandos_serv
607
         .connect_to_signal("ClientNotFound",
608
                            self.client_not_found,
609
                            dbus_interface=server_interface,
610
                            byte_arrays=True))
617 by Teddy Hogeborn
* mandos: Bug fix: Don't print output from checkers when running in
611
        for path, client in mandos_clients.items():
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
612
            client_proxy_object = self.bus.get_object(self.busname,
613
                                                      path)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
614
            self.add_client(MandosClientWidget(
615
                server_proxy_object=self.mandos_serv,
616
                proxy_object=client_proxy_object,
617
                properties=client,
618
                update_hook=self.refresh,
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
619
                delete_hook=self.remove_client),
537 by Björn Påhlsson
nicer stacktrace when mandos-monitor fail during startup
620
                            path=path)
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
621
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
622
        self.refresh()
1169 by Teddy Hogeborn
mandos-monitor: Use new GLib.io_add_watch() call signature
623
        self._input_callback_tag = (
624
            GLib.io_add_watch(
625
                GLib.IOChannel.unix_new(sys.stdin.fileno()),
626
                GLib.PRIORITY_DEFAULT, GLib.IO_IN,
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
627
                glib_safely(self.process_input)))
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
628
        self.main_loop.run()
629
        # Main loop has finished, we should close everything now
830 by Teddy Hogeborn
Server: Use python-gi instead of old python-gobject
630
        GLib.source_remove(self._input_callback_tag)
1174 by Teddy Hogeborn
Address Python 3 bytes/str warnings
631
        with warnings.catch_warnings():
632
            warnings.simplefilter("ignore", BytesWarning)
633
            self.screen.stop()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
634
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
635
    def stop(self):
636
        self.main_loop.quit()
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
637
        log.removeHandler(self.loghandler)
638
        log.propagate = self.orig_log_propagate
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
639
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
640
    def process_input(self, source, condition):
641
        keys = self.screen.get_input()
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
642
        translations = {"ctrl n": "down",       # Emacs
643
                        "ctrl p": "up",         # Emacs
644
                        "ctrl v": "page down",  # Emacs
645
                        "meta v": "page up",    # Emacs
646
                        " ": "page down",       # less
647
                        "f": "page down",       # less
648
                        "b": "page up",         # less
649
                        "j": "down",            # vi
650
                        "k": "up",              # vi
651
                        }
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
652
        for key in keys:
653
            try:
654
                key = translations[key]
655
            except KeyError:    # :-)
656
                pass
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
657
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
658
            if key == "q" or key == "Q":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
659
                self.stop()
660
                break
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
661
            elif key == "window resize":
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
662
                self.size = self.screen.get_cols_rows()
663
                self.refresh()
722 by Teddy Hogeborn
mandos-monitor: Bug fix: Make Ctrl-L actually redraw the screen.
664
            elif key == "ctrl l":
665
                self.screen.clear()
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
666
                self.refresh()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
667
            elif key == "l" or key == "D":
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
668
                self.toggle_log_display()
669
                self.refresh()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
670
            elif key == "w" or key == "i":
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
671
                self.change_log_display()
672
                self.refresh()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
673
            elif key == "?" or key == "f1" or key == "esc":
407 by Teddy Hogeborn
* mandos-monitor (MandosClientWidget): Change "StopChecker" key to "S"
674
                if not self.log_visible:
675
                    self.log_visible = True
676
                    self.rebuild()
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
677
                self.add_log_line(("bold",
678
                                   "  ".join(("q: Quit",
679
                                              "?: Help",
680
                                              "l: Log window toggle",
681
                                              "TAB: Switch window",
682
                                              "w: Wrap (log lines)",
683
                                              "v: Toggle verbose log",
684
                                   ))))
685
                self.add_log_line(("bold",
686
                                   "  ".join(("Clients:",
687
                                              "+: Enable",
688
                                              "-: Disable",
689
                                              "R: Remove",
690
                                              "s: Start new checker",
691
                                              "S: Stop checker",
692
                                              "C: Checker OK",
693
                                              "a: Approve",
694
                                              "d: Deny",
695
                                   ))))
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
696
                self.refresh()
463.1.2 by teddy at bsnet
* mandos-monitor: Use unicode string literals. Update __future__
697
            elif key == "tab":
405 by Teddy Hogeborn
* mandos-monitor (MandosClientPropertyCache): Remove conversion of
698
                if self.topwidget.get_focus() is self.logbox:
699
                    self.topwidget.set_focus(0)
700
                else:
701
                    self.topwidget.set_focus(self.logbox)
702
                self.refresh()
713 by Teddy Hogeborn
mandos-monitor: New "verbose" mode to see less important log messages.
703
            elif key == "v":
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
704
                if log.level < logging.INFO:
705
                    log.setLevel(logging.INFO)
706
                    log.info("Verbose mode: Off")
713 by Teddy Hogeborn
mandos-monitor: New "verbose" mode to see less important log messages.
707
                else:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
708
                    log.setLevel(logging.NOTSET)
709
                    log.info("Verbose mode: On")
873 by Teddy Hogeborn
PEP8 compliance: mandos-monitor
710
            # elif (key == "end" or key == "meta >" or key == "G"
711
            #       or key == ">"):
712
            #     pass            # xxx end-of-buffer
713
            # elif (key == "home" or key == "meta <" or key == "g"
714
            #       or key == "<"):
715
            #     pass            # xxx beginning-of-buffer
716
            # elif key == "ctrl e" or key == "$":
717
            #     pass            # xxx move-end-of-line
718
            # elif key == "ctrl a" or key == "^":
719
            #     pass            # xxx move-beginning-of-line
720
            # elif key == "ctrl b" or key == "meta (" or key == "h":
721
            #     pass            # xxx left
722
            # elif key == "ctrl f" or key == "meta )" or key == "l":
723
            #     pass            # xxx right
724
            # elif key == "a":
725
            #     pass            # scroll up log
726
            # elif key == "z":
727
            #     pass            # scroll down log
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
728
            elif self.topwidget.selectable():
729
                self.topwidget.keypress(self.size, key)
730
                self.refresh()
731
        return True
732
888 by Teddy Hogeborn
Add more PEP8 compliance (as per the "pycodestyle" tool).
733
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
734
class UILogHandler(logging.Handler):
735
    def __init__(self, ui, *args, **kwargs):
736
        self.ui = ui
737
        super(UILogHandler, self).__init__(*args, **kwargs)
738
        self.setFormatter(
739
            logging.Formatter("%(asctime)s: %(message)s"))
740
    def emit(self, record):
741
        msg = self.format(record)
742
        if record.levelno > logging.INFO:
743
            msg = ("bold", msg)
744
        self.ui.add_log_line(msg)
745
746
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
747
ui = UserInterface()
748
try:
749
    ui.run()
24.1.159 by Björn Påhlsson
added approval to mandos-ctl
750
except KeyboardInterrupt:
1183 by Teddy Hogeborn
mandos-monitor: Use Python's standard loggging module
751
    with warnings.catch_warnings():
752
        warnings.filterwarnings("ignore", "", BytesWarning)
753
        ui.screen.stop()
754
except Exception:
755
    with warnings.catch_warnings():
756
        warnings.filterwarnings("ignore", "", BytesWarning)
757
        ui.screen.stop()
404 by Teddy Hogeborn
* mandos-monitor: New prototype version of interactive server
758
    raise