add ToastCompat

Author: porum

defensor-gradle-plugin/src/main/java/com/panda912/defensor/plugin/internal/interceptor/ToastInterceptor.kt

@@ -0,0 +1,65 @@
+package com.panda912.defensor.plugin.internal.interceptor
+
+import com.panda912.defensor.plugin.internal.BytecodeInterceptor
+import com.panda912.defensor.plugin.internal.Output
+import com.panda912.defensor.plugin.internal.visitor.BaseClassReader
+import com.panda912.defensor.plugin.internal.visitor.BaseClassVisitor
+import com.panda912.defensor.plugin.internal.visitor.BaseMethodVisitor
+import com.panda912.defensor.plugin.utils.TOAST_COMPAT
+import com.panda912.defensor.plugin.utils.convertToStaticDescriptor
+import com.panda912.defensor.plugin.utils.toInternalName
+import org.objectweb.asm.ClassWriter
+import org.objectweb.asm.MethodVisitor
+import org.objectweb.asm.Opcodes
+
+/**
+ * Created by panda on 2022/5/24 19:00
+ */
+class ToastInterceptor : BytecodeInterceptor {
+
+  override fun intercept(chain: BytecodeInterceptor.Chain): Output {
+    val input = chain.request()
+
+    val cr = BaseClassReader(input.bytes)
+    val cw = ClassWriter(cr, ClassWriter.COMPUTE_MAXS)
+    cr.accept(object : BaseClassVisitor(cw) {
+      override fun visitMethod(
+        access: Int,
+        name: String?,
+        descriptor: String?,
+        signature: String?,
+        exceptions: Array<out String>?
+      ): MethodVisitor {
+        val mv = super.visitMethod(access, name, descriptor, signature, exceptions)
+        return ToastMethodVisitor(mv)
+      }
+    })
+    input.bytes = cw.toByteArray()
+    return chain.proceed(input)
+  }
+}
+
+class ToastMethodVisitor(mv: MethodVisitor) : BaseMethodVisitor(mv) {
+
+  override fun visitMethodInsn(
+    opcode: Int,
+    owner: String,
+    name: String,
+    descriptor: String,
+    isInterface: Boolean
+  ) {
+
+    if (owner == "android/widget/Toast" && name == "show" && opcode == Opcodes.INVOKEVIRTUAL && descriptor == "()V") {
+      super.visitMethodInsn(
+        Opcodes.INVOKESTATIC,
+        TOAST_COMPAT.toInternalName(),
+        name,
+        descriptor.convertToStaticDescriptor("Landroid/widget/Toast;"),
+        isInterface
+      )
+      return
+    }
+
+    super.visitMethodInsn(opcode, owner, name, descriptor, isInterface)
+  }
+}

defensor-gradle-plugin/src/main/java/com/panda912/defensor/plugin/internal/processor/BytecodeInjectorImpl.kt

@@ -38,6 +38,7 @@ class BytecodeInjectorImpl : BytecodeInjector {
       SafeDialogInterceptor(),
       StringInterceptor(),
       ThrowableInterceptor(),
+      ToastInterceptor(),
       UnboxingInterceptor(),
       UriInterceptor(),
       ValueAnimatorInterceptor(),

defensor-gradle-plugin/src/main/java/com/panda912/defensor/plugin/utils/Const.kt

@@ -1,7 +1,6 @@
 package com.panda912.defensor.plugin.utils
 
 import org.objectweb.asm.Type
-import java.io.File
 
 /**
  * Created by panda on 2021/8/13 17:25
@@ -85,6 +84,7 @@ const val SPANNABLE_STRING_BUILDER_DEFENSOR = "com.panda912.defensor.internal.Sp
 const val STRING_DEFENSOR = "com.panda912.defensor.internal.StringDefensor"
 const val TEXTVIEW_DEFENSOR = "com.panda912.defensor.internal.TextViewDefensor"
 const val THROWABLE_DEFENSOR = "com.panda912.defensor.internal.ThrowableDefensor"
+const val TOAST_COMPAT = "com.panda912.defensor.internal.ToastCompat"
 const val UNBOXING_DEFENSOR = "com.panda912.defensor.internal.UnboxingDefensor"
 const val URI_DEFENSOR = "com.panda912.defensor.internal.UriDefensor"
 const val VALUE_ANIMATOR_DEFENSOR = "com.panda912.defensor.internal.ValueAnimatorDefensor"

defensor/src/main/java/com/panda912/defensor/ErrorCode.java

@@ -21,4 +21,5 @@ public interface ErrorCode {
   int DialogException = 114;
   int URLDecoderException = 115;
   int IllegalStateException = 116;
+  int ToastException = 117;
 }

defensor/src/main/java/com/panda912/defensor/internal/ToastCompat.java

@@ -0,0 +1,69 @@
+package com.panda912.defensor.internal;
+
+import android.os.Handler;
+import android.os.Message;
+import android.widget.Toast;
+
+import androidx.annotation.NonNull;
+
+import com.panda912.defensor.CrashDefensor;
+import com.panda912.defensor.ErrorCode;
+
+import java.lang.reflect.Field;
+
+/**
+ * Created by panda on 2020/9/18 18:33
+ */
+public class ToastCompat {
+
+  private static Field mTNField;
+  private static Field mHandlerField;
+
+  static {
+    try {
+      mTNField = Toast.class.getDeclaredField("mTN");
+      mTNField.setAccessible(true);
+      mHandlerField = mTNField.getType().getDeclaredField("mHandler");
+      mHandlerField.setAccessible(true);
+    } catch (Throwable e) {
+      CrashDefensor.onCrash(ErrorCode.ToastException, "reflect get Toast mTN or mHandler failed", e);
+    }
+  }
+
+  private static void hook(Toast toast) {
+    try {
+      Object mTN = mTNField.get(toast);
+      mHandlerField.set(mTN, new HandlerProxy((Handler) mHandlerField.get(mTN)));
+    } catch (Throwable e) {
+      CrashDefensor.onCrash(ErrorCode.ToastException, "hook Toast mHandler failed", e);
+    }
+  }
+
+  private static class HandlerProxy extends Handler {
+
+    private final Handler mHandler;
+
+    HandlerProxy(Handler handler) {
+      mHandler = handler;
+    }
+
+    @Override
+    public void dispatchMessage(@NonNull Message msg) {
+      try {
+        super.dispatchMessage(msg);
+      } catch (Exception e) {
+        CrashDefensor.onCrash(ErrorCode.ToastException, "Android 7.1 BadTokenException OR other custom device has BadTokenException", e);
+      }
+    }
+
+    @Override
+    public void handleMessage(@NonNull Message msg) {
+      mHandler.handleMessage(msg);
+    }
+  }
+
+  public static void show(Toast toast) {
+    hook(toast);
+    toast.show();
+  }
+}

sample/src/main/java/com/panda912/defensor/sample/MainActivity.kt

@@ -42,6 +42,8 @@ class MainActivity : AppCompatActivity() {
       )
     )
 
+    Toast.makeText(this.applicationContext, "Hello defensor!", Toast.LENGTH_SHORT).show()
+
 //    val context: Context = this
 //    context.bindService(Intent(), object : ServiceConnection {
 //      override fun onServiceConnected(name: ComponentName?, service: IBinder?) {