espressif/esp-idf
1
0
Fork 1
mirror of https://github.com/espressif/esp-idf.git synced 2025-08-01 03:34:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

esp-tls: add api to free client session

Browse Source 
Free session with mbedtls api to avoid mem-leak
This commit is contained in:
Li Jingyi
2022-05-17 17:14:57 +08:00
parent a7664ecb27
commit 4be609a471
4 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
components/esp-tls/esp_tls.c
View File

@@ -39,6 +39,7 @@ static const char *TAG = "esp-tls";
#define _esp_tls_conn_delete esp_mbedtls_conn_delete #define _esp_tls_conn_delete esp_mbedtls_conn_delete
#define _esp_tls_net_init esp_mbedtls_net_init #define _esp_tls_net_init esp_mbedtls_net_init
#define _esp_tls_get_client_session esp_mbedtls_get_client_session #define _esp_tls_get_client_session esp_mbedtls_get_client_session
#define _esp_tls_free_client_session esp_mbedtls_free_client_session
#ifdef CONFIG_ESP_TLS_SERVER #ifdef CONFIG_ESP_TLS_SERVER
#define _esp_tls_server_session_create esp_mbedtls_server_session_create #define _esp_tls_server_session_create esp_mbedtls_server_session_create
#define _esp_tls_server_session_delete esp_mbedtls_server_session_delete #define _esp_tls_server_session_delete esp_mbedtls_server_session_delete
@@ -577,6 +578,11 @@ esp_tls_client_session_t *esp_tls_get_client_session(esp_tls_t *tls)
{ {
return _esp_tls_get_client_session(tls); return _esp_tls_get_client_session(tls);
} }
void esp_tls_free_client_session(esp_tls_client_session_t *client_session)
{
_esp_tls_free_client_session(client_session);
}
#endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */ #endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */

10
components/esp-tls/esp_tls.h
View File

@@ -688,6 +688,16 @@ esp_err_t esp_tls_plain_tcp_connect(const char *host, int hostlen, int port, con
* NULL on Failure * NULL on Failure
*/ */
esp_tls_client_session_t *esp_tls_get_client_session(esp_tls_t *tls); esp_tls_client_session_t *esp_tls_get_client_session(esp_tls_t *tls);
/**
* @brief Free the client session
*
* This function should be called after esp_tls_get_client_session().
*
* @param[in] client_session context as esp_tls_client_session_t
*
*/
void esp_tls_free_client_session(esp_tls_client_session_t *client_session);
#endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */ #endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */
#ifdef __cplusplus #ifdef __cplusplus
} }

8
components/esp-tls/esp_tls_mbedtls.c
View File

@@ -151,6 +151,14 @@ esp_tls_client_session_t *esp_mbedtls_get_client_session(esp_tls_t *tls)
return client_session; return client_session;
} }
void esp_mbedtls_free_client_session(esp_tls_client_session_t *client_session)
{
if (client_session) {
mbedtls_ssl_session_free(&(client_session->saved_session));
free(client_session);
}
}
#endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */ #endif /* CONFIG_ESP_TLS_CLIENT_SESSION_TICKETS */
int esp_mbedtls_handshake(esp_tls_t *tls, const esp_tls_cfg_t *cfg) int esp_mbedtls_handshake(esp_tls_t *tls, const esp_tls_cfg_t *cfg)

5
components/esp-tls/private_include/esp_tls_mbedtls.h
View File

@@ -104,6 +104,11 @@ esp_err_t set_client_config(const char *hostname, size_t hostlen, esp_tls_cfg_t
* Internal Callback for mbedtls_get_client_session * Internal Callback for mbedtls_get_client_session
*/ */
esp_tls_client_session_t *esp_mbedtls_get_client_session(esp_tls_t *tls); esp_tls_client_session_t *esp_mbedtls_get_client_session(esp_tls_t *tls);
/**
* Internal Callback for mbedtls_free_client_session
*/
void esp_mbedtls_free_client_session(esp_tls_client_session_t *client_session);
#endif #endif
/** /**