fix #222 (coloros proxied wakelock/broadcast) (#226)

* BroadcastFix: update condition

* add ColorOS15 wakelock/broadcast proxy fix

Author: yatli

app/src/main/java/com/kooritea/fcmfix/XposedMain.java

@@ -7,6 +7,7 @@
 import com.kooritea.fcmfix.xposed.GMSRestrictFix;
 import com.kooritea.fcmfix.xposed.KeepNotification;
 import com.kooritea.fcmfix.xposed.MiuiLocalNotificationFix;
+import com.kooritea.fcmfix.xposed.OplusProxyFix;
 import com.kooritea.fcmfix.xposed.PowerkeeperFix;
 import com.kooritea.fcmfix.xposed.ReconnectManagerFix;
 import com.kooritea.fcmfix.xposed.XposedModule;
@@ -41,6 +42,9 @@ public void handleLoadPackage(final XC_LoadPackage.LoadPackageParam loadPackageP
 
             XposedBridge.log("[fcmfix] com.android.server.notification.NotificationManagerService");
             new KeepNotification(loadPackageParam);
+
+            XposedBridge.log("[fcmfix] start hook com.android.server.power.OplusProxyWakeLock");
+            new OplusProxyFix(loadPackageParam);
         }
 
         if(loadPackageParam.packageName.equals("com.google.android.gms") && loadPackageParam.isFirstApplication){
@@ -54,7 +58,6 @@ public void handleLoadPackage(final XC_LoadPackage.LoadPackageParam loadPackageP
             XposedBridge.log("[fcmfix] start hook com.miui.powerkeeper");
             new PowerkeeperFix(loadPackageParam);
         }
-
     }
     private boolean fileIsExists(String strFile) {
         try {

app/src/main/java/com/kooritea/fcmfix/xposed/BroadcastFix.java

@@ -181,6 +181,8 @@ protected void beforeHookedMethod(MethodHookParam methodHookParam) {
                         }else{
                             printLog("Send Forced Start Broadcast: " + target, true);
                         }
+                        // cos15 unfreeze
+                        OplusProxyFix.unfreeze(target);
                     }
                 }
             }

app/src/main/java/com/kooritea/fcmfix/xposed/OplusProxyFix.java

@@ -0,0 +1,113 @@
+package com.kooritea.fcmfix.xposed;
+
+import android.content.pm.PackageManager;
+import android.os.WorkSource;
+
+import com.kooritea.fcmfix.util.XposedUtils;
+
+import de.robv.android.xposed.XC_MethodHook;
+import de.robv.android.xposed.XposedHelpers;
+import de.robv.android.xposed.callbacks.XC_LoadPackage;
+
+public class OplusProxyFix extends XposedModule {
+
+    private static Object s_oplusProxyWakeLock = null;
+
+    public OplusProxyFix(XC_LoadPackage.LoadPackageParam loadPackageParam) {
+        super(loadPackageParam);
+        try{
+            this.startHookOplusProxyWakeLock();
+            this.startHookOplusProxyBroadcast();
+        }catch(Throwable e) {
+            printLog("hook error OplusProxy:" + e.getMessage());
+        }
+    }
+
+    private void startHookOplusProxyBroadcast() throws Exception {
+        Class<?> oplusProxyBroadcastClass = XposedHelpers.findClass("com.android.server.am.OplusProxyBroadcast", loadPackageParam.classLoader);
+        Class<?> resultEnum = XposedHelpers.findClass("com.android.server.am.OplusProxyBroadcast$RESULT", loadPackageParam.classLoader);
+        Object notIncludeValue = XposedHelpers.getStaticObjectField(resultEnum, "NOT_INCLUDE");
+        Object proxyValue = XposedHelpers.getStaticObjectField(resultEnum, "PROXY");
+
+        /*
+        XXX only tested on OnePlus13T ColorOS 15
+        private RESULT shouldProxy( 8 args
+            00 Intent intent,
+            01 int callingPid,
+            02 int callingUid,
+            03 String callingPkg,
+            04 int uid,
+            05 String pkgName,
+            06 String action,
+            07 int appType) {
+         */
+
+        XposedUtils.findAndHookMethod(oplusProxyBroadcastClass, "shouldProxy", 8, new XC_MethodHook() {
+            @Override
+            protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
+                String callingPkg = (String)param.args[3];
+                String pkgName = (String)param.args[5];
+                String action = (String)param.args[6];
+                // positive sample: caller=com.google.android.gms, action=com.google.android.c2dm.intent.RECEIVE
+                if (isFCMAction(action) && targetIsAllow(pkgName)) {
+                    printLog("shouldProxy: bypass pkg="+pkgName+", caller="+callingPkg+", action="+action);
+                    param.setResult(notIncludeValue);
+                }
+            }
+        });
+    }
+
+    private void startHookOplusProxyWakeLock() throws Exception {
+        Class<?> oplusWakelockClass = XposedHelpers.findClass("com.android.server.power.OplusProxyWakeLock", loadPackageParam.classLoader);
+
+        XposedUtils.findAndHookConstructorAnyParam(oplusWakelockClass, new XC_MethodHook() {
+            @Override
+            protected void afterHookedMethod(MethodHookParam param) throws Throwable {
+                if (s_oplusProxyWakeLock != null) {
+                    printLog("warn: OplusProxyWakeLock constructed multiple times!");
+                    return;
+                }
+                s_oplusProxyWakeLock = param.thisObject;
+            }
+        });
+    }
+
+    private static int getTargetUidFromPackageName(String packageName) {
+        // Convert package name to UID
+        if (packageName != null) {
+            try {
+                PackageManager pm = context.getPackageManager();
+                return pm.getPackageUid(packageName, 0);
+            } catch (PackageManager.NameNotFoundException e) {
+                printLog("error: Package not found: " + packageName);
+            }
+        }
+
+        // Default to an invalid UID if we couldn't determine the target
+        return -1;
+    }
+
+    public static void unfreeze(String target) {
+        if (s_oplusProxyWakeLock == null) {
+            return;
+        }
+
+        int uid = getTargetUidFromPackageName(target);
+        if (uid < 0) {
+            return;
+        }
+
+        /*
+        XXX only tested on OnePlus13T ColorOS 15
+        unfreezeIfNeed: 3 args
+            00 int uid,
+            01 WorkSource ws,
+            02 String tag
+         */
+
+        printLog("unfreeze " + target + ", uid=" + uid);
+        WorkSource ws = new WorkSource();
+        XposedHelpers.callMethod(s_oplusProxyWakeLock, "unfreezeIfNeed", uid, ws, "FCMXX"); // 5 chars tag
+    }
+
+}

app/src/main/java/com/kooritea/fcmfix/xposed/XposedModule.java

@@ -309,13 +309,17 @@ protected void createFcmfixChannel(NotificationManagerCompat notificationManager
         }
     }
 
-    protected boolean isFCMIntent(Intent intent) {
-        String action = intent.getAction();
+    protected boolean isFCMAction(String action) {
         return action != null && (action.endsWith(".android.c2dm.intent.RECEIVE") ||
                 "com.google.firebase.MESSAGING_EVENT".equals(action) ||
                 "com.google.firebase.INSTANCE_ID_EVENT".equals(action));
     }
 
+    protected boolean isFCMIntent(Intent intent) {
+        String action = intent.getAction();
+        return isFCMAction(action);
+    }
+
      protected static String getSelfPackageName() {
         return staticLoadPackageParam == null ? "UNKNOWN":staticLoadPackageParam.packageName;
     }