| 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 <vector> | 5 #include <vector> |
| 6 | 6 |
| 7 #include "webkit/tools/test_shell/simple_socket_stream_bridge.h" | 7 #include "webkit/tools/test_shell/simple_socket_stream_bridge.h" |
| 8 | 8 |
| 9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
| 10 #include "base/ref_counted.h" | 10 #include "base/ref_counted.h" |
| 11 #include "googleurl/src/gurl.h" | 11 #include "googleurl/src/gurl.h" |
| 12 #include "net/socket_stream/socket_stream.h" | 12 #include "net/socket_stream/socket_stream.h" |
| 13 #include "net/url_request/url_request_context.h" | 13 #include "net/url_request/url_request_context.h" |
| 14 #include "webkit/api/public/WebSocketStreamHandle.h" | 14 #include "webkit/api/public/WebSocketStreamHandle.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 using webkit_glue::WebSocketStreamHandleBridge; | 18 using webkit_glue::WebSocketStreamHandleBridge; |
| 19 | 19 |
| 20 static const int kNoSocketId = 0; | 20 static const int kNoSocketId = 0; |
| 21 | 21 |
| 22 namespace { | 22 namespace { |
| 23 | 23 |
| 24 MessageLoop* g_io_thread; | 24 MessageLoop* g_io_thread; |
| 25 scoped_refptr<URLRequestContext> g_request_context; | 25 scoped_refptr<URLRequestContext> g_request_context; |
| 26 | 26 |
| 27 class WebSocketStreamHandleBridgeImpl | 27 class WebSocketStreamHandleBridgeImpl |
| 28 : public base::RefCountedThreadSafe<WebSocketStreamHandleBridgeImpl>, | 28 : public WebSocketStreamHandleBridge, |
| 29 public WebSocketStreamHandleBridge, | |
| 30 public net::SocketStream::Delegate { | 29 public net::SocketStream::Delegate { |
| 31 public: | 30 public: |
| 32 WebSocketStreamHandleBridgeImpl( | 31 WebSocketStreamHandleBridgeImpl( |
| 33 WebKit::WebSocketStreamHandle* handle, | 32 WebKit::WebSocketStreamHandle* handle, |
| 34 webkit_glue::WebSocketStreamHandleDelegate* delegate); | 33 webkit_glue::WebSocketStreamHandleDelegate* delegate); |
| 35 virtual ~WebSocketStreamHandleBridgeImpl(); | |
| 36 | 34 |
| 37 // WebSocketStreamHandleBridge methods. | 35 // WebSocketStreamHandleBridge methods. |
| 38 virtual void Connect(const GURL& url); | 36 virtual void Connect(const GURL& url); |
| 39 virtual bool Send(const std::vector<char>& data); | 37 virtual bool Send(const std::vector<char>& data); |
| 40 virtual void Close(); | 38 virtual void Close(); |
| 41 | 39 |
| 42 // net::SocketStream::Delegate methods. | 40 // net::SocketStream::Delegate methods. |
| 43 virtual void OnConnected(net::SocketStream* req, | 41 virtual void OnConnected(net::SocketStream* req, |
| 44 int max_pending_send_allowed); | 42 int max_pending_send_allowed); |
| 45 virtual void OnSentData(net::SocketStream* req, | 43 virtual void OnSentData(net::SocketStream* req, |
| 46 int amount_sent); | 44 int amount_sent); |
| 47 virtual void OnReceivedData(net::SocketStream* req, | 45 virtual void OnReceivedData(net::SocketStream* req, |
| 48 const char* data, int len); | 46 const char* data, int len); |
| 49 virtual void OnClose(net::SocketStream* req); | 47 virtual void OnClose(net::SocketStream* req); |
| 50 | 48 |
| 49 private: |
| 50 virtual ~WebSocketStreamHandleBridgeImpl(); |
| 51 | 51 |
| 52 private: | |
| 53 // Runs on |g_io_thread|; | 52 // Runs on |g_io_thread|; |
| 54 void DoConnect(const GURL& url); | 53 void DoConnect(const GURL& url); |
| 55 void DoSend(std::vector<char>* data); | 54 void DoSend(std::vector<char>* data); |
| 56 void DoClose(); | 55 void DoClose(); |
| 57 | 56 |
| 58 // Runs on |message_loop_|; | 57 // Runs on |message_loop_|; |
| 59 void DoOnConnected(int max_amount_send_allowed); | 58 void DoOnConnected(int max_amount_send_allowed); |
| 60 void DoOnSentData(int amount_sent); | 59 void DoOnSentData(int amount_sent); |
| 61 void DoOnReceivedData(std::vector<char>* data); | 60 void DoOnReceivedData(std::vector<char>* data); |
| 62 void DoOnClose(webkit_glue::WebSocketStreamHandleDelegate* delegate); | 61 void DoOnClose(webkit_glue::WebSocketStreamHandleDelegate* delegate); |
| (...skipping 155 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 218 namespace webkit_glue { | 217 namespace webkit_glue { |
| 219 | 218 |
| 220 /* static */ | 219 /* static */ |
| 221 WebSocketStreamHandleBridge* WebSocketStreamHandleBridge::Create( | 220 WebSocketStreamHandleBridge* WebSocketStreamHandleBridge::Create( |
| 222 WebKit::WebSocketStreamHandle* handle, | 221 WebKit::WebSocketStreamHandle* handle, |
| 223 WebSocketStreamHandleDelegate* delegate) { | 222 WebSocketStreamHandleDelegate* delegate) { |
| 224 return new WebSocketStreamHandleBridgeImpl(handle, delegate); | 223 return new WebSocketStreamHandleBridgeImpl(handle, delegate); |
| 225 } | 224 } |
| 226 | 225 |
| 227 } // namespace webkit_glue | 226 } // namespace webkit_glue |
| OLD | NEW |