diff options
author | Mark <mteffeteller@google.com> | 2023-06-21 23:30:18 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2023-06-21 23:30:18 +0000 |
commit | 54819157eaa66e14f2c68b54609bd6bfa360b708 (patch) | |
tree | 68cf332a40b94b2d28b256b19b916f99220bb0c4 /src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt | |
parent | ba37c2e361c2ba91bacc47fcae5383c52e50f6be (diff) | |
parent | e73be1680dae58cb83d869104def1c59102d59b2 (diff) | |
download | jazzer-api-54819157eaa66e14f2c68b54609bd6bfa360b708.tar.gz |
Sync jazzer in AOSP with upstream repo (new SHA: 30decf81a147c66fa5a098072c38ab6924ba0aa6) am: 9350e0ab03 am: 99d9a79746 am: 34a8e5c8aa am: e73be1680d
Original change: https://android-review.googlesource.com/c/platform/external/jazzer-api/+/2627336
Change-Id: I1b97ed5cdcf2adda4d443148cc0d447974e51785
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
Diffstat (limited to 'src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt')
-rw-r--r-- | src/main/java/com/code_intelligence/jazzer/instrumentor/ClassInstrumentor.kt | 55 |
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 + } +} |