/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release
24.1.116 by Björn Påhlsson
added a mandos list client program
1
#!/usr/bin/python
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
2
# -*- mode: python; coding: utf-8 -*-
24.1.116 by Björn Påhlsson
added a mandos list client program
3
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
4
import sys
24.1.116 by Björn Påhlsson
added a mandos list client program
5
import dbus
6
from optparse import OptionParser
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
7
import locale
8
9
locale.setlocale(locale.LC_ALL, u'')
24.1.116 by Björn Påhlsson
added a mandos list client program
10
11
tablewords = {
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
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',
24.1.116 by Björn Påhlsson
added a mandos list client program
23
    }
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
24
defaultkeywords = ('name', 'enabled', 'timeout', 'last_checked_ok',
25
                   'checker')
24.1.116 by Björn Påhlsson
added a mandos list client program
26
busname = 'org.mandos-system.Mandos'
24.1.118 by Björn Påhlsson
Added enable/disable
27
server_path = '/Mandos'
28
server_interface = 'org.mandos_system.Mandos'
29
client_interface = 'org.mandos_system.Mandos.Client'
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
30
version = "1.0.2"
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
31
24.1.118 by Björn Påhlsson
Added enable/disable
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
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
38
def print_clients():
24.1.118 by Björn Påhlsson
Added enable/disable
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
24.1.116 by Björn Påhlsson
added a mandos list client program
55
parser = OptionParser(version = "%%prog %s" % version)
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
56
parser.add_option("-a", "--all", action="store_true",
237.2.3 by Teddy Hogeborn
Merge "mandos-list" from belorn.
57
                  help="Print all fields")
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
58
parser.add_option("-e", "--enable", action="store_true",
24.1.118 by Björn Påhlsson
Added enable/disable
59
                  help="Enable specified client")
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
60
parser.add_option("-d", "--disable", action="store_true",
24.1.118 by Björn Påhlsson
Added enable/disable
61
                  help="disable specified client")
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
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:
24.1.116 by Björn Påhlsson
added a mandos list client program
84
    keywords = defaultkeywords
24.1.119 by Björn Påhlsson
Added more method support for mandos clients through mandos-ctl
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