82
86
    properties and calls a hook function when any of them are
 
85
 
    def __init__(self, proxy_object=None, *args, **kwargs):
 
 
89
    def __init__(self, proxy_object=None, properties=None, **kwargs):
 
86
90
        self.proxy = proxy_object # Mandos Client proxy object
 
88
 
        self.properties = dict()
 
89
 
        self.proxy.connect_to_signal("PropertyChanged",
 
90
 
                                     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__(
 
 
91
        self.properties = dict() if properties is None else properties
 
 
92
        self.property_changed_match = (
 
 
93
            self.proxy.connect_to_signal("PropertyChanged",
 
 
94
                                         self._property_changed,
 
 
98
        if properties is None:
 
 
99
            self.properties.update(
 
 
100
                self.proxy.GetAll(client_interface,
 
 
102
                                  = dbus.PROPERTIES_IFACE))
 
 
104
        super(MandosClientPropertyCache, self).__init__(**kwargs)
 
 
106
    def _property_changed(self, property, value):
 
 
107
        """Helper which takes positional arguments"""
 
 
108
        return self.property_changed(property=property, value=value)
 
102
110
    def property_changed(self, property=None, value=None):
 
103
111
        """This is called whenever we get a PropertyChanged signal
 
 
123
134
        self.logger = logger
 
125
136
        self._update_timer_callback_tag = None
 
126
 
        self._update_timer_callback_lock = 0
 
127
 
        self.last_checker_failed = False
 
129
138
        # The widget shown normally
 
130
139
        self._text_widget = urwid.Text("")
 
131
140
        # The widget shown when we have focus
 
132
141
        self._focus_text_widget = urwid.Text("")
 
133
 
        super(MandosClientWidget, self).__init__(
 
134
 
            update_hook=update_hook, delete_hook=delete_hook,
 
 
142
        super(MandosClientWidget, self).__init__(**kwargs)
 
137
144
        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",
 
158
 
                                     self.checker_completed,
 
161
 
        self.proxy.connect_to_signal("CheckerStarted",
 
162
 
                                     self.checker_started,
 
165
 
        self.proxy.connect_to_signal("GotSecret",
 
169
 
        self.proxy.connect_to_signal("NeedApproval",
 
173
 
        self.proxy.connect_to_signal("Rejected",
 
 
146
        self.match_objects = (
 
 
147
            self.proxy.connect_to_signal("CheckerCompleted",
 
 
148
                                         self.checker_completed,
 
 
151
            self.proxy.connect_to_signal("CheckerStarted",
 
 
152
                                         self.checker_started,
 
 
155
            self.proxy.connect_to_signal("GotSecret",
 
 
159
            self.proxy.connect_to_signal("NeedApproval",
 
 
163
            self.proxy.connect_to_signal("Rejected",
 
 
167
        #self.logger('Created client {0}'
 
 
168
        #            .format(self.properties["Name"]))
 
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
170
    def using_timer(self, flag):
 
185
171
        """Call this method with True or False when timer should be
 
186
172
        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:
 
 
174
        if flag and self._update_timer_callback_tag is None:
 
 
175
            # Will update the shown timer value every second
 
194
176
            self._update_timer_callback_tag = (gobject.timeout_add
 
196
178
                                                self.update_timer))
 
197
 
        elif old and self._update_timer_callback_lock == 0:
 
 
179
        elif not (flag or self._update_timer_callback_tag is None):
 
198
180
            gobject.source_remove(self._update_timer_callback_tag)
 
199
181
            self._update_timer_callback_tag = None
 
201
183
    def checker_completed(self, exitstatus, condition, command):
 
202
184
        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))
 
212
 
        if not self.last_checker_failed:
 
213
 
            self.last_checker_failed = True
 
214
 
            self.using_timer(True)
 
215
188
        if os.WIFEXITED(condition):
 
216
 
            self.logger('Checker for client %s (command "%s")'
 
217
 
                        ' failed with exit code %s'
 
218
 
                        % (self.properties["Name"], command,
 
219
 
                           os.WEXITSTATUS(condition)))
 
 
189
            self.logger('Checker for client {0} (command "{1}")'
 
 
190
                        ' failed with exit code {2}'
 
 
191
                        .format(self.properties["Name"], command,
 
 
192
                                os.WEXITSTATUS(condition)))
 
220
193
        elif os.WIFSIGNALED(condition):
 
221
 
            self.logger('Checker for client %s (command "%s")'
 
222
 
                        ' was killed by signal %s'
 
223
 
                        % (self.properties["Name"], command,
 
224
 
                           os.WTERMSIG(condition)))
 
 
194
            self.logger('Checker for client {0} (command "{1}") was'
 
 
195
                        ' killed by signal {2}'
 
 
196
                        .format(self.properties["Name"], command,
 
 
197
                                os.WTERMSIG(condition)))
 
225
198
        elif os.WCOREDUMP(condition):
 
226
 
            self.logger('Checker for client %s (command "%s")'
 
 
199
            self.logger('Checker for client {0} (command "{1}")'
 
228
 
                        % (self.properties["Name"], command))
 
 
201
                        .format(self.properties["Name"], command))
 
230
 
            self.logger('Checker for client %s completed'
 
 
203
            self.logger('Checker for client {0} completed'
 
 
205
                        .format(self.properties["Name"]))
 
234
208
    def checker_started(self, command):
 
235
 
        #self.logger('Client %s started checker "%s"'
 
236
 
        #            % (self.properties["Name"], unicode(command)))
 
 
209
        """Server signals that a checker started. This could be useful
 
 
210
           to log in the future. """
 
 
211
        #self.logger('Client {0} started checker "{1}"'
 
 
212
        #            .format(self.properties["Name"],
 
239
216
    def got_secret(self):
 
240
 
        self.last_checker_failed = False
 
241
 
        self.logger('Client %s received its secret'
 
242
 
                    % self.properties["Name"])
 
 
217
        self.logger('Client {0} received its secret'
 
 
218
                    .format(self.properties["Name"]))
 
244
220
    def need_approval(self, timeout, default):
 
246
 
            message = 'Client %s needs approval within %s seconds'
 
 
222
            message = 'Client {0} needs approval within {1} seconds'
 
248
 
            message = 'Client %s will get its secret in %s seconds'
 
250
 
                    % (self.properties["Name"], timeout/1000))
 
251
 
        self.using_timer(True)
 
 
224
            message = 'Client {0} will get its secret in {1} seconds'
 
 
225
        self.logger(message.format(self.properties["Name"],
 
253
228
    def rejected(self, reason):
 
254
 
        self.logger('Client %s was rejected; reason: %s'
 
255
 
                    % (self.properties["Name"], reason))
 
 
229
        self.logger('Client {0} was rejected; reason: {1}'
 
 
230
                    .format(self.properties["Name"], reason))
 
257
232
    def selectable(self):
 
258
233
        """Make this a "selectable" widget.
 
 
295
270
            last_approval_request = isoformat_to_datetime(
 
296
271
                self.properties["LastApprovalRequest"])
 
297
272
            if last_approval_request is not None:
 
298
 
                timer = timeout - (datetime.datetime.utcnow()
 
299
 
                                   - last_approval_request)
 
 
273
                timer = max(timeout - (datetime.datetime.utcnow()
 
 
274
                                       - last_approval_request),
 
 
275
                            datetime.timedelta())
 
301
277
                timer = datetime.timedelta()
 
302
278
            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)
 
 
279
                message = "Approval in {0}. (d)eny?"
 
 
281
                message = "Denial in {0}. (a)pprove?"
 
 
282
            message = message.format(unicode(timer).rsplit(".", 1)[0])
 
 
283
            self.using_timer(True)
 
 
284
        elif self.properties["LastCheckerStatus"] != 0:
 
 
285
            # When checker has failed, show timer until client expires
 
 
286
            expires = self.properties["Expires"]
 
 
288
                timer = datetime.timedelta(0)
 
 
290
                expires = (datetime.datetime.strptime
 
 
291
                           (expires, '%Y-%m-%dT%H:%M:%S.%f'))
 
 
292
                timer = max(expires - datetime.datetime.utcnow(),
 
 
293
                            datetime.timedelta())
 
316
294
            message = ('A checker has failed! Time until client'
 
318
 
                           % unicode(timer).rsplit(".", 1)[0])
 
 
295
                       ' gets disabled: {0}'
 
 
296
                       .format(unicode(timer).rsplit(".", 1)[0]))
 
 
297
            self.using_timer(True)
 
320
299
            message = "enabled"
 
321
 
        self._text = "%s%s" % (base, message)
 
 
300
            self.using_timer(False)
 
 
301
        self._text = "{0}{1}".format(base, message)
 
323
303
        if not urwid.supports_unicode():
 
324
304
            self._text = self._text.encode("ascii", "replace")
 
325
305
        textlist = [("normal", self._text)]
 
 
359
344
        This overrides the method from urwid.FlowWidget"""
 
361
 
            self.proxy.Enable(dbus_interface = client_interface)
 
 
346
            self.proxy.Enable(dbus_interface = client_interface,
 
363
 
            self.proxy.Disable(dbus_interface = client_interface)
 
 
349
            self.proxy.Disable(dbus_interface = client_interface,
 
365
352
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
366
 
                               dbus_interface = client_interface)
 
 
353
                               dbus_interface = client_interface,
 
368
356
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
369
 
                                  dbus_interface = client_interface)
 
 
357
                                  dbus_interface = client_interface,
 
370
359
        elif key == "R" or key == "_" or key == "ctrl k":
 
371
360
            self.server_proxy_object.RemoveClient(self.proxy
 
374
 
            self.proxy.StartChecker(dbus_interface = client_interface)
 
 
364
            self.proxy.StartChecker(dbus_interface = client_interface,
 
376
 
            self.proxy.StopChecker(dbus_interface = client_interface)
 
 
367
            self.proxy.StopChecker(dbus_interface = client_interface,
 
378
 
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
 
370
            self.proxy.CheckedOK(dbus_interface = client_interface,
 
380
373
#         elif key == "p" or key == "=":
 
381
374
#             self.proxy.pause()
 
 
475
467
        self.busname = domain + '.Mandos'
 
476
468
        self.main_loop = gobject.MainLoop()
 
477
 
        self.bus = dbus.SystemBus()
 
478
 
        mandos_dbus_objc = self.bus.get_object(
 
479
 
            self.busname, "/", follow_name_owner_changes=True)
 
480
 
        self.mandos_serv = dbus.Interface(mandos_dbus_objc,
 
484
 
            mandos_clients = (self.mandos_serv
 
485
 
                              .GetAllClientsWithProperties())
 
486
 
        except dbus.exceptions.DBusException:
 
487
 
            mandos_clients = dbus.Dictionary()
 
490
 
         .connect_to_signal("ClientRemoved",
 
491
 
                            self.find_and_remove_client,
 
492
 
                            dbus_interface=server_interface,
 
495
 
         .connect_to_signal("ClientAdded",
 
497
 
                            dbus_interface=server_interface,
 
500
 
         .connect_to_signal("ClientNotFound",
 
501
 
                            self.client_not_found,
 
502
 
                            dbus_interface=server_interface,
 
504
 
        for path, client in mandos_clients.iteritems():
 
505
 
            client_proxy_object = self.bus.get_object(self.busname,
 
507
 
            self.add_client(MandosClientWidget(server_proxy_object
 
510
 
                                               =client_proxy_object,
 
520
470
    def client_not_found(self, fingerprint, address):
 
521
 
        self.log_message(("Client with address %s and fingerprint %s"
 
522
 
                          " could not be found" % (address,
 
 
471
        self.log_message("Client with address {0} and fingerprint"
 
 
472
                         " {1} could not be found"
 
 
473
                         .format(address, fingerprint))
 
525
475
    def rebuild(self):
 
526
476
        """This rebuilds the User Interface.
 
 
626
578
        """Start the main loop and exit when it's done."""
 
 
579
        self.bus = dbus.SystemBus()
 
 
580
        mandos_dbus_objc = self.bus.get_object(
 
 
581
            self.busname, "/", follow_name_owner_changes=True)
 
 
582
        self.mandos_serv = dbus.Interface(mandos_dbus_objc,
 
 
586
            mandos_clients = (self.mandos_serv
 
 
587
                              .GetAllClientsWithProperties())
 
 
588
            if not mandos_clients:
 
 
589
                self.log_message_raw(("bold", "Note: Server has no clients."))
 
 
590
        except dbus.exceptions.DBusException:
 
 
591
            self.log_message_raw(("bold", "Note: No Mandos servers running."))
 
 
592
            mandos_clients = dbus.Dictionary()
 
 
595
         .connect_to_signal("ClientRemoved",
 
 
596
                            self.find_and_remove_client,
 
 
597
                            dbus_interface=server_interface,
 
 
600
         .connect_to_signal("ClientAdded",
 
 
602
                            dbus_interface=server_interface,
 
 
605
         .connect_to_signal("ClientNotFound",
 
 
606
                            self.client_not_found,
 
 
607
                            dbus_interface=server_interface,
 
 
609
        for path, client in mandos_clients.iteritems():
 
 
610
            client_proxy_object = self.bus.get_object(self.busname,
 
 
612
            self.add_client(MandosClientWidget(server_proxy_object
 
 
615
                                               =client_proxy_object,
 
628
626
        self._input_callback_tag = (gobject.io_add_watch
 
629
627
                                    (sys.stdin.fileno(),