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() } }