diff --git a/configure.ac b/configure.ac index cdc5fb41d..73ed499ce 100644 --- a/configure.ac +++ b/configure.ac @@ -115,7 +115,7 @@ AC_C_BIGENDIAN # check if functions of interest are linkable, but also check if # they're declared by the expected headers, and if not, supersede the # unusable positive from AC_CHECK_FUNCS(). -AC_CHECK_FUNCS([gethostbyname getaddrinfo gettimeofday gmtime_r inet_ntoa memset socket strftime]) +AC_CHECK_FUNCS([gethostbyname getaddrinfo gettimeofday gmtime_r inet_ntoa memset socket strftime atexit]) AC_CHECK_DECLS([gethostbyname, getaddrinfo, gettimeofday, gmtime_r, inet_ntoa, memset, socket, strftime], [], [ if test "$(eval echo \$"$(eval 'echo ac_cv_func_${as_decl_name}')")" = "yes" then diff --git a/src/ssl.c b/src/ssl.c index fb450b17c..1be42fc40 100644 --- a/src/ssl.c +++ b/src/ssl.c @@ -5092,6 +5092,16 @@ int AddCA(WOLFSSL_CERT_MANAGER* cm, DerBuffer** pDer, int type, int verify) static int wolfSSL_RAND_InitMutex(void); #endif +#if defined(OPENSSL_EXTRA) && defined(HAVE_ATEXIT) +static void AtExitCleanup(void) +{ + if (initRefCount > 0) { + initRefCount = 1; + (void)wolfSSL_Cleanup(); + } +} +#endif + WOLFSSL_ABI int wolfSSL_Init(void) { @@ -5116,6 +5126,14 @@ int wolfSSL_Init(void) #endif #ifdef OPENSSL_EXTRA + #ifdef HAVE_ATEXIT + /* OpenSSL registers cleanup using atexit */ + if (atexit(AtExitCleanup) != 0) { + WOLFSSL_MSG("Bad atexit registration"); + return WC_INIT_E; + } + #endif + #ifndef WOLFSSL_NO_OPENSSL_RAND_CB if (wolfSSL_RAND_InitMutex() != 0) { return BAD_MUTEX_E; @@ -57658,9 +57676,8 @@ int wolfSSL_CONF_cmd(WOLFSSL_CONF_CTX* cctx, const char* cmd, const char* value) bio->shutdown = BIO_CLOSE; /* default to close things */ bio->num = SOCKET_INVALID; /* Default to invalid socket */ bio->init = 1; - if (method->type != WOLFSSL_BIO_FILE && - method->type != WOLFSSL_BIO_SOCKET && - method->type != WOLFSSL_BIO_MD) { + if (method->type == WOLFSSL_BIO_MEMORY || + method->type == WOLFSSL_BIO_BIO) { bio->mem_buf =(WOLFSSL_BUF_MEM*)XMALLOC(sizeof(WOLFSSL_BUF_MEM), 0, DYNAMIC_TYPE_OPENSSL); if (bio->mem_buf == NULL) {