#!/bin/bash
# Genkernel v3

GK_V='3.1.0k'
TEMP='/var/tmp/genkernel'
TODEBUGCACHE=1 # Until an error occurs or DEBUGFILE is fully qualified.

small_die() {
  echo $1
  exit 1
}

source /etc/genkernel.conf || small_die "Could not read /etc/genkernel.conf"
source ${GK_BIN}/gen_funcs.sh || small_die "Could not read ${GK_BIN}/gen_funcs.sh"
source ${GK_BIN}/gen_cmdline.sh || gen_die "Could not read ${GK_BIN}/gen_cmdline.sh"
source ${GK_BIN}/gen_arch.sh || gen_die "Could not read ${GK_BIN}/gen_arch.sh"
source ${GK_BIN}/gen_determineargs.sh || gen_die "Could not read ${GK_BIN}/gen_determineargs.sh"
source ${GK_BIN}/gen_compile.sh || gen_die "Could not read ${GK_BIN}/gen_compile.sh"
source ${GK_BIN}/gen_configkernel.sh || gen_die "Could not read ${GK_BIN}/gen_configkernel.sh"
source ${GK_BIN}/gen_initrd.sh || gen_die "Could not read ${GK_BIN}/gen_initrd.sh"
source ${GK_BIN}/gen_moddeps.sh || gen_die "Could not read ${GK_BIN}/gen_moddeps.sh"
source ${GK_BIN}/gen_package.sh || gen_die "Could not read ${GK_BIN}/gen_package.sh"
source ${GK_BIN}/gen_bootloader.sh || gen_die "Could not read ${GK_BIN}/gen_bootloader.sh"

BUILD_KERNEL=0
BUILD_INITRD=0

