From be37082b87d18dc3ebac0d37329d3881328ed2c3 Mon Sep 17 00:00:00 2001
From: PoliEcho <PoliEcho@pupes.org>
Date: Fri, 1 Aug 2025 16:16:00 +0200
Subject: [PATCH] a

---
 src/client/main.rs | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/src/client/main.rs b/src/client/main.rs
index 5f1d522..103e481 100644
--- a/src/client/main.rs
+++ b/src/client/main.rs
@@ -90,6 +90,11 @@ fn main() -> std::io::Result<()> {
                 Ok(s) => s,
                 Err(e) => return Err(ServerErrorResponses::into_io_error(e)),
             };
+        println!(
+            "{} my bublic sockaddr: {}",
+            "[LOG]".blue(),
+            public_sock_addr_raw
+        );
 
         let mut salt: [u8; BLOCK_SIZE] = [0u8; BLOCK_SIZE];
         let mut iv: [u8; BLOCK_SIZE] = [0u8; BLOCK_SIZE];
@@ -195,6 +200,7 @@ fn main() -> std::io::Result<()> {
         let encrypted = network_write_lock.encrypted;
         let key = network_write_lock.key;
         network_write_lock.peers.iter_mut().for_each(|peer| {
+            let mut retry_count: usize = 0;
             loop {
                 match net::P2P_query(&mut buf, &peer.sock_addr, &socket, encrypted, key) {
                     Ok(ip) => {
@@ -202,12 +208,18 @@ fn main() -> std::io::Result<()> {
                         peer.private_ip = ip;
                         break;
                     }
-                    Err(e) => eprintln!(
-                        "{} while getting ip from peer: {}, Error: {}, Retrying!",
-                        "[ERROR]".red(),
-                        peer.sock_addr,
-                        e
-                    ),
+                    Err(e) => {
+                        eprintln!(
+                            "{} while getting ip from peer: {}, Error: {}, Retrying!",
+                            "[ERROR]".red(),
+                            peer.sock_addr,
+                            e
+                        );
+                        retry_count += 1
+                    }
+                }
+                if retry_count >= STANDARD_RETRY_MAX {
+                    break;
                 }
             }
         });