PackageInstaller: Fix crash when uninstalling apps with fragile data

* Fixes:
  - https://gitlab.com/LineageOS/issues/android/-/issues/2298
  - https://issuetracker.google.com/issues/148573843

Change-Id: I5a27d6f02d17a52e1a1a2aff4fb46327ce5e6127

Author: luk1337

packages/PackageInstaller/src/com/android/packageinstaller/handheld/UninstallAlertDialogFragment.java

@@ -35,7 +35,6 @@
 import android.os.UserHandle;
 import android.os.UserManager;
 import android.os.storage.StorageManager;
-import android.os.storage.StorageVolume;
 import android.util.Log;
 import android.view.LayoutInflater;
 import android.view.View;
@@ -64,29 +63,18 @@ public class UninstallAlertDialogFragment extends DialogFragment implements
      * @return The number of bytes.
      */
     private long getAppDataSizeForUser(@NonNull String pkg, @NonNull UserHandle user) {
-        StorageManager storageManager = getContext().getSystemService(StorageManager.class);
+        PackageManager packageManager = getContext().getPackageManager();
         StorageStatsManager storageStatsManager =
                 getContext().getSystemService(StorageStatsManager.class);
 
-        List<StorageVolume> volumes = storageManager.getStorageVolumes();
-        long appDataSize = 0;
-
-        int numVolumes = volumes.size();
-        for (int i = 0; i < numVolumes; i++) {
-            StorageStats stats;
-            try {
-                stats = storageStatsManager.queryStatsForPackage(convert(volumes.get(i).getUuid()),
-                        pkg, user);
-            } catch (PackageManager.NameNotFoundException | IOException e) {
-                Log.e(LOG_TAG, "Cannot determine amount of app data for " + pkg + " on "
-                        + volumes.get(i) + " (user " + user + ")", e);
-                continue;
-            }
-
-            appDataSize += stats.getDataBytes();
+        try {
+            ApplicationInfo info = packageManager.getApplicationInfo(pkg, 0);
+            return storageStatsManager.queryStatsForPackage(
+                    info.storageUuid, pkg, user).getDataBytes();
+        } catch (PackageManager.NameNotFoundException | IOException e) {
+            Log.e(LOG_TAG, "Cannot determine amount of app data for " + pkg, e);
+            return 0;
         }
-
-        return appDataSize;
     }
 
     /**