1 /* 2 * Copyright 2023 The OpenSSL Project Authors. All Rights Reserved. 3 * 4 * Licensed under the Apache License 2.0 (the "License"). You may not use 5 * this file except in compliance with the License. You can obtain a copy 6 * in the file LICENSE in the source distribution or at 7 * https://www.openssl.org/source/license.html 8 */ 9 10 #include "internal/quic_lcidm.h" 11 #include "testutil.h" 12 13 static char ptrs[8]; 14 15 static const QUIC_CONN_ID cid8_1 = { 8, { 1 } }; 16 static const QUIC_CONN_ID cid8_2 = { 8, { 2 } }; 17 static const QUIC_CONN_ID cid8_3 = { 8, { 3 } }; 18 static const QUIC_CONN_ID cid8_4 = { 8, { 4 } }; 19 static const QUIC_CONN_ID cid8_5 = { 8, { 5 } }; 20 21 static int test_lcidm(void) 22 { 23 int testresult = 0; 24 QUIC_LCIDM *lcidm; 25 size_t lcid_len = 10; /* != ODCID len */ 26 QUIC_CONN_ID lcid_1, lcid_dummy, lcid_init = {0}; 27 OSSL_QUIC_FRAME_NEW_CONN_ID ncid_frame_1, ncid_frame_2, ncid_frame_3; 28 void *opaque = NULL; 29 uint64_t seq_num = UINT64_MAX; 30 int did_retire = 0; 31 32 if (!TEST_ptr(lcidm = ossl_quic_lcidm_new(NULL, lcid_len))) 33 goto err; 34 35 if (!TEST_size_t_eq(ossl_quic_lcidm_get_lcid_len(lcidm), lcid_len)) 36 goto err; 37 38 if (!TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1)) 39 || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_2)) 40 || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_1)) 41 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 0) 42 || !TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_3)) 43 || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_4)) 44 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 1) 45 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 1) 46 || !TEST_true(ossl_quic_lcidm_retire_odcid(lcidm, ptrs + 0)) 47 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0) 48 || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1)) 49 || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_5)) 50 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0) 51 52 || !TEST_true(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_1)) 53 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 1) 54 || !TEST_false(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_init)) 55 || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_1)) 56 || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_2)) 57 || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_3)) 58 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 4) 59 || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &lcid_1, &seq_num, &opaque)) 60 || !TEST_ptr_eq(opaque, ptrs + 2) 61 || !TEST_uint64_t_eq(seq_num, 0) 62 || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id, 63 &seq_num, &opaque)) 64 || !TEST_ptr_eq(opaque, ptrs + 2) 65 || !TEST_uint64_t_eq(seq_num, 1) 66 || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id, 67 &seq_num, &opaque)) 68 || !TEST_ptr_eq(opaque, ptrs + 2) 69 || !TEST_uint64_t_eq(seq_num, 2) 70 || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_3.conn_id, 71 &seq_num, &opaque)) 72 || !TEST_ptr_eq(opaque, ptrs + 2) 73 || !TEST_uint64_t_eq(seq_num, 3) 74 || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL, 75 &lcid_dummy, &seq_num, &did_retire)) 76 || !TEST_true(did_retire) 77 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 3) 78 || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL, 79 &lcid_dummy, &seq_num, &did_retire)) 80 || !TEST_true(did_retire) 81 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2) 82 || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL, 83 &lcid_dummy, &seq_num, &did_retire)) 84 || !TEST_false(did_retire) 85 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2) 86 || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &lcid_init, 87 &seq_num, &opaque)) 88 || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id, 89 &seq_num, &opaque)) 90 || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id, 91 &seq_num, &opaque)) 92 || !TEST_true(ossl_quic_lcidm_cull(lcidm, ptrs + 2)) 93 || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 0)) 94 goto err; 95 96 testresult = 1; 97 err: 98 ossl_quic_lcidm_free(lcidm); 99 return testresult; 100 } 101 102 int setup_tests(void) 103 { 104 ADD_TEST(test_lcidm); 105 return 1; 106 } 107