| 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/common/socket_stream_dispatcher.h" | 5 #include "content/common/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.h" | 13 #include "base/message_loop.h" |
| 14 #include "content/common/child_thread.h" | 14 #include "content/common/child_thread.h" |
| 15 #include "content/common/socket_stream.h" | 15 #include "content/common/socket_stream.h" |
| 16 #include "content/common/socket_stream_handle_data.h" |
| 16 #include "content/common/socket_stream_messages.h" | 17 #include "content/common/socket_stream_messages.h" |
| 17 #include "googleurl/src/gurl.h" | 18 #include "googleurl/src/gurl.h" |
| 18 #include "webkit/glue/websocketstreamhandle_bridge.h" | 19 #include "webkit/glue/websocketstreamhandle_bridge.h" |
| 19 #include "webkit/glue/websocketstreamhandle_delegate.h" | 20 #include "webkit/glue/websocketstreamhandle_delegate.h" |
| 20 | 21 |
| 21 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. | 22 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. |
| 22 // It communicates with the main browser process via SocketStreamDispatcher. | 23 // It communicates with the main browser process via SocketStreamDispatcher. |
| 23 class IPCWebSocketStreamHandleBridge | 24 class IPCWebSocketStreamHandleBridge |
| 24 : public webkit_glue::WebSocketStreamHandleBridge { | 25 : public webkit_glue::WebSocketStreamHandleBridge { |
| 25 public: | 26 public: |
| (...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 139 } | 140 } |
| 140 | 141 |
| 141 void IPCWebSocketStreamHandleBridge::DoConnect(const GURL& url) { | 142 void IPCWebSocketStreamHandleBridge::DoConnect(const GURL& url) { |
| 142 DCHECK(child_thread_); | 143 DCHECK(child_thread_); |
| 143 DCHECK_EQ(socket_id_, content::kNoSocketId); | 144 DCHECK_EQ(socket_id_, content::kNoSocketId); |
| 144 if (delegate_) | 145 if (delegate_) |
| 145 delegate_->WillOpenStream(handle_, url); | 146 delegate_->WillOpenStream(handle_, url); |
| 146 | 147 |
| 147 socket_id_ = all_bridges.Get().Add(this); | 148 socket_id_ = all_bridges.Get().Add(this); |
| 148 DCHECK_NE(socket_id_, content::kNoSocketId); | 149 DCHECK_NE(socket_id_, content::kNoSocketId); |
| 150 int render_view_id = MSG_ROUTING_NONE; |
| 151 const SocketStreamHandleData* data = |
| 152 SocketStreamHandleData::ForHandle(handle_); |
| 153 if (data) |
| 154 render_view_id = data->render_view_id(); |
| 149 AddRef(); // Released in OnClosed(). | 155 AddRef(); // Released in OnClosed(). |
| 150 if (child_thread_->Send(new SocketStreamHostMsg_Connect(url, socket_id_))) { | 156 if (child_thread_->Send( |
| 157 new SocketStreamHostMsg_Connect(render_view_id, url, socket_id_))) { |
| 151 DVLOG(1) << "Connect socket_id=" << socket_id_; | 158 DVLOG(1) << "Connect socket_id=" << socket_id_; |
| 152 // TODO(ukai): timeout to OnConnected. | 159 // TODO(ukai): timeout to OnConnected. |
| 153 } else { | 160 } else { |
| 154 DLOG(ERROR) << "IPC SocketStream_Connect failed."; | 161 DLOG(ERROR) << "IPC SocketStream_Connect failed."; |
| 155 OnClosed(); | 162 OnClosed(); |
| 156 } | 163 } |
| 157 } | 164 } |
| 158 | 165 |
| 159 void IPCWebSocketStreamHandleBridge::DoClose() { | 166 void IPCWebSocketStreamHandleBridge::DoClose() { |
| 160 child_thread_->Send(new SocketStreamHostMsg_Close(socket_id_)); | 167 child_thread_->Send(new SocketStreamHostMsg_Close(socket_id_)); |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 217 } | 224 } |
| 218 | 225 |
| 219 void SocketStreamDispatcher::OnClosed(int socket_id) { | 226 void SocketStreamDispatcher::OnClosed(int socket_id) { |
| 220 IPCWebSocketStreamHandleBridge* bridge = | 227 IPCWebSocketStreamHandleBridge* bridge = |
| 221 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); | 228 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); |
| 222 if (bridge) | 229 if (bridge) |
| 223 bridge->OnClosed(); | 230 bridge->OnClosed(); |
| 224 else | 231 else |
| 225 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; | 232 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; |
| 226 } | 233 } |
| OLD | NEW |