|  | /* | 
|  | *  Key writing application | 
|  | * | 
|  | *  Copyright (C) 2006-2015, ARM Limited, All Rights Reserved | 
|  | *  SPDX-License-Identifier: Apache-2.0 | 
|  | * | 
|  | *  Licensed under the Apache License, Version 2.0 (the "License"); you may | 
|  | *  not use this file except in compliance with the License. | 
|  | *  You may obtain a copy of the License at | 
|  | * | 
|  | *  http://www.apache.org/licenses/LICENSE-2.0 | 
|  | * | 
|  | *  Unless required by applicable law or agreed to in writing, software | 
|  | *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | 
|  | *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | *  See the License for the specific language governing permissions and | 
|  | *  limitations under the License. | 
|  | * | 
|  | *  This file is part of mbed TLS (https://tls.mbed.org) | 
|  | */ | 
|  |  | 
|  | #if !defined(MBEDTLS_CONFIG_FILE) | 
|  | #include "mbedtls/config.h" | 
|  | #else | 
|  | #include MBEDTLS_CONFIG_FILE | 
|  | #endif | 
|  |  | 
|  | #if defined(MBEDTLS_PLATFORM_C) | 
|  | #include "mbedtls/platform.h" | 
|  | #else | 
|  | #include <stdio.h> | 
|  | #define mbedtls_printf     printf | 
|  | #endif | 
|  |  | 
|  | #if defined(MBEDTLS_PK_WRITE_C) && defined(MBEDTLS_FS_IO) | 
|  | #include "mbedtls/error.h" | 
|  | #include "mbedtls/pk.h" | 
|  | #include "mbedtls/error.h" | 
|  |  | 
|  | #include <stdio.h> | 
|  | #include <string.h> | 
|  | #endif | 
|  |  | 
|  | #if defined(MBEDTLS_PEM_WRITE_C) | 
|  | #define USAGE_OUT \ | 
|  | "    output_file=%%s      default: keyfile.pem\n"   \ | 
|  | "    output_format=pem|der default: pem\n" | 
|  | #else | 
|  | #define USAGE_OUT \ | 
|  | "    output_file=%%s      default: keyfile.der\n"   \ | 
|  | "    output_format=der     default: der\n" | 
|  | #endif | 
|  |  | 
|  | #if defined(MBEDTLS_PEM_WRITE_C) | 
|  | #define DFL_OUTPUT_FILENAME     "keyfile.pem" | 
|  | #define DFL_OUTPUT_FORMAT       OUTPUT_FORMAT_PEM | 
|  | #else | 
|  | #define DFL_OUTPUT_FILENAME     "keyfile.der" | 
|  | #define DFL_OUTPUT_FORMAT       OUTPUT_FORMAT_DER | 
|  | #endif | 
|  |  | 
|  | #define DFL_MODE                MODE_NONE | 
|  | #define DFL_FILENAME            "keyfile.key" | 
|  | #define DFL_DEBUG_LEVEL         0 | 
|  | #define DFL_OUTPUT_MODE         OUTPUT_MODE_NONE | 
|  |  | 
|  | #define MODE_NONE               0 | 
|  | #define MODE_PRIVATE            1 | 
|  | #define MODE_PUBLIC             2 | 
|  |  | 
|  | #define OUTPUT_MODE_NONE               0 | 
|  | #define OUTPUT_MODE_PRIVATE            1 | 
|  | #define OUTPUT_MODE_PUBLIC             2 | 
|  |  | 
|  | #define OUTPUT_FORMAT_PEM              0 | 
|  | #define OUTPUT_FORMAT_DER              1 | 
|  |  | 
|  | #define USAGE \ | 
|  | "\n usage: key_app_writer param=<>...\n"            \ | 
|  | "\n acceptable parameters:\n"                       \ | 
|  | "    mode=private|public default: none\n"           \ | 
|  | "    filename=%%s         default: keyfile.key\n"   \ | 
|  | "    output_mode=private|public default: none\n"    \ | 
|  | USAGE_OUT                                           \ | 
|  | "\n" | 
|  |  | 
|  | #if !defined(MBEDTLS_PK_WRITE_C) || !defined(MBEDTLS_FS_IO) | 
|  | int main( void ) | 
|  | { | 
|  | mbedtls_printf( "MBEDTLS_PK_WRITE_C and/or MBEDTLS_FS_IO not defined.\n" ); | 
|  | return( 0 ); | 
|  | } | 
|  | #else | 
|  | /* | 
|  | * global options | 
|  | */ | 
|  | struct options | 
|  | { | 
|  | int mode;                   /* the mode to run the application in   */ | 
|  | const char *filename;       /* filename of the key file             */ | 
|  | int output_mode;            /* the output mode to use               */ | 
|  | const char *output_file;    /* where to store the constructed key file  */ | 
|  | int output_format;          /* the output format to use             */ | 
|  | } opt; | 
|  |  | 
|  | static int write_public_key( mbedtls_pk_context *key, const char *output_file ) | 
|  | { | 
|  | int ret; | 
|  | FILE *f; | 
|  | unsigned char output_buf[16000]; | 
|  | unsigned char *c = output_buf; | 
|  | size_t len = 0; | 
|  |  | 
|  | memset(output_buf, 0, 16000); | 
|  |  | 
|  | #if defined(MBEDTLS_PEM_WRITE_C) | 
|  | if( opt.output_format == OUTPUT_FORMAT_PEM ) | 
|  | { | 
|  | if( ( ret = mbedtls_pk_write_pubkey_pem( key, output_buf, 16000 ) ) != 0 ) | 
|  | return( ret ); | 
|  |  | 
|  | len = strlen( (char *) output_buf ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | { | 
|  | if( ( ret = mbedtls_pk_write_pubkey_der( key, output_buf, 16000 ) ) < 0 ) | 
|  | return( ret ); | 
|  |  | 
|  | len = ret; | 
|  | c = output_buf + sizeof(output_buf) - len - 1; | 
|  | } | 
|  |  | 
|  | if( ( f = fopen( output_file, "w" ) ) == NULL ) | 
|  | return( -1 ); | 
|  |  | 
|  | if( fwrite( c, 1, len, f ) != len ) | 
|  | { | 
|  | fclose( f ); | 
|  | return( -1 ); | 
|  | } | 
|  |  | 
|  | fclose( f ); | 
|  |  | 
|  | return( 0 ); | 
|  | } | 
|  |  | 
|  | static int write_private_key( mbedtls_pk_context *key, const char *output_file ) | 
|  | { | 
|  | int ret; | 
|  | FILE *f; | 
|  | unsigned char output_buf[16000]; | 
|  | unsigned char *c = output_buf; | 
|  | size_t len = 0; | 
|  |  | 
|  | memset(output_buf, 0, 16000); | 
|  |  | 
|  | #if defined(MBEDTLS_PEM_WRITE_C) | 
|  | if( opt.output_format == OUTPUT_FORMAT_PEM ) | 
|  | { | 
|  | if( ( ret = mbedtls_pk_write_key_pem( key, output_buf, 16000 ) ) != 0 ) | 
|  | return( ret ); | 
|  |  | 
|  | len = strlen( (char *) output_buf ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | { | 
|  | if( ( ret = mbedtls_pk_write_key_der( key, output_buf, 16000 ) ) < 0 ) | 
|  | return( ret ); | 
|  |  | 
|  | len = ret; | 
|  | c = output_buf + sizeof(output_buf) - len - 1; | 
|  | } | 
|  |  | 
|  | if( ( f = fopen( output_file, "w" ) ) == NULL ) | 
|  | return( -1 ); | 
|  |  | 
|  | if( fwrite( c, 1, len, f ) != len ) | 
|  | { | 
|  | fclose( f ); | 
|  | return( -1 ); | 
|  | } | 
|  |  | 
|  | fclose( f ); | 
|  |  | 
|  | return( 0 ); | 
|  | } | 
|  |  | 
|  | int main( int argc, char *argv[] ) | 
|  | { | 
|  | int ret = 0; | 
|  | char buf[1024]; | 
|  | int i; | 
|  | char *p, *q; | 
|  |  | 
|  | mbedtls_pk_context key; | 
|  | mbedtls_mpi N, P, Q, D, E, DP, DQ, QP; | 
|  |  | 
|  | /* | 
|  | * Set to sane values | 
|  | */ | 
|  | mbedtls_pk_init( &key ); | 
|  | memset( buf, 0, sizeof( buf ) ); | 
|  |  | 
|  | mbedtls_mpi_init( &N ); mbedtls_mpi_init( &P ); mbedtls_mpi_init( &Q ); | 
|  | mbedtls_mpi_init( &D ); mbedtls_mpi_init( &E ); mbedtls_mpi_init( &DP ); | 
|  | mbedtls_mpi_init( &DQ ); mbedtls_mpi_init( &QP ); | 
|  |  | 
|  | if( argc == 0 ) | 
|  | { | 
|  | usage: | 
|  | ret = 1; | 
|  | mbedtls_printf( USAGE ); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | opt.mode                = DFL_MODE; | 
|  | opt.filename            = DFL_FILENAME; | 
|  | opt.output_mode         = DFL_OUTPUT_MODE; | 
|  | opt.output_file         = DFL_OUTPUT_FILENAME; | 
|  | opt.output_format       = DFL_OUTPUT_FORMAT; | 
|  |  | 
|  | for( i = 1; i < argc; i++ ) | 
|  | { | 
|  | p = argv[i]; | 
|  | if( ( q = strchr( p, '=' ) ) == NULL ) | 
|  | goto usage; | 
|  | *q++ = '\0'; | 
|  |  | 
|  | if( strcmp( p, "mode" ) == 0 ) | 
|  | { | 
|  | if( strcmp( q, "private" ) == 0 ) | 
|  | opt.mode = MODE_PRIVATE; | 
|  | else if( strcmp( q, "public" ) == 0 ) | 
|  | opt.mode = MODE_PUBLIC; | 
|  | else | 
|  | goto usage; | 
|  | } | 
|  | else if( strcmp( p, "output_mode" ) == 0 ) | 
|  | { | 
|  | if( strcmp( q, "private" ) == 0 ) | 
|  | opt.output_mode = OUTPUT_MODE_PRIVATE; | 
|  | else if( strcmp( q, "public" ) == 0 ) | 
|  | opt.output_mode = OUTPUT_MODE_PUBLIC; | 
|  | else | 
|  | goto usage; | 
|  | } | 
|  | else if( strcmp( p, "output_format" ) == 0 ) | 
|  | { | 
|  | #if defined(MBEDTLS_PEM_WRITE_C) | 
|  | if( strcmp( q, "pem" ) == 0 ) | 
|  | opt.output_format = OUTPUT_FORMAT_PEM; | 
|  | else | 
|  | #endif | 
|  | if( strcmp( q, "der" ) == 0 ) | 
|  | opt.output_format = OUTPUT_FORMAT_DER; | 
|  | else | 
|  | goto usage; | 
|  | } | 
|  | else if( strcmp( p, "filename" ) == 0 ) | 
|  | opt.filename = q; | 
|  | else if( strcmp( p, "output_file" ) == 0 ) | 
|  | opt.output_file = q; | 
|  | else | 
|  | goto usage; | 
|  | } | 
|  |  | 
|  | if( opt.mode == MODE_NONE && opt.output_mode != OUTPUT_MODE_NONE ) | 
|  | { | 
|  | mbedtls_printf( "\nCannot output a key without reading one.\n"); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | if( opt.mode == MODE_PUBLIC && opt.output_mode == OUTPUT_MODE_PRIVATE ) | 
|  | { | 
|  | mbedtls_printf( "\nCannot output a private key from a public key.\n"); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | if( opt.mode == MODE_PRIVATE ) | 
|  | { | 
|  | /* | 
|  | * 1.1. Load the key | 
|  | */ | 
|  | mbedtls_printf( "\n  . Loading the private key ..." ); | 
|  | fflush( stdout ); | 
|  |  | 
|  | ret = mbedtls_pk_parse_keyfile( &key, opt.filename, NULL ); | 
|  |  | 
|  | if( ret != 0 ) | 
|  | { | 
|  | mbedtls_strerror( ret, (char *) buf, sizeof(buf) ); | 
|  | mbedtls_printf( " failed\n  !  mbedtls_pk_parse_keyfile returned -0x%04x - %s\n\n", -ret, buf ); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | mbedtls_printf( " ok\n" ); | 
|  |  | 
|  | /* | 
|  | * 1.2 Print the key | 
|  | */ | 
|  | mbedtls_printf( "  . Key information    ...\n" ); | 
|  |  | 
|  | #if defined(MBEDTLS_RSA_C) | 
|  | if( mbedtls_pk_get_type( &key ) == MBEDTLS_PK_RSA ) | 
|  | { | 
|  | mbedtls_rsa_context *rsa = mbedtls_pk_rsa( key ); | 
|  |  | 
|  | if( ( ret = mbedtls_rsa_export    ( rsa, &N, &P, &Q, &D, &E ) ) != 0 || | 
|  | ( ret = mbedtls_rsa_export_crt( rsa, &DP, &DQ, &QP ) )      != 0 ) | 
|  | { | 
|  | mbedtls_printf( " failed\n  ! could not export RSA parameters\n\n" ); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | mbedtls_mpi_write_file( "N:  ",  &N,  16, NULL ); | 
|  | mbedtls_mpi_write_file( "E:  ",  &E,  16, NULL ); | 
|  | mbedtls_mpi_write_file( "D:  ",  &D,  16, NULL ); | 
|  | mbedtls_mpi_write_file( "P:  ",  &P,  16, NULL ); | 
|  | mbedtls_mpi_write_file( "Q:  ",  &Q,  16, NULL ); | 
|  | mbedtls_mpi_write_file( "DP: ",  &DP, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "DQ:  ", &DQ, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "QP:  ", &QP, 16, NULL ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | #if defined(MBEDTLS_ECP_C) | 
|  | if( mbedtls_pk_get_type( &key ) == MBEDTLS_PK_ECKEY ) | 
|  | { | 
|  | mbedtls_ecp_keypair *ecp = mbedtls_pk_ec( key ); | 
|  | mbedtls_mpi_write_file( "Q(X): ", &ecp->Q.X, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "Q(Y): ", &ecp->Q.Y, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "Q(Z): ", &ecp->Q.Z, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "D   : ", &ecp->d  , 16, NULL ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | mbedtls_printf("key type not supported yet\n"); | 
|  |  | 
|  | } | 
|  | else if( opt.mode == MODE_PUBLIC ) | 
|  | { | 
|  | /* | 
|  | * 1.1. Load the key | 
|  | */ | 
|  | mbedtls_printf( "\n  . Loading the public key ..." ); | 
|  | fflush( stdout ); | 
|  |  | 
|  | ret = mbedtls_pk_parse_public_keyfile( &key, opt.filename ); | 
|  |  | 
|  | if( ret != 0 ) | 
|  | { | 
|  | mbedtls_strerror( ret, (char *) buf, sizeof(buf) ); | 
|  | mbedtls_printf( " failed\n  !  mbedtls_pk_parse_public_key returned -0x%04x - %s\n\n", -ret, buf ); | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | mbedtls_printf( " ok\n" ); | 
|  |  | 
|  | /* | 
|  | * 1.2 Print the key | 
|  | */ | 
|  | mbedtls_printf( "  . Key information    ...\n" ); | 
|  |  | 
|  | #if defined(MBEDTLS_RSA_C) | 
|  | if( mbedtls_pk_get_type( &key ) == MBEDTLS_PK_RSA ) | 
|  | { | 
|  | mbedtls_rsa_context *rsa = mbedtls_pk_rsa( key ); | 
|  |  | 
|  | if( ( ret = mbedtls_rsa_export( rsa, &N, NULL, NULL, | 
|  | NULL, &E ) ) != 0 ) | 
|  | { | 
|  | mbedtls_printf( " failed\n  ! could not export RSA parameters\n\n" ); | 
|  | goto exit; | 
|  | } | 
|  | mbedtls_mpi_write_file( "N: ", &N, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "E: ", &E, 16, NULL ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | #if defined(MBEDTLS_ECP_C) | 
|  | if( mbedtls_pk_get_type( &key ) == MBEDTLS_PK_ECKEY ) | 
|  | { | 
|  | mbedtls_ecp_keypair *ecp = mbedtls_pk_ec( key ); | 
|  | mbedtls_mpi_write_file( "Q(X): ", &ecp->Q.X, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "Q(Y): ", &ecp->Q.Y, 16, NULL ); | 
|  | mbedtls_mpi_write_file( "Q(Z): ", &ecp->Q.Z, 16, NULL ); | 
|  | } | 
|  | else | 
|  | #endif | 
|  | mbedtls_printf("key type not supported yet\n"); | 
|  | } | 
|  | else | 
|  | goto usage; | 
|  |  | 
|  | if( opt.output_mode == OUTPUT_MODE_PUBLIC ) | 
|  | { | 
|  | write_public_key( &key, opt.output_file ); | 
|  | } | 
|  | if( opt.output_mode == OUTPUT_MODE_PRIVATE ) | 
|  | { | 
|  | write_private_key( &key, opt.output_file ); | 
|  | } | 
|  |  | 
|  | exit: | 
|  |  | 
|  | if( ret != 0 && ret != 1) | 
|  | { | 
|  | #ifdef MBEDTLS_ERROR_C | 
|  | mbedtls_strerror( ret, buf, sizeof( buf ) ); | 
|  | mbedtls_printf( " - %s\n", buf ); | 
|  | #else | 
|  | mbedtls_printf("\n"); | 
|  | #endif | 
|  | } | 
|  |  | 
|  | mbedtls_mpi_free( &N ); mbedtls_mpi_free( &P ); mbedtls_mpi_free( &Q ); | 
|  | mbedtls_mpi_free( &D ); mbedtls_mpi_free( &E ); mbedtls_mpi_free( &DP ); | 
|  | mbedtls_mpi_free( &DQ ); mbedtls_mpi_free( &QP ); | 
|  |  | 
|  | mbedtls_pk_free( &key ); | 
|  |  | 
|  | #if defined(_WIN32) | 
|  | mbedtls_printf( "  + Press Enter to exit this program.\n" ); | 
|  | fflush( stdout ); getchar(); | 
|  | #endif | 
|  |  | 
|  | return( ret ); | 
|  | } | 
|  | #endif /* MBEDTLS_PK_WRITE_C && MBEDTLS_FS_IO */ |