test/bio_meth_test.c

   1 /*
   2  * Copyright 2024 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 #include <openssl/bio.h>
  11 #include "testutil.h"
  12
  13 static int test_bio_meth(void)
  14 {
  15     int i, ret = 0, id;
  16     BIO_METHOD *meth1 = NULL, *meth2 = NULL, *meth3 = NULL;
  17     BIO *membio = NULL, *bio1 = NULL, *bio2 = NULL, *bio3 = NULL;
  18
  19     id = BIO_get_new_index();
  20     if (!TEST_int_eq(id, BIO_TYPE_START + 1))
  21         goto err;
  22
  23     if (!TEST_ptr(meth1 = BIO_meth_new(id, "Method1"))
  24         || !TEST_ptr(meth2 = BIO_meth_new(BIO_TYPE_NONE, "Method2"))
  25         || !TEST_ptr(meth3 = BIO_meth_new(BIO_TYPE_NONE|BIO_TYPE_FILTER, "Method3"))
  26         || !TEST_ptr(bio1 = BIO_new(meth1))
  27         || !TEST_ptr(bio2 = BIO_new(meth2))
  28         || !TEST_ptr(bio3 = BIO_new(meth3))
  29         || !TEST_ptr(membio = BIO_new(BIO_s_mem())))
  30         goto err;
  31
  32     BIO_set_next(bio3, bio2);
  33     BIO_set_next(bio2, bio1);
  34     BIO_set_next(bio1, membio);
  35
  36     /* Check that we get an error if we exhaust BIO_get_new_index() */
  37     for (i = id + 1; i <= BIO_TYPE_MASK; ++i) {
  38         if (!TEST_int_eq(BIO_get_new_index(), i))
  39             goto err;
  40     }
  41     if (!TEST_int_eq(BIO_get_new_index(), -1))
  42         goto err;
  43
  44     /* test searching works */
  45     if (!TEST_ptr_eq(BIO_find_type(bio3, BIO_TYPE_MEM), membio)
  46         || !TEST_ptr_eq(BIO_find_type(bio3, id), bio1))
  47         goto err;
  48
  49     /* Check searching for BIO_TYPE_NONE returns NULL */
  50     if (!TEST_ptr_null(BIO_find_type(bio3, BIO_TYPE_NONE)))
  51         goto err;
  52     /* Check searching for BIO_TYPE_NONE + BIO_TYPE_FILTER works */
  53     if (!TEST_ptr_eq(BIO_find_type(bio3, BIO_TYPE_FILTER), bio3))
  54         goto err;
  55     ret = 1;
  56 err:
  57     BIO_free(membio);
  58     BIO_free(bio3);
  59     BIO_free(bio2);
  60     BIO_free(bio1);
  61     BIO_meth_free(meth3);
  62     BIO_meth_free(meth2);
  63     BIO_meth_free(meth1);
  64     return ret;
  65 }
  66
  67 int setup_tests(void)
  68 {
  69     ADD_TEST(test_bio_meth);
  70     return 1;
  71 }