Project

General

Profile

Download (16.3 KB) Statistics
| Branch: | Tag: | Revision:
1 5b237745 Scott Ullrich
#!/bin/sh
2
3 84aa381e Scott Ullrich
# /etc/rc.firmware
4 79b9570c Scott Ullrich
# originally part of m0n0wall (http://neon1.net/m0n0wall)
5 498be2fc Scott Ullrich
# Copyright (C) 2005-2009 Scott Ullrich <sullrich@pfsense.org>.
6 84aa381e Scott Ullrich
# Copyright (C) 2003 Manuel Kasper <mk@neon1.net>.
7
# All rights reserved.
8 5b237745 Scott Ullrich
9 3e0621ca Scott Ullrich
# mount /cf
10
/etc/rc.conf_mount_rw
11
12 f22c94e0 Scott Ullrich
# Reset file(s)
13
echo "" >/conf/upgrade_log.txt
14 78d7e759 jim-p
echo "" >/conf/firmware_update_misc_log.txt
15 c5eb3a17 Scott Ullrich
echo "" >/conf/fdisk_upgrade_log.txt
16 f22c94e0 Scott Ullrich
17 78d7e759 jim-p
exec 3>&2 2>>/conf/firmware_update_misc_log.txt
18 51c9db03 Scott Ullrich
19 84aa381e Scott Ullrich
export ACTION=$1
20
export IMG=$2
21
if [ $# -eq 3 ]; then
22
	export CUSTOMIMG=$3
23
fi
24 2b61eeb1 Scott Ullrich
25 84aa381e Scott Ullrich
if [ $ACTION != "upgrade" ]; then
26
	/sbin/umount -f /ftmp > /dev/null 2>&1
27 fee835af Scott Ullrich
fi
28 5b237745 Scott Ullrich
29 62520171 Scott Ullrich
file_notice() {
30
	/usr/local/bin/php -q -d auto_prepend_file=config.inc <<ENDOFF
31
	<?php
32
		require_once("globals.inc");		
33
		require_once("functions.inc");
34
		file_notice("$1", "$2", "$1", "");
35
	?>
36
ENDOFF
37
}
38
39 79b9570c Scott Ullrich
output_env_to_log() {
40 6b3e4734 Scott Ullrich
	date >> /conf/upgrade_log.txt
41
	echo "" >> /conf/upgrade_log.txt
42 db7c7513 Scott Ullrich
	
43 6b3e4734 Scott Ullrich
	ls -lah /dev/ >> /conf/upgrade_log.txt
44
	echo "" >> /conf/upgrade_log.txt
45 79b9570c Scott Ullrich
46 6b3e4734 Scott Ullrich
	ls -lah $IMG >> /conf/upgrade_log.txt
47
	echo "" >> /conf/upgrade_log.txt
48 79b9570c Scott Ullrich
49 6b3e4734 Scott Ullrich
	md5 $IMG >> /conf/upgrade_log.txt
50
	echo "" >> /conf/upgrade_log.txt
51 79b9570c Scott Ullrich
52 6b3e4734 Scott Ullrich
	mount >> /conf/upgrade_log.txt
53
	echo "" >> /conf/upgrade_log.txt
54 79b9570c Scott Ullrich
55 6b3e4734 Scott Ullrich
	top >> /conf/upgrade_log.txt
56
	echo "" >> /conf/upgrade_log.txt
57 79b9570c Scott Ullrich
}
58
59 84aa381e Scott Ullrich
backup_chflags() {
60
	TOPROCESS="bin lib libexec sbin usr"
61
	for files in $TOPROCESS; do
62 d2307ffb Scott Ullrich
		/usr/sbin/mtree -Pcp /${files} | bzip2 -9 > /tmp/chflags.dist.${files}.bz2 2>> /conf/upgrade_log.txt 
63 84aa381e Scott Ullrich
	done
64
}
65 390c8e72 Scott Ullrich
66 84aa381e Scott Ullrich
restore_chflags() {
67
	TOPROCESS="bin lib libexec sbin usr"
68
	for files in $TOPROCESS; do
69 6b3e4734 Scott Ullrich
		cd / && /usr/bin/bzcat /tmp/chflags.dist.${files}.bz2 | /usr/sbin/mtree -PU -p /${files} >> /conf/upgrade_log.txt 2>&1
70 84aa381e Scott Ullrich
	done
71
}
72 8652a410 Scott Ullrich
73 84aa381e Scott Ullrich
remove_chflags() {
74
	TOPROCESS="bin lib libexec sbin usr"
75
	for files in $TOPROCESS; do
76
		/bin/chflags -R noschg /${files}
77
		/bin/chmod -R u+rw /${files}
78
	done
79
}
80 21fbee83 Scott Ullrich
81 84aa381e Scott Ullrich
binary_update() {
82
	TGZ=$1
83
	ERR_F="/tmp/bdiff.log"
84
	rm ${ERR_F} 2>/dev/null
85
	/bin/mkdir /tmp/patched /tmp/patches 2>>${ERR_F}
86 bf072179 jim-p
	# Save the old shutdown binary. If we switch from i386 to amd64 (or back) the reboot binary won't run at the end since it doesn't match up.
87 6fe72bc8 jim-p
	/bin/cp -p /sbin/shutdown /sbin/shutdown.old
88 84aa381e Scott Ullrich
	backup_chflags
89
	remove_chflags
90
	cd /tmp/patches
91
	for i in `/usr/bin/tar tvzf $TGZ | egrep -v "(^d|_md5)" | nawk '{print $9;}'`;
92
	 do
93
	   FILE=`basename ${i}`
94
	   echo "Working on ${i}"
95
	   # Untar patch file and md5 files
96
	   /usr/bin/tar xzf ${TGZ} ${i} ${i}.old_file_md5 ${i}.new_patch_md5 ${i}.new_file_md5 2>>${ERR_F}
97 d0b3b0b1 Scott Ullrich
98 84aa381e Scott Ullrich
	   # Apply patch - oldfile newfile patchfile
99
	   /usr/local/bin/bspatch /${i} /tmp/patched/${FILE} /tmp/patches/${i} 2>>${ERR_F}
100 d0b3b0b1 Scott Ullrich
101 84aa381e Scott Ullrich
	   OLD_FILE_MD5=`cat /tmp/patches/${i}.old_file_md5 2>/dev/null`
102
	   NEW_PATCH_MD5=`cat /tmp/patches/${i}.new_patch_md5 2>/dev/null`
103
	   NEW_FILE_MD5=`cat /tmp/patches/${i}.new_file_md5 2>/dev/null`
104
	   PATCHED_MD5=`/sbin/md5 -q /tmp/patched/${FILE} 2>/dev/null`
105 2b61eeb1 Scott Ullrich
106 84aa381e Scott Ullrich
	   if [ "$PATCHED_MD5" = "$NEW_PATCH_MD5" ]; then
107
		/usr/bin/install -S  /tmp/patched/${FILE} /${i}
108
	   else
109
		#echo "${i} file does not match intended final md5."
110
		echo "${i} file does not match intended final md5." >> ${ERR_F}
111
	   fi
112
113
	   /bin/rm /tmp/patched/${FILE} >> ${ERR_F}
114
	   /bin/rm /tmp/patches/${i} >> ${ERR_F}
115
	   /bin/rm /tmp/patches/${i}.* >> ${ERR_F}
116
	done
117
	/bin/rm -rf /tmp/patched /tmp/patches >> ${ERR_F}
118
	restore_chflags
119
}
120
121
case $ACTION in
122
enable)
123 6b3e4734 Scott Ullrich
	touch /conf/upgrade_log.txt
