refactor(ShareLimit): convert from Java to Kotlin and fix bug in latest WA

Author: Dev4Mod

app/src/main/java/com/wmods/wppenhacer/xposed/core/devkit/Unobfuscator.kt

@@ -736,34 +736,6 @@ object Unobfuscator {
         }
     }
 
-    @Throws(Exception::class)
-    @JvmStatic
-    fun loadShareLimitMethod(classLoader: ClassLoader): Method {
-        return UnobfuscatorCache.getInstance().getMethod(classLoader) {
-            findFirstMethodUsingStrings(
-                classLoader,
-                StringMatchType.Contains,
-                "send_max_video_duration"
-            )
-                ?: throw Exception("ShareLimit method not found")
-        }
-    }
-
-    @Throws(Exception::class)
-    @JvmStatic
-    fun loadShareMapItemField(classLoader: ClassLoader): Field {
-        return UnobfuscatorCache.getInstance().getField(classLoader) {
-            val shareLimitMethod = loadShareLimitMethod(classLoader)
-            val methodData = bridge.getMethodData(shareLimitMethod) ?: return@getField null
-            val usingFields = methodData.usingFields
-            for (ufield in usingFields) {
-                val field = ufield.field.getFieldInstance(classLoader)
-                if (field.type == Map::class.java) return@getField field
-            }
-            throw Exception("ShareItem field not found")
-        }
-    }
-
     @Throws(Exception::class)
     @JvmStatic
     fun loadMenuManagerClass(classLoader: ClassLoader): Class<*> {
@@ -3405,7 +3377,6 @@ object Unobfuscator {
     }
 
     @Throws(Exception::class)
-
     @JvmStatic
     fun loadOnConversationsListChangedMethod(classLoader: ClassLoader): Method? {
         return UnobfuscatorCache.getInstance().getMethod(classLoader) {
@@ -3417,4 +3388,14 @@ object Unobfuscator {
             )
         }
     }
+
+    fun loadMultiSelectionLimitInfoClass(classLoader: ClassLoader): Class<*> {
+        return UnobfuscatorCache.getInstance().getClass(classLoader) {
+            bridge.findClass {
+                matcher {
+                    usingStrings("MultiSelectionLimitInfo")
+                }
+            }.single().getInstance(classLoader)
+        }
+    }
 }

app/src/main/java/com/wmods/wppenhacer/xposed/features/general/ShareLimit.kt

@@ -1,58 +1,22 @@
-package com.wmods.wppenhacer.xposed.features.general;
-
-import androidx.annotation.NonNull;
-
-import com.wmods.wppenhacer.xposed.core.Feature;
-import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator;
-
-import java.util.HashMap;
-
-import de.robv.android.xposed.XC_MethodHook;
-import de.robv.android.xposed.XSharedPreferences;
-import de.robv.android.xposed.XposedBridge;
-
-public class ShareLimit extends Feature {
-    public ShareLimit(ClassLoader loader, XSharedPreferences preferences) {
-        super(loader, preferences);
+package com.wmods.wppenhacer.xposed.features.general
+
+import com.wmods.wppenhacer.xposed.core.Feature
+import com.wmods.wppenhacer.xposed.core.devkit.Unobfuscator
+import de.robv.android.xposed.XC_MethodHook
+import de.robv.android.xposed.XSharedPreferences
+import de.robv.android.xposed.XposedBridge
+
+class ShareLimit(classLoader: ClassLoader, prefs: XSharedPreferences) : Feature(classLoader, prefs) {
+
+    override fun doHook() {
+        if(!prefs.getBoolean("removeforwardlimit", false)) return
+        val multiSelectionLimitInfoClass = Unobfuscator.loadMultiSelectionLimitInfoClass(classLoader)
+        XposedBridge.hookAllConstructors(multiSelectionLimitInfoClass, object : XC_MethodHook() {
+            override fun beforeHookedMethod(param: MethodHookParam) {
+                param.args[0] = Int.MAX_VALUE
+            }
+        })
     }
 
-    public void doHook() throws Exception {
-        if (!prefs.getBoolean("removeforwardlimit", false)) return;
-        var shareLimitMethod = Unobfuscator.loadShareLimitMethod(classLoader);
-        logDebug(Unobfuscator.getMethodDescriptor(shareLimitMethod));
-        var shareItemField = Unobfuscator.loadShareMapItemField(classLoader);
-        logDebug(Unobfuscator.getFieldDescriptor(shareItemField));
-
-        XposedBridge.hookMethod(
-                shareLimitMethod,
-                new XC_MethodHook() {
-                    private HashMap<Object, Object> fakeMap;
-                    private HashMap<Object, Object> mMap;
-
-                    /**
-                     * @noinspection unchecked
-                     */
-                    @Override
-                    protected void beforeHookedMethod(MethodHookParam param) throws Throwable {
-                        fakeMap = new HashMap<>();
-                        mMap = (HashMap<Object, Object>) shareItemField.get(param.thisObject);
-                        shareItemField.set(param.thisObject, fakeMap);
-                    }
-
-                    @Override
-                    protected void afterHookedMethod(MethodHookParam param) throws Throwable {
-                        mMap.putAll(fakeMap);
-                        shareItemField.set(param.thisObject, mMap);
-                        fakeMap.clear();
-                    }
-                });
-
-
-    }
-
-    @NonNull
-    @Override
-    public String getPluginName() {
-        return "Share Limit";
-    }
-}
+    override fun getPluginName(): String = "Share Limit"
+}