blob: cb4659359607f3c6433af43af8d66db5b63e4d30 [file] [log] [blame]
Paul Bakkerfc36d162011-01-27 16:50:02 +00001/**
Paul Bakker6083fd22011-12-03 21:45:14 +00002 * \brief Use and generate random data into a file via the CTR_DBRG based on AES
Paul Bakkerfc36d162011-01-27 16:50:02 +00003 *
Paul Bakkercce9d772011-11-18 14:26:47 +00004 * Copyright (C) 2006-2011, Brainspark B.V.
Paul Bakkerfc36d162011-01-27 16:50:02 +00005 *
6 * This file is part of PolarSSL (http://www.polarssl.org)
7 * Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
8 *
9 * All rights reserved.
10 *
11 * This program is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation; either version 2 of the License, or
14 * (at your option) any later version.
15 *
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
20 *
21 * You should have received a copy of the GNU General Public License along
22 * with this program; if not, write to the Free Software Foundation, Inc.,
23 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 */
25
26#include "polarssl/config.h"
Paul Bakker5690efc2011-05-26 13:16:06 +000027
Paul Bakker6083fd22011-12-03 21:45:14 +000028#include "polarssl/entropy.h"
29#include "polarssl/ctr_drbg.h"
Paul Bakker5690efc2011-05-26 13:16:06 +000030
Paul Bakkerfc36d162011-01-27 16:50:02 +000031#include <stdio.h>
32
Paul Bakker6083fd22011-12-03 21:45:14 +000033#if !defined(POLARSSL_CTR_DRBG_C) || !defined(POLARSSL_ENTROPY_C)
Paul Bakkercce9d772011-11-18 14:26:47 +000034int main( int argc, char *argv[] )
Paul Bakker5690efc2011-05-26 13:16:06 +000035{
Paul Bakkercce9d772011-11-18 14:26:47 +000036 ((void) argc);
37 ((void) argv);
38
Paul Bakker6083fd22011-12-03 21:45:14 +000039 printf("POLARSSL_CTR_DRBG_C or POLARSSL_ENTROPY_C not defined.\n");
Paul Bakker5690efc2011-05-26 13:16:06 +000040 return( 0 );
41}
42#else
Paul Bakkerfc36d162011-01-27 16:50:02 +000043int main( int argc, char *argv[] )
44{
45 FILE *f;
Paul Bakker6083fd22011-12-03 21:45:14 +000046 int i, k, ret;
47 ctr_drbg_context ctr_drbg;
48 entropy_context entropy;
Paul Bakkerfc36d162011-01-27 16:50:02 +000049 unsigned char buf[1024];
50
51 if( argc < 2 )
52 {
53 fprintf( stderr, "usage: %s <output filename>\n", argv[0] );
54 return( 1 );
55 }
56
57 if( ( f = fopen( argv[1], "wb+" ) ) == NULL )
58 {
59 printf( "failed to open '%s' for writing.\n", argv[0] );
60 return( 1 );
61 }
62
Paul Bakker6083fd22011-12-03 21:45:14 +000063 entropy_init( &entropy );
Paul Bakkerfb3a83f2011-12-15 20:05:53 +000064 ret = ctr_drbg_init( &ctr_drbg, entropy_func, &entropy, (unsigned char *) "RANDOM_GEN", 10 );
65 if( ret != 0 )
66 {
67 printf( "failed in ctr_drbg_init: %d\n", ret );
68 goto cleanup;
69 }
Paul Bakker6083fd22011-12-03 21:45:14 +000070 ctr_drbg_set_prediction_resistance( &ctr_drbg, CTR_DRBG_PR_OFF );
Paul Bakkerfc36d162011-01-27 16:50:02 +000071
Paul Bakkerfc754a92011-12-05 13:23:51 +000072#if defined(POLARSSL_FS_IO)
73 ret = ctr_drbg_update_seed_file( &ctr_drbg, "seedfile" );
74
Paul Bakker3f9b6502011-12-15 19:50:22 +000075 if( ret == POLARSSL_ERR_CTR_DRBG_FILE_IO_ERROR )
Paul Bakkerfc754a92011-12-05 13:23:51 +000076 {
Paul Bakkerfb3a83f2011-12-15 20:05:53 +000077 printf( "Failed to open seedfile. Generating one.\n" );
Paul Bakkerfc754a92011-12-05 13:23:51 +000078 ret = ctr_drbg_write_seed_file( &ctr_drbg, "seedfile" );
79 if( ret != 0 )
80 {
Paul Bakkerfb3a83f2011-12-15 20:05:53 +000081 printf( "failed in ctr_drbg_write_seed_file: %d\n", ret );
Paul Bakkerfc754a92011-12-05 13:23:51 +000082 goto cleanup;
83 }
84 }
85 else if( ret != 0 )
86 {
Paul Bakkerfb3a83f2011-12-15 20:05:53 +000087 printf( "failed in ctr_drbg_update_seed_file: %d\n", ret );
Paul Bakkerfc754a92011-12-05 13:23:51 +000088 goto cleanup;
89 }
90#endif
91
Paul Bakkerfc36d162011-01-27 16:50:02 +000092 for( i = 0, k = 768; i < k; i++ )
93 {
Paul Bakker6083fd22011-12-03 21:45:14 +000094 ret = ctr_drbg_random( &ctr_drbg, buf, sizeof( buf ) );
95 if( ret != 0 )
Paul Bakkera3d195c2011-11-27 21:07:34 +000096 {
Paul Bakker6083fd22011-12-03 21:45:14 +000097 printf("failed!\n");
98 goto cleanup;
Paul Bakkera3d195c2011-11-27 21:07:34 +000099 }
Paul Bakkerfc36d162011-01-27 16:50:02 +0000100
Paul Bakker6083fd22011-12-03 21:45:14 +0000101 fwrite( buf, 1, sizeof( buf ), f );
Paul Bakkerfc36d162011-01-27 16:50:02 +0000102
103 printf( "Generating 32Mb of data in file '%s'... %04.1f" \
104 "%% done\r", argv[1], (100 * (float) (i + 1)) / k );
105 fflush( stdout );
106 }
107
Paul Bakker6083fd22011-12-03 21:45:14 +0000108 ret = 0;
109
110cleanup:
Paul Bakkerfc754a92011-12-05 13:23:51 +0000111 printf("\n");
Paul Bakkerfc36d162011-01-27 16:50:02 +0000112
113 fclose( f );
Paul Bakker6083fd22011-12-03 21:45:14 +0000114
115 return( ret );
Paul Bakkerfc36d162011-01-27 16:50:02 +0000116}
Paul Bakker508ad5a2011-12-04 17:09:26 +0000117#endif /* POLARSSL_CTR_DRBG_C && POLARSSL_ENTROPY_C */