| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #ifndef MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ | 5 #ifndef MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ |
| 6 #define MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ | 6 #define MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ |
| 7 | 7 |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "mojo/application/app_lifetime_helper.h" |
| 9 #include "mojo/services/network/public/interfaces/web_socket.mojom.h" | 10 #include "mojo/services/network/public/interfaces/web_socket.mojom.h" |
| 10 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_impl.h" | 11 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_impl.h" |
| 11 | 12 |
| 12 namespace net { | 13 namespace net { |
| 13 class WebSocketChannel; | 14 class WebSocketChannel; |
| 14 } // namespace net | 15 } // namespace net |
| 15 | 16 |
| 16 namespace mojo { | 17 namespace mojo { |
| 17 class NetworkContext; | 18 class NetworkContext; |
| 18 class WebSocketReadQueue; | 19 class WebSocketReadQueue; |
| 19 | 20 |
| 20 // Forms a bridge between the WebSocket mojo interface and the net::WebSocket | 21 // Forms a bridge between the WebSocket mojo interface and the net::WebSocket |
| 21 // implementation. | 22 // implementation. |
| 22 class WebSocketImpl : public InterfaceImpl<WebSocket> { | 23 class WebSocketImpl : public InterfaceImpl<WebSocket> { |
| 23 public: | 24 public: |
| 24 explicit WebSocketImpl(NetworkContext* context); | 25 WebSocketImpl(NetworkContext* context, |
| 26 scoped_ptr<mojo::AppRefCount> app_refcount); |
| 25 ~WebSocketImpl() override; | 27 ~WebSocketImpl() override; |
| 26 | 28 |
| 27 private: | 29 private: |
| 28 // WebSocket methods: | 30 // WebSocket methods: |
| 29 void Connect(const String& url, | 31 void Connect(const String& url, |
| 30 Array<String> protocols, | 32 Array<String> protocols, |
| 31 const String& origin, | 33 const String& origin, |
| 32 ScopedDataPipeConsumerHandle send_stream, | 34 ScopedDataPipeConsumerHandle send_stream, |
| 33 WebSocketClientPtr client) override; | 35 WebSocketClientPtr client) override; |
| 34 void Send(bool fin, WebSocket::MessageType type, uint32_t num_bytes) override; | 36 void Send(bool fin, WebSocket::MessageType type, uint32_t num_bytes) override; |
| 35 void FlowControl(int64_t quota) override; | 37 void FlowControl(int64_t quota) override; |
| 36 void Close(uint16_t code, const String& reason) override; | 38 void Close(uint16_t code, const String& reason) override; |
| 37 | 39 |
| 38 // Called with the data to send once it has been read from |send_stream_|. | 40 // Called with the data to send once it has been read from |send_stream_|. |
| 39 void DidReadFromSendStream(bool fin, | 41 void DidReadFromSendStream(bool fin, |
| 40 WebSocket::MessageType type, | 42 WebSocket::MessageType type, |
| 41 uint32_t num_bytes, | 43 uint32_t num_bytes, |
| 42 const char* data); | 44 const char* data); |
| 43 | 45 |
| 44 // The channel we use to send events to the network. | 46 // The channel we use to send events to the network. |
| 45 scoped_ptr<net::WebSocketChannel> channel_; | 47 scoped_ptr<net::WebSocketChannel> channel_; |
| 46 ScopedDataPipeConsumerHandle send_stream_; | 48 ScopedDataPipeConsumerHandle send_stream_; |
| 47 scoped_ptr<WebSocketReadQueue> read_queue_; | 49 scoped_ptr<WebSocketReadQueue> read_queue_; |
| 48 NetworkContext* context_; | 50 NetworkContext* context_; |
| 51 scoped_ptr<mojo::AppRefCount> app_refcount_; |
| 49 }; | 52 }; |
| 50 | 53 |
| 51 } // namespace mojo | 54 } // namespace mojo |
| 52 | 55 |
| 53 #endif // MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ | 56 #endif // MOJO_SERVICES_NETWORK_WEB_SOCKET_IMPL_H_ |
| OLD | NEW |