3 # Copyright (c) 2003 Intel Corporation
6 # Copyright (c) 2004-2007 The Trustees of Princeton University
8 # expected /proc/partitions format
10 import os, sys, string
15 from Exceptions import *
17 import BootServerRequest
23 Download core + extensions bootstrapfs tarballs and install on the hard drive
25 Expect the following variables from the store:
26 SYSIMG_PATH the path where the system image will be mounted
27 PARTITIONS dictionary of generic part. types (root/swap)
28 and their associated devices.
29 NODE_ID the id of this machine
31 Sets the following variables:
32 TEMP_BOOTCD_PATH where the boot cd is remounted in the temp
34 ROOT_MOUNTED set to 1 when the the base logical volumes
38 log.write( "\n\nStep: Install: bootstrapfs tarball.\n" )
40 # make sure we have the variables we need
42 SYSIMG_PATH= vars["SYSIMG_PATH"]
44 raise ValueError, "SYSIMG_PATH"
46 PARTITIONS= vars["PARTITIONS"]
47 if PARTITIONS == None:
48 raise ValueError, "PARTITIONS"
50 NODE_ID= vars["NODE_ID"]
52 raise ValueError, "NODE_ID"
55 raise BootManagerException, "Missing variable in vars: %s\n" % var
56 except ValueError, var:
57 raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
61 # make sure the required partitions exist
62 val= PARTITIONS["root"]
63 val= PARTITIONS["swap"]
64 val= PARTITIONS["vservers"]
65 except KeyError, part:
66 log.write( "Missing partition in PARTITIONS: %s\n" % part )
69 bs_request= BootServerRequest.BootServerRequest(vars)
71 log.write( "turning on swap space\n" )
72 utils.sysexec( "swapon %s" % PARTITIONS["swap"], log )
74 # make sure the sysimg dir is present
75 utils.makedirs( SYSIMG_PATH )
77 log.write( "mounting root file system\n" )
78 utils.sysexec( "mount -t ext3 %s %s" % (PARTITIONS["root"],SYSIMG_PATH), log )
80 log.write( "mounting vserver partition in root file system\n" )
81 utils.makedirs( SYSIMG_PATH + "/vservers" )
82 utils.sysexec( "mount -t ext3 %s %s/vservers" % (PARTITIONS["vservers"],
85 vars['ROOT_MOUNTED']= 1
89 node_flavour = BootAPI.call_api_function(vars, "GetNodeFlavour", (NODE_ID,) )
90 nodefamily = node_flavour['nodefamily']
91 extensions = node_flavour['extensions']
92 plain = node_flavour['plain']
94 raise BootManagerException ("Could not call GetNodeFlavour - need PLCAPI-5.0")
96 # the 'plain' option is for tests mostly
98 download_suffix=".tar"
100 log.write("Using plain bootstrapfs images\n")
102 download_suffix=".tar.bz2"
103 uncompress_option="-j"
104 log.write("Using compressed bootstrapfs images\n")
106 log.write ("Using nodefamily=%s\n"%(nodefamily))
108 log.write("Installing only core software\n")
110 log.write("Requested extensions %r" % extensions)
112 bootstrapfs_names = [ nodefamily ] + extensions
114 for name in bootstrapfs_names:
115 tarball = "bootstrapfs-%s%s"%(name,download_suffix)
116 source_file= "/boot/%s" % (tarball)
117 dest_file= "%s/%s" % (SYSIMG_PATH, tarball)
119 source_hash_file= "/boot/%s.sha1sum" % (tarball)
120 dest_hash_file= "%s/%s.sha1sum" % (SYSIMG_PATH, tarball)
122 # 30 is the connect timeout, 14400 is the max transfer time in
124 log.write( "downloading %s\n" % source_file )
125 result = bs_request.DownloadFile( source_file, None, None,
130 # Download SHA1 checksum file
131 result = bs_request.DownloadFile( source_hash_file, None, None,
132 1, 1, dest_hash_file,
135 if not utils.check_file_hash(dest_file, dest_hash_file):
136 raise BootManagerException, "FATAL: SHA1 checksum does not match between %s and %s" % (source_file, source_hash_file)
138 log.write( "extracting %s in %s\n" % (dest_file,SYSIMG_PATH) )
139 result = utils.sysexec( "tar -C %s -xpf %s %s" % (SYSIMG_PATH,dest_file,uncompress_option), log )
141 utils.removefile( dest_file )
143 # the main tarball is required
144 if name == nodefamily:
145 raise BootManagerException, "FATAL: Unable to download main tarball %s from server." % \
147 # for extensions, just print a warning
149 log.write("WARNING: tarball for extension %s not found\n"%(name))
151 # copy resolv.conf from the base system into our temp dir
152 # so DNS lookups work correctly while we are chrooted
153 log.write( "Copying resolv.conf to temp dir\n" )
154 utils.sysexec( "cp /etc/resolv.conf %s/etc/" % SYSIMG_PATH, log )
156 # Copy the boot server certificate(s) and GPG public key to
157 # /usr/boot in the temp dir.
158 log.write( "Copying boot server certificates and public key\n" )
160 if os.path.exists("/usr/boot"):
161 utils.makedirs(SYSIMG_PATH + "/usr")
162 shutil.copytree("/usr/boot", SYSIMG_PATH + "/usr/boot")
163 elif os.path.exists("/usr/bootme"):
164 utils.makedirs(SYSIMG_PATH + "/usr/boot")
165 boot_server = file("/usr/bootme/BOOTSERVER").readline().strip()
166 shutil.copy("/usr/bootme/cacert/" + boot_server + "/cacert.pem",
167 SYSIMG_PATH + "/usr/boot/cacert.pem")
168 file(SYSIMG_PATH + "/usr/boot/boot_server", "w").write(boot_server)
169 shutil.copy("/usr/bootme/pubring.gpg", SYSIMG_PATH + "/usr/boot/pubring.gpg")
171 # For backward compatibility
172 if os.path.exists("/usr/bootme"):
173 utils.makedirs(SYSIMG_PATH + "/mnt/cdrom")
174 shutil.copytree("/usr/bootme", SYSIMG_PATH + "/mnt/cdrom/bootme")
176 # Import the GPG key into the RPM database so that RPMS can be verified
177 utils.makedirs(SYSIMG_PATH + "/etc/pki/rpm-gpg")
178 utils.sysexec("gpg --homedir=/root --export --armor" \
179 " --no-default-keyring --keyring %s/usr/boot/pubring.gpg" \
180 " >%s/etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab" % (SYSIMG_PATH, SYSIMG_PATH))
181 utils.sysexec_chroot(SYSIMG_PATH, "rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab")