Add WakeLock during Plugin VM installation

Download for most URLs will already set WakeLock in ResourceDownloader,
but for Drive downloads, and then for the post-download installation we
want to set the WakeLock to avoid the devices sleeping.

Bug: b/155345403
Change-Id: I62a1b6abdc25c0f90ec26df4d5152e7e07d5414d
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2398747Reviewed-by: Jason Lin <lxj@google.com>
Commit-Queue: Joel Hockey <joelhockey@chromium.org>
Cr-Commit-Position: refs/heads/master@{#805620}

chrome/browser/chromeos/plugin_vm/plugin_vm_installer.cc

@@ -35,8 +35,10 @@
 #include "components/prefs/pref_service.h"
 #include "content/public/browser/browser_task_traits.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/device_service.h"
 #include "content/public/browser/network_service_instance.h"
 #include "net/traffic_annotation/network_traffic_annotation.h"
+#include "services/device/public/mojom/wake_lock_provider.mojom.h"
 
 namespace plugin_vm {
 
@@ -171,6 +173,9 @@ void PluginVmInstaller::Cancel() {
 }
 
 void PluginVmInstaller::CheckLicense() {
+  // Request wake lock when state_ goes to kInstalling, and cancel it when state
+  // goes back to kIdle.
+  GetWakeLock()->RequestWakeLock();
   state_ = State::kInstalling;
   UpdateInstallingState(InstallingState::kCheckingLicense);
 
@@ -920,6 +925,7 @@ void PluginVmInstaller::OnTemporaryImageRemoved(bool success) {
 void PluginVmInstaller::CancelFinished() {
   DCHECK_EQ(state_, State::kCancelling);
   state_ = State::kIdle;
+  GetWakeLock()->CancelWakeLock();
   installing_state_ = InstallingState::kInactive;
 
   if (observer_)
@@ -928,6 +934,7 @@ void PluginVmInstaller::CancelFinished() {
 
 void PluginVmInstaller::InstallFailed(FailureReason reason) {
   state_ = State::kIdle;
+  GetWakeLock()->CancelWakeLock();
   installing_state_ = InstallingState::kInactive;
   base::UmaHistogramEnumeration(kFailureReasonHistogram, reason);
   RecordPluginVmSetupResultHistogram(PluginVmSetupResult::kError);
@@ -939,7 +946,21 @@ void PluginVmInstaller::InstallFinished() {
   base::UmaHistogramLongTimes(kSetupTimeHistogram,
                               base::TimeTicks::Now() - setup_start_tick_);
   state_ = State::kIdle;
+  GetWakeLock()->CancelWakeLock();
   installing_state_ = InstallingState::kInactive;
 }
 
+device::mojom::WakeLock* PluginVmInstaller::GetWakeLock() {
+  if (!wake_lock_) {
+    mojo::Remote<device::mojom::WakeLockProvider> wake_lock_provider;
+    content::GetDeviceService().BindWakeLockProvider(
+        wake_lock_provider.BindNewPipeAndPassReceiver());
+    wake_lock_provider->GetWakeLockWithoutContext(
+        device::mojom::WakeLockType::kPreventAppSuspension,
+        device::mojom::WakeLockReason::kOther, "Plugin VM Installer",
+        wake_lock_.BindNewPipeAndPassReceiver());
+  }
+  return wake_lock_.get();
+}
+
 }  // namespace plugin_vm

chrome/browser/chromeos/plugin_vm/plugin_vm_installer.h

@@ -16,6 +16,8 @@
 #include "chromeos/dbus/dlcservice/dlcservice_client.h"
 #include "components/download/public/background_service/download_params.h"
 #include "components/keyed_service/core/keyed_service.h"
+#include "mojo/public/cpp/bindings/remote.h"
+#include "services/device/public/mojom/wake_lock.mojom.h"
 
 namespace download {
 class DownloadService;
@@ -265,6 +267,10 @@ class PluginVmInstaller : public KeyedService,
   void RemoveTemporaryImageIfExists();
   void OnTemporaryImageRemoved(bool success);
 
+  // Keep the system awake during installation.
+  device::mojom::WakeLock* GetWakeLock();
+  mojo::Remote<device::mojom::WakeLock> wake_lock_;
+
   base::WeakPtrFactory<PluginVmInstaller> weak_ptr_factory_{this};
 
   DISALLOW_COPY_AND_ASSIGN(PluginVmInstaller);