1 /*
2 * Copyright (c) 2013 Nicolas George
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public License
8 * as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with FFmpeg; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21 /* Optional external libraries; can be enabled using:
22 * make VERSUS=crypto+gcrypt+tomcrypt+mbedcrypto tools/crypto_bench */
23 #define USE_crypto 0x01 /* OpenSSL's libcrypto */
24 #define USE_gcrypt 0x02 /* GnuTLS's libgcrypt */
25 #define USE_tomcrypt 0x04 /* LibTomCrypt */
26 #define USE_mbedcrypto 0x08 /* mbed TLS */
27
28 #include <stdlib.h>
29 #include <math.h>
30
37
38 #ifndef AV_READ_TIME
39 #define AV_READ_TIME(x) 0
40 #endif
41
42 #if HAVE_UNISTD_H
43 #include <unistd.h> /* for getopt */
44 #endif
45 #if !HAVE_GETOPT
47 #endif
48
49 #define MAX_INPUT_SIZE 1048576
50 #define MAX_OUTPUT_SIZE 128
51
55
56 static const uint8_t *
hardcoded_key =
"FFmpeg is the best program ever.";
57
59 {
61 exit(1);
62 }
63
69 };
70
71 /***************************************************************************
72 * lavu: libavutil
73 ***************************************************************************/
74
87
88 #define IMPL_USE_lavu IMPL_USE
89
92 {
94 }
95
96 #define DEFINE_LAVU_MD(suffix, type, namespace, hsize) \
97 static void run_lavu_ ## suffix(uint8_t *output, \
98 const uint8_t *input, unsigned size) \
99 { \
100 static struct type *h; \
101 if (!h && !(h = av_ ## namespace ## _alloc())) \
102 fatal_error("out of memory"); \
103 av_ ## namespace ## _init(h, hsize); \
104 av_ ## namespace ## _update(h, input, size); \
105 av_ ## namespace ## _final(h, output); \
106 }
107
113
116 {
117 static struct AVAES *aes;
122 }
123
126 {
132 }
133
136 {
142 }
143
146 {
152 }
153
156 {
157 static struct AVDES *des;
162 }
163
166 {
172 }
173
176 {
177 static struct AVRC4 *rc4;
182 }
183
186 {
187 static struct AVXTEA *xtea;
192 }
193
194 /***************************************************************************
195 * crypto: OpenSSL's libcrypto
196 ***************************************************************************/
197
198 #if (USE_EXT_LIBS) & USE_crypto
199
200 #define OPENSSL_DISABLE_OLD_DES_SUPPORT
201 #include <openssl/md5.h>
202 #include <openssl/sha.h>
203 #include <openssl/ripemd.h>
204 #include <openssl/aes.h>
205 #include <openssl/blowfish.h>
206 #include <openssl/camellia.h>
207 #include <openssl/cast.h>
208 #include <openssl/des.h>
209 #include <openssl/rc4.h>
210
211 #define DEFINE_CRYPTO_WRAPPER(suffix, function) \
212 static void run_crypto_ ## suffix(uint8_t *output, \
213 const uint8_t *input, unsigned size) \
214 { \
215 function(input, size, output); \
216 }
217
218 DEFINE_CRYPTO_WRAPPER(
md5, MD5)
219 DEFINE_CRYPTO_WRAPPER(sha1, SHA1)
220 DEFINE_CRYPTO_WRAPPER(sha256, SHA256)
221 DEFINE_CRYPTO_WRAPPER(sha512, SHA512)
222 DEFINE_CRYPTO_WRAPPER(ripemd160, RIPEMD160)
223
224 static void run_crypto_aes128(uint8_t *
output,
226 {
227 AES_KEY aes;
229
234 }
235
236 static void run_crypto_blowfish(uint8_t *
output,
238 {
239 BF_KEY blowfish;
241
245 }
246
247 static void run_crypto_camellia(uint8_t *
output,
249 {
250 CAMELLIA_KEY camellia;
252
257 }
258
259 static void run_crypto_cast128(uint8_t *
output,
261 {
264
268 }
269
270 static void run_crypto_des(uint8_t *
output,
272 {
273 DES_key_schedule des;
275
279 }
280
281 static void run_crypto_rc4(uint8_t *
output,
283 {
284 RC4_KEY rc4;
285
288 }
289
290 #define IMPL_USE_crypto(...) IMPL_USE(__VA_ARGS__)
291 #else
292 #define IMPL_USE_crypto(...) /* ignore */
293 #endif
294
295 /***************************************************************************
296 * gcrypt: GnuTLS's libgcrypt
297 ***************************************************************************/
298
299 #if (USE_EXT_LIBS) & USE_gcrypt
300
301 #include <gcrypt.h>
302
303 #define DEFINE_GCRYPT_WRAPPER(suffix, algo) \
304 static void run_gcrypt_ ## suffix(uint8_t *output, \
305 const uint8_t *input, unsigned size) \
306 { \
307 gcry_md_hash_buffer(GCRY_MD_ ## algo, output, input, size); \
308 }
309
310 DEFINE_GCRYPT_WRAPPER(
md5, MD5)
311 DEFINE_GCRYPT_WRAPPER(sha1, SHA1)
312 DEFINE_GCRYPT_WRAPPER(sha256, SHA256)
313 DEFINE_GCRYPT_WRAPPER(sha512, SHA512)
314 DEFINE_GCRYPT_WRAPPER(ripemd160, RMD160)
315
316 #define DEFINE_GCRYPT_CYPHER_WRAPPER(suffix, cypher, mode, sz) \
317 static void run_gcrypt_ ## suffix(uint8_t *output, \
318 const uint8_t *input, unsigned size) \
319 { \
320 static gcry_cipher_hd_t suffix; \
321 if (!suffix) \
322 gcry_cipher_open(&suffix, GCRY_CIPHER_ ## cypher, GCRY_CIPHER_MODE_ ## mode, 0); \
323 gcry_cipher_setkey(suffix, hardcoded_key, sz); \
324 gcry_cipher_encrypt(suffix, output, size, input, size); \
325 }
326
327 DEFINE_GCRYPT_CYPHER_WRAPPER(aes128, AES128, ECB, 16)
328 DEFINE_GCRYPT_CYPHER_WRAPPER(blowfish, BLOWFISH, ECB, 16)
329 DEFINE_GCRYPT_CYPHER_WRAPPER(camellia, CAMELLIA128, ECB, 16)
330 DEFINE_GCRYPT_CYPHER_WRAPPER(cast128, CAST5, ECB, 16)
331 DEFINE_GCRYPT_CYPHER_WRAPPER(des, DES, ECB, 8)
332 DEFINE_GCRYPT_CYPHER_WRAPPER(twofish, TWOFISH128, ECB, 16)
333 DEFINE_GCRYPT_CYPHER_WRAPPER(rc4, ARCFOUR, STREAM, 16)
334
335 #define IMPL_USE_gcrypt(...) IMPL_USE(__VA_ARGS__)
336 #else
337 #define IMPL_USE_gcrypt(...) /* ignore */
338 #endif
339
340 /***************************************************************************
341 * mbedcrypto: mbed TLS
342 ***************************************************************************/
343
344 #if (USE_EXT_LIBS) & USE_mbedcrypto
345
346 #include <mbedtls/aes.h>
347 #include <mbedtls/arc4.h>
348 #include <mbedtls/blowfish.h>
349 #include <mbedtls/camellia.h>
350 #include <mbedtls/des.h>
351 #include <mbedtls/md5.h>
352 #include <mbedtls/ripemd160.h>
353 #include <mbedtls/sha1.h>
354 #include <mbedtls/sha256.h>
355 #include <mbedtls/sha512.h>
356 #include <mbedtls/xtea.h>
357
358 #define DEFINE_MBEDCRYPTO_WRAPPER(suffix) \
359 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
360 const uint8_t *input, unsigned size) \
361 { \
362 mbedtls_ ## suffix ## _ret(input, size, output); \
363 }
364
365 #define DEFINE_MBEDCRYPTO_WRAPPER_SHA2(suffix) \
366 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
367 const uint8_t *input, unsigned size) \
368 { \
369 mbedtls_ ## suffix ## _ret(input, size, output, 0); \
370 }
371
372 DEFINE_MBEDCRYPTO_WRAPPER(
md5)
373 DEFINE_MBEDCRYPTO_WRAPPER(ripemd160)
374 DEFINE_MBEDCRYPTO_WRAPPER(sha1)
375 DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha256)
376 DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha512)
377
378
379 #define DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(suffix, cypher, algo) \
380 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
381 const uint8_t *input, unsigned size) \
382 { \
383 mbedtls_ ## cypher ## _context cypher; \
384 \
385 mbedtls_ ## cypher ## _init(&cypher); \
386 mbedtls_ ## cypher ## _setkey_enc(&cypher, hardcoded_key, 128); \
387 for (int i = 0; i < size; i += 16) \
388 mbedtls_ ## cypher ## _crypt_ecb(&cypher, MBEDTLS_ ## algo ## _ENCRYPT, \
389 input + i, output + i); \
390 mbedtls_ ## cypher ## _free(&cypher); \
391 }
392
393 DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(aes128, aes, AES)
394 DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(camellia, camellia, CAMELLIA)
395
396 static void run_mbedcrypto_blowfish(uint8_t *
output,
398 {
399 mbedtls_blowfish_context blowfish;
400
401 mbedtls_blowfish_init(&blowfish);
403 for (
int i = 0;
i <
size;
i += 8)
404 mbedtls_blowfish_crypt_ecb(&blowfish, MBEDTLS_BLOWFISH_ENCRYPT,
406 mbedtls_blowfish_free(&blowfish);
407 }
408
409 static void run_mbedcrypto_des(uint8_t *
output,
411 {
412 mbedtls_des_context des;
413
414 mbedtls_des_init(&des);
416 for (
int i = 0;
i <
size;
i += 8)
418 mbedtls_des_free(&des);
419 }
420
421 static void run_mbedcrypto_rc4(uint8_t *
output,
423 {
424 mbedtls_arc4_context rc4;
425
426 mbedtls_arc4_init(&rc4);
429 mbedtls_arc4_free(&rc4);
430 }
431
432 static void run_mbedcrypto_xtea(uint8_t *
output,
434 {
435 mbedtls_xtea_context xtea;
436
437 mbedtls_xtea_init(&xtea);
439 for (
int i = 0;
i <
size;
i += 8)
440 mbedtls_xtea_crypt_ecb(&xtea, MBEDTLS_XTEA_ENCRYPT,
442 mbedtls_xtea_free(&xtea);
443 }
444
445 #define IMPL_USE_mbedcrypto(...) IMPL_USE(__VA_ARGS__)
446 #else
447 #define IMPL_USE_mbedcrypto(...) /* ignore */
448 #endif
449
450 /***************************************************************************
451 * tomcrypt: LibTomCrypt
452 ***************************************************************************/
453
454 #if (USE_EXT_LIBS) & USE_tomcrypt
455
456 #include <tomcrypt.h>
457
458 #define DEFINE_TOMCRYPT_WRAPPER(suffix, namespace, algo) \
459 static void run_tomcrypt_ ## suffix(uint8_t *output, \
460 const uint8_t *input, unsigned size) \
461 { \
462 hash_state md; \
463 namespace ## _init(&md); \
464 namespace ## _process(&md, input, size); \
465 namespace ## _done(&md, output); \
466 }
467
468 DEFINE_TOMCRYPT_WRAPPER(
md5,
md5, MD5)
469 DEFINE_TOMCRYPT_WRAPPER(sha1, sha1, SHA1)
470 DEFINE_TOMCRYPT_WRAPPER(sha256, sha256, SHA256)
471 DEFINE_TOMCRYPT_WRAPPER(sha512, sha512, SHA512)
472 DEFINE_TOMCRYPT_WRAPPER(ripemd128, rmd128, RIPEMD128)
473 DEFINE_TOMCRYPT_WRAPPER(ripemd160, rmd160, RIPEMD160)
474
475 static void run_tomcrypt_aes128(uint8_t *
output,
477 {
478 symmetric_key aes;
480
485 }
486
487 static void run_tomcrypt_blowfish(uint8_t *
output,
489 {
490 symmetric_key blowfish;
492
496 }
497
498 static void run_tomcrypt_camellia(uint8_t *
output,
500 {
501 symmetric_key camellia;
503
508 }
509
510 static void run_tomcrypt_cast128(uint8_t *
output,
512 {
515
519 }
520
521 static void run_tomcrypt_des(uint8_t *
output,
523 {
524 symmetric_key des;
526
530 }
531
532 static void run_tomcrypt_rc4(uint8_t *
output,
534 {
535 rc4_state rc4;
536
539 rc4_stream_done(&rc4);
540 }
541
542 static void run_tomcrypt_twofish(uint8_t *
output,
544 {
545 symmetric_key twofish;
547
552 }
553
554 static void run_tomcrypt_xtea(uint8_t *
output,
556 {
557 symmetric_key xtea;
559
563 }
564
565
566 #define IMPL_USE_tomcrypt(...) IMPL_USE(__VA_ARGS__)
567 #else
568 #define IMPL_USE_tomcrypt(...) /* ignore */
569 #endif
570
571 /***************************************************************************
572 * Driver code
573 ***************************************************************************/
574
576 {
578 }
579
582 {
583 uint64_t t0, t1;
585 unsigned outlen = 0, outcrc = 0;
587 double mtime, ttime = 0, ttime2 = 0, stime;
589
592 return;
593 if (!sscanf(impl->
output,
"crc:%x", &outcrc)) {
594 outlen = strlen(impl->
output) / 2;
595 for (
i = 0;
i < outlen;
i++) {
598 }
599 }
600 for (
i = 0;
i < 8;
i++)
/* heat caches */
602 for (
i = 0;
i < nruns;
i++) {
603 memset(
output, 0,
size);
/* avoid leftovers from previous runs */
607 if (outlen ? memcmp(
output, outref, outlen) :
609 fprintf(stderr, "Expected: ");
610 if (outlen)
611 for (j = 0; j < outlen; j++)
612 fprintf(stderr,
"%02x",
output[j]);
613 else
615 fprintf(stderr, "\n");
617 }
619 ttime += mtime;
620 ttime2 += mtime * mtime;
621 }
622
623 ttime /= nruns;
624 ttime2 /= nruns;
625 stime = sqrt(ttime2 - ttime * ttime);
626 printf(
"%-10s %-12s size: %7d runs: %6d time: %8.3f +- %.3f\n",
628 fflush(stdout);
629 }
630
631 #define IMPL_USE(lib, name, symbol, output) \
632 { #lib, name, run_ ## lib ## _ ## symbol, output },
633 #define IMPL(lib, ...) IMPL_USE_ ## lib(lib, __VA_ARGS__)
634 #define IMPL_ALL(...) \
635 IMPL(lavu, __VA_ARGS__) \
636 IMPL(crypto, __VA_ARGS__) \
637 IMPL(gcrypt, __VA_ARGS__) \
638 IMPL(mbedcrypto, __VA_ARGS__) \
639 IMPL(tomcrypt, __VA_ARGS__)
640
642 IMPL_ALL(
"MD5",
md5,
"aa26ff5b895356bcffd9292ba9f89e66")
643 IMPL_ALL(
"SHA-1", sha1,
"1fd8bd1fa02f5b0fe916b0d71750726b096c5744")
644 IMPL_ALL(
"SHA-256", sha256,
"14028ac673b3087e51a1d407fbf0df4deeec8f217119e13b07bf2138f93db8c5")
645 IMPL_ALL(
"SHA-512", sha512,
"3afdd44a80d99af15c87bd724cb717243193767835ce866dd5d58c02d674bb57"
646 "7c25b9e118c200a189fcd5a01ef106a4e200061f3e97dbf50ba065745fd46bef")
647 IMPL(lavu,
"RIPEMD-128", ripemd128,
"9ab8bfba2ddccc5d99c9d4cdfb844a5f")
648 IMPL(tomcrypt,
"RIPEMD-128", ripemd128,
"9ab8bfba2ddccc5d99c9d4cdfb844a5f")
649 IMPL_ALL(
"RIPEMD-160", ripemd160,
"62a5321e4fc8784903bb43ab7752c75f8b25af00")
650 IMPL_ALL(
"AES-128", aes128,
"crc:ff6bc888")
651 IMPL_ALL(
"CAMELLIA", camellia,
"crc:7abb59a7")
652 IMPL(lavu,
"CAST-128", cast128,
"crc:456aa584")
653 IMPL(crypto,
"CAST-128", cast128,
"crc:456aa584")
654 IMPL(gcrypt,
"CAST-128", cast128,
"crc:456aa584")
655 IMPL(tomcrypt,
"CAST-128", cast128,
"crc:456aa584")
656 IMPL_ALL(
"BLOWFISH", blowfish,
"crc:33e8aa74")
657 IMPL_ALL(
"DES", des,
"crc:31291e0b")
658 IMPL(lavu,
"TWOFISH", twofish,
"crc:9edbd5c1")
659 IMPL(gcrypt,
"TWOFISH", twofish,
"crc:9edbd5c1")
660 IMPL(tomcrypt,
"TWOFISH", twofish,
"crc:9edbd5c1")
661 IMPL_ALL(
"RC4", rc4,
"crc:538d37b2")
662 IMPL(lavu,
"XTEA", xtea,
"crc:931fc270")
663 IMPL(mbedcrypto,
"XTEA", xtea,
"crc:931fc270")
664 IMPL(tomcrypt,
"XTEA", xtea,
"crc:931fc270")
665 };
666
667 int main(
int argc,
char **argv)
668 {
671 unsigned i, impl,
size;
672 int opt;
673
674 while ((opt =
getopt(argc, argv,
"hl:a:r:")) != -1) {
675 switch (opt) {
676 case 'l':
678 break;
679 case 'a':
681 break;
682 case 'r':
684 break;
685 case 'h':
686 default:
687 fprintf(stderr, "Usage: %s [-l libs] [-a algos] [-r runs]\n",
688 argv[0]);
689 if ((USE_EXT_LIBS)) {
690 char buf[1024];
691 snprintf(buf,
sizeof(buf),
"%s%s%s%s",
692 ((USE_EXT_LIBS) &
USE_crypto) ?
"+crypto" :
"",
693 ((USE_EXT_LIBS) &
USE_gcrypt) ?
"+gcrypt" :
"",
696 fprintf(stderr, "Built with the following external libraries:\n"
697 "make VERSUS=%s\n", buf + 1);
698 } else {
699 fprintf(stderr, "Built without external libraries; use\n"
700 "make VERSUS=crypto+gcrypt+mbedcrypto+tomcrypt tools/crypto_bench\n"
701 "to enable them.\n");
702 }
703 exit(opt != 'h');
704 }
705 }
711
713
717
719
720 return 0;
721 }