From 274ecc0248a37909f641201e4dadd89e5a289e56 Mon Sep 17 00:00:00 2001
From: dmiller <dmiller@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Wed, 13 Nov 2024 19:15:13 +0000
Subject: [PATCH] simplify a loop

---
 nselib/dns.lua | 20 ++++++--------------
 1 file changed, 6 insertions(+), 14 deletions(-)

diff --git a/nselib/dns.lua b/nselib/dns.lua
index b4ba4f92c..d52e97a8c 100644
--- a/nselib/dns.lua
+++ b/nselib/dns.lua
@@ -106,21 +106,13 @@ local function sendPacketsUDP(data, host, port, timeout, cnt, multiple)
 
     local response
 
-    if ( multiple ) then
-      while(true) do
-        status, response = socket:receive()
-        if( not(status) ) then break end
-
-        local status, _, _, ip, _ = socket:get_info()
-        table.insert(responses, { data = response, peer = ip } )
-      end
-    else
+    repeat
       status, response = socket:receive()
-      if ( status ) then
-        local status, _, _, ip, _ = socket:get_info()
-        table.insert(responses, { data = response, peer = ip } )
-      end
-    end
+      if( not(status) ) then break end
+
+      local status, _, _, ip, _ = socket:get_info()
+      table.insert(responses, { data = response, peer = ip } )
+    until not multiple
 
     if (#responses>0) then
       socket:close()