Remove authenticated_ fields from stubs.

remoting/host/client_session.cc

 // 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/host/client_session.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "base/task.h"
+#include "media/base/callback.h"

[Alpha Left Google, 2011/03/28 16:22:26]

do you mean base/callback.h?

[simonmorris, 2011/03/29 17:07:51]

I wanted media/base/callback.h for AutoTaskRunner.

 #include "remoting/host/user_authenticator.h"
 #include "remoting/proto/auth.pb.h"
 
 namespace remoting {
 
 ClientSession::ClientSession(
     EventHandler* event_handler,
-    scoped_refptr<protocol::ConnectionToClient> connection)
+    const base::Callback<UserAuthenticator*(void)>& auth_factory,
+    scoped_refptr<protocol::ConnectionToClient> connection,
+    protocol::InputStub* input_stub)
     : event_handler_(event_handler),
-      connection_(connection) {
+      auth_factory_(auth_factory),
+      connection_(connection),
+      input_stub_(input_stub),
+      authenticated_(false) {
 }
 
 ClientSession::~ClientSession() {
 }
 
 void ClientSession::SuggestResolution(
     const protocol::SuggestResolutionRequest* msg, Task* done) {
-  done->Run();
-  delete done;
+  media::AutoTaskRunner task_runner(done);
+
+  if (!authenticated_) {
+    LOG(WARNING) << "Invalid control message received "
+                 << "(client not authenticated).";
+    return;
+  }
 }
 
 void ClientSession::BeginSessionRequest(
     const protocol::LocalLoginCredentials* credentials, Task* done) {
   DCHECK(event_handler_);
 
   bool success = false;
-  scoped_ptr<UserAuthenticator> authenticator(UserAuthenticator::Create());
+  scoped_ptr<UserAuthenticator> authenticator(auth_factory_.Run());
   switch (credentials->type()) {
     case protocol::PASSWORD:
       success = authenticator->Authenticate(credentials->username(),
                                             credentials->credential());
       break;
 
     default:
       LOG(ERROR) << "Invalid credentials type " << credentials->type();
       break;
   }
 
   if (success) {
+    authenticated_ = true;
     event_handler_->LocalLoginSucceeded(connection_.get());
   } else {
     LOG(WARNING) << "Login failed for user " << credentials->username();
     event_handler_->LocalLoginFailed(connection_.get());
   }
 
   done->Run();
   delete done;
 }
 
+void ClientSession::InjectKeyEvent(const protocol::KeyEvent* event,
+                                   Task* done) {
+  if (authenticated_) {
+    input_stub_->InjectKeyEvent(event, done);
+    return;
+  }
+  done->Run();
+  delete done;
+}
+
+void ClientSession::InjectMouseEvent(const protocol::MouseEvent* event,
+                                     Task* done) {
+  if (authenticated_) {
+    input_stub_->InjectMouseEvent(event, done);
+    return;
+  }
+  done->Run();
+  delete done;
+}
+
 void ClientSession::Disconnect() {
   connection_->Disconnect();
+  authenticated_ = false;
 }
 
 protocol::ConnectionToClient* ClientSession::connection() const {
   return connection_.get();
 }
 
+bool ClientSession::authenticated() const {
+  return authenticated_;
+}
+
 }  // namespace remoting