使用 openssl 计算 base64(防止内存泄露)

#include "openssl/bio.h"
#include "openssl/evp.h"
#include "openssl/buffer.h"

int base64(const unsigned char *src, int src_len, unsigned char *dst, int *dst_len) 
{
    BIO *bio_hnd_mem = NULL;
    BIO *bio_hnd_meth = NULL;
    BUF_MEM *buf_mem_ptr = NULL;

    if ((src == NULL) || (dst == NULL) || (dst_len == NULL)) {
        return -1;
    }

    bio_hnd_meth = BIO_new(BIO_f_base64());
    BIO_set_flags(bio_hnd_meth, BIO_FLAGS_BASE64_NO_NL);
    bio_hnd_mem = BIO_new(BIO_s_mem());
    bio_hnd_meth = BIO_push(bio_hnd_meth, bio_hnd_mem);

    BIO_write(bio_hnd_meth, src, src_len);
    BIO_flush(bio_hnd_meth);
    BIO_get_mem_ptr(bio_hnd_meth, &buf_mem_ptr);
    BIO_set_close(bio_hnd_meth, BIO_NOCLOSE);

    if (*dst_len < buf_mem_ptr->length) {
        return -1;
    }

    memcpy(dst, buf_mem_ptr->data, buf_mem_ptr->length);
    *dst_len = buf_mem_ptr->length;

    BUF_MEM_free(buf_mem_ptr);
    BIO_free(bio_hnd_meth);
    BIO_free(bio_hnd_mem);

    return *dst_len;
}

 

  注意:

结尾处的释放资源时我看到网上好多帖子上只调用了一个函数:BIO_free_all(bio_hnd_meth)。

但是,我在实际使用中发现如果只调用一个BIO_free_all(),会产生内存泄露,主要是没有释放掉buf_mem_ptr指向的内存。

由于我还没看openssl库源码,因此具体原因尚不明确。猜测可能是低版本的openssl库有bug,可能高版本解决了这个bug,也可能是openssl本身设计就是如此只是使用者没有搞明白而已。

 

 

上一篇:PyTorch模型转TensorRT


下一篇:bwa mem 比对分值参数测试