1 files changed, 31 insertions, 21 deletions
diff --git a/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/PatchTestUtils.kt b/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/PatchTestUtils.kt
index f286d03f..00279c35 100644
--- a/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/PatchTestUtils.kt
+++ b/agent/src/test/java/com/code_intelligence/jazzer/instrumentor/PatchTestUtils.kt
@@ -14,30 +14,40 @@
 
 package com.code_intelligence.jazzer.instrumentor
 
-fun classToBytecode(targetClass: Class<*>): ByteArray {
-    return ClassLoader
-        .getSystemClassLoader()
-        .getResourceAsStream("${targetClass.name.replace('.', '/')}.class")!!
-        .use {
-            it.readBytes()
-        }
-}
+import java.io.FileOutputStream
 
-fun bytecodeToClass(name: String, bytecode: ByteArray): Class<*> {
-    return BytecodeClassLoader(name, bytecode).loadClass(name)
-}
+object PatchTestUtils {
+    @JvmStatic
+    fun classToBytecode(targetClass: Class<*>): ByteArray {
+        return ClassLoader
+            .getSystemClassLoader()
+            .getResourceAsStream("${targetClass.name.replace('.', '/')}.class")!!
+            .use {
+                it.readBytes()
+            }
+    }
 
-/**
- * A ClassLoader that dynamically loads a single specified class from byte code and delegates all other class loads to
- * its own ClassLoader.
- */
-class BytecodeClassLoader(val className: String, private val classBytecode: ByteArray) :
-    ClassLoader(BytecodeClassLoader::class.java.classLoader) {
-    override fun loadClass(name: String): Class<*> {
-        if (name != className)
-            return super.loadClass(name)
+    @JvmStatic
+    fun bytecodeToClass(name: String, bytecode: ByteArray): Class<*> {
+        return BytecodeClassLoader(name, bytecode).loadClass(name)
+    }
+
+    @JvmStatic
+    public fun dumpBytecode(outDir: String, name: String, originalBytecode: ByteArray) {
+        FileOutputStream("$outDir/$name.class").use { fos -> fos.write(originalBytecode) }
+    }
 
-        return defineClass(className, classBytecode, 0, classBytecode.size)
+    /**
+     * A ClassLoader that dynamically loads a single specified class from byte code and delegates all other class loads to
+     * its own ClassLoader.
+     */
+    class BytecodeClassLoader(val className: String, private val classBytecode: ByteArray) :
+        ClassLoader(BytecodeClassLoader::class.java.classLoader) {
+        override fun loadClass(name: String): Class<*> {
+            if (name != className)
+                return super.loadClass(name)
+            return defineClass(className, classBytecode, 0, classBytecode.size)
+        }
     }
 }