#!/bin/bash
# Dual-WAN Switching script
# Automatic switching to backup line if internet connection fails and back to preferred one if it starts working.
# Works on Linux and FreeBSD.
# (c) .kkursor, 2010
# This script is licensed under GNU GPLv3 license and you are free to redistribute it as long as you agree with the license.
# See http://www.gnu.org/licenses/gpl.html for details.

# We define 2 default gateways of WAN interfaces
gw1=192.168.50.1 # Gateway 1, preferred
gw2=192.168.100.1 # Gateway 2, backup
ping_hostname=ya.ru             # Host to ping
ping_ip=`host $ping_hostname | grep -v mail | awk '{ print $4 }' | tail -n 1`   # get an IP of required host
hold_preferred=YES      # YES if you want to use preferred line as much as possible. For example, if backup line is more expensive. If it does not matter, comment this line.
LOG=/var/log/change_gateway.log   # Logs

# DO NOT EDIT BELOW LINES
# Check what OS we use to modify route add commands
if [ `uname` == "FreeBSD" ]; then
        echo "[`date`] FreeBSD detected."
        gw_word=""
        os="bsd"
elif [ `uname` == "Linux" ]; then
        echo "[`date`] Linux detected."
        gw_word="gw"
        os="linux"
fi

# check_inet - check internet connection
function check_inet
{
        ping -c 1 $ping_hostname >> /dev/null 2>/dev/null
        echo $?
}

# See which gateway is currently in use
echo "[`date`] Определение текущего шлюза ..."
function current_gw
{
        if [ $os == "bsd" ]; then
                gw=`netstat -rn | grep default | awk '{ print $2 }'`
        elif [ $os == "linux" ]; then
                gw=`route -n 2>/dev/null | grep 0.0.0.0 | grep G | awk '{ print $2 }'`
        fi

        echo $gw
}

echo "[`date`] Текущий шлюз по умолчанию $(current_gw)."

echo "[`date`] Проверка интернет-соединения ..."
status=$(check_inet)

if [ $status -eq 0 -a "$(current_gw)" == "$gw1" ]; then
        echo "[`date`] Интернет-соединение - OK.";
elif [ $status -eq 0 -a "$(current_gw)" == "$gw2" ]; then
        echo "[`date`] Вы используете резервную линию."
                if [ "$hold_preferred" == "YES" ]; then
                        echo "[`date`] Проверка основной линии ..."
                        route add $ping_ip/32 $gw_word $gw1 >> /dev/null # add a route for ping hostname over primary line
                        ping -c 1 $ping_ip >> /dev/null
                        if [ $? -eq 0 ]; then
                                echo "[`date`] Основная линия исправна. Переключение на основную линию." >> $LOG
                                route del default >> /dev/null
                                route add default $gw_word $gw1 >> /dev/null
                        else
                                echo "[`date`] Основная линия не работает. Работает резервная линия."
                        fi
                        route del $ping_ip/32 >> /dev/null
                fi
elif [ $status -ne 0 ]; then
        if [ "$(current_gw)" == "$gw1" ]; then
                echo "[`date`] Основная линия неисправна. Переключение на резервную линию." >> $LOG
                route del default >> /dev/null
                route add default $gw_word $gw2 >> /dev/null
                if [ $(check_inet) -eq 0 ]; then
                        echo "[`date`] Резервная линия работоспособна."
                else
                        echo "[`date`] Резервная линия НЕ в рабочем состоянии. Позвоните своему провайдеру."
                        echo "[`date`] Переключение обратно на основную линию."
                        route del default >> /dev/null
                        route add default $gw_word $gw1 >> /dev/null
                        exit
                fi
        elif [ "$(current_gw)" == "$gw2" ]; then
                echo "[`date`] Резервная линия неисправна. Переключение на основную линию." >> $LOG
                route del default >> /dev/null
                route add default $gw_word $gw1 >> /dev/null
                if [ $(check_inet) -eq 0 ]; then
                        echo "[`date`] Основная линия работоспособна."
                else
                        echo "[`date`] Резервная линия НЕ в рабочем состоянии. Позвоните своему провайдеру."
                        echo "[`date`] Переключение обратно на резервную линию."
                        route del default >> /dev/null
                        route add default $gw_word $gw2 >> /dev/null
                fi
        else
                echo "[`date`] No default route set or current router is different from $gw1 and $gw2."
                echo "[`date`] Adding $gw1 as default router."
                route del default > /dev/null
                route add default $gw_word $gw1 > /dev/null
        fi
fi

echo "[`date`] Текущий маршрут по умолчанию $(current_gw)."

if [ $(check_inet) -eq 0 ]; then
        state="- РАБОТАЕТ"
else
        state="- НЕ РАБОТАЕТ"
fi

echo "[`date`] Интернет-соединение $state."