2
2
# -*- mode: python; coding: utf-8 -*-
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2009-2011 Teddy Hogeborn
7
# Copyright © 2009-2011 Björn Påhlsson
9
# This program is free software: you can redistribute it and/or modify
10
# it under the terms of the GNU General Public License as published by
11
# the Free Software Foundation, either version 3 of the License, or
12
# (at your option) any later version.
14
# This program is distributed in the hope that it will be useful,
15
# but WITHOUT ANY WARRANTY; without even the implied warranty of
16
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
# GNU General Public License for more details.
19
# You should have received a copy of the GNU General Public License
20
# along with this program. If not, see <http://www.gnu.org/licenses/>.
22
# Contact the authors at <mandos@fukt.bsnet.se>.
4
from __future__ import division, absolute_import, with_statement
25
from __future__ import (division, absolute_import, print_function,
102
130
self.logger = logger
104
132
self._update_timer_callback_tag = None
133
self._update_timer_callback_lock = 0
105
134
self.last_checker_failed = False
107
136
# The widget shown normally
108
self._text_widget = urwid.Text(u"")
137
self._text_widget = urwid.Text("")
109
138
# The widget shown when we have focus
110
self._focus_text_widget = urwid.Text(u"")
139
self._focus_text_widget = urwid.Text("")
111
140
super(MandosClientWidget, self).__init__(
112
141
update_hook=update_hook, delete_hook=delete_hook,
115
144
self.opened = False
116
self.proxy.connect_to_signal(u"CheckerCompleted",
117
self.checker_completed,
120
self.proxy.connect_to_signal(u"CheckerStarted",
121
self.checker_started,
124
self.proxy.connect_to_signal(u"GotSecret",
128
self.proxy.connect_to_signal(u"NeedApproval",
132
self.proxy.connect_to_signal(u"Rejected",
136
146
last_checked_ok = isoformat_to_datetime(self.properties
138
148
if last_checked_ok is None:
139
149
self.last_checker_failed = True
142
152
- last_checked_ok)
143
153
> datetime.timedelta
145
self.properties["interval"]))
146
158
if self.last_checker_failed:
159
self.using_timer(True)
161
if self.need_approval:
162
self.using_timer(True)
164
self.match_objects = (
165
self.proxy.connect_to_signal("CheckerCompleted",
166
self.checker_completed,
169
self.proxy.connect_to_signal("CheckerStarted",
170
self.checker_started,
173
self.proxy.connect_to_signal("GotSecret",
177
self.proxy.connect_to_signal("NeedApproval",
181
self.proxy.connect_to_signal("Rejected",
185
#self.logger('Created client %s' % (self.properties["Name"]))
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
def using_timer(self, flag):
194
"""Call this method with True or False when timer should be
195
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:
147
203
self._update_timer_callback_tag = (gobject.timeout_add
149
205
self.update_timer))
206
elif old and self._update_timer_callback_lock == 0:
207
gobject.source_remove(self._update_timer_callback_tag)
208
self._update_timer_callback_tag = None
151
210
def checker_completed(self, exitstatus, condition, command):
152
211
if exitstatus == 0:
153
212
if self.last_checker_failed:
154
213
self.last_checker_failed = False
155
gobject.source_remove(self._update_timer_callback_tag)
156
self._update_timer_callback_tag = None
157
self.logger(u'Checker for client %s (command "%s")'
159
% (self.properties[u"name"], command))
214
self.using_timer(False)
215
#self.logger('Checker for client %s (command "%s")'
217
# % (self.properties["Name"], command))
163
221
if not self.last_checker_failed:
164
222
self.last_checker_failed = True
165
self._update_timer_callback_tag = (gobject.timeout_add
223
self.using_timer(True)
168
224
if os.WIFEXITED(condition):
169
self.logger(u'Checker for client %s (command "%s")'
170
u' failed with exit code %s'
171
% (self.properties[u"name"], command,
225
self.logger('Checker for client %s (command "%s")'
226
' failed with exit code %s'
227
% (self.properties["Name"], command,
172
228
os.WEXITSTATUS(condition)))
173
229
elif os.WIFSIGNALED(condition):
174
self.logger(u'Checker for client %s (command "%s")'
175
u' was killed by signal %s'
176
% (self.properties[u"name"], command,
230
self.logger('Checker for client %s (command "%s")'
231
' was killed by signal %s'
232
% (self.properties["Name"], command,
177
233
os.WTERMSIG(condition)))
178
234
elif os.WCOREDUMP(condition):
179
self.logger(u'Checker for client %s (command "%s")'
181
% (self.properties[u"name"], command))
235
self.logger('Checker for client %s (command "%s")'
237
% (self.properties["Name"], command))
183
self.logger(u'Checker for client %s completed mysteriously')
239
self.logger('Checker for client %s completed'
186
243
def checker_started(self, command):
187
#self.logger(u'Client %s started checker "%s"'
188
# % (self.properties[u"name"], unicode(command)))
244
#self.logger('Client %s started checker "%s"'
245
# % (self.properties["Name"], unicode(command)))
191
248
def got_secret(self):
192
self.logger(u'Client %s received its secret'
193
% self.properties[u"name"])
249
self.last_checker_failed = False
250
self.logger('Client %s received its secret'
251
% self.properties["Name"])
195
253
def need_approval(self, timeout, default):
197
message = u'Client %s needs approval within %s seconds'
255
message = 'Client %s needs approval within %s seconds'
199
message = u'Client %s will get its secret in %s seconds'
257
message = 'Client %s will get its secret in %s seconds'
200
258
self.logger(message
201
% (self.properties[u"name"], timeout/1000))
259
% (self.properties["Name"], timeout/1000))
260
self.using_timer(True)
203
262
def rejected(self, reason):
204
self.logger(u'Client %s was rejected; reason: %s'
205
% (self.properties[u"name"], reason))
263
self.logger('Client %s was rejected; reason: %s'
264
% (self.properties["Name"], reason))
207
266
def selectable(self):
208
267
"""Make this a "selectable" widget.
209
268
This overrides the method from urwid.FlowWidget."""
212
def rows(self, (maxcol,), focus=False):
271
def rows(self, maxcolrow, focus=False):
213
272
"""How many rows this widget will occupy might depend on
214
273
whether we have focus or not.
215
274
This overrides the method from urwid.FlowWidget"""
216
return self.current_widget(focus).rows((maxcol,), focus=focus)
275
return self.current_widget(focus).rows(maxcolrow, focus=focus)
218
277
def current_widget(self, focus=False):
219
278
if focus or self.opened:
223
282
def update(self):
224
283
"Called when what is visible on the screen should be updated."
225
284
# How to add standout mode to a style
226
with_standout = { u"normal": u"standout",
227
u"bold": u"bold-standout",
229
u"underline-blink-standout",
230
u"bold-underline-blink":
231
u"bold-underline-blink-standout",
285
with_standout = { "normal": "standout",
286
"bold": "bold-standout",
288
"underline-blink-standout",
289
"bold-underline-blink":
290
"bold-underline-blink-standout",
234
293
# Rebuild focus and non-focus widgets using current properties
236
295
# Base part of a client. Name!
237
base = (u'%(name)s: '
238
% {u"name": self.properties[u"name"]})
239
if not self.properties[u"enabled"]:
240
message = u"DISABLED"
297
% {"name": self.properties["Name"]})
298
if not self.properties["Enabled"]:
300
elif self.properties["ApprovalPending"]:
301
timeout = datetime.timedelta(milliseconds
304
last_approval_request = isoformat_to_datetime(
305
self.properties["LastApprovalRequest"])
306
if last_approval_request is not None:
307
timer = timeout - (datetime.datetime.utcnow()
308
- last_approval_request)
310
timer = datetime.timedelta()
311
if self.properties["ApprovedByDefault"]:
312
message = "Approval in %s. (d)eny?"
314
message = "Denial in %s. (a)pprove?"
315
message = message % unicode(timer).rsplit(".", 1)[0]
241
316
elif self.last_checker_failed:
242
317
timeout = datetime.timedelta(milliseconds
243
= self.properties[u"timeout"])
244
320
last_ok = isoformat_to_datetime(
245
max((self.properties["last_checked_ok"]
246
or self.properties["created"]),
247
self.properties[u"last_enabled"]))
321
max((self.properties["LastCheckedOK"]
322
or self.properties["Created"]),
323
self.properties["LastEnabled"]))
248
324
timer = timeout - (datetime.datetime.utcnow() - last_ok)
250
message = (u'A checker has failed! Time until client gets diabled: %s'
252
elif self.properties[u"approved_pending"]:
253
if self.properties[u"approved_by_default"]:
254
message = u"Connection established to client. (d)eny?"
256
message = u"Seeks approval to send secret. (a)pprove?"
325
message = ('A checker has failed! Time until client'
327
% unicode(timer).rsplit(".", 1)[0])
259
330
self._text = "%s%s" % (base, message)
261
332
if not urwid.supports_unicode():
262
333
self._text = self._text.encode("ascii", "replace")
263
textlist = [(u"normal", self._text)]
334
textlist = [("normal", self._text)]
264
335
self._text_widget.set_text(textlist)
265
336
self._focus_text_widget.set_text([(with_standout[text[0]],
280
351
return True # Keep calling this
353
def delete(self, *args, **kwargs):
283
354
if self._update_timer_callback_tag is not None:
284
355
gobject.source_remove(self._update_timer_callback_tag)
285
356
self._update_timer_callback_tag = None
357
for match in self.match_objects:
359
self.match_objects = ()
286
360
if self.delete_hook is not None:
287
361
self.delete_hook(self)
362
return super(MandosClientWidget, self).delete(*args, **kwargs)
289
def render(self, (maxcol,), focus=False):
364
def render(self, maxcolrow, focus=False):
290
365
"""Render differently if we have focus.
291
366
This overrides the method from urwid.FlowWidget"""
292
return self.current_widget(focus).render((maxcol,),
367
return self.current_widget(focus).render(maxcolrow,
295
def keypress(self, (maxcol,), key):
370
def keypress(self, maxcolrow, key):
297
372
This overrides the method from urwid.FlowWidget"""
299
self.proxy.Enable(dbus_interface = client_interface)
301
self.proxy.Disable(dbus_interface = client_interface)
374
self.proxy.Enable(dbus_interface = client_interface,
377
self.proxy.Disable(dbus_interface = client_interface,
303
380
self.proxy.Approve(dbus.Boolean(True, variant_level=1),
304
dbus_interface = client_interface)
381
dbus_interface = client_interface,
306
384
self.proxy.Approve(dbus.Boolean(False, variant_level=1),
307
dbus_interface = client_interface)
308
elif key == u"r" or key == u"_" or key == u"ctrl k":
385
dbus_interface = client_interface,
387
elif key == "R" or key == "_" or key == "ctrl k":
309
388
self.server_proxy_object.RemoveClient(self.proxy
312
self.proxy.StartChecker(dbus_interface = client_interface)
314
self.proxy.StopChecker(dbus_interface = client_interface)
316
self.proxy.CheckedOK(dbus_interface = client_interface)
392
self.proxy.StartChecker(dbus_interface = client_interface,
395
self.proxy.StopChecker(dbus_interface = client_interface,
398
self.proxy.CheckedOK(dbus_interface = client_interface,
318
# elif key == u"p" or key == "=":
401
# elif key == "p" or key == "=":
319
402
# self.proxy.pause()
320
# elif key == u"u" or key == ":":
403
# elif key == "u" or key == ":":
321
404
# self.proxy.unpause()
322
# elif key == u"RET":
325
# self.proxy.Approve(True)
327
# self.proxy.Approve(False)
362
441
self.screen = urwid.curses_display.Screen()
364
443
self.screen.register_palette((
366
u"default", u"default", None),
368
u"default", u"default", u"bold"),
370
u"default", u"default", u"underline"),
372
u"default", u"default", u"standout"),
373
(u"bold-underline-blink",
374
u"default", u"default", (u"bold", u"underline")),
376
u"default", u"default", (u"bold", u"standout")),
377
(u"underline-blink-standout",
378
u"default", u"default", (u"underline", u"standout")),
379
(u"bold-underline-blink-standout",
380
u"default", u"default", (u"bold", u"underline",
445
"default", "default", None),
447
"default", "default", "bold"),
449
"default", "default", "underline"),
451
"default", "default", "standout"),
452
("bold-underline-blink",
453
"default", "default", ("bold", "underline")),
455
"default", "default", ("bold", "standout")),
456
("underline-blink-standout",
457
"default", "default", ("underline", "standout")),
458
("bold-underline-blink-standout",
459
"default", "default", ("bold", "underline",
384
463
if urwid.supports_unicode():
385
self.divider = u"─" # \u2500
386
#self.divider = u"━" # \u2501
464
self.divider = "─" # \u2500
465
#self.divider = "━" # \u2501
388
#self.divider = u"-" # \u002d
389
self.divider = u"_" # \u005f
467
#self.divider = "-" # \u002d
468
self.divider = "_" # \u005f
391
470
self.screen.start()
406
485
# This keeps track of whether self.uilist currently has
407
486
# self.logbox in it or not
408
487
self.log_visible = True
409
self.log_wrap = u"any"
488
self.log_wrap = "any"
412
self.log_message_raw((u"bold",
413
u"Mandos Monitor version " + version))
414
self.log_message_raw((u"bold",
491
self.log_message_raw(("bold",
492
"Mandos Monitor version " + version))
493
self.log_message_raw(("bold",
417
496
self.busname = domain + '.Mandos'
418
497
self.main_loop = gobject.MainLoop()
419
498
self.bus = dbus.SystemBus()
420
499
mandos_dbus_objc = self.bus.get_object(
421
self.busname, u"/", follow_name_owner_changes=True)
500
self.busname, "/", follow_name_owner_changes=True)
422
501
self.mandos_serv = dbus.Interface(mandos_dbus_objc,
424
503
= server_interface)
489
571
and len(self.log) > self.max_log_length):
490
572
del self.log[0:len(self.log)-self.max_log_length-1]
491
573
self.logbox.set_focus(len(self.logbox.body.contents),
492
coming_from=u"above")
495
577
def toggle_log_display(self):
496
578
"""Toggle visibility of the log buffer."""
497
579
self.log_visible = not self.log_visible
499
self.log_message(u"Log visibility changed to: "
500
+ unicode(self.log_visible))
581
#self.log_message("Log visibility changed to: "
582
# + unicode(self.log_visible))
502
584
def change_log_display(self):
503
585
"""Change type of log display.
504
586
Currently, this toggles wrapping of text lines."""
505
if self.log_wrap == u"clip":
506
self.log_wrap = u"any"
587
if self.log_wrap == "clip":
588
self.log_wrap = "any"
508
self.log_wrap = u"clip"
590
self.log_wrap = "clip"
509
591
for textwidget in self.log:
510
592
textwidget.set_wrap_mode(self.log_wrap)
511
self.log_message(u"Wrap mode: " + self.log_wrap)
593
#self.log_message("Wrap mode: " + self.log_wrap)
513
595
def find_and_remove_client(self, path, name):
514
"""Find an client from its object path and remove it.
596
"""Find a client by its object path and remove it.
516
598
This is connected to the ClientRemoved signal from the
517
599
Mandos server object."""
594
678
except KeyError: # :-)
597
if key == u"q" or key == u"Q":
681
if key == "q" or key == "Q":
600
elif key == u"window resize":
684
elif key == "window resize":
601
685
self.size = self.screen.get_cols_rows()
603
elif key == u"\f": # Ctrl-L
687
elif key == "\f": # Ctrl-L
605
elif key == u"l" or key == u"D":
689
elif key == "l" or key == "D":
606
690
self.toggle_log_display()
608
elif key == u"w" or key == u"i":
692
elif key == "w" or key == "i":
609
693
self.change_log_display()
611
elif key == u"?" or key == u"f1" or key == u"esc":
695
elif key == "?" or key == "f1" or key == "esc":
612
696
if not self.log_visible:
613
697
self.log_visible = True
615
self.log_message_raw((u"bold",
619
u"l: Log window toggle",
620
u"TAB: Switch window",
622
self.log_message_raw((u"bold",
628
u"s: Start new checker",
699
self.log_message_raw(("bold",
703
"l: Log window toggle",
704
"TAB: Switch window",
706
self.log_message_raw(("bold",
712
"s: Start new checker",
635
719
if self.topwidget.get_focus() is self.logbox:
636
720
self.topwidget.set_focus(0)
638
722
self.topwidget.set_focus(self.logbox)
640
#elif (key == u"end" or key == u"meta >" or key == u"G"
724
#elif (key == "end" or key == "meta >" or key == "G"
642
726
# pass # xxx end-of-buffer
643
#elif (key == u"home" or key == u"meta <" or key == u"g"
727
#elif (key == "home" or key == "meta <" or key == "g"
645
729
# pass # xxx beginning-of-buffer
646
#elif key == u"ctrl e" or key == u"$":
730
#elif key == "ctrl e" or key == "$":
647
731
# pass # xxx move-end-of-line
648
#elif key == u"ctrl a" or key == u"^":
732
#elif key == "ctrl a" or key == "^":
649
733
# pass # xxx move-beginning-of-line
650
#elif key == u"ctrl b" or key == u"meta (" or key == u"h":
734
#elif key == "ctrl b" or key == "meta (" or key == "h":
651
735
# pass # xxx left
652
#elif key == u"ctrl f" or key == u"meta )" or key == u"l":
736
#elif key == "ctrl f" or key == "meta )" or key == "l":
653
737
# pass # xxx right
655
739
# pass # scroll up log
657
741
# pass # scroll down log
658
742
elif self.topwidget.selectable():
659
743
self.topwidget.keypress(self.size, key)