Returning offset for multiple object ASN.1 stream with separated records

wbond · Feb 3, 2020 · b6c592f · b6c592f
1 parent c09b87c
commit b6c592f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/asn1crypto/core.py b/asn1crypto/core.py
@@ -205,7 +205,7 @@ class Asn1Value(object):
     _native = None
 
     @classmethod
-    def load(cls, encoded_data, strict=False, **kwargs):
+    def load(cls, encoded_data, strict=False, return_offset=False, **kwargs):
         """
         Loads a BER/DER-encoded byte string using the current class as the spec
 
@@ -227,8 +227,12 @@ def load(cls, encoded_data, strict=False, **kwargs):
         if cls.tag is not None:
             spec = cls
 
-        value, _ = _parse_build(encoded_data, spec=spec, spec_params=kwargs, strict=strict)
-        return value
+        if return_offset:
+            value, new_pointer = _parse_build(encoded_data, spec=spec, spec_params=kwargs, strict=strict)
+            return value, new_pointer
+        else:
+            value, _ = _parse_build(encoded_data, spec=spec, spec_params=kwargs, strict=strict)
+            return value
 
     def __init__(self, explicit=None, implicit=None, no_explicit=False, tag_type=None, class_=None, tag=None,
                  optional=None, default=None, contents=None, method=None):