-#!/usr/bin/python2
-# $Id$
+#!/usr/bin/python
#
# Copyright (c) 2003 Intel Corporation
# All rights reserved.
import BootServerRequest
import ModelOptions
import BootAPI
-
-def Run( vars, log ):
+import plnet
+
+class BootAPIWrap:
+ def __init__(self, vars):
+ self.vars = vars
+ def call(self, func, *args):
+ return BootAPI.call_api_function(self.vars, func, args)
+ def __getattr__(self, func):
+ return lambda *args: self.call(func, *args)
+
+class logger:
+ def __init__(self, log):
+ self._log = log
+ def log(self, msg, level=3):
+ self._log.write(msg + "\n")
+ def verbose(self, msg):
+ self.log(msg, 0)
+
+def Run(vars, log):
"""
Write out the network configuration for this machine:
/etc/hosts
SYSIMG_PATH the path where the system image will be mounted
(always starts with TEMP_PATH)
INTERFACES All the interfaces associated with this node
- INTERFACE_SETTINGS dictionary
+ INTERFACE_SETTINGS dictionary
Sets the following variables:
None
"""
- log.write( "\n\nStep: Install: Writing Network Configuration files.\n" )
+ log.write("\n\nStep: Install: Writing Network Configuration files.\n")
try:
- SYSIMG_PATH= vars["SYSIMG_PATH"]
+ SYSIMG_PATH = vars["SYSIMG_PATH"]
if SYSIMG_PATH == "":
- raise ValueError, "SYSIMG_PATH"
-
- except KeyError, var:
- raise BootManagerException, "Missing variable in vars: %s\n" % var
- except ValueError, var:
- raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
+ raise ValueError("SYSIMG_PATH")
+ except KeyError as var:
+ raise BootManagerException("Missing variable in vars: {}\n".format(var))
+ except ValueError as var:
+ raise BootManagerException("Variable in vars, shouldn't be: {}\n".format(var))
- try:
- interface_settings= vars['INTERFACE_SETTINGS']
- except KeyError, e:
- raise BootManagerException, "No interface settings found in vars."
try:
- hostname= interface_settings['hostname']
- domainname= interface_settings['domainname']
- method= interface_settings['method']
- ip= interface_settings['ip']
- gateway= interface_settings['gateway']
- network= interface_settings['network']
- netmask= interface_settings['netmask']
- dns1= interface_settings['dns1']
- mac= interface_settings['mac']
- except KeyError, e:
- raise BootManagerException, "Missing value %s in interface settings." % str(e)
+ INTERFACE_SETTINGS = vars['INTERFACE_SETTINGS']
+ except KeyError as e:
+ raise BootManagerException("No interface settings found in vars.")
try:
- dns2= ''
- dns2= interface_settings['dns2']
- except KeyError, e:
- pass
-
+ hostname = INTERFACE_SETTINGS['hostname']
+ domainname = INTERFACE_SETTINGS['domainname']
+ method = INTERFACE_SETTINGS['method']
+ ip = INTERFACE_SETTINGS['ip']
+ gateway = INTERFACE_SETTINGS['gateway']
+ network = INTERFACE_SETTINGS['network']
+ netmask = INTERFACE_SETTINGS['netmask']
+ dns1 = INTERFACE_SETTINGS['dns1']
+ mac = INTERFACE_SETTINGS['mac']
+ except KeyError as e:
+ raise BootManagerException("Missing value {} in interface settings.".format(e))
+
+ # dns2 is not required to be set
+ dns2 = INTERFACE_SETTINGS.get('dns2','')
# Node Manager needs at least PLC_API_HOST and PLC_BOOT_HOST
log.write("Writing /etc/planetlab/plc_config\n")
- utils.makedirs("%s/etc/planetlab" % SYSIMG_PATH)
- plc_config = file("%s/etc/planetlab/plc_config" % SYSIMG_PATH, "w")
-
- api_url = vars['BOOT_API_SERVER']
- (scheme, netloc, path, params, query, fragment) = urlparse.urlparse(api_url)
- parts = netloc.split(':')
- host = parts[0]
- if len(parts) > 1:
- port = parts[1]
- else:
- port = '80'
- try:
- log.write("getting via https://%s/PlanetLabConf/get_plc_config.php " % host)
- bootserver = httplib.HTTPSConnection(host, int(port))
- bootserver.connect()
- bootserver.request("GET","https://%s/PlanetLabConf/get_plc_config.php" % host)
- plc_config.write("%s" % bootserver.getresponse().read())
- bootserver.close()
- log.write("Done\n")
- except :
- log.write(" .. Failed. Using old method. -- stack trace follows\n")
- traceback.print_exc(file=log.OutputFile)
- bs= BootServerRequest.BootServerRequest()
- if bs.BOOTSERVER_CERTS:
- print >> plc_config, "PLC_BOOT_HOST='%s'" % bs.BOOTSERVER_CERTS.keys()[0]
- print >> plc_config, "PLC_API_HOST='%s'" % host
- print >> plc_config, "PLC_API_PORT='%s'" % port
- print >> plc_config, "PLC_API_PATH='%s'" % path
-
- plc_config.close()
-
-
- log.write( "Writing /etc/hosts\n" )
- hosts_file= file("%s/etc/hosts" % SYSIMG_PATH, "w" )
- hosts_file.write( "127.0.0.1 localhost\n" )
- if method == "static":
- hosts_file.write( "%s %s.%s\n" % (ip, hostname, domainname) )
- hosts_file.close()
- hosts_file= None
-
-
- if method == "static":
- log.write( "Writing /etc/resolv.conf\n" )
- resolv_file= file("%s/etc/resolv.conf" % SYSIMG_PATH, "w" )
- if dns1 != "":
- resolv_file.write( "nameserver %s\n" % dns1 )
- if dns2 != "":
- resolv_file.write( "nameserver %s\n" % dns2 )
- resolv_file.write( "search %s\n" % domainname )
- resolv_file.close()
- resolv_file= None
-
- log.write( "Writing /etc/sysconfig/network\n" )
- network_file= file("%s/etc/sysconfig/network" % SYSIMG_PATH, "w" )
- network_file.write( "NETWORKING=yes\n" )
- network_file.write( "HOSTNAME=%s.%s\n" % (hostname, domainname) )
- if method == "static":
- network_file.write( "GATEWAY=%s\n" % gateway )
- network_file.close()
- network_file= None
-
- interfaces = {}
- interface_count = 1
- for interface in vars['INTERFACES']:
- if method == "static" or method == "dhcp":
- if interface['is_primary'] == 1:
- ifnum = 0
- else:
- ifnum = interface_count
- interface_count += 1
-
- inter = {}
- if interface['mac']:
- inter['HWADDR'] = interface['mac']
-
- if interface['method'] == "static":
- inter['BOOTPROTO'] = "static"
- inter['IPADDR'] = interface['ip']
- inter['NETMASK'] = interface['netmask']
-
- elif interface['method'] == "dhcp":
- inter['BOOTPROTO'] = "dhcp"
- if interface['hostname']:
- inter['DHCP_HOSTNAME'] = interface['hostname']
- else:
- inter['DHCP_HOSTNAME'] = hostname
- if not interface['is_primary']:
- inter['DHCLIENTARGS'] = "-R subnet-mask"
-
- alias = ""
- ifname=None
- if len(interface['interface_setting_ids']) > 0:
- settings = \
- BootAPI.call_api_function(vars, "GetInterfaceSettings",
- ({'interface_setting_id':
- interface['interface_setting_ids']},))
- for setting in settings:
- # to explicitly set interface name
- if setting['name'].upper() == "IFNAME":
- ifname=setting['value']
- elif setting['name'].upper() == "DRIVER":
- # xxx not sure how to do that yet - probably add a line in modprobe.conf
- pass
- elif setting['name'].upper() == "ALIAS":
- alias = ":" + setting['value']
-
- # a hack for testing before a new setting is hardcoded here
- # use the backdoor setting and put as a value 'var=value'
- elif setting['name'].upper() == "BACKDOOR":
- [var,value]=setting['value'].split('=',1)
- inter[var]=value
-
- elif setting['name'].lower() in \
- [ "mode", "essid", "nw", "freq", "channel", "sens", "rate",
- "key", "key1", "key2", "key3", "key4", "securitymode",
- "iwconfig", "iwpriv" ] :
- inter [setting['name'].upper()] = setting['value']
- inter ['TYPE']='Wireless'
- else:
- log.write("Warning - ignored setting named %s\n"%setting['name'])
-
- if alias and 'HWADDR' in inter:
- for (dev, i) in interfaces.iteritems():
- if i['HWADDR'] == inter['HWADDR']:
- break
- del inter['HWADDR']
- interfaces[dev + alias] =inter
- interface_count -= 1
- else:
- if not ifname:
- ifname="eth%d" % ifnum
- else:
- interface_count -= 1
- interfaces[ifname] =inter
-
- for (dev, inter) in interfaces.iteritems():
- path = "%s/etc/sysconfig/network-scripts/ifcfg-%s" % (
- SYSIMG_PATH, dev)
- f = file(path, "w")
- log.write("Writing %s\n" % path.replace(SYSIMG_PATH, ""))
-
- f.write("DEVICE=%s\n" % dev)
- f.write("ONBOOT=yes\n")
- f.write("USERCTL=no\n")
- for (key, val) in inter.iteritems():
- f.write('%s="%s"\n' % (key, val))
-
- f.close()
-
+ utils.makedirs("{}/etc/planetlab".format(SYSIMG_PATH))
+ with open("{}/etc/planetlab/plc_config".format(SYSIMG_PATH), "w") as plc_config:
+
+ api_url = vars['BOOT_API_SERVER']
+ (scheme, netloc, path, params, query, fragment) = urlparse.urlparse(api_url)
+ parts = netloc.split(':')
+ host = parts[0]
+ if len(parts) > 1:
+ port = parts[1]
+ else:
+ port = '80'
+ try:
+ log.write("getting via https://{}/PlanetLabConf/get_plc_config.php ".format(host))
+ bootserver = httplib.HTTPSConnection(host, int(port))
+ bootserver.connect()
+ bootserver.request("GET","https://{}/PlanetLabConf/get_plc_config.php".format(host))
+ plc_config.write("{}".format(bootserver.getresponse().read()))
+ bootserver.close()
+ log.write("Done\n")
+ except:
+ log.write(" .. Failed. Using old method. -- stack trace follows\n")
+ traceback.print_exc(file=log.OutputFile)
+ bs = BootServerRequest.BootServerRequest(vars)
+ if bs.BOOTSERVER_CERTS:
+ plc_config.write("PLC_BOOT_HOST='{}'\n".format(bs.BOOTSERVER_CERTS.keys()[0]))
+ plc_config.write("PLC_API_HOST='{}'\n".format(host))
+ plc_config.write("PLC_API_PORT='{}'\n".format(port))
+ plc_config.write("PLC_API_PATH='{}'\n".format(path))
+
+ log.write("Writing /etc/hosts\n")
+ with open("{}/etc/hosts".format(SYSIMG_PATH), "w") as hosts_file:
+ hosts_file.write("127.0.0.1 localhost\n")
+ if method == "static":
+ hosts_file.write("{} {}.{}\n".format(ip, hostname, domainname))
+
+ data = {'hostname': '{}.{}'.format(hostname, domainname),
+ 'interfaces': vars['INTERFACES']}
+ plnet.InitInterfaces(
+ logger(log), BootAPIWrap(vars),
+ data, SYSIMG_PATH,
+ files_only=True, program="BootManager")