mirror of
https://github.com/openssl/openssl.git
synced 2024-12-09 05:51:54 +08:00
11efa7d45b
The providers have to call up_ref to keep the cbio pointer, just like the internal bio_prov.c does. OSSL_STORE_attach passes a cbio pointer to the provider and then calls ossl_core_bio_free(cbio). If up_ref is not called, the cbio gets freed way too early. Reviewed-by: Matt Caswell <matt@openssl.org> Reviewed-by: Paul Dale <pauli@openssl.org> (Merged from https://github.com/openssl/openssl/pull/15163)
120 lines
3.4 KiB
C
120 lines
3.4 KiB
C
/*
|
|
* Copyright 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
|
|
*/
|
|
|
|
#include <string.h>
|
|
#include <openssl/bio.h>
|
|
#include "testutil.h"
|
|
|
|
struct ossl_core_bio_st {
|
|
int dummy;
|
|
BIO *bio;
|
|
};
|
|
|
|
static int tst_bio_core_read_ex(OSSL_CORE_BIO *bio, char *data, size_t data_len,
|
|
size_t *bytes_read)
|
|
{
|
|
return BIO_read_ex(bio->bio, data, data_len, bytes_read);
|
|
}
|
|
|
|
static int tst_bio_core_write_ex(OSSL_CORE_BIO *bio, const char *data,
|
|
size_t data_len, size_t *written)
|
|
{
|
|
return BIO_write_ex(bio->bio, data, data_len, written);
|
|
}
|
|
|
|
static int tst_bio_core_gets(OSSL_CORE_BIO *bio, char *buf, int size)
|
|
{
|
|
return BIO_gets(bio->bio, buf, size);
|
|
}
|
|
|
|
static int tst_bio_core_puts(OSSL_CORE_BIO *bio, const char *str)
|
|
{
|
|
return BIO_puts(bio->bio, str);
|
|
}
|
|
|
|
static long tst_bio_core_ctrl(OSSL_CORE_BIO *bio, int cmd, long num, void *ptr)
|
|
{
|
|
return BIO_ctrl(bio->bio, cmd, num, ptr);
|
|
}
|
|
|
|
static int tst_bio_core_up_ref(OSSL_CORE_BIO *bio)
|
|
{
|
|
return BIO_up_ref(bio->bio);
|
|
}
|
|
|
|
static int tst_bio_core_free(OSSL_CORE_BIO *bio)
|
|
{
|
|
return BIO_free(bio->bio);
|
|
}
|
|
|
|
static const OSSL_DISPATCH biocbs[] = {
|
|
{ OSSL_FUNC_BIO_READ_EX, (void (*)(void))tst_bio_core_read_ex },
|
|
{ OSSL_FUNC_BIO_WRITE_EX, (void (*)(void))tst_bio_core_write_ex },
|
|
{ OSSL_FUNC_BIO_GETS, (void (*)(void))tst_bio_core_gets },
|
|
{ OSSL_FUNC_BIO_PUTS, (void (*)(void))tst_bio_core_puts },
|
|
{ OSSL_FUNC_BIO_CTRL, (void (*)(void))tst_bio_core_ctrl },
|
|
{ OSSL_FUNC_BIO_UP_REF, (void (*)(void))tst_bio_core_up_ref },
|
|
{ OSSL_FUNC_BIO_FREE, (void (*)(void))tst_bio_core_free },
|
|
{ 0, NULL }
|
|
};
|
|
|
|
static int test_bio_core(void)
|
|
{
|
|
BIO *cbio = NULL, *cbiobad = NULL;
|
|
OSSL_LIB_CTX *libctx = OSSL_LIB_CTX_new_from_dispatch(NULL, biocbs);
|
|
int testresult = 0;
|
|
OSSL_CORE_BIO corebio;
|
|
const char *msg = "Hello world";
|
|
char buf[80];
|
|
|
|
corebio.bio = BIO_new(BIO_s_mem());
|
|
if (!TEST_ptr(corebio.bio)
|
|
|| !TEST_ptr(libctx)
|
|
/*
|
|
* Attempting to create a corebio in a libctx that was not
|
|
* created via OSSL_LIB_CTX_new_from_dispatch() should fail.
|
|
*/
|
|
|| !TEST_ptr_null((cbiobad = BIO_new_from_core_bio(NULL, &corebio)))
|
|
|| !TEST_ptr((cbio = BIO_new_from_core_bio(libctx, &corebio))))
|
|
goto err;
|
|
|
|
if (!TEST_int_gt(BIO_puts(corebio.bio, msg), 0)
|
|
/* Test a ctrl via BIO_eof */
|
|
|| !TEST_false(BIO_eof(cbio))
|
|
|| !TEST_int_gt(BIO_gets(cbio, buf, sizeof(buf)), 0)
|
|
|| !TEST_true(BIO_eof(cbio))
|
|
|| !TEST_str_eq(buf, msg))
|
|
goto err;
|
|
|
|
buf[0] = '\0';
|
|
if (!TEST_int_gt(BIO_write(cbio, msg, strlen(msg) + 1), 0)
|
|
|| !TEST_int_gt(BIO_read(cbio, buf, sizeof(buf)), 0)
|
|
|| !TEST_str_eq(buf, msg))
|
|
goto err;
|
|
|
|
testresult = 1;
|
|
err:
|
|
BIO_free(cbiobad);
|
|
BIO_free(cbio);
|
|
BIO_free(corebio.bio);
|
|
OSSL_LIB_CTX_free(libctx);
|
|
return testresult;
|
|
}
|
|
|
|
int setup_tests(void)
|
|
{
|
|
if (!test_skip_common_options()) {
|
|
TEST_error("Error parsing test options\n");
|
|
return 0;
|
|
}
|
|
|
|
ADD_TEST(test_bio_core);
|
|
return 1;
|
|
}
|