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 enough files to run rpm and yum from a chroot in
24 the system image directory
26 Expect the following variables from the store:
27 SYSIMG_PATH the path where the system image will be mounted
28 PARTITIONS dictionary of generic part. types (root/swap)
29 and their associated devices.
30 SUPPORT_FILE_DIR directory on the boot servers containing
31 scripts and support files
32 NODE_ID the id of this machine
34 Sets the following variables:
35 TEMP_BOOTCD_PATH where the boot cd is remounted in the temp
37 ROOT_MOUNTED set to 1 when the the base logical volumes
41 log.write( "\n\nStep: Install: bootstrapfs tarball.\n" )
43 # make sure we have the variables we need
45 SYSIMG_PATH= vars["SYSIMG_PATH"]
47 raise ValueError, "SYSIMG_PATH"
49 PARTITIONS= vars["PARTITIONS"]
50 if PARTITIONS == None:
51 raise ValueError, "PARTITIONS"
53 SUPPORT_FILE_DIR= vars["SUPPORT_FILE_DIR"]
54 if SUPPORT_FILE_DIR == None:
55 raise ValueError, "SUPPORT_FILE_DIR"
57 NODE_ID= vars["NODE_ID"]
59 raise ValueError, "NODE_ID"
62 raise BootManagerException, "Missing variable in vars: %s\n" % var
63 except ValueError, var:
64 raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
68 # make sure the required partitions exist
69 val= PARTITIONS["root"]
70 val= PARTITIONS["swap"]
71 val= PARTITIONS["vservers"]
72 except KeyError, part:
73 log.write( "Missing partition in PARTITIONS: %s\n" % part )
76 bs_request= BootServerRequest.BootServerRequest()
78 log.write( "turning on swap space\n" )
79 utils.sysexec( "swapon %s" % PARTITIONS["swap"], log )
81 # make sure the sysimg dir is present
82 utils.makedirs( SYSIMG_PATH )
84 log.write( "mounting root file system\n" )
85 utils.sysexec( "mount -t ext3 %s %s" % (PARTITIONS["root"],SYSIMG_PATH), log )
87 log.write( "mounting vserver partition in root file system\n" )
88 utils.makedirs( SYSIMG_PATH + "/vservers" )
89 utils.sysexec( "mount -t ext3 %s %s/vservers" % (PARTITIONS["vservers"],
92 vars['ROOT_MOUNTED']= 1
94 # which extensions are we part of ?
95 utils.breakpoint("Getting the extension tag on node")
98 extension_tag = BootAPI.call_api_function(vars, "GetNodeExtensions", (NODE_ID,) )
100 extensions = extension_tag.split()
103 log.write("WARNING : Failed to query tag 'extensions'")
104 log.write(traceback.format_exc())
107 log.write("installing only core software\n")
109 # check if the plain-bootstrapfs tag is set
110 download_suffix=".tar.bz2"
113 if BootAPI.call_api_function (vars, "GetNodePlainBootstrapfs", (NODE_ID,) ):
114 download_suffix=".tar"
117 log.write("WARNING : Failed to query tag 'plain-bootstrapfs'")
118 log.write(traceback.format_exc())
121 log.write("Using uncompressed bootstrapfs images\n")
123 # see also GetBootMedium in PLCAPI that does similar things
124 # figuring the default node family:
125 # (1) look at /etc/planetlab/nodefamily on the bootcd
126 # (2) if that fails, set to planetlab-i386
128 (pldistro,arch) = file("/etc/planetlab/nodefamily").read().strip().split("-")
130 # what arch should be used for this node
131 utils.breakpoint("Getting the arch tag on node")
135 arch = BootAPI.call_api_function(vars, "GetNodeArch", (NODE_ID,) )
137 log.write("WARNING : Failed to query tag 'arch'")
138 log.write(traceback.format_exc())
141 log.write ("Using arch=%s\n"%arch)
143 bootstrapfs_names = [ pldistro ] + extensions
145 # download and extract support tarball for this step, which has
146 # everything we need to successfully run
148 # we first try to find a tarball, if it is not found we use yum instead
150 # download and extract support tarball for this step,
151 for bootstrapfs_name in bootstrapfs_names:
152 tarball = "bootstrapfs-%s-%s%s"%(bootstrapfs_name,arch,download_suffix)
153 source_file= "%s/%s" % (SUPPORT_FILE_DIR,tarball)
154 dest_file= "%s/%s" % (SYSIMG_PATH, tarball)
156 # 30 is the connect timeout, 14400 is the max transfer time in
158 log.write( "downloading %s\n" % source_file )
159 result= bs_request.DownloadFile( source_file, None, None,
163 log.write( "extracting %s in %s\n" % (dest_file,SYSIMG_PATH) )
164 result= utils.sysexec( "tar -C %s -xpf %s %s" % (SYSIMG_PATH,dest_file,untar_option), log )
166 utils.removefile( dest_file )
168 # the main tarball is required
169 if bootstrapfs_name == pldistro:
170 raise BootManagerException, "Unable to download main tarball %s from server." % \
173 log.write("tarball for %s-%s not found, scheduling a yum attempt\n"%(bootstrapfs_name,arch))
174 yum_extensions.append(bootstrapfs_name)
176 # copy resolv.conf from the base system into our temp dir
177 # so DNS lookups work correctly while we are chrooted
178 log.write( "Copying resolv.conf to temp dir\n" )
179 utils.sysexec( "cp /etc/resolv.conf %s/etc/" % SYSIMG_PATH, log )
181 # Copy the boot server certificate(s) and GPG public key to
182 # /usr/boot in the temp dir.
183 log.write( "Copying boot server certificates and public key\n" )
185 if os.path.exists("/usr/boot"):
186 utils.makedirs(SYSIMG_PATH + "/usr")
187 shutil.copytree("/usr/boot", SYSIMG_PATH + "/usr/boot")
188 elif os.path.exists("/usr/bootme"):
189 utils.makedirs(SYSIMG_PATH + "/usr/boot")
190 boot_server = file("/usr/bootme/BOOTSERVER").readline().strip()
191 shutil.copy("/usr/bootme/cacert/" + boot_server + "/cacert.pem",
192 SYSIMG_PATH + "/usr/boot/cacert.pem")
193 file(SYSIMG_PATH + "/usr/boot/boot_server", "w").write(boot_server)
194 shutil.copy("/usr/bootme/pubring.gpg", SYSIMG_PATH + "/usr/boot/pubring.gpg")
196 # For backward compatibility
197 if os.path.exists("/usr/bootme"):
198 utils.makedirs(SYSIMG_PATH + "/mnt/cdrom")
199 shutil.copytree("/usr/bootme", SYSIMG_PATH + "/mnt/cdrom/bootme")
201 # Import the GPG key into the RPM database so that RPMS can be verified
202 utils.makedirs(SYSIMG_PATH + "/etc/pki/rpm-gpg")
203 utils.sysexec("gpg --homedir=/root --export --armor" \
204 " --no-default-keyring --keyring %s/usr/boot/pubring.gpg" \
205 " >%s/etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab" % (SYSIMG_PATH, SYSIMG_PATH))
206 utils.sysexec("chroot %s rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab" % \
209 # the yum config has changed entirely;
210 # in addition yum installs have more or less never worked - let's forget about this
211 # maybe NodeManager could profitably do the job instead
213 log.write('WARNING : yum installs for node extensions are not supported anymore')