Home | History | Annotate | Download | only in child 
      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 WebSerializedOrigin;
     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   virtual void connect(const blink::WebURL& url,
     35                        const blink::WebVector<blink::WebString>& protocols,
     36                        const blink::WebSerializedOrigin& origin,
     37                        blink::WebSocketHandleClient* client) OVERRIDE;
     38   virtual void send(bool fin,
     39                     WebSocketHandle::MessageType type,
     40                     const char* data,
     41                     size_t size) OVERRIDE;
     42   virtual void flowControl(int64_t quota) OVERRIDE;
     43   virtual void close(unsigned short code,
     44                      const blink::WebString& reason) OVERRIDE;
     45 
     46   virtual void Disconnect();
     47 
     48   void set_render_frame_id(int id) {
     49     render_frame_id_ = id;
     50   }
     51 
     52  private:
     53   virtual ~WebSocketBridge();
     54 
     55   void DidConnect(bool fail,
     56                   const std::string& selected_protocol,
     57                   const std::string& extensions);
     58   void DidStartOpeningHandshake(const WebSocketHandshakeRequest& request);
     59   void DidFinishOpeningHandshake(const WebSocketHandshakeResponse& response);
     60   void DidFail(const std::string& message);
     61   void DidReceiveData(bool fin,
     62                       WebSocketMessageType type,
     63                       const std::vector<char>& data);
     64   void DidReceiveFlowControl(int64_t quota);
     65   void DidClose(bool was_clean, unsigned short code, const std::string& reason);
     66   void DidStartClosingHandshake();
     67 
     68   int channel_id_;
     69   int render_frame_id_;
     70   blink::WebSocketHandleClient* client_;
     71 
     72   static const int kInvalidChannelId = -1;
     73 };
     74 
     75 }  // namespace content
     76 
     77 #endif  // CONTENT_CHILD_WEBSOCKET_BRIDGE_H_
     78