Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 1 | /* |
| 2 | * RSA simple data encryption program |
| 3 | * |
Bence Szépkúti | 1e14827 | 2020-08-07 13:07:28 +0200 | [diff] [blame^] | 4 | * Copyright The Mbed TLS Contributors |
Manuel Pégourié-Gonnard | 37ff140 | 2015-09-04 14:21:07 +0200 | [diff] [blame] | 5 | * SPDX-License-Identifier: Apache-2.0 |
| 6 | * |
| 7 | * Licensed under the Apache License, Version 2.0 (the "License"); you may |
| 8 | * not use this file except in compliance with the License. |
| 9 | * You may obtain a copy of the License at |
| 10 | * |
| 11 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 12 | * |
| 13 | * Unless required by applicable law or agreed to in writing, software |
| 14 | * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| 15 | * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 16 | * See the License for the specific language governing permissions and |
| 17 | * limitations under the License. |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 18 | */ |
| 19 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 20 | #if !defined(MBEDTLS_CONFIG_FILE) |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 21 | #include "mbedtls/config.h" |
Manuel Pégourié-Gonnard | cef4ad2 | 2014-04-29 12:39:06 +0200 | [diff] [blame] | 22 | #else |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 23 | #include MBEDTLS_CONFIG_FILE |
Manuel Pégourié-Gonnard | cef4ad2 | 2014-04-29 12:39:06 +0200 | [diff] [blame] | 24 | #endif |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 25 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 26 | #if defined(MBEDTLS_PLATFORM_C) |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 27 | #include "mbedtls/platform.h" |
Rich Evans | f90016a | 2015-01-19 14:26:37 +0000 | [diff] [blame] | 28 | #else |
Rich Evans | 18b78c7 | 2015-02-11 14:06:19 +0000 | [diff] [blame] | 29 | #include <stdio.h> |
Simon Butcher | 1ef918d | 2016-04-13 11:56:27 +0100 | [diff] [blame] | 30 | #include <stdlib.h> |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 31 | #define mbedtls_fprintf fprintf |
| 32 | #define mbedtls_printf printf |
| 33 | #define mbedtls_exit exit |
| 34 | #define MBEDTLS_EXIT_SUCCESS EXIT_SUCCESS |
| 35 | #define MBEDTLS_EXIT_FAILURE EXIT_FAILURE |
| 36 | #endif /* MBEDTLS_PLATFORM_C */ |
Rich Evans | f90016a | 2015-01-19 14:26:37 +0000 | [diff] [blame] | 37 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 38 | #if defined(MBEDTLS_BIGNUM_C) && defined(MBEDTLS_RSA_C) && \ |
| 39 | defined(MBEDTLS_ENTROPY_C) && defined(MBEDTLS_FS_IO) && \ |
| 40 | defined(MBEDTLS_CTR_DRBG_C) |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 41 | #include "mbedtls/rsa.h" |
| 42 | #include "mbedtls/entropy.h" |
| 43 | #include "mbedtls/ctr_drbg.h" |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 44 | |
Rich Evans | 18b78c7 | 2015-02-11 14:06:19 +0000 | [diff] [blame] | 45 | #include <string.h> |
| 46 | #endif |
| 47 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 48 | #if !defined(MBEDTLS_BIGNUM_C) || !defined(MBEDTLS_RSA_C) || \ |
| 49 | !defined(MBEDTLS_ENTROPY_C) || !defined(MBEDTLS_FS_IO) || \ |
| 50 | !defined(MBEDTLS_CTR_DRBG_C) |
Rich Evans | 85b05ec | 2015-02-12 11:37:29 +0000 | [diff] [blame] | 51 | int main( void ) |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 52 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 53 | mbedtls_printf("MBEDTLS_BIGNUM_C and/or MBEDTLS_RSA_C and/or " |
| 54 | "MBEDTLS_ENTROPY_C and/or MBEDTLS_FS_IO and/or " |
| 55 | "MBEDTLS_CTR_DRBG_C not defined.\n"); |
Krzysztof Stachowiak | 5e1b195 | 2019-04-24 14:24:46 +0200 | [diff] [blame] | 56 | mbedtls_exit( 0 ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 57 | } |
| 58 | #else |
Simon Butcher | 63cb97e | 2018-12-06 17:43:31 +0000 | [diff] [blame] | 59 | |
Simon Butcher | 63cb97e | 2018-12-06 17:43:31 +0000 | [diff] [blame] | 60 | |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 61 | int main( int argc, char *argv[] ) |
| 62 | { |
| 63 | FILE *f; |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 64 | int ret = 1; |
| 65 | int exit_code = MBEDTLS_EXIT_FAILURE; |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 66 | size_t i; |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 67 | mbedtls_rsa_context rsa; |
| 68 | mbedtls_entropy_context entropy; |
| 69 | mbedtls_ctr_drbg_context ctr_drbg; |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 70 | unsigned char input[1024]; |
| 71 | unsigned char buf[512]; |
Paul Bakker | ef3f8c7 | 2013-06-24 13:01:08 +0200 | [diff] [blame] | 72 | const char *pers = "rsa_encrypt"; |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 73 | mbedtls_mpi N, E; |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 74 | |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 75 | if( argc != 2 ) |
| 76 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 77 | mbedtls_printf( "usage: rsa_encrypt <string of max 100 characters>\n" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 78 | |
Paul Bakker | cce9d77 | 2011-11-18 14:26:47 +0000 | [diff] [blame] | 79 | #if defined(_WIN32) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 80 | mbedtls_printf( "\n" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 81 | #endif |
| 82 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 83 | mbedtls_exit( exit_code ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 84 | } |
| 85 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 86 | mbedtls_printf( "\n . Seeding the random number generator..." ); |
Paul Bakker | 508ad5a | 2011-12-04 17:09:26 +0000 | [diff] [blame] | 87 | fflush( stdout ); |
| 88 | |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 89 | mbedtls_mpi_init( &N ); mbedtls_mpi_init( &E ); |
Simon Butcher | 6b46c62 | 2016-04-12 13:25:08 +0100 | [diff] [blame] | 90 | mbedtls_rsa_init( &rsa, MBEDTLS_RSA_PKCS_V15, 0 ); |
| 91 | mbedtls_ctr_drbg_init( &ctr_drbg ); |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 92 | mbedtls_entropy_init( &entropy ); |
Simon Butcher | 6b46c62 | 2016-04-12 13:25:08 +0100 | [diff] [blame] | 93 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 94 | ret = mbedtls_ctr_drbg_seed( &ctr_drbg, mbedtls_entropy_func, |
| 95 | &entropy, (const unsigned char *) pers, |
| 96 | strlen( pers ) ); |
| 97 | if( ret != 0 ) |
Paul Bakker | 508ad5a | 2011-12-04 17:09:26 +0000 | [diff] [blame] | 98 | { |
Simon Butcher | 6b46c62 | 2016-04-12 13:25:08 +0100 | [diff] [blame] | 99 | mbedtls_printf( " failed\n ! mbedtls_ctr_drbg_seed returned %d\n", |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 100 | ret ); |
Paul Bakker | 508ad5a | 2011-12-04 17:09:26 +0000 | [diff] [blame] | 101 | goto exit; |
| 102 | } |
| 103 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 104 | mbedtls_printf( "\n . Reading public key from rsa_pub.txt" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 105 | fflush( stdout ); |
| 106 | |
Paul Bakker | d246ed3 | 2011-10-06 13:18:27 +0000 | [diff] [blame] | 107 | if( ( f = fopen( "rsa_pub.txt", "rb" ) ) == NULL ) |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 108 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 109 | mbedtls_printf( " failed\n ! Could not open rsa_pub.txt\n" \ |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 110 | " ! Please run rsa_genkey first\n\n" ); |
| 111 | goto exit; |
| 112 | } |
| 113 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 114 | if( ( ret = mbedtls_mpi_read_file( &N, 16, f ) ) != 0 || |
| 115 | ( ret = mbedtls_mpi_read_file( &E, 16, f ) ) != 0 ) |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 116 | { |
Simon Butcher | 6b46c62 | 2016-04-12 13:25:08 +0100 | [diff] [blame] | 117 | mbedtls_printf( " failed\n ! mbedtls_mpi_read_file returned %d\n\n", |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 118 | ret ); |
Janos Follath | 98e28a7 | 2016-05-31 14:03:54 +0100 | [diff] [blame] | 119 | fclose( f ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 120 | goto exit; |
| 121 | } |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 122 | fclose( f ); |
| 123 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 124 | if( ( ret = mbedtls_rsa_import( &rsa, &N, NULL, NULL, NULL, &E ) ) != 0 ) |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 125 | { |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 126 | mbedtls_printf( " failed\n ! mbedtls_rsa_import returned %d\n\n", |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 127 | ret ); |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 128 | goto exit; |
| 129 | } |
| 130 | |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 131 | if( strlen( argv[1] ) > 100 ) |
| 132 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 133 | mbedtls_printf( " Input data larger than 100 characters.\n\n" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 134 | goto exit; |
| 135 | } |
| 136 | |
| 137 | memcpy( input, argv[1], strlen( argv[1] ) ); |
| 138 | |
| 139 | /* |
| 140 | * Calculate the RSA encryption of the hash. |
| 141 | */ |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 142 | mbedtls_printf( "\n . Generating the RSA encrypted value" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 143 | fflush( stdout ); |
| 144 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 145 | ret = mbedtls_rsa_pkcs1_encrypt( &rsa, mbedtls_ctr_drbg_random, |
| 146 | &ctr_drbg, MBEDTLS_RSA_PUBLIC, |
| 147 | strlen( argv[1] ), input, buf ); |
| 148 | if( ret != 0 ) |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 149 | { |
Simon Butcher | 6b46c62 | 2016-04-12 13:25:08 +0100 | [diff] [blame] | 150 | mbedtls_printf( " failed\n ! mbedtls_rsa_pkcs1_encrypt returned %d\n\n", |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 151 | ret ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 152 | goto exit; |
| 153 | } |
| 154 | |
| 155 | /* |
| 156 | * Write the signature into result-enc.txt |
| 157 | */ |
| 158 | if( ( f = fopen( "result-enc.txt", "wb+" ) ) == NULL ) |
| 159 | { |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 160 | mbedtls_printf( " failed\n ! Could not create %s\n\n", "result-enc.txt" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 161 | goto exit; |
| 162 | } |
| 163 | |
| 164 | for( i = 0; i < rsa.len; i++ ) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 165 | mbedtls_fprintf( f, "%02X%s", buf[i], |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 166 | ( i + 1 ) % 16 == 0 ? "\r\n" : " " ); |
| 167 | |
| 168 | fclose( f ); |
| 169 | |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 170 | mbedtls_printf( "\n . Done (created \"%s\")\n\n", "result-enc.txt" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 171 | |
Andres Amaya Garcia | 25b5af5 | 2018-04-30 22:08:36 +0100 | [diff] [blame] | 172 | exit_code = MBEDTLS_EXIT_SUCCESS; |
| 173 | |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 174 | exit: |
Hanno Becker | 0c26393 | 2017-08-23 06:47:06 +0100 | [diff] [blame] | 175 | mbedtls_mpi_free( &N ); mbedtls_mpi_free( &E ); |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 176 | mbedtls_ctr_drbg_free( &ctr_drbg ); |
| 177 | mbedtls_entropy_free( &entropy ); |
Janos Follath | f713b0a | 2016-03-17 15:21:39 +0000 | [diff] [blame] | 178 | mbedtls_rsa_free( &rsa ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 179 | |
Paul Bakker | cce9d77 | 2011-11-18 14:26:47 +0000 | [diff] [blame] | 180 | #if defined(_WIN32) |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 181 | mbedtls_printf( " + Press Enter to exit this program.\n" ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 182 | fflush( stdout ); getchar(); |
| 183 | #endif |
| 184 | |
Krzysztof Stachowiak | 5e1b195 | 2019-04-24 14:24:46 +0200 | [diff] [blame] | 185 | mbedtls_exit( exit_code ); |
Paul Bakker | 7bc05ff | 2011-08-09 10:30:36 +0000 | [diff] [blame] | 186 | } |
Manuel Pégourié-Gonnard | 2cf5a7c | 2015-04-08 12:49:31 +0200 | [diff] [blame] | 187 | #endif /* MBEDTLS_BIGNUM_C && MBEDTLS_RSA_C && MBEDTLS_ENTROPY_C && |
| 188 | MBEDTLS_FS_IO && MBEDTLS_CTR_DRBG_C */ |