4b0dc76312c3efe989e428b64db3cfcc59cdcbad
[pyplnet.git] / plnet.py
1 # $Id$
2
3 import os
4 import socket
5 import time
6
7 import sioc
8 import modprobe
9
10 def InitInterfaces(logger, plc, data, root="", files_only=False):
11     sysconfig = "%s/etc/sysconfig/network-scripts" % root
12
13     # query running network interfaces
14     devs = sioc.gifconf()
15     ips = dict(zip(devs.values(), devs.keys()))
16     macs = {}
17     for dev in devs:
18         macs[sioc.gifhwaddr(dev).lower()] = dev
19
20     # assume data['networks'] contains this node's NodeNetworks
21     interfaces = {}
22     interface = 1
23     hostname = data.get('hostname',socket.gethostname())
24     networks = data['networks']
25     failedToGetSettings = False
26     for network in networks:
27         logger.verbose('net:InitInterfaces interface %d: %s'%(interface,network))
28         logger.verbose('net:InitInterfaces macs = %s' % macs)
29         logger.verbose('net:InitInterfaces ips = %s' % ips)
30         # Get interface name preferably from MAC address, falling back
31         # on IP address.
32         hwaddr=network['mac']
33         if hwaddr <> None: hwaddr=hwaddr.lower()
34         if hwaddr in macs:
35             orig_ifname = macs[hwaddr]
36         elif network['ip'] in ips:
37             orig_ifname = ips[network['ip']]
38         else:
39             orig_ifname = None
40
41         if orig_ifname:
42                 logger.verbose('net:InitInterfaces orig_ifname = %s' % orig_ifname)
43         
44         inter = {}
45         inter['ONBOOT']='yes'
46         inter['USERCTL']='no'
47         if network['mac']:
48             inter['HWADDR'] = network['mac']
49
50         if network['method'] == "static":
51             inter['BOOTPROTO'] = "static"
52             inter['IPADDR'] = network['ip']
53             inter['NETMASK'] = network['netmask']
54
55         elif network['method'] == "dhcp":
56             inter['BOOTPROTO'] = "dhcp"
57             if network['hostname']:
58                 inter['DHCP_HOSTNAME'] = network['hostname']
59             else:
60                 inter['DHCP_HOSTNAME'] = hostname 
61             if not network['is_primary']:
62                 inter['DHCLIENTARGS'] = "-R subnet-mask"
63
64         if len(network['interface_tag_ids']) > 0:
65             try:
66                 settings = plc.GetInterfaceTags({'interface_tag_id':
67                                                  network['interface_tag_ids']})
68             except:
69                 logger.log("net:InitInterfaces FATAL: failed call GetInterfaceTags({'interface_tag_id':{%s})"% \
70                            network['interface_tag_ids'])
71                 failedToGetSettings = True
72                 continue # on to the next network
73
74             for setting in settings:
75                 # to explicitly set interface name
76                 settingname = setting['name'].upper()
77                 if settingname in ('IFNAME','ALIAS','CFGOPTIONS','DRIVER'):
78                     inter[settingname]=setting['value']
79                 else:
80                     logger.log("net:InitInterfaces WARNING: ignored setting named %s"%setting['name'])
81
82         # support aliases to interfaces either by name or HWADDR
83         if 'ALIAS' in inter:
84             if 'HWADDR' in inter:
85                 hwaddr = inter['HWADDR'].lower()
86                 del inter['HWADDR']
87                 if hwaddr in macs:
88                     hwifname = macs[hwaddr]
89                     if ('IFNAME' in inter) and inter['IFNAME'] <> hwifname:
90                         logger.log("net:InitInterfaces WARNING: alias ifname (%s) and hwaddr ifname (%s) do not match"%\
91                                        (inter['IFNAME'],hwifname))
92                         inter['IFNAME'] = hwifname
93                 else:
94                     logger.log('net:InitInterfaces WARNING: mac addr %s for alias not found' %(hwaddr,alias))
95
96             if 'IFNAME' in inter:
97                 # stupid RH /etc/sysconfig/network-scripts/ifup-aliases:new_interface()
98                 # checks if the "$DEVNUM" only consists of '^[0-9A-Za-z_]*$'. Need to make
99                 # our aliases compliant.
100                 parts = inter['ALIAS'].split('_')
101                 isValid=True
102                 for part in parts:
103                     isValid=isValid and part.isalnum()
104
105                 if isValid:
106                     interfaces["%s:%s" % (inter['IFNAME'],inter['ALIAS'])] = inter 
107                 else:
108                     logger.log("net:InitInterfaces WARNING: interface alias (%s) not a valid string for RH ifup-aliases"% inter['ALIAS'])
109             else:
110                 logger.log("net:InitInterfaces WARNING: interface alias (%s) not matched to an interface"% inter['ALIAS'])
111             interface -= 1
112         else:
113             if ('IFNAME' not in inter) and not orig_ifname:
114                 ifname="eth%d" % (interface-1)
115                 # should check if $ifname is an eth already defines
116                 if os.path.exists("%s/ifcfg-%s"%(sysconfig,ifname)):
117                     logger.log("net:InitInterfaces WARNING: possibly blowing away %s configuration"%ifname)
118             else:
119                 if ('IFNAME' not in inter) and orig_ifname:
120                     ifname = orig_ifname
121                 else:
122                     ifname = inter['IFNAME']
123                 interface -= 1
124             interfaces[ifname] = inter
125                 
126     m = modprobe.Modprobe()
127     m.input("%s/etc/modprobe.conf" % root)
128     for (dev, inter) in interfaces.iteritems():
129         # get the driver string "moduleName option1=a option2=b"
130         driver=inter.get('DRIVER','')
131         if driver <> '':
132             driver=driver.split()
133             kernelmodule=driver[0]
134             m.aliasset(dev,kernelmodule)
135             options=" ".join(driver[1:])
136             if options <> '':
137                 m.optionsset(dev,options)
138     m.output("%s/etc/modprobe.conf" % root)
139
140     # clean up after any ifcfg-$dev script that's no longer listed as
141     # part of the NodeNetworks associated with this node
142
143     # list all network-scripts
144     files = os.listdir(sysconfig)
145
146     # filter out the ifcfg-* files
147     ifcfgs=[]
148     for f in files:
149         if f.find("ifcfg-") == 0:
150             ifcfgs.append(f)
151
152     # remove loopback (lo) from ifcfgs list
153     lo = "ifcfg-lo"
154     if lo in ifcfgs: ifcfgs.remove(lo)
155
156     # remove known devices from icfgs list
157     for (dev, inter) in interfaces.iteritems():
158         ifcfg = 'ifcfg-'+dev
159         if ifcfg in ifcfgs: ifcfgs.remove(ifcfg)
160
161     # delete the remaining ifcfgs from 
162     deletedSomething = False
163
164     if not failedToGetSettings:
165         for ifcfg in ifcfgs:
166             dev = ifcfg[len('ifcfg-'):]
167             path = "%s/ifcfg-%s" % (sysconfig,dev)
168             logger.verbose("net:InitInterfaces removing %s %s"%(dev,path))
169             os.system("/sbin/ifdown %s" % dev)
170             deletedSomething=True
171             os.unlink(path)
172
173     # wait a bit for the one or more ifdowns to have taken effect
174     if deletedSomething:
175         time.sleep(2)
176
177     # Process ifcg-$dev changes / additions
178     newdevs = []
179     for (dev, inter) in interfaces.iteritems():
180         tmpnam = os.tmpnam()
181         f = file(tmpnam, "w")
182         f.write("# Autogenerated by NodeManager/net.py.... do not edit!\n")
183         if 'DRIVER' in inter:
184             f.write("# using %s driver for device %s\n" % (inter['DRIVER'],dev))
185         f.write('DEVICE="%s"\n' % dev)
186         
187         # print the configuration values
188         for (key, val) in inter.iteritems():
189             if key not in ('IFNAME','ALIAS','CFGOPTIONS','DRIVER'):
190                 f.write('%s="%s"\n' % (key, val))
191
192         # print the configuration specific option values (if any)
193         if 'CFGOPTIONS' in inter:
194             cfgoptions = inter['CFGOPTIONS']
195             f.write('#CFGOPTIONS are %s\n' % cfgoptions)
196             for cfgoption in cfgoptions.split():
197                 key,val = cfgoption.split('=')
198                 key=key.strip()
199                 key=key.upper()
200                 val=val.strip()
201                 f.write('%s="%s"\n' % (key,val))
202         f.close()
203
204         # compare whether two files are the same
205         def comparefiles(a,b):
206             try:
207                 logger.verbose("net:InitInterfaces comparing %s with %s" % (a,b))
208                 if not os.path.exists(a): return False
209                 fb = open(a)
210                 buf_a = fb.read()
211                 fb.close()
212
213                 if not os.path.exists(b): return False
214                 fb = open(b)
215                 buf_b = fb.read()
216                 fb.close()
217
218                 return buf_a == buf_b
219             except IOError, e:
220                 return False
221
222         path = "%s/ifcfg-%s" % (sysconfig,dev)
223         if not os.path.exists(path):
224             logger.verbose('net:InitInterfaces adding configuration for %s' % dev)
225             # add ifcfg-$dev configuration file
226             os.rename(tmpnam,path)
227             os.chmod(path,0644)
228             newdevs.append(dev)
229             
230         elif not comparefiles(tmpnam,path):
231             logger.verbose('net:InitInterfaces Configuration change for %s' % dev)
232             logger.verbose('net:InitInterfaces ifdown %s' % dev)
233             # invoke ifdown for the old configuration
234             os.system("/sbin/ifdown %s" % dev)
235             # wait a few secs for ifdown to complete
236             time.sleep(2)
237
238             logger.log('replacing configuration for %s' % dev)
239             # replace ifcfg-$dev configuration file
240             os.rename(tmpnam,path)
241             os.chmod(path,0644)
242             newdevs.append(dev)
243         else:
244             # tmpnam & path are identical
245             os.unlink(tmpnam)
246
247     for dev in newdevs:
248         cfgvariables = {}
249         fb = file("%s/ifcfg-%s"%(sysconfig,dev),"r")
250         for line in fb.readlines():
251             parts = line.split()
252             if parts[0][0]=="#":continue
253             if parts[0].find('='):
254                 name,value = parts[0].split('=')
255                 # clean up name & value
256                 name = name.strip()
257                 value = value.strip()
258                 value = value.strip("'")
259                 value = value.strip('"')
260                 cfgvariables[name]=value
261         fb.close()
262
263         def getvar(name):
264             if name in cfgvariables:
265                 value=cfgvariables[name]
266                 value = value.lower()
267                 return value
268             return ''
269
270         # skip over device configs with ONBOOT=no
271         if getvar("ONBOOT") == 'no': continue
272
273         # don't bring up slave devices, the network scripts will
274         # handle those correctly
275         if getvar("SLAVE") == 'yes': continue
276
277         logger.verbose('net:InitInterfaces bringing up %s' % dev)
278         os.system("/sbin/ifup %s" % dev)
279