#!/bin/bash

ngxvcl=${BR}/_build/nginx_vcl
ngxldp=${BR}/_build/nginx_ldp

tls_tcp=tls

function test_vcl(){
    echo ""
    echo "===================================================================="
    echo "Testing ..."
    echo ""
    export LD_LIBRARY_PATH=/usr/local/ssl/lib

    cp ${BR}/configs/vppset-test /tmp/.
    ${BR}/vpp/build-root/install-vpp-native/vpp/bin/vpp -c ${BR}/configs/startup-test.conf &
    vpp_pid=$!

    sleep 5

    export VCL_CONFIG=${BR}/configs/vcl.conf

    export NGXVCL_TLS_ON=1
    export NGXVCL_TLS_CERT=${BR}/configs/tls-test-cert
    export NGXVCL_TLS_KEY=${BR}/configs/tls-test-key

    export LD_LIBRARY_PATH=${BR}/vpp/build-root/install-vpp-native/vpp/lib

    sudo killall -v -s 9 nginx || echo "continue execute"

    mdir=0
    if [ ! -d /usr/local/nginx/logs ]; then
        mdir=2
        if [ ! -d /usr/local/nginx ]; then
            mdir=1
            fi
    fi
    if [ ${mdir} -ne 0 ]; then
        mkdir -p /usr/local/nginx/logs
    fi
    ${ngxvcl}/objs/nginx -c ${BR}/configs/nginx-test.conf &
    nginx_pid=$!
    sleep 5

    echo "===================================================================="
    ret=0
    v=`ps -A|grep "${vpp_pid}" | wc -l`
    if [ ${v} -eq 1 ]; then
        echo "VCL test: vpp                                                 OK"
    else
        echo "VCL test: vpp                                                 FAIL"
        ret=1
    fi

    v=`ps -A|grep "${nginx_pid}" | wc -l`
    if [ ${v} -eq 1 ]; then
        echo "VCL test: nginx                                               OK"
    else
        echo "VCL test: nginx                                               FAIL"
        ret=1
    fi
    echo "===================================================================="

    sudo killall -v -s 9 nginx || echo ""
    sudo kill -9 ${vpp_pid} || echo ""

    rm /tmp/vppset-test
    if [ ${mdir} -eq 1 ]; then
        rm -rf /usr/local/nginx
    elif [ ${mdir} -eq 2 ]; then
        rm -rf /usr/local/nginx/logs
    fi

    if [ ${ret} -eq 1 ]; then
        exit 1;
    fi
}

function test_ldp(){
    echo ""
    echo "===================================================================="
    echo "Testing ..."
    echo ""

    export LD_LIBRARY_PATH=/usr/local/ssl/lib

    cp ${BR}/configs/vppset-test /tmp/.
    ${BR}/vpp/build-root/install-vpp-native/vpp/bin/vpp -c ${BR}/configs/startup-test.conf &
    vpp_pid=$!

    sleep 5

    mdir=0
    if [ ! -d /usr/local/nginx/logs ]; then
        mdir=2
        if [ ! -d /usr/local/nginx ]; then
            mdir=1
            fi
    fi
    if [ ${mdir} -ne 0 ]; then
        mkdir -p /usr/local/nginx/logs
    fi

    export VCL_CONFIG=${BR}/configs/vcl.conf
    export LDP_TRANSPARENT_TLS=1
    export LDP_TLS_CERT_FILE=${BR}/configs/tls-test-cert
    export LDP_TLS_KEY_FILE=${BR}/configs/tls-test-key
    LD_PRELOAD=${BR}/vpp/build-root/install-vpp-native/vpp/lib/libvcl_ldpreload.so \
        ${ngxldp}/objs/nginx -c ${BR}/configs/nginx-test.conf &

    nginx_pid=$!
    sleep 5

    echo "===================================================================="
    ret=0
    v=`ps -A|grep "${vpp_pid}" | wc -l`
    if [ ${v} -eq 1 ]; then
        echo "LDP test: vpp                                                 OK"
    else
        echo "LDP test: vpp                                                 FAIL"
        ret=1
    fi

    v=`ps -A|grep "${nginx_pid}" | wc -l`
    if [ ${v} -eq 1 ]; then
        echo "LDP test: nginx                                               OK"
    else
        echo "LDP test: nginx                                               FAIL"
        ret=1
    fi
    echo "===================================================================="

    sudo killall -v -s 9 nginx || echo ""
    sudo kill -9 ${vpp_pid} || echo ""

    rm /tmp/vppset-test
    if [ ${mdir} -eq 1 ]; then
        rm -rf /usr/local/nginx
    elif [ ${mdir} -eq 2 ]; then
        rm -rf /usr/local/nginx/logs
    fi

    if [ ${ret} -eq 1 ]; then
        exit 1
    fi
}

args=("$@")
case ${1} in
    vcl)
        test_vcl
        ;;
    ldp)
        test_ldp
        ;;
esac