1 /*
2 * replay_driver.c
3 *
4 * A driver for the replay_database implementation
5 *
6 * David A. McGrew
7 * Cisco Systems, Inc.
8 */
9
10 /*
11 *
12 * Copyright (c) 2001-2006, Cisco Systems, Inc.
13 * All rights reserved.
14 *
15 * Redistribution and use in source and binary forms, with or without
16 * modification, are permitted provided that the following conditions
17 * are met:
18 *
19 * Redistributions of source code must retain the above copyright
20 * notice, this list of conditions and the following disclaimer.
21 *
22 * Redistributions in binary form must reproduce the above
23 * copyright notice, this list of conditions and the following
24 * disclaimer in the documentation and/or other materials provided
25 * with the distribution.
26 *
27 * Neither the name of the Cisco Systems, Inc. nor the names of its
28 * contributors may be used to endorse or promote products derived
29 * from this software without specific prior written permission.
30 *
31 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
32 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
33 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
34 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
35 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
36 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
37 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
38 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
39 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
40 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
41 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
42 * OF THE POSSIBILITY OF SUCH DAMAGE.
43 *
44 */
45
46 #include <stdio.h>
47
48 #include "rdb.h"
49 #include "ut_sim.h"
50
51 /*
52 * num_trials defines the number of trials that are used in the
53 * validation functions below
54 */
55
56 unsigned num_trials = 1 << 16;
57
58 err_status_t
59 test_rdb_db(void);
60
61 double
62 rdb_check_adds_per_second(void);
63
64 int
main(void)65 main (void) {
66 err_status_t err;
67
68 printf("testing anti-replay database (rdb_t)...\n");
69 err = test_rdb_db();
70 if (err) {
71 printf("failed\n");
72 exit(1);
73 }
74 printf("done\n");
75
76 printf("rdb_check/rdb_adds per second: %e\n",
77 rdb_check_adds_per_second());
78
79 return 0;
80 }
81
82
83 void
print_rdb(rdb_t * rdb)84 print_rdb(rdb_t *rdb) {
85 printf("rdb: {%u, %s}\n", rdb->window_start, v128_bit_string(&rdb->bitmask));
86 }
87
88 err_status_t
rdb_check_add(rdb_t * rdb,uint32_t idx)89 rdb_check_add(rdb_t *rdb, uint32_t idx) {
90
91 if (rdb_check(rdb, idx) != err_status_ok) {
92 printf("rdb_check failed at index %u\n", idx);
93 return err_status_fail;
94 }
95 if (rdb_add_index(rdb, idx) != err_status_ok) {
96 printf("rdb_add_index failed at index %u\n", idx);
97 return err_status_fail;
98 }
99
100 return err_status_ok;
101 }
102
103 err_status_t
rdb_check_expect_failure(rdb_t * rdb,uint32_t idx)104 rdb_check_expect_failure(rdb_t *rdb, uint32_t idx) {
105 err_status_t err;
106
107 err = rdb_check(rdb, idx);
108 if ((err != err_status_replay_old) && (err != err_status_replay_fail)) {
109 printf("rdb_check failed at index %u (false positive)\n", idx);
110 return err_status_fail;
111 }
112
113 return err_status_ok;
114 }
115
116 err_status_t
rdb_check_unordered(rdb_t * rdb,uint32_t idx)117 rdb_check_unordered(rdb_t *rdb, uint32_t idx) {
118 err_status_t rstat;
119
120 /* printf("index: %u\n", idx); */
121 rstat = rdb_check(rdb, idx);
122 if ((rstat != err_status_ok) && (rstat != err_status_replay_old)) {
123 printf("rdb_check_unordered failed at index %u\n", idx);
124 return rstat;
125 }
126 return err_status_ok;
127 }
128
129 err_status_t
test_rdb_db()130 test_rdb_db() {
131 rdb_t rdb;
132 uint32_t idx, ircvd;
133 ut_connection utc;
134 err_status_t err;
135
136 if (rdb_init(&rdb) != err_status_ok) {
137 printf("rdb_init failed\n");
138 return err_status_init_fail;
139 }
140
141 /* test sequential insertion */
142 for (idx=0; idx < num_trials; idx++) {
143 err = rdb_check_add(&rdb, idx);
144 if (err)
145 return err;
146 }
147
148 /* test for false positives */
149 for (idx=0; idx < num_trials; idx++) {
150 err = rdb_check_expect_failure(&rdb, idx);
151 if (err)
152 return err;
153 }
154
155 /* re-initialize */
156 if (rdb_init(&rdb) != err_status_ok) {
157 printf("rdb_init failed\n");
158 return err_status_fail;
159 }
160
161 /* test non-sequential insertion */
162 ut_init(&utc);
163
164 for (idx=0; idx < num_trials; idx++) {
165 ircvd = ut_next_index(&utc);
166 err = rdb_check_unordered(&rdb, ircvd);
167 if (err)
168 return err;
169 }
170
171 return err_status_ok;
172 }
173
174 #include <time.h> /* for clock() */
175 #include <stdlib.h> /* for random() */
176
177 #define REPLAY_NUM_TRIALS 10000000
178
179 double
rdb_check_adds_per_second(void)180 rdb_check_adds_per_second(void) {
181 uint32_t i;
182 rdb_t rdb;
183 clock_t timer;
184 int failures; /* count number of failures */
185
186 if (rdb_init(&rdb) != err_status_ok) {
187 printf("rdb_init failed\n");
188 exit(1);
189 }
190
191 timer = clock();
192 for(i=0; i < REPLAY_NUM_TRIALS; i+=3) {
193 if (rdb_check(&rdb, i+2) != err_status_ok)
194 ++failures;
195 if (rdb_add_index(&rdb, i+2) != err_status_ok)
196 ++failures;
197 if (rdb_check(&rdb, i+1) != err_status_ok)
198 ++failures;
199 if (rdb_add_index(&rdb, i+1) != err_status_ok)
200 ++failures;
201 if (rdb_check(&rdb, i) != err_status_ok)
202 ++failures;
203 if (rdb_add_index(&rdb, i) != err_status_ok)
204 ++failures;
205 }
206 timer = clock() - timer;
207
208 return (double) CLOCKS_PER_SEC * REPLAY_NUM_TRIALS / timer;
209 }
210