2017-09-08 02:51:28 -03:00
|
|
|
#ifndef Py_INTERNAL_PYSTATE_H
|
|
|
|
#define Py_INTERNAL_PYSTATE_H
|
|
|
|
#ifdef __cplusplus
|
|
|
|
extern "C" {
|
|
|
|
#endif
|
|
|
|
|
2019-04-17 18:02:26 -03:00
|
|
|
#ifndef Py_BUILD_CORE
|
|
|
|
# error "this header requires Py_BUILD_CORE define"
|
2018-10-31 21:51:40 -03:00
|
|
|
#endif
|
|
|
|
|
2022-04-06 08:58:07 -03:00
|
|
|
#include "pycore_runtime.h" /* PyRuntimeState */
|
2017-09-08 02:51:28 -03:00
|
|
|
|
|
|
|
|
2020-03-20 09:38:58 -03:00
|
|
|
/* Check if the current thread is the main thread.
|
|
|
|
Use _Py_IsMainInterpreter() to check if it's the main interpreter. */
|
|
|
|
static inline int
|
|
|
|
_Py_IsMainThread(void)
|
|
|
|
{
|
|
|
|
unsigned long thread = PyThread_get_thread_ident();
|
|
|
|
return (thread == _PyRuntime.main_thread);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2021-12-07 21:56:06 -04:00
|
|
|
static inline PyInterpreterState *
|
|
|
|
_PyInterpreterState_Main(void)
|
|
|
|
{
|
|
|
|
return _PyRuntime.interpreters.main;
|
|
|
|
}
|
|
|
|
|
2020-03-20 09:38:58 -03:00
|
|
|
static inline int
|
2021-02-19 08:33:31 -04:00
|
|
|
_Py_IsMainInterpreter(PyInterpreterState *interp)
|
2020-03-20 09:38:58 -03:00
|
|
|
{
|
2021-12-07 21:56:06 -04:00
|
|
|
return (interp == _PyInterpreterState_Main());
|
2020-03-20 09:38:58 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2021-09-27 13:00:32 -03:00
|
|
|
static inline const PyConfig *
|
|
|
|
_Py_GetMainConfig(void)
|
|
|
|
{
|
2021-12-07 21:56:06 -04:00
|
|
|
PyInterpreterState *interp = _PyInterpreterState_Main();
|
2021-09-27 13:00:32 -03:00
|
|
|
if (interp == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return _PyInterpreterState_GetConfig(interp);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2020-03-20 09:38:58 -03:00
|
|
|
/* Only handle signals on the main thread of the main interpreter. */
|
|
|
|
static inline int
|
2020-04-08 18:35:05 -03:00
|
|
|
_Py_ThreadCanHandleSignals(PyInterpreterState *interp)
|
2020-03-20 09:38:58 -03:00
|
|
|
{
|
2021-12-07 21:56:06 -04:00
|
|
|
return (_Py_IsMainThread() && _Py_IsMainInterpreter(interp));
|
2020-03-20 09:38:58 -03:00
|
|
|
}
|
2019-11-20 12:34:39 -04:00
|
|
|
|
2017-09-08 02:51:28 -03:00
|
|
|
|
2020-03-20 10:50:35 -03:00
|
|
|
/* Only execute pending calls on the main thread. */
|
|
|
|
static inline int
|
|
|
|
_Py_ThreadCanHandlePendingCalls(void)
|
|
|
|
{
|
|
|
|
return _Py_IsMainThread();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2018-10-31 21:51:40 -03:00
|
|
|
/* Variable and macro for in-line access to current thread
|
|
|
|
and interpreter state */
|
|
|
|
|
2020-05-05 14:56:48 -03:00
|
|
|
static inline PyThreadState*
|
|
|
|
_PyRuntimeState_GetThreadState(_PyRuntimeState *runtime)
|
|
|
|
{
|
2020-03-06 19:24:23 -04:00
|
|
|
return (PyThreadState*)_Py_atomic_load_relaxed(&runtime->gilstate.tstate_current);
|
|
|
|
}
|
2019-05-10 18:39:09 -03:00
|
|
|
|
2018-10-31 21:51:40 -03:00
|
|
|
/* Get the current Python thread state.
|
|
|
|
|
|
|
|
Efficient macro reading directly the 'gilstate.tstate_current' atomic
|
|
|
|
variable. The macro is unsafe: it does not check for error and it can
|
|
|
|
return NULL.
|
|
|
|
|
|
|
|
The caller must hold the GIL.
|
|
|
|
|
2021-10-13 09:09:13 -03:00
|
|
|
See also PyThreadState_Get() and _PyThreadState_UncheckedGet(). */
|
2020-05-05 14:56:48 -03:00
|
|
|
static inline PyThreadState*
|
|
|
|
_PyThreadState_GET(void)
|
|
|
|
{
|
2020-04-08 18:35:05 -03:00
|
|
|
return _PyRuntimeState_GetThreadState(&_PyRuntime);
|
|
|
|
}
|
2018-10-31 21:51:40 -03:00
|
|
|
|
2020-06-01 11:02:40 -03:00
|
|
|
PyAPI_FUNC(void) _Py_NO_RETURN _Py_FatalError_TstateNULL(const char *func);
|
|
|
|
|
|
|
|
static inline void
|
|
|
|
_Py_EnsureFuncTstateNotNULL(const char *func, PyThreadState *tstate)
|
|
|
|
{
|
|
|
|
if (tstate == NULL) {
|
|
|
|
_Py_FatalError_TstateNULL(func);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Call Py_FatalError() if tstate is NULL
|
|
|
|
#define _Py_EnsureTstateNotNULL(tstate) \
|
2022-06-20 11:04:52 -03:00
|
|
|
_Py_EnsureFuncTstateNotNULL(__func__, (tstate))
|
2020-06-01 11:02:40 -03:00
|
|
|
|
|
|
|
|
2018-10-31 21:51:40 -03:00
|
|
|
/* Get the current interpreter state.
|
|
|
|
|
|
|
|
The macro is unsafe: it does not check for error and it can return NULL.
|
|
|
|
|
|
|
|
The caller must hold the GIL.
|
|
|
|
|
|
|
|
See also _PyInterpreterState_Get()
|
|
|
|
and _PyGILState_GetInterpreterStateUnsafe(). */
|
2020-04-14 10:14:01 -03:00
|
|
|
static inline PyInterpreterState* _PyInterpreterState_GET(void) {
|
2020-04-08 18:35:05 -03:00
|
|
|
PyThreadState *tstate = _PyThreadState_GET();
|
2020-06-01 11:02:40 -03:00
|
|
|
#ifdef Py_DEBUG
|
|
|
|
_Py_EnsureTstateNotNULL(tstate);
|
|
|
|
#endif
|
2020-04-08 18:35:05 -03:00
|
|
|
return tstate->interp;
|
|
|
|
}
|
2018-10-31 21:51:40 -03:00
|
|
|
|
|
|
|
|
2021-10-15 11:06:30 -03:00
|
|
|
// PyThreadState functions
|
2017-09-08 02:51:28 -03:00
|
|
|
|
2021-12-07 20:26:29 -04:00
|
|
|
PyAPI_FUNC(void) _PyThreadState_SetCurrent(PyThreadState *tstate);
|
|
|
|
// We keep this around exclusively for stable ABI compatibility.
|
2019-06-03 22:15:09 -03:00
|
|
|
PyAPI_FUNC(void) _PyThreadState_Init(
|
|
|
|
PyThreadState *tstate);
|
|
|
|
PyAPI_FUNC(void) _PyThreadState_DeleteExcept(
|
|
|
|
_PyRuntimeState *runtime,
|
|
|
|
PyThreadState *tstate);
|
2019-05-10 18:39:09 -03:00
|
|
|
|
2021-10-15 11:06:30 -03:00
|
|
|
|
|
|
|
static inline void
|
2022-03-15 14:06:21 -03:00
|
|
|
_PyThreadState_UpdateTracingState(PyThreadState *tstate)
|
2021-10-15 11:06:30 -03:00
|
|
|
{
|
2022-07-28 06:17:22 -03:00
|
|
|
bool use_tracing =
|
|
|
|
(tstate->tracing == 0) &&
|
|
|
|
(tstate->c_tracefunc != NULL || tstate->c_profilefunc != NULL);
|
2021-10-15 11:06:30 -03:00
|
|
|
tstate->cframe->use_tracing = (use_tracing ? 255 : 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* Other */
|
|
|
|
|
2019-05-10 18:39:09 -03:00
|
|
|
PyAPI_FUNC(PyThreadState *) _PyThreadState_Swap(
|
|
|
|
struct _gilstate_runtime_state *gilstate,
|
|
|
|
PyThreadState *newts);
|
2019-04-24 11:47:40 -03:00
|
|
|
|
2019-05-27 11:39:22 -03:00
|
|
|
PyAPI_FUNC(PyStatus) _PyInterpreterState_Enable(_PyRuntimeState *runtime);
|
2019-04-24 12:14:33 -03:00
|
|
|
|
2020-06-02 10:51:37 -03:00
|
|
|
#ifdef HAVE_FORK
|
|
|
|
extern PyStatus _PyInterpreterState_DeleteExceptMain(_PyRuntimeState *runtime);
|
|
|
|
extern PyStatus _PyGILState_Reinit(_PyRuntimeState *runtime);
|
|
|
|
extern void _PySignal_AfterFork(void);
|
|
|
|
#endif
|
2017-09-08 02:51:28 -03:00
|
|
|
|
2019-11-22 13:52:27 -04:00
|
|
|
|
|
|
|
PyAPI_FUNC(int) _PyState_AddModule(
|
|
|
|
PyThreadState *tstate,
|
|
|
|
PyObject* module,
|
2022-03-03 18:08:07 -04:00
|
|
|
PyModuleDef* def);
|
2019-11-22 13:52:27 -04:00
|
|
|
|
2020-06-03 09:39:59 -03:00
|
|
|
|
|
|
|
PyAPI_FUNC(int) _PyOS_InterruptOccurred(PyThreadState *tstate);
|
|
|
|
|
2017-09-08 02:51:28 -03:00
|
|
|
#ifdef __cplusplus
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
#endif /* !Py_INTERNAL_PYSTATE_H */
|