plnet.py

   1 #!/usr/bin/python /usr/bin/plcsh
   2 # $Id$
   3
   4 import os
   5 import socket
   6 import time
   7 import tempfile
   8 import errno
   9
  10 import sioc
  11 import modprobe
  12
  13 global version
  14 version = 4.3
  15
  16 def InitInterfaces(logger, plc, data, root="", files_only=False, program="NodeManager"):
  17     global version
  18
  19     sysconfig = "%s/etc/sysconfig/network-scripts" % root
  20     try:
  21         os.makedirs(sysconfig)
  22     except OSError, e:
  23         if e.errno != errno.EEXIST:
  24             raise e
  25
  26     # query running network interfaces
  27     devs = sioc.gifconf()
  28     ips = dict(zip(devs.values(), devs.keys()))
  29     macs = {}
  30     for dev in devs:
  31         macs[sioc.gifhwaddr(dev).lower()] = dev
  32
  33     # assume data['networks'] contains this node's NodeNetworks
  34     interfaces = {}
  35     interface = 1
  36     hostname = data.get('hostname',socket.gethostname())
  37     gateway = None
  38     networks = data['networks']
  39     failedToGetSettings = False
  40     for network in networks:
  41         logger.verbose('net:InitInterfaces interface %d: %s'%(interface,network))
  42         logger.verbose('net:InitInterfaces macs = %s' % macs)
  43         logger.verbose('net:InitInterfaces ips = %s' % ips)
  44         # Get interface name preferably from MAC address, falling back
  45         # on IP address.
  46         hwaddr=network['mac']
  47         if hwaddr <> None: hwaddr=hwaddr.lower()
  48         if hwaddr in macs:
  49             orig_ifname = macs[hwaddr]
  50         elif network['ip'] in ips:
  51             orig_ifname = ips[network['ip']]
  52         else:
  53             orig_ifname = None
  54
  55         if orig_ifname:
  56                 logger.verbose('net:InitInterfaces orig_ifname = %s' % orig_ifname)
  57
  58         inter = {}
  59         inter['ONBOOT']='yes'
  60         inter['USERCTL']='no'
  61         if network['mac']:
  62             inter['HWADDR'] = network['mac']
  63         if network['is_primary']:
  64             inter['PRIMARY']='yes'
  65
  66         if network['method'] == "static":
  67             inter['BOOTPROTO'] = "static"
  68             inter['IPADDR'] = network['ip']
  69             inter['NETMASK'] = network['netmask']
  70             inter['DNS1'] = network['dns1']
  71             inter['DNS2'] = network['dns2']
  72             if network['is_primary']:
  73                 gateway = network['gateway']
  74
  75         elif network['method'] == "dhcp":
  76             inter['BOOTPROTO'] = "dhcp"
  77             inter['PERSISTENT_DHCLIENT'] = "yes"
  78             if network['hostname']:
  79                 inter['DHCP_HOSTNAME'] = network['hostname']
  80             else:
  81                 inter['DHCP_HOSTNAME'] = hostname
  82             if not network['is_primary']:
  83                 inter['DHCLIENTARGS'] = "-R subnet-mask"
  84
  85         try:
  86             plc.GetInterfaceTags()
  87             version = 4.3
  88         except AttributeError:
  89             version = 4.2
  90
  91         if version == 4.3:
  92             interface_tag_ids = "interface_tag_ids"
  93             interface_tag_id = "interface_tag_id"
  94         else:
  95             interface_tag_ids = "nodenetwork_setting_ids"
  96             interface_tag_id = "nodenetwork_setting_id"
  97
  98         if len(network[interface_tag_ids]) > 0:
  99             try:
 100                 if version == 4.3:
 101                     settings = plc.GetInterfaceTags({interface_tag_id:network[interface_tag_ids]})
 102                 else:
 103                     settings = plc.GetNodeNetworkSettings({interface_tag_id:network[interface_tag_ids]})
 104             except:
 105                 logger.log("net:InitInterfaces FATAL: failed call GetInterfaceTags({'interface_tag_id':{%s})"% \
 106                            network['interface_tag_ids'])
 107                 failedToGetSettings = True
 108                 continue # on to the next network
 109
 110             for setting in settings:
 111                 # to explicitly set interface name
 112                 settingname = setting['name'].upper()
 113                 if settingname in ('IFNAME','ALIAS','CFGOPTIONS','DRIVER'):
 114                     inter[settingname]=setting['value']
 115                 else:
 116                     logger.log("net:InitInterfaces WARNING: ignored setting named %s"%setting['name'])
 117
 118         # support aliases to interfaces either by name or HWADDR
 119         if 'ALIAS' in inter:
 120             if 'HWADDR' in inter:
 121                 hwaddr = inter['HWADDR'].lower()
 122                 del inter['HWADDR']
 123                 if hwaddr in macs:
 124                     hwifname = macs[hwaddr]
 125                     if ('IFNAME' in inter) and inter['IFNAME'] <> hwifname:
 126                         logger.log("net:InitInterfaces WARNING: alias ifname (%s) and hwaddr ifname (%s) do not match"%\
 127                                        (inter['IFNAME'],hwifname))
 128                         inter['IFNAME'] = hwifname
 129                 else:
 130                     logger.log('net:InitInterfaces WARNING: mac addr %s for alias not found' %(hwaddr,alias))
 131
 132             if 'IFNAME' in inter:
 133                 # stupid RH /etc/sysconfig/network-scripts/ifup-aliases:new_interface()
 134                 # checks if the "$DEVNUM" only consists of '^[0-9A-Za-z_]*$'. Need to make
 135                 # our aliases compliant.
 136                 parts = inter['ALIAS'].split('_')
 137                 isValid=True
 138                 for part in parts:
 139                     isValid=isValid and part.isalnum()
 140
 141                 if isValid:
 142                     interfaces["%s:%s" % (inter['IFNAME'],inter['ALIAS'])] = inter
 143                 else:
 144                     logger.log("net:InitInterfaces WARNING: interface alias (%s) not a valid string for RH ifup-aliases"% inter['ALIAS'])
 145             else:
 146                 logger.log("net:InitInterfaces WARNING: interface alias (%s) not matched to an interface"% inter['ALIAS'])
 147             interface -= 1
 148         else:
 149             if ('IFNAME' not in inter) and not orig_ifname:
 150                 ifname="eth%d" % (interface-1)
 151                 # should check if $ifname is an eth already defines
 152                 if os.path.exists("%s/ifcfg-%s"%(sysconfig,ifname)):
 153                     logger.log("net:InitInterfaces WARNING: possibly blowing away %s configuration"%ifname)
 154             else:
 155                 if ('IFNAME' not in inter) and orig_ifname:
 156                     ifname = orig_ifname
 157                 else:
 158                     ifname = inter['IFNAME']
 159                 interface -= 1
 160             interfaces[ifname] = inter
 161
 162     m = modprobe.Modprobe()
 163     try:
 164         m.input("%s/etc/modprobe.conf" % root)
 165     except:
 166         pass
 167     for (dev, inter) in interfaces.iteritems():
 168         # get the driver string "moduleName option1=a option2=b"
 169         driver=inter.get('DRIVER','')
 170         if driver <> '':
 171             driver=driver.split()
 172             kernelmodule=driver[0]
 173             m.aliasset(dev,kernelmodule)
 174             options=" ".join(driver[1:])
 175             if options <> '':
 176                 m.optionsset(dev,options)
 177     m.output("%s/etc/modprobe.conf" % root, program)
 178
 179     # clean up after any ifcfg-$dev script that's no longer listed as
 180     # part of the NodeNetworks associated with this node
 181
 182     # list all network-scripts
 183     files = os.listdir(sysconfig)
 184
 185     # filter out the ifcfg-* files
 186     ifcfgs=[]
 187     for f in files:
 188         if f.find("ifcfg-") == 0:
 189             ifcfgs.append(f)
 190
 191     # remove loopback (lo) from ifcfgs list
 192     lo = "ifcfg-lo"
 193     if lo in ifcfgs: ifcfgs.remove(lo)
 194
 195     # remove known devices from icfgs list
 196     for (dev, inter) in interfaces.iteritems():
 197         ifcfg = 'ifcfg-'+dev
 198         if ifcfg in ifcfgs: ifcfgs.remove(ifcfg)
 199
 200     # delete the remaining ifcfgs from
 201     deletedSomething = False
 202
 203     if not failedToGetSettings:
 204         for ifcfg in ifcfgs:
 205             dev = ifcfg[len('ifcfg-'):]
 206             path = "%s/ifcfg-%s" % (sysconfig,dev)
 207             if not files_only:
 208                 logger.verbose("net:InitInterfaces removing %s %s"%(dev,path))
 209                 os.system("/sbin/ifdown %s" % dev)
 210             deletedSomething=True
 211             os.unlink(path)
 212
 213     # wait a bit for the one or more ifdowns to have taken effect
 214     if deletedSomething:
 215         time.sleep(2)
 216
 217     # Write network configuration file
 218     networkconf = file("%s/etc/sysconfig/network" % root, "w")
 219     networkconf.write("NETWORKING=yes\nHOSTNAME=%s\n" % hostname)
 220     if gateway is not None:
 221         networkconf.write("GATEWAY=%s\n" % gateway)
 222     networkconf.close()
 223
 224     # Process ifcfg-$dev changes / additions
 225     newdevs = []
 226     for (dev, inter) in interfaces.iteritems():
 227         (fd, tmpnam) = tempfile.mkstemp(dir=sysconfig)
 228         f = os.fdopen(fd, "w")
 229         f.write("# Autogenerated by pyplnet... do not edit!\n")
 230         if 'DRIVER' in inter:
 231             f.write("# using %s driver for device %s\n" % (inter['DRIVER'],dev))
 232         f.write('DEVICE=%s\n' % dev)
 233
 234         # print the configuration values
 235         for (key, val) in inter.iteritems():
 236             if key not in ('IFNAME','ALIAS','CFGOPTIONS','DRIVER'):
 237                 f.write('%s=%s\n' % (key, val))
 238
 239         # print the configuration specific option values (if any)
 240         if 'CFGOPTIONS' in inter:
 241             cfgoptions = inter['CFGOPTIONS']
 242             f.write('#CFGOPTIONS are %s\n' % cfgoptions)
 243             for cfgoption in cfgoptions.split():
 244                 key,val = cfgoption.split('=')
 245                 key=key.strip()
 246                 key=key.upper()
 247                 val=val.strip()
 248                 f.write('%s="%s"\n' % (key,val))
 249         f.close()
 250
 251         # compare whether two files are the same
 252         def comparefiles(a,b):
 253             try:
 254                 logger.verbose("net:InitInterfaces comparing %s with %s" % (a,b))
 255                 if not os.path.exists(a): return False
 256                 fb = open(a)
 257                 buf_a = fb.read()
 258                 fb.close()
 259
 260                 if not os.path.exists(b): return False
 261                 fb = open(b)
 262                 buf_b = fb.read()
 263                 fb.close()
 264
 265                 return buf_a == buf_b
 266             except IOError, e:
 267                 return False
 268
 269         path = "%s/ifcfg-%s" % (sysconfig,dev)
 270         if not os.path.exists(path):
 271             logger.verbose('net:InitInterfaces adding configuration for %s' % dev)
 272             # add ifcfg-$dev configuration file
 273             os.rename(tmpnam,path)
 274             os.chmod(path,0644)
 275             newdevs.append(dev)
 276
 277         elif not comparefiles(tmpnam,path):
 278             logger.verbose('net:InitInterfaces Configuration change for %s' % dev)
 279             if not files_only:
 280                 logger.verbose('net:InitInterfaces ifdown %s' % dev)
 281                 # invoke ifdown for the old configuration
 282                 os.system("/sbin/ifdown %s" % dev)
 283                 # wait a few secs for ifdown to complete
 284                 time.sleep(2)
 285
 286             logger.log('replacing configuration for %s' % dev)
 287             # replace ifcfg-$dev configuration file
 288             os.rename(tmpnam,path)
 289             os.chmod(path,0644)
 290             newdevs.append(dev)
 291         else:
 292             # tmpnam & path are identical
 293             os.unlink(tmpnam)
 294
 295     for dev in newdevs:
 296         cfgvariables = {}
 297         fb = file("%s/ifcfg-%s"%(sysconfig,dev),"r")
 298         for line in fb.readlines():
 299             parts = line.split()
 300             if parts[0][0]=="#":continue
 301             if parts[0].find('='):
 302                 name,value = parts[0].split('=')
 303                 # clean up name & value
 304                 name = name.strip()
 305                 value = value.strip()
 306                 value = value.strip("'")
 307                 value = value.strip('"')
 308                 cfgvariables[name]=value
 309         fb.close()
 310
 311         def getvar(name):
 312             if name in cfgvariables:
 313                 value=cfgvariables[name]
 314                 value = value.lower()
 315                 return value
 316             return ''
 317
 318         # skip over device configs with ONBOOT=no
 319         if getvar("ONBOOT") == 'no': continue
 320
 321         # don't bring up slave devices, the network scripts will
 322         # handle those correctly
 323         if getvar("SLAVE") == 'yes': continue
 324
 325         if not files_only:
 326             logger.verbose('net:InitInterfaces bringing up %s' % dev)
 327             os.system("/sbin/ifup %s" % dev)
 328
 329 if __name__ == "__main__":
 330     import optparse
 331     import sys
 332
 333     parser = optparse.OptionParser(usage="plnet [-v] [-f] [-p <program>] -r root node_id")
 334     parser.add_option("-v", "--verbose", action="store_true", dest="verbose")
 335     parser.add_option("-r", "--root", action="store", type="string",
 336                       dest="root", default=None)
 337     parser.add_option("-f", "--files-only", action="store_true",
 338                       dest="files_only")
 339     parser.add_option("-p", "--program", action="store", type="string",
 340                       dest="program", default="plnet")
 341     (options, args) = parser.parse_args()
 342     if len(args) != 1 or options.root is None:
 343         print >>sys.stderr, "Missing root or node_id"
 344         parser.print_help()
 345         sys.exit(1)
 346
 347     node = shell.GetNodes({'node_id': [int(args[0])]})
 348     try:
 349         networks = shell.GetInterfaces({'interface_id': node[0]['interface_ids']})
 350     except AttributeError:
 351         networks = shell.GetNodeNetworks({'nodenetwork_id':node[0]['nodenetwork_ids']})
 352         version = 4.2
 353
 354
 355     data = {'hostname': node[0]['hostname'], 'networks': networks}
 356     class logger:
 357         def __init__(self, verbose):
 358             self.verbosity = verbose
 359         def log(self, msg, loglevel=2):
 360             if self.verbosity:
 361                 print msg
 362         def verbose(self, msg):
 363             self.log(msg, 1)
 364     l = logger(options.verbose)
 365     InitInterfaces(l, shell, data, options.root, options.files_only)