diff --git a/cyassl/ssl.h b/cyassl/ssl.h index 1f9643d2e..b8aa8a965 100644 --- a/cyassl/ssl.h +++ b/cyassl/ssl.h @@ -886,8 +886,9 @@ CYASSL_API void CyaSSL_SetIOWriteFlags(CYASSL* ssl, int flags); typedef int (*CallbackGenCookie)(CYASSL* ssl, unsigned char* buf, int sz, void* ctx); -CYASSL_API void CyaSSL_CTX_SetGenCookie(CYASSL_CTX*, CallbackGenCookie); -CYASSL_API void CyaSSL_SetCookieCtx(CYASSL* ssl, void *ctx); +CYASSL_API void CyaSSL_CTX_SetGenCookie(CYASSL_CTX*, CallbackGenCookie); +CYASSL_API void CyaSSL_SetCookieCtx(CYASSL* ssl, void *ctx); +CYASSL_API void* CyaSSL_GetCookieCtx(CYASSL* ssl); typedef int (*CallbackIOOcsp)(void*, const char*, int, unsigned char*, int, unsigned char**); diff --git a/src/io.c b/src/io.c index 2442de433..4511d4f8a 100644 --- a/src/io.c +++ b/src/io.c @@ -896,6 +896,15 @@ CYASSL_API void CyaSSL_SetCookieCtx(CYASSL* ssl, void *ctx) ssl->IOCB_CookieCtx = ctx; } + +CYASSL_API void* CyaSSL_GetCookieCtx(CYASSL* ssl) +{ + if (ssl) + return ssl->IOCB_CookieCtx; + + return NULL; +} + #endif /* CYASSL_DTLS */