source/steps/InstallWriteConfig.py

   1 # Copyright (c) 2003 Intel Corporation
   2 # All rights reserved.
   3
   4 # Redistribution and use in source and binary forms, with or without
   5 # modification, are permitted provided that the following conditions are
   6 # met:
   7
   8 #     * Redistributions of source code must retain the above copyright
   9 #       notice, this list of conditions and the following disclaimer.
  10
  11 #     * Redistributions in binary form must reproduce the above
  12 #       copyright notice, this list of conditions and the following
  13 #       disclaimer in the documentation and/or other materials provided
  14 #       with the distribution.
  15
  16 #     * Neither the name of the Intel Corporation nor the names of its
  17 #       contributors may be used to endorse or promote products derived
  18 #       from this software without specific prior written permission.
  19
  20 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE INTEL OR
  24 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  25 # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  26 # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  27 # PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  28 # LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  29 # NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  30 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31
  32 # EXPORT LAWS: THIS LICENSE ADDS NO RESTRICTIONS TO THE EXPORT LAWS OF
  33 # YOUR JURISDICTION. It is licensee's responsibility to comply with any
  34 # export regulations applicable in licensee's jurisdiction. Under
  35 # CURRENT (May 2000) U.S. export regulations this software is eligible
  36 # for export from the U.S. and can be downloaded by or otherwise
  37 # exported or reexported worldwide EXCEPT to U.S. embargoed destinations
  38 # which include Cuba, Iraq, Libya, North Korea, Iran, Syria, Sudan,
  39 # Afghanistan and any other country to which the U.S. has embargoed
  40 # goods and services.
  41
  42
  43 import os, string
  44
  45 from Exceptions import *
  46 import utils
  47 from systeminfo import systeminfo
  48 import BootAPI
  49
  50
  51 def Run( vars, log ):
  52
  53     """
  54     Writes out the following configuration files for the node:
  55     /etc/fstab
  56     /etc/hosts
  57     /etc/sysconfig/network-scripts/ifcfg-eth0
  58     /etc/resolv.conf (if applicable)
  59     /etc/sysconfig/network
  60     /etc/modprobe.conf
  61     /etc/ssh/ssh_host_key
  62     /etc/ssh/ssh_host_rsa_key
  63     /etc/ssh/ssh_host_dsa_key
  64
  65     Expect the following variables from the store:
  66     VERSION                 the version of the install
  67     SYSIMG_PATH             the path where the system image will be mounted
  68                             (always starts with TEMP_PATH)
  69     PARTITIONS              dictionary of generic part. types (root/swap)
  70                             and their associated devices.
  71     PLCONF_DIR              The directory to store the configuration file in
  72     NETWORK_SETTINGS  A dictionary of the values from the network
  73                                 configuration file
  74     BOOT_CD_VERSION          A tuple of the current bootcd version
  75
  76     Sets the following variables:
  77     None
  78
  79     """
  80
  81     log.write( "\n\nStep: Install: Writing configuration files.\n" )
  82
  83     # make sure we have the variables we need
  84     try:
  85         VERSION= vars["VERSION"]
  86         if VERSION == "":
  87             raise ValueError, "VERSION"
  88
  89         SYSIMG_PATH= vars["SYSIMG_PATH"]
  90         if SYSIMG_PATH == "":
  91             raise ValueError, "SYSIMG_PATH"
  92
  93         PARTITIONS= vars["PARTITIONS"]
  94         if PARTITIONS == None:
  95             raise ValueError, "PARTITIONS"
  96
  97         PLCONF_DIR= vars["PLCONF_DIR"]
  98         if PLCONF_DIR == "":
  99             raise ValueError, "PLCONF_DIR"
 100
 101         NETWORK_SETTINGS= vars["NETWORK_SETTINGS"]
 102         if NETWORK_SETTINGS == "":
 103             raise ValueError, "NETWORK_SETTINGS"
 104
 105         BOOT_CD_VERSION= vars["BOOT_CD_VERSION"]
 106         if BOOT_CD_VERSION == "":
 107             raise ValueError, "BOOT_CD_VERSION"
 108
 109     except KeyError, var:
 110         raise BootManagerException, "Missing variable in vars: %s\n" % var
 111     except ValueError, var:
 112         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
 113
 114     try:
 115         # we need to keys in PARTITIONS, root and swap, make sure
 116         # they exist
 117         val= PARTITIONS["root"]
 118         val= PARTITIONS["swap"]
 119         val= PARTITIONS["vservers"]
 120         val= PARTITIONS["mapper-root"]
 121         val= PARTITIONS["mapper-swap"]
 122         val= PARTITIONS["mapper-vservers"]
 123     except KeyError, part:
 124         log.write( "Missing partition in PARTITIONS: %s\n" % part )
 125         return 0
 126
 127
 128     log.write( "Setting local time to UTC\n" )
 129     utils.sysexec( "chroot %s ln -sf /usr/share/zoneinfo/UTC /etc/localtime" % \
 130                    SYSIMG_PATH, log )
 131
 132
 133     log.write( "Enabling ntp at boot\n" )
 134     utils.sysexec( "chroot %s chkconfig ntpd on" % SYSIMG_PATH, log )
 135
 136     log.write( "Creating system directory %s\n" % PLCONF_DIR )
 137     if not utils.makedirs( "%s/%s" % (SYSIMG_PATH,PLCONF_DIR) ):
 138         log.write( "Unable to create directory\n" )
 139         return 0
 140
 141
 142     log.write( "Writing network configuration\n" )
 143     write_network_configuration( vars, log )
 144
 145     # write out the modprobe.conf file for the system. make sure
 146     # the order of the ethernet devices are listed in the same order
 147     # as the boot cd loaded the modules. this is found in /tmp/loadedmodules
 148     # ultimately, the order will only match the boot cd order if
 149     # the kernel modules have the same name - which should be true for the later
 150     # version boot cds because they use the same kernel version.
 151     # older boot cds use a 2.4.19 kernel, and its possible some of the network
 152     # module names have changed, in which case the system might not boot
 153     # if the network modules are activated in a different order that the
 154     # boot cd.
 155     log.write( "Writing /etc/modprobe.conf\n" )
 156     write_modprobeconf_file( vars, log )
 157
 158     # dump the modprobe.conf file to the log (not to screen)
 159     log.write( "Contents of new modprobe.conf file:\n" )
 160     modulesconf_file= file("%s/etc/modprobe.conf" % SYSIMG_PATH, "r" )
 161     contents= modulesconf_file.read()
 162     log.write( contents + "\n" )
 163     modulesconf_file.close()
 164     modulesconf_file= None
 165     log.write( "End contents of new modprobe.conf file.\n" )
 166
 167     log.write( "Writing system /etc/fstab\n" )
 168     fstab= file( "%s/etc/fstab" % SYSIMG_PATH, "w" )
 169     fstab.write( "%s           none        swap      sw        0 0\n" % \
 170                  PARTITIONS["mapper-swap"] )
 171     fstab.write( "%s           /           ext3      defaults  0 0\n" % \
 172                  PARTITIONS["mapper-root"] )
 173     fstab.write( "%s           /vservers   ext3      tagxid,defaults  0 0\n" % \
 174                  PARTITIONS["mapper-vservers"] )
 175     fstab.write( "none         /proc       proc      defaults  0 0\n" )
 176     fstab.write( "none         /dev/shm    tmpfs     defaults  0 0\n" )
 177     fstab.write( "none         /dev/pts    devpts    defaults  0 0\n" )
 178     # no longer needed
 179     # fstab.write( "none         /rcfs       rcfs      defaults  0 0\n" )
 180     fstab.close()
 181
 182
 183     log.write( "Writing system /etc/issue\n" )
 184     issue= file( "%s/etc/issue" % SYSIMG_PATH, "w" )
 185     issue.write( "PlanetLab Node: \\n\n" )
 186     issue.write( "Kernel \\r on an \\m\n" )
 187     issue.write( "http://www.planet-lab.org\n\n" )
 188     issue.close()
 189
 190     log.write( "Setting up authentication (non-ssh)\n" )
 191     utils.sysexec( "chroot %s authconfig --nostart --kickstart --enablemd5 " \
 192                    "--enableshadow" % SYSIMG_PATH, log )
 193     utils.sysexec( "sed -e 's/^root\:\:/root\:*\:/g' " \
 194                    "%s/etc/shadow > %s/etc/shadow.new" % \
 195                    (SYSIMG_PATH,SYSIMG_PATH), log )
 196     utils.sysexec( "chroot %s mv " \
 197                    "/etc/shadow.new /etc/shadow" % SYSIMG_PATH, log )
 198     utils.sysexec( "chroot %s chmod 400 /etc/shadow" % SYSIMG_PATH, log )
 199
 200     # if we are setup with dhcp, copy the current /etc/resolv.conf into
 201     # the system image so we can run programs inside that need network access
 202     method= ""
 203     try:
 204         method= vars['NETWORK_SETTINGS']['method']
 205     except:
 206         pass
 207
 208     if method == "dhcp":
 209         utils.sysexec( "cp /etc/resolv.conf %s/etc/" % SYSIMG_PATH, log )
 210
 211     # the kernel rpm should have already done this, so don't fail the
 212     # install if it fails
 213     log.write( "Mounting /proc in system image\n" )
 214     utils.sysexec_noerr( "mount -t proc proc %s/proc" % SYSIMG_PATH, log )
 215
 216     # mkinitrd references both /etc/modprobe.conf and /etc/fstab
 217     # as well as /proc/lvm/global. The kernel RPM installation
 218     # likely created an improper initrd since these files did not
 219     # yet exist. Re-create the initrd here.
 220     log.write( "Making initrd\n" )
 221
 222     # trick mkinitrd in case the current environment does not have device mapper
 223     fake_root_lvm= 0
 224     if not os.path.exists( "%s/%s" % (SYSIMG_PATH,PARTITIONS["mapper-root"]) ):
 225         fake_root_lvm= 1
 226         utils.makedirs( "%s/dev/mapper" % SYSIMG_PATH )
 227         rootdev= file( "%s/%s" % (SYSIMG_PATH,PARTITIONS["mapper-root"]), "w" )
 228         rootdev.close()
 229
 230     initrd= os.readlink( "%s/boot/initrd-boot" % SYSIMG_PATH )
 231     kernel_version= initrd.replace("initrd-", "").replace(".img", "")
 232     utils.removefile( "%s/boot/%s" % (SYSIMG_PATH, initrd) )
 233     utils.sysexec( "chroot %s mkinitrd /boot/initrd-%s.img %s" % \
 234                    (SYSIMG_PATH, kernel_version, kernel_version), log )
 235
 236     if fake_root_lvm == 1:
 237         utils.removefile( "%s/%s" % (SYSIMG_PATH,PARTITIONS["mapper-root"]) )
 238
 239     log.write( "Writing node install version\n" )
 240     utils.makedirs( "%s/etc/planetlab" % SYSIMG_PATH )
 241     ver= file( "%s/etc/planetlab/install_version" % SYSIMG_PATH, "w" )
 242     ver.write( "%s\n" % VERSION )
 243     ver.close()
 244
 245     log.write( "Creating ssh host keys\n" )
 246     key_gen_prog= "/usr/bin/ssh-keygen"
 247
 248     log.write( "Generating SSH1 RSA host key:\n" )
 249     key_file= "/etc/ssh/ssh_host_key"
 250     utils.sysexec( "chroot %s %s -q -t rsa1 -f %s -C '' -N ''" %
 251                    (SYSIMG_PATH,key_gen_prog,key_file), log )
 252     utils.sysexec( "chmod 600 %s/%s" % (SYSIMG_PATH,key_file), log )
 253     utils.sysexec( "chmod 644 %s/%s.pub" % (SYSIMG_PATH,key_file), log )
 254
 255     log.write( "Generating SSH2 RSA host key:\n" )
 256     key_file= "/etc/ssh/ssh_host_rsa_key"
 257     utils.sysexec( "chroot %s %s -q -t rsa -f %s -C '' -N ''" %
 258                    (SYSIMG_PATH,key_gen_prog,key_file), log )
 259     utils.sysexec( "chmod 600 %s/%s" % (SYSIMG_PATH,key_file), log )
 260     utils.sysexec( "chmod 644 %s/%s.pub" % (SYSIMG_PATH,key_file), log )
 261
 262     log.write( "Generating SSH2 DSA host key:\n" )
 263     key_file= "/etc/ssh/ssh_host_dsa_key"
 264     utils.sysexec( "chroot %s %s -q -t dsa -f %s -C '' -N ''" %
 265                    (SYSIMG_PATH,key_gen_prog,key_file), log )
 266     utils.sysexec( "chmod 600 %s/%s" % (SYSIMG_PATH,key_file), log )
 267     utils.sysexec( "chmod 644 %s/%s.pub" % (SYSIMG_PATH,key_file), log )
 268
 269     return 1
 270
 271
 272
 273 def write_network_configuration( vars, log ):
 274     """
 275     Write out the network configuration for this machine:
 276     /etc/hosts
 277     /etc/sysconfig/network-scripts/ifcfg-eth0
 278     /etc/resolv.conf (if applicable)
 279     /etc/sysconfig/network
 280
 281     It is assumed the caller mounted the root partition and the vserver partition
 282     starting on SYSIMG_PATH - it is not checked here.
 283
 284     The values to be used for the network settings are to be set in vars
 285     in the variable 'NETWORK_SETTINGS', which is a dictionary
 286     with keys:
 287
 288      Key               Used by this function
 289      -----------------------------------------------
 290      node_id
 291      node_key
 292      method            x
 293      ip                x
 294      mac               x (optional)
 295      gateway           x
 296      network           x
 297      broadcast         x
 298      netmask           x
 299      dns1              x
 300      dns2              x (optional)
 301      hostname          x
 302      domainname        x
 303     """
 304
 305     try:
 306         SYSIMG_PATH= vars["SYSIMG_PATH"]
 307         if SYSIMG_PATH == "":
 308             raise ValueError, "SYSIMG_PATH"
 309
 310     except KeyError, var:
 311         raise BootManagerException, "Missing variable in vars: %s\n" % var
 312     except ValueError, var:
 313         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
 314
 315
 316     try:
 317         network_settings= vars['NETWORK_SETTINGS']
 318     except KeyError, e:
 319         raise BootManagerException, "No network settings found in vars."
 320
 321     try:
 322         hostname= network_settings['hostname']
 323         domainname= network_settings['domainname']
 324         method= network_settings['method']
 325         ip= network_settings['ip']
 326         gateway= network_settings['gateway']
 327         network= network_settings['network']
 328         netmask= network_settings['netmask']
 329         dns1= network_settings['dns1']
 330         mac= network_settings['mac']
 331     except KeyError, e:
 332         raise BootManagerException, "Missing value %s in network settings." % str(e)
 333
 334     try:
 335         dns2= ''
 336         dns2= network_settings['dns2']
 337     except KeyError, e:
 338         pass
 339
 340
 341     log.write( "Writing /etc/hosts\n" )
 342     hosts_file= file("%s/etc/hosts" % SYSIMG_PATH, "w" )
 343     hosts_file.write( "127.0.0.1       localhost\n" )
 344     if method == "static":
 345         hosts_file.write( "%s %s.%s\n" % (ip, hostname, domainname) )
 346     hosts_file.close()
 347     hosts_file= None
 348
 349
 350     log.write( "Writing /etc/sysconfig/network-scripts/ifcfg-eth0\n" )
 351     eth0_file= file("%s/etc/sysconfig/network-scripts/ifcfg-eth0" %
 352                     SYSIMG_PATH, "w" )
 353     eth0_file.write( "DEVICE=eth0\n" )
 354     if method == "static":
 355         eth0_file.write( "BOOTPROTO=static\n" )
 356         eth0_file.write( "IPADDR=%s\n" % ip )
 357         eth0_file.write( "NETMASK=%s\n" % netmask )
 358         eth0_file.write( "GATEWAY=%s\n" % gateway )
 359     else:
 360         eth0_file.write( "BOOTPROTO=dhcp\n" )
 361         eth0_file.write( "DHCP_HOSTNAME=%s\n" % hostname )
 362     if mac != "":
 363         eth0_file.write( "HWADDR=%s\n" % mac )
 364     eth0_file.write( "ONBOOT=yes\n" )
 365     eth0_file.write( "USERCTL=no\n" )
 366     eth0_file.close()
 367     eth0_file= None
 368
 369     if method == "static":
 370         log.write( "Writing /etc/resolv.conf\n" )
 371         resolv_file= file("%s/etc/resolv.conf" % SYSIMG_PATH, "w" )
 372         if dns1 != "":
 373             resolv_file.write( "nameserver %s\n" % dns1 )
 374         if dns2 != "":
 375             resolv_file.write( "nameserver %s\n" % dns2 )
 376         resolv_file.write( "search %s\n" % domainname )
 377         resolv_file.close()
 378         resolv_file= None
 379
 380     log.write( "Writing /etc/sysconfig/network\n" )
 381     network_file= file("%s/etc/sysconfig/network" % SYSIMG_PATH, "w" )
 382     network_file.write( "NETWORKING=yes\n" )
 383     network_file.write( "HOSTNAME=%s.%s\n" % (hostname, domainname) )
 384     if method == "static":
 385         network_file.write( "GATEWAY=%s\n" % gateway )
 386     network_file.close()
 387     network_file= None
 388
 389
 390
 391 def write_modprobeconf_file( vars, log, filename = "/etc/modprobe.conf"):
 392     """
 393     write out the system file /etc/modprobe.conf with the current
 394     set of modules.
 395
 396     returns a tuple of the number of network driver lines and storage
 397     driver lines written as (networkcount,storagecount)
 398     """
 399
 400     # make sure we have this class loaded
 401     from systeminfo import systeminfo
 402
 403     try:
 404         SYSIMG_PATH= vars["SYSIMG_PATH"]
 405         if SYSIMG_PATH == "":
 406             raise ValueError, "SYSIMG_PATH"
 407
 408     except KeyError, var:
 409         raise BootManagerException, "Missing variable in vars: %s\n" % var
 410     except ValueError, var:
 411         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
 412
 413
 414     # get the kernel version
 415     initrd= os.readlink( "%s/boot/initrd-boot" % SYSIMG_PATH )
 416     kernel_version= initrd.replace("initrd-", "").replace(".img", "")
 417
 418     sysinfo= systeminfo()
 419     sysmods= sysinfo.get_system_modules(SYSIMG_PATH, kernel_version)
 420     if sysmods is None:
 421         raise BootManagerException, "Unable to get list of system modules."
 422
 423     eth_count= 0
 424     scsi_count= 0
 425
 426     modulesconf_file= file("%s/%s" % (SYSIMG_PATH,filename), "w" )
 427
 428     for type in sysmods:
 429         if type == sysinfo.MODULE_CLASS_SCSI:
 430             for a_mod in sysmods[type]:
 431                 if scsi_count == 0:
 432                     modulesconf_file.write( "alias scsi_hostadapter %s\n" %
 433                                             a_mod )
 434                 else:
 435                     modulesconf_file.write( "alias scsi_hostadapter%d %s\n" %
 436                                             (scsi_count,a_mod) )
 437                 scsi_count= scsi_count + 1
 438
 439         elif type == sysinfo.MODULE_CLASS_NETWORK:
 440             for a_mod in sysmods[type]:
 441                 modulesconf_file.write( "alias eth%d %s\n" %
 442                                         (eth_count,a_mod) )
 443                 eth_count= eth_count + 1
 444
 445     modulesconf_file.close()
 446     modulesconf_file= None
 447
 448     return (eth_count,scsi_count)
 449