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
|
#!/usr/bin/env python
# $Id: $
from socket import gethostbyaddr, herror
from string import split
from syslog import syslog, LOG_INFO, LOG_ERR
from sys import argv, exit, exc_info
from time import time
from traceback import extract_tb
from ConfigParser import ConfigParser
from elementtree.ElementTree import Element, tostring
from Bcfg2.Server.Core import Core
from Bcfg2.Server.Metadata import MetadataStore
from sss.daemonize import daemonize
from sss.restriction import DataSet, Data
from sss.server import Server
class BcfgServer(Server):
__implementation__ = 'Bcfg2'
__component__ = 'bcfg2'
__dispatch__ = {'get-config':'BuildConfig', 'get-probes':'GetProbes', 'probe-data':'CommitProbeData'}
__validate__ = 0
def __setup__(self):
c = ConfigParser()
c.read([self.kwargs.get('configfile', '/etc/bcfg2.conf')])
repo = c.get('server','repository')
generators = split(c.get('server','generators'),',')
structures = split(c.get('server', 'structures'),',')
mpath = c.get('server','metadata')
self.core = Core(repo, structures, generators)
self.metadata = MetadataStore("%s/metadata.xml"%(mpath), self.core.fam)
self.__progress__()
def __progress__(self):
while self.core.fam.fm.pending():
self.core.fam.HandleEvent()
try:
self.core.RunCronTasks()
except:
self.LogFailure("Cron")
return 0
def BuildConfig(self, xml, (peer,port)):
try:
client = gethostbyaddr(peer)[0].split('.')[0]
except herror:
return Element("error", type='host resolution error')
t = time()
config = Element("Configuration", version='2.0')
# get metadata for host
if xml.attrib.has_key("profile") and xml.attrib.has_key("image"):
m = self.metadata.FetchMetadata(client, image=xml.attrib['image'], profile=xml.attrib['profile'])
else:
m = self.metadata.FetchMetadata(client)
try:
structures = self.core.GetStructures(m)
except:
self.LogFailure("GetStructures")
return Element("error", type='structure error')
for s in structures:
try:
self.core.BindStructure(s, m)
config.append(s)
except:
self.LogFailure("BindStructure")
#for x in s.getchildren():
# print x.attrib['name'], '\000' in tostring(x)
syslog(LOG_INFO, "Generated config for %s in %s seconds"%(client, time()-t))
return config
def GetProbes(self, xml, (peer,port)):
r = Element('probes')
try:
client = gethostbyaddr(peer)[0].split('.')[0]
except herror:
return Element("error", type='host resolution error')
m = self.GetMetadata(client)
for g in self.core.generators:
for p in g.GetProbes(m):
r.append(p)
return r
def CommitProbeData(self, xml, (peer,port)):
try:
client = gethostbyaddr(peer)[0].split('.')[0]
except herror:
return Element("error", type='host resolution error')
for data in xml.findall(".//probe-data"):
try:
[g] = [x for x in self.core.generators if x.__name__ == data.attrib['source']]
g.AcceptProbeData(client, data)
except:
self.LogFailure("CommitProbeData")
return Element("OK")
def LogFailure(self, failure):
(t,v,tb)=exc_info()
syslog(LOG_ERR, "Unexpected failure in %s"%(failure))
for line in extract_tb(tb):
syslog(LOG_ERR, ' File "%s", line %i, in %s\n %s\n'%line)
syslog(LOG_ERR, "%s: %s\n"%(t,v))
del t,v,tb
if __name__ == '__main__':
if '--daemon' in argv:
daemonize('/dev/null')
server = BcfgServer()
for i in range(10):
server.__progress__()
server.ServeForever()
|
