--- trunk/grubby/new-kernel-pkg	2011/07/19 17:02:36	1431
+++ trunk/grubby/new-kernel-pkg	2012/10/01 12:33:17	1937
@@ -25,6 +25,32 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 
+read_os_release()
+{
+	local NAME
+	local ID
+	local Version
+	local Version_ID
+	local PRETTY_NAME
+	local ANSI_COLOR
+	local CPE_NAME
+
+	if [[ -f /etc/os-release ]]
+	then
+		source /etc/os-release
+
+		case $1 in
+			name) echo "${NAME}" ;;
+			id) echo "${ID}" ;;
+			version) echo "${Version}" ;;
+			version_id) echo "${Version_ID}" ;;
+			pretty_name) echo "${PRETTY_NAME}" ;;
+			ansi_color) echo "${ANSI_COLOR}" ;;
+			cpe_name) echo "${CPE_NAME}" ;;
+		esac
+	fi
+}
+
 PATH=/sbin:/bin:$PATH
 
 lilo=/sbin/lilo
@@ -40,11 +66,19 @@
 fi
 
 [ -f /etc/conf.d/grubby ] && . /etc/conf.d/grubby
+[ -f /etc/conf.d/grubby-uboot ] && . /etc/conf.d/grubby-uboot
 
+cfgGrub2=""
+cfgGrub2Efi=""
 cfgGrub=""
 cfgLilo=""
+cfgExtlinux=""
+cfguBoot=""
 runLilo=""
 grubConfig=""
+grub2Config=""
+grub2EfiConfig=""
+extlinuxConfig=""
 
 ARCH=$(uname -m)
 
@@ -57,6 +91,7 @@
 elif [[ ${ARCH} = ppc64 ]] || [[ ${ARCH} = ppc ]]
 then
 	liloConfig=/etc/yaboot.conf
+	grub2Config=/boot/grub/grub.cfg
 	bootPrefix=/boot
 	lilo=/sbin/ybin
 	liloFlag=yaboot
@@ -77,10 +112,27 @@
 	lilo=/sbin/zipl
 	runLilo="yes"
 	isx86=""
+elif [[ ${ARCH} =~ armv[5|7].*l ]]
+then
+	liloConfig=""
+	bootPrefix=/boot
+	ubootDir=${UBOOT_DIR:-"/boot"}
+	ubootScript=$ubootDir/${UBOOT_SCR:-"boot.scr"}
+	ubootKList=${UBOOT_KLIST:-"klist.txt"}
+	ubootDevice=/dev/${UBOOT_DEVICE:-"mmcblk0p1"}
+	ubootDefaultImage=${UBOOT_UIMAGE:-"uImage"}
+	ubootDefaultInitrd=${UBOOT_UINITRD:-"uInitrd"}
+	ubootAddress=${UBOOT_IMGADDR:-"0x00008000"}
+	mounted=""
+	liloFlag=""
+	isx86=""
 else
 	# this leaves i?86 and x86_64
 	liloConfig=/etc/lilo.conf
 	grubConfig=/boot/grub/grub.conf
+	grub2Config=/boot/grub/grub.cfg
+	grub2EfiConfig=/boot/grub/grub-efi.cfg
+	extlinuxConfig=/boot/extlinux/extlinux.conf
 	bootPrefix=/boot
 	liloFlag=lilo
 	isx86="yes"
@@ -96,8 +148,8 @@
 verbose=""
 makedefault=""
 package=""
-mbkernel=""
-mbargs=""
+mbkernel="${HYPERVISOR}"
+mbargs="${HYPERVISOR_ARGS}"
 adddracutargs=""
 addplymouthinitrd=""
 
@@ -115,15 +167,6 @@
 	exit 1
 }
 
