@samitouri / QOSamiQemu
Code Commits Issues Pulls Sessions CI/CD
master
master
c 208 lines 6.18 KB
Raw
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(size_t chunk_size,
19 QCryptoCipherMode mode,
20 QCryptoCipherAlgo alg)
21 {
22 QCryptoCipher *cipher;
23 Error *err = NULL;
24 uint8_t *key = NULL, *iv = NULL;
25 uint8_t *plaintext = NULL, *ciphertext = NULL;
26 size_t nkey;
27 size_t niv;
28 const size_t total = 2 * GiB;
29 size_t remain;
30
31 if (!qcrypto_cipher_supports(alg, mode)) {
32 return;
33 }
34
35 nkey = qcrypto_cipher_get_key_len(alg);
36 niv = qcrypto_cipher_get_iv_len(alg, mode);
37 if (mode == QCRYPTO_CIPHER_MODE_XTS) {
38 nkey *= 2;
39 }
40
41 key = g_new0(uint8_t, nkey);
42 memset(key, g_test_rand_int(), nkey);
43
44 iv = g_new0(uint8_t, niv);
45 memset(iv, g_test_rand_int(), niv);
46
47 ciphertext = g_new0(uint8_t, chunk_size);
48
49 plaintext = g_new0(uint8_t, chunk_size);
50 memset(plaintext, g_test_rand_int(), chunk_size);
51
52 cipher = qcrypto_cipher_new(alg, mode,
53 key, nkey, &err);
54 g_assert(cipher != NULL);
55
56 if (mode != QCRYPTO_CIPHER_MODE_ECB)
57 g_assert(qcrypto_cipher_setiv(cipher,
58 iv, niv,
59 &err) == 0);
60
61 g_test_timer_start();
62 remain = total;
63 while (remain) {
64 g_assert(qcrypto_cipher_encrypt(cipher,
65 plaintext,
66 ciphertext,
67 chunk_size,
68 &err) == 0);
69 remain -= chunk_size;
70 }
71 g_test_timer_elapsed();
72
73 g_test_message("enc(%s-%s) chunk %zu bytes %.2f MB/sec ",
74 QCryptoCipherAlgo_str(alg),
75 QCryptoCipherMode_str(mode),
76 chunk_size, (double)total / MiB / g_test_timer_last());
77
78 g_test_timer_start();
79 remain = total;
80 while (remain) {
81 g_assert(qcrypto_cipher_decrypt(cipher,
82 plaintext,
83 ciphertext,
84 chunk_size,
85 &err) == 0);
86 remain -= chunk_size;
87 }
88 g_test_timer_elapsed();
89
90 g_test_message("dec(%s-%s) chunk %zu bytes %.2f MB/sec ",
91 QCryptoCipherAlgo_str(alg),
92 QCryptoCipherMode_str(mode),
93 chunk_size, (double)total / MiB / g_test_timer_last());
94
95 qcrypto_cipher_free(cipher);
96 g_free(plaintext);
97 g_free(ciphertext);
98 g_free(iv);
99 g_free(key);
100 }
101
102
103 static void test_cipher_speed_ecb_aes_128(const void *opaque)
104 {
105 size_t chunk_size = (size_t)opaque;
106 test_cipher_speed(chunk_size,
107 QCRYPTO_CIPHER_MODE_ECB,
108 QCRYPTO_CIPHER_ALGO_AES_128);
109 }
110
111 static void test_cipher_speed_ecb_aes_256(const void *opaque)
112 {
113 size_t chunk_size = (size_t)opaque;
114 test_cipher_speed(chunk_size,
115 QCRYPTO_CIPHER_MODE_ECB,
116 QCRYPTO_CIPHER_ALGO_AES_256);
117 }
118
119 static void test_cipher_speed_cbc_aes_128(const void *opaque)
120 {
121 size_t chunk_size = (size_t)opaque;
122 test_cipher_speed(chunk_size,
123 QCRYPTO_CIPHER_MODE_CBC,
124 QCRYPTO_CIPHER_ALGO_AES_128);
125 }
126
127 static void test_cipher_speed_cbc_aes_256(const void *opaque)
128 {
129 size_t chunk_size = (size_t)opaque;
130 test_cipher_speed(chunk_size,
131 QCRYPTO_CIPHER_MODE_CBC,
132 QCRYPTO_CIPHER_ALGO_AES_256);
133 }
134
135 static void test_cipher_speed_ctr_aes_128(const void *opaque)
136 {
137 size_t chunk_size = (size_t)opaque;
138 test_cipher_speed(chunk_size,
139 QCRYPTO_CIPHER_MODE_CTR,
140 QCRYPTO_CIPHER_ALGO_AES_128);
141 }
142
143 static void test_cipher_speed_ctr_aes_256(const void *opaque)
144 {
145 size_t chunk_size = (size_t)opaque;
146 test_cipher_speed(chunk_size,
147 QCRYPTO_CIPHER_MODE_CTR,
148 QCRYPTO_CIPHER_ALGO_AES_256);
149 }
150
151 static void test_cipher_speed_xts_aes_128(const void *opaque)
152 {
153 size_t chunk_size = (size_t)opaque;
154 test_cipher_speed(chunk_size,
155 QCRYPTO_CIPHER_MODE_XTS,
156 QCRYPTO_CIPHER_ALGO_AES_128);
157 }
158
159 static void test_cipher_speed_xts_aes_256(const void *opaque)
160 {
161 size_t chunk_size = (size_t)opaque;
162 test_cipher_speed(chunk_size,
163 QCRYPTO_CIPHER_MODE_XTS,
164 QCRYPTO_CIPHER_ALGO_AES_256);
165 }
166
167
168 int main(int argc, char **argv)
169 {
170 char *alg = NULL;
171 char *size = NULL;
172 g_test_init(&argc, &argv, NULL);
173 g_assert(qcrypto_init(NULL) == 0);
174
175 #define ADD_TEST(mode, cipher, keysize, chunk) \
176 if ((!alg || g_str_equal(alg, #mode)) && \
177 (!size || g_str_equal(size, #chunk))) \
178 g_test_add_data_func( \
179 "/crypto/cipher/" #mode "-" #cipher "-" #keysize "/chunk-" #chunk, \
180 (void *)chunk, \
181 test_cipher_speed_ ## mode ## _ ## cipher ## _ ## keysize)
182
183 if (argc >= 2) {
184 alg = argv[1];
185 }
186 if (argc >= 3) {
187 size = argv[2];
188 }
189
190 #define ADD_TESTS(chunk) \
191 do { \
192 ADD_TEST(ecb, aes, 128, chunk); \
193 ADD_TEST(ecb, aes, 256, chunk); \
194 ADD_TEST(cbc, aes, 128, chunk); \
195 ADD_TEST(cbc, aes, 256, chunk); \
196 ADD_TEST(ctr, aes, 128, chunk); \
197 ADD_TEST(ctr, aes, 256, chunk); \
198 ADD_TEST(xts, aes, 128, chunk); \
199 ADD_TEST(xts, aes, 256, chunk); \
200 } while (0)
201
202 ADD_TESTS(512);
203 ADD_TESTS(4096);
204 ADD_TESTS(16384);
205 ADD_TESTS(65536);
206
207 return g_test_run();
208 }