From ef98fd9f96179cf3e55b1f65c4b2ea0b8d3203dd Mon Sep 17 00:00:00 2001
From: Scott Ullrich <sullrich@pfsense.org>
Date: Sat, 27 Jun 2009 13:47:15 -0400
Subject: [PATCH] Adding NanoBSD upgrade bits

---
 etc/rc.firmware | 149 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 149 insertions(+)

diff --git a/etc/rc.firmware b/etc/rc.firmware
index 8c4eab5121..9adb7450e1 100755
--- a/etc/rc.firmware
+++ b/etc/rc.firmware
@@ -147,6 +147,155 @@ upgrade)
 		rm -f /var/run/config.lock
 		sh /etc/rc.reboot
 	fi
+	;;
+pfSenseNanoBSDupgrade)
+
+	# Sanity check - bail early if there's no firmware file!
+	if [ ! -r $IMG ]; then
+		echo "2nd parameter has not been passed or file does not exist. Exiting." >> /cf/upgrade_log.txt 2>&1
+		exit 1
+	fi
+
+	touch /var/run/firmware.lock
+
+	echo "Firmware upgrade in progress..."  >> /cf/upgrade_log.txt 2>&1
+	echo "Firmware upgrade in progress..." | wall
+
+	# mount /cf
+	/etc/rc.conf_mount_rw
+
+	# backup config
+	/bin/mkdir -p /tmp/configbak
+	cp -p /conf/* /tmp/configbak 2>/dev/null
+
+	touch /cf/upgrade_log.txt
+	echo "" > /cf/upgrade_log.txt
+
+	echo "Installing $IMG." >> /cf/upgrade_log.txt 2>&1
+	echo "Installing $IMG." >> /cf/upgrade_log.txt
+
+	# resolve glabel label that we booted from
+	BOOT_DEVICE=`/sbin/mount | /usr/bin/grep pfsense | /usr/bin/cut -d'/' -f4 | /usr/bin/cut -d' ' -f1`
+	# resolve glabel to the real boot dev entry
+	REAL_BOOT_DEVICE=`/sbin/glabel list | /usr/bin/grep -B2 ufs/${BOOT_DEVICE} | /usr/bin/head -n 1 | /usr/bin/cut -f3 -d' '`
+	# grab the boot device, example ad1, ad0
+	BOOT_DRIVE=`/sbin/glabel list | /usr/bin/grep -B2 ufs/pfsense | /usr/bin/head -n 1 | /usr/bin/cut -f3 -d' ' | /usr/bin/cut -d's' -f1`
+	# test the slice.  if we are on slice 1 we need to flash 2 and vica versa
+	if [ `echo $REAL_BOOT_DEVICE | /usr/bin/grep "s1"` ]; then 
+		SLICE="2"
+		OLDSLICE="1"
+		TOFLASH="${BOOT_DRIVE}s${SLICE}"
+		COMPLETE_PATH="${BOOT_DRIVE}s${SLICE}a"
+		GLABEL_SLICE="pfsense1"
+	else
+		SLICE="1"
+		OLDSLICE="2"		
+		TOFLASH="${BOOT_DRIVE}s${SLICE}"
+		COMPLETE_PATH="${BOOT_DRIVE}s${SLICE}a"
+		GLABEL_SLICE="pfsense0"		
+	fi
+
+	# Output specifc information that this script is using
+	echo "SLICE $SLICE" >> /cf/upgrade_log.txt
+	echo "OLDSLICE $OLDSLICE" >> /cf/upgrade_log.txt
+	echo "TOFLASH $TOFLASH" >> /cf/upgrade_log.txt
+	echo "COMPLETE_PATH $COMPLETE_PATH" >> /cf/upgrade_log.txt
+	echo "GLABEL_SLICE $GLABEL_SLICE" >> /cf/upgrade_log.txt
+
+	# Output environment information to log file
+	output_env_to_log
+
+	# Foot shooting is fun!
+	echo "" >> /cf/upgrade_log.txt
+	echo "/sbin/sysctl kern.geom.debugflags=16" >> /cf/upgrade_log.txt
+	/sbin/sysctl kern.geom.debugflags=16 >> /cf/upgrade_log.txt 2>&1
+
+	# Remove TOFLASH and get ready for new flash image
+	echo "" >> /cf/upgrade_log.txt
+	echo "dd if=/dev/zero of=/dev/${TOFLASH} bs=1m count=1" >> /cf/upgrade_log.txt	
+	dd if=/dev/zero of=/dev/${TOFLASH} bs=1m count=1 >> /cf/upgrade_log.txt 2>&1
+
+	# Stream gzipped image to dd and explode image to new area
+	echo "" >> /cf/upgrade_log.txt
+	echo "/usr/bin/gunzip -S '' -c $IMG | /bin/dd of=/dev/${TOFLASH} bs=16k" >> /cf/upgrade_log.txt
+	/usr/bin/gunzip -S "" -c $IMG | /bin/dd of=/dev/${TOFLASH} bs=16k >> /cf/upgrade_log.txt 2>&1
+
+	# Ensure that our new system is sound and bail if it is not and file a notice
+	echo "" >> /cf/upgrade_log.txt
+	echo "/sbin/fsck_ffs -y /dev/$COMPLETE_PATH" >> /cf/upgrade_log.txt
+	/sbin/fsck_ffs -y /dev/$COMPLETE_PATH >> /cf/upgrade_log.txt 2>&1
+	if [ $? != 0 ]; then
+		/usr/local/bin/php -q -d auto_prepend_file=config.inc <<ENDOFF
+		<?php
+			require_once("globals.inc");		
+			require_once("functions.inc");
+			file_notice("UpgradeFailure", "{\$g['product_name']} upgrade has failed.   Your system has been left in a usable state.", "UpgradeFailure", "");
+		?>
+ENDOFF
+		rm /var/run/firmware.lock
+		exit 1
+	fi
+
+	# Add back the corresponding glabel
+	echo "" >> /cf/upgrade_log.txt
+	echo "/sbin/tunefs -L ${GLABEL_SLICE} /dev/$COMPLETE_PATH" >> /cf/upgrade_log.txt
+	/sbin/tunefs -L ${GLABEL_SLICE} /dev/$COMPLETE_PATH >> /cf/upgrade_log.txt 2>&1
+
+	# Set active boot source
+	echo "" >> /cf/upgrade_log.txt
+	echo "/usr/sbin/boot0cfg -s ${SLICE} -v /dev/${BOOT_DRIVE}" >> /cf/upgrade_log.txt
+	/usr/sbin/boot0cfg -s ${SLICE} -v /dev/${BOOT_DRIVE} >> /cf/upgrade_log.txt 2>&1
+
+	# Set active mount slice in fdisk
+	echo "" >> /cf/upgrade_log.txt
+	echo "gpart set -a active -i ${SLICE} ${BOOT_DRIVE}" >> /cf/upgrade_log.txt
+	gpart set -a active -i ${SLICE} ${BOOT_DRIVE}
+
+	# restore config
+	cp -p /tmp/configbak/* /conf 2>/dev/null
+
+	# Remove upgrade file
+	rm -f $IMG
+
+	# Mount newly prepared slice
+	mkdir /tmp/$GLABEL_SLICE
+	mount /dev/ufs/$GLABEL_SLICE /tmp/$GLABEL_SLICE
+
+	# If /tmp/$TOFLASH/tmp/post_upgrade_command exists 
+	# after update then execute the command.
+	if [ -f /tmp/$TOFLASH/tmp/post_upgrade_command ]; then
+		sh /tmp/$TOFLASH/tmp/post_upgrade_command >> /cf/upgrade_log.txt 2>&1
+	fi
+
+	# Update fstab
+	sed -i "" "s/pfsense${OLDSLICE}/pfsense${SLICE}/g" /tmp/$GLABEL_SLICE/etc/fstab
+	echo "" >> /cf/upgrade_log.txt
+	cat /tmp/$GLABEL_SLICE/etc/fstab >> /cf/upgrade_log.txt
+
+	# Unmount newly prepared slice
+	umount /tmp/$GLABEL_SLICE
+
+	# Remove extra stuff
+	rm -rf /etc/rc.conf
+	rm -rf /etc/motd
+	rm -rf /usr/savecore/*
+
+	date >> /cf/upgrade_log.txt
+	echo "" >> /cf/upgrade_log.txt
+
+	# remount /cf ro
+	/etc/rc.conf_mount_ro
+	/bin/sync
+	echo "Done." >> /cf/upgrade_log.txt 2>&1
+
+	# Enough fun for now.
+	echo "" >> /cf/upgrade_log.txt
+	echo "/sbin/sysctl kern.geom.debugflags=0" >> /cf/upgrade_log.txt
+	/sbin/sysctl kern.geom.debugflags=0 >> /cf/upgrade_log.txt 2>&1
+
+	rm -f /var/run/firmware.lock
+	sh /etc/rc.reboot
+
 	;;
 pfSenseupgrade)
 	# wait 5 seconds before beginning