source/steps/WriteNetworkConfig.py

   1 #!/usr/bin/python2
   2
   3 # Copyright (c) 2003 Intel Corporation
   4 # All rights reserved.
   5 #
   6 # Copyright (c) 2004-2006 The Trustees of Princeton University
   7 # All rights reserved.
   8 # expected /proc/partitions format
   9
  10 import os, string
  11
  12 from Exceptions import *
  13 import utils
  14 import BootServerRequest
  15 import ModelOptions
  16 import urlparse
  17 import BootAPI
  18
  19 def Run( vars, log ):
  20     """
  21     Write out the network configuration for this machine:
  22     /etc/hosts
  23     /etc/sysconfig/network-scripts/ifcfg-eth0
  24     /etc/resolv.conf (if applicable)
  25     /etc/sysconfig/network
  26
  27     The values to be used for the network settings are to be set in vars
  28     in the variable 'NETWORK_SETTINGS', which is a dictionary
  29     with keys:
  30
  31      Key               Used by this function
  32      -----------------------------------------------
  33      node_id
  34      node_key
  35      method            x
  36      ip                x
  37      mac               x (optional)
  38      gateway           x
  39      network           x
  40      broadcast         x
  41      netmask           x
  42      dns1              x
  43      dns2              x (optional)
  44      hostname          x
  45      domainname        x
  46
  47     Expect the following variables from the store:
  48     SYSIMG_PATH             the path where the system image will be mounted
  49                             (always starts with TEMP_PATH)
  50     NETWORK_SETTINGS  A dictionary of the values from the network
  51                                 configuration file
  52     NODE_NETWORKS           All the network associated with this node
  53     Sets the following variables:
  54     None
  55     """
  56
  57     log.write( "\n\nStep: Install: Writing Network Configuration files.\n" )
  58
  59     try:
  60         SYSIMG_PATH= vars["SYSIMG_PATH"]
  61         if SYSIMG_PATH == "":
  62             raise ValueError, "SYSIMG_PATH"
  63
  64     except KeyError, var:
  65         raise BootManagerException, "Missing variable in vars: %s\n" % var
  66     except ValueError, var:
  67         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
  68
  69
  70     try:
  71         network_settings= vars['NETWORK_SETTINGS']
  72     except KeyError, e:
  73         raise BootManagerException, "No network settings found in vars."
  74
  75     try:
  76         hostname= network_settings['hostname']
  77         domainname= network_settings['domainname']
  78         method= network_settings['method']
  79         ip= network_settings['ip']
  80         gateway= network_settings['gateway']
  81         network= network_settings['network']
  82         netmask= network_settings['netmask']
  83         dns1= network_settings['dns1']
  84         mac= network_settings['mac']
  85     except KeyError, e:
  86         raise BootManagerException, "Missing value %s in network settings." % str(e)
  87
  88     try:
  89         dns2= ''
  90         dns2= network_settings['dns2']
  91     except KeyError, e:
  92         pass
  93
  94
  95     # Node Manager needs at least PLC_API_HOST and PLC_BOOT_HOST
  96     log.write("Writing /etc/planetlab/plc_config\n")
  97     utils.makedirs("%s/etc/planetlab" % SYSIMG_PATH)
  98     plc_config = file("%s/etc/planetlab/plc_config" % SYSIMG_PATH, "w")
  99
 100     bs= BootServerRequest.BootServerRequest()
 101     if bs.BOOTSERVER_CERTS:
 102         print >> plc_config, "PLC_BOOT_HOST='%s'" % bs.BOOTSERVER_CERTS.keys()[0]
 103
 104     api_url = vars['BOOT_API_SERVER']
 105     (scheme, netloc, path, params, query, fragment) = urlparse.urlparse(api_url)
 106     parts = netloc.split(':')
 107     host = parts[0]
 108     if len(parts) > 1:
 109         port = parts[1]
 110     else:
 111         port = '80'
 112     print >> plc_config, "PLC_API_HOST='%s'" % host
 113     print >> plc_config, "PLC_API_PORT='%s'" % port
 114     print >> plc_config, "PLC_API_PATH='%s'" % path
 115
 116     plc_config.close()
 117
 118
 119     log.write( "Writing /etc/hosts\n" )
 120     hosts_file= file("%s/etc/hosts" % SYSIMG_PATH, "w" )
 121     hosts_file.write( "127.0.0.1       localhost\n" )
 122     if method == "static":
 123         hosts_file.write( "%s %s.%s\n" % (ip, hostname, domainname) )
 124     hosts_file.close()
 125     hosts_file= None
 126
 127
 128     if method == "static":
 129         log.write( "Writing /etc/resolv.conf\n" )
 130         resolv_file= file("%s/etc/resolv.conf" % SYSIMG_PATH, "w" )
 131         if dns1 != "":
 132             resolv_file.write( "nameserver %s\n" % dns1 )
 133         if dns2 != "":
 134             resolv_file.write( "nameserver %s\n" % dns2 )
 135         resolv_file.write( "search %s\n" % domainname )
 136         resolv_file.close()
 137         resolv_file= None
 138
 139     log.write( "Writing /etc/sysconfig/network\n" )
 140     network_file= file("%s/etc/sysconfig/network" % SYSIMG_PATH, "w" )
 141     network_file.write( "NETWORKING=yes\n" )
 142     network_file.write( "HOSTNAME=%s.%s\n" % (hostname, domainname) )
 143     if method == "static":
 144         network_file.write( "GATEWAY=%s\n" % gateway )
 145     network_file.close()
 146     network_file= None
 147
 148     interface = 1
 149     for network in vars['NODE_NETWORKS']:
 150         if method == "static" or method == "dhcp":
 151             if network['is_primary'] == 1:
 152                 ifnum = 0
 153             else:
 154                 ifnum = interface
 155                 interface += 1
 156
 157             path = "%s/etc/sysconfig/network-scripts/ifcfg-eth%d" % (
 158                    SYSIMG_PATH, ifnum)
 159             f = file(path, "w")
 160             log.write("Writing %s\n" % path.replace(SYSIMG_PATH, ""))
 161
 162             f.write("DEVICE=eth%d\n" % ifnum)
 163             f.write("HWADDR=%s\n" % network['mac'])
 164             f.write("ONBOOT=yes\n")
 165             f.write("USERCTL=no\n")
 166
 167             if network['method'] == "static":
 168                 f.write("BOOTPROTO=static\n")
 169                 f.write("IPADDR=%s\n" % network['ip'])
 170                 f.write("NETMASK=%s\n" % network['netmask'])
 171
 172             elif network['method'] == "dhcp":
 173                 f.write("BOOTPROTO=dhcp\n")
 174                 if network['hostname']:
 175                     f.write("DHCP_HOSTNAME=%s\n" % network['hostname'])
 176                 else:
 177                     f.write("DHCP_HOSTNAME=%s\n" % hostname)
 178                 if network['is_primary'] != 1:
 179                     f.write("DHCLIENTARGS='-R subnet-mask'\n")
 180
 181             if len(network['nodenetwork_setting_ids']) > 0:
 182                 settings = BootAPI.call_api_function(vars, "GetNodeNetworkSettings",
 183                     ({'nodenetwork_setting_id': network['nodenetwork_setting_ids']},))
 184                 for setting in settings:
 185                     if setting['category'].upper() != "WLAN":
 186                         continue
 187                     if setting['name'].upper() == "SSID":
 188                         f.write("ESSID=%s\n" % setting['value'])
 189                     elif setting['name'].upper() == "IWCONFIG":
 190                         f.write("IWCONFIG=%s\n" % setting['value'])
 191                     elif setting['name'].upper() == "MODE":
 192                         f.write("MODE=%s\n" % setting['value'])
 193
 194             f.close()
 195