Commit | Line | Data |
---|---|---|
3454dce4 | 1 | #! /usr/bin/env python |
3454dce4 | 2 | # |
c215a4bc IJ |
3 | # This file is part of secnet. |
4 | # See README for full list of copyright holders. | |
5 | # | |
6 | # secnet is free software; you can redistribute it and/or modify it | |
7 | # under the terms of the GNU General Public License as published by | |
8 | # the Free Software Foundation; either version d of the License, or | |
3454dce4 SE |
9 | # (at your option) any later version. |
10 | # | |
c215a4bc IJ |
11 | # secnet is distributed in the hope that it will be useful, but |
12 | # WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | # General Public License for more details. | |
3454dce4 SE |
15 | # |
16 | # You should have received a copy of the GNU General Public License | |
c215a4bc IJ |
17 | # version 3 along with secnet; if not, see |
18 | # https://www.gnu.org/licenses/gpl.html. | |
3454dce4 SE |
19 | |
20 | """VPN sites file manipulation. | |
21 | ||
22 | This program enables VPN site descriptions to be submitted for | |
23 | inclusion in a central database, and allows the resulting database to | |
24 | be turned into a secnet configuration file. | |
25 | ||
26 | A database file can be turned into a secnet configuration file simply: | |
27 | make-secnet-sites.py [infile [outfile]] | |
28 | ||
29 | It would be wise to run secnet with the "--just-check-config" option | |
30 | before installing the output on a live system. | |
31 | ||
32 | The program expects to be invoked via userv to manage the database; it | |
33 | relies on the USERV_USER and USERV_GROUP environment variables. The | |
34 | command line arguments for this invocation are: | |
35 | ||
36 | make-secnet-sites.py -u header-filename groupfiles-directory output-file \ | |
37 | group | |
38 | ||
39 | All but the last argument are expected to be set by userv; the 'group' | |
40 | argument is provided by the user. A suitable userv configuration file | |
41 | fragment is: | |
42 | ||
43 | reset | |
44 | no-disconnect-hup | |
45 | no-suppress-args | |
46 | cd ~/secnet/sites-test/ | |
08f344d3 | 47 | execute ~/secnet/make-secnet-sites.py -u vpnheader groupfiles sites |
3454dce4 SE |
48 | |
49 | This program is part of secnet. It relies on the "ipaddr" library from | |
50 | Cendio Systems AB. | |
51 | ||
52 | """ | |
53 | ||
54 | import string | |
55 | import time | |
56 | import sys | |
57 | import os | |
3b83c932 | 58 | import getopt |
040040f3 | 59 | import re |
8dea8d37 | 60 | |
71d65e4c IJ |
61 | import ipaddr |
62 | ||
90ad8cd4 IJ |
63 | sys.path.insert(0,"/usr/local/share/secnet") |
64 | sys.path.insert(0,"/usr/share/secnet") | |
71d65e4c | 65 | import ipaddrset |
3454dce4 | 66 | |
00152558 | 67 | VERSION="0.1.18" |
3b83c932 SE |
68 | |
69 | # Classes describing possible datatypes in the configuration file | |
70 | ||
71 | class single_ipaddr: | |
72 | "An IP address" | |
73 | def __init__(self,w): | |
71d65e4c | 74 | self.addr=ipaddr.IPAddress(w[1]) |
3b83c932 | 75 | def __str__(self): |
71d65e4c | 76 | return '"%s"'%self.addr |
3b83c932 SE |
77 | |
78 | class networks: | |
79 | "A set of IP addresses specified as a list of networks" | |
3454dce4 | 80 | def __init__(self,w): |
71d65e4c | 81 | self.set=ipaddrset.IPAddressSet() |
3454dce4 | 82 | for i in w[1:]: |
71d65e4c IJ |
83 | x=ipaddr.IPNetwork(i,strict=True) |
84 | self.set.append([x]) | |
3b83c932 | 85 | def __str__(self): |
71d65e4c | 86 | return ",".join(map((lambda n: '"%s"'%n), self.set.networks())) |
3454dce4 SE |
87 | |
88 | class dhgroup: | |
3b83c932 | 89 | "A Diffie-Hellman group" |
3454dce4 | 90 | def __init__(self,w): |
b2a56f7c SE |
91 | self.mod=w[1] |
92 | self.gen=w[2] | |
3b83c932 SE |
93 | def __str__(self): |
94 | return 'diffie-hellman("%s","%s")'%(self.mod,self.gen) | |
3454dce4 SE |
95 | |
96 | class hash: | |
3b83c932 | 97 | "A choice of hash function" |
3454dce4 | 98 | def __init__(self,w): |
b2a56f7c SE |
99 | self.ht=w[1] |
100 | if (self.ht!='md5' and self.ht!='sha1'): | |
101 | complain("unknown hash type %s"%(self.ht)) | |
3b83c932 SE |
102 | def __str__(self): |
103 | return '%s'%(self.ht) | |
3454dce4 SE |
104 | |
105 | class email: | |
3b83c932 | 106 | "An email address" |
3454dce4 | 107 | def __init__(self,w): |
b2a56f7c | 108 | self.addr=w[1] |
3b83c932 SE |
109 | def __str__(self): |
110 | return '<%s>'%(self.addr) | |
3454dce4 | 111 | |
040040f3 IJ |
112 | class boolean: |
113 | "A boolean" | |
114 | def __init__(self,w): | |
115 | if re.match('[TtYy1]',w[1]): | |
116 | self.b=True | |
117 | elif re.match('[FfNn0]',w[1]): | |
118 | self.b=False | |
119 | else: | |
120 | complain("invalid boolean value"); | |
121 | def __str__(self): | |
122 | return ['False','True'][self.b] | |
123 | ||
3454dce4 | 124 | class num: |
3b83c932 | 125 | "A decimal number" |
3454dce4 | 126 | def __init__(self,w): |
b2a56f7c | 127 | self.n=string.atol(w[1]) |
3b83c932 SE |
128 | def __str__(self): |
129 | return '%d'%(self.n) | |
3454dce4 SE |
130 | |
131 | class address: | |
3b83c932 | 132 | "A DNS name and UDP port number" |
3454dce4 | 133 | def __init__(self,w): |
b2a56f7c SE |
134 | self.adr=w[1] |
135 | self.port=string.atoi(w[2]) | |
136 | if (self.port<1 or self.port>65535): | |
137 | complain("invalid port number") | |
3b83c932 SE |
138 | def __str__(self): |
139 | return '"%s"; port %d'%(self.adr,self.port) | |
3454dce4 SE |
140 | |
141 | class rsakey: | |
3b83c932 | 142 | "An RSA public key" |
3454dce4 | 143 | def __init__(self,w): |
b2a56f7c SE |
144 | self.l=string.atoi(w[1]) |
145 | self.e=w[2] | |
146 | self.n=w[3] | |
3b83c932 SE |
147 | def __str__(self): |
148 | return 'rsa-public("%s","%s")'%(self.e,self.n) | |
149 | ||
150 | # Possible properties of configuration nodes | |
151 | keywords={ | |
152 | 'contact':(email,"Contact address"), | |
153 | 'dh':(dhgroup,"Diffie-Hellman group"), | |
154 | 'hash':(hash,"Hash function"), | |
155 | 'key-lifetime':(num,"Maximum key lifetime (ms)"), | |
156 | 'setup-timeout':(num,"Key setup timeout (ms)"), | |
157 | 'setup-retries':(num,"Maximum key setup packet retries"), | |
158 | 'wait-time':(num,"Time to wait after unsuccessful key setup (ms)"), | |
159 | 'renegotiate-time':(num,"Time after key setup to begin renegotiation (ms)"), | |
160 | 'restrict-nets':(networks,"Allowable networks"), | |
161 | 'networks':(networks,"Claimed networks"), | |
162 | 'pubkey':(rsakey,"RSA public site key"), | |
163 | 'peer':(single_ipaddr,"Tunnel peer IP address"), | |
a25b1149 | 164 | 'address':(address,"External contact address and port"), |
040040f3 | 165 | 'mobile':(boolean,"Site is mobile"), |
3b83c932 SE |
166 | } |
167 | ||
168 | def sp(name,value): | |
169 | "Simply output a property - the default case" | |
170 | return "%s %s;\n"%(name,value) | |
171 | ||
172 | # All levels support these properties | |
173 | global_properties={ | |
174 | 'contact':(lambda name,value:"# Contact email address: %s\n"%(value)), | |
175 | 'dh':sp, | |
176 | 'hash':sp, | |
177 | 'key-lifetime':sp, | |
178 | 'setup-timeout':sp, | |
179 | 'setup-retries':sp, | |
180 | 'wait-time':sp, | |
181 | 'renegotiate-time':sp, | |
a25b1149 | 182 | 'restrict-nets':(lambda name,value:"# restrict-nets %s\n"%value), |
3b83c932 SE |
183 | } |
184 | ||
185 | class level: | |
186 | "A level in the configuration hierarchy" | |
187 | depth=0 | |
188 | leaf=0 | |
189 | allow_properties={} | |
190 | require_properties={} | |
191 | def __init__(self,w): | |
192 | self.name=w[1] | |
193 | self.properties={} | |
194 | self.children={} | |
195 | def indent(self,w,t): | |
196 | w.write(" "[:t]) | |
197 | def prop_out(self,n): | |
198 | return self.allow_properties[n](n,str(self.properties[n])) | |
199 | def output_props(self,w,ind): | |
200 | for i in self.properties.keys(): | |
201 | if self.allow_properties[i]: | |
202 | self.indent(w,ind) | |
203 | w.write("%s"%self.prop_out(i)) | |
204 | def output_data(self,w,ind,np): | |
205 | self.indent(w,ind) | |
206 | w.write("%s {\n"%(self.name)) | |
207 | self.output_props(w,ind+2) | |
208 | if self.depth==1: w.write("\n"); | |
209 | for c in self.children.values(): | |
210 | c.output_data(w,ind+2,np+self.name+"/") | |
211 | self.indent(w,ind) | |
212 | w.write("};\n") | |
213 | ||
214 | class vpnlevel(level): | |
215 | "VPN level in the configuration hierarchy" | |
216 | depth=1 | |
217 | leaf=0 | |
218 | type="vpn" | |
219 | allow_properties=global_properties.copy() | |
220 | require_properties={ | |
221 | 'contact':"VPN admin contact address" | |
222 | } | |
223 | def __init__(self,w): | |
224 | level.__init__(self,w) | |
225 | def output_vpnflat(self,w,ind,h): | |
226 | "Output flattened list of site names for this VPN" | |
227 | self.indent(w,ind) | |
228 | w.write("%s {\n"%(self.name)) | |
229 | for i in self.children.keys(): | |
230 | self.children[i].output_vpnflat(w,ind+2, | |
231 | h+"/"+self.name+"/"+i) | |
232 | w.write("\n") | |
233 | self.indent(w,ind+2) | |
234 | w.write("all-sites %s;\n"% | |
235 | string.join(self.children.keys(),',')) | |
236 | self.indent(w,ind) | |
237 | w.write("};\n") | |
238 | ||
239 | class locationlevel(level): | |
240 | "Location level in the configuration hierarchy" | |
241 | depth=2 | |
242 | leaf=0 | |
243 | type="location" | |
244 | allow_properties=global_properties.copy() | |
245 | require_properties={ | |
246 | 'contact':"Location admin contact address", | |
247 | } | |
248 | def __init__(self,w): | |
249 | level.__init__(self,w) | |
250 | self.group=w[2] | |
251 | def output_vpnflat(self,w,ind,h): | |
252 | self.indent(w,ind) | |
253 | # The "h=h,self=self" abomination below exists because | |
254 | # Python didn't support nested_scopes until version 2.1 | |
255 | w.write("%s %s;\n"%(self.name,string.join( | |
256 | map(lambda x,h=h,self=self: | |
257 | h+"/"+x,self.children.keys()),','))) | |
258 | ||
259 | class sitelevel(level): | |
260 | "Site level (i.e. a leafnode) in the configuration hierarchy" | |
261 | depth=3 | |
262 | leaf=1 | |
263 | type="site" | |
264 | allow_properties=global_properties.copy() | |
265 | allow_properties.update({ | |
266 | 'address':sp, | |
267 | 'networks':None, | |
268 | 'peer':None, | |
a25b1149 | 269 | 'pubkey':(lambda n,v:"key %s;\n"%v), |
2489e9eb | 270 | 'address':(lambda n,v:"address %s;\n"%v), |
040040f3 | 271 | 'mobile':sp, |
3b83c932 SE |
272 | }) |
273 | require_properties={ | |
274 | 'dh':"Diffie-Hellman group", | |
275 | 'contact':"Site admin contact address", | |
3b83c932 SE |
276 | 'networks':"Networks claimed by the site", |
277 | 'hash':"hash function", | |
278 | 'peer':"Gateway address of the site", | |
a25b1149 | 279 | 'pubkey':"RSA public key of the site", |
3b83c932 | 280 | } |
3454dce4 | 281 | def __init__(self,w): |
3b83c932 SE |
282 | level.__init__(self,w) |
283 | def output_data(self,w,ind,np): | |
284 | self.indent(w,ind) | |
285 | w.write("%s {\n"%(self.name)) | |
286 | self.indent(w,ind+2) | |
287 | w.write("name \"%s\";\n"%(np+self.name)) | |
288 | self.output_props(w,ind+2) | |
289 | self.indent(w,ind+2) | |
290 | w.write("link netlink {\n"); | |
291 | self.indent(w,ind+4) | |
292 | w.write("routes %s;\n"%str(self.properties["networks"])) | |
293 | self.indent(w,ind+4) | |
294 | w.write("ptp-address %s;\n"%str(self.properties["peer"])) | |
295 | self.indent(w,ind+2) | |
296 | w.write("};\n") | |
297 | self.indent(w,ind) | |
298 | w.write("};\n") | |
299 | ||
300 | # Levels in the configuration file | |
301 | # (depth,properties) | |
302 | levels={'vpn':vpnlevel, 'location':locationlevel, 'site':sitelevel} | |
303 | ||
304 | # Reserved vpn/location/site names | |
305 | reserved={'all-sites':None} | |
306 | reserved.update(keywords) | |
307 | reserved.update(levels) | |
3454dce4 SE |
308 | |
309 | def complain(msg): | |
3b83c932 | 310 | "Complain about a particular input line" |
3454dce4 SE |
311 | global complaints |
312 | print ("%s line %d: "%(file,line))+msg | |
313 | complaints=complaints+1 | |
314 | def moan(msg): | |
3b83c932 | 315 | "Complain about something in general" |
3454dce4 SE |
316 | global complaints |
317 | print msg; | |
318 | complaints=complaints+1 | |
319 | ||
3b83c932 SE |
320 | root=level(['root','root']) # All vpns are children of this node |
321 | obstack=[root] | |
322 | allow_defs=0 # Level above which new definitions are permitted | |
26f727b9 | 323 | prefix='' |
3b83c932 SE |
324 | |
325 | def set_property(obj,w): | |
326 | "Set a property on a configuration node" | |
327 | if obj.properties.has_key(w[0]): | |
328 | complain("%s %s already has property %s defined"% | |
329 | (obj.type,obj.name,w[0])) | |
330 | else: | |
331 | obj.properties[w[0]]=keywords[w[0]][0](w) | |
3454dce4 | 332 | |
c4497add | 333 | def pline(i,allow_include=False): |
3b83c932 SE |
334 | "Process a configuration file line" |
335 | global allow_defs, obstack, root | |
6d8cd9b2 | 336 | w=string.split(i.rstrip('\n')) |
433b0ae8 | 337 | if len(w)==0: return [i] |
3454dce4 | 338 | keyword=w[0] |
3b83c932 | 339 | current=obstack[len(obstack)-1] |
3454dce4 | 340 | if keyword=='end-definitions': |
3b83c932 SE |
341 | allow_defs=sitelevel.depth |
342 | obstack=[root] | |
433b0ae8 | 343 | return [i] |
c4497add IJ |
344 | if keyword=='include': |
345 | if not allow_include: | |
346 | complain("include not permitted here") | |
433b0ae8 | 347 | return [] |
c4497add IJ |
348 | if len(w) != 2: |
349 | complain("include requires one argument") | |
433b0ae8 | 350 | return [] |
c4497add | 351 | newfile=os.path.join(os.path.dirname(file),w[1]) |
433b0ae8 | 352 | return pfilepath(newfile,allow_include=allow_include) |
3b83c932 SE |
353 | if levels.has_key(keyword): |
354 | # We may go up any number of levels, but only down by one | |
355 | newdepth=levels[keyword].depth | |
356 | currentdepth=len(obstack) # actually +1... | |
357 | if newdepth<=currentdepth: | |
358 | obstack=obstack[:newdepth] | |
359 | if newdepth>currentdepth: | |
360 | complain("May not go from level %d to level %d"% | |
361 | (currentdepth-1,newdepth)) | |
362 | # See if it's a new one (and whether that's permitted) | |
363 | # or an existing one | |
364 | current=obstack[len(obstack)-1] | |
365 | if current.children.has_key(w[1]): | |
366 | # Not new | |
367 | current=current.children[w[1]] | |
368 | if service and group and current.depth==2: | |
369 | if group!=current.group: | |
370 | complain("Incorrect group!") | |
3454dce4 | 371 | else: |
3b83c932 SE |
372 | # New |
373 | # Ignore depth check for now | |
374 | nl=levels[keyword](w) | |
375 | if nl.depth<allow_defs: | |
376 | complain("New definitions not allowed at " | |
377 | "level %d"%nl.depth) | |
4a9b680b IJ |
378 | # we risk crashing if we continue |
379 | sys.exit(1) | |
3b83c932 SE |
380 | current.children[w[1]]=nl |
381 | current=nl | |
382 | obstack.append(current) | |
433b0ae8 | 383 | return [i] |
3b83c932 SE |
384 | if current.allow_properties.has_key(keyword): |
385 | set_property(current,w) | |
433b0ae8 | 386 | return [i] |
3454dce4 | 387 | else: |
3b83c932 SE |
388 | complain("Property %s not allowed at %s level"% |
389 | (keyword,current.type)) | |
433b0ae8 | 390 | return [] |
3b83c932 SE |
391 | |
392 | complain("unknown keyword '%s'"%(keyword)) | |
3454dce4 | 393 | |
c4497add | 394 | def pfilepath(pathname,allow_include=False): |
9b8369e0 | 395 | f=open(pathname) |
433b0ae8 | 396 | outlines=pfile(pathname,f.readlines(),allow_include=allow_include) |
9b8369e0 | 397 | f.close() |
433b0ae8 | 398 | return outlines |
9b8369e0 | 399 | |
c4497add | 400 | def pfile(name,lines,allow_include=False): |
3b83c932 | 401 | "Process a file" |
3454dce4 SE |
402 | global file,line |
403 | file=name | |
404 | line=0 | |
433b0ae8 | 405 | outlines=[] |
3454dce4 SE |
406 | for i in lines: |
407 | line=line+1 | |
408 | if (i[0]=='#'): continue | |
433b0ae8 IJ |
409 | outlines += pline(i,allow_include=allow_include) |
410 | return outlines | |
3454dce4 SE |
411 | |
412 | def outputsites(w): | |
3b83c932 SE |
413 | "Output include file for secnet configuration" |
414 | w.write("# secnet sites file autogenerated by make-secnet-sites " | |
3454dce4 | 415 | +"version %s\n"%VERSION) |
3b83c932 SE |
416 | w.write("# %s\n"%time.asctime(time.localtime(time.time()))) |
417 | w.write("# Command line: %s\n\n"%string.join(sys.argv)) | |
3454dce4 SE |
418 | |
419 | # Raw VPN data section of file | |
26f727b9 | 420 | w.write(prefix+"vpn-data {\n") |
3b83c932 SE |
421 | for i in root.children.values(): |
422 | i.output_data(w,2,"") | |
3454dce4 SE |
423 | w.write("};\n") |
424 | ||
425 | # Per-VPN flattened lists | |
26f727b9 | 426 | w.write(prefix+"vpn {\n") |
3b83c932 | 427 | for i in root.children.values(): |
26f727b9 | 428 | i.output_vpnflat(w,2,prefix+"vpn-data") |
3454dce4 SE |
429 | w.write("};\n") |
430 | ||
431 | # Flattened list of sites | |
26f727b9 IJ |
432 | w.write(prefix+"all-sites %s;\n"%string.join( |
433 | map(lambda x:"%svpn/%s/all-sites"%(prefix,x), | |
434 | root.children.keys()),",")) | |
3454dce4 SE |
435 | |
436 | # Are we being invoked from userv? | |
437 | service=0 | |
438 | # If we are, which group does the caller want to modify? | |
439 | group=None | |
440 | ||
3454dce4 SE |
441 | line=0 |
442 | file=None | |
443 | complaints=0 | |
444 | ||
3454dce4 SE |
445 | if len(sys.argv)<2: |
446 | pfile("stdin",sys.stdin.readlines()) | |
447 | of=sys.stdout | |
448 | else: | |
449 | if sys.argv[1]=='-u': | |
450 | if len(sys.argv)!=6: | |
451 | print "Wrong number of arguments" | |
452 | sys.exit(1) | |
453 | service=1 | |
454 | header=sys.argv[2] | |
455 | groupfiledir=sys.argv[3] | |
456 | sitesfile=sys.argv[4] | |
457 | group=sys.argv[5] | |
458 | if not os.environ.has_key("USERV_USER"): | |
459 | print "Environment variable USERV_USER not found" | |
460 | sys.exit(1) | |
461 | user=os.environ["USERV_USER"] | |
462 | # Check that group is in USERV_GROUP | |
463 | if not os.environ.has_key("USERV_GROUP"): | |
464 | print "Environment variable USERV_GROUP not found" | |
465 | sys.exit(1) | |
466 | ugs=os.environ["USERV_GROUP"] | |
467 | ok=0 | |
468 | for i in string.split(ugs): | |
469 | if group==i: ok=1 | |
470 | if not ok: | |
471 | print "caller not in group %s"%group | |
472 | sys.exit(1) | |
5b77d1a9 | 473 | headerinput=pfilepath(header,allow_include=True) |
3454dce4 SE |
474 | userinput=sys.stdin.readlines() |
475 | pfile("user input",userinput) | |
476 | else: | |
26f727b9 IJ |
477 | if sys.argv[1]=='-P': |
478 | prefix=sys.argv[2] | |
479 | sys.argv[1:3]=[] | |
3454dce4 SE |
480 | if len(sys.argv)>3: |
481 | print "Too many arguments" | |
482 | sys.exit(1) | |
21fd3a92 | 483 | pfilepath(sys.argv[1]) |
3454dce4 SE |
484 | of=sys.stdout |
485 | if len(sys.argv)>2: | |
486 | of=open(sys.argv[2],'w') | |
487 | ||
488 | # Sanity check section | |
3b83c932 SE |
489 | # Delete nodes where leaf=0 that have no children |
490 | ||
491 | def live(n): | |
492 | "Number of leafnodes below node n" | |
493 | if n.leaf: return 1 | |
494 | for i in n.children.keys(): | |
495 | if live(n.children[i]): return 1 | |
496 | return 0 | |
497 | def delempty(n): | |
498 | "Delete nodes that have no leafnode children" | |
499 | for i in n.children.keys(): | |
500 | delempty(n.children[i]) | |
501 | if not live(n.children[i]): | |
502 | del n.children[i] | |
503 | delempty(root) | |
504 | ||
505 | # Check that all constraints are met (as far as I can tell | |
506 | # restrict-nets/networks/peer are the only special cases) | |
507 | ||
508 | def checkconstraints(n,p,ra): | |
509 | new_p=p.copy() | |
510 | new_p.update(n.properties) | |
511 | for i in n.require_properties.keys(): | |
512 | if not new_p.has_key(i): | |
513 | moan("%s %s is missing property %s"% | |
514 | (n.type,n.name,i)) | |
515 | for i in new_p.keys(): | |
516 | if not n.allow_properties.has_key(i): | |
517 | moan("%s %s has forbidden property %s"% | |
518 | (n.type,n.name,i)) | |
519 | # Check address range restrictions | |
520 | if n.properties.has_key("restrict-nets"): | |
521 | new_ra=ra.intersection(n.properties["restrict-nets"].set) | |
3454dce4 | 522 | else: |
3b83c932 SE |
523 | new_ra=ra |
524 | if n.properties.has_key("networks"): | |
71d65e4c | 525 | if not n.properties["networks"].set <= new_ra: |
3b83c932 SE |
526 | moan("%s %s networks out of bounds"%(n.type,n.name)) |
527 | if n.properties.has_key("peer"): | |
528 | if not n.properties["networks"].set.contains( | |
529 | n.properties["peer"].addr): | |
530 | moan("%s %s peer not in networks"%(n.type,n.name)) | |
531 | for i in n.children.keys(): | |
532 | checkconstraints(n.children[i],new_p,new_ra) | |
533 | ||
71d65e4c | 534 | checkconstraints(root,{},ipaddrset.complete_set()) |
3454dce4 SE |
535 | |
536 | if complaints>0: | |
537 | if complaints==1: print "There was 1 problem." | |
538 | else: print "There were %d problems."%(complaints) | |
539 | sys.exit(1) | |
540 | ||
541 | if service: | |
542 | # Put the user's input into their group file, and rebuild the main | |
543 | # sites file | |
08f344d3 | 544 | f=open(groupfiledir+"/T"+group,'w') |
3454dce4 SE |
545 | f.write("# Section submitted by user %s, %s\n"% |
546 | (user,time.asctime(time.localtime(time.time())))) | |
3b83c932 | 547 | f.write("# Checked by make-secnet-sites version %s\n\n"%VERSION) |
3454dce4 SE |
548 | for i in userinput: f.write(i) |
549 | f.write("\n") | |
550 | f.close() | |
08f344d3 SE |
551 | os.rename(groupfiledir+"/T"+group,groupfiledir+"/R"+group) |
552 | f=open(sitesfile+"-tmp",'w') | |
ff05a229 | 553 | f.write("# sites file autogenerated by make-secnet-sites\n") |
08f344d3 SE |
554 | f.write("# generated %s, invoked by %s\n"% |
555 | (time.asctime(time.localtime(time.time())),user)) | |
ff05a229 | 556 | f.write("# use make-secnet-sites to turn this file into a\n") |
08f344d3 SE |
557 | f.write("# valid /etc/secnet/sites.conf file\n\n") |
558 | for i in headerinput: f.write(i) | |
559 | files=os.listdir(groupfiledir) | |
560 | for i in files: | |
561 | if i[0]=='R': | |
562 | j=open(groupfiledir+"/"+i) | |
563 | f.write(j.read()) | |
564 | j.close() | |
565 | f.write("# end of sites file\n") | |
566 | f.close() | |
567 | os.rename(sitesfile+"-tmp",sitesfile) | |
3454dce4 SE |
568 | else: |
569 | outputsites(of) |