summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBob Halley <halley@dnspython.org>2020-07-02 12:02:29 -0700
committerBob Halley <halley@dnspython.org>2020-07-02 12:02:45 -0700
commit03aed078022f128bfa1362532182ea65aaec7d44 (patch)
tree6fb8e18983e065c95fd62038be9f490dce17ea07
parent6bd9615accfbc9a9d6aa67294090fdba7f1aeee3 (diff)
downloaddnspython-03aed078022f128bfa1362532182ea65aaec7d44.tar.gz
add get_counted_bytes()
-rw-r--r--dns/_binary.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/dns/_binary.py b/dns/_binary.py
index b7c9959..f626db1 100644
--- a/dns/_binary.py
+++ b/dns/_binary.py
@@ -26,6 +26,10 @@ class Parser:
self.furthest = max(self.furthest, self.current)
return output
+ def get_counted_bytes(self, length_size=1):
+ length = int.from_bytes(self.get_bytes(length_size), 'big')
+ return self.get_bytes(length)
+
def get_remaining(self):
return self.get_bytes(self.remaining())