#!/bin/sh
#
# rc.restore_ramdisk_store
#
# part of pfSense (https://www.pfsense.org)
# Copyright (c) 2004-2016 Rubicon Communications, LLC (Netgate)
# All rights reserved.
#
# Based on src/etc/rc.d/savecore from FreeBSD
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# Wildcard file existence check function
wildcard_file_exists() {
	for file in "$1"; do
		if [ -f "$file" ]; then
			return 0
		fi
	done
	return 1
}

: ${DBPATH:=/var/dhcpd/var/db}
: ${CF_CONF_PATH:=/cf/conf}

: ${RAM_Disk_Store:=${CF_CONF_PATH}/RAM_Disk_Store}

# See if there are any backup files (*.tgz) in the RAM disk store.  If so, restore them.
if wildcard_file_exists "${RAM_Disk_Store}/"*".tgz"; then

	mesg="Restoring contents of RAM disk store..."
	echo -n "$mesg"

	USE_MFS_TMPVAR=$(/usr/local/sbin/read_xml_tag.sh boolean system/use_mfs_tmpvar)

	# Restore the ram disk
	for backup_file in "${RAM_Disk_Store}/"*".tgz"; do
		if [ -f "$backup_file" ]; then
			/usr/bin/tar -xzf "${backup_file}" -C / 2>&1
			exit_code=$?
			if [ $exit_code -ne 0 ]; then
				mesg="$mesg\nRAM disk restore failed: exit code $exit_code: $backup_file"
				error=1;
				continue
			fi

			mesg="$mesg\nRAM disk restore succeeded: $backup_file"

			#If this backup is still there on a full install, but we aren't going to use ram disks, remove the archive since this is a transition.
			if [ "${USE_MFS_TMPVAR}" != "true" ]; then
				/bin/rm -f "${backup_file}"
			fi
		fi
	done

	if [ $error ]; then
		mesg="$mesg\nRAM disk restore failed."
		echo " error.";
	else
		mesg="$mesg\nRAM disk restore succeeded."
		echo " done.";
	fi

	/bin/mkdir -p /var/log/
	printf "$mesg" > /var/log/restore_ramdisk_store.boot

	# See if there are any backup files (*.tgz) in the RAM disk store.  If not, and no error, clean up.
	if ! wildcard_file_exists "${RAM_Disk_Store}/"*".tgz" && [ ! $error ]; then
		/bin/rmdir "$RAM_Disk_Store" >/dev/null 2>&1
		/bin/rm -f /var/log/restore_ramdisk_store.boot
	fi
fi