X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=source%2Fsteps%2FWriteNetworkConfig.py;h=610bff8faf491e15ae92b78c9dbc7ffb3251fa59;hb=621647f8fad5b114a8bda1557ff1657c3efc7e22;hp=524622c5c56251adf1384000bb4a3a7cad757d1e;hpb=006d0f8434d4fdfda74b46897500cd29a5ccfdfc;p=bootmanager.git diff --git a/source/steps/WriteNetworkConfig.py b/source/steps/WriteNetworkConfig.py index 524622c..610bff8 100644 --- a/source/steps/WriteNetworkConfig.py +++ b/source/steps/WriteNetworkConfig.py @@ -1,5 +1,5 @@ -#!/usr/bin/python2 - +#!/usr/bin/python +# # Copyright (c) 2003 Intel Corporation # All rights reserved. # @@ -8,24 +8,44 @@ # expected /proc/partitions format import os, string +import traceback -from Exceptions import * import utils +import urlparse +import httplib + +from Exceptions import * import BootServerRequest import ModelOptions -import urlparse 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 - /etc/sysconfig/network-scripts/ifcfg-eth0 + /etc/sysconfig/network-scripts/ifcfg- /etc/resolv.conf (if applicable) /etc/sysconfig/network The values to be used for the network settings are to be set in vars - in the variable 'NETWORK_SETTINGS', which is a dictionary + in the variable 'INTERFACE_SETTINGS', which is a dictionary with keys: Key Used by this function @@ -46,60 +66,51 @@ def Run( vars, log ): Expect the following variables from the store: SYSIMG_PATH the path where the system image will be mounted - (always starts with TEMP_PATH) - NETWORK_SETTINGS A dictionary of the values from the network - configuration file - NODE_NETWORKS All the network associated with this node + (always starts with TEMP_PATH) + INTERFACES All the interfaces associated with this node + 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" + 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 + 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: - network_settings= vars['NETWORK_SETTINGS'] - except KeyError, e: - raise BootManagerException, "No network settings found in vars." + INTERFACE_SETTINGS = vars['INTERFACE_SETTINGS'] + except KeyError as e: + raise BootManagerException("No interface settings found in vars.") try: - hostname= network_settings['hostname'] - domainname= network_settings['domainname'] - method= network_settings['method'] - ip= network_settings['ip'] - gateway= network_settings['gateway'] - network= network_settings['network'] - netmask= network_settings['netmask'] - dns1= network_settings['dns1'] - mac= network_settings['mac'] - except KeyError, e: - raise BootManagerException, "Missing value %s in network settings." % str(e) - - try: - dns2= '' - dns2= network_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") - - bs= BootServerRequest.BootServerRequest() - if bs.BOOTSERVER_CERTS: - print >> plc_config, "PLC_BOOT_HOST='%s'" % bs.BOOTSERVER_CERTS.keys()[0] + utils.makedirs("{}/etc/planetlab".format(SYSIMG_PATH)) + plc_config = file("{}/etc/planetlab/plc_config".format(SYSIMG_PATH), "w") api_url = vars['BOOT_API_SERVER'] (scheme, netloc, path, params, query, fragment) = urlparse.urlparse(api_url) @@ -109,125 +120,37 @@ def Run( vars, log ): port = parts[1] else: port = '80' - print >> plc_config, "PLC_API_HOST='%s'" % host - print >> plc_config, "PLC_API_PORT='%s'" % port - print >> plc_config, "PLC_API_PATH='%s'" % path + 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: + print >> plc_config, "PLC_BOOT_HOST='{}'".format(bs.BOOTSERVER_CERTS.keys()[0]) + print >> plc_config, "PLC_API_HOST='{}'".format(host) + print >> plc_config, "PLC_API_PORT='{}'".format(port) + print >> plc_config, "PLC_API_PATH='{}'".format(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" ) + log.write("Writing /etc/hosts\n") + hosts_file = file("{}/etc/hosts".format(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.write("{} {}.{}\n".format(ip, hostname, domainname)) hosts_file.close() - hosts_file= None + 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 = 1 - for network in vars['NODE_NETWORKS']: - if method == "static" or method == "dhcp": - if network['is_primary'] == 1: - ifnum = 0 - else: - ifnum = interface - interface += 1 - - int = {} - if network['mac']: - int['HWADDR'] = network['mac'] - - if network['method'] == "static": - int['BOOTPROTO'] = "static" - int['IPADDR'] = network['ip'] - int['NETMASK'] = network['netmask'] - - elif network['method'] == "dhcp": - int['BOOTPROTO'] = "dhcp" - if network['hostname']: - int['DHCP_HOSTNAME'] = network['hostname'] - else: - int['DHCP_HOSTNAME'] = hostname - if not network['is_primary']: - int['DHCLIENTARGS'] = "-R subnet-mask" - - alias = "" - ifname=None - if len(network['nodenetwork_setting_ids']) > 0: - settings = BootAPI.call_api_function(vars, "GetNodeNetworkSettings", - ({'nodenetwork_setting_id': network['nodenetwork_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) - int[var]=value - - elif setting['name'].upper() in \ - [ "mode", "essid", "nw", "freq", "channel", "sens", "rate", - "key", "key1", "key2", "key3", "key4", "securitymode", - "iwconfig", "iwpriv" ] : - int [setting['name'].upper()] = setting['value'] - int ['TYPE']=Wireless - else: - log.write("Warning - ignored setting named %s"%setting['name']) - - if alias and 'HWADDR' in int: - for (dev, i) in interfaces.iteritems(): - if i['HWADDR'] == int['HWADDR']: - break - del int['HWADDR'] - interfaces[dev + alias] = int - interface -= 1 - else: - if not ifname: - ifname="eth%d" % ifnum - else: - interface -= 1 - interfaces[ifname] = int - - for (dev, int) 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 int.iteritems(): - f.write('%s="%s"\n' % (key, val)) - - f.close() + data = {'hostname': '{}.{}'.format(hostname, domainname), + 'networks': vars['INTERFACES']} + plnet.InitInterfaces(logger(log), BootAPIWrap(vars), data, SYSIMG_PATH, + True, "BootManager")