opt: sync gms doze

Author: lingqiqi5211

library/hidden-api/src/main/java/android/os/PowerExemptionManager.java

@@ -0,0 +1,12 @@
+package android.os;
+
+import android.content.Context;
+
+public class PowerExemptionManager {
+    public PowerExemptionManager(Context context) {
+    }
+
+    public void addToTemporaryAllowList(String packageName, int reasonCode,
+                                        String reason, long durationMs) {
+    }
+}

library/libhook/src/main/java/com/sevtinge/hyperceiler/libhook/rules/powerkeeper/GmsDozeFix.java

@@ -21,10 +21,13 @@
 
 import static com.sevtinge.hyperceiler.libhook.utils.hookapi.tool.EzxHelpUtils.deoptimize;
 
+import android.os.Bundle;
+
 import com.sevtinge.hyperceiler.common.log.XposedLog;
 import com.sevtinge.hyperceiler.libhook.base.BaseHook;
 
 import java.lang.reflect.Method;
+import java.util.List;
 
 import io.github.libxposed.api.XposedInterface;
 
@@ -33,6 +36,8 @@
  * https://github.com/Howard20181/HyperOS_FCM_Live/blob/main/HyperFCMLive/src/main/java/io/github/howard20181/hyperos/fcmlive/Hooker.java
  */
 public class GmsDozeFix extends BaseHook {
+    private static final String GMS_PACKAGE_NAME = "com.google.android.gms";
+
     private static final String[] GMS_OBSERVER_METHODS = new String[]{
         "updateGmsNetWork",
         "updateGmsAlarm",
@@ -49,6 +54,29 @@ public class GmsDozeFix extends BaseHook {
 
     @Override
     public void init() {
+        hookGmsObserver();
+        hookGlobalFeatureConfigureHelper();
+    }
+
+    private void hookGmsObserver() {
+        Class<?> netdExecutorClass = findClassIfExists("com.miui.powerkeeper.utils.NetdExecutor");
+        if (netdExecutorClass != null) {
+            try {
+                Method initGmsChainMethod = netdExecutorClass.getDeclaredMethod(
+                    "initGmsChain", String.class, int.class, String.class);
+                chain(initGmsChainMethod, chain -> {
+                    Object[] args = chain.getArgs().toArray();
+                    args[2] = "ACCEPT";
+                    return chain.proceed(args);
+                });
+                deoptimize(initGmsChainMethod);
+            } catch (NoSuchMethodException e) {
+                XposedLog.w(TAG, getPackageName(), "Skip missing method: NetdExecutor#initGmsChain", e);
+            } catch (Throwable t) {
+                XposedLog.e(TAG, getPackageName(), "Hook failed in initGmsChain", t);
+            }
+        }
+
         Class<?> gmsObserverClass = findClassIfExists("com.miui.powerkeeper.utils.GmsObserver");
         if (gmsObserverClass == null) return;
 
@@ -57,6 +85,33 @@ public void init() {
         }
     }
 
+    @SuppressWarnings("unchecked")
+    private void hookGlobalFeatureConfigureHelper() {
+        Class<?> helperClass = findClassIfExists(
+            "com.miui.powerkeeper.provider.GlobalFeatureConfigureHelper");
+        if (helperClass == null) return;
+
+        try {
+            Method getDozeWhiteListAppsMethod = helperClass.getDeclaredMethod(
+                "getDozeWhiteListApps", Bundle.class);
+            chain(getDozeWhiteListAppsMethod, chain -> {
+                Object result = chain.proceed();
+                if (result instanceof List<?>) {
+                    List<String> whiteList = (List<String>) result;
+                    if (!whiteList.contains(GMS_PACKAGE_NAME)) {
+                        whiteList.add(GMS_PACKAGE_NAME);
+                    }
+                }
+                return result;
+            });
+        } catch (NoSuchMethodException e) {
+            XposedLog.w(TAG, getPackageName(),
+                "Skip missing method: GlobalFeatureConfigureHelper#getDozeWhiteListApps", e);
+        } catch (Throwable t) {
+            XposedLog.e(TAG, getPackageName(), "Hook failed in getDozeWhiteListApps", t);
+        }
+    }
+
     private void hookObserverMethod(Class<?> observerClass, String methodName) {
         try {
             Method method = observerClass.getDeclaredMethod(methodName, boolean.class);