Merge pull request didi#952 from Leifzhang/feature/pthread_hook_kit

[feat] 基于matrix开发了一个pthread hook的调试组件

Author: jtsky

Android/app/build.gradle

@@ -209,6 +209,7 @@ dependencies {
     implementation rootProject.ext.dependencies["tbs"]
     debugImplementation rootProject.ext.dependencies["leakcanary_android"]
     implementation 'io.reactivex.rxjava2:rxandroid:2.0.1'
+    implementation project(":dokit-pthread-hook")
 }
 
 

Android/dokit-pthread-hook/.gitignore

@@ -0,0 +1 @@
+/build

Android/dokit-pthread-hook/build.gradle

@@ -0,0 +1,81 @@
+apply plugin: 'com.android.library'
+apply plugin: 'kotlin-android'
+apply plugin: 'kotlin-kapt'
+apply from: '../upload.gradle'
+
+android {
+    compileSdkVersion rootProject.ext.android["compileSdkVersion"]
+
+    defaultConfig {
+        minSdkVersion rootProject.ext.android["minSdkVersion_21"]
+        targetSdkVersion rootProject.ext.android["targetSdkVersion"]
+        versionCode rootProject.ext.android["versionCode"]
+        versionName rootProject.ext.android["versionName"]
+
+        lintOptions {
+            abortOnError false
+        }
+    }
+
+    buildTypes {
+        debug {
+            minifyEnabled false
+            proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro'
+        }
+        release {
+            minifyEnabled false
+            proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro'
+        }
+    }
+
+    lintOptions {
+        abortOnError false
+    }
+
+    compileOptions {
+        sourceCompatibility JavaVersion.VERSION_1_8
+        targetCompatibility JavaVersion.VERSION_1_8
+    }
+
+    kotlinOptions {
+        jvmTarget = "1.8"
+    }
+    /**
+     * 支持ViewBinding
+     */
+    buildFeatures {
+        viewBinding = true
+    }
+
+}
+
+
+dependencies {
+    implementation fileTree(dir: 'libs', include: ['*.jar'])
+    testImplementation 'junit:junit:4.12'
+    androidTestImplementation 'junit:junit:4.12'
+    compileOnly project(':dokit-util')
+    compileOnly project(':dokit-okhttp-api')
+    //此处需要使用api的形式 向上暴露内部api
+    implementation project(':dokit')
+
+    if (needKotlinV14()) {
+        implementation rootProject.ext.dependencies["kotlin_v14"]
+    } else {
+        implementation rootProject.ext.dependencies["kotlin_v13"]
+    }
+    def matrixVersion = "2.0.2"
+    implementation "com.tencent.matrix:matrix-hooks:$matrixVersion"
+    implementation "com.tencent.matrix:matrix-fd:$matrixVersion"
+    implementation('com.github.kirich1409:viewbindingpropertydelegate-noreflection:1.5.2') {
+        transitive = false
+    }
+    implementation rootProject.ext.dependencies["core-ktx"]
+    implementation rootProject.ext.dependencies["fragment-ktx"]
+    implementation rootProject.ext.dependencies["appcompat"]
+    implementation rootProject.ext.dependencies["recyclerview"]
+    //okhttp wrap
+    //auto-service
+    implementation rootProject.ext.dependencies["auto_service"]
+    kapt rootProject.ext.dependencies["auto_service"]
+}

Android/dokit-pthread-hook/consumer-rules.pro

@@ -0,0 +1 @@
+ARTIFACT_ID=dokitx-pthread-hook

Android/dokit-pthread-hook/gradle.properties

@@ -0,0 +1,21 @@
+# Add project specific ProGuard rules here.
+# You can control the set of applied configuration files using the
+# proguardFiles setting in build.gradle.
+#
+# For more details, see
+#   http://developer.android.com/guide/developing/tools/proguard.html
+
+# If your project uses WebView with JS, uncomment the following
+# and specify the fully qualified class name to the JavaScript interface
+# class:
+#-keepclassmembers class fqcn.of.javascript.interface.for.webview {
+#   public *;
+#}
+
+# Uncomment this to preserve the line number information for
+# debugging stack traces.
+#-keepattributes SourceFile,LineNumberTable
+
+# If you keep the line number information, uncomment this to
+# hide the original source file name.
+#-renamesourcefileattribute SourceFile

Android/dokit-pthread-hook/proguard-rules.pro

@@ -0,0 +1,24 @@
+package com.kronos.dokit.pthread
+
+import androidx.test.platform.app.InstrumentationRegistry
+import androidx.test.ext.junit.runners.AndroidJUnit4
+
+import org.junit.Test
+import org.junit.runner.RunWith
+
+import org.junit.Assert.*
+
+/**
+ * Instrumented test, which will execute on an Android device.
+ *
+ * See [testing documentation](http://d.android.com/tools/testing).
+ */
+@RunWith(AndroidJUnit4::class)
+class ExampleInstrumentedTest {
+    @Test
+    fun useAppContext() {
+        // Context of the app under test.
+        val appContext = InstrumentationRegistry.getInstrumentation().targetContext
+        assertEquals("com.kronos.dokit.pthread.test", appContext.packageName)
+    }
+}

Android/dokit-pthread-hook/src/androidTest/java/com/kronos/dokit/pthread/ExampleInstrumentedTest.kt

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="utf-8"?>
+<manifest xmlns:android="http://schemas.android.com/apk/res/android"
+    package="com.kronos.dokit.pthread">
+
+    <application>
+        <activity
+            android:name=".ui.PThreadHookUiActivity"
+            android:theme="@style/Theme.AppCompat.Light.NoActionBar" />
+    </application>
+</manifest>

