Set correct uid when loading app icon in data usage UI.
work profile apps belongs to a different user. Set the correct uid when loading the icon so the work profile badge is displayed. Change-Id: I2ed27d6b83c1b42961500dc8d63475bf1f7eabcd Fix: 38372681 Test: make RunSettingsRoboTests
This commit is contained in:
@@ -36,12 +36,15 @@ import android.support.v7.preference.PreferenceCategory;
|
||||
import android.text.format.Formatter;
|
||||
import android.util.ArraySet;
|
||||
import android.util.IconDrawableFactory;
|
||||
import android.util.Log;
|
||||
import android.view.View;
|
||||
import android.widget.AdapterView;
|
||||
|
||||
import com.android.internal.logging.nano.MetricsProto.MetricsEvent;
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.applications.AppInfoBase;
|
||||
import com.android.settings.applications.PackageManagerWrapper;
|
||||
import com.android.settings.applications.PackageManagerWrapperImpl;
|
||||
import com.android.settings.widget.EntityHeaderController;
|
||||
import com.android.settingslib.AppItem;
|
||||
import com.android.settingslib.net.ChartData;
|
||||
@@ -71,6 +74,7 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
private static final int LOADER_CHART_DATA = 2;
|
||||
private static final int LOADER_APP_PREF = 3;
|
||||
|
||||
private PackageManagerWrapper mPackageManagerWrapper;
|
||||
private final ArraySet<String> mPackages = new ArraySet<>();
|
||||
private Preference mTotalUsage;
|
||||
private Preference mForegroundUsage;
|
||||
@@ -99,6 +103,7 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
@Override
|
||||
public void onCreate(Bundle icicle) {
|
||||
super.onCreate(icicle);
|
||||
mPackageManagerWrapper = new PackageManagerWrapperImpl(getPackageManager());
|
||||
final Bundle args = getArguments();
|
||||
|
||||
try {
|
||||
@@ -142,11 +147,11 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
|
||||
if (mAppItem.key > 0) {
|
||||
if (mPackages.size() != 0) {
|
||||
PackageManager pm = getPackageManager();
|
||||
try {
|
||||
ApplicationInfo info = pm.getApplicationInfo(mPackages.valueAt(0), 0);
|
||||
ApplicationInfo info = mPackageManagerWrapper.getApplicationInfoAsUser(
|
||||
mPackages.valueAt(0), 0, UserHandle.getUserId(mAppItem.key));
|
||||
mIcon = IconDrawableFactory.newInstance(getActivity()).getBadgedIcon(info);
|
||||
mLabel = info.loadLabel(pm);
|
||||
mLabel = info.loadLabel(mPackageManagerWrapper.getPackageManager());
|
||||
mPackageName = info.packageName;
|
||||
} catch (PackageManager.NameNotFoundException e) {
|
||||
}
|
||||
@@ -317,9 +322,13 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
|
||||
String pkg = mPackages.size() != 0 ? mPackages.valueAt(0) : null;
|
||||
int uid = 0;
|
||||
try {
|
||||
uid = pkg != null ? getPackageManager().getPackageUid(pkg, 0) : 0;
|
||||
} catch (PackageManager.NameNotFoundException e) {
|
||||
if (pkg != null) {
|
||||
try {
|
||||
uid = mPackageManagerWrapper.getPackageUidAsUser(pkg,
|
||||
UserHandle.getUserId(mAppItem.key));
|
||||
} catch (PackageManager.NameNotFoundException e) {
|
||||
Log.w(TAG, "Skipping UID because cannot find package " + pkg);
|
||||
}
|
||||
}
|
||||
|
||||
final boolean showInfoButton = mAppItem.key > 0;
|
||||
@@ -328,6 +337,7 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
final Preference pref = EntityHeaderController
|
||||
.newInstance(activity, this, null /* header */)
|
||||
.setRecyclerView(getListView(), getLifecycle())
|
||||
.setUid(uid)
|
||||
.setButtonActions(showInfoButton
|
||||
? EntityHeaderController.ActionType.ACTION_APP_INFO
|
||||
: EntityHeaderController.ActionType.ACTION_NONE,
|
||||
@@ -335,7 +345,6 @@ public class AppDataUsage extends DataUsageBase implements Preference.OnPreferen
|
||||
.setIcon(mIcon)
|
||||
.setLabel(mLabel)
|
||||
.setPackageName(pkg)
|
||||
.setUid(uid)
|
||||
.done(activity, getPrefContext());
|
||||
getPreferenceScreen().addPreference(pref);
|
||||
}
|
||||
|
Reference in New Issue
Block a user