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 )