Index: remoting/protocol/client_control_dispatcher.h |
diff --git a/remoting/protocol/client_control_dispatcher.h b/remoting/protocol/client_control_dispatcher.h |
index 005cc0d0ff6b4a7dec8bbfa99bbef0902e59ebfd..6d91c5a384b29389ef4ce96471c69dd4f63a187f 100644 |
--- a/remoting/protocol/client_control_dispatcher.h |
+++ b/remoting/protocol/client_control_dispatcher.h |
@@ -5,8 +5,8 @@ |
#ifndef REMOTING_PROTOCOL_CLIENT_CONTROL_DISPATCHER_H_ |
#define REMOTING_PROTOCOL_CLIENT_CONTROL_DISPATCHER_H_ |
-#include "base/basictypes.h" |
#include "base/memory/ref_counted.h" |
+#include "remoting/protocol/channel_dispatcher_base.h" |
#include "remoting/protocol/host_stub.h" |
#include "remoting/protocol/message_reader.h" |
@@ -25,20 +25,20 @@ class Session; |
// ClientControlDispatcher dispatches incoming messages on the control |
// channel to ClientStub, and also implements HostStub for outgoing |
// messages. |
-class ClientControlDispatcher : public HostStub { |
+class ClientControlDispatcher : public ChannelDispatcherBase, public HostStub { |
public: |
ClientControlDispatcher(); |
virtual ~ClientControlDispatcher(); |
- // Initialize the control channel and the dispatcher for the |
- // |session|. Doesn't take ownership of |session|. |
- void Init(protocol::Session* session); |
- |
// Sets ClientStub that will be called for each incoming control |
// message. Doesn't take ownershipt of |client_stub|. It must outlive |
// this dispatcher. |
void set_client_stub(ClientStub* client_stub) { client_stub_ = client_stub; } |
+ protected: |
+ // ChannelDispatcherBase overrides. |
+ virtual void OnInitialized() OVERRIDE; |
+ |
private: |
void OnMessageReceived(ControlMessage* message, |
const base::Closure& done_task); |