#! /bin/sh
# Copyright 2000, International Business Machines Corporation and others.
# All Rights Reserved.
# 
# This software has been released under the terms of the IBM Public
# License.  For details, see the LICENSE file in the top-level source
# directory or online at http://www.openafs.org/dl/license10.html

# AFS	Start and stop AFS components
# 
# 
# chkconfig: 35 60 20
# description:  AFS is a distributed file system which provides location
#		transparency, caching and secure authentication.
#		Additional configuration can be done in the /etc/sysconfig/afs
#		file. Read the documentation in that file for more information.
#
# The following section is used at least by SuSE insserv(8)
### BEGIN INIT INFO
# Provides: afs
# Required-Start: $network
# Required-Stop: $network
# Default-Start: 3 5
# Default-Stop: 0 1 2 6
# Description:  AFS is a distributed file system which provides location
#              transparency, caching and secure authentication.
#              Additional configuration can be done in the /etc/sysconfig/afs
#              file. Read the documentation in that file for more information.
### END INIT INFO
#
# Note that AFS does not use a pid file in /var/run. It is turned off by
# unmounting /afs.


# check for special bootup functions
if [ -f /etc/rc.d/init.d/functions ] ; then
	. /etc/rc.d/init.d/functions
	afs_rh=1
else
	# special (RedHat) functions not available...
	# so I provide neccessary replacements (e.g. for SuSE)

	function echo_failure () { echo -n " - failed." ; }

	function echo_success () { echo -n " - successful." ; }

	# these are hopefully all functions I use...
fi

runcmd() {
   echo -n "$1 "
   shift
   if [ "$BOOTUP" = "color" ]; then
      $* && echo_success || echo_failure
   else
      $*
   fi
   echo
}

SYSCNF=/etc/sysconfig/afs

# Gather up options and post startup script name, if present
if [ -f $SYSCNF ] ; then
	. $SYSCNF
fi

CACHEINFO=${CACHEINFO:-/usr/vice/etc/cacheinfo}
CACHE=${CACHEDIR:-/usr/vice/cache}
AFS=${AFSDIR:-/afs}

# is_on returns 1 if value of arg is "on"
is_on() {
	if  test "$1" = "on" ; then return 0
	else return 1
        fi
}

on_network() {
	#*******************************************
	# Begin of modifications done by Troy Dawson
	#*******************************************	
	FOUNDSERVER="no"
	FINISHED="no"
	CELL="$(cat /usr/vice/etc/ThisCell)"
	cat /usr/vice/etc/CellServDB | {
		while [ "$FINISHED" = "no" ] 
		do
			read line
			CHAR="$(echo $line | cut -c1)"
			if [ "$CHAR" = ">" ] ; then
				if [ "$FOUNDSERVER" = "yes" ] ; then
					FINISHED="yes"
				else
					LINECELL="$(echo $line | cut -d' ' -f1 | cut -d'>' -f2)"
					if [ "$CELL" = "$LINECELL" ] ; then
						FOUNDSERVER="yes"
					fi
				fi
			else
				if [ "$FOUNDSERVER" = "yes" ] ; then
					SERVER="$(echo $line | cut -d' ' -f1)"
					ping -c 1 -w 5 $SERVER > /dev/null
					if [ $? -eq 0 ] ; then
						return 0
					fi

				fi
			fi
		done
		return 1
	}
	return
	#*******************************************
	# End of modifications done by Troy Dawson
	#*******************************************
}

# If choose_client can't correctly determine which client to use, set
# LIBAFS manually.
choose_client() {

        # Use the associated script to determine the module name
	if test -f /usr/vice/etc/afsmodname && \
	    test -f /usr/vice/etc/modload/SymTable ; then
	    
            LIBAFS=`/usr/vice/etc/afsmodname`
	    if test "x$LIBAFS" = "x" ; then
		echo Did not find matching module in SymTable
	    else
		echo Found $LIBAFS from SymTable... Loading...
		return 0
	    fi
	fi

	# Use the second field of the uname -v output instead of just
	# doing a match on the whole thing to protect against matching
	# a timezone named SMP -- I don't know of one, but let's be
	# paranoid.

	set X `uname -v`; shift
	case $2 in
	SMP) MP=.mp ;;	# MP system
	*)   MP= ;;	# SP system
	esac

	# For now, just use uname -r to get the module version. 
	VERSION=`uname -r`

	LIBAFS=libafs-$VERSION$MP.o
}

