mirror of
https://github.com/yuzu-mirror/mbedtls.git
synced 2026-04-04 22:19:05 +00:00
Switch to the new code style
Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
parent
fd13a0f851
commit
449bd8303e
442 changed files with 86735 additions and 89438 deletions
|
|
@ -29,43 +29,47 @@
|
|||
#endif
|
||||
|
||||
#if !defined(MBEDTLS_MD_C) || !defined(MBEDTLS_FS_IO)
|
||||
int main( void )
|
||||
int main(void)
|
||||
{
|
||||
mbedtls_printf("MBEDTLS_MD_C and/or MBEDTLS_FS_IO not defined.\n");
|
||||
mbedtls_exit( 0 );
|
||||
mbedtls_exit(0);
|
||||
}
|
||||
#else
|
||||
|
||||
|
||||
static int generic_wrapper( const mbedtls_md_info_t *md_info, char *filename, unsigned char *sum )
|
||||
static int generic_wrapper(const mbedtls_md_info_t *md_info, char *filename, unsigned char *sum)
|
||||
{
|
||||
int ret = mbedtls_md_file( md_info, filename, sum );
|
||||
int ret = mbedtls_md_file(md_info, filename, sum);
|
||||
|
||||
if( ret == 1 )
|
||||
mbedtls_fprintf( stderr, "failed to open: %s\n", filename );
|
||||
if (ret == 1) {
|
||||
mbedtls_fprintf(stderr, "failed to open: %s\n", filename);
|
||||
}
|
||||
|
||||
if( ret == 2 )
|
||||
mbedtls_fprintf( stderr, "failed to read: %s\n", filename );
|
||||
if (ret == 2) {
|
||||
mbedtls_fprintf(stderr, "failed to read: %s\n", filename);
|
||||
}
|
||||
|
||||
return( ret );
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int generic_print( const mbedtls_md_info_t *md_info, char *filename )
|
||||
static int generic_print(const mbedtls_md_info_t *md_info, char *filename)
|
||||
{
|
||||
int i;
|
||||
unsigned char sum[MBEDTLS_MD_MAX_SIZE];
|
||||
|
||||
if( generic_wrapper( md_info, filename, sum ) != 0 )
|
||||
return( 1 );
|
||||
if (generic_wrapper(md_info, filename, sum) != 0) {
|
||||
return 1;
|
||||
}
|
||||
|
||||
for( i = 0; i < mbedtls_md_get_size( md_info ); i++ )
|
||||
mbedtls_printf( "%02x", sum[i] );
|
||||
for (i = 0; i < mbedtls_md_get_size(md_info); i++) {
|
||||
mbedtls_printf("%02x", sum[i]);
|
||||
}
|
||||
|
||||
mbedtls_printf( " %s\n", filename );
|
||||
return( 0 );
|
||||
mbedtls_printf(" %s\n", filename);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int generic_check( const mbedtls_md_info_t *md_info, char *filename )
|
||||
static int generic_check(const mbedtls_md_info_t *md_info, char *filename)
|
||||
{
|
||||
int i;
|
||||
size_t n;
|
||||
|
|
@ -81,141 +85,137 @@ static int generic_check( const mbedtls_md_info_t *md_info, char *filename )
|
|||
char buf[MBEDTLS_MD_MAX_SIZE * 2 + 1];
|
||||
#endif
|
||||
|
||||
if( ( f = fopen( filename, "rb" ) ) == NULL )
|
||||
{
|
||||
mbedtls_printf( "failed to open: %s\n", filename );
|
||||
return( 1 );
|
||||
if ((f = fopen(filename, "rb")) == NULL) {
|
||||
mbedtls_printf("failed to open: %s\n", filename);
|
||||
return 1;
|
||||
}
|
||||
|
||||
nb_err1 = nb_err2 = 0;
|
||||
nb_tot1 = nb_tot2 = 0;
|
||||
|
||||
memset( line, 0, sizeof( line ) );
|
||||
memset(line, 0, sizeof(line));
|
||||
|
||||
n = sizeof( line );
|
||||
n = sizeof(line);
|
||||
|
||||
while( fgets( line, (int) n - 1, f ) != NULL )
|
||||
{
|
||||
n = strlen( line );
|
||||
while (fgets(line, (int) n - 1, f) != NULL) {
|
||||
n = strlen(line);
|
||||
|
||||
if( n < (size_t) 2 * mbedtls_md_get_size( md_info ) + 4 )
|
||||
{
|
||||
mbedtls_printf("No '%s' hash found on line.\n", mbedtls_md_get_name( md_info ));
|
||||
if (n < (size_t) 2 * mbedtls_md_get_size(md_info) + 4) {
|
||||
mbedtls_printf("No '%s' hash found on line.\n", mbedtls_md_get_name(md_info));
|
||||
continue;
|
||||
}
|
||||
|
||||
if( line[2 * mbedtls_md_get_size( md_info )] != ' ' || line[2 * mbedtls_md_get_size( md_info ) + 1] != ' ' )
|
||||
{
|
||||
mbedtls_printf("No '%s' hash found on line.\n", mbedtls_md_get_name( md_info ));
|
||||
if (line[2 * mbedtls_md_get_size(md_info)] != ' ' ||
|
||||
line[2 * mbedtls_md_get_size(md_info) + 1] != ' ') {
|
||||
mbedtls_printf("No '%s' hash found on line.\n", mbedtls_md_get_name(md_info));
|
||||
continue;
|
||||
}
|
||||
|
||||
if( line[n - 1] == '\n' ) { n--; line[n] = '\0'; }
|
||||
if( line[n - 1] == '\r' ) { n--; line[n] = '\0'; }
|
||||
if (line[n - 1] == '\n') {
|
||||
n--; line[n] = '\0';
|
||||
}
|
||||
if (line[n - 1] == '\r') {
|
||||
n--; line[n] = '\0';
|
||||
}
|
||||
|
||||
nb_tot1++;
|
||||
|
||||
if( generic_wrapper( md_info, line + 2 + 2 * mbedtls_md_get_size( md_info ), sum ) != 0 )
|
||||
{
|
||||
if (generic_wrapper(md_info, line + 2 + 2 * mbedtls_md_get_size(md_info), sum) != 0) {
|
||||
nb_err1++;
|
||||
continue;
|
||||
}
|
||||
|
||||
nb_tot2++;
|
||||
|
||||
for( i = 0; i < mbedtls_md_get_size( md_info ); i++ )
|
||||
sprintf( buf + i * 2, "%02x", sum[i] );
|
||||
for (i = 0; i < mbedtls_md_get_size(md_info); i++) {
|
||||
sprintf(buf + i * 2, "%02x", sum[i]);
|
||||
}
|
||||
|
||||
/* Use constant-time buffer comparison */
|
||||
diff = 0;
|
||||
for( i = 0; i < 2 * mbedtls_md_get_size( md_info ); i++ )
|
||||
for (i = 0; i < 2 * mbedtls_md_get_size(md_info); i++) {
|
||||
diff |= line[i] ^ buf[i];
|
||||
|
||||
if( diff != 0 )
|
||||
{
|
||||
nb_err2++;
|
||||
mbedtls_fprintf( stderr, "wrong checksum: %s\n", line + 66 );
|
||||
}
|
||||
|
||||
n = sizeof( line );
|
||||
if (diff != 0) {
|
||||
nb_err2++;
|
||||
mbedtls_fprintf(stderr, "wrong checksum: %s\n", line + 66);
|
||||
}
|
||||
|
||||
n = sizeof(line);
|
||||
}
|
||||
|
||||
if( nb_err1 != 0 )
|
||||
{
|
||||
mbedtls_printf( "WARNING: %d (out of %d) input files could "
|
||||
"not be read\n", nb_err1, nb_tot1 );
|
||||
if (nb_err1 != 0) {
|
||||
mbedtls_printf("WARNING: %d (out of %d) input files could "
|
||||
"not be read\n", nb_err1, nb_tot1);
|
||||
}
|
||||
|
||||
if( nb_err2 != 0 )
|
||||
{
|
||||
mbedtls_printf( "WARNING: %d (out of %d) computed checksums did "
|
||||
"not match\n", nb_err2, nb_tot2 );
|
||||
if (nb_err2 != 0) {
|
||||
mbedtls_printf("WARNING: %d (out of %d) computed checksums did "
|
||||
"not match\n", nb_err2, nb_tot2);
|
||||
}
|
||||
|
||||
fclose( f );
|
||||
fclose(f);
|
||||
|
||||
return( nb_err1 != 0 || nb_err2 != 0 );
|
||||
return nb_err1 != 0 || nb_err2 != 0;
|
||||
}
|
||||
|
||||
int main( int argc, char *argv[] )
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int ret = 1, i;
|
||||
int exit_code = MBEDTLS_EXIT_FAILURE;
|
||||
const mbedtls_md_info_t *md_info;
|
||||
mbedtls_md_context_t md_ctx;
|
||||
|
||||
mbedtls_md_init( &md_ctx );
|
||||
mbedtls_md_init(&md_ctx);
|
||||
|
||||
if( argc == 1 )
|
||||
{
|
||||
if (argc == 1) {
|
||||
const int *list;
|
||||
|
||||
mbedtls_printf( "print mode: generic_sum <mbedtls_md> <file> <file> ...\n" );
|
||||
mbedtls_printf( "check mode: generic_sum <mbedtls_md> -c <checksum file>\n" );
|
||||
mbedtls_printf("print mode: generic_sum <mbedtls_md> <file> <file> ...\n");
|
||||
mbedtls_printf("check mode: generic_sum <mbedtls_md> -c <checksum file>\n");
|
||||
|
||||
mbedtls_printf( "\nAvailable message digests:\n" );
|
||||
mbedtls_printf("\nAvailable message digests:\n");
|
||||
list = mbedtls_md_list();
|
||||
while( *list )
|
||||
{
|
||||
md_info = mbedtls_md_info_from_type( *list );
|
||||
mbedtls_printf( " %s\n", mbedtls_md_get_name( md_info ) );
|
||||
while (*list) {
|
||||
md_info = mbedtls_md_info_from_type(*list);
|
||||
mbedtls_printf(" %s\n", mbedtls_md_get_name(md_info));
|
||||
list++;
|
||||
}
|
||||
|
||||
mbedtls_exit( exit_code );
|
||||
mbedtls_exit(exit_code);
|
||||
}
|
||||
|
||||
/*
|
||||
* Read the MD from the command line
|
||||
*/
|
||||
md_info = mbedtls_md_info_from_string( argv[1] );
|
||||
if( md_info == NULL )
|
||||
{
|
||||
mbedtls_fprintf( stderr, "Message Digest '%s' not found\n", argv[1] );
|
||||
mbedtls_exit( exit_code );
|
||||
md_info = mbedtls_md_info_from_string(argv[1]);
|
||||
if (md_info == NULL) {
|
||||
mbedtls_fprintf(stderr, "Message Digest '%s' not found\n", argv[1]);
|
||||
mbedtls_exit(exit_code);
|
||||
}
|
||||
if( mbedtls_md_setup( &md_ctx, md_info, 0 ) )
|
||||
{
|
||||
mbedtls_fprintf( stderr, "Failed to initialize context.\n" );
|
||||
mbedtls_exit( exit_code );
|
||||
if (mbedtls_md_setup(&md_ctx, md_info, 0)) {
|
||||
mbedtls_fprintf(stderr, "Failed to initialize context.\n");
|
||||
mbedtls_exit(exit_code);
|
||||
}
|
||||
|
||||
ret = 0;
|
||||
if( argc == 4 && strcmp( "-c", argv[2] ) == 0 )
|
||||
{
|
||||
ret |= generic_check( md_info, argv[3] );
|
||||
if (argc == 4 && strcmp("-c", argv[2]) == 0) {
|
||||
ret |= generic_check(md_info, argv[3]);
|
||||
goto exit;
|
||||
}
|
||||
|
||||
for( i = 2; i < argc; i++ )
|
||||
ret |= generic_print( md_info, argv[i] );
|
||||
for (i = 2; i < argc; i++) {
|
||||
ret |= generic_print(md_info, argv[i]);
|
||||
}
|
||||
|
||||
if ( ret == 0 )
|
||||
if (ret == 0) {
|
||||
exit_code = MBEDTLS_EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
exit:
|
||||
mbedtls_md_free( &md_ctx );
|
||||
mbedtls_md_free(&md_ctx);
|
||||
|
||||
mbedtls_exit( exit_code );
|
||||
mbedtls_exit(exit_code);
|
||||
}
|
||||
#endif /* MBEDTLS_MD_C && MBEDTLS_FS_IO */
|
||||
|
|
|
|||
|
|
@ -26,30 +26,32 @@
|
|||
#endif
|
||||
|
||||
#if !defined(MBEDTLS_MD5_C)
|
||||
int main( void )
|
||||
int main(void)
|
||||
{
|
||||
mbedtls_printf("MBEDTLS_MD5_C not defined.\n");
|
||||
mbedtls_exit( 0 );
|
||||
mbedtls_exit(0);
|
||||
}
|
||||
#else
|
||||
|
||||
|
||||
int main( void )
|
||||
int main(void)
|
||||
{
|
||||
int i, ret;
|
||||
unsigned char digest[16];
|
||||
char str[] = "Hello, world!";
|
||||
|
||||
mbedtls_printf( "\n MD5('%s') = ", str );
|
||||
mbedtls_printf("\n MD5('%s') = ", str);
|
||||
|
||||
if( ( ret = mbedtls_md5( (unsigned char *) str, 13, digest ) ) != 0 )
|
||||
mbedtls_exit( MBEDTLS_EXIT_FAILURE );
|
||||
if ((ret = mbedtls_md5((unsigned char *) str, 13, digest)) != 0) {
|
||||
mbedtls_exit(MBEDTLS_EXIT_FAILURE);
|
||||
}
|
||||
|
||||
for( i = 0; i < 16; i++ )
|
||||
mbedtls_printf( "%02x", digest[i] );
|
||||
for (i = 0; i < 16; i++) {
|
||||
mbedtls_printf("%02x", digest[i]);
|
||||
}
|
||||
|
||||
mbedtls_printf( "\n\n" );
|
||||
mbedtls_printf("\n\n");
|
||||
|
||||
mbedtls_exit( MBEDTLS_EXIT_SUCCESS );
|
||||
mbedtls_exit(MBEDTLS_EXIT_SUCCESS);
|
||||
}
|
||||
#endif /* MBEDTLS_MD5_C */
|
||||
|
|
|
|||
|
|
@ -49,10 +49,10 @@
|
|||
|
||||
/* If the build options we need are not enabled, compile a placeholder. */
|
||||
#if !defined(MBEDTLS_MD_C)
|
||||
int main( void )
|
||||
int main(void)
|
||||
{
|
||||
printf( "MBEDTLS_MD_C not defined\r\n" );
|
||||
return( 0 );
|
||||
printf("MBEDTLS_MD_C not defined\r\n");
|
||||
return 0;
|
||||
}
|
||||
#else
|
||||
|
||||
|
|
@ -69,30 +69,31 @@ const unsigned char msg2_part2[] = { 0x06, 0x06 };
|
|||
const unsigned char key_bytes[32] = { 0 };
|
||||
|
||||
/* Print the contents of a buffer in hex */
|
||||
void print_buf( const char *title, unsigned char *buf, size_t len )
|
||||
void print_buf(const char *title, unsigned char *buf, size_t len)
|
||||
{
|
||||
printf( "%s:", title );
|
||||
for( size_t i = 0; i < len; i++ )
|
||||
printf( " %02x", buf[i] );
|
||||
printf( "\n" );
|
||||
printf("%s:", title);
|
||||
for (size_t i = 0; i < len; i++) {
|
||||
printf(" %02x", buf[i]);
|
||||
}
|
||||
printf("\n");
|
||||
}
|
||||
|
||||
/* Run an Mbed TLS function and bail out if it fails.
|
||||
* A string description of the error code can be recovered with:
|
||||
* programs/util/strerror <value> */
|
||||
#define CHK( expr ) \
|
||||
#define CHK(expr) \
|
||||
do \
|
||||
{ \
|
||||
ret = ( expr ); \
|
||||
if( ret != 0 ) \
|
||||
ret = (expr); \
|
||||
if (ret != 0) \
|
||||
{ \
|
||||
printf( "Error %d at line %d: %s\n", \
|
||||
ret, \
|
||||
__LINE__, \
|
||||
#expr ); \
|
||||
printf("Error %d at line %d: %s\n", \
|
||||
ret, \
|
||||
__LINE__, \
|
||||
#expr); \
|
||||
goto exit; \
|
||||
} \
|
||||
} while( 0 )
|
||||
} while (0)
|
||||
|
||||
/*
|
||||
* This function demonstrates computation of the HMAC of two messages using
|
||||
|
|
@ -106,42 +107,42 @@ int hmac_demo(void)
|
|||
|
||||
mbedtls_md_context_t ctx;
|
||||
|
||||
mbedtls_md_init( &ctx );
|
||||
mbedtls_md_init(&ctx);
|
||||
|
||||
/* prepare context and load key */
|
||||
// the last argument to setup is 1 to enable HMAC (not just hashing)
|
||||
const mbedtls_md_info_t *info = mbedtls_md_info_from_type( alg );
|
||||
CHK( mbedtls_md_setup( &ctx, info, 1 ) );
|
||||
CHK( mbedtls_md_hmac_starts( &ctx, key_bytes, sizeof( key_bytes ) ) );
|
||||
const mbedtls_md_info_t *info = mbedtls_md_info_from_type(alg);
|
||||
CHK(mbedtls_md_setup(&ctx, info, 1));
|
||||
CHK(mbedtls_md_hmac_starts(&ctx, key_bytes, sizeof(key_bytes)));
|
||||
|
||||
/* compute HMAC(key, msg1_part1 | msg1_part2) */
|
||||
CHK( mbedtls_md_hmac_update( &ctx, msg1_part1, sizeof( msg1_part1 ) ) );
|
||||
CHK( mbedtls_md_hmac_update( &ctx, msg1_part2, sizeof( msg1_part2 ) ) );
|
||||
CHK( mbedtls_md_hmac_finish( &ctx, out ) );
|
||||
print_buf( "msg1", out, mbedtls_md_get_size( info ) );
|
||||
CHK(mbedtls_md_hmac_update(&ctx, msg1_part1, sizeof(msg1_part1)));
|
||||
CHK(mbedtls_md_hmac_update(&ctx, msg1_part2, sizeof(msg1_part2)));
|
||||
CHK(mbedtls_md_hmac_finish(&ctx, out));
|
||||
print_buf("msg1", out, mbedtls_md_get_size(info));
|
||||
|
||||
/* compute HMAC(key, msg2_part1 | msg2_part2) */
|
||||
CHK( mbedtls_md_hmac_reset( &ctx ) ); // prepare for new operation
|
||||
CHK( mbedtls_md_hmac_update( &ctx, msg2_part1, sizeof( msg2_part1 ) ) );
|
||||
CHK( mbedtls_md_hmac_update( &ctx, msg2_part2, sizeof( msg2_part2 ) ) );
|
||||
CHK( mbedtls_md_hmac_finish( &ctx, out ) );
|
||||
print_buf( "msg2", out, mbedtls_md_get_size( info ) );
|
||||
CHK(mbedtls_md_hmac_reset(&ctx)); // prepare for new operation
|
||||
CHK(mbedtls_md_hmac_update(&ctx, msg2_part1, sizeof(msg2_part1)));
|
||||
CHK(mbedtls_md_hmac_update(&ctx, msg2_part2, sizeof(msg2_part2)));
|
||||
CHK(mbedtls_md_hmac_finish(&ctx, out));
|
||||
print_buf("msg2", out, mbedtls_md_get_size(info));
|
||||
|
||||
exit:
|
||||
mbedtls_md_free( &ctx );
|
||||
mbedtls_platform_zeroize( out, sizeof( out ) );
|
||||
mbedtls_md_free(&ctx);
|
||||
mbedtls_platform_zeroize(out, sizeof(out));
|
||||
|
||||
return( ret );
|
||||
return ret;
|
||||
}
|
||||
|
||||
int main(void)
|
||||
{
|
||||
int ret;
|
||||
|
||||
CHK( hmac_demo() );
|
||||
CHK(hmac_demo());
|
||||
|
||||
exit:
|
||||
return( ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE );
|
||||
return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
|
||||
}
|
||||
|
||||
#endif
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue