| Index: remoting/protocol/host_control_dispatcher.h
|
| diff --git a/remoting/protocol/host_control_dispatcher.h b/remoting/protocol/host_control_dispatcher.h
|
| index 2e52ba6096fe32e7913e60fb2effa325461ba614..45c04fbf86d5cbc918662f48bbd435879f62beb6 100644
|
| --- a/remoting/protocol/host_control_dispatcher.h
|
| +++ b/remoting/protocol/host_control_dispatcher.h
|
| @@ -5,8 +5,8 @@
|
| #ifndef REMOTING_PROTOCOL_HOST_CONTROL_DISPATCHER_H_
|
| #define REMOTING_PROTOCOL_HOST_CONTROL_DISPATCHER_H_
|
|
|
| -#include "base/basictypes.h"
|
| #include "base/memory/ref_counted.h"
|
| +#include "remoting/protocol/channel_dispatcher_base.h"
|
| #include "remoting/protocol/client_stub.h"
|
| #include "remoting/protocol/message_reader.h"
|
|
|
| @@ -14,6 +14,10 @@ namespace base {
|
| class MessageLoopProxy;
|
| } // namespace base
|
|
|
| +namespace net {
|
| +class StreamSocket;
|
| +} // namespace net
|
| +
|
| namespace remoting {
|
| namespace protocol {
|
|
|
| @@ -25,22 +29,21 @@ class Session;
|
| // HostControlDispatcher dispatches incoming messages on the control
|
| // channel to HostStub, and also implements ClientStub for outgoing
|
| // messages.
|
| -class HostControlDispatcher : public ClientStub {
|
| +class HostControlDispatcher : public ChannelDispatcherBase, public ClientStub {
|
| public:
|
| HostControlDispatcher();
|
| virtual ~HostControlDispatcher();
|
|
|
| - // Initialize the control channel and the dispatcher for the
|
| - // |session|. Doesn't take ownership of |session|.
|
| - void Init(Session* session);
|
| -
|
| // Sets HostStub that will be called for each incoming control
|
| // message. Doesn't take ownership of |host_stub|. It must outlive
|
| // this dispatcher.
|
| void set_host_stub(HostStub* host_stub) { host_stub_ = host_stub; }
|
|
|
| + protected:
|
| + // ChannelDispatcherBase overrides.
|
| + virtual void OnInitialized() OVERRIDE;
|
| +
|
| private:
|
| - // This method is called by |reader_| when a message is received.
|
| void OnMessageReceived(ControlMessage* message,
|
| const base::Closure& done_task);
|
|
|
|
|