net.adns: Add support for cancelling a non-blocking lookup, optionally calling the handler

1 files changed, 8 insertions, 1 deletions

diff --git a/net/adns.lua b/net/adns.lua
index 07d4fe54..6062da73 100644
--- a/net/adns.lua
+++ b/net/adns.lua
@@ -16,7 +16,7 @@ function lookup(handler, qname, qtype, qclass)
 				end
 				log("debug", "Records for %s not in cache, sending query (%s)...", qname, tostring(coroutine.running()));
 				dns.query(qname, qtype, qclass);
-				coroutine.yield(nil); -- Wait for reply
+				coroutine.yield({ qclass or "IN", qtype or "A", qname, coroutine.running()}); -- Wait for reply
 				log("debug", "Reply for %s (%s)", qname, tostring(coroutine.running()));
 				local ok, err = pcall(handler, dns.peek(qname, qtype, qclass));
 				if not ok then
@@ -25,6 +25,13 @@ function lookup(handler, qname, qtype, qclass)
 			end)(dns.peek(qname, qtype, qclass));
 end
 
+function cancel(handle, call_handler)
+	dns.cancel(handle);
+	if call_handler then
+		handle[4]()
+	end
+end
+
 function new_async_socket(sock)
 	local newconn = {};
 	local listener = {};