#!/bin/bash
clear
m_s="https://file.lihanzhang.cn/frp/"
echo -e "\e[32mWelcome to use Penetration tools\e[0m"
echo '1.linux amd64'
echo '2.linux arm'
echo '3.linux arm64'
echo ''

read -p 'Chose your version(1, 2, 3): ' version
read -p 'Client or Server?(c, s): ' version2
read -p 'Whether to enable Authentication(y/n): ' auth
if [[ "$auth" = y ]]; then
    read -sp "Enter token: " token
    echo ''
fi
echo ''
echo -e "\e[32mDownloading......\e[0m"

if [ ! -d /frp ]; then
    mkdir /frp
fi
if [ "$version2" == s ]; then
    if [ "$version" == 1 ]; then
        curl "${m_s}linux_amd64_s.tar.gz" -o /frp/frps.tar.gz
    fi
    if [ "$version" == 2 ]; then
        curl "${m_s}linux_arm_s.tar.gz" -o /frp/frps.tar.gz
    fi
    if [ "$version" == 3 ]; then
        curl "${m_s}linux_arm64_s.tar.gz" -o /frp/frps.tar.gz
    fi
    echo 'bindPort = 7000' >/frp/frps.toml
    if [[ "$auth" = y ]]; then
        sed -i "1a auth.token = \"${token}\"" /frp/frps.toml
    fi
    echo ''
    echo -e "\e[32mDecompressing......\e[0m"
    tar -xzvf /frp/frps.tar.gz -C /frp
    rm -rf /frp/frps.tar.gz
fi

if [ "$version2" == c ]; then
    if [ "$version" == 1 ]; then
        curl "${m_s}linux_amd64_c.tar.gz" -o /frp/frpc.tar.gz
    fi
    if [ "$version" == 2 ]; then
        curl "${m_s}linux_arm_c.tar.gz" -o /frp/frpc.tar.gz
    fi
    if [ "$version" == 3 ]; then
        curl "${m_s}linux_arm64_c.tar.gz" -o /frp/frpc.tar.gz
    fi
    echo ''
    echo -e "\e[32mDecompressing......\e[0m"
    tar -xzvf /frp/frpc.tar.gz -C /frp
    rm -rf /frp/frpc.tar.gz
fi

if [ "$version2" == c ]; then
    echo ''
    echo -e "\e[32mConfiguring......\e[0m"
    read -p 'IP or Domain: ' domain
    read -p 'Local   port: ' local_port
    read -p 'Remote  port: ' remote_port
    echo "serveraddr = \"${domain}\"" >/frp/frpc.toml
    sed -i "1a serverport = 7000" /frp/frpc.toml
    sed -i "2a [[proxies]]" /frp/frpc.toml
    sed -i "3a name = \"proxies\"" /frp/frpc.toml
    sed -i "4a type = \"tcp\"" /frp/frpc.toml
    sed -i "5a localip = \"127.0.0.1\"" /frp/frpc.toml
    sed -i "6a localport = ${local_port}" /frp/frpc.toml
    sed -i "7a remoteport = ${remote_port}" /frp/frpc.toml
    if [[ "$auth" = y ]]; then
        sed -i "2a auth.token = \"${token}\"" /frp/frpc.toml
    fi
fi
echo ''