blob: cfb2e4c16007a5677e0bbe8dd4e8dbc0e03ad81b [file] [log] [blame]
// 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 <utility>
#include "base/logging.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<std::string>& requested_subprotocols)
: requested_subprotocols_(requested_subprotocols),
connect_delegate_(connect_delegate),
request_(nullptr) {
DCHECK(connect_delegate_);
}
WebSocketHandshakeStreamCreateHelper::~WebSocketHandshakeStreamCreateHelper() =
default;
std::unique_ptr<WebSocketHandshakeStreamBase>
WebSocketHandshakeStreamCreateHelper::CreateBasicStream(
std::unique_ptr<ClientSocketHandle> connection,
bool using_proxy,
WebSocketEndpointLockManager* websocket_endpoint_lock_manager) {
DCHECK(request_) << "set_request() must be called";
// The list of supported extensions and parameters is hard-coded.
// TODO(ricea): If more extensions are added, consider a more flexible
// method.
std::vector<std::string> extensions(
1, "permessage-deflate; client_max_window_bits");
auto stream = std::make_unique<WebSocketBasicHandshakeStream>(
std::move(connection), connect_delegate_, using_proxy,
requested_subprotocols_, extensions, request_,
websocket_endpoint_lock_manager);
request_->OnBasicHandshakeStreamCreated(stream.get());
return std::move(stream);
}
std::unique_ptr<WebSocketHandshakeStreamBase>
WebSocketHandshakeStreamCreateHelper::CreateHttp2Stream(
base::WeakPtr<SpdySession> session) {
DCHECK(request_) << "set_request() must be called";
std::vector<std::string> extensions(
1, "permessage-deflate; client_max_window_bits");
auto stream = std::make_unique<WebSocketHttp2HandshakeStream>(
session, connect_delegate_, requested_subprotocols_, extensions,
request_);
request_->OnHttp2HandshakeStreamCreated(stream.get());
#if defined(STARBOARD)
return std::unique_ptr<WebSocketHandshakeStreamBase>(stream.release());
#else
return stream;
#endif
}
} // namespace net