Method: WebSocket#handshake

Defined in:
lib/web/web-socket-ruby/lib/web_socket.rb

#handshake(status = nil, header = {}) ⇒ Object



135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
# File 'lib/web/web-socket-ruby/lib/web_socket.rb', line 135

def handshake(status = nil, header = {})
  if @handshaked
    raise(WebSocket::Error, "handshake has already been done")
  end
  status ||= "101 Switching Protocols"
  def_header = {}
  case @web_socket_version
    when "hixie-75"
      def_header["WebSocket-Origin"] = self.origin
      def_header["WebSocket-Location"] = self.location
      extra_bytes = ""
    when "hixie-76"
      def_header["Sec-WebSocket-Origin"] = self.origin
      def_header["Sec-WebSocket-Location"] = self.location
      extra_bytes = hixie_76_security_digest(
        @header["Sec-WebSocket-Key1"], @header["Sec-WebSocket-Key2"], @key3)
    else
      def_header["Sec-WebSocket-Accept"] = security_digest(@header["sec-websocket-key"])
      extra_bytes = ""
  end
  header = def_header.merge(header)
  header_str = header.map(){ |k, v| "#{k}: #{v}\r\n" }.join("")
  # Note that Upgrade and Connection must appear in this order.
  write(
    "HTTP/1.1 #{status}\r\n" +
    "Upgrade: websocket\r\n" +
    "Connection: Upgrade\r\n" +
    "#{header_str}\r\n#{extra_bytes}")
  flush()
  @handshaked = true
end