-#!/bin/sh
+###############################################################################
+# #
+# IPFire.org - A linux based firewall #
+# Copyright (C) 2007-2012 IPFire Team <info@ipfire.org> #
+# #
+# This program is free software: you can redistribute it and/or modify #
+# it under the terms of the GNU General Public License as published by #
+# the Free Software Foundation, either version 3 of the License, or #
+# (at your option) any later version. #
+# #
+# This program is distributed in the hope that it will be useful, #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
+# GNU General Public License for more details. #
+# #
+# You should have received a copy of the GNU General Public License #
+# along with this program. If not, see <http://www.gnu.org/licenses/>. #
+# #
+###############################################################################
+
+# Set histchars to an empty string so we are able to replace an
+# exclamation mark.
+histchars=
echo "Scanning for possible destination drives"
-# scan IDE devices
-echo "--> IDE"
-for DEVICE in $(kudzu -qps -t 30 -c HD -b IDE | grep device: | cut -d ' ' -f 2 | sort | uniq); do
- echo -n "---> $DEVICE"
- mount /dev/${DEVICE}1 /harddisk 2> /dev/null
- if [ -n "$(ls /harddisk/ipfire-*.tbz2 2>/dev/null)" ]; then
- umount /harddisk 2> /dev/null
- echo "${DEVICE} is source drive - skipping"
- continue
- else
- umount /harddisk 2> /dev/null
- echo -n "$DEVICE" > /tmp/dest_device
- echo " - yes, it is our destination"
- exit 0
+function _mount() {
+ local what=${1}
+
+ # Don't mount if the device does not exist.
+ [ -e "${what}" ] || return 1
+
+ mount ${what} /harddisk 2>/dev/null
+}
+
+function _umount() {
+ umount -l /harddisk 2>/dev/null
+}
+
+function check_source_drive() {
+ local device="/dev/${1}"
+
+ local ret=1
+ local dev
+ for dev in ${device} ${device}1; do
+ # Mount the device (if possible).
+ _mount ${dev} || continue
+
+ if [ -n "$(ls /harddisk/ipfire-*.tlz 2>/dev/null)" ]; then
+ ret=0
fi
-done
- mount /dev/${DEVICE}1 /cdrom 2> /dev/null
- if [ ]; then
- echo -n ${DEVICE} > /tmp/source_device
- echo "Found Sources in ${DEVICE}"
- else
- umount /cdrom 2> /dev/null
- echo "Found no Sources in ${DEVICE} skipping"
- fi
- umount /cdrom 2> /dev/null
-
-
-
-# scan USB/SCSI devices
-echo "--> USB/SCSI"
-for DEVICE in $(kudzu -qps -t 30 -c HD -b SCSI | grep device: | cut -d ' ' -f 2 | sort | uniq); do
- echo -n "---> $DEVICE"
- mount /dev/${DEVICE}1 /harddisk 2> /dev/null
- if [ -n "$(ls /harddisk/ipfire-*.tbz2 2>/dev/null)" ]; then
- umount /harddisk 2> /dev/null
- echo "${DEVICE} is source drive - skipping"
+ _umount
+
+ # Stop if the device has been detected as a source drive.
+ [ "${ret}" = "0" ] && break
+ done
+
+ return ${ret}
+}
+
+for path in /sys/block/*; do
+ device=$(basename ${path})
+
+ # Skip devices which cannot be used.
+ case "${device}" in
+ # Virtual devices.
+ loop*|ram*)
+ continue
+ ;;
+ # Floppy.
+ fd*)
continue
- else
- umount /harddisk 2> /dev/null
- echo -n "$DEVICE" > /tmp/dest_device
- echo " - yes, it is our destination"
- exit 1
+ ;;
+ esac
+
+ # Replace any exclamation marks (e.g. cciss!c0d0).
+ device=${device//!/\/}
+
+ # Guess if this could be a raid device.
+ for dev in ${device} ${device}p1; do
+ if [ -e "/dev/${dev}" ]; then
+ device=${dev}
+ break
fi
-done
+ done
-# scan RAID devices
-echo "--> RAID"
-for DEVICE in $(kudzu -qps -t 30 -c HD -b RAID | grep device: | cut -d ' ' -f 2 | sort | uniq); do
- echo -n "---> $DEVICE"
- mount /dev/${DEVICE}p1 /harddisk 2> /dev/null
- if [ -n "$(ls /harddisk/ipfire-*.tbz2 2>/dev/null)" ]; then
- umount /harddisk 2> /dev/null
- echo "${DEVICE} is source drive - skipping"
- echo " is source drive"
- continue
- else
- umount /harddisk 2> /dev/null
- echo -n "$DEVICE" > /tmp/dest_device
- echo " - yes, it is our destination"
+ echo "Checking ${device}"
+ if check_source_drive ${device}; then
+ echo " is source drive - skipping"
+ continue
+ fi
+
+ # Found it.
+ echo " OK, this is it..."
+ echo -n "${device}" > /tmp/dest_device
+
+ # Exit code table:
+ # 1: sda
+ # 2: RAID
+ # 10: nothing found
+ case "${device}" in
+ *p1|*c0d0)
exit 2
- fi
+ ;;
+ *)
+ exit 1
+ ;;
+ esac
done
-exit 10 # Nothing found
+# Nothing found.
+exit 10