]>
Commit | Line | Data |
---|---|---|
90a11157 HL |
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_srtm.h" | |
11 | #include "testutil.h" | |
12 | ||
13 | static char ptrs[8]; | |
14 | ||
15 | static const QUIC_STATELESS_RESET_TOKEN token_1 = {{ | |
16 | 0x01, 0x02, 0x03, 0x04 | |
17 | }}; | |
18 | ||
19 | static const QUIC_STATELESS_RESET_TOKEN token_2 = {{ | |
20 | 0x01, 0x02, 0x03, 0x05 | |
21 | }}; | |
22 | ||
23 | static int test_srtm(void) | |
24 | { | |
25 | int testresult = 0; | |
26 | QUIC_SRTM *srtm; | |
27 | void *opaque = NULL; | |
28 | uint64_t seq_num = 0; | |
29 | ||
30 | if (!TEST_ptr(srtm = ossl_quic_srtm_new(NULL, NULL))) | |
31 | goto err; | |
32 | ||
33 | if (!TEST_true(ossl_quic_srtm_add(srtm, ptrs + 0, 0, &token_1)) | |
34 | || !TEST_false(ossl_quic_srtm_add(srtm, ptrs + 0, 0, &token_1)) | |
35 | || !TEST_false(ossl_quic_srtm_remove(srtm, ptrs + 0, 1)) | |
36 | || !TEST_false(ossl_quic_srtm_remove(srtm, ptrs + 3, 0)) | |
37 | || !TEST_true(ossl_quic_srtm_cull(srtm, ptrs + 3)) | |
38 | || !TEST_true(ossl_quic_srtm_cull(srtm, ptrs + 3)) | |
39 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 0, 1, &token_1)) | |
40 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 0, 2, &token_1)) | |
41 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 0, 3, &token_1)) | |
42 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 1, 0, &token_1)) | |
43 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 2, 0, &token_2)) | |
44 | || !TEST_true(ossl_quic_srtm_add(srtm, ptrs + 3, 3, &token_2)) | |
45 | || !TEST_true(ossl_quic_srtm_remove(srtm, ptrs + 3, 3)) | |
46 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 0, &opaque, &seq_num)) | |
47 | || !TEST_ptr_eq(opaque, ptrs + 1) | |
48 | || !TEST_uint64_t_eq(seq_num, 0) | |
49 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 1, &opaque, &seq_num)) | |
50 | || !TEST_ptr_eq(opaque, ptrs + 0) | |
51 | || !TEST_uint64_t_eq(seq_num, 3) | |
52 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 2, &opaque, &seq_num)) | |
53 | || !TEST_ptr_eq(opaque, ptrs + 0) | |
54 | || !TEST_uint64_t_eq(seq_num, 2) | |
55 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 3, &opaque, &seq_num)) | |
56 | || !TEST_ptr_eq(opaque, ptrs + 0) | |
57 | || !TEST_uint64_t_eq(seq_num, 1) | |
58 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 4, &opaque, &seq_num)) | |
59 | || !TEST_ptr_eq(opaque, ptrs + 0) | |
60 | || !TEST_uint64_t_eq(seq_num, 0) | |
61 | || !TEST_false(ossl_quic_srtm_lookup(srtm, &token_1, 5, &opaque, &seq_num)) | |
62 | || !TEST_true(ossl_quic_srtm_cull(srtm, ptrs + 0)) | |
63 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_1, 0, &opaque, &seq_num)) | |
64 | || !TEST_ptr_eq(opaque, ptrs + 1) | |
65 | || !TEST_uint64_t_eq(seq_num, 0) | |
66 | || !TEST_true(ossl_quic_srtm_lookup(srtm, &token_2, 0, &opaque, &seq_num)) | |
67 | || !TEST_ptr_eq(opaque, ptrs + 2) | |
68 | || !TEST_uint64_t_eq(seq_num, 0) | |
69 | || !TEST_true(ossl_quic_srtm_remove(srtm, ptrs + 2, 0)) | |
70 | || !TEST_false(ossl_quic_srtm_lookup(srtm, &token_2, 0, &opaque, &seq_num)) | |
71 | ) | |
72 | goto err; | |
73 | ||
74 | testresult = 1; | |
75 | err: | |
76 | ossl_quic_srtm_free(srtm); | |
77 | return testresult; | |
78 | } | |
79 | ||
80 | int setup_tests(void) | |
81 | { | |
82 | ADD_TEST(test_srtm); | |
83 | return 1; | |
84 | } |