From 0884bcaf1c873036ee444b7a4e7aeef1d902d89d Mon Sep 17 00:00:00 2001
From: anulax1225 <anulax1225@icloud.com>
Date: Mon, 4 Mar 2024 11:33:17 +0100
Subject: [PATCH] Cleaned File

---
 bakanet/src/bakanet/dns_lookup.cpp | 14 ++++++++------
 bakanet/src/bakanet/dns_lookup.h   |  1 +
 bakanet/src/bakanet/ip_address.cpp |  4 ++--
 bakanet/src/bakanet/ip_address.h   |  4 ++--
 4 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/bakanet/src/bakanet/dns_lookup.cpp b/bakanet/src/bakanet/dns_lookup.cpp
index c59e072..70a1303 100644
--- a/bakanet/src/bakanet/dns_lookup.cpp
+++ b/bakanet/src/bakanet/dns_lookup.cpp
@@ -14,22 +14,24 @@ namespace Bk::Net {
 	    hints.ai_family = ai_family; 
 	    hints.ai_socktype = SOCK_STREAM;
 	 
-	    if ((status = getaddrinfo(host_name.c_str(), NULL, &hints, &res)) != 0) {
+	    if ((status = getaddrinfo(host_name.c_str(), NULL, &hints, &res)) != 0) 
+		{
 	        output.push_back("");
 	        return output;
 	    }
 	 
-	    for(p = res;p != NULL; p = p->ai_next) {
+	    for(p = res;p != NULL; p = p->ai_next) 
+		{
 	        void *addr;
-	        if (p->ai_family == AF_INET) { // IPv4
+	        if (p->ai_family == AF_INET) 
+			{
 	            struct sockaddr_in *ipv4 = (struct sockaddr_in *)p->ai_addr;
 	            addr = &(ipv4->sin_addr);
-	        } else { // IPv6
+	        } else 
+			{
 	            struct sockaddr_in6 *ipv6 = (struct sockaddr_in6 *)p->ai_addr;
 	            addr = &(ipv6->sin6_addr);
 	        }
-	 
-	        // convert the IP to a string
 	        inet_ntop(p->ai_family, addr, ip_address, sizeof ip_address);
 	        output.push_back(ip_address);
 	    }
diff --git a/bakanet/src/bakanet/dns_lookup.h b/bakanet/src/bakanet/dns_lookup.h
index e93d2cb..5f487c9 100644
--- a/bakanet/src/bakanet/dns_lookup.h
+++ b/bakanet/src/bakanet/dns_lookup.h
@@ -8,6 +8,7 @@
 #include <arpa/inet.h>
 
 #include "ip_version.h"
+#include "ip_address.h"
 
 namespace Bk::Net {
 	std::vector<std::string> dns_lookup(const std::string &host_name, IpVersion ipv);
diff --git a/bakanet/src/bakanet/ip_address.cpp b/bakanet/src/bakanet/ip_address.cpp
index cb17a9b..c7a78ee 100644
--- a/bakanet/src/bakanet/ip_address.cpp
+++ b/bakanet/src/bakanet/ip_address.cpp
@@ -2,8 +2,8 @@
 #include <string.h>
 
 namespace Bk::Net {
-    IpAddress::IpAddress(const char* ip)
-    : str(ip)
+    IpAddress::IpAddress(const char* ip, IpVersion ipv)
+    : str(ip), version(ipv)
     {
         if (inet_pton(AF_INET, str, &bytes) <= 0) perror("Bad IP");
     }
diff --git a/bakanet/src/bakanet/ip_address.h b/bakanet/src/bakanet/ip_address.h
index c274ebb..9495ed0 100644
--- a/bakanet/src/bakanet/ip_address.h
+++ b/bakanet/src/bakanet/ip_address.h
@@ -12,12 +12,12 @@ namespace Bk::Net {
 	class IpAddress 
 	{
 	public:
-		IpAddress(const char* ip);
+		IpAddress(const char* ip, IpVersion ipv = IpVersion::IPv4);
 
 		static void from_dns(char* dns);
 		
 		const char* str;
 		in_addr bytes;
-		IpVersion version = IpVersion::IPv4;
+		IpVersion version;
 	};
 }
\ No newline at end of file