# Parse all command line options...
while [ $# -gt 0 ]
do
	Option=$1; shift
	parse_cmdline $Option
done

# Check if no action is specified...
if [ "${BUILD_KERNEL}" -eq 0 -a "${BUILD_INITRD}" -eq 0 ]
then
	usage
	exit 1
fi

clear_log
NORMAL=${GOOD} print_info 1 "Gentoo Linux Genkernel; Version ${GK_V}${NORMAL}"

# Set ${ARCH}
get_official_arch

# Read arch-specific config
source ${ARCH_CONFIG} || gen_die "Could not read ${ARCH_CONFIG}"
source ${GK_SHARE}/${ARCH}/modules_load || gen_die "Could not read ${GK_SHARE}/${ARCH}/modules_load"

[ ! -f "${TEMP}" ] && mkdir -p "${TEMP}"
[ ! -f "${CACHE_DIR}" ] && mkdir -p "${CACHE_DIR}"

# Based on genkernel.conf, arch-specific configs, and commandline options,
# get the real arguments for usage...

determine_real_args
dump_debugcache

NORMAL=${BOLD} print_info 1 "Compiling Linux Kernel ${BOLD}${KV}${NORMAL} for ${BOLD}${ARCH}${NORMAL}..."
echo

if [ "${BUILD_INITRD}" -ne '0' ]
then
	if ! has_loop
	then
		modprobe loop 2>/dev/null
		if ! has_loop
		then
			print_error 1 'Your kernel does not appear to have loop device support.'
			print_error 1 'Please load loop support before running genkernel!'
			gen_die 'Load loop support!'
		else
			print_info 1 'loop: "loop" module loaded successfully...'
		fi
	fi
fi

# Check /boot is mounted
if isTrue ${CMD_NOINSTALL}
then
	isTrue ${MOUNTBOOT} && print_info 2 'Skipping automatic mount of boot'
else
	if ! egrep -q ' /boot ' /proc/mounts
	then
		if egrep -q '^[^#].+	/boot	' /etc/fstab
		then
			if isTrue ${MOUNTBOOT}
			then
				if ! mount /boot
				then
					print_warning 1 "${BOLD}WARNING${NORMAL}: Failed to mount /boot!"
					echo
				else
					print_info 1 'mount: /boot mounted successfully!'
				fi
			else
				print_warning 1 "${BOLD}WARNING${NORMAL}: No mounted /boot partition detected!"
				print_warning 1 '         Run ``mount /boot`` to mount it!'
				echo
			fi
		fi
	elif isBootRO
	then
		if isTrue ${MOUNTBOOT}
		then
			if ! mount -o remount,rw /boot
			then
				print_warning 1 "${BOLD}WARNING${NORMAL}: Failed to remount /boot RW!"
				echo
			else
				print_info 1 "mount: /boot remounted read/write successfully!"
				BOOTRW=1
			fi
		fi
	fi
fi

# Check whether another Genkernel is running
GENPIDS="`ps -C genkernel --no-headers -o pid | wc -l`"
if [ "${GENPIDS}" -gt '3' ]
then
	[ "${GENPIDS}" -gt '4' ] && EX='s'
	print_warning 1 "${BOLD}WARNING${NORMAL}: Another Genkernel instance is running under"
	print_warning 1 "         process ID${EX} " 0
	GENPIDS=`ps -C genkernel --no-headers -o pid`
	echo -n "${GENPIDS}" | sed -e "s/$$//; s/ /, /g"
	echo 'halting...'
	echo
	print_warning 1 'Running multiple genkernels on the same source tree will cause data loss!'
	print_info 1 "Press ^C to halt; ^D to continue [ ${BOLD}if${NORMAL} you know what you're doing! ]"
	cat
	echo
	CTEMP="${TEMP}"
	TEMP="/var/tmp/genkernel-`dd if=/dev/random count=1 2>/dev/null | md5sum | cut -d\  -f1`"
	print_info 1 'thread: Running multiple genkernels may cause problems!'
	print_info 1 "thread: Temporary files reallocated to ${TEMP}..."
	echo
fi

if [ ${BUILD_KERNEL} -eq 1 ]
then
	# Configure kernel
	config_kernel

	# Make deps
	compile_dep

	# Make prepare [2.6]
	[ "${VER}" -gt '2' ] || [ "${VER}" -eq '2' -a "${PAT}" -ge '6' ] && compile_generic prepare kernel

	# Run postconf
	if [ "${CMD_POSTCONF}" != '' ]
	then
		print_info 1 '' 1 0
		print_info 1 "Preparing to run postconf: \"${CMD_POSTCONF}\"" 0

		CALLBACK_ESCAPE=0
		CALLBACK_COUNT=0

		trap "CALLBACK_ESCAPE=1" TERM KILL INT QUIT ABRT
		while [[ ${CALLBACK_ESCAPE} -eq 0 && ${CALLBACK_COUNT} -lt 5 ]]
		do
			sleep 1; echo -n '.';
			let CALLBACK_COUNT=${CALLBACK_COUNT}+1
		done

		if [ "${CALLBACK_ESCAPE}" -eq 0 ]
		then
			echo
			echo
			eval ${CMD_POSTCONF} | tee -a ${DEBUGFILE}
			CMD_STATUS="${PIPESTATUS[0]}"
			echo
			print_info 1 "<<< Postconf callback exit status: ${CMD_STATUS}"
			[ "${CMD_STATUS}" -ne 0 ] && gen_die '--postconf failed!'
		else
			echo
			print_info 1 '>>> Postconf cancelled...'
		fi
		trap - TERM KILL INT QUIT ABRT
		print_info 1 '' 1 0
	fi

	if [ "${GENSPLASH}" -eq '1' ]
	then
		if [ -x /sbin/splash ]
		then
			[ -z "${GENSPLASH_THEME}" ] && [ -e /etc/conf.d/splash ] && source /etc/conf.d/splash
			[ -z "${GENSPLASH_THEME}" ] && GENSPLASH_THEME=default
			print_info 1 "	>> Installing gensplash [ using the ${GENSPLASH_THEME} theme ]..."
			# 2.6.9
			sed -i -e 's:$(obj)/gen_init_cpio $(initramfs-y) FORCE:$(if $(shell [ ! -f $(obj)/initramfs_data.cpio ] \&\& echo 1 ), $(obj)/gen_init_cpio $(initramfs-y)):' ${KERNEL_DIR}/usr/Makefile
			# 2.6.10
			sed -i -e 's:$(obj)/gen_init_cpio \\:$(if $(shell [ ! -f $(obj)/initramfs_data.cpio ] \&\& echo 1 ), $(obj)/gen_init_cpio $(initramfs-y) $(obj)/initramfs_list):' -e 's:$(initramfs-y) $(obj)/initramfs_list FORCE::' ${KERNEL_DIR}/usr/Makefile
			splash_geninitramfs -g ${KERNEL_DIR}/usr/initramfs_data.cpio.gz ${GENSPLASH_THEME}
			gunzip ${KERNEL_DIR}/usr/initramfs_data.cpio.gz
		else
			print_warning 1 '		>> No splash detected; skipping!'
		fi
	fi

	# Compile kernel
	compile_kernel

	# Compile modules
	compile_modules

	if [ ${SAVE_CONFIG} -eq 1 ]
	then
		print_info 1 "Copying config for successful build to /etc/kernels/kernel-config-${ARCH}-${KV}"
		[ ! -e '/etc/kernels' ] && mkdir -p /etc/kernels
		cp "${KERNEL_DIR}/.config" "/etc/kernels/kernel-config-${ARCH}-${KV}"
	fi
fi

# Run callback
if [ "${CMD_CALLBACK}" != "" ]
then
	print_info 1 "" 1 0
	print_info 1 "Preparing to run callback: \"${CMD_CALLBACK}\"" 0

	CALLBACK_ESCAPE=0
	CALLBACK_COUNT=0

	trap "CALLBACK_ESCAPE=1" TERM KILL INT QUIT ABRT
	while [[ ${CALLBACK_ESCAPE} -eq 0 && ${CALLBACK_COUNT} -lt 5 ]]
	do
		sleep 1; echo -n '.';
		let CALLBACK_COUNT=${CALLBACK_COUNT}+1
	done

	if [ "${CALLBACK_ESCAPE}" -eq 0 ]
	then
		echo
		echo
		eval ${CMD_CALLBACK} | tee -a ${DEBUGFILE}
		CMD_STATUS="${PIPESTATUS[0]}"
		echo
		print_info 1 "<<< Callback exit status: ${CMD_STATUS}"
		[ "${CMD_STATUS}" -ne 0 ] && gen_die '--callback failed!'
	else
		echo
		print_info 1 ">>> Callback cancelled..."
	fi
	trap - TERM KILL INT QUIT ABRT
	print_info 1 "" 1 0
fi

if [ "${BUILD_INITRD}" -eq '1' ]
then
	# Compile dietlibc
	if [ "${USE_DIETLIBC}" = '1' ]
	then
		compile_dietlibc
	fi

	# Compile Busybox
	compile_busybox

	# Only compile insmod if we're installing modules onto the initrd
	if [ "${NOINITRDMODULES}" = '' ]
	then
		if [ "${PAT}" -gt '4' ]
		then
			# Compile module-init-tools
			compile_module_init_tools
		else
			compile_modutils
		fi
	fi

	compile_devfsd
	if [ "${DEVFS}" -eq '0' ]
	then
		[ "${VER}" -gt '2' ] || [ "${VER}" -eq '2' -a "${PAT}" -ge '6' ] && UDEV=1
	fi
	[ "${UDEV}" -eq '1' ] && compile_udev

	# Create initrd
	create_initrd
else
	print_info 1 'initrd: Not building since only the kernel was requested...'
fi
[ "${MINKERNPACKAGE}" != "" ] && gen_minkernpackage

# Clean up...
[ -n "${CTEMP}" ] && rm -rf "${TEMP}"

if [ "${BUILD_KERNEL}" -eq '1' ]
then
	set_bootloader
	print_info 1 ''
	print_info 1 "Kernel compiled successfully!"
	print_info 1 ''
	print_info 1 'Required Kernel Parameters:'
	if [ "${BUILD_INITRD}" -eq '0' ]
	then
		print_info 1 '    root=/dev/$ROOT'
		print_info 1 '    [ And "vga=0x317 splash=verbose" if you use a framebuffer ]'
		print_info 1 ''
		print_info 1 '    Where $ROOT is the device node for your root partition as the'
		print_info 1 '    one specified in /etc/fstab'
	else	
		print_info 1 '    root=/dev/ram0 real_root=/dev/$ROOT init=/linuxrc'
		[ "${INITRD_SIZE}" -ge 4096 ] && print_info 1 "    ramdisk_size=8192"
		print_info 1 '    [ And "vga=0x317 splash=verbose" if you use a framebuffer ]'
		print_info 1 ''
		print_info 1 '    Where $ROOT is the device node for your root partition as the'
		print_info 1 '    one specified in /etc/fstab'
		print_info 1 ''
		print_info 1 "If you require Genkernel's hardware detection features; you MUST"
		print_info 1 'tell your bootloader to use the provided initrd file. Otherwise;'
		print_info 1 'substitute the root argument for the real_root argument if you are'
		print_info 1 'not planning to use the initrd...'
	fi
else
	print_info 1 ''
	print_info 1 'initrd compiled successfully!'
fi

[ "${BOOTRW}" != '' ] && mount -o remount,ro /boot

print_info 1 ''
print_info 1 'Do NOT report kernel bugs as genkernel bugs unless your bug'
print_info 1 'is about the default genkernel configuration...'
print_info 1 ''
print_info 1 'Make sure you have the latest genkernel before reporting bugs.'