#
# Find prefix symbol to use with insmod.  We find the unregister_filesystem
# string from /proc/ksyms since we know it's there.  If /proc/ksyms does not
# exist, we print that info to the console and use the uname -v output to
# decide on a prefix.
# unregister_filesystem_Rsmp_b240cad8 is a typcial SMP version string from
# a kernel built from ftp.kernel.org
#

KSYMS_FILE=/proc/ksyms
SEARCH_STR="unregister_filesystem"
DEFAULT_SMP_PREFIX="smp_" # Redhat kernels need "smp" instead
PREFIX="" # none needed for UP with <= 1Gig memory

set_prefix()
{
	h='[0-9a-fA-F]'
	h8="$h$h$h$h$h$h$h$h"
	prefix_set=0

	set X `fgrep $SEARCH_STR $KSYMS_FILE 2> /dev/null`; shift
	str=$2
	case $str in
	${SEARCH_STR}_R$h8)
		# No prefix required
		;;
	$SEARCH_STR)
		# No versioning in kernel symbols
		;;
	${SEARCH_STR}_R*$h8)
		suffix=${str#${SEARCH_STR}_R}
		PREFIX=${suffix%$h8}
		;;
	*)
		case $str in
		'')
			echo afsd: Cannot find \"$SEARCH_STR\" in file $KSYMS_FILE
			;;
		*)
			echo afsd: Malformed kernel version symbol \"$str\"
			;;
		esac

		echo Guessing prefix from output of uname -v
		set X `uname -v`; shift
		case $2 in
		SMP)
			PREFIX=$DEFAULT_SMP_PREFIX
			;;
		esac
		;;
	esac
}

MODLOADDIR=/usr/vice/etc/modload
# load_client loads the AFS client module if it's not already loaded. 
load_client() {
	# If LIBAFS is set, use it.
	if [ -z "$LIBAFS" ] ; then
		# Try to determine the right client.
		choose_client
	fi
    
	if [ ! -f "$MODLOADDIR/$LIBAFS" ] ; then
		echo AFS module $MODLOADDIR/$LIBAFS does not exist. Not starting AFS.
		exit 1
	fi

	# use the prefix command if required
	set_prefix
	/sbin/insmod ${PREFIX:+-P $PREFIX} -f -m $MODLOADDIR/$LIBAFS > $MODLOADDIR/libafs.map 2>&1
}

generate_cacheinfo() {
    if [ "$CACHESIZE" = "AUTOMATIC" ]; then
	LINE=`df -k $CACHE | tail -1`
	PART=`echo $LINE | awk '{ if ( ($NF != "/usr")  && ($NF != "/") ) print $NF; else print "NONE";}'`
	if [ "$PART" = "NONE" ]; then
	    echo "$CACHE or /usr/vice is not a separate partition"
	    echo "you have to change the cachesize in $SYSCNF by hand"
	    exit 1
	else
	    PARTSIZE=`echo $LINE | awk '{print $2}'`
	    CACHESIZE=`echo $PARTSIZE | awk '{printf "%d",int(($1*.8)/1000)*1000}'`
	fi
    fi
    if [ "x$CACHESIZE" != "x" ]; then
	echo $AFS:$CACHE:$CACHESIZE >$CACHEINFO
	chmod 0644 $CACHEINFO
    else
	CACHESIZE=`awk -F: '{print $3}' < $CACHEINFO`
    fi
}

