Slight refactor of SafetyCenterResourcesContext.

Only cache the resources context, not every field.

Ideally it should implement ContextWrapper for the resources APK
context directly, I may follow up with another CL to do that (it's
a bit more risky and involved).

Also guard it by the API lock to make sure the resources context
is loaded at most once.

Bug: 283100177
Test: atest CtsSafetyCenterTestCases
Relnote: N/A
Change-Id: Ia54faa909105566fd5382c68b3e9046e1628267b

4 files changed, 172 insertions, 172 deletions

diff --git a/SafetyCenter/ResourcesLib/java/com/android/safetycenter/resources/SafetyCenterResourcesContext.java b/SafetyCenter/ResourcesLib/java/com/android/safetycenter/resources/SafetyCenterResourcesContext.java
index 9a77296e2..215e0f6d7 100644
--- a/SafetyCenter/ResourcesLib/java/com/android/safetycenter/resources/SafetyCenterResourcesContext.java
+++ b/SafetyCenter/ResourcesLib/java/com/android/safetycenter/resources/SafetyCenterResourcesContext.java
@@ -39,10 +39,12 @@ import java.io.InputStream;
 import java.util.List;
 
 /**
- * Wrapper for context to override getResources method. Resources for the Safety Center that need to
- * be fetched from the dedicated resources APK.
+ * Wrapper for a base context to expose Safety Center resources that need to be fetched from a
+ * dedicated APK.
+ *
+ * <p>This class isn't thread safe. Thread safety must be handled by the caller.
  */
