From d871c050f72bff5e155dca32590e73890985ec9b Mon Sep 17 00:00:00 2001
From: sushen339 <sushen339@gmail.com>
Date: Tue, 18 Nov 2025 04:25:33 +0800
Subject: [PATCH]

---
 block-ip.sh | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/block-ip.sh b/block-ip.sh
index 755cd99..367529b 100644
--- a/block-ip.sh
+++ b/block-ip.sh
@@ -162,10 +162,11 @@ ban_ip() {
         # 新IP保存后，趁API可用时为文件中没有国家信息的旧IP补充查询（最多补充3个）
         if [ "$SHOULD_QUERY" -eq 1 ] && command -v curl >/dev/null 2>&1; then
             if [ -f "$PERSIST_FILE" ] && [ -s "$PERSIST_FILE" ]; then
-                UPDATE_COUNT=0
-                TEMP_UPDATE="/tmp/block_ip_update_$$"
-                cp "$PERSIST_FILE" "$TEMP_UPDATE"
+                # 创建备份用于读取
+                PERSIST_BACKUP="${PERSIST_FILE}.reading_$$"
+                cp "$PERSIST_FILE" "$PERSIST_BACKUP"
                 
+                UPDATE_COUNT=0
                 while IFS= read -r line; do
                     [ "$UPDATE_COUNT" -ge 3 ] && break
                     # 只处理不含'|'的IPv4单IP行，且不是刚添加的IP
@@ -173,18 +174,18 @@ ban_ip() {
                         OLD_IP="$line"
                         OLD_COUNTRY=$(curl -s --max-time 2 "https://ipinfo.io/$OLD_IP/country" 2>/dev/null | tr -d '\n\r ')
                         if [ -n "$OLD_COUNTRY" ] && [ ${#OLD_COUNTRY} -eq 2 ]; then
-                            # 使用awk替换整行，避免sed正则表达式问题
-                            awk -v ip="$OLD_IP" -v country="$OLD_COUNTRY" \
-                                'BEGIN{FS=OFS="|"} $1==ip && NF==1 {print ip,country; next} {print}' \
-                                "$TEMP_UPDATE" > "${TEMP_UPDATE}.new" && mv "${TEMP_UPDATE}.new" "$TEMP_UPDATE"
+                            # 使用awk替换：匹配整行等于OLD_IP的行，替换为IP|COUNTRY
+                            awk -v target="$OLD_IP" -v country="$OLD_COUNTRY" '
+                                $0 == target { print target "|" country; next }
+                                { print }
+                            ' "$PERSIST_FILE" > "${PERSIST_FILE}.tmp" && mv "${PERSIST_FILE}.tmp" "$PERSIST_FILE"
                             UPDATE_COUNT=$((UPDATE_COUNT + 1))
                             log "[补充地区] IP=$OLD_IP 国家=$(get_country_name "$OLD_COUNTRY")"
                         fi
                     fi
-                done < "$PERSIST_FILE"
+                done < "$PERSIST_BACKUP"
                 
-                # 如果有更新则覆盖原文件
-                [ "$UPDATE_COUNT" -gt 0 ] && mv "$TEMP_UPDATE" "$PERSIST_FILE" || rm -f "$TEMP_UPDATE"
+                rm -f "$PERSIST_BACKUP"
             fi
         fi
     elif [ "$SAVE_DISK" -ne 2 ]; then