Refactor byte-code transformation

src/jvm/main/org/jetbrains/kotlinx/lincheck/AtomicFieldUpdaterNames.kt

@@ -0,0 +1,62 @@
+/*
+ * Lincheck
+ *
+ * Copyright (C) 2019 - 2024 JetBrains s.r.o.
+ *
+ * This Source Code Form is subject to the terms of the
+ * Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed
+ * with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.jetbrains.kotlinx.lincheck
+
+import sun.misc.Unsafe
+import java.lang.reflect.Modifier
+import java.util.concurrent.atomic.AtomicIntegerFieldUpdater
+import java.util.concurrent.atomic.AtomicLongFieldUpdater
+import java.util.concurrent.atomic.AtomicReferenceFieldUpdater
+
+/**
+ * [AtomicFieldUpdaterNames] is used to map atomic objects to their names (usually, the corresponding field names)
+ * and class owners. The weak identity hash map ensures that atomic objects are compared using reference
+ * equality and does not prevent them from being garbage collected.
+ */
+internal object AtomicFieldUpdaterNames {
+    private val unsafe: Unsafe = try {
+        val unsafeField = Unsafe::class.java.getDeclaredField("theUnsafe")
+        unsafeField.isAccessible = true
+        unsafeField.get(null) as Unsafe
+    } catch (ex: Exception) {
+        throw RuntimeException("Can't get the Unsafe instance, please report it to the Lincheck team", ex)
+    }
+
+    fun getName(updater: Any): String? {
+        if (updater !is AtomicIntegerFieldUpdater<*> && updater !is AtomicLongFieldUpdater<*> && updater !is AtomicReferenceFieldUpdater<*, *>) {
+            throw IllegalArgumentException("Provided object is not a recognized Atomic*FieldUpdater type.")
+        }
+        // Extract the private offset value and find the matching field.
+        try {
+            // Cannot use neither reflection not MethodHandles.Lookup, as they lead to a warning.
+            val tclassField = updater.javaClass.getDeclaredField("tclass")
+            val targetType = unsafe.getObject(updater, unsafe.objectFieldOffset(tclassField)) as Class<*>
+
+            val offsetField = updater.javaClass.getDeclaredField("offset")
+            val offset = unsafe.getLong(updater, unsafe.objectFieldOffset(offsetField))
+
+            for (field in targetType.declaredFields) {
+                try {
+                    if (Modifier.isNative(field.modifiers)) continue
+                    val fieldOffset = if (Modifier.isStatic(field.modifiers)) unsafe.staticFieldOffset(field)
+                    else unsafe.objectFieldOffset(field)
+                    if (fieldOffset == offset) return field.name
+                } catch (t: Throwable) {
+                    t.printStackTrace()
+                }
+            }
+        } catch (t: Throwable) {
+            t.printStackTrace()
+        }
+
+        return null // Field not found
+    }
+}

src/jvm/main/org/jetbrains/kotlinx/lincheck/CTestConfiguration.kt

@@ -13,7 +13,6 @@ import org.jetbrains.kotlinx.lincheck.CTestConfiguration.Companion.DEFAULT_TIMEO
 import org.jetbrains.kotlinx.lincheck.execution.*
 import org.jetbrains.kotlinx.lincheck.strategy.*
 import org.jetbrains.kotlinx.lincheck.strategy.managed.*
-import org.jetbrains.kotlinx.lincheck.strategy.managed.ManagedCTestConfiguration.Companion.DEFAULT_ELIMINATE_LOCAL_OBJECTS
 import org.jetbrains.kotlinx.lincheck.strategy.managed.modelchecking.*
 import org.jetbrains.kotlinx.lincheck.strategy.stress.*
 import org.jetbrains.kotlinx.lincheck.verifier.*
@@ -96,7 +95,6 @@ internal fun createFromTestClassAnnotations(testClass: Class<*>): List<CTestConf
                         testClass
                     ),
                     timeoutMs = DEFAULT_TIMEOUT_MS,
-                    eliminateLocalObjects = DEFAULT_ELIMINATE_LOCAL_OBJECTS,
                     customScenarios = emptyList()
                 )
             }

src/jvm/main/org/jetbrains/kotlinx/lincheck/EventTracker.kt

@@ -0,0 +1,49 @@
+/*
+ * Lincheck
+ *
+ * Copyright (C) 2019 - 2024 JetBrains s.r.o.
+ *
+ * This Source Code Form is subject to the terms of the
+ * Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed
+ * with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.jetbrains.kotlinx.lincheck
+
+import java.util.*
+
+/**
+ * Methods of this interface are called from the instrumented tested code during model-checking.
+ * See [Injections] for the documentation.
+ */
+internal interface EventTracker {
+    fun lock(monitor: Any, codeLocation: Int)
+    fun unlock(monitor: Any, codeLocation: Int)
+
+    fun park(codeLocation: Int)
+    fun unpark(thread: Thread, codeLocation: Int)
+
+    fun wait(monitor: Any, codeLocation: Int, withTimeout: Boolean)
+    fun notify(monitor: Any, codeLocation: Int, notifyAll: Boolean)
+
+    fun beforeReadField(obj: Any, className: String, fieldName: String, codeLocation: Int)
+    fun beforeReadFieldStatic(className: String, fieldName: String, codeLocation: Int)
+    fun beforeReadArrayElement(array: Any, index: Int, codeLocation: Int)
+    fun afterRead(value: Any?)
+
+    fun beforeWriteField(obj: Any, className: String, fieldName: String, value: Any?, codeLocation: Int)
+    fun beforeWriteFieldStatic(className: String, fieldName: String, value: Any?, codeLocation: Int)
+    fun beforeWriteArrayElement(array: Any, index: Int, value: Any?, codeLocation: Int)
+    fun afterWrite()
+
+    fun beforeMethodCall(owner: Any?, className: String, methodName: String, codeLocation: Int, params: Array<Any?>)
+    fun beforeAtomicMethodCall(owner: Any?, methodName: String, codeLocation: Int, params: Array<Any?>)
+    fun onMethodCallFinishedSuccessfully(result: Any?)
+    fun onMethodCallThrewException(t: Throwable)
+
+    fun getThreadLocalRandom(): Random
+    fun randomNextInt(): Int
+
+    fun onNewObjectCreation(obj: Any)
+    fun onWriteToObjectFieldOrArrayCell(obj: Any, fieldOrArrayCellValue: Any?)
+}