Subversion Repositories ALCASAR

Rev

Details | Last modification | View Log

Rev Author Line No. Line
219 jeremy 1
#!/bin/bash
328 franck 2
# $Id: alcasar-load_balancing.sh 2454 2017-12-09 18:59:31Z tom.houdayer $
221 franck 3
 
1068 franck 4
# Generic Load balancer for multiple WAN links - version 1.1 (04 Feb 2011)
5
# (c) 2011 Pau Oliva Fora - http://pof.eslack.org
6
#
7
# Licensed under GPLv3 - for full terms see:
8
# http://www.gnu.org/licenses/gpl-3.0.html
9
#
10
# Adapted and debugged (adr et ping -S) by ALCASAR Team (3abtux@alcasar.net)
11
# (c) 2013  3abtux - http://www.alcasar.net
12
#
13
# Specify each WAN link in a separate column, example:
14
# In this example we have 3 wan links (vlanXXX interfaces) attached to a single
15
# physical interface because we use a vlan-enabled switch between the balancer
16
# machine and the ADSL routers we want to balance. The weight parameter should
1078 franck 17
# be kept to a low integer.
1068 franck 18
#
19
#
20
# Modified by ALCASAR team :
219 jeremy 21
 
672 richard 22
 
1078 franck 23
prog="alcasar-load_balancing.sh"
24
pidfile="/var/run/alcasar-load_balancing.pid"
25
 
1068 franck 26
###############################
27
# MAIN PARAMETERs Configuration
28
###############################
219 jeremy 29
 
1078 franck 30
DIR_ETC="/usr/local/etc"
31
CONF_FILE="$DIR_ETC/alcasar.conf"
32
MULTIWAN=`grep MULTIWAN= $CONF_FILE|cut -d"=" -f2`
33
MULTIWAN=${MULTIWAN:=off}
34
FAILOVER=`grep FAILOVER= $CONF_FILE|cut -d"=" -f2`
35
FAILOVER=${FAILOVER:=30}
1789 franck 36
EXTIF=`grep ^EXTIF= $CONF_FILE|cut -d"=" -f2`				# INTernal InterFace
1078 franck 37
 
38
 
1068 franck 39
# space separated list of public IPs to ping in watchdog mode
40
# set this to some public ip addresses pingable and always on.
41
TESTIPS="8.8.8.8 192.0.32.10"
219 jeremy 42
 
1068 franck 43
# set to 1 when testing, set to 0 when happy with the results
1078 franck 44
VERBOSE=0
219 jeremy 45
 
1068 franck 46
# CONFIGURATION ENDS HERE
47
###############################
219 jeremy 48
 
49
 
1068 franck 50
if [ $(whoami) != "root" ]; then
2454 tom.houday 51
	echo "You must be root to run this!" ; echo ; exit 1
219 jeremy 52
fi
53
 
1068 franck 54
# Adapter for ALCASAR project
55
CONF_FILE="/usr/local/etc/alcasar.conf"
219 jeremy 56
 
1068 franck 57
# Virtual interfaces creating
58
function create_eth () {
59
	routecmd="ip route replace default scope global"
60
	NBIFACE=`grep "^WAN" $CONF_FILE | wc -l`	# Nbre interfaces virtuelles
61
	i=0
62
	while [ $i -le $NBIFACE ]
63
	do
64
		INT="WAN$i"
65
		echo $INT
66
		ACTIVE=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $1}'`	# Active
67
		WT=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $5}'`		# WEIGHT
68
		WT=${WT:-1}
69
		IP=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $3}' | cut -d"/" -f1`	# @IP
937 franck 70
 
1068 franck 71
		if [ $i -ne 0 ]; then
1789 franck 72
			[ -e /etc/sysconfig/network-scripts/ifcfg-$EXTIF:$i ] && ifdown $EXTIF:$i && rm -f /etc/sysconfig/network-scripts/ifcfg-$EXTIF:$i
