UnivNautes

#!/bin/sh

#

# Common functions to be used by build scripts

#

#  build_iso.sh

#  Copyright (C) 2004-2009 Scott Ullrich

#  All rights reserved.

#  

#  Redistribution and use in source and binary forms, with or without

#  modification, are permitted provided that the following conditions are met:

#  

#  1. Redistributions of source code must retain the above copyright notice,

#     this list of conditions and the following disclaimer.

#  

#  2. Redistributions in binary form must reproduce the above copyright

#     notice, this list of conditions and the following disclaimer in the

#     documentation and/or other materials provided with the distribution.

#  

#  THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,

#  INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY

#  AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE

#  AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,

#  OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

#  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

#  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

#  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

#  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

#  POSSIBILITY OF SUCH DAMAGE.

#

# Crank up error reporting, debugging.

#  set -e 

#  set -x

set +e

usage() {

	echo "Usage $0 [options] [ iso | nanobsd | ova | nanobsd-vga | memstick | memstickserial | fullupdate | all ]"

	echo "		all = iso nanobsd nanobsd-vga memstick memstickserial fullupdate"

	echo "	[ options ]: "

	echo "		--flash-size|-f size(s) - a list of flash sizes to build with nanobsd i.e. '512m 1g'. Default: 512m"

	echo "		--flash-vendor|-m vendor - flash vendor type. check the code for supported vendors. Default: sandisk"

	echo "		--no-buildworld|-c - Will set NO_BUILDWORLD NO_BUILDKERNEL to not build kernel and world"

	echo "		--no-cleanobjdir|--no-cleanrepos|-d - Will not clean FreeBSD object built dir to allow restarting a build with NO_CLEAN"

	echo "		--resume-image-build|-r - Includes -c -d and also will just move directly to image creation using pre-staged data"

	echo "		--apply-patches - Fetch FreeBSD sources and apply patches"

	echo "		--build-pfPorts - Rebuild all ports required to generate an image"

	echo "		--build-pfPort port - Rebuild a single port required to generate an image"

	echo "		--builder-required-ports - Build ports required from builder tools to operate"

	echo "		--update-sources - Refetch all sources"

	echo "		--print-flags - Show current builder configuration"

	echo "		--clean-builder - clean all builder used data/resources"

	echo "		--configure - generate a config file to be used by builder tools"

	echo "		--build-kernels - build all configured kernels"

	echo "		--build-kernel argument - build specifed kernel. Exampe --build-kernel pfSense_SMP.10"

	echo "		--install-extra-kernels argument - Put extra kernel(s) under /kernel image directory. Exampe --install-extra-kernels pfSense_wrap.10.i386"

	echo "		--enable-memorydisks - This will put $PFSENSESTAGEDIR and $PFSENSEISODIR as MFS filesystems"

	echo "		--disable-memorydisks - Will just teardown these filesystsm created by --enable-memorydisks"

	exit 1

}

unset _SKIP_REBUILD_PRESTAGE

unset pfPORTTOBUILD

unset IMAGETYPE

BUILDACTION="images"

# Maybe use options for nocleans etc?

