diff --git a/src/proxy/tinymitm/raai-helper.cpp b/src/proxy/tinymitm/raai-helper.cpp index b1a2a5f..03f8523 100644 --- a/src/proxy/tinymitm/raai-helper.cpp +++ b/src/proxy/tinymitm/raai-helper.cpp @@ -6,7 +6,7 @@ #include #include -void WolfSSLDeleter::operator()(WC_RNG* rng) +void RAAIHelper::operator()(WC_RNG* rng) { if (rng) { @@ -14,7 +14,7 @@ void WolfSSLDeleter::operator()(WC_RNG* rng) delete rng; } } -void WolfSSLDeleter::operator()(RsaKey* key) +void RAAIHelper::operator()(RsaKey* key) { if (key) { @@ -22,12 +22,17 @@ void WolfSSLDeleter::operator()(RsaKey* key) delete key; } } -void WolfSSLDeleter::operator()(WOLFSSL_CTX* ctx) +void RAAIHelper::operator()(WOLFSSL_CTX* ctx) { if (ctx) wolfSSL_CTX_free(ctx); } -void WolfSSLDeleter::operator()(WOLFSSL* ssl) +void RAAIHelper::operator()(WOLFSSL* ssl) { if (ssl) wolfSSL_free(ssl); } + +void RAAIHelper::operator()(addrinfo* addrinfo) +{ + if (addrinfo) freeaddrinfo(addrinfo); +} diff --git a/src/proxy/tinymitm/raai-helper.h b/src/proxy/tinymitm/raai-helper.h index aec8098..3dc33d4 100644 --- a/src/proxy/tinymitm/raai-helper.h +++ b/src/proxy/tinymitm/raai-helper.h @@ -4,11 +4,13 @@ struct WC_RNG; struct RsaKey; struct WOLFSSL_CTX; struct WOLFSSL; +struct addrinfo; -struct WolfSSLDeleter +struct RAAIHelper { void operator()(WC_RNG* rng); void operator()(RsaKey* key); void operator()(WOLFSSL_CTX* ctx); void operator()(WOLFSSL* ssl); + void operator()(addrinfo* addrinfo); };