1068 franck 73
			IFACE=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $2}'`	# IFACE
74
			IP_NET=`grep "^$INT=" $CONF_FILE | awk -F'"' '{print $2}' | awk -F, '{ print $3}'`	# IP
75
			NET="`ipcalc -n $IP_NET | cut -d"=" -f2`/`ipcalc -p $IP_NET|cut -d"=" -f2`"
76
			GW=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $4}'`		# @GW
77
			MTU=`grep "$INT=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $6}'`	# MTU
219 jeremy 78
 
1789 franck 79
			# Config $EXTIF:$i (Internet)
80
			cat <<EOF > /etc/sysconfig/network-scripts/ifcfg-$EXTIF:$i
1068 franck 81
DEVICE=$IFACE
82
BOOTPROTO=static
83
IPADDR=`echo $IP | cut -d"/" -f1`
84
NETMASK=`ipcalc -m $IP_NET | cut -d= -f2`
85
NETWORK=`ipcalc -n $IP_NET | cut -d= -f2`
86
MTU=$MTU
87
ONBOOT=yes
88
NOZEROCONF=yes
89
MII_NOT_SUPPORTED=yes
90
IPV6INIT=no
91
IPV6TO4INIT=no
92
ACCOUNTING=no
93
USERCTL=no
94
EOF
1789 franck 95
			echo "ifup $EXTIF:$i"
96
			ifup $EXTIF:$i
1068 franck 97
			NET="`ipcalc -n $IP_NET | cut -d"=" -f2`/`ipcalc -p $IP_NET|cut -d"=" -f2`"
98
		else
1789 franck 99
			IFACE="$EXTIF"
1068 franck 100
			IP_NET=`grep "^PUBLIC_IP=" $CONF_FILE | awk -F'=' '{print $2}'`			# IP/MSK
101
			IP=`grep "^PUBLIC_IP=" $CONF_FILE | awk -F= '{ print $2 }' | cut -d"/" -f1`	# @IP
102
			GW=`grep "^GW=" $CONF_FILE | awk -F= '{print $2}'`				# @GW
103
#			MTU=`grep "^PUBLIC_MTU=" $CONF_FILE | awk -F= '{print $2}'`			# MTU
104
		fi # End
219 jeremy 105
 
1068 franck 106
		NET="`ipcalc -n $IP_NET | cut -d"=" -f2`/`ipcalc -p $IP_NET|cut -d"=" -f2`"
2454 tom.houday 107
		if [ "$PARAM" == "add" ]; then
1068 franck 108
			set -x
109
			table=$(($i + 1))
110
			ip route ${PARAM} ${NET} dev ${IFACE} src ${IP} table $table
111
			ip route ${PARAM} default via ${GW} table $table
112
			ip rule ${PARAM} from ${IP} table $table
113
			set +x
114
		fi
115
		echo "	Iface: ${IFACE}"
116
		echo "	IP: ${IP}"
117
		echo "	IP_NET: ${IP_NET}"
118
		echo "	NET: ${NET}"
119
		echo "	GW: ${GW}"
120
		echo "	Weight: ${WT}"
121
		echo "	MTU : ${MTU}"
122
		echo
123
		routecmd="${routecmd} nexthop via ${GW} dev ${IFACE} weight ${WT}"
124
		i=$(($i + 1))
125
	done # End While
219 jeremy 126
 
2454 tom.houday 127
	if [ "$PARAM" == "add" ]; then
1068 franck 128
		echo "[] Balanced routing:"
129
		# suppress default route
130
		ip route del default scope global
131
		set -x
132
		${routecmd}
133
		set +x
134
		echo
135
	fi
2454 tom.houday 136
 
1068 franck 137
} # end create_eth
219 jeremy 138
 
