From 9f0ac830f2be531b0424b89351ab6374cecec9f8 Mon Sep 17 00:00:00 2001 From: nexy7574 Date: Wed, 18 Sep 2024 18:15:55 +0100 Subject: [PATCH] Catch more types of query errors --- app/modules/drill.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/modules/drill.py b/app/modules/drill.py index 0b7bc38..dd41cc4 100644 --- a/app/modules/drill.py +++ b/app/modules/drill.py @@ -10,6 +10,7 @@ import dns.asyncquery import dns.nameserver import dns.resolver import dns.message +from dns.quic._common import UnexpectedEOF from urllib.parse import urlparse @@ -116,7 +117,7 @@ class DNSCommand(niobot.Module): end_ts = time.perf_counter() if isinstance(response, dns.message.Message): response = (response, False) - except dns.exception.DNSException as err: + except (dns.exception.DNSException, UnexpectedEOF, httpx.HTTPError, ValueError) as err: return await m.edit( "There was an error while querying the DNS server:\n```\n%s\n```" % err )