From 03aed078022f128bfa1362532182ea65aaec7d44 Mon Sep 17 00:00:00 2001 From: Bob Halley Date: Thu, 2 Jul 2020 12:02:29 -0700 Subject: add get_counted_bytes() --- dns/_binary.py | 4 ++++ 1 file changed, 4 insertions(+) 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()) -- cgit v1.2.1