-run_grub2()
-{
-	if [ -d /boot/grub -a -x /sbin/grub-mkconfig ]
-	then
-		cp /boot/grub/grub.cfg /boot/grub/grub.cfg.old
-		/sbin/grub-mkconfig -o /boot/grub/grub.cfg
-	fi
-}
-
 install()
 {
 	# XXX kernel should be able to be specified also (or work right on ia64)
@@ -161,9 +204,6 @@
 		return
 	fi
 
-	# Run grub2's configuration update in parallel, if it is installed
-	run_grub2
-
 	# get the root filesystem to use
 	rootdevice=$(awk '{ if ($1 !~ /^[ \t]*#/ && $2 == "/") { print $1; }}' /etc/fstab)
 
@@ -179,19 +219,73 @@
 
 		if [[ -n ${banner} ]]
 		then
-			title="${banner} (${version})"
+			title="${banner} [ ${version} ]"
 		elif [ -f /etc/mageversion ]
 		then
-			title="Magellan Linux $(< /etc/mageversion) (${version})"
+			title="Magellan Linux $(< /etc/mageversion) [ ${version} ]"
+		elif [[ $(read_os_release id) = magellan ]]
+		then
+			title="$(read_os_release name) $(read_os_release version_id) [ ${version} ]"
 		else
-			title="Magellan Linux (${version})"
+			title="Magellan Linux [ ${version} ]"
 		fi
-		${grubby} --add-kernel=${bootPrefix}/${kernelName}-${version} \
+		${grubby} --grub -c ${grubConfig} \
+			--add-kernel=${bootPrefix}/${kernelName}-${version} \
 			${INITRD} --copy-default ${makedefault} --title "${title}" \
 			${mbkernel:+--add-multiboot="${mbkernel}"} ${mbargs:+--mbargs="${mbargs}"} \
 			--args="root=${rootdevice} ${kernargs}" --remove-kernel="TITLE=${title}"
 	else
-		[[ -n ${verbose} ]] && echo "${grubConfig} does not exist, not running grubby"
+		[[ -n ${verbose} ]] && echo "${grubConfig} does not exist, not running grubby for grub 0.97"
+	fi
+	if [[ -n ${cfgGrub2} ]]
+	then
+		[[ -n ${verbose} ]] && echo "adding ${version} to ${grub2Config}"
+
+		if [[ -n ${banner} ]]
+		then
+			title="${banner} [ ${version} ]"
+		elif [ -f /etc/mageversion ]
+		then
+			title="Magellan Linux $(< /etc/mageversion) [ ${version} ]"
+		elif [[ $(read_os_release id) = magellan ]]
+		then
+			title="$(read_os_release name) $(read_os_release version_id) [ ${version} ]"
+		else
+			title="Magellan Linux [ ${version} ]"
+		fi
+		${grubby} --grub2 -c ${grub2Config} \
+			--add-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} --copy-default ${makedefault} --title "${title}" \
+			${mbkernel:+--add-multiboot="${mbkernel}"} \
+			${mbargs:+--mbargs="${mbargs}"} \
+			--args="root=${rootdevice} ${kernargs}" --remove-kernel="TITLE=${title}"
+	else
+		[[ -n ${verbose} ]] && echo "${grub2Config} does not exist, not running grubby for grub 2"
+	fi
+	if [[ -n ${cfgGrub2Efi} ]]
+	then
+		[[ -n ${verbose} ]] && echo "adding ${version} to ${grub2EfiConfig}"
+
+		if [[ -n ${banner} ]]
+		then
+			title="${banner} [ ${version} ]"
+		elif [ -f /etc/mageversion ]
+		then
+			title="Magellan Linux $(< /etc/mageversion) [ ${version} ]"
+		elif [[ $(read_os_release id) = magellan ]]
+		then
+			title="$(read_os_release name) $(read_os_release version_id) [ ${version} ]"
+		else
+			title="Magellan Linux [ ${version} ]"
+		fi
+		${grubby} --grub2 -c ${grub2EfiConfig} \
+			--add-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} --copy-default ${makedefault} --title "${title}" \
+			${mbkernel:+--add-multiboot="${mbkernel}"} \
+			${mbargs:+--mbargs="${mbargs}"} \
+			--args="root=${rootdevice} ${kernargs}" --remove-kernel="TITLE=${title}"
+	else
+		[[ -n ${verbose} ]] && echo "${grub2EfiConfig} does not exist, not running grubby for grub 2 with UEFI"
 	fi
 
 	if [[ -n ${cfgLilo} ]]
@@ -217,6 +311,31 @@
 	else
 		[[ -n ${verbose} ]] && echo "${liloConfig} does not exist, not running grubby"
 	fi
