| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "remoting/protocol/client_control_dispatcher.h" | 5 #include "remoting/protocol/client_control_dispatcher.h" |
| 6 | 6 |
| 7 #include "base/memory/ref_counted.h" | |
| 8 #include "base/message_loop_proxy.h" | 7 #include "base/message_loop_proxy.h" |
| 9 #include "net/base/io_buffer.h" | 8 #include "net/socket/stream_socket.h" |
| 9 #include "remoting/base/constants.h" |
| 10 #include "remoting/proto/control.pb.h" | 10 #include "remoting/proto/control.pb.h" |
| 11 #include "remoting/proto/event.pb.h" | |
| 12 #include "remoting/proto/internal.pb.h" | 11 #include "remoting/proto/internal.pb.h" |
| 12 #include "remoting/protocol/buffered_socket_writer.h" |
| 13 #include "remoting/protocol/client_stub.h" | 13 #include "remoting/protocol/client_stub.h" |
| 14 #include "remoting/protocol/input_stub.h" | |
| 15 #include "remoting/protocol/message_reader.h" | |
| 16 #include "remoting/protocol/session.h" | |
| 17 | 14 |
| 18 namespace remoting { | 15 namespace remoting { |
| 19 namespace protocol { | 16 namespace protocol { |
| 20 | 17 |
| 21 ClientControlDispatcher::ClientControlDispatcher() | 18 ClientControlDispatcher::ClientControlDispatcher() |
| 22 : client_stub_(NULL), | 19 : ChannelDispatcherBase(kControlChannelName), |
| 20 client_stub_(NULL), |
| 23 writer_(new BufferedSocketWriter(base::MessageLoopProxy::current())) { | 21 writer_(new BufferedSocketWriter(base::MessageLoopProxy::current())) { |
| 24 } | 22 } |
| 25 | 23 |
| 26 ClientControlDispatcher::~ClientControlDispatcher() { | 24 ClientControlDispatcher::~ClientControlDispatcher() { |
| 27 writer_->Close(); | 25 writer_->Close(); |
| 28 } | 26 } |
| 29 | 27 |
| 30 void ClientControlDispatcher::Init(protocol::Session* session) { | 28 void ClientControlDispatcher::OnInitialized() { |
| 31 DCHECK(session); | |
| 32 | |
| 33 // TODO(garykac): Set write failed callback. | 29 // TODO(garykac): Set write failed callback. |
| 34 writer_->Init(session->control_channel(), | 30 writer_->Init(channel(), BufferedSocketWriter::WriteFailedCallback()); |
| 35 BufferedSocketWriter::WriteFailedCallback()); | 31 reader_.Init(channel(), base::Bind( |
| 36 reader_.Init(session->control_channel(), base::Bind( | |
| 37 &ClientControlDispatcher::OnMessageReceived, base::Unretained(this))); | 32 &ClientControlDispatcher::OnMessageReceived, base::Unretained(this))); |
| 38 } | 33 } |
| 39 | 34 |
| 40 void ClientControlDispatcher::OnMessageReceived( | 35 void ClientControlDispatcher::OnMessageReceived( |
| 41 ControlMessage* message, const base::Closure& done_task) { | 36 ControlMessage* message, const base::Closure& done_task) { |
| 42 DCHECK(client_stub_); | 37 DCHECK(client_stub_); |
| 43 | 38 |
| 44 if (message->has_begin_session_deprecated()) { | 39 if (message->has_begin_session_deprecated()) { |
| 45 // Host sends legacy BeginSession message for compatibility with | 40 // Host sends legacy BeginSession message for compatibility with |
| 46 // older clients. Ignore it. | 41 // older clients. Ignore it. |
| 47 } else { | 42 } else { |
| 48 LOG(WARNING) << "Unknown control message received."; | 43 LOG(WARNING) << "Unknown control message received."; |
| 49 } | 44 } |
| 50 done_task.Run(); | 45 done_task.Run(); |
| 51 } | 46 } |
| 52 | 47 |
| 53 } // namespace protocol | 48 } // namespace protocol |
| 54 } // namespace remoting | 49 } // namespace remoting |
| OLD | NEW |