If memory debugging enabled return error on leaks.
[oweals/openssl.git] / test / p5_crpt2_test.c
1 /* Written by Christian Heimes, 2013 */
2 /*
3  * Copyright (c) 2013 The OpenSSL Project.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  *
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in
14  *    the documentation and/or other materials provided with the
15  *    distribution.
16  *
17  * 3. All advertising materials mentioning features or use of this
18  *    software must display the following acknowledgment:
19  *    "This product includes software developed by the OpenSSL Project
20  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
21  *
22  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
23  *    endorse or promote products derived from this software without
24  *    prior written permission. For written permission, please contact
25  *    openssl-core@openssl.org.
26  *
27  * 5. Products derived from this software may not be called "OpenSSL"
28  *    nor may "OpenSSL" appear in their names without prior written
29  *    permission of the OpenSSL Project.
30  *
31  * 6. Redistributions of any form whatsoever must retain the following
32  *    acknowledgment:
33  *    "This product includes software developed by the OpenSSL Project
34  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
35  *
36  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
37  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
39  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
40  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
41  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
42  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
43  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
44  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
45  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
46  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
47  * OF THE POSSIBILITY OF SUCH DAMAGE.
48  */
49
50 #include <stdio.h>
51 #include <string.h>
52
53 #include "../e_os.h"
54
55 #include <openssl/opensslconf.h>
56 #include <openssl/evp.h>
57 #ifndef OPENSSL_NO_ENGINE
58 # include <openssl/engine.h>
59 #endif
60 #include <openssl/err.h>
61 #include <openssl/conf.h>
62
63 typedef struct {
64     const char *pass;
65     int passlen;
66     const char *salt;
67     int saltlen;
68     int iter;
69 } testdata;
70
71 static testdata test_cases[] = {
72     {"password", 8, "salt", 4, 1},
73     {"password", 8, "salt", 4, 2},
74     {"password", 8, "salt", 4, 4096},
75     {"passwordPASSWORDpassword", 24,
76      "saltSALTsaltSALTsaltSALTsaltSALTsalt", 36, 4096},
77     {"pass\0word", 9, "sa\0lt", 5, 4096},
78     {NULL},
79 };
80
81 static const char *sha1_results[] = {
82     "0c60c80f961f0e71f3a9b524af6012062fe037a6",
83     "ea6c014dc72d6f8ccd1ed92ace1d41f0d8de8957",
84     "4b007901b765489abead49d926f721d065a429c1",
85     "3d2eec4fe41c849b80c8d83662c0e44a8b291a964cf2f07038",
86     "56fa6aa75548099dcc37d7f03425e0c3",
87 };
88
89 static const char *sha256_results[] = {
90     "120fb6cffcf8b32c43e7225256c4f837a86548c92ccc35480805987cb70be17b",
91     "ae4d0c95af6b46d32d0adff928f06dd02a303f8ef3c251dfd6e2d85a95474c43",
92     "c5e478d59288c841aa530db6845c4c8d962893a001ce4e11a4963873aa98134a",
93     "348c89dbcbd32b2f32d814b8116e84cf2b17347ebc1800181c4e2a1fb8dd53e1c63551"
94         "8c7dac47e9",
95     "89b69d0516f829893c696226650a8687",
96 };
97
98 static const char *sha512_results[] = {
99     "867f70cf1ade02cff3752599a3a53dc4af34c7a669815ae5d513554e1c8cf252c02d47"
100         "0a285a0501bad999bfe943c08f050235d7d68b1da55e63f73b60a57fce",
101     "e1d9c16aa681708a45f5c7c4e215ceb66e011a2e9f0040713f18aefdb866d53cf76cab"
102         "2868a39b9f7840edce4fef5a82be67335c77a6068e04112754f27ccf4e",
103     "d197b1b33db0143e018b12f3d1d1479e6cdebdcc97c5c0f87f6902e072f457b5143f30"
104         "602641b3d55cd335988cb36b84376060ecd532e039b742a239434af2d5",
105     "8c0511f4c6e597c6ac6315d8f0362e225f3c501495ba23b868c005174dc4ee71115b59"
106         "f9e60cd9532fa33e0f75aefe30225c583a186cd82bd4daea9724a3d3b8",
107     "9d9e9c4cd21fe4be24d5b8244c759665",
108 };
109
110 static void hexdump(FILE *f, const char *title, const unsigned char *s, int l)
111 {
112     int i;
113     fprintf(f, "%s", title);
114     for (i = 0; i < l; i++) {
115         fprintf(f, "%02x", s[i]);
116     }
117     fprintf(f, "\n");
118 }
119
120 static void convert(unsigned char *dst, const unsigned char *src, int len)
121 {
122     int i;
123     for (i = 0; i < len; i++, dst++, src += 2) {
124         unsigned int n;
125         sscanf((char *)src, "%2x", &n);
126         *dst = (unsigned char)n;
127     }
128     *dst = 0;
129 }
130
131 static void
132 test_p5_pbkdf2(int i, char *digestname, testdata *test, const char *hex)
133 {
134     const EVP_MD *digest;
135     unsigned char *out;
136     unsigned char *expected;
137     int keylen, r;
138
139     digest = EVP_get_digestbyname(digestname);
140     if (digest == NULL) {
141         fprintf(stderr, "unknown digest %s\n", digestname);
142         EXIT(5);
143     }
144
145     if ((strlen(hex) % 2) != 0) {
146         fprintf(stderr, "odd hex digest %s %i\n", digestname, i);
147         EXIT(5);
148     }
149     keylen = strlen(hex) / 2;
150     expected = OPENSSL_malloc(keylen + 1);
151     out = OPENSSL_malloc(keylen + 1);
152     if ((expected == NULL) || (out == NULL)) {
153         fprintf(stderr, "malloc() failed\n");
154         EXIT(5);
155     }
156     convert(expected, (const unsigned char *)hex, keylen);
157
158     r = PKCS5_PBKDF2_HMAC(test->pass, test->passlen,
159                           (const unsigned char *)test->salt, test->saltlen,
160                           test->iter, digest, keylen, out);
161
162     if (r == 0) {
163         fprintf(stderr, "PKCS5_PBKDF2_HMAC(%s) failure test %i\n",
164                 digestname, i);
165         EXIT(3);
166     }
167     if (memcmp(expected, out, keylen) != 0) {
168         fprintf(stderr, "Wrong result for PKCS5_PBKDF2_HMAC(%s) test %i\n",
169                 digestname, i);
170         hexdump(stderr, "expected: ", expected, keylen);
171         hexdump(stderr, "result:   ", out, keylen);
172         EXIT(2);
173     }
174     OPENSSL_free(expected);
175     OPENSSL_free(out);
176 }
177
178 int main(int argc, char **argv)
179 {
180     int i;
181     testdata *test = test_cases;
182
183     CRYPTO_set_mem_debug(1);
184     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
185
186     OpenSSL_add_all_digests();
187 # ifndef OPENSSL_NO_ENGINE
188     ENGINE_load_builtin_engines();
189     ENGINE_register_all_digests();
190 # endif
191
192     printf("PKCS5_PBKDF2_HMAC() tests ");
193     for (i = 0; test->pass != NULL; i++, test++) {
194         test_p5_pbkdf2(i, "sha1", test, sha1_results[i]);
195         test_p5_pbkdf2(i, "sha256", test, sha256_results[i]);
196         test_p5_pbkdf2(i, "sha512", test, sha512_results[i]);
197         printf(".");
198     }
199     printf(" done\n");
200
201 # ifndef OPENSSL_NO_ENGINE
202     ENGINE_cleanup();
203 # endif
204     EVP_cleanup();
205     CRYPTO_cleanup_all_ex_data();
206     ERR_remove_thread_state(NULL);
207     ERR_free_strings();
208 #ifndef OPENSSL_NO_CRYPTO_MDEBUG
209     if (CRYPTO_mem_leaks_fp(stderr) <= 0)
210         return 1;
211 # endif
212     return 0;
213 }