From 5708ab26c0de8fc1be3e96a0f3f092da0938169e Mon Sep 17 00:00:00 2001
From: Marvin Borner
Date: Fri, 20 Nov 2020 15:58:32 +0100
Subject: Fixed some network race conditions

---
 libnet/dns.c | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

(limited to 'libnet/dns.c')

diff --git a/libnet/dns.c b/libnet/dns.c
index fc6b06f..6682831 100644
--- a/libnet/dns.c
+++ b/libnet/dns.c
@@ -59,8 +59,6 @@ static u32 dns_handle_packet(struct dns_packet *packet)
 u32 dns_request(const char *name, const char *tld)
 {
 	struct socket *socket = net_open(S_UDP);
-	/* if (socket) */
-	/* 	socket->src_port = 50053; */
 	if (!socket || !net_connect(socket, dns_ip_addr, 53))
 		return 0;
 
@@ -71,8 +69,9 @@ u32 dns_request(const char *name, const char *tld)
 	net_send(socket, packet, length);
 	free(packet);
 
-	u8 buf[128] = { 0 };
-	int l = net_receive(socket, buf, 128);
+	u8 buf[1024] = { 0 };
+	int l = net_receive(socket, buf, 1024);
+	net_close(socket);
 	if (l > 0)
 		return dns_handle_packet((void *)buf);
 	else
-- 
cgit v1.2.3