82
88
    properties and calls a hook function when any of them are
 
85
 
    def __init__(self, proxy_object=None, *args, **kwargs):
 
 
91
    def __init__(self, proxy_object=None, properties=None, **kwargs):
 
86
92
        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__(
 
 
93
        self.properties = dict() if properties is None else properties
 
 
94
        self.property_changed_match = (
 
 
95
            self.proxy.connect_to_signal("PropertiesChanged",
 
 
96
                                         self.properties_changed,
 
 
97
                                         dbus.PROPERTIES_IFACE,
 
 
100
        if properties is None:
 
 
101
            self.properties.update(
 
 
102
                self.proxy.GetAll(client_interface,
 
 
104
                                  = dbus.PROPERTIES_IFACE))
 
 
106
        super(MandosClientPropertyCache, self).__init__(**kwargs)
 
102
 
    def property_changed(self, property=None, value=None):
 
103
 
        """This is called whenever we get a PropertyChanged signal
 
104
 
        It updates the changed property in the "properties" dict.
 
 
108
    def properties_changed(self, interface, properties, invalidated):
 
 
109
        """This is called whenever we get a PropertiesChanged signal
 
 
110
        It updates the changed properties in the "properties" dict.
 
106
112
        # Update properties dict with new value
 
107
 
        self.properties[property] = value
 
 
113
        if interface == client_interface:
 
 
114
            self.properties.update(properties)
 
 
117
        self.property_changed_match.remove()
 
110
120
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
 
 
123
133
        self.logger = logger
 
125
135
        self._update_timer_callback_tag = None
 
126
 
        self._update_timer_callback_lock = 0
 
127
 
        self.last_checker_failed = False
 
129
137
        # The widget shown normally
 
130
138
        self._text_widget = urwid.Text("")
 
131
139
        # The widget shown when we have focus
 
132
140
        self._focus_text_widget = urwid.Text("")
 
133
 
        super(MandosClientWidget, self).__init__(
 
134
 
            update_hook=update_hook, delete_hook=delete_hook,
 
 
141
        super(MandosClientWidget, self).__init__(**kwargs)
 
137
143
        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",
 
 
145
        self.match_objects = (
 
 
146
            self.proxy.connect_to_signal("CheckerCompleted",
 
 
147
                                         self.checker_completed,
 
 
150
            self.proxy.connect_to_signal("CheckerStarted",
 
 
151
                                         self.checker_started,
 
 
154
            self.proxy.connect_to_signal("GotSecret",
 
 
158
            self.proxy.connect_to_signal("NeedApproval",
 
 
162
            self.proxy.connect_to_signal("Rejected",
 
 
166
        self.logger('Created client {}'
 
 
167
                    .format(self.properties["Name"]), level=0)
 
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
169
    def using_timer(self, flag):
 
185
170
        """Call this method with True or False when timer should be
 
186
171
        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:
 
 
173
        if flag and self._update_timer_callback_tag is None:
 
 
174
            # Will update the shown timer value every second
 
194
175
            self._update_timer_callback_tag = (gobject.timeout_add
 
196
177
                                                self.update_timer))
 
197
 
        elif old and self._update_timer_callback_lock == 0:
 
 
178
        elif not (flag or self._update_timer_callback_tag is None):
 
198
179
            gobject.source_remove(self._update_timer_callback_tag)
 
199
180
            self._update_timer_callback_tag = None
 
201
182
    def checker_completed(self, exitstatus, condition, command):
 
202
183
        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))
 
 
184
            self.logger('Checker for client {} (command "{}")'
 
 
185
                        ' succeeded'.format(self.properties["Name"],
 
212
 
        if not self.last_checker_failed:
 
213
 
            self.last_checker_failed = True
 
214
 
            self.using_timer(True)
 
215
190
        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)))
 
 
191
            self.logger('Checker for client {} (command "{}") failed'
 
 
193
                        .format(self.properties["Name"], command,
 
 
194
                                os.WEXITSTATUS(condition)))
 
220
195
        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)))
 
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'
 
 
196
            self.logger('Checker for client {} (command "{}") was'
 
 
197
                        ' killed by signal {}'
 
 
198
                        .format(self.properties["Name"], command,
 
 
199
                                os.WTERMSIG(condition)))
 
234
202
    def checker_started(self, command):
 
235
 
        #self.logger('Client %s started checker "%s"'
 
236
 
        #            % (self.properties["Name"], unicode(command)))
 
 
203
        """Server signals that a checker started."""
 
 
204
        self.logger('Client {} started checker "{}"'
 
 
205
                    .format(self.properties["Name"],
 
239
208
    def got_secret(self):
 
240
 
        self.last_checker_failed = False
 
241
 
        self.logger('Client %s received its secret'
 
242
 
                    % self.properties["Name"])
 
 
209
        self.logger('Client {} received its secret'
 
 
210
                    .format(self.properties["Name"]))
 
244
212
    def need_approval(self, timeout, default):
 
246
 
            message = 'Client %s needs approval within %s seconds'
 
 
214
            message = 'Client {} needs approval within {} seconds'
 
248
 
            message = 'Client %s will get its secret in %s seconds'
 
250
 
                    % (self.properties["Name"], timeout/1000))
 
251
 
        self.using_timer(True)
 
 
216
            message = 'Client {} will get its secret in {} seconds'
 
 
217
        self.logger(message.format(self.properties["Name"],
 
253
220
    def rejected(self, reason):
 
254
 
        self.logger('Client %s was rejected; reason: %s'
 
255
 
                    % (self.properties["Name"], reason))
 
 
221
        self.logger('Client {} was rejected; reason: {}'
 
 
222
                    .format(self.properties["Name"], reason))
 
257
224
    def selectable(self):
 
258
225
        """Make this a "selectable" widget.
 
 
295
262
            last_approval_request = isoformat_to_datetime(
 
296
263
                self.properties["LastApprovalRequest"])
 
297
264
            if last_approval_request is not None:
 
298
 
                timer = timeout - (datetime.datetime.utcnow()
 
299
 
                                   - last_approval_request)
 
 
265
                timer = max(timeout - (datetime.datetime.utcnow()
 
 
266
                                       - last_approval_request),
 
 
267
                            datetime.timedelta())
 
301
269
                timer = datetime.timedelta()
 
302
270
            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)
 
 
271
                message = "Approval in {}. (d)eny?"
 
 
273
                message = "Denial in {}. (a)pprove?"
 
 
274
            message = message.format(str(timer).rsplit(".", 1)[0])
 
 
275
            self.using_timer(True)
 
 
276
        elif self.properties["LastCheckerStatus"] != 0:
 
 
277
            # When checker has failed, show timer until client expires
 
 
278
            expires = self.properties["Expires"]
 
 
280
                timer = datetime.timedelta(0)
 
 
282
                expires = (datetime.datetime.strptime
 
 
283
                           (expires, '%Y-%m-%dT%H:%M:%S.%f'))
 
 
284
                timer = max(expires - datetime.datetime.utcnow(),
 
 
285
                            datetime.timedelta())
 
316
286
            message = ('A checker has failed! Time until client'
 
318
 
                           % unicode(timer).rsplit(".", 1)[0])
 
 
288
                       .format(str(timer).rsplit(".", 1)[0]))
 
 
289
            self.using_timer(True)
 
320
291
            message = "enabled"
 
321
 
        self._text = "%s%s" % (base, message)
 
 
292
            self.using_timer(False)
 
 
293
        self._text = "{}{}".format(base, message)
 
323
295
        if not urwid.supports_unicode():
 
324
296
            self._text = self._text.encode("ascii", "replace")
 
325
297
        textlist = [("normal", self._text)]
 
 
397
 
    def property_changed(self, property=None, value=None,
 
399
 
        """Call self.update() if old value is not new value.
 
 
378
    def properties_changed(self, interface, properties, invalidated):
 
 
379
        """Call self.update() if any properties changed.
 
400
380
        This overrides the method from MandosClientPropertyCache"""
 
401
 
        property_name = unicode(property)
 
402
 
        old_value = self.properties.get(property_name)
 
403
 
        super(MandosClientWidget, self).property_changed(
 
404
 
            property=property, value=value, *args, **kwargs)
 
405
 
        if self.properties.get(property_name) != old_value:
 
 
381
        old_values = { key: self.properties.get(key)
 
 
382
                       for key in properties.keys() }
 
 
383
        super(MandosClientWidget, self).properties_changed(
 
 
384
            interface, properties, invalidated)
 
 
385
        if any(old_values[key] != self.properties.get(key)
 
 
386
               for key in old_values):
 
 
432
413
                 "default", "default", None),
 
434
 
                 "default", "default", "bold"),
 
 
415
                 "bold", "default", "bold"),
 
435
416
                ("underline-blink",
 
436
 
                 "default", "default", "underline"),
 
 
417
                 "underline,blink", "default", "underline,blink"),
 
438
 
                 "default", "default", "standout"),
 
 
419
                 "standout", "default", "standout"),
 
439
420
                ("bold-underline-blink",
 
440
 
                 "default", "default", ("bold", "underline")),
 
 
421
                 "bold,underline,blink", "default", "bold,underline,blink"),
 
441
422
                ("bold-standout",
 
442
 
                 "default", "default", ("bold", "standout")),
 
 
423
                 "bold,standout", "default", "bold,standout"),
 
443
424
                ("underline-blink-standout",
 
444
 
                 "default", "default", ("underline", "standout")),
 
 
425
                 "underline,blink,standout", "default",
 
 
426
                 "underline,blink,standout"),
 
445
427
                ("bold-underline-blink-standout",
 
446
 
                 "default", "default", ("bold", "underline",
 
 
428
                 "bold,underline,blink,standout", "default",
 
 
429
                 "bold,underline,blink,standout"),
 
450
432
        if urwid.supports_unicode():
 
 
483
467
        self.busname = domain + '.Mandos'
 
484
468
        self.main_loop = gobject.MainLoop()
 
485
 
        self.bus = dbus.SystemBus()
 
486
 
        mandos_dbus_objc = self.bus.get_object(
 
487
 
            self.busname, "/", follow_name_owner_changes=True)
 
488
 
        self.mandos_serv = dbus.Interface(mandos_dbus_objc,
 
492
 
            mandos_clients = (self.mandos_serv
 
493
 
                              .GetAllClientsWithProperties())
 
494
 
        except dbus.exceptions.DBusException:
 
495
 
            mandos_clients = dbus.Dictionary()
 
498
 
         .connect_to_signal("ClientRemoved",
 
499
 
                            self.find_and_remove_client,
 
500
 
                            dbus_interface=server_interface,
 
503
 
         .connect_to_signal("ClientAdded",
 
505
 
                            dbus_interface=server_interface,
 
508
 
         .connect_to_signal("ClientNotFound",
 
509
 
                            self.client_not_found,
 
510
 
                            dbus_interface=server_interface,
 
512
 
        for path, client in mandos_clients.iteritems():
 
513
 
            client_proxy_object = self.bus.get_object(self.busname,
 
515
 
            self.add_client(MandosClientWidget(server_proxy_object
 
518
 
                                               =client_proxy_object,
 
528
470
    def client_not_found(self, fingerprint, address):
 
529
 
        self.log_message(("Client with address %s and fingerprint %s"
 
530
 
                          " could not be found" % (address,
 
 
471
        self.log_message("Client with address {} and fingerprint {}"
 
 
472
                         " could not be found"
 
 
473
                         .format(address, fingerprint))
 
533
475
    def rebuild(self):
 
534
476
        """This rebuilds the User Interface.
 
 
577
524
            self.log_wrap = "clip"
 
578
525
        for textwidget in self.log:
 
579
526
            textwidget.set_wrap_mode(self.log_wrap)
 
580
 
        #self.log_message("Wrap mode: " + self.log_wrap)
 
 
527
        self.log_message("Wrap mode: {}".format(self.log_wrap),
 
582
 
    def find_and_remove_client(self, path, name):
 
583
 
        """Find an client from its object path and remove it.
 
 
530
    def find_and_remove_client(self, path, interfaces):
 
 
531
        """Find a client by its object path and remove it.
 
585
 
        This is connected to the ClientRemoved signal from the
 
 
533
        This is connected to the InterfacesRemoved signal from the
 
586
534
        Mandos server object."""
 
 
535
        if client_interface not in interfaces:
 
 
536
            # Not a Mandos client object; ignore
 
588
539
            client = self.clients_dict[path]
 
 
542
            self.log_message("Unknown client {!r} removed"
 
592
 
        self.remove_client(client, path)
 
594
 
    def add_new_client(self, path):
 
 
547
    def add_new_client(self, path, ifs_and_props):
 
 
548
        """Find a client by its object path and remove it.
 
 
550
        This is connected to the InterfacesAdded signal from the
 
 
551
        Mandos server object.
 
 
553
        if client_interface not in ifs_and_props:
 
 
554
            # Not a Mandos client object; ignore
 
595
556
        client_proxy_object = self.bus.get_object(self.busname, path)
 
596
557
        self.add_client(MandosClientWidget(server_proxy_object
 
597
558
                                           =self.mandos_serv,
 
 
634
593
        """Start the main loop and exit when it's done."""
 
 
594
        self.bus = dbus.SystemBus()
 
 
595
        mandos_dbus_objc = self.bus.get_object(
 
 
596
            self.busname, "/", follow_name_owner_changes=True)
 
 
597
        self.mandos_serv = dbus.Interface(mandos_dbus_objc,
 
 
601
            mandos_clients = (self.mandos_serv
 
 
602
                              .GetAllClientsWithProperties())
 
 
603
            if not mandos_clients:
 
 
604
                self.log_message_raw(("bold", "Note: Server has no clients."))
 
 
605
        except dbus.exceptions.DBusException:
 
 
606
            self.log_message_raw(("bold", "Note: No Mandos server running."))
 
 
607
            mandos_clients = dbus.Dictionary()
 
 
610
         .connect_to_signal("InterfacesRemoved",
 
 
611
                            self.find_and_remove_client,
 
 
613
                            = dbus.OBJECT_MANAGER_IFACE,
 
 
616
         .connect_to_signal("InterfacesAdded",
 
 
619
                            = dbus.OBJECT_MANAGER_IFACE,
 
 
622
         .connect_to_signal("ClientNotFound",
 
 
623
                            self.client_not_found,
 
 
624
                            dbus_interface=server_interface,
 
 
626
        for path, client in mandos_clients.items():
 
 
627
            client_proxy_object = self.bus.get_object(self.busname,
 
 
629
            self.add_client(MandosClientWidget(server_proxy_object
 
 
632
                                               =client_proxy_object,
 
636
643
        self._input_callback_tag = (gobject.io_add_watch
 
637
644
                                    (sys.stdin.fileno(),