blob: 2e40191d9029f403b00754795a26d214bbb414e2 [file] [log] [blame]
// Copyright 2018 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.
#ifndef NET_WEBSOCKETS_WEBSOCKET_HTTP2_HANDSHAKE_STREAM_H_
#define NET_WEBSOCKETS_WEBSOCKET_HTTP2_HANDSHAKE_STREAM_H_
#include <stdint.h>
#include <memory>
#include <string>
#include <vector>
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "net/base/completion_once_callback.h"
#include "net/base/io_buffer.h"
#include "net/base/net_export.h"
#include "net/base/request_priority.h"
#include "net/log/net_log_with_source.h"
#include "net/third_party/quiche/src/spdy/core/spdy_header_block.h"
#include "net/websockets/websocket_basic_stream_adapters.h"
#include "net/websockets/websocket_handshake_stream_base.h"
#include "net/websockets/websocket_stream.h"
namespace net {
struct LoadTimingInfo;
class SSLInfo;
class IOBuffer;
class SSLCertRequestInfo;
class IPEndPoint;
class HttpNetworkSession;
struct NetErrorDetails;
class HttpStream;
class HttpResponseHeaders;
struct HttpRequestInfo;
class HttpResponseInfo;
class SpdySession;
struct AlternativeService;
class SpdyStreamRequest;
struct WebSocketExtensionParams;
class NET_EXPORT_PRIVATE WebSocketHttp2HandshakeStream
: public WebSocketHandshakeStreamBase,
public WebSocketSpdyStreamAdapter::Delegate {
public:
// |connect_delegate| and |request| must out-live this object.
WebSocketHttp2HandshakeStream(
base::WeakPtr<SpdySession> session,
WebSocketStream::ConnectDelegate* connect_delegate,
std::vector<std::string> requested_sub_protocols,
std::vector<std::string> requested_extensions,
WebSocketStreamRequestAPI* request);
~WebSocketHttp2HandshakeStream() override;
// HttpStream methods.
int InitializeStream(const HttpRequestInfo* request_info,
bool can_send_early,
RequestPriority priority,
const NetLogWithSource& net_log,
CompletionOnceCallback callback) override;
int SendRequest(const HttpRequestHeaders& request_headers,
HttpResponseInfo* response,
CompletionOnceCallback callback) override;
int ReadResponseHeaders(CompletionOnceCallback callback) override;
int ReadResponseBody(IOBuffer* buf,
int buf_len,
CompletionOnceCallback callback) override;
void Close(bool not_reusable) override;
bool IsResponseBodyComplete() const override;
bool IsConnectionReused() const override;
void SetConnectionReused() override;
bool CanReuseConnection() const override;
int64_t GetTotalReceivedBytes() const override;
int64_t GetTotalSentBytes() const override;
bool GetAlternativeService(
AlternativeService* alternative_service) const override;
bool GetLoadTimingInfo(LoadTimingInfo* load_timing_info) const override;
void GetSSLInfo(SSLInfo* ssl_info) override;
void GetSSLCertRequestInfo(SSLCertRequestInfo* cert_request_info) override;
bool GetRemoteEndpoint(IPEndPoint* endpoint) override;
void Drain(HttpNetworkSession* session) override;
void SetPriority(RequestPriority priority) override;
void PopulateNetErrorDetails(NetErrorDetails* details) override;
HttpStream* RenewStreamForAuth() override;
// WebSocketHandshakeStreamBase methods.
// This is called from the top level once correct handshake response headers
// have been received. It creates an appropriate subclass of WebSocketStream
// depending on what extensions were negotiated. This object is unusable after
// Upgrade() has been called and should be disposed of as soon as possible.
std::unique_ptr<WebSocketStream> Upgrade() override;
base::WeakPtr<WebSocketHandshakeStreamBase> GetWeakPtr() override;
// WebSocketSpdyStreamAdapter::Delegate methods.
void OnHeadersSent() override;
void OnHeadersReceived(
const spdy::SpdyHeaderBlock& response_headers) override;
void OnClose(int status) override;
// Called by |spdy_stream_request_| when requested stream is ready.
void StartRequestCallback(int rv);
private:
// Validates the response and sends the finished handshake event.
int ValidateResponse();
// Check that the headers are well-formed and have a 200 status code,
// in which case returns OK, otherwise returns ERR_INVALID_RESPONSE.
int ValidateUpgradeResponse(const HttpResponseHeaders* headers);
void OnFailure(const std::string& message);
HandshakeResult result_;
// The connection to open the Websocket stream on.
base::WeakPtr<SpdySession> session_;
// Owned by another object.
// |connect_delegate| will live during the lifetime of this object.
WebSocketStream::ConnectDelegate* const connect_delegate_;
HttpResponseInfo* http_response_info_;
spdy::SpdyHeaderBlock http2_request_headers_;
// The sub-protocols we requested.
std::vector<std::string> requested_sub_protocols_;
// The extensions we requested.
std::vector<std::string> requested_extensions_;
WebSocketStreamRequestAPI* const stream_request_;
const HttpRequestInfo* request_info_;
RequestPriority priority_;
NetLogWithSource net_log_;
// SpdyStreamRequest that will create the stream.
std::unique_ptr<SpdyStreamRequest> spdy_stream_request_;
// SpdyStream corresponding to the request.
base::WeakPtr<SpdyStream> stream_;
// WebSocketSpdyStreamAdapter holding a WeakPtr to |stream_|.
// This can be passed on to WebSocketBasicStream when created.
std::unique_ptr<WebSocketSpdyStreamAdapter> stream_adapter_;
// True if |stream_| has been created then closed.
bool stream_closed_;
// The error code corresponding to the reason for closing the stream.
// Only meaningful if |stream_closed_| is true.
int stream_error_;
// True if complete response headers have been received.
bool response_headers_complete_;
// Save callback provided in asynchronous HttpStream methods.
CompletionOnceCallback callback_;
// The sub-protocol selected by the server.
std::string sub_protocol_;
// The extension(s) selected by the server.
std::string extensions_;
// The extension parameters. The class is defined in the implementation file
// to avoid including extension-related header files here.
std::unique_ptr<WebSocketExtensionParams> extension_params_;
base::WeakPtrFactory<WebSocketHttp2HandshakeStream> weak_ptr_factory_{this};
DISALLOW_COPY_AND_ASSIGN(WebSocketHttp2HandshakeStream);
};
} // namespace net
#endif // NET_WEBSOCKETS_WEBSOCKET_HTTP2_HANDSHAKE_STREAM_H_