blob: b84d047d8456f84986267c116ab747b4b2f75bd9 [file] [log] [blame]
Paul Bakker940f9ce2013-09-18 15:34:57 +02001/*
2 * Public key-based signature creation program
3 *
Manuel Pégourié-Gonnarda658a402015-01-23 09:45:19 +00004 * Copyright (C) 2006-2013, ARM Limited, All Rights Reserved
Paul Bakker940f9ce2013-09-18 15:34:57 +02005 *
Manuel Pégourié-Gonnardfe446432015-03-06 13:17:10 +00006 * This file is part of mbed TLS (https://tls.mbed.org)
Paul Bakker940f9ce2013-09-18 15:34:57 +02007 *
Paul Bakker940f9ce2013-09-18 15:34:57 +02008 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 */
22
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020023#if !defined(MBEDTLS_CONFIG_FILE)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000024#include "mbedtls/config.h"
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020025#else
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020026#include MBEDTLS_CONFIG_FILE
Manuel Pégourié-Gonnardcef4ad22014-04-29 12:39:06 +020027#endif
Paul Bakker940f9ce2013-09-18 15:34:57 +020028
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020029#if defined(MBEDTLS_PLATFORM_C)
Manuel Pégourié-Gonnard7f809972015-03-09 17:05:11 +000030#include "mbedtls/platform.h"
Rich Evansf90016a2015-01-19 14:26:37 +000031#else
Rich Evans18b78c72015-02-11 14:06:19 +000032#include <stdio.h>
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020033#define mbedtls_snprintf snprintf
34#define mbedtls_printf printf
35#define mbedtls_snprintf snprintf
Rich Evansf90016a2015-01-19 14:26:37 +000036#endif
37
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020038#if !defined(MBEDTLS_BIGNUM_C) || !defined(MBEDTLS_ENTROPY_C) || \
Manuel Pégourié-Gonnard06d5d612015-05-28 16:23:18 +020039 !defined(MBEDTLS_SHA256_C) || !defined(MBEDTLS_MD_C) || \
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020040 !defined(MBEDTLS_PK_PARSE_C) || !defined(MBEDTLS_FS_IO) || \
41 !defined(MBEDTLS_CTR_DRBG_C)
Rich Evans85b05ec2015-02-12 11:37:29 +000042int main( void )
Paul Bakker940f9ce2013-09-18 15:34:57 +020043{
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020044 mbedtls_printf("MBEDTLS_BIGNUM_C and/or MBEDTLS_ENTROPY_C and/or "
Manuel Pégourié-Gonnard06d5d612015-05-28 16:23:18 +020045 "MBEDTLS_SHA256_C and/or MBEDLTS_MD_C and/or "
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020046 "MBEDTLS_PK_PARSE_C and/or MBEDTLS_FS_IO and/or "
47 "MBEDTLS_CTR_DRBG_C not defined.\n");
Paul Bakker940f9ce2013-09-18 15:34:57 +020048 return( 0 );
49}
50#else
Manuel Pégourié-Gonnard06d5d612015-05-28 16:23:18 +020051
52#include "mbedtls/error.h"
53#include "mbedtls/entropy.h"
54#include "mbedtls/ctr_drbg.h"
55#include "mbedtls/md.h"
56#include "mbedtls/pk.h"
57
58#include <stdio.h>
59#include <string.h>
60
61#if defined _MSC_VER && !defined snprintf
62#define snprintf _snprintf
63#endif
64
Paul Bakker940f9ce2013-09-18 15:34:57 +020065int main( int argc, char *argv[] )
66{
67 FILE *f;
Paul Bakker0c226102014-04-17 16:02:36 +020068 int ret = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020069 mbedtls_pk_context pk;
70 mbedtls_entropy_context entropy;
71 mbedtls_ctr_drbg_context ctr_drbg;
Paul Bakker940f9ce2013-09-18 15:34:57 +020072 unsigned char hash[20];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020073 unsigned char buf[MBEDTLS_MPI_MAX_SIZE];
Paul Bakker940f9ce2013-09-18 15:34:57 +020074 char filename[512];
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020075 const char *pers = "mbedtls_pk_sign";
Paul Bakker940f9ce2013-09-18 15:34:57 +020076 size_t olen = 0;
77
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020078 mbedtls_entropy_init( &entropy );
Manuel Pégourié-Gonnardec160c02015-04-28 22:52:30 +020079 mbedtls_ctr_drbg_init( &ctr_drbg );
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020080 mbedtls_pk_init( &pk );
Paul Bakker940f9ce2013-09-18 15:34:57 +020081
82 if( argc != 3 )
83 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020084 mbedtls_printf( "usage: mbedtls_pk_sign <key_file> <filename>\n" );
Paul Bakker940f9ce2013-09-18 15:34:57 +020085
86#if defined(_WIN32)
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020087 mbedtls_printf( "\n" );
Paul Bakker940f9ce2013-09-18 15:34:57 +020088#endif
89
90 goto exit;
91 }
92
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +020093 mbedtls_printf( "\n . Seeding the random number generator..." );
Paul Bakker940f9ce2013-09-18 15:34:57 +020094 fflush( stdout );
95
Manuel Pégourié-Gonnardec160c02015-04-28 22:52:30 +020096 if( ( ret = mbedtls_ctr_drbg_seed( &ctr_drbg, mbedtls_entropy_func, &entropy,
Paul Bakker940f9ce2013-09-18 15:34:57 +020097 (const unsigned char *) pers,
98 strlen( pers ) ) ) != 0 )
99 {
Manuel Pégourié-Gonnardec160c02015-04-28 22:52:30 +0200100 mbedtls_printf( " failed\n ! mbedtls_ctr_drbg_seed returned -0x%04x\n", -ret );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200101 goto exit;
102 }
103
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200104 mbedtls_printf( "\n . Reading private key from '%s'", argv[1] );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200105 fflush( stdout );
106
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200107 if( ( ret = mbedtls_pk_parse_keyfile( &pk, argv[1], "" ) ) != 0 )
Paul Bakker940f9ce2013-09-18 15:34:57 +0200108 {
109 ret = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200110 mbedtls_printf( " failed\n ! Could not open '%s'\n", argv[1] );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200111 goto exit;
112 }
113
114 /*
Manuel Pégourié-Gonnard6f60cd82015-02-10 10:47:03 +0000115 * Compute the SHA-256 hash of the input file,
Paul Bakker940f9ce2013-09-18 15:34:57 +0200116 * then calculate the signature of the hash.
117 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200118 mbedtls_printf( "\n . Generating the SHA-256 signature" );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200119 fflush( stdout );
120
Manuel Pégourié-Gonnard06d5d612015-05-28 16:23:18 +0200121 if( ( ret = mbedtls_md_file(
122 mbedtls_md_info_from_type( MBEDTLS_MD_SHA256 ),
123 argv[2], hash ) ) != 0 )
Paul Bakker940f9ce2013-09-18 15:34:57 +0200124 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200125 mbedtls_printf( " failed\n ! Could not open or read %s\n\n", argv[2] );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200126 goto exit;
127 }
128
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200129 if( ( ret = mbedtls_pk_sign( &pk, MBEDTLS_MD_SHA256, hash, 0, buf, &olen,
130 mbedtls_ctr_drbg_random, &ctr_drbg ) ) != 0 )
Paul Bakker940f9ce2013-09-18 15:34:57 +0200131 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200132 mbedtls_printf( " failed\n ! mbedtls_pk_sign returned -0x%04x\n", -ret );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200133 goto exit;
134 }
135
136 /*
137 * Write the signature into <filename>-sig.txt
138 */
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200139 mbedtls_snprintf( filename, sizeof(filename), "%s.sig", argv[2] );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200140
141 if( ( f = fopen( filename, "wb+" ) ) == NULL )
142 {
143 ret = 1;
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200144 mbedtls_printf( " failed\n ! Could not create %s\n\n", filename );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200145 goto exit;
146 }
147
148 if( fwrite( buf, 1, olen, f ) != olen )
149 {
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200150 mbedtls_printf( "failed\n ! fwrite failed\n\n" );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200151 goto exit;
152 }
153
154 fclose( f );
155
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200156 mbedtls_printf( "\n . Done (created \"%s\")\n\n", filename );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200157
158exit:
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200159 mbedtls_pk_free( &pk );
160 mbedtls_ctr_drbg_free( &ctr_drbg );
161 mbedtls_entropy_free( &entropy );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200162
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200163#if defined(MBEDTLS_ERROR_C)
164 mbedtls_strerror( ret, (char *) buf, sizeof(buf) );
165 mbedtls_printf( " ! Last error was: %s\n", buf );
Manuel Pégourié-Gonnard92e5b592013-09-18 18:57:10 +0200166#endif
167
Paul Bakker940f9ce2013-09-18 15:34:57 +0200168#if defined(_WIN32)
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200169 mbedtls_printf( " + Press Enter to exit this program.\n" );
Paul Bakker940f9ce2013-09-18 15:34:57 +0200170 fflush( stdout ); getchar();
171#endif
172
173 return( ret );
174}
Manuel Pégourié-Gonnard2cf5a7c2015-04-08 12:49:31 +0200175#endif /* MBEDTLS_BIGNUM_C && MBEDTLS_ENTROPY_C &&
176 MBEDTLS_SHA256_C && MBEDTLS_PK_PARSE_C && MBEDTLS_FS_IO &&
177 MBEDTLS_CTR_DRBG_C */