From 81e7adb409199ac6594a18ca1bd88bfe89fbfc3e Mon Sep 17 00:00:00 2001
From: Luna <90072930+LunaWasFlaggedAgain@users.noreply.github.com>
Date: Sat, 11 Mar 2023 19:11:35 -0300
Subject: Fix infinite recursion in the DNS resolver (#82)

---
 azalea-protocol/src/resolver.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/azalea-protocol/src/resolver.rs b/azalea-protocol/src/resolver.rs
index d12654f2..f99cec05 100755
--- a/azalea-protocol/src/resolver.rs
+++ b/azalea-protocol/src/resolver.rs
@@ -50,6 +50,15 @@ pub async fn resolve_address(address: &ServerAddress) -> Result<SocketAddr, Reso
             port: redirect_srv.port(),
         };
 
+        if redirect_address.host == address.host {
+            let lookup_ip_result = resolver.lookup_ip(redirect_address.host).await;
+            let lookup_ip = lookup_ip_result.map_err(|_| ResolverError::NoIp)?;
+            return Ok(SocketAddr::new(
+                lookup_ip.iter().next().unwrap(),
+                redirect_address.port,
+            ))
+        }
+
         // debug!("redirecting to {:?}", redirect_address);
 
         return resolve_address(&redirect_address).await;
-- 
cgit v1.2.3