Implemented a switch for bypassing platform verification consent UI.

chrome/browser/chromeos/attestation/platform_verification_flow.cc

 // Copyright 2013 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 "platform_verification_flow.h"
 
+#include "base/command_line.h"
 #include "base/logging.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/chromeos/attestation/attestation_ca_client.h"
 #include "chrome/browser/chromeos/login/user_manager.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/system/statistics_provider.h"
 #include "chrome/browser/prefs/scoped_user_pref_update.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/attestation/attestation_flow.h"
 #include "chromeos/cryptohome/async_method_caller.h"
 #include "chromeos/dbus/cryptohome_client.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "components/user_prefs/pref_registry_syncable.h"
 #include "components/user_prefs/user_prefs.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/web_contents.h"
 
 namespace {
+// A switch which allows consent to be given on the command line.
+// TODO(dkrahn): Remove this when UI has been implemented.

[Mattias Nissler (ping if slow), 2013/09/06 13:40:21]

nit: Can you reference the bug that tracks UI implementation here?

[Darren Krahn, 2013/09/06 15:52:59]

Done.

+const char kAutoApproveSwitch[] =
+    "auto-approve-platform-verification-consent-prompts";
+
 // A callback method to handle DBus errors.
 void DBusCallback(const base::Callback<void(bool)>& on_success,
                   const base::Closure& on_failure,
                   chromeos::DBusMethodCallStatus call_status,
                   bool result) {
   if (call_status == chromeos::DBUS_METHOD_CALL_SUCCESS) {
     on_success.Run(result);
   } else {
     LOG(ERROR) << "PlatformVerificationFlow: DBus call failed!";
     on_failure.Run();
   }
 }
 }  // namespace
 
 namespace chromeos {
 namespace attestation {
 
+// A default implementation of the Delegate interface.
+class DefaultDelegate : public PlatformVerificationFlow::Delegate {
+ public:
+  DefaultDelegate() {}
+  virtual ~DefaultDelegate() {}
+
+  virtual void ShowConsentPrompt(
+      PlatformVerificationFlow::ConsentType type,
+      content::WebContents* web_contents,
+      const PlatformVerificationFlow::Delegate::ConsentCallback& callback) {
+    if (CommandLine::ForCurrentProcess()->HasSwitch(kAutoApproveSwitch)) {
+      LOG(WARNING) << "PlatformVerificationFlow: Automatic approval enabled.";
+      callback.Run(PlatformVerificationFlow::CONSENT_RESPONSE_ALLOW);
+    } else {
+      LOG(ERROR) << "PlatformVerificationFlow: Consent prompt not implemented.";

[Mattias Nissler (ping if slow), 2013/09/06 13:40:21]

NOTIMPLEMENTED() might be the thing you're looking for :)

[Darren Krahn, 2013/09/06 15:52:59]

Done.

+      callback.Run(PlatformVerificationFlow::CONSENT_RESPONSE_NONE);
+    }
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(DefaultDelegate);
+};
+
+PlatformVerificationFlow::PlatformVerificationFlow()
+    : attestation_flow_(NULL),
+      async_caller_(cryptohome::AsyncMethodCaller::GetInstance()),
+      cryptohome_client_(DBusThreadManager::Get()->GetCryptohomeClient()),
+      user_manager_(UserManager::Get()),
+      delegate_(NULL),
+      testing_prefs_(NULL),
+      weak_factory_(this) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  scoped_ptr<ServerProxy> attestation_ca_client(new AttestationCAClient());
+  default_attestation_flow_.reset(new AttestationFlow(
+      async_caller_,
+      cryptohome_client_,
+      attestation_ca_client.Pass()));
+  attestation_flow_ = default_attestation_flow_.get();
+  default_delegate_.reset(new DefaultDelegate());
+  delegate_ = default_delegate_.get();
+}
+
 PlatformVerificationFlow::PlatformVerificationFlow(
     AttestationFlow* attestation_flow,
     cryptohome::AsyncMethodCaller* async_caller,
     CryptohomeClient* cryptohome_client,
     UserManager* user_manager,
     system::StatisticsProvider* statistics_provider,
     Delegate* delegate)
     : attestation_flow_(attestation_flow),
       async_caller_(async_caller),
       cryptohome_client_(cryptohome_client),
       user_manager_(user_manager),
       statistics_provider_(statistics_provider),
       delegate_(delegate),
+      testing_prefs_(NULL),
       weak_factory_(this) {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 }
 
 PlatformVerificationFlow::~PlatformVerificationFlow() {
 }
 
 void PlatformVerificationFlow::ChallengePlatformKey(
     content::WebContents* web_contents,
     const std::string& service_id,
@@ skipping 279 matching lines @@
     bool allow_domain) {
   PrefService* pref_service = GetPrefs(web_contents);
   CHECK(pref_service);
   DictionaryPrefUpdate updater(pref_service, prefs::kRAConsentDomains);
   const GURL& url = GetURL(web_contents);
   updater->SetBoolean(url.host(), allow_domain);
 }
 
 }  // namespace attestation
 }  // namespace chromeos