content/child/websocket_bridge.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_CHILD_WEBSOCKET_BRIDGE_H_
   6 #define CONTENT_CHILD_WEBSOCKET_BRIDGE_H_
   7
   8 #include <stdint.h>
   9 #include <string>
  10 #include <vector>
  11
  12 #include "base/basictypes.h"
  13 #include "content/common/websocket.h"
  14 #include "ipc/ipc_message.h"
  15 #include "third_party/WebKit/public/platform/WebSocketHandle.h"
  16 #include "third_party/WebKit/public/platform/WebVector.h"
  17
  18 namespace blink {
  19 class WebSecurityOrigin;
  20 class WebString;
  21 class WebURL;
  22 }  // namespace blink
  23
  24 namespace content {
  25
  26 class WebSocketBridge : public blink::WebSocketHandle {
  27  public:
  28   WebSocketBridge();
  29
  30   // Handles an IPC message from the browser process.
  31   bool OnMessageReceived(const IPC::Message& message);
  32
  33   // WebSocketHandle functions.
  34   void connect(const blink::WebURL& url,
  35                const blink::WebVector<blink::WebString>& protocols,
  36                const blink::WebSecurityOrigin& origin,
  37                blink::WebSocketHandleClient* client) override;
  38   void send(bool fin,
  39             WebSocketHandle::MessageType type,
  40             const char* data,
  41             size_t size) override;
  42   void flowControl(int64_t quota) override;
  43   void close(unsigned short code, const blink::WebString& reason) override;
  44
  45   void Disconnect();
  46
  47   void set_render_frame_id(int id) {
  48     render_frame_id_ = id;
  49   }
  50
  51  private:
  52   ~WebSocketBridge() override;
  53
  54   void DidConnect(const std::string& selected_protocol,
  55                   const std::string& extensions);
  56   void DidStartOpeningHandshake(const WebSocketHandshakeRequest& request);
  57   void DidFinishOpeningHandshake(const WebSocketHandshakeResponse& response);
  58   void DidFail(const std::string& message);
  59   void DidReceiveData(bool fin,
  60                       WebSocketMessageType type,
  61                       const std::vector<char>& data);
  62   void DidReceiveFlowControl(int64_t quota);
  63   void DidClose(bool was_clean, unsigned short code, const std::string& reason);
  64   void DidStartClosingHandshake();
  65
  66   int channel_id_;
  67   int render_frame_id_;
  68   blink::WebSocketHandleClient* client_;
  69
  70   static const int kInvalidChannelId = -1;
  71 };
  72
  73 }  // namespace content
  74
  75 #endif  // CONTENT_CHILD_WEBSOCKET_BRIDGE_H_