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
|
import binascii, lxml.etree, time
import Bcfg2.Server.Admin
class Pull(Bcfg2.Server.Admin.Mode):
'''Pull mode retrieves entries from clients and integrates the information into the repository'''
__shorthelp__ = 'bcfg2-admin pull <client> <entry type> <entry name>'
__longhelp__ = __shorthelp__ + '\n\tIntegrate configuration information from clients into the server repository'
def __init__(self, configfile):
Bcfg2.Server.Admin.Mode.__init__(self, configfile)
def __call__(self, args):
Bcfg2.Server.Admin.Mode.__call__(self, args)
self.PullEntry(args[0], args[1], args[2])
def PullEntry(self, client, etype, ename):
'''Make currently recorded client state correct for entry'''
# FIXME Pull.py is _way_ too interactive
sdata = self.load_stats(client)
if sdata.xpath('.//Statistics[@state="dirty"]'):
state = 'dirty'
else:
state = 'clean'
# need to pull entry out of statistics
sxpath = ".//Statistics[@state='%s']/Bad/ConfigFile[@name='%s']/../.." % (state, ename)
sentries = sdata.xpath(sxpath)
if not len(sentries):
self.errExit("Found %d entries for %s:%s:%s" % \
(len(sentries), client, etype, ename))
else:
print "Found %d entries for %s:%s:%s" % \
(len(sentries), client, etype, ename)
maxtime = max([time.strptime(stat.get('time')) for stat in sentries])
print "Found entry from", time.strftime("%c", maxtime)
statblock = [stat for stat in sentries \
if time.strptime(stat.get('time')) == maxtime]
entry = statblock[0].xpath('.//Bad/ConfigFile[@name="%s"]' % ename)
if not entry:
self.errExit("Could not find state data for entry; rerun bcfg2 on client system")
cfentry = entry[-1]
badfields = [field for field in ['perms', 'owner', 'group'] \
if cfentry.get(field) != cfentry.get('current_' + field) and \
cfentry.get('current_' + field)]
if badfields:
m_updates = dict([(field, cfentry.get('current_' + field)) \
for field in badfields])
print "got metadata_updates", m_updates
else:
m_updates = {}
if 'current_bdiff' in cfentry.attrib:
data = False
diff = binascii.a2b_base64(cfentry.get('current_bdiff'))
elif 'current_diff' in cfentry.attrib:
data = False
diff = cfentry.get('current_diff')
elif 'current_bfile' in cfentry.attrib:
data = binascii.a2b_base64(cfentry.get('current_bfile'))
diff = False
else:
if not m_updates:
self.errExit("having trouble processing entry. Entry is:\n" \
+ lxml.etree.tostring(cfentry))
else:
data = False
diff = False
if diff:
print "Located diff:\n %s" % diff
elif data:
print "Found full (binary) file data"
if m_updates:
print "Found metadata updates"
if not diff and not data and not m_updates:
self.errExit("Failed to locate diff or full data or metadata updates\nStatistics entry was:\n%s" % lxml.etree.tostring(cfentry))
try:
bcore = Bcfg2.Server.Core.Core({}, self.configfile)
except Bcfg2.Server.Core.CoreInitError, msg:
self.errExit("Core load failed because %s" % msg)
[bcore.fam.Service() for _ in range(10)]
while bcore.fam.Service():
pass
m = bcore.metadata.get_metadata(client)
# find appropriate plugin in bcore
glist = [gen for gen in bcore.generators if
gen.Entries.get(etype, {}).has_key(ename)]
if len(glist) != 1:
self.errExit("Got wrong numbers of matching generators for entry:" \
+ "%s" % ([g.__name__ for g in glist]))
plugin = glist[0]
try:
plugin.AcceptEntry(m, 'ConfigFile', ename, diff, data, m_updates)
except Bcfg2.Server.Plugin.PluginExecutionError:
self.errExit("Configuration upload not supported by plugin %s" \
% (plugin.__name__))
# svn commit if running under svn
|