124 0e50dda6 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
125
	echo "Enable" >> /conf/upgrade_log.txt
126
	echo "" >> /conf/upgrade_log.txt		
127 b2f626b2 Ermal Lu?i
	/etc/rc.conf_mount_ro
128 84aa381e Scott Ullrich
	;;
129
auto)
130 a368a026 Ermal Lu?i
	touch /var/run/firmwarelock.dirty
131 84aa381e Scott Ullrich
	backup_chflags
132
	remove_chflags
133
	/etc/rc.firmware_auto
134
	restore_chflags
135 b2f626b2 Ermal Lu?i
	/etc/rc.conf_mount_ro
136 84aa381e Scott Ullrich
	;;
137 c2aff9da Scott Ullrich
pfSenseNanoBSDupgrade)
138
139
	# Sanity check - bail early if there's no firmware file!
140
	if [ ! -r $IMG ]; then
141 6b3e4734 Scott Ullrich
		echo "2nd parameter has not been passed or file does not exist. Exiting." >> /conf/upgrade_log.txt 2>&1
142 bc43f0ee Scott Ullrich
		/etc/rc.conf_mount_ro
143 ac8de020 Scott Ullrich
		exit 1
144 c2aff9da Scott Ullrich
	fi
145
146 9581e85b Scott Ullrich
	# Prevent full upgrade file from being used to upgrade
147
	if [ `echo $IMG | grep "full"` ]; then
