Consider globals_override.inc and replace hard coded pfSense string by product_name

1 files changed, 15 insertions, 3 deletions

diff --git a/src/etc/rc.reboot b/src/etc/rc.reboot
index afdc93c..3b77520 100755
--- a/src/etc/rc.reboot
+++ b/src/etc/rc.reboot
@@ -9,10 +9,22 @@ fi
 
 sleep 1
 
-# If PLATFORM is pfSense then remove
+globals_inc="/etc/inc/globals.inc"
+if [ -f /etc/inc/globals_override.inc ]; then
+	globals_inc="/etc/inc/globals_override.inc ${globals_inc}"
+fi
+product=$(cat ${globals_inc} | \
+	grep product_name | \
+	head -n 1 | \
+	sed 's/^.*=>* *//; s/["\;,]*//g')
+
+# product cannot be blank
+product=${product:-"pfSense"}
+
+# If PLATFORM is $product then remove
 # temporary files on shutdown from /tmp/
 PLATFORM=`cat /etc/platform`
-if [ "$PLATFORM" = "pfSense" ]; then
+if [ "$PLATFORM" = "${product}" ]; then
 	rm -rf /tmp/*
 fi
 
@@ -20,7 +32,7 @@ USE_MFS_TMPVAR=`/usr/bin/grep -c use_mfs_tmpvar /cf/conf/config.xml`
 DISK_NAME=`/bin/df /var/db/rrd | /usr/bin/tail -1 | /usr/bin/awk '{print $1;}'`
 DISK_TYPE=`/usr/bin/basename ${DISK_NAME} | /usr/bin/cut -c1-2`
 # If we are not on a full install, or if the full install wants RAM disks, or if the full install _was_ using RAM disks, but isn't for the next boot...
-if [ "${PLATFORM}" != "pfSense" ] || [ ${USE_MFS_TMPVAR} -gt 0 ] || [ "${DISK_TYPE}" = "md" ]; then
+if [ "${PLATFORM}" != "${product}" ] || [ ${USE_MFS_TMPVAR} -gt 0 ] || [ "${DISK_TYPE}" = "md" ]; then
 	/etc/rc.backup_rrd.sh
 	/etc/rc.backup_dhcpleases.sh
 fi