1
2
3
4
5
6
7
8
9
10
11
12
13
14
15#include "qemu/osdep.h"
16#include "crypto/init.h"
17#include "crypto/hmac.h"
18
19#define INPUT_TEXT1 "ABCDEFGHIJKLMNOPQRSTUVWXY"
20#define INPUT_TEXT2 "Zabcdefghijklmnopqrstuvwx"
21#define INPUT_TEXT3 "yz0123456789"
22#define INPUT_TEXT INPUT_TEXT1 \
23 INPUT_TEXT2 \
24 INPUT_TEXT3
25
26#define KEY "monkey monkey monkey monkey"
27
28typedef struct QCryptoHmacTestData QCryptoHmacTestData;
29struct QCryptoHmacTestData {
30 QCryptoHashAlgorithm alg;
31 const char *hex_digest;
32};
33
34static QCryptoHmacTestData test_data[] = {
35 {
36 .alg = QCRYPTO_HASH_ALG_MD5,
37 .hex_digest =
38 "ede9cb83679ba82d88fbeae865b3f8fc",
39 },
40 {
41 .alg = QCRYPTO_HASH_ALG_SHA1,
42 .hex_digest =
43 "c7b5a631e3aac975c4ededfcd346e469"
44 "dbc5f2d1",
45 },
46 {
47 .alg = QCRYPTO_HASH_ALG_SHA224,
48 .hex_digest =
49 "5f768179dbb29ca722875d0f461a2e2f"
50 "597d0210340a84df1a8e9c63",
51 },
52 {
53 .alg = QCRYPTO_HASH_ALG_SHA256,
54 .hex_digest =
55 "3798f363c57afa6edaffe39016ca7bad"
56 "efd1e670afb0e3987194307dec3197db",
57 },
58 {
59 .alg = QCRYPTO_HASH_ALG_SHA384,
60 .hex_digest =
61 "d218680a6032d33dccd9882d6a6a7164"
62 "64f26623be257a9b2919b185294f4a49"
63 "9e54b190bfd6bc5cedd2cd05c7e65e82",
64 },
65 {
66 .alg = QCRYPTO_HASH_ALG_SHA512,
67 .hex_digest =
68 "835a4f5b3750b4c1fccfa88da2f746a4"
69 "900160c9f18964309bb736c13b59491b"
70 "8e32d37b724cc5aebb0f554c6338a3b5"
71 "94c4ba26862b2dadb59b7ede1d08d53e",
72 },
73 {
74 .alg = QCRYPTO_HASH_ALG_RIPEMD160,
75 .hex_digest =
76 "94964ed4c1155b62b668c241d67279e5"
77 "8a711676",
78 },
79};
80
81static const char hex[] = "0123456789abcdef";
82
83static void test_hmac_alloc(void)
84{
85 size_t i;
86
87 for (i = 0; i < G_N_ELEMENTS(test_data); i++) {
88 QCryptoHmacTestData *data = &test_data[i];
89 QCryptoHmac *hmac = NULL;
90 uint8_t *result = NULL;
91 size_t resultlen = 0;
92 Error *err = NULL;
93 const char *exp_output = NULL;
94 int ret;
95 size_t j;
96
97 if (!qcrypto_hmac_supports(data->alg)) {
98 return;
99 }
100
101 exp_output = data->hex_digest;
102
103 hmac = qcrypto_hmac_new(data->alg, (const uint8_t *)KEY,
104 strlen(KEY), &err);
105 g_assert(err == NULL);
106 g_assert(hmac != NULL);
107
108 ret = qcrypto_hmac_bytes(hmac, (const char *)INPUT_TEXT,
109 strlen(INPUT_TEXT), &result,
110 &resultlen, &err);
111 g_assert(err == NULL);
112 g_assert(ret == 0);
113
114 for (j = 0; j < resultlen; j++) {
115 g_assert(exp_output[j * 2] == hex[(result[j] >> 4) & 0xf]);
116 g_assert(exp_output[j * 2 + 1] == hex[result[j] & 0xf]);
117 }
118
119 qcrypto_hmac_free(hmac);
120
121 g_free(result);
122 }
123}
124
125static void test_hmac_prealloc(void)
126{
127 size_t i;
128
129 for (i = 0; i < G_N_ELEMENTS(test_data); i++) {
130 QCryptoHmacTestData *data = &test_data[i];
131 QCryptoHmac *hmac = NULL;
132 uint8_t *result = NULL;
133 size_t resultlen = 0;
134 Error *err = NULL;
135 const char *exp_output = NULL;
136 int ret;
137 size_t j;
138
139 if (!qcrypto_hmac_supports(data->alg)) {
140 return;
141 }
142
143 exp_output = data->hex_digest;
144
145 resultlen = strlen(exp_output) / 2;
146 result = g_new0(uint8_t, resultlen);
147
148 hmac = qcrypto_hmac_new(data->alg, (const uint8_t *)KEY,
149 strlen(KEY), &err);
150 g_assert(err == NULL);
151 g_assert(hmac != NULL);
152
153 ret = qcrypto_hmac_bytes(hmac, (const char *)INPUT_TEXT,
154 strlen(INPUT_TEXT), &result,
155 &resultlen, &err);
156 g_assert(err == NULL);
157 g_assert(ret == 0);
158
159 exp_output = data->hex_digest;
160 for (j = 0; j < resultlen; j++) {
161 g_assert(exp_output[j * 2] == hex[(result[j] >> 4) & 0xf]);
162 g_assert(exp_output[j * 2 + 1] == hex[result[j] & 0xf]);
163 }
164
165 qcrypto_hmac_free(hmac);
166
167 g_free(result);
168 }
169}
170
171static void test_hmac_iov(void)
172{
173 size_t i;
174
175 for (i = 0; i < G_N_ELEMENTS(test_data); i++) {
176 QCryptoHmacTestData *data = &test_data[i];
177 QCryptoHmac *hmac = NULL;
178 uint8_t *result = NULL;
179 size_t resultlen = 0;
180 Error *err = NULL;
181 const char *exp_output = NULL;
182 int ret;
183 size_t j;
184 struct iovec iov[3] = {
185 { .iov_base = (char *)INPUT_TEXT1, .iov_len = strlen(INPUT_TEXT1) },
186 { .iov_base = (char *)INPUT_TEXT2, .iov_len = strlen(INPUT_TEXT2) },
187 { .iov_base = (char *)INPUT_TEXT3, .iov_len = strlen(INPUT_TEXT3) },
188 };
189
190 if (!qcrypto_hmac_supports(data->alg)) {
191 return;
192 }
193
194 exp_output = data->hex_digest;
195
196 hmac = qcrypto_hmac_new(data->alg, (const uint8_t *)KEY,
197 strlen(KEY), &err);
198 g_assert(err == NULL);
199 g_assert(hmac != NULL);
200
201 ret = qcrypto_hmac_bytesv(hmac, iov, 3, &result,
202 &resultlen, &err);
203 g_assert(err == NULL);
204 g_assert(ret == 0);
205
206 for (j = 0; j < resultlen; j++) {
207 g_assert(exp_output[j * 2] == hex[(result[j] >> 4) & 0xf]);
208 g_assert(exp_output[j * 2 + 1] == hex[result[j] & 0xf]);
209 }
210
211 qcrypto_hmac_free(hmac);
212
213 g_free(result);
214 }
215}
216
217static void test_hmac_digest(void)
218{
219 size_t i;
220
221 for (i = 0; i < G_N_ELEMENTS(test_data); i++) {
222 QCryptoHmacTestData *data = &test_data[i];
223 QCryptoHmac *hmac = NULL;
224 uint8_t *result = NULL;
225 Error *err = NULL;
226 const char *exp_output = NULL;
227 int ret;
228
229 if (!qcrypto_hmac_supports(data->alg)) {
230 return;
231 }
232
233 exp_output = data->hex_digest;
234
235 hmac = qcrypto_hmac_new(data->alg, (const uint8_t *)KEY,
236 strlen(KEY), &err);
237 g_assert(err == NULL);
238 g_assert(hmac != NULL);
239
240 ret = qcrypto_hmac_digest(hmac, (const char *)INPUT_TEXT,
241 strlen(INPUT_TEXT), (char **)&result,
242 &err);
243 g_assert(err == NULL);
244 g_assert(ret == 0);
245
246 g_assert_cmpstr((const char *)result, ==, exp_output);
247
248 qcrypto_hmac_free(hmac);
249
250 g_free(result);
251 }
252}
253
254int main(int argc, char **argv)
255{
256 g_test_init(&argc, &argv, NULL);
257
258 g_assert(qcrypto_init(NULL) == 0);
259
260 g_test_add_func("/crypto/hmac/iov", test_hmac_iov);
261 g_test_add_func("/crypto/hmac/alloc", test_hmac_alloc);
262 g_test_add_func("/crypto/hmac/prealloc", test_hmac_prealloc);
263 g_test_add_func("/crypto/hmac/digest", test_hmac_digest);
264
265 return g_test_run();
266}
267
