Updates following review of SSL_stateless() code
[openssl.git] / test / hmactest.c
1 /*
2  * Copyright 1995-2017 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (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     unsigned char key[16];
31     int key_len;
32     unsigned char data[64];
33     int data_len;
34     unsigned char *digest;
35 } test[8] = {
36     {
37         "", 0, "More text test vectors to stuff up EBCDIC machines :-)", 54,
38         (unsigned char *)"e9139d1e6ee064ef8cf514fc7dc83e86",
39     },
40     {
41         {
42             0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b, 0x0b,
43             0x0b, 0x0b, 0x0b, 0x0b, 0x0b,
44         }, 16, "Hi There", 8,
45         (unsigned char *)"9294727a3638bb1c13f48ef8158bfc9d",
46     },
47     {
48         "Jefe", 4, "what do ya want for nothing?", 28,
49         (unsigned char *)"750c783e6ab0b503eaa86e310a5db738",
50     },
51     {
52         {
53             0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
54             0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
55         }, 16, {
56             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
57             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
58             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
59             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd,
60             0xdd, 0xdd, 0xdd, 0xdd, 0xdd, 0xdd
61         }, 50, (unsigned char *)"56be34521d144c88dbb8c733f0e8b3f6",
62     },
63     {
64         "", 0, "My test data", 12,
65         (unsigned char *)"61afdecb95429ef494d61fdee15990cabf0826fc"
66     },
67     {
68         "", 0, "My test data", 12,
69         (unsigned char *)"2274b195d90ce8e03406f4b526a47e0787a88a65479938f1a5baa3ce0f079776"
70     },
71     {
72         "123456", 6, "My test data", 12,
73         (unsigned char *)"bab53058ae861a7f191abe2d0145cbb123776a6369ee3f9d79ce455667e411dd"
74     },
75     {
76         "12345", 5, "My test data again", 18,
77         (unsigned char *)"a12396ceddd2a85f4c656bc1e0aa50c78cffde3e"
78     }
79 };
80 # endif
81
82 static char *pt(unsigned char *md, unsigned int len);
83
84
85 # ifndef OPENSSL_NO_MD5
86 static int test_hmac_md5(int idx)
87 {
88     char *p;
89 #  ifdef CHARSET_EBCDIC
90     ebcdic2ascii(test[0].data, test[0].data, test[0].data_len);
91     ebcdic2ascii(test[1].data, test[1].data, test[1].data_len);
92     ebcdic2ascii(test[2].key, test[2].key, test[2].key_len);
93     ebcdic2ascii(test[2].data, test[2].data, test[2].data_len);
94 #  endif
95
96     p = pt(HMAC(EVP_md5(),
97                 test[idx].key, test[idx].key_len,
98                 test[idx].data, test[idx].data_len, NULL, NULL),
99                 MD5_DIGEST_LENGTH);
100
101     if (!TEST_str_eq(p, (char *)test[idx].digest))
102       return 0;
103
104     return 1;
105 }
106 # endif
107
108 static int test_hmac_bad(void)
109 {
110     HMAC_CTX *ctx = NULL;
111     int ret = 0;
112
113     ctx = HMAC_CTX_new();
114     if (!TEST_ptr(ctx)
115         || !TEST_ptr_null(HMAC_CTX_get_md(ctx))
116         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, NULL, NULL))
117         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len))
118         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, EVP_sha1(), NULL))
119         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len)))
120         goto err;
121
122     ret = 1;
123 err:
124     HMAC_CTX_free(ctx);
125     return ret;
126 }
127
128 static int test_hmac_run(void)
129 {
130     char *p;
131     HMAC_CTX *ctx = NULL;
132     unsigned char buf[EVP_MAX_MD_SIZE];
133     unsigned int len;
134     int ret = 0;
135
136     ctx = HMAC_CTX_new();
137     HMAC_CTX_reset(ctx);
138
139     if (!TEST_ptr(ctx)
140         || !TEST_ptr_null(HMAC_CTX_get_md(ctx))
141         || !TEST_false(HMAC_Init_ex(ctx, NULL, 0, NULL, NULL))
142         || !TEST_false(HMAC_Update(ctx, test[4].data, test[4].data_len))
143         || !TEST_false(HMAC_Init_ex(ctx, test[4].key, -1, EVP_sha1(), NULL)))
144         goto err;
145
146     if (!TEST_true(HMAC_Init_ex(ctx, test[4].key, test[4].key_len, EVP_sha1(), NULL))
147         || !TEST_true(HMAC_Update(ctx, test[4].data, test[4].data_len))
148         || !TEST_true(HMAC_Final(ctx, buf, &len)))
149         goto err;
150
151     p = pt(buf, len);
152     if (!TEST_str_eq(p, (char *)test[4].digest))
153         goto err;
154
155     if (!TEST_false(HMAC_Init_ex(ctx, NULL, 0, EVP_sha256(), NULL)))
156         goto err;
157
158     if (!TEST_true(HMAC_Init_ex(ctx, test[5].key, test[5].key_len, EVP_sha256(), NULL))
159         || !TEST_ptr_eq(HMAC_CTX_get_md(ctx), EVP_sha256())
160         || !TEST_true(HMAC_Update(ctx, test[5].data, test[5].data_len))
161         || !TEST_true(HMAC_Final(ctx, buf, &len)))
162         goto err;
163
164     p = pt(buf, len);
165     if (!TEST_str_eq(p, (char *)test[5].digest))
166         goto err;
167
168     if (!TEST_true(HMAC_Init_ex(ctx, test[6].key, test[6].key_len, NULL, NULL))
169         || !TEST_true(HMAC_Update(ctx, test[6].data, test[6].data_len))
170         || !TEST_true(HMAC_Final(ctx, buf, &len)))
171         goto err;
172     p = pt(buf, len);
173     if (!TEST_str_eq(p, (char *)test[6].digest))
174         goto err;
175
176     ret = 1;
177 err:
178     HMAC_CTX_free(ctx);
179     return ret;
180 }
181
182
183 static int test_hmac_single_shot(void)
184 {
185     char *p;
186
187     /* Test single-shot with an empty key. */
188     p = pt(HMAC(EVP_sha1(), NULL, 0, test[4].data, test[4].data_len,
189                 NULL, NULL), SHA_DIGEST_LENGTH);
190     if (!TEST_str_eq(p, (char *)test[4].digest))
191         return 0;
192
193     return 1;
194 }
195
196
197 static int test_hmac_copy(void)
198 {
199     char *p;
200     HMAC_CTX *ctx = NULL, *ctx2 = NULL;
201     unsigned char buf[EVP_MAX_MD_SIZE];
202     unsigned int len;
203     int ret = 0;
204
205     ctx = HMAC_CTX_new();
206     ctx2 = HMAC_CTX_new();
207     if (!TEST_ptr(ctx) || !TEST_ptr(ctx2))
208         goto err;
209
210     if (!TEST_true(HMAC_Init_ex(ctx, test[7].key, test[7].key_len, EVP_sha1(), NULL))
211         || !TEST_true(HMAC_Update(ctx, test[7].data, test[7].data_len))
212         || !TEST_true(HMAC_CTX_copy(ctx2, ctx))
213         || !TEST_true(HMAC_Final(ctx2, buf, &len)))
214         goto err;
215
216     p = pt(buf, len);
217     if (!TEST_str_eq(p, (char *)test[7].digest))
218         goto err;
219
220     ret = 1;
221 err:
222     HMAC_CTX_free(ctx2);
223     HMAC_CTX_free(ctx);
224     return ret;
225 }
226
227 # ifndef OPENSSL_NO_MD5
228 static char *pt(unsigned char *md, unsigned int len)
229 {
230     unsigned int i;
231     static char buf[80];
232
233     for (i = 0; i < len; i++)
234         sprintf(&(buf[i * 2]), "%02x", md[i]);
235     return buf;
236 }
237 # endif
238
239 int setup_tests(void)
240 {
241     ADD_ALL_TESTS(test_hmac_md5, 4);
242     ADD_TEST(test_hmac_single_shot);
243     ADD_TEST(test_hmac_bad);
244     ADD_TEST(test_hmac_run);
245     ADD_TEST(test_hmac_copy);
246     return 1;
247 }
248