src/init/ossec-server.sh

   1 #!/bin/sh
   2 # ossec-control        This shell script takes care of starting
   3 #                      or stopping ossec-hids
   4 # Author: Daniel B. Cid <daniel.cid@gmail.com>
   5
   6
   7 # Getting where we are installed
   8 LOCAL=`dirname $0`;
   9 cd ${LOCAL}
  10 PWD=`pwd`
  11 DIR=`dirname $PWD`;
  12 PLIST=${DIR}/bin/.process_list;
  13
  14
  15 ###  Do not modify bellow here ###
  16
  17 # Getting additional processes
  18 ls -la ${PLIST} > /dev/null 2>&1
  19 if [ $? = 0 ]; then
  20 . ${PLIST};
  21 fi
  22
  23
  24 NAME="OSSEC HIDS"
  25 VERSION="v2.5.1"
  26 AUTHOR="Trend Micro Inc."
  27 DAEMONS="ossec-monitord ossec-logcollector ossec-remoted ossec-syscheckd ossec-analysisd ossec-maild ossec-execd ${DB_DAEMON} ${CSYSLOG_DAEMON} ${AGENTLESS_DAEMON}"
  28
  29
  30 ## Locking for the start/stop
  31 LOCK="${DIR}/var/start-script-lock"
  32 LOCK_PID="${LOCK}/pid"
  33
  34
  35 # This number should be more than enough (even if it is
  36 # started multiple times together). It will try for up
  37 # to 10 attempts (or 10 seconds) to execute.
  38 MAX_ITERATION="10"
  39
  40
  41
  42 # Check pid
  43 checkpid()
  44 {
  45     for i in ${DAEMONS}; do
  46         for j in `cat ${DIR}/var/run/${i}*.pid 2>/dev/null`; do
  47             ps -p $j |grep ossec >/dev/null 2>&1
  48             if [ ! $? = 0 ]; then
  49                 echo "Deleting PID file '${DIR}/var/run/${i}-${j}.pid' not used..."
  50                 rm ${DIR}/var/run/${i}-${j}.pid
  51             fi
  52         done
  53     done
  54 }
  55
  56
  57
  58 # Lock function
  59 lock()
  60 {
  61     i=0;
  62
  63     # Providing a lock.
  64     while [ 1 ]; do
  65         mkdir ${LOCK} > /dev/null 2>&1
  66         MSL=$?
  67         if [ "${MSL}" = "0" ]; then
  68             # Lock aquired (setting the pid)
  69             echo "$$" > ${LOCK_PID}
  70             return;
  71         fi
  72
  73         # Waiting 1 second before trying again
  74         sleep 1;
  75         i=`expr $i + 1`;
  76
  77         # If PID is not present, speed things a bit.
  78         kill -0 `cat ${LOCK_PID}` >/dev/null 2>&1
  79         if [ ! $? = 0 ]; then
  80             # Pid is not present.
  81             i=`expr $i + 1`;
  82         fi
  83
  84         # We tried 10 times to acquire the lock.
  85         if [ "$i" = "${MAX_ITERATION}" ]; then
  86             # Unlocking and executing
  87             unlock;
  88             mkdir ${LOCK} > /dev/null 2>&1
  89             echo "$$" > ${LOCK_PID}
  90             return;
  91         fi
  92     done
  93 }
  94
  95
  96 # Unlock function
  97 unlock()
  98 {
  99     rm -rf ${LOCK}
 100 }
 101
 102
 103 # Help message
 104 help()
 105 {
 106     # Help message
 107     echo ""
 108     echo "Usage: $0 {start|stop|restart|status|enable|disable}";
 109     exit 1;
 110 }
 111
 112
 113 # Enables/disables additional daemons
 114 enable()
 115 {
 116     if [ "X$2" = "X" ]; then
 117         echo ""
 118         echo "Enable options: database, client-syslog, agentless, debug"
 119         echo "Usage: $0 enable [database|client-syslog|agentless|debug]"
 120         exit 1;
 121     fi
 122
 123     if [ "X$2" = "Xdatabase" ]; then
 124         echo "DB_DAEMON=ossec-dbd" >> ${PLIST};
 125     elif [ "X$2" = "Xclient-syslog" ]; then
 126         echo "CSYSLOG_DAEMON=ossec-csyslogd" >> ${PLIST};
 127     elif [ "X$2" = "Xagentless" ]; then
 128         echo "AGENTLESS_DAEMON=ossec-agentlessd" >> ${PLIST};
 129     elif [ "X$2" = "Xdebug" ]; then
 130         echo "DEBUG_CLI=\"-d\"" >> ${PLIST};
 131     else
 132         echo ""
 133         echo "Invalid enable option."
 134         echo ""
 135         echo "Enable options: database, client-syslog, agentless, debug"
 136         echo "Usage: $0 enable [database|client-syslog|agentless|debug]"
 137         exit 1;
 138     fi
 139
 140
 141 }
 142
 143
 144
 145 # Enables/disables additional daemons
 146 disable()
 147 {
 148     if [ "X$2" = "X" ]; then
 149         echo ""
 150         echo "Disable options: database, client-syslog, agentless, debug"
 151         echo "Usage: $0 disable [database|client-syslog|agentless|debug]"
 152         exit 1;
 153     fi
 154
 155     if [ "X$2" = "Xdatabase" ]; then
 156         echo "DB_DAEMON=\"\"" >> ${PLIST};
 157     elif [ "X$2" = "Xclient-syslog" ]; then
 158         echo "CSYSLOG_DAEMON=\"\"" >> ${PLIST};
 159     elif [ "X$2" = "Xagentless" ]; then
 160         echo "AGENTLESS_DAEMON=\"\"" >> ${PLIST};
 161     elif [ "X$2" = "Xdebug" ]; then
 162         echo "DEBUG_CLI=\"\"" >> ${PLIST};
 163     else
 164         echo ""
 165         echo "Invalid disable option."
 166         echo ""
 167         echo "Disable options: database, client-syslog, agentless, debug"
 168         echo "Usage: $0 disable [database|client-syslog|agentless|debug]"
 169         exit 1;
 170     fi
 171
 172
 173 }
 174
 175
 176
 177 # Status function
 178 status()
 179 {
 180     for i in ${DAEMONS}; do
 181         pstatus ${i};
 182         if [ $? = 0 ]; then
 183             echo "${i} not running..."
 184         else
 185             echo "${i} is running..."
 186         fi
 187     done
 188 }
 189
 190 testconfig()
 191 {
 192     # We first loop to check the config.
 193     for i in ${SDAEMONS}; do
 194         ${DIR}/bin/${i} -t ${DEBUG_CLI};
 195         if [ $? != 0 ]; then
 196             echo "${i}: Configuration error. Exiting"
 197             unlock;
 198             exit 1;
 199         fi
 200     done
 201 }
 202
 203 # Start function
 204 start()
 205 {
 206     SDAEMONS="${DB_DAEMON} ${CSYSLOG_DAEMON} ${AGENTLESS_DAEMON} ossec-maild ossec-execd ossec-analysisd ossec-logcollector ossec-remoted ossec-syscheckd ossec-monitord"
 207
 208     echo "Starting $NAME $VERSION (by $AUTHOR)..."
 209     ${DIR}/bin/ossec-logtest -t
 210     if [ ! $? = 0 ]; then
 211         echo "ossec-analysisd: Configuration error. Exiting."
 212     fi
 213     lock;
 214     checkpid;
 215
 216
 217     # We actually start them now.
 218     for i in ${SDAEMONS}; do
 219         pstatus ${i};
 220         if [ $? = 0 ]; then
 221             ${DIR}/bin/${i} ${DEBUG_CLI};
 222             if [ $? != 0 ]; then
 223                 unlock;
 224                 exit 1;
 225             fi
 226
 227             echo "Started ${i}..."
 228         else
 229             echo "${i} already running..."
 230         fi
 231
 232     done
 233
 234     # After we start we give 2 seconds for the daemons
 235     # to internally create their PID files.
 236     sleep 2;
 237     unlock;
 238     echo "Completed."
 239 }
 240
 241 # Process status
 242 pstatus()
 243 {
 244     pfile=$1;
 245
 246     # pfile must be set
 247     if [ "X${pfile}" = "X" ]; then
 248         return 0;
 249     fi
 250
 251     ls ${DIR}/var/run/${pfile}*.pid > /dev/null 2>&1
 252     if [ $? = 0 ]; then
 253         for j in `cat ${DIR}/var/run/${pfile}*.pid 2>/dev/null`; do
 254             ps -p $j |grep ossec >/dev/null 2>&1
 255             if [ ! $? = 0 ]; then
 256                 echo "${pfile}: Process $j not used by ossec, removing .."
 257                 rm -f ${DIR}/var/run/${pfile}-$j.pid
 258                 continue;
 259             fi
 260
 261             kill -0 $j > /dev/null 2>&1
 262             if [ $? = 0 ]; then
 263                 return 1;
 264             fi
 265         done
 266     fi
 267
 268     return 0;
 269 }
 270
 271
 272 # Stop all
 273 stopa()
 274 {
 275     lock;
 276     checkpid;
 277     for i in ${DAEMONS}; do
 278         pstatus ${i};
 279         if [ $? = 1 ]; then
 280             echo "Killing ${i} .. ";
 281
 282             kill `cat ${DIR}/var/run/${i}*.pid`;
 283         else
 284             echo "${i} not running ..";
 285         fi
 286
 287         rm -f ${DIR}/var/run/${i}*.pid
 288
 289      done
 290
 291     unlock;
 292     echo "$NAME $VERSION Stopped"
 293 }
 294
 295
 296 ### MAIN HERE ###
 297
 298 case "$1" in
 299   start)
 300     testconfig
 301         start
 302         ;;
 303   stop)
 304         stopa
 305         ;;
 306   restart)
 307     testconfig
 308         stopa
 309         start
 310         ;;
 311   status)
 312     status
 313         ;;
 314   help)
 315     help
 316     ;;
 317   enable)
 318     enable $1 $2;
 319     ;;
 320   disable)
 321     disable $1 $2;
 322     ;;
 323   *)
 324     help
 325 esac