2
2
# -*- mode: python; coding: utf-8 -*-
 
4
 
# Mandos Monitor - Control and monitor the Mandos server
 
6
 
# Copyright © 2009-2011 Teddy Hogeborn
 
7
 
# Copyright © 2009-2011 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
 
11
 
# the Free Software Foundation, either version 3 of the License, or
 
12
 
# (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
 
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
17
 
#     GNU General Public License for more details.
 
19
 
# You should have received a copy of the GNU General Public License
 
20
 
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
22
 
# Contact the authors at <mandos@fukt.bsnet.se>.
 
25
 
from __future__ import division, absolute_import, print_function, unicode_literals
 
 
4
from __future__ import division, absolute_import, with_statement
 
 
61
33
urwid.curses_display.curses.A_UNDERLINE |= (
 
62
34
    urwid.curses_display.curses.A_BLINK)
 
64
 
def isoformat_to_datetime(iso):
 
65
 
    "Parse an ISO 8601 date string to a datetime.datetime()"
 
68
 
    d, t = iso.split("T", 1)
 
69
 
    year, month, day = d.split("-", 2)
 
70
 
    hour, minute, second = t.split(":", 2)
 
71
 
    second, fraction = divmod(float(second), 1)
 
72
 
    return datetime.datetime(int(year),
 
77
 
                             int(second),           # Whole seconds
 
78
 
                             int(fraction*1000000)) # Microseconds
 
80
36
class MandosClientPropertyCache(object):
 
81
37
    """This wraps a Mandos Client D-Bus proxy object, caches the
 
82
38
    properties and calls a hook function when any of them are
 
85
 
    def __init__(self, proxy_object=None, *args, **kwargs):
 
 
41
    def __init__(self, proxy_object=None, properties=None, *args,
 
86
43
        self.proxy = proxy_object # Mandos Client proxy object
 
88
 
        self.properties = dict()
 
89
 
        self.proxy.connect_to_signal("PropertyChanged",
 
 
45
        if properties is None:
 
 
46
            self.properties = dict()
 
 
48
            self.properties = properties
 
 
49
        self.proxy.connect_to_signal(u"PropertyChanged",
 
90
50
                                     self.property_changed,
 
94
 
        self.properties.update(
 
95
 
            self.proxy.GetAll(client_interface,
 
96
 
                              dbus_interface = dbus.PROPERTIES_IFACE))
 
98
 
        #XXX This break good super behaviour!
 
99
 
#        super(MandosClientPropertyCache, self).__init__(
 
 
54
        if properties is None:
 
 
55
            self.properties.update(self.proxy.GetAll(client_interface,
 
 
57
                                                     dbus.PROPERTIES_IFACE))
 
 
58
        super(MandosClientPropertyCache, self).__init__(
 
 
59
            proxy_object=proxy_object,
 
 
60
            properties=properties, *args, **kwargs)
 
102
62
    def property_changed(self, property=None, value=None):
 
103
63
        """This is called whenever we get a PropertyChanged signal
 
 
123
83
        self.logger = logger
 
125
 
        self._update_timer_callback_tag = None
 
126
 
        self._update_timer_callback_lock = 0
 
127
 
        self.last_checker_failed = False
 
129
85
        # The widget shown normally
 
130
 
        self._text_widget = urwid.Text("")
 
 
86
        self._text_widget = urwid.Text(u"")
 
131
87
        # The widget shown when we have focus
 
132
 
        self._focus_text_widget = urwid.Text("")
 
 
88
        self._focus_text_widget = urwid.Text(u"")
 
