Issue 43: HTTPS reads after RD end

paulc · Oct 17, 2022 · becc233 · becc233
1 parent d1d1f38
commit becc233
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dnslib/dns.py b/dnslib/dns.py
@@ -1822,6 +1822,7 @@ def __init__(self, priority, target, params):
     @classmethod
     def parse(cls,buffer,length):
         try:
+            end = buffer.offset + length
             priority, = buffer.unpack("!H")
             target = []
             while True:
@@ -1831,7 +1832,7 @@ def parse(cls,buffer,length):
                 seg = bytearray(buffer.get(n))
                 target.append(seg)
             params = []
-            while buffer.remaining() > 0:
+            while buffer.offset < end:
                 k, = buffer.unpack("!H")
                 n, = buffer.unpack("!H")
                 v = bytearray(buffer.get(n))