summaryrefslogtreecommitdiff
path: root/test/packettest.c
diff options
context:
space:
mode:
Diffstat (limited to 'test/packettest.c')
-rw-r--r--test/packettest.c106
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;
}