1068 franck 139
###########################
140
# Fonction virtual Interfaces deleting
141
###########################
142
delete_eth () {
1789 franck 143
	IFACE_COUNT=`ls -l /etc/sysconfig/network-scripts/ifcfg-$EXTIF:* | wc -l`
1068 franck 144
	echo $IFACE_COUNT
145
	while [ $IFACE_COUNT -ne 0 ]
146
	do
2454 tom.houday 147
		i=$IFACE_COUNT
1789 franck 148
		echo "ifdown $EXTIF:$i"
149
		ifdown $EXTIF:$i
150
		rm -f /etc/sysconfig/network-scripts/ifcfg-$EXTIF:$i
1068 franck 151
		IFACE_COUNT=$(($IFACE_COUNT - 1))
152
	done
153
	ip route del default scope global
154
#	ip route add default gw 192.168.1.1
155
}
219 jeremy 156
 
2454 tom.houday 157
 
1068 franck 158
# do not modify below this line unless you know what you're doing :)
159
function getvalue() {
2454 tom.houday 160
	index=$1
161
	VAR=$2
219 jeremy 162
 
2454 tom.houday 163
	n=1
164
	for f in ${VAR} ; do
165
		if [ "${n}" == "${index}" ]; then
166
			echo "$f"
167
			break
168
		fi
169
		n=$(($n++))
170
	done
1068 franck 171
}
172
 
