From 62f81db5617c1a879031ad5aed9707e3931691e3 Mon Sep 17 00:00:00 2001
From: Jensfrank <everett7623@gmail.com>
Date: Sun, 21 Jul 2024 10:35:10 +0800
Subject: [PATCH] Update Nlbench.sh

---
 Nlbench.sh | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/Nlbench.sh b/Nlbench.sh
index dc76028..1d62879 100644
--- a/Nlbench.sh
+++ b/Nlbench.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 
 # 定义版本
-CURRENT_VERSION="2024-07-19 v1.0.6" # 最新版本号
+CURRENT_VERSION="2024-07-21 v1.0.7" # 最新版本号
 SCRIPT_URL="https://raw.githubusercontent.com/everett7623/nodeloc_vps_test/main/Nlbench.sh"
 VERSION_URL="https://raw.githubusercontent.com/everett7623/nodeloc_vps_test/main/version.sh"
 
@@ -266,10 +266,11 @@ TEST_DURATION=30
 run_iperf3_test() {
     echo -e "${GREEN}服务端VPS位于美国拉斯维加斯${NC}"
     echo -e "${GREEN}连接到服务端进行iperf3测试。。。${NC}"
-    if iperf3 -c $AUXILIARY_VPS -p $IPERF_PORT -t $TEST_DURATION; then
-    echo -e "${YELLOW}iperf3 测试完成${NC}"
+    timeout ${TEST_DURATION}s iperf3 -c $AUXILIARY_VPS -p $IPERF_PORT -t $TEST_DURATION
+    if [ $? -eq 0 ]; then
+        echo -e "${YELLOW}iperf3 测试完成${NC}"
     else
-    echo -e "${RED}iperf3 测试失败${NC}"
+        echo -e "${RED}iperf3 测试失败或超时${NC}"
     fi
 }