From: Matt Caswell <matt@openssl.org>
Date: Fri, 16 Apr 2021 11:21:50 +0000 (+0100)
Subject: Add a test for OSSL_LIB_CTX_set0_default
X-Git-Tag: openssl-3.0.0-alpha15~36
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=ee203a87ff1ff1af46a5ff11f761bdd07a5503e4;p=thirdparty%2Fopenssl.git

Add a test for OSSL_LIB_CTX_set0_default

Also includes testing for OSSL_LIB_CTX_get0_global_default().

Reviewed-by: Paul Dale <pauli@openssl.org>
Reviewed-by: Tim Hudson <tjh@openssl.org>
Reviewed-by: Tomas Mraz <tomas@openssl.org>
(Merged from https://github.com/openssl/openssl/pull/14890)
---

diff --git a/test/context_internal_test.c b/test/context_internal_test.c
index 0b786adf675..a8759084695 100644
--- a/test/context_internal_test.c
+++ b/test/context_internal_test.c
@@ -73,9 +73,64 @@ static int test_def_context(void)
     return test_context(NULL);
 }
 
+static int test_set0_default(void)
+{
+    OSSL_LIB_CTX *global = OSSL_LIB_CTX_get0_global_default();
+    OSSL_LIB_CTX *local = OSSL_LIB_CTX_new();
+    OSSL_LIB_CTX *prev;
+    int testresult = 0;
+    FOO *data = NULL;
+
+    if (!TEST_ptr(global)
+            || !TEST_ptr(local)
+            || !TEST_ptr_eq(global, OSSL_LIB_CTX_set0_default(NULL))
+            || !TEST_ptr(data = ossl_lib_ctx_get_data(local, 0, &foo_method)))
+        goto err;
+
+    /* Set local "i" value to 43. Global "i" should be 42 */
+    data->i++;
+    if (!TEST_int_eq(data->i, 43))
+        goto err;
+
+    /* The default context should still be the "global" default */
+    if (!TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
+            || !TEST_int_eq(data->i, 42))
+        goto err;
+
+    /* Check we can change the local default context */
+    if (!TEST_ptr(prev = OSSL_LIB_CTX_set0_default(local))
+            || !TEST_ptr_eq(global, prev)
+            || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
+            || !TEST_int_eq(data->i, 43))
+        goto err;
+
+    /* Calling OSSL_LIB_CTX_set0_default() with a NULL should be a no-op */
+    if (!TEST_ptr_eq(local, OSSL_LIB_CTX_set0_default(NULL))
+            || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
+            || !TEST_int_eq(data->i, 43))
+        goto err;
+
+    /* Global default should be unchanged */
+    if (!TEST_ptr_eq(global, OSSL_LIB_CTX_get0_global_default()))
+        goto err;
+
+    /* Check we can swap back to the global default */
+   if (!TEST_ptr(prev = OSSL_LIB_CTX_set0_default(global))
+            || !TEST_ptr_eq(local, prev)
+            || !TEST_ptr(data = ossl_lib_ctx_get_data(NULL, 0, &foo_method))
+            || !TEST_int_eq(data->i, 42))
+        goto err;
+
+    testresult = 1;
+ err:
+    OSSL_LIB_CTX_free(local);
+    return testresult;
+}
+
 int setup_tests(void)
 {
     ADD_TEST(test_app_context);
     ADD_TEST(test_def_context);
+    ADD_TEST(test_set0_default);
     return 1;
 }