Home | History | Annotate | Line # | Download | only in test 
      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 }
     72