1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

net.mbedtls: add explicit closing of the tcp connection in the shutdown method (#16027)

This commit is contained in:
Thomas Peißl 2022-10-10 18:30:58 +02:00 committed by GitHub
parent 82593338fa
commit 0f229874a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -37,6 +37,8 @@ mut:
handle int
duration time.Duration
opened bool
owns_socket bool
}
[params]
@ -77,6 +79,15 @@ pub fn (mut s SSLConn) shutdown() ? {
}
C.mbedtls_ssl_free(&s.ssl)
C.mbedtls_ssl_config_free(&s.conf)
if s.owns_socket {
$if windows {
C.shutdown(s.handle, C.SD_BOTH)
net.socket_error(C.closesocket(s.handle))?
} $else {
C.shutdown(s.handle, C.SHUT_RDWR)
net.socket_error(C.close(s.handle))?
}
}
}
// connect to server using mbedtls
@ -173,6 +184,7 @@ pub fn (mut s SSLConn) connect(mut tcp_conn net.TcpConn, hostname string) ? {
// dial opens an ssl connection on hostname:port
pub fn (mut s SSLConn) dial(hostname string, port int) ? {
s.owns_socket = true
if s.opened {
return error('ssl connection already open')
}