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