148 a331bc2a sullrich
		echo "You cannot use a full file for upgrade.  Please use a file labeled nanobsd upgrade."
149 62520171 Scott Ullrich
		file_notice "NanoBSDUpgradeFailure" "You have attemped to use a full NanoBSD installation file as an upgrade.  Please use a NanoBSD file labeled 'upgrade' instead."
150 9974dae5 Scott Ullrich
		rm -f $IMG
151 bc43f0ee Scott Ullrich
		/etc/rc.conf_mount_ro		
152 9581e85b Scott Ullrich
		exit 1
153
	fi
154
155 a368a026 Ermal Lu?i
	touch /var/run/firmwarelock.dirty
156 6a581b52 Scott Ullrich
157 6b3e4734 Scott Ullrich
	echo "NanoBSD Firmware upgrade in progress..."  >> /conf/upgrade_log.txt 2>&1
158 e3ddf385 Scott Ullrich
	echo "NanoBSD Firmware upgrade in progress..." | wall
159 476aa14b Scott Ullrich
	/etc/rc.firmware_notify -e -g -m "NanoBSD Firmware upgrade in progress..."
160
	
161 c2aff9da Scott Ullrich
	# backup config
162
	/bin/mkdir -p /tmp/configbak
163 ce09186c jim-p
	cp -Rp /conf/* /tmp/configbak 2>/dev/null
164 c2aff9da Scott Ullrich
165 2cd85631 jim-p
	# Remove logs from backup dir to avoid clobbering upon restore.
166
	rm /tmp/configbak/*_log.txt 2>/dev/null
167
168 0e50dda6 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
169 36fcc0ad Scott Ullrich
170 e588bd54 jim-p
	echo "Installing ${IMG}." >> /conf/upgrade_log.txt 2>&1
171
	echo "Installing ${IMG}." | wall
172 f93a2816 Scott Ullrich
173 15533c1e Scott Ullrich
	# resolve glabel label that we booted from
174 69025059 Scott Ullrich
	BOOT_DEVICE=`/sbin/mount | /usr/bin/grep pfsense | /usr/bin/cut -d'/' -f4 | /usr/bin/cut -d' ' -f1`
175 15533c1e Scott Ullrich
	# resolve glabel to the real boot dev entry
176 004675b6 Scott Ullrich
	REAL_BOOT_DEVICE=`/sbin/glabel list | /usr/bin/grep -B2 ufs/${BOOT_DEVICE} | /usr/bin/head -n 1 | /usr/bin/cut -f3 -d' '`
177 15533c1e Scott Ullrich
	# grab the boot device, example ad1, ad0
178 bbae22d6 Scott Ullrich
	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`
179 15533c1e Scott Ullrich
	# test the slice.  if we are on slice 1 we need to flash 2 and vica versa
180 004675b6 Scott Ullrich
	if [ `echo $REAL_BOOT_DEVICE | /usr/bin/grep "s1"` ]; then 
181
		SLICE="2"
182 63f29ad5 Scott Ullrich
		OLDSLICE="1"
183 99760ae6 Scott Ullrich
		TOFLASH="${BOOT_DRIVE}s${SLICE}"
184 63f29ad5 Scott Ullrich
		COMPLETE_PATH="${BOOT_DRIVE}s${SLICE}a"
185 93585de9 Scott Ullrich
		GLABEL_SLICE="pfsense1"
186 feff69c9 Scott Ullrich
		UFS_ID="1"
187 36bb9bea Scott Ullrich
		OLD_UFS_ID="0"
188 e348af75 Scott Ullrich
	else
189 004675b6 Scott Ullrich
		SLICE="1"
190 63f29ad5 Scott Ullrich
		OLDSLICE="2"		
191 99760ae6 Scott Ullrich
		TOFLASH="${BOOT_DRIVE}s${SLICE}"
192 63f29ad5 Scott Ullrich
		COMPLETE_PATH="${BOOT_DRIVE}s${SLICE}a"
193 feff69c9 Scott Ullrich
		GLABEL_SLICE="pfsense0"
194
		UFS_ID="0"
195 36bb9bea Scott Ullrich
		OLD_UFS_ID="1"
196 e348af75 Scott Ullrich
	fi
197
198 79b9570c Scott Ullrich
	# Output specifc information that this script is using
199 e588bd54 jim-p
	echo "SLICE         ${SLICE}" >> /conf/upgrade_log.txt
200
	echo "OLDSLICE      ${OLDSLICE}" >> /conf/upgrade_log.txt
201
	echo "TOFLASH       ${TOFLASH}" >> /conf/upgrade_log.txt
202
	echo "COMPLETE_PATH ${COMPLETE_PATH}" >> /conf/upgrade_log.txt
203
	echo "GLABEL_SLICE  ${GLABEL_SLICE}" >> /conf/upgrade_log.txt
204 a3b8edea Scott Ullrich
205 b90885ff Scott Ullrich
	# First ensure the new file can fit inside the 
206
	# slice that we are going to be operating on.
207
	NEW_IMG_SIZE=`echo $((\`gzip -l ${IMG} | grep -v compressed | awk '{ print $2}'\` / 1024 / 1024))`
208
	SIZE=`/sbin/fdisk ${COMPLETE_PATH} | /usr/bin/grep Meg | /usr/bin/awk '{ print $5 }' | /usr/bin/cut -d"(" -f2`
209 80682f55 Scott Ullrich
	SIZE=`expr $SIZE + 1`
210 b90885ff Scott Ullrich
	if [ "$SIZE" -lt "$NEW_IMG_SIZE" ]; then
211 5d15e968 Scott Ullrich
		file_notice "UpgradeFailure" "Upgrade failed due to the upgrade image being larger than the partition that is configured on disk.  Halting. Size on disk: $SIZE < Size of new image: $NEW_IMG_SIZE"
212 b90885ff Scott Ullrich
		echo "Upgrade failed.  Please check the system log file for more information" | wall
213 9974dae5 Scott Ullrich
		rm -f $IMG
214 7335c4cf Scott Ullrich
		rm -f /var/run/firmwarelock.dirty
215
		rm -f /var/run/firmware.lock
216 3bcd5555 Renato Botelho
		rm -f ${IMG}
217 b90885ff Scott Ullrich
		/etc/rc.conf_mount_ro		
218
		exit 1
219
	fi
220
221 79b9570c Scott Ullrich
	# Output environment information to log file
222
	output_env_to_log
223 6b3e4734 Scott Ullrich
	
224
	# Grab a before upgrade look at fdisk
225 c5eb3a17 Scott Ullrich
	echo "" >> /conf/fdisk_upgrade_log.txt
226
	echo "Before upgrade fdisk/bsdlabel" >> /conf/fdisk_upgrade_log.txt
227
	fdisk $BOOT_DRIVE >> /conf/fdisk_upgrade_log.txt
228 e588bd54 jim-p
	bsdlabel -A ${BOOT_DRIVE}s1 >> /conf/fdisk_upgrade_log.txt
229
	bsdlabel -A ${BOOT_DRIVE}s2 >> /conf/fdisk_upgrade_log.txt
230
	bsdlabel -A ${BOOT_DRIVE}s3 >> /conf/fdisk_upgrade_log.txt
231 c5eb3a17 Scott Ullrich
	echo "---------------------------------------------------------------" >> /conf/fdisk_upgrade_log.txt
232
	echo "" >> /conf/fdisk_upgrade_log.txt
233 6b3e4734 Scott Ullrich
	
234 024445ee Scott Ullrich
	# Log that we are really doing a NanoBSD upgrade
235 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
236
	echo "NanoBSD upgrade starting" >> /conf/upgrade_log.txt
237
	echo "" >> /conf/upgrade_log.txt
238 a3b8edea Scott Ullrich
239 e348af75 Scott Ullrich
	# Remove TOFLASH and get ready for new flash image
240 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
241
	echo "dd if=/dev/zero of=/dev/${TOFLASH} bs=1m count=1" >> /conf/upgrade_log.txt	
242
	dd if=/dev/zero of=/dev/${TOFLASH} bs=1m count=1 >> /conf/upgrade_log.txt 2>&1
243 e348af75 Scott Ullrich
244
	# Stream gzipped image to dd and explode image to new area
245 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
246 ac2bed30 Scott Ullrich
	echo "/usr/bin/gzip -dc $IMG | /bin/dd of=/dev/${TOFLASH} obs=64k" >> /conf/upgrade_log.txt
247
	/usr/bin/gzip -dc $IMG | /bin/dd of=/dev/${TOFLASH} obs=64k >> /conf/upgrade_log.txt 2>&1
248 6b3e4734 Scott Ullrich
249
	# Grab a after upgrade look at fdisk
250 c5eb3a17 Scott Ullrich
	echo "" >> /conf/fdisk_upgrade_log.txt
251 0e50dda6 Scott Ullrich
	echo "After upgrade fdisk/bsdlabel" >> /conf/upgrade_log.txt
252 c5eb3a17 Scott Ullrich
	fdisk $BOOT_DRIVE >> /conf/fdisk_upgrade_log.txt
253 e588bd54 jim-p
	bsdlabel -A ${BOOT_DRIVE}s1 >> /conf/fdisk_upgrade_log.txt
254
	bsdlabel -A ${BOOT_DRIVE}s2 >> /conf/fdisk_upgrade_log.txt
255
	bsdlabel -A ${BOOT_DRIVE}s3 >> /conf/fdisk_upgrade_log.txt
256 c5eb3a17 Scott Ullrich
	echo "---------------------------------------------------------------" >> /conf/fdisk_upgrade_log.txt
257
	echo "" >> /conf/fdisk_upgrade_log.txt
258 e348af75 Scott Ullrich
	
259
	# Ensure that our new system is sound and bail if it is not and file a notice
260 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
261 e588bd54 jim-p
	echo "/sbin/fsck_ufs -y /dev/${COMPLETE_PATH}" >> /conf/upgrade_log.txt
262
	/sbin/fsck_ufs -y /dev/${COMPLETE_PATH} >> /conf/upgrade_log.txt 2>&1
263 e348af75 Scott Ullrich
	if [ $? != 0 ]; then
264 62520171 Scott Ullrich
		file_notice "UpgradeFailure" "{\$g['product_name']} upgrade has failed.   Your system has been left in a usable state."
265 9974dae5 Scott Ullrich
		rm -f $IMG
266 7335c4cf Scott Ullrich
		rm -f /var/run/firmwarelock.dirty
267
		rm -f /var/run/firmware.lock
268 bc43f0ee Scott Ullrich
		/etc/rc.conf_mount_ro		
269 9492ad9d Scott Ullrich
		exit 1
270 e348af75 Scott Ullrich
	fi
271
272 0bc8c0b0 Scott Ullrich
	# Enable foot shooting
273 32a6b30c Scott Ullrich
	sysctl kern.geom.debugflags=16
274
275 63f29ad5 Scott Ullrich
	# Add back the corresponding glabel
276 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
277 e588bd54 jim-p
	echo "/sbin/tunefs -L ${GLABEL_SLICE} /dev/${COMPLETE_PATH}" >> /conf/upgrade_log.txt
278
	/sbin/tunefs -L ${GLABEL_SLICE} /dev/${COMPLETE_PATH} >> /conf/upgrade_log.txt 2>&1
279 63f29ad5 Scott Ullrich
280 c2aff9da Scott Ullrich
	# restore config
281 ce09186c jim-p
	cp -Rp /tmp/configbak/* /conf 2>/dev/null
282 c2aff9da Scott Ullrich
283
	# Remove upgrade file
284
	rm -f $IMG
285
286 5924c7f8 Scott Ullrich
	# Mount newly prepared slice
287 db7c7513 Scott Ullrich
	mkdir /tmp/$GLABEL_SLICE
288
	mount /dev/ufs/$GLABEL_SLICE /tmp/$GLABEL_SLICE
289 5924c7f8 Scott Ullrich
290 ba8e0870 Erik Fonnesbeck
	# If /boot/loader.conf.local exists
291
	# copy to the other slice.
292
	if [ -f /boot/loader.conf.local ]; then
293
		cp /boot/loader.conf.local /tmp/$GLABEL_SLICE/boot/loader.conf.local
294
	fi
295
296 8eaa9957 Scott Ullrich
	# If /tmp/$GLABEL_SLICE/tmp/post_upgrade_command exists 
297 baabd21d Scott Ullrich
	# after update then execute the command.
298 8eaa9957 Scott Ullrich
	if [ -f /tmp/$GLABEL_SLICE/tmp/post_upgrade_command ]; then
299 6b3e4734 Scott Ullrich
		sh /tmp/$GLABEL_SLICE/tmp/post_upgrade_command >> /conf/upgrade_log.txt 2>&1
300 c2aff9da Scott Ullrich
	fi
301 5924c7f8 Scott Ullrich
302 63f29ad5 Scott Ullrich
	# Update fstab
303 507960cb Scott Ullrich
	cp /etc/fstab /tmp/$GLABEL_SLICE/etc/fstab
304 36bb9bea Scott Ullrich
	sed -i "" "s/pfsense${OLD_UFS_ID}/pfsense${UFS_ID}/g" /tmp/$GLABEL_SLICE/etc/fstab
305 45c710da Scott Ullrich
	if [ $? != 0 ]; then
306
		echo "Something went wrong when trying to update the fstab entry.  Aborting upgrade."
307
		file_notice "UpgradeFailure" "Something went wrong when trying to update the fstab entry.  Aborting upgrade."
308 9974dae5 Scott Ullrich
		rm -f $IMG
309 7335c4cf Scott Ullrich
		rm -f /var/run/firmwarelock.dirty
310
		rm -f /var/run/firmware.lock
311 45c710da Scott Ullrich
		umount /tmp/$GLABEL_SLICE
312 bc43f0ee Scott Ullrich
		/etc/rc.conf_mount_ro
313 45c710da Scott Ullrich
		exit 1
314
	fi
315 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
316
	cat /tmp/$GLABEL_SLICE/etc/fstab >> /conf/upgrade_log.txt
317 5924c7f8 Scott Ullrich
318 53e85f50 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
319 af214f1d Scott Ullrich
	find /tmp/$GLABEL_SLICE >/conf/file_upgrade_log.txt
320 53e85f50 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
321
322 5924c7f8 Scott Ullrich
	# Unmount newly prepared slice
323 db7c7513 Scott Ullrich
	umount /tmp/$GLABEL_SLICE
324 7f38e934 Scott Ullrich
	
325
	sync
326 c2aff9da Scott Ullrich
327 45c710da Scott Ullrich
	# Set active mount slice in fdisk
328 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
329
	echo "gpart set -a active -i ${SLICE} ${BOOT_DRIVE}" >> /conf/upgrade_log.txt
330 0bc8c0b0 Scott Ullrich
	gpart set -a active -i ${SLICE} ${BOOT_DRIVE} >> /conf/upgrade_log.txt 2>&1
331 45c710da Scott Ullrich
332 7f38e934 Scott Ullrich
	sync
333
334 e213b6fc Scott Ullrich
	# Set active boot source - NanoBSD does not do this but otherwise we
335
	# end up with the wrong partition being active.
336 b4047760 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
337
	echo "/usr/sbin/boot0cfg -s ${SLICE} -v /dev/${BOOT_DRIVE}" >> /conf/upgrade_log.txt
338
	/usr/sbin/boot0cfg -s ${SLICE} -v /dev/${BOOT_DRIVE} >> /conf/upgrade_log.txt 2>&1
339 6b3e4734 Scott Ullrich
340 0bc8c0b0 Scott Ullrich
	# Disable foot shooting
341 32a6b30c Scott Ullrich
	sysctl kern.geom.debugflags=0
342
343 6b3e4734 Scott Ullrich
	# Grab a final look at fdisk
344 c5eb3a17 Scott Ullrich
	echo "" >> /conf/fdisk_upgrade_log.txt
345
	echo "Final upgrade fdisk/bsdlabel" >> /conf/fdisk_upgrade_log.txt
346
	fdisk $BOOT_DRIVE >> /conf/fdisk_upgrade_log.txt
347 e588bd54 jim-p
	bsdlabel -A ${BOOT_DRIVE}s1 >> /conf/fdisk_upgrade_log.txt
348
	bsdlabel -A ${BOOT_DRIVE}s2 >> /conf/fdisk_upgrade_log.txt
349
	bsdlabel -A ${BOOT_DRIVE}s3 >> /conf/fdisk_upgrade_log.txt
350 c5eb3a17 Scott Ullrich
	echo "---------------------------------------------------------------" >> /conf/fdisk_upgrade_log.txt
351
	echo "" >> /conf/fdisk_upgrade_log.txt
352 45c710da Scott Ullrich
353 5924c7f8 Scott Ullrich
	# Remove extra stuff
354 c2aff9da Scott Ullrich
	rm -rf /etc/rc.conf
355
	rm -rf /etc/motd
356
	rm -rf /usr/savecore/*
357
358 6b3e4734 Scott Ullrich
	date >> /conf/upgrade_log.txt
359
	echo "" >> /conf/upgrade_log.txt
360 db7c7513 Scott Ullrich
361 359d9ec9 Scott Ullrich
	# Trigger a package reinstallation on reobot
362 65c79f55 Scott Ullrich
	touch /conf/needs_package_sync
363 359d9ec9 Scott Ullrich
364 5924c7f8 Scott Ullrich
	# remount /cf ro
365
	/etc/rc.conf_mount_ro
366 c2aff9da Scott Ullrich
	/bin/sync
367 903f12d4 Scott Ullrich
368 d29d05ab Scott Ullrich
	echo "NanoBSD Firmware upgrade is complete.  Rebooting in 10 seconds."  >> /conf/upgrade_log.txt 2>&1
369
	echo "NanoBSD Firmware upgrade is complete.  Rebooting in 10 seconds." | wall
370 476aa14b Scott Ullrich
	/etc/rc.firmware_notify -e -g -m "NanoBSD Firmware upgrade is complete.  Rebooting in 10 seconds."
371 d29d05ab Scott Ullrich
372 d4b8f9a8 Scott Ullrich
	sleep 10
373
374 a368a026 Ermal Lu?i
	rm -f /var/run/firmwarelock.dirty
375 7335c4cf Scott Ullrich
	rm -f /var/run/firmware.lock
376 5423f1a0 Scott Ullrich
	sh /etc/rc.reboot
377 c2aff9da Scott Ullrich
378 84aa381e Scott Ullrich
	;;
379
pfSenseupgrade)
380 50ba07ea Scott Ullrich
381 84aa381e Scott Ullrich
	# Sanity check - bail early if there's no firmware file!
382
	if [ ! -r $IMG ]; then
383 6b3e4734 Scott Ullrich
		echo "2nd parameter has not been passed or file does not exist. Exiting." >> /conf/upgrade_log.txt 2>&1
384 50ba07ea Scott Ullrich
		/etc/rc.conf_mount_ro
385 84aa381e Scott Ullrich
		exit
386
	fi
387
388 6a581b52 Scott Ullrich
	# wait 1 seconds before beginning
389
	sleep 1
390
391 d2307ffb Scott Ullrich
	# Log that we are really doing a pfSense upgrade
392 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
393 efdf0b9c sullrich
	echo "pfSenseupgrade upgrade starting" >> /conf/upgrade_log.txt
394 6b3e4734 Scott Ullrich
	echo "" >> /conf/upgrade_log.txt
395 50ba07ea Scott Ullrich
396 a368a026 Ermal Lu?i
	touch /var/run/firmwarelock.dirty
397 6a581b52 Scott Ullrich
398 6b3e4734 Scott Ullrich
	touch /conf/upgrade_log.txt
399
	echo "" >> /conf/upgrade_log.txt
400 79b9570c Scott Ullrich
401
	# Output environment information to log file
402
	output_env_to_log
403
404 84aa381e Scott Ullrich
	backup_chflags
405
	remove_chflags
406
407 4a8e802d Scott Ullrich
	# Do we have a pre-upgrade hook in the update file?
408
	if [ `tar tvzf $IMG | grep /tmp/pre_upgrade_command | wc -l` -gt 0 ]; then 
409 6b3e4734 Scott Ullrich
		tar xzvf $IMG -C / ./tmp/pre_upgrade_command >> /conf/upgrade_log.txt 2>&1
410
		chmod a+rx /tmp/pre_upgrade_command >> /conf/upgrade_log.txt 2>&1
411
		sh /tmp/pre_upgrade_command >> /conf/upgrade_log.txt 2>&1
412 4a8e802d Scott Ullrich
	fi
413
414 6b3e4734 Scott Ullrich
	echo "Firmware upgrade in progress..."  >> /conf/upgrade_log.txt 2>&1
415 79b9570c Scott Ullrich
	echo "Firmware upgrade in progress..."  | wall
416 476aa14b Scott Ullrich
	/etc/rc.firmware_notify -e -g -m "Firmware upgrade in progress..."
417
	
418 84aa381e Scott Ullrich
	# backup config
419
	/bin/mkdir -p /tmp/configbak
420 ce09186c jim-p
	cp -Rp /conf/* /tmp/configbak 2>/dev/null
421 84aa381e Scott Ullrich
422 2cd85631 jim-p
	# Remove logs from backup dir to avoid clobbering upon restore.
423
	rm /tmp/configbak/*_log.txt 2>/dev/null
424
425 84aa381e Scott Ullrich
	# tar explode image onto hd
426 6b3e4734 Scott Ullrich
	echo "Installing $IMG." >> /conf/upgrade_log.txt 2>&1
427
	cd / && /usr/bin/tar xzUPf $IMG >> /conf/upgrade_log.txt 2>&1
428 84aa381e Scott Ullrich
	/usr/bin/find / -name CVS -exec rm -fr {} \;
429 6b3e4734 Scott Ullrich
	echo "Image installed $IMG." >> /conf/upgrade_log.txt 2>&1
430 84aa381e Scott Ullrich
431
    # process custom image if its passed
432
    if [ $# -eq 3 ]; then
433
	    if [ -f $CUSTOMIMG ]; then
434 6b3e4734 Scott Ullrich
	        echo "Custom image $CUSTOMIMG found." >> /conf/upgrade_log.txt 2>&1
435
	        echo "Custom image ($CUSTOMIMG) found." >> /conf/upgrade_log.txt 2>&1
436 84aa381e Scott Ullrich
	        PWD_DIR=`pwd`
437 6b3e4734 Scott Ullrich
	        cd / && /usr/bin/tar xzPUf $CUSTOMIMG >> /conf/upgrade_log.txt 2>&1
438 84aa381e Scott Ullrich
	        cd $PWD_DIR
439 6b3e4734 Scott Ullrich
	        echo "Custom image $CUSTOMIMG installed." >> /conf/upgrade_log.txt 2>&1
440 84aa381e Scott Ullrich
	    fi
441
    fi
442
443
	# restore config
444 ce09186c jim-p
	cp -Rp /tmp/configbak/* /conf 2>/dev/null
445 84aa381e Scott Ullrich
446
	# restore /etc symlinks
447
	rm /etc/hosts
448
	ln -s /var/etc/hosts /etc/hosts
449
450
	restore_chflags
451
452
	# Remove upgrade file
453
	rm -f $IMG
454
455 6e75ac00 Scott Ullrich
	if [ -e /etc/init_bootloader.sh ]; then
456 6b3e4734 Scott Ullrich
		sh /etc/init_bootloader.sh >> /conf/upgrade_log.txt 2>&1
457 6e75ac00 Scott Ullrich
	fi
458 84aa381e Scott Ullrich
459 6e75ac00 Scott Ullrich
	# If /tmp/post_upgrade_command exists after update
460
	# then execute the command.
461
	if [ -f /tmp/post_upgrade_command ]; then
462 6b3e4734 Scott Ullrich
		sh /tmp/post_upgrade_command >> /conf/upgrade_log.txt 2>&1
463 6e75ac00 Scott Ullrich
	fi
464
465 79b9570c Scott Ullrich
	# remove unused files
466 84aa381e Scott Ullrich
	rm -rf /etc/rc.conf
467
	rm -rf /etc/motd
468
	rm -rf /usr/savecore/*
469 79b9570c Scott Ullrich
470 6b3e4734 Scott Ullrich
	date >> /conf/upgrade_log.txt
471
	echo "" >> /conf/upgrade_log.txt
472 db7c7513 Scott Ullrich
473 79b9570c Scott Ullrich
	# remount /cf ro
474 84aa381e Scott Ullrich
	/etc/rc.conf_mount_ro
475 6e75ac00 Scott Ullrich
476 79b9570c Scott Ullrich
	# release the firmware lock
477 a368a026 Ermal Lu?i
	rm -f /var/run/firmwarelock.dirty
478 7335c4cf Scott Ullrich
	rm -f /var/run/firmware.lock
479 84aa381e Scott Ullrich
	/bin/sync
480
481 d29d05ab Scott Ullrich
	echo "Firmware upgrade is complete. Rebooting in 10 seconds."  >> /conf/upgrade_log.txt 2>&1
482
	echo "Firmware upgrade is complete. Rebooting in 10 seconds."  | wall
483 476aa14b Scott Ullrich
	/etc/rc.firmware_notify -e -g -m "Firmware upgrade is complete. Rebooting in 10 seconds."
484 d29d05ab Scott Ullrich
485 d4b8f9a8 Scott Ullrich
	# Sleep and allow disks to catch up
486
	sleep 10
487
488 84aa381e Scott Ullrich
	# If the archive has unpacked a file called
489
	# /tmp/no_upgrade_reboot_required then do
490
	# not reboot after upgrade.
491
	if [ -f /tmp/no_upgrade_reboot_required ]; then
492
		rm /tmp/no_upgrade_reboot_required
493
	else
494
		sh /etc/rc.reboot
495
	fi
496 6e75ac00 Scott Ullrich
497 84aa381e Scott Ullrich
	;;
498
delta_update)
499 a368a026 Ermal Lu?i
	touch /var/run/firmwarelock.dirty
500 84aa381e Scott Ullrich
	backup_chflags
501
	remove_chflags
502
	binary_update $IMG
503
	restore_chflags
504
	rm -rf /etc/rc.conf
505
	rm -rf /etc/motd
506
	find / -name CVS -type d -exec rm {} \;
507
	rm -rf /usr/savecore/*
508
	/etc/rc.conf_mount_ro
509
	/sbin/umount -f /cf 2>/dev/null
510
	/sbin/mount -r /cf 2>/dev/null
511
	/sbin/umount -f / 2>/dev/null
512
	/sbin/mount -r / 2>/dev/null
513
	if [ -e /etc/init_bootloader.sh ]; then
514
		sh /etc/init_bootloader.sh
515
	fi
516
517
	;;
518
esac