Block event processing on host/client until the client has authenticated.

remoting/protocol/connection_to_host.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_host.h"
 
 #include "base/callback.h"
 #include "base/message_loop.h"
 #include "remoting/base/constants.h"
 #include "remoting/jingle_glue/jingle_thread.h"
 #include "remoting/proto/auth.pb.h"
 #include "remoting/protocol/client_message_dispatcher.h"
 #include "remoting/protocol/client_stub.h"
 #include "remoting/protocol/host_control_sender.h"
 #include "remoting/protocol/input_sender.h"
 #include "remoting/protocol/jingle_session_manager.h"
 #include "remoting/protocol/video_reader.h"
 #include "remoting/protocol/video_stub.h"
 #include "remoting/protocol/util.h"
 
 namespace remoting {
 namespace protocol {
 
 ConnectionToHost::ConnectionToHost(JingleThread* thread)
-    : thread_(thread),
+    : client_authenticated_(false),
+      thread_(thread),
       event_callback_(NULL),
       dispatcher_(new ClientMessageDispatcher()) {
 }
 
 ConnectionToHost::~ConnectionToHost() {
 }
 
 InputStub* ConnectionToHost::input_stub() {
   return input_stub_.get();
 }
@@ skipping 153 matching lines @@
       dispatcher_->Initialize(session_.get(), client_stub_);
       event_callback_->OnConnectionOpened(this);
       break;
 
     default:
       // Ignore the other states by default.
       break;
   }
 }
 
+void ConnectionToHost::SetClientAuthenticated(bool auth) {
+  client_authenticated_ = auth;
+
+  // Enable/disable each of the channels.
+  if (input_stub_.get())
+    input_stub_->SetEnabled(auth);
+  if (host_stub_.get())
+    host_stub_->SetEnabled(auth);
+  if (client_stub_)
+    client_stub_->SetEnabled(auth);
+}
+
 }  // namespace protocol
 }  // namespace remoting