A new authenticated connection evicts an old one.

remoting/protocol/connection_to_client.cc

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// 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/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"
 #include "remoting/protocol/host_stub.h"
 #include "remoting/protocol/input_stub.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)
     : client_authenticated_(false),
       loop_(message_loop),
       handler_(handler),
-      host_stub_(host_stub),
+      host_stub_(NULL),
       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.
 }
 
@@ skipping 27 matching lines @@
 
 VideoStub* ConnectionToClient::video_stub() {
   return video_writer_.get();
 }
 
 // Return pointer to ClientStub.
 ClientStub* ConnectionToClient::client_stub() {
   return client_stub_.get();
 }
 
+protocol::HostStub* ConnectionToClient::host_stub() {
+  return host_stub_;
+}
+
+void ConnectionToClient::set_host_stub(protocol::HostStub* host_stub) {
+  host_stub_ = host_stub;
+}
+
 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_);
   }
 
@@ skipping 49 matching lines @@
 
   // Enable/disable each of the channels.
   if (input_stub_)
     input_stub_->OnAuthenticated();
   if (host_stub_)
     host_stub_->OnAuthenticated();
   if (client_stub_.get())
     client_stub_->OnAuthenticated();
 }
 
+bool ConnectionToClient::client_authenticated() {
+  return client_authenticated_;
+}
+
 }  // namespace protocol
 }  // namespace remoting