| /* |
| * Key reading application |
| * |
| * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved |
| * |
| * This file is part of mbed TLS (https://tls.mbed.org) |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License along |
| * with this program; if not, write to the Free Software Foundation, Inc., |
| * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| */ |
| |
| #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_BIGNUM_C) && \ |
| defined(MBEDTLS_PK_PARSE_C) && defined(MBEDTLS_FS_IO) |
| #include "mbedtls/error.h" |
| #include "mbedtls/rsa.h" |
| #include "mbedtls/x509.h" |
| |
| #include <string.h> |
| #endif |
| |
| #define MODE_NONE 0 |
| #define MODE_PRIVATE 1 |
| #define MODE_PUBLIC 2 |
| |
| #define DFL_MODE MODE_NONE |
| #define DFL_FILENAME "keyfile.key" |
| #define DFL_PASSWORD "" |
| #define DFL_PASSWORD_FILE "" |
| #define DFL_DEBUG_LEVEL 0 |
| |
| #define USAGE \ |
| "\n usage: key_app param=<>...\n" \ |
| "\n acceptable parameters:\n" \ |
| " mode=private|public default: none\n" \ |
| " filename=%%s default: keyfile.key\n" \ |
| " password=%%s default: \"\"\n" \ |
| " password_file=%%s default: \"\"\n" \ |
| "\n" |
| |
| |
| #if !defined(MBEDTLS_BIGNUM_C) || \ |
| !defined(MBEDTLS_PK_PARSE_C) || !defined(MBEDTLS_FS_IO) |
| int main( void ) |
| { |
| mbedtls_printf("MBEDTLS_BIGNUM_C and/or " |
| "MBEDTLS_PK_PARSE_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 */ |
| const char *password; /* password for the private key */ |
| const char *password_file; /* password_file for the private key */ |
| } opt; |
| |
| int main( int argc, char *argv[] ) |
| { |
| int ret = 0; |
| mbedtls_pk_context pk; |
| char buf[1024]; |
| int i; |
| char *p, *q; |
| |
| /* |
| * Set to sane values |
| */ |
| mbedtls_pk_init( &pk ); |
| memset( buf, 0, sizeof(buf) ); |
| |
| if( argc == 0 ) |
| { |
| usage: |
| mbedtls_printf( USAGE ); |
| goto exit; |
| } |
| |
| opt.mode = DFL_MODE; |
| opt.filename = DFL_FILENAME; |
| opt.password = DFL_PASSWORD; |
| opt.password_file = DFL_PASSWORD_FILE; |
| |
| 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, "filename" ) == 0 ) |
| opt.filename = q; |
| else if( strcmp( p, "password" ) == 0 ) |
| opt.password = q; |
| else if( strcmp( p, "password_file" ) == 0 ) |
| opt.password_file = q; |
| else |
| goto usage; |
| } |
| |
| if( opt.mode == MODE_PRIVATE ) |
| { |
| if( strlen( opt.password ) && strlen( opt.password_file ) ) |
| { |
| mbedtls_printf( "Error: cannot have both password and password_file\n" ); |
| goto usage; |
| } |
| |
| if( strlen( opt.password_file ) ) |
| { |
| FILE *f; |
| |
| mbedtls_printf( "\n . Loading the password file ..." ); |
| if( ( f = fopen( opt.password_file, "rb" ) ) == NULL ) |
| { |
| mbedtls_printf( " failed\n ! fopen returned NULL\n" ); |
| goto exit; |
| } |
| if( fgets( buf, sizeof(buf), f ) == NULL ) |
| { |
| fclose( f ); |
| mbedtls_printf( "Error: fgets() failed to retrieve password\n" ); |
| goto exit; |
| } |
| fclose( f ); |
| |
| i = (int) strlen( buf ); |
| if( buf[i - 1] == '\n' ) buf[i - 1] = '\0'; |
| if( buf[i - 2] == '\r' ) buf[i - 2] = '\0'; |
| opt.password = buf; |
| } |
| |
| /* |
| * 1.1. Load the key |
| */ |
| mbedtls_printf( "\n . Loading the private key ..." ); |
| fflush( stdout ); |
| |
| ret = mbedtls_pk_parse_keyfile( &pk, opt.filename, opt.password ); |
| |
| if( ret != 0 ) |
| { |
| mbedtls_printf( " failed\n ! mbedtls_pk_parse_keyfile returned -0x%04x\n", -ret ); |
| 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( &pk ) == MBEDTLS_PK_RSA ) |
| { |
| mbedtls_rsa_context *rsa = mbedtls_pk_rsa( pk ); |
| mbedtls_mpi_write_file( "N: ", &rsa->N, 16, NULL ); |
| mbedtls_mpi_write_file( "E: ", &rsa->E, 16, NULL ); |
| mbedtls_mpi_write_file( "D: ", &rsa->D, 16, NULL ); |
| mbedtls_mpi_write_file( "P: ", &rsa->P, 16, NULL ); |
| mbedtls_mpi_write_file( "Q: ", &rsa->Q, 16, NULL ); |
| mbedtls_mpi_write_file( "DP: ", &rsa->DP, 16, NULL ); |
| mbedtls_mpi_write_file( "DQ: ", &rsa->DQ, 16, NULL ); |
| mbedtls_mpi_write_file( "QP: ", &rsa->QP, 16, NULL ); |
| } |
| else |
| #endif |
| #if defined(MBEDTLS_ECP_C) |
| if( mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_ECKEY ) |
| { |
| mbedtls_ecp_keypair *ecp = mbedtls_pk_ec( pk ); |
| 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("Do not know how to print key information for this type\n" ); |
| goto exit; |
| } |
| } |
| 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( &pk, opt.filename ); |
| |
| if( ret != 0 ) |
| { |
| mbedtls_printf( " failed\n ! mbedtls_pk_parse_public_keyfile returned -0x%04x\n", -ret ); |
| goto exit; |
| } |
| |
| mbedtls_printf( " ok\n" ); |
| |
| mbedtls_printf( " . Key information ...\n" ); |
| #if defined(MBEDTLS_RSA_C) |
| if( mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_RSA ) |
| { |
| mbedtls_rsa_context *rsa = mbedtls_pk_rsa( pk ); |
| mbedtls_mpi_write_file( "N: ", &rsa->N, 16, NULL ); |
| mbedtls_mpi_write_file( "E: ", &rsa->E, 16, NULL ); |
| } |
| else |
| #endif |
| #if defined(MBEDTLS_ECP_C) |
| if( mbedtls_pk_get_type( &pk ) == MBEDTLS_PK_ECKEY ) |
| { |
| mbedtls_ecp_keypair *ecp = mbedtls_pk_ec( pk ); |
| 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("Do not know how to print key information for this type\n" ); |
| goto exit; |
| } |
| } |
| else |
| goto usage; |
| |
| exit: |
| |
| #if defined(MBEDTLS_ERROR_C) |
| mbedtls_strerror( ret, buf, sizeof(buf) ); |
| mbedtls_printf( " ! Last error was: %s\n", buf ); |
| #endif |
| |
| mbedtls_pk_free( &pk ); |
| |
| #if defined(_WIN32) |
| mbedtls_printf( " + Press Enter to exit this program.\n" ); |
| fflush( stdout ); getchar(); |
| #endif |
| |
| return( ret ); |
| } |
| #endif /* MBEDTLS_BIGNUM_C && MBEDTLS_PK_PARSE_C && MBEDTLS_FS_IO */ |