cros: Prevent shut down on lid-close during migration.

chromeos/dbus/power_policy_controller.cc

 // Copyright (c) 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 "chromeos/dbus/power_policy_controller.h"
 
 #include <stdint.h>
 
 #include <utility>
 
@@ skipping 258 matching lines @@
   SendCurrentPolicy();
 }
 
 void PowerPolicyController::NotifyChromeIsExiting() {
   if (chrome_is_exiting_)
     return;
   chrome_is_exiting_ = true;
   SendCurrentPolicy();
 }
 
+void PowerPolicyController::SetEncryptionMigrationActive(bool active) {
+  if (encryption_migration_active_ == active)
+    return;
+
+  encryption_migration_active_ = active;
+  SendCurrentPolicy();
+}
+
 PowerPolicyController::PowerPolicyController(PowerManagerClient* client)
     : client_(client),
       prefs_were_set_(false),
       honor_screen_wake_locks_(true),
       next_wake_lock_id_(1),
-      chrome_is_exiting_(false) {
+      chrome_is_exiting_(false),
+      encryption_migration_active_(false) {
   DCHECK(client_);
   client_->AddObserver(this);
 }
 
 PowerPolicyController::~PowerPolicyController() {
   client_->RemoveObserver(this);
 }
 
 PowerPolicyController::WakeLock::WakeLock(Type type,
                                           WakeLockReason reason,
@@ skipping 65 matching lines @@
       policy.set_ac_idle_action(
           power_manager::PowerManagementPolicy_Action_DO_NOTHING);
     }
     if (!policy.has_battery_idle_action() || policy.battery_idle_action() ==
         power_manager::PowerManagementPolicy_Action_SUSPEND) {
       policy.set_battery_idle_action(
           power_manager::PowerManagementPolicy_Action_DO_NOTHING);
     }
   }
 
+  if (encryption_migration_active_ &&
+      policy.lid_closed_action() !=
+          power_manager::PowerManagementPolicy_Action_DO_NOTHING) {
+    policy.set_lid_closed_action(
+        power_manager::PowerManagementPolicy_Action_SUSPEND);
+    causes +=
+        std::string((causes.empty() ? "" : ", ")) + "encryption migration";
+  }
+
   // To avoid a race in the case where the user asks Chrome to sign out
   // and then immediately closes the lid, override the lid-closed action
   // so the system will stay awake while Chrome is exiting. When Chrome
   // restarts to display the login screen, it will send an updated
   // policy that powerd can act on.
   if (chrome_is_exiting_ &&
       (!policy.has_lid_closed_action() ||
        policy.lid_closed_action() ==
            power_manager::PowerManagementPolicy_Action_SUSPEND ||
        policy.lid_closed_action() ==
            power_manager::PowerManagementPolicy_Action_SHUT_DOWN)) {
     policy.set_lid_closed_action(
         power_manager::PowerManagementPolicy_Action_DO_NOTHING);
   }
 
   if (!causes.empty())
     policy.set_reason(causes);
   client_->SetPolicy(policy);
 }
 
 }  // namespace chromeos