13bba6f8fbe2c0f549c2a85059d5e7c9149dac3c
[sahlberg/ctdb.git] / config / functions
1 # utility functions for ctdb event scripts
2
3 #######################################
4 # pull in a system config file, if any
5 loadconfig() {
6     name="$1"
7     if [ -f /etc/sysconfig/$name ]; then
8         . /etc/sysconfig/$name
9     elif [ -f /etc/default/$name ]; then
10         . /etc/default/$name
11     elif [ -f $CTDB_BASE/sysconfig/$name ]; then
12         . $CTDB_BASE/sysconfig/$name
13     fi
14 }
15
16
17 ######################################################
18 # simulate /sbin/service on platforms that don't have it
19 service() { 
20   service_name="$1"
21   op="$2"
22   if [ -x /sbin/service ]; then
23       /sbin/service "$service_name" "$op"
24   elif [ -x /etc/init.d/$service_name ]; then
25       /etc/init.d/$service_name "$op"
26   elif [ -x /etc/rc.d/init.d/$service_name ]; then
27       /etc/rc.d/init.d/$service_name "$op"
28   fi
29 }
30
31 ######################################################
32 # simulate /sbin/service (niced) on platforms that don't have it
33 nice_service() { 
34   service_name="$1"
35   op="$2"
36   if [ -x /sbin/service ]; then
37       nice /sbin/service "$service_name" "$op"
38   elif [ -x /etc/init.d/$service_name ]; then
39       nice /etc/init.d/$service_name "$op"
40   elif [ -x /etc/rc.d/init.d/$service_name ]; then
41       nice /etc/rc.d/init.d/$service_name "$op"
42   fi
43 }
44
45 ######################################################
46 # wait for a command to return a zero exit status
47 # usage: ctdb_wait_command SERVICE_NAME <command>
48 ######################################################
49 ctdb_wait_command() {
50   service_name="$1"
51   wait_cmd="$2"
52   [ -z "$wait_cmd" ] && return;
53   all_ok=0
54   echo "Waiting for service $service_name to start"
55   while [ $all_ok -eq 0 ]; do
56           $wait_cmd > /dev/null 2>&1 && all_ok=1
57           ctdb status > /dev/null 2>&1 || {
58                 echo "ctdb daemon has died. Exiting wait for $service_name"
59                 exit 1
60           }
61           [ $all_ok -eq 1 ] || sleep 1
62   done
63   echo "Local service $service_name is up"
64 }
65
66
67 ######################################################
68 # wait for a set of tcp ports
69 # usage: ctdb_wait_tcp_ports SERVICE_NAME <ports...>
70 ######################################################
71 ctdb_wait_tcp_ports() {
72   service_name="$1"
73   shift
74   wait_ports="$*"
75   [ -z "$wait_ports" ] && return;
76   all_ok=0
77   echo "Waiting for tcp service $service_name to start"
78   while [ $all_ok -eq 0 ]; do
79           all_ok=1
80           for p in $wait_ports; do
81               if [ -x /usr/bin/netcat ]; then
82                   /usr/bin/netcat -z 127.0.0.1 $p > /dev/null || all_ok=0
83               elif [ -x /usr/bin/nc ]; then
84                   /usr/bin/nc -z 127.0.0.1 $p > /dev/null || all_ok=0
85               elif [ -x /usr/bin/netstat ]; then
86                   (netstat -a -n | egrep "0.0.0.0:$p\s*LISTEN" > /dev/null) || all_ok=0
87               elif [ -x /bin/netstat ]; then
88                   (netstat -a -n | egrep "0.0.0.0:$p\s*LISTEN" > /dev/null) || all_ok=0
89               else 
90                   echo "No tool to check tcp ports availabe. can not check in ctdb_wait_tcp_ports"
91                   return
92               fi
93           done
94           [ $all_ok -eq 1 ] || sleep 1
95           ctdb status > /dev/null 2>&1 || {
96                 echo "ctdb daemon has died. Exiting tcp wait $service_name"
97                 exit 1
98           }
99   done
100   echo "Local tcp services for $service_name are up"
101 }
102
103
104
105 ######################################################
106 # wait for a set of directories
107 # usage: ctdb_wait_directories SERVICE_NAME <directories...>
108 ######################################################
109 ctdb_wait_directories() {
110   service_name="$1"
111   shift
112   wait_dirs="$*"
113   [ -z "$wait_dirs" ] && return;
114   all_ok=0
115   echo "Waiting for local directories for $service_name"
116   while [ $all_ok -eq 0 ]; do
117           all_ok=1
118           for d in $wait_dirs; do
119               [ -d $d ] || all_ok=0
120           done
121           [ $all_ok -eq 1 ] || sleep 1
122           ctdb status > /dev/null 2>&1 || {
123                 echo "ctdb daemon has died. Exiting directory wait for $service_name"
124                 exit 1
125           }
126   done
127   echo "Local directories for $service_name are available"
128 }
129
130
131 ######################################################
132 # check that a rpc server is registered with portmap
133 # and responding to requests
134 # usage: ctdb_check_rpc SERVICE_NAME PROGNUM VERSION
135 ######################################################
136 ctdb_check_rpc() {
137     service_name="$1"
138     prognum="$2"
139     version="$3"
140     rpcinfo -u localhost $prognum $version > /dev/null || {
141             echo "ERROR: $service_name not responding to rpc requests"
142             exit 1
143     }
144 }
145
146 ######################################################
147 # check a set of directories is available
148 # usage: ctdb_check_directories SERVICE_NAME <directories...>
149 ######################################################
150 ctdb_check_directories() {
151   service_name="$1"
152   shift
153   wait_dirs="$*"
154   [ -z "$wait_dirs" ] && return;
155   for d in $wait_dirs; do
156       [ -d $d ] || {
157           echo "ERROR: $service_name directory $d not available"
158           exit 1
159       }
160   done
161 }
162
163 ######################################################
164 # check a set of tcp ports
165 # usage: ctdb_check_tcp_ports SERVICE_NAME <ports...>
166 ######################################################
167 ctdb_check_tcp_ports() {
168   service_name="$1"
169   shift
170   wait_ports="$*"
171   [ -z "$wait_ports" ] && return;
172   for p in $wait_ports; do
173       all_ok=1
174       if [ -x /usr/bin/netcat ]; then
175           /usr/bin/netcat -z 127.0.0.1 $p > /dev/null || all_ok=0
176       elif [ -x /usr/bin/nc ]; then
177           /usr/bin/nc -z 127.0.0.1 $p > /dev/null || all_ok=0
178       elif [ -x /usr/bin/netstat ]; then
179           (netstat -a -n | egrep "0.0.0.0:$p .*LISTEN" > /dev/null ) || all_ok=0
180       elif [ -x /bin/netstat ]; then
181           (netstat -a -n | egrep "0.0.0.0:$p .*LISTEN" > /dev/null ) || all_ok=0
182       fi
183       [ $all_ok -eq 1 ] || {
184           echo "ERROR: $service_name tcp port $p is not responding"
185           exit 1
186       }
187   done
188 }
189
190 ######################################################
191 # check a command returns zero status
192 # usage: ctdb_check_command SERVICE_NAME <command>
193 ######################################################
194 ctdb_check_command() {
195   service_name="$1"
196   wait_cmd="$2"
197   [ -z "$wait_cmd" ] && return;
198   $wait_cmd > /dev/null 2>&1 || {
199       echo "ERROR: $service_name - $wait_cmd returned error"
200       exit 1
201   }
202 }
203
204 ################################################
205 # kill off any TCP connections with the given IP
206 ################################################
207 kill_tcp_connections() {
208     _IP="$1"    
209     _failed=0
210
211     _killcount=0
212     connfile="$CTDB_BASE/state/connections.$_IP"
213     netstat -tn |egrep "^tcp.*\s+$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' > $connfile
214     while read dest src; do
215         srcip=`echo $src | cut -d: -f1`
216         srcport=`echo $src | cut -d: -f2`
217         destip=`echo $dest | cut -d: -f1`
218         destport=`echo $dest | cut -d: -f2`
219         ctdb killtcp $srcip:$srcport $destip:$destport >/dev/null 2>&1 || _failed=1
220         echo "Killing TCP connection $srcip:$srcport $destip:$destport"
221         _killcount=`expr $_killcount + 1`
222     done < $connfile
223     /bin/rm -f $connfile
224     [ $_failed = 0 ] || {
225         echo "Failed to send killtcp control"
226         return;
227     }
228     [ $_killcount -gt 0 ] || {
229         return;
230     }
231     _count=0
232     while netstat -tn |egrep "^tcp.*\s+$_IP:.*ESTABLISHED" > /dev/null; do
233         sleep 1
234         _count=`expr $_count + 1`
235         [ $_count -gt 3 ] && {
236             echo "Timed out killing tcp connections for IP $_IP"
237             return;
238         }
239     done
240     echo "killed $_killcount TCP connections to released IP $_IP"
241 }
242
243 ########################################################
244 # start/stop the nfs service on different platforms
245 ########################################################
246 startstop_nfs() {
247         PLATFORM="unknown"
248         [ -x /etc/init.d/nfsserver ] && {
249                 PLATFORM="sles"
250         }
251         [ -x /etc/init.d/nfslock ] && {
252                 PLATFORM="rhel"
253         }
254
255         case $PLATFORM in
256         sles)
257                 case $1 in
258                 start)
259                         service nfsserver start
260                         ;;
261                 stop)
262                         service nfsserver stop > /dev/null 2>&1
263                         ;;
264                 esac
265                 ;;
266         rhel)
267                 case $1 in
268                 start)
269                         service nfslock start
270                         service nfs start
271                         ;;
272                 stop)
273                         service nfs stop > /dev/null 2>&1
274                         service nfslock stop > /dev/null 2>&1
275                         ;;
276                 esac
277                 ;;
278         *)
279                 echo "Unknown platform. NFS is not supported with ctdb"
280                 exit 1
281                 ;;
282         esac
283 }
284
285 ########################################################
286 # start/stop the nfs lockmanager service on different platforms
287 ########################################################
288 startstop_nfslock() {
289         PLATFORM="unknown"
290         [ -x /etc/init.d/nfsserver ] && {
291                 PLATFORM="sles"
292         }
293         [ -x /etc/init.d/nfslock ] && {
294                 PLATFORM="rhel"
295         }
296
297         case $PLATFORM in
298         sles)
299                 # for sles there is no service for lockmanager
300                 # so we instead just shutdown/restart nfs
301                 case $1 in
302                 start)
303                         service nfsserver start
304                         ;;
305                 stop)
306                         service nfsserver stop > /dev/null 2>&1
307                         ;;
308                 esac
309                 ;;
310         rhel)
311                 case $1 in
312                 start)
313                         service nfslock start
314                         ;;
315                 stop)
316                         service nfslock stop > /dev/null 2>&1
317                         ;;
318                 esac
319                 ;;
320         *)
321                 echo "Unknown platform. NFS locking is not supported with ctdb"
322                 exit 1
323                 ;;
324         esac
325 }
326
327 ########################################################
328 # load a site local config file
329 ########################################################
330
331 [ -x $CTDB_BASE/rc.local ] && {
332         . $CTDB_BASE/rc.local
333 }