+
+	if [[ -n ${cfgExtlinux} ]]
+	then
+		[[ -n ${verbose} ]] && echo "adding ${version} to ${extlinuxConfig}"
+
+		if [[ -n ${banner} ]]
+		then
+			title="${banner} [ ${version} ]"
+		elif [ -f /etc/mageversion ]
+		then
+			title="Magellan Linux $(< /etc/mageversion) [ ${version} ]"
+		elif [[ $(read_os_release id) = magellan ]]
+		then
+			title="$(read_os_release name) $(read_os_release version_id) [ ${version} ]"
+		else
+			title="Magellan Linux [ ${version} ]"
+		fi
+		${grubby} --extlinux -c ${extlinuxConfig} \
+			--add-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} --copy-default ${makedefault} --title "${title}" \
+			${mbkernel:+--add-multiboot="${mbkernel}"} ${mbargs:+--mbargs="${mbargs}"} \
+			--args="root=${rootdevice} ${kernargs}" --remove-kernel="TITLE=${title}"
+	else
+		[[ -n ${verbose} ]] && echo "${extlinuxConfig} does not exist, not running grubby for extlinux"
+	fi
 }
 
 remove()
@@ -229,15 +348,29 @@
 		return
 	fi
 
-	# Run grub2's configuration update in parallel, if it is installed
-	run_grub2
-
 	if [[ -n ${cfgGrub} ]]
 	then
 		[[ -n ${verbose} ]] && echo "removing ${version} from ${grubConfig}"
-		${grubby} --remove-kernel=${bootPrefix}/${kernelName}-${version}
+		${grubby} --grub -c ${grubConfig} \
+			--remove-kernel=${bootPrefix}/${kernelName}-${version}
 	else
-		[[ -n ${verbose} ]] && echo "${grubConfig} does not exist, not running grubby"
+		[[ -n ${verbose} ]] && echo "${grubConfig} does not exist, not running grubby for grub 0.97"
+	fi
+	if [[ -n ${cfgGrub2} ]]
+	then
+		[[ -n ${verbose} ]] && echo "removing ${version} from ${grub2Config}"
+		${grubby} --grub2 -c ${grub2Config} \
+			--remove-kernel=${bootPrefix}/${kernelName}-${version}
+	else
+		[[ -n ${verbose} ]] && echo "${grub2Config} does not exist, not running grubby for grub 2"
+	fi
+	if [[ -n ${cfgGrub2Efi} ]]
+	then
+		[[ -n ${verbose} ]] && echo "removing ${version} from ${grub2EfiConfig}"
+		${grubby} --grub2 -c ${grub2EfiConfig} \
+			--remove-kernel=${bootPrefix}/${kernelName}-${version}
+	else
+		[[ -n ${verbose} ]] && echo "${grub2EfiConfig} does not exist, not running grubby grub 2 with UEFI"
 	fi
 
 	if [[ -n ${cfgLilo} ]]
@@ -258,6 +391,69 @@
 	else
 		[[ -n ${verbose} ]] && echo "${liloConfig} does not exist, not running grubby"
 	fi
+
+	if [[ -n ${cfguBoot} ]]
+	then
+		[[ -n ${verbose} ]] && echo "removing ${version} from ${ubootDir}..."
+
+		if [ -f ${ubootDir}/${ubootKList} ]
+		then
+			tmpKList=$(mktemp ${ubootDir}/${ubootKList}.XXXX)
+			curversion=$(tail -n1 ${ubootDir}/${ubootKList})
+			sed "/$version$/d" ${ubootDir}/${ubootKList} > ${tmpKList}
+			newversion=$(tail -n1 ${tmpKList})
+			if [ -f ${ubootDir}/uImage-${newversion} ] && [ -f ${ubootDir}/uInitrd-${newversion} ]
+			then
+				if [[ ${curversion} != ${newversion} ]]
+				then
+					cp -fp ${ubootDir}/uImage-${newversion} ${ubootDir}/${ubootDefaultImage}
+					if [ $? -ne 0 ]
+					then
+						[[ -n ${verbose} ]] && echo "copy uImage-${newversion} error, default kernel not replaced!" && exit
+					fi
+					cp -fp ${ubootDir}/uInitrd-${newversion} ${ubootDir}/${ubootDefaultInitrd}
+					if [ $? -ne 0 ]
+					then
+						[[ -n ${verbose} ]] && echo "copy uInitrd-${newversion} error, default Initrd not replaced!" && exit
+					fi
+				fi
+
+				[[ -n ${verbose} ]] && echo "removing uImage-${version}"
+				if [ -f ${ubootDir}/uImage-${version} ]
+				then
+					rm -f ${ubootDir}/uImage-${version}
+				else
+					[[ -n ${verbose} ]] && echo "uImage-${version} did not exist!"
+				fi
+
+				[[ -n ${verbose} ]] && echo "removing uInitrd-${version}"
+				if [ -f ${ubootDir}/uInitrd-${version} ]
+				then
+					rm -f ${ubootDir}/uInitrd-${version}
+				else
+					[[ -n ${verbose} ]] && echo "uInitrd-${version} did not exist!"
+				fi
+
+				mv ${tmpKList} ${ubootDir}/${ubootKList}
+			else
+				[[ -n ${verbose} ]] && echo "uImage ${newversion} does not exist!"
+				[ -f ${tmpKList} ] && rm -f ${tmpKList}
+			fi
+		else
+			[[ -n ${verbose} ]] && echo "No previous kernel version.  U-Boot images not removed!"
+		fi
+	else
+		[[ -n ${verbose} ]] && echo "${ubootScript} does not exist, not modifying ${ubootDir}"
+	fi
+
+	if [[ -n ${cfgExtlinux} ]]
+	then
+		[[ -n ${verbose} ]] && echo "removing ${version} from ${extlinuxConfig}"
+		${grubby} --extlinux -c ${extlinuxConfig} \
+			--remove-kernel=${bootPrefix}/${kernelName}-${version}
+	else
+		[[ -n ${verbose} ]] && echo "${extlinuxConfig} does not exist, not running grubby for extlinux"
+	fi
 }
 
 update()
