From fb6c7e2688bf67ca87ca267bccfc8e7447189ad0 Mon Sep 17 00:00:00 2001 From: Paul Bakker Date: Fri, 21 Jan 2011 10:21:11 +0000 Subject: [PATCH] - Added generic_sum example application to show use of generic message digest layer. --- programs/Makefile | 6 +- programs/hash/CMakeLists.txt | 5 +- programs/hash/generic_sum.c | 206 +++++++++++++++++++++++++++++++++++ 3 files changed, 215 insertions(+), 2 deletions(-) create mode 100644 programs/hash/generic_sum.c diff --git a/programs/Makefile b/programs/Makefile index 7380ad2ee..37c867a51 100644 --- a/programs/Makefile +++ b/programs/Makefile @@ -8,7 +8,7 @@ OFLAGS = -O LDFLAGS = -L../library -lpolarssl APPS = aes/aescrypt2 aes/crypt_and_hash \ - hash/hello \ + hash/hello hash/generic_sum \ hash/md5sum hash/sha1sum \ hash/sha2sum pkey/dh_client \ pkey/dh_genprime pkey/dh_server \ @@ -35,6 +35,10 @@ hash/hello: hash/hello.c ../library/libpolarssl.a echo " CC hash/hello.c" $(CC) $(CFLAGS) $(OFLAGS) hash/hello.c $(LDFLAGS) -o $@ +hash/generic_sum: hash/generic_sum.c ../library/libpolarssl.a + echo " CC hash/generic_sum.c" + $(CC) $(CFLAGS) $(OFLAGS) hash/generic_sum.c $(LDFLAGS) -o $@ + hash/md5sum: hash/md5sum.c ../library/libpolarssl.a echo " CC hash/md5sum.c" $(CC) $(CFLAGS) $(OFLAGS) hash/md5sum.c $(LDFLAGS) -o $@ diff --git a/programs/hash/CMakeLists.txt b/programs/hash/CMakeLists.txt index 0341c8b3c..367dae15b 100644 --- a/programs/hash/CMakeLists.txt +++ b/programs/hash/CMakeLists.txt @@ -1,6 +1,9 @@ add_executable(hello hello.c) target_link_libraries(hello polarssl) +add_executable(generic_sum generic_sum.c) +target_link_libraries(generic_sum polarssl) + add_executable(md5sum md5sum.c) target_link_libraries(md5sum polarssl) @@ -10,6 +13,6 @@ target_link_libraries(sha1sum polarssl) add_executable(sha2sum sha2sum.c) target_link_libraries(sha2sum polarssl) -INSTALL(TARGETS hello md5sum sha1sum sha2sum +INSTALL(TARGETS hello md5sum sha1sum sha2sum generic_sum DESTINATION "bin" PERMISSIONS OWNER_READ OWNER_WRITE OWNER_EXECUTE GROUP_READ GROUP_EXECUTE WORLD_READ WORLD_EXECUTE) diff --git a/programs/hash/generic_sum.c b/programs/hash/generic_sum.c new file mode 100644 index 000000000..add0ecd6e --- /dev/null +++ b/programs/hash/generic_sum.c @@ -0,0 +1,206 @@ +/* + * generic message digest layer demonstration program + * + * Copyright (C) 2006-2010, Brainspark B.V. + * + * This file is part of PolarSSL (http://www.polarssl.org) + * Lead Maintainer: Paul Bakker + * + * All rights reserved. + * + * 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. + */ + +#ifndef _CRT_SECURE_NO_DEPRECATE +#define _CRT_SECURE_NO_DEPRECATE 1 +#endif + +#include +#include + +#include "polarssl/md.h" + +static int generic_wrapper( const md_info_t *md_info, char *filename, unsigned char *sum ) +{ + int ret = md_file( md_info, filename, sum ); + + if( ret == 1 ) + fprintf( stderr, "failed to open: %s\n", filename ); + + if( ret == 2 ) + fprintf( stderr, "failed to read: %s\n", filename ); + + return( ret ); +} + +static int generic_print( const md_info_t *md_info, char *filename ) +{ + int i; + unsigned char sum[POLARSSL_MD_MAX_SIZE]; + + if( generic_wrapper( md_info, filename, sum ) != 0 ) + return( 1 ); + + for( i = 0; i < md_info->size; i++ ) + printf( "%02x", sum[i] ); + + printf( " %s\n", filename ); + return( 0 ); +} + +static int generic_check( const md_info_t *md_info, char *filename ) +{ + int i; + size_t n; + FILE *f; + int nb_err1, nb_err2; + int nb_tot1, nb_tot2; + unsigned char sum[POLARSSL_MD_MAX_SIZE]; + char buf[POLARSSL_MD_MAX_SIZE * 2 + 1], line[1024]; + + if( ( f = fopen( filename, "rb" ) ) == NULL ) + { + printf( "failed to open: %s\n", filename ); + return( 1 ); + } + + nb_err1 = nb_err2 = 0; + nb_tot1 = nb_tot2 = 0; + + memset( line, 0, sizeof( line ) ); + + n = sizeof( line ); + + while( fgets( line, n - 1, f ) != NULL ) + { + n = strlen( line ); + + if( n < (size_t) 2 * md_info->size + 4 ) + { + printf("No '%s' hash found on line.\n", md_info->name); + continue; + } + + if( line[2 * md_info->size] != ' ' || line[2 * md_info->size + 1] != ' ' ) + { + printf("No '%s' hash found on line.\n", md_info->name); + continue; + } + + 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 * md_info->size, sum ) != 0 ) + { + nb_err1++; + continue; + } + + nb_tot2++; + + for( i = 0; i < md_info->size; i++ ) + sprintf( buf + i * 2, "%02x", sum[i] ); + + if( memcmp( line, buf, 2 * md_info->size ) != 0 ) + { + nb_err2++; + fprintf( stderr, "wrong checksum: %s\n", line + 66 ); + } + + n = sizeof( line ); + } + + if( nb_err1 != 0 ) + { + printf( "WARNING: %d (out of %d) input files could " + "not be read\n", nb_err1, nb_tot1 ); + } + + if( nb_err2 != 0 ) + { + printf( "WARNING: %d (out of %d) computed checksums did " + "not match\n", nb_err2, nb_tot2 ); + } + + return( nb_err1 != 0 || nb_err2 != 0 ); +} + +int main( int argc, char *argv[] ) +{ + int ret, i; + const md_info_t *md_info; + md_context_t md_ctx; + + memset( &md_ctx, 0, sizeof( md_context_t )); + + if( argc == 1 ) + { + const int *list; + + printf( "print mode: generic_sum ...\n" ); + printf( "check mode: generic_sum -c \n" ); + + printf( "\nAvailable message digests:\n" ); + list = md_list(); + while( *list ) + { + md_info = md_info_from_type( *list ); + printf( " %s\n", md_info->name ); + list++; + } + +#ifdef WIN32 + printf( "\n Press Enter to exit this program.\n" ); + fflush( stdout ); getchar(); +#endif + + return( 1 ); + } + + /* + * Read the MD from the command line + */ + md_info = md_info_from_string( argv[1] ); + if( md_info == NULL ) + { + fprintf( stderr, "Message Digest '%s' not found\n", argv[1] ); + return( 1 ); + } + if( md_init_ctx( &md_ctx, md_info) ) + { + fprintf( stderr, "Failed to initialize context.\n" ); + return( 1 ); + } + printf("%08x\n", (unsigned int) md_ctx.md_ctx ); + + ret = 0; + if( argc == 4 && strcmp( "-c", argv[2] ) == 0 ) + { + ret |= generic_check( md_info, argv[3] ); + goto exit; + } + + printf("%08x\n", (unsigned int) md_ctx.md_ctx ); + for( i = 2; i < argc; i++ ) + ret |= generic_print( md_info, argv[i] ); + +exit: + printf("%08x\n", (unsigned int) md_ctx.md_ctx ); + md_free_ctx( &md_ctx ); + + return( ret ); +}