intercept isInterrupted method.

cmu-pasta · Jun 21, 2024 · ec0a924 · ec0a924
1 parent c362f50
commit ec0a924
Show file tree

Hide file tree

Showing 6 changed files with 22 additions and 1 deletion.
diff --git a/core/src/main/kotlin/cmu/pasta/fray/core/GlobalContext.kt b/core/src/main/kotlin/cmu/pasta/fray/core/GlobalContext.kt
@@ -234,6 +234,10 @@ object GlobalContext {
     registeredThreads[t.id]?.block()
   }
 
+  fun threadIsInterrupted(t: Thread, result: Boolean): Boolean {
+    return result || registeredThreads[t.id]!!.interruptSignaled
+  }
+
   fun threadGetState(t: Thread, state: Thread.State): Thread.State {
     if (state == Thread.State.WAITING ||
         state == Thread.State.TIMED_WAITING ||

diff --git a/core/src/main/kotlin/cmu/pasta/fray/core/RuntimeDelegate.kt b/core/src/main/kotlin/cmu/pasta/fray/core/RuntimeDelegate.kt
@@ -683,4 +683,11 @@ class RuntimeDelegate : Delegate() {
       return true
     }
   }
+
+  override fun onThreadIsInterrupted(result: Boolean, t: Thread): Boolean {
+    if (checkEntered()) return result
+    val isInterrupted = GlobalContext.threadIsInterrupted(t, result)
+    entered.set(false)
+    return isInterrupted
+  }
 }
diff --git a/examples/src/main/kotlin/JUnitRunner.kt b/examples/src/main/kotlin/JUnitRunner.kt
@@ -13,7 +13,7 @@ fun main(args: Array<String>) {
 
   val result = JUnitCore().run(request)
   if (!result.wasSuccessful()) {
-    val failureReport = result.failures.joinToString { "\n =========== \n" }
+    val failureReport = result.failures.joinToString("\n =========== \n")
     throw RuntimeException(failureReport)
   }
 }
diff --git a/...rumentation/src/main/kotlin/cmu/pasta/fray/instrumentation/visitors/ThreadInstrumenter.kt b/...rumentation/src/main/kotlin/cmu/pasta/fray/instrumentation/visitors/ThreadInstrumenter.kt
@@ -30,6 +30,10 @@ class ThreadInstrumenter(cv: ClassVisitor) : ClassVisitorBase(cv, Thread::class.
       return MethodExitVisitor(
           mv, Runtime::onThreadGetAndClearInterrupt, access, name, descriptor, true, false, false)
     }
+    if (name == "isInterrupted") {
+      return MethodExitVisitor(
+          mv, Runtime::onThreadIsInterrupted, access, name, descriptor, true, false, false)
+    }
     if (name == "clearInterrupt") {
       return MethodExitVisitor(
           mv, Runtime::onThreadClearInterrupt, access, name, descriptor, true, false, false)

diff --git a/runtime/src/main/java/cmu/pasta/fray/runtime/Delegate.java b/runtime/src/main/java/cmu/pasta/fray/runtime/Delegate.java
@@ -242,5 +242,8 @@ public void onConditionAwaitUninterruptibly(Condition object) {
     public void onConditionAwaitUninterruptiblyDone(Condition object) {
     }
 
+    public boolean onThreadIsInterrupted(boolean result, Thread t) {
+        return result;
+    }
 }
 
diff --git a/runtime/src/main/java/cmu/pasta/fray/runtime/Runtime.java b/runtime/src/main/java/cmu/pasta/fray/runtime/Runtime.java
@@ -330,4 +330,7 @@ public static void onConditionAwaitUninterruptiblyDone(Condition object) {
         DELEGATE.onConditionAwaitUninterruptiblyDone(object);
     }
 
+    public static boolean onThreadIsInterrupted(boolean result, Thread t) {
+        return DELEGATE.onThreadIsInterrupted(result, t);
+    }
 }