1 files changed, 55 insertions, 0 deletions
diff --git a/src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt b/src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt
new file mode 100644
index 00000000..a93e29c7
--- /dev/null
+++ b/src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt
@@ -0,0 +1,55 @@
+// Copyright 2021 Code Intelligence GmbH
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package com.code_intelligence.jazzer.instrumentor
+
+import com.code_intelligence.jazzer.runtime.CoverageMap
+
+fun extractClassFileMajorVersion(classfileBuffer: ByteArray): Int {
+    return ((classfileBuffer[6].toInt() and 0xff) shl 8) or (classfileBuffer[7].toInt() and 0xff)
+}
+
+class ClassInstrumentor(private val internalClassName: String, bytecode: ByteArray) {
+
+    var instrumentedBytecode = bytecode
+        private set
+
+    fun coverage(initialEdgeId: Int): Int {
+        val edgeCoverageInstrumentor = EdgeCoverageInstrumentor(
+            defaultEdgeCoverageStrategy,
+            defaultCoverageMap,
+            initialEdgeId,
+        )
+        instrumentedBytecode = edgeCoverageInstrumentor.instrument(internalClassName, instrumentedBytecode)
+        return edgeCoverageInstrumentor.numEdges
+    }
+
+    fun traceDataFlow(instrumentations: Set<InstrumentationType>) {
+        instrumentedBytecode =
+            TraceDataFlowInstrumentor(instrumentations).instrument(internalClassName, instrumentedBytecode)
+    }
+
+    fun hooks(hooks: Iterable<Hook>, classWithHooksEnabledField: String?) {
+        instrumentedBytecode = HookInstrumentor(
+            hooks,
+            java6Mode = extractClassFileMajorVersion(instrumentedBytecode) < 51,
+            classWithHooksEnabledField = classWithHooksEnabledField,
+        ).instrument(internalClassName, instrumentedBytecode)
+    }
+
+    companion object {
+        val defaultEdgeCoverageStrategy = StaticMethodStrategy()
+        val defaultCoverageMap = CoverageMap::class.java
+    }
+}