cros: Prevent casual shut down and power save during migration.

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

 // 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.
 
 #ifndef CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_ENCRYPTION_MIGRATION_SCREEN_HANDLER_H_
 #define CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_ENCRYPTION_MIGRATION_SCREEN_HANDLER_H_
 
+#include <memory>
+
 #include "base/macros.h"
 #include "chrome/browser/chromeos/login/screens/encryption_migration_screen_view.h"
 #include "chrome/browser/ui/webui/chromeos/login/base_screen_handler.h"
 #include "chromeos/cryptohome/cryptohome_parameters.h"
 #include "chromeos/dbus/power_manager_client.h"
 #include "chromeos/login/auth/user_context.h"
 #include "third_party/cros_system_api/dbus/cryptohome/dbus-constants.h"
 
+namespace device {
+class PowerSaveBlocker;
+}  // namespace device
+
 namespace chromeos {
 
 // WebUI implementation of EncryptionMigrationScreenView
 class EncryptionMigrationScreenHandler : public EncryptionMigrationScreenView,
                                          public BaseScreenHandler,
                                          public PowerManagerClient::Observer {
  public:
   EncryptionMigrationScreenHandler();
   ~EncryptionMigrationScreenHandler() override;
 
@@ skipping 36 matching lines @@
   // Updates UI state.
   void UpdateUIState(UIState state);
 
   void CheckAvailableStorage();
   void OnGetAvailableStorage(int64_t size);
   void WaitBatteryAndMigrate();
   void StartMigration();
   void OnMountExistingVault(bool success,
                             cryptohome::MountError return_code,
                             const std::string& mount_hash);
+  void StartBlockingPowerSave();
+  void StopBlockingPowerSave();
+
   // Creates authorization key for MountEx method using |user_context_|.
   cryptohome::KeyDefinition GetAuthKey();
 
   // Handlers for cryptohome API callbacks.
   void OnMigrationProgress(cryptohome::DircryptoMigrationStatus status,
                            uint64_t current,
                            uint64_t total);
   void OnMigrationRequested(bool success);
 
   Delegate* delegate_ = nullptr;
@@ skipping 12 matching lines @@
   // True if the system should resume the previous incomplete migration.
   bool should_resume_ = false;
 
   // The current battery level.
   double current_battery_percent_ = 0.0;
 
   // True if the migration should start immediately once the battery level gets
   // sufficient.
   bool should_migrate_on_enough_battery_ = false;
 
+  std::unique_ptr<device::PowerSaveBlocker> power_save_blocker_;
+
   base::WeakPtrFactory<EncryptionMigrationScreenHandler> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(EncryptionMigrationScreenHandler);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_UI_WEBUI_CHROMEOS_LOGIN_ENCRYPTION_MIGRATION_SCREEN_HANDLER_H_