Skip to content

Commit 0893b27

Browse files
committed
extmod/modussl_mbedtls: Make socket.close() free all TLS resources.
Also, use mp_stream_close() helper to close the underlying socket.
1 parent 363087a commit 0893b27

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

extmod/modussl_mbedtls.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,15 +227,15 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_setblocking_obj, socket_setblocking);
227227
STATIC mp_obj_t socket_close(mp_obj_t self_in) {
228228
mp_obj_ssl_socket_t *self = MP_OBJ_TO_PTR(self_in);
229229

230+
mbedtls_pk_free(&self->pkey);
231+
mbedtls_x509_crt_free(&self->cert);
230232
mbedtls_x509_crt_free(&self->cacert);
231233
mbedtls_ssl_free(&self->ssl);
232234
mbedtls_ssl_config_free(&self->conf);
233235
mbedtls_ctr_drbg_free(&self->ctr_drbg);
234236
mbedtls_entropy_free(&self->entropy);
235237

236-
mp_obj_t dest[2];
237-
mp_load_method(self->sock, MP_QSTR_close, dest);
238-
return mp_call_method_n_kw(0, 0, dest);
238+
return mp_stream_close(self->sock);
239239
}
240240
STATIC MP_DEFINE_CONST_FUN_OBJ_1(socket_close_obj, socket_close);
241241

0 commit comments

Comments
 (0)