Force encryption migration if the device supports ARC.

chrome/browser/ui/webui/chromeos/login/encryption_migration_screen_handler.cc

 // Copyright 2017 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 "chrome/browser/ui/webui/chromeos/login/encryption_migration_screen_handler.h"
 
+#include <string>
+#include <utility>
+
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chromeos/cryptohome/homedir_methods.h"
 #include "chromeos/dbus/cryptohome_client.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 
 namespace {
 
 constexpr char kJsScreenPath[] = "login.EncryptionMigrationScreen";
 
 // JS API callbacks names.
 constexpr char kJsApiStartMigration[] = "startMigration";
+constexpr char kJsApiSkipMigration[] = "skipMigration";
 constexpr char kJsApiRequestRestart[] = "requestRestart";
 
 }  // namespace
 
 namespace chromeos {
 
 EncryptionMigrationScreenHandler::EncryptionMigrationScreenHandler()
     : BaseScreenHandler(kScreenId), weak_ptr_factory_(this) {
   set_call_js_prefix(kJsScreenPath);
 }
@@ skipping 19 matching lines @@
   delegate_ = delegate;
   if (page_is_ready())
     Initialize();
 }
 
 void EncryptionMigrationScreenHandler::SetUserContext(
     const UserContext& user_context) {
   user_context_ = user_context;
 }
 
+void EncryptionMigrationScreenHandler::SetContinueLoginCallback(
+    ContinueLoginCallback callback) {
+  continue_login_callback_ = std::move(callback);
+}
+
 void EncryptionMigrationScreenHandler::DeclareLocalizedValues(
     ::login::LocalizedValuesBuilder* builder) {}
 
 void EncryptionMigrationScreenHandler::Initialize() {
   if (!page_is_ready() || !delegate_)
     return;
 
   if (show_on_init_) {
     Show();
     show_on_init_ = false;
   }
 }
 
 void EncryptionMigrationScreenHandler::RegisterMessages() {
   AddCallback(kJsApiStartMigration,
               &EncryptionMigrationScreenHandler::HandleStartMigration);
+  AddCallback(kJsApiSkipMigration,
+              &EncryptionMigrationScreenHandler::HandleSkipMigration);
   AddCallback(kJsApiRequestRestart,
               &EncryptionMigrationScreenHandler::HandleRequestRestart);
 }
 
 void EncryptionMigrationScreenHandler::HandleStartMigration() {
   StartMigration();
 }
 
+void EncryptionMigrationScreenHandler::HandleSkipMigration() {
+  // If the user skips migration, we mount the cryptohome without performing the
+  // migration by reusing UserContext and LoginPerformer which were used in the
+  // previous attempt and dropping |is_forcing_dircrypto| flag in UserContext.
+  // In this case, the user can not launch ARC apps in the session, and will be
+  // asked to do the migration again in the next log-in attempt.
+  if (!continue_login_callback_.is_null()) {
+    user_context_.SetIsForcingDircrypto(false);
+    std::move(continue_login_callback_).Run(user_context_);
+  }
+}
+
 void EncryptionMigrationScreenHandler::HandleRequestRestart() {
   // TODO(fukino): If the migration finished successfully, we don't need to
   // restart the device. Let's sign in to the desktop using the already-provided
   // user credential.
   chrome::AttemptRestart();
 }
 
 void EncryptionMigrationScreenHandler::UpdateUIState(UIState state) {
   if (state == current_ui_state_)
     return;
@@ skipping 19 matching lines @@
   // Chrome OS M38 and older will have a legacy key with no label while those
   // created by Chrome OS M39 and newer will have a key with the label
   // kCryptohomeGAIAKeyLabel.
   const cryptohome::KeyDefinition auth_key(key->GetSecret(), std::string(),
                                            cryptohome::PRIV_DEFAULT);
   cryptohome::HomedirMethods::GetInstance()->MigrateToDircrypto(
       cryptohome::Identification(user_context_.GetAccountId()),
       cryptohome::Authorization(auth_key),
       base::Bind(&EncryptionMigrationScreenHandler::OnMigrationRequested,
                  weak_ptr_factory_.GetWeakPtr()));
+  UpdateUIState(UIState::MIGRATING);
 }
 
 void EncryptionMigrationScreenHandler::OnMigrationProgress(
     cryptohome::DircryptoMigrationStatus status,
     uint64_t current,
     uint64_t total) {
   switch (status) {
     case cryptohome::DIRCRYPTO_MIGRATION_INITIALIZING:
       UpdateUIState(UIState::MIGRATING);
       break;
@@ skipping 20 matching lines @@
 void EncryptionMigrationScreenHandler::OnMigrationRequested(bool success) {
   // This function is called when MigrateToDircrypto is correctly requested.
   // It does not mean that the migration is completed. We should know the
   // completion by DircryptoMigrationProgressHandler. success == false means a
   // failure in DBus communication.
   // TODO(fukino): Handle this case. Should we retry or restart?
   DCHECK(success);
 }
 
 }  // namespace chromeos