hmac_drbg.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630
  1. /*
  2. * HMAC_DRBG implementation (NIST SP 800-90)
  3. *
  4. * Copyright The Mbed TLS Contributors
  5. * 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.
  18. */
  19. /*
  20. * The NIST SP 800-90A DRBGs are described in the following publication.
  21. * http://csrc.nist.gov/publications/nistpubs/800-90A/SP800-90A.pdf
  22. * References below are based on rev. 1 (January 2012).
  23. */
  24. #include "common.h"
  25. #if defined(MBEDTLS_HMAC_DRBG_C)
  26. #include "mbedtls/hmac_drbg.h"
  27. #include "mbedtls/platform_util.h"
  28. #include "mbedtls/error.h"
  29. #include <string.h>
  30. #if defined(MBEDTLS_FS_IO)
  31. #include <stdio.h>
  32. #endif
  33. #if defined(MBEDTLS_SELF_TEST)
  34. #if defined(MBEDTLS_PLATFORM_C)
  35. #include "mbedtls/platform.h"
  36. #else
  37. #include <stdio.h>
  38. #define mbedtls_printf printf
  39. #endif /* MBEDTLS_SELF_TEST */
  40. #endif /* MBEDTLS_PLATFORM_C */
  41. /*
  42. * HMAC_DRBG context initialization
  43. */
  44. void mbedtls_hmac_drbg_init( mbedtls_hmac_drbg_context *ctx )
  45. {
  46. memset( ctx, 0, sizeof( mbedtls_hmac_drbg_context ) );
  47. ctx->reseed_interval = MBEDTLS_HMAC_DRBG_RESEED_INTERVAL;
  48. }
  49. /*
  50. * HMAC_DRBG update, using optional additional data (10.1.2.2)
  51. */
  52. int mbedtls_hmac_drbg_update_ret( mbedtls_hmac_drbg_context *ctx,
  53. const unsigned char *additional,
  54. size_t add_len )
  55. {
  56. size_t md_len = mbedtls_md_get_size( ctx->md_ctx.md_info );
  57. unsigned char rounds = ( additional != NULL && add_len != 0 ) ? 2 : 1;
  58. unsigned char sep[1];
  59. unsigned char K[MBEDTLS_MD_MAX_SIZE];
  60. int ret = MBEDTLS_ERR_MD_BAD_INPUT_DATA;
  61. for( sep[0] = 0; sep[0] < rounds; sep[0]++ )
  62. {
  63. /* Step 1 or 4 */
  64. if( ( ret = mbedtls_md_hmac_reset( &ctx->md_ctx ) ) != 0 )
  65. goto exit;
  66. if( ( ret = mbedtls_md_hmac_update( &ctx->md_ctx,
  67. ctx->V, md_len ) ) != 0 )
  68. goto exit;
  69. if( ( ret = mbedtls_md_hmac_update( &ctx->md_ctx,
  70. sep, 1 ) ) != 0 )
  71. goto exit;
  72. if( rounds == 2 )
  73. {
  74. if( ( ret = mbedtls_md_hmac_update( &ctx->md_ctx,
  75. additional, add_len ) ) != 0 )
  76. goto exit;
  77. }
  78. if( ( ret = mbedtls_md_hmac_finish( &ctx->md_ctx, K ) ) != 0 )
  79. goto exit;
  80. /* Step 2 or 5 */
  81. if( ( ret = mbedtls_md_hmac_starts( &ctx->md_ctx, K, md_len ) ) != 0 )
  82. goto exit;
  83. if( ( ret = mbedtls_md_hmac_update( &ctx->md_ctx,
  84. ctx->V, md_len ) ) != 0 )
  85. goto exit;
  86. if( ( ret = mbedtls_md_hmac_finish( &ctx->md_ctx, ctx->V ) ) != 0 )
  87. goto exit;
  88. }
  89. exit:
  90. mbedtls_platform_zeroize( K, sizeof( K ) );
  91. return( ret );
  92. }
  93. #if !defined(MBEDTLS_DEPRECATED_REMOVED)
  94. void mbedtls_hmac_drbg_update( mbedtls_hmac_drbg_context *ctx,
  95. const unsigned char *additional,
  96. size_t add_len )
  97. {
  98. (void) mbedtls_hmac_drbg_update_ret( ctx, additional, add_len );
  99. }
  100. #endif /* MBEDTLS_DEPRECATED_REMOVED */
  101. /*
  102. * Simplified HMAC_DRBG initialisation (for use with deterministic ECDSA)
  103. */
  104. int mbedtls_hmac_drbg_seed_buf( mbedtls_hmac_drbg_context *ctx,
  105. const mbedtls_md_info_t * md_info,
  106. const unsigned char *data, size_t data_len )
  107. {
  108. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  109. if( ( ret = mbedtls_md_setup( &ctx->md_ctx, md_info, 1 ) ) != 0 )
  110. return( ret );
  111. #if defined(MBEDTLS_THREADING_C)
  112. mbedtls_mutex_init( &ctx->mutex );
  113. #endif
  114. /*
  115. * Set initial working state.
  116. * Use the V memory location, which is currently all 0, to initialize the
  117. * MD context with an all-zero key. Then set V to its initial value.
  118. */
  119. if( ( ret = mbedtls_md_hmac_starts( &ctx->md_ctx, ctx->V,
  120. mbedtls_md_get_size( md_info ) ) ) != 0 )
  121. return( ret );
  122. memset( ctx->V, 0x01, mbedtls_md_get_size( md_info ) );
  123. if( ( ret = mbedtls_hmac_drbg_update_ret( ctx, data, data_len ) ) != 0 )
  124. return( ret );
  125. return( 0 );
  126. }
  127. /*
  128. * Internal function used both for seeding and reseeding the DRBG.
  129. * Comments starting with arabic numbers refer to section 10.1.2.4
  130. * of SP800-90A, while roman numbers refer to section 9.2.
  131. */
  132. static int hmac_drbg_reseed_core( mbedtls_hmac_drbg_context *ctx,
  133. const unsigned char *additional, size_t len,
  134. int use_nonce )
  135. {
  136. unsigned char seed[MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT];
  137. size_t seedlen = 0;
  138. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  139. {
  140. size_t total_entropy_len;
  141. if( use_nonce == 0 )
  142. total_entropy_len = ctx->entropy_len;
  143. else
  144. total_entropy_len = ctx->entropy_len * 3 / 2;
  145. /* III. Check input length */
  146. if( len > MBEDTLS_HMAC_DRBG_MAX_INPUT ||
  147. total_entropy_len + len > MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT )
  148. {
  149. return( MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG );
  150. }
  151. }
  152. memset( seed, 0, MBEDTLS_HMAC_DRBG_MAX_SEED_INPUT );
  153. /* IV. Gather entropy_len bytes of entropy for the seed */
  154. if( ( ret = ctx->f_entropy( ctx->p_entropy,
  155. seed, ctx->entropy_len ) ) != 0 )
  156. {
  157. return( MBEDTLS_ERR_HMAC_DRBG_ENTROPY_SOURCE_FAILED );
  158. }
  159. seedlen += ctx->entropy_len;
  160. /* For initial seeding, allow adding of nonce generated
  161. * from the entropy source. See Sect 8.6.7 in SP800-90A. */
  162. if( use_nonce )
  163. {
  164. /* Note: We don't merge the two calls to f_entropy() in order
  165. * to avoid requesting too much entropy from f_entropy()
  166. * at once. Specifically, if the underlying digest is not
  167. * SHA-1, 3 / 2 * entropy_len is at least 36 Bytes, which
  168. * is larger than the maximum of 32 Bytes that our own
  169. * entropy source implementation can emit in a single
  170. * call in configurations disabling SHA-512. */
  171. if( ( ret = ctx->f_entropy( ctx->p_entropy,
  172. seed + seedlen,
  173. ctx->entropy_len / 2 ) ) != 0 )
  174. {
  175. return( MBEDTLS_ERR_HMAC_DRBG_ENTROPY_SOURCE_FAILED );
  176. }
  177. seedlen += ctx->entropy_len / 2;
  178. }
  179. /* 1. Concatenate entropy and additional data if any */
  180. if( additional != NULL && len != 0 )
  181. {
  182. memcpy( seed + seedlen, additional, len );
  183. seedlen += len;
  184. }
  185. /* 2. Update state */
  186. if( ( ret = mbedtls_hmac_drbg_update_ret( ctx, seed, seedlen ) ) != 0 )
  187. goto exit;
  188. /* 3. Reset reseed_counter */
  189. ctx->reseed_counter = 1;
  190. exit:
  191. /* 4. Done */
  192. mbedtls_platform_zeroize( seed, seedlen );
  193. return( ret );
  194. }
  195. /*
  196. * HMAC_DRBG reseeding: 10.1.2.4 + 9.2
  197. */
  198. int mbedtls_hmac_drbg_reseed( mbedtls_hmac_drbg_context *ctx,
  199. const unsigned char *additional, size_t len )
  200. {
  201. return( hmac_drbg_reseed_core( ctx, additional, len, 0 ) );
  202. }
  203. /*
  204. * HMAC_DRBG initialisation (10.1.2.3 + 9.1)
  205. *
  206. * The nonce is not passed as a separate parameter but extracted
  207. * from the entropy source as suggested in 8.6.7.
  208. */
  209. int mbedtls_hmac_drbg_seed( mbedtls_hmac_drbg_context *ctx,
  210. const mbedtls_md_info_t * md_info,
  211. int (*f_entropy)(void *, unsigned char *, size_t),
  212. void *p_entropy,
  213. const unsigned char *custom,
  214. size_t len )
  215. {
  216. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  217. size_t md_size;
  218. if( ( ret = mbedtls_md_setup( &ctx->md_ctx, md_info, 1 ) ) != 0 )
  219. return( ret );
  220. /* The mutex is initialized iff the md context is set up. */
  221. #if defined(MBEDTLS_THREADING_C)
  222. mbedtls_mutex_init( &ctx->mutex );
  223. #endif
  224. md_size = mbedtls_md_get_size( md_info );
  225. /*
  226. * Set initial working state.
  227. * Use the V memory location, which is currently all 0, to initialize the
  228. * MD context with an all-zero key. Then set V to its initial value.
  229. */
  230. if( ( ret = mbedtls_md_hmac_starts( &ctx->md_ctx, ctx->V, md_size ) ) != 0 )
  231. return( ret );
  232. memset( ctx->V, 0x01, md_size );
  233. ctx->f_entropy = f_entropy;
  234. ctx->p_entropy = p_entropy;
  235. if( ctx->entropy_len == 0 )
  236. {
  237. /*
  238. * See SP800-57 5.6.1 (p. 65-66) for the security strength provided by
  239. * each hash function, then according to SP800-90A rev1 10.1 table 2,
  240. * min_entropy_len (in bits) is security_strength.
  241. *
  242. * (This also matches the sizes used in the NIST test vectors.)
  243. */
  244. ctx->entropy_len = md_size <= 20 ? 16 : /* 160-bits hash -> 128 bits */
  245. md_size <= 28 ? 24 : /* 224-bits hash -> 192 bits */
  246. 32; /* better (256+) -> 256 bits */
  247. }
  248. if( ( ret = hmac_drbg_reseed_core( ctx, custom, len,
  249. 1 /* add nonce */ ) ) != 0 )
  250. {
  251. return( ret );
  252. }
  253. return( 0 );
  254. }
  255. /*
  256. * Set prediction resistance
  257. */
  258. void mbedtls_hmac_drbg_set_prediction_resistance( mbedtls_hmac_drbg_context *ctx,
  259. int resistance )
  260. {
  261. ctx->prediction_resistance = resistance;
  262. }
  263. /*
  264. * Set entropy length grabbed for seeding
  265. */
  266. void mbedtls_hmac_drbg_set_entropy_len( mbedtls_hmac_drbg_context *ctx, size_t len )
  267. {
  268. ctx->entropy_len = len;
  269. }
  270. /*
  271. * Set reseed interval
  272. */
  273. void mbedtls_hmac_drbg_set_reseed_interval( mbedtls_hmac_drbg_context *ctx, int interval )
  274. {
  275. ctx->reseed_interval = interval;
  276. }
  277. /*
  278. * HMAC_DRBG random function with optional additional data:
  279. * 10.1.2.5 (arabic) + 9.3 (Roman)
  280. */
  281. int mbedtls_hmac_drbg_random_with_add( void *p_rng,
  282. unsigned char *output, size_t out_len,
  283. const unsigned char *additional, size_t add_len )
  284. {
  285. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  286. mbedtls_hmac_drbg_context *ctx = (mbedtls_hmac_drbg_context *) p_rng;
  287. size_t md_len = mbedtls_md_get_size( ctx->md_ctx.md_info );
  288. size_t left = out_len;
  289. unsigned char *out = output;
  290. /* II. Check request length */
  291. if( out_len > MBEDTLS_HMAC_DRBG_MAX_REQUEST )
  292. return( MBEDTLS_ERR_HMAC_DRBG_REQUEST_TOO_BIG );
  293. /* III. Check input length */
  294. if( add_len > MBEDTLS_HMAC_DRBG_MAX_INPUT )
  295. return( MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG );
  296. /* 1. (aka VII and IX) Check reseed counter and PR */
  297. if( ctx->f_entropy != NULL && /* For no-reseeding instances */
  298. ( ctx->prediction_resistance == MBEDTLS_HMAC_DRBG_PR_ON ||
  299. ctx->reseed_counter > ctx->reseed_interval ) )
  300. {
  301. if( ( ret = mbedtls_hmac_drbg_reseed( ctx, additional, add_len ) ) != 0 )
  302. return( ret );
  303. add_len = 0; /* VII.4 */
  304. }
  305. /* 2. Use additional data if any */
  306. if( additional != NULL && add_len != 0 )
  307. {
  308. if( ( ret = mbedtls_hmac_drbg_update_ret( ctx,
  309. additional, add_len ) ) != 0 )
  310. goto exit;
  311. }
  312. /* 3, 4, 5. Generate bytes */
  313. while( left != 0 )
  314. {
  315. size_t use_len = left > md_len ? md_len : left;
  316. if( ( ret = mbedtls_md_hmac_reset( &ctx->md_ctx ) ) != 0 )
  317. goto exit;
  318. if( ( ret = mbedtls_md_hmac_update( &ctx->md_ctx,
  319. ctx->V, md_len ) ) != 0 )
  320. goto exit;
  321. if( ( ret = mbedtls_md_hmac_finish( &ctx->md_ctx, ctx->V ) ) != 0 )
  322. goto exit;
  323. memcpy( out, ctx->V, use_len );
  324. out += use_len;
  325. left -= use_len;
  326. }
  327. /* 6. Update */
  328. if( ( ret = mbedtls_hmac_drbg_update_ret( ctx,
  329. additional, add_len ) ) != 0 )
  330. goto exit;
  331. /* 7. Update reseed counter */
  332. ctx->reseed_counter++;
  333. exit:
  334. /* 8. Done */
  335. return( ret );
  336. }
  337. /*
  338. * HMAC_DRBG random function
  339. */
  340. int mbedtls_hmac_drbg_random( void *p_rng, unsigned char *output, size_t out_len )
  341. {
  342. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  343. mbedtls_hmac_drbg_context *ctx = (mbedtls_hmac_drbg_context *) p_rng;
  344. #if defined(MBEDTLS_THREADING_C)
  345. if( ( ret = mbedtls_mutex_lock( &ctx->mutex ) ) != 0 )
  346. return( ret );
  347. #endif
  348. ret = mbedtls_hmac_drbg_random_with_add( ctx, output, out_len, NULL, 0 );
  349. #if defined(MBEDTLS_THREADING_C)
  350. if( mbedtls_mutex_unlock( &ctx->mutex ) != 0 )
  351. return( MBEDTLS_ERR_THREADING_MUTEX_ERROR );
  352. #endif
  353. return( ret );
  354. }
  355. /*
  356. * This function resets HMAC_DRBG context to the state immediately
  357. * after initial call of mbedtls_hmac_drbg_init().
  358. */
  359. void mbedtls_hmac_drbg_free( mbedtls_hmac_drbg_context *ctx )
  360. {
  361. if( ctx == NULL )
  362. return;
  363. #if defined(MBEDTLS_THREADING_C)
  364. /* The mutex is initialized iff the md context is set up. */
  365. if( ctx->md_ctx.md_info != NULL )
  366. mbedtls_mutex_free( &ctx->mutex );
  367. #endif
  368. mbedtls_md_free( &ctx->md_ctx );
  369. mbedtls_platform_zeroize( ctx, sizeof( mbedtls_hmac_drbg_context ) );
  370. ctx->reseed_interval = MBEDTLS_HMAC_DRBG_RESEED_INTERVAL;
  371. }
  372. #if defined(MBEDTLS_FS_IO)
  373. int mbedtls_hmac_drbg_write_seed_file( mbedtls_hmac_drbg_context *ctx, const char *path )
  374. {
  375. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  376. FILE *f;
  377. unsigned char buf[ MBEDTLS_HMAC_DRBG_MAX_INPUT ];
  378. if( ( f = fopen( path, "wb" ) ) == NULL )
  379. return( MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR );
  380. if( ( ret = mbedtls_hmac_drbg_random( ctx, buf, sizeof( buf ) ) ) != 0 )
  381. goto exit;
  382. if( fwrite( buf, 1, sizeof( buf ), f ) != sizeof( buf ) )
  383. {
  384. ret = MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR;
  385. goto exit;
  386. }
  387. ret = 0;
  388. exit:
  389. fclose( f );
  390. mbedtls_platform_zeroize( buf, sizeof( buf ) );
  391. return( ret );
  392. }
  393. int mbedtls_hmac_drbg_update_seed_file( mbedtls_hmac_drbg_context *ctx, const char *path )
  394. {
  395. int ret = 0;
  396. FILE *f = NULL;
  397. size_t n;
  398. unsigned char buf[ MBEDTLS_HMAC_DRBG_MAX_INPUT ];
  399. unsigned char c;
  400. if( ( f = fopen( path, "rb" ) ) == NULL )
  401. return( MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR );
  402. n = fread( buf, 1, sizeof( buf ), f );
  403. if( fread( &c, 1, 1, f ) != 0 )
  404. {
  405. ret = MBEDTLS_ERR_HMAC_DRBG_INPUT_TOO_BIG;
  406. goto exit;
  407. }
  408. if( n == 0 || ferror( f ) )
  409. {
  410. ret = MBEDTLS_ERR_HMAC_DRBG_FILE_IO_ERROR;
  411. goto exit;
  412. }
  413. fclose( f );
  414. f = NULL;
  415. ret = mbedtls_hmac_drbg_update_ret( ctx, buf, n );
  416. exit:
  417. mbedtls_platform_zeroize( buf, sizeof( buf ) );
  418. if( f != NULL )
  419. fclose( f );
  420. if( ret != 0 )
  421. return( ret );
  422. return( mbedtls_hmac_drbg_write_seed_file( ctx, path ) );
  423. }
  424. #endif /* MBEDTLS_FS_IO */
  425. #if defined(MBEDTLS_SELF_TEST)
  426. #if !defined(MBEDTLS_SHA1_C)
  427. /* Dummy checkup routine */
  428. int mbedtls_hmac_drbg_self_test( int verbose )
  429. {
  430. (void) verbose;
  431. return( 0 );
  432. }
  433. #else
  434. #define OUTPUT_LEN 80
  435. /* From a NIST PR=true test vector */
  436. static const unsigned char entropy_pr[] = {
  437. 0xa0, 0xc9, 0xab, 0x58, 0xf1, 0xe2, 0xe5, 0xa4, 0xde, 0x3e, 0xbd, 0x4f,
  438. 0xf7, 0x3e, 0x9c, 0x5b, 0x64, 0xef, 0xd8, 0xca, 0x02, 0x8c, 0xf8, 0x11,
  439. 0x48, 0xa5, 0x84, 0xfe, 0x69, 0xab, 0x5a, 0xee, 0x42, 0xaa, 0x4d, 0x42,
  440. 0x17, 0x60, 0x99, 0xd4, 0x5e, 0x13, 0x97, 0xdc, 0x40, 0x4d, 0x86, 0xa3,
  441. 0x7b, 0xf5, 0x59, 0x54, 0x75, 0x69, 0x51, 0xe4 };
  442. static const unsigned char result_pr[OUTPUT_LEN] = {
  443. 0x9a, 0x00, 0xa2, 0xd0, 0x0e, 0xd5, 0x9b, 0xfe, 0x31, 0xec, 0xb1, 0x39,
  444. 0x9b, 0x60, 0x81, 0x48, 0xd1, 0x96, 0x9d, 0x25, 0x0d, 0x3c, 0x1e, 0x94,
  445. 0x10, 0x10, 0x98, 0x12, 0x93, 0x25, 0xca, 0xb8, 0xfc, 0xcc, 0x2d, 0x54,
  446. 0x73, 0x19, 0x70, 0xc0, 0x10, 0x7a, 0xa4, 0x89, 0x25, 0x19, 0x95, 0x5e,
  447. 0x4b, 0xc6, 0x00, 0x1d, 0x7f, 0x4e, 0x6a, 0x2b, 0xf8, 0xa3, 0x01, 0xab,
  448. 0x46, 0x05, 0x5c, 0x09, 0xa6, 0x71, 0x88, 0xf1, 0xa7, 0x40, 0xee, 0xf3,
  449. 0xe1, 0x5c, 0x02, 0x9b, 0x44, 0xaf, 0x03, 0x44 };
  450. /* From a NIST PR=false test vector */
  451. static const unsigned char entropy_nopr[] = {
  452. 0x79, 0x34, 0x9b, 0xbf, 0x7c, 0xdd, 0xa5, 0x79, 0x95, 0x57, 0x86, 0x66,
  453. 0x21, 0xc9, 0x13, 0x83, 0x11, 0x46, 0x73, 0x3a, 0xbf, 0x8c, 0x35, 0xc8,
  454. 0xc7, 0x21, 0x5b, 0x5b, 0x96, 0xc4, 0x8e, 0x9b, 0x33, 0x8c, 0x74, 0xe3,
  455. 0xe9, 0x9d, 0xfe, 0xdf };
  456. static const unsigned char result_nopr[OUTPUT_LEN] = {
  457. 0xc6, 0xa1, 0x6a, 0xb8, 0xd4, 0x20, 0x70, 0x6f, 0x0f, 0x34, 0xab, 0x7f,
  458. 0xec, 0x5a, 0xdc, 0xa9, 0xd8, 0xca, 0x3a, 0x13, 0x3e, 0x15, 0x9c, 0xa6,
  459. 0xac, 0x43, 0xc6, 0xf8, 0xa2, 0xbe, 0x22, 0x83, 0x4a, 0x4c, 0x0a, 0x0a,
  460. 0xff, 0xb1, 0x0d, 0x71, 0x94, 0xf1, 0xc1, 0xa5, 0xcf, 0x73, 0x22, 0xec,
  461. 0x1a, 0xe0, 0x96, 0x4e, 0xd4, 0xbf, 0x12, 0x27, 0x46, 0xe0, 0x87, 0xfd,
  462. 0xb5, 0xb3, 0xe9, 0x1b, 0x34, 0x93, 0xd5, 0xbb, 0x98, 0xfa, 0xed, 0x49,
  463. 0xe8, 0x5f, 0x13, 0x0f, 0xc8, 0xa4, 0x59, 0xb7 };
  464. /* "Entropy" from buffer */
  465. static size_t test_offset;
  466. static int hmac_drbg_self_test_entropy( void *data,
  467. unsigned char *buf, size_t len )
  468. {
  469. const unsigned char *p = data;
  470. memcpy( buf, p + test_offset, len );
  471. test_offset += len;
  472. return( 0 );
  473. }
  474. #define CHK( c ) if( (c) != 0 ) \
  475. { \
  476. if( verbose != 0 ) \
  477. mbedtls_printf( "failed\n" ); \
  478. return( 1 ); \
  479. }
  480. /*
  481. * Checkup routine for HMAC_DRBG with SHA-1
  482. */
  483. int mbedtls_hmac_drbg_self_test( int verbose )
  484. {
  485. mbedtls_hmac_drbg_context ctx;
  486. unsigned char buf[OUTPUT_LEN];
  487. const mbedtls_md_info_t *md_info = mbedtls_md_info_from_type( MBEDTLS_MD_SHA1 );
  488. mbedtls_hmac_drbg_init( &ctx );
  489. /*
  490. * PR = True
  491. */
  492. if( verbose != 0 )
  493. mbedtls_printf( " HMAC_DRBG (PR = True) : " );
  494. test_offset = 0;
  495. CHK( mbedtls_hmac_drbg_seed( &ctx, md_info,
  496. hmac_drbg_self_test_entropy, (void *) entropy_pr,
  497. NULL, 0 ) );
  498. mbedtls_hmac_drbg_set_prediction_resistance( &ctx, MBEDTLS_HMAC_DRBG_PR_ON );
  499. CHK( mbedtls_hmac_drbg_random( &ctx, buf, OUTPUT_LEN ) );
  500. CHK( mbedtls_hmac_drbg_random( &ctx, buf, OUTPUT_LEN ) );
  501. CHK( memcmp( buf, result_pr, OUTPUT_LEN ) );
  502. mbedtls_hmac_drbg_free( &ctx );
  503. mbedtls_hmac_drbg_free( &ctx );
  504. if( verbose != 0 )
  505. mbedtls_printf( "passed\n" );
  506. /*
  507. * PR = False
  508. */
  509. if( verbose != 0 )
  510. mbedtls_printf( " HMAC_DRBG (PR = False) : " );
  511. mbedtls_hmac_drbg_init( &ctx );
  512. test_offset = 0;
  513. CHK( mbedtls_hmac_drbg_seed( &ctx, md_info,
  514. hmac_drbg_self_test_entropy, (void *) entropy_nopr,
  515. NULL, 0 ) );
  516. CHK( mbedtls_hmac_drbg_reseed( &ctx, NULL, 0 ) );
  517. CHK( mbedtls_hmac_drbg_random( &ctx, buf, OUTPUT_LEN ) );
  518. CHK( mbedtls_hmac_drbg_random( &ctx, buf, OUTPUT_LEN ) );
  519. CHK( memcmp( buf, result_nopr, OUTPUT_LEN ) );
  520. mbedtls_hmac_drbg_free( &ctx );
  521. mbedtls_hmac_drbg_free( &ctx );
  522. if( verbose != 0 )
  523. mbedtls_printf( "passed\n" );
  524. if( verbose != 0 )
  525. mbedtls_printf( "\n" );
  526. return( 0 );
  527. }
  528. #endif /* MBEDTLS_SHA1_C */
  529. #endif /* MBEDTLS_SELF_TEST */
  530. #endif /* MBEDTLS_HMAC_DRBG_C */