QUIC wire format support

Reviewed-by: Tim Hudson <tjh@openssl.org>
Reviewed-by: Tomas Mraz <tomas@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/18382)

1 files changed, 106 insertions, 0 deletions

diff --git a/test/packettest.c b/test/packettest.c
index e8aec47446..c62247f9be 100644
--- a/test/packettest.c
+++ b/test/packettest.c
@@ -465,6 +465,110 @@ static int test_PACKET_as_length_prefixed_2(void)
     return 1;
 }
 
+static int test_PACKET_get_quic_vlint(void)
+{
+    struct quic_test_case {
+        unsigned char buf[16];
+        size_t expected_read_count;
+        uint64_t value;
+    };
+
+    static const struct quic_test_case cases[] = {
+        { {0x00}, 1, 0  },
+        { {0x01}, 1, 1  },
+        { {0x3e}, 1, 62 },
+        { {0x3f}, 1, 63 },
+        { {0x40,0x00}, 2, 0 },
+        { {0x40,0x01}, 2, 1 },
+        { {0x40,0x02}, 2, 2 },
+        { {0x40,0xff}, 2, 255 },
+        { {0x41,0x00}, 2, 256 },
+        { {0x7f,0xfe}, 2, 16382 },
+        { {0x7f,0xff}, 2, 16383 },
+        { {0x80,0x00,0x00,0x00}, 4, 0 },
+        { {0x80,0x00,0x00,0x01}, 4, 1 },
+        { {0x80,0x00,0x01,0x02}, 4, 258 },
+        { {0x80,0x18,0x49,0x65}, 4, 1591653 },
+        { {0xbe,0x18,0x49,0x65}, 4, 1041779045 },
+        { {0xbf,0xff,0xff,0xff}, 4, 1073741823 },
+        { {0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x00}, 8, 0 },
+        { {0xc0,0x00,0x00,0x00,0x00,0x00,0x01,0x02}, 8, 258 },
+        { {0xfd,0x1f,0x59,0x8d,0xc9,0xf8,0x71,0x8a}, 8, 4404337426105397642 },
+    };
+
+    PACKET pkt;
+    size_t i;
+    uint64_t v;
+
+    for (i = 0; i < OSSL_NELEM(cases); ++i) {
+        memset(&pkt, 0, sizeof(pkt));
+        v = 55;
+
+        if (!TEST_true(PACKET_buf_init(&pkt, cases[i].buf, sizeof(cases[i].buf)))
+                || !TEST_true(PACKET_get_quic_vlint(&pkt, &v))
+                || !TEST_uint64_t_eq(v, cases[i].value)
+                || !TEST_size_t_eq(PACKET_remaining(&pkt),
+                                   sizeof(cases[i].buf) - cases[i].expected_read_count)
+           )
+            return 0;
+    }
+
+    return 1;
+}
+
+static int test_PACKET_get_quic_length_prefixed(void)
+{
+    struct quic_test_case {
+        unsigned char buf[16];
+        size_t enclen, len;
+        int fail;
+    };
+
+    static const struct quic_test_case cases[] = {
+        /* success cases */
+        { {0x00}, 1, 0, 0 },
+        { {0x01}, 1, 1, 0 },
+        { {0x02}, 1, 2, 0 },
+        { {0x03}, 1, 3, 0 },
+        { {0x04}, 1, 4, 0 },
+        { {0x05}, 1, 5, 0 },
+
+        /* failure cases */
+        { {0x10}, 1, 0, 1 },
+        { {0x3f}, 1, 0, 1 },
+    };
+
+    size_t i;
+    PACKET pkt, subpkt = {0};
+
+    for (i = 0; i < OSSL_NELEM(cases); ++i) {
+        memset(&pkt, 0, sizeof(pkt));
+
+        if (!TEST_true(PACKET_buf_init(&pkt, cases[i].buf,
+                                       cases[i].fail
+                                         ? sizeof(cases[i].buf)
+                                         : cases[i].enclen + cases[i].len)))
+            return 0;
+
+        if (!TEST_int_eq(PACKET_get_quic_length_prefixed(&pkt, &subpkt), !cases[i].fail))
+            return 0;
+
+        if (cases[i].fail) {
+            if (!TEST_ptr_eq(pkt.curr, cases[i].buf))
+                return 0;
+            continue;
+        }
+
+        if (!TEST_ptr_eq(subpkt.curr, cases[i].buf + cases[i].enclen))
+            return 0;
+
+        if (!TEST_size_t_eq(subpkt.remaining, cases[i].len))
+            return 0;
+    }
+
+    return 1;
+}
+
 int setup_tests(void)
 {
     unsigned int i;
@@ -495,5 +599,7 @@ int setup_tests(void)
     ADD_TEST(test_PACKET_get_length_prefixed_3);
     ADD_TEST(test_PACKET_as_length_prefixed_1);
     ADD_TEST(test_PACKET_as_length_prefixed_2);
+    ADD_TEST(test_PACKET_get_quic_vlint);
+    ADD_TEST(test_PACKET_get_quic_length_prefixed);
     return 1;
 }