choose_afsdoptions() {
	if [ -z "$OPTIONS" -o "$OPTIONS" = "AUTOMATIC" ]; then
		if [ $CACHESIZE -lt 131072 ]; then
			OPTIONS=$SMALL
		elif [ $CACHESIZE -lt 524288 ]; then
			OPTIONS=$MEDIUM
		elif [ $CACHESIZE -lt 1048576 ]; then
			OPTIONS=$LARGE
		elif [ $CACHESIZE -lt 2097152 ]; then
			OPTIONS=$XLARGE
		else
			OPTIONS=$XXLARGE
		fi
	fi
	AFSD_OPTIONS="$OPTIONS $VERBOSE"
}

case "$1" in 
  start)
    if [ ! "$afs_rh" -o ! -f /var/lock/subsys/afs ]; then
        if [ `echo "$OPTIONS" | grep -c dynroot` = 0 ]; then
            on_network || exit 1
        fi
	# Load kernel extensions
	if  load_client  ; then :
	else
		echo Failed to load AFS client, not starting AFS services.
		exit 1
	fi

	echo "Starting AFS services..... "
	# Start bosserver, it if exists
	if  is_on $AFS_SERVER && test -x /usr/afs/bin/bosserver  ; then
		/usr/afs/bin/bosserver 
		test "$afs_rh" && touch /var/lock/subsys/afs
	fi

	# Start AFS client
	if  is_on $AFS_CLIENT && test -x /usr/vice/etc/afsd  ; then
		generate_cacheinfo
		choose_afsdoptions
		/usr/vice/etc/afsd ${AFSD_OPTIONS}
		test "$afs_rh" && touch /var/lock/subsys/afs
		$AFS_POST_INIT
	fi
	#*******************************************
	# Begin of modifications done by Troy Dawson
	#*******************************************	
	# Link the update_afs_cellservdb.pl to run daily
	if [ -s /usr/vice/etc/update_afs_cellservdb.pl ] ; then
		ln -sf /usr/vice/etc/update_afs_cellservdb.pl /etc/cron.daily/update_afs_cellservdb.pl
	fi
	#*******************************************
	# End of modifications done by Troy Dawson
	#*******************************************
    fi
	;;

  stop)
    if [ ! "$afs_rh" -o -f /var/lock/subsys/afs ]; then
	# Stop AFS
	echo "Stopping AFS services..... "

	if  is_on $AFS_CLIENT  ; then
		if [ -x /usr/vice/etc/killafs ] ; then
			runcmd "Sending all processes using /afs the TERM signal ..." /usr/vice/etc/killafs TERM
			runcmd "Sending all processes using /afs the KILL signal ..." /usr/vice/etc/killafs KILL
		fi
		umount /afs
	fi

	if  is_on $AFS_SERVER && test -x /usr/bin/bos ; then
		echo "Stopping AFS bosserver"
		/usr/bin/bos shutdown localhost -localauth -wait
		killall -HUP bosserver
	fi

	LIBAFS=`/sbin/lsmod | fgrep libafs`
	if [ -n "$LIBAFS" ] ; then
		LIBAFS=`echo $LIBAFS | awk 'BEGIN { FS = " " } { print $1 }'`
		/sbin/rmmod $LIBAFS
	fi
	#*******************************************
	# Begin of modifications done by Troy Dawson
	#*******************************************	
	#sync the hardware clock to AFS
	/sbin/hwclock --systohc

	# unLink the update_afs_cellservdb.pl to run daily
	if [ -a /etc/cron.daily/update_afs_cellservdb.pl ] ; then
		rm -f /etc/cron.daily/update_afs_cellservdb.pl
	fi
	#*******************************************
	# End of modifications done by Troy Dawson
	#*******************************************

	rm -f /var/lock/subsys/afs
    fi
	;;

  restart)
	# Restart AFS
	$0 stop
	$0 start
	;;

  *)
	echo Usage: 'afs <start|stop|restart>'

esac

exit 0

action fool the Red Hat initscripts
