86
123
        self.logger = logger
 
 
125
        self._update_timer_callback_tag = None
 
 
126
        self._update_timer_callback_lock = 0
 
 
127
        self.last_checker_failed = False
 
88
129
        # The widget shown normally
 
89
 
        self._text_widget = urwid.Text(u"")
 
 
130
        self._text_widget = urwid.Text("")
 
90
131
        # The widget shown when we have focus
 
91
 
        self._focus_text_widget = urwid.Text(u"")
 
 
132
        self._focus_text_widget = urwid.Text("")
 
92
133
        super(MandosClientWidget, self).__init__(
 
93
134
            update_hook=update_hook, delete_hook=delete_hook,
 
96
137
        self.opened = False
 
97
 
        self.proxy.connect_to_signal(u"CheckerCompleted",
 
 
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",
 
98
158
                                     self.checker_completed,
 
100
160
                                     byte_arrays=True)
 
101
 
        self.proxy.connect_to_signal(u"CheckerStarted",
 
 
161
        self.proxy.connect_to_signal("CheckerStarted",
 
102
162
                                     self.checker_started,
 
103
163
                                     client_interface,
 
104
164
                                     byte_arrays=True)
 
105
 
        self.proxy.connect_to_signal(u"GotSecret",
 
 
165
        self.proxy.connect_to_signal("GotSecret",
 
107
167
                                     client_interface,
 
108
168
                                     byte_arrays=True)
 
109
 
        self.proxy.connect_to_signal(u"NeedApproval",
 
 
169
        self.proxy.connect_to_signal("NeedApproval",
 
110
170
                                     self.need_approval,
 
111
171
                                     client_interface,
 
112
172
                                     byte_arrays=True)
 
113
 
        self.proxy.connect_to_signal(u"Rejected",
 
 
173
        self.proxy.connect_to_signal("Rejected",
 
115
175
                                     client_interface,
 
116
176
                                     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
 
118
201
    def checker_completed(self, exitstatus, condition, command):
 
119
202
        if exitstatus == 0:
 
120
 
            #self.logger(u'Checker for client %s (command "%s")'
 
122
 
            #            % (self.properties[u"name"], command))
 
 
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))
 
 
212
        if not self.last_checker_failed:
 
 
213
            self.last_checker_failed = True
 
 
214
            self.using_timer(True)
 
124
215
        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,
 
 
216
            self.logger('Checker for client %s (command "%s")'
 
 
217
                        ' failed with exit code %s'
 
 
218
                        % (self.properties["Name"], command,
 
128
219
                           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,
 
 
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,
 
134
224
                           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')
 
 
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'
 
142
234
    def checker_started(self, command):
 
143
 
        #self.logger(u'Client %s started checker "%s"'
 
144
 
        #            % (self.properties[u"name"], unicode(command)))
 
 
235
        #self.logger('Client %s started checker "%s"'
 
 
236
        #            % (self.properties["Name"], unicode(command)))
 
147
239
    def got_secret(self):
 
148
 
        self.logger(u'Client %s received its secret'
 
149
 
                    % self.properties[u"name"])
 
 
240
        self.last_checker_failed = False
 
 
241
        self.logger('Client %s received its secret'
 
 
242
                    % self.properties["Name"])
 
151
244
    def need_approval(self, timeout, default):
 
153
 
            message = u'Client %s needs approval within %s seconds'
 
 
246
            message = 'Client %s needs approval within %s seconds'
 
155
 
            message = u'Client %s will get its secret in %s seconds'
 
 
248
            message = 'Client %s will get its secret in %s seconds'
 
156
249
        self.logger(message
 
157
 
                    % (self.properties[u"name"], timeout/1000))
 
 
250
                    % (self.properties["Name"], timeout/1000))
 
 
251
        self.using_timer(True)
 
159
253
    def rejected(self, reason):
 
160
 
        self.logger(u'Client %s was rejected; reason: %s'
 
161
 
                    % (self.properties[u"name"], reason))
 
 
254
        self.logger('Client %s was rejected; reason: %s'
 
 
255
                    % (self.properties["Name"], reason))
 
163
257
    def selectable(self):
 
164
258
        """Make this a "selectable" widget.
 
165
259
        This overrides the method from urwid.FlowWidget."""
 
168
 
    def rows(self, (maxcol,), focus=False):
 
 
262
    def rows(self, maxcolrow, focus=False):
 
169
263
        """How many rows this widget will occupy might depend on
 
170
264
        whether we have focus or not.
 
171
265
        This overrides the method from urwid.FlowWidget"""
 
172
 
        return self.current_widget(focus).rows((maxcol,), focus=focus)
 
 
266
        return self.current_widget(focus).rows(maxcolrow, focus=focus)
 
174
268
    def current_widget(self, focus=False):
 
175
269
        if focus or self.opened:
 
 
179
273
    def update(self):
 
180
274
        "Called when what is visible on the screen should be updated."
 
181
275
        # 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",
 
 
276
        with_standout = { "normal": "standout",
 
 
277
                          "bold": "bold-standout",
 
 
279
                              "underline-blink-standout",
 
 
280
                          "bold-underline-blink":
 
 
281
                              "bold-underline-blink-standout",
 
190
284
        # Rebuild focus and non-focus widgets using current properties
 
192
286
        # 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?"
 
 
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])
 
202
 
            self._text += (u'%(enabled)s'
 
205
 
                                if self.properties[u"enabled"]
 
 
321
        self._text = "%s%s" % (base, message)
 
207
323
        if not urwid.supports_unicode():
 
208
324
            self._text = self._text.encode("ascii", "replace")
 
209
 
        textlist = [(u"normal", self._text)]
 
 
325
        textlist = [("normal", self._text)]
 
210
326
        self._text_widget.set_text(textlist)
 
211
327
        self._focus_text_widget.set_text([(with_standout[text[0]],
 
 
220
336
        if self.update_hook is not None:
 
221
337
            self.update_hook()
 
 
339
    def update_timer(self):
 
 
342
        return True             # Keep calling this
 
223
344
    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
 
224
348
        if self.delete_hook is not None:
 
225
349
            self.delete_hook(self)
 
227
 
    def render(self, (maxcol,), focus=False):
 
 
351
    def render(self, maxcolrow, focus=False):
 
228
352
        """Render differently if we have focus.
 
229
353
        This overrides the method from urwid.FlowWidget"""
 
230
 
        return self.current_widget(focus).render((maxcol,),
 
 
354
        return self.current_widget(focus).render(maxcolrow,
 
233
 
    def keypress(self, (maxcol,), key):
 
 
357
    def keypress(self, maxcolrow, key):
 
235
359
        This overrides the method from urwid.FlowWidget"""
 
237
361
            self.proxy.Enable(dbus_interface = client_interface)
 
239
363
            self.proxy.Disable(dbus_interface = client_interface)
 
241
365
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
242
366
                               dbus_interface = client_interface)
 
244
368
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
245
369
                                  dbus_interface = client_interface)
 
246
 
        elif key == u"r" or key == u"_" or key == u"ctrl k":
 
 
370
        elif key == "R" or key == "_" or key == "ctrl k":
 
247
371
            self.server_proxy_object.RemoveClient(self.proxy
 
250
374
            self.proxy.StartChecker(dbus_interface = client_interface)
 
252
376
            self.proxy.StopChecker(dbus_interface = client_interface)
 
254
378
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
256
 
#         elif key == u"p" or key == "=":
 
 
380
#         elif key == "p" or key == "=":
 
257
381
#             self.proxy.pause()
 
258
 
#         elif key == u"u" or key == ":":
 
 
382
#         elif key == "u" or key == ":":
 
259
383
#             self.proxy.unpause()
 
260
 
#         elif key == u"RET":
 
263
 
#            self.proxy.Approve(True)
 
265
 
#            self.proxy.Approve(False)
 
 
300
420
        self.screen = urwid.curses_display.Screen()
 
302
422
        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",
 
 
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",
 
322
442
        if urwid.supports_unicode():
 
323
 
            self.divider = u"─" # \u2500
 
324
 
            #self.divider = u"━" # \u2501
 
 
443
            self.divider = "─" # \u2500
 
 
444
            #self.divider = "━" # \u2501
 
326
 
            #self.divider = u"-" # \u002d
 
327
 
            self.divider = u"_" # \u005f
 
 
446
            #self.divider = "-" # \u002d
 
 
447
            self.divider = "_" # \u005f
 
329
449
        self.screen.start()
 
 
427
550
            and len(self.log) > self.max_log_length):
 
428
551
            del self.log[0:len(self.log)-self.max_log_length-1]
 
429
552
        self.logbox.set_focus(len(self.logbox.body.contents),
 
430
 
                              coming_from=u"above")
 
433
556
    def toggle_log_display(self):
 
434
557
        """Toggle visibility of the log buffer."""
 
435
558
        self.log_visible = not self.log_visible
 
437
 
        self.log_message(u"Log visibility changed to: "
 
438
 
                         + unicode(self.log_visible))
 
 
560
        #self.log_message("Log visibility changed to: "
 
 
561
        #                 + unicode(self.log_visible))
 
440
563
    def change_log_display(self):
 
441
564
        """Change type of log display.
 
442
565
        Currently, this toggles wrapping of text lines."""
 
443
 
        if self.log_wrap == u"clip":
 
444
 
            self.log_wrap = u"any"
 
 
566
        if self.log_wrap == "clip":
 
 
567
            self.log_wrap = "any"
 
446
 
            self.log_wrap = u"clip"
 
 
569
            self.log_wrap = "clip"
 
447
570
        for textwidget in self.log:
 
448
571
            textwidget.set_wrap_mode(self.log_wrap)
 
449
 
        self.log_message(u"Wrap mode: " + self.log_wrap)
 
 
572
        #self.log_message("Wrap mode: " + self.log_wrap)
 
451
574
    def find_and_remove_client(self, path, name):
 
452
575
        """Find an client from its object path and remove it.
 
 
532
655
            except KeyError:    # :-)
 
535
 
            if key == u"q" or key == u"Q":
 
 
658
            if key == "q" or key == "Q":
 
538
 
            elif key == u"window resize":
 
 
661
            elif key == "window resize":
 
539
662
                self.size = self.screen.get_cols_rows()
 
541
 
            elif key == u"\f":  # Ctrl-L
 
 
664
            elif key == "\f":  # Ctrl-L
 
543
 
            elif key == u"l" or key == u"D":
 
 
666
            elif key == "l" or key == "D":
 
544
667
                self.toggle_log_display()
 
546
 
            elif key == u"w" or key == u"i":
 
 
669
            elif key == "w" or key == "i":
 
547
670
                self.change_log_display()
 
549
 
            elif key == u"?" or key == u"f1" or key == u"esc":
 
 
672
            elif key == "?" or key == "f1" or key == "esc":
 
550
673
                if not self.log_visible:
 
551
674
                    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",
 
 
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",
 
573
696
                if self.topwidget.get_focus() is self.logbox:
 
574
697
                    self.topwidget.set_focus(0)
 
576
699
                    self.topwidget.set_focus(self.logbox)
 
578
 
            #elif (key == u"end" or key == u"meta >" or key == u"G"
 
 
701
            #elif (key == "end" or key == "meta >" or key == "G"
 
580
703
            #    pass            # xxx end-of-buffer
 
581
 
            #elif (key == u"home" or key == u"meta <" or key == u"g"
 
 
704
            #elif (key == "home" or key == "meta <" or key == "g"
 
583
706
            #    pass            # xxx beginning-of-buffer
 
584
 
            #elif key == u"ctrl e" or key == u"$":
 
 
707
            #elif key == "ctrl e" or key == "$":
 
585
708
            #    pass            # xxx move-end-of-line
 
586
 
            #elif key == u"ctrl a" or key == u"^":
 
 
709
            #elif key == "ctrl a" or key == "^":
 
587
710
            #    pass            # xxx move-beginning-of-line
 
588
 
            #elif key == u"ctrl b" or key == u"meta (" or key == u"h":
 
 
711
            #elif key == "ctrl b" or key == "meta (" or key == "h":
 
589
712
            #    pass            # xxx left
 
590
 
            #elif key == u"ctrl f" or key == u"meta )" or key == u"l":
 
 
713
            #elif key == "ctrl f" or key == "meta )" or key == "l":
 
591
714
            #    pass            # xxx right
 
593
716
            #    pass            # scroll up log
 
595
718
            #    pass            # scroll down log
 
596
719
            elif self.topwidget.selectable():
 
597
720
                self.topwidget.keypress(self.size, key)