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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright (c) 2010 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/connection_to_client.h"
#include "google/protobuf/message.h"
#include "net/base/io_buffer.h"
#include "remoting/protocol/client_control_sender.h"
#include "remoting/protocol/host_message_dispatcher.h"
// TODO(hclam): Remove this header once MessageDispatcher is used.
#include "remoting/base/compound_buffer.h"
namespace remoting {
namespace protocol {
// Determine how many update streams we should count to find the size of
// average update stream.
static const size_t kAverageUpdateStream = 10;
ConnectionToClient::ConnectionToClient(MessageLoop* message_loop,
EventHandler* handler,
HostStub* host_stub,
InputStub* input_stub)
: loop_(message_loop),
handler_(handler),
host_stub_(host_stub),
input_stub_(input_stub) {
DCHECK(loop_);
DCHECK(handler_);
}
ConnectionToClient::~ConnectionToClient() {
// TODO(hclam): When we shut down the viewer we may have to close the
// connection.
}
void ConnectionToClient::Init(protocol::Session* session) {
DCHECK_EQ(session->message_loop(), MessageLoop::current());
session_ = session;
session_->SetStateChangeCallback(
NewCallback(this, &ConnectionToClient::OnSessionStateChange));
}
protocol::Session* ConnectionToClient::session() {
return session_;
}
void ConnectionToClient::Disconnect() {
// This method can be called from main thread so perform threading switching.
if (MessageLoop::current() != loop_) {
loop_->PostTask(
FROM_HERE,
NewRunnableMethod(this, &ConnectionToClient::Disconnect));
return;
}
// If there is a channel then close it and release the reference.
if (session_) {
session_->Close(NewRunnableMethod(this, &ConnectionToClient::OnClosed));
session_ = NULL;
}
}
VideoStub* ConnectionToClient::video_stub() {
return video_writer_.get();
}
// Return pointer to ClientStub.
ClientStub* ConnectionToClient::client_stub() {
return client_stub_.get();
}
ConnectionToClient::ConnectionToClient() {
}
void ConnectionToClient::OnSessionStateChange(protocol::Session::State state) {
if (state == protocol::Session::CONNECTED) {
client_stub_.reset(new ClientControlSender(session_->control_channel()));
video_writer_.reset(VideoWriter::Create(session_->config()));
video_writer_->Init(session_);
dispatcher_.reset(new HostMessageDispatcher());
dispatcher_->Initialize(session_.get(), host_stub_, input_stub_);
}
// This method can be called from main thread so perform threading switching.
if (MessageLoop::current() != loop_) {
loop_->PostTask(
FROM_HERE,
NewRunnableMethod(this, &ConnectionToClient::StateChangeTask, state));
} else {
StateChangeTask(state);
}
}
void ConnectionToClient::StateChangeTask(protocol::Session::State state) {
DCHECK_EQ(loop_, MessageLoop::current());
DCHECK(handler_);
switch(state) {
case protocol::Session::CONNECTING:
break;
// Don't care about this message.
case protocol::Session::CONNECTED:
handler_->OnConnectionOpened(this);
break;
case protocol::Session::CLOSED:
handler_->OnConnectionClosed(this);
break;
case protocol::Session::FAILED:
handler_->OnConnectionFailed(this);
break;
default:
// We shouldn't receive other states.
NOTREACHED();
}
}
// OnClosed() is used as a callback for protocol::Session::Close().
void ConnectionToClient::OnClosed() {
}
} // namespace protocol
} // namespace remoting
|
