#!/bin/bash

# This version of the script builds:
# - x86 gmp 5.0.5, mpfr 3.1.1 and mpc 1.0.1
# - x86->x86 binutils 2.23 and gcc 4.7.3
# - x86->x86_64 binutils 2.23 and gcc 4.7.3

# The script will:
# 1) Create x86 and x86_64 sysroots based on the unix-build env
# 2) Download the required packages
# 3) Apply patches from patches/*:
#    gcc-4.7.3/gcc-4.6.0bpw.diff - disable a bogus warning (taken from //sw/tools/unix/hosts/Linux-x86/targets/Linux-ARMv7-gnueabi/gcc-4.6.0/gcc-4.6.0bpw.diff, applies with offsets)
# 4) Configure, build and install the packages into install_x86 and install_x86_64

# To install the toolchain into //sw/tools you have to uncomment
# install_common, install_x86 and install_x86_64 at the end of the script.

# REPO and JOBS should be set for your env
REPO=/data/p4ws/
JOBS=16

# The rest should just work, but feel free to play with it

NV_TOOLS=${REPO}/sw/tools/
NV_TOOLS_TARGETS=$NV_TOOLS/unix/hosts/Linux-x86/targets/
export MAKE=${NV_TOOLS}/linux/gnumake/fc2/gmake
#export MAKE=gmake
PREFIX=$(readlink -f $(dirname $0))

UTIL_CONFIG_X86=${NV_TOOLS}/unix/hosts/Linux-x86/targets/Linux-x86/configs/gcc-4.1.1as2-with-glibc-2.2.2-10.mk
UTIL_PREFIX_X86=GCC_4_1_1_AS2_HOST_Linux_x86_TARGET_Linux_x86_
#UTIL_CONFIG_X86=${NV_TOOLS}/unix/experimental/hosts/Linux-x86/targets/Linux-x86/configs/gcc-4.7.2.mk
#UTIL_PREFIX_X86=GCC_4_7_2_HOST_Linux_x86_TARGET_Linux_x86_
UTIL_PREFIX_X86_64=GCC_4_1_1_AS2_HOST_Linux_x86_TARGET_Linux_x86_64_
UTIL_CONFIG_X86_64=${NV_TOOLS}/unix/hosts/Linux-x86/targets/Linux-x86_64/configs/gcc-4.1.1as2.mk

X86_64_SYSROOT_DIR=$PREFIX/x86_64-sysroot/
X86_SYSROOT_DIR=$PREFIX/x86-sysroot/

binutils_BASE_SRC="http://ftp.gnu.org/gnu/binutils/"
binutils_VER=2.31.1
binutils_PKG_SUF=.tar.gz

gmp_VER=5.0.5
gmp_setup() {
    PKG_SRC="ftp://ftp.gmplib.org/pub/gmp-$gmp_VER/gmp-$gmp_VER.tar.bz2"
}

mpfr_VER=3.1.1
mpfr_BASE_SRC="http://www.mpfr.org/mpfr-3.1.1/"

mpc_VER=1.0.1
mpc_BASE_SRC="https://ftp.gnu.org/gnu/mpc/"
mpc_PKG_SUF=".tar.gz"

gcc_stage1_VER=4.7.3
gcc_stage2_VER=4.7.3

gcc_setup() {
    PKG_SRC="http://mirrors.kernel.org/gnu/gcc/gcc-$VER/gcc-$VER.tar.bz2"
}

die() {
    echo $@
    exit 1
}

edo() {
    echo $@
    "$@" || die "$@ failed"
}

get_config() {
    local config=$1 prefix=$2 var=$3
#cat <<EOM
    $MAKE -f - print-${prefix}${var} <<EOM
    NV_TOOLS=${NV_TOOLS}
include ${config}

print-%:
	@echo \$(\$*)
EOM
}

set_build() {
    local host=$1 build=$2 target=$3

    if [[ $host == 32 ]]; then
        HOST=i686-pc-linux-gnu
    else
        HOST=x86_64-pc-linux-gnu
    fi

    if [[ $build == 32 ]]; then
        BUILD=i686-pc-linux-gnu
    else
        BUILD=x86_64-pc-linux-gnu
    fi

    if [[ $target == 32 ]]; then
        INSTALL_PREFIX=install_x86/
        SYSROOT_DIR=$X86_SYSROOT_DIR
        TARGET=i686-pc-linux-gnu
    else
        INSTALL_PREFIX=install_x86_64/
        SYSROOT_DIR=$X86_64_SYSROOT_DIR
        TARGET=x86_64-pc-linux-gnu
    fi
}

