4b8ecc8b4d6c83756c42a4eff2b94cc679473ba2
[bootmanager.git] / source / steps / InstallBootstrapRPM.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
44 import os, sys, string
45 import popen2
46
47 from Exceptions import *
48 import utils
49 import BootServerRequest
50
51
52 def Run( vars, log ):
53     """
54     Download enough files to run rpm and yum from a chroot in
55     the system image directory
56     
57     Expect the following variables from the store:
58     SYSIMG_PATH          the path where the system image will be mounted
59     PARTITIONS           dictionary of generic part. types (root/swap)
60                          and their associated devices.
61     SUPPORT_FILE_DIR     directory on the boot servers containing
62                          scripts and support files
63     NODE_ID              the id of this machine
64     
65     Sets the following variables:
66     TEMP_BOOTCD_PATH     where the boot cd is remounted in the temp
67                          path
68     ROOT_MOUNTED         set to 1 when the the base logical volumes
69                          are mounted.
70     """
71
72     log.write( "\n\nStep: Install: Bootstrapping RPM.\n" )
73
74     # make sure we have the variables we need
75     try:
76         SYSIMG_PATH= vars["SYSIMG_PATH"]
77         if SYSIMG_PATH == "":
78             raise ValueError, "SYSIMG_PATH"
79
80         PARTITIONS= vars["PARTITIONS"]
81         if PARTITIONS == None:
82             raise ValueError, "PARTITIONS"
83
84         SUPPORT_FILE_DIR= vars["SUPPORT_FILE_DIR"]
85         if SUPPORT_FILE_DIR == None:
86             raise ValueError, "SUPPORT_FILE_DIR"
87
88         NODE_ID= vars["NODE_ID"]
89         if NODE_ID == "":
90             raise ValueError, "NODE_ID"
91
92     except KeyError, var:
93         raise BootManagerException, "Missing variable in vars: %s\n" % var
94     except ValueError, var:
95         raise BootManagerException, "Variable in vars, shouldn't be: %s\n" % var
96
97
98     try:
99         # make sure the required partitions exist
100         val= PARTITIONS["root"]
101         val= PARTITIONS["swap"]
102         val= PARTITIONS["vservers"]
103     except KeyError, part:
104         log.write( "Missing partition in PARTITIONS: %s\n" % part )
105         return 0   
106
107     bs_request= BootServerRequest.BootServerRequest()
108     
109     log.write( "turning on swap space\n" )
110     utils.sysexec( "swapon %s" % PARTITIONS["swap"], log )
111
112     # make sure the sysimg dir is present
113     utils.makedirs( SYSIMG_PATH )
114
115     log.write( "mounting root file system\n" )
116     utils.sysexec( "mount -t ext3 %s %s" % (PARTITIONS["root"],SYSIMG_PATH), log )
117
118     log.write( "mounting vserver partition in root file system\n" )
119     utils.makedirs( SYSIMG_PATH + "/vservers" )
120     utils.sysexec( "mount -t ext3 %s %s/vservers" % (PARTITIONS["vservers"],
121                                                      SYSIMG_PATH), log )
122
123     vars['ROOT_MOUNTED']= 1
124     
125
126     # download and extract support tarball for
127     # this step, which has everything
128     # we need to successfully run
129     for step_support_file in [ "PlanetLab-Bootstrap.tar.bz2",
130                                "alpina-BootstrapRPM.tar.bz2" ]: 
131         source_file= "%s/%s" % (SUPPORT_FILE_DIR,step_support_file)
132         dest_file= "%s/%s" % (SYSIMG_PATH, step_support_file)
133
134         # 30 is the connect timeout, 7200 is the max transfer time
135         # in seconds (2 hours)
136         log.write( "downloading %s\n" % step_support_file )
137         result= bs_request.DownloadFile( source_file, None, None,
138                                          1, 1, dest_file,
139                                          30, 7200)
140         if result:
141             # New bootstrap tarball contains everything necessary to
142             # boot, no need to bootstrap further.
143             vars['SKIP_INSTALL_BASE']= (step_support_file == "PlanetLab-Bootstrap.tar.bz2")
144             break
145
146     if not result:
147         raise BootManagerException, "Unable to download %s from server." % \
148               source_file
149
150     log.write( "extracting %s in %s\n" % (dest_file,SYSIMG_PATH) )
151     result= utils.sysexec( "tar -C %s -xpjf %s" % (SYSIMG_PATH,dest_file), log )
152     utils.removefile( dest_file )
153
154     # get the yum configuration file for this node (yum.conf).
155     # this needs to come from the configuration file service,
156     # so, if its a beta node, it'll install the beta rpms from
157     # the beginning. The configuration file service will return
158     # the url for the file we need to request to get the actual
159     # conf file, so two requests need to be made.
160
161     # the only changes we will need to make to it are to change
162     # the cache and log directories, so when we run yum from
163     # the chrooted tempfs mount, it'll cache the rpms on the
164     # sysimg partition
165
166     log.write( "Fetching URL for yum.conf from configuration file service\n" )
167
168     postVars= {"node_id" : NODE_ID,
169                "file" : "/etc/yum.conf"}
170
171     yum_conf_url_file= "/tmp/yumconf.url"
172
173     result= bs_request.DownloadFile(
174         "/db/plnodeconf/getsinglefile.php",
175         None, postVars, 1, 1, yum_conf_url_file)
176     
177     if result == 0:
178         log.write( "Unable to make request to get url for yum.conf\n" )
179         return 0
180
181     try:
182         yum_conf_url= file(yum_conf_url_file,"r").read()
183         yum_conf_url= string.strip(yum_conf_url)
184         if yum_conf_url == "":
185             raise BootManagerException, \
186                   "Downloaded yum configuration file URL is empty."
187     except IOError:
188         raise BootManagerException, \
189               "Unable to open downloaded yum configuration file URL."
190
191     # now, get the actual contents of yum.conf for this node
192     log.write( "Fetching yum.conf contents from configuration file service\n" )
193
194     postVars= {}
195     download_file_loc= "%s/etc/yum.conf" % SYSIMG_PATH
196
197     result= bs_request.DownloadFile( yum_conf_url,
198                                      None, postVars, 1, 1,
199                                      download_file_loc)
200
201     if result == 0:
202         log.write( "Unable to make request to get yum.conf\n" )
203         return 0
204
205     # copy resolv.conf from the base system into our temp dir
206     # so DNS lookups work correctly while we are chrooted
207     log.write( "Copying resolv.conf to temp dir\n" )
208     utils.sysexec( "cp /etc/resolv.conf %s/etc/" % SYSIMG_PATH, log )
209
210     # mount the boot cd in the temp path, under /mnt/cdrom. this way,
211     # we can use the certs when programs are running
212     # chrooted in the temp path
213     cdrom_mount_point= "%s/mnt/cdrom" % SYSIMG_PATH
214     utils.makedirs( cdrom_mount_point )
215     log.write( "Copying contents of /usr/bootme to /mnt/cdrom\n" )
216     utils.sysexec( "cp -r /usr/bootme %s/mnt/cdrom/" % SYSIMG_PATH, log )
217
218     return 1