83
83
    properties and calls a hook function when any of them are
 
86
 
    def __init__(self, proxy_object=None, *args, **kwargs):
 
 
86
    def __init__(self, proxy_object=None, properties=None, **kwargs):
 
87
87
        self.proxy = proxy_object # Mandos Client proxy object
 
89
 
        self.properties = dict()
 
 
88
        self.properties = dict() if properties is None else properties
 
90
89
        self.property_changed_match = (
 
91
 
            self.proxy.connect_to_signal("PropertyChanged",
 
92
 
                                         self.property_changed,
 
 
90
            self.proxy.connect_to_signal("PropertiesChanged",
 
 
91
                                         self.properties_changed,
 
 
92
                                         dbus.PROPERTIES_IFACE,
 
96
 
        self.properties.update(
 
97
 
            self.proxy.GetAll(client_interface,
 
98
 
                              dbus_interface = dbus.PROPERTIES_IFACE))
 
100
 
        #XXX This breaks good super behaviour
 
101
 
#        super(MandosClientPropertyCache, self).__init__(
 
 
95
        if properties is None:
 
 
96
            self.properties.update(
 
 
97
                self.proxy.GetAll(client_interface,
 
 
99
                                  = dbus.PROPERTIES_IFACE))
 
 
101
        super(MandosClientPropertyCache, self).__init__(**kwargs)
 
104
 
    def property_changed(self, property=None, value=None):
 
105
 
        """This is called whenever we get a PropertyChanged signal
 
106
 
        It updates the changed property in the "properties" dict.
 
 
103
    def properties_changed(self, interface, properties, invalidated):
 
 
104
        """This is called whenever we get a PropertiesChanged signal
 
 
105
        It updates the changed properties in the "properties" dict.
 
108
107
        # Update properties dict with new value
 
109
 
        self.properties[property] = value
 
 
108
        self.properties.update(properties)
 
111
 
    def delete(self, *args, **kwargs):
 
112
111
        self.property_changed_match.remove()
 
113
 
        super(MandosClientPropertyCache, self).__init__(
 
117
114
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
 
 
130
127
        self.logger = logger
 
132
129
        self._update_timer_callback_tag = None
 
133
 
        self._update_timer_callback_lock = 0
 
134
 
        self.last_checker_failed = False
 
136
131
        # The widget shown normally
 
137
132
        self._text_widget = urwid.Text("")
 
138
133
        # The widget shown when we have focus
 
139
134
        self._focus_text_widget = urwid.Text("")
 
140
 
        super(MandosClientWidget, self).__init__(
 
141
 
            update_hook=update_hook, delete_hook=delete_hook,
 
 
135
        super(MandosClientWidget, self).__init__(**kwargs)
 
144
137
        self.opened = False
 
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
139
        self.match_objects = (
 
165
140
            self.proxy.connect_to_signal("CheckerCompleted",
 
166
141
                                         self.checker_completed,
 
 
183
158
                                         client_interface,
 
184
159
                                         byte_arrays=True))
 
185
 
        #self.logger('Created client %s' % (self.properties["Name"]))
 
 
160
        self.logger('Created client {}'
 
 
161
                    .format(self.properties["Name"]), level=0)
 
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
163
    def using_timer(self, flag):
 
194
164
        """Call this method with True or False when timer should be
 
195
165
        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:
 
 
167
        if flag and self._update_timer_callback_tag is None:
 
203
168
            # Will update the shown timer value every second
 
204
169
            self._update_timer_callback_tag = (gobject.timeout_add
 
206
171
                                                self.update_timer))
 
207
 
        elif old and self._update_timer_callback_lock == 0:
 
 
172
        elif not (flag or self._update_timer_callback_tag is None):
 
208
173
            gobject.source_remove(self._update_timer_callback_tag)
 
209
174
            self._update_timer_callback_tag = None
 
211
 
    def checker_completed(self, exitstatus, condition, command):
 
 
176
    def checker_completed(self, exitstatus, signal, command):
 
212
177
        if exitstatus == 0:
 
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))
 
 
178
            self.logger('Checker for client {} (command "{}")'
 
 
179
                        ' succeeded'.format(self.properties["Name"],
 
222
 
        if not self.last_checker_failed:
 
223
 
            self.last_checker_failed = True
 
224
 
            self.using_timer(True)
 
225
 
        if os.WIFEXITED(condition):
 
226
 
            self.logger('Checker for client %s (command "%s")'
 
227
 
                        ' failed with exit code %s'
 
228
 
                        % (self.properties["Name"], command,
 
229
 
                           os.WEXITSTATUS(condition)))
 
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,
 
234
 
                           os.WTERMSIG(condition)))
 
235
 
        elif os.WCOREDUMP(condition):
 
236
 
            self.logger('Checker for client %s (command "%s")'
 
238
 
                        % (self.properties["Name"], command))
 
 
185
            self.logger('Checker for client {} (command "{}") failed'
 
 
187
                        .format(self.properties["Name"], command,
 
 
190
            self.logger('Checker for client {} (command "{}") was'
 
 
191
                        ' killed by signal {}'
 
 
192
                        .format(self.properties["Name"], command,
 
240
 
            self.logger('Checker for client %s completed'
 
 
195
            self.logger('Checker for client {} completed'
 
 
197
                        .format(self.properties["Name"]))
 
244
200
    def checker_started(self, command):
 
245
 
        """Server signals that a checker started. This could be useful
 
246
 
           to log in the future. """
 
247
 
        #self.logger('Client %s started checker "%s"'
 
248
 
        #            % (self.properties["Name"], unicode(command)))
 
 
201
        """Server signals that a checker started."""
 
 
202
        self.logger('Client {} started checker "{}"'
 
 
203
                    .format(self.properties["Name"],
 
251
206
    def got_secret(self):
 
252
 
        self.last_checker_failed = False
 
253
 
        self.logger('Client %s received its secret'
 
254
 
                    % self.properties["Name"])
 
 
207
        self.logger('Client {} received its secret'
 
 
208
                    .format(self.properties["Name"]))
 
256
210
    def need_approval(self, timeout, default):
 
258
 
            message = 'Client %s needs approval within %s seconds'
 
 
212
            message = 'Client {} needs approval within {} seconds'
 
260
 
            message = 'Client %s will get its secret in %s seconds'
 
262
 
                    % (self.properties["Name"], timeout/1000))
 
263
 
        self.using_timer(True)
 
 
214
            message = 'Client {} will get its secret in {} seconds'
 
 
215
        self.logger(message.format(self.properties["Name"],
 
265
218
    def rejected(self, reason):
 
266
 
        self.logger('Client %s was rejected; reason: %s'
 
267
 
                    % (self.properties["Name"], reason))
 
 
219
        self.logger('Client {} was rejected; reason: {}'
 
 
220
                    .format(self.properties["Name"], reason))
 
269
222
    def selectable(self):
 
270
223
        """Make this a "selectable" widget.
 
 
307
260
            last_approval_request = isoformat_to_datetime(
 
308
261
                self.properties["LastApprovalRequest"])
 
309
262
            if last_approval_request is not None:
 
310
 
                timer = timeout - (datetime.datetime.utcnow()
 
311
 
                                   - last_approval_request)
 
 
263
                timer = max(timeout - (datetime.datetime.utcnow()
 
 
264
                                       - last_approval_request),
 
 
265
                            datetime.timedelta())
 
313
267
                timer = datetime.timedelta()
 
314
268
            if self.properties["ApprovedByDefault"]:
 
315
 
                message = "Approval in %s. (d)eny?"
 
 
269
                message = "Approval in {}. (d)eny?"
 
317
 
                message = "Denial in %s. (a)pprove?"
 
318
 
            message = message % unicode(timer).rsplit(".", 1)[0]
 
319
 
        elif self.last_checker_failed:
 
320
 
            # When checker has failed, print a timer until client expires
 
 
271
                message = "Denial in {}. (a)pprove?"
 
 
272
            message = message.format(str(timer).rsplit(".", 1)[0])
 
 
273
            self.using_timer(True)
 
 
274
        elif self.properties["LastCheckerStatus"] != 0:
 
 
275
            # When checker has failed, show timer until client expires
 
321
276
            expires = self.properties["Expires"]
 
322
277
            if expires == "":
 
323
278
                timer = datetime.timedelta(0)
 
325
 
                expires = datetime.datetime.strptime(expires,
 
326
 
                                                     '%Y-%m-%dT%H:%M:%S.%f')
 
327
 
                timer = expires - datetime.datetime.utcnow()
 
 
280
                expires = (datetime.datetime.strptime
 
 
281
                           (expires, '%Y-%m-%dT%H:%M:%S.%f'))
 
 
282
                timer = max(expires - datetime.datetime.utcnow(),
 
 
283
                            datetime.timedelta())
 
328
284
            message = ('A checker has failed! Time until client'
 
330
 
                           % unicode(timer).rsplit(".", 1)[0])
 
 
286
                       .format(str(timer).rsplit(".", 1)[0]))
 
 
287
            self.using_timer(True)
 
332
289
            message = "enabled"
 
333
 
        self._text = "%s%s" % (base, message)
 
 
290
            self.using_timer(False)
 
 
291
        self._text = "{}{}".format(base, message)
 
335
293
        if not urwid.supports_unicode():
 
336
294
            self._text = self._text.encode("ascii", "replace")
 
337
295
        textlist = [("normal", self._text)]
 
 
414
 
    def property_changed(self, property=None, value=None,
 
416
 
        """Call self.update() if old value is not new value.
 
 
372
    def properties_changed(self, interface, properties, invalidated):
 
 
373
        """Call self.update() if any properties changed.
 
417
374
        This overrides the method from MandosClientPropertyCache"""
 
418
 
        property_name = unicode(property)
 
419
 
        old_value = self.properties.get(property_name)
 
420
 
        super(MandosClientWidget, self).property_changed(
 
421
 
            property=property, value=value, *args, **kwargs)
 
422
 
        if self.properties.get(property_name) != old_value:
 
 
375
        old_values = { key: self.properties.get(key)
 
 
376
                       for key in properties.keys() }
 
 
377
        super(MandosClientWidget, self).properties_changed(
 
 
378
            interface, properties, invalidated)
 
 
379
        if any(old_values[key] != self.properties.get(key)
 
 
380
               for key in old_values):
 
 
449
407
                 "default", "default", None),
 
451
 
                 "default", "default", "bold"),
 
 
409
                 "bold", "default", "bold"),
 
452
410
                ("underline-blink",
 
453
 
                 "default", "default", "underline"),
 
 
411
                 "underline,blink", "default", "underline,blink"),
 
455
 
                 "default", "default", "standout"),
 
 
413
                 "standout", "default", "standout"),
 
456
414
                ("bold-underline-blink",
 
457
 
                 "default", "default", ("bold", "underline")),
 
 
415
                 "bold,underline,blink", "default", "bold,underline,blink"),
 
458
416
                ("bold-standout",
 
459
 
                 "default", "default", ("bold", "standout")),
 
 
417
                 "bold,standout", "default", "bold,standout"),
 
460
418
                ("underline-blink-standout",
 
461
 
                 "default", "default", ("underline", "standout")),
 
 
419
                 "underline,blink,standout", "default",
 
 
420
                 "underline,blink,standout"),
 
462
421
                ("bold-underline-blink-standout",
 
463
 
                 "default", "default", ("bold", "underline",
 
 
422
                 "bold,underline,blink,standout", "default",
 
 
423
                 "bold,underline,blink,standout"),
 
467
426
        if urwid.supports_unicode():
 
 
521
482
            self.uilist.append(self.logbox)
 
522
483
        self.topwidget = urwid.Pile(self.uilist)
 
524
 
    def log_message(self, message):
 
 
485
    def log_message(self, message, level=1):
 
 
486
        """Log message formatted with timestamp"""
 
 
487
        if level < self.log_level:
 
525
489
        timestamp = datetime.datetime.now().isoformat()
 
526
 
        self.log_message_raw(timestamp + ": " + message)
 
 
490
        self.log_message_raw("{}: {}".format(timestamp, message),
 
528
 
    def log_message_raw(self, markup):
 
 
493
    def log_message_raw(self, markup, level=1):
 
529
494
        """Add a log message to the log buffer."""
 
 
495
        if level < self.log_level:
 
530
497
        self.log.append(urwid.Text(markup, wrap=self.log_wrap))
 
531
498
        if (self.max_log_length
 
532
499
            and len(self.log) > self.max_log_length):