2
2
# -*- mode: python; coding: utf-8 -*-
4
4
# Mandos Monitor - Control and monitor the Mandos server
6
# Copyright © 2008-2014 Teddy Hogeborn
7
# Copyright © 2008-2014 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
6
# Copyright © 2008-2018 Teddy Hogeborn
7
# Copyright © 2008-2018 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
11
13
# the Free Software Foundation, either version 3 of the License, or
12
14
# (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
# Mandos is distributed in the hope that it will be useful, but
17
# WITHOUT ANY WARRANTY; without even the implied warranty of
16
18
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17
19
# GNU General Public License for more details.
19
21
# You should have received a copy of the GNU General Public License
20
# along with this program. If not, see
21
# <http://www.gnu.org/licenses/>.
22
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
23
24
# Contact the authors at <mandos@recompile.se>.
26
27
from __future__ import (division, absolute_import, print_function,
64
66
"ApprovalDelay": "Approval Delay",
65
67
"ApprovalDuration": "Approval Duration",
66
68
"Checker": "Checker",
67
"ExtendedTimeout" : "Extended Timeout"
69
"ExtendedTimeout": "Extended Timeout",
71
"LastCheckerStatus": "Last Checker Status",
69
73
defaultkeywords = ("Name", "Enabled", "Timeout", "LastCheckedOK")
70
74
domain = "se.recompile"
71
75
busname = domain + ".Mandos"
73
77
server_interface = domain + ".Mandos"
74
78
client_interface = domain + ".Mandos.Client"
77
def timedelta_to_milliseconds(td):
78
"""Convert a datetime.timedelta object to milliseconds"""
79
return ((td.days * 24 * 60 * 60 * 1000)
81
+ (td.microseconds // 1000))
83
dbus.OBJECT_MANAGER_IFACE
84
except AttributeError:
85
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
83
88
def milliseconds_to_string(ms):
84
89
td = datetime.timedelta(0, 0, 0, ms)
85
90
return ("{days}{hours:02}:{minutes:02}:{seconds:02}"
86
.format(days = "{0}T".format(td.days) if td.days else "",
87
hours = td.seconds // 3600,
88
minutes = (td.seconds % 3600) // 60,
89
seconds = td.seconds % 60,
91
.format(days="{}T".format(td.days) if td.days else "",
92
hours=td.seconds // 3600,
93
minutes=(td.seconds % 3600) // 60,
94
seconds=td.seconds % 60))
93
97
def rfc3339_duration_to_delta(duration):
94
98
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
96
100
>>> rfc3339_duration_to_delta("P7D")
97
101
datetime.timedelta(7)
98
102
>>> rfc3339_duration_to_delta("PT60S")
108
112
>>> rfc3339_duration_to_delta("P1DT3M20S")
109
113
datetime.timedelta(1, 200)
112
116
# Parsing an RFC 3339 duration with regular expressions is not
113
117
# possible - there would have to be multiple places for the same
114
118
# values, like seconds. The current code, while more esoteric, is
115
119
# cleaner without depending on a parsing library. If Python had a
116
120
# built-in library for parsing we would use it, but we'd like to
117
121
# avoid excessive use of external libraries.
119
123
# New type for defining tokens, syntax, and semantics all-in-one
120
Token = collections.namedtuple("Token",
121
("regexp", # To match token; if
122
# "value" is not None,
123
# must have a "group"
125
"value", # datetime.timedelta or
127
"followers")) # Tokens valid after
124
Token = collections.namedtuple("Token", (
125
"regexp", # To match token; if "value" is not None, must have
126
# a "group" containing digits
127
"value", # datetime.timedelta or None
128
"followers")) # Tokens valid after this token
129
129
# RFC 3339 "duration" tokens, syntax, and semantics; taken from
130
130
# the "duration" ABNF definition in RFC 3339, Appendix A.
131
131
token_end = Token(re.compile(r"$"), None, frozenset())
132
132
token_second = Token(re.compile(r"(\d+)S"),
133
133
datetime.timedelta(seconds=1),
134
frozenset((token_end,)))
134
frozenset((token_end, )))
135
135
token_minute = Token(re.compile(r"(\d+)M"),
136
136
datetime.timedelta(minutes=1),
137
137
frozenset((token_second, token_end)))
186
189
# No currently valid tokens were found
187
raise ValueError("Invalid RFC 3339 duration")
190
raise ValueError("Invalid RFC 3339 duration: {!r}"
188
192
# End token found
192
196
def string_to_delta(interval):
193
197
"""Parse a string and return a datetime.timedelta
195
>>> string_to_delta("7d")
199
>>> string_to_delta('7d')
196
200
datetime.timedelta(7)
197
>>> string_to_delta("60s")
201
>>> string_to_delta('60s')
198
202
datetime.timedelta(0, 60)
199
>>> string_to_delta("60m")
203
>>> string_to_delta('60m')
200
204
datetime.timedelta(0, 3600)
201
>>> string_to_delta("24h")
205
>>> string_to_delta('24h')
202
206
datetime.timedelta(1)
203
>>> string_to_delta("1w")
207
>>> string_to_delta('1w')
204
208
datetime.timedelta(7)
205
>>> string_to_delta("5m 30s")
209
>>> string_to_delta('5m 30s')
206
210
datetime.timedelta(0, 330)
210
214
return rfc3339_duration_to_delta(interval)
211
215
except ValueError:
214
218
value = datetime.timedelta(0)
215
219
regexp = re.compile(r"(\d+)([dsmhw]?)")
217
221
for num, suffix in regexp.findall(interval):
218
222
if suffix == "d":
219
223
value += datetime.timedelta(int(num))
278
283
parser = argparse.ArgumentParser()
279
284
parser.add_argument("--version", action="version",
280
version = "%(prog)s {0}".format(version),
285
version="%(prog)s {}".format(version),
281
286
help="show version number and exit")
282
287
parser.add_argument("-a", "--all", action="store_true",
283
288
help="Select all clients")
284
289
parser.add_argument("-v", "--verbose", action="store_true",
285
290
help="Print all fields")
291
parser.add_argument("-j", "--dump-json", action="store_true",
292
help="Dump client data in JSON format")
286
293
parser.add_argument("-e", "--enable", action="store_true",
287
294
help="Enable client")
288
295
parser.add_argument("-d", "--disable", action="store_true",
327
334
help="Run self-test")
328
335
parser.add_argument("client", nargs="*", help="Client name")
329
336
options = parser.parse_args()
331
338
if has_actions(options) and not (options.client or options.all):
332
339
parser.error("Options require clients names or --all.")
333
340
if options.verbose and has_actions(options):
334
parser.error("--verbose can only be used alone or with"
341
parser.error("--verbose can only be used alone.")
342
if options.dump_json and (options.verbose
343
or has_actions(options)):
344
parser.error("--dump-json can only be used alone.")
336
345
if options.all and not has_actions(options):
337
346
parser.error("--all requires an action.")
339
348
if options.check:
340
350
fail_count, test_count = doctest.testmod()
341
351
sys.exit(os.EX_OK if fail_count == 0 else 1)
344
354
bus = dbus.SystemBus()
345
355
mandos_dbus_objc = bus.get_object(busname, server_path)
346
356
except dbus.exceptions.DBusException:
347
print("Could not connect to Mandos server",
357
print("Could not connect to Mandos server", file=sys.stderr)
351
360
mandos_serv = dbus.Interface(mandos_dbus_objc,
352
dbus_interface = server_interface)
354
#block stderr since dbus library prints to stderr
361
dbus_interface=server_interface)
362
mandos_serv_object_manager = dbus.Interface(
363
mandos_dbus_objc, dbus_interface=dbus.OBJECT_MANAGER_IFACE)
365
# block stderr since dbus library prints to stderr
355
366
null = os.open(os.path.devnull, os.O_RDWR)
356
367
stderrcopy = os.dup(sys.stderr.fileno())
357
368
os.dup2(null, sys.stderr.fileno())
361
mandos_clients = mandos_serv.GetAllClientsWithProperties()
372
mandos_clients = {path: ifs_and_props[client_interface]
373
for path, ifs_and_props in
374
mandos_serv_object_manager
375
.GetManagedObjects().items()
376
if client_interface in ifs_and_props}
364
379
os.dup2(stderrcopy, sys.stderr.fileno())
365
380
os.close(stderrcopy)
366
except dbus.exceptions.DBusException:
367
print("Access denied: Accessing mandos server through dbus.",
381
except dbus.exceptions.DBusException as e:
382
print("Access denied: "
383
"Accessing mandos server through D-Bus: {}".format(e),
371
387
# Compile dict of (clients: properties) to process
374
390
if options.all or not options.client:
375
clients = dict((bus.get_object(busname, path), properties)
376
for path, properties in
377
mandos_clients.items())
391
clients = {bus.get_object(busname, path): properties
392
for path, properties in mandos_clients.items()}
379
394
for name in options.client:
380
for path, client in mandos_clients.iteritems():
395
for path, client in mandos_clients.items():
381
396
if client["Name"] == name:
382
397
client_objc = bus.get_object(busname, path)
383
398
clients[client_objc] = client
386
print("Client not found on server: {0!r}"
401
print("Client not found on server: {!r}"
387
402
.format(name), file=sys.stderr)
390
405
if not has_actions(options) and clients:
392
keywords = ("Name", "Enabled", "Timeout",
393
"LastCheckedOK", "Created", "Interval",
394
"Host", "Fingerprint", "CheckerRunning",
406
if options.verbose or options.dump_json:
407
keywords = ("Name", "Enabled", "Timeout", "LastCheckedOK",
408
"Created", "Interval", "Host", "KeyID",
409
"Fingerprint", "CheckerRunning",
395
410
"LastEnabled", "ApprovalPending",
397
"LastApprovalRequest", "ApprovalDelay",
398
"ApprovalDuration", "Checker",
411
"ApprovedByDefault", "LastApprovalRequest",
412
"ApprovalDelay", "ApprovalDuration",
413
"Checker", "ExtendedTimeout", "Expires",
401
416
keywords = defaultkeywords
403
print_clients(clients.values(), keywords)
418
if options.dump_json:
419
json.dump({client["Name"]: {key:
421
if isinstance(client[key],
425
for client in clients.values()},
426
fp=sys.stdout, indent=4,
427
separators=(',', ': '))
430
print_clients(clients.values(), keywords)
405
432
# Process each client in the list by all selected options
406
433
for client in clients:
407
435
def set_client_prop(prop, value):
408
436
"""Set a Client D-Bus property"""
409
437
client.Set(client_interface, prop, value,
410
438
dbus_interface=dbus.PROPERTIES_IFACE)
411
440
def set_client_prop_ms(prop, value):
412
441
"""Set a Client D-Bus property, converted
413
442
from a string to milliseconds."""
414
443
set_client_prop(prop,
415
timedelta_to_milliseconds
416
(string_to_delta(value)))
444
string_to_delta(value).total_seconds()
417
447
if options.remove:
418
448
mandos_serv.RemoveClient(client.__dbus_object_path__)
419
449
if options.enable: