1 /**
2 * \brief Generate random data into a file
3 *
4 * Copyright The Mbed TLS Contributors
5 * SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
6 */
7
8 #if !defined(MBEDTLS_CONFIG_FILE)
9 #include "mbedtls/config.h"
10 #else
11 #include MBEDTLS_CONFIG_FILE
12 #endif
13
14 #include "mbedtls/platform.h"
15
16 #if defined(MBEDTLS_HAVEGE_C) && defined(MBEDTLS_FS_IO)
17 #include "mbedtls/havege.h"
18
19 #include <stdio.h>
20 #include <time.h>
21 #endif
22
23 #if !defined(MBEDTLS_HAVEGE_C) || !defined(MBEDTLS_FS_IO)
main(void)24 int main(void)
25 {
26 mbedtls_printf("MBEDTLS_HAVEGE_C not defined.\n");
27 mbedtls_exit(0);
28 }
29 #else
30
31
main(int argc,char * argv[])32 int main(int argc, char *argv[])
33 {
34 FILE *f;
35 time_t t;
36 int i, k, ret = 1;
37 int exit_code = MBEDTLS_EXIT_FAILURE;
38 mbedtls_havege_state hs;
39 unsigned char buf[1024];
40
41 if (argc < 2) {
42 mbedtls_fprintf(stderr, "usage: %s <output filename>\n", argv[0]);
43 mbedtls_exit(exit_code);
44 }
45
46 if ((f = fopen(argv[1], "wb+")) == NULL) {
47 mbedtls_printf("failed to open '%s' for writing.\n", argv[1]);
48 mbedtls_exit(exit_code);
49 }
50
51 mbedtls_havege_init(&hs);
52
53 t = time(NULL);
54
55 for (i = 0, k = 768; i < k; i++) {
56 if ((ret = mbedtls_havege_random(&hs, buf, sizeof(buf))) != 0) {
57 mbedtls_printf(" failed\n ! mbedtls_havege_random returned -0x%04X",
58 (unsigned int) -ret);
59 goto exit;
60 }
61
62 fwrite(buf, sizeof(buf), 1, f);
63
64 mbedtls_printf("Generating %ldkb of data in file '%s'... %04.1f" \
65 "%% done\r",
66 (long) (sizeof(buf) * k / 1024),
67 argv[1],
68 (100 * (float) (i + 1)) / k);
69 fflush(stdout);
70 }
71
72 if (t == time(NULL)) {
73 t--;
74 }
75
76 mbedtls_printf(" \n ");
77
78 exit_code = MBEDTLS_EXIT_SUCCESS;
79
80 exit:
81 mbedtls_havege_free(&hs);
82 fclose(f);
83 mbedtls_exit(exit_code);
84 }
85 #endif /* MBEDTLS_HAVEGE_C */
86