Use ClassReader.EXPAND_FRAMES as dictated by the AnalyzerAdapter …

…documentation (#368) Signed-off-by: Evgeniy Moiseenko <[email protected]>
JetBrains · Aug 28, 2024 · d852db6 · d852db6
1 parent 6163a8f
commit d852db6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/jvm/main/org/jetbrains/kotlinx/lincheck/transformation/LincheckJavaAgent.kt b/src/jvm/main/org/jetbrains/kotlinx/lincheck/transformation/LincheckJavaAgent.kt
@@ -359,8 +359,9 @@ internal object LincheckClassFileTransformer : ClassFileTransformer {
     ): ByteArray = transformedClassesCache.computeIfAbsent(internalClassName.canonicalClassName) {
         val reader = ClassReader(classBytes)
         val writer = SafeClassWriter(reader, loader, ClassWriter.COMPUTE_FRAMES)
+        val visitor = LincheckClassVisitor(instrumentationMode, writer)
         try {
-            reader.accept(LincheckClassVisitor(instrumentationMode, writer), ClassReader.SKIP_FRAMES)
+            reader.accept(visitor, ClassReader.EXPAND_FRAMES)
             writer.toByteArray()
         } catch (e: Throwable) {
             System.err.println("Unable to transform $internalClassName")