1 /*
2  *  Certificate reading application
3  *
4  *  Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
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.
18  *
19  *  This file is part of mbed TLS (https://tls.mbed.org)
20  */
21 
22 #if !defined(MBEDTLS_CONFIG_FILE)
23 #include "mbedtls/config.h"
24 #else
25 #include MBEDTLS_CONFIG_FILE
26 #endif
27 
28 #if defined(MBEDTLS_PLATFORM_C)
29 #include "mbedtls/platform.h"
30 #else
31 #include <stdio.h>
32 #include <stdlib.h>
33 #define mbedtls_time       time
34 #define mbedtls_time_t     time_t
35 #define mbedtls_fprintf    fprintf
36 #define mbedtls_printf     printf
37 #endif
38 
39 #if !defined(MBEDTLS_BIGNUM_C) || !defined(MBEDTLS_ENTROPY_C) ||  \
40     !defined(MBEDTLS_SSL_TLS_C) || !defined(MBEDTLS_SSL_CLI_C) || \
41     !defined(MBEDTLS_NET_C) || !defined(MBEDTLS_RSA_C) ||         \
42     !defined(MBEDTLS_X509_CRT_PARSE_C) || !defined(MBEDTLS_FS_IO) ||  \
43     !defined(MBEDTLS_CTR_DRBG_C)
main(void)44 int main( void )
45 {
46     mbedtls_printf("MBEDTLS_BIGNUM_C and/or MBEDTLS_ENTROPY_C and/or "
47            "MBEDTLS_SSL_TLS_C and/or MBEDTLS_SSL_CLI_C and/or "
48            "MBEDTLS_NET_C and/or MBEDTLS_RSA_C and/or "
49            "MBEDTLS_X509_CRT_PARSE_C and/or MBEDTLS_FS_IO and/or "
50            "MBEDTLS_CTR_DRBG_C not defined.\n");
51     return( 0 );
52 }
53 #else
54 
55 #include "mbedtls/entropy.h"
56 #include "mbedtls/ctr_drbg.h"
57 #include "mbedtls/net_sockets.h"
58 #include "mbedtls/ssl.h"
59 #include "mbedtls/x509.h"
60 #include "mbedtls/debug.h"
61 
62 #include <stdio.h>
63 #include <stdlib.h>
64 #include <string.h>
65 
66 #define MODE_NONE               0
67 #define MODE_FILE               1
68 #define MODE_SSL                2
69 
70 #define DFL_MODE                MODE_NONE
71 #define DFL_FILENAME            "cert.crt"
72 #define DFL_CA_FILE             ""
73 #define DFL_CRL_FILE            ""
74 #define DFL_CA_PATH             ""
75 #define DFL_SERVER_NAME         "localhost"
76 #define DFL_SERVER_PORT         "4433"
77 #define DFL_DEBUG_LEVEL         0
78 #define DFL_PERMISSIVE          0
79 
80 #define USAGE_IO \
81     "    ca_file=%%s          The single file containing the top-level CA(s) you fully trust\n" \
82     "                        default: \"\" (none)\n" \
83     "    crl_file=%%s         The single CRL file you want to use\n" \
84     "                        default: \"\" (none)\n" \
85     "    ca_path=%%s          The path containing the top-level CA(s) you fully trust\n" \
86     "                        default: \"\" (none) (overrides ca_file)\n"
87 
88 #define USAGE \
89     "\n usage: cert_app param=<>...\n"                  \
90     "\n acceptable parameters:\n"                       \
91     "    mode=file|ssl       default: none\n"           \
92     "    filename=%%s         default: cert.crt\n"      \
93     USAGE_IO                                            \
94     "    server_name=%%s      default: localhost\n"     \
95     "    server_port=%%d      default: 4433\n"          \
96     "    debug_level=%%d      default: 0 (disabled)\n"  \
97     "    permissive=%%d       default: 0 (disabled)\n"  \
98     "\n"
99 
100 /*
101  * global options
102  */
103 struct options
104 {
105     int mode;                   /* the mode to run the application in   */
106     const char *filename;       /* filename of the certificate file     */
107     const char *ca_file;        /* the file with the CA certificate(s)  */
108     const char *crl_file;       /* the file with the CRL to use         */
109     const char *ca_path;        /* the path with the CA certificate(s) reside */
110     const char *server_name;    /* hostname of the server (client only) */
111     const char *server_port;    /* port on which the ssl service runs   */
112     int debug_level;            /* level of debugging                   */
113     int permissive;             /* permissive parsing                   */
114 } opt;
115 
my_debug(void * ctx,int level,const char * file,int line,const char * str)116 static void my_debug( void *ctx, int level,
117                       const char *file, int line,
118                       const char *str )
119 {
120     ((void) level);
121 
122     mbedtls_fprintf( (FILE *) ctx, "%s:%04d: %s", file, line, str );
123     fflush(  (FILE *) ctx  );
124 }
125 
my_verify(void * data,mbedtls_x509_crt * crt,int depth,uint32_t * flags)126 static int my_verify( void *data, mbedtls_x509_crt *crt, int depth, uint32_t *flags )
127 {
128     char buf[1024];
129     ((void) data);
130 
131     mbedtls_printf( "\nVerify requested for (Depth %d):\n", depth );
132     mbedtls_x509_crt_info( buf, sizeof( buf ) - 1, "", crt );
133     mbedtls_printf( "%s", buf );
134 
135     if ( ( *flags ) == 0 )
136         mbedtls_printf( "  This certificate has no flags\n" );
137     else
138     {
139         mbedtls_x509_crt_verify_info( buf, sizeof( buf ), "  ! ", *flags );
140         mbedtls_printf( "%s\n", buf );
141     }
142 
143     return( 0 );
144 }
145 
main(int argc,char * argv[])146 int main( int argc, char *argv[] )
147 {
148     int ret = 0;
149     mbedtls_net_context server_fd;
150     unsigned char buf[1024];
151     mbedtls_entropy_context entropy;
152     mbedtls_ctr_drbg_context ctr_drbg;
153     mbedtls_ssl_context ssl;
154     mbedtls_ssl_config conf;
155     mbedtls_x509_crt cacert;
156     mbedtls_x509_crl cacrl;
157     int i, j;
158     uint32_t flags;
159     int verify = 0;
160     char *p, *q;
161     const char *pers = "cert_app";
162 
163     /*
164      * Set to sane values
165      */
166     mbedtls_net_init( &server_fd );
167     mbedtls_ctr_drbg_init( &ctr_drbg );
168     mbedtls_ssl_init( &ssl );
169     mbedtls_ssl_config_init( &conf );
170     mbedtls_x509_crt_init( &cacert );
171 #if defined(MBEDTLS_X509_CRL_PARSE_C)
172     mbedtls_x509_crl_init( &cacrl );
173 #else
174     /* Zeroize structure as CRL parsing is not supported and we have to pass
175        it to the verify function */
176     memset( &cacrl, 0, sizeof(mbedtls_x509_crl) );
177 #endif
178 
179     if( argc == 0 )
180     {
181     usage:
182         mbedtls_printf( USAGE );
183         ret = 2;
184         goto exit;
185     }
186 
187     opt.mode                = DFL_MODE;
188     opt.filename            = DFL_FILENAME;
189     opt.ca_file             = DFL_CA_FILE;
190     opt.crl_file            = DFL_CRL_FILE;
191     opt.ca_path             = DFL_CA_PATH;
192     opt.server_name         = DFL_SERVER_NAME;
193     opt.server_port         = DFL_SERVER_PORT;
194     opt.debug_level         = DFL_DEBUG_LEVEL;
195     opt.permissive          = DFL_PERMISSIVE;
196 
197     for( i = 1; i < argc; i++ )
198     {
199         p = argv[i];
200         if( ( q = strchr( p, '=' ) ) == NULL )
201             goto usage;
202         *q++ = '\0';
203 
204         for( j = 0; p + j < q; j++ )
205         {
206             if( argv[i][j] >= 'A' && argv[i][j] <= 'Z' )
207                 argv[i][j] |= 0x20;
208         }
209 
210         if( strcmp( p, "mode" ) == 0 )
211         {
212             if( strcmp( q, "file" ) == 0 )
213                 opt.mode = MODE_FILE;
214             else if( strcmp( q, "ssl" ) == 0 )
215                 opt.mode = MODE_SSL;
216             else
217                 goto usage;
218         }
219         else if( strcmp( p, "filename" ) == 0 )
220             opt.filename = q;
221         else if( strcmp( p, "ca_file" ) == 0 )
222             opt.ca_file = q;
223         else if( strcmp( p, "crl_file" ) == 0 )
224             opt.crl_file = q;
225         else if( strcmp( p, "ca_path" ) == 0 )
226             opt.ca_path = q;
227         else if( strcmp( p, "server_name" ) == 0 )
228             opt.server_name = q;
229         else if( strcmp( p, "server_port" ) == 0 )
230             opt.server_port = q;
231         else if( strcmp( p, "debug_level" ) == 0 )
232         {
233             opt.debug_level = atoi( q );
234             if( opt.debug_level < 0 || opt.debug_level > 65535 )
235                 goto usage;
236         }
237         else if( strcmp( p, "permissive" ) == 0 )
238         {
239             opt.permissive = atoi( q );
240             if( opt.permissive < 0 || opt.permissive > 1 )
241                 goto usage;
242         }
243         else
244             goto usage;
245     }
246 
247     /*
248      * 1.1. Load the trusted CA
249      */
250     mbedtls_printf( "  . Loading the CA root certificate ..." );
251     fflush( stdout );
252 
253     if( strlen( opt.ca_path ) )
254     {
255         ret = mbedtls_x509_crt_parse_path( &cacert, opt.ca_path );
256         verify = 1;
257     }
258     else if( strlen( opt.ca_file ) )
259     {
260         ret = mbedtls_x509_crt_parse_file( &cacert, opt.ca_file );
261         verify = 1;
262     }
263 
264     if( ret < 0 )
265     {
266         mbedtls_printf( " failed\n  !  mbedtls_x509_crt_parse returned -0x%x\n\n", -ret );
267         goto exit;
268     }
269 
270     mbedtls_printf( " ok (%d skipped)\n", ret );
271 
272 #if defined(MBEDTLS_X509_CRL_PARSE_C)
273     if( strlen( opt.crl_file ) )
274     {
275         if( ( ret = mbedtls_x509_crl_parse_file( &cacrl, opt.crl_file ) ) != 0 )
276         {
277             mbedtls_printf( " failed\n  !  mbedtls_x509_crl_parse returned -0x%x\n\n", -ret );
278             goto exit;
279         }
280 
281         verify = 1;
282     }
283 #endif
284 
285     if( opt.mode == MODE_FILE )
286     {
287         mbedtls_x509_crt crt;
288         mbedtls_x509_crt *cur = &crt;
289         mbedtls_x509_crt_init( &crt );
290 
291         /*
292          * 1.1. Load the certificate(s)
293          */
294         mbedtls_printf( "\n  . Loading the certificate(s) ..." );
295         fflush( stdout );
296 
297         ret = mbedtls_x509_crt_parse_file( &crt, opt.filename );
298 
299         if( ret < 0 )
300         {
301             mbedtls_printf( " failed\n  !  mbedtls_x509_crt_parse_file returned %d\n\n", ret );
302             mbedtls_x509_crt_free( &crt );
303             goto exit;
304         }
305 
306         if( opt.permissive == 0 && ret > 0 )
307         {
308             mbedtls_printf( " failed\n  !  mbedtls_x509_crt_parse failed to parse %d certificates\n\n", ret );
309             mbedtls_x509_crt_free( &crt );
310             goto exit;
311         }
312 
313         mbedtls_printf( " ok\n" );
314 
315         /*
316          * 1.2 Print the certificate(s)
317          */
318         while( cur != NULL )
319         {
320             mbedtls_printf( "  . Peer certificate information    ...\n" );
321             ret = mbedtls_x509_crt_info( (char *) buf, sizeof( buf ) - 1, "      ",
322                                  cur );
323             if( ret == -1 )
324             {
325                 mbedtls_printf( " failed\n  !  mbedtls_x509_crt_info returned %d\n\n", ret );
326                 mbedtls_x509_crt_free( &crt );
327                 goto exit;
328             }
329 
330             mbedtls_printf( "%s\n", buf );
331 
332             cur = cur->next;
333         }
334 
335         ret = 0;
336 
337         /*
338          * 1.3 Verify the certificate
339          */
340         if( verify )
341         {
342             mbedtls_printf( "  . Verifying X.509 certificate..." );
343 
344             if( ( ret = mbedtls_x509_crt_verify( &crt, &cacert, &cacrl, NULL, &flags,
345                                          my_verify, NULL ) ) != 0 )
346             {
347                 char vrfy_buf[512];
348 
349                 mbedtls_printf( " failed\n" );
350 
351                 mbedtls_x509_crt_verify_info( vrfy_buf, sizeof( vrfy_buf ), "  ! ", flags );
352 
353                 mbedtls_printf( "%s\n", vrfy_buf );
354             }
355             else
356                 mbedtls_printf( " ok\n" );
357         }
358 
359         mbedtls_x509_crt_free( &crt );
360     }
361     else if( opt.mode == MODE_SSL )
362     {
363         /*
364          * 1. Initialize the RNG and the session data
365          */
366         mbedtls_printf( "\n  . Seeding the random number generator..." );
367         fflush( stdout );
368 
369         mbedtls_entropy_init( &entropy );
370         if( ( ret = mbedtls_ctr_drbg_seed( &ctr_drbg, mbedtls_entropy_func, &entropy,
371                                    (const unsigned char *) pers,
372                                    strlen( pers ) ) ) != 0 )
373         {
374             mbedtls_printf( " failed\n  ! mbedtls_ctr_drbg_seed returned %d\n", ret );
375             goto ssl_exit;
376         }
377 
378         mbedtls_printf( " ok\n" );
379 
380 #if defined(MBEDTLS_DEBUG_C)
381         mbedtls_debug_set_threshold( opt.debug_level );
382 #endif
383 
384         /*
385          * 2. Start the connection
386          */
387         mbedtls_printf( "  . SSL connection to tcp/%s/%s...", opt.server_name,
388                                                               opt.server_port );
389         fflush( stdout );
390 
391         if( ( ret = mbedtls_net_connect( &server_fd, opt.server_name,
392                                  opt.server_port, MBEDTLS_NET_PROTO_TCP ) ) != 0 )
393         {
394             mbedtls_printf( " failed\n  ! mbedtls_net_connect returned %d\n\n", ret );
395             goto ssl_exit;
396         }
397 
398         /*
399          * 3. Setup stuff
400          */
401         if( ( ret = mbedtls_ssl_config_defaults( &conf,
402                         MBEDTLS_SSL_IS_CLIENT,
403                         MBEDTLS_SSL_TRANSPORT_STREAM,
404                         MBEDTLS_SSL_PRESET_DEFAULT ) ) != 0 )
405         {
406             mbedtls_printf( " failed\n  ! mbedtls_ssl_config_defaults returned %d\n\n", ret );
407             goto exit;
408         }
409 
410         if( verify )
411         {
412             mbedtls_ssl_conf_authmode( &conf, MBEDTLS_SSL_VERIFY_REQUIRED );
413             mbedtls_ssl_conf_ca_chain( &conf, &cacert, NULL );
414             mbedtls_ssl_conf_verify( &conf, my_verify, NULL );
415         }
416         else
417             mbedtls_ssl_conf_authmode( &conf, MBEDTLS_SSL_VERIFY_NONE );
418 
419         mbedtls_ssl_conf_rng( &conf, mbedtls_ctr_drbg_random, &ctr_drbg );
420         mbedtls_ssl_conf_dbg( &conf, my_debug, stdout );
421 
422         if( ( ret = mbedtls_ssl_setup( &ssl, &conf ) ) != 0 )
423         {
424             mbedtls_printf( " failed\n  ! mbedtls_ssl_setup returned %d\n\n", ret );
425             goto ssl_exit;
426         }
427 
428         if( ( ret = mbedtls_ssl_set_hostname( &ssl, opt.server_name ) ) != 0 )
429         {
430             mbedtls_printf( " failed\n  ! mbedtls_ssl_set_hostname returned %d\n\n", ret );
431             goto ssl_exit;
432         }
433 
434         mbedtls_ssl_set_bio( &ssl, &server_fd, mbedtls_net_send, mbedtls_net_recv, NULL );
435 
436         /*
437          * 4. Handshake
438          */
439         while( ( ret = mbedtls_ssl_handshake( &ssl ) ) != 0 )
440         {
441             if( ret != MBEDTLS_ERR_SSL_WANT_READ && ret != MBEDTLS_ERR_SSL_WANT_WRITE )
442             {
443                 mbedtls_printf( " failed\n  ! mbedtls_ssl_handshake returned %d\n\n", ret );
444                 goto ssl_exit;
445             }
446         }
447 
448         mbedtls_printf( " ok\n" );
449 
450         /*
451          * 5. Print the certificate
452          */
453         mbedtls_printf( "  . Peer certificate information    ...\n" );
454         ret = mbedtls_x509_crt_info( (char *) buf, sizeof( buf ) - 1, "      ",
455                              ssl.session->peer_cert );
456         if( ret == -1 )
457         {
458             mbedtls_printf( " failed\n  !  mbedtls_x509_crt_info returned %d\n\n", ret );
459             goto ssl_exit;
460         }
461 
462         mbedtls_printf( "%s\n", buf );
463 
464         mbedtls_ssl_close_notify( &ssl );
465 
466 ssl_exit:
467         mbedtls_ssl_free( &ssl );
468         mbedtls_ssl_config_free( &conf );
469     }
470     else
471         goto usage;
472 
473 exit:
474 
475     mbedtls_net_free( &server_fd );
476     mbedtls_x509_crt_free( &cacert );
477 #if defined(MBEDTLS_X509_CRL_PARSE_C)
478     mbedtls_x509_crl_free( &cacrl );
479 #endif
480     mbedtls_ctr_drbg_free( &ctr_drbg );
481     mbedtls_entropy_free( &entropy );
482 
483 #if defined(_WIN32)
484     mbedtls_printf( "  + Press Enter to exit this program.\n" );
485     fflush( stdout ); getchar();
486 #endif
487 
488     if( ret < 0 )
489         ret = 1;
490 
491     return( ret );
492 }
493 #endif /* MBEDTLS_BIGNUM_C && MBEDTLS_ENTROPY_C && MBEDTLS_SSL_TLS_C &&
494           MBEDTLS_SSL_CLI_C && MBEDTLS_NET_C && MBEDTLS_RSA_C &&
495           MBEDTLS_X509_CRT_PARSE_C && MBEDTLS_FS_IO && MBEDTLS_CTR_DRBG_C */
496