/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to mandos-ctl

merge

Show diffs side-by-side

added added

removed removed

Lines of Context:
mandos-ctl
1
 
#!/usr/bin/python
2
 
# -*- mode: python; coding: utf-8 -*-
3
 
 
4
 
import sys
5
 
import dbus
6
 
from optparse import OptionParser
7
 
import locale
8
 
 
9
 
locale.setlocale(locale.LC_ALL, u'')
10
 
 
11
 
tablewords = {
12
 
    'name': u'Name',
13
 
    'enabled': u'Enabled',
14
 
    'timeout': u'Timeout',
15
 
    'last_checked_ok': u'Last Successful Check',
16
 
    'created': u'Created',
17
 
    'interval': u'Interval',
18
 
    'host': u'Host',
19
 
    'fingerprint': u'Fingerprint',
20
 
    'checker_running': u'Check Is Running',
21
 
    'last_enabled': u'Last Enabled',
22
 
    'checker': u'Checker',
23
 
    }
24
 
defaultkeywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
25
 
                   'checker')
26
 
busname = 'org.mandos-system.Mandos'
27
 
server_path = '/Mandos'
28
 
server_interface = 'org.mandos_system.Mandos'
29
 
client_interface = 'org.mandos_system.Mandos.Client'
30
 
version = "1.0.2"
31
 
 
32
 
bus = dbus.SystemBus()
33
 
mandos_dbus_objc = bus.get_object(busname, server_path)
34
 
mandos_serv = dbus.Interface(mandos_dbus_objc,
35
 
                             dbus_interface = server_interface)
36
 
mandos_clients = mandos_serv.GetAllClientsWithProperties()
37
 
 
38
 
def print_clients():
39
 
    def valuetostring(x):
40
 
        if type(x) is dbus.Boolean:
41
 
            return u"Yes" if x else u"No"
42
 
        else:
43
 
            return unicode(x)
44
 
    format_string = u' '.join(u'%%-%ds' %
45
 
                              max(len(tablewords[key]),
46
 
                                  max(len(valuetostring(client[key]))
47
 
                                      for client in
48
 
                                      mandos_clients.itervalues()))
49
 
                              for key in keywords)
50
 
    print format_string % tuple(tablewords[key] for key in keywords) 
51
 
    for client in mandos_clients.itervalues():
52
 
        print format_string % tuple(valuetostring(client[key])
53
 
                                    for key in keywords)
54
 
 
55
 
parser = OptionParser(version = "%%prog %s" % version)
56
 
parser.add_option("-a", "--all", action="store_true",
57
 
                  help="Print all fields")
58
 
parser.add_option("-e", "--enable", action="store_true",
59
 
                  help="Enable specified client")
60
 
parser.add_option("-d", "--disable", action="store_true",
61
 
                  help="disable specified client")
62
 
parser.add_option("-b", "--bumptimeout", action="store_true",
63
 
                  help="Bump timeout of specified client")
64
 
parser.add_option("-s", "--startchecker", action="store_true",
65
 
                  help="Start checker for specified client")
66
 
parser.add_option("-S", "--stopchecker", action="store_true",
67
 
                  help="Stop checker for specified client")
68
 
options, client_names = parser.parse_args()
69
 
 
70
 
clients=[]
71
 
for name in client_names:
72
 
    for path, client in mandos_clients.iteritems():
73
 
        if client['name'] == name:
74
 
            client_objc = bus.get_object(busname, path)
75
 
            clients.append(dbus.Interface(client_objc,
76
 
                                          dbus_interface
77
 
                                          = client_interface))
78
 
            break
79
 
    else:
80
 
        print >> sys.stderr, "Client not found on server: %r" % name
81
 
        sys.exit(1)
82
 
 
83
 
if not clients:
84
 
    keywords = defaultkeywords
85
 
    if options.all:
86
 
        keywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
87
 
                    'created', 'interval', 'host', 'fingerprint',
88
 
                    'checker_running', 'last_enabled', 'checker')
89
 
    print_clients()
90
 
 
91
 
for client in clients:
92
 
    if options.enable:
93
 
        client.Enable()
94
 
    if options.disable:
95
 
        client.Disable()
96
 
    if options.bumptimeout:
97
 
        client.BumpTimeout()
98
 
    if options.startchecker:
99
 
        client.StartChecker()
100
 
    if options.stopchecker:
101
 
        client.StopChecker()
102