From 15631073f8dcd8a8fe39d7229d378ec7616b1fcd Mon Sep 17 00:00:00 2001
From: Jensfrank <everett7623@gmail.com>
Date: Tue, 2 Jul 2024 21:23:19 +0800
Subject: [PATCH] Update Nlbench_beta.sh

---
 Nlbench_beta.sh | 502 +++++++++++++++++++++---------------------------
 1 file changed, 223 insertions(+), 279 deletions(-)

diff --git a/Nlbench_beta.sh b/Nlbench_beta.sh
index ff07a49..a90b9cb 100644
--- a/Nlbench_beta.sh
+++ b/Nlbench_beta.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # 定义版本
-VERSION="1.0.0"
+VERSION="2.1.0"
 
 # 定义颜色
 RED='\033[0;31m'
@@ -10,30 +10,22 @@ YELLOW='\033[1;33m'
 NC='\033[0m' # No Color
 
 # 检查 root 权限并获取 sudo 权限
-if [ "$(id -u)" != "0" ]; then
-    echo "此脚本需要 root 权限运行。"
-    if ! sudo -v; then
-        echo "无法获取 sudo 权限，退出脚本。"
-        exit 1
+check_root() {
+    if [ "$(id -u)" != "0" ]; then
+        echo "此脚本需要 root 权限运行。"
+        if ! sudo -v; then
+            echo "无法获取 sudo 权限，退出脚本。"
+            exit 1
+        fi
+        echo "已获取 sudo 权限。"
     fi
-    echo "已获取 sudo 权限。"
-fi
+}
 
 # 检查并安装依赖
 install_dependencies() {
     echo -e "${YELLOW}正在检查并安装必要的依赖项...${NC}"
     
-    # 更新包列表
-    if ! sudo apt-get update; then
-        echo -e "${RED}无法更新包列表。请检查您的网络连接和系统设置。${NC}"
-        exit 1
-    fi
-    
-    # 安装依赖
-    local dependencies=(
-        "curl"
-        "wget"
-    )
+    local dependencies=("curl" "wget" "iperf3" "awk" "sed")
     
     for dep in "${dependencies[@]}"; do
         if ! command -v "$dep" &> /dev/null; then
@@ -47,7 +39,15 @@ install_dependencies() {
     done
     
     echo -e "${GREEN}依赖项检查和安装完成。${NC}"
-    clear
+}
+
+# 获取IP地址
+get_ip_address() {
+    ipv4_address=$(curl -s --max-time 5 ipv4.ip.sb)
+    [ -z "$ipv4_address" ] && ipv4_address=$(ip -4 addr show | grep -oP '(?<=inet\s)\d+(\.\d+){3}' | grep -v '127.0.0.1' | head -n1)
+
+    ipv6_address=$(curl -s --max-time 5 ipv6.ip.sb)
+    [ -z "$ipv6_address" ] && ipv6_address=$(ip -6 addr show | grep -oP '(?<=inet6\s)[\da-f:]+' | grep -v '^::1' | grep -v '^fe80' | head -n1)
 }
 
 # 检测VPS地理位置
@@ -84,10 +84,7 @@ sum_run_times() {
     fi
 }
 
-# 调用函数获取统计数据
-sum_run_times
-
-# 输出欢迎信息
+# 显示欢迎信息
 show_welcome() {
     echo ""
     echo -e "${RED}---------------------------------By'Jensfrank---------------------------------${NC}"
@@ -109,7 +106,7 @@ show_welcome() {
     echo ""
     echo -e "${RED}---------------------------------By'Jensfrank---------------------------------${NC}"
     echo ""
-    echo "一键脚本将测试以下项目："
+    echo "一键脚本将测试以下项目，可以自动全部测试，或者自定义选择测试项目："
     echo "1. Yabs"
     echo "2. 融合怪"
     echo "3. IP质量"
@@ -124,290 +121,237 @@ show_welcome() {
     clear
 }
 
-# 定义一个数组来存储每个命令的输出
-declare -a test_results
-
-# 在每个命令执行后保存结果
-run_and_capture() {
-    local command_output=$(eval "$1" 2>&1)
-    test_results+=("$command_output")
-    echo "$command_output"
+# Markdown 转义函数
+escape_markdown() {
+    sed 's/```/\\`\\`\\`/g' | sed 's/\[/\\[/g' | sed 's/\]/\\]/g' | sed 's/\*/\\*/g' | sed 's/_/\\_/g'
 }
 
-# 去除Yabs板块ANSI转义码
-yabs_process_output() {
-    local input="$1"
-    echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g'
+# 执行YABS测试
+run_yabs() {
+    echo "执行YABS测试..."
+    yabs_result=$(wget -qO- yabs.sh | bash)
+    echo "$yabs_result" > yabs_result.txt
 }
 
-# 去除融合怪板块ANSI转义码并截取
-fusion_process_output() {
-    local input="$1"
-    # 使用更全面的 sed 命令去除所有 ANSI 转义码
-    echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g' | awk '/A Bench Script/{f=1} f; /短链/{f=0}'
+# 执行融合怪测试
+run_fusion() {
+    echo "执行融合怪测试..."
+    fusion_result=$(curl -L https://gitlab.com/spiritysdx/za/-/raw/main/ecs.sh -o ecs.sh && chmod +x ecs.sh && bash ecs.sh)
+    fusion_result=$(echo "$fusion_result" | awk '/A Bench Script/{f=1} f; /短链/{f=0}')
+    echo "$fusion_result" > fusion_result.txt
 }
 
-# 去除IP质量板块ANSI转义码并截取
-ip_process_output() {
-    local input="$1"
-    local start_line=$(echo "$input" | grep -n '正在检测黑名单数据库' | tail -n 1 | cut -d ':' -f 1)
+# 执行IP质量测试
+run_ip_quality() {
+    echo "执行IP质量测试..."
+    ip_quality_result=$(bash <(curl -Ls IP.Check.Place))
+    
+    local start_line=$(echo "$ip_quality_result" | grep -n '正在检测黑名单数据库' | tail -n 1 | cut -d ':' -f 1)
     start_line=$((start_line + 1))  # 移动到下一行
-    local end_line=$(echo "$input" | grep -n '按回车键返回主菜单' | head -n 1 | cut -d ':' -f 1)
+    local end_line=$(echo "$ip_quality_result" | grep -n '按回车键返回主菜单' | head -n 1 | cut -d ':' -f 1)
     
     if [ -n "$start_line" ] && [ -n "$end_line" ]; then
-        tail -n +"$start_line" <<< "$input" | head -n $(($end_line - $start_line)) | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g'
+        ip_quality_result=$(tail -n +"$start_line" <<< "$ip_quality_result" | head -n $(($end_line - $start_line)) | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g')
     elif [ -n "$start_line" ]; then
-        tail -n +"$start_line" <<< "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g'
-    else
-        echo "未找到合适的日志内容。"
+        ip_quality_result=$(tail -n +"$start_line" <<< "$ip_quality_result" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g')
     fi
+    
+    echo "$ip_quality_result" > ip_quality_result.txt
 }
 
-# 去除流媒体板块ANSI转义码并截取
-streaming_process_output() {
-    local input="$1"
-    echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g' | awk '/项目地址/{f=1} f; /检测脚本当天运行次数/{f=0}'
+# 执行流媒体解锁测试
+run_streaming() {
+    echo "执行流媒体解锁测试..."
+    region=$(detect_region)
+    streaming_result=$(bash <(curl -L -s media.ispvps.com) $region)
+    streaming_result=$(echo "$streaming_result" | awk '/项目地址/{f=1} f; /检测脚本当天运行次数/{f=0}')
+    echo "$streaming_result" > streaming_result.txt
 }
 
-# 去除响应板块ANSI转义码
-response_process_output() {
-    local input="$1"
-    echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g'
+# 执行响应测试
+run_response() {
+    echo "执行响应测试..."
+    response_result=$(bash <(curl -sL https://nodebench.mereith.com/scripts/curltime.sh))
+    echo "$response_result" > response_result.txt
 }
 
-# 去除三网测速板块板块ANSI转义码并截取（多线程）
-speedtest_multi_process_output() {
-    local input="$1"
-    # Step 1: 去除 ANSI 转义码
-    local no_ansi
-    no_ansi=$(echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g')
-    echo "去除 ANSI 转义码后的输出: $no_ansi" >&2
-
-    # Step 2: 过滤掉包含 "测试进行中" 的行
-    local no_progress=$(echo "$no_ansi" | grep -v "^ *测试进行中")
-    echo "过滤掉包含 '测试进行中' 的行后的输出: $no_progress" >&2
-
-    # Step 3: 截取所需的测试结果
-    local speedtest_multi_process_output_result=$(echo "$no_progress" | awk '/大陆三网\+教育网 IPv4 多线程测速/{f=1} f; /北京时间/{f=0}')
-    echo "$speedtest_multi_process_output_result"
+# 执行三网测速（多线程）
+run_speedtest_multi() {
+    echo "执行三网测速（多线程）..."
+    speedtest_multi_result=$(bash <(curl -sL bash.icu/speedtest) 1)
+    speedtest_multi_result=$(echo "$speedtest_multi_result" | sed -e '1,/序号\:/d' -e '/测试进行中/d' -e '/^\s*$/d')
+    echo "$speedtest_multi_result" > speedtest_multi_result.txt
 }
 
-# 去除三网测速板块板块ANSI转义码并截取（单线程）
-speedtest_single_process_output() {
-    local input="$1"
-    # Step 1: 去除 ANSI 转义码
-    local no_ansi
-    no_ansi=$(echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g')
-    echo "去除 ANSI 转义码后的输出: $no_ansi" >&2
-
-    # Step 2: 过滤掉包含 "测试进行中" 的行
-    local no_progress
-    no_progress=$(echo "$no_ansi" | grep -v "^ *测试进行中")
-    echo "过滤掉包含 '测试进行中' 的行后的输出: $no_progress" >&2
-
-    # Step 3: 截取所需的测试结果
-    local speedtest_single_process_output_result=$(echo "$no_progress" | awk '/大陆三网\+教育网 IPv4 单线程测速/{f=1} f; /北京时间/{f=0}')
-    echo "$speedtest_single_process_output_result"
+# 执行三网测速（单线程）
+run_speedtest_single() {
+    echo "执行三网测速（单线程）..."
+    speedtest_single_result=$(bash <(curl -sL bash.icu/speedtest) 2)
+    speedtest_single_result=$(echo "$speedtest_single_result" | sed -e '1,/序号\:/d' -e '/测试进行中/d' -e '/^\s*$/d')
+    echo "$speedtest_single_result" > speedtest_single_result.txt
 }
 
-# 去除回程路由板块板块ANSI转义码并截取
-autotrace_process_output() {
-    local input="$1"
-    # 使用更全面的 sed 命令去除所有 ANSI 转义码
-    echo "$input" | sed -E 's/\x1b\[[0-9;]*[a-zA-Z]//g' | awk '/No:1/9 Traceroute/{f=1} f; /[信息] 已删除 Nexttrace 文件/{f=0}'
+# 执行回程路由测试
+run_traceroute() {
+    echo "执行回程路由测试..."
+    traceroute_result=$(wget -N --no-check-certificate https://raw.githubusercontent.com/Chennhaoo/Shell_Bash/master/AutoTrace.sh && chmod +x AutoTrace.sh && bash AutoTrace.sh)
+    traceroute_result=$(echo "$traceroute_result" | sed -e '/测试项/,+9d' -e '/信息/d' -e '/^\s*$/d')
+    echo "$traceroute_result" > traceroute_result.txt
 }
 
-# 根据编号执行特定测试
-perform_test() {
-    case $1 in
-        # YABS
-        1) echo -e "运行${YELLOW}YABS...${NC}"
-           yabs_result=$(run_and_capture "wget -qO- yabs.sh | bash")
-           ;;
-        # 融合怪
-        2) echo -e "运行${YELLOW}融合怪...${NC}"
-           fusion_result=$(run_and_capture "curl -L https://gitlab.com/spiritysdx/za/-/raw/main/ecs.sh -o ecs.sh && chmod +x ecs.sh && echo '1' | bash ecs.sh")
-           ;;
-        # IP质量
-        3) echo -e "运行${YELLOW}IP质量测试...${NC}"
-           ip_quality_result=$(run_and_capture "bash <(curl -Ls IP.Check.Place)")
-           ;;
-        # 流媒体解锁
-        4) echo -e "运行${YELLOW}流媒体解锁测试...${NC}"
-           local region=$(detect_region)
-           streaming_result=$(run_and_capture "echo '$region' | bash <(curl -L -s media.ispvps.com)")
-           ;;
-        # 响应测试
-        5) echo -e "运行${YELLOW}响应测试...${NC}"
-           response_result=$(run_and_capture "bash <(curl -sL https://nodebench.mereith.com/scripts/curltime.sh)")
-           ;;
-        # 多线程测试
-        6) echo -e "运行${YELLOW}多线程测试...${NC}"
-           speedtest_multi_result=$(run_and_capture "echo '1' | bash <(curl -sL bash.icu/speedtest)")
-           ;;
-        # 单线程测试
-        7) echo -e "运行${YELLOW}单线程测试...${NC}"
-           speedtest_single_result=$(run_and_capture "echo '2' | bash <(curl -sL bash.icu/speedtest)")
-           ;;
-        # 回程路由
-        8) echo -e "运行${YELLOW}回程路由...${NC}"
-           autotrace_result=$(run_and_capture "wget -N --no-check-certificate https://raw.githubusercontent.com/Chennhaoo/Shell_Bash/master/AutoTrace.sh && chmod +x AutoTrace.sh && echo '1' | bash AutoTrace.sh")
-           ;;
-        *)
-           echo -e "${RED}未知测试编号：$1${NC}"
-           ;;
-    esac
-}
+# 生成Markdown文件
+generate_markdown() {
+    local output_file="vps_test_results_$(date +%Y%m%d_%H%M%S).md"
+    {
+        echo "# VPS 测试结果"
+        echo "测试时间：$(date)"
+        echo ""
+        echo "## 系统信息"
+        echo "- IPv4: $ipv4_address"
+        echo "- IPv6: $ipv6_address"
+        echo ""
+        echo "[tabs]"
+        
+        echo "[tab=\"YABS\"]"
+        echo "\`\`\`"
+        cat yabs_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"融合怪\"]"
+        echo "\`\`\`"
+        cat fusion_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"IP质量\"]"
+        cat ip_quality_result.txt | escape_markdown
+        echo "[/tab]"
+        
+        echo "[tab=\"流媒体\"]"
+        echo "\`\`\`"
+        cat streaming_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"响应\"]"
+        echo "\`\`\`"
+        cat response_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"多线程测速\"]"
+        echo "\`\`\`"
+        cat speedtest_multi_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"单线程测速\"]"
+        echo "\`\`\`"
+        cat speedtest_single_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"回程路由\"]"
+        echo "\`\`\`"
+        cat traceroute_result.txt | escape_markdown
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"去程路由\"]"
+        echo "[/tab]"
+        
+        echo "[tab=\"iperf3\"]"
+        echo "\`\`\`"
+        echo "\`\`\`"
+        echo "[/tab]"
+        
+        echo "[tab=\"Ping.pe\"]"
+        echo "[/tab]"
+        
+        echo "[tab=\"哪吒 ICMP\"]"
+        echo "[/tab]"
+        
+        echo "[tab=\"其他\"]"
+        echo "[/tab]"
+        
+        echo "[/tabs]"
+    } > "$output_file"
 
-# 运行所有测试
-run_all_tests() {
-    echo -e "${RED}请选择测试模式：${NC}"
-    echo "1. 全部测试"
-    echo "2. 选择性测试（输入数字，用英文逗号分隔，如：1,2,3）"
-    echo "    [1] Yabs"
-    echo "    [2] 融合怪"
-    echo "    [3] IP质量"
-    echo "    [4] 流媒体解锁"
-    echo "    [5] 响应测试"
-    echo "    [6] 多线程测试"
-    echo "    [7] 单线程测试"
-    echo "    [8] 回程路由"
-    read -r test_choice
-
-    case "$test_choice" in
-        1)
-            echo -e "${RED}执行全部测试...${NC}"
-            echo -e "${RED}开始测试，测试时间较长，请耐心等待...${NC}"
-            perform_test 1
-            perform_test 2
-            perform_test 3
-            perform_test 4
-            perform_test 5
-            perform_test 6
-            perform_test 7
-            perform_test 8
-            format_results
-            ;;
-        2)
-            echo "请输入要测试的编号，用逗号分隔（如：1,2,3）："
-            read -r test_numbers
-            IFS=',' read -ra selected_tests <<< "$test_numbers"
-            echo -e "${RED}开始测试，测试时间较长，请耐心等待...${NC}"
-            for test_num in "${selected_tests[@]}"; do
-                perform_test "$test_num"
-            done
-
-            # 格式化结果
-            echo -e "${YELLOW}此报告由Nodeloc_VPS_自动脚本测试生成...${NC}"
-            format_results
-            ;;
-        *)
-            echo -e "${RED}无效的选项：$test_choice${NC}"
-            ;;
-    esac
-}
-
-# 格式化结果为 Markdown
-format_results() {
-
-# 处理yabs测试结果
-local processed_yabs_result=$(yabs_process_output "$yabs_result")
-
-# 处理融合怪结果
-local processed_fusion_result=$(fusion_process_output "$fusion_result")
-
-# 处理IP质量结果
-local processed_ip_result=$(ip_process_output "$ip_quality_result")
-
-# 处理流媒体解锁结果
-local processed_streaming_result=$(streaming_process_output "$streaming_result")
-
-# 处理响应测试结果
-local processed_response_result=$(response_process_output "$response_result")
-
-# 处理三网测速结果
-local processed_speedtest_multi_result=$(speedtest_multi_process_output "$speedtest_multi_result")
-local processed_speedtest_single_result=$(speedtest_single_process_output "$speedtest_single_result")
-
-# 处理回程路由结果
-local processed_autotrace_result=$(autotrace_process_output "$autotrace_result")
-
-# Tabs分栏输出结果，用于复制到Nodeloc论坛
-
-result="[tabs]
-[tab=\"YABS\"]
-\`\`\`
-$processed_yabs_result
-\`\`\`
-[/tab]
-[tab=\"融合怪\"]
-\`\`\`
-$processed_fusion_result
-\`\`\`
-[/tab]
-[tab=\"IP质量\"]
-\`\`\`
-########################################################################
-$processed_ip_result
-\`\`\`
-[/tab]
-[tab=\"流媒体\"]
-\`\`\`
-$processed_streaming_result
-\`\`\`
-[/tab]
-[tab=\"响应\"]
-\`\`\`
-$processed_response_result
-\`\`\`
-[/tab]
-[tab=\"多线程测速\"]
-\`\`\`
-$processed_speedtest_multi_result
-\`\`\`
-[/tab]
-[tab=\"单线程测速\"]
-\`\`\`
-$processed_speedtest_single_result
-\`\`\`
-[/tab]
-[tab=\"回程路由\"]
-\`\`\`
-$processed_autotrace_result
-\`\`\`
-[/tab]
-[tab=\"去程路由\"]
-
-[/tab]
-[tab=\"iperf3\"]
-\`\`\`
-
-\`\`\`
-[/tab]
-[tab=\"Ping.pe\"]
-
-[/tab]
-[tab=\"哪吒 ICMP\"]
-
-[/tab]
-[tab=\"其他\"]
-
-[/tab]
-[/tabs]"
-
-    echo "$result" > results.md
-    echo -e "${YELLOW}结果已保存到 results.md 文件中。${NC}"
+    echo "测试结果已保存到 $output_file"
 }
 
 # 主函数
 main() {
+    check_root
     install_dependencies
+    get_ip_address
+    sum_run_times
     show_welcome
-    run_all_tests
-    echo -e "${YELLOW}所有测试完成，可到results.md复制到Nodeloc使用。${NC}"
-    read -n 1 -s
-    echo "最终测试结果如下:" >&2
-    cat results.md >&2
+
+    while true; do
+        echo "请选择测试选项："
+        echo "1. 测试全部脚本"
+        echo "2. 选择特定脚本测试"
+        echo "0. 退出"
+        read -p "请输入选项 (0-2): " option
+
+        case $option in
+            1)
+                run_yabs
+                run_fusion
+                run_ip_quality
+                run_streaming
+                run_response
+                run_speedtest_multi
+                run_speedtest_single
+                run_traceroute
+                break
+                ;;
+            2)
+                echo "请输入要测试的脚本编号（用逗号分隔，如1,2,3）:"
+                echo "1. Yabs"
+                echo "2. 融合怪"
+                echo "3. IP质量"
+                echo "4. 流媒体解锁"
+                echo "5. 响应测试"
+                echo "6. 多线程测试"
+                echo "7. 单线程测试"
+                echo "8. 回程路由"
+                read -p "输入选择: " scripts
+                IFS=',' read -ra ADDR <<< "$scripts"
+                for i in "${ADDR[@]}"; do
+                    case $i in
+                        1) run_yabs ;;
+                        2) run_fusion ;;
+                        3) run_ip_quality ;;
+                        4) run_streaming ;;
+                        5) run_response ;;
+                        6) run_speedtest_multi ;;
+                        7) run_speedtest_single ;;
+                        8) run_traceroute ;;
+                        *) echo "无效的选项: $i" ;;
+                    esac
+                done
+                break
+                ;;
+            0)
+                echo "感谢使用，再见！"
+                exit 0
+                ;;
+            *)
+                echo "无效的选项，请重新输入。"
+                ;;
+        esac
+    done
+
+    generate_markdown
+    
+    # 清理临时文件
+    rm -f yabs_result.txt fusion_result.txt ip_quality_result.txt streaming_result.txt response_result.txt speedtest_multi_result.txt speedtest_single_result.txt traceroute_result.txt
+
+    echo "测试完成！结果已保存到 Markdown 文件中。"
 }
 
+# 执行主函数
 main