programs/test/benchmark.c - security/mbedtls - Rivoreo Source Code Repositories

 /*
  *  Benchmark demonstration program
  *
  *  Copyright (C) 2006-2013, Brainspark B.V.
  *
  *  This file is part of PolarSSL (http://www.polarssl.org)
  *  Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
  *
  *  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 <string.h>
 #include <stdlib.h>
 #include <stdio.h>

 #include "polarssl/config.h"

 #include "polarssl/md4.h"
 #include "polarssl/md5.h"
 #include "polarssl/sha1.h"
 #include "polarssl/sha256.h"
 #include "polarssl/sha512.h"
 #include "polarssl/arc4.h"
 #include "polarssl/des.h"
 #include "polarssl/aes.h"
 #include "polarssl/blowfish.h"
 #include "polarssl/camellia.h"
 #include "polarssl/gcm.h"
 #include "polarssl/rsa.h"
 #include "polarssl/dhm.h"
 #include "polarssl/timing.h"
 #include "polarssl/havege.h"
 #include "polarssl/ctr_drbg.h"

 #define BUFSIZE         1024
 #define HEADER_FORMAT   "  %-15s :  "

 #if defined(POLARSSL_TIMING_C)
 static int myrand( void *rng_state, unsigned char *output, size_t len )
 {
     size_t use_len;
     int rnd;

     if( rng_state != NULL )
         rng_state  = NULL;

     while( len > 0 )
     {
         use_len = len;
         if( use_len > sizeof(int) )
             use_len = sizeof(int);

         rnd = rand();
         memcpy( output, &rnd, use_len );
         output += use_len;
         len -= use_len;
     }

     return( 0 );
 }

 unsigned char buf[BUFSIZE];
 #endif

 #if !defined(POLARSSL_TIMING_C)
 int main( int argc, char *argv[] )
 {
     ((void) argc);
     ((void) argv);

     printf("POLARSSL_TIMING_C not defined.\n");
     return( 0 );
 }
 #else
 int main( int argc, char *argv[] )
 {
     int ret, keysize;
     unsigned long i, j, tsc;
     unsigned char tmp[64];
 #if defined(POLARSSL_ARC4_C)
     arc4_context arc4;
 #endif
 #if defined(POLARSSL_DES_C)
     des3_context des3;
     des_context des;
 #endif
 #if defined(POLARSSL_AES_C)
     aes_context aes;
 #if defined(POLARSSL_GCM_C)
     gcm_context gcm;
 #endif
 #endif
 #if defined(POLARSSL_BLOWFISH_C)
     blowfish_context blowfish;
 #endif
 #if defined(POLARSSL_CAMELLIA_C)
     camellia_context camellia;
 #endif
 #if defined(POLARSSL_RSA_C) && defined(POLARSSL_BIGNUM_C) &&    \
     defined(POLARSSL_GENPRIME)
     rsa_context rsa;
 #endif
 #if defined(POLARSSL_HAVEGE_C)
     havege_state hs;
 #endif
 #if defined(POLARSSL_CTR_DRBG_C)
     ctr_drbg_context    ctr_drbg;
 #endif
 #if defined(POLARSSL_DHM_C) && defined(POLARSSL_BIGNUM_C)
     dhm_context dhm;
     size_t olen = BUFSIZE;
 #endif
     ((void) argc);
     ((void) argv);

     memset( buf, 0xAA, sizeof( buf ) );

     printf( "\n" );

 #if defined(POLARSSL_MD4_C)
     printf( HEADER_FORMAT, "MD4" );
     fflush( stdout );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         md4( buf, BUFSIZE, tmp );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         md4( buf, BUFSIZE, tmp );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_MD5_C)
     printf( HEADER_FORMAT, "MD5" );
     fflush( stdout );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         md5( buf, BUFSIZE, tmp );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         md5( buf, BUFSIZE, tmp );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_SHA1_C)
     printf( HEADER_FORMAT, "SHA-1" );
     fflush( stdout );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         sha1( buf, BUFSIZE, tmp );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         sha1( buf, BUFSIZE, tmp );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_SHA256_C)
     printf( HEADER_FORMAT, "SHA-256" );
     fflush( stdout );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         sha256( buf, BUFSIZE, tmp, 0 );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         sha256( buf, BUFSIZE, tmp, 0 );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_SHA512_C)
     printf( HEADER_FORMAT, "SHA-512" );
     fflush( stdout );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         sha512( buf, BUFSIZE, tmp, 0 );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         sha512( buf, BUFSIZE, tmp, 0 );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_ARC4_C)
     printf( HEADER_FORMAT, "ARC4" );
     fflush( stdout );

     arc4_setup( &arc4, tmp, 32 );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         arc4_crypt( &arc4, BUFSIZE, buf, buf );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         arc4_crypt( &arc4, BUFSIZE, buf, buf );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_DES_C)
     printf( HEADER_FORMAT, "3DES" );
     fflush( stdout );

     des3_set3key_enc( &des3, tmp );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         des3_crypt_cbc( &des3, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         des3_crypt_cbc( &des3, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );

     printf( HEADER_FORMAT, "DES" );
     fflush( stdout );

     des_setkey_enc( &des, tmp );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         des_crypt_cbc( &des, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

     tsc = hardclock();
     for( j = 0; j < 1024; j++ )
         des_crypt_cbc( &des, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_AES_C)
     for( keysize = 128; keysize <= 256; keysize += 64 )
     {
         printf( "  AES-CBC-%d     :  ", keysize );
         fflush( stdout );

         memset( buf, 0, sizeof( buf ) );
         memset( tmp, 0, sizeof( tmp ) );
         aes_setkey_enc( &aes, tmp, keysize );

         set_alarm( 1 );

         for( i = 1; ! alarmed; i++ )
             aes_crypt_cbc( &aes, AES_ENCRYPT, BUFSIZE, tmp, buf, buf );

         tsc = hardclock();
         for( j = 0; j < 4096; j++ )
             aes_crypt_cbc( &aes, AES_ENCRYPT, BUFSIZE, tmp, buf, buf );

         printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                         ( hardclock() - tsc ) / ( j * BUFSIZE ) );
     }
 #if defined(POLARSSL_GCM_C)
     for( keysize = 128; keysize <= 256; keysize += 64 )
     {
         printf( "  AES-GCM-%d     :  ", keysize );
         fflush( stdout );

         memset( buf, 0, sizeof( buf ) );
         memset( tmp, 0, sizeof( tmp ) );
         gcm_init( &gcm, tmp, keysize );

         set_alarm( 1 );

         for( i = 1; ! alarmed; i++ )
             gcm_crypt_and_tag( &gcm, GCM_ENCRYPT, BUFSIZE, tmp, 12, NULL, 0, buf, buf, 16, tmp );

         tsc = hardclock();
         for( j = 0; j < 4096; j++ )
             gcm_crypt_and_tag( &gcm, GCM_ENCRYPT, BUFSIZE, tmp, 12, NULL, 0, buf, buf, 16, tmp );

         printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                         ( hardclock() - tsc ) / ( j * BUFSIZE ) );
     }
 #endif
 #endif

 #if defined(POLARSSL_CAMELLIA_C)
     for( keysize = 128; keysize <= 256; keysize += 64 )
     {
         printf( "  CAMELLIA-CBC-%d:  ", keysize );
         fflush( stdout );

         memset( buf, 0, sizeof( buf ) );
         memset( tmp, 0, sizeof( tmp ) );
         camellia_setkey_enc( &camellia, tmp, keysize );

         set_alarm( 1 );

         for( i = 1; ! alarmed; i++ )
             camellia_crypt_cbc( &camellia, CAMELLIA_ENCRYPT, BUFSIZE, tmp, buf, buf );

         tsc = hardclock();
         for( j = 0; j < 4096; j++ )
             camellia_crypt_cbc( &camellia, CAMELLIA_ENCRYPT, BUFSIZE, tmp, buf, buf );

         printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                         ( hardclock() - tsc ) / ( j * BUFSIZE ) );
     }
 #endif

 #if defined(POLARSSL_BLOWFISH_C)
     for( keysize = 128; keysize <= 256; keysize += 64 )
     {
         printf( "  BLOWFISH-CBC-%d:  ", keysize );
         fflush( stdout );

         memset( buf, 0, sizeof( buf ) );
         memset( tmp, 0, sizeof( tmp ) );
         blowfish_setkey( &blowfish, tmp, keysize );

         set_alarm( 1 );

         for( i = 1; ! alarmed; i++ )
             blowfish_crypt_cbc( &blowfish, BLOWFISH_ENCRYPT, BUFSIZE, tmp, buf, buf );

         tsc = hardclock();
         for( j = 0; j < 4096; j++ )
             blowfish_crypt_cbc( &blowfish, BLOWFISH_ENCRYPT, BUFSIZE, tmp, buf, buf );

         printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                         ( hardclock() - tsc ) / ( j * BUFSIZE ) );
     }
 #endif

 #if defined(POLARSSL_HAVEGE_C)
     printf( HEADER_FORMAT, "HAVEGE" );
     fflush( stdout );

     havege_init( &hs );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         havege_random( &hs, buf, BUFSIZE );

     tsc = hardclock();
     for( j = 1; j < 1024; j++ )
         havege_random( &hs, buf, BUFSIZE );

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_CTR_DRBG_C)
     printf( HEADER_FORMAT, "CTR_DRBG (NOPR)" );
     fflush( stdout );

     if( ctr_drbg_init( &ctr_drbg, myrand, NULL, NULL, 0 ) != 0 )
         exit(1);

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
             exit(1);

     tsc = hardclock();
     for( j = 1; j < 1024; j++ )
         if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
             exit(1);

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );

     printf( HEADER_FORMAT, "CTR_DRBG (PR)" );
     fflush( stdout );

     if( ctr_drbg_init( &ctr_drbg, myrand, NULL, NULL, 0 ) != 0 )
         exit(1);

     ctr_drbg_set_prediction_resistance( &ctr_drbg, CTR_DRBG_PR_ON );

     set_alarm( 1 );
     for( i = 1; ! alarmed; i++ )
         if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
             exit(1);

     tsc = hardclock();
     for( j = 1; j < 1024; j++ )
         if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
             exit(1);

     printf( "%9lu Kb/s,  %9lu cycles/byte\n", i * BUFSIZE / 1024,
                     ( hardclock() - tsc ) / ( j * BUFSIZE ) );
 #endif

 #if defined(POLARSSL_RSA_C) && defined(POLARSSL_BIGNUM_C) &&    \
     defined(POLARSSL_GENPRIME)
     rsa_init( &rsa, RSA_PKCS_V15, 0 );
     rsa_gen_key( &rsa, myrand, NULL, 1024, 65537 );

     printf( HEADER_FORMAT, "RSA-1024" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_public( &rsa, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu  public/s\n", i / 3 );

     printf( HEADER_FORMAT, "RSA-1024" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_private( &rsa, myrand, NULL, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu private/s\n", i / 3 );

     rsa_free( &rsa );

     rsa_init( &rsa, RSA_PKCS_V15, 0 );
     rsa_gen_key( &rsa, myrand, NULL, 2048, 65537 );

     printf( HEADER_FORMAT, "RSA-2048" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_public( &rsa, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu  public/s\n", i / 3 );

     printf( HEADER_FORMAT, "RSA-2048" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_private( &rsa, myrand, NULL, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu private/s\n", i / 3 );

     rsa_free( &rsa );

     rsa_init( &rsa, RSA_PKCS_V15, 0 );
     rsa_gen_key( &rsa, myrand, NULL, 4096, 65537 );

     printf( HEADER_FORMAT, "RSA-4096" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_public( &rsa, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu  public/s\n", i / 3 );

     printf( HEADER_FORMAT, "RSA-4096" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         buf[0] = 0;
         ret = rsa_private( &rsa, myrand, NULL, buf, buf );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu private/s\n", i / 3 );

     rsa_free( &rsa );
 #endif

 #if defined(POLARSSL_DHM_C) && defined(POLARSSL_BIGNUM_C)
     memset( &dhm, 0, sizeof( dhm_context ) );

     mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC5114_MODP_1024_P );
     mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC5114_MODP_1024_G );
     dhm.len = mpi_size( &dhm.P );
     dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
     mpi_copy( &dhm.GY, &dhm.GX );

     printf( HEADER_FORMAT, "DHM-1024" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
         ret |= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     printf( HEADER_FORMAT, "fixed-DHM-1024" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     dhm_free( &dhm );

     memset( &dhm, 0, sizeof( dhm_context ) );

     mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC3526_MODP_2048_P );
     mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC3526_MODP_2048_G );
     dhm.len = mpi_size( &dhm.P );
     dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
     mpi_copy( &dhm.GY, &dhm.GX );

     printf( HEADER_FORMAT, "DHM-2048" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
         ret |= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     printf( HEADER_FORMAT, "fixed-DHM-2048" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     dhm_free( &dhm );
     memset( &dhm, 0, sizeof( dhm_context ) );

     mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC3526_MODP_3072_P );
     mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC3526_MODP_3072_G );
     dhm.len = mpi_size( &dhm.P );
     dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
     mpi_copy( &dhm.GY, &dhm.GX );

     printf( HEADER_FORMAT, "DHM-3072" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
         ret |= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     printf( HEADER_FORMAT, "fixed-DHM-3072" );
     fflush( stdout );
     set_alarm( 3 );

     ret = 0;
     for( i = 1; ! alarmed && ! ret ; i++ )
     {
         olen = sizeof( buf );
         ret |= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
     }

     if( ret != 0 )
         printf( "FAILED\n" );
     else
         printf( "%9lu handshake/s\n", i / 3 );

     dhm_free( &dhm );
 #endif

     printf( "\n" );

 #if defined(_WIN32)
     printf( "  Press Enter to exit this program.\n" );
     fflush( stdout ); getchar();
 #endif

     return( 0 );
 }
 #endif /* POLARSSL_TIMING_C */
	/*
	* Benchmark demonstration program
	*
	* Copyright (C) 2006-2013, Brainspark B.V.
	*
	* This file is part of PolarSSL (http://www.polarssl.org)
	* Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
	*
	* 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 <string.h>
	#include <stdlib.h>
	#include <stdio.h>

	#include "polarssl/config.h"

	#include "polarssl/md4.h"
	#include "polarssl/md5.h"
	#include "polarssl/sha1.h"
	#include "polarssl/sha256.h"
	#include "polarssl/sha512.h"
	#include "polarssl/arc4.h"
	#include "polarssl/des.h"
	#include "polarssl/aes.h"
	#include "polarssl/blowfish.h"
	#include "polarssl/camellia.h"
	#include "polarssl/gcm.h"
	#include "polarssl/rsa.h"
	#include "polarssl/dhm.h"
	#include "polarssl/timing.h"
	#include "polarssl/havege.h"
	#include "polarssl/ctr_drbg.h"

	#define BUFSIZE 1024
	#define HEADER_FORMAT " %-15s : "

	#if defined(POLARSSL_TIMING_C)
	static int myrand( void rng_state, unsigned char output, size_t len )
	{
	size_t use_len;
	int rnd;

	if( rng_state != NULL )
	rng_state = NULL;

	while( len > 0 )
	{
	use_len = len;
	if( use_len > sizeof(int) )
	use_len = sizeof(int);

	rnd = rand();
	memcpy( output, &rnd, use_len );
	output += use_len;
	len -= use_len;
	}

	return( 0 );
	}

	unsigned char buf[BUFSIZE];
	#endif

	#if !defined(POLARSSL_TIMING_C)
	int main( int argc, char *argv[] )
	{
	((void) argc);
	((void) argv);

	printf("POLARSSL_TIMING_C not defined.\n");
	return( 0 );
	}
	#else
	int main( int argc, char *argv[] )
	{
	int ret, keysize;
	unsigned long i, j, tsc;
	unsigned char tmp[64];
	#if defined(POLARSSL_ARC4_C)
	arc4_context arc4;
	#endif
	#if defined(POLARSSL_DES_C)
	des3_context des3;
	des_context des;
	#endif
	#if defined(POLARSSL_AES_C)
	aes_context aes;
	#if defined(POLARSSL_GCM_C)
	gcm_context gcm;
	#endif
	#endif
	#if defined(POLARSSL_BLOWFISH_C)
	blowfish_context blowfish;
	#endif
	#if defined(POLARSSL_CAMELLIA_C)
	camellia_context camellia;
	#endif
	#if defined(POLARSSL_RSA_C) && defined(POLARSSL_BIGNUM_C) && \
	defined(POLARSSL_GENPRIME)
	rsa_context rsa;
	#endif
	#if defined(POLARSSL_HAVEGE_C)
	havege_state hs;
	#endif
	#if defined(POLARSSL_CTR_DRBG_C)
	ctr_drbg_context ctr_drbg;
	#endif
	#if defined(POLARSSL_DHM_C) && defined(POLARSSL_BIGNUM_C)
	dhm_context dhm;
	size_t olen = BUFSIZE;
	#endif
	((void) argc);
	((void) argv);

	memset( buf, 0xAA, sizeof( buf ) );

	printf( "\n" );

	#if defined(POLARSSL_MD4_C)
	printf( HEADER_FORMAT, "MD4" );
	fflush( stdout );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	md4( buf, BUFSIZE, tmp );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	md4( buf, BUFSIZE, tmp );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_MD5_C)
	printf( HEADER_FORMAT, "MD5" );
	fflush( stdout );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	md5( buf, BUFSIZE, tmp );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	md5( buf, BUFSIZE, tmp );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_SHA1_C)
	printf( HEADER_FORMAT, "SHA-1" );
	fflush( stdout );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	sha1( buf, BUFSIZE, tmp );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	sha1( buf, BUFSIZE, tmp );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_SHA256_C)
	printf( HEADER_FORMAT, "SHA-256" );
	fflush( stdout );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	sha256( buf, BUFSIZE, tmp, 0 );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	sha256( buf, BUFSIZE, tmp, 0 );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_SHA512_C)
	printf( HEADER_FORMAT, "SHA-512" );
	fflush( stdout );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	sha512( buf, BUFSIZE, tmp, 0 );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	sha512( buf, BUFSIZE, tmp, 0 );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_ARC4_C)
	printf( HEADER_FORMAT, "ARC4" );
	fflush( stdout );

	arc4_setup( &arc4, tmp, 32 );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	arc4_crypt( &arc4, BUFSIZE, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	arc4_crypt( &arc4, BUFSIZE, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_DES_C)
	printf( HEADER_FORMAT, "3DES" );
	fflush( stdout );

	des3_set3key_enc( &des3, tmp );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	des3_crypt_cbc( &des3, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	des3_crypt_cbc( &des3, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );

	printf( HEADER_FORMAT, "DES" );
	fflush( stdout );

	des_setkey_enc( &des, tmp );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	des_crypt_cbc( &des, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 1024; j++ )
	des_crypt_cbc( &des, DES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_AES_C)
	for( keysize = 128; keysize <= 256; keysize += 64 )
	{
	printf( " AES-CBC-%d : ", keysize );
	fflush( stdout );

	memset( buf, 0, sizeof( buf ) );
	memset( tmp, 0, sizeof( tmp ) );
	aes_setkey_enc( &aes, tmp, keysize );

	set_alarm( 1 );

	for( i = 1; ! alarmed; i++ )
	aes_crypt_cbc( &aes, AES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 4096; j++ )
	aes_crypt_cbc( &aes, AES_ENCRYPT, BUFSIZE, tmp, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	}
	#if defined(POLARSSL_GCM_C)
	for( keysize = 128; keysize <= 256; keysize += 64 )
	{
	printf( " AES-GCM-%d : ", keysize );
	fflush( stdout );

	memset( buf, 0, sizeof( buf ) );
	memset( tmp, 0, sizeof( tmp ) );
	gcm_init( &gcm, tmp, keysize );

	set_alarm( 1 );

	for( i = 1; ! alarmed; i++ )
	gcm_crypt_and_tag( &gcm, GCM_ENCRYPT, BUFSIZE, tmp, 12, NULL, 0, buf, buf, 16, tmp );

	tsc = hardclock();
	for( j = 0; j < 4096; j++ )
	gcm_crypt_and_tag( &gcm, GCM_ENCRYPT, BUFSIZE, tmp, 12, NULL, 0, buf, buf, 16, tmp );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	}
	#endif
	#endif

	#if defined(POLARSSL_CAMELLIA_C)
	for( keysize = 128; keysize <= 256; keysize += 64 )
	{
	printf( " CAMELLIA-CBC-%d: ", keysize );
	fflush( stdout );

	memset( buf, 0, sizeof( buf ) );
	memset( tmp, 0, sizeof( tmp ) );
	camellia_setkey_enc( &camellia, tmp, keysize );

	set_alarm( 1 );

	for( i = 1; ! alarmed; i++ )
	camellia_crypt_cbc( &camellia, CAMELLIA_ENCRYPT, BUFSIZE, tmp, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 4096; j++ )
	camellia_crypt_cbc( &camellia, CAMELLIA_ENCRYPT, BUFSIZE, tmp, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	}
	#endif

	#if defined(POLARSSL_BLOWFISH_C)
	for( keysize = 128; keysize <= 256; keysize += 64 )
	{
	printf( " BLOWFISH-CBC-%d: ", keysize );
	fflush( stdout );

	memset( buf, 0, sizeof( buf ) );
	memset( tmp, 0, sizeof( tmp ) );
	blowfish_setkey( &blowfish, tmp, keysize );

	set_alarm( 1 );

	for( i = 1; ! alarmed; i++ )
	blowfish_crypt_cbc( &blowfish, BLOWFISH_ENCRYPT, BUFSIZE, tmp, buf, buf );

	tsc = hardclock();
	for( j = 0; j < 4096; j++ )
	blowfish_crypt_cbc( &blowfish, BLOWFISH_ENCRYPT, BUFSIZE, tmp, buf, buf );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	}
	#endif

	#if defined(POLARSSL_HAVEGE_C)
	printf( HEADER_FORMAT, "HAVEGE" );
	fflush( stdout );

	havege_init( &hs );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	havege_random( &hs, buf, BUFSIZE );

	tsc = hardclock();
	for( j = 1; j < 1024; j++ )
	havege_random( &hs, buf, BUFSIZE );

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_CTR_DRBG_C)
	printf( HEADER_FORMAT, "CTR_DRBG (NOPR)" );
	fflush( stdout );

	if( ctr_drbg_init( &ctr_drbg, myrand, NULL, NULL, 0 ) != 0 )
	exit(1);

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
	exit(1);

	tsc = hardclock();
	for( j = 1; j < 1024; j++ )
	if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
	exit(1);

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );

	printf( HEADER_FORMAT, "CTR_DRBG (PR)" );
	fflush( stdout );

	if( ctr_drbg_init( &ctr_drbg, myrand, NULL, NULL, 0 ) != 0 )
	exit(1);

	ctr_drbg_set_prediction_resistance( &ctr_drbg, CTR_DRBG_PR_ON );

	set_alarm( 1 );
	for( i = 1; ! alarmed; i++ )
	if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
	exit(1);

	tsc = hardclock();
	for( j = 1; j < 1024; j++ )
	if( ctr_drbg_random( &ctr_drbg, buf, BUFSIZE ) != 0 )
	exit(1);

	printf( "%9lu Kb/s, %9lu cycles/byte\n", i * BUFSIZE / 1024,
	( hardclock() - tsc ) / ( j * BUFSIZE ) );
	#endif

	#if defined(POLARSSL_RSA_C) && defined(POLARSSL_BIGNUM_C) && \
	defined(POLARSSL_GENPRIME)
	rsa_init( &rsa, RSA_PKCS_V15, 0 );
	rsa_gen_key( &rsa, myrand, NULL, 1024, 65537 );

	printf( HEADER_FORMAT, "RSA-1024" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_public( &rsa, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu public/s\n", i / 3 );

	printf( HEADER_FORMAT, "RSA-1024" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_private( &rsa, myrand, NULL, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu private/s\n", i / 3 );

	rsa_free( &rsa );

	rsa_init( &rsa, RSA_PKCS_V15, 0 );
	rsa_gen_key( &rsa, myrand, NULL, 2048, 65537 );

	printf( HEADER_FORMAT, "RSA-2048" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_public( &rsa, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu public/s\n", i / 3 );

	printf( HEADER_FORMAT, "RSA-2048" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_private( &rsa, myrand, NULL, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu private/s\n", i / 3 );

	rsa_free( &rsa );

	rsa_init( &rsa, RSA_PKCS_V15, 0 );
	rsa_gen_key( &rsa, myrand, NULL, 4096, 65537 );

	printf( HEADER_FORMAT, "RSA-4096" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_public( &rsa, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu public/s\n", i / 3 );

	printf( HEADER_FORMAT, "RSA-4096" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	buf[0] = 0;
	ret = rsa_private( &rsa, myrand, NULL, buf, buf );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu private/s\n", i / 3 );

	rsa_free( &rsa );
	#endif

	#if defined(POLARSSL_DHM_C) && defined(POLARSSL_BIGNUM_C)
	memset( &dhm, 0, sizeof( dhm_context ) );

	mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC5114_MODP_1024_P );
	mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC5114_MODP_1024_G );
	dhm.len = mpi_size( &dhm.P );
	dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	mpi_copy( &dhm.GY, &dhm.GX );

	printf( HEADER_FORMAT, "DHM-1024" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	printf( HEADER_FORMAT, "fixed-DHM-1024" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	dhm_free( &dhm );

	memset( &dhm, 0, sizeof( dhm_context ) );

	mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC3526_MODP_2048_P );
	mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC3526_MODP_2048_G );
	dhm.len = mpi_size( &dhm.P );
	dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	mpi_copy( &dhm.GY, &dhm.GX );

	printf( HEADER_FORMAT, "DHM-2048" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	printf( HEADER_FORMAT, "fixed-DHM-2048" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	dhm_free( &dhm );
	memset( &dhm, 0, sizeof( dhm_context ) );

	mpi_read_string( &dhm.P, 16, POLARSSL_DHM_RFC3526_MODP_3072_P );
	mpi_read_string( &dhm.G, 16, POLARSSL_DHM_RFC3526_MODP_3072_G );
	dhm.len = mpi_size( &dhm.P );
	dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	mpi_copy( &dhm.GY, &dhm.GX );

	printf( HEADER_FORMAT, "DHM-3072" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_make_public( &dhm, dhm.len, buf, dhm.len, myrand, NULL );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, NULL, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	printf( HEADER_FORMAT, "fixed-DHM-3072" );
	fflush( stdout );
	set_alarm( 3 );

	ret = 0;
	for( i = 1; ! alarmed && ! ret ; i++ )
	{
	olen = sizeof( buf );
	ret \|= dhm_calc_secret( &dhm, buf, &olen, myrand, NULL );
	}

	if( ret != 0 )
	printf( "FAILED\n" );
	else
	printf( "%9lu handshake/s\n", i / 3 );

	dhm_free( &dhm );
	#endif

	printf( "\n" );

	#if defined(_WIN32)
	printf( " Press Enter to exit this program.\n" );
	fflush( stdout ); getchar();
	#endif

	return( 0 );
	}
	#endif /* POLARSSL_TIMING_C */