mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
websocket: move library files to root of module
This commit is contained in:
72
vlib/net/websocket/handshake.v
Normal file
72
vlib/net/websocket/handshake.v
Normal file
@ -0,0 +1,72 @@
|
||||
module websocket
|
||||
|
||||
fn (ws mut Client) read_handshake(seckey string){
|
||||
l.d("reading handshake...")
|
||||
mut bytes_read := 0
|
||||
max_buffer := 256
|
||||
buffer_size := 1
|
||||
mut buffer := malloc(max_buffer)
|
||||
|
||||
for bytes_read <= max_buffer {
|
||||
res := ws.read_from_server(buffer + bytes_read, buffer_size)
|
||||
if res == 0 || res == -1 {
|
||||
l.f("read_handshake: Failed to read handshake.")
|
||||
}
|
||||
if buffer[bytes_read] == `\n` && buffer[bytes_read-1] == `\r` && buffer[bytes_read-2] == `\n` && buffer[bytes_read-3] == `\r` {
|
||||
break
|
||||
}
|
||||
bytes_read += buffer_size
|
||||
}
|
||||
buffer[max_buffer+1] = `\0`
|
||||
ws.handshake_handler(string(byteptr(buffer)), seckey)
|
||||
}
|
||||
|
||||
fn (ws mut Client) handshake_handler(handshake_response, seckey string){
|
||||
l.d("handshake_handler:\r\n${handshake_response}")
|
||||
lines := handshake_response.split_into_lines()
|
||||
|
||||
header := lines[0]
|
||||
if !header.starts_with("HTTP/1.1 101") && !header.starts_with("HTTP/1.0 101") {
|
||||
l.f("handshake_handler: invalid HTTP status response code")
|
||||
}
|
||||
|
||||
for i in 1..lines.len {
|
||||
if lines[i].len <= 0 || lines[i] == "\r\n" {
|
||||
continue
|
||||
}
|
||||
keys := lines[i].split(":")
|
||||
|
||||
match keys[0] {
|
||||
"Upgrade" {
|
||||
ws.flags << Flag.has_upgrade
|
||||
}
|
||||
"Connection" {
|
||||
ws.flags << Flag.has_connection
|
||||
}
|
||||
"Sec-WebSocket-Accept" {
|
||||
l.d("comparing hashes")
|
||||
response := create_key_challenge_response(seckey)
|
||||
if keys[1].trim_space() != response {
|
||||
l.e("handshake_handler: Sec-WebSocket-Accept header does not match computed sha1/base64 response.")
|
||||
}
|
||||
ws.flags << Flag.has_accept
|
||||
unsafe {
|
||||
response.free()
|
||||
}
|
||||
} else {}
|
||||
}
|
||||
unsafe {
|
||||
keys.free()
|
||||
}
|
||||
}
|
||||
if ws.flags.len < 3 {
|
||||
ws.close(1002, "invalid websocket HTTP headers")
|
||||
l.e("invalid websocket HTTP headers")
|
||||
}
|
||||
l.i("handshake successful!")
|
||||
unsafe {
|
||||
handshake_response.free()
|
||||
lines.free()
|
||||
header.free()
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user