test/context_internal_test.c

   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
  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 }
  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
  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
  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
  72 static int test_def_context(void)
  73 {
  74     return test_context(NULL);
  75 }
  76
  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
 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 }