From 0037e9c816a6086bdf80019f59c130f0f0eb9fc0 Mon Sep 17 00:00:00 2001
From: Gavin Corkery <gavincorkery@google.com>
Date: Mon, 15 Jun 2020 14:28:14 +0100
Subject: [PATCH] Mark boot completion via apexservice

Use the StagingManager lifecycle class to call into ApexManager
when the boot completed broadcast is received. ApexManager will
then call into apexservice to inform apexd of the boot completion,
instead of apexd relying on the sys.boot_completed property.

Test: atest StagedInstallTest
Test: atest ApexRollbackTests
Bug: 158467962
Change-Id: Ibcb4cb784243530eb2cf8c2633651342e8115d14
Merged-In: Ibcb4cb784243530eb2cf8c2633651342e8115d14
(cherry picked from commit 221f3ec59c9e774d3bcdceb7e1058dcb303b1dcf)
---
 .../com/android/server/pm/ApexManager.java    | 19 +++++++++++++++++++
 .../com/android/server/pm/StagingManager.java |  5 +++++
 2 files changed, 24 insertions(+)

diff --git a/services/core/java/com/android/server/pm/ApexManager.java b/services/core/java/com/android/server/pm/ApexManager.java
index e12cb8f533a7b..f116a24a76e16 100644
--- a/services/core/java/com/android/server/pm/ApexManager.java
+++ b/services/core/java/com/android/server/pm/ApexManager.java
@@ -351,6 +351,11 @@ public abstract class ApexManager {
      */
     public abstract boolean destroyCeSnapshotsNotSpecified(int userId, int[] retainRollbackIds);
 
+    /**
+     * Inform apexd that the boot has completed.
+     */
+    public abstract void markBootCompleted();
+
     /**
      * Dumps various state information to the provided {@link PrintWriter} object.
      *
@@ -883,6 +888,15 @@ public abstract class ApexManager {
             }
         }
 
+        @Override
+        public void markBootCompleted() {
+            try {
+                waitForApexService().markBootCompleted();
+            } catch (RemoteException re) {
+                Slog.e(TAG, "Unable to contact apexservice", re);
+            }
+        }
+
         /**
          * Dump information about the packages contained in a particular cache
          * @param packagesCache the cache to print information about.
@@ -1129,6 +1143,11 @@ public abstract class ApexManager {
             return true;
         }
 
+        @Override
+        public void markBootCompleted() {
+            // No-op
+        }
+
         @Override
         void dump(PrintWriter pw, String packageName) {
             // No-op
diff --git a/services/core/java/com/android/server/pm/StagingManager.java b/services/core/java/com/android/server/pm/StagingManager.java
index e2bd1f1119caf..0c96f592c0dbb 100644
--- a/services/core/java/com/android/server/pm/StagingManager.java
+++ b/services/core/java/com/android/server/pm/StagingManager.java
@@ -164,6 +164,7 @@ public class StagingManager {
         public void onBootPhase(int phase) {
             if (phase == SystemService.PHASE_BOOT_COMPLETED && sStagingManager != null) {
                 sStagingManager.markStagedSessionsAsSuccessful();
+                sStagingManager.markBootCompleted();
             }
         }
     }
@@ -179,6 +180,10 @@ public class StagingManager {
         }
     }
 
+    private void markBootCompleted() {
+        mApexManager.markBootCompleted();
+    }
+
     /**
      * Validates the signature used to sign the container of the new apex package
      *