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.7"
  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     RETVAL=0
 181     for i in ${DAEMONS}; do
 182         pstatus ${i};
 183         if [ $? = 0 ]; then
 184             echo "${i} not running..."
 185             RETVAL=1
 186         else
 187             echo "${i} is running..."
 188         fi
 189     done
 190     exit $RETVAL
 191 }
 192
 193 testconfig()
 194 {
 195     # We first loop to check the config.
 196     for i in ${SDAEMONS}; do
 197         ${DIR}/bin/${i} -t ${DEBUG_CLI};
 198         if [ $? != 0 ]; then
 199             echo "${i}: Configuration error. Exiting"
 200             unlock;
 201             exit 1;
 202         fi
 203     done
 204 }
 205
 206 # Start function
 207 start()
 208 {
 209     SDAEMONS="${DB_DAEMON} ${CSYSLOG_DAEMON} ${AGENTLESS_DAEMON} ossec-maild ossec-execd ossec-analysisd ossec-logcollector ossec-remoted ossec-syscheckd ossec-monitord"
 210
 211     echo "Starting $NAME $VERSION (by $AUTHOR)..."
 212     echo | ${DIR}/bin/ossec-logtest > /dev/null 2>&1;
 213     if [ ! $? = 0 ]; then
 214         echo "OSSEC analysisd: Testing rules failed. Configuration error. Exiting."
 215         exit 1;
 216     fi
 217     lock;
 218     checkpid;
 219
 220
 221     # We actually start them now.
 222     for i in ${SDAEMONS}; do
 223         pstatus ${i};
 224         if [ $? = 0 ]; then
 225             ${DIR}/bin/${i} ${DEBUG_CLI};
 226             if [ $? != 0 ]; then
 227                 unlock;
 228                 exit 1;
 229             fi
 230
 231             echo "Started ${i}..."
 232         else
 233             echo "${i} already running..."
 234         fi
 235
 236     done
 237
 238     # After we start we give 2 seconds for the daemons
 239     # to internally create their PID files.
 240     sleep 2;
 241     unlock;
 242     echo "Completed."
 243 }
 244
 245 # Process status
 246 pstatus()
 247 {
 248     pfile=$1;
 249
 250     # pfile must be set
 251     if [ "X${pfile}" = "X" ]; then
 252         return 0;
 253     fi
 254
 255     ls ${DIR}/var/run/${pfile}*.pid > /dev/null 2>&1
 256     if [ $? = 0 ]; then
 257         for j in `cat ${DIR}/var/run/${pfile}*.pid 2>/dev/null`; do
 258             ps -p $j |grep ossec >/dev/null 2>&1
 259             if [ ! $? = 0 ]; then
 260                 echo "${pfile}: Process $j not used by ossec, removing .."
 261                 rm -f ${DIR}/var/run/${pfile}-$j.pid
 262                 continue;
 263             fi
 264
 265             kill -0 $j > /dev/null 2>&1
 266             if [ $? = 0 ]; then
 267                 return 1;
 268             fi
 269         done
 270     fi
 271
 272     return 0;
 273 }
 274
 275
 276 # Stop all
 277 stopa()
 278 {
 279     lock;
 280     checkpid;
 281     for i in ${DAEMONS}; do
 282         pstatus ${i};
 283         if [ $? = 1 ]; then
 284             echo "Killing ${i} .. ";
 285
 286             kill `cat ${DIR}/var/run/${i}*.pid`;
 287         else
 288             echo "${i} not running ..";
 289         fi
 290
 291         rm -f ${DIR}/var/run/${i}*.pid
 292
 293      done
 294
 295     unlock;
 296     echo "$NAME $VERSION Stopped"
 297 }
 298
 299
 300 ### MAIN HERE ###
 301
 302 case "$1" in
 303   start)
 304     testconfig
 305         start
 306         ;;
 307   stop)
 308         stopa
 309         ;;
 310   restart)
 311     testconfig
 312         stopa
 313         sleep 1;
 314         start
 315         ;;
 316   reload)
 317         DAEMONS="ossec-monitord ossec-logcollector ossec-remoted ossec-syscheckd ossec-analysisd ossec-maild ${DB_DAEMON} ${CSYSLOG_DAEMON} ${AGENTLESS_DAEMON}"
 318         stopa
 319         start
 320         ;;
 321   status)
 322     status
 323         ;;
 324   help)
 325     help
 326     ;;
 327   enable)
 328     enable $1 $2;
 329     ;;
 330   disable)
 331     disable $1 $2;
 332     ;;
 333   *)
 334     help
 335 esac