feat: add test functions for java

archguard · Mar 28, 2022 · fecf421 · fecf421
1 parent 9082f2d
commit fecf421
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/scan_bytecode/src/main/kotlin/org/archguard/scanner/bytecode/ByteCodeParser.kt b/scan_bytecode/src/main/kotlin/org/archguard/scanner/bytecode/ByteCodeParser.kt
@@ -1,9 +1,11 @@
 package org.archguard.scanner.bytecode
 
 import chapi.domain.core.CodeDataStruct
+import chapi.domain.core.CodeFunction
 import org.objectweb.asm.ClassReader
 import org.objectweb.asm.Type
 import org.objectweb.asm.tree.ClassNode
+import org.objectweb.asm.tree.MethodNode
 import org.slf4j.LoggerFactory
 import java.io.File
 import java.io.FileInputStream
@@ -34,9 +36,18 @@ class ByteCodeParser {
         val ds = CodeDataStruct()
         ds.NodeName = getClassName(classNode.name).toString()
 
+        classNode.methods.forEach {
+            ds.Functions += this.createMethod(it, ds.NodeName, classNode)
+        }
+
         return ds
     }
 
+    private fun createMethod(methodNode: MethodNode, nodeName: String, classNode: ClassNode): CodeFunction {
+        val codeFunction = CodeFunction(Name = methodNode.name)
+        return codeFunction
+    }
+
 
     private fun getClassName(internalName: String): String? {
         return Type.getObjectType(internalName).className

diff --git a/scan_bytecode/src/test/kotlin/org/archguard/scanner/bytecode/ByteCodeParserTest.kt b/scan_bytecode/src/test/kotlin/org/archguard/scanner/bytecode/ByteCodeParserTest.kt
@@ -12,5 +12,8 @@ internal class ByteCodeParserTest {
 
         val ds = ByteCodeParser().parseClassFile(path)
         assertEquals("org.archguard.demo.HelloWorld", ds.NodeName)
+        assertEquals(2, ds.Functions.size)
+        assertEquals("<init>", ds.Functions[0].Name)
+        assertEquals("main", ds.Functions[1].Name)
     }
 }