Don't store an HMAC key for longer than we need
[openssl.git] / test / hmactest.c
1 /*
2  * Copyright 1995-2017 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the Apache License 2.0 (the "License").  You may not use
5  * this file except in compliance with the License.  You can obtain a copy
6  * in the file LICENSE in the source distribution or at
7  * https://www.openssl.org/source/license.html
8  */
9
10 #include <stdio.h>
11 #include <string.h>
12 #include <stdlib.h>
13
14 #include "internal/nelem.h"
15
16 # include <openssl/hmac.h>
17 # include <openssl/sha.h>
18 # ifndef OPENSSL_NO_MD5
19 #  include <openssl/md5.h>
20 # endif
21
22 # ifdef CHARSET_EBCDIC
23 #  include <openssl/ebcdic.h>
24 # endif
25
26 #include "testutil.h"
27
28 # ifndef OPENSSL_NO_MD5
29 static struct test_st {
30     const char key[16];
31     int key_len;
32     const unsigned char data[64];
33     int data_len;
34     const char *digest;
35 } test[8] = {
36     {
37         "", 0, "More text test vectors to stuff up EBCDIC machines :-)", 54,
38         "e9139d1e6ee064ef8cf514fc7dc83e86",
39     },
40     {
41         "\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b",
42         16, "Hi There", 8,
43         "9294727a3638bb1c13f48ef8158bfc9d",
44     },
45     {
46         "Jefe", 4, "what do ya want for nothing?", 28,
47         "750c783e6ab0b503eaa86e310a5db738",
48     },
49     {
50         "\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa\xaa",
51         16, {
52             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
53             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
54             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
55             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
56             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd
57         }, 50, "56be34521d144c88dbb8c733f0e8b3f6",
58     },
59     {
60         "", 0, "My test data", 12,
61         "61afdecb95429ef494d61fdee15990cabf0826fc"
62     },
63     {
64         "", 0, "My test data", 12,
65         "2274b195d90ce8e03406f4b526a47e0787a88a65479938f1a5baa3ce0f079776"
66     },
67     {
68         "123456", 6, "My test data", 12,
69         "bab53058ae861a7f191abe2d0145cbb123776a6369ee3f9d79ce455667e411dd"
70     },
71     {
72         "12345", 5, "My test data again", 18,
73         "a12396ceddd2a85f4c656bc1e0aa50c78cffde3e"
74     }
75 };
76 # endif
77
78 static char *pt(unsigned char *md, unsigned int len);
79
80
81 # ifndef OPENSSL_NO_MD5
82 static int test_hmac_md5(int idx)
83 {
84     char *p;
85 #  ifdef CHARSET_EBCDIC
86     ebcdic2ascii(test[0].data, test[0].data, test[0].data_len);
87     ebcdic2ascii(test[1].data, test[1].data, test[1].data_len);
88     ebcdic2ascii(test[2].key, test[2].key, test[2].key_len);
89     ebcdic2ascii(test[2].data, test[2].data, test[2].data_len);
90 #  endif
91
92     p = pt(HMAC(EVP_md5(),
93                 test[idx].key, test[idx].key_len,
94                 test[idx].data, test[idx].data_len, NULL, NULL),
95                 MD5_DIGEST_LENGTH);
96
97     if (!TEST_str_eq(p, test[idx].digest))
98       return 0;
99
100     return 1;
101 }
102 # endif
103
104 static int test_hmac_bad(void)
105 {
106     HMAC_CTX *ctx = NULL;
107     int ret = 0;
108
109     ctx = HMAC_CTX_new();
110     if (!TEST_ptr(ctx)
111         || !TEST_ptr_null(HMAC_CTX_get_md(ctx))
112         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, NULL, NULL))
113         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len))
114         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, EVP_sha1(), NULL))
115         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len)))
116         goto err;
117
118     ret = 1;
119 err:
120     HMAC_CTX_free(ctx);
121     return ret;
122 }
123
124 static int test_hmac_run(void)
125 {
126     char *p;
127     HMAC_CTX *ctx = NULL;
128     unsigned char buf[EVP_MAX_MD_SIZE];
129     unsigned int len;
130     int ret = 0;
131
132     ctx = HMAC_CTX_new();
133     HMAC_CTX_reset(ctx);
134
135     if (!TEST_ptr(ctx)
136         || !TEST_ptr_null(HMAC_CTX_get_md(ctx))
137         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, NULL, NULL))
138         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len))
139         || !TEST_false(HMAC_Init_ex(ctx, test[4].key, -1, EVP_sha1(), NULL)))
140         goto err;
141
142     if (!TEST_true(HMAC_Init_ex(ctx, test[4].key, test[4].key_len, EVP_sha1(), NULL))
143         || !TEST_true(HMAC_Update(ctx, test[4].data, test[4].data_len))
144         || !TEST_true(HMAC_Final(ctx, buf, &len)))
145         goto err;
146
147     p = pt(buf, len);
148     if (!TEST_str_eq(p, test[4].digest))
149         goto err;
150
151     if (!TEST_false(HMAC_Init_ex(ctx, NULL, 0, EVP_sha256(), NULL)))
152         goto err;
153
154     if (!TEST_true(HMAC_Init_ex(ctx, test[5].key, test[5].key_len, EVP_sha256(), NULL))
155         || !TEST_ptr_eq(HMAC_CTX_get_md(ctx), EVP_sha256())
156         || !TEST_true(HMAC_Update(ctx, test[5].data, test[5].data_len))
157         || !TEST_true(HMAC_Final(ctx, buf, &len)))
158         goto err;
159
160     p = pt(buf, len);
161     if (!TEST_str_eq(p, test[5].digest))
162         goto err;
163
164     if (!TEST_true(HMAC_Init_ex(ctx, test[6].key, test[6].key_len, NULL, NULL))
165         || !TEST_true(HMAC_Update(ctx, test[6].data, test[6].data_len))
166         || !TEST_true(HMAC_Final(ctx, buf, &len)))
167         goto err;
168     p = pt(buf, len);
169     if (!TEST_str_eq(p, test[6].digest))
170         goto err;
171
172     /* Test reusing a key */
173     if (!TEST_true(HMAC_Init_ex(ctx, NULL, 0, NULL, NULL))
174         || !TEST_true(HMAC_Update(ctx, test[6].data, test[6].data_len))
175         || !TEST_true(HMAC_Final(ctx, buf, &len)))
176         goto err;
177     p = pt(buf, len);
178     if (!TEST_str_eq(p, test[6].digest))
179         goto err;
180
181     /*
182      * Test reusing a key where the digest is provided again but is the same as
183      * last time
184      */
185     if (!TEST_true(HMAC_Init_ex(ctx, NULL, 0, EVP_sha256(), NULL))
186         || !TEST_true(HMAC_Update(ctx, test[6].data, test[6].data_len))
187         || !TEST_true(HMAC_Final(ctx, buf, &len)))
188         goto err;
189     p = pt(buf, len);
190     if (!TEST_str_eq(p, test[6].digest))
191         goto err;
192
193     ret = 1;
194 err:
195     HMAC_CTX_free(ctx);
196     return ret;
197 }
198
199
200 static int test_hmac_single_shot(void)
201 {
202     char *p;
203
204     /* Test single-shot with an empty key. */
205     p = pt(HMAC(EVP_sha1(), NULL, 0, test[4].data, test[4].data_len,
206                 NULL, NULL), SHA_DIGEST_LENGTH);
207     if (!TEST_str_eq(p, test[4].digest))
208         return 0;
209
210     return 1;
211 }
212
213
214 static int test_hmac_copy(void)
215 {
216     char *p;
217     HMAC_CTX *ctx = NULL, *ctx2 = NULL;
218     unsigned char buf[EVP_MAX_MD_SIZE];
219     unsigned int len;
220     int ret = 0;
221
222     ctx = HMAC_CTX_new();
223     ctx2 = HMAC_CTX_new();
224     if (!TEST_ptr(ctx) || !TEST_ptr(ctx2))
225         goto err;
226
227     if (!TEST_true(HMAC_Init_ex(ctx, test[7].key, test[7].key_len, EVP_sha1(), NULL))
228         || !TEST_true(HMAC_Update(ctx, test[7].data, test[7].data_len))
229         || !TEST_true(HMAC_CTX_copy(ctx2, ctx))
230         || !TEST_true(HMAC_Final(ctx2, buf, &len)))
231         goto err;
232
233     p = pt(buf, len);
234     if (!TEST_str_eq(p, test[7].digest))
235         goto err;
236
237     ret = 1;
238 err:
239     HMAC_CTX_free(ctx2);
240     HMAC_CTX_free(ctx);
241     return ret;
242 }
243
244 # ifndef OPENSSL_NO_MD5
245 static char *pt(unsigned char *md, unsigned int len)
246 {
247     unsigned int i;
248     static char buf[80];
249
250     for (i = 0; i < len; i++)
251         sprintf(&(buf[i * 2]), "%02x", md[i]);
252     return buf;
253 }
254 # endif
255
256 int setup_tests(void)
257 {
258     ADD_ALL_TESTS(test_hmac_md5, 4);
259     ADD_TEST(test_hmac_single_shot);
260     ADD_TEST(test_hmac_bad);
261     ADD_TEST(test_hmac_run);
262     ADD_TEST(test_hmac_copy);
263     return 1;
264 }
265