133
89
        super(MandosClientWidget, self).__init__(
 
134
90
            update_hook=update_hook, delete_hook=delete_hook,
 
137
93
        self.opened = False
 
139
 
        last_checked_ok = isoformat_to_datetime(self.properties
 
141
 
        if last_checked_ok is None:
 
142
 
            self.last_checker_failed = True
 
144
 
            self.last_checker_failed = ((datetime.datetime.utcnow()
 
151
 
        if self.last_checker_failed:
 
152
 
            self.using_timer(True)
 
154
 
        if self.need_approval:
 
155
 
            self.using_timer(True)
 
157
 
        self.proxy.connect_to_signal("CheckerCompleted",
 
 
94
        self.proxy.connect_to_signal(u"CheckerCompleted",
 
158
95
                                     self.checker_completed,
 
161
 
        self.proxy.connect_to_signal("CheckerStarted",
 
 
98
        self.proxy.connect_to_signal(u"CheckerStarted",
 
162
99
                                     self.checker_started,
 
163
100
                                     client_interface,
 
164
101
                                     byte_arrays=True)
 
165
 
        self.proxy.connect_to_signal("GotSecret",
 
 
102
        self.proxy.connect_to_signal(u"GotSecret",
 
167
104
                                     client_interface,
 
168
105
                                     byte_arrays=True)
 
169
 
        self.proxy.connect_to_signal("NeedApproval",
 
173
 
        self.proxy.connect_to_signal("Rejected",
 
 
106
        self.proxy.connect_to_signal(u"Rejected",
 
175
108
                                     client_interface,
 
176
109
                                     byte_arrays=True)
 
178
 
    def property_changed(self, property=None, value=None):
 
179
 
        super(self, MandosClientWidget).property_changed(property,
 
181
 
        if property == "ApprovalPending":
 
182
 
            using_timer(bool(value))
 
184
 
    def using_timer(self, flag):
 
185
 
        """Call this method with True or False when timer should be
 
186
 
        activated or deactivated.
 
188
 
        old = self._update_timer_callback_lock
 
190
 
            self._update_timer_callback_lock += 1
 
192
 
            self._update_timer_callback_lock -= 1
 
193
 
        if old == 0 and self._update_timer_callback_lock:
 
194
 
            self._update_timer_callback_tag = (gobject.timeout_add
 
197
 
        elif old and self._update_timer_callback_lock == 0:
 
198
 
            gobject.source_remove(self._update_timer_callback_tag)
 
199
 
            self._update_timer_callback_tag = None
 
201
111
    def checker_completed(self, exitstatus, condition, command):
 
202
112
        if exitstatus == 0:
 
203
 
            if self.last_checker_failed:
 
204
 
                self.last_checker_failed = False
 
205
 
                self.using_timer(False)
 
206
 
            #self.logger('Checker for client %s (command "%s")'
 
208
 
            #            % (self.properties["Name"], command))
 
 
113
            self.logger(u'Checker for client %s (command "%s")'
 
 
115
                        % (self.properties[u"name"], command))
 
212
 
        if not self.last_checker_failed:
 
213
 
            self.last_checker_failed = True
 
214
 
            self.using_timer(True)
 
215
117
        if os.WIFEXITED(condition):
 
216
 
            self.logger('Checker for client %s (command "%s")'
 
217
 
                        ' failed with exit code %s'
 
218
 
                        % (self.properties["Name"], command,
 
 
118
            self.logger(u'Checker for client %s (command "%s")'
 
 
119
                        u' failed with exit code %s'
 
 
120
                        % (self.properties[u"name"], command,
 
219
121
                           os.WEXITSTATUS(condition)))
 
220
 
        elif os.WIFSIGNALED(condition):
 
221
 
            self.logger('Checker for client %s (command "%s")'
 
222
 
                        ' was killed by signal %s'
 
223
 
                        % (self.properties["Name"], command,
 
 
123
        if os.WIFSIGNALED(condition):
 
 
124
            self.logger(u'Checker for client %s (command "%s")'
 
 
125
                        u' was killed by signal %s'
 
 
126
                        % (self.properties[u"name"], command,
 
224
127
                           os.WTERMSIG(condition)))
 
225
 
        elif os.WCOREDUMP(condition):
 
226
 
            self.logger('Checker for client %s (command "%s")'
 
228
 
                        % (self.properties["Name"], command))
 
230
 
            self.logger('Checker for client %s completed'
 
 
129
        if os.WCOREDUMP(condition):
 
 
130
            self.logger(u'Checker for client %s (command "%s")'
 
 
132
                        % (self.properties[u"name"], command))
 
 
133
        self.logger(u'Checker for client %s completed mysteriously')
 
234
135
    def checker_started(self, command):
 
235
 
        #self.logger('Client %s started checker "%s"'
 
236
 
        #            % (self.properties["Name"], unicode(command)))
 
 
136
        self.logger(u'Client %s started checker "%s"'
 
 
137
                    % (self.properties[u"name"], unicode(command)))
 
239
139
    def got_secret(self):
 
240
 
        self.last_checker_failed = False
 
241
 
        self.logger('Client %s received its secret'
 
242
 
                    % self.properties["Name"])
 
244
 
    def need_approval(self, timeout, default):
 
246
 
            message = 'Client %s needs approval within %s seconds'
 
248
 
            message = 'Client %s will get its secret in %s seconds'
 
250
 
                    % (self.properties["Name"], timeout/1000))
 
251
 
        self.using_timer(True)
 
253
 
    def rejected(self, reason):
 
254
 
        self.logger('Client %s was rejected; reason: %s'
 
255
 
                    % (self.properties["Name"], reason))
 
 
140
        self.logger(u'Client %s received its secret'
 
 
141
                    % self.properties[u"name"])
 
 
144
        self.logger(u'Client %s was rejected'
 
 
145
                    % self.properties[u"name"])
 
257
147
    def selectable(self):
 
258
148
        """Make this a "selectable" widget.
 
259
149
        This overrides the method from urwid.FlowWidget."""
 
262
 
    def rows(self, maxcolrow, focus=False):
 
 
152
    def rows(self, (maxcol,), focus=False):
 
263
153
        """How many rows this widget will occupy might depend on
 
264
154
        whether we have focus or not.
 
265
155
        This overrides the method from urwid.FlowWidget"""
 
266
 
        return self.current_widget(focus).rows(maxcolrow, focus=focus)
 
 
156
        return self.current_widget(focus).rows((maxcol,), focus=focus)
 
268
158
    def current_widget(self, focus=False):
 
269
159
        if focus or self.opened:
 
 
273
163
    def update(self):
 
274
164
        "Called when what is visible on the screen should be updated."
 
275
165
        # How to add standout mode to a style
 
276
 
        with_standout = { "normal": "standout",
 
277
 
                          "bold": "bold-standout",
 
279
 
                              "underline-blink-standout",
 
280
 
                          "bold-underline-blink":
 
281
 
                              "bold-underline-blink-standout",
 
 
166
        with_standout = { u"normal": u"standout",
 
 
167
                          u"bold": u"bold-standout",
 
 
169
                              u"underline-blink-standout",
 
 
170
                          u"bold-underline-blink":
 
 
171
                              u"bold-underline-blink-standout",
 
284
174
        # Rebuild focus and non-focus widgets using current properties
 
286
 
        # Base part of a client. Name!
 
288
 
                      % {"name": self.properties["Name"]})
 
289
 
        if not self.properties["Enabled"]:
 
291
 
        elif self.properties["ApprovalPending"]:
 
292
 
            timeout = datetime.timedelta(milliseconds
 
295
 
            last_approval_request = isoformat_to_datetime(
 
296
 
                self.properties["LastApprovalRequest"])
 
297
 
            if last_approval_request is not None:
 
298
 
                timer = timeout - (datetime.datetime.utcnow()
 
299
 
                                   - last_approval_request)
 
301
 
                timer = datetime.timedelta()
 
302
 
            if self.properties["ApprovedByDefault"]:
 
303
 
                message = "Approval in %s. (d)eny?"
 
305
 
                message = "Denial in %s. (a)pprove?"
 
306
 
            message = message % unicode(timer).rsplit(".", 1)[0]
 
307
 
        elif self.last_checker_failed:
 
308
 
            timeout = datetime.timedelta(milliseconds
 
311
 
            last_ok = isoformat_to_datetime(
 
312
 
                max((self.properties["LastCheckedOK"]
 
313
 
                     or self.properties["Created"]),
 
314
 
                    self.properties["LastEnabled"]))
 
315
 
            timer = timeout - (datetime.datetime.utcnow() - last_ok)
 
316
 
            message = ('A checker has failed! Time until client'
 
318
 
                           % unicode(timer).rsplit(".", 1)[0])
 
321
 
        self._text = "%s%s" % (base, message)
 
 
175
        self._text = (u'%(name)s: %(enabled)s'
 
 
176
                      % { u"name": self.properties[u"name"],
 
 
179
                               if self.properties[u"enabled"]
 
323
181
        if not urwid.supports_unicode():
 
324
182
            self._text = self._text.encode("ascii", "replace")
 
325
 
        textlist = [("normal", self._text)]
 
 
183
        textlist = [(u"normal", self._text)]
 
326
184
        self._text_widget.set_text(textlist)
 
327
185
        self._focus_text_widget.set_text([(with_standout[text[0]],
 
 
336
194
        if self.update_hook is not None:
 
337
195
            self.update_hook()
 
339
 
    def update_timer(self):
 
342
 
        return True             # Keep calling this
 
344
197
    def delete(self):
 
345
 
        if self._update_timer_callback_tag is not None:
 
346
 
            gobject.source_remove(self._update_timer_callback_tag)
 
347
 
            self._update_timer_callback_tag = None
 
348
198
        if self.delete_hook is not None:
 
349
199
            self.delete_hook(self)
 
351
 
    def render(self, maxcolrow, focus=False):
 
 
201
    def render(self, (maxcol,), focus=False):
 
352
202
        """Render differently if we have focus.
 
353
203
        This overrides the method from urwid.FlowWidget"""
 
354
 
        return self.current_widget(focus).render(maxcolrow,
 
 
204
        return self.current_widget(focus).render((maxcol,),
 
357
 
    def keypress(self, maxcolrow, key):
 
 
207
    def keypress(self, (maxcol,), key):
 
359
209
        This overrides the method from urwid.FlowWidget"""
 
361
 
            self.proxy.Enable(dbus_interface = client_interface)
 
363
 
            self.proxy.Disable(dbus_interface = client_interface)
 
365
 
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
366
 
                               dbus_interface = client_interface)
 
368
 
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
369
 
                                  dbus_interface = client_interface)
 
370
 
        elif key == "R" or key == "_" or key == "ctrl k":
 
 
210
        if key == u"e" or key == u"+":
 
 
212
        elif key == u"d" or key == u"-":
 
 
214
        elif key == u"r" or key == u"_" or key == u"ctrl k":
 
371
215
            self.server_proxy_object.RemoveClient(self.proxy
 
374
 
            self.proxy.StartChecker(dbus_interface = client_interface)
 
376
 
            self.proxy.StopChecker(dbus_interface = client_interface)
 
378
 
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
 
218
            self.proxy.StartChecker()
 
 
220
            self.proxy.StopChecker()
 
 
222
            self.proxy.CheckedOK()
 
380
 
#         elif key == "p" or key == "=":
 
 
224
#         elif key == u"p" or key == "=":
 
381
225
#             self.proxy.pause()
 
382
 
#         elif key == "u" or key == ":":
 
 
226
#         elif key == u"u" or key == ":":
 
383
227
#             self.proxy.unpause()
 
 
228
#         elif key == u"RET":
 
 
420
264
        self.screen = urwid.curses_display.Screen()
 
422
266
        self.screen.register_palette((
 
424
 
                 "default", "default", None),
 
426
 
                 "default", "default", "bold"),
 
428
 
                 "default", "default", "underline"),
 
430
 
                 "default", "default", "standout"),
 
431
 
                ("bold-underline-blink",
 
432
 
                 "default", "default", ("bold", "underline")),
 
434
 
                 "default", "default", ("bold", "standout")),
 
435
 
                ("underline-blink-standout",
 
436
 
                 "default", "default", ("underline", "standout")),
 
437
 
                ("bold-underline-blink-standout",
 
438
 
                 "default", "default", ("bold", "underline",
 
 
268
                 u"default", u"default", None),
 
 
270
                 u"default", u"default", u"bold"),
 
 
272
                 u"default", u"default", u"underline"),
 
 
274
                 u"default", u"default", u"standout"),
 
 
275
                (u"bold-underline-blink",
 
 
276
                 u"default", u"default", (u"bold", u"underline")),
 
 
278
                 u"default", u"default", (u"bold", u"standout")),
 
 
279
                (u"underline-blink-standout",
 
 
280
                 u"default", u"default", (u"underline", u"standout")),
 
 
281
                (u"bold-underline-blink-standout",
 
 
282
                 u"default", u"default", (u"bold", u"underline",
 
442
286
        if urwid.supports_unicode():
 
443
 
            self.divider = "─" # \u2500
 
444
 
            #self.divider = "━" # \u2501
 
 
287
            self.divider = u"─" # \u2500
 
 
288
            #self.divider = u"━" # \u2501
 
446
 
            #self.divider = "-" # \u002d
 
447
 
            self.divider = "_" # \u005f
 
 
290
            #self.divider = u"-" # \u002d
 
 
291
            self.divider = u"_" # \u005f
 
449
293
        self.screen.start()
 
 
550
391
            and len(self.log) > self.max_log_length):
 
551
392
            del self.log[0:len(self.log)-self.max_log_length-1]
 
552
393
        self.logbox.set_focus(len(self.logbox.body.contents),
 
 
394
                              coming_from=u"above")
 
556
397
    def toggle_log_display(self):
 
557
398
        """Toggle visibility of the log buffer."""
 
558
399
        self.log_visible = not self.log_visible
 
560
 
        #self.log_message("Log visibility changed to: "
 
561
 
        #                 + unicode(self.log_visible))
 
 
401
        self.log_message(u"Log visibility changed to: "
 
 
402
                         + unicode(self.log_visible))
 
563
404
    def change_log_display(self):
 
564
405
        """Change type of log display.
 
565
406
        Currently, this toggles wrapping of text lines."""
 
566
 
        if self.log_wrap == "clip":
 
567
 
            self.log_wrap = "any"
 
 
407
        if self.log_wrap == u"clip":
 
 
408
            self.log_wrap = u"any"
 
569
 
            self.log_wrap = "clip"
 
 
410
            self.log_wrap = u"clip"
 
570
411
        for textwidget in self.log:
 
571
412
            textwidget.set_wrap_mode(self.log_wrap)
 
572
 
        #self.log_message("Wrap mode: " + self.log_wrap)
 
 
413
        self.log_message(u"Wrap mode: " + self.log_wrap)
 
574
415
    def find_and_remove_client(self, path, name):
 
575
416
        """Find an client from its object path and remove it.
 
 
655
495
            except KeyError:    # :-)
 
658
 
            if key == "q" or key == "Q":
 
 
498
            if key == u"q" or key == u"Q":
 
661
 
            elif key == "window resize":
 
 
501
            elif key == u"window resize":
 
662
502
                self.size = self.screen.get_cols_rows()
 
664
 
            elif key == "\f":  # Ctrl-L
 
 
504
            elif key == u"\f":  # Ctrl-L
 
666
 
            elif key == "l" or key == "D":
 
 
506
            elif key == u"l" or key == u"D":
 
667
507
                self.toggle_log_display()
 
669
 
            elif key == "w" or key == "i":
 
 
509
            elif key == u"w" or key == u"i":
 
670
510
                self.change_log_display()
 
672
 
            elif key == "?" or key == "f1" or key == "esc":
 
 
512
            elif key == u"?" or key == u"f1" or key == u"esc":
 
673
513
                if not self.log_visible:
 
674
514
                    self.log_visible = True
 
676
 
                self.log_message_raw(("bold",
 
680
 
                                            "l: Log window toggle",
 
681
 
                                            "TAB: Switch window",
 
683
 
                self.log_message_raw(("bold",
 
689
 
                                             "s: Start new checker",
 
 
516
                self.log_message_raw((u"bold",
 
 
520
                                            u"l: Log window toggle",
 
 
521
                                            u"TAB: Switch window",
 
 
523
                self.log_message_raw((u"bold",
 
 
529
                                             u"s: Start new checker",
 
696
534
                if self.topwidget.get_focus() is self.logbox:
 
697
535
                    self.topwidget.set_focus(0)
 
699
537
                    self.topwidget.set_focus(self.logbox)
 
701
 
            #elif (key == "end" or key == "meta >" or key == "G"
 
 
539
            #elif (key == u"end" or key == u"meta >" or key == u"G"
 
703
541
            #    pass            # xxx end-of-buffer
 
704
 
            #elif (key == "home" or key == "meta <" or key == "g"
 
 
542
            #elif (key == u"home" or key == u"meta <" or key == u"g"
 
706
544
            #    pass            # xxx beginning-of-buffer
 
707
 
            #elif key == "ctrl e" or key == "$":
 
 
545
            #elif key == u"ctrl e" or key == u"$":
 
708
546
            #    pass            # xxx move-end-of-line
 
709
 
            #elif key == "ctrl a" or key == "^":
 
 
547
            #elif key == u"ctrl a" or key == u"^":
 
710
548
            #    pass            # xxx move-beginning-of-line
 
711
 
            #elif key == "ctrl b" or key == "meta (" or key == "h":
 
 
549
            #elif key == u"ctrl b" or key == u"meta (" or key == u"h":
 
712
550
            #    pass            # xxx left
 
713
 
            #elif key == "ctrl f" or key == "meta )" or key == "l":
 
 
551
            #elif key == u"ctrl f" or key == u"meta )" or key == u"l":
 
714
552
            #    pass            # xxx right
 
716
554
            #    pass            # scroll up log
 
718
556
            #    pass            # scroll down log
 
719
557
            elif self.topwidget.selectable():
 
720
558
                self.topwidget.keypress(self.size, key)