CustoMIUIzer/app/src/main/java/name/mikanoshi/customiuizer/mods/PackagePermissions.java

package name.mikanoshi.customiuizer.mods;

import android.content.pm.ApplicationInfo;
import android.content.pm.ResolveInfo;

import static de.robv.android.xposed.XposedHelpers.findClass;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import de.robv.android.xposed.callbacks.XC_LoadPackage.LoadPackageParam;
import de.robv.android.xposed.XposedHelpers;

import name.mikanoshi.customiuizer.utils.Helpers;
import name.mikanoshi.customiuizer.utils.Helpers.MethodHook;

public class PackagePermissions {

	private static final ArrayList<String> systemPackages = new ArrayList<String>();

//	@SuppressWarnings("unchecked")
//	private static void doBefore(MethodHookParam param) {
//		ArrayList requestedPermissions = (ArrayList)getObjectField(param.args[0], "requestedPermissions");
//		param.setObjectExtra("orig_requested_permissions", requestedPermissions);
//		//ArrayList requestedPermissionsRequired = (ArrayList)getObjectField(param.args[0], "requestedPermissionsRequired");
//		//param.setObjectExtra("orig_requested_permissions_required", requestedPermissionsRequired);
//
//		String pkgName = (String)getObjectField(param.args[0], "packageName");
//		if (pkgName.equalsIgnoreCase(Helpers.modulePkg)) {
//			requestedPermissions.add("miui.permission.READ_LOGS");
//			requestedPermissions.add("miui.permission.DUMP_CACHED_LOG");
//		}
//
//		setObjectField(param.args[0], "requestedPermissions", requestedPermissions);
//		//setObjectField(param.args[0], "requestedPermissionsRequired", requestedPermissionsRequired);
//	}
//
//	@SuppressWarnings("unchecked")
//	private static void doAfter(MethodHookParam param) {
//		ArrayList origRequestedPermissions = (ArrayList) param.getObjectExtra("orig_requested_permissions");
//		if (origRequestedPermissions != null) setObjectField(param.args[0], "requestedPermissions", origRequestedPermissions);
//		//ArrayList origRequestedPermissionsRequired = (ArrayList) param.getObjectExtra("orig_requested_permissions_required");
//		//if (origRequestedPermissionsRequired != null) setObjectField(param.args[0], "requestedPermissionsRequired", origRequestedPermissionsRequired);
//	}

