2
2
# -*- mode: python; coding: utf-8 -*-
4
from __future__ import division
7
5
from optparse import OptionParser
12
8
locale.setlocale(locale.LC_ALL, u'')
16
'Enabled': u'Enabled',
17
'Timeout': u'Timeout',
18
'LastCheckedOK': u'Last Successful Check',
19
'Created': u'Created',
20
'Interval': u'Interval',
22
'Fingerprint': u'Fingerprint',
23
'CheckerRunning': u'Check Is Running',
24
'LastEnabled': u'Last Enabled',
25
'Checker': u'Checker',
12
'enabled': u'Enabled',
13
'timeout': u'Timeout',
14
'last_checked_ok': u'Last Successful Check',
15
'created': u'Created',
16
'interval': u'Interval',
18
'fingerprint': u'Fingerprint',
19
'checker_running': u'Check Is Running',
20
'last_enabled': u'Last Enabled',
21
'checker': u'Checker',
27
defaultkeywords = ('Name', 'Enabled', 'Timeout', 'LastCheckedOK')
28
domain = 'se.bsnet.fukt'
29
busname = domain + '.Mandos'
31
server_interface = domain + '.Mandos'
32
client_interface = domain + '.Mandos.Client'
35
bus = dbus.SystemBus()
36
mandos_dbus_objc = bus.get_object(busname, server_path)
37
except dbus.exceptions.DBusException:
23
busname = 'org.mandos-system.Mandos'
24
object_path = '/Mandos'
25
interface = 'org.mandos_system.Mandos'
27
defaultkeywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
30
parser = OptionParser(version = "%%prog %s" % version)
31
parser.add_option("-a", "--all", action="store_true", default=False,
32
help="Print all fields")
33
options = parser.parse_args()[0]
35
keywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
36
'created', 'interval', 'host', 'fingerprint',
37
'checker_running', 'last_enabled', 'checker')
39
keywords = defaultkeywords
42
bus = dbus.SystemBus()
43
mandos_dbus_objc = bus.get_object(busname, object_path)
40
44
mandos_serv = dbus.Interface(mandos_dbus_objc,
41
dbus_interface = server_interface)
45
dbus_interface = interface)
42
46
mandos_clients = mandos_serv.GetAllClientsWithProperties()
44
def timedelta_to_milliseconds(td):
45
"Convert a datetime.timedelta object to milliseconds"
46
return ((td.days * 24 * 60 * 60 * 1000)
48
+ (td.microseconds // 1000))
50
def milliseconds_to_string(ms):
51
td = datetime.timedelta(0, 0, 0, ms)
52
return (u"%(days)s%(hours)02d:%(minutes)02d:%(seconds)02d"
53
% { "days": "%dT" % td.days if td.days else "",
54
"hours": td.seconds // 3600,
55
"minutes": (td.seconds % 3600) // 60,
56
"seconds": td.seconds % 60,
60
def string_to_delta(interval):
61
"""Parse a string and return a datetime.timedelta
63
>>> string_to_delta('7d')
65
>>> string_to_delta('60s')
66
datetime.timedelta(0, 60)
67
>>> string_to_delta('60m')
68
datetime.timedelta(0, 3600)
69
>>> string_to_delta('24h')
71
>>> string_to_delta(u'1w')
73
>>> string_to_delta('5m 30s')
74
datetime.timedelta(0, 330)
76
timevalue = datetime.timedelta(0)
77
regexp = re.compile("\d+[dsmhw]")
79
for s in regexp.findall(interval):
81
suffix = unicode(s[-1])
84
delta = datetime.timedelta(value)
86
delta = datetime.timedelta(0, value)
88
delta = datetime.timedelta(0, 0, 0, 0, value)
90
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
92
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
95
except (ValueError, IndexError):
100
def print_clients(clients):
101
def valuetostring(value, keyword):
102
if type(value) is dbus.Boolean:
103
return u"Yes" if value else u"No"
104
if keyword in (u"timeout", u"interval"):
105
return milliseconds_to_string(value)
106
return unicode(value)
108
# Create format string to print table rows
109
format_string = u' '.join(u'%%-%ds' %
110
max(len(tablewords[key]),
111
max(len(valuetostring(client[key],
117
print format_string % tuple(tablewords[key] for key in keywords)
118
for client in clients:
119
print format_string % tuple(valuetostring(client[key], key)
122
parser = OptionParser(version = "%%prog %s" % version)
123
parser.add_option("-a", "--all", action="store_true",
124
help="Print all fields")
125
parser.add_option("-e", "--enable", action="store_true",
126
help="Enable client")
127
parser.add_option("-d", "--disable", action="store_true",
128
help="disable client")
129
parser.add_option("-b", "--bump-timeout", action="store_true",
130
help="Bump timeout for client")
131
parser.add_option("--start-checker", action="store_true",
132
help="Start checker for client")
133
parser.add_option("--stop-checker", action="store_true",
134
help="Stop checker for client")
135
parser.add_option("-V", "--is-enabled", action="store_true",
136
help="Check if client is enabled")
137
parser.add_option("-r", "--remove", action="store_true",
138
help="Remove client")
139
parser.add_option("-c", "--checker", type="string",
140
help="Set checker command for client")
141
parser.add_option("-t", "--timeout", type="string",
142
help="Set timeout for client")
143
parser.add_option("-i", "--interval", type="string",
144
help="Set checker interval for client")
145
parser.add_option("-H", "--host", type="string",
146
help="Set host for client")
147
parser.add_option("-s", "--secret", type="string",
148
help="Set password blob (file) for client")
149
parser.add_option("-A", "--approve", action="store_true",
150
help="Approve any current client request")
151
parser.add_option("-D", "--deny", action="store_true",
152
help="Deny any current client request")
153
options, client_names = parser.parse_args()
155
# Compile list of clients to process
157
for name in client_names:
158
for path, client in mandos_clients.iteritems():
159
if client['name'] == name:
160
client_objc = bus.get_object(busname, path)
161
clients.append(client_objc)
49
if type(x) is dbus.Boolean:
50
return u"Yes" if x else u"No"
164
print >> sys.stderr, "Client not found on server: %r" % name
167
if not clients and mandos_clients.values():
168
keywords = defaultkeywords
170
keywords = ('Name', 'Enabled', 'Timeout', 'LastCheckedOK',
171
'Created', 'Interval', 'Host', 'Fingerprint',
172
'CheckerRunning', 'LastEnabled', 'Checker')
173
print_clients(mandos_clients.values())
175
# Process each client in the list by all selected options
176
for client in clients:
178
mandos_serv.RemoveClient(client.__dbus_object_path__)
180
client.Enable(dbus_interface=client_interface)
182
client.Disable(dbus_interface=client_interface)
183
if options.bump_timeout:
184
client.CheckedOK(dbus_interface=client_interface)
185
if options.start_checker:
186
client.StartChecker(dbus_interface=client_interface)
187
if options.stop_checker:
188
client.StopChecker(dbus_interface=client_interface)
189
if options.is_enabled:
190
sys.exit(0 if client.Get(client_interface,
192
dbus_interface=dbus.PROPERTIES_IFACE)
195
client.Set(client_interface, u"Checker", options.checker,
196
dbus_interface=dbus.PROPERTIES_IFACE)
198
client.Set(client_interface, u"Host", options.host,
199
dbus_interface=dbus.PROPERTIES_IFACE)
201
client.Set(client_interface, u"Interval",
202
timedelta_to_milliseconds
203
(string_to_delta(options.interval)),
204
dbus_interface=dbus.PROPERTIES_IFACE)
206
client.Set(client_interface, u"Timeout",
207
timedelta_to_milliseconds(string_to_delta
209
dbus_interface=dbus.PROPERTIES_IFACE)
211
client.Set(client_interface, u"Secret",
212
dbus.ByteArray(open(options.secret, u'rb').read()),
213
dbus_interface=dbus.PROPERTIES_IFACE)
215
client.Approve(dbus.Boolean(True),
216
dbus_interface=client_interface)
218
client.Approve(dbus.Boolean(False),
219
dbus_interface=client_interface)
54
format_string = u' '.join(u'%%-%ds'
55
% max(len(tablewords[key]),
56
max(len(valuetostring(client[key]))
58
in mandos_clients.itervalues()))
60
print format_string % tuple(tablewords[key] for key in keywords)
61
for client in mandos_clients.itervalues():
62
print format_string % tuple(valuetostring(client[key])
added
removed
mandos-list
