From 09f738ced164b3b347cd4865c11a4e7c45de25e3 Mon Sep 17 00:00:00 2001

From: =?UTF-8?q?S=C3=BCmer=20Cip?= <sumer.cip@platform.sh>

Date: Thu, 4 Aug 2022 14:04:47 +0300

Subject: [PATCH 04/11] 6 failures in TCs

---

 yappi/_yappi.c | 57 ++++++++++++++++++++++++++------------------------

 1 file changed, 30 insertions(+), 27 deletions(-)

diff --git a/yappi/_yappi.c b/yappi/_yappi.c

index 00fdbd4..d2f0af4 100644

--- a/yappi/_yappi.c

+++ b/yappi/_yappi.c

@@ -1292,17 +1292,37 @@ _resume_greenlet_ctx(_ctx *ctx)

     ydprintf("resuming context: %ld, shift: %lld", ctx->id, shift);

 }

+static void 

+_eval_setprofile(PyThreadState *ts)

+{

+#if PY_VERSION_HEX < 0x030a00b1

+    ts->use_tracing = 1;

+#else

+    ts->cframe->use_tracing = 1;

+#endif

+    ts->c_profilefunc = _yapp_callback;

+

+    //_update_tracing_state()

+

+//  todo: do this only for 3.11

+    ts->cframe->use_tracing = 255;

+}

+

+static void

+_eval_unsetprofile(PyThreadState *ts)

+{

+#if PY_VERSION_HEX < 0x030a00b1

+    ts->use_tracing = 0;

+#else

+    ts->cframe->use_tracing = 0;

+#endif

+    ts->c_profilefunc = NULL;

+}

+

 static _ctx *

 _bootstrap_thread(PyThreadState *ts)

 {

-// #if PY_VERSION_HEX < 0x030a00b1

-//     ts->use_tracing = 1;

-// #else

-//     ts->cframe->use_tracing = 1;

-// #endif

-//     ts->c_profilefunc = _yapp_callback;

-    PyEval_SetProfile(_yapp_callback, NULL);

-

+    _eval_setprofile(ts);

     return NULL;

 }

@@ -1331,16 +1351,7 @@ _profile_thread(PyThreadState *ts)

     } else {

         ctx = (_ctx *)it->val;

     }

-    

-// #if PY_VERSION_HEX < 0x030a00b1

-//     ts->use_tracing = 1;

-// #else

-//     ts->cframe->use_tracing = 1;

-// #endif

-//     ts->c_profilefunc = _yapp_callback;

-    // TODO: How to do this for all threads?

-    PyEval_SetProfile(_yapp_callback, NULL);

-

+    _eval_setprofile(ts);

     ctx->id = ctx_id;

     ctx->tid = ts->thread_id;

     ctx->ts_ptr = ts;

@@ -1356,15 +1367,7 @@ _profile_thread(PyThreadState *ts)

 static _ctx*

 _unprofile_thread(PyThreadState *ts)

 {

-// #if PY_VERSION_HEX < 0x030a00b1

-//     ts->use_tracing = 0;

-// #else

-//     ts->cframe->use_tracing = 0;

-// #endif

-//     ts->c_profilefunc = NULL;

-

-    // TODO: How to do this for all threads?

-    PyEval_SetProfile(NULL, NULL);

+    _eval_unsetprofile(ts);

     return NULL; //dummy return for enum_threads() func. prototype

 }

-- 

2.34.1