content/browser/renderer_host/websocket_host.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
   6 #define CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/memory/scoped_ptr.h"
  12 #include "content/common/content_export.h"
  13 #include "content/common/websocket.h"
  14
  15 class GURL;
  16
  17 namespace url {
  18 class Origin;
  19 }  // namespace url
  20
  21 namespace net {
  22 class WebSocketChannel;
  23 class URLRequestContext;
  24 }  // namespace net
  25
  26 namespace IPC {
  27 class Message;
  28 }  // namespace IPC
  29
  30 namespace content {
  31
  32 class WebSocketDispatcherHost;
  33
  34 // Host of net::WebSocketChannel. The lifetime of an instance of this class is
  35 // completely controlled by the WebSocketDispatcherHost object.
  36 class CONTENT_EXPORT WebSocketHost {
  37  public:
  38   WebSocketHost(int routing_id,
  39                 WebSocketDispatcherHost* dispatcher,
  40                 net::URLRequestContext* url_request_context);
  41   virtual ~WebSocketHost();
  42
  43   // General message dispatch. WebSocketDispatcherHost::OnMessageReceived
  44   // delegates to this method after looking up the |routing_id|.
  45   virtual bool OnMessageReceived(const IPC::Message& message);
  46
  47  private:
  48   // Handlers for each message type, dispatched by OnMessageReceived(), as
  49   // defined in content/common/websocket_messages.h
  50
  51   void OnAddChannelRequest(const GURL& socket_url,
  52                            const std::vector<std::string>& requested_protocols,
  53                            const url::Origin& origin,
  54                            int render_frame_id);
  55
  56   void OnSendFrame(bool fin,
  57                    WebSocketMessageType type,
  58                    const std::vector<char>& data);
  59
  60   void OnFlowControl(int64 quota);
  61
  62   void OnDropChannel(bool was_clean, uint16 code, const std::string& reason);
  63
  64   // The channel we use to send events to the network.
  65   scoped_ptr<net::WebSocketChannel> channel_;
  66
  67   // The WebSocketHostDispatcher that created this object.
  68   WebSocketDispatcherHost* const dispatcher_;
  69
  70   // The URL request context for the channel.
  71   net::URLRequestContext* const url_request_context_;
  72
  73   // The ID used to route messages.
  74   const int routing_id_;
  75
  76   DISALLOW_COPY_AND_ASSIGN(WebSocketHost);
  77 };
  78
  79 }  // namespace content
  80
  81 #endif  // CONTENT_BROWSER_RENDERER_HOST_WEBSOCKET_HOST_H_