diff --git a/frpc.sh b/frpc.sh index 169f991..57689ec 100644 --- a/frpc.sh +++ b/frpc.sh @@ -1,9 +1,21 @@ #!/bin/bash -# FRP Client 一键安装脚本 -# 自动解压、配置和安装 frpc -# 使用方法: chmod +x frpc.sh && ./frpc.sh -# curl -sS -O https://gitea.tohub.top/Share/script/raw/branch/main/frpc.sh && chmod +x frpc.sh && ./frpc.sh +# FRP Client 自动安装/卸载脚本 +# 支持自动检测架构(amd64/arm64)并下载对应安装包 +# +# 使用方法: +# chmod +x frpc.sh && ./frpc.sh # 交互式主菜单 +# ./frpc.sh manage # 进入服务管理界面 +# +# 快速安装: +# curl -sS -O https://gitea.tohub.top/Share/script/raw/branch/main/frpc.sh && chmod +x frpc.sh && ./frpc.sh +# +# 功能说明: +# 1. 自动检测系统架构(amd64/arm64) +# 2. 自动下载对应架构的安装包 +# 3. 验证安装包完整性 +# 4. 安装并配置 frpc 服务 +# 5. 提供完整的服务管理功能 # 彩色输出 RED='\033[0;31m' @@ -15,7 +27,7 @@ NC='\033[0m' # No Color # 配置变量 INSTALL_DIR="/root/data/docker_data/frpc" -BASE_URL="https://gitea.tohub.top/Share/script/raw/branch/main" +BASE_URL="https://gitea.tohub.top/Share/script/raw/branch/main/frpc" FRPC_VERSION="0.65.0" @@ -23,39 +35,158 @@ set -e # 日志函数 +print_info() { + echo -e "${BLUE}[INFO]${NC} $1" +} + +print_success() { + echo -e "${GREEN}[SUCCESS]${NC} $1" +} + +print_warning() { + echo -e "${YELLOW}[WARNING]${NC} $1" +} + +print_error() { + echo -e "${RED}[ERROR]${NC} $1" +} + +# 兼容旧函数名 log_info() { - echo -e "${GREEN}[INFO]${NC} $1" + print_info "$1" } log_warn() { - echo -e "${YELLOW}[WARN]${NC} $1" + print_warning "$1" } log_error() { - echo -e "${RED}[ERROR]${NC} $1" + print_error "$1" } # 检查是否为root用户 check_root() { if [[ $EUID -ne 0 ]]; then - log_error "此脚本需要root权限运行" + print_error "此脚本需要root权限运行" + print_info "请使用: sudo $0" exit 1 fi } -# 检查文件是否存在 -check_files() { - if [[ ! -f "frp_0.64.0_linux_amd64.tar.gz" ]]; then - log_error "找不到 frp_0.64.0_linux_amd64.tar.gz 文件" +# 检测系统架构 +detect_arch() { + local arch=$(uname -m) + case "$arch" in + x86_64) + echo "amd64" + ;; + aarch64) + echo "arm64" + ;; + *) + print_error "不支持的系统架构: $arch" + print_info "仅支持 x86_64(amd64) 和 aarch64(arm64)" + exit 1 + ;; + esac +} + +# 创建安装目录 +create_install_dir() { + if [[ ! -d "$INSTALL_DIR" ]]; then + print_info "创建安装目录: $INSTALL_DIR" + mkdir -p "$INSTALL_DIR" + if [[ $? -ne 0 ]]; then + print_error "创建目录失败" + exit 1 + fi + fi +} + +# 下载文件 +download_file() { + local url=$1 + local output=$2 + + print_info "正在下载: $url" + + # 如果存在旧的不完整文件,先删除 + if [[ -f "$output" ]]; then + rm -f "$output" + fi + + if command -v wget &> /dev/null; then + wget -q --show-progress "$url" -O "$output" + elif command -v curl &> /dev/null; then + curl -# -L "$url" -o "$output" + else + print_error "未找到 wget 或 curl,请先安装" exit 1 fi - - if [[ ! -f "frpc.toml" ]]; then - log_error "找不到 frpc.toml 配置文件" - exit 1 + + # 检查下载是否成功 + if [[ $? -eq 0 ]] && [[ -f "$output" ]] && [[ -s "$output" ]]; then + print_success "下载完成: $output" + return 0 + else + print_error "下载失败: $url" + rm -f "$output" + return 1 fi } +# 验证tar.gz文件是否有效 +verify_tar_file() { + local file=$1 + + if [[ ! -f "$file" ]]; then + return 1 + fi + + if [[ ! -s "$file" ]]; then + print_warning "文件为空: $file" + return 1 + fi + + # 尝试列出tar文件内容来验证完整性 + if ! tar -tzf "$file" >/dev/null 2>&1; then + print_warning "文件不是有效的 tar.gz 包: $file" + return 1 + fi + + return 0 +} + +# 下载安装包 +download_packages() { + local arch=$(detect_arch) + print_info "检测到系统架构: $arch" + + create_install_dir + + local frp_file="frp_${FRPC_VERSION}_linux_${arch}.tar.gz" + local download_failed=0 + + # 下载 FRP + if [[ ! -f "$INSTALL_DIR/$frp_file" ]]; then + if ! download_file "$BASE_URL/$frp_file" "$INSTALL_DIR/$frp_file"; then + download_failed=1 + fi + else + print_warning "文件已存在,跳过下载: $frp_file" + fi + + # 下载配置文件(如果不存在) + if [[ ! -f "$INSTALL_DIR/frpc.toml" ]]; then + print_info "下载配置文件模板..." + if ! download_file "$BASE_URL/frpc.toml" "$INSTALL_DIR/frpc.toml"; then + print_warning "配置文件下载失败,需要手动创建" + fi + fi + + return $download_failed +} + # 创建frp用户 create_frp_user() { if ! id "frp" &>/dev/null; then @@ -68,49 +199,75 @@ create_frp_user() { # 解压安装包 extract_archive() { - log_info "解压 frp_0.64.0_linux_amd64.tar.gz..." - + local arch=$(detect_arch) + local frp_file="$INSTALL_DIR/frp_${FRPC_VERSION}_linux_${arch}.tar.gz" + + # 检查文件是否存在 + if [[ ! -f "$frp_file" ]]; then + print_warning "安装包不存在,开始下载..." + if ! download_packages; then + print_error "下载失败,无法继续安装" + return 1 + fi + fi + + # 验证文件完整性 + print_info "验证安装包完整性..." + if ! verify_tar_file "$frp_file"; then + print_error "安装包验证失败,请重新下载" + rm -f "$frp_file" + return 1 + fi + print_success "安装包验证通过" + + print_info "解压 frp_${FRPC_VERSION}_linux_${arch}.tar.gz..." + # 创建临时目录 TEMP_DIR=$(mktemp -d) - + # 解压到临时目录 - tar -xzf frp_0.64.0_linux_amd64.tar.gz -C "$TEMP_DIR" - - # 创建安装目录 - mkdir -p /root/frpc - - # 复制文件 - cp "$TEMP_DIR"/frp_0.64.0_linux_amd64/frpc /root/frpc/ - + tar -xzf "$frp_file" -C "$TEMP_DIR" + + # 复制文件到安装目录 + cp "$TEMP_DIR"/frp_${FRPC_VERSION}_linux_${arch}/frpc "$INSTALL_DIR/" + # 清理临时目录 rm -rf "$TEMP_DIR" - - # 设置权限 - root目录需要特殊权限处理 - chmod 755 /root # 确保frp用户可以访问root目录 - chown -R root:frp /root/frpc - chmod 750 /root/frpc - chmod +x /root/frpc/frpc - - log_info "frpc 已安装到 /root/frpc/" + + # 设置权限 + chown -R root:frp "$INSTALL_DIR" + chmod 750 "$INSTALL_DIR" + chmod +x "$INSTALL_DIR/frpc" + + print_success "frpc 已安装到 $INSTALL_DIR/" } # 配置frpc configure_frpc() { - log_info "配置 frpc..." - - # 复制配置文件 - cp frpc.toml /root/frpc/ - chown root:frp /root/frpc/frpc.toml - chmod 640 /root/frpc/frpc.toml - - log_info "配置文件已复制到 /root/frpc/frpc.toml" + print_info "配置 frpc..." + + # 检查配置文件是否已存在 + if [[ -f "$INSTALL_DIR/frpc.toml" ]]; then + print_success "配置文件已存在: $INSTALL_DIR/frpc.toml" + elif [[ -f "frpc.toml" ]]; then + # 如果当前目录有配置文件,复制它 + cp frpc.toml "$INSTALL_DIR/" + print_success "配置文件已复制到 $INSTALL_DIR/frpc.toml" + else + print_error "找不到配置文件 frpc.toml" + print_info "请手动创建配置文件: $INSTALL_DIR/frpc.toml" + return 1 + fi + + chown root:frp "$INSTALL_DIR/frpc.toml" + chmod 640 "$INSTALL_DIR/frpc.toml" } # 创建systemd服务 create_service() { - log_info "创建 systemd 服务..." - - cat > /etc/systemd/system/frpc.service << 'EOF' + print_info "创建 systemd 服务..." + + cat > /etc/systemd/system/frpc.service << EOF [Unit] Description=Frp Client Service After=network.target @@ -120,8 +277,8 @@ Type=simple User=frp Restart=on-failure RestartSec=5s -ExecStart=/root/frpc/frpc -c /root/frpc/frpc.toml -ExecReload=/bin/kill -s HUP $MAINPID +ExecStart=${INSTALL_DIR}/frpc -c ${INSTALL_DIR}/frpc.toml +ExecReload=/bin/kill -s HUP \$MAINPID KillMode=mixed TimeoutStopSec=5s @@ -131,46 +288,172 @@ EOF # 重新加载systemd systemctl daemon-reload - - log_info "systemd 服务已创建" + + print_success "systemd 服务已创建" } # 启动服务 start_service() { - log_info "启动 frpc 服务..." - + print_info "启动 frpc 服务..." + # 启用开机自启 systemctl enable frpc - + # 启动服务 systemctl start frpc - + # 检查服务状态 if systemctl is-active --quiet frpc; then - log_info "frpc 服务启动成功" + print_success "frpc 服务启动成功" else - log_error "frpc 服务启动失败" - log_error "请检查日志: journalctl -u frpc -f" - exit 1 + print_error "frpc 服务启动失败" + print_error "请检查日志: journalctl -u frpc -f" + return 1 fi } # 显示状态信息 show_status() { - log_info "安装完成!" + echo "" + print_success "============================================" + print_success "FRP Client 安装完成!" + print_success "============================================" echo "" echo "服务状态:" systemctl status frpc --no-pager echo "" - echo "常用命令:" + print_info "常用命令:" echo " 启动服务: systemctl start frpc" echo " 停止服务: systemctl stop frpc" echo " 重启服务: systemctl restart frpc" echo " 查看状态: systemctl status frpc" echo " 查看日志: journalctl -u frpc -f" echo "" - echo "配置文件位置: /root/frpc/frpc.toml" - echo "程序位置: /root/frpc/frpc" + print_info "配置文件位置: $INSTALL_DIR/frpc.toml" + print_info "程序位置: $INSTALL_DIR/frpc" + print_success "============================================" + echo "" +} + +# 安装frpc主函数 +install_frpc() { + print_info "开始安装 FRP Client..." + + create_frp_user + extract_archive || return 1 + configure_frpc || return 1 + create_service + start_service || return 1 + show_status + + print_success "FRP Client 安装完成!" +} + +# 卸载frpc +uninstall_frpc() { + print_warning "即将卸载 FRP Client" + read -p "是否继续?[y/N] " -n 1 -r + echo + + if [[ ! $REPLY =~ ^[Yy]$ ]]; then + print_info "已取消卸载" + return + fi + + # 停止服务 + print_info "停止 frpc 服务..." + systemctl stop frpc 2>/dev/null || true + + # 禁用开机自启 + print_info "禁用开机自启..." + systemctl disable frpc 2>/dev/null || true + + # 删除服务文件 + print_info "删除服务文件..." + rm -f /etc/systemd/system/frpc.service + systemctl daemon-reload + + # 删除程序文件 + print_info "删除程序文件..." + rm -rf "$INSTALL_DIR" + + # 删除frp用户 + print_info "删除 frp 用户..." + userdel frp 2>/dev/null || true + + echo "" + print_success "FRP Client 已完全卸载" +} + +# 检查安装状态 +check_status() { + local arch=$(detect_arch) + local frp_file="$INSTALL_DIR/frp_${FRPC_VERSION}_linux_${arch}.tar.gz" + + echo "" + print_info "============================================" + print_info "检查 FRP Client 状态" + print_info "============================================" + + # 检查程序是否安装 + if [[ -f "$INSTALL_DIR/frpc" ]]; then + print_success "FRP Client 已安装" + print_info " 程序位置: $INSTALL_DIR/frpc" + + # 获取程序版本 + if [[ -x "$INSTALL_DIR/frpc" ]]; then + local version=$("$INSTALL_DIR/frpc" --version 2>/dev/null | head -n 1 || echo "未知") + print_info " 版本信息: $version" + fi + + # 检查服务状态 + if systemctl is-active --quiet frpc; then + print_success "frpc 服务运行中" + else + print_warning "frpc 服务未运行" + fi + + if systemctl is-enabled --quiet frpc; then + print_success "开机自启已启用" + else + print_warning "开机自启未启用" + fi + + # 检查配置文件 + if [[ -f "$INSTALL_DIR/frpc.toml" ]]; then + print_success "配置文件存在" + else + print_warning "配置文件不存在" + fi + else + print_warning "FRP Client 未安装" + fi + + # 检查frp用户 + if id "frp" &>/dev/null; then + print_success "frp 用户存在" + else + print_warning "frp 用户不存在" + fi + + # 检查安装包状态 + echo "" + print_info "安装包状态:" + if [[ -f "$frp_file" ]]; then + local size=$(du -h "$frp_file" 2>/dev/null | cut -f1) + print_info " FRP 压缩包: ${CYAN}已下载${NC} (${size})" + else + print_warning " FRP 压缩包: 未下载" + fi + + if [[ -f "$INSTALL_DIR/frpc.toml" ]]; then + print_info " 配置文件: ${CYAN}存在${NC}" + else + print_warning " 配置文件: 不存在" + fi + + print_info "============================================" + echo "" } # 交互式服务管理菜单 @@ -231,39 +514,29 @@ interactive_menu() { 6|config) echo "配置文件内容:" echo "----------------------------------------" - cat /root/frpc/frpc.toml + cat "$INSTALL_DIR/frpc.toml" echo "----------------------------------------" ;; 7|edit) - log_info "编辑配置文件..." + print_info "编辑配置文件..." if command -v nano >/dev/null 2>&1; then - nano /root/frpc/frpc.toml + nano "$INSTALL_DIR/frpc.toml" elif command -v vi >/dev/null 2>&1; then - vi /root/frpc/frpc.toml + vi "$INSTALL_DIR/frpc.toml" else - log_error "未找到文本编辑器,请手动编辑: /root/frpc/frpc.toml" + print_error "未找到文本编辑器,请手动编辑: $INSTALL_DIR/frpc.toml" fi - + read -p "是否重启服务使配置生效? [y/N]: " restart_choice if [[ $restart_choice =~ ^[Yy]$ ]]; then systemctl restart frpc - log_info "服务已重启" + print_info "服务已重启" fi ;; 8|uninstall) - read -p "确定要卸载 frpc 服务吗? [y/N]: " confirm - if [[ $confirm =~ ^[Yy]$ ]]; then - log_warn "正在卸载 frpc 服务..." - systemctl stop frpc 2>/dev/null || true - systemctl disable frpc 2>/dev/null || true - rm -f /etc/systemd/system/frpc.service - systemctl daemon-reload - rm -rf /root/frpc - userdel -r frp 2>/dev/null || true - log_info "frpc 服务已完全卸载" + uninstall_frpc + if [[ $? -eq 0 ]]; then exit 0 - else - log_info "取消卸载操作" fi ;; 0|exit|quit|q) @@ -279,37 +552,92 @@ interactive_menu() { done } +# 主菜单 +show_main_menu() { + clear + echo -e "${CYAN}" + echo "============================================" + echo " FRP Client 自动安装/卸载脚本" + echo "============================================" + echo -e "${NC}" + echo -e "${GREEN}1.${NC} 安装 FRP Client" + echo -e "${GREEN}2.${NC} 卸载 FRP Client" + echo -e "${GREEN}3.${NC} 检查状态" + echo -e "${GREEN}4.${NC} 下载安装包" + echo -e "${GREEN}5.${NC} 服务管理" + echo -e "${GREEN}0.${NC} 退出" + echo "" + echo -e "${CYAN}============================================${NC}" +} + # 主函数 main() { + check_root + # 检查是否为管理模式 if [[ "$1" == "manage" || "$1" == "menu" || "$1" == "-m" ]]; then interactive_menu return fi - - log_info "开始安装 FRP Client..." - - check_root - check_files - create_frp_user - extract_archive - configure_frpc - create_service - start_service - show_status - - echo "" - log_info "FRP Client 安装完成!" - - # 询问是否进入交互管理界面 - read -p "是否进入交互式管理界面? [y/N]: " enter_menu - if [[ $enter_menu =~ ^[Yy]$ ]]; then - interactive_menu + + # 如果没有参数,显示交互式主菜单 + if [[ $# -eq 0 ]]; then + while true; do + show_main_menu + read -p "请选择操作 [0-5]: " choice + + case $choice in + 1) + install_frpc + read -p "按任意键继续..." -n 1 + ;; + 2) + uninstall_frpc + read -p "按任意键继续..." -n 1 + ;; + 3) + check_status + read -p "按任意键继续..." -n 1 + ;; + 4) + print_info "开始下载安装包..." + if download_packages; then + print_success "下载完成!" + else + print_error "下载失败" + fi + read -p "按任意键继续..." -n 1 + ;; + 5) + interactive_menu + ;; + 0) + print_info "退出脚本" + exit 0 + ;; + *) + print_error "无效的选择,请重新输入" + sleep 2 + ;; + esac + done else + # 兼容旧的直接安装模式(如果有参数的话) + print_info "开始安装 FRP Client..." + install_frpc + echo "" - log_info "可使用以下命令管理服务:" - echo " ./install_frpc.sh manage # 进入交互管理界面" - echo " systemctl [start|stop|restart|status] frpc" + # 询问是否进入交互管理界面 + read -p "是否进入服务管理界面? [y/N]: " enter_menu + if [[ $enter_menu =~ ^[Yy]$ ]]; then + interactive_menu + else + echo "" + print_info "可使用以下命令:" + echo " $0 # 进入主菜单" + echo " $0 manage # 进入服务管理界面" + echo " systemctl [start|stop|restart|status] frpc" + fi fi }