OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "net/base/io_buffer.h" | 5 #include "net/base/io_buffer.h" |
6 #include "remoting/base/multiple_array_input_stream.h" | 6 #include "remoting/base/multiple_array_input_stream.h" |
7 #include "remoting/proto/control.pb.h" | 7 #include "remoting/proto/control.pb.h" |
8 #include "remoting/proto/event.pb.h" | 8 #include "remoting/proto/event.pb.h" |
9 #include "remoting/proto/video.pb.h" | |
10 #include "remoting/protocol/chromotocol_connection.h" | |
11 #include "remoting/protocol/host_message_dispatcher.h" | 9 #include "remoting/protocol/host_message_dispatcher.h" |
12 #include "remoting/protocol/host_stub.h" | 10 #include "remoting/protocol/host_stub.h" |
13 #include "remoting/protocol/input_stub.h" | 11 #include "remoting/protocol/input_stub.h" |
14 #include "remoting/protocol/message_reader.h" | 12 #include "remoting/protocol/message_reader.h" |
| 13 #include "remoting/protocol/session.h" |
15 | 14 |
16 namespace { | 15 namespace { |
17 | 16 |
18 // A single protobuf can contain multiple messages that will be handled by | 17 // A single protobuf can contain multiple messages that will be handled by |
19 // different message handlers. We use this wrapper to ensure that the | 18 // different message handlers. We use this wrapper to ensure that the |
20 // protobuf is only deleted after all the handlers have finished executing. | 19 // protobuf is only deleted after all the handlers have finished executing. |
21 template <typename T> | 20 template <typename T> |
22 class RefCountedMessage : public base::RefCounted<RefCountedMessage<T> > { | 21 class RefCountedMessage : public base::RefCounted<RefCountedMessage<T> > { |
23 public: | 22 public: |
24 RefCountedMessage(T* message) : message_(message) { } | 23 RefCountedMessage(T* message) : message_(message) { } |
(...skipping 20 matching lines...) Expand all Loading... |
45 | 44 |
46 HostMessageDispatcher::HostMessageDispatcher() : | 45 HostMessageDispatcher::HostMessageDispatcher() : |
47 host_stub_(NULL), | 46 host_stub_(NULL), |
48 input_stub_(NULL) { | 47 input_stub_(NULL) { |
49 } | 48 } |
50 | 49 |
51 HostMessageDispatcher::~HostMessageDispatcher() { | 50 HostMessageDispatcher::~HostMessageDispatcher() { |
52 } | 51 } |
53 | 52 |
54 bool HostMessageDispatcher::Initialize( | 53 bool HostMessageDispatcher::Initialize( |
55 ChromotocolConnection* connection, | 54 protocol::Session* session, |
56 HostStub* host_stub, InputStub* input_stub) { | 55 HostStub* host_stub, InputStub* input_stub) { |
57 if (!connection || !host_stub || !input_stub || | 56 if (!session || !host_stub || !input_stub || |
58 !connection->event_channel() || !connection->control_channel()) { | 57 !session->event_channel() || !session->control_channel()) { |
59 return false; | 58 return false; |
60 } | 59 } |
61 | 60 |
62 control_message_reader_.reset(new MessageReader()); | 61 control_message_reader_.reset(new MessageReader()); |
63 event_message_reader_.reset(new MessageReader()); | 62 event_message_reader_.reset(new MessageReader()); |
64 host_stub_ = host_stub; | 63 host_stub_ = host_stub; |
65 input_stub_ = input_stub; | 64 input_stub_ = input_stub; |
66 | 65 |
67 // Initialize the readers on the sockets provided by channels. | 66 // Initialize the readers on the sockets provided by channels. |
68 event_message_reader_->Init<EventMessage>( | 67 event_message_reader_->Init<EventMessage>( |
69 connection->event_channel(), | 68 session->event_channel(), |
70 NewCallback(this, &HostMessageDispatcher::OnEventMessageReceived)); | 69 NewCallback(this, &HostMessageDispatcher::OnEventMessageReceived)); |
71 control_message_reader_->Init<ControlMessage>( | 70 control_message_reader_->Init<ControlMessage>( |
72 connection->control_channel(), | 71 session->control_channel(), |
73 NewCallback(this, &HostMessageDispatcher::OnControlMessageReceived)); | 72 NewCallback(this, &HostMessageDispatcher::OnControlMessageReceived)); |
74 return true; | 73 return true; |
75 } | 74 } |
76 | 75 |
77 void HostMessageDispatcher::OnControlMessageReceived(ControlMessage* message) { | 76 void HostMessageDispatcher::OnControlMessageReceived(ControlMessage* message) { |
78 scoped_refptr<RefCountedMessage<ControlMessage> > ref_msg = | 77 scoped_refptr<RefCountedMessage<ControlMessage> > ref_msg = |
79 new RefCountedMessage<ControlMessage>(message); | 78 new RefCountedMessage<ControlMessage>(message); |
80 if (message->has_suggest_resolution()) { | 79 if (message->has_suggest_resolution()) { |
81 host_stub_->SuggestResolution( | 80 host_stub_->SuggestResolution( |
82 message->suggest_resolution(), NewDeleteTask(ref_msg)); | 81 message->suggest_resolution(), NewDeleteTask(ref_msg)); |
83 } | 82 } |
84 } | 83 } |
85 | 84 |
86 void HostMessageDispatcher::OnEventMessageReceived( | 85 void HostMessageDispatcher::OnEventMessageReceived( |
87 EventMessage* message) { | 86 EventMessage* message) { |
88 // TODO(hclam): Implement. | 87 // TODO(hclam): Implement. |
89 } | 88 } |
90 | 89 |
91 } // namespace protocol | 90 } // namespace protocol |
92 } // namespace remoting | 91 } // namespace remoting |
OLD | NEW |