OLD | NEW |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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/ref_counted.h" | 10 #include "base/ref_counted.h" |
11 #include "chrome/common/child_thread.h" | 11 #include "base/mp/mp_child_thread.h" |
12 #include "chrome/common/render_messages.h" | 12 #include "chrome/common/render_messages.h" |
13 #include "chrome/common/net/socket_stream.h" | 13 #include "chrome/common/net/socket_stream.h" |
14 #include "googleurl/src/gurl.h" | 14 #include "googleurl/src/gurl.h" |
15 #include "webkit/glue/websocketstreamhandle_bridge.h" | 15 #include "webkit/glue/websocketstreamhandle_bridge.h" |
16 #include "webkit/glue/websocketstreamhandle_delegate.h" | 16 #include "webkit/glue/websocketstreamhandle_delegate.h" |
17 | 17 |
18 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. | 18 // IPCWebSocketStreamHandleBridge is owned by each SocketStreamHandle. |
19 // It communicates with the main browser process via SocketStreamDispatcher. | 19 // It communicates with the main browser process via SocketStreamDispatcher. |
20 class IPCWebSocketStreamHandleBridge | 20 class IPCWebSocketStreamHandleBridge |
21 : public webkit_glue::WebSocketStreamHandleBridge { | 21 : public webkit_glue::WebSocketStreamHandleBridge { |
22 public: | 22 public: |
23 IPCWebSocketStreamHandleBridge( | 23 IPCWebSocketStreamHandleBridge( |
24 ChildThread* child_thread, | 24 base::MpChildThread* child_thread, |
25 WebKit::WebSocketStreamHandle* handle, | 25 WebKit::WebSocketStreamHandle* handle, |
26 webkit_glue::WebSocketStreamHandleDelegate* delegate) | 26 webkit_glue::WebSocketStreamHandleDelegate* delegate) |
27 : socket_id_(chrome_common_net::kNoSocketId), | 27 : socket_id_(chrome_common_net::kNoSocketId), |
28 child_thread_(child_thread), | 28 child_thread_(child_thread), |
29 handle_(handle), | 29 handle_(handle), |
30 delegate_(delegate) {} | 30 delegate_(delegate) {} |
31 | 31 |
32 // Returns the handle having given id or NULL if there is no such handle. | 32 // Returns the handle having given id or NULL if there is no such handle. |
33 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); | 33 static IPCWebSocketStreamHandleBridge* FromSocketId(int id); |
34 | 34 |
35 // webkit_glue::WebSocketStreamHandleBridge methods. | 35 // webkit_glue::WebSocketStreamHandleBridge methods. |
36 virtual void Connect(const GURL& url); | 36 virtual void Connect(const GURL& url); |
37 virtual bool Send(const std::vector<char>& data); | 37 virtual bool Send(const std::vector<char>& data); |
38 virtual void Close(); | 38 virtual void Close(); |
39 | 39 |
40 // Called by SocketStreamDispatcher. | 40 // Called by SocketStreamDispatcher. |
41 void OnConnected(int max_amount_send_allowed); | 41 void OnConnected(int max_amount_send_allowed); |
42 void OnSentData(int amount_sent); | 42 void OnSentData(int amount_sent); |
43 void OnReceivedData(const std::vector<char>& data); | 43 void OnReceivedData(const std::vector<char>& data); |
44 void OnClosed(); | 44 void OnClosed(); |
45 | 45 |
46 private: | 46 private: |
47 virtual ~IPCWebSocketStreamHandleBridge(); | 47 virtual ~IPCWebSocketStreamHandleBridge(); |
48 | 48 |
49 void DoConnect(const GURL& url); | 49 void DoConnect(const GURL& url); |
50 int socket_id_; | 50 int socket_id_; |
51 | 51 |
52 ChildThread* child_thread_; | 52 base::MpChildThread* child_thread_; |
53 WebKit::WebSocketStreamHandle* handle_; | 53 WebKit::WebSocketStreamHandle* handle_; |
54 webkit_glue::WebSocketStreamHandleDelegate* delegate_; | 54 webkit_glue::WebSocketStreamHandleDelegate* delegate_; |
55 | 55 |
56 static IDMap<IPCWebSocketStreamHandleBridge> all_bridges; | 56 static IDMap<IPCWebSocketStreamHandleBridge> all_bridges; |
57 }; | 57 }; |
58 | 58 |
59 IDMap<IPCWebSocketStreamHandleBridge> | 59 IDMap<IPCWebSocketStreamHandleBridge> |
60 IPCWebSocketStreamHandleBridge::all_bridges; | 60 IPCWebSocketStreamHandleBridge::all_bridges; |
61 | 61 |
62 /* static */ | 62 /* static */ |
(...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
152 | 152 |
153 SocketStreamDispatcher::SocketStreamDispatcher() { | 153 SocketStreamDispatcher::SocketStreamDispatcher() { |
154 } | 154 } |
155 | 155 |
156 /* static */ | 156 /* static */ |
157 webkit_glue::WebSocketStreamHandleBridge* | 157 webkit_glue::WebSocketStreamHandleBridge* |
158 SocketStreamDispatcher::CreateBridge( | 158 SocketStreamDispatcher::CreateBridge( |
159 WebKit::WebSocketStreamHandle* handle, | 159 WebKit::WebSocketStreamHandle* handle, |
160 webkit_glue::WebSocketStreamHandleDelegate* delegate) { | 160 webkit_glue::WebSocketStreamHandleDelegate* delegate) { |
161 return new IPCWebSocketStreamHandleBridge( | 161 return new IPCWebSocketStreamHandleBridge( |
162 ChildThread::current(), handle, delegate); | 162 base::MpChildThread::current(), handle, delegate); |
163 } | 163 } |
164 | 164 |
165 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { | 165 bool SocketStreamDispatcher::OnMessageReceived(const IPC::Message& msg) { |
166 bool handled = true; | 166 bool handled = true; |
167 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) | 167 IPC_BEGIN_MESSAGE_MAP(SocketStreamDispatcher, msg) |
168 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Connected, OnConnected) | 168 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Connected, OnConnected) |
169 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_SentData, OnSentData) | 169 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_SentData, OnSentData) |
170 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_ReceivedData, OnReceivedData) | 170 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_ReceivedData, OnReceivedData) |
171 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Closed, OnClosed) | 171 IPC_MESSAGE_HANDLER(ViewMsg_SocketStream_Closed, OnClosed) |
172 IPC_MESSAGE_UNHANDLED(handled = false) | 172 IPC_MESSAGE_UNHANDLED(handled = false) |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
206 } | 206 } |
207 | 207 |
208 void SocketStreamDispatcher::OnClosed(int socket_id) { | 208 void SocketStreamDispatcher::OnClosed(int socket_id) { |
209 IPCWebSocketStreamHandleBridge* bridge = | 209 IPCWebSocketStreamHandleBridge* bridge = |
210 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); | 210 IPCWebSocketStreamHandleBridge::FromSocketId(socket_id); |
211 if (bridge) | 211 if (bridge) |
212 bridge->OnClosed(); | 212 bridge->OnClosed(); |
213 else | 213 else |
214 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; | 214 DLOG(ERROR) << "No SocketStreamHandleBridge for socket_id=" << socket_id; |
215 } | 215 } |
OLD | NEW |