#! /bin/bash

# Paranoid check to make sure we don't reinvoke ourselves, effectively

# making a fork()bomb

if [ -n "$LD_HUGETLBFS_RECURSION" ]; then

    exit 99

fi

export LD_HUGETLBFS_RECURSION=1

### SET DEFAULT LDSCRIPT PATH HERE ###

if [ -z "$HUGETLB_LDSCRIPT_PATH" ]; then

    # Assume this script is running from the libhugetlbfs source tree,

    # and look for the ldscripts accordingly

    HUGETLB_LDSCRIPT_PATH=$(dirname $(readlink $0))/ldscripts

fi

### SET CUSTOM_LDSCRIPTS HERE ###

if [ -z "$CUSTOM_LDSCRIPTS" ]; then

    # Assume this script is running from the libhugetlbfs source tree,

    # and set CUSTOM_LDSCRIPTS to default "yes"

    CUSTOM_LDSCRIPTS="yes"

fi

# Try to figure out what's the underlying linker to invoke

if [ -z "$LD" ]; then

    for x in $(which -a ld); do

	if [ "$x" != "$0" ]; then

	    LD="$x"

	    break

	fi

    done

fi

i=0

while [ -n "$1" ]; do

    arg="$1"

    case "$arg" in

	-m*)

	    EMU="${arg#-m}"

	    args[$i]="$arg"

	    i=$[i+1]

	    if [ -z "$EMU" ]; then

		shift

		EMU="$1"

		args[$i]="$1"

		i=$[i+1]

	    fi

	    ;;

	--hugetlbfs-link=*)

	    if [ -z "$HUGETLB_DEPRECATED_LINK" ]; then

		echo -n "ld.hugetlbfs: --hugetlbfs-link is deprecated. " 1>&2

		echo "Migrate to --hugetlbfs-align." 1>&2

	    fi

	    HTLB_LINK="${arg#--hugetlbfs-link=}"

	    ;;

	--hugetlbfs-script-path=*)

	    HUGETLB_LDSCRIPT_PATH="${arg#--hugetlbfs-script-path=}"

	    ;;

	--hugetlbfs-align)

	    HTLB_ALIGN="slice"

	    ;;

	--)

	    args=("${args[@]}" "$@")

	    break

	    ;;

	*)

	    args[$i]="$arg"

	    i=$[i+1]

	    ;;

    esac

    shift

done

if [ -n "$HTLB_LINK" ]; then

    if [ "$CUSTOM_LDSCRIPTS" == "no" ]; then

        echo -n "ld.hugetlbfs: --hugetlbfs-link is not supported on this " 1>&2

        echo "platform. Use --hugetlbfs-align instead." 1>&2

    fi

    HTLB_ALIGN="" # --hugetlbfs-link overrides --hugetlbfs-align

    LDSCRIPT="$EMU.x$HTLB_LINK"

    HTLBOPTS="-T${HUGETLB_LDSCRIPT_PATH}/${LDSCRIPT}"

fi

# if -m is not present on command line

if [ -z "$EMU" ]; then

	if [ -n "$LDEMULATION" ]; then

		# try env. variable

		EMU="$LDEMULATION"

	else

		# pick first supported

		EMU="$(ld -V | sed -n '/Supported emulations/{n;p}' | tr -d ' ')"

	fi

fi

# if -m is not present on command line

if [ -z "$EMU" ]; then

	if [ -n "$LDEMULATION" ]; then

		# try env. variable

		EMU="$LDEMULATION"

	else

		# pick first supported

		EMU="$(ld -V | sed -n '/Supported emulations/{n;p}' | tr -d ' ')"

	fi

fi

MB=$((1024*1024))

case "$EMU" in

elf32ppclinux)		HPAGE_SIZE=$((16*$MB)) SLICE_SIZE=$((256*$MB)) ;;

elf64ppc|elf64lppc)

	hpage_kb=$(cat /proc/meminfo  | grep Hugepagesize: | awk '{print $2}')

	MMU_TYPE=$(cat /proc/cpuinfo  | grep MMU | awk '{ print $3}')

	HPAGE_SIZE=$((hpage_kb * 1024))

	if [ "$MMU_TYPE" == "Hash" ] ; then

		SLICE_SIZE=$((256*$MB))

	else

		SLICE_SIZE=$HPAGE_SIZE

	fi ;;

elf_i386|elf_x86_64)	HPAGE_SIZE=$((4*$MB)) SLICE_SIZE=$HPAGE_SIZE ;;

elf_s390|elf64_s390)	HPAGE_SIZE=$((1*$MB)) SLICE_SIZE=$HPAGE_SIZE ;;

armelf*_linux_eabi|aarch64elf*|aarch64linux*)

	hpage_kb=$(cat /proc/meminfo  | grep Hugepagesize: | awk '{print $2}')

	HPAGE_SIZE=$((hpage_kb * 1024))

	SLICE_SIZE=$HPAGE_SIZE ;;

esac

if [ "$HTLB_ALIGN" == "slice" ]; then

	HTLBOPTS="-zcommon-page-size=$SLICE_SIZE -zmax-page-size=$SLICE_SIZE"

	HTLBOPTS="$HTLBOPTS -lhugetlbfs"

	# targeting the ARM platform one needs to explicitly set the text segment offset

	# otherwise it will be NULL.

	case "$EMU" in

	armelf*_linux_eabi|aarch64elf*|aarch64linux*)	HTLBOPTS="$HTLBOPTS -Ttext-segment=$SLICE_SIZE" ;;

	elf_i386)		HTLBOPTS="$HTLBOPTS -Ttext-segment=0x08000000" ;;

	elf64ppc|elf64lppc)

		if [ "$MMU_TYPE" == "Hash" ] ; then

			printf -v TEXTADDR "%x" "$SLICE_SIZE"

			HTLBOPTS="$HTLBOPTS -Ttext-segment=$TEXTADDR"

		fi ;;

	esac

fi

${LD} "${args[@]}" ${HTLBOPTS}