borealis: Refactor tasks to be sequence-mutually exclusive

Tasks are deleted on completion (even before crrev.com/c/2507234), this
caused a UAF when a member-object of the AwaitBorealisStartup task was
transitively deleted but its callee.

The chosen solution is to post completion as a separate task rather than
call it from the current completing BorealisTask, we do this is a
mutually-exclusive way so that the deletion of the task is
exclusively-after its completion.

Bug: b/172605172
Change-Id: Iaf23e9ead37196988b24753cef451da318c48436
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2521876Reviewed-by: Daniel Ng <danielng@google.com>
Commit-Queue: Nic Hollingum <hollingum@google.com>
Cr-Commit-Position: refs/heads/master@{#824745}

chrome/browser/chromeos/borealis/borealis_context_manager_unittest.cc

@@ -36,14 +36,13 @@ MATCHER(IsFailureResult, "") {
 class MockTask : public BorealisTask {
  public:
   explicit MockTask(bool success) : success_(success) {}
-  void Run(BorealisContext* context,
-           BorealisTask::CompletionStatusCallback callback) override {
+  void RunInternal(BorealisContext* context) override {
     if (success_) {
       context->set_vm_name("test_vm_name");
-      std::move(callback).Run(BorealisContextManager::Status::kSuccess, "");
+      Complete(BorealisContextManager::Status::kSuccess, "");
     } else {
       // Just use a random error.
-      std::move(callback).Run(BorealisContextManager::Status::kStartVmFailed,
+      Complete(BorealisContextManager::Status::kStartVmFailed,
                "Something went wrong!");
     }
   }
@@ -211,8 +210,7 @@ class NeverCompletingContextManager : public BorealisContextManagerImpl {
  private:
   class NeverCompletingTask : public BorealisTask {
    public:
-    void Run(BorealisContext* context,
-             CompletionStatusCallback callback) override {}
+    void RunInternal(BorealisContext* context) override {}
   };
 
   base::queue<std::unique_ptr<BorealisTask>> GetTasks() override {

chrome/browser/chromeos/borealis/borealis_task.cc

@@ -5,8 +5,10 @@
 #include "chrome/browser/chromeos/borealis/borealis_task.h"
 #include <string>
 
+#include "base/location.h"
 #include "base/logging.h"
 #include "base/strings/string_number_conversions.h"
+#include "base/threading/sequenced_task_runner_handle.h"
 #include "chrome/browser/chromeos/borealis/borealis_context.h"
 #include "chrome/browser/chromeos/borealis/borealis_util.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
@@ -17,39 +19,54 @@
 
 namespace borealis {
 
+BorealisTask::BorealisTask() = default;
+
+BorealisTask::~BorealisTask() = default;
+
+void BorealisTask::Run(BorealisContext* context,
+                       CompletionStatusCallback callback) {
+  callback_ = std::move(callback);
+  RunInternal(context);
+}
+
+void BorealisTask::Complete(BorealisContextManager::Status status,
+                            std::string message) {
+  // Task completion is self-mutually-exclusive, because tasks are deleted once
+  // complete.
+  base::SequencedTaskRunnerHandle::Get()->PostTask(
+      FROM_HERE,
+      base::BindOnce(std::move(callback_), status, std::move(message)));
+}
+
 MountDlc::MountDlc() = default;
 MountDlc::~MountDlc() = default;
 
-void MountDlc::Run(BorealisContext* context,
-                   CompletionStatusCallback callback) {
+void MountDlc::RunInternal(BorealisContext* context) {
   // TODO(b/172279567): Ensure the DLC is present before trying to install,
   // otherwise we will silently download borealis here.
   chromeos::DlcserviceClient::Get()->Install(
       kBorealisDlcName,
-      base::BindOnce(&MountDlc::OnMountDlc, weak_factory_.GetWeakPtr(), context,
-                     std::move(callback)),
+      base::BindOnce(&MountDlc::OnMountDlc, weak_factory_.GetWeakPtr(),
+                     context),
       base::DoNothing());
 }
 
 void MountDlc::OnMountDlc(
     BorealisContext* context,
-    CompletionStatusCallback callback,
     const chromeos::DlcserviceClient::InstallResult& install_result) {
   if (install_result.error != dlcservice::kErrorNone) {
-    std::move(callback).Run(
-        BorealisContextManager::Status::kMountFailed,
+    Complete(BorealisContextManager::Status::kMountFailed,
              "Mounting the DLC for Borealis failed: " + install_result.error);
   } else {
     context->set_root_path(install_result.root_path);
-    std::move(callback).Run(BorealisContextManager::Status::kSuccess, "");
+    Complete(BorealisContextManager::Status::kSuccess, "");
   }
 }
 
 CreateDiskImage::CreateDiskImage() = default;
 CreateDiskImage::~CreateDiskImage() = default;
 
-void CreateDiskImage::Run(BorealisContext* context,
-                          CompletionStatusCallback callback) {
+void CreateDiskImage::RunInternal(BorealisContext* context) {
   vm_tools::concierge::CreateDiskImageRequest request;
   request.set_disk_path(context->vm_name());
   request.set_cryptohome_id(
@@ -59,19 +76,16 @@ void CreateDiskImage::Run(BorealisContext* context,
   request.set_disk_size(0);
 
   chromeos::DBusThreadManager::Get()->GetConciergeClient()->CreateDiskImage(
-      std::move(request),
-      base::BindOnce(&CreateDiskImage::OnCreateDiskImage,
-                     weak_factory_.GetWeakPtr(), context, std::move(callback)));
+      std::move(request), base::BindOnce(&CreateDiskImage::OnCreateDiskImage,
+                                         weak_factory_.GetWeakPtr(), context));
 }
 
 void CreateDiskImage::OnCreateDiskImage(
     BorealisContext* context,
-    CompletionStatusCallback callback,
     base::Optional<vm_tools::concierge::CreateDiskImageResponse> response) {
   if (!response) {
     context->set_disk_path(base::FilePath());
-    std::move(callback).Run(
-        BorealisContextManager::Status::kDiskImageFailed,
+    Complete(BorealisContextManager::Status::kDiskImageFailed,
              "Failed to create disk image for Borealis: Empty response.");
     return;
   }
@@ -79,20 +93,19 @@ void CreateDiskImage::OnCreateDiskImage(
   if (response->status() != vm_tools::concierge::DISK_STATUS_EXISTS &&
       response->status() != vm_tools::concierge::DISK_STATUS_CREATED) {
     context->set_disk_path(base::FilePath());
-    std::move(callback).Run(BorealisContextManager::Status::kDiskImageFailed,
+    Complete(BorealisContextManager::Status::kDiskImageFailed,
              "Failed to create disk image for Borealis: " +
                  response->failure_reason());
     return;
   }
   context->set_disk_path(base::FilePath(response->disk_path()));
-  std::move(callback).Run(BorealisContextManager::Status::kSuccess, "");
+  Complete(BorealisContextManager::Status::kSuccess, "");
 }
 
 StartBorealisVm::StartBorealisVm() = default;
 StartBorealisVm::~StartBorealisVm() = default;
 
-void StartBorealisVm::Run(BorealisContext* context,
-                          CompletionStatusCallback callback) {
+void StartBorealisVm::RunInternal(BorealisContext* context) {
   vm_tools::concierge::StartVmRequest request;
   vm_tools::concierge::VirtualMachineSpec* vm = request.mutable_vm();
   vm->set_kernel(context->root_path() + "/vm_kernel");
@@ -119,31 +132,28 @@ void StartBorealisVm::Run(BorealisContext* context,
                        ? "enabled"
                        : "disabled");
   chromeos::DBusThreadManager::Get()->GetConciergeClient()->StartTerminaVm(
-      std::move(request),
-      base::BindOnce(&StartBorealisVm::OnStartBorealisVm,
-                     weak_factory_.GetWeakPtr(), context, std::move(callback)));
+      std::move(request), base::BindOnce(&StartBorealisVm::OnStartBorealisVm,
+                                         weak_factory_.GetWeakPtr(), context));
 }
 
 void StartBorealisVm::OnStartBorealisVm(
     BorealisContext* context,
-    CompletionStatusCallback callback,
     base::Optional<vm_tools::concierge::StartVmResponse> response) {
   if (!response) {
-    std::move(callback).Run(BorealisContextManager::Status::kStartVmFailed,
+    Complete(BorealisContextManager::Status::kStartVmFailed,
              "Failed to start Borealis VM: Empty response.");
     return;
   }
 
   if (response->status() == vm_tools::concierge::VM_STATUS_RUNNING ||
       response->status() == vm_tools::concierge::VM_STATUS_STARTING) {
-    std::move(callback).Run(BorealisContextManager::Status::kSuccess, "");
+    Complete(BorealisContextManager::Status::kSuccess, "");
     return;
   }
 
-  std::move(callback).Run(
-      BorealisContextManager::Status::kStartVmFailed,
-      "Failed to start Borealis VM: " + response->failure_reason() + " (code " +
-          base::NumberToString(response->status()) + ")");
+  Complete(BorealisContextManager::Status::kStartVmFailed,
+           "Failed to start Borealis VM: " + response->failure_reason() +
+               " (code " + base::NumberToString(response->status()) + ")");
 }
 
 AwaitBorealisStartup::AwaitBorealisStartup(Profile* profile,
@@ -151,11 +161,10 @@ AwaitBorealisStartup::AwaitBorealisStartup(Profile* profile,
     : watcher_(profile, vm_name) {}
 AwaitBorealisStartup::~AwaitBorealisStartup() = default;
 
-void AwaitBorealisStartup::Run(BorealisContext* context,
-                               CompletionStatusCallback callback) {
+void AwaitBorealisStartup::RunInternal(BorealisContext* context) {
   watcher_.AwaitLaunch(
       base::BindOnce(&AwaitBorealisStartup::OnAwaitBorealisStartup,
-                     weak_factory_.GetWeakPtr(), context, std::move(callback)));
+                     weak_factory_.GetWeakPtr(), context));
 }
 
 BorealisLaunchWatcher& AwaitBorealisStartup::GetWatcherForTesting() {
@@ -164,15 +173,13 @@ BorealisLaunchWatcher& AwaitBorealisStartup::GetWatcherForTesting() {
 
 void AwaitBorealisStartup::OnAwaitBorealisStartup(
     BorealisContext* context,
-    CompletionStatusCallback callback,
     base::Optional<std::string> container) {
   if (!container) {
-    std::move(callback).Run(
-        BorealisContextManager::Status::kAwaitBorealisStartupFailed,
+    Complete(BorealisContextManager::Status::kAwaitBorealisStartupFailed,
              "Awaiting for Borealis launch failed: timed out");
     return;
   }
   context->set_container_name(container.value());
-  std::move(callback).Run(BorealisContextManager::Status::kSuccess, "");
+  Complete(BorealisContextManager::Status::kSuccess, "");
 }
 }  // namespace borealis

chrome/browser/chromeos/borealis/borealis_task.h

@@ -24,12 +24,21 @@ class BorealisTask {
   // different status should be used, and an error string provided.
   using CompletionStatusCallback =
       base::OnceCallback<void(BorealisContextManager::Status, std::string)>;
-  BorealisTask() = default;
+
+  BorealisTask();
   BorealisTask(const BorealisTask&) = delete;
   BorealisTask& operator=(const BorealisTask&) = delete;
-  virtual void Run(BorealisContext* context,
-                   CompletionStatusCallback callback) = 0;
-  virtual ~BorealisTask() = default;
+  virtual ~BorealisTask();
+
+  void Run(BorealisContext* context, CompletionStatusCallback callback);
+
+ protected:
+  virtual void RunInternal(BorealisContext* context) = 0;
+
+  void Complete(BorealisContextManager::Status status, std::string message);
+
+ private:
+  CompletionStatusCallback callback_;
 };
 
 // Mounts the Borealis DLC.
@@ -37,13 +46,11 @@ class MountDlc : public BorealisTask {
  public:
   MountDlc();
   ~MountDlc() override;
-  void Run(BorealisContext* context,
-           CompletionStatusCallback callback) override;
+  void RunInternal(BorealisContext* context) override;
 
  private:
   void OnMountDlc(
       BorealisContext* context,
-      CompletionStatusCallback callback,
       const chromeos::DlcserviceClient::InstallResult& install_result);
   base::WeakPtrFactory<MountDlc> weak_factory_{this};
 };
@@ -53,13 +60,11 @@ class CreateDiskImage : public BorealisTask {
  public:
   CreateDiskImage();
   ~CreateDiskImage() override;
-  void Run(BorealisContext* context,
-           CompletionStatusCallback callback) override;
+  void RunInternal(BorealisContext* context) override;
 
  private:
   void OnCreateDiskImage(
       BorealisContext* context,
-      CompletionStatusCallback callback,
       base::Optional<vm_tools::concierge::CreateDiskImageResponse> response);
   base::WeakPtrFactory<CreateDiskImage> weak_factory_{this};
 };
@@ -69,13 +74,11 @@ class StartBorealisVm : public BorealisTask {
  public:
   StartBorealisVm();
   ~StartBorealisVm() override;
-  void Run(BorealisContext* context,
-           CompletionStatusCallback callback) override;
+  void RunInternal(BorealisContext* context) override;
 
  private:
   void OnStartBorealisVm(
       BorealisContext* context,
-      CompletionStatusCallback callback,
       base::Optional<vm_tools::concierge::StartVmResponse> response);
   base::WeakPtrFactory<StartBorealisVm> weak_factory_{this};
 };
@@ -85,13 +88,11 @@ class AwaitBorealisStartup : public BorealisTask {
  public:
   AwaitBorealisStartup(Profile* profile, std::string vm_name);
   ~AwaitBorealisStartup() override;
-  void Run(BorealisContext* context,
-           CompletionStatusCallback callback) override;
+  void RunInternal(BorealisContext* context) override;
   BorealisLaunchWatcher& GetWatcherForTesting();
 
  private:
   void OnAwaitBorealisStartup(BorealisContext* context,
-                              CompletionStatusCallback callback,
                               base::Optional<std::string> container);
   BorealisLaunchWatcher watcher_;
   base::WeakPtrFactory<AwaitBorealisStartup> weak_factory_{this};