Android/dokit-pthread-hook/src/main/AndroidManifest.xml

@@ -0,0 +1,89 @@
+package com.kronos.dokit.pthread
+
+import android.app.Activity
+import android.app.Application
+import android.os.Bundle
+import com.didichuxing.doraemonkit.DoKit
+
+/**
+ *
+ *  @Author LiABao
+ *  @Since 2021/11/17
+ *
+ */
+class AutoDumpListener : Application.ActivityLifecycleCallbacks {
+
+    init {
+        DoKit.APPLICATION.registerActivityLifecycleCallbacks(this)
+    }
+
+
+    private fun getThreadCount(): Int {
+        return getAllThreads().size
+    }
+
+    private var count = 0
+
+    /**
+     * ps -p `self` -t
+     * See http://man7.org/linux/man-pages/man1/ps.1.html
+     */
+    private fun getAllThreads(): MutableList<Thread> {
+        var group = Thread.currentThread().threadGroup
+        var system: ThreadGroup?
+        do {
+            system = group
+            group = group?.parent
+        } while (group != null)
+        val count = system?.activeCount() ?: 0
+        val threads = arrayOfNulls<Thread>(count)
+        system?.enumerate(threads)
+        val list = mutableListOf<Thread>()
+        threads.forEach {
+            it?.let { it1 -> list.add(it1) }
+        }
+        return list
+    }
+
+    override fun onActivityCreated(activity: Activity, savedInstanceState: Bundle?) {
+
+    }
+
+    override fun onActivityStarted(activity: Activity) {
+
+    }
+
+    override fun onActivityResumed(activity: Activity) {
+
+    }
+
+    override fun onActivityPaused(activity: Activity) {
+
+    }
+
+    override fun onActivityStopped(activity: Activity) {
+    }
+
+    override fun onActivitySaveInstanceState(activity: Activity, outState: Bundle) {
+
+    }
+
+    override fun onActivityDestroyed(activity: Activity) {
+        if (count > 3) {
+            return
+        }
+        if (getThreadCount() > Threshold) {
+            DoKit.APPLICATION.dump {
+                forEach {
+
+                }
+
+            }
+        }
+    }
+
+
+    companion object {
+        const val Threshold = 200
+    }
+}

Android/dokit-pthread-hook/src/main/java/com/kronos/dokit/pthread/AutoDumpListener.kt

@@ -0,0 +1,54 @@
+package com.kronos.dokit.pthread
+
+import android.content.Context
+import android.util.Log
+import com.tencent.matrix.fd.FDDumpBridge
+import com.tencent.matrix.hook.pthread.PthreadHook
+import java.io.File
+
+/**
+ *
+ *  @Author LiABao
+ *  @Since 2021/11/16
+ *
+ */
+
+// 当前没有删除这部分文件 后续需要考虑的
+fun Context.dump(invoke: MutableList<PThreadEntity>.() -> Unit = {}) {
+    try {
+        val parent = "$cacheDir/pthread"
+        val output = "$parent/pthread_hook_${System.currentTimeMillis() / 1000L}.log"
+        parent.createDirectory()
+        PthreadHook.INSTANCE.dump(output)
+        fdLimit()
+        val pthreads = getJson(output).parserPThread()
+        pthreads.forEach {
+            Log.i(TAG, "pthread error :$it\n")
+        }
+        invoke.invoke(pthreads)
+    } catch (e: Exception) {
+
+    }
+}
+
+fun String.createDirectory() {
+    val file = File(this)
+    if (!file.exists()) {
+        file.mkdir()
+    }
+}
+
+fun fdLimit() {
+    Log.i(TAG, "FD limit = " + FDDumpBridge.getFDLimit())
+}
+
+fun getJson(path: String): String {
+    val stream = File(path).bufferedReader().readText()
+    return stream.apply {
+        Log.i(TAG, "pThread:$this")
+    }
+}
+
+const val TAG = "PThreadDumpHelper"
+
+

Android/dokit-pthread-hook/src/main/java/com/kronos/dokit/pthread/PThreadDumpHelper.kt

@@ -0,0 +1,51 @@
+package com.kronos.dokit.pthread
+
+import org.json.JSONObject
+
+/**
+ *
+ *  @Author LiABao
+ *  @Since 2021/11/17
+ *
+ */
+data class PThreadEntity(
+    val hash: String, val native: String, val java: String,
+    val count: Int, val threads: MutableList<ThreadNameEntity>
+) {
+
+
+}
+
+data class ThreadNameEntity(val tid: String, val name: String)
+
+
+fun JSONObject.parserThreads(): MutableList<ThreadNameEntity> {
+    val jsonArray = optJSONArray("threads")
+    val index = jsonArray?.length() ?: 0
+    val threads = mutableListOf<ThreadNameEntity>()
+    for (index in 0 until index) {
+        val thread = jsonArray.optJSONObject(index)
+        threads.add(ThreadNameEntity(thread.optString("tid"), thread.optString("name")))
+    }
+    return threads
+}
+
+fun String.parserPThread(): MutableList<PThreadEntity> {
+    val jsonObject = JSONObject(this)
+    val pThreadArray = jsonObject.optJSONArray("PthreadHook_not_exited")
+    val pThreads = mutableListOf<PThreadEntity>()
+    pThreadArray?.apply {
+        for (index in 0 until length()) {
+            val jsonObject = optJSONObject(index)
+            pThreads.add(
+                PThreadEntity(
+                    jsonObject.optString("hash"), jsonObject.optString("native"),
+                    jsonObject.optString("java"), jsonObject.optInt("count", 0),
+                    jsonObject.parserThreads()
+                )
+            )
+        }
+    }
+    return pThreads
+}
+