tests/crypto: Use the IEC binary prefix definitions
[qemu.git] / tests / benchmark-crypto-cipher.c
1 /*
2 * QEMU Crypto cipher speed benchmark
3 *
4 * Copyright (c) 2017 HUAWEI TECHNOLOGIES CO., LTD.
5 *
6 * Authors:
7 * Longpeng(Mike) <longpeng2@huawei.com>
8 *
9 * This work is licensed under the terms of the GNU GPL, version 2 or
10 * (at your option) any later version. See the COPYING file in the
11 * top-level directory.
12 */
13 #include "qemu/osdep.h"
14 #include "qemu/units.h"
15 #include "crypto/init.h"
16 #include "crypto/cipher.h"
17
18 static void test_cipher_speed(const void *opaque)
19 {
20 QCryptoCipher *cipher;
21 Error *err = NULL;
22 double total = 0.0;
23 size_t chunk_size = (size_t)opaque;
24 uint8_t *key = NULL, *iv = NULL;
25 uint8_t *plaintext = NULL, *ciphertext = NULL;
26 size_t nkey = qcrypto_cipher_get_key_len(QCRYPTO_CIPHER_ALG_AES_128);
27 size_t niv = qcrypto_cipher_get_iv_len(QCRYPTO_CIPHER_ALG_AES_128,
28 QCRYPTO_CIPHER_MODE_CBC);
29
30 key = g_new0(uint8_t, nkey);
31 memset(key, g_test_rand_int(), nkey);
32
33 iv = g_new0(uint8_t, niv);
34 memset(iv, g_test_rand_int(), niv);
35
36 ciphertext = g_new0(uint8_t, chunk_size);
37
38 plaintext = g_new0(uint8_t, chunk_size);
39 memset(plaintext, g_test_rand_int(), chunk_size);
40
41 cipher = qcrypto_cipher_new(QCRYPTO_CIPHER_ALG_AES_128,
42 QCRYPTO_CIPHER_MODE_CBC,
43 key, nkey, &err);
44 g_assert(cipher != NULL);
45
46 g_assert(qcrypto_cipher_setiv(cipher,
47 iv, niv,
48 &err) == 0);
49
50 g_test_timer_start();
51 do {
52 g_assert(qcrypto_cipher_encrypt(cipher,
53 plaintext,
54 ciphertext,
55 chunk_size,
56 &err) == 0);
57 total += chunk_size;
58 } while (g_test_timer_elapsed() < 5.0);
59
60 total /= MiB;
61 g_print("cbc(aes128): ");
62 g_print("Testing chunk_size %zu bytes ", chunk_size);
63 g_print("done: %.2f MB in %.2f secs: ", total, g_test_timer_last());
64 g_print("%.2f MB/sec\n", total / g_test_timer_last());
65
66 qcrypto_cipher_free(cipher);
67 g_free(plaintext);
68 g_free(ciphertext);
69 g_free(iv);
70 g_free(key);
71 }
72
73 int main(int argc, char **argv)
74 {
75 size_t i;
76 char name[64];
77
78 g_test_init(&argc, &argv, NULL);
79 g_assert(qcrypto_init(NULL) == 0);
80
81 for (i = 512; i <= 64 * KiB; i *= 2) {
82 memset(name, 0 , sizeof(name));
83 snprintf(name, sizeof(name), "/crypto/cipher/speed-%zu", i);
84 g_test_add_data_func(name, (void *)i, test_cipher_speed);
85 }
86
87 return g_test_run();
88 }