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
135
131
# The widget shown normally
136
132
self._text_widget = urwid.Text("")
137
133
# The widget shown when we have focus
138
134
self._focus_text_widget = urwid.Text("")
139
super(MandosClientWidget, self).__init__(
140
update_hook=update_hook, delete_hook=delete_hook,
135
super(MandosClientWidget, self).__init__(**kwargs)
143
137
self.opened = False
145
last_checked_ok = isoformat_to_datetime(self.properties
148
if self.properties ["LastCheckerStatus"] != 0:
149
self.using_timer(True)
151
if self.need_approval:
152
self.using_timer(True)
154
139
self.match_objects = (
155
140
self.proxy.connect_to_signal("CheckerCompleted",
156
141
self.checker_completed,
173
158
client_interface,
174
159
byte_arrays=True))
175
#self.logger('Created client {0}'
176
# .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))
183
if property == "LastCheckerStatus":
184
using_timer(value != 0)
185
#self.logger('Checker for client {0} (command "{1}") was '
186
# ' successful'.format(self.properties["Name"],
160
self.logger('Created client {}'
161
.format(self.properties["Name"]), level=0)
189
163
def using_timer(self, flag):
190
164
"""Call this method with True or False when timer should be
191
165
activated or deactivated.
193
old = self._update_timer_callback_lock
195
self._update_timer_callback_lock += 1
197
self._update_timer_callback_lock -= 1
198
if old == 0 and self._update_timer_callback_lock:
167
if flag and self._update_timer_callback_tag is None:
199
168
# Will update the shown timer value every second
200
169
self._update_timer_callback_tag = (gobject.timeout_add
202
171
self.update_timer))
203
elif old and self._update_timer_callback_lock == 0:
172
elif not (flag or self._update_timer_callback_tag is None):
204
173
gobject.source_remove(self._update_timer_callback_tag)
205
174
self._update_timer_callback_tag = None
207
176
def checker_completed(self, exitstatus, condition, command):
208
177
if exitstatus == 0:
178
self.logger('Checker for client {} (command "{}")'
179
' succeeded'.format(self.properties["Name"],
212
184
if os.WIFEXITED(condition):
213
self.logger('Checker for client {0} (command "{1}")'
214
' failed with exit code {2}'
185
self.logger('Checker for client {} (command "{}") failed'
215
187
.format(self.properties["Name"], command,
216
188
os.WEXITSTATUS(condition)))
217
189
elif os.WIFSIGNALED(condition):
218
self.logger('Checker for client {0} (command "{1}") was'
219
' killed by signal {2}'
190
self.logger('Checker for client {} (command "{}") was'
191
' killed by signal {}'
220
192
.format(self.properties["Name"], command,
221
193
os.WTERMSIG(condition)))
222
194
elif os.WCOREDUMP(condition):
223
self.logger('Checker for client {0} (command "{1}")'
225
.format(self.properties["Name"], command))
195
self.logger('Checker for client {} (command "{}") dumped'
196
' core'.format(self.properties["Name"],
227
self.logger('Checker for client {0} completed'
199
self.logger('Checker for client {} completed'
229
201
.format(self.properties["Name"]))
232
204
def checker_started(self, command):
233
"""Server signals that a checker started. This could be useful
234
to log in the future. """
235
#self.logger('Client {0} started checker "{1}"'
236
# .format(self.properties["Name"],
205
"""Server signals that a checker started."""
206
self.logger('Client {} started checker "{}"'
207
.format(self.properties["Name"],
240
210
def got_secret(self):
241
self.logger('Client {0} received its secret'
211
self.logger('Client {} received its secret'
242
212
.format(self.properties["Name"]))
244
214
def need_approval(self, timeout, default):
246
message = 'Client {0} needs approval within {1} seconds'
216
message = 'Client {} needs approval within {} seconds'
248
message = 'Client {0} will get its secret in {1} seconds'
218
message = 'Client {} will get its secret in {} seconds'
249
219
self.logger(message.format(self.properties["Name"],
251
self.using_timer(True)
253
222
def rejected(self, reason):
254
self.logger('Client {0} was rejected; reason: {1}'
223
self.logger('Client {} was rejected; reason: {}'
255
224
.format(self.properties["Name"], reason))
257
226
def selectable(self):
294
264
last_approval_request = isoformat_to_datetime(
295
265
self.properties["LastApprovalRequest"])
296
266
if last_approval_request is not None:
297
timer = timeout - (datetime.datetime.utcnow()
298
- last_approval_request)
267
timer = max(timeout - (datetime.datetime.utcnow()
268
- last_approval_request),
269
datetime.timedelta())
300
271
timer = datetime.timedelta()
301
272
if self.properties["ApprovedByDefault"]:
302
message = "Approval in {0}. (d)eny?"
273
message = "Approval in {}. (d)eny?"
304
message = "Denial in {0}. (a)pprove?"
305
message = message.format(unicode(timer).rsplit(".", 1)[0])
275
message = "Denial in {}. (a)pprove?"
276
message = message.format(str(timer).rsplit(".", 1)[0])
277
self.using_timer(True)
306
278
elif self.properties["LastCheckerStatus"] != 0:
307
# When checker has failed, print a timer until client expires
279
# When checker has failed, show timer until client expires
308
280
expires = self.properties["Expires"]
309
281
if expires == "":
310
282
timer = datetime.timedelta(0)
312
expires = datetime.datetime.strptime(expires,
313
'%Y-%m-%dT%H:%M:%S.%f')
314
timer = expires - datetime.datetime.utcnow()
284
expires = (datetime.datetime.strptime
285
(expires, '%Y-%m-%dT%H:%M:%S.%f'))
286
timer = max(expires - datetime.datetime.utcnow(),
287
datetime.timedelta())
315
288
message = ('A checker has failed! Time until client'
316
' gets disabled: {0}'
317
.format(unicode(timer).rsplit(".", 1)[0]))
290
.format(str(timer).rsplit(".", 1)[0]))
291
self.using_timer(True)
319
293
message = "enabled"
320
self._text = "{0}{1}".format(base, message)
294
self.using_timer(False)
295
self._text = "{}{}".format(base, message)
322
297
if not urwid.supports_unicode():
323
298
self._text = self._text.encode("ascii", "replace")
324
299
textlist = [("normal", self._text)]
401
def property_changed(self, property=None, value=None,
403
"""Call self.update() if old value is not new value.
376
def properties_changed(self, interface, properties, invalidated):
377
"""Call self.update() if any properties changed.
404
378
This overrides the method from MandosClientPropertyCache"""
405
property_name = unicode(property)
406
old_value = self.properties.get(property_name)
407
super(MandosClientWidget, self).property_changed(
408
property=property, value=value, *args, **kwargs)
409
if self.properties.get(property_name) != old_value:
379
old_values = { key: self.properties.get(key)
380
for key in properties.keys() }
381
super(MandosClientWidget, self).properties_changed(
382
interface, properties, invalidated)
383
if any(old_values[key] != self.properties.get(key)
384
for key in old_values):
436
411
"default", "default", None),
438
"default", "default", "bold"),
413
"bold", "default", "bold"),
439
414
("underline-blink",
440
"default", "default", "underline"),
415
"underline,blink", "default", "underline,blink"),
442
"default", "default", "standout"),
417
"standout", "default", "standout"),
443
418
("bold-underline-blink",
444
"default", "default", ("bold", "underline")),
419
"bold,underline,blink", "default", "bold,underline,blink"),
445
420
("bold-standout",
446
"default", "default", ("bold", "standout")),
421
"bold,standout", "default", "bold,standout"),
447
422
("underline-blink-standout",
448
"default", "default", ("underline", "standout")),
423
"underline,blink,standout", "default",
424
"underline,blink,standout"),
449
425
("bold-underline-blink-standout",
450
"default", "default", ("bold", "underline",
426
"bold,underline,blink,standout", "default",
427
"bold,underline,blink,standout"),
454
430
if urwid.supports_unicode():
508
486
self.uilist.append(self.logbox)
509
487
self.topwidget = urwid.Pile(self.uilist)
511
def log_message(self, message):
489
def log_message(self, message, level=1):
490
"""Log message formatted with timestamp"""
491
if level < self.log_level:
512
493
timestamp = datetime.datetime.now().isoformat()
513
self.log_message_raw(timestamp + ": " + message)
494
self.log_message_raw("{}: {}".format(timestamp, message),
515
def log_message_raw(self, markup):
497
def log_message_raw(self, markup, level=1):
516
498
"""Add a log message to the log buffer."""
499
if level < self.log_level:
517
501
self.log.append(urwid.Text(markup, wrap=self.log_wrap))
518
502
if (self.max_log_length
519
503
and len(self.log) > self.max_log_length):