[mirror_qemu.git] / tests / benchmark-crypto-cipher.c

/*
 * QEMU Crypto cipher speed benchmark
 *
 * Copyright (c) 2017 HUAWEI TECHNOLOGIES CO., LTD.
 *
 * Authors:
 *    Longpeng(Mike) <longpeng2@huawei.com>
 *
 * This work is licensed under the terms of the GNU GPL, version 2 or
 * (at your option) any later version.  See the COPYING file in the
 * top-level directory.
 */
#include "qemu/osdep.h"
#include "qemu/units.h"
#include "crypto/init.h"
#include "crypto/cipher.h"

static void test_cipher_speed(size_t chunk_size,
                              QCryptoCipherMode mode,
                              QCryptoCipherAlgorithm alg)
{
    QCryptoCipher *cipher;
    Error *err = NULL;
    uint8_t *key = NULL, *iv = NULL;
    uint8_t *plaintext = NULL, *ciphertext = NULL;
    size_t nkey;
    size_t niv;
    const size_t total = 2 * GiB;
    size_t remain;

    if (!qcrypto_cipher_supports(alg, mode)) {
        return;
    }

    nkey = qcrypto_cipher_get_key_len(alg);
    niv = qcrypto_cipher_get_iv_len(alg, mode);
    if (mode == QCRYPTO_CIPHER_MODE_XTS) {
        nkey *= 2;
    }

    key = g_new0(uint8_t, nkey);
    memset(key, g_test_rand_int(), nkey);

    iv = g_new0(uint8_t, niv);
    memset(iv, g_test_rand_int(), niv);

    ciphertext = g_new0(uint8_t, chunk_size);

    plaintext = g_new0(uint8_t, chunk_size);
    memset(plaintext, g_test_rand_int(), chunk_size);

    cipher = qcrypto_cipher_new(alg, mode,
                                key, nkey, &err);
    g_assert(cipher != NULL);

    if (mode != QCRYPTO_CIPHER_MODE_ECB)
        g_assert(qcrypto_cipher_setiv(cipher,
                                      iv, niv,
                                      &err) == 0);

    g_test_timer_start();
    remain = total;
    while (remain) {
        g_assert(qcrypto_cipher_encrypt(cipher,
                                        plaintext,
                                        ciphertext,
                                        chunk_size,
                                        &err) == 0);
        remain -= chunk_size;
    }
    g_test_timer_elapsed();

    g_test_message("enc(%s-%s) chunk %zu bytes %.2f MB/sec ",
                   QCryptoCipherAlgorithm_str(alg),
                   QCryptoCipherMode_str(mode),
                   chunk_size, (double)total / MiB / g_test_timer_last());

    g_test_timer_start();
    remain = total;
    while (remain) {
        g_assert(qcrypto_cipher_decrypt(cipher,
                                        plaintext,
                                        ciphertext,
                                        chunk_size,
                                        &err) == 0);
        remain -= chunk_size;
    }
    g_test_timer_elapsed();

    g_test_message("dec(%s-%s) chunk %zu bytes %.2f MB/sec ",
                   QCryptoCipherAlgorithm_str(alg),
                   QCryptoCipherMode_str(mode),
                   chunk_size, (double)total / MiB / g_test_timer_last());

    qcrypto_cipher_free(cipher);
    g_free(plaintext);
    g_free(ciphertext);
    g_free(iv);
    g_free(key);
}


static void test_cipher_speed_ecb_aes_128(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_ECB,
                      QCRYPTO_CIPHER_ALG_AES_128);
}

static void test_cipher_speed_ecb_aes_256(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_ECB,
                      QCRYPTO_CIPHER_ALG_AES_256);
}

static void test_cipher_speed_cbc_aes_128(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_CBC,
                      QCRYPTO_CIPHER_ALG_AES_128);
}

static void test_cipher_speed_cbc_aes_256(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_CBC,
                      QCRYPTO_CIPHER_ALG_AES_256);
}

static void test_cipher_speed_ctr_aes_128(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_CTR,
                      QCRYPTO_CIPHER_ALG_AES_128);
}

static void test_cipher_speed_ctr_aes_256(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_CTR,
                      QCRYPTO_CIPHER_ALG_AES_256);
}

static void test_cipher_speed_xts_aes_128(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_XTS,
                      QCRYPTO_CIPHER_ALG_AES_128);
}

static void test_cipher_speed_xts_aes_256(const void *opaque)
{
    size_t chunk_size = (size_t)opaque;
    test_cipher_speed(chunk_size,
                      QCRYPTO_CIPHER_MODE_XTS,
                      QCRYPTO_CIPHER_ALG_AES_256);
}


