// Copyright 2013 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "net/websockets/websocket_handshake_stream_create_helper.h" #include #include #include "base/check.h" #include "base/memory/weak_ptr.h" #include "net/socket/client_socket_handle.h" #include "net/websockets/websocket_basic_handshake_stream.h" #include "net/websockets/websocket_http2_handshake_stream.h" namespace net { WebSocketHandshakeStreamCreateHelper::WebSocketHandshakeStreamCreateHelper( WebSocketStream::ConnectDelegate* connect_delegate, const std::vector& requested_subprotocols, WebSocketStreamRequestAPI* request) : connect_delegate_(connect_delegate), requested_subprotocols_(requested_subprotocols), request_(request) { DCHECK(connect_delegate_); DCHECK(request_); } WebSocketHandshakeStreamCreateHelper::~WebSocketHandshakeStreamCreateHelper() = default; std::unique_ptr WebSocketHandshakeStreamCreateHelper::CreateBasicStream( std::unique_ptr connection, bool using_proxy, WebSocketEndpointLockManager* websocket_endpoint_lock_manager) { // The list of supported extensions and parameters is hard-coded. // TODO(ricea): If more extensions are added, consider a more flexible // method. std::vector extensions( 1, "permessage-deflate; client_max_window_bits"); auto stream = std::make_unique( std::move(connection), connect_delegate_, using_proxy, requested_subprotocols_, extensions, request_, websocket_endpoint_lock_manager); request_->OnBasicHandshakeStreamCreated(stream.get()); return stream; } std::unique_ptr WebSocketHandshakeStreamCreateHelper::CreateHttp2Stream( base::WeakPtr session, std::set dns_aliases) { std::vector extensions( 1, "permessage-deflate; client_max_window_bits"); auto stream = std::make_unique( session, connect_delegate_, requested_subprotocols_, extensions, request_, std::move(dns_aliases)); request_->OnHttp2HandshakeStreamCreated(stream.get()); return stream; } } // namespace net