mirror/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
vlib/net/mbedtls/ssl_connection.v
View File

@ -37,6 +37,8 @@ mut:
handle int handle int
duration time.Duration duration time.Duration
opened bool opened bool
owns_socket bool
} }
[params] [params]
@ -77,6 +79,15 @@ pub fn (mut s SSLConn) shutdown() ? {
} }
C.mbedtls_ssl_free(&s.ssl) C.mbedtls_ssl_free(&s.ssl)
C.mbedtls_ssl_config_free(&s.conf) 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 // 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 // dial opens an ssl connection on hostname:port
pub fn (mut s SSLConn) dial(hostname string, port int) ? { pub fn (mut s SSLConn) dial(hostname string, port int) ? {
s.owns_socket = true
if s.opened { if s.opened {
return error('ssl connection already open') return error('ssl connection already open')
} }