173
######################
174
# Fonction de FailOver
175
######################
176
function failover () {
177
 
178
	echo "[] Watchdog started"
179
	# 0 == all links ok, 1 == some link down
180
	STATE=0
2454 tom.houday 181
 
1068 franck 182
	DOWNCOUNT_BAK=0
183
	DOWN_BAK=""
184
	NBIFACE=`grep "^WAN" $CONF_FILE | wc -l`	# Nbre interfaces virtuelles
185
	echo "Nombre interfaces =  "$NBIFACE
1789 franck 186
	WANIFACE[0]="$EXTIF"
1068 franck 187
	c=0
188
	while [ $c -le $NBIFACE ]; do
189
		ITH=(`grep "WAN$c=" $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $2}'`)	# IFACE
190
		echo $ITH
191
		WANIFACE="${WANIFACE} $ITH"
192
		echo $WANIFACE
193
		c=$(($c + 1))
194
	done
195
	echo "Liste des interfaces : "${WANIFACE[*]}
196
	# Failover test
197
	while : ; do
2454 tom.houday 198
 
1068 franck 199
		if [ $VERBOSE -eq 1 ]; then
200
			echo "[] Sleeping, state=$STATE"
201
		fi
1078 franck 202
		sleep $FAILOVER
2454 tom.houday 203
 
1068 franck 204
		IFINDEX=1
205
		DOWN=""			# liste des interfaces down
206
		DOWNCOUNT=0		# nombre d'interface down
207
		for iface in $WANIFACE ; do
208
			COUNT=0		# compteur de test
209
			FAIL=0		# Nombre de fois down
1469 richard 210
			# Recup de l'adresse IP dynamiquement
1068 franck 211
			IP=`ifconfig $iface |grep "inet adr" |cut -f 2 -d ":" |awk '{print $1}'`
212
			if [ $i -ne 0 ]; then
213
				GW=`grep "$iface," $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $4}'`		# @GW
214
				WT=`grep "$iface," $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $5}'`		# @WT
215
			else
216
				GW=`grep "^GW=" $CONF_FILE | awk -F= '{print $2}'`			# @GW
2454 tom.houday 217
			fi
1068 franck 218
			for TESTIP in $TESTIPS ; do
219
				COUNT=$(($COUNT + 1))
220
				ping -W 3 -I $IP -c 1 $TESTIP > /dev/null 2>&1
221
#				ping -W 3 -I $IP -c 1 $TESTIP
222
				# Si ping de la première adresse --> ok  --> stop du test pour l'interface testée
223
				if [ $? -eq 0 ]; then
224
					break
2454 tom.houday 225
				else
1068 franck 226
					# sinon on compte une erreur
227
					FAIL=$(($FAIL + 1))
228
				fi
229
			done # End of test sur un serveur Internet
230
			# Affichage du nombre de down
231
			echo "FAIL=$FAIL"
232
			# Si nombre de fois down = nombre de tests -->  Iface down --> log dans fichier log avec l'heure
233
			if [ $FAIL -eq $COUNT ]; then
234
				echo "`date +%F-%Hh%mm%Ss` : [WARN] $iface is down!"
235
				# Si etat différent de 1 (déjà tombé) --> changement de l'état général en default
236
				if [ $STATE -ne 1 ]; then
237
					echo "Switching state $STATE -> 1"
238
					STATE=1
239
				fi
240
				# Rajout de l'iface dans la liste des interfaces down
241
				DOWN="${DOWN} $IFINDEX"
242
				echo "DOWN=$DOWN"
243
				# Nombre d'interface down
244
				DOWNCOUNT=$(($DOWNCOUNT + 1))
245
				echo "DOWNCOUNT=$DOWNCOUNT"
246
			fi
247
			IFINDEX=$(($IFINDEX + 1))
248
			echo "IFINDEX =$IFINDEX"
249
		done # End Test Interface in WANIFACE
250
 
2454 tom.houday 251
		#  0 Passerelle down et état précédent différent (retour à la normale)) --> mise à la normale des passerelles
1068 franck 252
#		if [ $DOWNCOUNT -eq 0 ] && [ $DOWNCOUNT -ne $DOWNCOUNT_BAK ]; then
253
		if [ $DOWNCOUNT -eq 0 ] ; then
254
			if [ $STATE -eq 1 ]; then
255
				echo
256
				echo "[] All links up and running :)"
257
				set -x
258
				${routecmd}
259
				set +x
260
				# Changement de l'état en normal
261
				STATE=0
262
				echo "Switching state 1 -> 0"
263
			fi # End retour etat normal
264
			# if no interface is down, go to the next cycle
265
			continue
266
		# cas ou au moins une passerelle down mais état identique au précédent Test --> rien à changer
267
		else
268
			if [ "$DOWN_BAK" == "$DOWN" ]; then
269
			echo "DOWN_BAK == DOWN = $DOWN"
270
				continue	# --> état identique test precedent --> boucle suivante
271
		# cas ou au moins une passerelle down mais état différent de test précédent --> remplacement par nouvelle règle
272
			else
273
				cmd="ip route replace default scope global"
274
				IFINDEX=1
275
				suffix=""
276
				# Pour chaque interface --> traitement et application de la règle de routage
277
				for iface in $WANIFACE ; do
278
					echo "-------------------------"
279
					echo "iface=$iface"
280
					echo "Index = " $IFINDEX
281
					FAILIF=0
2454 tom.houday 282
					# Pour chaque interface down -->
1068 franck 283
					echo "Interfaces DOWN = $DOWN"
284
					for lnkdwn in $DOWN ; do
285
						echo "LINKDOWN = "$lnkdown
286
						if [ $lnkdwn -eq $IFINDEX ]; then
287
							FAILIF=1
2454 tom.houday 288
							break
1068 franck 289
						else
290
							continue
291
						fi
292
					done # End linkdown in DOWN
293
					# Interface en etat normal --> rajout de la règle en mode nexthop
294
					if [ $FAILIF -eq 0 ]; then
295
						IP=`ifconfig $iface |grep "inet adr" |cut -f 2 -d ":" |awk '{print $1}'`
1789 franck 296
						if [ $iface != "$EXTIF" ]; then
1068 franck 297
							GW=`grep "$iface," $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $4}'`		# @GW
298
							WT=`grep "$iface," $CONF_FILE | awk -F'"' '{ print $2 }' | awk -F, '{ print $5}'`		# @GW
299
						else
300
							GW=`grep "^GW=" $CONF_FILE | awk -F= '{print $2}'`			# @GW
2454 tom.houday 301
						fi
1068 franck 302
						echo "GW=$GW"
303
						echo "WT=$WT"
304
						echo "suffix=$sufix"
305
						suffix="${suffix} nexthop via ${GW} dev ${iface} weight ${WT:-1}"
306
					fi # End interface = noFAIL
307
					IFINDEX=$(($IFINDEX + 1))
308
				done # End  iface IN WANIFACE
309
				# Commande globale
310
				cmd="ip route replace default scope global $suffix"
2454 tom.houday 311
 
1068 franck 312
				if [ $VERBOSE -eq 1 ]; then
313
					set -x
314
			#		echo "Avec commentaire : " ${cmd}
315
					${cmd}
316
					set +x
317
					echo
318
				else
319
					${cmd} 2>/dev/null
320
					echo ${cmd}
321
				fi # end Application de la commande de routage globale
322
			fi #
323
			DOWN_BAK=$DOWN	# Enregistrement de l'etat
2454 tom.houday 324
		fi # End
1068 franck 325
	done
326
} # End of Failover
327
 
