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 80 matching lines @@
   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 (encryption_migration_active_ &&

[Daniel Erat, 2017/05/30 13:39:42]

thinking about this a bit more, i think i misspoke earlier, and this block
should come immediately before the chrome_is_exiting_ block. that is, if the lid
is closed while chrome is both migrating and exiting, we want to do nothing (so
the exit can finish) instead of suspending. sorry, mind moving this up a bit?

[dspaid, 2017/05/30 23:37:22]

Went ahead and moved it up, though in this case I don't think it matters since
if chrome_is_exiting_ is true then it will set the action to DO_NOTHING, and we
only suspend if the current policy is not DO_NOTHING.

+      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";
+  }
+
   if (!causes.empty())
     policy.set_reason(causes);
   client_->SetPolicy(policy);
 }
 
 }  // namespace chromeos