#!/bin/bash
#
# Init file for the ARC gridftp server
#
# chkconfig: - 55 25
# description: ARC gridftpd
#
# config: /etc/sysconfig/nordugrid
# config: /etc/sysconfig/gridftpd
# config: /usr/etc/arc.conf
# config: /etc/arc.conf

### BEGIN INIT INFO
# Provides:          gridftpd
# Required-Start:    $local_fs $remote_fs
# Required-Stop:     $local_fs $remote_fs
# Default-Stop:      0 1 2 3 4 5 6
# Short-Description: ARC gridftpd
# Description:       ARC gridftp server
### END INIT INFO

# source function library
if [ -f /etc/init.d/functions ]; then
    . /etc/init.d/functions
    log_success_msg() {
	echo -n "$@"
	success "$@"
	echo
    }
    log_warning_msg() {
	echo -n "$@"
	warning "$@"
	echo
    }
    log_failure_msg() {
	echo -n "$@"
	failure "$@"
	echo
    }
elif [ -f /lib/lsb/init-functions ]; then
    . /lib/lsb/init-functions
else
    echo "Error: Cannot source neither init.d nor lsb functions"
    exit 1
fi

add_library_path() {
    location="$1"
    if [ ! "x$location" = "x" ] ; then
	if [ ! "$location" = "/usr" ] ; then
	    libdir="$location/lib"
	    libdir64="$location/lib64"
	    if [ -d "$libdir64" ] ; then
		if [ "x$LD_LIBRARY_PATH" = "x" ]; then
		    LD_LIBRARY_PATH="$libdir64"
		else
		    LD_LIBRARY_PATH="$libdir64:$LD_LIBRARY_PATH"
		fi
	    fi
	    if [ -d "$libdir" ] ; then
		if [ "x$LD_LIBRARY_PATH" = "x" ]; then
		    LD_LIBRARY_PATH="$libdir"
		else
		    LD_LIBRARY_PATH="$libdir:$LD_LIBRARY_PATH"
		fi
	    fi
	fi
    fi
}

prog=gridftpd
RUN=yes

# sysconfig files
if [ -r /etc/sysconfig/nordugrid ]; then
    . /etc/sysconfig/nordugrid
elif [ -r /etc/default/nordugrid ]; then
    . /etc/default/nordugrid
fi
if [ -r /etc/sysconfig/${prog} ]; then
    . /etc/sysconfig/${prog}
elif [ -r /etc/default/${prog} ]; then
    . /etc/default/${prog}
fi

if [ "$RUN" != "yes" ] ; then
    log_warning_msg "$prog disabled, please adjust the configuration to your"
    log_warning_msg "needs and then set RUN to 'yes' in /etc/default/$prog to enable it."
    exit 0
fi

# GLOBUS_LOCATION
GLOBUS_LOCATION=${GLOBUS_LOCATION:-/usr}
if [ ! -d "$GLOBUS_LOCATION" ]; then
    log_failure_msg "GLOBUS_LOCATION ($GLOBUS_LOCATION) not found"
    exit 1
fi
export GLOBUS_LOCATION

# ARC_LOCATION
ARC_LOCATION=${ARC_LOCATION:-/usr}
if [ ! -d "$ARC_LOCATION" ]; then
    log_failure_msg "ARC_LOCATION ($ARC_LOCATION) not found"
    exit 1
fi
export ARC_LOCATION

readconfigvar() {
    fname=$1
    if [ ! -r "$fname" ]; then
	return
    fi
    bname="[$2]"
    vname=$3
    value=
    cat "$fname" | grep -e '^\[' -e "^${vname}=" | {
	while true; do
	    read line
	    if [ ! $? = 0 ] ; then
		return
	    fi
	    if [ "$line" = "$bname" ] ; then
		while true ; do
		    read line
		    if [ ! $? = 0 ] ; then
			return
		    fi
		    lstart=`echo "$line" | head -c 1`
		    if [ "$lstart" = '[' ] ; then
			return
		    fi
		    vlname=`echo "$line" | sed 's/=.*//;t;s/.*//'`
		    if [ "$vlname" = "$vname" ] ; then
			val=`echo "$line" | sed 's/[^=]*=//'`
			eval "echo $val"
			return
		    fi
		done
	    fi
	done
    }
}

check_cert() {
  X509_USER_CERT=`readconfigvar "$ARC_CONFIG" gridftpd x509_user_cert`
  X509_USER_KEY=`readconfigvar "$ARC_CONFIG" gridftpd x509_user_key`
  if [ -z "$X509_USER_CERT" ] ; then
    X509_USER_CERT=`readconfigvar "$ARC_CONFIG" common x509_user_cert`
  fi
  if [ -z "$X509_USER_KEY" ] ; then
    X509_USER_KEY=`readconfigvar "$ARC_CONFIG" common x509_user_key`
  fi
  if [ -z "$X509_USER_CERT" ] ; then
    X509_USER_CERT=/etc/grid-security/hostcert.pem
  fi
  if [ -z "$X509_USER_KEY" ] ; then
    X509_USER_KEY=/etc/grid-security/hostkey.pem
  fi
  if [ ! -f "$X509_USER_CERT" ] ; then
    log_failure_msg "Host certificate not found"
    exit 1
  fi
  if [ ! -f "$X509_USER_KEY" ] ; then
    log_failure_msg "Host key not found"
    exit 1
  fi
  # check permissions on key
  perms=`stat -L -c %a "$X509_USER_KEY"`
  if [ "$perms" != "600" ] && [ "$perms" != "400" ] ; then
    log_failure_msg "Host key must be readable only by user"
    exit 1
  fi
}

