#!/bin/bash

#

# /etc/init.d/iscsi

#

### BEGIN INIT INFO

# Provides:          iscsi

# Required-Start:    $network

# Should-Start:      iscsitarget multipathd

# Required-Stop:     $network

# Should-Stop:       multipathd

# Default-Start:     3 5

# Default-Stop:      

# Short-Description: iSCSI initiator daemon

# Description:       The iSCSI initator is used to create and

#                    manage iSCSI connections to an iSCSI Target.

#

### END INIT INFO

CONFIG_FILE=/etc/iscsi/iscsid.conf

DAEMON=/sbin/iscsid

ISCSIADM=/sbin/iscsiadm

BRCM_ISCSIUIO=/sbin/brcm_iscsiuio

ARGS="-c $CONFIG_FILE -n"

# Source LSB init functions

. /etc/rc.status

# Reset status of this service

rc_reset

DM_MAJOR=$(sed -n 's/\(.*\) device-mapper/\1/p' /proc/devices)

iscsi_login_all_nodes()

{

	echo -n "Setting up iSCSI targets: "

	$ISCSIADM -m node --loginall=automatic 2> /dev/null

	if [ $? == 21 ] ; then

	    rc_failed 6

	fi

	rc_status -v

}

#

# Try to load all required modules prior to startup

#

iscsi_load_transport_modules()

{

    loaded=$(sed -n "/^iscsi_tcp/p" /proc/modules)

    if [ -z "$loaded" ] ; then

	modprobe iscsi_tcp

	if [ $? = 0 ] ; then

	    echo -n " tcp"

	fi

    fi

    for iface in /etc/iscsi/ifaces/*; do

	[ -f "$iface" ] || continue

	[ "$iface" = "iface.example" ] && continue

	# Check if the iface has been configured

	result=$(sed '/#.*/D;/iface.iscsi_ifacename/D;/iface.hwaddress/D;/iface.transport_name/D' $iface)

	if [ "$result" ] ; then

	    mod=$(sed -n 's/iface.transport_name *= *\(.*\)/\1/p' $iface)

	    loaded=$(sed -n "/^$mod/p" /proc/modules)

	    if [ -z "$loaded" ] ; then

		modprobe $mod

		if [ $? = 0 ] ; then

		    echo -n " $mod"

		fi

	    fi

	fi

    done

}

#

# Set a temporary startmode for ifdown

#

iscsi_modify_if_startmode()

{

    local ifname=$1

    local tmp_ifcfg=/dev/.sysconfig/network/if-$ifname

    if [ -e "$tmp_ifcfg" ] ; then

	. $tmp_ifcfg

	if [ "$startmode" ] ; then

	    return

	fi

    fi

    : disabling shutdown on $ifname

    echo "startmode=nfsroot" >> $tmp_ifcfg

}

iscsi_get_ifacename_from_session()

{

    local session=$1

    local ifacename

    ifacename=$(iscsiadm -m session -r ${session##.*/session} 2> /dev/null | \

	sed -n 's/iface.iscsi_ifacename = \(.*\)/\1/p')

    if [ -z "$ifacename" ] ; then

	# Check for iBFT

	ifacename=$(iscsiadm -m fw 2> /dev/null)

	if [ -n "$ifacename" ] ; then

	    ifacename="fw"

	fi

    fi

    echo $ifacename

}

iscsi_get_hwaddress_from_iface()

{

    local iface=$1

    local hwaddress

    hwaddress=$(iscsiadm -m iface -I "$iface" 2> /dev/null | sed -n 's/iface.hwaddress = \(.*\)/\1/p')

    [ "$hwaddress" = "<empty>" ] && hwaddress=

    echo $hwaddress

}

iscsi_get_ifname_from_iface()

{

    local iface=$1

    local ifname

    ifname=$(iscsiadm -m iface -I "$iface" 2> /dev/null | sed -n 's/iface.net_ifacename = \(.*\)/\1/p')

    [ "$ifname" = "<empty>" ] && ifname=

    echo $ifname

}

iscsi_get_ipaddr_from_iface()

{

    local iface=$1

    local ipaddr

    ipaddr=$(iscsiadm -m iface -I "$iface" 2> /dev/null | sed -n 's/iface.ipaddress = \(.*\)/\1/p')

    [ "$ipaddr" = "<empty>" ] && ipaddr=

    echo $ipaddr

}

iscsi_get_ifname_from_firmware()

{

    local hwaddress

    hwaddress=$(iscsiadm -m fw 2> /dev/null | sed -n 's/iface.net_ifacename = \(.*\)/\1/p')

    echo $hwaddress

}

#

# cxgb3i is using the HWAddress to select

# the correct interface

#

iscsi_get_ifname_from_hwaddress()

{

    local hwaddress=$1

    for if in /sys/class/net/*; do

	[ -e "$if" ] || continue

	read mac < $if/address

	[ "$mac" = "$hwaddress" ] || continue

	echo ${if##*/}

	break

    done

}

