X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=tools%2Fcrypto_bench.c;h=aca8bbb1a85d6d2eda7875c1250afd18cdd1bc5f;hb=1cf6129d13a33ab43c4e403f342a5756b69482ea;hp=203bffe19832866451d76c1ee363a33414e6647e;hpb=d28eb0e34d80f12b779c7b6befc5b350ffcdf476;p=ffmpeg diff --git a/tools/crypto_bench.c b/tools/crypto_bench.c index 203bffe1983..aca8bbb1a85 100644 --- a/tools/crypto_bench.c +++ b/tools/crypto_bench.c @@ -23,6 +23,7 @@ #define USE_crypto 0x01 /* OpenSSL's libcrypto */ #define USE_gcrypt 0x02 /* GnuTLS's libgcrypt */ #define USE_tomcrypt 0x04 /* LibTomCrypt */ +#define USE_mbedcrypto 0x08 /* mbed TLS */ #include #include @@ -311,29 +312,140 @@ DEFINE_GCRYPT_WRAPPER(sha256, SHA256) DEFINE_GCRYPT_WRAPPER(sha512, SHA512) DEFINE_GCRYPT_WRAPPER(ripemd160, RMD160) -#define DEFINE_GCRYPT_CYPHER_WRAPPER(suffix, cypher, sz) \ +#define DEFINE_GCRYPT_CYPHER_WRAPPER(suffix, cypher, mode, sz) \ static void run_gcrypt_ ## suffix(uint8_t *output, \ const uint8_t *input, unsigned size) \ { \ static gcry_cipher_hd_t suffix; \ if (!suffix) \ - gcry_cipher_open(&suffix, GCRY_CIPHER_ ## cypher, GCRY_CIPHER_MODE_ECB, 0); \ + gcry_cipher_open(&suffix, GCRY_CIPHER_ ## cypher, GCRY_CIPHER_MODE_ ## mode, 0); \ gcry_cipher_setkey(suffix, hardcoded_key, sz); \ gcry_cipher_encrypt(suffix, output, size, input, size); \ } -DEFINE_GCRYPT_CYPHER_WRAPPER(aes128, AES128, 16) -DEFINE_GCRYPT_CYPHER_WRAPPER(blowfish, BLOWFISH, 16) -DEFINE_GCRYPT_CYPHER_WRAPPER(camellia, CAMELLIA128, 16) -DEFINE_GCRYPT_CYPHER_WRAPPER(cast128, CAST5, 16) -DEFINE_GCRYPT_CYPHER_WRAPPER(des, DES, 8) -DEFINE_GCRYPT_CYPHER_WRAPPER(twofish, TWOFISH128, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(aes128, AES128, ECB, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(blowfish, BLOWFISH, ECB, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(camellia, CAMELLIA128, ECB, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(cast128, CAST5, ECB, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(des, DES, ECB, 8) +DEFINE_GCRYPT_CYPHER_WRAPPER(twofish, TWOFISH128, ECB, 16) +DEFINE_GCRYPT_CYPHER_WRAPPER(rc4, ARCFOUR, STREAM, 16) #define IMPL_USE_gcrypt(...) IMPL_USE(__VA_ARGS__) #else #define IMPL_USE_gcrypt(...) /* ignore */ #endif +/*************************************************************************** + * mbedcrypto: mbed TLS + ***************************************************************************/ + +#if (USE_EXT_LIBS) & USE_mbedcrypto + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define DEFINE_MBEDCRYPTO_WRAPPER(suffix) \ +static void run_mbedcrypto_ ## suffix(uint8_t *output, \ + const uint8_t *input, unsigned size) \ +{ \ + mbedtls_ ## suffix ## _ret(input, size, output); \ +} + +#define DEFINE_MBEDCRYPTO_WRAPPER_SHA2(suffix) \ +static void run_mbedcrypto_ ## suffix(uint8_t *output, \ + const uint8_t *input, unsigned size) \ +{ \ + mbedtls_ ## suffix ## _ret(input, size, output, 0); \ +} + +DEFINE_MBEDCRYPTO_WRAPPER(md5) +DEFINE_MBEDCRYPTO_WRAPPER(ripemd160) +DEFINE_MBEDCRYPTO_WRAPPER(sha1) +DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha256) +DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha512) + + +#define DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(suffix, cypher, algo) \ +static void run_mbedcrypto_ ## suffix(uint8_t *output, \ + const uint8_t *input, unsigned size) \ +{ \ + mbedtls_ ## cypher ## _context cypher; \ + \ + mbedtls_ ## cypher ## _init(&cypher); \ + mbedtls_ ## cypher ## _setkey_enc(&cypher, hardcoded_key, 128); \ + for (int i = 0; i < size; i += 16) \ + mbedtls_ ## cypher ## _crypt_ecb(&cypher, MBEDTLS_ ## algo ## _ENCRYPT, \ + input + i, output + i); \ + mbedtls_ ## cypher ## _free(&cypher); \ +} + +DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(aes128, aes, AES) +DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(camellia, camellia, CAMELLIA) + +static void run_mbedcrypto_blowfish(uint8_t *output, + const uint8_t *input, unsigned size) +{ + mbedtls_blowfish_context blowfish; + + mbedtls_blowfish_init(&blowfish); + mbedtls_blowfish_setkey(&blowfish, hardcoded_key, 128); + for (int i = 0; i < size; i += 8) + mbedtls_blowfish_crypt_ecb(&blowfish, MBEDTLS_BLOWFISH_ENCRYPT, + input + i, output + i); + mbedtls_blowfish_free(&blowfish); +} + +static void run_mbedcrypto_des(uint8_t *output, + const uint8_t *input, unsigned size) +{ + mbedtls_des_context des; + + mbedtls_des_init(&des); + mbedtls_des_setkey_enc(&des, hardcoded_key); + for (int i = 0; i < size; i += 8) + mbedtls_des_crypt_ecb(&des, input + i, output + i); + mbedtls_des_free(&des); +} + +static void run_mbedcrypto_rc4(uint8_t *output, + const uint8_t *input, unsigned size) +{ + mbedtls_arc4_context rc4; + + mbedtls_arc4_init(&rc4); + mbedtls_arc4_setup(&rc4, hardcoded_key, 16); + mbedtls_arc4_crypt(&rc4, size, input, output); + mbedtls_arc4_free(&rc4); +} + +static void run_mbedcrypto_xtea(uint8_t *output, + const uint8_t *input, unsigned size) +{ + mbedtls_xtea_context xtea; + + mbedtls_xtea_init(&xtea); + mbedtls_xtea_setup(&xtea, hardcoded_key); + for (int i = 0; i < size; i += 8) + mbedtls_xtea_crypt_ecb(&xtea, MBEDTLS_XTEA_ENCRYPT, + input + i, output + i); + mbedtls_xtea_free(&xtea); +} + +#define IMPL_USE_mbedcrypto(...) IMPL_USE(__VA_ARGS__) +#else +#define IMPL_USE_mbedcrypto(...) /* ignore */ +#endif + /*************************************************************************** * tomcrypt: LibTomCrypt ***************************************************************************/ @@ -416,6 +528,16 @@ static void run_tomcrypt_des(uint8_t *output, des_ecb_encrypt(input + i, output + i, &des); } +static void run_tomcrypt_rc4(uint8_t *output, + const uint8_t *input, unsigned size) +{ + rc4_state rc4; + + rc4_stream_setup(&rc4, hardcoded_key, 16); + rc4_stream_crypt(&rc4, input, size, output); + rc4_stream_done(&rc4); +} + static void run_tomcrypt_twofish(uint8_t *output, const uint8_t *input, unsigned size) { @@ -512,6 +634,7 @@ static void run_implementation(const uint8_t *input, uint8_t *output, IMPL(lavu, __VA_ARGS__) \ IMPL(crypto, __VA_ARGS__) \ IMPL(gcrypt, __VA_ARGS__) \ + IMPL(mbedcrypto, __VA_ARGS__) \ IMPL(tomcrypt, __VA_ARGS__) struct hash_impl implementations[] = { @@ -525,15 +648,18 @@ struct hash_impl implementations[] = { IMPL_ALL("RIPEMD-160", ripemd160, "62a5321e4fc8784903bb43ab7752c75f8b25af00") IMPL_ALL("AES-128", aes128, "crc:ff6bc888") IMPL_ALL("CAMELLIA", camellia, "crc:7abb59a7") - IMPL_ALL("CAST-128", cast128, "crc:456aa584") + IMPL(lavu, "CAST-128", cast128, "crc:456aa584") + IMPL(crypto, "CAST-128", cast128, "crc:456aa584") + IMPL(gcrypt, "CAST-128", cast128, "crc:456aa584") + IMPL(tomcrypt, "CAST-128", cast128, "crc:456aa584") IMPL_ALL("BLOWFISH", blowfish, "crc:33e8aa74") IMPL_ALL("DES", des, "crc:31291e0b") IMPL(lavu, "TWOFISH", twofish, "crc:9edbd5c1") IMPL(gcrypt, "TWOFISH", twofish, "crc:9edbd5c1") IMPL(tomcrypt, "TWOFISH", twofish, "crc:9edbd5c1") - IMPL(lavu, "RC4", rc4, "crc:538d37b2") - IMPL(crypto, "RC4", rc4, "crc:538d37b2") + IMPL_ALL("RC4", rc4, "crc:538d37b2") IMPL(lavu, "XTEA", xtea, "crc:931fc270") + IMPL(mbedcrypto, "XTEA", xtea, "crc:931fc270") IMPL(tomcrypt, "XTEA", xtea, "crc:931fc270") }; @@ -561,15 +687,16 @@ int main(int argc, char **argv) argv[0]); if ((USE_EXT_LIBS)) { char buf[1024]; - snprintf(buf, sizeof(buf), "%s%s%s", + snprintf(buf, sizeof(buf), "%s%s%s%s", ((USE_EXT_LIBS) & USE_crypto) ? "+crypto" : "", ((USE_EXT_LIBS) & USE_gcrypt) ? "+gcrypt" : "", + ((USE_EXT_LIBS) & USE_mbedcrypto) ? "+mbedcrypto" : "", ((USE_EXT_LIBS) & USE_tomcrypt) ? "+tomcrypt" : ""); fprintf(stderr, "Built with the following external libraries:\n" "make VERSUS=%s\n", buf + 1); } else { fprintf(stderr, "Built without external libraries; use\n" - "make VERSUS=crypto+gcrypt+tomcrypt tools/crypto_bench\n" + "make VERSUS=crypto+gcrypt+mbedcrypto+tomcrypt tools/crypto_bench\n" "to enable them.\n"); } exit(opt != 'h');