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.8.3"
  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                 echo "${i} did not start correctly.";
 228                 unlock;
 229                 exit 1;
 230             fi
 231
 232             echo "Started ${i}..."
 233         else
 234             echo "${i} already running..."
 235         fi
 236
 237     done
 238
 239     # After we start we give 2 seconds for the daemons
 240     # to internally create their PID files.
 241     sleep 2;
 242     unlock;
 243     echo "Completed."
 244 }
 245
 246 # Process status
 247 pstatus()
 248 {
 249     pfile=$1;
 250
 251     # pfile must be set
 252     if [ "X${pfile}" = "X" ]; then
 253         return 0;
 254     fi
 255
 256     ls ${DIR}/var/run/${pfile}*.pid > /dev/null 2>&1
 257     if [ $? = 0 ]; then
 258         for j in `cat ${DIR}/var/run/${pfile}*.pid 2>/dev/null`; do
 259             ps -p $j |grep ossec >/dev/null 2>&1
 260             if [ ! $? = 0 ]; then
 261                 echo "${pfile}: Process $j not used by ossec, removing .."
 262                 rm -f ${DIR}/var/run/${pfile}-$j.pid
 263                 continue;
 264             fi
 265
 266             kill -0 $j > /dev/null 2>&1
 267             if [ $? = 0 ]; then
 268                 return 1;
 269             fi
 270         done
 271     fi
 272
 273     return 0;
 274 }
 275
 276
 277 # Stop all
 278 stopa()
 279 {
 280     lock;
 281     checkpid;
 282     for i in ${DAEMONS}; do
 283         pstatus ${i};
 284         if [ $? = 1 ]; then
 285             echo "Killing ${i} .. ";
 286
 287             kill `cat ${DIR}/var/run/${i}*.pid`;
 288         else
 289             echo "${i} not running ..";
 290         fi
 291
 292         rm -f ${DIR}/var/run/${i}*.pid
 293
 294      done
 295
 296     unlock;
 297     echo "$NAME $VERSION Stopped"
 298 }
 299
 300
 301 ### MAIN HERE ###
 302
 303 case "$1" in
 304   start)
 305     testconfig
 306         start
 307         ;;
 308   stop)
 309         stopa
 310         ;;
 311   restart)
 312     testconfig
 313         stopa
 314         sleep 1;
 315         start
 316         ;;
 317   reload)
 318         DAEMONS="ossec-monitord ossec-logcollector ossec-remoted ossec-syscheckd ossec-analysisd ossec-maild ${DB_DAEMON} ${CSYSLOG_DAEMON} ${AGENTLESS_DAEMON}"
 319         stopa
 320         start
 321         ;;
 322   status)
 323     status
 324         ;;
 325   help)
 326     help
 327     ;;
 328   enable)
 329     enable $1 $2;
 330     ;;
 331   disable)
 332     disable $1 $2;
 333     ;;
 334   *)
 335     help
 336 esac