| Index: content/browser/renderer_host/websocket_host.h
|
| diff --git a/content/browser/renderer_host/websocket_host.h b/content/browser/renderer_host/websocket_host.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3488ef06b522de65228c176045d349126273952b
|
| --- /dev/null
|
| +++ b/content/browser/renderer_host/websocket_host.h
|
| @@ -0,0 +1,126 @@
|
| +// 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.
|
| +
|
| +#ifndef CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
|
| +#define CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
|
| +
|
| +#include <stdint.h>
|
| +
|
| +#include <memory>
|
| +#include <string>
|
| +#include <vector>
|
| +
|
| +#include "base/macros.h"
|
| +#include "base/memory/weak_ptr.h"
|
| +#include "base/time/time.h"
|
| +#include "content/common/content_export.h"
|
| +#include "content/common/websocket.h"
|
| +
|
| +class GURL;
|
| +struct WebSocketHostMsg_AddChannelRequest_Params;
|
| +
|
| +namespace url {
|
| +class Origin;
|
| +} // namespace url
|
| +
|
| +namespace net {
|
| +class WebSocketChannel;
|
| +class URLRequestContext;
|
| +} // namespace net
|
| +
|
| +namespace IPC {
|
| +class Message;
|
| +} // namespace IPC
|
| +
|
| +namespace content {
|
| +
|
| +class WebSocketBlobSender;
|
| +class WebSocketDispatcherHost;
|
| +
|
| +// Host of net::WebSocketChannel. The lifetime of an instance of this class is
|
| +// completely controlled by the WebSocketDispatcherHost object.
|
| +class CONTENT_EXPORT WebSocketHost {
|
| + public:
|
| + WebSocketHost(int routing_id,
|
| + WebSocketDispatcherHost* dispatcher,
|
| + net::URLRequestContext* url_request_context,
|
| + base::TimeDelta delay);
|
| + virtual ~WebSocketHost();
|
| +
|
| + // The renderer process is going away.
|
| + // This function is virtual for testing.
|
| + virtual void GoAway();
|
| +
|
| + // General message dispatch. WebSocketDispatcherHost::OnMessageReceived
|
| + // delegates to this method after looking up the |routing_id|.
|
| + virtual bool OnMessageReceived(const IPC::Message& message);
|
| +
|
| + int routing_id() const { return routing_id_; }
|
| +
|
| + bool handshake_succeeded() const { return handshake_succeeded_; }
|
| + void OnHandshakeSucceeded() { handshake_succeeded_ = true; }
|
| +
|
| + private:
|
| + class WebSocketEventHandler;
|
| +
|
| + // Handlers for each message type, dispatched by OnMessageReceived(), as
|
| + // defined in content/common/websocket_messages.h
|
| +
|
| + void OnAddChannelRequest(
|
| + const WebSocketHostMsg_AddChannelRequest_Params& request);
|
| +
|
| + void AddChannel(const GURL& socket_url,
|
| + const std::vector<std::string>& requested_protocols,
|
| + const url::Origin& origin,
|
| + const GURL& first_party_for_cookies,
|
| + const std::string& user_agent_override,
|
| + int render_frame_id);
|
| +
|
| + void OnSendBlob(const std::string& uuid, uint64_t expected_size);
|
| +
|
| + void OnSendFrame(bool fin,
|
| + WebSocketMessageType type,
|
| + const std::vector<char>& data);
|
| +
|
| + void OnFlowControl(int64_t quota);
|
| +
|
| + void OnDropChannel(bool was_clean, uint16_t code, const std::string& reason);
|
| +
|
| + void BlobSendComplete(int result);
|
| +
|
| + // non-NULL if and only if this object is currently in "blob sending mode".
|
| + std::unique_ptr<WebSocketBlobSender> blob_sender_;
|
| +
|
| + // The channel we use to send events to the network.
|
| + std::unique_ptr<net::WebSocketChannel> channel_;
|
| +
|
| + // The WebSocketHostDispatcher that created this object.
|
| + WebSocketDispatcherHost* const dispatcher_;
|
| +
|
| + // The URL request context for the channel.
|
| + net::URLRequestContext* const url_request_context_;
|
| +
|
| + // The ID used to route messages.
|
| + const int routing_id_;
|
| +
|
| + // Delay used for per-renderer WebSocket throttling.
|
| + base::TimeDelta delay_;
|
| +
|
| + // SendFlowControl() is delayed when OnFlowControl() is called before
|
| + // AddChannel() is called.
|
| + // Zero indicates there is no pending SendFlowControl().
|
| + int64_t pending_flow_control_quota_;
|
| +
|
| + // handshake_succeeded_ is set and used by WebSocketDispatcherHost
|
| + // to manage counters for per-renderer WebSocket throttling.
|
| + bool handshake_succeeded_;
|
| +
|
| + base::WeakPtrFactory<WebSocketHost> weak_ptr_factory_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(WebSocketHost);
|
| +};
|
| +
|
| +} // namespace content
|
| +
|
| +#endif // CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
|
|
|