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
36
37 #ifndef AV_READ_TIME
38 #define AV_READ_TIME(x) 0
39 #endif
40
41 #if HAVE_UNISTD_H
42 #include <unistd.h> /* for getopt */
43 #endif
44 #if !HAVE_GETOPT
46 #endif
47
48 #define MAX_INPUT_SIZE 1048576
49 #define MAX_OUTPUT_SIZE 128
50
54
55 static const uint8_t *
hardcoded_key =
"FFmpeg is the best program ever.";
56
58 {
60 exit(1);
61 }
62
68 };
69
70 /***************************************************************************
71 * lavu: libavutil
72 ***************************************************************************/
73
86
87 #define IMPL_USE_lavu IMPL_USE
88
91 {
93 }
94
95 #define DEFINE_LAVU_MD(suffix, type, namespace, hsize) \
96 static void run_lavu_ ## suffix(uint8_t *output, \
97 const uint8_t *input, unsigned size) \
98 { \
99 static struct type *h; \
100 if (!h && !(h = av_ ## namespace ## _alloc())) \
101 fatal_error("out of memory"); \
102 av_ ## namespace ## _init(h, hsize); \
103 av_ ## namespace ## _update(h, input, size); \
104 av_ ## namespace ## _final(h, output); \
105 }
106
112
115 {
116 static struct AVAES *aes;
121 }
122
125 {
131 }
132
135 {
141 }
142
145 {
151 }
152
155 {
156 static struct AVDES *des;
161 }
162
165 {
171 }
172
175 {
176 static struct AVRC4 *rc4;
181 }
182
185 {
186 static struct AVXTEA *xtea;
191 }
192
193 /***************************************************************************
194 * crypto: OpenSSL's libcrypto
195 ***************************************************************************/
196
197 #if (USE_EXT_LIBS) & USE_crypto
198
199 #define OPENSSL_DISABLE_OLD_DES_SUPPORT
200 #include <openssl/md5.h>
201 #include <openssl/sha.h>
202 #include <openssl/ripemd.h>
203 #include <openssl/aes.h>
204 #include <openssl/blowfish.h>
205 #include <openssl/camellia.h>
206 #include <openssl/cast.h>
207 #include <openssl/des.h>
208 #include <openssl/rc4.h>
209
210 #define DEFINE_CRYPTO_WRAPPER(suffix, function) \
211 static void run_crypto_ ## suffix(uint8_t *output, \
212 const uint8_t *input, unsigned size) \
213 { \
214 function(input, size, output); \
215 }
216
217 DEFINE_CRYPTO_WRAPPER(
md5,
MD5)
218 DEFINE_CRYPTO_WRAPPER(sha1, SHA1)
219 DEFINE_CRYPTO_WRAPPER(sha256,
SHA256)
220 DEFINE_CRYPTO_WRAPPER(sha512,
SHA512)
221 DEFINE_CRYPTO_WRAPPER(ripemd160,
RIPEMD160)
222
223 static void run_crypto_aes128(uint8_t *
output,
225 {
226 AES_KEY aes;
228
233 }
234
235 static void run_crypto_blowfish(uint8_t *
output,
237 {
238 BF_KEY blowfish;
240
244 }
245
246 static void run_crypto_camellia(uint8_t *
output,
248 {
249 CAMELLIA_KEY camellia;
251
256 }
257
258 static void run_crypto_cast128(uint8_t *
output,
260 {
263
267 }
268
269 static void run_crypto_des(uint8_t *
output,
271 {
272 DES_key_schedule des;
274
278 }
279
280 static void run_crypto_rc4(uint8_t *
output,
282 {
283 RC4_KEY rc4;
284
287 }
288
289 #define IMPL_USE_crypto(...) IMPL_USE(__VA_ARGS__)
290 #else
291 #define IMPL_USE_crypto(...) /* ignore */
292 #endif
293
294 /***************************************************************************
295 * gcrypt: GnuTLS's libgcrypt
296 ***************************************************************************/
297
298 #if (USE_EXT_LIBS) & USE_gcrypt
299
300 #include <gcrypt.h>
301
302 #define DEFINE_GCRYPT_WRAPPER(suffix, algo) \
303 static void run_gcrypt_ ## suffix(uint8_t *output, \
304 const uint8_t *input, unsigned size) \
305 { \
306 gcry_md_hash_buffer(GCRY_MD_ ## algo, output, input, size); \
307 }
308
309 DEFINE_GCRYPT_WRAPPER(
md5,
MD5)
310 DEFINE_GCRYPT_WRAPPER(sha1, SHA1)
311 DEFINE_GCRYPT_WRAPPER(sha256,
SHA256)
312 DEFINE_GCRYPT_WRAPPER(sha512,
SHA512)
313 DEFINE_GCRYPT_WRAPPER(ripemd160, RMD160)
314
315 #define DEFINE_GCRYPT_CYPHER_WRAPPER(suffix, cypher, mode, sz) \
316 static void run_gcrypt_ ## suffix(uint8_t *output, \
317 const uint8_t *input, unsigned size) \
318 { \
319 static gcry_cipher_hd_t suffix; \
320 if (!suffix) \
321 gcry_cipher_open(&suffix, GCRY_CIPHER_ ## cypher, GCRY_CIPHER_MODE_ ## mode, 0); \
322 gcry_cipher_setkey(suffix, hardcoded_key, sz); \
323 gcry_cipher_encrypt(suffix, output, size, input, size); \
324 }
325
326 DEFINE_GCRYPT_CYPHER_WRAPPER(aes128, AES128, ECB, 16)
327 DEFINE_GCRYPT_CYPHER_WRAPPER(blowfish, BLOWFISH, ECB, 16)
328 DEFINE_GCRYPT_CYPHER_WRAPPER(camellia, CAMELLIA128, ECB, 16)
329 DEFINE_GCRYPT_CYPHER_WRAPPER(cast128, CAST5, ECB, 16)
330 DEFINE_GCRYPT_CYPHER_WRAPPER(des, DES, ECB, 8)
331 DEFINE_GCRYPT_CYPHER_WRAPPER(twofish, TWOFISH128, ECB, 16)
332 DEFINE_GCRYPT_CYPHER_WRAPPER(rc4, ARCFOUR, STREAM, 16)
333
334 #define IMPL_USE_gcrypt(...) IMPL_USE(__VA_ARGS__)
335 #else
336 #define IMPL_USE_gcrypt(...) /* ignore */
337 #endif
338
339 /***************************************************************************
340 * mbedcrypto: mbed TLS
341 ***************************************************************************/
342
343 #if (USE_EXT_LIBS) & USE_mbedcrypto
344
345 #include <mbedtls/aes.h>
346 #include <mbedtls/arc4.h>
347 #include <mbedtls/blowfish.h>
348 #include <mbedtls/camellia.h>
349 #include <mbedtls/des.h>
350 #include <mbedtls/md5.h>
351 #include <mbedtls/ripemd160.h>
352 #include <mbedtls/sha1.h>
353 #include <mbedtls/sha256.h>
354 #include <mbedtls/sha512.h>
355 #include <mbedtls/xtea.h>
356
357 #define DEFINE_MBEDCRYPTO_WRAPPER(suffix) \
358 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
359 const uint8_t *input, unsigned size) \
360 { \
361 mbedtls_ ## suffix ## _ret(input, size, output); \
362 }
363
364 #define DEFINE_MBEDCRYPTO_WRAPPER_SHA2(suffix) \
365 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
366 const uint8_t *input, unsigned size) \
367 { \
368 mbedtls_ ## suffix ## _ret(input, size, output, 0); \
369 }
370
371 DEFINE_MBEDCRYPTO_WRAPPER(
md5)
372 DEFINE_MBEDCRYPTO_WRAPPER(ripemd160)
373 DEFINE_MBEDCRYPTO_WRAPPER(sha1)
374 DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha256)
375 DEFINE_MBEDCRYPTO_WRAPPER_SHA2(sha512)
376
377
378 #define DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(suffix, cypher, algo) \
379 static void run_mbedcrypto_ ## suffix(uint8_t *output, \
380 const uint8_t *input, unsigned size) \
381 { \
382 mbedtls_ ## cypher ## _context cypher; \
383 \
384 mbedtls_ ## cypher ## _init(&cypher); \
385 mbedtls_ ## cypher ## _setkey_enc(&cypher, hardcoded_key, 128); \
386 for (int i = 0; i < size; i += 16) \
387 mbedtls_ ## cypher ## _crypt_ecb(&cypher, MBEDTLS_ ## algo ## _ENCRYPT, \
388 input + i, output + i); \
389 mbedtls_ ## cypher ## _free(&cypher); \
390 }
391
392 DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(aes128, aes, AES)
393 DEFINE_MBEDCRYPTO_CYPHER_WRAPPER(camellia, camellia, CAMELLIA)
394
395 static void run_mbedcrypto_blowfish(uint8_t *
output,
397 {
398 mbedtls_blowfish_context blowfish;
399
400 mbedtls_blowfish_init(&blowfish);
402 for (
int i = 0;
i <
size;
i += 8)
403 mbedtls_blowfish_crypt_ecb(&blowfish, MBEDTLS_BLOWFISH_ENCRYPT,
405 mbedtls_blowfish_free(&blowfish);
406 }
407
408 static void run_mbedcrypto_des(uint8_t *
output,
410 {
411 mbedtls_des_context des;
412
413 mbedtls_des_init(&des);
415 for (
int i = 0;
i <
size;
i += 8)
417 mbedtls_des_free(&des);
418 }
419
420 static void run_mbedcrypto_rc4(uint8_t *
output,
422 {
423 mbedtls_arc4_context rc4;
424
425 mbedtls_arc4_init(&rc4);
428 mbedtls_arc4_free(&rc4);
429 }
430
431 static void run_mbedcrypto_xtea(uint8_t *
output,
433 {
434 mbedtls_xtea_context xtea;
435
436 mbedtls_xtea_init(&xtea);
438 for (
int i = 0;
i <
size;
i += 8)
439 mbedtls_xtea_crypt_ecb(&xtea, MBEDTLS_XTEA_ENCRYPT,
441 mbedtls_xtea_free(&xtea);
442 }
443
444 #define IMPL_USE_mbedcrypto(...) IMPL_USE(__VA_ARGS__)
445 #else
446 #define IMPL_USE_mbedcrypto(...) /* ignore */
447 #endif
448
449 /***************************************************************************
450 * tomcrypt: LibTomCrypt
451 ***************************************************************************/
452
453 #if (USE_EXT_LIBS) & USE_tomcrypt
454
455 #include <tomcrypt.h>
456
457 #define DEFINE_TOMCRYPT_WRAPPER(suffix, namespace, algo) \
458 static void run_tomcrypt_ ## suffix(uint8_t *output, \
459 const uint8_t *input, unsigned size) \
460 { \
461 hash_state md; \
462 namespace ## _init(&md); \
463 namespace ## _process(&md, input, size); \
464 namespace ## _done(&md, output); \
465 }
466
468 DEFINE_TOMCRYPT_WRAPPER(sha1, sha1, SHA1)
469 DEFINE_TOMCRYPT_WRAPPER(sha256, sha256,
SHA256)
470 DEFINE_TOMCRYPT_WRAPPER(sha512, sha512,
SHA512)
471 DEFINE_TOMCRYPT_WRAPPER(ripemd128, rmd128,
RIPEMD128)
472 DEFINE_TOMCRYPT_WRAPPER(ripemd160, rmd160,
RIPEMD160)
473
474 static void run_tomcrypt_aes128(uint8_t *
output,
476 {
477 symmetric_key aes;
479
484 }
485
486 static void run_tomcrypt_blowfish(uint8_t *
output,
488 {
489 symmetric_key blowfish;
491
495 }
496
497 static void run_tomcrypt_camellia(uint8_t *
output,
499 {
500 symmetric_key camellia;
502
507 }
508
509 static void run_tomcrypt_cast128(uint8_t *
output,
511 {
514
518 }
519
520 static void run_tomcrypt_des(uint8_t *
output,
522 {
523 symmetric_key des;
525
529 }
530
531 static void run_tomcrypt_rc4(uint8_t *
output,
533 {
534 rc4_state rc4;
535
538 rc4_stream_done(&rc4);
539 }
540
541 static void run_tomcrypt_twofish(uint8_t *
output,
543 {
544 symmetric_key twofish;
546
551 }
552
553 static void run_tomcrypt_xtea(uint8_t *
output,
555 {
556 symmetric_key xtea;
558
562 }
563
564
565 #define IMPL_USE_tomcrypt(...) IMPL_USE(__VA_ARGS__)
566 #else
567 #define IMPL_USE_tomcrypt(...) /* ignore */
568 #endif
569
570 /***************************************************************************
571 * Driver code
572 ***************************************************************************/
573
575 {
577 }
578
581 {
584 unsigned outlen = 0, outcrc = 0;
586 double mtime, ttime = 0, ttime2 = 0, stime;
588
591 return;
592 if (!sscanf(impl->
output,
"crc:%x", &outcrc)) {
593 outlen = strlen(impl->
output) / 2;
594 for (
i = 0;
i < outlen;
i++) {
597 }
598 }
599 for (
i = 0;
i < 8;
i++)
/* heat caches */
601 for (
i = 0;
i < nruns;
i++) {
602 memset(
output, 0,
size);
/* avoid leftovers from previous runs */
606 if (outlen ? memcmp(
output, outref, outlen) :
608 fprintf(stderr, "Expected: ");
609 if (outlen)
610 for (j = 0; j < outlen; j++)
611 fprintf(stderr,
"%02x",
output[j]);
612 else
614 fprintf(stderr, "\n");
616 }
618 ttime += mtime;
619 ttime2 += mtime * mtime;
620 }
621
622 ttime /= nruns;
623 ttime2 /= nruns;
624 stime = sqrt(ttime2 - ttime * ttime);
625 printf(
"%-10s %-12s size: %7d runs: %6d time: %8.3f +- %.3f\n",
627 fflush(stdout);
628 }
629
630 #define IMPL_USE(lib, name, symbol, output) \
631 { #lib, name, run_ ## lib ## _ ## symbol, output },
632 #define IMPL(lib, ...) IMPL_USE_ ## lib(lib, __VA_ARGS__)
633 #define IMPL_ALL(...) \
634 IMPL(lavu, __VA_ARGS__) \
635 IMPL(crypto, __VA_ARGS__) \
636 IMPL(gcrypt, __VA_ARGS__) \
637 IMPL(mbedcrypto, __VA_ARGS__) \
638 IMPL(tomcrypt, __VA_ARGS__)
639
641 IMPL_ALL(
"MD5",
md5,
"aa26ff5b895356bcffd9292ba9f89e66")
642 IMPL_ALL(
"SHA-1", sha1,
"1fd8bd1fa02f5b0fe916b0d71750726b096c5744")
643 IMPL_ALL(
"SHA-256", sha256,
"14028ac673b3087e51a1d407fbf0df4deeec8f217119e13b07bf2138f93db8c5")
644 IMPL_ALL(
"SHA-512", sha512,
"3afdd44a80d99af15c87bd724cb717243193767835ce866dd5d58c02d674bb57"
645 "7c25b9e118c200a189fcd5a01ef106a4e200061f3e97dbf50ba065745fd46bef")
646 IMPL(lavu,
"RIPEMD-128", ripemd128,
"9ab8bfba2ddccc5d99c9d4cdfb844a5f")
647 IMPL(tomcrypt,
"RIPEMD-128", ripemd128,
"9ab8bfba2ddccc5d99c9d4cdfb844a5f")
648 IMPL_ALL(
"RIPEMD-160", ripemd160,
"62a5321e4fc8784903bb43ab7752c75f8b25af00")
649 IMPL_ALL(
"AES-128", aes128,
"crc:ff6bc888")
650 IMPL_ALL(
"CAMELLIA", camellia,
"crc:7abb59a7")
651 IMPL(lavu,
"CAST-128", cast128,
"crc:456aa584")
652 IMPL(crypto,
"CAST-128", cast128,
"crc:456aa584")
653 IMPL(gcrypt,
"CAST-128", cast128,
"crc:456aa584")
654 IMPL(tomcrypt,
"CAST-128", cast128,
"crc:456aa584")
655 IMPL_ALL(
"BLOWFISH", blowfish,
"crc:33e8aa74")
656 IMPL_ALL(
"DES", des,
"crc:31291e0b")
657 IMPL(lavu,
"TWOFISH", twofish,
"crc:9edbd5c1")
658 IMPL(gcrypt,
"TWOFISH", twofish,
"crc:9edbd5c1")
659 IMPL(tomcrypt,
"TWOFISH", twofish,
"crc:9edbd5c1")
660 IMPL_ALL(
"RC4", rc4,
"crc:538d37b2")
661 IMPL(lavu,
"XTEA", xtea,
"crc:931fc270")
662 IMPL(mbedcrypto,
"XTEA", xtea,
"crc:931fc270")
663 IMPL(tomcrypt,
"XTEA", xtea,
"crc:931fc270")
664 };
665
666 int main(
int argc,
char **argv)
667 {
670 unsigned i, impl,
size;
671 int opt;
672
673 while ((opt =
getopt(argc, argv,
"hl:a:r:")) != -1) {
674 switch (opt) {
675 case 'l':
677 break;
678 case 'a':
680 break;
681 case 'r':
683 break;
684 case 'h':
685 default:
686 fprintf(stderr, "Usage: %s [-l libs] [-a algos] [-r runs]\n",
687 argv[0]);
688 if ((USE_EXT_LIBS)) {
689 char buf[1024];
690 snprintf(buf,
sizeof(buf),
"%s%s%s%s",
691 ((USE_EXT_LIBS) &
USE_crypto) ?
"+crypto" :
"",
692 ((USE_EXT_LIBS) &
USE_gcrypt) ?
"+gcrypt" :
"",
695 fprintf(stderr, "Built with the following external libraries:\n"
696 "make VERSUS=%s\n", buf + 1);
697 } else {
698 fprintf(stderr, "Built without external libraries; use\n"
699 "make VERSUS=crypto+gcrypt+mbedcrypto+tomcrypt tools/crypto_bench\n"
700 "to enable them.\n");
701 }
702 exit(opt != 'h');
703 }
704 }
710
712
716
718
719 return 0;
720 }