fix: update computation for payload length offset (#63)

anmonteiro · Aug 1, 2024 · f982c11 · f982c11
1 parent a3db363
commit f982c11
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/parse.ml b/lib/parse.ml
@@ -63,10 +63,10 @@ let payload_offset_of_bits bits =
   let initial_offset = 2 in
   let mask_offset    = (bits land (1 lsl 7)) lsr (7 - 2) in
   let length_offset  =
-    let length = bits land 0b01111111 in
-    if length < 126
-    then 0
-    else 2 lsl ((length land 0b1) lsl 2)
+    match bits land 0b01111111 with
+    | 127 -> 8
+    | 126 -> 2
+    | _   -> 0
   in
   initial_offset + mask_offset + length_offset
 ;;