QUIC RCIDM: Add test

Reviewed-by: Matt Caswell <matt@openssl.org>
Reviewed-by: Tomas Mraz <tomas@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/23022)

3 files changed, 151 insertions, 1 deletions

diff --git a/test/build.info b/test/build.info
index 88620b86b3..7e1ce28522 100644
--- a/test/build.info
+++ b/test/build.info
@@ -344,6 +344,10 @@ IF[{- !$disabled{tests} -}]
   INCLUDE[quic_lcidm_test]=../include ../apps/include
   DEPEND[quic_lcidm_test]=../libcrypto.a ../libssl.a libtestutil.a
 
+  SOURCE[quic_rcidm_test]=quic_rcidm_test.c
+  INCLUDE[quic_rcidm_test]=../include ../apps/include
+  DEPEND[quic_rcidm_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
@@ -1160,7 +1164,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 quic_lcidm_test
+    PROGRAMS{noinst}=quic_srtm_test quic_lcidm_test quic_rcidm_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_rcidm_test.c b/test/quic_rcidm_test.c
new file mode 100644
index 0000000000..3e841bed56
--- /dev/null
+++ b/test/quic_rcidm_test.c
@@ -0,0 +1,127 @@
+/*
+ * 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_rcidm.h"
+#include "testutil.h"
+
+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_rcidm(int idx)
+{
+    int testresult = 0;
+    QUIC_RCIDM *rcidm;
+    OSSL_QUIC_FRAME_NEW_CONN_ID ncid_frame_1 = {0}, ncid_frame_2 = {0};
+    QUIC_CONN_ID dcid_out;
+    const QUIC_CONN_ID *odcid = NULL;
+    uint64_t seq_num_out;
+
+    ncid_frame_1.seq_num        = 2;
+    ncid_frame_1.conn_id.id_len = 8;
+    ncid_frame_1.conn_id.id[0]  = 3;
+
+    ncid_frame_2.seq_num        = 3;
+    ncid_frame_2.conn_id.id_len = 8;
+    ncid_frame_2.conn_id.id[0]  = 4;
+
+    odcid = ((idx == 2) ? NULL : &cid8_1);
+    if (!TEST_ptr(rcidm = ossl_quic_rcidm_new(odcid)))
+        goto err;
+
+    if (idx != 2) {
+        if (/* ODCID not counted */
+            !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+            || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
+            || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+
+            || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_1))
+            || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
+            goto err;
+    } else {
+        if (!TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+            || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
+            goto err;
+    }
+
+    if (idx == 1) {
+        if (!TEST_true(ossl_quic_rcidm_add_from_server_retry(rcidm, &cid8_5))
+            || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+            || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
+            || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+
+            || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_5))
+            || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
+            goto err;
+    }
+
+    if (!TEST_true(ossl_quic_rcidm_add_from_initial(rcidm, &cid8_2))
+        /* Initial SCID (seq=0) is counted */
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 1)
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+        || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+        || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2))
+
+        || !TEST_true(ossl_quic_rcidm_add_from_ncid(rcidm, &ncid_frame_1))
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 2)
+        /* Not changed over yet - handshake not confirmed */
+        || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+        || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2))
+
+        || !TEST_true(ossl_quic_rcidm_add_from_ncid(rcidm, &ncid_frame_2))
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 3)
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 0)
+        || !TEST_false(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
+        /* Not changed over yet - handshake not confirmed */
+        || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+        || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2)))
+        goto err;
+
+    ossl_quic_rcidm_on_handshake_complete(rcidm);
+
+    if (!TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+        || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+        || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_3))
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 1)
+        || !TEST_true(ossl_quic_rcidm_peek_retire_seq_num(rcidm, &seq_num_out))
+        || !TEST_uint64_t_eq(seq_num_out, 0))
+        goto err;
+
+    ossl_quic_rcidm_request_roll(rcidm);
+
+    if (!TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+        || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
+        || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
+        || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_4))
+        || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 2)
+        || !TEST_true(ossl_quic_rcidm_peek_retire_seq_num(rcidm, &seq_num_out))
+        || !TEST_uint64_t_eq(seq_num_out, 0)
+        || !TEST_true(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
+        || !TEST_uint64_t_eq(seq_num_out, 0)
+        || !TEST_true(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
+        || !TEST_uint64_t_eq(seq_num_out, 2))
+        goto err;
+
+    testresult = 1;
+err:
+    ossl_quic_rcidm_free(rcidm);
+    return testresult;
+}
+
+int setup_tests(void)
+{
+    ADD_ALL_TESTS(test_rcidm, 3);
+    return 1;
+}
diff --git a/test/recipes/70-test_quic_rcidm.t b/test/recipes/70-test_quic_rcidm.t
new file mode 100644
index 0000000000..a00f381eb6
--- /dev/null
+++ b/test/recipes/70-test_quic_rcidm.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_rcidm");
+
+plan skip_all => "QUIC protocol is not supported by this OpenSSL build"
+    if disabled('quic');
+
+plan tests => 1;
+
+ok(run(test(["quic_rcidm_test"])));