123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136 |
- /*
- * Copyright 2019-2021 The OpenSSL Project Authors. All Rights Reserved.
- *
- * Licensed under the Apache License 2.0 (the "License"). You may not use
- * this file except in compliance with the License. You can obtain a copy
- * in the file LICENSE in the source distribution or at
- * https://www.openssl.org/source/license.html
- */
- /* Internal tests for the OpenSSL library context */
- #include "internal/cryptlib.h"
- #include "testutil.h"
- /*
- * Everything between BEGIN EXAMPLE and END EXAMPLE is copied from
- * doc/internal/man3/ossl_lib_ctx_get_data.pod
- */
- /*
- * ======================================================================
- * BEGIN EXAMPLE
- */
- typedef struct foo_st {
- int i;
- void *data;
- } FOO;
- static void *foo_new(OSSL_LIB_CTX *ctx)
- {
- FOO *ptr = OPENSSL_zalloc(sizeof(*ptr));
- if (ptr != NULL)
- ptr->i = 42;
- return ptr;
- }
- static void foo_free(void *ptr)
- {
- OPENSSL_free(ptr);
- }
- static const OSSL_LIB_CTX_METHOD foo_method = {
- foo_new,
- foo_free
- };
- /*
- * END EXAMPLE
- * ======================================================================
- */
- static int test_context(OSSL_LIB_CTX *ctx)
- {
- FOO *data = NULL;
- return TEST_ptr(data = ossl_lib_ctx_get_data(ctx, 0, &foo_method))
- /* OPENSSL_zalloc in foo_new() initialized it to zero */
- && TEST_int_eq(data->i, 42);
- }
- static int test_app_context(void)
- {
- OSSL_LIB_CTX *ctx = NULL;
- int result =
- TEST_ptr(ctx = OSSL_LIB_CTX_new())
- && test_context(ctx);
- OSSL_LIB_CTX_free(ctx);
- return result;
- }
- 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;
- }
|