From 2af9f65d1168c2755cbd16a3e691bd656048c902 Mon Sep 17 00:00:00 2001
From: Arne Welzel <arne.welzel@corelight.com>
Date: Thu, 12 Oct 2023 11:15:09 +0200
Subject: [PATCH] quic: Avoid bytes for VariableLengthInteger

Allocation of bytes objects due to parsing and usage of pack and the
invocation of to_uint() showed significantly in profiles (3.3% sample
matches). Switch to a more procedural approach to avoid the allocation
overhead.
---
 analyzer/QUIC.spicy | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

diff --git a/analyzer/QUIC.spicy b/analyzer/QUIC.spicy
index e7276f2..56dc0c9 100644
--- a/analyzer/QUIC.spicy
+++ b/analyzer/QUIC.spicy
@@ -140,17 +140,13 @@ type VariableLengthInteger = unit {
   #
   # https://datatracker.ietf.org/doc/rfc9000/
   # Section 16 and Appendix A
-  #
-  first_byte: bytes &size=1 {
-    local uint8_val = uint8($$.to_uint(spicy::ByteOrder::Big));
-    self.bytes_to_parse = 2**((0xC0 & uint8_val) >> 6);
-    # Re-pack without most significant two bits for later use.
-    self.first_byte = pack(0x3f & uint8_val, spicy::ByteOrder::Big);
+  : uint8 {
+    self.bytes_to_parse = 2**((0xC0 & $$) >> 6);
+    self.result = $$ & 0x3F;
   }
-  remaining_bytes: bytes &size=self.bytes_to_parse  - 1;
 
-  on %done {
-    self.result = (self.first_byte + self.remaining_bytes).to_uint(spicy::ByteOrder::Big);
+  : uint8[self.bytes_to_parse - 1] if (self.bytes_to_parse > 1) foreach {
+    self.result = (self.result << 8) | $$;
   }
 };