#!/usr/bin/python
-
+#
# Copyright (c) 2003 Intel Corporation
# All rights reserved.
#
# All rights reserved.
# expected /proc/partitions format
-import os, sys, string
+import os, string
import popen2
import shutil
import traceback
+import time
from Exceptions import *
import utils
def Run( vars, log ):
"""
- Download enough files to run rpm and yum from a chroot in
- the system image directory
+ Download core + extensions bootstrapfs tarballs and install on the hard drive
Expect the following variables from the store:
SYSIMG_PATH the path where the system image will be mounted
PARTITIONS dictionary of generic part. types (root/swap)
and their associated devices.
- SUPPORT_FILE_DIR directory on the boot servers containing
- scripts and support files
NODE_ID the id of this machine
Sets the following variables:
if PARTITIONS == None:
raise ValueError, "PARTITIONS"
- SUPPORT_FILE_DIR= vars["SUPPORT_FILE_DIR"]
- if SUPPORT_FILE_DIR == None:
- raise ValueError, "SUPPORT_FILE_DIR"
-
NODE_ID= vars["NODE_ID"]
if NODE_ID == "":
raise ValueError, "NODE_ID"
+ VERSION=vars['VERSION'] or 'unknown'
+
except KeyError, var:
raise BootManagerException, "Missing variable in vars: %s\n" % var
except ValueError, var:
log.write( "mounting root file system\n" )
utils.sysexec( "mount -t ext3 %s %s" % (PARTITIONS["root"],SYSIMG_PATH), log )
- log.write( "mounting vserver partition in root file system\n" )
- utils.makedirs( SYSIMG_PATH + "/vservers" )
- utils.sysexec( "mount -t ext3 %s %s/vservers" % (PARTITIONS["vservers"],
- SYSIMG_PATH), log )
+ fstype = 'ext3' if vars['virt']=='vs' else 'btrfs'
- vars['ROOT_MOUNTED']= 1
+ one_partition = vars['ONE_PARTITION']=='1'
- # which extensions are we part of ?
- utils.breakpoint("Checking for the extension(s) tags")
- extensions = []
- try:
- extension_tag = BootAPI.call_api_function(vars, "GetNodeExtensions", (NODE_ID,) )
- if extension_tag:
- extensions = extension_tag.split()
+ if (not one_partition):
+ log.write( "mounting vserver partition in root file system (type %s)\n"%fstype )
+ utils.makedirs( SYSIMG_PATH + "/vservers" )
+ utils.sysexec( "mount -t %s %s %s/vservers" % \
+ (fstype, PARTITIONS["vservers"], SYSIMG_PATH), log )
- except:
- log.write("WARNING : Failed to query tag 'extensions'\n")
- log.write(traceback.format_exc())
+ if vars['virt']=='lxc':
+ # NOTE: btrfs quota is supported from version: >= btrfs-progs-0.20 (f18+)
+ # older versions will not recongize the 'quota' command.
+ log.write( "Enabling btrfs quota on %s/vservers\n"%SYSIMG_PATH )
+ utils.sysexec_noerr( "btrfs quota enable %s/vservers" % SYSIMG_PATH )
- if not extensions:
- log.write("installing only core software\n")
-
- # check if the plain-bootstrapfs tag is set
- download_suffix=".tar.bz2"
- untar_option="-j"
- try:
- if BootAPI.call_api_function (vars, "GetNodePlainBootstrapfs", (NODE_ID,) ):
- download_suffix=".tar"
- untar_option=""
- except:
- log.write("WARNING : Failed to query tag 'plain-bootstrapfs'\n")
- log.write(traceback.format_exc())
-
- if not untar_option:
- log.write("Using uncompressed bootstrapfs images\n")
-
- # see also GetBootMedium in PLCAPI that does similar things
- # figuring the default node family:
- # (1) get node's tags 'arch' and 'pldistro'
- # (2) if unsuccessful search /etc/planetlab/nodefamily on the bootcd
- # (3) if that fails, set to planetlab-i386
+ vars['ROOT_MOUNTED']= 1
- try:
- api_pldistro = BootAPI.call_api_function(vars, "GetNodePldistro", (NODE_ID,) )
- except:
- log.write("WARNING : Failed to query tag 'pldistro'\n")
- api_pldistro = None
- try:
- api_arch = BootAPI.call_api_function(vars, "GetNodeArch", (NODE_ID,) )
- except:
- log.write("WARNING : Failed to query tag 'arch'\n")
- api_arch = None
- try:
- (etc_pldistro,etc_arch) = file("/etc/planetlab/nodefamily").read().strip().split("-")
- except:
- log.write("WARNING : Failed to parse /etc/planetlab/nodefamily\n")
- (etc_pldistro,etc_arch)=(None,None)
- default_pldistro="planetlab"
- default_arch="i386"
-
- if api_pldistro:
- pldistro = api_pldistro
- log.write ("Using pldistro from pldistro API tag\n")
- elif etc_pldistro:
- pldistro = etc_pldistro
- log.write ("Using pldistro from /etc/planetlab/nodefamily\n")
- else:
- pldistro = default_pldistro
- log.write ("Using default pldistro\n")
-
- if api_arch:
- arch = api_arch
- log.write ("Using arch from arch API tag\n")
- elif etc_arch:
- arch = etc_arch
- log.write ("Using arch from /etc/planetlab/nodefamily\n")
+ # this is now retrieved in GetAndUpdateNodeDetails
+ nodefamily = vars['nodefamily']
+ extensions = vars['extensions']
+ # the 'plain' option is for tests mostly
+ plain = vars['plain']
+ if plain:
+ download_suffix=".tar"
+ uncompress_option=""
+ log.write("Using plain bootstrapfs images\n")
else:
- arch = default_arch
- log.write ("Using default arch\n")
-
- log.write ("Using nodefamily=%s-%s\n"%(pldistro,arch))
+ download_suffix=".tar.bz2"
+ uncompress_option="-j"
+ log.write("Using compressed bootstrapfs images\n")
- bootstrapfs_names = [ pldistro ] + extensions
-
- # download and extract support tarball for this step, which has
- # everything we need to successfully run
+ log.write ("Using nodefamily=%s\n"%(nodefamily))
+ if not extensions:
+ log.write("Installing only core software\n")
+ else:
+ log.write("Requested extensions %r\n" % extensions)
+
+ bootstrapfs_names = [ nodefamily ] + extensions
- # we first try to find a tarball, if it is not found we use yum instead
- yum_extensions = []
- # download and extract support tarball for this step,
- for bootstrapfs_name in bootstrapfs_names:
- tarball = "bootstrapfs-%s-%s%s"%(bootstrapfs_name,arch,download_suffix)
- source_file= "%s/%s" % (SUPPORT_FILE_DIR,tarball)
+ for name in bootstrapfs_names:
+ tarball = "bootstrapfs-%s%s"%(name,download_suffix)
+ source_file= "/boot/%s" % (tarball)
dest_file= "%s/%s" % (SYSIMG_PATH, tarball)
+ source_hash_file= "/boot/%s.sha1sum" % (tarball)
+ dest_hash_file= "%s/%s.sha1sum" % (SYSIMG_PATH, tarball)
+
+ time_beg=time.time()
+ log.write( "downloading %s\n" % source_file )
# 30 is the connect timeout, 14400 is the max transfer time in
# seconds (4 hours)
- log.write( "downloading %s\n" % source_file )
- result= bs_request.DownloadFile( source_file, None, None,
+ result = bs_request.DownloadFile( source_file, None, None,
1, 1, dest_file,
30, 14400)
+ time_end=time.time()
+ duration=int(time_end-time_beg)
+ log.write("Done downloading (%s seconds)\n"%duration)
if result:
+ # Download SHA1 checksum file
+ log.write( "downloading sha1sum for %s\n"%source_file)
+ result = bs_request.DownloadFile( source_hash_file, None, None,
+ 1, 1, dest_hash_file,
+ 30, 14400)
+
+ log.write( "verifying sha1sum for %s\n"%source_file)
+ if not utils.check_file_hash(dest_file, dest_hash_file):
+ raise BootManagerException, "FATAL: SHA1 checksum does not match between %s and %s" % (source_file, source_hash_file)
+
+
+ time_beg=time.time()
log.write( "extracting %s in %s\n" % (dest_file,SYSIMG_PATH) )
- result= utils.sysexec( "tar -C %s -xpf %s %s" % (SYSIMG_PATH,dest_file,untar_option), log )
- log.write( "Done\n")
+ result = utils.sysexec( "tar -C %s -xpf %s %s" % (SYSIMG_PATH,dest_file,uncompress_option), log )
+ time_end=time.time()
+ duration=int(time_end-time_beg)
+ log.write( "Done extracting (%s seconds)\n"%duration)
utils.removefile( dest_file )
else:
# the main tarball is required
- if bootstrapfs_name == pldistro:
- raise BootManagerException, "Unable to download main tarball %s from server." % \
+ if name == nodefamily:
+ raise BootManagerException, "FATAL: Unable to download main tarball %s from server." % \
source_file
+ # for extensions, just print a warning
else:
- log.write("tarball for %s-%s not found, scheduling a yum attempt\n"%(bootstrapfs_name,arch))
- yum_extensions.append(bootstrapfs_name)
+ log.write("WARNING: tarball for extension %s not found\n"%(name))
# copy resolv.conf from the base system into our temp dir
# so DNS lookups work correctly while we are chrooted
utils.makedirs(SYSIMG_PATH + "/mnt/cdrom")
shutil.copytree("/usr/bootme", SYSIMG_PATH + "/mnt/cdrom/bootme")
- # Import the GPG key into the RPM database so that RPMS can be verified
- utils.makedirs(SYSIMG_PATH + "/etc/pki/rpm-gpg")
- utils.sysexec("gpg --homedir=/root --export --armor" \
- " --no-default-keyring --keyring %s/usr/boot/pubring.gpg" \
- " >%s/etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab" % (SYSIMG_PATH, SYSIMG_PATH))
- utils.sysexec_chroot(SYSIMG_PATH, "rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab")
-
- # the yum config has changed entirely;
- # in addition yum installs have more or less never worked - let's forget about this
- # maybe NodeManager could profitably do the job instead
- if yum_extensions:
- log.write("WARNING : yum installs for node extensions are not supported anymore\n")
+ # ONE_PARTITION => new distribution type
+ if (vars['ONE_PARTITION']!='1'):
+ # Import the GPG key into the RPM database so that RPMS can be verified
+ utils.makedirs(SYSIMG_PATH + "/etc/pki/rpm-gpg")
+ utils.sysexec("gpg --homedir=/root --export --armor" \
+ " --no-default-keyring --keyring %s/usr/boot/pubring.gpg" \
+ " >%s/etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab" % (SYSIMG_PATH, SYSIMG_PATH), log)
+ utils.sysexec_chroot(SYSIMG_PATH, "rpm --import /etc/pki/rpm-gpg/RPM-GPG-KEY-planetlab", log)
+
+ # keep a log on the installed hdd
+ stamp=file(SYSIMG_PATH + "/bm-install.txt",'w')
+ now=time.strftime("%Y-%b-%d @ %H:%M %Z", time.gmtime())
+ stamp.write("Hard drive installed by BootManager %s\n"%VERSION)
+ stamp.write("Finished extraction of bootstrapfs on %s\n"%now)
+ stamp.write("Using nodefamily %s\n"%nodefamily)
+ stamp.close()
return 1