blob: e17512e779abc452c8d40d0d3e9e659679e99d58 [file] [log] [blame]
Paul Bakker6083fd22011-12-03 21:45:14 +00001/*
2 * Entropy accumulator implementation
3 *
Paul Bakker9988d6b2016-06-01 11:29:42 +01004 * Copyright (C) 2006-2016, ARM Limited, All Rights Reserved
Manuel Pégourié-Gonnard37ff1402015-09-04 14:21:07 +02005 * SPDX-License-Identifier: Apache-2.0
6 *
7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
8 * not use this file except in compliance with the License.
9 * You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
Paul Bakker6083fd22011-12-03 21:45:14 +000018 *
Manuel Pégourié-Gonnardfe446432015-03-06 13:17:10 +000019 * This file is part of mbed TLS (https://tls.mbed.org)
Paul Bakker6083fd22011-12-03 21:45:14 +000020 */
21
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020022#if !defined(MBEDTLS_CONFIG_FILE)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000023#include "mbedtls/config.h"
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020024#else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020025#include MBEDTLS_CONFIG_FILE
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020026#endif
Paul Bakker6083fd22011-12-03 21:45:14 +000027
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020028#if defined(MBEDTLS_ENTROPY_C)
Paul Bakker6083fd22011-12-03 21:45:14 +000029
Simon Butcherab5df402016-06-11 02:31:21 +010030#if defined(MBEDTLS_TEST_NULL_ENTROPY)
Simon Butcher1ceab6e2016-06-21 10:14:00 +010031#warning "**** WARNING! MBEDTLS_TEST_NULL_ENTROPY defined! "
32#warning "**** THIS BUILD HAS NO DEFINED ENTROPY SOURCES "
33#warning "**** THIS BUILD IS *NOT* SUITABLE FOR PRODUCTION USE "
Simon Butcherab5df402016-06-11 02:31:21 +010034#endif
35
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000036#include "mbedtls/entropy.h"
37#include "mbedtls/entropy_poll.h"
Paul Bakker6083fd22011-12-03 21:45:14 +000038
Rich Evans00ab4702015-02-06 13:43:58 +000039#include <string.h>
40
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020041#if defined(MBEDTLS_FS_IO)
Paul Bakker66ff70d2014-03-26 11:54:05 +010042#include <stdio.h>
43#endif
44
Paul Bakker217efbc2016-07-14 14:30:03 +010045#if defined(MBEDTLS_ENTROPY_NV_SEED)
46#include "mbedtls/platform.h"
47#endif
48
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020049#if defined(MBEDTLS_SELF_TEST)
50#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000051#include "mbedtls/platform.h"
Rich Evans00ab4702015-02-06 13:43:58 +000052#else
53#include <stdio.h>
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020054#define mbedtls_printf printf
55#endif /* MBEDTLS_PLATFORM_C */
56#endif /* MBEDTLS_SELF_TEST */
Rich Evans00ab4702015-02-06 13:43:58 +000057
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020058#if defined(MBEDTLS_HAVEGE_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000059#include "mbedtls/havege.h"
Paul Bakker28c7e7f2011-12-15 19:49:30 +000060#endif
61
Paul Bakker34617722014-06-13 17:20:13 +020062/* Implementation that should never be optimized out by the compiler */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020063static void mbedtls_zeroize( void *v, size_t n ) {
Paul Bakker34617722014-06-13 17:20:13 +020064 volatile unsigned char *p = v; while( n-- ) *p++ = 0;
65}
66
Paul Bakker6083fd22011-12-03 21:45:14 +000067#define ENTROPY_MAX_LOOP 256 /**< Maximum amount to loop before error */
68
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020069void mbedtls_entropy_init( mbedtls_entropy_context *ctx )
Paul Bakker6083fd22011-12-03 21:45:14 +000070{
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +010071 ctx->source_count = 0;
72 memset( ctx->source, 0, sizeof( ctx->source ) );
Paul Bakker6083fd22011-12-03 21:45:14 +000073
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020074#if defined(MBEDTLS_THREADING_C)
75 mbedtls_mutex_init( &ctx->mutex );
Paul Bakkerf4e7dc52013-09-28 15:23:57 +020076#endif
77
Andres Amaya Garcia95869c42017-06-29 16:31:44 +010078 ctx->accumulator_started = 0;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020079#if defined(MBEDTLS_ENTROPY_SHA512_ACCUMULATOR)
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +010080 mbedtls_sha512_init( &ctx->accumulator );
Paul Bakkerfb08fd22013-08-27 15:06:26 +020081#else
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +010082 mbedtls_sha256_init( &ctx->accumulator );
Paul Bakkerfb08fd22013-08-27 15:06:26 +020083#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084#if defined(MBEDTLS_HAVEGE_C)
85 mbedtls_havege_init( &ctx->havege_data );
Paul Bakker43655f42011-12-15 20:11:16 +000086#endif
Paul Bakker6083fd22011-12-03 21:45:14 +000087
Hanno Beckerd4a872e2017-09-07 08:09:33 +010088 /* Reminder: Update ENTROPY_HAVE_STRONG in the test files
Hanno Becker47deec42017-07-24 12:27:09 +010089 * when adding more strong entropy sources here. */
Hanno Beckerc6deafc2017-07-23 14:06:42 +010090
Simon Butcherab5df402016-06-11 02:31:21 +010091#if defined(MBEDTLS_TEST_NULL_ENTROPY)
Simon Butcher4157b602016-06-12 00:31:33 +010092 mbedtls_entropy_add_source( ctx, mbedtls_null_entropy_poll, NULL,
Janos Follath53de7842016-06-08 15:29:18 +010093 1, MBEDTLS_ENTROPY_SOURCE_STRONG );
94#endif
95
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020096#if !defined(MBEDTLS_NO_DEFAULT_ENTROPY_SOURCES)
97#if !defined(MBEDTLS_NO_PLATFORM_ENTROPY)
98 mbedtls_entropy_add_source( ctx, mbedtls_platform_entropy_poll, NULL,
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +020099 MBEDTLS_ENTROPY_MIN_PLATFORM,
100 MBEDTLS_ENTROPY_SOURCE_STRONG );
Paul Bakker6083fd22011-12-03 21:45:14 +0000101#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200102#if defined(MBEDTLS_TIMING_C)
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200103 mbedtls_entropy_add_source( ctx, mbedtls_hardclock_poll, NULL,
104 MBEDTLS_ENTROPY_MIN_HARDCLOCK,
105 MBEDTLS_ENTROPY_SOURCE_WEAK );
Paul Bakker6083fd22011-12-03 21:45:14 +0000106#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200107#if defined(MBEDTLS_HAVEGE_C)
108 mbedtls_entropy_add_source( ctx, mbedtls_havege_poll, &ctx->havege_data,
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200109 MBEDTLS_ENTROPY_MIN_HAVEGE,
110 MBEDTLS_ENTROPY_SOURCE_STRONG );
Paul Bakker28c7e7f2011-12-15 19:49:30 +0000111#endif
Manuel Pégourié-Gonnard3f77dfb2015-06-19 10:06:21 +0200112#if defined(MBEDTLS_ENTROPY_HARDWARE_ALT)
Manuel Pégourié-Gonnardfc2ccfe2015-07-10 11:15:50 +0100113 mbedtls_entropy_add_source( ctx, mbedtls_hardware_poll, NULL,
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200114 MBEDTLS_ENTROPY_MIN_HARDWARE,
115 MBEDTLS_ENTROPY_SOURCE_STRONG );
Manuel Pégourié-Gonnard3f77dfb2015-06-19 10:06:21 +0200116#endif
Paul Bakker9988d6b2016-06-01 11:29:42 +0100117#if defined(MBEDTLS_ENTROPY_NV_SEED)
118 mbedtls_entropy_add_source( ctx, mbedtls_nv_seed_poll, NULL,
119 MBEDTLS_ENTROPY_BLOCK_SIZE,
120 MBEDTLS_ENTROPY_SOURCE_STRONG );
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +0100121 ctx->initial_entropy_run = 0;
Paul Bakker9988d6b2016-06-01 11:29:42 +0100122#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200123#endif /* MBEDTLS_NO_DEFAULT_ENTROPY_SOURCES */
Paul Bakker6083fd22011-12-03 21:45:14 +0000124}
125
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200126void mbedtls_entropy_free( mbedtls_entropy_context *ctx )
Paul Bakker1ffefac2013-09-28 15:23:03 +0200127{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200128#if defined(MBEDTLS_HAVEGE_C)
129 mbedtls_havege_free( &ctx->havege_data );
Paul Bakkera317a982014-06-18 16:44:11 +0200130#endif
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200131#if defined(MBEDTLS_THREADING_C)
132 mbedtls_mutex_free( &ctx->mutex );
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200133#endif
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +0100134#if defined(MBEDTLS_ENTROPY_SHA512_ACCUMULATOR)
135 mbedtls_sha512_free( &ctx->accumulator );
136#else
137 mbedtls_sha256_free( &ctx->accumulator );
138#endif
139#if defined(MBEDTLS_ENTROPY_NV_SEED)
140 ctx->initial_entropy_run = 0;
141#endif
142 ctx->source_count = 0;
143 mbedtls_zeroize( ctx->source, sizeof( ctx->source ) );
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100144 ctx->accumulator_started = 0;
Paul Bakker1ffefac2013-09-28 15:23:03 +0200145}
146
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200147int mbedtls_entropy_add_source( mbedtls_entropy_context *ctx,
148 mbedtls_entropy_f_source_ptr f_source, void *p_source,
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200149 size_t threshold, int strong )
Paul Bakker6083fd22011-12-03 21:45:14 +0000150{
Hanno Becker61937d42017-04-26 15:01:23 +0100151 int idx, ret = 0;
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000152
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200153#if defined(MBEDTLS_THREADING_C)
154 if( ( ret = mbedtls_mutex_lock( &ctx->mutex ) ) != 0 )
Paul Bakker47703a02014-02-06 15:01:20 +0100155 return( ret );
156#endif
157
Hanno Becker61937d42017-04-26 15:01:23 +0100158 idx = ctx->source_count;
159 if( idx >= MBEDTLS_ENTROPY_MAX_SOURCES )
Paul Bakker47703a02014-02-06 15:01:20 +0100160 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200161 ret = MBEDTLS_ERR_ENTROPY_MAX_SOURCES;
Paul Bakker47703a02014-02-06 15:01:20 +0100162 goto exit;
163 }
Paul Bakker6083fd22011-12-03 21:45:14 +0000164
Hanno Becker61937d42017-04-26 15:01:23 +0100165 ctx->source[idx].f_source = f_source;
166 ctx->source[idx].p_source = p_source;
167 ctx->source[idx].threshold = threshold;
168 ctx->source[idx].strong = strong;
Paul Bakker6083fd22011-12-03 21:45:14 +0000169
170 ctx->source_count++;
171
Paul Bakker47703a02014-02-06 15:01:20 +0100172exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200173#if defined(MBEDTLS_THREADING_C)
174 if( mbedtls_mutex_unlock( &ctx->mutex ) != 0 )
175 return( MBEDTLS_ERR_THREADING_MUTEX_ERROR );
Paul Bakker47703a02014-02-06 15:01:20 +0100176#endif
177
178 return( ret );
Paul Bakker6083fd22011-12-03 21:45:14 +0000179}
180
181/*
182 * Entropy accumulator update
183 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200184static int entropy_update( mbedtls_entropy_context *ctx, unsigned char source_id,
Paul Bakkerb6c5d2e2013-06-25 16:25:17 +0200185 const unsigned char *data, size_t len )
Paul Bakker6083fd22011-12-03 21:45:14 +0000186{
187 unsigned char header[2];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200188 unsigned char tmp[MBEDTLS_ENTROPY_BLOCK_SIZE];
Paul Bakker6083fd22011-12-03 21:45:14 +0000189 size_t use_len = len;
190 const unsigned char *p = data;
Jaeden Amero66954e12018-01-25 16:05:54 +0000191 int ret = 0;
Paul Bakkerb6c5d2e2013-06-25 16:25:17 +0200192
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200193 if( use_len > MBEDTLS_ENTROPY_BLOCK_SIZE )
Paul Bakker6083fd22011-12-03 21:45:14 +0000194 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200195#if defined(MBEDTLS_ENTROPY_SHA512_ACCUMULATOR)
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100196 if( ( ret = mbedtls_sha512_ret( data, len, tmp, 0 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000197 goto cleanup;
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200198#else
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100199 if( ( ret = mbedtls_sha256_ret( data, len, tmp, 0 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000200 goto cleanup;
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200201#endif
Paul Bakker6083fd22011-12-03 21:45:14 +0000202 p = tmp;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200203 use_len = MBEDTLS_ENTROPY_BLOCK_SIZE;
Paul Bakker6083fd22011-12-03 21:45:14 +0000204 }
205
206 header[0] = source_id;
207 header[1] = use_len & 0xFF;
208
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100209 /*
210 * Start the accumulator if this has not already happened. Note that
211 * it is sufficient to start the accumulator here only because all calls to
212 * gather entropy eventually execute this code.
213 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200214#if defined(MBEDTLS_ENTROPY_SHA512_ACCUMULATOR)
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100215 if( ctx->accumulator_started == 0 &&
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100216 ( ret = mbedtls_sha512_starts_ret( &ctx->accumulator, 0 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000217 goto cleanup;
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100218 else
219 ctx->accumulator_started = 1;
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100220 if( ( ret = mbedtls_sha512_update_ret( &ctx->accumulator, header, 2 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000221 goto cleanup;
222 ret = mbedtls_sha512_update_ret( &ctx->accumulator, p, use_len );
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200223#else
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100224 if( ctx->accumulator_started == 0 &&
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100225 ( ret = mbedtls_sha256_starts_ret( &ctx->accumulator, 0 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000226 goto cleanup;
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100227 else
228 ctx->accumulator_started = 1;
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100229 if( ( ret = mbedtls_sha256_update_ret( &ctx->accumulator, header, 2 ) ) != 0 )
Jaeden Amero66954e12018-01-25 16:05:54 +0000230 goto cleanup;
231 ret = mbedtls_sha256_update_ret( &ctx->accumulator, p, use_len );
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200232#endif
Paul Bakkerb6c5d2e2013-06-25 16:25:17 +0200233
Jaeden Amero66954e12018-01-25 16:05:54 +0000234cleanup:
Andres Amaya Garcia65121932017-07-05 15:45:47 +0100235 mbedtls_zeroize( tmp, sizeof( tmp ) );
236
Jaeden Amero66954e12018-01-25 16:05:54 +0000237 return( ret );
Paul Bakker6083fd22011-12-03 21:45:14 +0000238}
239
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200240int mbedtls_entropy_update_manual( mbedtls_entropy_context *ctx,
Paul Bakker6083fd22011-12-03 21:45:14 +0000241 const unsigned char *data, size_t len )
242{
Paul Bakker47703a02014-02-06 15:01:20 +0100243 int ret;
244
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200245#if defined(MBEDTLS_THREADING_C)
246 if( ( ret = mbedtls_mutex_lock( &ctx->mutex ) ) != 0 )
Paul Bakker47703a02014-02-06 15:01:20 +0100247 return( ret );
248#endif
249
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200250 ret = entropy_update( ctx, MBEDTLS_ENTROPY_SOURCE_MANUAL, data, len );
Paul Bakker47703a02014-02-06 15:01:20 +0100251
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200252#if defined(MBEDTLS_THREADING_C)
253 if( mbedtls_mutex_unlock( &ctx->mutex ) != 0 )
254 return( MBEDTLS_ERR_THREADING_MUTEX_ERROR );
Paul Bakker47703a02014-02-06 15:01:20 +0100255#endif
256
Paul Bakkerd8bb8262014-06-17 14:06:49 +0200257 return( ret );
Paul Bakker6083fd22011-12-03 21:45:14 +0000258}
259
260/*
261 * Run through the different sources to add entropy to our accumulator
262 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200263static int entropy_gather_internal( mbedtls_entropy_context *ctx )
Paul Bakker6083fd22011-12-03 21:45:14 +0000264{
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200265 int ret, i, have_one_strong = 0;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200266 unsigned char buf[MBEDTLS_ENTROPY_MAX_GATHER];
Paul Bakker6083fd22011-12-03 21:45:14 +0000267 size_t olen;
Paul Bakker47703a02014-02-06 15:01:20 +0100268
Paul Bakker43655f42011-12-15 20:11:16 +0000269 if( ctx->source_count == 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200270 return( MBEDTLS_ERR_ENTROPY_NO_SOURCES_DEFINED );
Paul Bakker43655f42011-12-15 20:11:16 +0000271
Paul Bakker6083fd22011-12-03 21:45:14 +0000272 /*
273 * Run through our entropy sources
274 */
275 for( i = 0; i < ctx->source_count; i++ )
276 {
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200277 if( ctx->source[i].strong == MBEDTLS_ENTROPY_SOURCE_STRONG )
278 have_one_strong = 1;
279
Paul Bakker6083fd22011-12-03 21:45:14 +0000280 olen = 0;
Paul Bakker66d5d072014-06-17 16:39:18 +0200281 if( ( ret = ctx->source[i].f_source( ctx->source[i].p_source,
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200282 buf, MBEDTLS_ENTROPY_MAX_GATHER, &olen ) ) != 0 )
Paul Bakker6083fd22011-12-03 21:45:14 +0000283 {
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100284 goto cleanup;
Paul Bakker6083fd22011-12-03 21:45:14 +0000285 }
286
287 /*
288 * Add if we actually gathered something
289 */
290 if( olen > 0 )
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000291 {
Andres Amaya Garcia95869c42017-06-29 16:31:44 +0100292 if( ( ret = entropy_update( ctx, (unsigned char) i,
293 buf, olen ) ) != 0 )
294 return( ret );
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000295 ctx->source[i].size += olen;
296 }
Paul Bakker6083fd22011-12-03 21:45:14 +0000297 }
298
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200299 if( have_one_strong == 0 )
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100300 ret = MBEDTLS_ERR_ENTROPY_NO_STRONG_SOURCE;
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200301
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100302cleanup:
303 mbedtls_zeroize( buf, sizeof( buf ) );
304
305 return( ret );
Paul Bakker6083fd22011-12-03 21:45:14 +0000306}
307
Paul Bakker47703a02014-02-06 15:01:20 +0100308/*
309 * Thread-safe wrapper for entropy_gather_internal()
310 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200311int mbedtls_entropy_gather( mbedtls_entropy_context *ctx )
Paul Bakker47703a02014-02-06 15:01:20 +0100312{
Paul Bakkerddd427a2014-04-09 14:47:58 +0200313 int ret;
Paul Bakker47703a02014-02-06 15:01:20 +0100314
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200315#if defined(MBEDTLS_THREADING_C)
316 if( ( ret = mbedtls_mutex_lock( &ctx->mutex ) ) != 0 )
Paul Bakkerddd427a2014-04-09 14:47:58 +0200317 return( ret );
Paul Bakker47703a02014-02-06 15:01:20 +0100318#endif
319
Paul Bakkerddd427a2014-04-09 14:47:58 +0200320 ret = entropy_gather_internal( ctx );
Paul Bakker47703a02014-02-06 15:01:20 +0100321
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200322#if defined(MBEDTLS_THREADING_C)
323 if( mbedtls_mutex_unlock( &ctx->mutex ) != 0 )
324 return( MBEDTLS_ERR_THREADING_MUTEX_ERROR );
Paul Bakker47703a02014-02-06 15:01:20 +0100325#endif
326
Paul Bakkerddd427a2014-04-09 14:47:58 +0200327 return( ret );
Paul Bakker47703a02014-02-06 15:01:20 +0100328}
329
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200330int mbedtls_entropy_func( void *data, unsigned char *output, size_t len )
Paul Bakker6083fd22011-12-03 21:45:14 +0000331{
Manuel Pégourié-Gonnardbf82ff02015-06-19 09:40:51 +0200332 int ret, count = 0, i, done;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200333 mbedtls_entropy_context *ctx = (mbedtls_entropy_context *) data;
334 unsigned char buf[MBEDTLS_ENTROPY_BLOCK_SIZE];
Paul Bakker6083fd22011-12-03 21:45:14 +0000335
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200336 if( len > MBEDTLS_ENTROPY_BLOCK_SIZE )
337 return( MBEDTLS_ERR_ENTROPY_SOURCE_FAILED );
Paul Bakker6083fd22011-12-03 21:45:14 +0000338
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100339#if defined(MBEDTLS_ENTROPY_NV_SEED)
340 /* Update the NV entropy seed before generating any entropy for outside
341 * use.
342 */
Paul Bakkerfc9c7c82016-06-01 15:25:50 +0100343 if( ctx->initial_entropy_run == 0 )
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100344 {
Paul Bakkerfc9c7c82016-06-01 15:25:50 +0100345 ctx->initial_entropy_run = 1;
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100346 if( ( ret = mbedtls_entropy_update_nv_seed( ctx ) ) != 0 )
347 return( ret );
348 }
349#endif
350
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200351#if defined(MBEDTLS_THREADING_C)
352 if( ( ret = mbedtls_mutex_lock( &ctx->mutex ) ) != 0 )
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200353 return( ret );
354#endif
355
Paul Bakker6083fd22011-12-03 21:45:14 +0000356 /*
357 * Always gather extra entropy before a call
358 */
359 do
360 {
361 if( count++ > ENTROPY_MAX_LOOP )
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200362 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200363 ret = MBEDTLS_ERR_ENTROPY_SOURCE_FAILED;
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200364 goto exit;
365 }
Paul Bakker6083fd22011-12-03 21:45:14 +0000366
Paul Bakker47703a02014-02-06 15:01:20 +0100367 if( ( ret = entropy_gather_internal( ctx ) ) != 0 )
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200368 goto exit;
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000369
Manuel Pégourié-Gonnardbf82ff02015-06-19 09:40:51 +0200370 done = 1;
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000371 for( i = 0; i < ctx->source_count; i++ )
Manuel Pégourié-Gonnardbf82ff02015-06-19 09:40:51 +0200372 if( ctx->source[i].size < ctx->source[i].threshold )
373 done = 0;
Paul Bakker6083fd22011-12-03 21:45:14 +0000374 }
Manuel Pégourié-Gonnardbf82ff02015-06-19 09:40:51 +0200375 while( ! done );
Paul Bakker6083fd22011-12-03 21:45:14 +0000376
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200377 memset( buf, 0, MBEDTLS_ENTROPY_BLOCK_SIZE );
Paul Bakker6083fd22011-12-03 21:45:14 +0000378
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200379#if defined(MBEDTLS_ENTROPY_SHA512_ACCUMULATOR)
Andres Amaya Garciab2b063f2017-07-20 16:45:24 +0100380 /*
381 * Note that at this stage it is assumed that the accumulator was started
382 * in a previous call to entropy_update(). If this is not guaranteed, the
383 * code below will fail.
384 */
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100385 if( ( ret = mbedtls_sha512_finish_ret( &ctx->accumulator, buf ) ) != 0 )
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100386 goto exit;
Paul Bakker9e36f042013-06-30 14:34:05 +0200387
Paul Bakker6083fd22011-12-03 21:45:14 +0000388 /*
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000389 * Reset accumulator and counters and recycle existing entropy
Paul Bakker6083fd22011-12-03 21:45:14 +0000390 */
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +0100391 mbedtls_sha512_free( &ctx->accumulator );
392 mbedtls_sha512_init( &ctx->accumulator );
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100393 if( ( ret = mbedtls_sha512_starts_ret( &ctx->accumulator, 0 ) ) != 0 )
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100394 goto exit;
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100395 if( ( ret = mbedtls_sha512_update_ret( &ctx->accumulator, buf,
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100396 MBEDTLS_ENTROPY_BLOCK_SIZE ) ) != 0 )
397 goto exit;
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200398
399 /*
Paul Bakkerb13d3ff2014-03-26 12:51:25 +0100400 * Perform second SHA-512 on entropy
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200401 */
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100402 if( ( ret = mbedtls_sha512_ret( buf, MBEDTLS_ENTROPY_BLOCK_SIZE,
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100403 buf, 0 ) ) != 0 )
404 goto exit;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200405#else /* MBEDTLS_ENTROPY_SHA512_ACCUMULATOR */
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100406 if( ( ret = mbedtls_sha256_finish_ret( &ctx->accumulator, buf ) ) != 0 )
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100407 goto exit;
Paul Bakkerfb08fd22013-08-27 15:06:26 +0200408
409 /*
410 * Reset accumulator and counters and recycle existing entropy
411 */
Andres Amaya Garciaa7559cb2017-06-29 16:12:31 +0100412 mbedtls_sha256_free( &ctx->accumulator );
413 mbedtls_sha256_init( &ctx->accumulator );
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100414 if( ( ret = mbedtls_sha256_starts_ret( &ctx->accumulator, 0 ) ) != 0 )
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100415 goto exit;
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100416 if( ( ret = mbedtls_sha256_update_ret( &ctx->accumulator, buf,
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100417 MBEDTLS_ENTROPY_BLOCK_SIZE ) ) != 0 )
418 goto exit;
Paul Bakkerb13d3ff2014-03-26 12:51:25 +0100419
420 /*
421 * Perform second SHA-256 on entropy
422 */
Gilles Peskine9e4f77c2018-01-22 11:48:08 +0100423 if( ( ret = mbedtls_sha256_ret( buf, MBEDTLS_ENTROPY_BLOCK_SIZE,
Andres Amaya Garcia207cea52017-06-29 13:28:13 +0100424 buf, 0 ) ) != 0 )
425 goto exit;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200426#endif /* MBEDTLS_ENTROPY_SHA512_ACCUMULATOR */
Paul Bakkerbd4a9d02011-12-10 17:02:19 +0000427
428 for( i = 0; i < ctx->source_count; i++ )
429 ctx->source[i].size = 0;
Paul Bakker6083fd22011-12-03 21:45:14 +0000430
431 memcpy( output, buf, len );
432
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200433 ret = 0;
434
435exit:
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100436 mbedtls_zeroize( buf, sizeof( buf ) );
437
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200438#if defined(MBEDTLS_THREADING_C)
439 if( mbedtls_mutex_unlock( &ctx->mutex ) != 0 )
440 return( MBEDTLS_ERR_THREADING_MUTEX_ERROR );
Paul Bakkerf4e7dc52013-09-28 15:23:57 +0200441#endif
442
443 return( ret );
Paul Bakker6083fd22011-12-03 21:45:14 +0000444}
445
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100446#if defined(MBEDTLS_ENTROPY_NV_SEED)
447int mbedtls_entropy_update_nv_seed( mbedtls_entropy_context *ctx )
448{
449 int ret = MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR;
Andres Amaya Garciaaf0b31d2017-07-05 14:23:54 +0100450 unsigned char buf[MBEDTLS_ENTROPY_BLOCK_SIZE];
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100451
452 /* Read new seed and write it to NV */
453 if( ( ret = mbedtls_entropy_func( ctx, buf, MBEDTLS_ENTROPY_BLOCK_SIZE ) ) != 0 )
454 return( ret );
455
456 if( mbedtls_nv_seed_write( buf, MBEDTLS_ENTROPY_BLOCK_SIZE ) < 0 )
457 return( MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR );
458
459 /* Manually update the remaining stream with a separator value to diverge */
460 memset( buf, 0, MBEDTLS_ENTROPY_BLOCK_SIZE );
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100461 ret = mbedtls_entropy_update_manual( ctx, buf, MBEDTLS_ENTROPY_BLOCK_SIZE );
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100462
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100463 return( ret );
Paul Bakkerd5c9f6d2016-06-01 11:30:54 +0100464}
465#endif /* MBEDTLS_ENTROPY_NV_SEED */
466
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200467#if defined(MBEDTLS_FS_IO)
468int mbedtls_entropy_write_seed_file( mbedtls_entropy_context *ctx, const char *path )
Paul Bakker66ff70d2014-03-26 11:54:05 +0100469{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200470 int ret = MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR;
Paul Bakker66ff70d2014-03-26 11:54:05 +0100471 FILE *f;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200472 unsigned char buf[MBEDTLS_ENTROPY_BLOCK_SIZE];
Paul Bakker66ff70d2014-03-26 11:54:05 +0100473
474 if( ( f = fopen( path, "wb" ) ) == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200475 return( MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR );
Paul Bakker66ff70d2014-03-26 11:54:05 +0100476
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200477 if( ( ret = mbedtls_entropy_func( ctx, buf, MBEDTLS_ENTROPY_BLOCK_SIZE ) ) != 0 )
Paul Bakker66ff70d2014-03-26 11:54:05 +0100478 goto exit;
479
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200480 if( fwrite( buf, 1, MBEDTLS_ENTROPY_BLOCK_SIZE, f ) != MBEDTLS_ENTROPY_BLOCK_SIZE )
Paul Bakker66ff70d2014-03-26 11:54:05 +0100481 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200482 ret = MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR;
Paul Bakker66ff70d2014-03-26 11:54:05 +0100483 goto exit;
484 }
485
486 ret = 0;
487
488exit:
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100489 mbedtls_zeroize( buf, sizeof( buf ) );
490
Paul Bakker66ff70d2014-03-26 11:54:05 +0100491 fclose( f );
492 return( ret );
493}
494
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200495int mbedtls_entropy_update_seed_file( mbedtls_entropy_context *ctx, const char *path )
Paul Bakker66ff70d2014-03-26 11:54:05 +0100496{
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100497 int ret = 0;
Paul Bakker66ff70d2014-03-26 11:54:05 +0100498 FILE *f;
499 size_t n;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200500 unsigned char buf[ MBEDTLS_ENTROPY_MAX_SEED_SIZE ];
Paul Bakker66ff70d2014-03-26 11:54:05 +0100501
502 if( ( f = fopen( path, "rb" ) ) == NULL )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200503 return( MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR );
Paul Bakker66ff70d2014-03-26 11:54:05 +0100504
505 fseek( f, 0, SEEK_END );
506 n = (size_t) ftell( f );
507 fseek( f, 0, SEEK_SET );
508
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200509 if( n > MBEDTLS_ENTROPY_MAX_SEED_SIZE )
510 n = MBEDTLS_ENTROPY_MAX_SEED_SIZE;
Paul Bakker66ff70d2014-03-26 11:54:05 +0100511
512 if( fread( buf, 1, n, f ) != n )
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100513 ret = MBEDTLS_ERR_ENTROPY_FILE_IO_ERROR;
514 else
515 ret = mbedtls_entropy_update_manual( ctx, buf, n );
Paul Bakker66ff70d2014-03-26 11:54:05 +0100516
517 fclose( f );
518
Andres Amaya Garcia1adcd952017-06-26 09:58:59 +0100519 mbedtls_zeroize( buf, sizeof( buf ) );
520
521 if( ret != 0 )
522 return( ret );
Paul Bakker66ff70d2014-03-26 11:54:05 +0100523
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200524 return( mbedtls_entropy_write_seed_file( ctx, path ) );
Paul Bakker66ff70d2014-03-26 11:54:05 +0100525}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200526#endif /* MBEDTLS_FS_IO */
Paul Bakker66ff70d2014-03-26 11:54:05 +0100527
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200528#if defined(MBEDTLS_SELF_TEST)
Simon Butcher669c6352016-09-15 18:57:34 +0100529#if !defined(MBEDTLS_TEST_NULL_ENTROPY)
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200530/*
531 * Dummy source function
532 */
533static int entropy_dummy_source( void *data, unsigned char *output,
534 size_t len, size_t *olen )
535{
536 ((void) data);
537
538 memset( output, 0x2a, len );
539 *olen = len;
540
541 return( 0 );
542}
Simon Butcher669c6352016-09-15 18:57:34 +0100543#endif /* !MBEDTLS_TEST_NULL_ENTROPY */
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200544
Andres AGe7723ec2016-08-25 10:18:50 +0100545#if defined(MBEDTLS_ENTROPY_HARDWARE_ALT)
Andres AGb34e42e2016-08-22 11:08:50 +0100546
Andres AGe7723ec2016-08-25 10:18:50 +0100547static int mbedtls_entropy_source_self_test_gather( unsigned char *buf, size_t buf_len )
548{
549 int ret = 0;
550 size_t entropy_len = 0;
551 size_t olen = 0;
552 size_t attempts = buf_len;
553
554 while( attempts > 0 && entropy_len < buf_len )
555 {
556 if( ( ret = mbedtls_hardware_poll( NULL, buf + entropy_len,
557 buf_len - entropy_len, &olen ) ) != 0 )
558 return( ret );
559
560 entropy_len += olen;
561 attempts--;
562 }
563
564 if( entropy_len < buf_len )
565 {
566 ret = 1;
567 }
568
569 return( ret );
570}
571
572
573static int mbedtls_entropy_source_self_test_check_bits( const unsigned char *buf,
574 size_t buf_len )
575{
576 unsigned char set= 0xFF;
577 unsigned char unset = 0x00;
578 size_t i;
579
580 for( i = 0; i < buf_len; i++ )
581 {
582 set &= buf[i];
583 unset |= buf[i];
584 }
585
586 return( set == 0xFF || unset == 0x00 );
587}
Andres AGb34e42e2016-08-22 11:08:50 +0100588
589/*
Andres AGe7723ec2016-08-25 10:18:50 +0100590 * A test to ensure hat the entropy sources are functioning correctly
591 * and there is no obvious failure. The test performs the following checks:
Andres AGb34e42e2016-08-22 11:08:50 +0100592 * - The entropy source is not providing only 0s (all bits unset) or 1s (all
593 * bits set).
594 * - The entropy source is not providing values in a pattern. Because the
Andres AGe7723ec2016-08-25 10:18:50 +0100595 * hardware could be providing data in an arbitrary length, this check polls
596 * the hardware entropy source twice and compares the result to ensure they
597 * are not equal.
Andres AGb34e42e2016-08-22 11:08:50 +0100598 * - The error code returned by the entropy source is not an error.
599 */
Andres AGe7723ec2016-08-25 10:18:50 +0100600int mbedtls_entropy_source_self_test( int verbose )
Andres AGb34e42e2016-08-22 11:08:50 +0100601{
602 int ret = 0;
Andres AGe7723ec2016-08-25 10:18:50 +0100603 unsigned char buf0[2 * sizeof( unsigned long long int )];
604 unsigned char buf1[2 * sizeof( unsigned long long int )];
Andres AGb34e42e2016-08-22 11:08:50 +0100605
606 if( verbose != 0 )
607 mbedtls_printf( " ENTROPY_BIAS test: " );
608
Andres AGe7723ec2016-08-25 10:18:50 +0100609 memset( buf0, 0x00, sizeof( buf0 ) );
610 memset( buf1, 0x00, sizeof( buf1 ) );
Andres AGb34e42e2016-08-22 11:08:50 +0100611
Andres AGe7723ec2016-08-25 10:18:50 +0100612 if( ( ret = mbedtls_entropy_source_self_test_gather( buf0, sizeof( buf0 ) ) ) != 0 )
Andres AGb34e42e2016-08-22 11:08:50 +0100613 goto cleanup;
Andres AGe7723ec2016-08-25 10:18:50 +0100614 if( ( ret = mbedtls_entropy_source_self_test_gather( buf1, sizeof( buf1 ) ) ) != 0 )
615 goto cleanup;
616
617 /* Make sure that the returned values are not all 0 or 1 */
618 if( ( ret = mbedtls_entropy_source_self_test_check_bits( buf0, sizeof( buf0 ) ) ) != 0 )
619 goto cleanup;
620 if( ( ret = mbedtls_entropy_source_self_test_check_bits( buf1, sizeof( buf1 ) ) ) != 0 )
621 goto cleanup;
Andres AGb34e42e2016-08-22 11:08:50 +0100622
623 /* Make sure that the entropy source is not returning values in a
624 * pattern */
Andres AGe7723ec2016-08-25 10:18:50 +0100625 ret = memcmp( buf0, buf1, sizeof( buf0 ) ) == 0;
Andres AGb34e42e2016-08-22 11:08:50 +0100626
627cleanup:
Andres AGb34e42e2016-08-22 11:08:50 +0100628 if( verbose != 0 )
629 {
630 if( ret != 0 )
631 mbedtls_printf( "failed\n" );
632 else
633 mbedtls_printf( "passed\n" );
634
635 mbedtls_printf( "\n" );
636 }
637
638 return( ret != 0 );
639}
Andres AGe7723ec2016-08-25 10:18:50 +0100640
641#endif /* MBEDTLS_ENTROPY_HARDWARE_ALT */
Andres AGb34e42e2016-08-22 11:08:50 +0100642
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200643/*
644 * The actual entropy quality is hard to test, but we can at least
645 * test that the functions don't cause errors and write the correct
646 * amount of data to buffers.
647 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200648int mbedtls_entropy_self_test( int verbose )
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200649{
Andres Amaya Garciaa928e672016-09-13 13:30:02 +0100650 int ret = 1;
Simon Butcher669c6352016-09-15 18:57:34 +0100651#if !defined(MBEDTLS_TEST_NULL_ENTROPY)
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200652 mbedtls_entropy_context ctx;
653 unsigned char buf[MBEDTLS_ENTROPY_BLOCK_SIZE] = { 0 };
654 unsigned char acc[MBEDTLS_ENTROPY_BLOCK_SIZE] = { 0 };
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200655 size_t i, j;
Simon Butcher669c6352016-09-15 18:57:34 +0100656#endif /* !MBEDTLS_TEST_NULL_ENTROPY */
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200657
658 if( verbose != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200659 mbedtls_printf( " ENTROPY test: " );
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200660
Andres Amaya Garciaa928e672016-09-13 13:30:02 +0100661#if !defined(MBEDTLS_TEST_NULL_ENTROPY)
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200662 mbedtls_entropy_init( &ctx );
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200663
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200664 /* First do a gather to make sure we have default sources */
Manuel Pégourié-Gonnarde94bfe62015-05-14 13:57:50 +0200665 if( ( ret = mbedtls_entropy_gather( &ctx ) ) != 0 )
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200666 goto cleanup;
667
Manuel Pégourié-Gonnard7580ba42015-06-19 10:26:32 +0200668 ret = mbedtls_entropy_add_source( &ctx, entropy_dummy_source, NULL, 16,
669 MBEDTLS_ENTROPY_SOURCE_WEAK );
Manuel Pégourié-Gonnarde94bfe62015-05-14 13:57:50 +0200670 if( ret != 0 )
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200671 goto cleanup;
672
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200673 if( ( ret = mbedtls_entropy_update_manual( &ctx, buf, sizeof buf ) ) != 0 )
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200674 goto cleanup;
675
676 /*
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200677 * To test that mbedtls_entropy_func writes correct number of bytes:
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200678 * - use the whole buffer and rely on ASan to detect overruns
679 * - collect entropy 8 times and OR the result in an accumulator:
680 * any byte should then be 0 with probably 2^(-64), so requiring
681 * each of the 32 or 64 bytes to be non-zero has a false failure rate
682 * of at most 2^(-58) which is acceptable.
683 */
684 for( i = 0; i < 8; i++ )
685 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200686 if( ( ret = mbedtls_entropy_func( &ctx, buf, sizeof( buf ) ) ) != 0 )
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200687 goto cleanup;
688
689 for( j = 0; j < sizeof( buf ); j++ )
690 acc[j] |= buf[j];
691 }
692
693 for( j = 0; j < sizeof( buf ); j++ )
694 {
695 if( acc[j] == 0 )
696 {
697 ret = 1;
698 goto cleanup;
699 }
700 }
701
Andres AGe7723ec2016-08-25 10:18:50 +0100702#if defined(MBEDTLS_ENTROPY_HARDWARE_ALT)
703 if( ( ret = mbedtls_entropy_source_self_test( 0 ) ) != 0 )
704 goto cleanup;
705#endif
706
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200707cleanup:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200708 mbedtls_entropy_free( &ctx );
Andres Amaya Garciaa928e672016-09-13 13:30:02 +0100709#endif /* !MBEDTLS_TEST_NULL_ENTROPY */
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200710
711 if( verbose != 0 )
712 {
713 if( ret != 0 )
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200714 mbedtls_printf( "failed\n" );
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200715 else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200716 mbedtls_printf( "passed\n" );
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200717
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200718 mbedtls_printf( "\n" );
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200719 }
720
721 return( ret != 0 );
722}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200723#endif /* MBEDTLS_SELF_TEST */
Manuel Pégourié-Gonnard4dd73922014-05-30 10:34:15 +0200724
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200725#endif /* MBEDTLS_ENTROPY_C */