| OLD | NEW | 
|---|
|  | (Empty) | 
| 1 /* |  | 
| 2  * rdbx.c |  | 
| 3  * |  | 
| 4  * a replay database with extended range, using a rollover counter |  | 
| 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 #ifdef HAVE_CONFIG_H |  | 
| 47     #include <config.h> |  | 
| 48 #endif |  | 
| 49 |  | 
| 50 #include "rdbx.h" |  | 
| 51 |  | 
| 52 |  | 
| 53 /* |  | 
| 54  * from RFC 3711: |  | 
| 55  * |  | 
| 56  * A receiver reconstructs the index i of a packet with sequence |  | 
| 57  *  number SEQ using the estimate |  | 
| 58  * |  | 
| 59  * i = 2^16 * v + SEQ, |  | 
| 60  * |  | 
| 61  * where v is chosen from the set { ROC-1, ROC, ROC+1 } such that i is |  | 
| 62  * closest to the value 2^16 * ROC + s_l.  If the value r+1 is used, |  | 
| 63  * then the rollover counter r in the cryptographic context is |  | 
| 64  * incremented by one (if the packet containing s is authentic). |  | 
| 65  */ |  | 
| 66 |  | 
| 67 |  | 
| 68 |  | 
| 69 /* |  | 
| 70  * rdbx implementation notes |  | 
| 71  * |  | 
| 72  * A xtd_seq_num_t is essentially a sequence number for which some of |  | 
| 73  * the data on the wire are implicit.  It logically consists of a |  | 
| 74  * rollover counter and a sequence number; the sequence number is the |  | 
| 75  * explicit part, and the rollover counter is the implicit part. |  | 
| 76  * |  | 
| 77  * Upon receiving a sequence_number (e.g. in a newly received SRTP |  | 
| 78  * packet), the complete xtd_seq_num_t can be estimated by using a |  | 
| 79  * local xtd_seq_num_t as a basis.  This is done using the function |  | 
| 80  * index_guess(&local, &guess, seq_from_packet).  This function |  | 
| 81  * returns the difference of the guess and the local value.  The local |  | 
| 82  * xtd_seq_num_t can be moved forward to the guess using the function |  | 
| 83  * index_advance(&guess, delta), where delta is the difference. |  | 
| 84  * |  | 
| 85  * |  | 
| 86  * A rdbx_t consists of a xtd_seq_num_t and a bitmask.  The index is highest |  | 
| 87  * sequence number that has been received, and the bitmask indicates |  | 
| 88  * which of the recent indicies have been received as well.  The |  | 
| 89  * highest bit in the bitmask corresponds to the index in the bitmask. |  | 
| 90  */ |  | 
| 91 |  | 
| 92 |  | 
| 93 void |  | 
| 94 index_init(xtd_seq_num_t *pi) { |  | 
| 95 #ifdef NO_64BIT_MATH |  | 
| 96   *pi = make64(0,0); |  | 
| 97 #else |  | 
| 98   *pi = 0; |  | 
| 99 #endif |  | 
| 100 } |  | 
| 101 |  | 
| 102 void |  | 
| 103 index_advance(xtd_seq_num_t *pi, sequence_number_t s) { |  | 
| 104 #ifdef NO_64BIT_MATH |  | 
| 105   /* a > ~b means a+b will generate a carry */ |  | 
| 106   /* s is uint16 here */ |  | 
| 107   *pi = make64(high32(*pi) + (s > ~low32(*pi) ? 1 : 0),low32(*pi) + s); |  | 
| 108 #else |  | 
| 109   *pi += s; |  | 
| 110 #endif |  | 
| 111 } |  | 
| 112 |  | 
| 113 |  | 
| 114 /* |  | 
| 115  * index_guess(local, guess, s) |  | 
| 116  * |  | 
| 117  * given a xtd_seq_num_t local (which represents the last |  | 
| 118  * known-to-be-good received xtd_seq_num_t) and a sequence number s |  | 
| 119  * (from a newly arrived packet), sets the contents of *guess to |  | 
| 120  * contain the best guess of the packet index to which s corresponds, |  | 
| 121  * and returns the difference between *guess and *local |  | 
| 122  * |  | 
| 123  * nota bene - the output is a signed integer, DON'T cast it to a |  | 
| 124  * unsigned integer! |  | 
| 125  */ |  | 
| 126 |  | 
| 127 int |  | 
| 128 index_guess(const xtd_seq_num_t *local, |  | 
| 129                    xtd_seq_num_t *guess, |  | 
| 130                    sequence_number_t s) { |  | 
| 131 #ifdef NO_64BIT_MATH |  | 
| 132   uint32_t local_roc = ((high32(*local) << 16) | |  | 
| 133                                                 (low32(*local) >> 16)); |  | 
| 134   uint16_t local_seq = (uint16_t) (low32(*local)); |  | 
| 135 #else |  | 
| 136   uint32_t local_roc = (uint32_t)(*local >> 16); |  | 
| 137   uint16_t local_seq = (uint16_t) *local; |  | 
| 138 #endif |  | 
| 139 #ifdef NO_64BIT_MATH |  | 
| 140   uint32_t guess_roc = ((high32(*guess) << 16) | |  | 
| 141                                                 (low32(*guess) >> 16)); |  | 
| 142   uint16_t guess_seq = (uint16_t) (low32(*guess)); |  | 
| 143 #else |  | 
| 144   uint32_t guess_roc = (uint32_t)(*guess >> 16); |  | 
| 145   uint16_t guess_seq = (uint16_t) *guess; |  | 
| 146 #endif |  | 
| 147   int difference; |  | 
| 148 |  | 
| 149   if (local_seq < seq_num_median) { |  | 
| 150     if (s - local_seq > seq_num_median) { |  | 
| 151       guess_roc = local_roc - 1; |  | 
| 152       difference = s - local_seq - seq_num_max; |  | 
| 153     } else { |  | 
| 154       guess_roc = local_roc; |  | 
| 155       difference = s - local_seq; |  | 
| 156     } |  | 
| 157   } else { |  | 
| 158     if (local_seq - seq_num_median > s) { |  | 
| 159       guess_roc = local_roc + 1; |  | 
| 160       difference = s - local_seq + seq_num_max; |  | 
| 161     } else { |  | 
| 162       guess_roc = local_roc; |  | 
| 163       difference = s - local_seq; |  | 
| 164     } |  | 
| 165   } |  | 
| 166   guess_seq = s; |  | 
| 167 |  | 
| 168   /* Note: guess_roc is 32 bits, so this generates a 48-bit result! */ |  | 
| 169 #ifdef NO_64BIT_MATH |  | 
| 170   *guess = make64(guess_roc >> 16, |  | 
| 171                                   (guess_roc << 16) | guess_seq); |  | 
| 172 #else |  | 
| 173   *guess = (((uint64_t) guess_roc) << 16) | guess_seq; |  | 
| 174 #endif |  | 
| 175 |  | 
| 176   return difference; |  | 
| 177 } |  | 
| 178 |  | 
| 179 /* |  | 
| 180  * rdbx |  | 
| 181  * |  | 
| 182  */ |  | 
| 183 |  | 
| 184 |  | 
| 185 /* |  | 
| 186  *  rdbx_init(&r, ws) initializes the rdbx_t pointed to by r with window size ws |  | 
| 187  */ |  | 
| 188 |  | 
| 189 err_status_t |  | 
| 190 rdbx_init(rdbx_t *rdbx, unsigned long ws) { |  | 
| 191   if (ws == 0) |  | 
| 192     return err_status_bad_param; |  | 
| 193 |  | 
| 194   if (bitvector_alloc(&rdbx->bitmask, ws) != 0) |  | 
| 195     return err_status_alloc_fail; |  | 
| 196 |  | 
| 197   index_init(&rdbx->index); |  | 
| 198 |  | 
| 199   return err_status_ok; |  | 
| 200 } |  | 
| 201 |  | 
| 202 /* |  | 
| 203  *  rdbx_dealloc(&r) frees memory for the rdbx_t pointed to by r |  | 
| 204  */ |  | 
| 205 |  | 
| 206 err_status_t |  | 
| 207 rdbx_dealloc(rdbx_t *rdbx) { |  | 
| 208   bitvector_dealloc(&rdbx->bitmask); |  | 
| 209 |  | 
| 210   return err_status_ok; |  | 
| 211 } |  | 
| 212 |  | 
| 213 /* |  | 
| 214  * rdbx_set_roc(rdbx, roc) initalizes the rdbx_t at the location rdbx |  | 
| 215  * to have the rollover counter value roc.  If that value is less than |  | 
| 216  * the current rollover counter value, then the function returns |  | 
| 217  * err_status_replay_old; otherwise, err_status_ok is returned. |  | 
| 218  * |  | 
| 219  */ |  | 
| 220 |  | 
| 221 err_status_t |  | 
| 222 rdbx_set_roc(rdbx_t *rdbx, uint32_t roc) { |  | 
| 223   bitvector_set_to_zero(&rdbx->bitmask); |  | 
| 224 |  | 
| 225 #ifdef NO_64BIT_MATH |  | 
| 226   #error not yet implemented |  | 
| 227 #else |  | 
| 228 |  | 
| 229   /* make sure that we're not moving backwards */ |  | 
| 230   if (roc < (rdbx->index >> 16)) |  | 
| 231     return err_status_replay_old; |  | 
| 232 |  | 
| 233   rdbx->index &= 0xffff;   /* retain lowest 16 bits */ |  | 
| 234   rdbx->index |= ((uint64_t)roc) << 16;  /* set ROC */ |  | 
| 235 #endif |  | 
| 236 |  | 
| 237   return err_status_ok; |  | 
| 238 } |  | 
| 239 |  | 
| 240 /* |  | 
| 241  * rdbx_get_packet_index(rdbx) returns the value of the packet index |  | 
| 242  * for the rdbx_t pointed to by rdbx |  | 
| 243  * |  | 
| 244  */ |  | 
| 245 |  | 
| 246 xtd_seq_num_t |  | 
| 247 rdbx_get_packet_index(const rdbx_t *rdbx) { |  | 
| 248   return rdbx->index; |  | 
| 249 } |  | 
| 250 |  | 
| 251 /* |  | 
| 252  * rdbx_get_window_size(rdbx) returns the value of the window size |  | 
| 253  * for the rdbx_t pointed to by rdbx |  | 
| 254  * |  | 
| 255  */ |  | 
| 256 |  | 
| 257 unsigned long |  | 
| 258 rdbx_get_window_size(const rdbx_t *rdbx) { |  | 
| 259   return bitvector_get_length(&rdbx->bitmask); |  | 
| 260 } |  | 
| 261 |  | 
| 262 /* |  | 
| 263  * rdbx_check(&r, delta) checks to see if the xtd_seq_num_t |  | 
| 264  * which is at rdbx->index + delta is in the rdb |  | 
| 265  */ |  | 
| 266 |  | 
| 267 err_status_t |  | 
| 268 rdbx_check(const rdbx_t *rdbx, int delta) { |  | 
| 269 |  | 
| 270   if (delta > 0) {       /* if delta is positive, it's good */ |  | 
| 271     return err_status_ok; |  | 
| 272   } else if ((int)(bitvector_get_length(&rdbx->bitmask) - 1) + delta < 0) { |  | 
| 273                          /* if delta is lower than the bitmask, it's bad */ |  | 
| 274     return err_status_replay_old; |  | 
| 275   } else if (bitvector_get_bit(&rdbx->bitmask, |  | 
| 276                                (int)(bitvector_get_length(&rdbx->bitmask) - 1) +
      delta) == 1) { |  | 
| 277                          /* delta is within the window, so check the bitmask */ |  | 
| 278     return err_status_replay_fail; |  | 
| 279   } |  | 
| 280  /* otherwise, the index is okay */ |  | 
| 281 |  | 
| 282   return err_status_ok; |  | 
| 283 } |  | 
| 284 |  | 
| 285 /* |  | 
| 286  * rdbx_add_index adds the xtd_seq_num_t at rdbx->window_start + d to |  | 
| 287  * replay_db (and does *not* check if that xtd_seq_num_t appears in db) |  | 
| 288  * |  | 
| 289  * this function should be called only after replay_check has |  | 
| 290  * indicated that the index does not appear in the rdbx, e.g., a mutex |  | 
| 291  * should protect the rdbx between these calls if need be |  | 
| 292  */ |  | 
| 293 |  | 
| 294 err_status_t |  | 
| 295 rdbx_add_index(rdbx_t *rdbx, int delta) { |  | 
| 296 |  | 
| 297   if (delta > 0) { |  | 
| 298     /* shift forward by delta */ |  | 
| 299     index_advance(&rdbx->index, delta); |  | 
| 300     bitvector_left_shift(&rdbx->bitmask, delta); |  | 
| 301     bitvector_set_bit(&rdbx->bitmask, bitvector_get_length(&rdbx->bitmask) - 1); |  | 
| 302   } else { |  | 
| 303     /* delta is in window */ |  | 
| 304     bitvector_set_bit(&rdbx->bitmask, bitvector_get_length(&rdbx->bitmask) -1 + 
     delta); |  | 
| 305   } |  | 
| 306 |  | 
| 307   /* note that we need not consider the case that delta == 0 */ |  | 
| 308 |  | 
| 309   return err_status_ok; |  | 
| 310 } |  | 
| 311 |  | 
| 312 |  | 
| 313 |  | 
| 314 /* |  | 
| 315  * rdbx_estimate_index(rdbx, guess, s) |  | 
| 316  * |  | 
| 317  * given an rdbx and a sequence number s (from a newly arrived packet), |  | 
| 318  * sets the contents of *guess to contain the best guess of the packet |  | 
| 319  * index to which s corresponds, and returns the difference between |  | 
| 320  * *guess and the locally stored synch info |  | 
| 321  */ |  | 
| 322 |  | 
| 323 int |  | 
| 324 rdbx_estimate_index(const rdbx_t *rdbx, |  | 
| 325                     xtd_seq_num_t *guess, |  | 
| 326                     sequence_number_t s) { |  | 
| 327 |  | 
| 328   /* |  | 
| 329    * if the sequence number and rollover counter in the rdbx are |  | 
| 330    * non-zero, then use the index_guess(...) function, otherwise, just |  | 
| 331    * set the rollover counter to zero (since the index_guess(...) |  | 
| 332    * function might incorrectly guess that the rollover counter is |  | 
| 333    * 0xffffffff) |  | 
| 334    */ |  | 
| 335 |  | 
| 336 #ifdef NO_64BIT_MATH |  | 
| 337   /* seq_num_median = 0x8000 */ |  | 
| 338   if (high32(rdbx->index) > 0 || |  | 
| 339           low32(rdbx->index) > seq_num_median) |  | 
| 340 #else |  | 
| 341   if (rdbx->index > seq_num_median) |  | 
| 342 #endif |  | 
| 343     return index_guess(&rdbx->index, guess, s); |  | 
| 344 |  | 
| 345 #ifdef NO_64BIT_MATH |  | 
| 346   *guess = make64(0,(uint32_t) s); |  | 
| 347 #else |  | 
| 348   *guess = s; |  | 
| 349 #endif |  | 
| 350 |  | 
| 351 #ifdef NO_64BIT_MATH |  | 
| 352   return s - (uint16_t) low32(rdbx->index); |  | 
| 353 #else |  | 
| 354   return s - (uint16_t) rdbx->index; |  | 
| 355 #endif |  | 
| 356 } |  | 
| OLD | NEW | 
|---|