iscsi_get_ifname_from_ipaddr()

{

    local ipaddr=$1

    local ifname

    ifname=$(ip addr show to $ipaddr | sed -n 's/[0-9]*: \([^ :]*\): .*/\1/p')

    return $ifname

}

#

# Handle 'default' interface:

# It is basically impossible to determine via which

# interface the iSCSI traffic will flow, so we take

# the easy option and ignore _all_ active interfaces

# during shutdown

#

iscsi_modify_all_interfaces()

{

    ip link show up | sed -n '/.*LOOPBACK.*/d;s/[0-9]*: \(.*\): .*/\1/p' | while read ifname; do

	iscsi_modify_if_startmode $ifname

    done

}

#

# Check iface setting and disable

# affected network interfaces

#

iscsi_check_interface()

{

    local session=$1

    local i h n

    i=$(iscsi_get_ifacename_from_session $session)

    [ -z "$i" ] && continue

    if [ "$i" = "default" ] ; then

	iscsi_modify_all_interfaces

    elif [ "$i" = "fw" ] ; then

	n=$(iscsi_get_ifname_from_firmware)

    else

	n=$(iscsi_get_ifname_from_iface $i)

	if [ -z "$n" ] ; then

	    h=$(iscsi_get_hwaddress_from_iface $i)

	    if [ -n "$h" ] ; then

		n=$(iscsi_get_ifname_from_hwaddress $h)

	    fi

	fi

	if [ -z "$n" ] ; then

	    h=$(iscsi_get_ipaddr_from_iface $i)

	    if [ -n "$h" ] ; then

		n=$(iscsi_get_ifname_from_ipaddr $h)

	    fi

	fi

    fi

    if [ "$n" ] ; then

	iscsi_modify_if_startmode $n

    fi

}

#

# Check if device 'dev' is mounted

# Returns the mount point on success

#

iscsi_check_if_mounted()

