1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#!/usr/bin/env python
'''Bcfg2 Client'''
__revision__ = '$Revision$'
from getopt import getopt, GetoptError
from os import popen, chmod, unlink
from sys import argv, exit as sysexit
from tempfile import mktemp
from elementtree.ElementTree import Element, XML, tostring
from sss.ssslib import comm_lib
from Bcfg2.Client.Debian import Debian
def RunProbe(probe):
'''Execute probe'''
ret = Element("probe-data", name=probe.attrib['name'], source=probe.attrib['source'])
script = open(mktemp(), 'w+')
script.write("#!%s\n"%(probe.attrib.get('interpreter', '/bin/sh')))
script.write(probe.text)
script.close()
chmod(script.name, 0755)
ret.text = popen(script.name).read()
unlink(script.name)
return ret
def dgetopt(arglist, opt, vopt):
'''parse options into a dictionary'''
ret = {}
for optname in opt.values() + vopt.values():
ret[optname] = False
gstr = "".join(opt.keys()) + "".join([xy+':' for xy in vopt.keys()])
try:
(o, a) = getopt(arglist, gstr)
except GetoptError, gerr:
print gerr
print "bcfg2 Usage:"
for arg in opt.iteritems():
print " -%s %s" % arg
for arg in vopt.iteritems():
print " -%s <%s>" % arg
sysexit(1)
for (gopt, garg) in o:
option = gopt[1:]
if opt.has_key(option):
ret[opt[option]] = True
else:
ret[vopt[option]] = garg
return ret
if __name__ == '__main__':
# parse command line options
options = {'v':'verbose', 'q':'quick', 'd':'debug', 'n':'dryrun', 'B':'build', 'p':'paranoid'}
doptions = {'b':'bundle', 'f':'file', 'c':'cache', 'p':'profile', 'i':'image'}
setup = dgetopt(argv[1:], options, doptions)
# connect to bcfg2d
comm = comm_lib()
h = comm.ClientInit("bcfg2")
# get probes
comm.SendMessage(h, "<get-probes/>")
data = comm.RecvMessage(h)
#if setup['verbose']: print data
probes = XML(data)
# execute probes
cpd = Element("probe-data")
[cpd.append(RunProbe(x)) for x in probes.findall(".//probe")]
# upload probe responses
comm.SendMessage(h, tostring(cpd))
r = comm.RecvMessage(h)
msg = Element("get-config")
if setup['profile']:
msg.attrib['profile'] = setup['profile']
if setup['image']:
msg.attrib['image'] = setup['image']
# get config
comm.SendMessage(h, tostring(msg))
r = comm.RecvMessage(h)
if setup['cache']:
try:
open(setup['cache'], 'w').write(r)
except IOError:
print "failed to write config cache file %s" % (setup['cache'])
cfg = XML(r)
if cfg.tag == 'error':
print "got error from server"
sysexit(1)
if setup['bundle']:
c = XML("<Configuration version='2.0/>")
for child in cfg.getroot().getchildren():
if ((child.tag == 'Bundle') and (child.attrib['name'] == setup['bundle'])):
c.append(child)
cfg = c
client = Debian(cfg, setup)
# verify state
client.Inventory()
correct = client.states.values().count(True)
total = len(client.states.values())
if correct < total:
# summarize current state
client.CondPrint('verbose', "--> %s of %s config elements correct" % (correct, total))
# install incorrect aspects of configuration
client.Install()
client.CondPrint('verbose', "--> %s of %s config elements correct"%(client.states.values().count(True), total))
print "Failing Entries:"
for k, v in client.states.iteritems():
if not v:
print "%s:%s" % (k.tag, k.attrib['name'])
else:
client.CondPrint("verbose", "All entries correct")
# upload statistics
m = Element("upload-statistics")
stats = client.GenerateStats()
m.append(stats)
comm.SendMessage(h, tostring(m))
r = comm.RecvMessage(h)
# clean up
comm.ClientClose(h)
|