#!/bin/sh /etc/rc.common
# campus portal login as an init.d service for OpenWrt (with procd monitor)

START=95
STOP=10
USE_PROCD=1

#  配置区 
USER_ID="" # 用户名
PASSWORD="" # 密码
SERVICE="DianXin" # Service 字段，没有可以留空

INTERFACE="eth1" # 网络端口，用于指定 Macvlan 多账号登录

PING_URL="http://baidu.com"
PORTAL_HOST="10.30.252.71" # 记得修改这个为自己学校的登录网址 IP
INDEX_URL="http://$PORTAL_HOST/eportal/index.jsp"
LOGIN_API="http://$PORTAL_HOST/eportal/InterFace.do?method=login"

CURL_UA="Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/68.0.3440.84 Safari/537.36"
CURL_OPTS="-fsS --connect-timeout 5 --max-time 10 --interface $INTERFACE"

# 监控间隔（秒）
MONITOR_INTERVAL=60
# 断线判定失败后重试次数/间隔
RETRY_MAX=3
RETRY_SLEEP=2

EXTRA_COMMANDS="login sstatus"
EXTRA_HELP="        login   Run a one-shot portal login now
        sstatus  Print current connectivity/portal status"

url_encode() {
    local s="${1}" i c out=""
    for i in $(seq 1 ${#s}); do
        c="$(printf '%s' "$s" | cut -c $i)"
        case "$c" in
            [a-zA-Z0-9.~_-]) out="${out}${c}" ;;
            *) out="${out}$(printf '%%%02X' "'$c")" ;;
        esac
    done
    printf '%s' "$out"
}

is_online() {
    local resp
    resp="$(curl $CURL_OPTS "$PING_URL" -A "$CURL_UA" || true)"
    echo "$resp" | grep -q "http://www\.baidu\.com/" && return 0 || return 1
}

extract_query_string() {
    local resp="$1"
    echo "$resp" \
    | grep -o "top\.self\.location\.href='http://$PORTAL_HOST/eportal/index\.jsp?[^']\+'" \
    | sed "s#.*index.jsp?##; s/'\$//"
}

do_login_once() {
    echo "[campus_portal_eth1] Probing connectivity..."
    if is_online; then
        echo "[campus_portal_eth1] Already online."
        return 0
    fi

    echo "[campus_portal_eth1] Fetching portal redirect page..."
    local resp qs enc1 enc2 post_data login_json
    resp="$(curl $CURL_OPTS "$PING_URL" -A "$CURL_UA" || true)"
    qs="$(extract_query_string "$resp")"
    if [ -z "$qs" ]; then
        echo "[campus_portal_eth1] ERROR: failed to get queryString."
        return 1
    fi
    echo "[campus_portal_eth1] Got queryString (len=${#qs})."

    enc1="$(url_encode "$qs")"
    enc2="$(url_encode "$enc1")"
    post_data="userId=$(url_encode "$USER_ID")&password=$(url_encode "$PASSWORD")&service=$(url_encode "$SERVICE")&queryString=$enc2&operatorPwd=&operatorUserId=&validcode=&passwordEncrypt=false"

    echo "[campus_portal_eth1] Logging in..."
    login_json="$(curl $CURL_OPTS -X POST "$LOGIN_API" \
        -H "User-Agent: $CURL_UA" \
        -H "Referer: $INDEX_URL" \
        -H "Accept: */*" \
        -H "Accept-Encoding: gzip, deflate" \
        -H "Accept-Language: zh-CN,zh;q=0.9,en;q=0.8" \
        -H "Content-Type: application/x-www-form-urlencoded; charset=UTF-8" \
        --data "$post_data" || true)"

    echo "$login_json" | grep -q '"result":"success"' && {
        echo "[campus_portal_eth1] Login success."
        return 0
    }
    echo "[campus_portal_eth1] Login failed."
    echo "$login_json" | sed -n 's/.*"message":"\([^"]*\)".*/[campus_portal_eth1] reason: \1/p'
    return 2
}

# procd 管理的常驻监控
_monitor_loop() {
    # 注意：这里不要用 'EOF'，让变量在这里就展开
    cat <<EOF
MONITOR_INTERVAL=${MONITOR_INTERVAL}
RETRY_MAX=${RETRY_MAX}
RETRY_SLEEP=${RETRY_SLEEP}
PING_URL="${PING_URL}"
CURL_UA="${CURL_UA}"

is_online() {
    resp="\$(curl -fsS --connect-timeout 5 --max-time 10 --interface $INTERFACE "\$PING_URL" -A "$CURL_UA" || true)"
    echo "\$resp" | grep -q "http://www\\.baidu\\.com/"
}

while true; do
    if is_online; then
        sleep "\$MONITOR_INTERVAL"
        continue
    fi

    echo "[campus_portal_eth1] offline detected, trying to login..."
    tries=1
    rc=1
    while [ "\$tries" -le "\$RETRY_MAX" ]; do
        # 请注意，一定要保证这里的文件名和你的文件名一致。
        /etc/init.d/campus_portal_eth1 login && { rc=0; break; }
        sleep "\$RETRY_SLEEP"
        tries=\$((tries+1))
    done
    if [ "\$rc" -ne 0 ]; then
        echo "[campus_portal_eth1] auto-login failed after \$RETRY_MAX tries."
    fi
    sleep "\$MONITOR_INTERVAL"
done
EOF
}

start_service() {
    procd_open_instance
    # 用 /bin/sh -c 启一个 loop，当作守护进程
    procd_set_param command /bin/sh -c "$(_monitor_loop)"
    # 异常退出自动重启（首次延迟 5s，之后指数回退 60s 封顶）
    procd_set_param respawn 5 1 60
    procd_set_param stdout 1
    procd_set_param stderr 1
    # 可按需添加：当网络配置变更时重启该实例
    procd_close_instance
}

stop_service() {
    # procd 会杀掉守护进程
    return 0
}

reload_service() { stop; start; }

service_triggers() {
    # 当 network / wireless 配置被 commit 时触发
    procd_add_reload_trigger "network" "wireless"
}

# rc.common 默认 boot() 会调用 start()
# 我们保留一次性手动命令
login() { do_login_once; }
sstatus() { if is_online; then echo "[campus_portal_eth1] Online"; else echo "[campus_portal_eth1] Offline"; fi; }

# 开机启动
/etc/init.d/campus_portal_eth1 enable
# 启动
/etc/init.d/campus_portal_eth1 start