Add packet parsing and error handling

yasasa · Sep 27, 2019 · 5fe9c76 · 5fe9c76
1 parent 914c03f
commit 5fe9c76
Show file tree

Hide file tree

Showing 7 changed files with 462 additions and 180 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,3 @@
 *.pyc
 __pycache__
+.vscode
diff --git a/dns.py b/dns.py
diff --git a/dns/__init__.py b/dns/__init__.py
@@ -0,0 +1,3 @@
+from dns.resolver import Resolver
+from dns.message import Query, Response, QUERY_TYPE_A, QUERY_TYPE_MX, QUERY_TYPE_NS
+from dns.errors import *
diff --git a/dns/errors.py b/dns/errors.py
@@ -0,0 +1,28 @@
+class DNSError(Exception):
+    def __init__(self, payload, message=""):
+        self.payload = payload
+        self.message = message
+
+
+class FormatError(DNSError):
+    pass
+
+
+class ServerError(DNSError):
+    pass
+
+
+class InvalidNameError(DNSError):
+    pass
+
+
+class ServerRefusedError(DNSError):
+    pass
+
+
+class ResponsePacketError(DNSError):
+    pass
+
+
+class ResponseTruncatedError(DNSError):
+    pass