int main(int argc, char **argv)
{
    char *alg = NULL;
    char *size = NULL;
    g_test_init(&argc, &argv, NULL);
    g_assert(qcrypto_init(NULL) == 0);

#define ADD_TEST(mode, cipher, keysize, chunk)                          \
    if ((!alg || g_str_equal(alg, #mode)) &&                            \
        (!size || g_str_equal(size, #chunk)))                           \
        g_test_add_data_func(                                           \
        "/crypto/cipher/" #mode "-" #cipher "-" #keysize "/chunk-" #chunk, \
        (void *)chunk,                                                  \
        test_cipher_speed_ ## mode ## _ ## cipher ## _ ## keysize)

    if (argc >= 2) {
        alg = argv[1];
    }
    if (argc >= 3) {
        size = argv[2];
    }

#define ADD_TESTS(chunk)                        \
    do {                                        \
        ADD_TEST(ecb, aes, 128, chunk);         \
        ADD_TEST(ecb, aes, 256, chunk);         \
        ADD_TEST(cbc, aes, 128, chunk);         \
        ADD_TEST(cbc, aes, 256, chunk);         \
        ADD_TEST(ctr, aes, 128, chunk);         \
        ADD_TEST(ctr, aes, 256, chunk);         \
        ADD_TEST(xts, aes, 128, chunk);         \
        ADD_TEST(xts, aes, 256, chunk);         \
    } while (0)

    ADD_TESTS(512);
    ADD_TESTS(4096);
    ADD_TESTS(16384);
    ADD_TESTS(65536);

    return g_test_run();
}
Commit	Line	Data
1efd9d5e LM	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"
68dbb6d0	14	#include "qemu/units.h"
1efd9d5e LM	15	#include "crypto/init.h"
	16	#include "crypto/cipher.h"
	17
a9e08155 DB	18	static void test_cipher_speed(size_t chunk_size,
	19	QCryptoCipherMode mode,
	20	QCryptoCipherAlgorithm alg)
1efd9d5e LM	21	{
	22	QCryptoCipher *cipher;
	23	Error *err = NULL;
1efd9d5e LM	24	uint8_t key = NULL, iv = NULL;
1efd9d5e LM	25	uint8_t plaintext = NULL, ciphertext = NULL;
a9e08155 DB	26	size_t nkey;
a9e08155 DB	27	size_t niv;
b4296d7f DB	28	const size_t total = 2 * GiB;
b4296d7f DB	29	size_t remain;
a9e08155 DB	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	}
1efd9d5e LM	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
a9e08155	52	cipher = qcrypto_cipher_new(alg, mode,
1efd9d5e LM	53	key, nkey, &err);
	54	g_assert(cipher != NULL);
	55
a9e08155 DB	56	if (mode != QCRYPTO_CIPHER_MODE_ECB)
	57	g_assert(qcrypto_cipher_setiv(cipher,
	58	iv, niv,
	59	&err) == 0);
1efd9d5e LM	60
1efd9d5e LM	61	g_test_timer_start();
b4296d7f DB	62	remain = total;
b4296d7f DB	63	while (remain) {
1efd9d5e LM	64	g_assert(qcrypto_cipher_encrypt(cipher,
	65	plaintext,
	66	ciphertext,
	67	chunk_size,
	68	&err) == 0);
b4296d7f DB	69	remain -= chunk_size;
	70	}
	71	g_test_timer_elapsed();
1efd9d5e	72
ca04c3cf DB	73	g_test_message("enc(%s-%s) chunk %zu bytes %.2f MB/sec ",
	74	QCryptoCipherAlgorithm_str(alg),
	75	QCryptoCipherMode_str(mode),
	76	chunk_size, (double)total / MiB / g_test_timer_last());
a9e08155	77
a9e08155	78	g_test_timer_start();
b4296d7f DB	79	remain = total;
b4296d7f DB	80	while (remain) {
a9e08155 DB	81	g_assert(qcrypto_cipher_decrypt(cipher,
	82	plaintext,
	83	ciphertext,
	84	chunk_size,
	85	&err) == 0);
b4296d7f DB	86	remain -= chunk_size;
	87	}
	88	g_test_timer_elapsed();
a9e08155	89
ca04c3cf DB	90	g_test_message("dec(%s-%s) chunk %zu bytes %.2f MB/sec ",
	91	QCryptoCipherAlgorithm_str(alg),
	92	QCryptoCipherMode_str(mode),
	93	chunk_size, (double)total / MiB / g_test_timer_last());
1efd9d5e LM	94
	95	qcrypto_cipher_free(cipher);
	96	g_free(plaintext);
	97	g_free(ciphertext);
	98	g_free(iv);
	99	g_free(key);
	100	}
	101
a9e08155 DB	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_ALG_AES_128);
	109	}
	110
	111	static void test_cipher_speed_ecb_aes_256(const void *opaque)
1efd9d5e	112	{
a9e08155 DB	113	size_t chunk_size = (size_t)opaque;
	114	test_cipher_speed(chunk_size,
	115	QCRYPTO_CIPHER_MODE_ECB,
	116	QCRYPTO_CIPHER_ALG_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_ALG_AES_128);
	125	}
1efd9d5e	126
a9e08155 DB	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_ALG_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_ALG_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_ALG_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_ALG_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_ALG_AES_256);
	165	}
	166
	167
	168	int main(int argc, char **argv)
	169	{
e2186a36 DB	170	char *alg = NULL;
e2186a36 DB	171	char *size = NULL;
1efd9d5e LM	172	g_test_init(&argc, &argv, NULL);
	173	g_assert(qcrypto_init(NULL) == 0);
	174
a9e08155	175	#define ADD_TEST(mode, cipher, keysize, chunk) \
e2186a36 DB	176	if ((!alg \|\| g_str_equal(alg, #mode)) && \
	177	(!size \|\| g_str_equal(size, #chunk))) \
	178	g_test_add_data_func( \
a9e08155 DB	179	"/crypto/cipher/" #mode "-" #cipher "-" #keysize "/chunk-" #chunk, \
	180	(void *)chunk, \
	181	test_cipher_speed_ ## mode ## _ ## cipher ## _ ## keysize)
	182
e2186a36 DB	183	if (argc >= 2) {
	184	alg = argv[1];
	185	}
	186	if (argc >= 3) {
	187	size = argv[2];
	188	}
	189
a9e08155 DB	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);
1efd9d5e LM	206
	207	return g_test_run();
	208	}