diff --git a/benchmark.c b/benchmark.c index 10492a4..e433698 100644 --- a/benchmark.c +++ b/benchmark.c @@ -70,14 +70,38 @@ uint8_t PKT_CONTENT[]={ /* Prototypes */ static ssize_t do_encode_packet( uint8_t * pktbuf, size_t bufsize, const n2n_community_t c ); +static void run_transop_benchmark(const char *op_name, n2n_trans_op_t *op_fn, uint8_t *pktbuf, n2n_community_t c); int main(int argc, char * argv[]) { uint8_t pktbuf[N2N_PKT_BUF_SIZE]; - n2n_trans_op_t transop_null; + n2n_community_t c; + n2n_trans_op_t transop_null, transop_twofish, transop_aes_cbc; + u_char encrypt_pwd[] = "SoMEVer!S$cUREPassWORD"; + memset(c,0,sizeof(N2N_COMMUNITY_SIZE)); + memcpy(c, "abc123def456", 12); + + /* Init transopts */ + memset(&transop_null, 0, sizeof(transop_null)); + transop_null_init(&transop_null); + memset(&transop_twofish, 0, sizeof(transop_twofish)); + transop_twofish_init(&transop_twofish); + transop_twofish_setup_psk(&transop_twofish, 0, encrypt_pwd, sizeof(encrypt_pwd)-1); + memset(&transop_aes_cbc, 0, sizeof(transop_aes_cbc)); + transop_twofish_init(&transop_aes_cbc); + transop_twofish_setup_psk(&transop_aes_cbc, 0, encrypt_pwd, sizeof(encrypt_pwd)-1); + + /* Run the tests */ + run_transop_benchmark("transop_null", &transop_null, pktbuf, c); + run_transop_benchmark("transop_twofish", &transop_twofish, pktbuf, c); + run_transop_benchmark("transop_aes", &transop_aes_cbc, pktbuf, c); + + return 0; +} + +static void run_transop_benchmark(const char *op_name, n2n_trans_op_t *op_fn, uint8_t *pktbuf, n2n_community_t c) { n2n_common_t cmn; n2n_PACKET_t pkt; - n2n_community_t c; struct timeval t1; struct timeval t2; @@ -89,18 +113,14 @@ int main(int argc, char * argv[]) { ssize_t nw; ssize_t tdiff; - transop_null_init( &transop_null ); - memset(c,0,sizeof(N2N_COMMUNITY_SIZE)); - n=10000; - memcpy( c, "abc123def456", 12 ); gettimeofday( &t1, NULL ); for(i=0; ifwd( op_fn, pktbuf+nw, N2N_PKT_BUF_SIZE-nw, PKT_CONTENT, sizeof(PKT_CONTENT) ); @@ -110,7 +130,7 @@ int main(int argc, char * argv[]) { decode_common( &cmn, pktbuf, &rem, &idx); decode_PACKET( &pkt, &cmn, pktbuf, &rem, &idx ); - if ( 0 == (i%1000) ) + if ( 0 == (i%(n/10)) ) { fprintf(stderr,"."); } @@ -119,13 +139,12 @@ int main(int argc, char * argv[]) { tdiff = ((t2.tv_sec - t1.tv_sec) * 1000000) + (t2.tv_usec - t1.tv_usec); - fprintf( stderr, "\nrun %u times. (%u -> %u nsec each) %u.%06u -> %u.%06u.\n", + fprintf( stderr, "\n[%s] %u times: (%u -> %u nsec each) %u.%06u -> %u.%06u.\n", + op_name, (unsigned int)i, (unsigned int)tdiff, (unsigned int)((tdiff *1000)/i), (uint32_t)t1.tv_sec, (uint32_t)t1.tv_usec, (uint32_t)t2.tv_sec, (uint32_t)t2.tv_usec ); - - return 0; } static ssize_t do_encode_packet( uint8_t * pktbuf, size_t bufsize, const n2n_community_t c ) @@ -154,3 +173,4 @@ static ssize_t do_encode_packet( uint8_t * pktbuf, size_t bufsize, const n2n_com return idx; } +