test/bio_core_test.c

   1 /*
   2  * Copyright 2021-2023 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 <string.h>
  11 #include <openssl/bio.h>
  12 #include "testutil.h"
  13
  14 struct ossl_core_bio_st {
  15     int dummy;
  16     BIO *bio;
  17 };
  18
  19 static int tst_bio_core_read_ex(OSSL_CORE_BIO *bio, char *data, size_t data_len,
  20                                 size_t *bytes_read)
  21 {
  22     return BIO_read_ex(bio->bio, data, data_len, bytes_read);
  23 }
  24
  25 static int tst_bio_core_write_ex(OSSL_CORE_BIO *bio, const char *data,
  26                                  size_t data_len, size_t *written)
  27 {
  28     return BIO_write_ex(bio->bio, data, data_len, written);
  29 }
  30
  31 static int tst_bio_core_gets(OSSL_CORE_BIO *bio, char *buf, int size)
  32 {
  33     return BIO_gets(bio->bio, buf, size);
  34 }
  35
  36 static int tst_bio_core_puts(OSSL_CORE_BIO *bio, const char *str)
  37 {
  38     return BIO_puts(bio->bio, str);
  39 }
  40
  41 static long tst_bio_core_ctrl(OSSL_CORE_BIO *bio, int cmd, long num, void *ptr)
  42 {
  43     return BIO_ctrl(bio->bio, cmd, num, ptr);
  44 }
  45
  46 static int tst_bio_core_up_ref(OSSL_CORE_BIO *bio)
  47 {
  48     return BIO_up_ref(bio->bio);
  49 }
  50
  51 static int tst_bio_core_free(OSSL_CORE_BIO *bio)
  52 {
  53     return BIO_free(bio->bio);
  54 }
  55
  56 static const OSSL_DISPATCH biocbs[] = {
  57     { OSSL_FUNC_BIO_READ_EX, (void (*)(void))tst_bio_core_read_ex },
  58     { OSSL_FUNC_BIO_WRITE_EX, (void (*)(void))tst_bio_core_write_ex },
  59     { OSSL_FUNC_BIO_GETS, (void (*)(void))tst_bio_core_gets },
  60     { OSSL_FUNC_BIO_PUTS, (void (*)(void))tst_bio_core_puts },
  61     { OSSL_FUNC_BIO_CTRL, (void (*)(void))tst_bio_core_ctrl },
  62     { OSSL_FUNC_BIO_UP_REF, (void (*)(void))tst_bio_core_up_ref },
  63     { OSSL_FUNC_BIO_FREE, (void (*)(void))tst_bio_core_free },
  64     OSSL_DISPATCH_END
  65 };
  66
  67 static int test_bio_core(void)
  68 {
  69     BIO *cbio = NULL, *cbiobad = NULL;
  70     OSSL_LIB_CTX *libctx = OSSL_LIB_CTX_new_from_dispatch(NULL, biocbs);
  71     int testresult = 0;
  72     OSSL_CORE_BIO corebio;
  73     const char *msg = "Hello world";
  74     char buf[80];
  75
  76     corebio.bio = BIO_new(BIO_s_mem());
  77     if (!TEST_ptr(corebio.bio)
  78             || !TEST_ptr(libctx)
  79                /*
  80                 * Attempting to create a corebio in a libctx that was not
  81                 * created via OSSL_LIB_CTX_new_from_dispatch() should fail.
  82                 */
  83             || !TEST_ptr_null((cbiobad = BIO_new_from_core_bio(NULL, &corebio)))
  84             || !TEST_ptr((cbio = BIO_new_from_core_bio(libctx, &corebio))))
  85         goto err;
  86
  87     if (!TEST_int_gt(BIO_puts(corebio.bio, msg), 0)
  88                /* Test a ctrl via BIO_eof */
  89             || !TEST_false(BIO_eof(cbio))
  90             || !TEST_int_gt(BIO_gets(cbio, buf, sizeof(buf)), 0)
  91             || !TEST_true(BIO_eof(cbio))
  92             || !TEST_str_eq(buf, msg))
  93         goto err;
  94
  95     buf[0] = '\0';
  96     if (!TEST_int_gt(BIO_write(cbio, msg, strlen(msg) + 1), 0)
  97             || !TEST_int_gt(BIO_read(cbio, buf, sizeof(buf)), 0)
  98             || !TEST_str_eq(buf, msg))
  99         goto err;
 100
 101     testresult = 1;
 102  err:
 103     BIO_free(cbiobad);
 104     BIO_free(cbio);
 105     BIO_free(corebio.bio);
 106     OSSL_LIB_CTX_free(libctx);
 107     return testresult;
 108 }
 109
 110 int setup_tests(void)
 111 {
 112     if (!test_skip_common_options()) {
 113         TEST_error("Error parsing test options\n");
 114         return 0;
 115     }
 116
 117     ADD_TEST(test_bio_core);
 118     return 1;
 119 }