diff --git a/src/runtime/lock_wasip1.go b/src/runtime/lock_wasip1.go
index 55153c3a05f542..237d1f0431ceb5 100644
--- a/src/runtime/lock_wasip1.go
+++ b/src/runtime/lock_wasip1.go
@@ -105,8 +105,16 @@ func notetsleepg(n *note, ns int64) bool {
 	}
 }
 
+var onIdle = func() bool {
+	return false
+}
+
+func wasiOnIdle(callback func() bool) {
+	onIdle = callback
+}
+
 func beforeIdle(int64, int64) (*g, bool) {
-	return nil, false
+	return nil, onIdle()
 }
 
 func checkTimeouts() {}