From 516430d2e51850c9c49e427ad52803f2f6baabbd Mon Sep 17 00:00:00 2001
From: darkclip <darkclip@users.noreply.github.com>
Date: Tue, 2 Jan 2024 05:27:49 +0800
Subject: [PATCH] update

---
 wg-watch.sh | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/wg-watch.sh b/wg-watch.sh
index 06f49bde..f9345ff0 100755
--- a/wg-watch.sh
+++ b/wg-watch.sh
@@ -92,21 +92,20 @@ generate_ips(){
 
     set --;
     num=0;
-    while true; do
+    while [ $num -lt $max ]; do
         ip=$($prog);
         unique=true;
         for item in $@; do
             if [ "$item" = "$ip" ]; then
                 unique=false;
+                break;
             fi;
         done;
-        if [ $unique = 'true' ]; then
-            set -- $@ $ip;
-            num=$(($num + 1));
-        fi;
-        if [ $num -ge $max ]; then
-            break;
+        if [ $unique != 'true' ]; then
+            continue;
         fi;
+        set -- $@ $ip;
+        num=$(($num + 1));
     done;
     echo $@;
 };
@@ -161,11 +160,13 @@ resolve_ip_addresses(){
             for item in $@; do
                 if [ "$item" = "$ip:$port" ]; then
                     unique=false;
+                    break;
                 fi;
             done;
-            if [ $unique = 'true' ]; then
-                set -- $@ "$ip:$port";
+            if [ $unique != 'true' ]; then
+                continue;
             fi;
+            set -- $@ "$ip:$port";
         done;
     done;
     echo $@;