blob: 0f27aaa30d059521579e0dade9bde464effdc29d [file] [log] [blame]
Paul Bakkerfc36d162011-01-27 16:50:02 +00001/**
Paul Bakker6083fd22011-12-03 21:45:14 +00002 * \brief Use and generate multiple entropies calls into a file
Paul Bakkerfc36d162011-01-27 16:50:02 +00003 *
Bence Szépkúti1e148272020-08-07 13:07:28 +02004 * Copyright The Mbed TLS Contributors
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 Bakkerfc36d162011-01-27 16:50:02 +000018 */
19
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020020#if !defined(MBEDTLS_CONFIG_FILE)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000021#include "mbedtls/config.h"
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020022#else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020023#include MBEDTLS_CONFIG_FILE
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020024#endif
Paul Bakker5690efc2011-05-26 13:16:06 +000025
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020026#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000027#include "mbedtls/platform.h"
Rich Evansf90016a2015-01-19 14:26:37 +000028#else
Rich Evans18b78c72015-02-11 14:06:19 +000029#include <stdio.h>
Andres Amaya Garcia55a0d562018-04-29 20:40:36 +010030#include <stdlib.h>
31#define mbedtls_fprintf fprintf
32#define mbedtls_printf printf
Manuel Pégourié-Gonnard3ef6a6d2018-12-10 14:31:45 +010033#define mbedtls_exit exit
Andres Amaya Garcia7d429652018-04-30 22:42:33 +010034#define MBEDTLS_EXIT_SUCCESS EXIT_SUCCESS
Andres Amaya Garcia55a0d562018-04-29 20:40:36 +010035#define MBEDTLS_EXIT_FAILURE EXIT_FAILURE
36#endif /* MBEDTLS_PLATFORM_C */
Rich Evansf90016a2015-01-19 14:26:37 +000037
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020038#if defined(MBEDTLS_ENTROPY_C) && defined(MBEDTLS_FS_IO)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000039#include "mbedtls/entropy.h"
Paul Bakker5690efc2011-05-26 13:16:06 +000040
Paul Bakkerfc36d162011-01-27 16:50:02 +000041#include <stdio.h>
Rich Evans18b78c72015-02-11 14:06:19 +000042#endif
Paul Bakkerfc36d162011-01-27 16:50:02 +000043
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020044#if !defined(MBEDTLS_ENTROPY_C) || !defined(MBEDTLS_FS_IO)
Rich Evans85b05ec2015-02-12 11:37:29 +000045int main( void )
Paul Bakker5690efc2011-05-26 13:16:06 +000046{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020047 mbedtls_printf("MBEDTLS_ENTROPY_C and/or MBEDTLS_FS_IO not defined.\n");
Krzysztof Stachowiak5e1b1952019-04-24 14:24:46 +020048 mbedtls_exit( 0 );
Paul Bakker5690efc2011-05-26 13:16:06 +000049}
50#else
Manuel Pégourié-Gonnard3ef6a6d2018-12-10 14:31:45 +010051
Manuel Pégourié-Gonnard3ef6a6d2018-12-10 14:31:45 +010052
Paul Bakkerfc36d162011-01-27 16:50:02 +000053int main( int argc, char *argv[] )
54{
55 FILE *f;
Andres Amaya Garcia55a0d562018-04-29 20:40:36 +010056 int i, k, ret = 1;
57 int exit_code = MBEDTLS_EXIT_FAILURE;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020058 mbedtls_entropy_context entropy;
59 unsigned char buf[MBEDTLS_ENTROPY_BLOCK_SIZE];
Paul Bakkerfc36d162011-01-27 16:50:02 +000060
61 if( argc < 2 )
62 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020063 mbedtls_fprintf( stderr, "usage: %s <output filename>\n", argv[0] );
Krzysztof Stachowiak5e1b1952019-04-24 14:24:46 +020064 mbedtls_exit( exit_code );
Paul Bakkerfc36d162011-01-27 16:50:02 +000065 }
66
67 if( ( f = fopen( argv[1], "wb+" ) ) == NULL )
68 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020069 mbedtls_printf( "failed to open '%s' for writing.\n", argv[1] );
Krzysztof Stachowiak5e1b1952019-04-24 14:24:46 +020070 mbedtls_exit( exit_code );
Paul Bakkerfc36d162011-01-27 16:50:02 +000071 }
72
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020073 mbedtls_entropy_init( &entropy );
Paul Bakkerfc36d162011-01-27 16:50:02 +000074
75 for( i = 0, k = 768; i < k; i++ )
76 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020077 ret = mbedtls_entropy_func( &entropy, buf, sizeof( buf ) );
Paul Bakker6083fd22011-12-03 21:45:14 +000078 if( ret != 0 )
Paul Bakkera3d195c2011-11-27 21:07:34 +000079 {
Andres Amaya Garcia55a0d562018-04-29 20:40:36 +010080 mbedtls_printf( " failed\n ! mbedtls_entropy_func returned -%04X\n",
Kenneth Soerensen518d4352020-04-01 17:22:45 +020081 (unsigned int) ret );
Paul Bakker6083fd22011-12-03 21:45:14 +000082 goto cleanup;
Paul Bakkera3d195c2011-11-27 21:07:34 +000083 }
Paul Bakkerfc36d162011-01-27 16:50:02 +000084
Paul Bakker6083fd22011-12-03 21:45:14 +000085 fwrite( buf, 1, sizeof( buf ), f );
Paul Bakkerfc36d162011-01-27 16:50:02 +000086
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020087 mbedtls_printf( "Generating %ldkb of data in file '%s'... %04.1f" \
veggie64a57992015-01-26 10:35:25 -050088 "%% done\r", (long)(sizeof(buf) * k / 1024), argv[1], (100 * (float) (i + 1)) / k );
Paul Bakkerfc36d162011-01-27 16:50:02 +000089 fflush( stdout );
90 }
91
Andres Amaya Garcia55a0d562018-04-29 20:40:36 +010092 exit_code = MBEDTLS_EXIT_SUCCESS;
Paul Bakker6083fd22011-12-03 21:45:14 +000093
94cleanup:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020095 mbedtls_printf( "\n" );
Paul Bakkerfc36d162011-01-27 16:50:02 +000096
97 fclose( f );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020098 mbedtls_entropy_free( &entropy );
Paul Bakker6083fd22011-12-03 21:45:14 +000099
Krzysztof Stachowiak5e1b1952019-04-24 14:24:46 +0200100 mbedtls_exit( exit_code );
Paul Bakkerfc36d162011-01-27 16:50:02 +0000101}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200102#endif /* MBEDTLS_ENTROPY_C */