blob: daac39278a9323bfb27e9785ec88a2c808687755 (
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
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
|
// 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.
#ifndef REMOTING_HOST_CLIENT_CONNECTION_H_
#define REMOTING_HOST_CLIENT_CONNECTION_H_
#include <deque>
#include <vector>
#include "base/message_loop.h"
#include "base/ref_counted.h"
#include "base/scoped_ptr.h"
#include "remoting/proto/internal.pb.h"
#include "remoting/protocol/chromoting_connection.h"
#include "remoting/protocol/stream_reader.h"
#include "remoting/protocol/stream_writer.h"
namespace media {
class DataBuffer;
} // namespace media
namespace remoting {
// This class represents a remote viewer connected to the chromoting host
// through a libjingle connection. A viewer object is responsible for sending
// screen updates and other messages to the remote viewer. It is also
// responsible for receiving and parsing data from the remote viewer and
// delegating events to the event handler.
class ClientConnection : public base::RefCountedThreadSafe<ClientConnection> {
public:
class EventHandler {
public:
virtual ~EventHandler() {}
// Handles an event received by the ClientConnection. Receiver will own the
// ClientMessages in ClientMessageList and needs to delete them.
// Note that the sender of messages will not reference messages
// again so it is okay to clear |messages| in this method.
virtual void HandleMessage(ClientConnection* viewer,
ChromotingClientMessage* message) = 0;
// Called when the network connection is opened.
virtual void OnConnectionOpened(ClientConnection* viewer) = 0;
// Called when the network connection is closed.
virtual void OnConnectionClosed(ClientConnection* viewer) = 0;
// Called when the network connection has failed.
virtual void OnConnectionFailed(ClientConnection* viewer) = 0;
};
// Constructs a ClientConnection object. |message_loop| is the message loop
// that this object runs on. A viewer object receives events and messages from
// a libjingle channel, these events are delegated to |handler|.
// It is guranteed that |handler| is called only on the |message_loop|.
ClientConnection(MessageLoop* message_loop,
EventHandler* handler);
virtual ~ClientConnection();
virtual void Init(ChromotingConnection* connection);
// Returns the connection in use.
virtual ChromotingConnection* connection() { return connection_; }
// Send information to the client for initialization.
virtual void SendInitClientMessage(int width, int height);
// Send encoded update stream data to the viewer.
virtual void SendUpdateStreamPacketMessage(
const ChromotingHostMessage& message);
// Gets the number of update stream messages not yet transmitted.
// Note that the value returned is an estimate using average size of the
// most recent update streams.
// TODO(hclam): Report this number accurately.
virtual int GetPendingUpdateStreamMessages();
// Disconnect the client connection. This method is allowed to be called
// more than once and calls after the first one will be ignored.
//
// After this method is called all the send method calls will be ignored.
virtual void Disconnect();
protected:
// Protected constructor used by unit test.
ClientConnection();
private:
// Callback for ChromotingConnection.
void OnConnectionStateChange(ChromotingConnection::State state);
// Callback for EventsStreamReader.
void OnMessageReceived(ChromotingClientMessage* message);
// Process a libjingle state change event on the |loop_|.
void StateChangeTask(ChromotingConnection::State state);
// Process a data buffer received from libjingle.
void MessageReceivedTask(ChromotingClientMessage* message);
void OnClosed();
// The libjingle channel used to send and receive data from the remote client.
scoped_refptr<ChromotingConnection> connection_;
EventsStreamReader events_reader_;
VideoStreamWriter video_writer_;
// The message loop that this object runs on.
MessageLoop* loop_;
// Event handler for handling events sent from this object.
EventHandler* handler_;
DISALLOW_COPY_AND_ASSIGN(ClientConnection);
};
} // namespace remoting
#endif // REMOTING_HOST_CLIENT_CONNECTION_H_
|