linux 服务脚本

#!/bin/bash
#
# chkconfig: 2345 58 74
# description: my_SERVICE_NAME is a my Service
#

# common function
. /etc/init.d/functions

# service bash flag
IVMS_SERVICE=0

CORE_DUMP_DIR=/var/my_core
DAEMON_COREFILE_LIMIT=unlimited
SVC_SYS_DIR=/etc/init.d

SVC_PROG=my_SERVICE_PROG
SVC_NAME=my_SERVICE_NAME
SVC_DIR=my_SERVICE_DIR
SVC_LOCK_FILE=/var/lock/subsys/$SVC_PROG
SVC_PID_FILE=/var/run/$SVC_NAME.pid

# install service, $1 service name, $2 executable name
InstallSvc()
{
    if [ -z "$1" ]; then
        echo $"service name is needed by myservice."
        return -1
    fi

    SVC_NAME=$1
    SVC_PROG=${2:-$SVC_NAME}
    SVC_DIR=`pwd`
    SVC_SYS_FILE=$SVC_SYS_DIR/$SVC_NAME

    if [ ! -f $SVC_DIR/$SVC_PROG ]; then
        echo $"$SVC_DIR/$SVC_PROG does not exist."
        return -1
    fi

    echo $"installing $SVC_NAME service, executable file $SVC_DIR/$SVC_PROG ..."

    # set suid_dumpable on
    if [ -e /proc/sys/kernel/suid_dumpable ]; then
        echo 1 > /proc/sys/kernel/suid_dumpable
    else
        echo 1 > /proc/sys/fs/suid_dumpable
    fi

    # create core directory
    if [ -d $CORE_DUMP_DIR ]; then
        echo $"$CORE_DUMP_DIR/core-%e-%p-%t" > /proc/sys/kernel/core_pattern
    else
        mkdir $CORE_DUMP_DIR
        echo $"$CORE_DUMP_DIR/core-%e-%p-%t" > /proc/sys/kernel/core_pattern
    fi

    # config sysctl to enable core
    grep "$CORE_DUMP_DIR/core-%e-%p-%t" /etc/sysctl.conf
    if [ $? -ne 0 ]; then
        echo $"fs.suid_dumpable = 1" >> /etc/sysctl.conf
        echo $"kernel.core_pattern = $CORE_DUMP_DIR/core-%e-%p-%t" >> /etc/sysctl.conf
    fi

    # create service bash
    sed -e "s%my_SERVICE=0%my_SERVICE=1%g" $0 | \
    sed -e "s%my_SERVICE_PROG%$SVC_PROG%g" | \
    sed -e "s%my_SERVICE_NAME%$SVC_NAME%g" | \
    sed -e "s%my_SERVICE_DIR%$SVC_DIR%g" > $SVC_SYS_FILE

    chmod u+x $SVC_SYS_FILE
    chmod u+x $SVC_DIR/$SVC_PROG
    chkconfig --add $SVC_NAME

    echo $"install $SVC_NAME service successfully."
    return 0
}

# uninstall service, $1 service name
UninstallSvc()
{
    if [ -z "$1" ]; then
        echo $"service name is needed by myservice."
        return -1
    fi

    SVC_NAME=$1
    SVC_SYS_FILE=$SVC_SYS_DIR/$SVC_NAME

    echo $"uninstalling $SVC_NAME service ..."

    # rm service bash
    if [ -f $SVC_SYS_FILE ]; then
        chkconfig --del $SVC_NAME
        rm -f $SVC_SYS_FILE
    else
        warning $"$SVC_NAME service does not exist."
    fi

    echo $"uninstall $SVC_NAME service successfully."
    return 0
}

# start service
StartSvc()
{
    echo $"starting $SVC_NAME service ..."

    # set core unlimited, replaced by DAEMON_COREFILE_LIMIT
    #ulimit -c unlimited

    # create lockfile, run program
    touch $SVC_LOCK_FILE
    cd $SVC_DIR
    daemon --pidfile=$SVC_PID_FILE $SVC_DIR/$SVC_PROG -service
    if [ $? -eq 0 ]; then
        # created by the program
        #pidof $SVC_DIR/$SVC_PROG > $SVC_PID_FILE
        echo $"start $SVC_NAME service successfully."
        return 0
    else
        echo $"start $SVC_NAME service failure."
        return -1
    fi
}

# get service status
IsSvcRunning()
{
    local pid
    __pids_var_run $SVC_NAME $SVC_PID_FILE
    [ -n "$pid" ] && return 0 || return 1
}

# stop service
StopSvc()
{
    echo $"stopping $SVC_NAME service ..."

    # remove lockfile
    rm -f $SVC_LOCK_FILE

    # wait for exit
    local i RC
    for (( i = 0; i < 10; i++ )); do
        if IsSvcRunning; then
            sleep 1
        else
            break
        fi
    done

    if [ $i -eq 10 ] && IsSvcRunning; then
        killproc -p $SVC_PID_FILE $SVC_NAME
        RC=$?
    else
        rm -f $SVC_PID_FILE
        RC=0
    fi

    if [ $RC -eq 0 ]; then
        echo $"stop $SVC_NAME service successfully."
        return 0
    else
        echo $"stop $SVC_NAME service failure."
        return -1
    fi
}

# restart service
RestartSvc()
{
    echo $"restarting $SVC_NAME service ..."

    StopSvc
    if [ $? -eq 0 ]; then
        StartSvc
        if [ $? -eq 0 ]; then
            echo $"restart $SVC_NAME service successfully."
            return 0
        fi
    fi

    echo $"restart $SVC_NAME service failure."
    return -1
}

RETVAL=0

if [ $IVMS_SERVICE -eq 0 ]; then
    # setup
    case $1 in
        install | i)
            InstallSvc $2 $3
            RETVAL=$?
            ;;
        uninstall | u)
            UninstallSvc $2
            RETVAL=$?
            ;;
        *)
            echo $"Usage: $0 {i, install NAME [EXEC] | u, uninstall NAME}"
            ;;
    esac
else
    # service
    case $1 in
        start | r)
            StartSvc
            RETVAL=$?
            ;;
        stop | p)
            StopSvc
            RETVAL=$?
            ;;
        restart | e)
            RestartSvc
            RETVAL=$?
            ;;
        status | s)
            status -p $SVC_PID_FILE $SVC_NAME
            RETVAL=$?
            ;;
        status2)
            IsSvcRunning && echo $"$SVC_NAME service is running." || echo $"$SVC_NAME service has been stopped."
            ;;
        *)
            echo $$"Usage: $0 {start|stop|status|restart}"
            ;;
    esac
fi

exit $RETVAL