86
130
        self.logger = logger
 
 
132
        self._update_timer_callback_tag = None
 
 
133
        self._update_timer_callback_lock = 0
 
 
134
        self.last_checker_failed = False
 
88
136
        # The widget shown normally
 
89
 
        self._text_widget = urwid.Text(u"")
 
 
137
        self._text_widget = urwid.Text("")
 
90
138
        # The widget shown when we have focus
 
91
 
        self._focus_text_widget = urwid.Text(u"")
 
 
139
        self._focus_text_widget = urwid.Text("")
 
92
140
        super(MandosClientWidget, self).__init__(
 
93
141
            update_hook=update_hook, delete_hook=delete_hook,
 
96
144
        self.opened = False
 
97
 
        self.proxy.connect_to_signal(u"CheckerCompleted",
 
98
 
                                     self.checker_completed,
 
101
 
        self.proxy.connect_to_signal(u"CheckerStarted",
 
102
 
                                     self.checker_started,
 
105
 
        self.proxy.connect_to_signal(u"GotSecret",
 
109
 
        self.proxy.connect_to_signal(u"NeedApproval",
 
113
 
        self.proxy.connect_to_signal(u"Rejected",
 
 
146
        last_checked_ok = isoformat_to_datetime(self.properties
 
 
148
        if last_checked_ok is None:
 
 
149
            self.last_checker_failed = True
 
 
151
            self.last_checker_failed = ((datetime.datetime.utcnow()
 
 
158
        if self.last_checker_failed:
 
 
159
            self.using_timer(True)
 
 
161
        if self.need_approval:
 
 
162
            self.using_timer(True)
 
 
164
        self.match_objects = (
 
 
165
            self.proxy.connect_to_signal("CheckerCompleted",
 
 
166
                                         self.checker_completed,
 
 
169
            self.proxy.connect_to_signal("CheckerStarted",
 
 
170
                                         self.checker_started,
 
 
173
            self.proxy.connect_to_signal("GotSecret",
 
 
177
            self.proxy.connect_to_signal("NeedApproval",
 
 
181
            self.proxy.connect_to_signal("Rejected",
 
 
185
        #self.logger('Created client %s' % (self.properties["Name"]))
 
 
187
    def property_changed(self, property=None, value=None):
 
 
188
        super(self, MandosClientWidget).property_changed(property,
 
 
190
        if property == "ApprovalPending":
 
 
191
            using_timer(bool(value))
 
 
193
    def using_timer(self, flag):
 
 
194
        """Call this method with True or False when timer should be
 
 
195
        activated or deactivated.
 
 
197
        old = self._update_timer_callback_lock
 
 
199
            self._update_timer_callback_lock += 1
 
 
201
            self._update_timer_callback_lock -= 1
 
 
202
        if old == 0 and self._update_timer_callback_lock:
 
 
203
            # Will update the shown timer value every second
 
 
204
            self._update_timer_callback_tag = (gobject.timeout_add
 
 
207
        elif old and self._update_timer_callback_lock == 0:
 
 
208
            gobject.source_remove(self._update_timer_callback_tag)
 
 
209
            self._update_timer_callback_tag = None
 
118
211
    def checker_completed(self, exitstatus, condition, command):
 
119
212
        if exitstatus == 0:
 
120
 
            #self.logger(u'Checker for client %s (command "%s")'
 
122
 
            #            % (self.properties[u"name"], command))
 
 
213
            if self.last_checker_failed:
 
 
214
                self.last_checker_failed = False
 
 
215
                self.using_timer(False)
 
 
216
            #self.logger('Checker for client %s (command "%s")'
 
 
218
            #            % (self.properties["Name"], command))
 
 
222
        if not self.last_checker_failed:
 
 
223
            self.last_checker_failed = True
 
 
224
            self.using_timer(True)
 
124
225
        if os.WIFEXITED(condition):
 
125
 
            self.logger(u'Checker for client %s (command "%s")'
 
126
 
                        u' failed with exit code %s'
 
127
 
                        % (self.properties[u"name"], command,
 
 
226
            self.logger('Checker for client %s (command "%s")'
 
 
227
                        ' failed with exit code %s'
 
 
228
                        % (self.properties["Name"], command,
 
128
229
                           os.WEXITSTATUS(condition)))
 
130
 
        if os.WIFSIGNALED(condition):
 
131
 
            self.logger(u'Checker for client %s (command "%s")'
 
132
 
                        u' was killed by signal %s'
 
133
 
                        % (self.properties[u"name"], command,
 
 
230
        elif os.WIFSIGNALED(condition):
 
 
231
            self.logger('Checker for client %s (command "%s")'
 
 
232
                        ' was killed by signal %s'
 
 
233
                        % (self.properties["Name"], command,
 
134
234
                           os.WTERMSIG(condition)))
 
136
 
        if os.WCOREDUMP(condition):
 
137
 
            self.logger(u'Checker for client %s (command "%s")'
 
139
 
                        % (self.properties[u"name"], command))
 
140
 
        self.logger(u'Checker for client %s completed mysteriously')
 
 
235
        elif os.WCOREDUMP(condition):
 
 
236
            self.logger('Checker for client %s (command "%s")'
 
 
238
                        % (self.properties["Name"], command))
 
 
240
            self.logger('Checker for client %s completed'
 
142
244
    def checker_started(self, command):
 
143
 
        #self.logger(u'Client %s started checker "%s"'
 
144
 
        #            % (self.properties[u"name"], unicode(command)))
 
 
245
        #self.logger('Client %s started checker "%s"'
 
 
246
        #            % (self.properties["Name"], unicode(command)))
 
147
249
    def got_secret(self):
 
148
 
        self.logger(u'Client %s received its secret'
 
149
 
                    % self.properties[u"name"])
 
 
250
        self.last_checker_failed = False
 
 
251
        self.logger('Client %s received its secret'
 
 
252
                    % self.properties["Name"])
 
151
254
    def need_approval(self, timeout, default):
 
153
 
            message = u'Client %s needs approval within %s seconds'
 
 
256
            message = 'Client %s needs approval within %s seconds'
 
155
 
            message = u'Client %s will get its secret in %s seconds'
 
 
258
            message = 'Client %s will get its secret in %s seconds'
 
156
259
        self.logger(message
 
157
 
                    % (self.properties[u"name"], timeout/1000))
 
 
260
                    % (self.properties["Name"], timeout/1000))
 
 
261
        self.using_timer(True)
 
159
263
    def rejected(self, reason):
 
160
 
        self.logger(u'Client %s was rejected; reason: %s'
 
161
 
                    % (self.properties[u"name"], reason))
 
 
264
        self.logger('Client %s was rejected; reason: %s'
 
 
265
                    % (self.properties["Name"], reason))
 
163
267
    def selectable(self):
 
164
268
        """Make this a "selectable" widget.
 
165
269
        This overrides the method from urwid.FlowWidget."""
 
168
 
    def rows(self, (maxcol,), focus=False):
 
 
272
    def rows(self, maxcolrow, focus=False):
 
169
273
        """How many rows this widget will occupy might depend on
 
170
274
        whether we have focus or not.
 
171
275
        This overrides the method from urwid.FlowWidget"""
 
172
 
        return self.current_widget(focus).rows((maxcol,), focus=focus)
 
 
276
        return self.current_widget(focus).rows(maxcolrow, focus=focus)
 
174
278
    def current_widget(self, focus=False):
 
175
279
        if focus or self.opened:
 
 
179
283
    def update(self):
 
180
284
        "Called when what is visible on the screen should be updated."
 
181
285
        # How to add standout mode to a style
 
182
 
        with_standout = { u"normal": u"standout",
 
183
 
                          u"bold": u"bold-standout",
 
185
 
                              u"underline-blink-standout",
 
186
 
                          u"bold-underline-blink":
 
187
 
                              u"bold-underline-blink-standout",
 
 
286
        with_standout = { "normal": "standout",
 
 
287
                          "bold": "bold-standout",
 
 
289
                              "underline-blink-standout",
 
 
290
                          "bold-underline-blink":
 
 
291
                              "bold-underline-blink-standout",
 
190
294
        # Rebuild focus and non-focus widgets using current properties
 
192
296
        # Base part of a client. Name!
 
193
 
        self._text = (u'%(name)s: '
 
194
 
                      % {u"name": self.properties[u"name"]})
 
196
 
        if self.properties[u"approved_pending"]:
 
197
 
            if self.properties[u"approved_by_default"]:
 
198
 
                self._text += u"Connection established to client. (d)eny?"
 
200
 
                self._text += u"Seeks approval to send secret. (a)pprove?"
 
 
298
                      % {"name": self.properties["Name"]})
 
 
299
        if not self.properties["Enabled"]:
 
 
301
        elif self.properties["ApprovalPending"]:
 
 
302
            timeout = datetime.timedelta(milliseconds
 
 
305
            last_approval_request = isoformat_to_datetime(
 
 
306
                self.properties["LastApprovalRequest"])
 
 
307
            if last_approval_request is not None:
 
 
308
                timer = timeout - (datetime.datetime.utcnow()
 
 
309
                                   - last_approval_request)
 
 
311
                timer = datetime.timedelta()
 
 
312
            if self.properties["ApprovedByDefault"]:
 
 
313
                message = "Approval in %s. (d)eny?"
 
 
315
                message = "Denial in %s. (a)pprove?"
 
 
316
            message = message % unicode(timer).rsplit(".", 1)[0]
 
 
317
        elif self.last_checker_failed:
 
 
318
            # When checker has failed, print a timer until client expires
 
 
319
            expires = self.properties["Expires"]
 
 
321
                timer = datetime.timedelta(0)
 
 
323
                expires = datetime.datetime.strptime(expires,
 
 
324
                                                     '%Y-%m-%dT%H:%M:%S.%f')
 
 
325
                timer = expires - datetime.datetime.utcnow()
 
 
326
            message = ('A checker has failed! Time until client'
 
 
328
                           % unicode(timer).rsplit(".", 1)[0])
 
202
 
            self._text += (u'%(enabled)s'
 
205
 
                                if self.properties[u"enabled"]
 
 
331
        self._text = "%s%s" % (base, message)
 
207
333
        if not urwid.supports_unicode():
 
208
334
            self._text = self._text.encode("ascii", "replace")
 
209
 
        textlist = [(u"normal", self._text)]
 
 
335
        textlist = [("normal", self._text)]
 
210
336
        self._text_widget.set_text(textlist)
 
211
337
        self._focus_text_widget.set_text([(with_standout[text[0]],
 
 
220
346
        if self.update_hook is not None:
 
221
347
            self.update_hook()
 
 
349
    def update_timer(self):
 
 
350
        """called by gobject. Will indefinitely loop until
 
 
351
        gobject.source_remove() on tag is called"""
 
 
353
        return True             # Keep calling this
 
 
355
    def delete(self, *args, **kwargs):
 
 
356
        if self._update_timer_callback_tag is not None:
 
 
357
            gobject.source_remove(self._update_timer_callback_tag)
 
 
358
            self._update_timer_callback_tag = None
 
 
359
        for match in self.match_objects:
 
 
361
        self.match_objects = ()
 
224
362
        if self.delete_hook is not None:
 
225
363
            self.delete_hook(self)
 
 
364
        return super(MandosClientWidget, self).delete(*args, **kwargs)
 
227
 
    def render(self, (maxcol,), focus=False):
 
 
366
    def render(self, maxcolrow, focus=False):
 
228
367
        """Render differently if we have focus.
 
229
368
        This overrides the method from urwid.FlowWidget"""
 
230
 
        return self.current_widget(focus).render((maxcol,),
 
 
369
        return self.current_widget(focus).render(maxcolrow,
 
233
 
    def keypress(self, (maxcol,), key):
 
 
372
    def keypress(self, maxcolrow, key):
 
235
374
        This overrides the method from urwid.FlowWidget"""
 
237
 
            self.proxy.Enable(dbus_interface = client_interface)
 
239
 
            self.proxy.Disable(dbus_interface = client_interface)
 
 
376
            self.proxy.Enable(dbus_interface = client_interface,
 
 
379
            self.proxy.Disable(dbus_interface = client_interface,
 
241
382
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
242
 
                               dbus_interface = client_interface)
 
 
383
                               dbus_interface = client_interface,
 
244
386
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
245
 
                                  dbus_interface = client_interface)
 
246
 
        elif key == u"r" or key == u"_" or key == u"ctrl k":
 
 
387
                                  dbus_interface = client_interface,
 
 
389
        elif key == "R" or key == "_" or key == "ctrl k":
 
247
390
            self.server_proxy_object.RemoveClient(self.proxy
 
250
 
            self.proxy.StartChecker(dbus_interface = client_interface)
 
252
 
            self.proxy.StopChecker(dbus_interface = client_interface)
 
254
 
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
 
394
            self.proxy.StartChecker(dbus_interface = client_interface,
 
 
397
            self.proxy.StopChecker(dbus_interface = client_interface,
 
 
400
            self.proxy.CheckedOK(dbus_interface = client_interface,
 
256
 
#         elif key == u"p" or key == "=":
 
 
403
#         elif key == "p" or key == "=":
 
257
404
#             self.proxy.pause()
 
258
 
#         elif key == u"u" or key == ":":
 
 
405
#         elif key == "u" or key == ":":
 
259
406
#             self.proxy.unpause()
 
260
 
#         elif key == u"RET":
 
263
 
#            self.proxy.Approve(True)
 
265
 
#            self.proxy.Approve(False)
 
 
300
443
        self.screen = urwid.curses_display.Screen()
 
302
445
        self.screen.register_palette((
 
304
 
                 u"default", u"default", None),
 
306
 
                 u"default", u"default", u"bold"),
 
308
 
                 u"default", u"default", u"underline"),
 
310
 
                 u"default", u"default", u"standout"),
 
311
 
                (u"bold-underline-blink",
 
312
 
                 u"default", u"default", (u"bold", u"underline")),
 
314
 
                 u"default", u"default", (u"bold", u"standout")),
 
315
 
                (u"underline-blink-standout",
 
316
 
                 u"default", u"default", (u"underline", u"standout")),
 
317
 
                (u"bold-underline-blink-standout",
 
318
 
                 u"default", u"default", (u"bold", u"underline",
 
 
447
                 "default", "default", None),
 
 
449
                 "default", "default", "bold"),
 
 
451
                 "default", "default", "underline"),
 
 
453
                 "default", "default", "standout"),
 
 
454
                ("bold-underline-blink",
 
 
455
                 "default", "default", ("bold", "underline")),
 
 
457
                 "default", "default", ("bold", "standout")),
 
 
458
                ("underline-blink-standout",
 
 
459
                 "default", "default", ("underline", "standout")),
 
 
460
                ("bold-underline-blink-standout",
 
 
461
                 "default", "default", ("bold", "underline",
 
322
465
        if urwid.supports_unicode():
 
323
 
            self.divider = u"─" # \u2500
 
324
 
            #self.divider = u"━" # \u2501
 
 
466
            self.divider = "─" # \u2500
 
 
467
            #self.divider = "━" # \u2501
 
326
 
            #self.divider = u"-" # \u002d
 
327
 
            self.divider = u"_" # \u005f
 
 
469
            #self.divider = "-" # \u002d
 
 
470
            self.divider = "_" # \u005f
 
329
472
        self.screen.start()
 
 
427
573
            and len(self.log) > self.max_log_length):
 
428
574
            del self.log[0:len(self.log)-self.max_log_length-1]
 
429
575
        self.logbox.set_focus(len(self.logbox.body.contents),
 
430
 
                              coming_from=u"above")
 
433
579
    def toggle_log_display(self):
 
434
580
        """Toggle visibility of the log buffer."""
 
435
581
        self.log_visible = not self.log_visible
 
437
 
        self.log_message(u"Log visibility changed to: "
 
438
 
                         + unicode(self.log_visible))
 
 
583
        #self.log_message("Log visibility changed to: "
 
 
584
        #                 + unicode(self.log_visible))
 
440
586
    def change_log_display(self):
 
441
587
        """Change type of log display.
 
442
588
        Currently, this toggles wrapping of text lines."""
 
443
 
        if self.log_wrap == u"clip":
 
444
 
            self.log_wrap = u"any"
 
 
589
        if self.log_wrap == "clip":
 
 
590
            self.log_wrap = "any"
 
446
 
            self.log_wrap = u"clip"
 
 
592
            self.log_wrap = "clip"
 
447
593
        for textwidget in self.log:
 
448
594
            textwidget.set_wrap_mode(self.log_wrap)
 
449
 
        self.log_message(u"Wrap mode: " + self.log_wrap)
 
 
595
        #self.log_message("Wrap mode: " + self.log_wrap)
 
451
597
    def find_and_remove_client(self, path, name):
 
452
 
        """Find an client from its object path and remove it.
 
 
598
        """Find a client by its object path and remove it.
 
454
600
        This is connected to the ClientRemoved signal from the
 
455
601
        Mandos server object."""
 
 
532
680
            except KeyError:    # :-)
 
535
 
            if key == u"q" or key == u"Q":
 
 
683
            if key == "q" or key == "Q":
 
538
 
            elif key == u"window resize":
 
 
686
            elif key == "window resize":
 
539
687
                self.size = self.screen.get_cols_rows()
 
541
 
            elif key == u"\f":  # Ctrl-L
 
 
689
            elif key == "\f":  # Ctrl-L
 
543
 
            elif key == u"l" or key == u"D":
 
 
691
            elif key == "l" or key == "D":
 
544
692
                self.toggle_log_display()
 
546
 
            elif key == u"w" or key == u"i":
 
 
694
            elif key == "w" or key == "i":
 
547
695
                self.change_log_display()
 
549
 
            elif key == u"?" or key == u"f1" or key == u"esc":
 
 
697
            elif key == "?" or key == "f1" or key == "esc":
 
550
698
                if not self.log_visible:
 
551
699
                    self.log_visible = True
 
553
 
                self.log_message_raw((u"bold",
 
557
 
                                            u"l: Log window toggle",
 
558
 
                                            u"TAB: Switch window",
 
560
 
                self.log_message_raw((u"bold",
 
566
 
                                             u"s: Start new checker",
 
 
701
                self.log_message_raw(("bold",
 
 
705
                                            "l: Log window toggle",
 
 
706
                                            "TAB: Switch window",
 
 
708
                self.log_message_raw(("bold",
 
 
714
                                             "s: Start new checker",
 
573
721
                if self.topwidget.get_focus() is self.logbox:
 
574
722
                    self.topwidget.set_focus(0)
 
576
724
                    self.topwidget.set_focus(self.logbox)
 
578
 
            #elif (key == u"end" or key == u"meta >" or key == u"G"
 
 
726
            #elif (key == "end" or key == "meta >" or key == "G"
 
580
728
            #    pass            # xxx end-of-buffer
 
581
 
            #elif (key == u"home" or key == u"meta <" or key == u"g"
 
 
729
            #elif (key == "home" or key == "meta <" or key == "g"
 
583
731
            #    pass            # xxx beginning-of-buffer
 
584
 
            #elif key == u"ctrl e" or key == u"$":
 
 
732
            #elif key == "ctrl e" or key == "$":
 
585
733
            #    pass            # xxx move-end-of-line
 
586
 
            #elif key == u"ctrl a" or key == u"^":
 
 
734
            #elif key == "ctrl a" or key == "^":
 
587
735
            #    pass            # xxx move-beginning-of-line
 
588
 
            #elif key == u"ctrl b" or key == u"meta (" or key == u"h":
 
 
736
            #elif key == "ctrl b" or key == "meta (" or key == "h":
 
589
737
            #    pass            # xxx left
 
590
 
            #elif key == u"ctrl f" or key == u"meta )" or key == u"l":
 
 
738
            #elif key == "ctrl f" or key == "meta )" or key == "l":
 
591
739
            #    pass            # xxx right
 
593
741
            #    pass            # scroll up log
 
595
743
            #    pass            # scroll down log
 
596
744
            elif self.topwidget.selectable():
 
597
745
                self.topwidget.keypress(self.size, key)