while test "$1" != ""; do

        case "${1}" in

	--no-buildworld|-c)

		export NO_BUILDWORLD=YES

		export NO_BUILDKERNEL=YES

	;;

	--no-cleanobjdir|--no-cleanrepos|-d)

		export NO_CLEANFREEBSDOBJDIR=YES

		export NO_CLEANREPOS=YES

	;;

	--flash-size|-f)

		shift

		if [ $# -eq 0 ]; then

			echo "--flash-model needs extra parameter."

			echo

			usage

		fi

		export FLASH_SIZE="${1}"

	;;

	--flash-model|-m)

		shift

		if [ $# -eq 0 ]; then

			echo "--flash-model needs extra parameter."

			echo

			usage

		fi

		export FLASH_MODEL="${1}"

	;;

	--resume-image-build|-r)

		export NO_BUILDWORLD=YES

                export NO_BUILDKERNEL=YES

		export NO_CLEANFREEBSDOBJDIR=YES

                export NO_CLEANREPOS=YES

		_SKIP_REBUILD_PRESTAGE=YES

	;;

	--apply-patches)

		BUILDACTION="applypatches"

	;;

	--build-pfPorts)

		BUILDACTION="buildpfports"

	;;

	--build-pfPort)

		BUILDACTION="buildpfport"

		shift

		if [ $# -eq 0 ]; then

			echo "--build-pfPort needs extra parameter."

			echo 

			usage

		fi

		pfPORTTOBUILD="$1"

	;;

	--builder-required-ports)

		BUILDACTION="builderrequiredports"

	;;

	--build-kernels)

		BUILDACTION="buildkernels"

	;;

	--install-extra-kernels)

		shift

		if [ $# -eq 0 ]; then

			echo "--build-kernel needs extra parameter."

			echo 

			usage

		fi

		export INSTALL_EXTRA_KERNELS="${1}"

	;;

	--build-kernel)

		BUILDACTION="buildkernel"

		shift

		if [ $# -eq 0 ]; then

			echo "--build-kernel needs extra parameter."

			echo 

			usage

		fi

		export BUILD_KERNELS="${1}"

	;;

	--update-sources)

		BUILDACTION="updatesources"

	;;

	--print-flags)

		BUILDACTION="printflags"

	;;

	--clean-builder)

		BUILDACTION="cleanbuilder"

	;;

	--configure)

		BUILDACTION="configure"

	;;

	--test-tweet)

		BUILDACTION="testtweet"

	;;

	--enable-memorydisks)

		BUILDACTION="enablememorydisk"

	;;

	--disable-memorydisks)

		BUILDACTION="disablememorydisk"

	;;

	all|*iso*|*ova*|*memstick*|*memstickserial*|*nanobsd*|*nanobsd-vga*|*fullupdate*)

		BUILDACTION="images"

		IMAGETYPE="${1}"

	;;

	*)

		usage

        esac

	shift

done

# Suck in local vars

. ./builder_defaults.sh

_TMPPWD=$(cd .. && basename `pwd`)

if [ "${TOOLS_DIR}" != "${_TMPPWD}" ]; then

	echo "ERROR: builder tools checkout on ${_TMPPWD} not the same as TOOLS_DIR=${TOOLS_DIR} setting."

	exit 1

fi

unset _TMPPWD

# Suck in script helper functions

. ./builder_common.sh

# Define src.conf

if [ -z "${SRC_CONF:-}" ]; then

	export SRC_CONF="${BUILDER_SCRIPTS}/conf/src/src.conf.$FREEBSD_VERSION"

	export SRC_CONF_INSTALL="${BUILDER_SCRIPTS}/conf/src/src.conf.$FREEBSD_VERSION.install"

fi

if [ ! -z "${CUSTOM_REMOVE_LIST:-}" ]; then

	export PRUNE_LIST="${CUSTOM_REMOVE_LIST:-}"

else

	export PRUNE_LIST="${BUILDER_SCRIPTS}/conf/rmlist/remove.list.iso.$FREEBSD_VERSION"

fi

# This should be run first

launch

case $BUILDACTION in

	applypatches)

		update_freebsd_sources_and_apply_patches

	;;

	buildpfports)

		recompile_pfPorts

	;;

	buildpfport)

		recompile_pfPorts $pfPORTTOBUILD

	;;

	builderrequiredports)

		install_required_builder_system_ports

	;;

	buildkernels)

		update_freebsd_sources_and_apply_patches

		build_all_kernels

	;;

	buildkernel)

		update_freebsd_sources_and_apply_patches

		build_all_kernels

	;;

	configure)

		./scripts/menu.sh

	;;

	cleanbuilder)

		pfSense_clean_obj_dir

	;;

	printflags)

		print_flags

	;;

	images)

		# It will be handled below

	;;

	testtweet)

		post_tweet "Robots need loving too."

	;;

	updatesources)

		ensure_source_directories_present

	;;

	enablememorydisk)

		prestage_on_ram_setup

	;;

	disablememorydisk)

		prestage_on_ram_cleanup

	;;

	*)

		usage

	;;

esac

if [ "${BUILDACTION}" != "images" ]; then

	# Email that the operation has completed

	email_operation_completed

	finish

	exit 0

fi

