Respect lockdown mode

Author: null-dev

universalauth-xposed/src/main/java/ax/nd/universalauth/xposed/Module.java

@@ -4,12 +4,14 @@
 import static ax.nd.universalauth.xposed.common.XposedConstants.EXTRA_UNLOCK_MODE;
 import static ax.nd.universalauth.xposed.common.XposedConstants.MODE_UNLOCK_FADING;
 
+import android.app.ActivityManager;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Build;
 
 import java.lang.reflect.AccessibleObject;
 import java.lang.reflect.Field;
+import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.util.Arrays;
 import java.util.Objects;
@@ -36,6 +38,8 @@ public class Module implements IXposedHookLoadPackage {
     // com.android.systemui.statusbar.StatusBarState.SHADE_LOCKED
     private static final int SHADE_LOCKED = 2;
 
+    private Method isUserInLockdownMethod;
+
     /**
      * This method is called when an app is loaded. It's called very early, even before
      * {@link Application#onCreate} is called.
@@ -48,6 +52,9 @@ public class Module implements IXposedHookLoadPackage {
     public void handleLoadPackage(XC_LoadPackage.LoadPackageParam lpparam) throws Throwable {
         // Hook SystemUi
         if (Objects.equals(lpparam.packageName, "com.android.systemui")) {
+            Class<?> kumClazz = lpparam.classLoader.loadClass(KEYGUARD_UPDATE_MONITOR_CLASS);
+            isUserInLockdownMethod = getIsUserInLockdownMethod(kumClazz);
+
             try {
                 // Hook com.android.systemui.statusbar.phone.StatusBar.start
                 Class<?> statusBarClass = lpparam.classLoader.loadClass(STATUS_BAR_CLASS);
@@ -66,7 +73,6 @@ public void handleLoadPackage(XC_LoadPackage.LoadPackageParam lpparam) throws Th
                 }
             }
 
-            Class<?> kumClazz = lpparam.classLoader.loadClass(KEYGUARD_UPDATE_MONITOR_CLASS);
             // Hook com.android.keyguard.KeyguardUpdateMonitor.updateFaceListeningState
             try {
                 addHookEarlyUnlock(kumClazz, lpparam);
@@ -98,6 +104,19 @@ protected void afterHookedMethod(MethodHookParam param) throws Throwable {
         );
     }
 
+    private Method getIsUserInLockdownMethod(Class<?> kumClazz) {
+        try {
+            return asAccessible(kumClazz.getDeclaredMethod("isUserInLockdown", int.class));
+        } catch (NoSuchMethodException ignored) {
+            return null;
+        }
+    }
+
+    private boolean isUserInLockdown(Object kum) throws InvocationTargetException, IllegalAccessException {
+        return isUserInLockdownMethod != null
+                && (boolean) isUserInLockdownMethod.invoke(kum, Util.INSTANCE.getCurrentUser());
+    }
+
     private void addHookEarlyUnlock(Class<?> kumClazz, XC_LoadPackage.LoadPackageParam lpparam) throws Throwable {
         Field mStatusBarStateControllerField = asAccessible(kumClazz.getDeclaredField("mStatusBarStateController"));
         Field mKeyguardIsVisibleField = asAccessible(kumClazz.getDeclaredField("mKeyguardIsVisible"));
@@ -112,6 +131,10 @@ private void addHookEarlyUnlock(Class<?> kumClazz, XC_LoadPackage.LoadPackagePar
             @Override
             protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
                 Object kum = param.thisObject;
+                if(isUserInLockdown(kum)) {
+                    return;
+                }
+
                 Object sbsc = mStatusBarStateControllerField.get(kum);
                 boolean mKeyguardIsVisible = mKeyguardIsVisibleField.getBoolean(kum);
                 boolean mDeviceInteractive = mDeviceInteractiveField.getBoolean(kum);
@@ -188,12 +211,15 @@ private void hookStatusBar(Class<?> statusBarClass, ClassLoader classLoader, XC_
             systemUiClass = classLoader.loadClass(SYSTEM_UI_CLASS);
         }
         Context context = (Context) asAccessible(systemUiClass.getDeclaredField("mContext")).get(statusBar);
+        Object kum = asAccessible(statusBarClass.getDeclaredField("mKeyguardUpdateMonitor")).get(statusBar);
 
         UnlockMethod method = hookStatusBarBiometricUnlock(statusBar, statusBarClass);
 
         UnlockReceiver.INSTANCE.setup(context, statusBar, intent -> {
             try {
-                method.unlock(intent);
+                if(!isUserInLockdown(kum)) {
+                    method.unlock(intent);
+                }
             } catch (Throwable throwable) {
                 throw new RuntimeException(throwable);
             }

universalauth-xposed/src/main/java/ax/nd/universalauth/xposed/TrustHook.kt

@@ -14,13 +14,11 @@ object TrustHook {
             Int::class.javaPrimitiveType!!,
             Boolean::class.javaPrimitiveType!!
         ).asAccessible()
-        val activityManagerGetCurrentUser =
-            ActivityManager::class.java.getDeclaredMethod("getCurrentUser").asAccessible()
         XposedHelpersExt.runAfterClassConstructed(kumClazz) { param ->
             val context = contextField.get(param.thisObject) as Context
             UnlockReceiver.setup(context, param.thisObject) { intent ->
                 if (!intent.getBooleanExtra(EXTRA_BYPASS_KEYGUARD, true)) {
-                    val currentUser = activityManagerGetCurrentUser.invoke(null)
+                    val currentUser = Util.getCurrentUser()
                     onFaceAuthenticated.invoke(param.thisObject, currentUser, true)
                 }
             }

universalauth-xposed/src/main/java/ax/nd/universalauth/xposed/Util.kt

@@ -0,0 +1,13 @@
+package ax.nd.universalauth.xposed
+
+import android.app.ActivityManager
+import ax.nd.xposedutil.asAccessible
+
+object Util {
+    private val activityManagerGetCurrentUser =
+        ActivityManager::class.java.getDeclaredMethod("getCurrentUser").asAccessible()
+
+    fun getCurrentUser(): Int {
+        return activityManagerGetCurrentUser.invoke(null) as Int
+    }
+}