56b04ada203d4d96ace759604b4a4961a0302a4c
[ossec-hids.git] / 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