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.socket: windows fixes and test

Browse Source 
* added missing ai_canonname in addrinfo structure
* ai_canonname, ai_addr and ai_addrlen must be zeroed before getaddrinfo call
* write() must not be used on windows sockets
* added (Socket).get_port() function which is useful when socket was initialized with 0 (random) port
* tests is fixed, test server started listening on random port to avoid conflicts
This commit is contained in:
vitalyster 2019-09-20 17:04:39 +03:00 committed by Alexander Medvednikov
parent f1bb25bfaf
commit a42b4e219c
2 changed files with 40 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vlib/net/socket.v
View File

@ -45,9 +45,10 @@ mut:
ai_socktype int
ai_flags int
ai_protocol int
ai_addrlen int
ai_next voidptr
ai_addrlen int
ai_addr voidptr
ai_canonname voidptr
ai_next voidptr
}
struct C.sockaddr_storage {}
@ -175,6 +176,10 @@ pub fn (s Socket) connect(address string, port int) ?int {
hints.ai_family = C.AF_UNSPEC
hints.ai_socktype = C.SOCK_STREAM
hints.ai_flags = C.AI_PASSIVE
hints.ai_addrlen = 0
hints.ai_canonname = C.NULL
hints.ai_addr = C.NULL
info := &C.addrinfo{!}
sport := '$port'
@ -263,7 +268,7 @@ const (
)
pub fn (s Socket) write(str string) {
line := '$str\r\n'
C.write(s.sockfd, line.str, line.len)
C.send(s.sockfd, line.str, line.len, 0)
}
pub fn (s Socket) read_line() string {
@ -306,4 +311,11 @@ pub fn (s Socket) read_line() string {
return res
}
pub fn (s Socket) get_port() int {
mut addr := C.sockaddr_in {}
size := 16 // sizeof(sockaddr_in)
sockname_res := C.getsockname(s.sockfd, &addr, &size)
return int(C.ntohs(addr.sin_port))
}

51
vlib/net/socket_test.v
View File

@ -1,30 +1,29 @@
import net
fn test_socket() {
// server := net.listen(8080) or {
// println(err)
// return
// }
// println(server)
// client := net.dial('127.0.0.1', 8080) or {
// println(err)
// return
// }
// println(client)
// socket := server.accept() or {
// println(err)
// return
// }
// println(socket)
//
// message := 'Hello World'
// socket.send(message.str, message.len)
// println('Sent: ' + message)
//
// bytes := client.recv(1024)
// println('Received: ' + tos(bytes, message.len))
//
// server.close()
// client.close()
// socket.close()
mut server := net.listen(0) or {
println(err)
return
}
server_port := server.get_port()
mut client := net.dial('127.0.0.1', server_port) or {
println(err)
return
}
mut socket := server.accept() or {
println(err)
return
}
message := 'Hello World'
socket.send(message.str, message.len)
bytes := client.recv(1024)
received := tos(bytes, message.len)
assert message == received
server.close()
client.close()
socket.close()
}