install_dir() {
    local pkg=$1
    if [[ -z $pkg ]]; then
        pkg=$PKG
    fi

    case $pkg in
        gmp) pkg=gmm;;
        mpfr) pkg=gmm;;
        mpc) pkg=gmm;;
    esac

    if [[ $pkg == gmm ]]; then
        echo $PREFIX/install_x86/gmm
    else
        echo $PREFIX/$INSTALL_PREFIX/$pkg
    fi
}

CONFIG_VARS=("CC" "CXX" "CPP" "CFLAGS" "LDFLAGS" "BIN_LDFLAGS" "INCLUDE_DIRS" "LD" "AR" "AS" "NM" "OBJDUMP" "OBJCOPY" "READELF" "STRIP")

load_config() {
    local config=$1 prefix=$2 var_prefix=$3

    for var in "${CONFIG_VARS[@]}"; do
        declare -g ${var_prefix}${var}="$(get_config $config $prefix $var)"
    done
}

load_config $UTIL_CONFIG_X86 $UTIL_PREFIX_X86 "X86_"
load_config $UTIL_CONFIG_X86_64 $UTIL_PREFIX_X86_64 "X86_64_"

get_var() {
    local name=$1 suffix=$2
    local tmp=${name}_${suffix}
    echo ${!tmp}
}

get_pkg_var() {
    local var=$1
    local tmp=$(get_var ${PKG}_${TYPE} ${var})
    if [[ -z $tmp ]]; then
        get_var ${PKG} ${var}
    else
        echo $tmp
    fi
}

function_exists() {
    local name=$1
    if [[ $(type -t $name) == function ]]; then
        return 0
    else
        return 1
    fi
}

run_if_exists() {
    local name=$1
    if function_exists $name; then
        $name
    else
        return 0
    fi
}

setup() {
    PKG=$1
    TYPE=$2

    VER=$(get_pkg_var VER)
    PKG_VER=$(get_pkg_var PKG_VER)
    : ${PKG_VER:=$VER}
    BASE_SRC=$(get_pkg_var BASE_SRC)
    PKG_SUF=$(get_pkg_var PKG_SUF)
    : ${PKG_SUF:=.tar.bz2}
    PKG_TARBALL=$(get_pkg_var PKG_TARBALL)
    : ${PKG_TARBALL:=${PKG}-${PKG_VER}${PKG_SUF}}
    PKG_SRC="$BASE_SRC/${PKG_TARBALL}"

    SRC_DIR=$PREFIX/$PKG-$VER/
    BUILD_DIR=$PREFIX/$PKG-$VER-build/


    run_if_exists ${PKG}_setup
    run_if_exists ${PKG}_${TYPE}_setup
}

gogo() {
    setup $1 $2
    fetch
    unpack
    build
}

fetch() {
    if [[ ! -e $PKG_TARBALL ]]; then
        local tmp=$(mktemp /tmp/tmp.XXXXXX)
        edo wget ${PKG_SRC} --no-check-certificate --secure-protocol=TLSv1 -O $tmp
        edo mv $tmp $PKG_TARBALL
    fi
}

unpack() {
    if [[ $CLEAR_SRC == 1 ]]; then
        edo rm -rf $SRC_DIR
    fi
    if [[ ! -e $SRC_DIR ]]; then
        edo tar xf $PKG_TARBALL
        patch
    fi
}

