tools/tools/nanobsd/nanobsd.sh

   1 #!/bin/sh
   2 #
   3 # Copyright (c) 2005 Poul-Henning Kamp.
   4 # All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions
   8 # are met:
   9 # 1. Redistributions of source code must retain the above copyright
  10 #    notice, this list of conditions and the following disclaimer.
  11 # 2. Redistributions in binary form must reproduce the above copyright
  12 #    notice, this list of conditions and the following disclaimer in the
  13 #    documentation and/or other materials provided with the distribution.
  14 #
  15 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  19 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25 # SUCH DAMAGE.
  26 #
  27 # $FreeBSD$
  28 #
  29
  30 set -e
  31
  32 #######################################################################
  33 #
  34 # Setup default values for all controlling variables.
  35 # These values can be overridden from the config file(s)
  36 #
  37 #######################################################################
  38
  39 # Name of this NanoBSD build.  (Used to construct workdir names)
  40 NANO_NAME=full
  41
  42 # Source tree directory
  43 NANO_SRC=/usr/src
  44
  45 # Where nanobsd additional files live under the source tree
  46 NANO_TOOLS=tools/tools/nanobsd
  47
  48 # Where cust_pkg() finds packages to install
  49 NANO_PACKAGE_DIR=${NANO_SRC}/${NANO_TOOLS}/Pkg
  50 NANO_PACKAGE_LIST="*"
  51
  52 # Object tree directory
  53 # default is subdir of /usr/obj
  54 #NANO_OBJ=""
  55
  56 # The directory to put the final images
  57 # default is ${NANO_OBJ}
  58 #NANO_DISKIMGDIR=""
  59
  60 # Parallel Make
  61 NANO_PMAKE="make -j 3"
  62
  63 # The default name for any image we create.
  64 NANO_IMGNAME="_.disk.full"
  65
  66 # Options to put in make.conf during buildworld only
  67 CONF_BUILD=' '
  68
  69 # Options to put in make.conf during installworld only
  70 CONF_INSTALL=' '
  71
  72 # Options to put in make.conf during both build- & installworld.
  73 CONF_WORLD=' '
  74
  75 # Kernel config file to use
  76 NANO_KERNEL=GENERIC
  77
  78 # Customize commands.
  79 NANO_CUSTOMIZE=""
  80
  81 # Late customize commands.
  82 NANO_LATE_CUSTOMIZE=""
  83
  84 # Newfs paramters to use
  85 NANO_NEWFS="-b 4096 -f 512 -i 8192 -O1 -U"
  86
  87 # The drive name of the media at runtime
  88 NANO_DRIVE=ad0
  89
  90 # Target media size in 512 bytes sectors
  91 NANO_MEDIASIZE=1200000
  92
  93 # Number of code images on media (1 or 2)
  94 NANO_IMAGES=2
  95
  96 # 0 -> Leave second image all zeroes so it compresses better.
  97 # 1 -> Initialize second image with a copy of the first
  98 NANO_INIT_IMG2=1
  99
 100 # Size of code file system in 512 bytes sectors
 101 # If zero, size will be as large as possible.
 102 NANO_CODESIZE=0
 103
 104 # Size of configuration file system in 512 bytes sectors
 105 # Cannot be zero.
 106 NANO_CONFSIZE=2048
 107
 108 # Size of data file system in 512 bytes sectors
 109 # If zero: no partition configured.
 110 # If negative: max size possible
 111 NANO_DATASIZE=0
 112
 113 # Size of the /etc ramdisk in 512 bytes sectors
 114 NANO_RAM_ETCSIZE=10240
 115
 116 # Size of the /tmp+/var ramdisk in 512 bytes sectors
 117 NANO_RAM_TMPVARSIZE=10240
 118
 119 # Media geometry, only relevant if bios doesn't understand LBA.
 120 NANO_SECTS=63
 121 NANO_HEADS=16
 122
 123 # boot0 flags/options and configuration
 124 NANO_BOOT0CFG="-o packet -s 1 -m 3"
 125 NANO_BOOTLOADER="boot/boot0sio"
 126
 127 # Backing type of md(4) device
 128 # Can be "file" or "swap"
 129 NANO_MD_BACKING="file"
 130
 131 # Progress Print level
 132 PPLEVEL=3
 133
 134 #######################################################################
 135 # Not a variable at this time
 136
 137 NANO_ARCH=i386
 138
 139 #######################################################################
 140 #
 141 # The functions which do the real work.
 142 # Can be overridden from the config file(s)
 143 #
 144 #######################################################################
 145
 146 clean_build ( ) (
 147         pprint 2 "Clean and create object directory (${MAKEOBJDIRPREFIX})"
 148
 149         if ! rm -rf ${MAKEOBJDIRPREFIX} > /dev/null 2>&1 ; then
 150                 chflags -R noschg ${MAKEOBJDIRPREFIX}
 151                 rm -r ${MAKEOBJDIRPREFIX}
 152         fi
 153         mkdir -p ${MAKEOBJDIRPREFIX}
 154         printenv > ${MAKEOBJDIRPREFIX}/_.env
 155 )
 156
 157 make_conf_build ( ) (
 158         pprint 2 "Construct build make.conf ($NANO_MAKE_CONF_BUILD)"
 159
 160         echo "${CONF_WORLD}" > ${NANO_MAKE_CONF_BUILD}
 161         echo "${CONF_BUILD}" >> ${NANO_MAKE_CONF_BUILD}
 162 )
 163
 164 build_world ( ) (
 165         pprint 2 "run buildworld"
 166         pprint 3 "log: ${MAKEOBJDIRPREFIX}/_.bw"
 167
 168         cd ${NANO_SRC}
 169         env TARGET_ARCH=${NANO_ARCH} ${NANO_PMAKE} \
 170                 __MAKE_CONF=${NANO_MAKE_CONF_BUILD} buildworld \
 171                 > ${MAKEOBJDIRPREFIX}/_.bw 2>&1
 172 )
 173
 174 build_kernel ( ) (
 175         pprint 2 "build kernel ($NANO_KERNEL)"
 176         pprint 3 "log: ${MAKEOBJDIRPREFIX}/_.bk"
 177
 178         if [ -f ${NANO_KERNEL} ] ; then
 179                 cp ${NANO_KERNEL} ${NANO_SRC}/sys/${NANO_ARCH}/conf
 180         fi
 181
 182         (cd ${NANO_SRC};
 183         # unset these just in case to avoid compiler complaints
 184         # when cross-building
 185         unset TARGET_CPUTYPE
 186         unset TARGET_BIG_ENDIAN
 187         env TARGET_ARCH=${NANO_ARCH} ${NANO_PMAKE} buildkernel \
 188                 __MAKE_CONF=${NANO_MAKE_CONF_BUILD} KERNCONF=`basename ${NANO_KERNEL}` \
 189                 > ${MAKEOBJDIRPREFIX}/_.bk 2>&1
 190         )
 191 )
 192
 193 clean_world ( ) (
 194         if [ "${NANO_OBJ}" != "${MAKEOBJDIRPREFIX}" ]; then
 195                 pprint 2 "Clean and create object directory (${NANO_OBJ})"
 196                 if ! rm -rf ${NANO_OBJ} > /dev/null 2>&1 ; then
 197                         chflags -R noschg ${NANO_OBJ}
 198                         rm -r ${NANO_OBJ}
 199                 fi
 200                 mkdir -p ${NANO_OBJ} ${NANO_WORLDDIR}
 201                 printenv > ${NANO_OBJ}/_.env
 202         else
 203                 pprint 2 "Clean and create world directory (${NANO_WORLDDIR})"
 204                 if ! rm -rf ${NANO_WORLDDIR}/ > /dev/null 2>&1 ; then
 205                         chflags -R noschg ${NANO_WORLDDIR}
 206                         rm -rf ${NANO_WORLDDIR}
 207                 fi
 208                 mkdir -p ${NANO_WORLDDIR}
 209         fi
 210 )
 211
 212 make_conf_install ( ) (
 213         pprint 2 "Construct install make.conf ($NANO_MAKE_CONF_INSTALL)"
 214
 215         echo "${CONF_WORLD}" > ${NANO_MAKE_CONF_INSTALL}
 216         echo "${CONF_INSTALL}" >> ${NANO_MAKE_CONF_INSTALL}
 217 )
 218
 219 install_world ( ) (
 220         pprint 2 "installworld"
 221         pprint 3 "log: ${NANO_OBJ}/_.iw"
 222
 223         cd ${NANO_SRC}
 224         env TARGET_ARCH=${NANO_ARCH} \
 225         ${NANO_PMAKE} __MAKE_CONF=${NANO_MAKE_CONF_INSTALL} installworld \
 226                 DESTDIR=${NANO_WORLDDIR} \
 227                 > ${NANO_OBJ}/_.iw 2>&1
 228         chflags -R noschg ${NANO_WORLDDIR}
 229 )
 230
 231 install_etc ( ) (
 232
 233         pprint 2 "install /etc"
 234         pprint 3 "log: ${NANO_OBJ}/_.etc"
 235
 236         cd ${NANO_SRC}
 237         env TARGET_ARCH=${NANO_ARCH} \
 238         ${NANO_PMAKE} __MAKE_CONF=${NANO_MAKE_CONF_INSTALL} distribution \
 239                 DESTDIR=${NANO_WORLDDIR} \
 240                 > ${NANO_OBJ}/_.etc 2>&1
 241 )
 242
 243 install_kernel ( ) (
 244         pprint 2 "install kernel"
 245         pprint 3 "log: ${NANO_OBJ}/_.ik"
 246
 247         cd ${NANO_SRC}
 248         env TARGET_ARCH=${NANO_ARCH} ${NANO_PMAKE} installkernel \
 249                 DESTDIR=${NANO_WORLDDIR} \
 250                 __MAKE_CONF=${NANO_MAKE_CONF_INSTALL} KERNCONF=`basename ${NANO_KERNEL}` \
 251                 > ${NANO_OBJ}/_.ik 2>&1
 252 )
 253
 254 run_customize() (
 255
 256         pprint 2 "run customize scripts"
 257         for c in $NANO_CUSTOMIZE
 258         do
 259                 pprint 2 "customize \"$c\""
 260                 pprint 3 "log: ${NANO_OBJ}/_.cust.$c"
 261                 pprint 4 "`type $c`"
 262                 ( $c ) > ${NANO_OBJ}/_.cust.$c 2>&1
 263         done
 264 )
 265
 266 run_late_customize() (
 267
 268         pprint 2 "run late customize scripts"
 269         for c in $NANO_LATE_CUSTOMIZE
 270         do
 271                 pprint 2 "late customize \"$c\""
 272                 pprint 3 "log: ${NANO_OBJ}/_.late_cust.$c"
 273                 pprint 4 "`type $c`"
 274                 ( $c ) > ${NANO_OBJ}/_.late_cust.$c 2>&1
 275         done
 276 )
 277
 278 setup_nanobsd ( ) (
 279         pprint 2 "configure nanobsd setup"
 280         pprint 3 "log: ${NANO_OBJ}/_.dl"
 281
 282         (
 283         cd ${NANO_WORLDDIR}
 284
 285         # Move /usr/local/etc to /etc/local so that the /cfg stuff
 286         # can stomp on it.  Otherwise packages like ipsec-tools which
 287         # have hardcoded paths under ${prefix}/etc are not tweakable.
 288         if [ -d usr/local/etc ] ; then
 289                 (
 290                 mkdir -p etc/local
 291                 cd usr/local/etc
 292                 find . -print | cpio -dumpl ../../../etc/local
 293                 cd ..
 294                 rm -rf etc
 295                 ln -s ../../etc/local etc
 296                 )
 297         fi
 298
 299         for d in var etc
 300         do
 301                 # link /$d under /conf
 302                 # we use hard links so we have them both places.
 303                 # the files in /$d will be hidden by the mount.
 304                 # XXX: configure /$d ramdisk size
 305                 mkdir -p conf/base/$d conf/default/$d
 306                 find $d -print | cpio -dumpl conf/base/
 307         done
 308
 309         echo "$NANO_RAM_ETCSIZE" > conf/base/etc/md_size
 310         echo "$NANO_RAM_TMPVARSIZE" > conf/base/var/md_size
 311
 312         # pick up config files from the special partition
 313         echo "mount -o ro /dev/${NANO_DRIVE}s3" > conf/default/etc/remount
 314
 315         # Put /tmp on the /var ramdisk (could be symlink already)
 316         rmdir tmp || true
 317         rm tmp || true
 318         ln -s var/tmp tmp
 319
 320         ) > ${NANO_OBJ}/_.dl 2>&1
 321 )
 322
 323 setup_nanobsd_etc ( ) (
 324         pprint 2 "configure nanobsd /etc"
 325
 326         (
 327         cd ${NANO_WORLDDIR}
 328
 329         # create diskless marker file
 330         touch etc/diskless
 331
 332         # Make root filesystem R/O by default
 333         echo "root_rw_mount=NO" >> etc/defaults/rc.conf
 334
 335         # save config file for scripts
 336         echo "NANO_DRIVE=${NANO_DRIVE}" > etc/nanobsd.conf
 337
 338         echo "/dev/${NANO_DRIVE}s1a / ufs ro 1 1" > etc/fstab
 339         echo "/dev/${NANO_DRIVE}s3 /cfg ufs rw,noauto 2 2" >> etc/fstab
 340         mkdir -p cfg
 341         )
 342 )
 343
 344 prune_usr() (
 345
 346         # Remove all empty directories in /usr
 347         find ${NANO_WORLDDIR}/usr -type d -depth -print |
 348                 while read d
 349                 do
 350                         rmdir $d > /dev/null 2>&1 || true
 351                 done
 352 )
 353
 354 create_i386_diskimage ( ) (
 355         pprint 2 "build diskimage"
 356         pprint 3 "log: ${NANO_OBJ}/_.di"
 357
 358         (
 359         echo $NANO_MEDIASIZE $NANO_IMAGES \
 360                 $NANO_SECTS $NANO_HEADS \
 361                 $NANO_CODESIZE $NANO_CONFSIZE $NANO_DATASIZE |
 362         awk '
 363         {
 364                 printf "# %s\n", $0
 365
 366                 # size of cylinder in sectors
 367                 cs = $3 * $4
 368
 369                 # number of full cylinders on media
 370                 cyl = int ($1 / cs)
 371
 372                 # output fdisk geometry spec, truncate cyls to 1023
 373                 if (cyl <= 1023)
 374                         print "g c" cyl " h" $4 " s" $3
 375                 else
 376                         print "g c" 1023 " h" $4 " s" $3
 377
 378                 if ($7 > 0) {
 379                         # size of data partition in full cylinders
 380                         dsl = int (($7 + cs - 1) / cs)
 381                 } else {
 382                         dsl = 0;
 383                 }
 384
 385                 # size of config partition in full cylinders
 386                 csl = int (($6 + cs - 1) / cs)
 387
 388                 if ($5 == 0) {
 389                         # size of image partition(s) in full cylinders
 390                         isl = int ((cyl - dsl - csl) / $2)
 391                 } else {
 392                         isl = int (($5 + cs - 1) / cs)
 393                 }
 394
 395                 # First image partition start at second track
 396                 print "p 1 165 " $3, isl * cs - $3
 397                 c = isl * cs;
 398
 399                 # Second image partition (if any) also starts offset one
 400                 # track to keep them identical.
 401                 if ($2 > 1) {
 402                         print "p 2 165 " $3 + c, isl * cs - $3
 403                         c += isl * cs;
 404                 }
 405
 406                 # Config partition starts at cylinder boundary.
 407                 print "p 3 165 " c, csl * cs
 408                 c += csl * cs
 409
 410                 # Data partition (if any) starts at cylinder boundary.
 411                 if ($7 > 0) {
 412                         print "p 4 165 " c, dsl * cs
 413                 } else if ($7 < 0 && $1 > c) {
 414                         print "p 4 165 " c, $1 - c
 415                 } else if ($1 < c) {
 416                         print "Disk space overcommitted by", \
 417                             c - $1, "sectors" > "/dev/stderr"
 418                         exit 2
 419                 }
 420
 421                 # Force slice 1 to be marked active. This is necessary
 422                 # for booting the image from a USB device to work.
 423                 print "a 1"
 424         }
 425         ' > ${NANO_OBJ}/_.fdisk
 426
 427         IMG=${NANO_DISKIMGDIR}/${NANO_IMGNAME}
 428         MNT=${NANO_OBJ}/_.mnt
 429         mkdir -p ${MNT}
 430
 431         if [ "${NANO_MD_BACKING}" = "swap" ] ; then
 432                 MD=`mdconfig -a -t swap -s ${NANO_MEDIASIZE} -x ${NANO_SECTS} \
 433                         -y ${NANO_HEADS}`
 434         else
 435                 echo "Creating md backing file..."
 436                 dd if=/dev/zero of=${IMG} bs=${NANO_SECTS}b \
 437                         count=`expr ${NANO_MEDIASIZE} / ${NANO_SECTS}`
 438                 MD=`mdconfig -a -t vnode -f ${IMG} -x ${NANO_SECTS} \
 439                         -y ${NANO_HEADS}`
 440         fi
 441
 442         trap "df -i ${MNT} ; umount ${MNT} || true ; mdconfig -d -u $MD" 1 2 15 EXIT
 443
 444         fdisk -i -f ${NANO_OBJ}/_.fdisk ${MD}
 445         fdisk ${MD}
 446         # XXX: params
 447         # XXX: pick up cached boot* files, they may not be in image anymore.
 448         boot0cfg -B -b ${NANO_WORLDDIR}/${NANO_BOOTLOADER} ${NANO_BOOT0CFG} ${MD}
 449         bsdlabel -w -B -b ${NANO_WORLDDIR}/boot/boot ${MD}s1
 450         bsdlabel ${MD}s1
 451
 452         # Create first image
 453         newfs ${NANO_NEWFS} /dev/${MD}s1a
 454         mount /dev/${MD}s1a ${MNT}
 455         df -i ${MNT}
 456         echo "Copying worlddir..."
 457         ( cd ${NANO_WORLDDIR} && find . -print | cpio -dump ${MNT} )
 458         df -i ${MNT}
 459         echo "Generating mtree..."
 460         ( cd ${MNT} && mtree -c ) > ${NANO_OBJ}/_.mtree
 461         ( cd ${MNT} && du -k ) > ${NANO_OBJ}/_.du
 462         umount ${MNT}
 463
 464         if [ $NANO_IMAGES -gt 1 -a $NANO_INIT_IMG2 -gt 0 ] ; then
 465                 # Duplicate to second image (if present)
 466                 echo "Duplicating to second image..."
 467                 dd if=/dev/${MD}s1 of=/dev/${MD}s2 bs=64k
 468                 mount /dev/${MD}s2a ${MNT}
 469                 for f in ${MNT}/etc/fstab ${MNT}/conf/base/etc/fstab
 470                 do
 471                         sed -i "" "s/${NANO_DRIVE}s1/${NANO_DRIVE}s2/g" $f
 472                 done
 473                 umount ${MNT}
 474         fi
 475
 476         # Create Config slice
 477         newfs ${NANO_NEWFS} /dev/${MD}s3
 478         # XXX: fill from where ?
 479
 480         # Create Data slice, if any.
 481         if [ $NANO_DATASIZE -ne 0 ] ; then
 482                 newfs ${NANO_NEWFS} /dev/${MD}s4
 483                 # XXX: fill from where ?
 484         fi
 485
 486         if [ "${NANO_MD_BACKING}" = "swap" ] ; then
 487                 echo "Writing out ${NANO_IMGNAME}..."
 488                 dd if=/dev/${MD} of=${IMG} bs=64k
 489         fi
 490
 491         echo "Writing out _.disk.image..."
 492         dd if=/dev/${MD}s1 of=${NANO_DISKIMGDIR}/_.disk.image bs=64k
 493         mdconfig -d -u $MD
 494         ) > ${NANO_OBJ}/_.di 2>&1
 495 )
 496
 497 last_orders () (
 498         # Redefine this function with any last orders you may have
 499         # after the build completed, for instance to copy the finished
 500         # image to a more convenient place:
 501         # cp ${NANO_DISKIMGDIR}/_.disk.image /home/ftp/pub/nanobsd.disk
 502 )
 503
 504 #######################################################################
 505 #
 506 # Optional convenience functions.
 507 #
 508 #######################################################################
 509
 510 #######################################################################
 511 # Common Flash device geometries
 512 #
 513
 514 FlashDevice () {
 515         if [ -d ${NANO_TOOLS} ] ; then
 516                 . ${NANO_TOOLS}/FlashDevice.sub
 517         else
 518                 . ${NANO_SRC}/${NANO_TOOLS}/FlashDevice.sub
 519         fi
 520         sub_FlashDevice $1 $2
 521 }
 522
 523 #######################################################################
 524 # USB device geometries
 525 #
 526 # Usage:
 527 #       UsbDevice Generic 1000  # a generic flash key sold as having 1GB
 528 #
 529 # This function will set NANO_MEDIASIZE, NANO_HEADS and NANO_SECTS for you.
 530 #
 531 # Note that the capacity of a flash key is usually advertised in MB or
 532 # GB, *not* MiB/GiB. As such, the precise number of cylinders available
 533 # for C/H/S geometry may vary depending on the actual flash geometry.
 534 #
 535 # The following generic device layouts are understood:
 536 #  generic           An alias for generic-hdd.
 537 #  generic-hdd       255H 63S/T xxxxC with no MBR restrictions.
 538 #  generic-fdd       64H 32S/T xxxxC with no MBR restrictions.
 539 #
 540 # The generic-hdd device is preferred for flash devices larger than 1GB.
 541 #
 542
 543 UsbDevice () {
 544         a1=`echo $1 | tr '[:upper:]' '[:lower:]'`
 545         case $a1 in
 546         generic-fdd)
 547                 NANO_HEADS=64
 548                 NANO_SECTS=32
 549                 NANO_MEDIASIZE=$(( $2 * 1000 * 1000 / 512 ))
 550                 ;;
 551         generic|generic-hdd)
 552                 NANO_HEADS=255
 553                 NANO_SECTS=63
 554                 NANO_MEDIASIZE=$(( $2 * 1000 * 1000 / 512 ))
 555                 ;;
 556         *)
 557                 echo "Unknown USB flash device"
 558                 exit 2
 559                 ;;
 560         esac
 561 }
 562
 563 #######################################################################
 564 # Setup serial console
 565
 566 cust_comconsole () (
 567         # Enable getty on console
 568         sed -i "" -e /tty[du]0/s/off/on/ ${NANO_WORLDDIR}/etc/ttys
 569
 570         # Disable getty on syscons devices
 571         sed -i "" -e '/^ttyv[0-8]/s/    on/     off/' ${NANO_WORLDDIR}/etc/ttys
 572
 573         # Tell loader to use serial console early.
 574         echo " -h" > ${NANO_WORLDDIR}/boot.config
 575 )
 576
 577 #######################################################################
 578 # Allow root login via ssh
 579
 580 cust_allow_ssh_root () (
 581         sed -i "" -e '/PermitRootLogin/s/.*/PermitRootLogin yes/' \
 582             ${NANO_WORLDDIR}/etc/ssh/sshd_config
 583 )
 584
 585 #######################################################################
 586 # Install the stuff under ./Files
 587
 588 cust_install_files () (
 589         cd ${NANO_TOOLS}/Files
 590         find . -print | grep -Ev '/(CVS|\.svn)' | cpio -dumpv ${NANO_WORLDDIR}
 591 )
 592
 593 #######################################################################
 594 # Install packages from ${NANO_PACKAGE_DIR}
 595
 596 cust_pkg () (
 597
 598         # Copy packages into chroot
 599         mkdir -p ${NANO_WORLDDIR}/Pkg
 600         (
 601                 cd ${NANO_PACKAGE_DIR}
 602                 find ${NANO_PACKAGE_LIST} -print |
 603                     cpio -dumpv ${NANO_WORLDDIR}/Pkg
 604         )
 605
 606         # Count & report how many we have to install
 607         todo=`ls ${NANO_WORLDDIR}/Pkg | wc -l`
 608         echo "=== TODO: $todo"
 609         ls ${NANO_WORLDDIR}/Pkg
 610         echo "==="
 611         while true
 612         do
 613                 # Record how many we have now
 614                 have=`ls ${NANO_WORLDDIR}/var/db/pkg | wc -l`
 615
 616                 # Attempt to install more packages
 617                 # ...but no more than 200 at a time due to pkg_add's internal
 618                 # limitations.
 619                 chroot ${NANO_WORLDDIR} sh -c \
 620                         'ls Pkg/*tbz | xargs -n 200 pkg_add -F' || true
 621
 622                 # See what that got us
 623                 now=`ls ${NANO_WORLDDIR}/var/db/pkg | wc -l`
 624                 echo "=== NOW $now"
 625                 ls ${NANO_WORLDDIR}/var/db/pkg
 626                 echo "==="
 627
 628
 629                 if [ $now -eq $todo ] ; then
 630                         echo "DONE $now packages"
 631                         break
 632                 elif [ $now -eq $have ] ; then
 633                         echo "FAILED: Nothing happened on this pass"
 634                         exit 2
 635                 fi
 636         done
 637         rm -rf ${NANO_WORLDDIR}/Pkg
 638 )
 639
 640 #######################################################################
 641 # Convenience function:
 642 #       Register $1 as customize function.
 643
 644 customize_cmd () {
 645         NANO_CUSTOMIZE="$NANO_CUSTOMIZE $1"
 646 }
 647
 648 #######################################################################
 649 # Convenience function:
 650 #       Register $1 as late customize function to run just before
 651 #       image creation.
 652
 653 late_customize_cmd () {
 654         NANO_LATE_CUSTOMIZE="$NANO_LATE_CUSTOMIZE $1"
 655 }
 656
 657 #######################################################################
 658 #
 659 # All set up to go...
 660 #
 661 #######################################################################
 662
 663 # Progress Print
 664 #       Print $2 at level $1.
 665 pprint() {
 666     if [ "$1" -le $PPLEVEL ]; then
 667         runtime=$(( `date +%s` - $NANO_STARTTIME ))
 668         printf "%s %.${1}s %s\n" "`date -u -r $runtime +%H:%M:%S`" "#####" "$2" 1>&3
 669     fi
 670 }
 671
 672 usage () {
 673         (
 674         echo "Usage: $0 [-bikqvw] [-c config_file]"
 675         echo "  -b      suppress builds (both kernel and world)"
 676         echo "  -i      suppress disk image build"
 677         echo "  -k      suppress buildkernel"
 678         echo "  -n      add -DNO_CLEAN to buildworld, buildkernel, etc"
 679         echo "  -q      make output more quite"
 680         echo "  -v      make output more verbose"
 681         echo "  -w      suppress buildworld"
 682         echo "  -c      specify config file"
 683         ) 1>&2
 684         exit 2
 685 }
 686
 687 #######################################################################
 688 # Parse arguments
 689
 690 do_clean=true
 691 do_kernel=true
 692 do_world=true
 693 do_image=true
 694
 695 set +e
 696 args=`getopt bc:hiknqvw $*`
 697 if [ $? -ne 0 ] ; then
 698         usage
 699         exit 2
 700 fi
 701 set -e
 702
 703 set -- $args
 704 for i
 705 do
 706         case "$i"
 707         in
 708         -b)
 709                 do_world=false
 710                 do_kernel=false
 711                 shift
 712                 ;;
 713         -k)
 714                 do_kernel=false
 715                 shift
 716                 ;;
 717         -c)
 718                 . "$2"
 719                 shift
 720                 shift
 721                 ;;
 722         -h)
 723                 usage
 724                 ;;
 725         -i)
 726                 do_image=false
 727                 shift
 728                 ;;
 729         -n)
 730                 do_clean=false
 731                 shift
 732                 ;;
 733         -q)
 734                 PPLEVEL=$(($PPLEVEL - 1))
 735                 shift
 736                 ;;
 737         -v)
 738                 PPLEVEL=$(($PPLEVEL + 1))
 739                 shift
 740                 ;;
 741         -w)
 742                 do_world=false
 743                 shift
 744                 ;;
 745         --)
 746                 shift
 747                 break
 748         esac
 749 done
 750
 751 if [ $# -gt 0 ] ; then
 752         echo "$0: Extraneous arguments supplied"
 753         usage
 754 fi
 755
 756 #######################################################################
 757 # Setup and Export Internal variables
 758 #
 759 test -n "${NANO_OBJ}" || NANO_OBJ=/usr/obj/nanobsd.${NANO_NAME}/
 760 test -n "${MAKEOBJDIRPREFIX}" || MAKEOBJDIRPREFIX=${NANO_OBJ}
 761 test -n "${NANO_DISKIMGDIR}" || NANO_DISKIMGDIR=${NANO_OBJ}
 762
 763 NANO_WORLDDIR=${NANO_OBJ}/_.w
 764 NANO_MAKE_CONF_BUILD=${MAKEOBJDIRPREFIX}/make.conf.build
 765 NANO_MAKE_CONF_INSTALL=${NANO_OBJ}/make.conf.install
 766
 767 if [ -d ${NANO_TOOLS} ] ; then
 768         true
 769 elif [ -d ${NANO_SRC}/${NANO_TOOLS} ] ; then
 770         NANO_TOOLS=${NANO_SRC}/${NANO_TOOLS}
 771 else
 772         echo "NANO_TOOLS directory does not exist" 1>&2
 773         exit 1
 774 fi
 775
 776 if $do_clean ; then
 777         true
 778 else
 779         NANO_PMAKE="${NANO_PMAKE} -DNO_CLEAN"
 780 fi
 781
 782 export MAKEOBJDIRPREFIX
 783
 784 export NANO_ARCH
 785 export NANO_CODESIZE
 786 export NANO_CONFSIZE
 787 export NANO_CUSTOMIZE
 788 export NANO_DATASIZE
 789 export NANO_DRIVE
 790 export NANO_HEADS
 791 export NANO_IMAGES
 792 export NANO_IMGNAME
 793 export NANO_MAKE_CONF_BUILD
 794 export NANO_MAKE_CONF_INSTALL
 795 export NANO_MEDIASIZE
 796 export NANO_NAME
 797 export NANO_NEWFS
 798 export NANO_OBJ
 799 export NANO_PMAKE
 800 export NANO_SECTS
 801 export NANO_SRC
 802 export NANO_TOOLS
 803 export NANO_WORLDDIR
 804 export NANO_BOOT0CFG
 805 export NANO_BOOTLOADER
 806
 807 #######################################################################
 808 # And then it is as simple as that...
 809
 810 # File descriptor 3 is used for logging output, see pprint
 811 exec 3>&1
 812
 813 NANO_STARTTIME=`date +%s`
 814 pprint 1 "NanoBSD image ${NANO_NAME} build starting"
 815
 816 if $do_world ; then
 817         if $do_clean ; then
 818                 clean_build
 819         else
 820                 pprint 2 "Using existing build tree (as instructed)"
 821         fi
 822         make_conf_build
 823         build_world
 824 else
 825         pprint 2 "Skipping buildworld (as instructed)"
 826 fi
 827
 828 if $do_kernel ; then
 829         build_kernel
 830 else
 831         pprint 2 "Skipping buildkernel (as instructed)"
 832 fi
 833
 834 clean_world
 835 make_conf_install
 836 install_world
 837 install_etc
 838 setup_nanobsd_etc
 839 install_kernel
 840
 841 run_customize
 842 setup_nanobsd
 843 prune_usr
 844 run_late_customize
 845 if $do_image ; then
 846         create_${NANO_ARCH}_diskimage
 847 else
 848         pprint 2 "Skipping image build (as instructed)"
 849 fi
 850 last_orders
 851
 852 pprint 1 "NanoBSD image ${NANO_NAME} completed"