2 * Copyright 2000-2017 The OpenSSL Project Authors. All Rights Reserved.
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
13 #include <openssl/e_os2.h>
15 # include "testutil.h"
17 #ifndef OPENSSL_NO_ENGINE
18 # include <openssl/buffer.h>
19 # include <openssl/crypto.h>
20 # include <openssl/engine.h>
21 # include <openssl/rsa.h>
22 # include <openssl/err.h>
24 static void display_engine_list(void)
30 for (h
= ENGINE_get_first(); h
!= NULL
; h
= ENGINE_get_next(h
)) {
31 TEST_info("#%d: id = \"%s\", name = \"%s\"",
32 loop
++, ENGINE_get_id(h
), ENGINE_get_name(h
));
36 * ENGINE_get_first() increases the struct_ref counter, so we must call
37 * ENGINE_free() to decrease it again
44 static int test_engines(void)
46 ENGINE
*block
[NUMTOADD
];
48 char *ename
[NUMTOADD
];
53 ENGINE
*new_h1
= NULL
;
54 ENGINE
*new_h2
= NULL
;
55 ENGINE
*new_h3
= NULL
;
56 ENGINE
*new_h4
= NULL
;
58 memset(block
, 0, sizeof(block
));
59 if (!TEST_ptr(new_h1
= ENGINE_new())
60 || !TEST_true(ENGINE_set_id(new_h1
, "test_id0"))
61 || !TEST_true(ENGINE_set_name(new_h1
, "First test item"))
62 || !TEST_ptr(new_h2
= ENGINE_new())
63 || !TEST_true(ENGINE_set_id(new_h2
, "test_id1"))
64 || !TEST_true(ENGINE_set_name(new_h2
, "Second test item"))
65 || !TEST_ptr(new_h3
= ENGINE_new())
66 || !TEST_true(ENGINE_set_id(new_h3
, "test_id2"))
67 || !TEST_true(ENGINE_set_name(new_h3
, "Third test item"))
68 || !TEST_ptr(new_h4
= ENGINE_new())
69 || !TEST_true(ENGINE_set_id(new_h4
, "test_id3"))
70 || !TEST_true(ENGINE_set_name(new_h4
, "Fourth test item")))
72 TEST_info("Engines:");
73 display_engine_list();
75 if (!TEST_true(ENGINE_add(new_h1
)))
77 TEST_info("Engines:");
78 display_engine_list();
80 ptr
= ENGINE_get_first();
81 if (!TEST_true(ENGINE_remove(ptr
)))
84 TEST_info("Engines:");
85 display_engine_list();
87 if (!TEST_true(ENGINE_add(new_h3
))
88 || !TEST_true(ENGINE_add(new_h2
)))
90 TEST_info("Engines:");
91 display_engine_list();
93 if (!TEST_true(ENGINE_remove(new_h2
)))
95 TEST_info("Engines:");
96 display_engine_list();
98 if (!TEST_true(ENGINE_add(new_h4
)))
100 TEST_info("Engines:");
101 display_engine_list();
104 if (!TEST_false(ENGINE_add(new_h3
)))
109 if (!TEST_false(ENGINE_remove(new_h2
)))
113 if (!TEST_true(ENGINE_remove(new_h3
)))
115 TEST_info("Engines:");
116 display_engine_list();
118 if (!TEST_true(ENGINE_remove(new_h4
)))
120 TEST_info("Engines:");
121 display_engine_list();
124 * At this point, we should have an empty list, unless some hardware
125 * support engine got added. However, since we don't allow the config
126 * file to be loaded and don't otherwise load any built in engines,
127 * that is unlikely. Still, we check, if for nothing else, then to
128 * notify that something is a little off (and might mean that |new_h1|
129 * wasn't unloaded when it should have)
131 if ((ptr
= ENGINE_get_first()) != NULL
) {
132 if (!ENGINE_remove(ptr
))
133 TEST_info("Remove failed - probably no hardware support present");
136 TEST_info("Engines:");
137 display_engine_list();
139 if (!TEST_true(ENGINE_add(new_h1
))
140 || !TEST_true(ENGINE_remove(new_h1
)))
143 TEST_info("About to beef up the engine-type list");
144 for (loop
= 0; loop
< NUMTOADD
; loop
++) {
145 sprintf(buf
, "id%d", loop
);
146 eid
[loop
] = OPENSSL_strdup(buf
);
147 sprintf(buf
, "Fake engine type %d", loop
);
148 ename
[loop
] = OPENSSL_strdup(buf
);
149 if (!TEST_ptr(block
[loop
] = ENGINE_new())
150 || !TEST_true(ENGINE_set_id(block
[loop
], eid
[loop
]))
151 || !TEST_true(ENGINE_set_name(block
[loop
], ename
[loop
])))
154 for (loop
= 0; loop
< NUMTOADD
; loop
++) {
155 if (!TEST_true(ENGINE_add(block
[loop
]))) {
156 test_note("Adding stopped at %d, (%s,%s)",
157 loop
, ENGINE_get_id(block
[loop
]),
158 ENGINE_get_name(block
[loop
]));
163 TEST_info("About to empty the engine-type list");
164 while ((ptr
= ENGINE_get_first()) != NULL
) {
165 if (!TEST_true(ENGINE_remove(ptr
)))
169 for (loop
= 0; loop
< NUMTOADD
; loop
++) {
170 OPENSSL_free(eid
[loop
]);
171 OPENSSL_free(ename
[loop
]);
180 for (loop
= 0; loop
< NUMTOADD
; loop
++)
181 ENGINE_free(block
[loop
]);
185 /* Test EVP_PKEY method */
186 static EVP_PKEY_METHOD
*test_rsa
= NULL
;
188 static int called_encrypt
= 0;
190 /* Test function to check operation has been redirected */
191 static int test_encrypt(EVP_PKEY_CTX
*ctx
, unsigned char *sig
,
192 size_t *siglen
, const unsigned char *tbs
, size_t tbslen
)
198 static int test_pkey_meths(ENGINE
*e
, EVP_PKEY_METHOD
**pmeth
,
199 const int **pnids
, int nid
)
201 static const int rnid
= EVP_PKEY_RSA
;
207 if (nid
== EVP_PKEY_RSA
) {
216 /* Return a test EVP_PKEY value */
218 static EVP_PKEY
*get_test_pkey(void)
220 static unsigned char n
[] =
221 "\x00\xAA\x36\xAB\xCE\x88\xAC\xFD\xFF\x55\x52\x3C\x7F\xC4\x52\x3F"
222 "\x90\xEF\xA0\x0D\xF3\x77\x4A\x25\x9F\x2E\x62\xB4\xC5\xD9\x9C\xB5"
223 "\xAD\xB3\x00\xA0\x28\x5E\x53\x01\x93\x0E\x0C\x70\xFB\x68\x76\x93"
224 "\x9C\xE6\x16\xCE\x62\x4A\x11\xE0\x08\x6D\x34\x1E\xBC\xAC\xA0\xA1"
226 static unsigned char e
[] = "\x11";
228 RSA
*rsa
= RSA_new();
229 EVP_PKEY
*pk
= EVP_PKEY_new();
231 if (rsa
== NULL
|| pk
== NULL
|| !EVP_PKEY_assign_RSA(pk
, rsa
)) {
237 if (!RSA_set0_key(rsa
, BN_bin2bn(n
, sizeof(n
)-1, NULL
),
238 BN_bin2bn(e
, sizeof(e
)-1, NULL
), NULL
)) {
246 static int test_redirect(void)
248 const unsigned char pt
[] = "Hello World\n";
249 unsigned char *tmp
= NULL
;
251 EVP_PKEY_CTX
*ctx
= NULL
;
253 EVP_PKEY
*pkey
= NULL
;
257 if (!TEST_ptr(pkey
= get_test_pkey()))
260 len
= EVP_PKEY_size(pkey
);
261 if (!TEST_ptr(tmp
= OPENSSL_malloc(len
)))
264 if (!TEST_ptr(ctx
= EVP_PKEY_CTX_new(pkey
, NULL
)))
266 TEST_info("EVP_PKEY_encrypt test: no redirection");
267 /* Encrypt some data: should succeed but not be redirected */
268 if (!TEST_int_gt(EVP_PKEY_encrypt_init(ctx
), 0)
269 || !TEST_int_gt(EVP_PKEY_encrypt(ctx
, tmp
, &len
, pt
, sizeof(pt
)), 0)
270 || !TEST_false(called_encrypt
))
272 EVP_PKEY_CTX_free(ctx
);
275 /* Create a test ENGINE */
276 if (!TEST_ptr(e
= ENGINE_new())
277 || !TEST_true(ENGINE_set_id(e
, "Test redirect engine"))
278 || !TEST_true(ENGINE_set_name(e
, "Test redirect engine")))
282 * Try to create a context for this engine and test key.
283 * Try setting test key engine. Both should fail because the
284 * engine has no public key methods.
286 if (!TEST_ptr_null(ctx
= EVP_PKEY_CTX_new(pkey
, e
))
287 || !TEST_int_le(EVP_PKEY_set1_engine(pkey
, e
), 0))
290 /* Setup an empty test EVP_PKEY_METHOD and set callback to return it */
291 if (!TEST_ptr(test_rsa
= EVP_PKEY_meth_new(EVP_PKEY_RSA
, 0)))
293 ENGINE_set_pkey_meths(e
, test_pkey_meths
);
295 /* Getting a context for test ENGINE should now succeed */
296 if (!TEST_ptr(ctx
= EVP_PKEY_CTX_new(pkey
, e
)))
298 /* Encrypt should fail because operation is not supported */
299 if (!TEST_int_le(EVP_PKEY_encrypt_init(ctx
), 0))
301 EVP_PKEY_CTX_free(ctx
);
304 /* Add test encrypt operation to method */
305 EVP_PKEY_meth_set_encrypt(test_rsa
, 0, test_encrypt
);
307 TEST_info("EVP_PKEY_encrypt test: redirection via EVP_PKEY_CTX_new()");
308 if (!TEST_ptr(ctx
= EVP_PKEY_CTX_new(pkey
, e
)))
310 /* Encrypt some data: should succeed and be redirected */
311 if (!TEST_int_gt(EVP_PKEY_encrypt_init(ctx
), 0)
312 || !TEST_int_gt(EVP_PKEY_encrypt(ctx
, tmp
, &len
, pt
, sizeof(pt
)), 0)
313 || !TEST_true(called_encrypt
))
316 EVP_PKEY_CTX_free(ctx
);
320 /* Create context with default engine: should not be redirected */
321 if (!TEST_ptr(ctx
= EVP_PKEY_CTX_new(pkey
, NULL
))
322 || !TEST_int_gt(EVP_PKEY_encrypt_init(ctx
), 0)
323 || !TEST_int_gt(EVP_PKEY_encrypt(ctx
, tmp
, &len
, pt
, sizeof(pt
)), 0)
324 || !TEST_false(called_encrypt
))
327 EVP_PKEY_CTX_free(ctx
);
330 /* Set engine explicitly for test key */
331 if (!TEST_true(EVP_PKEY_set1_engine(pkey
, e
)))
334 TEST_info("EVP_PKEY_encrypt test: redirection via EVP_PKEY_set1_engine()");
336 /* Create context with default engine: should be redirected now */
337 if (!TEST_ptr(ctx
= EVP_PKEY_CTX_new(pkey
, NULL
))
338 || !TEST_int_gt(EVP_PKEY_encrypt_init(ctx
), 0)
339 || !TEST_int_gt(EVP_PKEY_encrypt(ctx
, tmp
, &len
, pt
, sizeof(pt
)), 0)
340 || !TEST_true(called_encrypt
))
346 EVP_PKEY_CTX_free(ctx
);
354 int global_init(void)
357 * If the config file gets loaded, the dynamic engine will be loaded,
358 * and that interferes with our test above.
360 return OPENSSL_init_crypto(OPENSSL_INIT_NO_LOAD_CONFIG
, NULL
);
363 int setup_tests(void)
365 #ifdef OPENSSL_NO_ENGINE
366 TEST_note("No ENGINE support");
368 ADD_TEST(test_engines
);
369 ADD_TEST(test_redirect
);