|  | /* | 
|  | *  HMAC_DRBG implementation (NIST SP 800-90) | 
|  | * | 
|  | *  Copyright The Mbed TLS Contributors | 
|  | *  SPDX-License-Identifier: Apache-2.0 | 
|  | * | 
|  | *  Licensed under the Apache License, Version 2.0 (the "License"); you may | 
|  | *  not use this file except in compliance with the License. | 
|  | *  You may obtain a copy of the License at | 
|  | * | 
|  | *  http://www.apache.org/licenses/LICENSE-2.0 | 
|  | * | 
|  | *  Unless required by applicable law or agreed to in writing, software | 
|  | *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | 
|  | *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | *  See the License for the specific language governing permissions and | 
|  | *  limitations under the License. | 
|  | */ | 
|  |  | 
|  | /* | 
|  | *  The NIST SP 800-90A DRBGs are described in the following publication. | 
|  | *  http://csrc.nist.gov/publications/nistpubs/800-90A/SP800-90A.pdf | 
|  | *  References below are based on rev. 1 (January 2012). | 
|  | */ | 
|  |  | 
|  | #include "common.h" | 
|  |  | 
|  | #if defined(MBEDTLS_HMAC_DRBG_C) | 
|  |  | 
|  | #include "mbedtls/hmac_drbg.h" | 
|  | #include "mbedtls/platform_util.h" | 
|  | #include "mbedtls/error.h" | 
|  |  | 
|  | #include <string.h> | 
|  |  | 
|  | #if defined(MBEDTLS_FS_IO) | 
|  | #include <stdio.h> | 
|  | #endif | 
|  |  | 
|  | #include "mbedtls/platform.h" | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG context initialization | 
|  | */ | 
|  | void mbedtls_hmac_drbg_init(mbedtls_hmac_drbg_context *ctx) | 
|  | { | 
|  | memset(ctx, 0, sizeof(mbedtls_hmac_drbg_context)); | 
|  |  | 
|  | ctx->reseed_interval = MBEDTLS_HMAC_DRBG_RESEED_INTERVAL; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG update, using optional additional data (10.1.2.2) | 
|  | */ | 
|  | int mbedtls_hmac_drbg_update(mbedtls_hmac_drbg_context *ctx, | 
|  | const unsigned char *additional, | 
|  | size_t add_len) | 
|  | { | 
|  | size_t md_len = mbedtls_md_get_size(ctx->md_ctx.md_info); | 
|  | unsigned char rounds = (additional != NULL && add_len != 0) ? 2 : 1; | 
|  | unsigned char sep[1]; | 
|  | unsigned char K[MBEDTLS_MD_MAX_SIZE]; | 
|  | int ret = MBEDTLS_ERR_MD_BAD_INPUT_DATA; | 
|  |  | 
|  | for (sep[0] = 0; sep[0] < rounds; sep[0]++) { | 
|  | /* Step 1 or 4 */ | 
|  | if ((ret = mbedtls_md_hmac_reset(&ctx->md_ctx)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_update(&ctx->md_ctx, | 
|  | ctx->V, md_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_update(&ctx->md_ctx, | 
|  | sep, 1)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if (rounds == 2) { | 
|  | if ((ret = mbedtls_md_hmac_update(&ctx->md_ctx, | 
|  | additional, add_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_finish(&ctx->md_ctx, K)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | /* Step 2 or 5 */ | 
|  | if ((ret = mbedtls_md_hmac_starts(&ctx->md_ctx, K, md_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_update(&ctx->md_ctx, | 
|  | ctx->V, md_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_finish(&ctx->md_ctx, ctx->V)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | } | 
|  |  | 
|  | exit: | 
|  | mbedtls_platform_zeroize(K, sizeof(K)); | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Simplified HMAC_DRBG initialisation (for use with deterministic ECDSA) | 
|  | */ | 
|  | int mbedtls_hmac_drbg_seed_buf(mbedtls_hmac_drbg_context *ctx, | 
|  | const mbedtls_md_info_t *md_info, | 
|  | const unsigned char *data, size_t data_len) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  |  | 
|  | if ((ret = mbedtls_md_setup(&ctx->md_ctx, md_info, 1)) != 0) { | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | #if defined(MBEDTLS_THREADING_C) | 
|  | mbedtls_mutex_init(&ctx->mutex); | 
|  | #endif | 
|  |  | 
|  | /* | 
|  | * Set initial working state. | 
|  | * Use the V memory location, which is currently all 0, to initialize the | 
|  | * MD context with an all-zero key. Then set V to its initial value. | 
|  | */ | 
|  | if ((ret = mbedtls_md_hmac_starts(&ctx->md_ctx, ctx->V, | 
|  | mbedtls_md_get_size(md_info))) != 0) { | 
|  | return ret; | 
|  | } | 
|  | memset(ctx->V, 0x01, mbedtls_md_get_size(md_info)); | 
|  |  | 
|  | if ((ret = mbedtls_hmac_drbg_update(ctx, data, data_len)) != 0) { | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Internal function used both for seeding and reseeding the DRBG. | 
|  | * Comments starting with arabic numbers refer to section 10.1.2.4 | 
|  | * of SP800-90A, while roman numbers refer to section 9.2. | 
|  | */ | 
|  | static int hmac_drbg_reseed_core(mbedtls_hmac_drbg_context *ctx, | 
|  | const unsigned char *additional, size_t len, | 
|  | int use_nonce) | 
|  | { | 
|  | unsigned char seed[MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT]; | 
|  | size_t seedlen = 0; | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  |  | 
|  | { | 
|  | size_t total_entropy_len; | 
|  |  | 
|  | if (use_nonce == 0) { | 
|  | total_entropy_len = ctx->entropy_len; | 
|  | } else { | 
|  | total_entropy_len = ctx->entropy_len * 3 / 2; | 
|  | } | 
|  |  | 
|  | /* III. Check input length */ | 
|  | if (len > MBEDTLS_HMAC_DRBG_MAX_INPUT || | 
|  | total_entropy_len + len > MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG; | 
|  | } | 
|  | } | 
|  |  | 
|  | memset(seed, 0, MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT); | 
|  |  | 
|  | /* IV. Gather entropy_len bytes of entropy for the seed */ | 
|  | if ((ret = ctx->f_entropy(ctx->p_entropy, | 
|  | seed, ctx->entropy_len)) != 0) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_ENTROPY_SOURCE_FAILED; | 
|  | } | 
|  | seedlen += ctx->entropy_len; | 
|  |  | 
|  | /* For initial seeding, allow adding of nonce generated | 
|  | * from the entropy source. See Sect 8.6.7 in SP800-90A. */ | 
|  | if (use_nonce) { | 
|  | /* Note: We don't merge the two calls to f_entropy() in order | 
|  | *       to avoid requesting too much entropy from f_entropy() | 
|  | *       at once. Specifically, if the underlying digest is not | 
|  | *       SHA-1, 3 / 2 * entropy_len is at least 36 Bytes, which | 
|  | *       is larger than the maximum of 32 Bytes that our own | 
|  | *       entropy source implementation can emit in a single | 
|  | *       call in configurations disabling SHA-512. */ | 
|  | if ((ret = ctx->f_entropy(ctx->p_entropy, | 
|  | seed + seedlen, | 
|  | ctx->entropy_len / 2)) != 0) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_ENTROPY_SOURCE_FAILED; | 
|  | } | 
|  |  | 
|  | seedlen += ctx->entropy_len / 2; | 
|  | } | 
|  |  | 
|  |  | 
|  | /* 1. Concatenate entropy and additional data if any */ | 
|  | if (additional != NULL && len != 0) { | 
|  | memcpy(seed + seedlen, additional, len); | 
|  | seedlen += len; | 
|  | } | 
|  |  | 
|  | /* 2. Update state */ | 
|  | if ((ret = mbedtls_hmac_drbg_update(ctx, seed, seedlen)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | /* 3. Reset reseed_counter */ | 
|  | ctx->reseed_counter = 1; | 
|  |  | 
|  | exit: | 
|  | /* 4. Done */ | 
|  | mbedtls_platform_zeroize(seed, seedlen); | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG reseeding: 10.1.2.4 + 9.2 | 
|  | */ | 
|  | int mbedtls_hmac_drbg_reseed(mbedtls_hmac_drbg_context *ctx, | 
|  | const unsigned char *additional, size_t len) | 
|  | { | 
|  | return hmac_drbg_reseed_core(ctx, additional, len, 0); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG initialisation (10.1.2.3 + 9.1) | 
|  | * | 
|  | * The nonce is not passed as a separate parameter but extracted | 
|  | * from the entropy source as suggested in 8.6.7. | 
|  | */ | 
|  | int mbedtls_hmac_drbg_seed(mbedtls_hmac_drbg_context *ctx, | 
|  | const mbedtls_md_info_t *md_info, | 
|  | int (*f_entropy)(void *, unsigned char *, size_t), | 
|  | void *p_entropy, | 
|  | const unsigned char *custom, | 
|  | size_t len) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  | size_t md_size; | 
|  |  | 
|  | if ((ret = mbedtls_md_setup(&ctx->md_ctx, md_info, 1)) != 0) { | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* The mutex is initialized iff the md context is set up. */ | 
|  | #if defined(MBEDTLS_THREADING_C) | 
|  | mbedtls_mutex_init(&ctx->mutex); | 
|  | #endif | 
|  |  | 
|  | md_size = mbedtls_md_get_size(md_info); | 
|  |  | 
|  | /* | 
|  | * Set initial working state. | 
|  | * Use the V memory location, which is currently all 0, to initialize the | 
|  | * MD context with an all-zero key. Then set V to its initial value. | 
|  | */ | 
|  | if ((ret = mbedtls_md_hmac_starts(&ctx->md_ctx, ctx->V, md_size)) != 0) { | 
|  | return ret; | 
|  | } | 
|  | memset(ctx->V, 0x01, md_size); | 
|  |  | 
|  | ctx->f_entropy = f_entropy; | 
|  | ctx->p_entropy = p_entropy; | 
|  |  | 
|  | if (ctx->entropy_len == 0) { | 
|  | /* | 
|  | * See SP800-57 5.6.1 (p. 65-66) for the security strength provided by | 
|  | * each hash function, then according to SP800-90A rev1 10.1 table 2, | 
|  | * min_entropy_len (in bits) is security_strength. | 
|  | * | 
|  | * (This also matches the sizes used in the NIST test vectors.) | 
|  | */ | 
|  | ctx->entropy_len = md_size <= 20 ? 16 : /* 160-bits hash -> 128 bits */ | 
|  | md_size <= 28 ? 24 : /* 224-bits hash -> 192 bits */ | 
|  | 32;  /* better (256+) -> 256 bits */ | 
|  | } | 
|  |  | 
|  | if ((ret = hmac_drbg_reseed_core(ctx, custom, len, | 
|  | 1 /* add nonce */)) != 0) { | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Set prediction resistance | 
|  | */ | 
|  | void mbedtls_hmac_drbg_set_prediction_resistance(mbedtls_hmac_drbg_context *ctx, | 
|  | int resistance) | 
|  | { | 
|  | ctx->prediction_resistance = resistance; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Set entropy length grabbed for seeding | 
|  | */ | 
|  | void mbedtls_hmac_drbg_set_entropy_len(mbedtls_hmac_drbg_context *ctx, size_t len) | 
|  | { | 
|  | ctx->entropy_len = len; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Set reseed interval | 
|  | */ | 
|  | void mbedtls_hmac_drbg_set_reseed_interval(mbedtls_hmac_drbg_context *ctx, int interval) | 
|  | { | 
|  | ctx->reseed_interval = interval; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG random function with optional additional data: | 
|  | * 10.1.2.5 (arabic) + 9.3 (Roman) | 
|  | */ | 
|  | int mbedtls_hmac_drbg_random_with_add(void *p_rng, | 
|  | unsigned char *output, size_t out_len, | 
|  | const unsigned char *additional, size_t add_len) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  | mbedtls_hmac_drbg_context *ctx = (mbedtls_hmac_drbg_context *) p_rng; | 
|  | size_t md_len = mbedtls_md_get_size(ctx->md_ctx.md_info); | 
|  | size_t left = out_len; | 
|  | unsigned char *out = output; | 
|  |  | 
|  | /* II. Check request length */ | 
|  | if (out_len > MBEDTLS_HMAC_DRBG_MAX_REQUEST) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_REQUEST_TOO_BIG; | 
|  | } | 
|  |  | 
|  | /* III. Check input length */ | 
|  | if (add_len > MBEDTLS_HMAC_DRBG_MAX_INPUT) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG; | 
|  | } | 
|  |  | 
|  | /* 1. (aka VII and IX) Check reseed counter and PR */ | 
|  | if (ctx->f_entropy != NULL && /* For no-reseeding instances */ | 
|  | (ctx->prediction_resistance == MBEDTLS_HMAC_DRBG_PR_ON || | 
|  | ctx->reseed_counter > ctx->reseed_interval)) { | 
|  | if ((ret = mbedtls_hmac_drbg_reseed(ctx, additional, add_len)) != 0) { | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | add_len = 0; /* VII.4 */ | 
|  | } | 
|  |  | 
|  | /* 2. Use additional data if any */ | 
|  | if (additional != NULL && add_len != 0) { | 
|  | if ((ret = mbedtls_hmac_drbg_update(ctx, | 
|  | additional, add_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | } | 
|  |  | 
|  | /* 3, 4, 5. Generate bytes */ | 
|  | while (left != 0) { | 
|  | size_t use_len = left > md_len ? md_len : left; | 
|  |  | 
|  | if ((ret = mbedtls_md_hmac_reset(&ctx->md_ctx)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_update(&ctx->md_ctx, | 
|  | ctx->V, md_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  | if ((ret = mbedtls_md_hmac_finish(&ctx->md_ctx, ctx->V)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | memcpy(out, ctx->V, use_len); | 
|  | out += use_len; | 
|  | left -= use_len; | 
|  | } | 
|  |  | 
|  | /* 6. Update */ | 
|  | if ((ret = mbedtls_hmac_drbg_update(ctx, | 
|  | additional, add_len)) != 0) { | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | /* 7. Update reseed counter */ | 
|  | ctx->reseed_counter++; | 
|  |  | 
|  | exit: | 
|  | /* 8. Done */ | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * HMAC_DRBG random function | 
|  | */ | 
|  | int mbedtls_hmac_drbg_random(void *p_rng, unsigned char *output, size_t out_len) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  | mbedtls_hmac_drbg_context *ctx = (mbedtls_hmac_drbg_context *) p_rng; | 
|  |  | 
|  | #if defined(MBEDTLS_THREADING_C) | 
|  | if ((ret = mbedtls_mutex_lock(&ctx->mutex)) != 0) { | 
|  | return ret; | 
|  | } | 
|  | #endif | 
|  |  | 
|  | ret = mbedtls_hmac_drbg_random_with_add(ctx, output, out_len, NULL, 0); | 
|  |  | 
|  | #if defined(MBEDTLS_THREADING_C) | 
|  | if (mbedtls_mutex_unlock(&ctx->mutex) != 0) { | 
|  | return MBEDTLS_ERR_THREADING_MUTEX_ERROR; | 
|  | } | 
|  | #endif | 
|  |  | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | /* | 
|  | *  This function resets HMAC_DRBG context to the state immediately | 
|  | *  after initial call of mbedtls_hmac_drbg_init(). | 
|  | */ | 
|  | void mbedtls_hmac_drbg_free(mbedtls_hmac_drbg_context *ctx) | 
|  | { | 
|  | if (ctx == NULL) { | 
|  | return; | 
|  | } | 
|  |  | 
|  | #if defined(MBEDTLS_THREADING_C) | 
|  | /* The mutex is initialized iff the md context is set up. */ | 
|  | if (ctx->md_ctx.md_info != NULL) { | 
|  | mbedtls_mutex_free(&ctx->mutex); | 
|  | } | 
|  | #endif | 
|  | mbedtls_md_free(&ctx->md_ctx); | 
|  | mbedtls_platform_zeroize(ctx, sizeof(mbedtls_hmac_drbg_context)); | 
|  | ctx->reseed_interval = MBEDTLS_HMAC_DRBG_RESEED_INTERVAL; | 
|  | } | 
|  |  | 
|  | #if defined(MBEDTLS_FS_IO) | 
|  | int mbedtls_hmac_drbg_write_seed_file(mbedtls_hmac_drbg_context *ctx, const char *path) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  | FILE *f; | 
|  | unsigned char buf[MBEDTLS_HMAC_DRBG_MAX_INPUT]; | 
|  |  | 
|  | if ((f = fopen(path, "wb")) == NULL) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR; | 
|  | } | 
|  |  | 
|  | /* Ensure no stdio buffering of secrets, as such buffers cannot be wiped. */ | 
|  | mbedtls_setbuf(f, NULL); | 
|  |  | 
|  | if ((ret = mbedtls_hmac_drbg_random(ctx, buf, sizeof(buf))) != 0) { | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | if (fwrite(buf, 1, sizeof(buf), f) != sizeof(buf)) { | 
|  | ret = MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR; | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | ret = 0; | 
|  |  | 
|  | exit: | 
|  | fclose(f); | 
|  | mbedtls_platform_zeroize(buf, sizeof(buf)); | 
|  |  | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | int mbedtls_hmac_drbg_update_seed_file(mbedtls_hmac_drbg_context *ctx, const char *path) | 
|  | { | 
|  | int ret = 0; | 
|  | FILE *f = NULL; | 
|  | size_t n; | 
|  | unsigned char buf[MBEDTLS_HMAC_DRBG_MAX_INPUT]; | 
|  | unsigned char c; | 
|  |  | 
|  | if ((f = fopen(path, "rb")) == NULL) { | 
|  | return MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR; | 
|  | } | 
|  |  | 
|  | /* Ensure no stdio buffering of secrets, as such buffers cannot be wiped. */ | 
|  | mbedtls_setbuf(f, NULL); | 
|  |  | 
|  | n = fread(buf, 1, sizeof(buf), f); | 
|  | if (fread(&c, 1, 1, f) != 0) { | 
|  | ret = MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG; | 
|  | goto exit; | 
|  | } | 
|  | if (n == 0 || ferror(f)) { | 
|  | ret = MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR; | 
|  | goto exit; | 
|  | } | 
|  | fclose(f); | 
|  | f = NULL; | 
|  |  | 
|  | ret = mbedtls_hmac_drbg_update(ctx, buf, n); | 
|  |  | 
|  | exit: | 
|  | mbedtls_platform_zeroize(buf, sizeof(buf)); | 
|  | if (f != NULL) { | 
|  | fclose(f); | 
|  | } | 
|  | if (ret != 0) { | 
|  | return ret; | 
|  | } | 
|  | return mbedtls_hmac_drbg_write_seed_file(ctx, path); | 
|  | } | 
|  | #endif /* MBEDTLS_FS_IO */ | 
|  |  | 
|  |  | 
|  | #if defined(MBEDTLS_SELF_TEST) | 
|  |  | 
|  | #if !defined(MBEDTLS_MD_CAN_SHA1) | 
|  | /* Dummy checkup routine */ | 
|  | int mbedtls_hmac_drbg_self_test(int verbose) | 
|  | { | 
|  | (void) verbose; | 
|  | return 0; | 
|  | } | 
|  | #else | 
|  |  | 
|  | #define OUTPUT_LEN  80 | 
|  |  | 
|  | /* From a NIST PR=true test vector */ | 
|  | static const unsigned char entropy_pr[] = { | 
|  | 0xa0, 0xc9, 0xab, 0x58, 0xf1, 0xe2, 0xe5, 0xa4, 0xde, 0x3e, 0xbd, 0x4f, | 
|  | 0xf7, 0x3e, 0x9c, 0x5b, 0x64, 0xef, 0xd8, 0xca, 0x02, 0x8c, 0xf8, 0x11, | 
|  | 0x48, 0xa5, 0x84, 0xfe, 0x69, 0xab, 0x5a, 0xee, 0x42, 0xaa, 0x4d, 0x42, | 
|  | 0x17, 0x60, 0x99, 0xd4, 0x5e, 0x13, 0x97, 0xdc, 0x40, 0x4d, 0x86, 0xa3, | 
|  | 0x7b, 0xf5, 0x59, 0x54, 0x75, 0x69, 0x51, 0xe4 | 
|  | }; | 
|  | static const unsigned char result_pr[OUTPUT_LEN] = { | 
|  | 0x9a, 0x00, 0xa2, 0xd0, 0x0e, 0xd5, 0x9b, 0xfe, 0x31, 0xec, 0xb1, 0x39, | 
|  | 0x9b, 0x60, 0x81, 0x48, 0xd1, 0x96, 0x9d, 0x25, 0x0d, 0x3c, 0x1e, 0x94, | 
|  | 0x10, 0x10, 0x98, 0x12, 0x93, 0x25, 0xca, 0xb8, 0xfc, 0xcc, 0x2d, 0x54, | 
|  | 0x73, 0x19, 0x70, 0xc0, 0x10, 0x7a, 0xa4, 0x89, 0x25, 0x19, 0x95, 0x5e, | 
|  | 0x4b, 0xc6, 0x00, 0x1d, 0x7f, 0x4e, 0x6a, 0x2b, 0xf8, 0xa3, 0x01, 0xab, | 
|  | 0x46, 0x05, 0x5c, 0x09, 0xa6, 0x71, 0x88, 0xf1, 0xa7, 0x40, 0xee, 0xf3, | 
|  | 0xe1, 0x5c, 0x02, 0x9b, 0x44, 0xaf, 0x03, 0x44 | 
|  | }; | 
|  |  | 
|  | /* From a NIST PR=false test vector */ | 
|  | static const unsigned char entropy_nopr[] = { | 
|  | 0x79, 0x34, 0x9b, 0xbf, 0x7c, 0xdd, 0xa5, 0x79, 0x95, 0x57, 0x86, 0x66, | 
|  | 0x21, 0xc9, 0x13, 0x83, 0x11, 0x46, 0x73, 0x3a, 0xbf, 0x8c, 0x35, 0xc8, | 
|  | 0xc7, 0x21, 0x5b, 0x5b, 0x96, 0xc4, 0x8e, 0x9b, 0x33, 0x8c, 0x74, 0xe3, | 
|  | 0xe9, 0x9d, 0xfe, 0xdf | 
|  | }; | 
|  | static const unsigned char result_nopr[OUTPUT_LEN] = { | 
|  | 0xc6, 0xa1, 0x6a, 0xb8, 0xd4, 0x20, 0x70, 0x6f, 0x0f, 0x34, 0xab, 0x7f, | 
|  | 0xec, 0x5a, 0xdc, 0xa9, 0xd8, 0xca, 0x3a, 0x13, 0x3e, 0x15, 0x9c, 0xa6, | 
|  | 0xac, 0x43, 0xc6, 0xf8, 0xa2, 0xbe, 0x22, 0x83, 0x4a, 0x4c, 0x0a, 0x0a, | 
|  | 0xff, 0xb1, 0x0d, 0x71, 0x94, 0xf1, 0xc1, 0xa5, 0xcf, 0x73, 0x22, 0xec, | 
|  | 0x1a, 0xe0, 0x96, 0x4e, 0xd4, 0xbf, 0x12, 0x27, 0x46, 0xe0, 0x87, 0xfd, | 
|  | 0xb5, 0xb3, 0xe9, 0x1b, 0x34, 0x93, 0xd5, 0xbb, 0x98, 0xfa, 0xed, 0x49, | 
|  | 0xe8, 0x5f, 0x13, 0x0f, 0xc8, 0xa4, 0x59, 0xb7 | 
|  | }; | 
|  |  | 
|  | /* "Entropy" from buffer */ | 
|  | static size_t test_offset; | 
|  | static int hmac_drbg_self_test_entropy(void *data, | 
|  | unsigned char *buf, size_t len) | 
|  | { | 
|  | const unsigned char *p = data; | 
|  | memcpy(buf, p + test_offset, len); | 
|  | test_offset += len; | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | #define CHK(c)    if ((c) != 0)                          \ | 
|  | {                                       \ | 
|  | if (verbose != 0)                  \ | 
|  | mbedtls_printf("failed\n");  \ | 
|  | return 1;                        \ | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Checkup routine for HMAC_DRBG with SHA-1 | 
|  | */ | 
|  | int mbedtls_hmac_drbg_self_test(int verbose) | 
|  | { | 
|  | mbedtls_hmac_drbg_context ctx; | 
|  | unsigned char buf[OUTPUT_LEN]; | 
|  | const mbedtls_md_info_t *md_info = mbedtls_md_info_from_type(MBEDTLS_MD_SHA1); | 
|  |  | 
|  | mbedtls_hmac_drbg_init(&ctx); | 
|  |  | 
|  | /* | 
|  | * PR = True | 
|  | */ | 
|  | if (verbose != 0) { | 
|  | mbedtls_printf("  HMAC_DRBG (PR = True) : "); | 
|  | } | 
|  |  | 
|  | test_offset = 0; | 
|  | CHK(mbedtls_hmac_drbg_seed(&ctx, md_info, | 
|  | hmac_drbg_self_test_entropy, (void *) entropy_pr, | 
|  | NULL, 0)); | 
|  | mbedtls_hmac_drbg_set_prediction_resistance(&ctx, MBEDTLS_HMAC_DRBG_PR_ON); | 
|  | CHK(mbedtls_hmac_drbg_random(&ctx, buf, OUTPUT_LEN)); | 
|  | CHK(mbedtls_hmac_drbg_random(&ctx, buf, OUTPUT_LEN)); | 
|  | CHK(memcmp(buf, result_pr, OUTPUT_LEN)); | 
|  | mbedtls_hmac_drbg_free(&ctx); | 
|  |  | 
|  | mbedtls_hmac_drbg_free(&ctx); | 
|  |  | 
|  | if (verbose != 0) { | 
|  | mbedtls_printf("passed\n"); | 
|  | } | 
|  |  | 
|  | /* | 
|  | * PR = False | 
|  | */ | 
|  | if (verbose != 0) { | 
|  | mbedtls_printf("  HMAC_DRBG (PR = False) : "); | 
|  | } | 
|  |  | 
|  | mbedtls_hmac_drbg_init(&ctx); | 
|  |  | 
|  | test_offset = 0; | 
|  | CHK(mbedtls_hmac_drbg_seed(&ctx, md_info, | 
|  | hmac_drbg_self_test_entropy, (void *) entropy_nopr, | 
|  | NULL, 0)); | 
|  | CHK(mbedtls_hmac_drbg_reseed(&ctx, NULL, 0)); | 
|  | CHK(mbedtls_hmac_drbg_random(&ctx, buf, OUTPUT_LEN)); | 
|  | CHK(mbedtls_hmac_drbg_random(&ctx, buf, OUTPUT_LEN)); | 
|  | CHK(memcmp(buf, result_nopr, OUTPUT_LEN)); | 
|  | mbedtls_hmac_drbg_free(&ctx); | 
|  |  | 
|  | mbedtls_hmac_drbg_free(&ctx); | 
|  |  | 
|  | if (verbose != 0) { | 
|  | mbedtls_printf("passed\n"); | 
|  | } | 
|  |  | 
|  | if (verbose != 0) { | 
|  | mbedtls_printf("\n"); | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  | #endif /* MBEDTLS_MD_CAN_SHA1 */ | 
|  | #endif /* MBEDTLS_SELF_TEST */ | 
|  |  | 
|  | #endif /* MBEDTLS_HMAC_DRBG_C */ |