@@ -280,13 +476,11 @@
 		fi
 	fi
 
-	# Run grub2's configuration update in parallel, if it is installed
-	run_grub2
-
 	if [[ -n ${cfgGrub} ]]
 	then
 		[[ -n ${verbose} ]] && echo "updating ${version} from ${grubConfig}"
-		${grubby} --update-kernel=${bootPrefix}/${kernelName}-${version} \
+		${grubby} --grub -c ${grubConfig} \
+			--update-kernel=${bootPrefix}/${kernelName}-${version} \
 			${INITRD} \
 			${kernargs:+--args="${kernargs}"} \
 			${removeargs:+--remove-args="${removeargs}"}
@@ -294,6 +488,30 @@
 		[[ -n ${verbose} ]] && echo "${grubConfig} does not exist, not running grubby"
 	fi
 
+	if [[ -n ${cfgGrub2} ]]
+	then
+		[[ -n ${verbose} ]] && echo "updating ${version} from ${grub2Config}"
+		${grubby} --grub2 -c ${grub2Config} \
+			--update-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} \
+			${kernargs:+--args="${kernargs}"} \
+			${removeargs:+--remove-args="${removeargs}"}
+	else
+		[[ -n ${verbose} ]] && echo "${grub2Config} does not exist, not running grubby"
+	fi
+
+	if [[ -n ${cfgGrub2Efi} ]]
+	then
+		[[ -n ${verbose} ]] && echo "updating ${version} from ${grub2EfiConfig}"
+		${grubby} --grub2 -c ${grub2EfiConfig} \
+			--update-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} \
+			${kernargs:+--args="${kernargs}"} \
+			${removeargs:+--remove-args="${removeargs}"}
+	else
+		[[ -n ${verbose} ]] && echo "${grub2EfiConfig} does not exist, not running grubby"
+	fi
+
 	if [[ -n ${cfgLilo} ]]
 	then
 		[[ -n ${verbose} ]] && echo "updating ${version} from ${liloConfig}"
@@ -316,15 +534,60 @@
 	else
 		[[ -n ${verbose} ]] && echo "${liloConfig} does not exist, not running grubby"
 	fi
