• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 
set_render_frame_id(int id)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