| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "chrome/common/socket_stream_dispatcher.h" | 5 #include "chrome/common/socket_stream_dispatcher.h" |
| 6 | 6 |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "base/id_map.h" | 9 #include "base/id_map.h" |
| 10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
| 11 #include "base/ref_counted.h" | 11 #include "base/ref_counted.h" |
| 12 #include "base/task.h" | 12 #include "base/task.h" |
| 13 #include "chrome/common/child_thread.h" | 13 #include "chrome/common/child_thread.h" |
| 14 #include "chrome/common/render_messages.h" | 14 #include "content/common/socket_stream.h" |
| 15 #include "chrome/common/net/socket_stream.h" | 15 #include "content/common/socket_stream_messages.h" |
| 16 #include "googleurl/src/gurl.h" | 16 #include "googleurl/src/gurl.h" |
| 17 #include "ipc/ipc_message.h" | |
| 18 #include "webkit/glue/websocketstreamhandle_bridge.h" | 17 #include "webkit/glue/websocketstreamhandle_bridge.h" |
| 19 #include "webkit/glue/websocketstreamhandle_delegate.h" | 18 #include "webkit/glue/websocketstreamhandle_delegate.h" |
| 20 | 19 |
| 21 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. | 20 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. |
| 22 // It communicates with the main browser process via SocketStreamDispatcher. | 21 // It communicates with the main browser process via SocketStreamDispatcher. |
| 23 class IPCWebSocketStreamHandleBridge | 22 class IPCWebSocketStreamHandleBridge |
| 24 : public webkit_glue::WebSocketStreamHandleBridge { | 23 : public webkit_glue::WebSocketStreamHandleBridge { |
| 25 public: | 24 public: |
| 26 IPCWebSocketStreamHandleBridge( | 25 IPCWebSocketStreamHandleBridge( |
| 27 ChildThread* child_thread, | 26 ChildThread* child_thread, |
| 28 WebKit::WebSocketStreamHandle* handle, | 27 WebKit::WebSocketStreamHandle* handle, |
| 29 webkit_glue::WebSocketStreamHandleDelegate* delegate) | 28 webkit_glue::WebSocketStreamHandleDelegate* delegate) |
| 30 : socket_id_(chrome_common_net::kNoSocketId), | 29 : socket_id_(content_common::kNoSocketId), |
| 31 child_thread_(child_thread), | 30 child_thread_(child_thread), |
| 32 handle_(handle), | 31 handle_(handle), |
| 33 delegate_(delegate) {} | 32 delegate_(delegate) {} |
| 34 | 33 |
| 35 // Returns the handle having given id or NULL if there is no such handle. | 34 // Returns the handle having given id or NULL if there is no such handle. |
| 36 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); | 35 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); |
| 37 | 36 |
| 38 // webkit_glue::WebSocketStreamHandleBridge methods. | 37 // webkit_glue::WebSocketStreamHandleBridge methods. |
| 39 virtual void Connect(const GURL& url); | 38 virtual void Connect(const GURL& url); |
| 40 virtual bool Send(const std::vector<char>& data); | 39 virtual bool Send(const std::vector<char>& data); |
| (...skipping 23 matching lines...) Expand all Loading... |
| 64 | 63 |
| 65 /* static */ | 64 /* static */ |
| 66 IPCWebSocketStreamHandleBridge* IPCWebSocketStreamHandleBridge::FromSocketId( | 65 IPCWebSocketStreamHandleBridge* IPCWebSocketStreamHandleBridge::FromSocketId( |
| 67 int id) { | 66 int id) { |
| 68 return all_bridges.Lookup(id); | 67 return all_bridges.Lookup(id); |
| 69 } | 68 } |
| 70 | 69 |
| 71 IPCWebSocketStreamHandleBridge::~IPCWebSocketStreamHandleBridge() { | 70 IPCWebSocketStreamHandleBridge::~IPCWebSocketStreamHandleBridge() { |
| 72 DVLOG(1) << "IPCWebSocketStreamHandleBridge destructor socket_id=" | 71 DVLOG(1) << "IPCWebSocketStreamHandleBridge destructor socket_id=" |
| 73 << socket_id_; | 72 << socket_id_; |
| 74 if (socket_id_ != chrome_common_net::kNoSocketId) { | 73 if (socket_id_ != content_common::kNoSocketId) { |
| 75 child_thread_->Send(new ViewHostMsg_Close(socket_id_)); | 74 child_thread_->Send(new SocketStreamHostMsg_Close(socket_id_)); |
| 76 socket_id_ = chrome_common_net::kNoSocketId; | 75 socket_id_ = content_common::kNoSocketId; |
| 77 } | 76 } |
| 78 } | 77 } |
| 79 | 78 |
| 80 void IPCWebSocketStreamHandleBridge::Connect(const GURL& url) { | 79 void IPCWebSocketStreamHandleBridge::Connect(const GURL& url) { |
| 81 DCHECK(child_thread_); | 80 DCHECK(child_thread_); |
| 82 DVLOG(1) << "Connect url=" << url; | 81 DVLOG(1) << "Connect url=" << url; |
| 83 child_thread_->message_loop()->PostTask( | 82 child_thread_->message_loop()->PostTask( |
| 84 FROM_HERE, | 83 FROM_HERE, |
| 85 NewRunnableMethod(this, &IPCWebSocketStreamHandleBridge::DoConnect, | 84 NewRunnableMethod(this, &IPCWebSocketStreamHandleBridge::DoConnect, |
| 86 url)); | 85 url)); |
| 87 } | 86 } |
| 88 | 87 |
| 89 bool IPCWebSocketStreamHandleBridge::Send( | 88 bool IPCWebSocketStreamHandleBridge::Send( |
| 90 const std::vector<char>& data) { | 89 const std::vector<char>& data) { |
| 91 DVLOG(1) << "Send data.size=" << data.size(); | 90 DVLOG(1) << "Send data.size=" << data.size(); |
| 92 if (child_thread_->Send( | 91 if (child_thread_->Send( |
| 93 new ViewHostMsg_SocketStream_SendData(socket_id_, data))) { | 92 new SocketStreamHostMsg_SendData(socket_id_, data))) { |
| 94 if (delegate_) | 93 if (delegate_) |
| 95 delegate_->WillSendData(handle_, &data[0], data.size()); | 94 delegate_->WillSendData(handle_, &data[0], data.size()); |
| 96 return true; | 95 return true; |
| 97 } | 96 } |
| 98 return false; | 97 return false; |
| 99 } | 98 } |
| 100 | 99 |
| 101 void IPCWebSocketStreamHandleBridge::Close() { | 100 void IPCWebSocketStreamHandleBridge::Close() { |
| 102 DVLOG(1) << "Close socket_id" << socket_id_; | 101 DVLOG(1) << "Close socket_id" << socket_id_; |
| 103 child_thread_->Send(new ViewHostMsg_SocketStream_Close(socket_id_)); | 102 child_thread_->Send(new SocketStreamHostMsg_Close(socket_id_)); |
| 104 } | 103 } |
| 105 | 104 |
| 106 void IPCWebSocketStreamHandleBridge::OnConnected(int max_pending_send_allowed) { | 105 void IPCWebSocketStreamHandleBridge::OnConnected(int max_pending_send_allowed) { |
| 107 DVLOG(1) << "IPCWebSocketStreamHandleBridge::OnConnected socket_id=" | 106 DVLOG(1) << "IPCWebSocketStreamHandleBridge::OnConnected socket_id=" |
| 108 << socket_id_; | 107 << socket_id_; |
| 109 if (delegate_) | 108 if (delegate_) |
| 110 delegate_->DidOpenStream(handle_, max_pending_send_allowed); | 109 delegate_->DidOpenStream(handle_, max_pending_send_allowed); |
| 111 } | 110 } |
| 112 | 111 |
| 113 void IPCWebSocketStreamHandleBridge::OnSentData(int amount_sent) { | 112 void IPCWebSocketStreamHandleBridge::OnSentData(int amount_sent) { |
| 114 if (delegate_) | 113 if (delegate_) |
| 115 delegate_->DidSendData(handle_, amount_sent); | 114 delegate_->DidSendData(handle_, amount_sent); |
| 116 } | 115 } |
| 117 | 116 |
| 118 void IPCWebSocketStreamHandleBridge::OnReceivedData( | 117 void IPCWebSocketStreamHandleBridge::OnReceivedData( |
| 119 const std::vector<char>& data) { | 118 const std::vector<char>& data) { |
| 120 if (delegate_) | 119 if (delegate_) |
| 121 delegate_->DidReceiveData(handle_, &data[0], data.size()); | 120 delegate_->DidReceiveData(handle_, &data[0], data.size()); |
| 122 } | 121 } |
| 123 | 122 |
| 124 void IPCWebSocketStreamHandleBridge::OnClosed() { | 123 void IPCWebSocketStreamHandleBridge::OnClosed() { |
| 125 DVLOG(1) << "IPCWebSocketStreamHandleBridge::OnClosed"; | 124 DVLOG(1) << "IPCWebSocketStreamHandleBridge::OnClosed"; |
| 126 if (socket_id_ != chrome_common_net::kNoSocketId) { | 125 if (socket_id_ != content_common::kNoSocketId) { |
| 127 all_bridges.Remove(socket_id_); | 126 all_bridges.Remove(socket_id_); |
| 128 socket_id_ = chrome_common_net::kNoSocketId; | 127 socket_id_ = content_common::kNoSocketId; |
| 129 } | 128 } |
| 130 if (delegate_) | 129 if (delegate_) |
| 131 delegate_->DidClose(handle_); | 130 delegate_->DidClose(handle_); |
| 132 delegate_ = NULL; | 131 delegate_ = NULL; |
| 133 Release(); | 132 Release(); |
| 134 } | 133 } |
| 135 | 134 |
| 136 void IPCWebSocketStreamHandleBridge::DoConnect(const GURL& url) { | 135 void IPCWebSocketStreamHandleBridge::DoConnect(const GURL& url) { |
| 137 DCHECK(child_thread_); | 136 DCHECK(child_thread_); |
| 138 DCHECK_EQ(socket_id_, chrome_common_net::kNoSocketId); | 137 DCHECK_EQ(socket_id_, content_common::kNoSocketId); |
| 139 if (delegate_) | 138 if (delegate_) |
| 140 delegate_->WillOpenStream(handle_, url); | 139 delegate_->WillOpenStream(handle_, url); |
| 141 | 140 |
| 142 socket_id_ = all_bridges.Add(this); | 141 socket_id_ = all_bridges.Add(this); |
| 143 DCHECK_NE(socket_id_, chrome_common_net::kNoSocketId); | 142 DCHECK_NE(socket_id_, content_common::kNoSocketId); |
| 144 AddRef(); // Released in OnClosed(). | 143 AddRef(); // Released in OnClosed(). |
| 145 if (child_thread_->Send( | 144 if (child_thread_->Send(new SocketStreamHostMsg_Connect(url, socket_id_))) { |
| 146 new ViewHostMsg_SocketStream_Connect(url, socket_id_))) { | |
| 147 DVLOG(1) << "Connect socket_id=" << socket_id_; | 145 DVLOG(1) << "Connect socket_id=" << socket_id_; |
| 148 // TODO(ukai): timeout to OnConnected. | 146 // TODO(ukai): timeout to OnConnected. |
| 149 } else { | 147 } else { |
| 150 LOG(ERROR) << "IPC SocketStream_Connect failed."; | 148 LOG(ERROR) << "IPC SocketStream_Connect failed."; |
| 151 OnClosed(); | 149 OnClosed(); |
| 152 } | 150 } |
| 153 } | 151 } |
| 154 | 152 |
| 155 SocketStreamDispatcher::SocketStreamDispatcher() { | 153 SocketStreamDispatcher::SocketStreamDispatcher() { |
| 156 } | 154 } |
| 157 | 155 |
| 158 /* static */ | 156 /* static */ |
| 159 webkit_glue::WebSocketStreamHandleBridge* | 157 webkit_glue::WebSocketStreamHandleBridge* |
| 160 SocketStreamDispatcher::CreateBridge( | 158 SocketStreamDispatcher::CreateBridge( |
| 161 WebKit::WebSocketStreamHandle* handle, | 159 WebKit::WebSocketStreamHandle* handle, |
| 162 webkit_glue::WebSocketStreamHandleDelegate* delegate) { | 160 webkit_glue::WebSocketStreamHandleDelegate* delegate) { |
| 163 return new IPCWebSocketStreamHandleBridge( | 161 return new IPCWebSocketStreamHandleBridge( |
| 164 ChildThread::current(), handle, delegate); | 162 ChildThread::current(), handle, delegate); |
| 165 } | 163 } |
| 166 | 164 |
| 167 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { | 165 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { |
| 168 bool handled = true; | 166 bool handled = true; |
| 169 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) | 167 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) |
| 170 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Connected, OnConnected) | 168 IPC_MESSAGE_HANDLER(SocketStreamMsg_Connected, OnConnected) |
| 171 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_SentData, OnSentData) | 169 IPC_MESSAGE_HANDLER(SocketStreamMsg_SentData, OnSentData) |
| 172 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_ReceivedData, OnReceivedData) | 170 IPC_MESSAGE_HANDLER(SocketStreamMsg_ReceivedData, OnReceivedData) |
| 173 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Closed, OnClosed) | 171 IPC_MESSAGE_HANDLER(SocketStreamMsg_Closed, OnClosed) |
| 174 IPC_MESSAGE_UNHANDLED(handled = false) | 172 IPC_MESSAGE_UNHANDLED(handled = false) |
| 175 IPC_END_MESSAGE_MAP() | 173 IPC_END_MESSAGE_MAP() |
| 176 return handled; | 174 return handled; |
| 177 } | 175 } |
| 178 | 176 |
| 179 void SocketStreamDispatcher::OnConnected(int socket_id, | 177 void SocketStreamDispatcher::OnConnected(int socket_id, |
| 180 int max_pending_send_allowed) { | 178 int max_pending_send_allowed) { |
| 181 DVLOG(1) << "SocketStreamDispatcher::OnConnected socket_id=" << socket_id | 179 DVLOG(1) << "SocketStreamDispatcher::OnConnected socket_id=" << socket_id |
| 182 << " max_pending_send_allowed=" << max_pending_send_allowed; | 180 << " max_pending_send_allowed=" << max_pending_send_allowed; |
| 183 IPCWebSocketStreamHandleBridge* bridge = | 181 IPCWebSocketStreamHandleBridge* bridge = |
| (...skipping 24 matching lines...) Expand all Loading... |
| 208 } | 206 } |
| 209 | 207 |
| 210 void SocketStreamDispatcher::OnClosed(int socket_id) { | 208 void SocketStreamDispatcher::OnClosed(int socket_id) { |
| 211 IPCWebSocketStreamHandleBridge* bridge = | 209 IPCWebSocketStreamHandleBridge* bridge = |
| 212 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); | 210 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); |
| 213 if (bridge) | 211 if (bridge) |
| 214 bridge->OnClosed(); | 212 bridge->OnClosed(); |
| 215 else | 213 else |
| 216 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; | 214 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; |
| 217 } | 215 } |
| OLD | NEW |