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) {
+  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);

[Daniel Erat, 2017/05/30 23:41:47]

just out of curiosity, are you currently seeing the system shut down when you
close the lid while the migration is in-progress? we usually shut down while at
the login screen and suspend after the user has logged in (although enterprise
policies can override this), so it'd depend on how far chrome thinks we've
gotten in the login process when the migration begins (which i'm not sure of).

[dspaid, 2017/05/30 23:46:39]

Yes, we do observe shut down on lid close for stock profiles.  We interrupt the
login process very early on, so user and enterprise policies have not yet been
applied.

+    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