Implement InputStub in the host side for chromoting

remoting/protocol/connection_to_client.cc

 // 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/multiple_array_input_stream.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)
+                                       EventHandler* handler,
+                                       HostStub* host_stub,
+                                       InputStub* input_stub)
     : loop_(message_loop),
-      handler_(handler) {
+      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) {
@@ skipping 32 matching lines @@
     session_ = NULL;
   }
 }
 
 ConnectionToClient::ConnectionToClient() {}
 
 void ConnectionToClient::OnSessionStateChange(
     protocol::Session::State state) {
   if (state == protocol::Session::CONNECTED) {
     client_stub_.reset(new ClientControlSender(session_->control_channel()));
-    event_reader_.Init<ChromotingClientMessage>(
-        session_->event_channel(),
-        NewCallback(this, &ConnectionToClient::OnMessageReceived));
     video_writer_.reset(VideoWriter::Create(session_->config()));
     video_writer_->Init(session_);
+
+    dispatcher_.reset(new HostMessageDispatcher());
+    dispatcher_->Initialize(session_.get(), host_stub_, input_stub_);
   }
 
   loop_->PostTask(FROM_HERE,
       NewRunnableMethod(this, &ConnectionToClient::StateChangeTask, state));
 }
 
-void ConnectionToClient::OnMessageReceived(ChromotingClientMessage* message) {
-  loop_->PostTask(FROM_HERE,
-      NewRunnableMethod(this, &ConnectionToClient::MessageReceivedTask,
-                        message));
-}
-
 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();
   }
 }
 
-void ConnectionToClient::MessageReceivedTask(ChromotingClientMessage* message) {
-  DCHECK_EQ(loop_, MessageLoop::current());
-  DCHECK(handler_);
-  handler_->HandleMessage(this, message);
-}
-
 // OnClosed() is used as a callback for protocol::Session::Close().
 void ConnectionToClient::OnClosed() {
 }
 
 }  // namespace protocol
 }  // namespace remoting