| Index: remoting/protocol/client_control_dispatcher.cc
|
| diff --git a/remoting/protocol/client_control_dispatcher.cc b/remoting/protocol/client_control_dispatcher.cc
|
| index 50897ac2dfba9e72e22e2c9dcd14b2a34033b03a..1e76e32c97975055c6b277ee37246d4af422f11b 100644
|
| --- a/remoting/protocol/client_control_dispatcher.cc
|
| +++ b/remoting/protocol/client_control_dispatcher.cc
|
| @@ -4,22 +4,20 @@
|
|
|
| #include "remoting/protocol/client_control_dispatcher.h"
|
|
|
| -#include "base/memory/ref_counted.h"
|
| #include "base/message_loop_proxy.h"
|
| -#include "net/base/io_buffer.h"
|
| +#include "net/socket/stream_socket.h"
|
| +#include "remoting/base/constants.h"
|
| #include "remoting/proto/control.pb.h"
|
| -#include "remoting/proto/event.pb.h"
|
| #include "remoting/proto/internal.pb.h"
|
| +#include "remoting/protocol/buffered_socket_writer.h"
|
| #include "remoting/protocol/client_stub.h"
|
| -#include "remoting/protocol/input_stub.h"
|
| -#include "remoting/protocol/message_reader.h"
|
| -#include "remoting/protocol/session.h"
|
|
|
| namespace remoting {
|
| namespace protocol {
|
|
|
| ClientControlDispatcher::ClientControlDispatcher()
|
| - : client_stub_(NULL),
|
| + : ChannelDispatcherBase(kControlChannelName),
|
| + client_stub_(NULL),
|
| writer_(new BufferedSocketWriter(base::MessageLoopProxy::current())) {
|
| }
|
|
|
| @@ -27,13 +25,10 @@ ClientControlDispatcher::~ClientControlDispatcher() {
|
| writer_->Close();
|
| }
|
|
|
| -void ClientControlDispatcher::Init(protocol::Session* session) {
|
| - DCHECK(session);
|
| -
|
| +void ClientControlDispatcher::OnInitialized() {
|
| // TODO(garykac): Set write failed callback.
|
| - writer_->Init(session->control_channel(),
|
| - BufferedSocketWriter::WriteFailedCallback());
|
| - reader_.Init(session->control_channel(), base::Bind(
|
| + writer_->Init(channel(), BufferedSocketWriter::WriteFailedCallback());
|
| + reader_.Init(channel(), base::Bind(
|
| &ClientControlDispatcher::OnMessageReceived, base::Unretained(this)));
|
| }
|
|
|
|
|