build.sh

   1 #!/bin/bash
   2 #
   3 # Builds custom BootCD ISO and USB images in the current
   4 # directory.
   5 #
   6 # Aaron Klingaman <alk@absarokasoft.com>
   7 # Mark Huang <mlhuang@cs.princeton.edu>
   8 # Copyright (C) 2004-2007 The Trustees of Princeton University
   9 #
  10 # Jan 2015 - f21 comes with isolinux 6.03 (was 4.05 in f20)
  11 # http://www.syslinux.org/wiki/index.php/ISOLINUX
  12
  13 COMMAND=$(basename $0)
  14 DIRNAME=$(dirname $0)
  15 PATH=/sbin:/bin:/usr/sbin:/usr/bin
  16
  17 # debugging flags
  18 # keep KERNEL_DEBUG_ARGS void for production
  19 KERNEL_DEBUG_ARGS=""
  20 # add more flags here for debugging
  21 # KERNEL_DEBUG_ARGS="$KERNEL_DEBUG_ARGS some_other_kernel_arg"
  22 # see also
  23 #  (*) GetBootMedium that has some provisions for common
  24 #      kargs, like e.g. for removing the hangcheck feature,
  25 #      or for turning on debug messages for systemd
  26 #      these can be turned on with tags on the node
  27 #  (*) tests default config, that uses this feature so
  28 #      the tests can benefit these features, without deploying
  29 #      them by default in production
  30
  31 # defaults
  32 DEFAULT_TYPES="usb iso"
  33 # Leave 4 MB of free space
  34 GRAPHIC_CONSOLE="graphic"
  35 SERIAL_CONSOLE="ttyS0:115200:n:8"
  36 CONSOLE_INFO=$GRAPHIC_CONSOLE
  37 MKISOFS_OPTS="-R -J -r -f -b isolinux.bin -c boot.cat -no-emul-boot -boot-load-size 4 -boot-info-table"
  38 # isolinux-debug.bin is supposedly helpful as well if available,
  39 # when things don't work as expected
  40 #MKISOFS_OPTS="-R -J -r -f -b isolinux-debug.bin -c boot.cat -no-emul-boot -boot-load-size 4 -boot-info-table"
  41
  42 FREE_SPACE=4096
  43
  44 # command-line settable args
  45 NODE_CONFIGURATION_FILE=
  46 CUSTOM_DIR=
  47 OUTPUT_BASE=
  48 DRY_RUN=""
  49 OUTPUT_NAME=""
  50 TYPES=""
  51 KERNEL_ARGS=""
  52
  53 # various globals
  54 BUILDTMP=""
  55 FULL_VERSION_STRING=""
  56 ISOREF=""
  57 ISOFS=""
  58 OVERLAY=""
  59 IS_SERIAL=""
  60 console_dev=""
  61 console_baud=""
  62 console_spec=""
  63 console_serial_line=""
  64
  65
  66 #################### compute all supported types
  67 # removing support for serial in the type
  68 # this is because kargs.txt goes in the overlay, that is computed only once
  69 # so we cannot handle serial and graphic modes within the same invokation of this script
  70
  71 ALL_TYPES=""
  72 for x in iso usb usb_partition; do for c in "" "_cramfs" ; do
  73   t="${x}${c}"
  74   case $t in
  75       usb_partition_cramfs)
  76           # unsupported
  77           ;;
  78       *)
  79           ALL_TYPES="$ALL_TYPES $t" ;;
  80   esac
  81 done; done
  82
  83 #################### cleanup utilities
  84 declare -a _CLEANUPS=()
  85 function do_cleanup() {
  86     cd / ; for i in "${_CLEANUPS[@]}"; do $i ; done
  87 }
  88 function push_cleanup() {
  89     _CLEANUPS=( "${_CLEANUPS[@]}" "$*" )
  90 }
  91 function pop_cleanup() {
  92     unset _CLEANUPS[$((${#_CLEANUPS[@]} - 1))]
  93 }
  94
  95 #################### initialization
  96 function init_and_check () {
  97
  98     # Change to our source directory
  99     local srcdir=$(cd $DIRNAME && pwd -P)
 100     pushd $srcdir
 101
 102     # Root of the isofs
 103     ISOREF=$PWD/${VARIANT}
 104
 105     # The reference image is expected to have been built by prep.sh (see .spec)
 106     # we disable the initial logic that called prep.sh if that was not the case
 107     # this is because prep.sh needs to know pldistro
 108     if [ ! -f $ISOREF/isofs/bootcd.img -o ! -f $ISOREF/version.txt ] ; then
 109         echo "Could not find isofs and version.txt in $ISOREF"
 110         if [ "$VARIANT" == "build" ] ; then
 111             echo "You have to run prep.sh prior to calling $COMMAND"
 112         else
 113             echo "You need to create your variant image, see kvariant.sh"
 114         fi
 115         echo "Exiting .."
 116         exit 1
 117     fi
 118
 119     # build/version.txt written by prep.sh
 120     BOOTCD_VERSION=$(cat ${VARIANT}/version.txt)
 121
 122     if [ -f /etc/planetlab/plc_config ] ; then
 123         # Source PLC configuration
 124         . /etc/planetlab/plc_config
 125     fi
 126
 127     # use /var/tmp that should be large enough on both chroot- or vserver-based myplc
 128     BUILDTMP=/var/tmp
 129
 130     FULL_VERSION_STRING="${PLC_NAME} BootCD ${BOOTCD_VERSION}"
 131
 132 }
 133
 134 # NOTE
 135 # the custom-dir feature is designed to let a myplc try/ship a patched bootcd
 136 # without the need for a full devel environment
 137 # for example, you would create /root/custom-bootcd/etc/rc.d/init.d/pl_hwinit
 138 # and run this script with -C /root/custom-bootcd
 139 # this creates a third .img image of the custom dir, that 'hides' the files from
 140 # bootcd.img in the resulting unionfs
 141 # it seems that this feature has not been used nor tested in a long time, use with care
 142
 143 usage() {
 144     echo "Usage: $COMMAND [OPTION]..."
 145     echo "    -f plnode.txt    Node to customize CD for (default: none)"
 146     echo "    -t 'types'       Build the specified images (default: $DEFAULT_TYPES)"
 147     echo "                     NOTE: mentioning 'serial' as part of the type is not supported anymore"
 148     echo "    -a               Build all known types as listed below"
 149     echo "    -s console-info  Enable a serial line as console and also bring up getty on that line"
 150     echo "                     console-info: tty:baud-rate:parity:bits"
 151     echo "                     or 'default' shortcut for $SERIAL_CONSOLE"
 152     echo "    -S               equivalent to -s default"
 153     echo "    -O output-base   The prefix of the generated files (default: PLC_NAME-BootCD-VERSION)"
 154     echo "                     useful when multiple types are provided"
 155     echo "                     can be a full path"
 156     echo "    -o output-name   The full name of the generated file"
 157     echo "    -C custom-dir    Custom directory"
 158     echo "    -V variant       Use a variant - see kvariant.sh"
 159     echo "    -n               Dry run - mostly for debug/test purposes"
 160     echo "    -k               Add additional parameters to the kargs.txt file"
 161     echo "    -h               This message"
 162     echo "All known types: $ALL_TYPES"
 163     exit 1
 164 }
 165
 166 ####################
 167 function parse_command_line () {
 168
 169     # init
 170     TYPES=""
 171     # Get options
 172     while getopts "f:t:as:SO:o:C:V:k:nh" opt ; do
 173         case $opt in
 174             f) NODE_CONFIGURATION_FILE=$OPTARG ;;
 175             t) TYPES="$TYPES $OPTARG" ;;
 176             a) TYPES="$ALL_TYPES" ;;
 177             s) CONSOLE_INFO="$OPTARG" ;;
 178             S) CONSOLE_INFO=$SERIAL_CONSOLE ;;
 179             O) OUTPUT_BASE="$OPTARG" ;;
 180             o) OUTPUT_NAME="$OPTARG" ;;
 181             C) CUSTOM_DIR="$OPTARG" ;;
 182             V) VARIANT="$OPTARG" ;;
 183             k) KERNEL_ARGS="$KERNEL_ARGS $OPTARG" ;;
 184             n) DRY_RUN=true ;;
 185             h|*) usage ;;
 186         esac
 187     done
 188
 189     # use defaults if not set
 190     [ -z "$TYPES" ] && TYPES="$DEFAULT_TYPES"
 191     [ -z "$VARIANT" ] && VARIANT="build"
 192     [ "$CONSOLE_INFO" == "default" ] && CONSOLE_INFO=$SERIAL_CONSOLE
 193
 194     if [ -n "$NODE_CONFIGURATION_FILE" ] ; then
 195     # check existence of NODE_CONFIGURATION_FILE and normalize as we will change directory
 196         if [ ! -f "$NODE_CONFIGURATION_FILE" ] ; then
 197             echo "Node configuration file $NODE_CONFIGURATION_FILE not found - exiting"
 198             exit 1
 199         fi
 200         cf_dir="$(dirname $NODE_CONFIGURATION_FILE)"
 201         cf_dir="$(cd $cf_dir; pwd -P)"
 202         cf_file="$(basename $NODE_CONFIGURATION_FILE)"
 203         NODE_CONFIGURATION_FILE="$cf_dir"/"$cf_file"
 204     fi
 205
 206     # check TYPES
 207     local matcher="XXX$(echo $ALL_TYPES | sed -e 's,\W,XXX,g')XXX"
 208     for t in $TYPES; do
 209         echo Checking type $t
 210         echo $matcher | grep XXX${t}XXX &> /dev/null
 211         if [ "$?" != 0 ] ; then
 212             echo Unknown type $t
 213             usage
 214         fi
 215     done
 216
 217 }
 218
 219 ####################
 220 function init_serial () {
 221     local console=$1; shift
 222     if [ "$console" == "$GRAPHIC_CONSOLE" ] ; then
 223         IS_SERIAL=
 224         console_spec=""
 225         echo "Standard, graphic, non-serial mode"
 226     else
 227         IS_SERIAL=true
 228         console_dev=$(echo "$console" | awk -F: ' {print $1}')
 229         console_baud=$(echo "$console" | awk -F: ' {print $2}')
 230         [ -z "$console_baud" ] && console_baud="115200"
 231         local console_parity=$(echo "$console" | awk -F: ' {print $3}')
 232         [ -z "$console_parity" ] && console_parity="n"
 233         local console_bits=$(echo "$console" | awk -F: ' {print $4}')
 234         [ -z "$console_bits" ] && console_bits="8"
 235         console_spec="console=${console_dev},${console_baud}${console_parity}${console_bits}"
 236         local tty_nb=$(echo $console_dev | sed -e 's,[a-zA-Z],,g')
 237         console_serial_line="SERIAL ${tty_nb} ${console_baud}"
 238         echo "Serial mode"
 239         echo "console_serial_line=${console_serial_line}"
 240         echo "console_spec=${console_spec}"
 241     fi
 242 }
 243
 244 #################### run once : build the overlay image
 245 function build_overlay () {
 246
 247     BUILDTMP=$(mktemp -d ${BUILDTMP}/bootcd.XXXXXX)
 248     push_cleanup rm -fr "${BUILDTMP}"
 249
 250     # initialize ISOFS
 251     ISOFS="${BUILDTMP}/isofs"
 252     mkdir -p "$ISOFS"
 253     for i in "$ISOREF"/isofs/{bootcd.img,kernel}; do
 254         ln -s "$i" "$ISOFS"
 255     done
 256     # use new location as of fedora 12
 257     # used to be in /usr/lib/syslinux/isolinux.bin
 258     # removed backward compat in jan. 2015
 259     # as of syslinux 6.05 (fedora 21) ldlinux.c32 is required by isolinux.bin
 260     # the debug version can come in handy at times, and is 40k as well
 261     isolinuxdir="/usr/share/syslinux"
 262     # ship only what is mandatory, and forget about
 263     # (*) isolinux-debug.bin as its name confuses mkisofs
 264     # (*) memdisk that is not useful
 265     isolinuxfiles="isolinux.bin ldlinux.c32"
 266     for isolinuxfile in $isolinuxfiles; do
 267         [ -f $isolinuxdir/$isolinuxfile ] && cp $isolinuxdir/$isolinuxfile "${BUILDTMP}/isofs"
 268     done
 269
 270     # Root of the ISO and USB images
 271     echo "* Populating root filesystem..."
 272     OVERLAY="${BUILDTMP}/overlay"
 273     install -d -m 755 $OVERLAY
 274     push_cleanup rm -fr $OVERLAY
 275
 276     # Create version files
 277     echo "* Creating version files"
 278
 279     # Boot Manager compares pl_version in both places to make sure that
 280     # the right CD is mounted. We used to boot from an initrd and mount
 281     # the CD on /usr. Now we just run everything out of the initrd.
 282     for file in $OVERLAY/pl_version $OVERLAY/usr/isolinux/pl_version ; do
 283         mkdir -p $(dirname $file)
 284         echo "$FULL_VERSION_STRING" >$file
 285     done
 286
 287     # Install boot server configuration files
 288     echo "* Installing boot server configuration files"
 289
 290     # We always intended to bring up and support backup boot servers,
 291     # but never got around to it. Just install the same parameters for
 292     # both for now.
 293     for dir in $OVERLAY/usr/boot $OVERLAY/usr/boot/backup ; do
 294         install -D -m 644 $PLC_BOOT_CA_SSL_CRT $dir/cacert.pem
 295         install -D -m 644 $PLC_ROOT_GPG_KEY_PUB $dir/pubring.gpg
 296         echo "$PLC_BOOT_HOST" >$dir/boot_server
 297         echo "$PLC_BOOT_SSL_PORT" >$dir/boot_server_port
 298         echo "/boot/" >$dir/boot_server_path
 299     done
 300
 301     # Install old-style boot server configuration files
 302     # as opposed to what a former comment suggested,
 303     # this is still required, somewhere in the bootmanager apparently
 304     install -D -m 644 $PLC_BOOT_CA_SSL_CRT $OVERLAY/usr/bootme/cacert/$PLC_BOOT_HOST/cacert.pem
 305     echo "$FULL_VERSION_STRING" >$OVERLAY/usr/bootme/ID
 306     echo "$PLC_BOOT_HOST" >$OVERLAY/usr/bootme/BOOTSERVER
 307     echo "$PLC_BOOT_HOST" >$OVERLAY/usr/bootme/BOOTSERVER_IP
 308     echo "$PLC_BOOT_SSL_PORT" >$OVERLAY/usr/bootme/BOOTPORT
 309
 310     # Generate /etc/issue
 311     echo "* Generating /etc/issue"
 312
 313     if [ "$PLC_WWW_PORT" = "443" ] ; then
 314         PLC_WWW_URL="https://$PLC_WWW_HOST/"
 315     elif [ "$PLC_WWW_PORT" != "80" ] ; then
 316         PLC_WWW_URL="http://$PLC_WWW_HOST:$PLC_WWW_PORT/"
 317     else
 318         PLC_WWW_URL="http://$PLC_WWW_HOST/"
 319     fi
 320
 321     mkdir -p $OVERLAY/etc
 322     cat >$OVERLAY/etc/issue <<EOF
 323 $FULL_VERSION_STRING
 324 $PLC_NAME Node: \n
 325 Kernel \r on an \m
 326 $PLC_WWW_URL
 327
 328 This machine is a node in the $PLC_NAME distributed network.  It has
 329 not fully booted yet. If you have cancelled the boot process at the
 330 request of $PLC_NAME Support, please follow the instructions provided
 331 to you. Otherwise, please contact $PLC_MAIL_SUPPORT_ADDRESS.
 332
 333 Console login at this point is restricted to root. Provide the root
 334 password of the default $PLC_NAME Central administrator account at the
 335 time that this CD was created.
 336
 337 EOF
 338
 339     # Set root password
 340     echo "* Setting root password"
 341
 342     if [ -z "$ROOT_PASSWORD" ] ; then
 343         # Generate an encrypted password with crypt() if not defined
 344         # in a static configuration.
 345         ROOT_PASSWORD=$(python <<EOF
 346 import crypt, random, string
 347 salt = [random.choice(string.letters + string.digits + "./") for i in range(0,8)]
 348 print crypt.crypt('$PLC_ROOT_PASSWORD', '\$1\$' + "".join(salt) + '\$')
 349 EOF
 350 )
 351     fi
 352
 353     # build/passwd copied out by prep.sh
 354     sed -e "s@^root:[^:]*:\(.*\)@root:$ROOT_PASSWORD:\1@" ${VARIANT}/passwd >$OVERLAY/etc/passwd
 355
 356 # this is more harmful than helpful
 357 # idea being, since we start a full-featured fedora system now, it would
 358 # have been nice to be able to enter sshd very early on - before bm has even been downloaded
 359 # however somehow it appears that these lines ruin all chances to enter ssh at all
 360 # either early or even later on;
 361 # plus, it is unclear what this would give on non=systemd nodes, so I am backing off for now
 362 #    # recent bootCDs rely on a standard systemd startup sequence
 363 #    # so allow debug key to enter in this context whenever that makes sense
 364 #    mkdir -p $OVERLAY/root/.ssh
 365 #    chmod 700 $OVERLAY/root/.ssh
 366 #    cp $PLC_DEBUG_SSH_KEY_PUB $OVERLAY/root/.ssh/authorized_keys
 367 #    chmod 600 $OVERLAY/root/.ssh/authorized_keys
 368
 369     # Install node configuration file (e.g., if node has no floppy disk or USB slot)
 370     if [ -f "$NODE_CONFIGURATION_FILE" ] ; then
 371         echo "* Installing node configuration file $NODE_CONFIGURATION_FILE -> /usr/boot/plnode.txt of the bootcd image"
 372         install -D -m 644 $NODE_CONFIGURATION_FILE $OVERLAY/usr/boot/plnode.txt
 373         NODE_ID=$(source $NODE_CONFIGURATION_FILE; echo $NODE_ID)
 374         echo "* Building network configuration for $NODE_ID"
 375         plnet -- --root $OVERLAY --files-only --program BootCD $NODE_ID
 376     fi
 377
 378     [ -n "$IS_SERIAL" ] && KERNEL_ARGS="$KERNEL_ARGS ${console_spec}"
 379
 380     # making sure the network interfaces are still numbered eth0 and above
 381     KERNEL_ARGS="$KERNEL_ARGS biosdevname=0"
 382     # this apparently is required instead (or in addition to) starting with f29
 383     KERNEL_ARGS="$KERNEL_ARGS net.ifnames=0"
 384     # making sure selinux is turned off - somehow this is needed with lxc/f14
 385     KERNEL_ARGS="$KERNEL_ARGS selinux=0"
 386     # add any debug flag if any (defined in the header of this script)
 387     KERNEL_ARGS="$KERNEL_ARGS $KERNEL_DEBUG_ARGS"
 388     # propagate kernel args for later boot stages
 389     [ -n "$KERNEL_ARGS" ] && echo "$KERNEL_ARGS" > $OVERLAY/kargs.txt
 390
 391     # Pack overlay files into a compressed archive
 392     echo "* Compressing overlay image"
 393     (cd $OVERLAY && find . | cpio --quiet -c -o) | gzip -9 >$ISOFS/overlay.img
 394
 395     rm -rf $OVERLAY
 396     pop_cleanup
 397
 398     if [ -n "$CUSTOM_DIR" ]; then
 399         echo "* Compressing custom image"
 400         (cd "$CUSTOM_DIR" && find . | cpio --quiet -c -o) | gzip -9 >$ISOFS/custom.img
 401     fi
 402
 403     # Calculate ramdisk size (total uncompressed size of both archives)
 404     ramdisk_size=$(gzip -l $ISOFS/bootcd.img $ISOFS/overlay.img ${CUSTOM_DIR:+$ISOFS/custom.img} | tail -1 | awk '{ print $2; }') # bytes
 405     ramdisk_size=$((($ramdisk_size + 1023) / 1024)) # kilobytes
 406
 407     echo "$FULL_VERSION_STRING" >$ISOFS/pl_version
 408
 409     popd
 410 }
 411
 412 #################### plain ISO
 413 function build_iso() {
 414     local iso="$1" ; shift
 415     local custom="$1"
 416
 417     # Write isolinux configuration
 418     cat >$ISOFS/isolinux.cfg <<EOF
 419 ${console_serial_line}
 420 PROMPT 0
 421 DEFAULT planetlab-bootcd
 422
 423 LABEL planetlab-bootcd
 424   DISPLAY pl_version
 425   LINUX kernel
 426   APPEND ramdisk_size=$ramdisk_size initrd=bootcd.img,overlay.img${custom:+,custom.img} root=/dev/ram0 rw ${KERNEL_ARGS}
 427 EOF
 428
 429     # Create ISO image
 430     echo "* Generated isolinux.cfg -------------------- BEG"
 431     cat $ISOFS/isolinux.cfg
 432     echo "* Generated isolinux.cfg -------------------- END"
 433     echo "* Creating ISO image in pwd=$(pwd)"
 434     echo "* with command mkisofs -o $iso $MKISOFS_OPTS $ISOFS"
 435     mkisofs -o "$iso" $MKISOFS_OPTS $ISOFS
 436 }
 437
 438 #################### USB with partitions
 439 function build_usb_partition() {
 440     echo -n "* Creating USB image with partitions..."
 441     local usb="$1" ; shift
 442     local custom="$1"
 443
 444     local size=$(($(du -Lsk $ISOFS | awk '{ print $1; }') + $FREE_SPACE))
 445     size=$(( $size / 1024 ))
 446
 447     local heads=64
 448     local sectors=32
 449     local cylinders=$(( ($size*1024*2)/($heads*$sectors) ))
 450     local offset=$(( $sectors*512 ))
 451
 452     if [ -f  /usr/lib/syslinux/mkdiskimage ] ; then
 453         /usr/lib/syslinux/mkdiskimage -M -4 "$usb" $size $heads $sectors
 454     else
 455         mkdiskimage -M -4 "$usb" $size $heads $sectors
 456     fi
 457
 458     cat >${BUILDTMP}/mtools.conf<<EOF
 459 drive z:
 460 file="${usb}"
 461 cylinders=$cylinders
 462 heads=$heads
 463 sectors=$sectors
 464 offset=$offset
 465 mformat_only
 466 mtools_skip_check=1
 467 EOF
 468     # environment variable for mtools
 469     export MTOOLSRC="${BUILDTMP}/mtools.conf"
 470
 471     ### COPIED FROM build_usb() below!!!!
 472     echo -n " populating USB image... "
 473     mcopy -bsQ -i "$usb" "$ISOFS"/* z:/
 474
 475     # Use syslinux instead of isolinux to make the image bootable
 476     tmp="${BUILDTMP}/syslinux.cfg"
 477     cat >$tmp <<EOF
 478 ${console_serial_line}
 479 PROMPT 0
 480 DEFAULT planetlab-bootcd
 481
 482 LABEL planetlab-bootcd
 483   DISPLAY pl_version
 484   LINUX kernel
 485   APPEND ramdisk_size=$ramdisk_size initrd=bootcd.img,overlay.img${custom:+,custom.img} root=/dev/ram0 rw ${KERNEL_ARGS}
 486 EOF
 487     mdel -i "$usb" z:/isolinux.cfg 2>/dev/null || :
 488     mcopy -i "$usb" "$tmp" z:/syslinux.cfg
 489     rm -f "$tmp"
 490     rm -f "${MTOOLSRC}"
 491     unset MTOOLSRC
 492
 493     echo "making USB image bootable."
 494     syslinux -o $offset "$usb"
 495
 496 }
 497
 498 #################### plain USB
 499 function build_usb() {
 500     echo -n "* Creating USB image... "
 501     local usb="$1" ; shift
 502     local custom="$1"
 503
 504     rm -f "$usb"
 505     mkfs.vfat -C "$usb" $(($(du -Lsk $ISOFS | awk '{ print $1; }') + $FREE_SPACE))
 506
 507     cat >${BUILDTMP}/mtools.conf<<EOF
 508 mtools_skip_check=1
 509 EOF
 510     # environment variable for mtools
 511     export MTOOLSRC="${BUILDTMP}/mtools.conf"
 512
 513     # Populate it
 514     echo -n " populating USB image... "
 515     mcopy -bsQ -i "$usb" "$ISOFS"/* ::/
 516
 517     # Use syslinux instead of isolinux to make the image bootable
 518     tmp="${BUILDTMP}/syslinux.cfg"
 519     cat >$tmp <<EOF
 520 ${console_serial_line}
 521 PROMPT 0
 522 DEFAULT planetlab-bootcd
 523
 524 LABEL planetlab-bootcd
 525   DISPLAY pl_version
 526   LINUX kernel
 527   APPEND ramdisk_size=$ramdisk_size initrd=bootcd.img,overlay.img${custom:+,custom.img} root=/dev/ram0 rw ${KERNEL_ARGS}
 528 EOF
 529     mdel -i "$usb" ::/isolinux.cfg 2>/dev/null || :
 530     mcopy -i "$usb" "$tmp" ::/syslinux.cfg
 531     rm -f "$tmp"
 532     rm -f "${MTOOLSRC}"
 533     unset MTOOLSRC
 534
 535     echo "making USB image bootable."
 536     syslinux "$usb"
 537 }
 538
 539 #################### utility to setup CRAMFS related support
 540 function prepare_cramfs() {
 541     [ -n "$CRAMFS_PREPARED" ] && return 0
 542     local custom=$1;
 543
 544     echo "* Setting up CRAMFS-based images"
 545     local tmp="${BUILDTMP}/cramfs-tree"
 546     mkdir -p "$tmp"
 547     push_cleanup rm -rf $tmp
 548     pushd $tmp
 549     gzip -d -c $ISOFS/bootcd.img     | cpio -diu
 550     gzip -d -c $ISOFS/overlay.img    | cpio -diu
 551     [ -n "$custom" ] && \
 552         gzip -d -c $ISOFS/custom.img | cpio -diu
 553
 554     # clean out unnecessary rpm lib
 555     echo "* clearing var/lib/rpm/*"
 556     rm -f var/lib/rpm/*
 557
 558     # bootcd requires this directory
 559     mkdir -p mnt/confdevice
 560
 561     # relocate various directory to /tmp
 562     rm -rf root
 563     ln -fs /tmp/root root
 564     ln -fs /sbin/init linuxrc
 565     ln -fs /tmp/resolv.conf etc/resolv.conf
 566     ln -fs /tmp/etc/mtab etc/mtab
 567
 568     # have pl_rsysinit copy over appropriate etc & var directories into /tmp/etc/
 569     # make /tmp/etc
 570     echo "* renaming dirs in ./etc"
 571     pushd etc
 572     for dir in `find * -type d -prune | grep -v rc.d`; do
 573         mv ${dir} ${dir}_o
 574         ln -fs /tmp/etc/${dir} ${dir}
 575     done
 576     popd
 577
 578     echo "* renaming dirs in ./var"
 579     # rename all top-level directories and put in a symlink to /tmp/var
 580     pushd var
 581     for dir in `find * -type d -prune`; do
 582         mv ${dir} ${dir}_o
 583         ln -fs /tmp/var/${dir} ${dir}
 584     done
 585     popd
 586
 587     # overwrite fstab to mount / as cramfs and /tmp as tmpfs
 588     echo "* Overwriting etc/fstab to use cramfs and tmpfs"
 589     rm -f ./etc/fstab
 590     cat >./etc/fstab <<EOF
 591 /dev/ram0     /              cramfs     ro              0 0
 592 none          /dev/pts       devpts     gid=5,mode=620  0 0
 593 none          /proc          proc       defaults        0 0
 594 none          /sys           sysfs      defaults        0 0
 595 EOF
 596
 597     pushd dev
 598     rm -f console
 599     mknod console c 5 1
 600     #for i in 0 1 2 3 4 5 6 7 8; do rm -f ram${i} ; done
 601     #for i in 0 1 2 3 4 5 6 7 8; do mknod ram${i} b 1 ${i} ; done
 602     #ln -fs ram1 ram
 603     #ln -fs ram0 ramdisk
 604     popd
 605
 606     # update etc/inittab to start with pl_rsysinit
 607     for file in etc/inittab etc/event.d/rcS etc/init/rcS.conf; do
 608         [ -f $file ] && sed -i 's,pl_sysinit,pl_rsysinit,' $file
 609     done
 610
 611     # modify inittab to have a serial console
 612     # xxx this might well be broken with f12 and above xxx
 613     if [ -n "$serial" ] ; then
 614         echo "T0:23:respawn:/sbin/agetty -L $console_dev $console_baud vt100" >> etc/inittab
 615         # and let root log in
 616         echo "$console_dev" >> etc/securetty
 617     fi
 618
 619     # calculate the size of /tmp based on the size of /etc & /var + 8MB slack
 620     etcsize=$(du -s ./etc | awk '{ print $1 }')
 621     varsize=$(du -s ./var | awk '{ print $1 }')
 622     let msize=($varsize+$etcsize+8192)/1024
 623
 624     # make dhclient happy
 625     for i in $(seq 0 9); do ln -fs /tmp/etc/dhclient-eth${i}.conf etc/dhclient-eth${i}.conf ; done
 626     ln -fs /tmp/etc/resolv.conf etc/resolv.conf
 627     ln -fs /tmp/etc/resolv.conf.predhclient etc/resolv.conf.predhclient
 628
 629     # generate pl_rsysinit
 630     cat > etc/rc.d/init.d/pl_rsysinit <<EOF
 631 #!/bin/sh
 632 # generated by $COMMAND
 633 echo -n "pl_rsysinit: preparing /etc and /var for pl_sysinit..."
 634 mount -t tmpfs -orw,size=${msize}M,mode=1777 tmpfs /tmp
 635 mkdir -p /tmp/root
 636 mkdir -p /tmp/etc
 637 touch /tmp/etc/resolv.conf
 638 touch /tmp/etc/mtab
 639 mkdir -p /tmp/var
 640
 641 # make mtab happy
 642 echo "tmpfs /tmp tmpfs rw,size=${msize}M,mode=1777 1 1" > /tmp/etc/mtab
 643
 644 # copy over directory contents of all _o directories from /etc and /var
 645 # /tmp/etc and /tmp/var
 646 pushd /etc
 647 for odir in \$(cd /etc && ls -d *_o); do dir=\$(echo \$odir | sed 's,\_o$,,'); (mkdir -p /tmp/etc/\$dir && cd \$odir && find . | cpio -p -d -u /tmp/etc/\$dir); done
 648 popd
 649 pushd /var
 650 for odir in \$(cd /var && ls -d *_o); do dir=\$(echo \$odir | sed 's,\_o$,,'); (mkdir -p /tmp/var/\$dir && cd \$odir && find . | cpio -p -d -u /tmp/var/\$dir); done
 651 popd
 652
 653 echo "done"
 654
 655 # hand over to pl_sysinit
 656 echo "pl_rsysinit: handing over to pl_sysinit"
 657 /etc/init.d/pl_sysinit
 658 EOF
 659     chmod +x etc/rc.d/init.d/pl_rsysinit
 660
 661     popd
 662
 663     # create the cramfs image
 664     echo "* Creating cramfs image"
 665     mkfs.cramfs $tmp/ ${BUILDTMP}/cramfs.img
 666     cramfs_size=$(($(du -sk ${BUILDTMP}/cramfs.img | awk '{ print $1; }') + 1))
 667     rm -rf $tmp
 668     pop_cleanup
 669 }
 670
 671 #################### Create ISO CRAMFS image
 672 function build_iso_cramfs() {
 673     local iso="$1" ; shift
 674     local custom="$1"
 675
 676     prepare_cramfs "$custom"
 677     echo "* Creating ISO CRAMFS-based image"
 678
 679     local tmp="${BUILDTMP}/cramfs-iso"
 680     mkdir -p "$tmp"
 681     push_cleanup rm -rf $tmp
 682     (cd $ISOFS && find . | grep -v "\.img$" | cpio -p -d -u $tmp/)
 683     cat >$tmp/isolinux.cfg <<EOF
 684 ${console_serial_line}
 685 PROMPT 0
 686 DEFAULT planetlab-bootcd
 687
 688 LABEL planetlab-bootcd
 689   DISPLAY pl_version
 690   LINUX kernel
 691   APPEND ramdisk_size=$ramdisk_size initrd=cramfs.img root=/dev/ram0 rw ${KERNEL_ARGS}
 692 EOF
 693
 694     cp ${BUILDTMP}/cramfs.img $tmp
 695     mkisofs -o "$iso" \
 696         $MKISOFS_OPTS \
 697         $tmp
 698
 699     rm -fr "$tmp"
 700     pop_cleanup
 701 }
 702
 703 #################### Create USB CRAMFS based image
 704 function build_usb_cramfs() {
 705     local usb="$1" ; shift
 706     local custom="$1"
 707
 708     prepare_cramfs "$custom"
 709     echo "* Creating USB CRAMFS based image"
 710
 711     let vfat_size=${cramfs_size}+$FREE_SPACE
 712
 713     # Make VFAT filesystem for USB
 714     mkfs.vfat -C "$usb" $vfat_size
 715
 716     # Populate it
 717     echo "* Populating USB with overlay images and cramfs"
 718     mcopy -bsQ -i "$usb" $ISOFS/kernel $ISOFS/pl_version ::/
 719     mcopy -bsQ -i "$usb" ${BUILDTMP}/cramfs.img ::/
 720
 721     # Use syslinux instead of isolinux to make the image bootable
 722     tmp="${BUILDTMP}/syslinux.cfg"
 723     cat >$tmp <<EOF
 724 ${console_serial_line}
 725 PROMPT 0
 726 DEFAULT planetlab-bootcd
 727
 728 LABEL planetlab-bootcd
 729   DISPLAY pl_version
 730   LINUX kernel
 731   APPEND ramdisk_size=$ramdisk_size initrd=cramfs.img root=/dev/ram0 rw ${KERNEL_ARGS}
 732 EOF
 733
 734     mcopy -bsQ -i "$usb" "$tmp" ::/syslinux.cfg
 735     rm -f "$tmp"
 736
 737     echo "* Making USB CRAMFS based image bootable"
 738     syslinux "$usb"
 739 }
 740
 741 #################### map on all types provided on the command-line and invoke one of the above functions
 742 function build_types () {
 743
 744     [ -z "$OUTPUT_BASE" ] && OUTPUT_BASE="$PLC_NAME-BootCD-$BOOTCD_VERSION"
 745
 746     # alter output filename to reflect serial settings
 747     if [ -n "$IS_SERIAL" ] ; then
 748         if [ "$CONSOLE_INFO" == "$SERIAL_CONSOLE" ] ; then
 749             serial="-serial"
 750         else
 751             serial="-serial-$(echo $CONSOLE_INFO | sed -e 's,:,,g')"
 752         fi
 753     else
 754         serial=""
 755     fi
 756
 757     function type_to_name() {
 758         echo $1 | sed '
 759         s/usb$/.usb/;
 760         s/usb_partition$/-partition.usb/;
 761         s/iso$/.iso/;
 762         s/usb_cramfs$/-cramfs.usb/;
 763         s/iso_cramfs$/-cramfs.iso/;
 764         '
 765     }
 766
 767     for t in $TYPES; do
 768         arg=$t
 769
 770         tname=`type_to_name $t`
 771         # if -o is specified (as it has no default)
 772         if [ -n "$OUTPUT_NAME" ] ; then
 773             output=$OUTPUT_NAME
 774         else
 775             output="${OUTPUT_BASE}${serial}${tname}"
 776         fi
 777
 778         echo "*** Dealing with type=$arg"
 779         echo '*' build_$t "$output" "$CUSTOM_DIR"
 780         [ -n "$DRY_RUN" ] || build_$t "$output" "$CUSTOM_DIR"
 781     done
 782 }
 783
 784 ####################
 785 function main () {
 786
 787     parse_command_line "$@"
 788
 789     init_and_check
 790
 791     echo "* Building bootcd images for $NODE_CONFIGURATION_FILE ($FULL_VERSION_STRING) - $(date +%H-%M:%S)"
 792     # Do not tolerate errors
 793     set -e
 794     trap "do_cleanup" ERR INT EXIT
 795
 796     init_serial $CONSOLE_INFO
 797     build_overlay
 798     build_types
 799
 800     echo "* Done with bootcd images for $NODE_CONFIGURATION_FILE - $(date +%H-%M:%S)"
 801     exit 0
 802 }
 803
 804 ####################
 805 main "$@"