#!/bin/sh

# xl2tpd.sh -- xl2tpd connect / disconnect
#
# Copyright (C) 2014 Zeno Zeng <zenoofzeng@gmail.com>
# Copyright (C) 2014 Zhang Hai <Dreaming.in.Code.ZH@Gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see
# <http://www.gnu.org/licenses/>.

PPP_LOG_FILE=/var/log/zjuvpn


xl2tpd_restart() {

    # for Openwrt / Debian / Ubuntu
    type systemctl >/dev/null 2>&1 || {
        /etc/init.d/xl2tpd restart
    }

    # for Arch Linux
    type systemctl >/dev/null 2>&1 && {
        systemctl xl2tpd restart
    }
    
    # wait until ready
    for i in $(seq 0 120); do
        if [ -e "/var/run/xl2tpd/l2tp-control" ]; then
            return 0
        fi
        sleep 1
    done
}

xl2tpd_create_lac() {
    username=$1
    password=$2

    LNS="10.5.1.9"
    LAC_NAME=zjuvpn-${username}
    PPP_OPT_FILE=/etc/ppp/peers/${LAC_NAME}
    L2TPD_CFG_FILE=/etc/xl2tpd/xl2tpd.conf

    touch $PPP_LOG_FILE

    cat > $PPP_OPT_FILE <<EOF
noauth
linkname $LAC_NAME
logfile $PPP_LOG_FILE
name $username
password $password
EOF
    chmod 600 $PPP_OPT_FILE

    touch $L2TPD_CFG_FILE
    if ! grep -q "\[lac ${LAC_NAME}\]" $L2TPD_CFG_FILE; then
        cat >> $L2TPD_CFG_FILE <<EOF
[lac ${LAC_NAME}]
lns = $LNS
redial = no
redial timeout = 5
require chap = yes
require authentication = no
ppp debug = no
pppoptfile = ${PPP_OPT_FILE}
require pap = no
autodial = yes
 
EOF
    fi
}

disconnect() {
    username=$1
    LAC_NAME=zjuvpn-$1
    xl2tpd-control disconnect ${LAC_NAME}
}

connect() {
    username=$1
    LAC_NAME=zjuvpn-$1
    xl2tpd-control disconnect ${LAC_NAME} > /dev/null
    xl2tpd-control connect ${LAC_NAME} > /dev/null

    echo -n > $PPP_LOG_FILE

    prev_count=$(ip addr show | grep 'inet.*ppp' | grep ' 10.5.' | wc -l)

    for i in $(seq 0 120); do

        tail $PPP_LOG_FILE

        count=$(ip addr show | grep 'inet.*ppp' | grep ' 10.5.' | wc -l)
        if [ ${count} -gt ${prev_count} ]; then
            echo "Bring up ppp, done."
            return
        fi

        sleep 1
    done

    echo "Fail to bring up ppp. Timeout."

    disconnect $1
}

case $1 in

    connect)
        xl2tpd_restart
        xl2tpd_create_lac $2 $3
        connect $2 $3
        ;;

    disconnect)
        disconnect $2 $3
        ;;

esac