328
 
329
#################
330
# Main
331
#################
332
 
333
echo "[] Load balancer for multiple WAN interfaces - v2.1"
334
echo "[] (c) 2011 Pau Oliva Fora <pof> @eslack.org"
335
echo "[] (c) 2013 3abtux ALCASAR  <3abtux> @alcasar.net"
336
echo
337
 
338
case $1 in
2454 tom.houday 339
	create)
340
		create_eth
1068 franck 341
	;;
2454 tom.houday 342
	delete)
343
		delete_eth
1068 franck 344
	;;
2454 tom.houday 345
	start)
346
		if [ "$MULTIWAN" != "on" ] && [ "$MULTIWAN" != "On" ]; then
1078 franck 347
		    echo "The MultiGateway is not activated !"
348
		    exit 0
1068 franck 349
		fi
2454 tom.houday 350
		PARAM="add"
351
		create_eth
352
		ip route flush cache
353
		if [ $FAILOVER -eq 0 ]; then
1078 franck 354
		      echo "The MultiWAN Mode is actived but not failover connectivity !"
355
		      exit 0
356
		fi
2454 tom.houday 357
		echo "Starting down $prog: "
358
		pid=`pidof -x "alcasar-load_balancing.sh"`
359
		if [ $pid != "" ]; then
360
			echo $pid > $pidfile
361
		fi
362
		touch /var/lock/subsys/alcasar-load_balancing
363
		failover
1068 franck 364
	;;
2454 tom.houday 365
	stop)
1068 franck 366
		PARAM="del"
1078 franck 367
		echo "Shutting down $prog: "
2454 tom.houday 368
		if [ -f $pidfile ]; then
369
			pid=`cat $pidfile`
370
			kill -9 $pid
371
		else
372
			echo "$prog is not running."
373
			exit 1
374
		fi
375
		RETVAL=$?
376
		echo
377
		[ $RETVAL -eq 0 ] && rm -f $pidfile && rm -f /var/lock/subsys/alcasar-load_balancing
378
		echo "Delete of virtual interfaces"
379
		delete_eth
380
		echo "Network restart"
381
		service network restart 2>&1 > /dev/null
382
		ip route
383
 
1068 franck 384
	;;
1078 franck 385
	status)
2454 tom.houday 386
		echo "Checking  $prog : "
387
		if [ -f $pidfile ]; then
388
			pid=`cat $pidfile`
389
			CHECK=`ps -p $pid --no-heading | awk {'printf $1'}`
390
			if [ "$CHECK" = "" ]; then
391
				echo "$prog is NOT running."
392
			else
393
				echo "$prog is running !"
394
			fi
395
		else
396
				echo "$prog is Not running."
397
		fi
1078 franck 398
	;;
2454 tom.houday 399
	fail)
400
		failover
1078 franck 401
	;;
2454 tom.houday 402
	*)
1078 franck 403
		echo "Usage: $0 [start|stop|status|create|delete]" ; echo ; exit 1
1068 franck 404
	;;
405
esac
406
 
407
exit 0