build.sh

#!/bin/bash

set -eu

SCRIPTS_DIR=$(cd `dirname $0`; pwd)
if [ -h $0 ]
then
        CMD=$(readlink $0)
        SCRIPTS_DIR=$(dirname $CMD)
fi
cd $SCRIPTS_DIR
cd ../
TOP_DIR=$(pwd)

SDFUSE_DIR=$TOP_DIR/scripts/sd-fuse

firsttime_usage()
{
	echo ""
	echo "# select board: "
	ALL_MK=`find ./device/friendlyelec -type f -name "*.mk" -printf "%f\n"`
	for mk in ${ALL_MK}; do
		echo "  ./build.sh $mk"
	done
	ALL_MK_LINK=`find ./device/friendlyelec -type l -name "*.mk" -printf "%f\n"`
        for mk in ${ALL_MK_LINK}; do
                echo "  ./build.sh $mk"
        done
	echo ""
}

usage()
{
	echo "USAGE: ./build.sh <parameter>"
	firsttime_usage
	echo "# build module: "
	echo "  ./build.sh all                -build all"
	echo "  ./build.sh uboot              -build uboot only"
	echo "  ./build.sh kernel             -build kernel only"
	echo "  ./build.sh friendlywrt        -build friendlywrt rootfs only"
	echo "  ./build.sh sd-img             -pack sd-card image, used to create bootable SD card"
	echo "  ./build.sh emmc-img           -pack sd-card image, used to write friendlywrt to emmc"
    echo "# clean"
    echo "  ./build.sh cleanall"
	echo ""
}

