1 /*
2 * Copyright 2019-2021 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 /* Internal tests for the OpenSSL library context */
11
12 #include "internal/cryptlib.h"
13 #include "testutil.h"
14
15 /*
16 * Everything between BEGIN EXAMPLE and END EXAMPLE is copied from
17 * doc/internal/man3/ossl_lib_ctx_get_data.pod
18 */
19
20 /*
21 * ======================================================================
22 * BEGIN EXAMPLE
23 */
24
25 typedef struct foo_st {
26 int i;
27 void *data;
28 } FOO;
29
foo_new(OSSL_LIB_CTX * ctx)30 static void *foo_new(OSSL_LIB_CTX *ctx)
31 {
32 FOO *ptr = OPENSSL_zalloc(sizeof(*ptr));
33 if (ptr != NULL)
34 ptr->i = 42;
35 return ptr;
36 }
foo_free(void * ptr)37 static void foo_free(void *ptr)
38 {
39 OPENSSL_free(ptr);
40 }
41 static const OSSL_LIB_CTX_METHOD foo_method = {
42 OSSL_LIB_CTX_METHOD_DEFAULT_PRIORITY,
43 foo_new,
44 foo_free
45 };
46
47 /*
48 * END EXAMPLE
49 * ======================================================================
50 */
51
test_context(OSSL_LIB_CTX * ctx)52 static int test_context(OSSL_LIB_CTX *ctx)
53 {
54 FOO *data = NULL;
55
56 return TEST_ptr(data = ossl_lib_ctx_get_data(ctx, 0, &foo_method))
57 /* OPENSSL_zalloc in foo_new() initialized it to zero */
58 && TEST_int_eq(data->i, 42);
59 }
60
test_app_context(void)61 static int test_app_context(void)
62 {
63 OSSL_LIB_CTX *ctx = NULL;
64 int result =
65 TEST_ptr(ctx = OSSL_LIB_CTX_new())
66 && test_context(ctx);
67
68 OSSL_LIB_CTX_free(ctx);
69 return result;
70 }
71
test_def_context(void)72 static int test_def_context(void)
73 {
74 return test_context(NULL);
75 }
76
test_set0_default(void)77 static int test_set0_default(void)
78 {
79 OSSL_LIB_CTX *global = OSSL_LIB_CTX_get0_global_default();
80 OSSL_LIB_CTX *local = OSSL_LIB_CTX_new();
81 OSSL_LIB_CTX *prev;
82 int testresult = 0;
83 FOO *data = NULL;
84
85 if (!TEST_ptr(global)
86 || !TEST_ptr(local)
87 || !TEST_ptr_eq(global, OSSL_LIB_CTX_set0_default(NULL))
88 || !TEST_ptr(data = ossl_lib_ctx_get_data(local, 0, &foo_method)))
89 goto err;
90
91 /* Set local "i" value to 43. Global "i" should be 42 */
92 data->i++;
93 if (!TEST_int_eq(data->i, 43))
94 goto err;
95
96 /* The default context should still be the "global" default */
97 if (!TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
98 || !TEST_int_eq(data->i, 42))
99 goto err;
100
101 /* Check we can change the local default context */
102 if (!TEST_ptr(prev = OSSL_LIB_CTX_set0_default(local))
103 || !TEST_ptr_eq(global, prev)
104 || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
105 || !TEST_int_eq(data->i, 43))
106 goto err;
107
108 /* Calling OSSL_LIB_CTX_set0_default() with a NULL should be a no-op */
109 if (!TEST_ptr_eq(local, OSSL_LIB_CTX_set0_default(NULL))
110 || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
111 || !TEST_int_eq(data->i, 43))
112 goto err;
113
114 /* Global default should be unchanged */
115 if (!TEST_ptr_eq(global, OSSL_LIB_CTX_get0_global_default()))
116 goto err;
117
118 /* Check we can swap back to the global default */
119 if (!TEST_ptr(prev = OSSL_LIB_CTX_set0_default(global))
120 || !TEST_ptr_eq(local, prev)
121 || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
122 || !TEST_int_eq(data->i, 42))
123 goto err;
124
125 testresult = 1;
126 err:
127 OSSL_LIB_CTX_free(local);
128 return testresult;
129 }
130
setup_tests(void)131 int setup_tests(void)
132 {
133 ADD_TEST(test_app_context);
134 ADD_TEST(test_def_context);
135 ADD_TEST(test_set0_default);
136 return 1;
137 }
138