gh-117657: Fix some simple races in instrumentation.c (GH-120118)

* stop the world when setting local events
python · Jun 13, 2024 · b1b61dc · b1b61dc
1 parent eebae2c
commit b1b61dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Python/instrumentation.c b/Python/instrumentation.c
@@ -1977,7 +1977,7 @@ _PyMonitoring_SetLocalEvents(PyCodeObject *code, int tool_id, _PyMonitoringEvent
     }
 
     int res;
-    LOCK_CODE(code);
+    _PyEval_StopTheWorld(interp);
     if (allocate_instrumentation_data(code)) {
         res = -1;
         goto done;
@@ -1994,7 +1994,7 @@ _PyMonitoring_SetLocalEvents(PyCodeObject *code, int tool_id, _PyMonitoringEvent
     res = force_instrument_lock_held(code, interp);
 
 done:
-    UNLOCK_CODE();
+    _PyEval_StartTheWorld(interp);
     return res;
 }