+
+	if [[ -n ${cfguBoot} ]]
+	then
+		[[ -n ${verbose} ]] && echo "adding $version to ${ubootDir}..."
+
+		[[ -n ${verbose} ]] && echo "creating uImage-${version}"
+		mkimage -A arm -O linux -T kernel -C none -a ${ubootAddress} \
+			-e ${ubootAddress} -n ${version} \
+			-d ${bootPrefix}/${kernelName}-${version} ${ubootDir}/uImage-${version}
+
+		[[ -n ${verbose} ]] && echo "creating uInitrd-${version}"
+		mkimage -A arm -O linux -T ramdisk -C none -a 0 -e 0 \
+			-n initramfs -d ${initrdfile} ${ubootDir}/uInitrd-${version}
+
+		if [ -f ${ubootDir}/uImage-${version} ] && [ -f ${ubootDir}/uInitrd-${version} ]
+		then
+			cp -fp ${ubootDir}/uImage-${version} ${ubootDir}/${ubootDefaultImage}
+			if [ $? -ne 0 ]
+			then
+				[[ -n ${verbose} ]] && echo "copy uImage-${version} error, kernel not installed!" && exit
+			fi
+			cp -fp ${ubootDir}/uInitrd-${version} ${ubootDir}/${ubootDefaultInitrd}
+			if [ $? -ne 0 ]
+			then
+				[[ -n ${verbose} ]] && echo "copy uInitrd-${version} error, kernel not installed!" && exit
+			fi
+			echo ${version} >> ${ubootDir}/${ubootKList}
+		else
+			[[ -n ${verbose} ]] && echo "cannot make ${version} the default"
+		fi
+	 else
+		[[ -n ${verbose} ]] && echo "${ubootScript} does not exist, not setting up ${ubootDir}"
+	fi
+
+	if [[ -n ${cfgExtlinux} ]]
+	then
+		[[ -n ${verbose} ]] && echo "updating ${version} from ${extlinuxConfig}"
+		${grubby} --extlinux -c ${extlinuxConfig} \
+			--update-kernel=${bootPrefix}/${kernelName}-${version} \
+			${INITRD} \
+			${kernargs:+--args="${kernargs}"} \
+			${removeargs:+--remove-args="${removeargs}"}
+	else
+		[[ -n ${verbose} ]] && echo "${extlinuxConfig} does not exist, not running grubby"
+	fi
 }
 
-mkinitrd()
+makeinitrd()
 {
 	if [[ -n ${dracut} ]]
 	then
-		tool="/sbin/dracut ${dracuthostonly} -f ${initrdfile} ${version}"
+		tool="dracut ${dracuthostonly} -f ${initrdfile} ${version}"
 	else
-		tool="/sbin/mkinitrd --allow-missing -f ${initrdfile} ${version}"
+		tool="mkinitrd --allow-missing -f ${initrdfile} ${version}"
 	fi
 	[[ -n ${verbose} ]] && echo "creating initrd ${initrdfile} using ${version}"
 	${tool}
@@ -565,7 +828,7 @@
 
 if [[ ${initrd} = make ]]
 then
-	mkinitrd
+	makeinitrd
 elif [[ ${initrd} = remove ]]
 then
 	rminitrd
@@ -578,7 +841,43 @@
 fi
 
 [[ -n ${grubConfig} ]] && [ -f ${grubConfig} ] && cfgGrub=1
+[[ -n ${grub2Config} ]] && [ -f ${grub2Config} ] && cfgGrub2=1
+[[ -n ${grub2EfiConfig} ]] && [ -f ${grub2EfiConfig} ] && cfgGrub2Efi=1
 [[ -n ${liloConfig} ]] && [ -f ${liloConfig} ] && cfgLilo=1
+[[ -n ${extlinuxConfig} ]] && [ -f ${extlinuxConfig} ] && cfgExtlinux=1
+
+# if we have a U-Boot directory, but no boot script, check if the directory
+# is mounted.  If not, mount it, and then check if a boot script exists.
+if [[ -n ${ubootDir} ]]
+then
+	if [ -f ${ubootScript} ]
+	then
+		cfguBoot=1
+	else
+		mountEntry=$(mount | grep ${ubootDir})
+		if [[ -z ${mountEntry} ]]
+		then
+			mount ${ubootDevice} ${ubootDir}
+			mounted=1
+		fi
+		[ -f ${ubootScript} ] && cfguBoot=1
+	fi
+fi
+
+# if we're using U-Boot, check if the default load address should change
+if [[ -n ${cfguBoot} ]] && [[ -z ${UBOOT_IMGADDR} ]]
+then
+	[[ ${version}  =~ .([^.]*)$ ]]
+	platform=${BASH_REMATCH[1]}
+	# A few platforms use an alternate kernel load address
+	if [[ ${platform} = omap ]]
+	then
+		ubootAddress=0x80008000
+	elif [[ ${platform} = imx ]]
+	then
+		ubootAddress=0x90008000
+	fi
+fi
 
 # if we have a lilo config on an x86 box, see if the default boot loader
 # is lilo to determine if it should be run
@@ -598,4 +897,7 @@
 	update
 fi
 
+# if we mounted the U-Boot directory, unmount it.
+[[ -n ${mounted} ]] && umount ${ubootDir}
+
 exit 0