1 /*
2  * QEMU Crypto hash 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/hash.h"
17 
18 typedef struct QCryptoHashOpts {
19     size_t chunk_size;
20     QCryptoHashAlgorithm alg;
21 } QCryptoHashOpts;
22 
23 static void test_hash_speed(const void *opaque)
24 {
25     const QCryptoHashOpts *opts = opaque;
26     uint8_t *in = NULL, *out = NULL;
27     size_t out_len = 0;
28     const size_t total = 2 * GiB;
29     size_t remain;
30     struct iovec iov;
31     int ret;
32 
33     in = g_new0(uint8_t, opts->chunk_size);
34     memset(in, g_test_rand_int(), opts->chunk_size);
35 
36     iov.iov_base = (char *)in;
37     iov.iov_len = opts->chunk_size;
38 
39     g_test_timer_start();
40     remain = total;
41     while (remain) {
42         ret = qcrypto_hash_bytesv(opts->alg,
43                                   &iov, 1, &out, &out_len,
44                                   NULL);
45         g_assert(ret == 0);
46 
47         remain -= opts->chunk_size;
48     }
49     g_test_timer_elapsed();
50 
51     g_test_message("hash(%s): chunk %zu bytes %.2f MB/sec",
52                    QCryptoHashAlgorithm_str(opts->alg),
53                    opts->chunk_size, total / g_test_timer_last());
54 
55     g_free(out);
56     g_free(in);
57 }
58 
59 int main(int argc, char **argv)
60 {
61     char name[64];
62 
63     g_test_init(&argc, &argv, NULL);
64     g_assert(qcrypto_init(NULL) == 0);
65 
66 #define TEST_ONE(a, c)                                          \
67     QCryptoHashOpts opts ## a ## c = {                          \
68         .alg = QCRYPTO_HASH_ALG_ ## a, .chunk_size = c,         \
69     };                                                          \
70     memset(name, 0 , sizeof(name));                             \
71     snprintf(name, sizeof(name),                                \
72              "/crypto/benchmark/hash/%s/bufsize-%d",            \
73              QCryptoHashAlgorithm_str(QCRYPTO_HASH_ALG_ ## a),  \
74              c);                                                \
75     if (qcrypto_hash_supports(QCRYPTO_HASH_ALG_ ## a))          \
76         g_test_add_data_func(name,                              \
77                              &opts ## a ## c,                   \
78                              test_hash_speed);
79 
80     TEST_ONE(MD5, 512);
81     TEST_ONE(MD5, 1024);
82     TEST_ONE(MD5, 4096);
83     TEST_ONE(MD5, 16384);
84 
85     TEST_ONE(SHA1, 512);
86     TEST_ONE(SHA1, 1024);
87     TEST_ONE(SHA1, 4096);
88     TEST_ONE(SHA1, 16384);
89 
90     TEST_ONE(SHA224, 512);
91     TEST_ONE(SHA224, 1024);
92     TEST_ONE(SHA224, 4096);
93     TEST_ONE(SHA224, 16384);
94 
95     TEST_ONE(SHA384, 512);
96     TEST_ONE(SHA384, 1024);
97     TEST_ONE(SHA384, 4096);
98     TEST_ONE(SHA384, 16384);
99 
100     TEST_ONE(SHA256, 512);
101     TEST_ONE(SHA256, 1024);
102     TEST_ONE(SHA256, 4096);
103     TEST_ONE(SHA256, 16384);
104 
105     TEST_ONE(SHA512, 512);
106     TEST_ONE(SHA512, 1024);
107     TEST_ONE(SHA512, 4096);
108     TEST_ONE(SHA512, 16384);
109 
110     TEST_ONE(RIPEMD160, 512);
111     TEST_ONE(RIPEMD160, 1024);
112     TEST_ONE(RIPEMD160, 4096);
113     TEST_ONE(RIPEMD160, 16384);
114 
115     return g_test_run();
116 }
117