{

    local dev=$1

    local d m t o x p

    cat /proc/mounts | sed -ne '/^\/dev\/.*/p' | while read d m t o x; do 

	if [ -L "$d" ] ; then

	    d=$(readlink -f $d)

	fi

	[ -b "$d" ] || continue

	b=$(ls -l $d | sed -n 's/.* \([0-9]*\), \([0-9]*\) .*/\1:\2/p')

	p=$(cd -P /sys/dev/block/$b ; echo $PWD)

	if [ -z "$p" ] ; then

	    d=${d##/dev}

	    p="/sys/block${d%%[0-9]*}"

	fi

	[ ! -d ${p} ] && continue

	if [ -e $p/partition ] ; then

	    p=$(cd -P $p/../; echo $PWD)

	fi

	if [ "$dev" = "${p##*/}" ] ; then

	    echo $m

	fi

    done

}

#

# Unwind block device stack

# 

# Stops unwinding if either no more 'holders'

# are found or if a device is mounted

# 

# Unmounts top-level device and deconfigures

# all devices down the stack

#

# Root fs is not unmounted

#

iscsi_unwind_stack()

{

    local p=$1

    local d=${p##*/}

    local u

    local m

    if [ ! -d ${p} ] ; then

	return;

    fi

    m=$(iscsi_check_if_mounted $d)

    if [ -z "$m" ] ; then

	for s in $p/holders/* ; do

	    [ -e $s ] || continue

	    p=$(cd -P $s; echo $PWD)

	    u=$(iscsi_unwind_stack $p)

	    if [ "$u" ] ; then

		echo -n "$u "

	    fi

	done

    else

	if [ "$m" = "/" ] ; then

	    echo -n "$d "

	    return 1

	fi

	if ! umount $m ; then

	    echo -n "$d "

	    return 1

	fi

    fi

    if [ "${d#dm-}" != "$d" ] ; then

	if ! dmsetup remove -j $DM_MAJOR -m ${d#dm-} 2> /dev/null ; then

	    echo -n "$d "

	    return 1

	fi

    fi

    if [ "${d#md}" != "$d" ] ; then

	if ! mdadm --manage /dev/$d --stop 2> /dev/null ; then

	    echo -n "$d "

	    return 1

	fi

    fi

    return 0

}

#

# Return all targets for a given session

#

iscsi_get_target()

{

    local session=$1

    local d

    for d in $session/device/target* ; do

	[ -e "$d" ] || continue

	echo "$d"

    done

}

#

# Checks all devices presented by a target

# and tries to umount them.

# Skip unmounting for the root fs.

# Stops on the first device which could not be unmounted

# and returns the mount device of that device.

#

iscsi_check_target()

{

    local t=$1

    local d b m

    for d in $t/* ; do

	[ -d $d/block ] || continue

	for b in $d/block/sd* ; do

	    [ -d "$b" ] || continue

	    m=$(iscsi_unwind_stack $b)

	    if [ -n "$m" ] ; then

		echo $m

		return 1

	    fi

	done

    done

}

#

# Check all sessions for mounted devices

# and shutdown the session if the affected

# devices could be umounted cleanly.

# If umount fails disable shutdown on all

# affected network interfaces

#

iscsi_stop_sessions()

{

    local t m s i

    i=0

    for session in /sys/class/iscsi_session/session* ; do

	[ -e "$session" ] || continue;

	[ -e $session/device ] || continue

	t=$(iscsi_get_target $session)

	m=$(iscsi_check_target $t)

	s=${session##*/session}

	if [ -z "$m" ] ; then

	    iscsiadm -m session -r ${s} -u

	    i=$(( $i + 1 ))

	else

	    iscsi_check_interface $s

	fi

    done

    echo $i

}

iscsi_list_all_nodes()

{

    # Check for active sessions

    if $ISCSIADM -m session > /dev/null; then

	return 0

    fi

    echo "Active connections:"

    $ISCSIADM -m session | while read proto num PORTAL TARGET ; do

	PORTAL=${PORTAL%,*}

	echo -e "\t$TARGET at $PORTAL"

    done

}

case "$1" in

    start)

	if checkproc $DAEMON ; then

	    RETVAL=0

	else

	    echo -n "Starting iSCSI initiator service: "

	    iscsi_load_transport_modules

	    if grep -q bnx2i /proc/modules && [ -x $BRCM_ISCSIUIO ] ; then

		startproc $BRCM_ISCSIUIO

	    fi

	    startproc $DAEMON $ARGS

	    RETVAL=$?

	    rc_status -v

	fi

	if [ "$RETVAL" == "0" ]; then

	    iscsi_login_all_nodes

	fi

	;;

    stop)

	n=$(iscsi_stop_sessions)

	echo -n "Stopping iSCSI initiator service: "

	if [ "$n" ] && [ "$n" != "0" ] ; then

	    m=$(iscsiadm -m session 2> /dev/null)

	    if [ -z "$m" ] ; then

		killproc -KILL $DAEMON

		RETVAL=$?

		if grep -q bnx2i /proc/modules && [ -x $BRCM_ISCSIUIO ]; then

		    killproc -KILL $BRCM_ISCSIUIO

		fi

 		RETVAL=$?

	    else

		RETVAL=1

	    fi

	    rc_failed $RETVAL

	    rc_status -v

	else

	    # umounting failed, leave initiator running

	    rc_status -s

	fi

	;;

    status)

	echo -n "Checking for iSCSI initiator service: "

	if checkproc $DAEMON ; then

	    rc_status -v

	    iscsi_list_all_nodes

	else

	    rc_failed 3

	    rc_status -v

	fi

	;;

    restart|reload)

	$0 stop

	RETVAL=$?

	if [ "$RETVAL" != "0" ]; then

		echo "Stopping iSCSI initiator service failed, not starting"

		exit $RETVAL

	fi

	sleep 1

	$0 start

	;;

    *)

	echo "Usage: $0 {start|stop|status|restart|reload}"

	exit 1

	;;

esac

rc_exit