	public static void hook(LoadPackageParam lpparam) {
		systemPackages.add(Helpers.modulePkg);
		//systemPackages.add("com.miui.packageinstaller");
		//systemPackages.add("pl.solidexplorer2");

		// Allow signature level permissions for module
		if (!Helpers.findAndHookMethodSilently("com.android.server.pm.permission.PermissionManagerService", lpparam.classLoader, "grantSignaturePermission",
			String.class, "android.content.pm.PackageParser.Package", "com.android.server.pm.permission.BasePermission", "com.android.server.pm.permission.PermissionsState",
			new MethodHook() {
				@Override
				protected void before(MethodHookParam param) throws Throwable {
					String pkgName = (String)XposedHelpers.getObjectField(param.args[1], "packageName");
					if (systemPackages.contains(pkgName)) param.setResult(true);
				}
			}
		)) Helpers.findAndHookMethod("com.android.server.pm.PackageManagerService", lpparam.classLoader, "grantSignaturePermission",
			String.class, "android.content.pm.PackageParser.Package", "com.android.server.pm.BasePermission", "com.android.server.pm.PermissionsState",
			new MethodHook() {
				@Override
				protected void before(MethodHookParam param) throws Throwable {
					String pkgName = (String)XposedHelpers.getObjectField(param.args[1], "packageName");
					if (systemPackages.contains(pkgName)) param.setResult(true);
				}
			}
		);

		if (!Helpers.findAndHookMethodSilently("com.android.server.pm.PackageManagerServiceUtils", lpparam.classLoader, "verifySignatures",
			"com.android.server.pm.PackageSetting", "com.android.server.pm.PackageSetting", "android.content.pm.PackageParser.SigningDetails", boolean.class, boolean.class,
			new MethodHook() {
				@Override
				protected void before(MethodHookParam param) throws Throwable {
					String pkgName = (String)XposedHelpers.getObjectField(param.args[0], "name");
					if (systemPackages.contains(pkgName)) param.setResult(true);
				}
			}
		)) Helpers.findAndHookMethod("com.android.server.pm.PackageManagerService", lpparam.classLoader, "verifySignaturesLP",
			"com.android.server.pm.PackageSetting", "android.content.pm.PackageParser.Package",
			new MethodHook() {
				@Override
				protected void before(MethodHookParam param) throws Throwable {
					String pkgName = (String)XposedHelpers.getObjectField(param.args[1], "packageName");
					if (systemPackages.contains(pkgName)) param.setResult(true);
				}
			}
		);

//		// Add custom permissions for module
//		if (!Helpers.findAndHookMethodSilently("com.android.server.pm.permission.PermissionManagerService", lpparam.classLoader, "grantRequestedRuntimePermissions",
//			"android.content.pm.PackageParser$Package", int[].class, String[].class, int.class, "com.android.server.pm.permission.PermissionManagerServiceInternal.PermissionCallback",
//			new MethodHook() {
//				@Override
//				protected void before(MethodHookParam param) throws Throwable {
//					doBefore(param);
//				}
//				@Override
//				protected void after(MethodHookParam param) throws Throwable {
//					doAfter(param);
//				}
//			}
//		)) if (!Helpers.findAndHookMethodSilently("com.android.server.pm.permission.PermissionManagerService", lpparam.classLoader, "grantPermissions",
//			"android.content.pm.PackageParser$Package", boolean.class, String.class, "com.android.server.pm.permission.PermissionManagerInternal.PermissionCallback",
//			new MethodHook() {
//				@Override
//				protected void before(MethodHookParam param) throws Throwable {
//					doBefore(param);
//				}
//				@Override
//				protected void after(MethodHookParam param) throws Throwable {
//					doAfter(param);
//				}
//			}
//		)) Helpers.findAndHookMethod("com.android.server.pm.PackageManagerService", lpparam.classLoader, "grantPermissionsLPw",
//			"android.content.pm.PackageParser$Package", boolean.class, String.class,
//			new MethodHook() {
//				@Override
//				protected void before(MethodHookParam param) throws Throwable {
//					doBefore(param);
//				}
//				@Override
//				protected void after(MethodHookParam param) throws Throwable {
//					doAfter(param);
//				}
//			}
//		);

		// Make module appear as system app
		Helpers.hookAllMethods("com.android.server.pm.PackageManagerService", lpparam.classLoader, "queryIntentActivitiesInternal", new MethodHook() {
			@Override
			@SuppressWarnings("unchecked")
			protected void after(MethodHookParam param) throws Throwable {
				List<ResolveInfo> infos = (List<ResolveInfo>)param.getResult();
				if (infos != null)
				for (ResolveInfo info: infos)
				if (info != null && info.activityInfo != null && systemPackages.contains(info.activityInfo.packageName))
				XposedHelpers.setObjectField(info, "system", true);
			}
		});

//		// Causes module removal by system on updates
//		Helpers.hookAllMethods("com.android.server.pm.PackageManagerService", lpparam.classLoader, "getApplicationInfoInternal", new MethodHook() {
//			@Override
//			protected void after(MethodHookParam param) throws Throwable {
//				ApplicationInfo info = (ApplicationInfo)param.getResult();
//				if (info != null && systemPackages.contains(info.packageName)) {
//					info.flags |= ApplicationInfo.FLAG_SYSTEM;
//					param.setResult(info);
//				}
//			}
//		});

		Helpers.findAndHookMethod("android.content.pm.ApplicationInfo", lpparam.classLoader, "isSystemApp", new MethodHook() {
			@Override
			protected void after(MethodHookParam param) throws Throwable {
				ApplicationInfo ai = (ApplicationInfo)param.thisObject;
				if (ai != null && systemPackages.contains(ai.packageName)) param.setResult(true);
			}
		});

		//noinspection ResultOfMethodCallIgnored
		Helpers.findAndHookMethodSilently("android.content.pm.ApplicationInfo", lpparam.classLoader, "isSignedWithPlatformKey", new MethodHook() {
			@Override
			protected void after(MethodHookParam param) throws Throwable {
				ApplicationInfo ai = (ApplicationInfo)param.thisObject;
				if (ai != null && systemPackages.contains(ai.packageName)) param.setResult(true);
			}
		});

		// Do not restrict background activity
		if (!Helpers.isNougat())
		Helpers.hookAllMethods("com.android.server.am.ActivityManagerService", lpparam.classLoader, "appRestrictedInBackgroundLocked", new MethodHook() {
			@Override
			protected void after(MethodHookParam param) throws Throwable {
				if (Helpers.modulePkg.equals(param.args[1])) param.setResult(0);
			}
		});

		if (!Helpers.isNougat())
		Helpers.hookAllMethods("com.android.server.am.ActivityManagerService", lpparam.classLoader, "appServicesRestrictedInBackgroundLocked", new MethodHook() {
			@Override
			protected void after(MethodHookParam param) throws Throwable {
				if (Helpers.modulePkg.equals(param.args[1])) param.setResult(0);
			}
		});

		Helpers.hookAllMethodsSilently("com.android.server.wm.ActivityRecordInjector", lpparam.classLoader, "canShowWhenLocked", new MethodHook() {
			@Override
			protected void before(MethodHookParam param) throws Throwable {
				param.setResult(true);
			}
		});

		try {
			Class<?> dpgpiClass = findClass("com.android.server.pm.DefaultPermissionGrantPolicyInjector", lpparam.classLoader);
			String[] MIUI_SYSTEM_APPS = (String[])XposedHelpers.getStaticObjectField(dpgpiClass, "MIUI_SYSTEM_APPS");
			ArrayList<String> mySystemApps = new ArrayList<String>(Arrays.asList(MIUI_SYSTEM_APPS));
			mySystemApps.addAll(systemPackages);
			XposedHelpers.setStaticObjectField(dpgpiClass, "MIUI_SYSTEM_APPS", mySystemApps.toArray(new String[0]));
		} catch (Throwable t) {}
	}

}