key.c 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /*
  2. * key.c
  3. *
  4. * key usage limits enforcement
  5. *
  6. * David A. Mcgrew
  7. * Cisco Systems, Inc.
  8. */
  9. /*
  10. *
  11. * Copyright (c) 2001-2017 Cisco Systems, Inc.
  12. * All rights reserved.
  13. *
  14. * Redistribution and use in source and binary forms, with or without
  15. * modification, are permitted provided that the following conditions
  16. * are met:
  17. *
  18. * Redistributions of source code must retain the above copyright
  19. * notice, this list of conditions and the following disclaimer.
  20. *
  21. * Redistributions in binary form must reproduce the above
  22. * copyright notice, this list of conditions and the following
  23. * disclaimer in the documentation and/or other materials provided
  24. * with the distribution.
  25. *
  26. * Neither the name of the Cisco Systems, Inc. nor the names of its
  27. * contributors may be used to endorse or promote products derived
  28. * from this software without specific prior written permission.
  29. *
  30. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  31. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  32. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  33. * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  34. * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  35. * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  36. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  37. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  38. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  39. * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  40. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  41. * OF THE POSSIBILITY OF SUCH DAMAGE.
  42. *
  43. */
  44. #ifdef HAVE_CONFIG_H
  45. #include <config.h>
  46. #endif
  47. #include "key.h"
  48. #define soft_limit 0x10000
  49. srtp_err_status_t srtp_key_limit_set(srtp_key_limit_t key,
  50. const srtp_xtd_seq_num_t s)
  51. {
  52. #ifdef NO_64BIT_MATH
  53. if (high32(s) == 0 && low32(s) < soft_limit) {
  54. return srtp_err_status_bad_param;
  55. }
  56. #else
  57. if (s < soft_limit) {
  58. return srtp_err_status_bad_param;
  59. }
  60. #endif
  61. key->num_left = s;
  62. key->state = srtp_key_state_normal;
  63. return srtp_err_status_ok;
  64. }
  65. srtp_err_status_t srtp_key_limit_clone(srtp_key_limit_t original,
  66. srtp_key_limit_t *new_key)
  67. {
  68. if (original == NULL) {
  69. return srtp_err_status_bad_param;
  70. }
  71. *new_key = original;
  72. return srtp_err_status_ok;
  73. }
  74. srtp_key_event_t srtp_key_limit_update(srtp_key_limit_t key)
  75. {
  76. #ifdef NO_64BIT_MATH
  77. if (low32(key->num_left) == 0) {
  78. // carry
  79. key->num_left =
  80. make64(high32(key->num_left) - 1, low32(key->num_left) - 1);
  81. } else {
  82. // no carry
  83. key->num_left = make64(high32(key->num_left), low32(key->num_left) - 1);
  84. }
  85. if (high32(key->num_left) != 0 || low32(key->num_left) >= soft_limit) {
  86. return srtp_key_event_normal; /* we're above the soft limit */
  87. }
  88. #else
  89. key->num_left--;
  90. if (key->num_left >= soft_limit) {
  91. return srtp_key_event_normal; /* we're above the soft limit */
  92. }
  93. #endif
  94. if (key->state == srtp_key_state_normal) {
  95. /* we just passed the soft limit, so change the state */
  96. key->state = srtp_key_state_past_soft_limit;
  97. }
  98. #ifdef NO_64BIT_MATH
  99. if (low32(key->num_left) == 0 && high32(key->num_left == 0))
  100. #else
  101. if (key->num_left < 1)
  102. #endif
  103. { /* we just hit the hard limit */
  104. key->state = srtp_key_state_expired;
  105. return srtp_key_event_hard_limit;
  106. }
  107. return srtp_key_event_soft_limit;
  108. }