if [ $# -gt 1 ]; then

	echo "ERROR: Too many argument given."

	echo

	usage

fi

if [ -z "${IMAGETYPE}" ]; then

	echo "ERROR: Need to specify image type to build."

	echo

	usage

fi

if [ "$IMAGETYPE" = "all" ]; then

	_IMAGESTOBUILD="iso fullupdate nanobsd nanobsd-vga memstick memstickserial"

else

	_IMAGESTOBUILD="${IMAGETYPE}"

fi

echo ">>> Building image type(s): ${_IMAGESTOBUILD}"

if [ -z "${_SKIP_REBUILD_PRESTAGE}" ]; then

	# Cleanup environment before start

	pfSense_clean_obj_dir

	# Make sure source directories are present.

	ensure_source_directories_present

	git_last_commit

	# Ensure binaries are present that builder system requires

	install_required_builder_system_ports

	# Output build flags

	print_flags

	# Check to see if pre-staging will be hosted on ram

	prestage_on_ram_setup

	# Build world, kernel and install

	echo ">>> Building world for ISO... $FREEBSD_VERSION  $FREEBSD_BRANCH ..."

	make_world

	# Build kernels

	echo ">>> Building kernel configs: $BUILD_KERNELS for FreeBSD: $FREEBSD_BRANCH ..."

	build_all_kernels

	# Prepare pre-final staging area

	clone_to_staging_area

	# Check if we need to force a ports rebuild

	check_for_forced_pfPorts_build

	# Install packages needed for livecd

	install_pkg_install_ports

	# Overlay host binaries

	cust_overlay_host_binaries

fi

export DEFAULT_KERNEL=${DEFAULT_KERNEL_ISO:-pfSense_SMP.${FREEBSD_VERSION}}

for _IMGTOBUILD in $_IMAGESTOBUILD; do

	# Clean up items that should be cleaned each run

        staginareas_clean_each_run

	# NOTE: This code is needed to support FreeBSD 8(2.*.*) builds that have extra kernels under /kernels than default

	if [ $FREEBSD_VERSION -lt 10 -a -z "${INSTALL_EXTRA_KERNELS}" ]; then

		if [ "${_IMGTOBUILD}" = "nanobsd-vga" ]; then

			export INSTALL_EXTRA_KERNELS="pfSense_SMP.${FREEBSD_VERSION}"

		else

			export INSTALL_EXTRA_KERNELS="pfSense_wrap.${FREEBSD_VERSION}.${TARGET}"

		fi

	fi

	if [ "${_IMGTOBUILD}" = "iso" ]; then

		(create_iso_image)

	elif [ "${_IMGTOBUILD}" = "memstick" ]; then

		(create_memstick_image)

	elif [ "${_IMGTOBUILD}" = "memstickserial" ]; then

		(create_memstick_serial_image)

	elif [ "${_IMGTOBUILD}" = "fullupdate" ]; then

		create_pfSense_Full_update_tarball

	elif [ "${_IMGTOBUILD}" = "nanobsd" -o "${_IMGTOBUILD}" = "nanobsd-vga" ]; then

		if [ "${TARGET}" = "i386" -a "${_IMGTOBUILD}" = "nanobsd" ]; then

			export DEFAULT_KERNEL=${DEFAULT_KERNEL_NANOBSD:-pfSense_wrap.${FREEBSD_VERSION}.${TARGET}}

		elif [ "${TARGET}" = "i386" -a "${_IMGTOBUILD}" = "nanobsd-vga" ]; then

			export DEFAULT_KERNEL=${DEFAULT_KERNEL_NANOBSDVGA:-pfSense_wrap_vga.${FREEBSD_VERSION}.${TARGET}}

		elif [ "${TARGET}" = "amd64" ]; then

			export DEFAULT_KERNEL=${DEFAULT_KERNEL_NANOBSD:-pfSense_SMP.${FREEBSD_VERSION}}

		fi

		# Create the NanoBSD disk image for i386

		if [ "$TARGET" = "i386" -o "$TARGET" = "amd64" ]; then

			create_nanobsd_diskimage ${_IMGTOBUILD} "${FLASH_SIZE}"

		elif [ "$TARGET" = "mips" ]; then

			# Create the NanoBSD disk image for mips

			export DEFAULT_KERNEL=${DEFAULT_KERNEL_MIPS:-pfSense_SMP.${FREEBSD_VERSION}}

			create_mips_diskimage ${_IMGTOBUILD} ${FLASH_SIZE}

		fi

	elif [ "${_IMGTOBUILD}" = "ova" ]; then

		(create_ova_image)

	fi

done

echo ">>> NOTE: waiting for jobs: `jobs -l` to finish..."

wait

# Email that the operation has completed

email_operation_completed

echo ">>> $MAKEOBJDIRPREFIXFINAL now contains:"

ls -lah $MAKEOBJDIRPREFIXFINAL

set -e

# Run final finish routines

finish