2
# -*- mode: python; coding: utf-8; after-save-hook: (lambda () (let ((command (if (and (boundp 'tramp-file-name-structure) (string-match (car tramp-file-name-structure) (buffer-file-name))) (tramp-file-name-localname (tramp-dissect-file-name (buffer-file-name))) (buffer-file-name)))) (if (= (shell-command (format "%s --check" (shell-quote-argument command)) "*Test*") 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w)) (kill-buffer "*Test*")) (display-buffer "*Test*")))); -*-
2
# -*- mode: python; coding: utf-8 -*-
4
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2008-2019 Teddy Hogeborn
7
# Copyright © 2008-2019 Björn Påhlsson
9
# This file is part of Mandos.
11
# Mandos is free software: you can redistribute it and/or modify it
12
# under the terms of the GNU General Public License as published by
6
# Copyright © 2008-2016 Teddy Hogeborn
7
# Copyright © 2008-2016 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
13
11
# the Free Software Foundation, either version 3 of the License, or
14
12
# (at your option) any later version.
16
# Mandos is distributed in the hope that it will be useful, but
17
# WITHOUT ANY WARRANTY; without even the implied warranty of
14
# This program is distributed in the hope that it will be useful,
15
# but WITHOUT ANY WARRANTY; without even the implied warranty of
18
16
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19
17
# GNU General Public License for more details.
21
19
# You should have received a copy of the GNU General Public License
22
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
20
# along with this program. If not, see
21
# <http://www.gnu.org/licenses/>.
24
23
# Contact the authors at <mandos@recompile.se>.
27
26
from __future__ import (division, absolute_import, print_function,
79
65
"ApprovalDelay": "Approval Delay",
80
66
"ApprovalDuration": "Approval Duration",
81
67
"Checker": "Checker",
82
"ExtendedTimeout": "Extended Timeout",
84
"LastCheckerStatus": "Last Checker Status",
68
"ExtendedTimeout": "Extended Timeout"
86
70
defaultkeywords = ("Name", "Enabled", "Timeout", "LastCheckedOK")
87
71
domain = "se.recompile"
97
81
except AttributeError:
98
82
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
101
84
def milliseconds_to_string(ms):
102
85
td = datetime.timedelta(0, 0, 0, ms)
103
return ("{days}{hours:02}:{minutes:02}:{seconds:02}"
104
.format(days="{}T".format(td.days) if td.days else "",
105
hours=td.seconds // 3600,
106
minutes=(td.seconds % 3600) // 60,
107
seconds=td.seconds % 60))
86
return ("{days}{hours:02}:{minutes:02}:{seconds:02}".format(
87
days = "{}T".format(td.days) if td.days else "",
88
hours = td.seconds // 3600,
89
minutes = (td.seconds % 3600) // 60,
90
seconds = td.seconds % 60))
110
93
def rfc3339_duration_to_delta(duration):
111
94
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
113
96
>>> rfc3339_duration_to_delta("P7D")
114
97
datetime.timedelta(7)
115
98
>>> rfc3339_duration_to_delta("PT60S")
125
108
>>> rfc3339_duration_to_delta("P1DT3M20S")
126
109
datetime.timedelta(1, 200)
129
112
# Parsing an RFC 3339 duration with regular expressions is not
130
113
# possible - there would have to be multiple places for the same
131
114
# values, like seconds. The current code, while more esoteric, is
132
115
# cleaner without depending on a parsing library. If Python had a
133
116
# built-in library for parsing we would use it, but we'd like to
134
117
# avoid excessive use of external libraries.
136
119
# New type for defining tokens, syntax, and semantics all-in-one
137
120
Token = collections.namedtuple("Token", (
138
121
"regexp", # To match token; if "value" is not None, must have
171
154
frozenset((token_year, token_month,
172
155
token_day, token_time,
174
# Define starting values:
176
value = datetime.timedelta()
157
# Define starting values
158
value = datetime.timedelta() # Value so far
177
159
found_token = None
178
# Following valid tokens
179
followers = frozenset((token_duration, ))
180
# String left to parse
160
followers = frozenset((token_duration, )) # Following valid tokens
161
s = duration # String left to parse
182
162
# Loop until end token is found
183
163
while found_token is not token_end:
184
164
# Search for any currently valid tokens
255
235
"ApprovalDuration", "ExtendedTimeout"):
256
236
return milliseconds_to_string(value)
257
237
return str(value)
259
239
# Create format string to print table rows
260
240
format_string = " ".join("{{{key}:{width}}}".format(
261
width=max(len(tablewords[key]),
262
max(len(valuetostring(client[key], key))
263
for client in clients)),
241
width = max(len(tablewords[key]),
242
max(len(valuetostring(client[key], key))
243
for client in clients)),
265
245
for key in keywords)
266
246
# Print header line
267
247
print(format_string.format(**tablewords))
268
248
for client in clients:
270
.format(**{key: valuetostring(client[key], key)
271
for key in keywords}))
249
print(format_string.format(**{
250
key: valuetostring(client[key], key)
251
for key in keywords }))
274
254
def has_actions(options):
296
276
parser = argparse.ArgumentParser()
297
277
parser.add_argument("--version", action="version",
298
version="%(prog)s {}".format(version),
278
version = "%(prog)s {}".format(version),
299
279
help="show version number and exit")
300
280
parser.add_argument("-a", "--all", action="store_true",
301
281
help="Select all clients")
347
327
help="Run self-test")
348
328
parser.add_argument("client", nargs="*", help="Client name")
349
329
options = parser.parse_args()
351
331
if has_actions(options) and not (options.client or options.all):
352
332
parser.error("Options require clients names or --all.")
353
333
if options.verbose and has_actions(options):
354
334
parser.error("--verbose can only be used alone.")
355
if options.dump_json and (options.verbose
356
or has_actions(options)):
335
if options.dump_json and (options.verbose or has_actions(options)):
357
336
parser.error("--dump-json can only be used alone.")
358
337
if options.all and not has_actions(options):
359
338
parser.error("--all requires an action.")
341
fail_count, test_count = doctest.testmod()
342
sys.exit(os.EX_OK if fail_count == 0 else 1)
362
345
bus = dbus.SystemBus()
363
346
mandos_dbus_objc = bus.get_object(busname, server_path)
364
347
except dbus.exceptions.DBusException:
365
log.critical("Could not connect to Mandos server")
348
print("Could not connect to Mandos server", file=sys.stderr)
368
351
mandos_serv = dbus.Interface(mandos_dbus_objc,
369
dbus_interface=server_interface)
352
dbus_interface = server_interface)
370
353
mandos_serv_object_manager = dbus.Interface(
371
mandos_dbus_objc, dbus_interface=dbus.OBJECT_MANAGER_IFACE)
373
# block stderr since dbus library prints to stderr
354
mandos_dbus_objc, dbus_interface = dbus.OBJECT_MANAGER_IFACE)
356
#block stderr since dbus library prints to stderr
374
357
null = os.open(os.path.devnull, os.O_RDWR)
375
358
stderrcopy = os.dup(sys.stderr.fileno())
376
359
os.dup2(null, sys.stderr.fileno())
380
mandos_clients = {path: ifs_and_props[client_interface]
381
for path, ifs_and_props in
382
mandos_serv_object_manager
383
.GetManagedObjects().items()
384
if client_interface in ifs_and_props}
363
mandos_clients = { path: ifs_and_props[client_interface]
364
for path, ifs_and_props in
365
mandos_serv_object_manager
366
.GetManagedObjects().items()
367
if client_interface in ifs_and_props }
387
370
os.dup2(stderrcopy, sys.stderr.fileno())
388
371
os.close(stderrcopy)
389
372
except dbus.exceptions.DBusException as e:
390
log.critical("Failed to access Mandos server through D-Bus:"
373
print("Access denied: Accessing mandos server through D-Bus: {}"
374
.format(e), file=sys.stderr)
394
377
# Compile dict of (clients: properties) to process
397
380
if options.all or not options.client:
398
clients = {bus.get_object(busname, path): properties
399
for path, properties in mandos_clients.items()}
381
clients = { bus.get_object(busname, path): properties
382
for path, properties in mandos_clients.items() }
401
384
for name in options.client:
402
385
for path, client in mandos_clients.items():
405
388
clients[client_objc] = client
408
log.critical("Client not found on server: %r", name)
391
print("Client not found on server: {!r}"
392
.format(name), file=sys.stderr)
411
395
if not has_actions(options) and clients:
412
396
if options.verbose or options.dump_json:
413
397
keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK",
414
"Created", "Interval", "Host", "KeyID",
415
"Fingerprint", "CheckerRunning",
416
"LastEnabled", "ApprovalPending",
417
"ApprovedByDefault", "LastApprovalRequest",
418
"ApprovalDelay", "ApprovalDuration",
419
"Checker", "ExtendedTimeout", "Expires",
398
"Created", "Interval", "Host", "Fingerprint",
399
"CheckerRunning", "LastEnabled",
400
"ApprovalPending", "ApprovedByDefault",
401
"LastApprovalRequest", "ApprovalDelay",
402
"ApprovalDuration", "Checker",
422
405
keywords = defaultkeywords
424
407
if options.dump_json:
425
408
json.dump({client["Name"]: {key:
426
409
bool(client[key])
427
410
if isinstance(client[key],
431
for client in clients.values()},
432
fp=sys.stdout, indent=4,
433
separators=(',', ': '))
413
for key in keywords }
414
for client in clients.values() },
415
fp = sys.stdout, indent = 4,
416
separators = (',', ': '))
436
419
print_clients(clients.values(), keywords)
438
421
# Process each client in the list by all selected options
439
422
for client in clients:
441
424
def set_client_prop(prop, value):
442
425
"""Set a Client D-Bus property"""
443
426
client.Set(client_interface, prop, value,
444
427
dbus_interface=dbus.PROPERTIES_IFACE)
446
429
def set_client_prop_ms(prop, value):
447
430
"""Set a Client D-Bus property, converted
448
431
from a string to milliseconds."""
449
432
set_client_prop(prop,
450
433
string_to_delta(value).total_seconds()
453
436
if options.remove:
454
437
mandos_serv.RemoveClient(client.__dbus_object_path__)
455
438
if options.enable:
463
446
if options.stop_checker:
464
447
set_client_prop("CheckerRunning", dbus.Boolean(False))
465
448
if options.is_enabled:
466
if client.Get(client_interface, "Enabled",
467
dbus_interface=dbus.PROPERTIES_IFACE):
449
sys.exit(0 if client.Get(client_interface,
452
dbus.PROPERTIES_IFACE)
471
454
if options.checker is not None:
472
455
set_client_prop("Checker", options.checker)
473
456
if options.host is not None:
499
482
client.Approve(dbus.Boolean(False),
500
483
dbus_interface=client_interface)
503
class Test_milliseconds_to_string(unittest.TestCase):
505
self.assertEqual(milliseconds_to_string(93785000),
507
def test_no_days(self):
508
self.assertEqual(milliseconds_to_string(7385000), "02:03:05")
509
def test_all_zero(self):
510
self.assertEqual(milliseconds_to_string(0), "00:00:00")
511
def test_no_fractional_seconds(self):
512
self.assertEqual(milliseconds_to_string(400), "00:00:00")
513
self.assertEqual(milliseconds_to_string(900), "00:00:00")
514
self.assertEqual(milliseconds_to_string(1900), "00:00:01")
517
def should_only_run_tests():
518
parser = argparse.ArgumentParser(add_help=False)
519
parser.add_argument("--check", action='store_true')
520
args, unknown_args = parser.parse_known_args()
521
run_tests = args.check
523
# Remove --check argument from sys.argv
524
sys.argv[1:] = unknown_args
527
# Add all tests from doctest strings
528
def load_tests(loader, tests, none):
530
tests.addTests(doctest.DocTestSuite())
533
486
if __name__ == "__main__":
534
if should_only_run_tests():
535
# Call using ./tdd-python-script --check [--verbose]