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 #ifndef REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ | 5 #ifndef REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ |
6 #define REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ | 6 #define REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/scoped_ptr.h" |
13 #include "remoting/base/buffered_socket_writer.h" | |
14 #include "remoting/protocol/errors.h" | 13 #include "remoting/protocol/errors.h" |
15 | 14 |
16 namespace remoting { | 15 namespace remoting { |
17 | 16 |
18 class CompoundBuffer; | 17 class CompoundBuffer; |
19 | 18 |
20 namespace protocol { | 19 namespace protocol { |
21 | 20 |
| 21 class MessageChannelFactory; |
22 class MessagePipe; | 22 class MessagePipe; |
23 class P2PStreamSocket; | |
24 class StreamChannelFactory; | |
25 | 23 |
26 // Base class for channel message dispatchers. It's responsible for | 24 // Base class for channel message dispatchers. It's responsible for |
27 // creating the named channel. Derived dispatchers then dispatch | 25 // creating the named channel. Derived dispatchers then dispatch |
28 // incoming messages on this channel as well as send outgoing | 26 // incoming messages on this channel as well as send outgoing |
29 // messages. | 27 // messages. |
30 class ChannelDispatcherBase { | 28 class ChannelDispatcherBase { |
31 public: | 29 public: |
32 class EventHandler { | 30 class EventHandler { |
33 public: | 31 public: |
34 EventHandler() {} | 32 EventHandler() {} |
35 virtual ~EventHandler() {} | 33 virtual ~EventHandler() {} |
36 | 34 |
37 virtual void OnChannelInitialized( | 35 virtual void OnChannelInitialized( |
38 ChannelDispatcherBase* channel_dispatcher) = 0; | 36 ChannelDispatcherBase* channel_dispatcher) = 0; |
39 virtual void OnChannelError(ChannelDispatcherBase* channel_dispatcher, | |
40 ErrorCode error) = 0; | |
41 }; | 37 }; |
42 | 38 |
43 // The callback is called when initialization is finished. The | 39 // The callback is called when initialization is finished. The |
44 // parameter is set to true on success. | 40 // parameter is set to true on success. |
45 typedef base::Callback<void(bool)> InitializedCallback; | 41 typedef base::Callback<void(bool)> InitializedCallback; |
46 | 42 |
47 virtual ~ChannelDispatcherBase(); | 43 virtual ~ChannelDispatcherBase(); |
48 | 44 |
49 // Creates and connects the channel in the specified | 45 // Creates and connects the channel in the specified |
50 // |session|. Caller retains ownership of the Session. | 46 // |session|. Caller retains ownership of the Session. |
51 void Init(StreamChannelFactory* channel_factory, EventHandler* event_handler); | 47 void Init(MessageChannelFactory* channel_factory, |
| 48 EventHandler* event_handler); |
52 | 49 |
53 const std::string& channel_name() { return channel_name_; } | 50 const std::string& channel_name() { return channel_name_; } |
54 | 51 |
55 // Returns true if the channel is currently connected. | 52 // Returns true if the channel is currently connected. |
56 bool is_connected() { return message_pipe() != nullptr; } | 53 bool is_connected() { return message_pipe() != nullptr; } |
57 | 54 |
58 protected: | 55 protected: |
59 explicit ChannelDispatcherBase(const char* channel_name); | 56 explicit ChannelDispatcherBase(const char* channel_name); |
60 | 57 |
61 MessagePipe* message_pipe() { return message_pipe_.get(); } | 58 MessagePipe* message_pipe() { return message_pipe_.get(); } |
62 | 59 |
63 // Child classes must override this method to handle incoming messages. | 60 // Child classes must override this method to handle incoming messages. |
64 virtual void OnIncomingMessage(scoped_ptr<CompoundBuffer> message) = 0; | 61 virtual void OnIncomingMessage(scoped_ptr<CompoundBuffer> message) = 0; |
65 | 62 |
66 private: | 63 private: |
67 void OnChannelReady(scoped_ptr<P2PStreamSocket> socket); | 64 void OnChannelReady(scoped_ptr<MessagePipe> message_pipe); |
68 void OnPipeError(int error); | 65 void OnPipeError(int error); |
69 | 66 |
70 std::string channel_name_; | 67 std::string channel_name_; |
71 StreamChannelFactory* channel_factory_; | 68 MessageChannelFactory* channel_factory_ = nullptr; |
72 EventHandler* event_handler_; | 69 EventHandler* event_handler_ = nullptr; |
73 | 70 |
74 scoped_ptr<MessagePipe> message_pipe_; | 71 scoped_ptr<MessagePipe> message_pipe_; |
75 | 72 |
76 DISALLOW_COPY_AND_ASSIGN(ChannelDispatcherBase); | 73 DISALLOW_COPY_AND_ASSIGN(ChannelDispatcherBase); |
77 }; | 74 }; |
78 | 75 |
79 } // namespace protocol | 76 } // namespace protocol |
80 } // namespace remoting | 77 } // namespace remoting |
81 | 78 |
82 #endif // REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ | 79 #endif // REMOTING_PROTOCOL_CHANNEL_DISPATCHER_BASE_H_ |
OLD | NEW |