CMD="$ARC_LOCATION/sbin/$prog"
if [ ! -x "$CMD" ]; then
    log_failure_msg "Missing executable"
    exit 1
fi

# ARC_CONFIG
if [ "x$ARC_CONFIG" = "x" ]; then
    if [ -r $ARC_LOCATION/etc/arc.conf ]; then
	ARC_CONFIG=$ARC_LOCATION/etc/arc.conf
    elif [ -r /etc/arc.conf ]; then
	ARC_CONFIG=/etc/arc.conf
    fi
fi

if [ ! -r "$ARC_CONFIG" ]; then
    log_failure_msg "ARC configuration not found (usually /etc/arc.conf)"
    exit 1
fi

CMD="$CMD -c '$ARC_CONFIG'"

# VOMS_LOCATION
VOMS_LOCATION=${VOMS_LOCATION:-@DEFAULT_VOMS_LOCATION@}

# GRIDSITE_LOCATION
GRIDSITE_LOCATION=${GRIDSITE_LOCATION:-@DEFAULT_GRIDSITE_LOCATION@}

add_library_path "$LFC_LOCATION"
add_library_path "$GRIDSITE_LOCATION"
add_library_path "$VOMS_LOCATION"
add_library_path "$GLOBUS_LOCATION"
if [ "x$LD_LIBRARY_PATH" = "x" ]; then
    LD_LIBRARY_PATH=$ARC_LOCATION/lib64
else
    LD_LIBRARY_PATH=$ARC_LOCATION/lib64:$LD_LIBRARY_PATH
fi
export LD_LIBRARY_PATH

PID_FILE=`readconfigvar "$ARC_CONFIG" gridftpd pidfile`

if [ `id -u` = 0 ] ; then
    # Debian does not have /var/lock/subsys
    if [ -d /var/lock/subsys ]; then
	LOCKFILE=/var/lock/subsys/$prog
    else
	LOCKFILE=/var/lock/$prog
    fi
    if [ "x$PID_FILE" = "x" ]; then
	PID_FILE=/var/run/$prog.pid
    fi
else
    LOCKFILE=$HOME/$prog.lock
    if [ "x$PID_FILE" = "x" ]; then
	PID_FILE=$HOME/$prog.pid
    fi
fi

logfile=`readconfigvar "$ARC_CONFIG" gridftpd logfile`
if [ "x$logfile" = "x" ]; then
    logfile=/var/log/arc/gridftpd.log
fi
if [ ! -d `dirname $logfile` ]; then
    mkdir -p `dirname $logfile`
fi

CMD="$CMD -P '$PID_FILE'"

start() {
    check_cert

    echo -n "Starting $prog: "

    # Check if we are already running
    if [ -f $PID_FILE ]; then
	read pid < $PID_FILE
	if [ "x$pid" != "x" ]; then
	    ps -p "$pid" -o comm 2>/dev/null | grep "^$prog$" 1>/dev/null 2>/dev/null
	    if [ $? -eq 0 ] ; then
		log_success_msg "already running (pid $pid)"
		return 0
	    fi
	fi
	rm -f "$PID_FILE" "$LOCKFILE"
    fi

    eval "$CMD"
    RETVAL=$?

    if [ $RETVAL -eq 0 ]; then
	touch $LOCKFILE
	log_success_msg
    else
	log_failure_msg
    fi
    return $RETVAL
}

stop() {
    echo -n "Stopping $prog: "

    if [ -f "$PID_FILE" ]; then
	read pid < "$PID_FILE"
	if [ ! -z "$pid" ] ; then
	    kill "$pid"
	    RETVAL=$?
	    if [ $RETVAL -eq 0 ]; then
		log_success_msg
	    else
		log_failure_msg
	    fi

	    sleep 1
	    kill -9 "$pid" 1>/dev/null 2>&1
	    rm -f "$PID_FILE" "$LOCKFILE"
	else
	    RETVAL=1
	    log_failure_msg "$prog shutdown - pidfile is empty"
	fi
    else
	RETVAL=0
	log_success_msg "$prog shutdown - already stopped"
    fi
    return $RETVAL
}

status() {
    if [ -f "$PID_FILE" ]; then
	read pid < "$PID_FILE"
	if [ "$pid" != "" ]; then
	    if ps -p "$pid" > /dev/null; then
		echo "$1 (pid $pid) is running..."
		return 0
	    fi
	    echo "$1 stopped but pid file exists"
	    return 1
	fi
    fi
    if [ -f $LOCKFILE ]; then
	echo "$1 stopped but lockfile exists"
	return 2
    fi
    echo "$1 is stopped"
    return 3
}

restart() {
    stop
    start
}

case "$1" in
    start)
	start
	;;
    stop)
	stop
	;;
    status)
	status $prog
	;;
    restart | force-reload)
	restart
	;;
    reload)
	;;
    condrestart | try-restart)
	[ -f $LOCKFILE ] && restart || :
	;;
    *)
	echo "Usage: $0 {start|stop|status|restart|force-reload|reload|condrestart|try-restart}"
	exit 1
	;;
esac

exit $?
