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