-public class SafetyCenterResourcesContext extends ContextWrapper {
+public final class SafetyCenterResourcesContext extends ContextWrapper {
     private static final String TAG = "SafetyCenterResContext";
 
     /** Intent action that is used to identify the Safety Center resources APK */
@@ -65,25 +67,22 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
     private final String mResourcesApkAction;
 
     /** The path where the Safety Center resources APK is expected to be installed */
-    @Nullable private final String mResourcesApkPath;
+    private final String mResourcesApkPath;
 
     /** Raw XML config resource name */
     private final String mConfigName;
 
-    /** Specific flags used for retrieving resolve info */
+    /** Specific flags used for retrieving resolve info. */
     private final int mFlags;
 
     /**
-     * Whether we should fallback with an empty string when calling {@link #getStringByName} for a
-     * string resource that does not exist.
+     * Whether we should fallback with an empty string / null values when calling the methods of
+     * this class for a resource that does not exist.
      */
     private final boolean mShouldFallbackIfNamedResourceNotFound;
 
-    // Cached package name and resources from the resources APK
-    @Nullable private String mResourcesApkPkgName;
-    @Nullable private AssetManager mAssetsFromApk;
-    @Nullable private Resources mResourcesFromApk;
-    @Nullable private Resources.Theme mThemeFromApk;
+    // Cached context from the resources APK
+    @Nullable private Context mResourcesApkContext;
 
     public SafetyCenterResourcesContext(Context contextBase) {
         this(contextBase, /* shouldFallbackIfNamedResourceNotFound */ true);
@@ -104,13 +103,13 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
     SafetyCenterResourcesContext(
             Context contextBase,
             String resourcesApkAction,
-            @Nullable String resourcesApkPath,
+            String resourcesApkPath,
             String configName,
             int flags,
             boolean shouldFallbackIfNamedResourceNotFound) {
         super(contextBase);
         mResourcesApkAction = requireNonNull(resourcesApkAction);
-        mResourcesApkPath = resourcesApkPath;
+        mResourcesApkPath = requireNonNull(resourcesApkPath);
         mConfigName = requireNonNull(configName);
         mFlags = flags;
         mShouldFallbackIfNamedResourceNotFound = shouldFallbackIfNamedResourceNotFound;
@@ -124,28 +123,32 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
     }
 
     /**
-     * Initializes the {@link Context}'s {@link AssetManager}, {@link Resources} and {@link
-     * Resources.Theme}.
+     * Initializes the resources APK {@link Context}, and returns whether this was successful.
      *
-     * <p>This call is optional as this can also be lazily instantiated. This is useful to ensure
-     * that resources are loaded prior to interacting with the {@link SafetyCenterResourcesContext},
-     * as this code needs to run for the same user as the provided base {@link Context}; which may
-     * not be the case with a binder call.
+     * <p>This call is optional as this can also be lazily instantiated. It can be used to ensure
+     * that the resources APK context is loaded prior to interacting with this class. This
+     * initialization code needs to run in the same user as the provided base {@link Context}. This
+     * may not be the case with a binder call, which is why it can be more appropriate to do this
+     * explicitly.
      */
-    public void init() {
-        mAssetsFromApk = getAssets();
-        mResourcesFromApk = getResources();
-        mThemeFromApk = getTheme();
+    public boolean init() {
+        return getResourcesApkContext() != null;
     }
 
-    /** Get the package name of the Safety Center resources APK. */
+    /** Gets the {@link Context} of the Safety Center resources APK. */
     @VisibleForTesting
     @Nullable
-    String getResourcesApkPkgName() {
-        if (mResourcesApkPkgName != null) {
-            return mResourcesApkPkgName;
+    Context getResourcesApkContext() {
+        if (mResourcesApkContext != null) {
+            return mResourcesApkContext;
         }
 
+        mResourcesApkContext = loadResourcesApkContext();
+        return mResourcesApkContext;
+    }
+
+    @Nullable
+    private Context loadResourcesApkContext() {
         List<ResolveInfo> resolveInfos =
                 getPackageManager().queryIntentActivities(new Intent(mResourcesApkAction), mFlags);
 
@@ -169,9 +172,7 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
         final int resolveInfosSize = resolveInfos.size();
         for (int i = 0; i < resolveInfosSize; i++) {
             ResolveInfo resolveInfo = resolveInfos.get(i);
-            if (mResourcesApkPath != null
-                    && !resolveInfo.activityInfo.applicationInfo.sourceDir.startsWith(
-                            mResourcesApkPath)) {
+            if (!resolveInfo.activityInfo.applicationInfo.sourceDir.startsWith(mResourcesApkPath)) {
                 // skip apps that don't live in the Permission apex
                 continue;
             }
@@ -188,9 +189,19 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
             return null;
         }
 
-        mResourcesApkPkgName = info.activityInfo.applicationInfo.packageName;
-        Log.i(TAG, "Found Safety Center resources APK at: " + mResourcesApkPkgName);
-        return mResourcesApkPkgName;
+        String resourcesApkPkgName = info.activityInfo.applicationInfo.packageName;
+        Log.i(TAG, "Found Safety Center resources APK at: " + resourcesApkPkgName);
+        return getPackageContext(resourcesApkPkgName);
+    }
+
+    @Nullable
+    private Context getPackageContext(String packageName) {
+        try {
+            return createPackageContext(packageName, 0);
+        } catch (PackageManager.NameNotFoundException e) {
+            Log.e(TAG, "Failed to load package context for: " + packageName, e);
+        }
+        return null;
     }
 
     /**
@@ -199,19 +210,11 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
      */
     @Nullable
     public InputStream getSafetyCenterConfig() {
-        String resourcePkgName = getResourcesApkPkgName();
-        if (resourcePkgName == null) {
-            return null;
-        }
-        Resources resources = getResources();
-        if (resources == null) {
-            return null;
-        }
-        int id = resources.getIdentifier(mConfigName, "raw", resourcePkgName);
+        int id = getResId(mConfigName, "raw");
         if (id == Resources.ID_NULL) {
             return null;
         }
-        return resources.openRawResource(id);
+        return getResources().openRawResource(id);
     }
 
     /**
@@ -240,7 +243,7 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
     /** Same as {@link #getOptionalString(int)} but using the string name rather than ID. */
     @Nullable
     public String getOptionalStringByName(String name) {
-        return getOptionalString(getStringRes(name));
+        return getOptionalString(getResId(name, "string"));
     }
 
     /**
@@ -249,97 +252,47 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
      * {@link #mShouldFallbackIfNamedResourceNotFound} is {@code false}).
      */
     public String getStringByName(String name) {
-        int id = getStringRes(name);
+        int id = getResId(name, "string");
         return maybeFallbackIfNamedResourceIsNull(name, getOptionalString(id));
     }
 
     /** Same as {@link #getStringByName(String)} but with the given {@code formatArgs}. */
     public String getStringByName(String name, Object... formatArgs) {
-        int id = getStringRes(name);
+        int id = getResId(name, "string");
         return maybeFallbackIfNamedResourceIsNull(name, getOptionalString(id, formatArgs));
     }
 
-    private String maybeFallbackIfNamedResourceIsNull(String name, @Nullable String value) {
-        if (value != null) {
-            return value;
-        }
-        if (!mShouldFallbackIfNamedResourceNotFound) {
-            throw new Resources.NotFoundException();
-        }
-        Log.w(TAG, "String resource " + name + " not found");
-        return "";
-    }
-
-    @StringRes
-    private int getStringRes(String name) {
-        return getResId(name, "string");
-    }
-
-    private int getResId(String name, String type) {
-        String resourcePkgName = getResourcesApkPkgName();
-        if (resourcePkgName == null) {
-            return Resources.ID_NULL;
-        }
-        Resources resources = getResources();
-        if (resources == null) {
-            return Resources.ID_NULL;
-        }
-        // TODO(b/227738283): profile the performance of this operation and consider adding caching
-        //  or finding some alternative solution.
-        return resources.getIdentifier(name, type, resourcePkgName);
-    }
-
-    @Nullable
-    private Context getResourcesApkContext() {
-        String name = getResourcesApkPkgName();
-        if (name == null) {
-            return null;
-        }
-        try {
-            return createPackageContext(name, 0);
-        } catch (PackageManager.NameNotFoundException e) {
-            Log.wtf(TAG, "Failed to load resources", e);
-        }
-        return null;
-    }
-
     /** Retrieve assets held in the Safety Center resources APK. */
     @Override
     @Nullable
     public AssetManager getAssets() {
-        if (mAssetsFromApk == null) {
-            Context resourcesApkContext = getResourcesApkContext();
-            if (resourcesApkContext != null) {
-                mAssetsFromApk = resourcesApkContext.getAssets();
-            }
+        Context resourcesApkContext = getResourcesApkContext();
+        if (resourcesApkContext == null) {
+            return null;
         }
-        return mAssetsFromApk;
+        return resourcesApkContext.getAssets();
     }
 
     /** Retrieve resources held in the Safety Center resources APK. */
     @Override
     @Nullable
     public Resources getResources() {
-        if (mResourcesFromApk == null) {
-            Context resourcesApkContext = getResourcesApkContext();
-            if (resourcesApkContext != null) {
-                mResourcesFromApk = resourcesApkContext.getResources();
-            }
+        Context resourcesApkContext = getResourcesApkContext();
+        if (resourcesApkContext == null) {
+            return null;
         }
-        return mResourcesFromApk;
+        return resourcesApkContext.getResources();
     }
 
     /** Retrieve theme held in the Safety Center resources APK. */
     @Override
     @Nullable
     public Resources.Theme getTheme() {
-        if (mThemeFromApk == null) {
-            Context resourcesApkContext = getResourcesApkContext();
-            if (resourcesApkContext != null) {
-                mThemeFromApk = resourcesApkContext.getTheme();
-            }
+        Context resourcesApkContext = getResourcesApkContext();
+        if (resourcesApkContext == null) {
+            return null;
         }
-        return mThemeFromApk;
+        return resourcesApkContext.getTheme();
     }
 
     /**
@@ -370,17 +323,17 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
      * drawable exists, returns {@code null} or throws {@link Resources.NotFoundException}.
      */
     @Nullable
-    public Icon getIconByDrawableName(String drawableResName) {
-        int resId = getResId(drawableResName, "drawable");
+    public Icon getIconByDrawableName(String name) {
+        int resId = getResId(name, "drawable");
         if (resId != Resources.ID_NULL) {
-            return Icon.createWithResource(getResourcesApkPkgName(), resId);
+            return Icon.createWithResource(getResourcesApkContext().getPackageName(), resId);
         }
 
         if (!mShouldFallbackIfNamedResourceNotFound) {
             throw new Resources.NotFoundException();
         }
 
-        Log.w(TAG, "Drawable resource " + drawableResName + " not found");
+        Log.w(TAG, "Drawable resource " + name + " not found");
         return null;
     }
 
@@ -400,4 +353,27 @@ public class SafetyCenterResourcesContext extends ContextWrapper {
         Log.w(TAG, "Color resource " + name + " not found");
         return null;
     }
+
+    private String maybeFallbackIfNamedResourceIsNull(String name, @Nullable String value) {
+        if (value != null) {
+            return value;
+        }
+        if (!mShouldFallbackIfNamedResourceNotFound) {
+            throw new Resources.NotFoundException();
+        }
+        Log.w(TAG, "String resource " + name + " not found");
+        return "";
+    }
+
+    private int getResId(String name, String type) {
+        Context resourcesApkContext = getResourcesApkContext();
+        if (resourcesApkContext == null) {
+            return Resources.ID_NULL;
+        }
+        // TODO(b/227738283): profile the performance of this operation and consider adding caching
+        //  or finding some alternative solution.
+        return resourcesApkContext
+                .getResources()
+                .getIdentifier(name, type, resourcesApkContext.getPackageName());
+    }
 }
diff --git a/SafetyCenter/ResourcesLib/tests/java/com/android/safetycenter/resources/SafetyCenterResourcesContextTest.kt b/SafetyCenter/ResourcesLib/tests/java/com/android/safetycenter/resources/SafetyCenterResourcesContextTest.kt
index 1a82460d2..c7905ea47 100644
--- a/SafetyCenter/ResourcesLib/tests/java/com/android/safetycenter/resources/SafetyCenterResourcesContextTest.kt
+++ b/SafetyCenter/ResourcesLib/tests/java/com/android/safetycenter/resources/SafetyCenterResourcesContextTest.kt
@@ -29,14 +29,14 @@ import org.junit.runner.RunWith
 @RunWith(AndroidJUnit4::class)
 class SafetyCenterResourcesContextTest {
     private val context: Context = getApplicationContext()
-    private val theme: Resources.Theme? = context.theme
 
     @Test
     fun validDataWithValidInputs() {
-        val resourcesContext =
-            SafetyCenterResourcesContext(context, RESOURCES_APK_ACTION, null, CONFIG_NAME, 0, false)
+        val resourcesContext = createNewResourcesContext()
 
-        assertThat(resourcesContext.resourcesApkPkgName).isEqualTo(RESOURCES_APK_PKG_NAME)
+        val resourcesApkContext = resourcesContext.resourcesApkContext
+        assertThat(resourcesApkContext).isNotNull()
+        assertThat(resourcesApkContext!!.packageName).isEqualTo(RESOURCES_APK_PKG_NAME)
 
         val configContent =
             resourcesContext.safetyCenterConfig?.bufferedReader().use { it?.readText() }
@@ -51,7 +51,7 @@ class SafetyCenterResourcesContextTest {
     fun nullDataWithWrongAction() {
         val resourcesContext = createNewResourcesContext(resourcesApkAction = "wrong")
 
-        assertThat(resourcesContext.resourcesApkPkgName).isNull()
+        assertThat(resourcesContext.resourcesApkContext).isNull()
         assertThat(resourcesContext.safetyCenterConfig).isNull()
         assertThat(resourcesContext.assets).isNull()
         assertThat(resourcesContext.resources).isNull()
@@ -63,7 +63,7 @@ class SafetyCenterResourcesContextTest {
         val resourcesContext =
             createNewResourcesContext(resourcesApkPath = "/apex/com.android.permission")
 
-        assertThat(resourcesContext.resourcesApkPkgName).isNull()
+        assertThat(resourcesContext.resourcesApkContext).isNull()
         assertThat(resourcesContext.safetyCenterConfig).isNull()
         assertThat(resourcesContext.assets).isNull()
         assertThat(resourcesContext.resources).isNull()
@@ -74,7 +74,7 @@ class SafetyCenterResourcesContextTest {
     fun nullDataWithWrongFlag() {
         val resourcesContext = createNewResourcesContext(flags = PackageManager.MATCH_SYSTEM_ONLY)
 
-        assertThat(resourcesContext.resourcesApkPkgName).isNull()
+        assertThat(resourcesContext.resourcesApkContext).isNull()
         assertThat(resourcesContext.safetyCenterConfig).isNull()
         assertThat(resourcesContext.assets).isNull()
         assertThat(resourcesContext.resources).isNull()
@@ -85,7 +85,7 @@ class SafetyCenterResourcesContextTest {
     fun nullConfigWithWrongConfigName() {
         val resourcesContext = createNewResourcesContext(configName = "wrong")
 
-        assertThat(resourcesContext.resourcesApkPkgName).isNotNull()
+        assertThat(resourcesContext.resourcesApkContext).isNotNull()
         assertThat(resourcesContext.safetyCenterConfig).isNull()
         assertThat(resourcesContext.assets).isNotNull()
         assertThat(resourcesContext.resources).isNotNull()
@@ -140,14 +140,14 @@ class SafetyCenterResourcesContextTest {
     fun getDrawableByName_validDrawable_returnsDrawable() {
         val resourcesContext = createNewResourcesContext()
 
-        assertThat(resourcesContext.getDrawableByName("valid_drawable", theme)).isNotNull()
+        assertThat(resourcesContext.getDrawableByName("valid_drawable", context.theme)).isNotNull()
     }
 
     @Test
     fun getDrawableByName_invalidDrawableWithFallback_returnsNull() {
         val resourcesContext = createNewResourcesContext(fallback = true)
 
-        assertThat(resourcesContext.getDrawableByName("invalid_drawable", theme)).isNull()
+        assertThat(resourcesContext.getDrawableByName("invalid_drawable", context.theme)).isNull()
     }
 
     @Test
@@ -155,7 +155,7 @@ class SafetyCenterResourcesContextTest {
         val resourcesContext = createNewResourcesContext(fallback = false)
 
         assertFailsWith(Resources.NotFoundException::class) {
-            resourcesContext.getDrawableByName("invalid_drawable", theme)
+            resourcesContext.getDrawableByName("invalid_drawable", context.theme)
         }
     }
 
@@ -207,13 +207,19 @@ class SafetyCenterResourcesContextTest {
 
     private fun createNewResourcesContext(
         resourcesApkAction: String = RESOURCES_APK_ACTION,
-        resourcesApkPath: String? = null,
+        resourcesApkPath: String = "",
         configName: String = CONFIG_NAME,
         flags: Int = 0,
         fallback: Boolean = false
     ) =
         SafetyCenterResourcesContext(
-            context, resourcesApkAction, resourcesApkPath, configName, flags, fallback)
+            context,
+            resourcesApkAction,
+            resourcesApkPath,
+            configName,
+            flags,
+            fallback
+        )
 
     companion object {
         const val RESOURCES_APK_ACTION =
diff --git a/service/java/com/android/safetycenter/SafetyCenterConfigReader.java b/service/java/com/android/safetycenter/SafetyCenterConfigReader.java
index ecb741a4b..555c1db11 100644
--- a/service/java/com/android/safetycenter/SafetyCenterConfigReader.java
+++ b/service/java/com/android/safetycenter/SafetyCenterConfigReader.java
@@ -76,7 +76,7 @@ public final class SafetyCenterConfigReader {
      * this method was {@code true}.
      */
     boolean loadConfig() {
-        SafetyCenterConfig safetyCenterConfig = readSafetyCenterConfig();
+        SafetyCenterConfig safetyCenterConfig = loadSafetyCenterConfig();
         if (safetyCenterConfig == null) {
             return false;
         }
@@ -225,26 +225,26 @@ public final class SafetyCenterConfigReader {
     }
 
     @Nullable
-    private SafetyCenterConfig readSafetyCenterConfig() {
-        InputStream in = mSafetyCenterResourcesContext.getSafetyCenterConfig();
-        if (in == null) {
-            Log.e(TAG, "Cannot get safety center config file, Safety Center will be disabled");
+    private SafetyCenterConfig loadSafetyCenterConfig() {
+        Resources resources = mSafetyCenterResourcesContext.getResources();
+        if (resources == null) {
+            Log.e(TAG, "Cannot access Safety Center resources");
             return null;
         }
 
-        Resources resources = mSafetyCenterResourcesContext.getResources();
-        if (resources == null) {
-            Log.e(TAG, "Cannot get safety center resources, Safety Center will be disabled");
+        InputStream in = mSafetyCenterResourcesContext.getSafetyCenterConfig();
+        if (in == null) {
+            Log.e(TAG, "Cannot access Safety Center config file");
             return null;
         }
 
         try {
             SafetyCenterConfig safetyCenterConfig =
                     SafetyCenterConfigParser.parseXmlResource(in, resources);
-            Log.d(TAG, "SafetyCenterConfig read successfully");
+            Log.d(TAG, "SafetyCenterConfig loaded successfully");
             return safetyCenterConfig;
         } catch (ParseException e) {
-            Log.e(TAG, "Cannot read SafetyCenterConfig, Safety Center will be disabled", e);
+            Log.e(TAG, "Cannot parse SafetyCenterConfig", e);
             return null;
         }
     }
diff --git a/service/java/com/android/safetycenter/SafetyCenterService.java b/service/java/com/android/safetycenter/SafetyCenterService.java
index 1a9f60781..969c43ce6 100644
--- a/service/java/com/android/safetycenter/SafetyCenterService.java
+++ b/service/java/com/android/safetycenter/SafetyCenterService.java
@@ -38,7 +38,6 @@ import static java.util.Objects.requireNonNull;
 import android.annotation.UserIdInt;
 import android.app.PendingIntent;
 import android.app.StatsManager;
-import android.app.StatsManager.StatsPullAtomCallback;
 import android.content.BroadcastReceiver;
 import android.content.Context;
 import android.content.Intent;
@@ -114,8 +113,10 @@ public final class SafetyCenterService extends SystemService {
     @GuardedBy("mApiLock")
     private final SafetyCenterTimeouts mSafetyCenterTimeouts = new SafetyCenterTimeouts();
 
+    @GuardedBy("mApiLock")
     private final SafetyCenterResourcesContext mSafetyCenterResourcesContext;
 
+    @GuardedBy("mApiLock")
     private final SafetyCenterNotificationChannels mNotificationChannels;
 
     @GuardedBy("mApiLock")
@@ -142,11 +143,10 @@ public final class SafetyCenterService extends SystemService {
     @GuardedBy("mApiLock")
     private final SafetyCenterDataChangeNotifier mSafetyCenterDataChangeNotifier;
 
-    private final StatsPullAtomCallback mPullAtomCallback;
     private final boolean mDeviceSupportsSafetyCenter;
 
     /** Whether the {@link SafetyCenterConfig} was successfully loaded. */
-    private volatile boolean mConfigAvailable;
+    private volatile boolean mConfigAvailable = false;
 
     public SafetyCenterService(Context context) {
         super(context);
@@ -178,13 +178,6 @@ public final class SafetyCenterService extends SystemService {
                         mSafetyCenterConfigReader,
                         mSafetyCenterRefreshTracker,
                         mSafetyCenterDataManager);
-        mPullAtomCallback =
-                new SafetyCenterPullAtomCallback(
-                        context,
-                        mApiLock,
-                        mSafetyCenterConfigReader,
-                        mSafetyCenterDataFactory,
-                        mSafetyCenterDataManager);
         mSafetyCenterDataChangeNotifier =
                 new SafetyCenterDataChangeNotifier(mNotificationSender, mSafetyCenterListeners);
         mDeviceSupportsSafetyCenter =
@@ -193,39 +186,52 @@ public final class SafetyCenterService extends SystemService {
                                 Resources.getSystem()
                                         .getIdentifier(
                                                 "config_enableSafetyCenter", "bool", "android"));
-        if (!mDeviceSupportsSafetyCenter) {
-            Log.i(TAG, "Device does not support Safety Center, it will be disabled");
-        }
     }
 
     @Override
     public void onStart() {
         publishBinderService(Context.SAFETY_CENTER_SERVICE, new Stub());
-        if (mDeviceSupportsSafetyCenter) {
-            synchronized (mApiLock) {
-                mSafetyCenterResourcesContext.init();
-                SafetyCenterFlags.init(mSafetyCenterResourcesContext);
-                mConfigAvailable = mSafetyCenterConfigReader.loadConfig();
-                if (mConfigAvailable) {
-                    mSafetyCenterDataManager.loadPersistableDataStateFromFile();
-                    new UserBroadcastReceiver().register(getContext());
-                    new SafetyCenterNotificationReceiver(
-                                    /* service= */ this,
-                                    mSafetyCenterDataManager,
-                                    mSafetyCenterDataChangeNotifier,
-                                    mApiLock)
-                            .register(getContext());
-                    new LocaleBroadcastReceiver().register(getContext());
-                }
+        if (!mDeviceSupportsSafetyCenter) {
+            Log.i(TAG, "Device does not support Safety Center, it will be disabled");
+            return;
+        }
+
+        synchronized (mApiLock) {
+            boolean safetyCenterResourcesInitialized = mSafetyCenterResourcesContext.init();
+            if (!safetyCenterResourcesInitialized) {
+                Log.e(TAG, "Cannot init Safety Center resources, Safety Center will be disabled");
+                return;
             }
+
+            SafetyCenterFlags.init(mSafetyCenterResourcesContext);
+
+            if (!mSafetyCenterConfigReader.loadConfig()) {
+                Log.e(TAG, "Cannot init Safety Center config, Safety Center will be disabled");
+                return;
+            }
+
+            mConfigAvailable = true;
+            mSafetyCenterDataManager.loadPersistableDataStateFromFile();
+            new UserBroadcastReceiver().register(getContext());
+            new SafetyCenterNotificationReceiver(
+                            /* service= */ this,
+                            mSafetyCenterDataManager,
+                            mSafetyCenterDataChangeNotifier,
+                            mApiLock)
+                    .register(getContext());
+            new LocaleBroadcastReceiver().register(getContext());
         }
     }
 
     @Override
     public void onBootPhase(int phase) {
-        if (phase == SystemService.PHASE_BOOT_COMPLETED && canUseSafetyCenter()) {
-            registerSafetyCenterEnabledListener();
-            registerSafetyCenterPullAtomCallback();
+        if (phase != SystemService.PHASE_BOOT_COMPLETED || !canUseSafetyCenter()) {
+            return;
+        }
+
+        registerSafetyCenterEnabledListener();
+        synchronized (mApiLock) {
+            registerSafetyCenterPullAtomCallbackLocked();
             mNotificationChannels.createAllChannelsForAllUsers(getContext());
         }
     }
@@ -239,14 +245,22 @@ public final class SafetyCenterService extends SystemService {
                 DeviceConfig.NAMESPACE_PRIVACY, foregroundThreadExecutor, listener);
     }
 
-    private void registerSafetyCenterPullAtomCallback() {
+    @GuardedBy("mApiLock")
+    private void registerSafetyCenterPullAtomCallbackLocked() {
         StatsManager statsManager =
                 requireNonNull(getContext().getSystemService(StatsManager.class));
+        SafetyCenterPullAtomCallback pullAtomCallback =
+                new SafetyCenterPullAtomCallback(
+                        getContext(),
+                        mApiLock,
+                        mSafetyCenterConfigReader,
+                        mSafetyCenterDataFactory,
+                        mSafetyCenterDataManager);
         statsManager.setPullAtomCallback(
                 SAFETY_STATE,
                 /* metadata= */ null,
                 BackgroundThread.getExecutor(),
-                mPullAtomCallback);
+                pullAtomCallback);
     }
 
     /** Service implementation of {@link ISafetyCenterManager.Stub}. */
@@ -992,7 +1006,9 @@ public final class SafetyCenterService extends SystemService {
         @Override
         public void onReceive(Context context, Intent intent) {
             Log.d(TAG, "Locale changed broadcast received");
-            mNotificationChannels.createAllChannelsForAllUsers(getContext());
+            synchronized (mApiLock) {
+                mNotificationChannels.createAllChannelsForAllUsers(getContext());
+            }
         }
     }
 
@@ -1056,7 +1072,9 @@ public final class SafetyCenterService extends SystemService {
                 case Intent.ACTION_MANAGED_PROFILE_ADDED:
                 case Intent.ACTION_MANAGED_PROFILE_AVAILABLE:
                     startRefreshingSafetySources(REFRESH_REASON_OTHER, userId);
-                    mNotificationChannels.createAllChannelsForUser(getContext(), userHandle);
+                    synchronized (mApiLock) {
+                        mNotificationChannels.createAllChannelsForUser(getContext(), userHandle);
+                    }
                     break;
             }
         }