if [ $# -ne 1 ]; then
    usage
    exit 1
fi

function log_error()
{
    local now=`date +%s`
    printf "\033[1;31m[ERROR]: $*\033[0m \n"
}

function log_warn()
{
    local now=`date +%s`
    printf "\033[1;31m[WARN]: $*\033[0m \n"
}

function log_info()
{
    local now=`date +%s`
    printf "\033[1;32m[INFO]: $* \033[0m \n"
}

function build_uboot(){
	# build uboot
	echo "============Start building uboot============"
	echo "SRC                = ${TOP_DIR}/u-boot"
        echo "TARGET_ARCH        = $TARGET_ARCH"
        echo "TARGET_PLAT        = $TARGET_PLAT"
	echo "TARGET_UBOOT_CONFIG=$TARGET_UBOOT_CONFIG"
	echo "========================================="

	(cd ${SDFUSE_DIR} && {
		DISABLE_MKIMG=1 UBOOT_SRC=${TOP_DIR}/u-boot ./build-uboot.sh friendlywrt_4.14_armhf
	})

	if [ $? -eq 0 ]; then
		echo "====Building uboot ok!===="
	else
		echo "====Building uboot failed!===="
		exit 1
	fi
}

function build_kernel(){
	# build kernel
	echo "============Start building kernel============"
	echo "SRC                  = ${TOP_DIR}/kernel"
	echo "TARGET_ARCH          = $TARGET_ARCH"
        echo "TARGET_PLAT          = $TARGET_PLAT"
	echo "TARGET_KERNEL_CONFIG = $TARGET_KERNEL_CONFIG"
	echo "=========================================="

	(cd ${SDFUSE_DIR} && {
		DISABLE_MKIMG=1 KERNEL_SRC=${TOP_DIR}/kernel ./build-kernel.sh friendlywrt_4.14_armhf
	})

	if [ $? -eq 0 ]; then
		echo "====Building kernel ok!===="
	else
		echo "====Building kernel failed!===="
		exit 1
	fi
}


function build_friendlywrt(){
	# build friendlywrt
	echo "==========Start build friendlywrt=========="
	echo "TARGET_FRIENDLYWRT_CONFIG=$TARGET_FRIENDLYWRT_CONFIG"
	echo "FRIENDLYWRT_SRC=$FRIENDLYWRT_SRC"
	echo "========================================="
	/usr/bin/time -f "you take %E to build friendlywrt" $SCRIPTS_DIR/mk-friendlywrt.sh $TARGET_FRIENDLYWRT_CONFIG $FRIENDLYWRT_SRC
	if [ $? -eq 0 ]; then
		echo "====Building friendlywrt ok!===="
	else
		echo "====Building friendlywrt failed!===="
		exit 1
	fi
}

function build_all() {
	build_uboot
	build_kernel
	build_friendlywrt
	build_sdimg
}

function clean_all(){
	echo "clean uboot, kernel, friendlywrt, img files"
	cd $TOP_DIR/u-boot/ && make distclean && cd -
	cd $TOP_DIR/kernel && make distclean && cd -
	cd $TOP_DIR/friendlywrt && 
	cd ${SDFUSE_DIR} && ./clean.sh && cd -
}

function copy_and_verify(){
	if [ ! -f $1 ]; then
        echo "not found: $1"
		echo "$3"
		exit 1
	fi
	cp $1 $2
}

function prepare_image_for_friendlyelec_eflasher(){
    local OS_DIR=$1
    local ROOTFS=$2
    if [ ! -d ${SDFUSE_DIR}/${OS_DIR} ]; then
        mkdir ${SDFUSE_DIR}/${OS_DIR}
    fi
    rm -rf ${SDFUSE_DIR}/${OS_DIR}/*

    # clean
    rm -rf ${SDFUSE_DIR}/out/rootfs.*
    rm -rf ${SDFUSE_DIR}/out/boot.*

    local ROOTFS_DIR=$(mktemp -d ${SDFUSE_DIR}/out/rootfs.XXXXXXXXX)
    log_info "Copying ${TOP_DIR}/${FRIENDLYWRT_ROOTFS} to ${ROOTFS_DIR}/"
    cp -af ${TOP_DIR}/${FRIENDLYWRT_SRC}/${FRIENDLYWRT_ROOTFS}/* ${ROOTFS_DIR}/
    for (( i=0; i<${#FRIENDLYWRT_FILES[@]}; i++ ));
    do
        # apply patch to rootfs
        if [ ! -z ${FRIENDLYWRT_FILES[$i]} ]; then
            log_info "Applying ${FRIENDLYWRT_FILES[$i]} to ${ROOTFS_DIR}"
	    if [ -f ${TOP_DIR}/${FRIENDLYWRT_FILES[$i]}/install.sh ]; then
		(cd ${TOP_DIR}/${FRIENDLYWRT_FILES[$i]} && {
			./install.sh ${ROOTFS_DIR}
		})
	    else
                rsync -a --no-o --no-g --exclude='.git' ${TOP_DIR}/${FRIENDLYWRT_FILES[$i]}/* ${ROOTFS_DIR}/
            fi
        fi
    done

    local BOOT_DIR=$(mktemp -d ${SDFUSE_DIR}/out/boot.XXXXXXXXX)

    # prepare uboot bin, boot.img and rootfs.img
    local UBOOT_DIR=${TOP_DIR}/u-boot
    local KERNEL_DIR=${TOP_DIR}/kernel
    (cd ${SDFUSE_DIR} && {
        ./tools/update_uboot_bin.sh ${UBOOT_DIR} ./${OS_DIR}
	if [ $? -ne 0 ]; then
                log_error "error: fail to copy uboot bin file."
                exit 1
        fi
        ./tools/setup_boot_and_rootfs.sh ${UBOOT_DIR} ${KERNEL_DIR} ${BOOT_DIR} ${ROOTFS_DIR} ./prebuilt
	if [ $? -ne 0 ]; then
                log_error "error: fail to copy kernel to rootfs.img."
                exit 1
        fi

	log_info "prepare boot.img ..."
        ./build-boot-img.sh ${BOOT_DIR} ./${OS_DIR}/boot.img
	if [ $? -ne 0 ]; then
		log_error "error: fail to gen boot.img."
		exit 1
	fi 

	log_info "prepare rootfs.img ..."
        ./build-rootfs-img.sh ${ROOTFS_DIR} ./${OS_DIR}/rootfs.img 0
	if [ $? -ne 0 ]; then
                log_error "error: fail to gen rootfs.img."
                exit 1
        fi

	cat > ./${OS_DIR}/info.conf << EOL
title=${OS_DIR}
require-board=${TARGET_PLAT}
version=$(date +%Y-%m-%d)
EOL
        ./tools/update_prebuilt.sh ./${OS_DIR} ./prebuilt

    })

    # clean
    rm -rf ${ROOTFS_DIR}
    rm -rf ${BOOT_DIR}
}

function clean_device_files()
{
    # create tmp dir
    if [ ! -d ${1}/tmp ]; then
        mkdir ${1}/tmp
    fi
    chmod 1777 ${1}/tmp
    chown root:root ${1}/tmp
    (cd ${1}/dev && find . ! -type d -exec rm {} \;)
}

function build_sdimg(){
    source ${SDFUSE_DIR}/tools/util.sh
    local HAS_BUILT_UBOOT=`has_built_uboot ${TOP_DIR}/u-boot ${SDFUSE_DIR}/out`
    local HAS_BUILD_KERN=`has_built_kernel ${TOP_DIR}/kernel ${SDFUSE_DIR}/out`
    local HAS_BUILD_KERN_MODULES=`has_built_kernel_modules ${TOP_DIR}/kernel ${SDFUSE_DIR}/out`

    # log_info "HAS_BUILT_UBOOT = ${HAS_BUILT_UBOOT}"
    # log_info "HAS_BUILD_KERN = ${HAS_BUILD_KERN}"
    # log_info "HAS_BUILD_KERN_MODULES = ${HAS_BUILD_KERN_MODULES}"

    if [ ${HAS_BUILT_UBOOT} -ne 1 ]; then
        log_error "error: please build u-boot first."
        exit 1
    fi
    
    if [ ${HAS_BUILD_KERN} -ne 1 ]; then
        log_error "error: please build kernel first."
        exit 1
    fi

    if [ ${HAS_BUILD_KERN_MODULES} -ne 1 ]; then
        log_error "error: please build kernel first (miss kernel modules)."
        exit 1
    fi

    local ROOTFS=${TOP_DIR}/${FRIENDLYWRT_SRC}/${FRIENDLYWRT_ROOTFS}
    prepare_image_for_friendlyelec_eflasher ${TARGET_IMAGE_DIRNAME} ${ROOTFS} && (cd ${SDFUSE_DIR} && {
	./mk-sd-image.sh ${TARGET_IMAGE_DIRNAME} ${TARGET_SD_RAW_FILENAME}
        echo "-----------------------------------------"
        echo "Run the following command for sdcard install:"
        echo "    sudo dd if=out/${TARGET_SD_RAW_FILENAME} bs=1M of=/dev/sdX"
        echo "-----------------------------------------"
    })
}

function install_toolchain() {
	if [ ! -d /opt/FriendlyARM/toolchain/4.9.3 ]; then
                log_info "installing toolchain: arm-linux-gcc 4.9.3"
                sudo su -c "mkdir -p /opt/FriendlyARM/toolchain && cat $TOP_DIR/toolchain/gcc-x64/toolchain-4.9.3-armhf.tar.gz* | sudo tar xz -C /"
        fi

	if [ ! -d /opt/FriendlyARM/toolchain/6.4-aarch64 ]; then
		log_info "installing toolchain: aarch-linux-gcc 6.4"
		sudo su -c "mkdir -p /opt/FriendlyARM/toolchain && cat $TOP_DIR/toolchain/gcc-x64/toolchain-6.4-aarch64.tar.gz* | sudo tar xz -C /"
	fi


}

function build_emmcimg() {
    local ROOTFS=${TOP_DIR}/${FRIENDLYWRT_SRC}/${FRIENDLYWRT_ROOTFS}
    prepare_image_for_friendlyelec_eflasher ${TARGET_IMAGE_DIRNAME} ${ROOTFS} && (cd ${SDFUSE_DIR} && {
	    ./mk-emmc-image.sh ${TARGET_IMAGE_DIRNAME} ${TARGET_EFLASHER_RAW_FILENAME}
        echo "-----------------------------------------"
        echo "Run the following command for sdcard install:"
        echo "    sudo dd if=out/${TARGET_EFLASHER_RAW_FILENAME} bs=1M of=/dev/sdX"
        echo "-----------------------------------------"
    })
}

##############################################

# These arrays will be populated in the.mk file
declare -a FRIENDLYWRT_PACKAGES=("")
declare -a FRIENDLYWRT_FILES=("")

MK_LINK=".current_config.mk"
FOUND_MK_FILE=`find device/friendlyelec -name ${1} | wc -l`
if [ $FOUND_MK_FILE -gt 0 ]; then
    MK_FILE=`ls device/friendlyelec/*/${1}`
    echo "using config ${MK_FILE}"
	rm -f ${MK_LINK}
	ln -s ${MK_FILE} ${MK_LINK}
	source ${MK_LINK}
	install_toolchain
	build_all
else
	BUILD_TARGET=${1}

	if [ -e "${MK_LINK}" ]; then
		source ${MK_LINK}

        	# display var
        	# ( set -o posix ; set ) | less
	else
		echo "no .current_config.mk, please select a board first."
        	firsttime_usage
		exit 1
	fi
	install_toolchain

	#=========================
	# build target
	#=========================
	if [ $BUILD_TARGET == uboot ];then
		build_uboot
		exit 0
	elif [ $BUILD_TARGET == kernel ];then
		build_kernel
		exit 0
	elif [ $BUILD_TARGET == friendlywrt ];then
		build_friendlywrt
		exit 0
	elif [ $BUILD_TARGET == sd-img ]; then
		build_sdimg
		exit 0
	elif [ $BUILD_TARGET == emmc-img ]; then
		build_emmcimg
		exit 0
	elif [ $BUILD_TARGET == all ];then
		build_all
		exit 0
 	elif [ $BUILD_TARGET == cleanall ];then
		clean_all
		exit 0
	else
		echo "Can't find a build config file, please check again"
		usage
		exit 1
	fi
fi

exit 0