1 /*
2  * Zeroize application for debugger-driven testing
3  *
4  * This is a simple test application used for debugger-driven testing to check
5  * whether calls to mbedtls_platform_zeroize() are being eliminated by compiler
6  * optimizations. This application is used by the GDB script at
7  * tests/scripts/test_zeroize.gdb: the script sets a breakpoint at the last
8  * return statement in the main() function of this program. The debugger
9  * facilities are then used to manually inspect the memory and verify that the
10  * call to mbedtls_platform_zeroize() was not eliminated.
11  *
12  *  Copyright The Mbed TLS Contributors
13  *  SPDX-License-Identifier: Apache-2.0
14  *
15  *  Licensed under the Apache License, Version 2.0 (the "License"); you may
16  *  not use this file except in compliance with the License.
17  *  You may obtain a copy of the License at
18  *
19  *  http://www.apache.org/licenses/LICENSE-2.0
20  *
21  *  Unless required by applicable law or agreed to in writing, software
22  *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
23  *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
24  *  See the License for the specific language governing permissions and
25  *  limitations under the License.
26  */
27 
28 #include "mbedtls/build_info.h"
29 
30 #include <stdio.h>
31 
32 #if defined(MBEDTLS_PLATFORM_C)
33 #include "mbedtls/platform.h"
34 #else
35 #include <stdlib.h>
36 #define mbedtls_printf     printf
37 #define mbedtls_exit       exit
38 #define MBEDTLS_EXIT_SUCCESS EXIT_SUCCESS
39 #define MBEDTLS_EXIT_FAILURE EXIT_FAILURE
40 #endif
41 
42 #include "mbedtls/platform_util.h"
43 
44 #define BUFFER_LEN 1024
45 
usage(void)46 void usage( void )
47 {
48     mbedtls_printf( "Zeroize is a simple program to assist with testing\n" );
49     mbedtls_printf( "the mbedtls_platform_zeroize() function by using the\n" );
50     mbedtls_printf( "debugger. This program takes a file as input and\n" );
51     mbedtls_printf( "prints the first %d characters. Usage:\n\n", BUFFER_LEN );
52     mbedtls_printf( "       zeroize <FILE>\n" );
53 }
54 
main(int argc,char ** argv)55 int main( int argc, char** argv )
56 {
57     int exit_code = MBEDTLS_EXIT_FAILURE;
58     FILE *fp;
59     char buf[BUFFER_LEN];
60     char *p = buf;
61     char *end = p + BUFFER_LEN;
62     int c;
63 
64     if( argc != 2 )
65     {
66         mbedtls_printf( "This program takes exactly 1 agument\n" );
67         usage();
68         mbedtls_exit( exit_code );
69     }
70 
71     fp = fopen( argv[1], "r" );
72     if( fp == NULL )
73     {
74         mbedtls_printf( "Could not open file '%s'\n", argv[1] );
75         mbedtls_exit( exit_code );
76     }
77 
78     while( ( c = fgetc( fp ) ) != EOF && p < end - 1 )
79         *p++ = (char)c;
80     *p = '\0';
81 
82     if( p - buf != 0 )
83     {
84         mbedtls_printf( "%s\n", buf );
85         exit_code = MBEDTLS_EXIT_SUCCESS;
86     }
87     else
88         mbedtls_printf( "The file is empty!\n" );
89 
90     fclose( fp );
91     mbedtls_platform_zeroize( buf, sizeof( buf ) );
92 
93     mbedtls_exit( exit_code ); // GDB_BREAK_HERE -- don't remove this comment!
94 }
95