Class: Async::WebSocket::UpgradeRequest

Inherits:
Protocol::HTTP::Request
  • Object
show all
Includes:
Protocol::WebSocket::Headers
Defined in:
lib/async/websocket/upgrade_request.rb

Overview

This is required for HTTP/1.x to upgrade the connection to the WebSocket protocol. See tools.ietf.org/html/rfc6455

Defined Under Namespace

Classes: Wrapper

Instance Method Summary collapse

Constructor Details

#initialize(request, protocols: [], version: 13, &block) ⇒ UpgradeRequest

Returns a new instance of UpgradeRequest.



57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
# File 'lib/async/websocket/upgrade_request.rb', line 57

def initialize(request, protocols: [], version: 13, &block)
	@key = Nounce.generate_key
	
	headers = ::Protocol::HTTP::Headers[request.headers]
	
	headers.add(SEC_WEBSOCKET_KEY, @key)
	headers.add(SEC_WEBSOCKET_VERSION, String(version))
	
	if protocols.any?
		headers.add(SEC_WEBSOCKET_PROTOCOL, protocols.join(","))
	end
	
	super(request.scheme, request.authority, ::Protocol::HTTP::Methods::GET, request.path, nil, headers, nil, PROTOCOL)
end

Instance Method Details

#call(connection) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/async/websocket/upgrade_request.rb', line 72

def call(connection)
	response = super
	
	if accept_digest = response.headers[SEC_WEBSOCKET_ACCEPT]&.first
		expected_accept_digest = Nounce.accept_digest(@key)
		
		unless accept_digest and accept_digest == expected_accept_digest
			raise ProtocolError, "Invalid accept digest, expected #{expected_accept_digest.inspect}, got #{accept_digest.inspect}!"
		end
	end
	
	verified = accept_digest && Array(response.protocol).map(&:downcase) == %w(websocket) && response.headers["connection"]&.include?("upgrade")
	
	return Wrapper.new(response, verified: verified)
end