123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217 |
- #ifdef HAVE_CONFIG_H
- #include <config.h>
- #endif
- #include "auth.h"
- #include "alloc.h"
- #include "err.h"
- #include "auth_test_cases.h"
- #include <mbedtls/md.h>
- #define SHA1_DIGEST_SIZE 20
- srtp_debug_module_t srtp_mod_hmac = {
- 0,
- "hmac sha-1 mbedtls"
- };
- static srtp_err_status_t srtp_hmac_mbedtls_alloc(srtp_auth_t **a,
- int key_len,
- int out_len)
- {
- extern const srtp_auth_type_t srtp_hmac;
- debug_print(srtp_mod_hmac, "allocating auth func with key length %d",
- key_len);
- debug_print(srtp_mod_hmac, " tag length %d",
- out_len);
-
- if (out_len > SHA1_DIGEST_SIZE) {
- return srtp_err_status_bad_param;
- }
- *a = (srtp_auth_t *)srtp_crypto_alloc(sizeof(srtp_auth_t));
- if (*a == NULL) {
- return srtp_err_status_alloc_fail;
- }
-
- (*a)->state = srtp_crypto_alloc(sizeof(mbedtls_md_context_t));
- if ((*a)->state == NULL) {
- srtp_crypto_free(*a);
- *a = NULL;
- return srtp_err_status_alloc_fail;
- }
- mbedtls_md_init((mbedtls_md_context_t *)(*a)->state);
-
- (*a)->type = &srtp_hmac;
- (*a)->out_len = out_len;
- (*a)->key_len = key_len;
- (*a)->prefix_len = 0;
- return srtp_err_status_ok;
- }
- static srtp_err_status_t srtp_hmac_mbedtls_dealloc(srtp_auth_t *a)
- {
- mbedtls_md_context_t *hmac_ctx;
- hmac_ctx = (mbedtls_md_context_t *)a->state;
- mbedtls_md_free(hmac_ctx);
- srtp_crypto_free(hmac_ctx);
-
- octet_string_set_to_zero(a, sizeof(srtp_auth_t));
-
- srtp_crypto_free(a);
- return srtp_err_status_ok;
- }
- static srtp_err_status_t srtp_hmac_mbedtls_start(void *statev)
- {
- mbedtls_md_context_t *state = (mbedtls_md_context_t *)statev;
- if (mbedtls_md_hmac_reset(state) != 0)
- return srtp_err_status_auth_fail;
- return srtp_err_status_ok;
- }
- static srtp_err_status_t srtp_hmac_mbedtls_init(void *statev,
- const uint8_t *key,
- int key_len)
- {
- mbedtls_md_context_t *state = (mbedtls_md_context_t *)statev;
- const mbedtls_md_info_t *info = NULL;
- info = mbedtls_md_info_from_type(MBEDTLS_MD_SHA1);
- if (info == NULL)
- return srtp_err_status_auth_fail;
- if (mbedtls_md_setup(state, info, 1) != 0)
- return srtp_err_status_auth_fail;
- debug_print(srtp_mod_hmac, "mbedtls setup, name: %s",
- mbedtls_md_get_name(info));
- debug_print(srtp_mod_hmac, "mbedtls setup, size: %d",
- mbedtls_md_get_size(info));
- if (mbedtls_md_hmac_starts(state, key, key_len) != 0)
- return srtp_err_status_auth_fail;
- return srtp_err_status_ok;
- }
- static srtp_err_status_t srtp_hmac_mbedtls_update(void *statev,
- const uint8_t *message,
- int msg_octets)
- {
- mbedtls_md_context_t *state = (mbedtls_md_context_t *)statev;
- debug_print(srtp_mod_hmac, "input: %s",
- srtp_octet_string_hex_string(message, msg_octets));
- if (mbedtls_md_hmac_update(state, message, msg_octets) != 0)
- return srtp_err_status_auth_fail;
- return srtp_err_status_ok;
- }
- static srtp_err_status_t srtp_hmac_mbedtls_compute(void *statev,
- const uint8_t *message,
- int msg_octets,
- int tag_len,
- uint8_t *result)
- {
- mbedtls_md_context_t *state = (mbedtls_md_context_t *)statev;
- uint8_t hash_value[SHA1_DIGEST_SIZE];
- int i;
-
- if (tag_len > SHA1_DIGEST_SIZE) {
- return srtp_err_status_bad_param;
- }
-
- if (mbedtls_md_hmac_update(statev, message, msg_octets) != 0)
- return srtp_err_status_auth_fail;
- if (mbedtls_md_hmac_finish(state, hash_value) != 0)
- return srtp_err_status_auth_fail;
-
- for (i = 0; i < tag_len; i++) {
- result[i] = hash_value[i];
- }
- debug_print(srtp_mod_hmac, "output: %s",
- srtp_octet_string_hex_string(hash_value, tag_len));
- return srtp_err_status_ok;
- }
- static const char srtp_hmac_mbedtls_description[] =
- "hmac sha-1 authentication function using mbedtls";
- const srtp_auth_type_t srtp_hmac = {
- srtp_hmac_mbedtls_alloc,
- srtp_hmac_mbedtls_dealloc,
- srtp_hmac_mbedtls_init,
- srtp_hmac_mbedtls_compute,
- srtp_hmac_mbedtls_update,
- srtp_hmac_mbedtls_start,
- srtp_hmac_mbedtls_description,
- &srtp_hmac_test_case_0,
- SRTP_HMAC_SHA1
- };
|