QUIC LCIDM: Add test

Reviewed-by: Neil Horman <nhorman@openssl.org>
Reviewed-by: Matt Caswell <matt@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/22673)

3 files changed, 130 insertions, 1 deletions

diff --git a/test/build.info b/test/build.info
index ef3d5dad95..bb35d09f33 100644
--- a/test/build.info
+++ b/test/build.info
@@ -332,6 +332,10 @@ IF[{- !$disabled{tests} -}]
   INCLUDE[quic_srtm_test]=../include ../apps/include
   DEPEND[quic_srtm_test]=../libcrypto.a ../libssl.a libtestutil.a
 
+  SOURCE[quic_lcidm_test]=quic_lcidm_test.c
+  INCLUDE[quic_lcidm_test]=../include ../apps/include
+  DEPEND[quic_lcidm_test]=../libcrypto.a ../libssl.a libtestutil.a
+
   SOURCE[quic_fifd_test]=quic_fifd_test.c cc_dummy.c
   INCLUDE[quic_fifd_test]=../include ../apps/include
   DEPEND[quic_fifd_test]=../libcrypto.a ../libssl.a libtestutil.a
@@ -1148,7 +1152,7 @@ ENDIF
   IF[{- !$disabled{'quic'} -}]
     PROGRAMS{noinst}=quic_wire_test quic_ackm_test quic_record_test
     PROGRAMS{noinst}=quic_fc_test quic_stream_test quic_cfq_test quic_txpim_test
-    PROGRAMS{noinst}=quic_srtm_test
+    PROGRAMS{noinst}=quic_srtm_test quic_lcidm_test
     PROGRAMS{noinst}=quic_fifd_test quic_txp_test quic_tserver_test
     PROGRAMS{noinst}=quic_client_test quic_cc_test quic_multistream_test
   ENDIF
diff --git a/test/quic_lcidm_test.c b/test/quic_lcidm_test.c
new file mode 100644
index 0000000000..d877991049
--- /dev/null
+++ b/test/quic_lcidm_test.c
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2023 The OpenSSL Project Authors. All Rights Reserved.
+ *
+ * Licensed under the Apache License 2.0 (the "License").  You may not use
+ * this file except in compliance with the License.  You can obtain a copy
+ * in the file LICENSE in the source distribution or at
+ * https://www.openssl.org/source/license.html
+ */
+
+#include "internal/quic_lcidm.h"
+#include "testutil.h"
+
+static char ptrs[8];
+
+static const QUIC_CONN_ID cid8_1 = { 8, { 1 } };
+static const QUIC_CONN_ID cid8_2 = { 8, { 2 } };
+static const QUIC_CONN_ID cid8_3 = { 8, { 3 } };
+static const QUIC_CONN_ID cid8_4 = { 8, { 4 } };
+static const QUIC_CONN_ID cid8_5 = { 8, { 5 } };
+
+static int test_lcidm(void)
+{
+    int testresult = 0;
+    QUIC_LCIDM *lcidm;
+    size_t lcid_len = 8;
+    QUIC_CONN_ID lcid_1, lcid_dummy, lcid_init;
+    OSSL_QUIC_FRAME_NEW_CONN_ID ncid_frame_1, ncid_frame_2, ncid_frame_3;
+    void *opaque = NULL;
+    uint64_t seq_num = UINT64_MAX;
+    int did_retire = 0;
+
+    if (!TEST_ptr(lcidm = ossl_quic_lcidm_new(NULL, lcid_len)))
+        goto err;
+
+    if (!TEST_size_t_eq(ossl_quic_lcidm_get_lcid_len(lcidm), lcid_len))
+        goto err;
+
+    if (!TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1))
+        || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_2))
+        || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_1))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 0)
+        || !TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_3))
+        || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_4))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 1)
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 1)
+        || !TEST_true(ossl_quic_lcidm_retire_odcid(lcidm, ptrs + 0))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0)
+        || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1))
+        || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_5))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0)
+
+        || !TEST_true(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_1))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 1)
+        || !TEST_false(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_init))
+        || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_1))
+        || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_2))
+        || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_3))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 4)
+        || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &lcid_1, &seq_num, &opaque))
+        || !TEST_ptr_eq(opaque, ptrs + 2)
+        || !TEST_uint64_t_eq(seq_num, 0)
+        || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id,
+                                             &seq_num, &opaque))
+        || !TEST_ptr_eq(opaque, ptrs + 2)
+        || !TEST_uint64_t_eq(seq_num, 1)
+        || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id,
+                                             &seq_num, &opaque))
+        || !TEST_ptr_eq(opaque, ptrs + 2)
+        || !TEST_uint64_t_eq(seq_num, 2)
+        || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_3.conn_id,
+                                             &seq_num, &opaque))
+        || !TEST_ptr_eq(opaque, ptrs + 2)
+        || !TEST_uint64_t_eq(seq_num, 3)
+        || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
+                                             &lcid_dummy, &seq_num, &did_retire))
+        || !TEST_true(did_retire)
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 3)
+        || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
+                                             &lcid_dummy, &seq_num, &did_retire))
+        || !TEST_true(did_retire)
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2)
+        || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
+                                             &lcid_dummy, &seq_num, &did_retire))
+        || !TEST_false(did_retire)
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2)
+        || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &lcid_init,
+                                              &seq_num, &opaque))
+        || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id,
+                                              &seq_num, &opaque))
+        || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id,
+                                             &seq_num, &opaque))
+        || !TEST_true(ossl_quic_lcidm_cull(lcidm, ptrs + 2))
+        || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 0))
+        goto err;
+
+    testresult = 1;
+err:
+    ossl_quic_lcidm_free(lcidm);
+    return testresult;
+}
+
+int setup_tests(void)
+{
+    ADD_TEST(test_lcidm);
+    return 1;
+}
diff --git a/test/recipes/70-test_quic_lcidm.t b/test/recipes/70-test_quic_lcidm.t
new file mode 100644
index 0000000000..b61705beb1
--- /dev/null
+++ b/test/recipes/70-test_quic_lcidm.t
@@ -0,0 +1,19 @@
+#! /usr/bin/env perl
+# Copyright 2023 The OpenSSL Project Authors. All Rights Reserved.
+#
+# Licensed under the Apache License 2.0 (the "License").  You may not use
+# this file except in compliance with the License.  You can obtain a copy
+# in the file LICENSE in the source distribution or at
+# https://www.openssl.org/source/license.html
+
+use OpenSSL::Test;
+use OpenSSL::Test::Utils;
+
+setup("test_quic_lcidm");
+
+plan skip_all => "QUIC protocol is not supported by this OpenSSL build"
+    if disabled('quic');
+
+plan tests => 1;
+
+ok(run(test(["quic_lcidm_test"])));