很早之前就对 WebSocket 协定十分感兴趣,明天有空时看了一下 RFC6455, 发现其实是一个很简略的协定。于是尝试着实现了一个客户端。这里摘取一些要害局部的代码。
WebSocket 和一般的 tcp 连贯很相似,能够双向发送音讯(区别于 http的request-response模式)。
首先第一步是建设 tcp 连贯,而后发送 http 协定降级音讯:
defp upgrade_msg(uri, nonce) do """ GET / HTTP/1.1\r Host: #{uri.authority}\r Upgrade: websocket\r Connection: Upgrade\r Sec-WebSocket-Key: #{nonce}\r Sec-WebSocket-Version: 13\r\n """ end
之后服务器会返回一些内容,咱们校验过后websocket连贯就算建设胜利了:
defp handle_tcp(:handshake, data, %{challenge: challenge}) do {:ok, {:http_response, _, 101, "Switching Protocols"}, rest} = :erlang.decode_packet(:http_bin, data, []) |> IO.inspect() case validate_headers(rest, fn {:Connection, up} -> String.downcase(up) == "upgrade" {:Upgrade, ws} -> String.downcase(ws) == "websocket" {"Sec-WebSocket-Accept", ch} -> ch == challenge _ -> true end) do :ok -> IO.inspect("goto data_framing") {:data_framing, ""} {:error, wrong_header} -> IO.inspect("falied because: #{inspect(wrong_header)}") {:failed, :close} end end
连贯建设之后就能够以特定的格局收发音讯了,音讯的最小单位是 frame,它的构造是这样的:
def encode(%{opcode: op, mask: mask} = meta) do op = enop(op) payload = if mask, do: meta.masked_payload, else: meta.payload mask_key = if mask, do: meta.mask_key, else: <<>> mask = if mask, do: 1, else: 0 <<1::size(1), 0::size(3), op::size(4), mask::size(1), encode_payload_length(byte_size(payload))::bitstring, mask_key::bytes, payload::bytes>> end
客户端发送给服务器的内容须要 mask,而服务端发给客户端的则不须要。
opcode 罕用的有 text,binary,ping,pong,close. 别离代表不同的音讯类型。