✨ Clinit

yhs0602 · May 15, 2024 · 49ba31f · 49ba31f
1 parent e609eed
commit 49ba31f
Show file tree

Hide file tree

Showing 6 changed files with 31 additions and 0 deletions.
diff --git a/src/main/kotlin/vm/classloader/DexClassLoader.kt b/src/main/kotlin/vm/classloader/DexClassLoader.kt
@@ -54,6 +54,7 @@ class DexClassLoader(
         )
         loadedTypes[typeId] = loadedType
         loadedMockedClasses.add(mockedClass.clazz)
+        loadedType.callClInit()
         return loadedType
     }
 

diff --git a/src/main/kotlin/vm/classloader/DexDefinedType.kt b/src/main/kotlin/vm/classloader/DexDefinedType.kt
@@ -1,6 +1,7 @@
 package com.yhs0602.vm.classloader
 
 import com.yhs0602.dex.ParsedClass
+import com.yhs0602.vm.Environment
 import com.yhs0602.vm.MethodWrapper
 import net.bytebuddy.ByteBuddy
 import net.bytebuddy.dynamic.loading.ClassLoadingStrategy
@@ -26,6 +27,8 @@ class DexDefinedType(
     override val clazz by lazy {
         makeClazz()
     }
+    private var clinit: MethodTableEntry? = null
+    private var hasCalledClinit = false
 
     init {
         // populate v-table and i-table of the class
@@ -55,6 +58,9 @@ class DexDefinedType(
                 )
                 if (method.accessFlags.isStatic) {
                     staticMethods[methodTableEntry] = MethodWrapper.Encoded(method)
+                    if (methodId.name == "<clinit>") {
+                        clinit = methodTableEntry
+                    }
                 } else if (method.accessFlags.isConstructor) {
                     constructors[methodTableEntry] = MethodWrapper.Encoded(method)
                 } else {
@@ -81,6 +87,26 @@ class DexDefinedType(
         }
     }
 
+    override fun callClInit() {
+        if (hasCalledClinit) {
+            return
+        }
+        clinit?.let {
+            val clInitMethod = getMethod(it)
+            if (clInitMethod !is MethodWrapper.Encoded) {
+                throw IllegalStateException("clinit method is not encoded")
+            }
+            hasCalledClinit = true
+            clInitMethod.execute(
+                arrayOf(),
+                Environment.getInstance(),
+                clInitMethod.encodedMethod.codeItem ?: throw IllegalStateException("clinit method has no code"),
+                true,
+                0
+            )
+        }
+    }
+
     private fun makeClazz(): Class<*> {
         // use bytebuddy to create a class
         var builder = ByteBuddy()

diff --git a/src/main/kotlin/vm/classloader/MockedType.kt b/src/main/kotlin/vm/classloader/MockedType.kt
@@ -20,6 +20,7 @@ class MockedType(
     override val methods = mutableMapOf<MethodTableEntry, MethodWrapper>()
     override val constructors = mutableMapOf<MethodTableEntry, MethodWrapper>()
     override val staticMethods = mutableMapOf<MethodTableEntry, MethodWrapper>()
+    override fun callClInit() {} // no clinit for Mocked type, as it is already initialized
 
     init {
         // populate v-table and i-table of the class

diff --git a/src/main/kotlin/vm/classloader/ObjectType.kt b/src/main/kotlin/vm/classloader/ObjectType.kt
@@ -23,6 +23,7 @@ data object ObjectType : Type() {
     override val staticMethods = mutableMapOf<MethodTableEntry, MethodWrapper>()
     override val constructors = mutableMapOf<MethodTableEntry, MethodWrapper>()
     override val clazz: Class<*> = java.lang.Object::class.java
+    override fun callClInit() {} // no clinit for Object
 
     init {
         // populate v-table and i-table of Object

diff --git a/src/main/kotlin/vm/classloader/PrimitiveType.kt b/src/main/kotlin/vm/classloader/PrimitiveType.kt
@@ -19,6 +19,7 @@ class Primitive(override val descriptor: String) : Type() {
     override val methods = emptyMap<MethodTableEntry, MethodWrapper>()
     override val constructors = emptyMap<MethodTableEntry, MethodWrapper>()
     override val staticMethods = emptyMap<MethodTableEntry, MethodWrapper>()
+    override fun callClInit() {} // no clinit for primitive types
 
     override val clazz: Class<*> by lazy {
         when (descriptor) {

diff --git a/src/main/kotlin/vm/classloader/Type.kt b/src/main/kotlin/vm/classloader/Type.kt
@@ -40,6 +40,7 @@ sealed class Type {
     abstract val staticMethods: Map<MethodTableEntry, MethodWrapper>
     abstract val constructors: Map<MethodTableEntry, MethodWrapper>
     abstract val clazz: Class<*>
+    abstract fun callClInit()
 
     // Use this method to get the method entry in the virtual table
     fun getVirtualMethodEntry(givenEntry: MethodTableEntry): MethodTableEntry {