fix(tlv): handle underruns and overruns during writes

wowserhq · Nov 24, 2023 · 846b2a1 · 846b2a1
1 parent 24d6f80
commit 846b2a1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/lib/type/TlvIo.mts b/src/lib/type/TlvIo.mts
@@ -87,14 +87,21 @@ class TlvIo implements IoType {
     context.root = context.root ?? null;
 
     this.#tagType.write(stream, value.tag, context);
+
     this.#lengthType.write(stream, value.length, context);
+    const expectedEndOffset = stream.offset + value.length;
 
     const valueType = this.#valueCallback(value.tag, value.length);
     if (valueType && valueType.write) {
       valueType.write(stream, value.value, context);
     } else {
       stream.writeBytes(value.value);
     }
+
+    // Account for underruns and overruns
+    if (stream.offset !== expectedEndOffset) {
+      stream.offset = expectedEndOffset;
+    }
   }
 }