| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/child/socket_stream_dispatcher.h" | 5 #include "content/child/socket_stream_dispatcher.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/id_map.h" | 10 #include "base/id_map.h" |
| 11 #include "base/lazy_instance.h" | 11 #include "base/lazy_instance.h" |
| 12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 13 #include "base/message_loop/message_loop.h" | 13 #include "base/message_loop/message_loop.h" |
| 14 #include "base/strings/string16.h" | 14 #include "base/strings/string16.h" |
| 15 #include "base/strings/utf_string_conversions.h" | 15 #include "base/strings/utf_string_conversions.h" |
| 16 #include "content/child/child_thread.h" | 16 #include "content/child/child_thread.h" |
| 17 #include "content/child/web_socket_stream_handle_bridge.h" | |
| 18 #include "content/child/web_socket_stream_handle_delegate.h" | |
| 19 #include "content/child/web_socket_stream_handle_impl.h" | |
| 20 #include "content/common/socket_stream.h" | 17 #include "content/common/socket_stream.h" |
| 21 #include "content/common/socket_stream_handle_data.h" | 18 #include "content/common/socket_stream_handle_data.h" |
| 22 #include "content/common/socket_stream_messages.h" | 19 #include "content/common/socket_stream_messages.h" |
| 23 #include "net/base/net_errors.h" | 20 #include "net/base/net_errors.h" |
| 24 #include "url/gurl.h" | 21 #include "url/gurl.h" |
| 22 #include "webkit/child/websocketstreamhandle_bridge.h" |
| 23 #include "webkit/child/websocketstreamhandle_delegate.h" |
| 25 | 24 |
| 26 namespace content { | 25 namespace content { |
| 27 | 26 |
| 28 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. | 27 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. |
| 29 // It communicates with the main browser process via SocketStreamDispatcher. | 28 // It communicates with the main browser process via SocketStreamDispatcher. |
| 30 class IPCWebSocketStreamHandleBridge : public WebSocketStreamHandleBridge { | 29 class IPCWebSocketStreamHandleBridge |
| 30 : public webkit_glue::WebSocketStreamHandleBridge { |
| 31 public: | 31 public: |
| 32 IPCWebSocketStreamHandleBridge(blink::WebSocketStreamHandle* handle, | 32 IPCWebSocketStreamHandleBridge( |
| 33 WebSocketStreamHandleDelegate* delegate) | 33 blink::WebSocketStreamHandle* handle, |
| 34 : socket_id_(kNoSocketId), handle_(handle), delegate_(delegate) {} | 34 webkit_glue::WebSocketStreamHandleDelegate* delegate) |
| 35 : socket_id_(kNoSocketId), |
| 36 handle_(handle), |
| 37 delegate_(delegate) {} |
| 35 | 38 |
| 36 // Returns the handle having given id or NULL if there is no such handle. | 39 // Returns the handle having given id or NULL if there is no such handle. |
| 37 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); | 40 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); |
| 38 | 41 |
| 39 // WebSocketStreamHandleBridge methods. | 42 // webkit_glue::WebSocketStreamHandleBridge methods. |
| 40 virtual void Connect(const GURL& url) OVERRIDE; | 43 virtual void Connect(const GURL& url) OVERRIDE; |
| 41 virtual bool Send(const std::vector<char>& data) OVERRIDE; | 44 virtual bool Send(const std::vector<char>& data) OVERRIDE; |
| 42 virtual void Close() OVERRIDE; | 45 virtual void Close() OVERRIDE; |
| 43 | 46 |
| 44 // Called by SocketStreamDispatcher. | 47 // Called by SocketStreamDispatcher. |
| 45 void OnConnected(int max_amount_send_allowed); | 48 void OnConnected(int max_amount_send_allowed); |
| 46 void OnSentData(int amount_sent); | 49 void OnSentData(int amount_sent); |
| 47 void OnReceivedData(const std::vector<char>& data); | 50 void OnReceivedData(const std::vector<char>& data); |
| 48 void OnClosed(); | 51 void OnClosed(); |
| 49 void OnFailed(int error_code, const char* error_msg); | 52 void OnFailed(int error_code, const char* error_msg); |
| 50 | 53 |
| 51 private: | 54 private: |
| 52 virtual ~IPCWebSocketStreamHandleBridge(); | 55 virtual ~IPCWebSocketStreamHandleBridge(); |
| 53 | 56 |
| 54 // The ID for this bridge and corresponding SocketStream instance in the | 57 // The ID for this bridge and corresponding SocketStream instance in the |
| 55 // browser process. | 58 // browser process. |
| 56 int socket_id_; | 59 int socket_id_; |
| 57 | 60 |
| 58 blink::WebSocketStreamHandle* handle_; | 61 blink::WebSocketStreamHandle* handle_; |
| 59 WebSocketStreamHandleDelegate* delegate_; | 62 webkit_glue::WebSocketStreamHandleDelegate* delegate_; |
| 60 | 63 |
| 61 // Map from ID to bridge instance. | 64 // Map from ID to bridge instance. |
| 62 static base::LazyInstance<IDMap<IPCWebSocketStreamHandleBridge> >::Leaky | 65 static base::LazyInstance<IDMap<IPCWebSocketStreamHandleBridge> >::Leaky |
| 63 all_bridges; | 66 all_bridges; |
| 64 }; | 67 }; |
| 65 | 68 |
| 66 // static | 69 // static |
| 67 base::LazyInstance<IDMap<IPCWebSocketStreamHandleBridge> >::Leaky | 70 base::LazyInstance<IDMap<IPCWebSocketStreamHandleBridge> >::Leaky |
| 68 IPCWebSocketStreamHandleBridge::all_bridges = LAZY_INSTANCE_INITIALIZER; | 71 IPCWebSocketStreamHandleBridge::all_bridges = LAZY_INSTANCE_INITIALIZER; |
| 69 | 72 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 87 void IPCWebSocketStreamHandleBridge::Connect(const GURL& url) { | 90 void IPCWebSocketStreamHandleBridge::Connect(const GURL& url) { |
| 88 DVLOG(1) << "Bridge (" << this << ") Connect (url=" << url << ")"; | 91 DVLOG(1) << "Bridge (" << this << ") Connect (url=" << url << ")"; |
| 89 | 92 |
| 90 DCHECK_EQ(socket_id_, kNoSocketId); | 93 DCHECK_EQ(socket_id_, kNoSocketId); |
| 91 if (delegate_) | 94 if (delegate_) |
| 92 delegate_->WillOpenStream(handle_, url); | 95 delegate_->WillOpenStream(handle_, url); |
| 93 | 96 |
| 94 socket_id_ = all_bridges.Get().Add(this); | 97 socket_id_ = all_bridges.Get().Add(this); |
| 95 DCHECK_NE(socket_id_, kNoSocketId); | 98 DCHECK_NE(socket_id_, kNoSocketId); |
| 96 int render_frame_id = MSG_ROUTING_NONE; | 99 int render_frame_id = MSG_ROUTING_NONE; |
| 97 WebSocketStreamHandleImpl* impl = | |
| 98 static_cast<WebSocketStreamHandleImpl*>(handle_); | |
| 99 const SocketStreamHandleData* data = | 100 const SocketStreamHandleData* data = |
| 100 static_cast<SocketStreamHandleData*>(impl->GetUserData(handle_)); | 101 SocketStreamHandleData::ForHandle(handle_); |
| 101 if (data) | 102 if (data) |
| 102 render_frame_id = data->render_frame_id(); | 103 render_frame_id = data->render_frame_id(); |
| 103 AddRef(); // Released in OnClosed(). | 104 AddRef(); // Released in OnClosed(). |
| 104 ChildThread::current()->Send( | 105 ChildThread::current()->Send( |
| 105 new SocketStreamHostMsg_Connect(render_frame_id, url, socket_id_)); | 106 new SocketStreamHostMsg_Connect(render_frame_id, url, socket_id_)); |
| 106 DVLOG(1) << "Bridge #" << socket_id_ << " sent IPC Connect"; | 107 DVLOG(1) << "Bridge #" << socket_id_ << " sent IPC Connect"; |
| 107 // TODO(ukai): timeout to OnConnected. | 108 // TODO(ukai): timeout to OnConnected. |
| 108 } | 109 } |
| 109 | 110 |
| 110 bool IPCWebSocketStreamHandleBridge::Send(const std::vector<char>& data) { | 111 bool IPCWebSocketStreamHandleBridge::Send(const std::vector<char>& data) { |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 166 const char* error_msg) { | 167 const char* error_msg) { |
| 167 DVLOG(1) << "Bridge #" << socket_id_ << " OnFailed (error_code=" << error_code | 168 DVLOG(1) << "Bridge #" << socket_id_ << " OnFailed (error_code=" << error_code |
| 168 << ")"; | 169 << ")"; |
| 169 if (delegate_) | 170 if (delegate_) |
| 170 delegate_->DidFail(handle_, error_code, base::ASCIIToUTF16(error_msg)); | 171 delegate_->DidFail(handle_, error_code, base::ASCIIToUTF16(error_msg)); |
| 171 } | 172 } |
| 172 | 173 |
| 173 SocketStreamDispatcher::SocketStreamDispatcher() { | 174 SocketStreamDispatcher::SocketStreamDispatcher() { |
| 174 } | 175 } |
| 175 | 176 |
| 176 // static | 177 /* static */ |
| 177 WebSocketStreamHandleBridge* SocketStreamDispatcher::CreateBridge( | 178 webkit_glue::WebSocketStreamHandleBridge* |
| 179 SocketStreamDispatcher::CreateBridge( |
| 178 blink::WebSocketStreamHandle* handle, | 180 blink::WebSocketStreamHandle* handle, |
| 179 WebSocketStreamHandleDelegate* delegate) { | 181 webkit_glue::WebSocketStreamHandleDelegate* delegate) { |
| 180 return new IPCWebSocketStreamHandleBridge(handle, delegate); | 182 return new IPCWebSocketStreamHandleBridge(handle, delegate); |
| 181 } | 183 } |
| 182 | 184 |
| 183 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { | 185 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { |
| 184 bool handled = true; | 186 bool handled = true; |
| 185 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) | 187 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) |
| 186 IPC_MESSAGE_HANDLER(SocketStreamMsg_Connected, OnConnected) | 188 IPC_MESSAGE_HANDLER(SocketStreamMsg_Connected, OnConnected) |
| 187 IPC_MESSAGE_HANDLER(SocketStreamMsg_SentData, OnSentData) | 189 IPC_MESSAGE_HANDLER(SocketStreamMsg_SentData, OnSentData) |
| 188 IPC_MESSAGE_HANDLER(SocketStreamMsg_ReceivedData, OnReceivedData) | 190 IPC_MESSAGE_HANDLER(SocketStreamMsg_ReceivedData, OnReceivedData) |
| 189 IPC_MESSAGE_HANDLER(SocketStreamMsg_Closed, OnClosed) | 191 IPC_MESSAGE_HANDLER(SocketStreamMsg_Closed, OnClosed) |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 245 void SocketStreamDispatcher::OnFailed(int socket_id, int error_code) { | 247 void SocketStreamDispatcher::OnFailed(int socket_id, int error_code) { |
| 246 IPCWebSocketStreamHandleBridge* bridge = | 248 IPCWebSocketStreamHandleBridge* bridge = |
| 247 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); | 249 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); |
| 248 if (bridge) | 250 if (bridge) |
| 249 bridge->OnFailed(error_code, net::ErrorToString(error_code)); | 251 bridge->OnFailed(error_code, net::ErrorToString(error_code)); |
| 250 else | 252 else |
| 251 DLOG(ERROR) << "No bridge for socket_id=" << socket_id; | 253 DLOG(ERROR) << "No bridge for socket_id=" << socket_id; |
| 252 } | 254 } |
| 253 | 255 |
| 254 } // namespace content | 256 } // namespace content |
| OLD | NEW |