blob: 43f4f8f95acc285e3786843f5fec31885333454f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "remoting/protocol/host_control_dispatcher.h"
#include "base/message_loop_proxy.h"
#include "net/socket/stream_socket.h"
#include "remoting/base/constants.h"
#include "remoting/proto/control.pb.h"
#include "remoting/proto/internal.pb.h"
#include "remoting/protocol/buffered_socket_writer.h"
#include "remoting/protocol/host_stub.h"
#include "remoting/protocol/util.h"
namespace remoting {
namespace protocol {
HostControlDispatcher::HostControlDispatcher()
: ChannelDispatcherBase(kControlChannelName),
host_stub_(NULL),
writer_(new BufferedSocketWriter(base::MessageLoopProxy::current())) {
}
HostControlDispatcher::~HostControlDispatcher() {
writer_->Close();
}
void HostControlDispatcher::OnInitialized() {
reader_.Init(channel(), base::Bind(
&HostControlDispatcher::OnMessageReceived, base::Unretained(this)));
writer_->Init(channel(), BufferedSocketWriter::WriteFailedCallback());
// Write legacy BeginSession message.
// TODO(sergeyu): Remove it. See http://crbug.com/104670 .
protocol::ControlMessage message;
message.mutable_begin_session_deprecated()->mutable_login_status()->
set_success(true);
writer_->Write(SerializeAndFrameMessage(message), base::Closure());
}
void HostControlDispatcher::OnMessageReceived(
ControlMessage* message, const base::Closure& done_task) {
DCHECK(host_stub_);
LOG(WARNING) << "Unknown control message received.";
done_task.Run();
}
} // namespace protocol
} // namespace remoting
|