patch() {
    pushd $SRC_DIR
    for p in $PREFIX/patches/$PKG-$VER/*; do
        echo Applying patch $p
        command patch -p1 < $p
    done
    popd
}

build() {
    edo rm -rf $BUILD_DIR
    edo mkdir -p $BUILD_DIR $(install_dir)

    edo pushd $BUILD_DIR

    if function_exists ${PKG}_${TYPE}_build; then
        ${PKG}_${TYPE}_build
    else
        ${PKG}_build
    fi

    edo popd
}

gmp_build() {
    edo $SRC_DIR/configure \
        --host=$HOST \
        --build=$BUILD \
        --prefix=$(install_dir) \
        --with-sysroot=$SYSROOT_DIR

    edo $MAKE -j$JOBS
    edo $MAKE install
}

mpfr_build() {
    edo $SRC_DIR/configure \
        --host=$HOST \
        --build=$BUILD \
        --prefix=$(install_dir) \
        --with-gmp=$(install_dir gmp)

    edo $MAKE -j$JOBS
    edo $MAKE install
}

mpc_build() {
    edo $SRC_DIR/configure \
        --host=$HOST \
        --build=$BUILD \
        --prefix=$(install_dir) \
        --with-gmp=$(install_dir gmp) \
        --with-mpfr=$(install_dir mpfr)

    edo $MAKE -j$JOBS
    edo $MAKE install
}

canonicalize_binaries() {
    local bindir=$1

    edo pushd $1

    # Prefix the binaries with $TARGET
    for f in *; do
        if [[ $f == ${f#$TARGET-} ]]; then
            mv $f $TARGET-$f
        fi
    done

    # Create generic symlinks to the $TARGET prefixed binaries
    for f in *; do
        if [[ $f != ${f#$TARGET-} ]]; then
            ln -s $f ${f#$TARGET-}
        fi
    done

    popd
}

strip_binaries() {
    local dir=$1

    find $dir -type f -executable \! -name \*.so\* \! -name \*.a | xargs -r -n1 strip --strip-unneeded
}

strip_libraries() {
    local dir=$1

    # debug symbols in shared libraries can be stripped as these libraries
    # won't be used at runtime
    find $dir -type f -name \*.so\* | xargs -r -n1 strip --strip-debug
    # debug symbols in static libraries might be useful as they get linked in
    # directly into the produced binaries
    # find $dir -type f -name \*.a | xargs -r -n1 strip --strip-debug
}

cleanup_install_dir() {
    local dir=$1

    rm -rf $dir/share
    strip_binaries $dir
    strip_libraries $dir
}

setup_x86_sysroot() {
    local sysroot=$X86_SYSROOT_DIR

    edo rm -rf $sysroot
    edo mkdir -p $sysroot/usr
    edo rsync -av $NV_TOOLS/unix/targets/Linux-x86/glibc-2.2.2-10/* $sysroot/usr/
    edo rsync -av $NV_TOOLS/unix/targets/Linux-x86/linux-libc-headers-2.6.12.0/* $sysroot/usr/include/
    edo pushd $sysroot/usr/lib/
    edo popd
}

setup_x86_64_sysroot() {
    local sysroot=$X86_64_SYSROOT_DIR

    edo rm -rf $sysroot
    edo mkdir -p $sysroot/usr
    edo rsync -av $NV_TOOLS/unix/targets/Linux-x86_64/glibc-2.2.5-0/* $sysroot/usr/
    #edo rsync -av $NV_TOOLS/unix/targets/Linux-x86/glibc-2.2.2-10/lib/ $sysroot/usr/lib32
}

binutils_build() {
    local config=(
        --prefix=$(install_dir)
        --host=$HOST
        --build=$BUILD
        --target=$TARGET
        --disable-werror
        --with-sysroot=$SYSROOT_DIR
    )
    edo $SRC_DIR/configure "${config[@]}"
    edo $MAKE -j$JOBS
    edo $MAKE install -j$JOBS

    canonicalize_binaries $(install_dir)/bin/
}

gcc_stage1_build() {
    local config=(
        --prefix=$(install_dir)
        --host=$HOST
        --build=$BUILD
        --target=$TARGET
        --enable-languages=c
        --disable-threads
        --disable-shared
        --disable-multilib
        --with-sysroot=$SYSROOT_DIR
        --without-headers
        --disable-libmudflap
        --disable-libssp
        --disable-lto
        --disable-decimal-float
        --disable-libgomp
        --disable-libquadmath
        --with-gmp=$(install_dir gmp)
        --with-mpfr=$(install_dir mpfr)
        --with-mpc=$(install_dir mpc)
    )

    edo $SRC_DIR/configure "${config[@]}"
    edo $MAKE -j$JOBS all-gcc
    edo $MAKE -j$JOBS all-target-libgcc
    edo $MAKE install-gcc
    edo $MAKE install-target-libgcc
}

gcc_stage2_build() {
    local config=(
        --prefix=$(install_dir)
        --host=$HOST
        --build=$BUILD
        --target=$TARGET
        --enable-languages=c,c++
        --enable-threads=posix
        --enable-shared
        --disable-multilib
        --disable-gnu-indirect-function
        --disable-gnu-unique-object
        --with-sysroot=$SYSROOT_DIR
        --with-gmp=$(install_dir gmp)
        --with-mpfr=$(install_dir mpfr)
        --with-mpc=$(install_dir mpc)
    )
    edo $SRC_DIR/configure "${config[@]}"
    edo $MAKE -j$JOBS
    edo $MAKE install

    canonicalize_binaries $(install_dir)/bin/
}

build_binutils() {
    export CC="$X86_CC $X86_CFLAGS $X86_LDFLAGS $X86_BIN_LDFLAGS"
    export CXX=$X86_CXX
    export CPP=$X86_CPP
    export LD=$X86_LD
    export AR=$X86_AR
    export AS=$X86_AS
    export NM=$X86_NM
    export STRIP=$X86_STRIP
    export READELF=$X86_READELF
    export OBJCOPY=$X86_OBJCOPY

    set_build 32 32 32
    gogo binutils

    set_build 32 32 64
    gogo binutils

    unset CC
    unset CXX
    unset CPP
    unset LD
    unset AR
    unset AS
    unset NM
    unset STRIP
    unset READELF
    unset OBJCOPY
}

build_gmm() {
    set_build 32 32 32

    export CC="$X86_CC $X86_CFLAGS $X86_LDFLAGS $X86_BIN_LDFLAGS"
    export CXX=$X86_CXX
    export CPP=$X86_CPP

    export PATH=$(install_dir binutils)/bin:$PATH

    gogo gmp
    gogo mpfr
    gogo mpc

    unset CC
    unset CXX
    unset CPP
}

build_gcc_32() {
    local saved_path=$PATH saved_ld_lib=$LD_LIBRARY_PATH

    export CC="$X86_CC $X86_CFLAGS $X86_LDFLAGS $X86_BIN_LDFLAGS"
    export CXX=$X86_CXX
    export CPP=$X86_CPP

    set_build 32 32 32

    export LD_LIBRARY_PATH=$(install_dir gmp)/lib
    export PATH=$(install_dir binutils)/bin:$PATH

    # Two stages required as the x86->x86 gcc 4.1.1 that we have most likely
    # has a bug that noone bothered to track down.
    # http://gcc.gnu.org/ml/gcc-help/2011-03/msg00352.html
    # Build a minimal gcc first
    gogo gcc stage1

    edo rm -rf $PREFIX/gcc-32-stage1
    edo mv $(install_dir gcc) $PREFIX/gcc-32-stage1
    export PATH=$PREFIX/gcc-32-stage1/bin:$PATH

    unset CC
    unset CXX
    unset CPP
    gogo gcc stage2

    export PATH=$saved_path
    export LD_LIBRARY_PATH=$saved_ld_lib
}

build_gcc_64() {
    local saved_path=$PATH saved_ld_lib=$LD_LIBRARY_PATH

    export CC="$X86_CC $X86_CFLAGS $X86_LDFLAGS $X86_BIN_LDFLAGS"
    export CXX=$X86_CXX
    export CPP=$X86_CPP

    set_build 32 32 64

    export LD_LIBRARY_PATH=$(install_dir gmp)/lib
    export PATH=$(install_dir binutils)/bin:$PATH
    gogo gcc stage2

    unset CC
    unset CXX
    unset CPP

    export PATH=$saved_path
    export LD_LIBRARY_PATH=$saved_ld_lib
}

clear_and_cp() {
    local src=$1 dst=$2

    if [[ -e $dst ]]; then
        edo rm -rf $dst
    fi

    edo mkdir -p $dst
    edo cp -a $src/* $dst/
}

install_common() {
    strip_libraries $(install_dir gmp)/lib/
    # copy only the versioned libraries
    edo cp -a $(install_dir gmp)/lib/*.so.* $NV_TOOLS/unix/hosts/Linux-x86/unix-build/lib/
}

install_x86() {
    set_build 32 32 32

    clear_and_cp $(install_dir binutils)/bin/ $NV_TOOLS_TARGETS/Linux-x86/binutils-${binutils_VER}/bin/
    clear_and_cp $(install_dir binutils)/$TARGET/ $NV_TOOLS_TARGETS/Linux-x86/binutils-${binutils_VER}/$TARGET/
    clear_and_cp $(install_dir gcc) $NV_TOOLS_TARGETS/Linux-x86/gcc-${gcc_stage2_VER}

    cleanup_install_dir $NV_TOOLS_TARGETS/Linux-x86/binutils-${binutils_VER}/
    cleanup_install_dir $NV_TOOLS_TARGETS/Linux-x86/gcc-${gcc_stage2_VER}/
}

install_x86_64() {
    set_build 32 32 64

    clear_and_cp $(install_dir binutils)/bin/ $NV_TOOLS_TARGETS/Linux-x86_64/binutils-${binutils_VER}/bin/
    clear_and_cp $(install_dir binutils)/$TARGET/ $NV_TOOLS_TARGETS/Linux-x86_64/binutils-${binutils_VER}/$TARGET/
    clear_and_cp $(install_dir gcc) $NV_TOOLS_TARGETS/Linux-x86_64/gcc-${gcc_stage2_VER}/

    cleanup_install_dir $NV_TOOLS_TARGETS/Linux-x86_64/binutils-${binutils_VER}/
    cleanup_install_dir $NV_TOOLS_TARGETS/Linux-x86_64/gcc-${gcc_stage2_VER}/
}

setup_x86_sysroot
setup_x86_64_sysroot

build_binutils
build_gmm
build_gcc_32
build_gcc_64

#install_common
#install_x86
#install_x86_64