summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJason Madden <jamadden@gmail.com>2020-11-18 09:42:56 -0600
committerJason Madden <jamadden@gmail.com>2020-11-18 09:42:56 -0600
commitfafa99c69564d966e75530dda7f8e7c83db53924 (patch)
treed1a54bdbcb2376c30abfbe84eca673ec1235ade1
parenta9a0c0749abe03dc8f34d4e31d278050f4080c4c (diff)
downloadgreenlet-issue206.tar.gz
Use clang-format and clang-tidy to PEP7 the source.issue206
Fixes #206
-rw-r--r--.clang-format29
-rw-r--r--MANIFEST.in1
-rw-r--r--src/greenlet/greenlet.c2898
-rw-r--r--src/greenlet/greenlet.h117
-rw-r--r--src/greenlet/tests/_test_extension.c306
-rw-r--r--src/greenlet/tests/_test_extension_cpp.cpp141
6 files changed, 1796 insertions, 1696 deletions
diff --git a/.clang-format b/.clang-format
new file mode 100644
index 0000000..807801b
--- /dev/null
+++ b/.clang-format
@@ -0,0 +1,29 @@
+# A clang-format style that approximates Python's PEP 7 -*- mode: yaml; -*-
+# Initially based on
+# https://gist.github.com/pganssle/0e3a5f828b4d07d79447f6ced8e7e4db
+BasedOnStyle: Google
+Language: Cpp
+
+AlignAfterOpenBracket: Align
+AllowShortBlocksOnASingleLine: false
+AllowShortIfStatementsOnASingleLine: Never
+AllowShortLoopsOnASingleLine: false
+AlwaysBreakAfterReturnType: All
+
+BinPackArguments: false
+BreakBeforeBraces: Stroustrup
+BreakBeforeTernaryOperators: true
+
+ColumnLimit: 79
+DerivePointerAlignment: false
+
+IndentWidth: 4
+IndentPPDirectives: AfterHash
+
+
+PointerAlignment: Left
+ReflowComments: true
+SpaceBeforeParens: ControlStatements
+SpacesInParentheses: false
+TabWidth: 4
+UseTab: Never
diff --git a/MANIFEST.in b/MANIFEST.in
index 6dd962c..a811098 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -27,6 +27,7 @@ include *.rst
include *.cfg
include *.py
include *.ini
+include .clang-format
recursive-include appveyor *.cmd
recursive-include appveyor *.ps1
diff --git a/src/greenlet/greenlet.c b/src/greenlet/greenlet.c
index 11c15bb..60580f6 100644
--- a/src/greenlet/greenlet.c
+++ b/src/greenlet/greenlet.c
@@ -1,17 +1,23 @@
-/* vim:set noet ts=8 sw=8 : */
-
+/* -*- indent-tabs-mode: nil; tab-width: 4; -*- */
+/* Format with:
+ * clang-format -i --style=file src/greenlet/greenlet.c
+ *
+ *
+ * Fix missing braces with:
+ * clang-tidy src/greenlet/greenlet.c -fix -checks="readability-braces-around-statements"
+*/
#define GREENLET_MODULE
#include "greenlet.h"
+
#include "structmember.h"
#ifdef __clang__
-#pragma clang diagnostic push
-#pragma clang diagnostic ignored "-Wunused-parameter"
-#pragma clang diagnostic ignored "-Wmissing-field-initializers"
+# pragma clang diagnostic push
+# pragma clang diagnostic ignored "-Wunused-parameter"
+# pragma clang diagnostic ignored "-Wmissing-field-initializers"
#endif
-
/***********************************************************
A PyGreenlet is a range of C stack addresses that must be
@@ -85,25 +91,26 @@ The running greenlet's stack_start is undefined but not NULL.
extern PyTypeObject PyGreenlet_Type;
#if PY_VERSION_HEX >= 0x030700A3
-# define GREENLET_PY37 1
+# define GREENLET_PY37 1
#else
-# define GREENLET_PY37 0
+# define GREENLET_PY37 0
#endif
#ifndef Py_SET_REFCNT
- /* Py_REFCNT and Py_SIZE macros are converted to functions
- https://bugs.python.org/issue39573 */
- #define Py_SET_REFCNT(obj, refcnt) Py_REFCNT(obj) = (refcnt)
+/* Py_REFCNT and Py_SIZE macros are converted to functions
+https://bugs.python.org/issue39573 */
+# define Py_SET_REFCNT(obj, refcnt) Py_REFCNT(obj) = (refcnt)
#endif
#ifndef _Py_DEC_REFTOTAL
- /* _Py_DEC_REFTOTAL macro has been removed from Python 3.9 by:
- https://github.com/python/cpython/commit/49932fec62c616ec88da52642339d83ae719e924 */
-# ifdef Py_REF_DEBUG
-# define _Py_DEC_REFTOTAL _Py_RefTotal--
-# else
-# define _Py_DEC_REFTOTAL
-# endif
+/* _Py_DEC_REFTOTAL macro has been removed from Python 3.9 by:
+ https://github.com/python/cpython/commit/49932fec62c616ec88da52642339d83ae719e924
+*/
+# ifdef Py_REF_DEBUG
+# define _Py_DEC_REFTOTAL _Py_RefTotal--
+# else
+# define _Py_DEC_REFTOTAL
+# endif
#endif
/* Weak reference to the switching-to greenlet during the slp switch */
@@ -119,8 +126,9 @@ static PyObject* volatile ts_passaround_kwargs = NULL;
/***********************************************************/
/* Thread-aware routines, switching global variables when needed */
-#define STATE_OK (ts_current->run_info == PyThreadState_GET()->dict \
- || !green_updatecurrent())
+#define STATE_OK \
+ (ts_current->run_info == PyThreadState_GET()->dict || \
+ !green_updatecurrent())
static PyObject* ts_curkey;
static PyObject* ts_delkey;
@@ -139,130 +147,137 @@ static PyObject* ts_empty_dict;
#define GREENLET_tp_clear green_clear
#define GREENLET_tp_is_gc green_is_gc
-static void green_clear_exc(PyGreenlet* g)
+static void
+green_clear_exc(PyGreenlet* g)
{
#if GREENLET_PY37
- g->exc_info = NULL;
- g->exc_state.exc_type = NULL;
- g->exc_state.exc_value = NULL;
- g->exc_state.exc_traceback = NULL;
- g->exc_state.previous_item = NULL;
+ g->exc_info = NULL;
+ g->exc_state.exc_type = NULL;
+ g->exc_state.exc_value = NULL;
+ g->exc_state.exc_traceback = NULL;
+ g->exc_state.previous_item = NULL;
#else
- g->exc_type = NULL;
- g->exc_value = NULL;
- g->exc_traceback = NULL;
+ g->exc_type = NULL;
+ g->exc_value = NULL;
+ g->exc_traceback = NULL;
#endif
}
-static PyGreenlet* green_create_main(void)
+static PyGreenlet*
+green_create_main(void)
{
- PyGreenlet* gmain;
- PyObject* dict = PyThreadState_GetDict();
- if (dict == NULL) {
- if (!PyErr_Occurred())
- PyErr_NoMemory();
- return NULL;
- }
-
- /* create the main greenlet for this thread */
- gmain = (PyGreenlet*) PyType_GenericAlloc(&PyGreenlet_Type, 0);
- if (gmain == NULL)
- return NULL;
- gmain->stack_start = (char*) 1;
- gmain->stack_stop = (char*) -1;
- gmain->run_info = dict;
- Py_INCREF(dict);
- return gmain;
+ PyGreenlet* gmain;
+ PyObject* dict = PyThreadState_GetDict();
+ if (dict == NULL) {
+ if (!PyErr_Occurred()) {
+ PyErr_NoMemory();
+ }
+ return NULL;
+ }
+
+ /* create the main greenlet for this thread */
+ gmain = (PyGreenlet*)PyType_GenericAlloc(&PyGreenlet_Type, 0);
+ if (gmain == NULL) {
+ return NULL;
+ }
+ gmain->stack_start = (char*)1;
+ gmain->stack_stop = (char*)-1;
+ gmain->run_info = dict;
+ Py_INCREF(dict);
+ return gmain;
}
-static int green_updatecurrent(void)
+static int
+green_updatecurrent(void)
{
- PyObject *exc, *val, *tb;
- PyThreadState* tstate;
- PyGreenlet* current;
- PyGreenlet* previous;
- PyObject* deleteme;
+ PyObject *exc, *val, *tb;
+ PyThreadState* tstate;
+ PyGreenlet* current;
+ PyGreenlet* previous;
+ PyObject* deleteme;
green_updatecurrent_restart:
- /* save current exception */
- PyErr_Fetch(&exc, &val, &tb);
-
- /* get ts_current from the active tstate */
- tstate = PyThreadState_GET();
- if (tstate->dict && (current =
- (PyGreenlet*) PyDict_GetItem(tstate->dict, ts_curkey))) {
- /* found -- remove it, to avoid keeping a ref */
- Py_INCREF(current);
- PyDict_DelItem(tstate->dict, ts_curkey);
- }
- else {
- /* first time we see this tstate */
- current = green_create_main();
- if (current == NULL) {
- Py_XDECREF(exc);
- Py_XDECREF(val);
- Py_XDECREF(tb);
- return -1;
- }
- }
- assert(current->run_info == tstate->dict);
+ /* save current exception */
+ PyErr_Fetch(&exc, &val, &tb);
+
+ /* get ts_current from the active tstate */
+ tstate = PyThreadState_GET();
+ if (tstate->dict &&
+ (current = (PyGreenlet*)PyDict_GetItem(tstate->dict, ts_curkey))) {
+ /* found -- remove it, to avoid keeping a ref */
+ Py_INCREF(current);
+ PyDict_DelItem(tstate->dict, ts_curkey);
+ }
+ else {
+ /* first time we see this tstate */
+ current = green_create_main();
+ if (current == NULL) {
+ Py_XDECREF(exc);
+ Py_XDECREF(val);
+ Py_XDECREF(tb);
+ return -1;
+ }
+ }
+ assert(current->run_info == tstate->dict);
green_updatecurrent_retry:
- /* update ts_current as soon as possible, in case of nested switches */
- Py_INCREF(current);
- previous = ts_current;
- ts_current = current;
-
- /* save ts_current as the current greenlet of its own thread */
- if (PyDict_SetItem(previous->run_info, ts_curkey, (PyObject*) previous)) {
- Py_DECREF(previous);
- Py_DECREF(current);
- Py_XDECREF(exc);
- Py_XDECREF(val);
- Py_XDECREF(tb);
- return -1;
- }
- Py_DECREF(previous);
-
- /* green_dealloc() cannot delete greenlets from other threads, so
- it stores them in the thread dict; delete them now. */
- deleteme = PyDict_GetItem(tstate->dict, ts_delkey);
- if (deleteme != NULL) {
- PyList_SetSlice(deleteme, 0, INT_MAX, NULL);
- }
-
- if (ts_current != current) {
- /* some Python code executed above and there was a thread switch,
- * so ts_current points to some other thread again. We need to
- * delete ts_curkey (it's likely there) and retry. */
- PyDict_DelItem(tstate->dict, ts_curkey);
- goto green_updatecurrent_retry;
- }
-
- /* release an extra reference */
- Py_DECREF(current);
-
- /* restore current exception */
- PyErr_Restore(exc, val, tb);
-
- /* thread switch could happen during PyErr_Restore, in that
- case there's nothing to do except restart from scratch. */
- if (ts_current->run_info != tstate->dict)
- goto green_updatecurrent_restart;
-
- return 0;
+ /* update ts_current as soon as possible, in case of nested switches */
+ Py_INCREF(current);
+ previous = ts_current;
+ ts_current = current;
+
+ /* save ts_current as the current greenlet of its own thread */
+ if (PyDict_SetItem(previous->run_info, ts_curkey, (PyObject*)previous)) {
+ Py_DECREF(previous);
+ Py_DECREF(current);
+ Py_XDECREF(exc);
+ Py_XDECREF(val);
+ Py_XDECREF(tb);
+ return -1;
+ }
+ Py_DECREF(previous);
+
+ /* green_dealloc() cannot delete greenlets from other threads, so
+ it stores them in the thread dict; delete them now. */
+ deleteme = PyDict_GetItem(tstate->dict, ts_delkey);
+ if (deleteme != NULL) {
+ PyList_SetSlice(deleteme, 0, INT_MAX, NULL);
+ }
+
+ if (ts_current != current) {
+ /* some Python code executed above and there was a thread switch,
+ * so ts_current points to some other thread again. We need to
+ * delete ts_curkey (it's likely there) and retry. */
+ PyDict_DelItem(tstate->dict, ts_curkey);
+ goto green_updatecurrent_retry;
+ }
+
+ /* release an extra reference */
+ Py_DECREF(current);
+
+ /* restore current exception */
+ PyErr_Restore(exc, val, tb);
+
+ /* thread switch could happen during PyErr_Restore, in that
+ case there's nothing to do except restart from scratch. */
+ if (ts_current->run_info != tstate->dict) {
+ goto green_updatecurrent_restart;
+ }
+
+ return 0;
}
-static PyObject* green_statedict(PyGreenlet* g)
+static PyObject*
+green_statedict(PyGreenlet* g)
{
- while (!PyGreenlet_STARTED(g)) {
- g = g->parent;
- if (g == NULL) {
- /* garbage collected greenlet in chain */
- return NULL;
- }
- }
- return g->run_info;
+ while (!PyGreenlet_STARTED(g)) {
+ g = g->parent;
+ if (g == NULL) {
+ /* garbage collected greenlet in chain */
+ return NULL;
+ }
+ }
+ return g->run_info;
}
/***********************************************************/
@@ -277,36 +292,40 @@ static PyObject* green_statedict(PyGreenlet* g)
own stack frame, leading to incomplete stack save/restore
*/
-#if defined(__GNUC__) && (__GNUC__ > 3 || (__GNUC__ == 3 && __GNUC_MINOR__ >= 4))
-#define GREENLET_NOINLINE_SUPPORTED
-#define GREENLET_NOINLINE(name) __attribute__((noinline)) name
+#if defined(__GNUC__) && \
+ (__GNUC__ > 3 || (__GNUC__ == 3 && __GNUC_MINOR__ >= 4))
+# define GREENLET_NOINLINE_SUPPORTED
+# define GREENLET_NOINLINE(name) __attribute__((noinline)) name
#elif defined(_MSC_VER) && (_MSC_VER >= 1300)
-#define GREENLET_NOINLINE_SUPPORTED
-#define GREENLET_NOINLINE(name) __declspec(noinline) name
+# define GREENLET_NOINLINE_SUPPORTED
+# define GREENLET_NOINLINE(name) __declspec(noinline) name
#endif
#ifdef GREENLET_NOINLINE_SUPPORTED
/* add forward declarations */
static void GREENLET_NOINLINE(slp_restore_state)(void);
static int GREENLET_NOINLINE(slp_save_state)(char*);
-#if !(defined(MS_WIN64) && defined(_M_X64))
+# if !(defined(MS_WIN64) && defined(_M_X64))
static int GREENLET_NOINLINE(slp_switch)(void);
-#endif
+# endif
static int GREENLET_NOINLINE(g_initialstub)(void*);
-#define GREENLET_NOINLINE_INIT() do {} while(0)
+# define GREENLET_NOINLINE_INIT() \
+ do { \
+ } while (0)
#else
/* force compiler to call functions via pointers */
static void (*slp_restore_state)(void);
static int (*slp_save_state)(char*);
static int (*slp_switch)(void);
static int (*g_initialstub)(void*);
-#define GREENLET_NOINLINE(name) cannot_inline_ ## name
-#define GREENLET_NOINLINE_INIT() do { \
- slp_restore_state = GREENLET_NOINLINE(slp_restore_state); \
- slp_save_state = GREENLET_NOINLINE(slp_save_state); \
- slp_switch = GREENLET_NOINLINE(slp_switch); \
- g_initialstub = GREENLET_NOINLINE(g_initialstub); \
-} while(0)
+# define GREENLET_NOINLINE(name) cannot_inline_##name
+# define GREENLET_NOINLINE_INIT() \
+ do { \
+ slp_restore_state = GREENLET_NOINLINE(slp_restore_state); \
+ slp_save_state = GREENLET_NOINLINE(slp_save_state); \
+ slp_switch = GREENLET_NOINLINE(slp_switch); \
+ g_initialstub = GREENLET_NOINLINE(g_initialstub); \
+ } while (0)
#endif
/*
@@ -314,15 +333,15 @@ static int (*g_initialstub)(void*);
* specific code, in order to simplify maintenance.
*/
-#define SLP_SAVE_STATE(stackref, stsizediff) \
- stackref += STACK_MAGIC; \
- if (slp_save_state((char*)stackref)) return -1; \
- if (!PyGreenlet_ACTIVE(ts_target)) return 1; \
- stsizediff = ts_target->stack_start - (char*)stackref
-
-#define SLP_RESTORE_STATE() \
- slp_restore_state()
+#define SLP_SAVE_STATE(stackref, stsizediff) \
+ stackref += STACK_MAGIC; \
+ if (slp_save_state((char*)stackref)) \
+ return -1; \
+ if (!PyGreenlet_ACTIVE(ts_target)) \
+ return 1; \
+ stsizediff = ts_target->stack_start - (char*)stackref
+#define SLP_RESTORE_STATE() slp_restore_state()
#define SLP_EVAL
#define slp_switch GREENLET_NOINLINE(slp_switch)
@@ -330,7 +349,8 @@ static int (*g_initialstub)(void*);
#undef slp_switch
#ifndef STACK_MAGIC
-#error "greenlet needs to be ported to this platform, or taught how to detect your compiler properly."
+# error \
+ "greenlet needs to be ported to this platform, or taught how to detect your compiler properly."
#endif /* !STACK_MAGIC */
#ifdef EXTERNAL_ASM
@@ -339,1531 +359,1567 @@ static int (*g_initialstub)(void*);
* EXTERNAL_ASM define so that this is included.
*/
-intptr_t slp_save_state_asm(intptr_t *ref) {
- intptr_t diff;
- SLP_SAVE_STATE(ref, diff);
- return diff;
+intptr_t
+slp_save_state_asm(intptr_t* ref)
+{
+ intptr_t diff;
+ SLP_SAVE_STATE(ref, diff);
+ return diff;
}
-void slp_restore_state_asm(void) {
- SLP_RESTORE_STATE();
+void
+slp_restore_state_asm(void)
+{
+ SLP_RESTORE_STATE();
}
-extern int slp_switch(void);
+extern int
+slp_switch(void);
#endif
/***********************************************************/
-static int g_save(PyGreenlet* g, char* stop)
+static int
+g_save(PyGreenlet* g, char* stop)
{
- /* Save more of g's stack into the heap -- at least up to 'stop'
-
- g->stack_stop |________|
- | |
- | __ stop . . . . .
- | | ==> . .
- |________| _______
- | | | |
- | | | |
- g->stack_start | | |_______| g->stack_copy
-
- */
- intptr_t sz1 = g->stack_saved;
- intptr_t sz2 = stop - g->stack_start;
- assert(g->stack_start != NULL);
- if (sz2 > sz1) {
- char* c = (char*)PyMem_Realloc(g->stack_copy, sz2);
- if (!c) {
- PyErr_NoMemory();
- return -1;
- }
- memcpy(c+sz1, g->stack_start+sz1, sz2-sz1);
- g->stack_copy = c;
- g->stack_saved = sz2;
- }
- return 0;
+ /* Save more of g's stack into the heap -- at least up to 'stop'
+
+ g->stack_stop |________|
+ | |
+ | __ stop . . . . .
+ | | ==> . .
+ |________| _______
+ | | | |
+ | | | |
+ g->stack_start | | |_______| g->stack_copy
+
+ */
+ intptr_t sz1 = g->stack_saved;
+ intptr_t sz2 = stop - g->stack_start;
+ assert(g->stack_start != NULL);
+ if (sz2 > sz1) {
+ char* c = (char*)PyMem_Realloc(g->stack_copy, sz2);
+ if (!c) {
+ PyErr_NoMemory();
+ return -1;
+ }
+ memcpy(c + sz1, g->stack_start + sz1, sz2 - sz1);
+ g->stack_copy = c;
+ g->stack_saved = sz2;
+ }
+ return 0;
}
static void GREENLET_NOINLINE(slp_restore_state)(void)
{
- PyGreenlet* g = ts_target;
- PyGreenlet* owner = ts_current;
+ PyGreenlet* g = ts_target;
+ PyGreenlet* owner = ts_current;
#ifdef SLP_BEFORE_RESTORE_STATE
- SLP_BEFORE_RESTORE_STATE();
+ SLP_BEFORE_RESTORE_STATE();
#endif
- /* Restore the heap copy back into the C stack */
- if (g->stack_saved != 0) {
- memcpy(g->stack_start, g->stack_copy, g->stack_saved);
- PyMem_Free(g->stack_copy);
- g->stack_copy = NULL;
- g->stack_saved = 0;
- }
- if (owner->stack_start == NULL)
- owner = owner->stack_prev; /* greenlet is dying, skip it */
- while (owner && owner->stack_stop <= g->stack_stop)
- owner = owner->stack_prev; /* find greenlet with more stack */
- g->stack_prev = owner;
+ /* Restore the heap copy back into the C stack */
+ if (g->stack_saved != 0) {
+ memcpy(g->stack_start, g->stack_copy, g->stack_saved);
+ PyMem_Free(g->stack_copy);
+ g->stack_copy = NULL;
+ g->stack_saved = 0;
+ }
+ if (owner->stack_start == NULL) {
+ owner = owner->stack_prev; /* greenlet is dying, skip it */
+ }
+ while (owner && owner->stack_stop <= g->stack_stop) {
+ owner = owner->stack_prev; /* find greenlet with more stack */
+ }
+ g->stack_prev = owner;
}
static int GREENLET_NOINLINE(slp_save_state)(char* stackref)
{
- /* must free all the C stack up to target_stop */
- char* target_stop = ts_target->stack_stop;
- PyGreenlet* owner = ts_current;
- assert(owner->stack_saved == 0);
- if (owner->stack_start == NULL)
- owner = owner->stack_prev; /* not saved if dying */
- else
- owner->stack_start = stackref;
+ /* must free all the C stack up to target_stop */
+ char* target_stop = ts_target->stack_stop;
+ PyGreenlet* owner = ts_current;
+ assert(owner->stack_saved == 0);
+ if (owner->stack_start == NULL) {
+ owner = owner->stack_prev; /* not saved if dying */
+ }
+ else {
+ owner->stack_start = stackref;
+ }
#ifdef SLP_BEFORE_SAVE_STATE
- SLP_BEFORE_SAVE_STATE();
+ SLP_BEFORE_SAVE_STATE();
#endif
- while (owner->stack_stop < target_stop) {
- /* ts_current is entierely within the area to free */
- if (g_save(owner, owner->stack_stop))
- return -1; /* XXX */
- owner = owner->stack_prev;
- }
- if (owner != ts_target) {
- if (g_save(owner, target_stop))
- return -1; /* XXX */
- }
- return 0;
+ while (owner->stack_stop < target_stop) {
+ /* ts_current is entierely within the area to free */
+ if (g_save(owner, owner->stack_stop)) {
+ return -1; /* XXX */
+ }
+ owner = owner->stack_prev;
+ }
+ if (owner != ts_target) {
+ if (g_save(owner, target_stop)) {
+ return -1; /* XXX */
+ }
+ }
+ return 0;
}
-static int g_switchstack(void)
+static int
+g_switchstack(void)
{
- /* Perform a stack switch according to some global variables
- that must be set before:
- - ts_current: current greenlet (holds a reference)
- - ts_target: greenlet to switch to (weak reference)
- - ts_passaround_args: NULL if PyErr_Occurred(),
- else a tuple of args sent to ts_target (holds a reference)
- - ts_passaround_kwargs: switch kwargs (holds a reference)
- On return results are passed via global variables as well:
- - ts_origin: originating greenlet (holds a reference)
- - ts_current: current greenlet (holds a reference)
- - ts_passaround_args: NULL if PyErr_Occurred(),
- else a tuple of args sent to ts_current (holds a reference)
- - ts_passaround_kwargs: switch kwargs (holds a reference)
- It is very important that stack switch is 'atomic', i.e. no
- calls into other Python code allowed (except very few that
- are safe), because global variables are very fragile.
- */
- int err;
- { /* save state */
- PyGreenlet* current = ts_current;
- PyThreadState* tstate = PyThreadState_GET();
- current->recursion_depth = tstate->recursion_depth;
- current->top_frame = tstate->frame;
+ /* Perform a stack switch according to some global variables
+ that must be set before:
+ - ts_current: current greenlet (holds a reference)
+ - ts_target: greenlet to switch to (weak reference)
+ - ts_passaround_args: NULL if PyErr_Occurred(),
+ else a tuple of args sent to ts_target (holds a reference)
+ - ts_passaround_kwargs: switch kwargs (holds a reference)
+ On return results are passed via global variables as well:
+ - ts_origin: originating greenlet (holds a reference)
+ - ts_current: current greenlet (holds a reference)
+ - ts_passaround_args: NULL if PyErr_Occurred(),
+ else a tuple of args sent to ts_current (holds a reference)
+ - ts_passaround_kwargs: switch kwargs (holds a reference)
+ It is very important that stack switch is 'atomic', i.e. no
+ calls into other Python code allowed (except very few that
+ are safe), because global variables are very fragile.
+ */
+ int err;
+ { /* save state */
+ PyGreenlet* current = ts_current;
+ PyThreadState* tstate = PyThreadState_GET();
+ current->recursion_depth = tstate->recursion_depth;
+ current->top_frame = tstate->frame;
#if GREENLET_PY37
- current->context = tstate->context;
+ current->context = tstate->context;
#endif
#if GREENLET_PY37
- current->exc_info = tstate->exc_info;
- current->exc_state = tstate->exc_state;
+ current->exc_info = tstate->exc_info;
+ current->exc_state = tstate->exc_state;
#else
- current->exc_type = tstate->exc_type;
- current->exc_value = tstate->exc_value;
- current->exc_traceback = tstate->exc_traceback;
+ current->exc_type = tstate->exc_type;
+ current->exc_value = tstate->exc_value;
+ current->exc_traceback = tstate->exc_traceback;
#endif
- }
- err = slp_switch();
- if (err < 0) { /* error */
- PyGreenlet* current = ts_current;
- current->top_frame = NULL;
+ }
+ err = slp_switch();
+ if (err < 0) { /* error */
+ PyGreenlet* current = ts_current;
+ current->top_frame = NULL;
#if GREENLET_PY37
- green_clear_exc(current);
+ green_clear_exc(current);
#else
- current->exc_type = NULL;
- current->exc_value = NULL;
- current->exc_traceback = NULL;
+ current->exc_type = NULL;
+ current->exc_value = NULL;
+ current->exc_traceback = NULL;
#endif
- assert(ts_origin == NULL);
- ts_target = NULL;
- }
- else {
- PyGreenlet* target = ts_target;
- PyGreenlet* origin = ts_current;
- PyThreadState* tstate = PyThreadState_GET();
- tstate->recursion_depth = target->recursion_depth;
- tstate->frame = target->top_frame;
- target->top_frame = NULL;
+ assert(ts_origin == NULL);
+ ts_target = NULL;
+ }
+ else {
+ PyGreenlet* target = ts_target;
+ PyGreenlet* origin = ts_current;
+ PyThreadState* tstate = PyThreadState_GET();
+ tstate->recursion_depth = target->recursion_depth;
+ tstate->frame = target->top_frame;
+ target->top_frame = NULL;
#if GREENLET_PY37
- tstate->context = target->context;
- target->context = NULL;
- /* Incrementing this value invalidates the contextvars cache,
- which would otherwise remain valid across switches */
- tstate->context_ver++;
+ tstate->context = target->context;
+ target->context = NULL;
+ /* Incrementing this value invalidates the contextvars cache,
+ which would otherwise remain valid across switches */
+ tstate->context_ver++;
#endif
#if GREENLET_PY37
- tstate->exc_state = target->exc_state;
- tstate->exc_info = target->exc_info ? target->exc_info : &tstate->exc_state;
+ tstate->exc_state = target->exc_state;
+ tstate->exc_info =
+ target->exc_info ? target->exc_info : &tstate->exc_state;
#else
- tstate->exc_type = target->exc_type;
- tstate->exc_value = target->exc_value;
- tstate->exc_traceback = target->exc_traceback;
+ tstate->exc_type = target->exc_type;
+ tstate->exc_value = target->exc_value;
+ tstate->exc_traceback = target->exc_traceback;
#endif
- green_clear_exc(target);
-
- assert(ts_origin == NULL);
- Py_INCREF(target);
- ts_current = target;
- ts_origin = origin;
- ts_target = NULL;
- }
- return err;
+ green_clear_exc(target);
+
+ assert(ts_origin == NULL);
+ Py_INCREF(target);
+ ts_current = target;
+ ts_origin = origin;
+ ts_target = NULL;
+ }
+ return err;
}
-
static int
-g_calltrace(PyObject* tracefunc, PyObject* event, PyGreenlet* origin, PyGreenlet* target)
+g_calltrace(PyObject* tracefunc, PyObject* event, PyGreenlet* origin,
+ PyGreenlet* target)
{
- PyObject *retval;
- PyObject *exc_type, *exc_val, *exc_tb;
- PyThreadState *tstate;
- PyErr_Fetch(&exc_type, &exc_val, &exc_tb);
- tstate = PyThreadState_GET();
- tstate->tracing++;
- tstate->use_tracing = 0;
- retval = PyObject_CallFunction(tracefunc, "O(OO)", event, origin, target);
- tstate->tracing--;
- tstate->use_tracing = (tstate->tracing <= 0 &&
- ((tstate->c_tracefunc != NULL) ||
- (tstate->c_profilefunc != NULL)));
- if (retval == NULL) {
- /* In case of exceptions trace function is removed */
- if (PyDict_GetItem(tstate->dict, ts_tracekey))
- PyDict_DelItem(tstate->dict, ts_tracekey);
- Py_XDECREF(exc_type);
- Py_XDECREF(exc_val);
- Py_XDECREF(exc_tb);
- return -1;
- } else
- Py_DECREF(retval);
- PyErr_Restore(exc_type, exc_val, exc_tb);
- return 0;
+ PyObject* retval;
+ PyObject *exc_type, *exc_val, *exc_tb;
+ PyThreadState* tstate;
+ PyErr_Fetch(&exc_type, &exc_val, &exc_tb);
+ tstate = PyThreadState_GET();
+ tstate->tracing++;
+ tstate->use_tracing = 0;
+ retval = PyObject_CallFunction(tracefunc, "O(OO)", event, origin, target);
+ tstate->tracing--;
+ tstate->use_tracing =
+ (tstate->tracing <= 0 &&
+ ((tstate->c_tracefunc != NULL) || (tstate->c_profilefunc != NULL)));
+ if (retval == NULL) {
+ /* In case of exceptions trace function is removed */
+ if (PyDict_GetItem(tstate->dict, ts_tracekey)) {
+ PyDict_DelItem(tstate->dict, ts_tracekey);
+ }
+ Py_XDECREF(exc_type);
+ Py_XDECREF(exc_val);
+ Py_XDECREF(exc_tb);
+ return -1;
+ }
+ else {
+ Py_DECREF(retval);
+ }
+ PyErr_Restore(exc_type, exc_val, exc_tb);
+ return 0;
}
-
-static PyObject *
+static PyObject*
g_switch(PyGreenlet* target, PyObject* args, PyObject* kwargs)
{
- /* _consumes_ a reference to the args tuple and kwargs dict,
- and return a new tuple reference */
- int err = 0;
- PyObject* run_info;
-
- /* check ts_current */
- if (!STATE_OK) {
- Py_XDECREF(args);
- Py_XDECREF(kwargs);
- return NULL;
- }
- run_info = green_statedict(target);
- if (run_info == NULL || run_info != ts_current->run_info) {
- Py_XDECREF(args);
- Py_XDECREF(kwargs);
- PyErr_SetString(PyExc_GreenletError, run_info
- ? "cannot switch to a different thread"
- : "cannot switch to a garbage collected greenlet");
- return NULL;
- }
-
- ts_passaround_args = args;
- ts_passaround_kwargs = kwargs;
-
- /* find the real target by ignoring dead greenlets,
- and if necessary starting a greenlet. */
- while (target) {
- if (PyGreenlet_ACTIVE(target)) {
- ts_target = target;
- err = g_switchstack();
- break;
- }
- if (!PyGreenlet_STARTED(target)) {
- void* dummymarker;
- ts_target = target;
- err = g_initialstub(&dummymarker);
- if (err == 1) {
- continue; /* retry the switch */
- }
- break;
- }
- target = target->parent;
- }
-
- /* For a very short time, immediately after the 'atomic'
- g_switchstack() call, global variables are in a known state.
- We need to save everything we need, before it is destroyed
- by calls into arbitrary Python code. */
- args = ts_passaround_args;
- ts_passaround_args = NULL;
- kwargs = ts_passaround_kwargs;
- ts_passaround_kwargs = NULL;
- if (err < 0) {
- /* Turn switch errors into switch throws */
- assert(ts_origin == NULL);
- Py_CLEAR(kwargs);
- Py_CLEAR(args);
- } else {
- PyGreenlet *origin;
- PyGreenlet *current;
- PyObject *tracefunc;
- origin = ts_origin;
- ts_origin = NULL;
-
- current = ts_current;
- if ((tracefunc = PyDict_GetItem(current->run_info, ts_tracekey)) != NULL) {
- Py_INCREF(tracefunc);
- if (g_calltrace(tracefunc, args ? ts_event_switch : ts_event_throw, origin, current) < 0) {
- /* Turn trace errors into switch throws */
- Py_CLEAR(kwargs);
- Py_CLEAR(args);
- }
- Py_DECREF(tracefunc);
- }
-
- Py_DECREF(origin);
- }
-
- /* We need to figure out what values to pass to the target greenlet
- based on the arguments that have been passed to greenlet.switch(). If
- switch() was just passed an arg tuple, then we'll just return that.
- If only keyword arguments were passed, then we'll pass the keyword
- argument dict. Otherwise, we'll create a tuple of (args, kwargs) and
- return both. */
- if (kwargs == NULL)
- {
- return args;
- }
- else if (PyDict_Size(kwargs) == 0)
- {
- Py_DECREF(kwargs);
- return args;
- }
- else if (PySequence_Length(args) == 0)
- {
- Py_DECREF(args);
- return kwargs;
- }
- else
- {
- PyObject *tuple = PyTuple_New(2);
- if (tuple == NULL) {
- Py_DECREF(args);
- Py_DECREF(kwargs);
- return NULL;
- }
- PyTuple_SET_ITEM(tuple, 0, args);
- PyTuple_SET_ITEM(tuple, 1, kwargs);
- return tuple;
- }
+ /* _consumes_ a reference to the args tuple and kwargs dict,
+ and return a new tuple reference */
+ int err = 0;
+ PyObject* run_info;
+
+ /* check ts_current */
+ if (!STATE_OK) {
+ Py_XDECREF(args);
+ Py_XDECREF(kwargs);
+ return NULL;
+ }
+ run_info = green_statedict(target);
+ if (run_info == NULL || run_info != ts_current->run_info) {
+ Py_XDECREF(args);
+ Py_XDECREF(kwargs);
+ PyErr_SetString(PyExc_GreenletError,
+ run_info ?
+ "cannot switch to a different thread" :
+ "cannot switch to a garbage collected greenlet");
+ return NULL;
+ }
+
+ ts_passaround_args = args;
+ ts_passaround_kwargs = kwargs;
+
+ /* find the real target by ignoring dead greenlets,
+ and if necessary starting a greenlet. */
+ while (target) {
+ if (PyGreenlet_ACTIVE(target)) {
+ ts_target = target;
+ err = g_switchstack();
+ break;
+ }
+ if (!PyGreenlet_STARTED(target)) {
+ void* dummymarker;
+ ts_target = target;
+ err = g_initialstub(&dummymarker);
+ if (err == 1) {
+ continue; /* retry the switch */
+ }
+ break;
+ }
+ target = target->parent;
+ }
+
+ /* For a very short time, immediately after the 'atomic'
+ g_switchstack() call, global variables are in a known state.
+ We need to save everything we need, before it is destroyed
+ by calls into arbitrary Python code. */
+ args = ts_passaround_args;
+ ts_passaround_args = NULL;
+ kwargs = ts_passaround_kwargs;
+ ts_passaround_kwargs = NULL;
+ if (err < 0) {
+ /* Turn switch errors into switch throws */
+ assert(ts_origin == NULL);
+ Py_CLEAR(kwargs);
+ Py_CLEAR(args);
+ }
+ else {
+ PyGreenlet* origin;
+ PyGreenlet* current;
+ PyObject* tracefunc;
+ origin = ts_origin;
+ ts_origin = NULL;
+
+ current = ts_current;
+ if ((tracefunc = PyDict_GetItem(current->run_info, ts_tracekey)) !=
+ NULL) {
+ Py_INCREF(tracefunc);
+ if (g_calltrace(tracefunc,
+ args ? ts_event_switch : ts_event_throw,
+ origin,
+ current) < 0) {
+ /* Turn trace errors into switch throws */
+ Py_CLEAR(kwargs);
+ Py_CLEAR(args);
+ }
+ Py_DECREF(tracefunc);
+ }
+
+ Py_DECREF(origin);
+ }
+
+ /* We need to figure out what values to pass to the target greenlet
+ based on the arguments that have been passed to greenlet.switch(). If
+ switch() was just passed an arg tuple, then we'll just return that.
+ If only keyword arguments were passed, then we'll pass the keyword
+ argument dict. Otherwise, we'll create a tuple of (args, kwargs) and
+ return both. */
+ if (kwargs == NULL) {
+ return args;
+ }
+ else if (PyDict_Size(kwargs) == 0) {
+ Py_DECREF(kwargs);
+ return args;
+ }
+ else if (PySequence_Length(args) == 0) {
+ Py_DECREF(args);
+ return kwargs;
+ }
+ else {
+ PyObject* tuple = PyTuple_New(2);
+ if (tuple == NULL) {
+ Py_DECREF(args);
+ Py_DECREF(kwargs);
+ return NULL;
+ }
+ PyTuple_SET_ITEM(tuple, 0, args);
+ PyTuple_SET_ITEM(tuple, 1, kwargs);
+ return tuple;
+ }
}
-static PyObject *
-g_handle_exit(PyObject *result)
+static PyObject*
+g_handle_exit(PyObject* result)
{
- if (result == NULL && PyErr_ExceptionMatches(PyExc_GreenletExit))
- {
- /* catch and ignore GreenletExit */
- PyObject *exc, *val, *tb;
- PyErr_Fetch(&exc, &val, &tb);
- if (val == NULL)
- {
- Py_INCREF(Py_None);
- val = Py_None;
- }
- result = val;
- Py_DECREF(exc);
- Py_XDECREF(tb);
- }
- if (result != NULL)
- {
- /* package the result into a 1-tuple */
- PyObject *r = result;
- result = PyTuple_New(1);
- if (result)
- {
- PyTuple_SET_ITEM(result, 0, r);
- }
- else
- {
- Py_DECREF(r);
- }
- }
- return result;
+ if (result == NULL && PyErr_ExceptionMatches(PyExc_GreenletExit)) {
+ /* catch and ignore GreenletExit */
+ PyObject *exc, *val, *tb;
+ PyErr_Fetch(&exc, &val, &tb);
+ if (val == NULL) {
+ Py_INCREF(Py_None);
+ val = Py_None;
+ }
+ result = val;
+ Py_DECREF(exc);
+ Py_XDECREF(tb);
+ }
+ if (result != NULL) {
+ /* package the result into a 1-tuple */
+ PyObject* r = result;
+ result = PyTuple_New(1);
+ if (result) {
+ PyTuple_SET_ITEM(result, 0, r);
+ }
+ else {
+ Py_DECREF(r);
+ }
+ }
+ return result;
}
static int GREENLET_NOINLINE(g_initialstub)(void* mark)
{
- int err;
- PyObject *o, *run;
- PyObject *exc, *val, *tb;
- PyObject *run_info;
- PyGreenlet* self = ts_target;
- PyObject* args = ts_passaround_args;
- PyObject* kwargs = ts_passaround_kwargs;
-
- /* save exception in case getattr clears it */
- PyErr_Fetch(&exc, &val, &tb);
- /* self.run is the object to call in the new greenlet */
- run = PyObject_GetAttrString((PyObject*) self, "run");
- if (run == NULL) {
- Py_XDECREF(exc);
- Py_XDECREF(val);
- Py_XDECREF(tb);
- return -1;
- }
- /* restore saved exception */
- PyErr_Restore(exc, val, tb);
-
- /* recheck the state in case getattr caused thread switches */
- if (!STATE_OK) {
- Py_DECREF(run);
- return -1;
- }
-
- /* recheck run_info in case greenlet reparented anywhere above */
- run_info = green_statedict(self);
- if (run_info == NULL || run_info != ts_current->run_info) {
- Py_DECREF(run);
- PyErr_SetString(PyExc_GreenletError, run_info
- ? "cannot switch to a different thread"
- : "cannot switch to a garbage collected greenlet");
- return -1;
- }
-
- /* by the time we got here another start could happen elsewhere,
- * that means it should now be a regular switch
- */
- if (PyGreenlet_STARTED(self)) {
- Py_DECREF(run);
- ts_passaround_args = args;
- ts_passaround_kwargs = kwargs;
- return 1;
- }
-
- /* start the greenlet */
- self->stack_start = NULL;
- self->stack_stop = (char*) mark;
- if (ts_current->stack_start == NULL) {
- /* ts_current is dying */
- self->stack_prev = ts_current->stack_prev;
- }
- else {
- self->stack_prev = ts_current;
- }
- self->top_frame = NULL;
- green_clear_exc(self);
- self->recursion_depth = PyThreadState_GET()->recursion_depth;
-
- /* restore arguments in case they are clobbered */
- ts_target = self;
- ts_passaround_args = args;
- ts_passaround_kwargs = kwargs;
-
- /* perform the initial switch */
- err = g_switchstack();
-
- /* returns twice!
- The 1st time with err=1: we are in the new greenlet
- The 2nd time with err=0: back in the caller's greenlet
- */
- if (err == 1) {
- /* in the new greenlet */
- PyGreenlet* origin;
- PyObject* tracefunc;
- PyObject* result;
- PyGreenlet* parent;
- self->stack_start = (char*) 1; /* running */
-
- /* grab origin while we still can */
- origin = ts_origin;
- ts_origin = NULL;
-
- /* now use run_info to store the statedict */
- o = self->run_info;
- self->run_info = green_statedict(self->parent);
- Py_INCREF(self->run_info);
- Py_XDECREF(o);
-
-
- if ((tracefunc = PyDict_GetItem(self->run_info, ts_tracekey)) != NULL) {
- Py_INCREF(tracefunc);
- if (g_calltrace(tracefunc, args ? ts_event_switch : ts_event_throw, origin, self) < 0) {
- /* Turn trace errors into switch throws */
- Py_CLEAR(kwargs);
- Py_CLEAR(args);
- }
- Py_DECREF(tracefunc);
- }
-
- Py_DECREF(origin);
-
- if (args == NULL) {
- /* pending exception */
- result = NULL;
- } else {
- /* call g.run(*args, **kwargs) */
- result = PyObject_Call(run, args, kwargs);
- Py_DECREF(args);
- Py_XDECREF(kwargs);
- }
- Py_DECREF(run);
- result = g_handle_exit(result);
-
- /* jump back to parent */
- self->stack_start = NULL; /* dead */
- for (parent = self->parent; parent != NULL; parent = parent->parent) {
- result = g_switch(parent, result, NULL);
- /* Return here means switch to parent failed,
- * in which case we throw *current* exception
- * to the next parent in chain.
- */
- assert(result == NULL);
- }
- /* We ran out of parents, cannot continue */
- PyErr_WriteUnraisable((PyObject *) self);
- Py_FatalError("greenlets cannot continue");
- }
- /* back in the parent */
- if (err < 0) {
- /* start failed badly, restore greenlet state */
- self->stack_start = NULL;
- self->stack_stop = NULL;
- self->stack_prev = NULL;
- }
- return err;
+ int err;
+ PyObject *o, *run;
+ PyObject *exc, *val, *tb;
+ PyObject* run_info;
+ PyGreenlet* self = ts_target;
+ PyObject* args = ts_passaround_args;
+ PyObject* kwargs = ts_passaround_kwargs;
+
+ /* save exception in case getattr clears it */
+ PyErr_Fetch(&exc, &val, &tb);
+ /* self.run is the object to call in the new greenlet */
+ run = PyObject_GetAttrString((PyObject*)self, "run");
+ if (run == NULL) {
+ Py_XDECREF(exc);
+ Py_XDECREF(val);
+ Py_XDECREF(tb);
+ return -1;
+ }
+ /* restore saved exception */
+ PyErr_Restore(exc, val, tb);
+
+ /* recheck the state in case getattr caused thread switches */
+ if (!STATE_OK) {
+ Py_DECREF(run);
+ return -1;
+ }
+
+ /* recheck run_info in case greenlet reparented anywhere above */
+ run_info = green_statedict(self);
+ if (run_info == NULL || run_info != ts_current->run_info) {
+ Py_DECREF(run);
+ PyErr_SetString(PyExc_GreenletError,
+ run_info ?
+ "cannot switch to a different thread" :
+ "cannot switch to a garbage collected greenlet");
+ return -1;
+ }
+
+ /* by the time we got here another start could happen elsewhere,
+ * that means it should now be a regular switch
+ */
+ if (PyGreenlet_STARTED(self)) {
+ Py_DECREF(run);
+ ts_passaround_args = args;
+ ts_passaround_kwargs = kwargs;
+ return 1;
+ }
+
+ /* start the greenlet */
+ self->stack_start = NULL;
+ self->stack_stop = (char*)mark;
+ if (ts_current->stack_start == NULL) {
+ /* ts_current is dying */
+ self->stack_prev = ts_current->stack_prev;
+ }
+ else {
+ self->stack_prev = ts_current;
+ }
+ self->top_frame = NULL;
+ green_clear_exc(self);
+ self->recursion_depth = PyThreadState_GET()->recursion_depth;
+
+ /* restore arguments in case they are clobbered */
+ ts_target = self;
+ ts_passaround_args = args;
+ ts_passaround_kwargs = kwargs;
+
+ /* perform the initial switch */
+ err = g_switchstack();
+
+ /* returns twice!
+ The 1st time with err=1: we are in the new greenlet
+ The 2nd time with err=0: back in the caller's greenlet
+ */
+ if (err == 1) {
+ /* in the new greenlet */
+ PyGreenlet* origin;
+ PyObject* tracefunc;
+ PyObject* result;
+ PyGreenlet* parent;
+ self->stack_start = (char*)1; /* running */
+
+ /* grab origin while we still can */
+ origin = ts_origin;
+ ts_origin = NULL;
+
+ /* now use run_info to store the statedict */
+ o = self->run_info;
+ self->run_info = green_statedict(self->parent);
+ Py_INCREF(self->run_info);
+ Py_XDECREF(o);
+
+ if ((tracefunc = PyDict_GetItem(self->run_info, ts_tracekey)) !=
+ NULL) {
+ Py_INCREF(tracefunc);
+ if (g_calltrace(tracefunc,
+ args ? ts_event_switch : ts_event_throw,
+ origin,
+ self) < 0) {
+ /* Turn trace errors into switch throws */
+ Py_CLEAR(kwargs);
+ Py_CLEAR(args);
+ }
+ Py_DECREF(tracefunc);
+ }
+
+ Py_DECREF(origin);
+
+ if (args == NULL) {
+ /* pending exception */
+ result = NULL;
+ }
+ else {
+ /* call g.run(*args, **kwargs) */
+ result = PyObject_Call(run, args, kwargs);
+ Py_DECREF(args);
+ Py_XDECREF(kwargs);
+ }
+ Py_DECREF(run);
+ result = g_handle_exit(result);
+
+ /* jump back to parent */
+ self->stack_start = NULL; /* dead */
+ for (parent = self->parent; parent != NULL; parent = parent->parent) {
+ result = g_switch(parent, result, NULL);
+ /* Return here means switch to parent failed,
+ * in which case we throw *current* exception
+ * to the next parent in chain.
+ */
+ assert(result == NULL);
+ }
+ /* We ran out of parents, cannot continue */
+ PyErr_WriteUnraisable((PyObject*)self);
+ Py_FatalError("greenlets cannot continue");
+ }
+ /* back in the parent */
+ if (err < 0) {
+ /* start failed badly, restore greenlet state */
+ self->stack_start = NULL;
+ self->stack_stop = NULL;
+ self->stack_prev = NULL;
+ }
+ return err;
}
-
/***********************************************************/
-
-static PyObject* green_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
+static PyObject*
+green_new(PyTypeObject* type, PyObject* args, PyObject* kwds)
{
- PyObject* o = PyBaseObject_Type.tp_new(type, ts_empty_tuple, ts_empty_dict);
- if (o != NULL) {
- if (!STATE_OK) {
- Py_DECREF(o);
- return NULL;
- }
- Py_INCREF(ts_current);
- ((PyGreenlet*) o)->parent = ts_current;
- }
- return o;
+ PyObject* o =
+ PyBaseObject_Type.tp_new(type, ts_empty_tuple, ts_empty_dict);
+ if (o != NULL) {
+ if (!STATE_OK) {
+ Py_DECREF(o);
+ return NULL;
+ }
+ Py_INCREF(ts_current);
+ ((PyGreenlet*)o)->parent = ts_current;
+ }
+ return o;
}
-static int green_setrun(PyGreenlet* self, PyObject* nrun, void* c);
-static int green_setparent(PyGreenlet* self, PyObject* nparent, void* c);
+static int
+green_setrun(PyGreenlet* self, PyObject* nrun, void* c);
+static int
+green_setparent(PyGreenlet* self, PyObject* nparent, void* c);
-static int green_init(PyGreenlet *self, PyObject *args, PyObject *kwargs)
+static int
+green_init(PyGreenlet* self, PyObject* args, PyObject* kwargs)
{
- PyObject *run = NULL;
- PyObject* nparent = NULL;
- static char *kwlist[] = {"run", "parent", 0};
- if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|OO:green", kwlist,
- &run, &nparent))
- return -1;
-
- if (run != NULL) {
- if (green_setrun(self, run, NULL))
- return -1;
- }
- if (nparent != NULL && nparent != Py_None)
- return green_setparent(self, nparent, NULL);
- return 0;
+ PyObject* run = NULL;
+ PyObject* nparent = NULL;
+ static char* kwlist[] = {"run", "parent", 0};
+ if (!PyArg_ParseTupleAndKeywords(
+ args, kwargs, "|OO:green", kwlist, &run, &nparent)) {
+ return -1;
+ }
+
+ if (run != NULL) {
+ if (green_setrun(self, run, NULL)) {
+ return -1;
+ }
+ }
+ if (nparent != NULL && nparent != Py_None) {
+ return green_setparent(self, nparent, NULL);
+ }
+ return 0;
}
-static int kill_greenlet(PyGreenlet* self)
+static int
+kill_greenlet(PyGreenlet* self)
{
- /* Cannot raise an exception to kill the greenlet if
- it is not running in the same thread! */
- if (self->run_info == PyThreadState_GET()->dict) {
- /* The dying greenlet cannot be a parent of ts_current
- because the 'parent' field chain would hold a
- reference */
- PyObject* result;
- PyGreenlet* oldparent;
- PyGreenlet* tmp;
- if (!STATE_OK) {
- return -1;
- }
- oldparent = self->parent;
- self->parent = ts_current;
- Py_INCREF(self->parent);
- /* Send the greenlet a GreenletExit exception. */
- PyErr_SetNone(PyExc_GreenletExit);
- result = g_switch(self, NULL, NULL);
- tmp = self->parent;
- self->parent = oldparent;
- Py_XDECREF(tmp);
- if (result == NULL)
- return -1;
- Py_DECREF(result);
- return 0;
- }
- else {
- /* Not the same thread! Temporarily save the greenlet
- into its thread's ts_delkey list. */
- PyObject* lst;
- lst = PyDict_GetItem(self->run_info, ts_delkey);
- if (lst == NULL) {
- lst = PyList_New(0);
- if (lst == NULL || PyDict_SetItem(self->run_info,
- ts_delkey, lst) < 0)
- return -1;
- }
- if (PyList_Append(lst, (PyObject*) self) < 0)
- return -1;
- if (!STATE_OK) /* to force ts_delkey to be reconsidered */
- return -1;
- return 0;
- }
+ /* Cannot raise an exception to kill the greenlet if
+ it is not running in the same thread! */
+ if (self->run_info == PyThreadState_GET()->dict) {
+ /* The dying greenlet cannot be a parent of ts_current
+ because the 'parent' field chain would hold a
+ reference */
+ PyObject* result;
+ PyGreenlet* oldparent;
+ PyGreenlet* tmp;
+ if (!STATE_OK) {
+ return -1;
+ }
+ oldparent = self->parent;
+ self->parent = ts_current;
+ Py_INCREF(self->parent);
+ /* Send the greenlet a GreenletExit exception. */
+ PyErr_SetNone(PyExc_GreenletExit);
+ result = g_switch(self, NULL, NULL);
+ tmp = self->parent;
+ self->parent = oldparent;
+ Py_XDECREF(tmp);
+ if (result == NULL) {
+ return -1;
+ }
+ Py_DECREF(result);
+ return 0;
+ }
+ else {
+ /* Not the same thread! Temporarily save the greenlet
+ into its thread's ts_delkey list. */
+ PyObject* lst;
+ lst = PyDict_GetItem(self->run_info, ts_delkey);
+ if (lst == NULL) {
+ lst = PyList_New(0);
+ if (lst == NULL ||
+ PyDict_SetItem(self->run_info, ts_delkey, lst) < 0) {
+ return -1;
+ }
+ }
+ if (PyList_Append(lst, (PyObject*)self) < 0) {
+ return -1;
+ }
+ if (!STATE_OK) { /* to force ts_delkey to be reconsidered */
+ return -1;
+ }
+ return 0;
+ }
}
-
static int
-green_traverse(PyGreenlet *self, visitproc visit, void *arg)
+green_traverse(PyGreenlet* self, visitproc visit, void* arg)
{
- /* We must only visit referenced objects, i.e. only objects
- Py_INCREF'ed by this greenlet (directly or indirectly):
- - stack_prev is not visited: holds previous stack pointer, but it's not referenced
- - frames are not visited: alive greenlets are not garbage collected anyway */
- Py_VISIT((PyObject*)self->parent);
- Py_VISIT(self->run_info);
+ /* We must only visit referenced objects, i.e. only objects
+ Py_INCREF'ed by this greenlet (directly or indirectly):
+ - stack_prev is not visited: holds previous stack pointer, but it's not
+ referenced
+ - frames are not visited: alive greenlets are not garbage collected
+ anyway */
+ Py_VISIT((PyObject*)self->parent);
+ Py_VISIT(self->run_info);
#if GREENLET_PY37
- Py_VISIT(self->context);
+ Py_VISIT(self->context);
#endif
#if GREENLET_PY37
- Py_VISIT(self->exc_state.exc_type);
- Py_VISIT(self->exc_state.exc_value);
- Py_VISIT(self->exc_state.exc_traceback);
+ Py_VISIT(self->exc_state.exc_type);
+ Py_VISIT(self->exc_state.exc_value);
+ Py_VISIT(self->exc_state.exc_traceback);
#else
- Py_VISIT(self->exc_type);
- Py_VISIT(self->exc_value);
- Py_VISIT(self->exc_traceback);
+ Py_VISIT(self->exc_type);
+ Py_VISIT(self->exc_value);
+ Py_VISIT(self->exc_traceback);
#endif
- Py_VISIT(self->dict);
- return 0;
+ Py_VISIT(self->dict);
+ return 0;
}
-static int green_is_gc(PyGreenlet* self)
+static int
+green_is_gc(PyGreenlet* self)
{
- /* Main greenlet can be garbage collected since it can only
- become unreachable if the underlying thread exited.
- Active greenlet cannot be garbage collected, however. */
- if (PyGreenlet_MAIN(self) || !PyGreenlet_ACTIVE(self))
- return 1;
- return 0;
+ /* Main greenlet can be garbage collected since it can only
+ become unreachable if the underlying thread exited.
+ Active greenlet cannot be garbage collected, however. */
+ if (PyGreenlet_MAIN(self) || !PyGreenlet_ACTIVE(self)) {
+ return 1;
+ }
+ return 0;
}
-static int green_clear(PyGreenlet* self)
+static int
+green_clear(PyGreenlet* self)
{
- /* Greenlet is only cleared if it is about to be collected.
- Since active greenlets are not garbage collectable, we can
- be sure that, even if they are deallocated during clear,
- nothing they reference is in unreachable or finalizers,
- so even if it switches we are relatively safe. */
- Py_CLEAR(self->parent);
- Py_CLEAR(self->run_info);
+ /* Greenlet is only cleared if it is about to be collected.
+ Since active greenlets are not garbage collectable, we can
+ be sure that, even if they are deallocated during clear,
+ nothing they reference is in unreachable or finalizers,
+ so even if it switches we are relatively safe. */
+ Py_CLEAR(self->parent);
+ Py_CLEAR(self->run_info);
#if GREENLET_PY37
- Py_CLEAR(self->context);
+ Py_CLEAR(self->context);
#endif
#if GREENLET_PY37
- Py_CLEAR(self->exc_state.exc_type);
- Py_CLEAR(self->exc_state.exc_value);
- Py_CLEAR(self->exc_state.exc_traceback);
+ Py_CLEAR(self->exc_state.exc_type);
+ Py_CLEAR(self->exc_state.exc_value);
+ Py_CLEAR(self->exc_state.exc_traceback);
#else
- Py_CLEAR(self->exc_type);
- Py_CLEAR(self->exc_value);
- Py_CLEAR(self->exc_traceback);
+ Py_CLEAR(self->exc_type);
+ Py_CLEAR(self->exc_value);
+ Py_CLEAR(self->exc_traceback);
#endif
- Py_CLEAR(self->dict);
- return 0;
+ Py_CLEAR(self->dict);
+ return 0;
}
-
-static void green_dealloc(PyGreenlet* self)
+static void
+green_dealloc(PyGreenlet* self)
{
- PyObject *error_type, *error_value, *error_traceback;
- Py_ssize_t refcnt;
-
- PyObject_GC_UnTrack(self);
-
- if (PyGreenlet_ACTIVE(self) && self->run_info != NULL && !PyGreenlet_MAIN(self)) {
- /* Hacks hacks hacks copied from instance_dealloc() */
- /* Temporarily resurrect the greenlet. */
- assert(Py_REFCNT(self) == 0);
- Py_SET_REFCNT(self, 1);
- /* Save the current exception, if any. */
- PyErr_Fetch(&error_type, &error_value, &error_traceback);
- if (kill_greenlet(self) < 0) {
- PyErr_WriteUnraisable((PyObject*) self);
- /* XXX what else should we do? */
- }
- /* Check for no resurrection must be done while we keep
- * our internal reference, otherwise PyFile_WriteObject
- * causes recursion if using Py_INCREF/Py_DECREF
- */
- if (Py_REFCNT(self) == 1 && PyGreenlet_ACTIVE(self)) {
- /* Not resurrected, but still not dead!
- XXX what else should we do? we complain. */
- PyObject* f = PySys_GetObject("stderr");
- Py_INCREF(self); /* leak! */
- if (f != NULL) {
- PyFile_WriteString("GreenletExit did not kill ",
- f);
- PyFile_WriteObject((PyObject*) self, f, 0);
- PyFile_WriteString("\n", f);
- }
- }
- /* Restore the saved exception. */
- PyErr_Restore(error_type, error_value, error_traceback);
- /* Undo the temporary resurrection; can't use DECREF here,
- * it would cause a recursive call.
- */
- assert(Py_REFCNT(self) > 0);
-
- refcnt = Py_REFCNT(self) - 1;
- Py_SET_REFCNT(self, refcnt);
- if (refcnt != 0) {
- /* Resurrected! */
- _Py_NewReference((PyObject*) self);
- Py_SET_REFCNT(self, refcnt);
-
- PyObject_GC_Track((PyObject *)self);
-
- _Py_DEC_REFTOTAL;
+ PyObject *error_type, *error_value, *error_traceback;
+ Py_ssize_t refcnt;
+
+ PyObject_GC_UnTrack(self);
+
+ if (PyGreenlet_ACTIVE(self) && self->run_info != NULL &&
+ !PyGreenlet_MAIN(self)) {
+ /* Hacks hacks hacks copied from instance_dealloc() */
+ /* Temporarily resurrect the greenlet. */
+ assert(Py_REFCNT(self) == 0);
+ Py_SET_REFCNT(self, 1);
+ /* Save the current exception, if any. */
+ PyErr_Fetch(&error_type, &error_value, &error_traceback);
+ if (kill_greenlet(self) < 0) {
+ PyErr_WriteUnraisable((PyObject*)self);
+ /* XXX what else should we do? */
+ }
+ /* Check for no resurrection must be done while we keep
+ * our internal reference, otherwise PyFile_WriteObject
+ * causes recursion if using Py_INCREF/Py_DECREF
+ */
+ if (Py_REFCNT(self) == 1 && PyGreenlet_ACTIVE(self)) {
+ /* Not resurrected, but still not dead!
+ XXX what else should we do? we complain. */
+ PyObject* f = PySys_GetObject("stderr");
+ Py_INCREF(self); /* leak! */
+ if (f != NULL) {
+ PyFile_WriteString("GreenletExit did not kill ", f);
+ PyFile_WriteObject((PyObject*)self, f, 0);
+ PyFile_WriteString("\n", f);
+ }
+ }
+ /* Restore the saved exception. */
+ PyErr_Restore(error_type, error_value, error_traceback);
+ /* Undo the temporary resurrection; can't use DECREF here,
+ * it would cause a recursive call.
+ */
+ assert(Py_REFCNT(self) > 0);
+
+ refcnt = Py_REFCNT(self) - 1;
+ Py_SET_REFCNT(self, refcnt);
+ if (refcnt != 0) {
+ /* Resurrected! */
+ _Py_NewReference((PyObject*)self);
+ Py_SET_REFCNT(self, refcnt);
+
+ PyObject_GC_Track((PyObject*)self);
+
+ _Py_DEC_REFTOTAL;
#ifdef COUNT_ALLOCS
- --Py_TYPE(self)->tp_frees;
- --Py_TYPE(self)->tp_allocs;
+ --Py_TYPE(self)->tp_frees;
+ --Py_TYPE(self)->tp_allocs;
#endif /* COUNT_ALLOCS */
- return;
- }
- }
- if (self->weakreflist != NULL)
- PyObject_ClearWeakRefs((PyObject *) self);
- Py_CLEAR(self->parent);
- Py_CLEAR(self->run_info);
+ return;
+ }
+ }
+ if (self->weakreflist != NULL) {
+ PyObject_ClearWeakRefs((PyObject*)self);
+ }
+ Py_CLEAR(self->parent);
+ Py_CLEAR(self->run_info);
#if GREENLET_PY37
- Py_CLEAR(self->context);
+ Py_CLEAR(self->context);
#endif
#if GREENLET_PY37
- Py_CLEAR(self->exc_state.exc_type);
- Py_CLEAR(self->exc_state.exc_value);
- Py_CLEAR(self->exc_state.exc_traceback);
+ Py_CLEAR(self->exc_state.exc_type);
+ Py_CLEAR(self->exc_state.exc_value);
+ Py_CLEAR(self->exc_state.exc_traceback);
#else
- Py_CLEAR(self->exc_type);
- Py_CLEAR(self->exc_value);
- Py_CLEAR(self->exc_traceback);
+ Py_CLEAR(self->exc_type);
+ Py_CLEAR(self->exc_value);
+ Py_CLEAR(self->exc_traceback);
#endif
- Py_CLEAR(self->dict);
- Py_TYPE(self)->tp_free((PyObject*)self);
+ Py_CLEAR(self->dict);
+ Py_TYPE(self)->tp_free((PyObject*)self);
}
-static PyObject* single_result(PyObject* results)
+static PyObject*
+single_result(PyObject* results)
{
- if (results != NULL && PyTuple_Check(results) &&
- PyTuple_GET_SIZE(results) == 1) {
- PyObject *result = PyTuple_GET_ITEM(results, 0);
- Py_INCREF(result);
- Py_DECREF(results);
- return result;
- }
- else
- return results;
+ if (results != NULL && PyTuple_Check(results) &&
+ PyTuple_GET_SIZE(results) == 1) {
+ PyObject* result = PyTuple_GET_ITEM(results, 0);
+ Py_INCREF(result);
+ Py_DECREF(results);
+ return result;
+ }
+ else {
+ return results;
+ }
}
-static PyObject *
-throw_greenlet(PyGreenlet *self, PyObject *typ, PyObject *val, PyObject *tb)
+static PyObject*
+throw_greenlet(PyGreenlet* self, PyObject* typ, PyObject* val, PyObject* tb)
{
- /* Note: _consumes_ a reference to typ, val, tb */
- PyObject *result = NULL;
- PyErr_Restore(typ, val, tb);
- if (PyGreenlet_STARTED(self) && !PyGreenlet_ACTIVE(self))
- {
- /* dead greenlet: turn GreenletExit into a regular return */
- result = g_handle_exit(result);
- }
- return single_result(g_switch(self, result, NULL));
+ /* Note: _consumes_ a reference to typ, val, tb */
+ PyObject* result = NULL;
+ PyErr_Restore(typ, val, tb);
+ if (PyGreenlet_STARTED(self) && !PyGreenlet_ACTIVE(self)) {
+ /* dead greenlet: turn GreenletExit into a regular return */
+ result = g_handle_exit(result);
+ }
+ return single_result(g_switch(self, result, NULL));
}
-PyDoc_STRVAR(green_switch_doc,
-"switch(*args, **kwargs)\n"
-"\n"
-"Switch execution to this greenlet.\n"
-"\n"
-"If this greenlet has never been run, then this greenlet\n"
-"will be switched to using the body of ``self.run(*args, **kwargs)``.\n"
-"\n"
-"If the greenlet is active (has been run, but was switch()'ed\n"
-"out before leaving its run function), then this greenlet will\n"
-"be resumed and the return value to its switch call will be\n"
-"None if no arguments are given, the given argument if one\n"
-"argument is given, or the args tuple and keyword args dict if\n"
-"multiple arguments are given.\n"
-"\n"
-"If the greenlet is dead, or is the current greenlet then this\n"
-"function will simply return the arguments using the same rules as\n"
-"above.\n");
-
-static PyObject* green_switch(
- PyGreenlet* self,
- PyObject* args,
- PyObject* kwargs)
+PyDoc_STRVAR(
+ green_switch_doc,
+ "switch(*args, **kwargs)\n"
+ "\n"
+ "Switch execution to this greenlet.\n"
+ "\n"
+ "If this greenlet has never been run, then this greenlet\n"
+ "will be switched to using the body of ``self.run(*args, **kwargs)``.\n"
+ "\n"
+ "If the greenlet is active (has been run, but was switch()'ed\n"
+ "out before leaving its run function), then this greenlet will\n"
+ "be resumed and the return value to its switch call will be\n"
+ "None if no arguments are given, the given argument if one\n"
+ "argument is given, or the args tuple and keyword args dict if\n"
+ "multiple arguments are given.\n"
+ "\n"
+ "If the greenlet is dead, or is the current greenlet then this\n"
+ "function will simply return the arguments using the same rules as\n"
+ "above.\n");
+
+static PyObject*
+green_switch(PyGreenlet* self, PyObject* args, PyObject* kwargs)
{
- Py_INCREF(args);
- Py_XINCREF(kwargs);
- return single_result(g_switch(self, args, kwargs));
+ Py_INCREF(args);
+ Py_XINCREF(kwargs);
+ return single_result(g_switch(self, args, kwargs));
}
-PyDoc_STRVAR(green_throw_doc,
-"Switches execution to this greenlet, but immediately raises the\n"
-"given exception in this greenlet. If no argument is provided, the exception\n"
-"defaults to `greenlet.GreenletExit`. The normal exception\n"
-"propagation rules apply, as described for `switch`. Note that calling this\n"
-"method is almost equivalent to the following::\n"
-"\n"
-" def raiser():\n"
-" raise typ, val, tb\n"
-" g_raiser = greenlet(raiser, parent=g)\n"
-" g_raiser.switch()\n"
-"\n"
-"except that this trick does not work for the\n"
-"`greenlet.GreenletExit` exception, which would not propagate\n"
-"from ``g_raiser`` to ``g``.\n");
-
-static PyObject *
-green_throw(PyGreenlet *self, PyObject *args)
+PyDoc_STRVAR(
+ green_throw_doc,
+ "Switches execution to this greenlet, but immediately raises the\n"
+ "given exception in this greenlet. If no argument is provided, the "
+ "exception\n"
+ "defaults to `greenlet.GreenletExit`. The normal exception\n"
+ "propagation rules apply, as described for `switch`. Note that calling "
+ "this\n"
+ "method is almost equivalent to the following::\n"
+ "\n"
+ " def raiser():\n"
+ " raise typ, val, tb\n"
+ " g_raiser = greenlet(raiser, parent=g)\n"
+ " g_raiser.switch()\n"
+ "\n"
+ "except that this trick does not work for the\n"
+ "`greenlet.GreenletExit` exception, which would not propagate\n"
+ "from ``g_raiser`` to ``g``.\n");
+
+static PyObject*
+green_throw(PyGreenlet* self, PyObject* args)
{
- PyObject *typ = PyExc_GreenletExit;
- PyObject *val = NULL;
- PyObject *tb = NULL;
-
- if (!PyArg_ParseTuple(args, "|OOO:throw", &typ, &val, &tb))
- {
- return NULL;
- }
-
- /* First, check the traceback argument, replacing None, with NULL */
- if (tb == Py_None)
- {
- tb = NULL;
- }
- else if (tb != NULL && !PyTraceBack_Check(tb))
- {
- PyErr_SetString(
- PyExc_TypeError,
- "throw() third argument must be a traceback object");
- return NULL;
- }
-
- Py_INCREF(typ);
- Py_XINCREF(val);
- Py_XINCREF(tb);
-
- if (PyExceptionClass_Check(typ))
- {
- PyErr_NormalizeException(&typ, &val, &tb);
- }
- else if (PyExceptionInstance_Check(typ))
- {
- /* Raising an instance. The value should be a dummy. */
- if (val && val != Py_None)
- {
- PyErr_SetString(
- PyExc_TypeError,
- "instance exception may not have a separate value");
- goto failed_throw;
- }
- else
- {
- /* Normalize to raise <class>, <instance> */
- Py_XDECREF(val);
- val = typ;
- typ = PyExceptionInstance_Class(typ);
- Py_INCREF(typ);
- }
- }
- else
- {
- /* Not something you can raise. throw() fails. */
- PyErr_Format(
- PyExc_TypeError,
- "exceptions must be classes, or instances, not %s",
- Py_TYPE(typ)->tp_name);
- goto failed_throw;
- }
-
- return throw_greenlet(self, typ, val, tb);
+ PyObject* typ = PyExc_GreenletExit;
+ PyObject* val = NULL;
+ PyObject* tb = NULL;
+
+ if (!PyArg_ParseTuple(args, "|OOO:throw", &typ, &val, &tb)) {
+ return NULL;
+ }
+
+ /* First, check the traceback argument, replacing None, with NULL */
+ if (tb == Py_None) {
+ tb = NULL;
+ }
+ else if (tb != NULL && !PyTraceBack_Check(tb)) {
+ PyErr_SetString(PyExc_TypeError,
+ "throw() third argument must be a traceback object");
+ return NULL;
+ }
+
+ Py_INCREF(typ);
+ Py_XINCREF(val);
+ Py_XINCREF(tb);
+
+ if (PyExceptionClass_Check(typ)) {
+ PyErr_NormalizeException(&typ, &val, &tb);
+ }
+ else if (PyExceptionInstance_Check(typ)) {
+ /* Raising an instance. The value should be a dummy. */
+ if (val && val != Py_None) {
+ PyErr_SetString(
+ PyExc_TypeError,
+ "instance exception may not have a separate value");
+ goto failed_throw;
+ }
+ else {
+ /* Normalize to raise <class>, <instance> */
+ Py_XDECREF(val);
+ val = typ;
+ typ = PyExceptionInstance_Class(typ);
+ Py_INCREF(typ);
+ }
+ }
+ else {
+ /* Not something you can raise. throw() fails. */
+ PyErr_Format(PyExc_TypeError,
+ "exceptions must be classes, or instances, not %s",
+ Py_TYPE(typ)->tp_name);
+ goto failed_throw;
+ }
+
+ return throw_greenlet(self, typ, val, tb);
failed_throw:
- /* Didn't use our arguments, so restore their original refcounts */
- Py_DECREF(typ);
- Py_XDECREF(val);
- Py_XDECREF(tb);
- return NULL;
+ /* Didn't use our arguments, so restore their original refcounts */
+ Py_DECREF(typ);
+ Py_XDECREF(val);
+ Py_XDECREF(tb);
+ return NULL;
}
-static int green_bool(PyGreenlet* self)
+static int
+green_bool(PyGreenlet* self)
{
- return PyGreenlet_ACTIVE(self);
+ return PyGreenlet_ACTIVE(self);
}
-static PyObject* green_getdict(PyGreenlet* self, void* c)
+static PyObject*
+green_getdict(PyGreenlet* self, void* c)
{
- if (self->dict == NULL) {
- self->dict = PyDict_New();
- if (self->dict == NULL)
- return NULL;
- }
- Py_INCREF(self->dict);
- return self->dict;
+ if (self->dict == NULL) {
+ self->dict = PyDict_New();
+ if (self->dict == NULL) {
+ return NULL;
+ }
+ }
+ Py_INCREF(self->dict);
+ return self->dict;
}
-static int green_setdict(PyGreenlet* self, PyObject* val, void* c)
+static int
+green_setdict(PyGreenlet* self, PyObject* val, void* c)
{
- PyObject* tmp;
-
- if (val == NULL) {
- PyErr_SetString(PyExc_TypeError, "__dict__ may not be deleted");
- return -1;
- }
- if (!PyDict_Check(val)) {
- PyErr_SetString(PyExc_TypeError, "__dict__ must be a dictionary");
- return -1;
- }
- tmp = self->dict;
- Py_INCREF(val);
- self->dict = val;
- Py_XDECREF(tmp);
- return 0;
+ PyObject* tmp;
+
+ if (val == NULL) {
+ PyErr_SetString(PyExc_TypeError, "__dict__ may not be deleted");
+ return -1;
+ }
+ if (!PyDict_Check(val)) {
+ PyErr_SetString(PyExc_TypeError, "__dict__ must be a dictionary");
+ return -1;
+ }
+ tmp = self->dict;
+ Py_INCREF(val);
+ self->dict = val;
+ Py_XDECREF(tmp);
+ return 0;
}
-static PyObject* green_getdead(PyGreenlet* self, void* c)
+static PyObject*
+green_getdead(PyGreenlet* self, void* c)
{
- if (PyGreenlet_ACTIVE(self) || !PyGreenlet_STARTED(self))
- Py_RETURN_FALSE;
- else
- Py_RETURN_TRUE;
+ if (PyGreenlet_ACTIVE(self) || !PyGreenlet_STARTED(self)) {
+ Py_RETURN_FALSE;
+ }
+ else {
+ Py_RETURN_TRUE;
+ }
}
-
-static PyObject* green_get_stack_saved(PyGreenlet* self, void* c)
+static PyObject*
+green_get_stack_saved(PyGreenlet* self, void* c)
{
- return PyLong_FromSsize_t(self->stack_saved);
+ return PyLong_FromSsize_t(self->stack_saved);
}
-
-static PyObject* green_getrun(PyGreenlet* self, void* c)
+static PyObject*
+green_getrun(PyGreenlet* self, void* c)
{
- if (PyGreenlet_STARTED(self) || self->run_info == NULL) {
- PyErr_SetString(PyExc_AttributeError, "run");
- return NULL;
- }
- Py_INCREF(self->run_info);
- return self->run_info;
+ if (PyGreenlet_STARTED(self) || self->run_info == NULL) {
+ PyErr_SetString(PyExc_AttributeError, "run");
+ return NULL;
+ }
+ Py_INCREF(self->run_info);
+ return self->run_info;
}
-static int green_setrun(PyGreenlet* self, PyObject* nrun, void* c)
+static int
+green_setrun(PyGreenlet* self, PyObject* nrun, void* c)
{
- PyObject* o;
- if (PyGreenlet_STARTED(self)) {
- PyErr_SetString(PyExc_AttributeError,
- "run cannot be set "
- "after the start of the greenlet");
- return -1;
- }
- o = self->run_info;
- self->run_info = nrun;
- Py_XINCREF(nrun);
- Py_XDECREF(o);
- return 0;
+ PyObject* o;
+ if (PyGreenlet_STARTED(self)) {
+ PyErr_SetString(PyExc_AttributeError,
+ "run cannot be set "
+ "after the start of the greenlet");
+ return -1;
+ }
+ o = self->run_info;
+ self->run_info = nrun;
+ Py_XINCREF(nrun);
+ Py_XDECREF(o);
+ return 0;
}
-static PyObject* green_getparent(PyGreenlet* self, void* c)
+static PyObject*
+green_getparent(PyGreenlet* self, void* c)
{
- PyObject* result = self->parent ? (PyObject*) self->parent : Py_None;
- Py_INCREF(result);
- return result;
+ PyObject* result = self->parent ? (PyObject*)self->parent : Py_None;
+ Py_INCREF(result);
+ return result;
}
-static int green_setparent(PyGreenlet* self, PyObject* nparent, void* c)
+static int
+green_setparent(PyGreenlet* self, PyObject* nparent, void* c)
{
- PyGreenlet* p;
- PyObject* run_info = NULL;
- if (nparent == NULL) {
- PyErr_SetString(PyExc_AttributeError, "can't delete attribute");
- return -1;
- }
- if (!PyGreenlet_Check(nparent)) {
- PyErr_SetString(PyExc_TypeError, "parent must be a greenlet");
- return -1;
- }
- for (p=(PyGreenlet*) nparent; p; p=p->parent) {
- if (p == self) {
- PyErr_SetString(PyExc_ValueError, "cyclic parent chain");
- return -1;
- }
- run_info = PyGreenlet_ACTIVE(p) ? p->run_info : NULL;
- }
- if (run_info == NULL) {
- PyErr_SetString(PyExc_ValueError, "parent must not be garbage collected");
- return -1;
- }
- if (PyGreenlet_STARTED(self) && self->run_info != run_info) {
- PyErr_SetString(PyExc_ValueError, "parent cannot be on a different thread");
- return -1;
- }
- p = self->parent;
- self->parent = (PyGreenlet*) nparent;
- Py_INCREF(nparent);
- Py_XDECREF(p);
- return 0;
+ PyGreenlet* p;
+ PyObject* run_info = NULL;
+ if (nparent == NULL) {
+ PyErr_SetString(PyExc_AttributeError, "can't delete attribute");
+ return -1;
+ }
+ if (!PyGreenlet_Check(nparent)) {
+ PyErr_SetString(PyExc_TypeError, "parent must be a greenlet");
+ return -1;
+ }
+ for (p = (PyGreenlet*)nparent; p; p = p->parent) {
+ if (p == self) {
+ PyErr_SetString(PyExc_ValueError, "cyclic parent chain");
+ return -1;
+ }
+ run_info = PyGreenlet_ACTIVE(p) ? p->run_info : NULL;
+ }
+ if (run_info == NULL) {
+ PyErr_SetString(PyExc_ValueError,
+ "parent must not be garbage collected");
+ return -1;
+ }
+ if (PyGreenlet_STARTED(self) && self->run_info != run_info) {
+ PyErr_SetString(PyExc_ValueError,
+ "parent cannot be on a different thread");
+ return -1;
+ }
+ p = self->parent;
+ self->parent = (PyGreenlet*)nparent;
+ Py_INCREF(nparent);
+ Py_XDECREF(p);
+ return 0;
}
#ifdef Py_CONTEXT_H
-#define GREENLET_NO_CONTEXTVARS_REASON "This build of greenlet"
+# define GREENLET_NO_CONTEXTVARS_REASON "This build of greenlet"
#else
-#define GREENLET_NO_CONTEXTVARS_REASON "This Python interpreter"
+# define GREENLET_NO_CONTEXTVARS_REASON "This Python interpreter"
#endif
-static PyObject* green_getcontext(PyGreenlet* self, void* c)
+static PyObject*
+green_getcontext(PyGreenlet* self, void* c)
{
#if GREENLET_PY37
- PyThreadState* tstate = PyThreadState_GET();
- PyObject* result;
-
- if (!STATE_OK) {
- return NULL;
- }
- if (PyGreenlet_ACTIVE(self) && self->top_frame == NULL) {
- /* Currently running greenlet: context is stored in the thread state,
- not the greenlet object. */
- if (self == ts_current) {
- result = tstate->context;
- }
- else {
- PyErr_SetString(PyExc_ValueError,
- "cannot get context of a "
- "greenlet that is running in a different thread");
- return NULL;
- }
- }
- else {
- /* Greenlet is not running: just return context. */
- result = self->context;
- }
- if (result == NULL) {
- result = Py_None;
- }
- Py_INCREF(result);
- return result;
+ PyThreadState* tstate = PyThreadState_GET();
+ PyObject* result;
+
+ if (!STATE_OK) {
+ return NULL;
+ }
+ if (PyGreenlet_ACTIVE(self) && self->top_frame == NULL) {
+ /* Currently running greenlet: context is stored in the thread state,
+ not the greenlet object. */
+ if (self == ts_current) {
+ result = tstate->context;
+ }
+ else {
+ PyErr_SetString(PyExc_ValueError,
+ "cannot get context of a "
+ "greenlet that is running in a different thread");
+ return NULL;
+ }
+ }
+ else {
+ /* Greenlet is not running: just return context. */
+ result = self->context;
+ }
+ if (result == NULL) {
+ result = Py_None;
+ }
+ Py_INCREF(result);
+ return result;
#else
- PyErr_SetString(PyExc_AttributeError,
- GREENLET_NO_CONTEXTVARS_REASON
- " does not support context variables");
- return NULL;
+ PyErr_SetString(PyExc_AttributeError,
+ GREENLET_NO_CONTEXTVARS_REASON
+ " does not support context variables");
+ return NULL;
#endif
}
-static int green_setcontext(PyGreenlet* self, PyObject* nctx, void* c)
+static int
+green_setcontext(PyGreenlet* self, PyObject* nctx, void* c)
{
#if GREENLET_PY37
- PyThreadState* tstate;
- PyObject* octx = NULL;
- if (!STATE_OK) {
- return -1;
- }
- if (nctx == NULL) {
- PyErr_SetString(PyExc_AttributeError, "can't delete attribute");
- return -1;
- }
- if (nctx == Py_None) {
- /* "Empty context" is stored as NULL, not None. */
- nctx = NULL;
- }
- else if (!PyContext_CheckExact(nctx)) {
- PyErr_SetString(PyExc_TypeError,
- "greenlet context must be a "
- "contextvars.Context or None");
- return -1;
- }
- tstate = PyThreadState_GET();
- if (PyGreenlet_ACTIVE(self) && self->top_frame == NULL) {
- /* Currently running greenlet: context is stored in the thread state,
- not the greenlet object. */
- if (self == ts_current) {
- octx = tstate->context;
- tstate->context = nctx;
- tstate->context_ver++;
- Py_XINCREF(nctx);
- }
- else {
- PyErr_SetString(PyExc_ValueError,
- "cannot set context of a "
- "greenlet that is running in a different thread");
- return -1;
- }
- }
- else {
- /* Greenlet is not running: just set context. */
- octx = self->context;
- self->context = nctx;
- Py_XINCREF(nctx);
- }
- Py_XDECREF(octx);
- return 0;
+ PyThreadState* tstate;
+ PyObject* octx = NULL;
+ if (!STATE_OK) {
+ return -1;
+ }
+ if (nctx == NULL) {
+ PyErr_SetString(PyExc_AttributeError, "can't delete attribute");
+ return -1;
+ }
+ if (nctx == Py_None) {
+ /* "Empty context" is stored as NULL, not None. */
+ nctx = NULL;
+ }
+ else if (!PyContext_CheckExact(nctx)) {
+ PyErr_SetString(PyExc_TypeError,
+ "greenlet context must be a "
+ "contextvars.Context or None");
+ return -1;
+ }
+ tstate = PyThreadState_GET();
+ if (PyGreenlet_ACTIVE(self) && self->top_frame == NULL) {
+ /* Currently running greenlet: context is stored in the thread state,
+ not the greenlet object. */
+ if (self == ts_current) {
+ octx = tstate->context;
+ tstate->context = nctx;
+ tstate->context_ver++;
+ Py_XINCREF(nctx);
+ }
+ else {
+ PyErr_SetString(PyExc_ValueError,
+ "cannot set context of a "
+ "greenlet that is running in a different thread");
+ return -1;
+ }
+ }
+ else {
+ /* Greenlet is not running: just set context. */
+ octx = self->context;
+ self->context = nctx;
+ Py_XINCREF(nctx);
+ }
+ Py_XDECREF(octx);
+ return 0;
#else
- PyErr_SetString(PyExc_AttributeError,
- GREENLET_NO_CONTEXTVARS_REASON
- " does not support context variables");
- return -1;
+ PyErr_SetString(PyExc_AttributeError,
+ GREENLET_NO_CONTEXTVARS_REASON
+ " does not support context variables");
+ return -1;
#endif
}
#undef GREENLET_NO_CONTEXTVARS_REASON
-static PyObject* green_getframe(PyGreenlet* self, void* c)
+static PyObject*
+green_getframe(PyGreenlet* self, void* c)
{
- PyObject* result = self->top_frame ? (PyObject*) self->top_frame : Py_None;
- Py_INCREF(result);
- return result;
+ PyObject* result = self->top_frame ? (PyObject*)self->top_frame : Py_None;
+ Py_INCREF(result);
+ return result;
}
-static PyObject* green_getstate(PyGreenlet* self)
+static PyObject*
+green_getstate(PyGreenlet* self)
{
- PyErr_Format(PyExc_TypeError,
- "cannot serialize '%s' object",
- Py_TYPE(self)->tp_name);
- return NULL;
+ PyErr_Format(PyExc_TypeError,
+ "cannot serialize '%s' object",
+ Py_TYPE(self)->tp_name);
+ return NULL;
}
-
/*****************************************************************************
* C interface
*
* These are exported using the CObject API
*/
-static PyGreenlet *
+static PyGreenlet*
PyGreenlet_GetCurrent(void)
{
- if (!STATE_OK) {
- return NULL;
- }
- Py_INCREF(ts_current);
- return ts_current;
+ if (!STATE_OK) {
+ return NULL;
+ }
+ Py_INCREF(ts_current);
+ return ts_current;
}
static int
-PyGreenlet_SetParent(PyGreenlet *g, PyGreenlet *nparent)
+PyGreenlet_SetParent(PyGreenlet* g, PyGreenlet* nparent)
{
- if (!PyGreenlet_Check(g)) {
- PyErr_SetString(PyExc_TypeError, "parent must be a greenlet");
- return -1;
- }
+ if (!PyGreenlet_Check(g)) {
+ PyErr_SetString(PyExc_TypeError, "parent must be a greenlet");
+ return -1;
+ }
- return green_setparent((PyGreenlet*) g, (PyObject *) nparent, NULL);
+ return green_setparent((PyGreenlet*)g, (PyObject*)nparent, NULL);
}
-static PyGreenlet *
-PyGreenlet_New(PyObject *run, PyGreenlet *parent)
+static PyGreenlet*
+PyGreenlet_New(PyObject* run, PyGreenlet* parent)
{
- PyGreenlet* g = NULL;
-
- g = (PyGreenlet *) PyType_GenericAlloc(&PyGreenlet_Type, 0);
- if (g == NULL) {
- return NULL;
- }
-
- if (run != NULL) {
- Py_INCREF(run);
- g->run_info = run;
- }
-
- if (parent != NULL) {
- if (PyGreenlet_SetParent(g, parent)) {
- Py_DECREF(g);
- return NULL;
- }
- } else {
- if ((g->parent = PyGreenlet_GetCurrent()) == NULL) {
- Py_DECREF(g);
- return NULL;
- }
- }
-
- return g;
+ PyGreenlet* g = NULL;
+
+ g = (PyGreenlet*)PyType_GenericAlloc(&PyGreenlet_Type, 0);
+ if (g == NULL) {
+ return NULL;
+ }
+
+ if (run != NULL) {
+ Py_INCREF(run);
+ g->run_info = run;
+ }
+
+ if (parent != NULL) {
+ if (PyGreenlet_SetParent(g, parent)) {
+ Py_DECREF(g);
+ return NULL;
+ }
+ }
+ else {
+ if ((g->parent = PyGreenlet_GetCurrent()) == NULL) {
+ Py_DECREF(g);
+ return NULL;
+ }
+ }
+
+ return g;
}
-static PyObject *
-PyGreenlet_Switch(PyGreenlet *g, PyObject *args, PyObject *kwargs)
+static PyObject*
+PyGreenlet_Switch(PyGreenlet* g, PyObject* args, PyObject* kwargs)
{
- PyGreenlet *self = (PyGreenlet *) g;
-
- if (!PyGreenlet_Check(self)) {
- PyErr_BadArgument();
- return NULL;
- }
-
- if (args == NULL) {
- args = Py_BuildValue("()");
- }
- else {
- Py_INCREF(args);
- }
-
- if (kwargs != NULL && PyDict_Check(kwargs)) {
- Py_INCREF(kwargs);
- }
- else {
- kwargs = NULL;
- }
-
- return single_result(g_switch(self, args, kwargs));
+ PyGreenlet* self = (PyGreenlet*)g;
+
+ if (!PyGreenlet_Check(self)) {
+ PyErr_BadArgument();
+ return NULL;
+ }
+
+ if (args == NULL) {
+ args = Py_BuildValue("()");
+ }
+ else {
+ Py_INCREF(args);
+ }
+
+ if (kwargs != NULL && PyDict_Check(kwargs)) {
+ Py_INCREF(kwargs);
+ }
+ else {
+ kwargs = NULL;
+ }
+
+ return single_result(g_switch(self, args, kwargs));
}
-static PyObject *
-PyGreenlet_Throw(PyGreenlet *self, PyObject *typ, PyObject *val, PyObject *tb)
+static PyObject*
+PyGreenlet_Throw(PyGreenlet* self, PyObject* typ, PyObject* val, PyObject* tb)
{
- if (!PyGreenlet_Check(self)) {
- PyErr_BadArgument();
- return NULL;
- }
- Py_INCREF(typ);
- Py_XINCREF(val);
- Py_XINCREF(tb);
- return throw_greenlet(self, typ, val, tb);
+ if (!PyGreenlet_Check(self)) {
+ PyErr_BadArgument();
+ return NULL;
+ }
+ Py_INCREF(typ);
+ Py_XINCREF(val);
+ Py_XINCREF(tb);
+ return throw_greenlet(self, typ, val, tb);
}
/** End C API ****************************************************************/
static PyMethodDef green_methods[] = {
- {"switch", (PyCFunction)green_switch,
- METH_VARARGS | METH_KEYWORDS, green_switch_doc},
- {"throw", (PyCFunction)green_throw, METH_VARARGS, green_throw_doc},
- {"__getstate__", (PyCFunction)green_getstate, METH_NOARGS, NULL},
- {NULL, NULL} /* sentinel */
+ {"switch",
+ (PyCFunction)green_switch,
+ METH_VARARGS | METH_KEYWORDS,
+ green_switch_doc},
+ {"throw", (PyCFunction)green_throw, METH_VARARGS, green_throw_doc},
+ {"__getstate__", (PyCFunction)green_getstate, METH_NOARGS, NULL},
+ {NULL, NULL} /* sentinel */
};
static PyGetSetDef green_getsets[] = {
- {"__dict__", (getter)green_getdict,
- (setter)green_setdict, /*XXX*/ NULL},
- {"run", (getter)green_getrun,
- (setter)green_setrun, /*XXX*/ NULL},
- {"parent", (getter)green_getparent,
- (setter)green_setparent, /*XXX*/ NULL},
- {"gr_frame", (getter)green_getframe,
- NULL, /*XXX*/ NULL},
- {"gr_context", (getter)green_getcontext,
- (setter)green_setcontext, /*XXX*/ NULL},
- {"dead", (getter)green_getdead,
- NULL, /*XXX*/ NULL},
- {"_stack_saved", (getter)green_get_stack_saved,
- NULL, /*XXX*/ NULL},
- {NULL}
-};
+ {"__dict__", (getter)green_getdict, (setter)green_setdict, /*XXX*/ NULL},
+ {"run", (getter)green_getrun, (setter)green_setrun, /*XXX*/ NULL},
+ {"parent", (getter)green_getparent, (setter)green_setparent, /*XXX*/ NULL},
+ {"gr_frame", (getter)green_getframe, NULL, /*XXX*/ NULL},
+ {"gr_context",
+ (getter)green_getcontext,
+ (setter)green_setcontext,
+ /*XXX*/ NULL},
+ {"dead", (getter)green_getdead, NULL, /*XXX*/ NULL},
+ {"_stack_saved", (getter)green_get_stack_saved, NULL, /*XXX*/ NULL},
+ {NULL}};
static PyNumberMethods green_as_number = {
- NULL, /* nb_add */
- NULL, /* nb_subtract */
- NULL, /* nb_multiply */
+ NULL, /* nb_add */
+ NULL, /* nb_subtract */
+ NULL, /* nb_multiply */
#if PY_MAJOR_VERSION < 3
- NULL, /* nb_divide */
+ NULL, /* nb_divide */
#endif
- NULL, /* nb_remainder */
- NULL, /* nb_divmod */
- NULL, /* nb_power */
- NULL, /* nb_negative */
- NULL, /* nb_positive */
- NULL, /* nb_absolute */
- (inquiry)green_bool, /* nb_bool */
+ NULL, /* nb_remainder */
+ NULL, /* nb_divmod */
+ NULL, /* nb_power */
+ NULL, /* nb_negative */
+ NULL, /* nb_positive */
+ NULL, /* nb_absolute */
+ (inquiry)green_bool, /* nb_bool */
};
-
PyTypeObject PyGreenlet_Type = {
- PyVarObject_HEAD_INIT(NULL, 0)
- "greenlet.greenlet", /* tp_name */
- sizeof(PyGreenlet), /* tp_basicsize */
- 0, /* tp_itemsize */
- /* methods */
- (destructor)green_dealloc, /* tp_dealloc */
- 0, /* tp_print */
- 0, /* tp_getattr */
- 0, /* tp_setattr */
- 0, /* tp_compare */
- 0, /* tp_repr */
- &green_as_number, /* tp_as _number*/
- 0, /* tp_as _sequence*/
- 0, /* tp_as _mapping*/
- 0, /* tp_hash */
- 0, /* tp_call */
- 0, /* tp_str */
- 0, /* tp_getattro */
- 0, /* tp_setattro */
- 0, /* tp_as_buffer*/
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | GREENLET_GC_FLAGS, /* tp_flags */
- "greenlet(run=None, parent=None) -> greenlet\n\n"
- "Creates a new greenlet object (without running it).\n\n"
- " - *run* -- The callable to invoke.\n"
- " - *parent* -- The parent greenlet. The default is the current "
- "greenlet.", /* tp_doc */
- (traverseproc)GREENLET_tp_traverse, /* tp_traverse */
- (inquiry)GREENLET_tp_clear, /* tp_clear */
- 0, /* tp_richcompare */
- offsetof(PyGreenlet, weakreflist), /* tp_weaklistoffset */
- 0, /* tp_iter */
- 0, /* tp_iternext */
- green_methods, /* tp_methods */
- 0, /* tp_members */
- green_getsets, /* tp_getset */
- 0, /* tp_base */
- 0, /* tp_dict */
- 0, /* tp_descr_get */
- 0, /* tp_descr_set */
- offsetof(PyGreenlet, dict), /* tp_dictoffset */
- (initproc)green_init, /* tp_init */
- GREENLET_tp_alloc, /* tp_alloc */
- green_new, /* tp_new */
- GREENLET_tp_free, /* tp_free */
- (inquiry)GREENLET_tp_is_gc, /* tp_is_gc */
+ PyVarObject_HEAD_INIT(NULL, 0) "greenlet.greenlet", /* tp_name */
+ sizeof(PyGreenlet), /* tp_basicsize */
+ 0, /* tp_itemsize */
+ /* methods */
+ (destructor)green_dealloc, /* tp_dealloc */
+ 0, /* tp_print */
+ 0, /* tp_getattr */
+ 0, /* tp_setattr */
+ 0, /* tp_compare */
+ 0, /* tp_repr */
+ &green_as_number, /* tp_as _number*/
+ 0, /* tp_as _sequence*/
+ 0, /* tp_as _mapping*/
+ 0, /* tp_hash */
+ 0, /* tp_call */
+ 0, /* tp_str */
+ 0, /* tp_getattro */
+ 0, /* tp_setattro */
+ 0, /* tp_as_buffer*/
+ Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE |
+ GREENLET_GC_FLAGS, /* tp_flags */
+ "greenlet(run=None, parent=None) -> greenlet\n\n"
+ "Creates a new greenlet object (without running it).\n\n"
+ " - *run* -- The callable to invoke.\n"
+ " - *parent* -- The parent greenlet. The default is the current "
+ "greenlet.", /* tp_doc */
+ (traverseproc)GREENLET_tp_traverse, /* tp_traverse */
+ (inquiry)GREENLET_tp_clear, /* tp_clear */
+ 0, /* tp_richcompare */
+ offsetof(PyGreenlet, weakreflist), /* tp_weaklistoffset */
+ 0, /* tp_iter */
+ 0, /* tp_iternext */
+ green_methods, /* tp_methods */
+ 0, /* tp_members */
+ green_getsets, /* tp_getset */
+ 0, /* tp_base */
+ 0, /* tp_dict */
+ 0, /* tp_descr_get */
+ 0, /* tp_descr_set */
+ offsetof(PyGreenlet, dict), /* tp_dictoffset */
+ (initproc)green_init, /* tp_init */
+ GREENLET_tp_alloc, /* tp_alloc */
+ green_new, /* tp_new */
+ GREENLET_tp_free, /* tp_free */
+ (inquiry)GREENLET_tp_is_gc, /* tp_is_gc */
};
PyDoc_STRVAR(mod_getcurrent_doc,
-"getcurrent() -> greenlet\n"
-"\n"
-"Returns the current greenlet (i.e. the one which called this function).\n"
-);
+ "getcurrent() -> greenlet\n"
+ "\n"
+ "Returns the current greenlet (i.e. the one which called this "
+ "function).\n");
-static PyObject* mod_getcurrent(PyObject* self)
+static PyObject*
+mod_getcurrent(PyObject* self)
{
- if (!STATE_OK)
- return NULL;
- Py_INCREF(ts_current);
- return (PyObject*) ts_current;
+ if (!STATE_OK) {
+ return NULL;
+ }
+ Py_INCREF(ts_current);
+ return (PyObject*)ts_current;
}
-
PyDoc_STRVAR(mod_settrace_doc,
-"settrace(callback) -> object\n"
-"\n"
-"Sets a new tracing function and returns the previous one.\n"
-);
-static PyObject* mod_settrace(PyObject* self, PyObject* args)
+ "settrace(callback) -> object\n"
+ "\n"
+ "Sets a new tracing function and returns the previous one.\n");
+static PyObject*
+mod_settrace(PyObject* self, PyObject* args)
{
- int err;
- PyObject* previous;
- PyObject* tracefunc;
- PyGreenlet* current;
- if (!PyArg_ParseTuple(args, "O", &tracefunc))
- return NULL;
- if (!STATE_OK)
- return NULL;
- current = ts_current;
- previous = PyDict_GetItem(current->run_info, ts_tracekey);
- if (previous == NULL)
- previous = Py_None;
- Py_INCREF(previous);
- if (tracefunc == Py_None)
- err = previous != Py_None ? PyDict_DelItem(current->run_info, ts_tracekey) : 0;
- else
- err = PyDict_SetItem(current->run_info, ts_tracekey, tracefunc);
- if (err < 0)
- Py_CLEAR(previous);
- return previous;
+ int err;
+ PyObject* previous;
+ PyObject* tracefunc;
+ PyGreenlet* current;
+ if (!PyArg_ParseTuple(args, "O", &tracefunc)) {
+ return NULL;
+ }
+ if (!STATE_OK) {
+ return NULL;
+ }
+ current = ts_current;
+ previous = PyDict_GetItem(current->run_info, ts_tracekey);
+ if (previous == NULL) {
+ previous = Py_None;
+ }
+ Py_INCREF(previous);
+ if (tracefunc == Py_None) {
+ err = previous != Py_None ?
+ PyDict_DelItem(current->run_info, ts_tracekey) :
+ 0;
+ }
+ else {
+ err = PyDict_SetItem(current->run_info, ts_tracekey, tracefunc);
+ }
+ if (err < 0) {
+ Py_CLEAR(previous);
+ }
+ return previous;
}
PyDoc_STRVAR(mod_gettrace_doc,
-"gettrace() -> object\n"
-"\n"
-"Returns the currently set tracing function, or None.\n"
-);
+ "gettrace() -> object\n"
+ "\n"
+ "Returns the currently set tracing function, or None.\n");
-static PyObject* mod_gettrace(PyObject* self)
+static PyObject*
+mod_gettrace(PyObject* self)
{
- PyObject* tracefunc;
- if (!STATE_OK)
- return NULL;
- tracefunc = PyDict_GetItem(ts_current->run_info, ts_tracekey);
- if (tracefunc == NULL)
- tracefunc = Py_None;
- Py_INCREF(tracefunc);
- return tracefunc;
+ PyObject* tracefunc;
+ if (!STATE_OK) {
+ return NULL;
+ }
+ tracefunc = PyDict_GetItem(ts_current->run_info, ts_tracekey);
+ if (tracefunc == NULL) {
+ tracefunc = Py_None;
+ }
+ Py_INCREF(tracefunc);
+ return tracefunc;
}
-
static PyMethodDef GreenMethods[] = {
- {"getcurrent", (PyCFunction)mod_getcurrent, METH_NOARGS, mod_getcurrent_doc},
- {"settrace", (PyCFunction)mod_settrace, METH_VARARGS, mod_settrace_doc},
- {"gettrace", (PyCFunction)mod_gettrace, METH_NOARGS, mod_gettrace_doc},
- {NULL, NULL} /* Sentinel */
+ {"getcurrent",
+ (PyCFunction)mod_getcurrent,
+ METH_NOARGS,
+ mod_getcurrent_doc},
+ {"settrace", (PyCFunction)mod_settrace, METH_VARARGS, mod_settrace_doc},
+ {"gettrace", (PyCFunction)mod_gettrace, METH_NOARGS, mod_gettrace_doc},
+ {NULL, NULL} /* Sentinel */
};
static char* copy_on_greentype[] = {
- "getcurrent",
- "error",
- "GreenletExit",
- "settrace",
- "gettrace",
- NULL
-};
+ "getcurrent", "error", "GreenletExit", "settrace", "gettrace", NULL};
#if PY_MAJOR_VERSION >= 3
-#define INITERROR return NULL
+# define INITERROR return NULL
static struct PyModuleDef greenlet_module_def = {
- PyModuleDef_HEAD_INIT,
- "greenlet._greenlet",
- NULL,
- -1,
- GreenMethods,
+ PyModuleDef_HEAD_INIT,
+ "greenlet._greenlet",
+ NULL,
+ -1,
+ GreenMethods,
};
PyMODINIT_FUNC
PyInit__greenlet(void)
#else
-#define INITERROR return
+# define INITERROR return
PyMODINIT_FUNC
init_greenlet(void)
#endif
{
- PyObject* m = NULL;
- char** p = NULL;
- PyObject *c_api_object;
- static void *_PyGreenlet_API[PyGreenlet_API_pointers];
+ PyObject* m = NULL;
+ char** p = NULL;
+ PyObject* c_api_object;
+ static void* _PyGreenlet_API[PyGreenlet_API_pointers];
- GREENLET_NOINLINE_INIT();
+ GREENLET_NOINLINE_INIT();
#if PY_MAJOR_VERSION >= 3
- m = PyModule_Create(&greenlet_module_def);
+ m = PyModule_Create(&greenlet_module_def);
#else
- m = Py_InitModule("greenlet._greenlet", GreenMethods);
+ m = Py_InitModule("greenlet._greenlet", GreenMethods);
#endif
- if (m == NULL)
- {
- INITERROR;
- }
+ if (m == NULL) {
+ INITERROR;
+ }
#if PY_MAJOR_VERSION >= 3
-# define Greenlet_Intern PyUnicode_InternFromString
+# define Greenlet_Intern PyUnicode_InternFromString
#else
-# define Greenlet_Intern PyString_InternFromString
+# define Greenlet_Intern PyString_InternFromString
#endif
- ts_curkey = Greenlet_Intern("__greenlet_ts_curkey");
- ts_delkey = Greenlet_Intern("__greenlet_ts_delkey");
- ts_tracekey = Greenlet_Intern("__greenlet_ts_tracekey");
- ts_event_switch = Greenlet_Intern("switch");
- ts_event_throw = Greenlet_Intern("throw");
+ ts_curkey = Greenlet_Intern("__greenlet_ts_curkey");
+ ts_delkey = Greenlet_Intern("__greenlet_ts_delkey");
+ ts_tracekey = Greenlet_Intern("__greenlet_ts_tracekey");
+ ts_event_switch = Greenlet_Intern("switch");
+ ts_event_throw = Greenlet_Intern("throw");
#undef Greenlet_Intern
- if (ts_curkey == NULL || ts_delkey == NULL)
- {
- INITERROR;
- }
- if (PyType_Ready(&PyGreenlet_Type) < 0)
- {
- INITERROR;
- }
- PyExc_GreenletError = PyErr_NewException("greenlet.error", NULL, NULL);
- if (PyExc_GreenletError == NULL)
- {
- INITERROR;
- }
- PyExc_GreenletExit = PyErr_NewException("greenlet.GreenletExit",
- PyExc_BaseException, NULL);
- if (PyExc_GreenletExit == NULL)
- {
- INITERROR;
- }
-
- ts_empty_tuple = PyTuple_New(0);
- if (ts_empty_tuple == NULL)
- {
- INITERROR;
- }
-
- ts_empty_dict = PyDict_New();
- if (ts_empty_dict == NULL)
- {
- INITERROR;
- }
-
- ts_current = green_create_main();
- if (ts_current == NULL)
- {
- INITERROR;
- }
-
- Py_INCREF(&PyGreenlet_Type);
- PyModule_AddObject(m, "greenlet", (PyObject*) &PyGreenlet_Type);
- Py_INCREF(PyExc_GreenletError);
- PyModule_AddObject(m, "error", PyExc_GreenletError);
- Py_INCREF(PyExc_GreenletExit);
- PyModule_AddObject(m, "GreenletExit", PyExc_GreenletExit);
-
- PyModule_AddObject(m, "GREENLET_USE_GC",
- PyBool_FromLong(1));
- PyModule_AddObject(m, "GREENLET_USE_TRACING",
- PyBool_FromLong(1));
- PyModule_AddObject(m, "GREENLET_USE_CONTEXT_VARS",
- PyBool_FromLong(GREENLET_PY37));
-
- /* also publish module-level data as attributes of the greentype. */
- /* XXX: Why? */
- for (p=copy_on_greentype; *p; p++) {
- PyObject* o = PyObject_GetAttrString(m, *p);
- if (!o) continue;
- PyDict_SetItemString(PyGreenlet_Type.tp_dict, *p, o);
- Py_DECREF(o);
- }
-
- /*
- * Expose C API
- */
-
- /* types */
- _PyGreenlet_API[PyGreenlet_Type_NUM] = (void *) &PyGreenlet_Type;
-
- /* exceptions */
- _PyGreenlet_API[PyExc_GreenletError_NUM] = (void *) PyExc_GreenletError;
- _PyGreenlet_API[PyExc_GreenletExit_NUM] = (void *) PyExc_GreenletExit;
-
- /* methods */
- _PyGreenlet_API[PyGreenlet_New_NUM] = (void *) PyGreenlet_New;
- _PyGreenlet_API[PyGreenlet_GetCurrent_NUM] =
- (void *) PyGreenlet_GetCurrent;
- _PyGreenlet_API[PyGreenlet_Throw_NUM] = (void *) PyGreenlet_Throw;
- _PyGreenlet_API[PyGreenlet_Switch_NUM] = (void *) PyGreenlet_Switch;
- _PyGreenlet_API[PyGreenlet_SetParent_NUM] =
- (void *) PyGreenlet_SetParent;
-
- /* XXX: Note that our module name is ``greenlet._greenlet``, but for
- backwards compatibility with existing C code, we need the _C_API to
- be directly in greenlet.
- */
- c_api_object = PyCapsule_New((void *) _PyGreenlet_API, "greenlet._C_API", NULL);
- if (c_api_object != NULL)
- {
- PyModule_AddObject(m, "_C_API", c_api_object);
- }
+ if (ts_curkey == NULL || ts_delkey == NULL) {
+ INITERROR;
+ }
+ if (PyType_Ready(&PyGreenlet_Type) < 0) {
+ INITERROR;
+ }
+ PyExc_GreenletError = PyErr_NewException("greenlet.error", NULL, NULL);
+ if (PyExc_GreenletError == NULL) {
+ INITERROR;
+ }
+ PyExc_GreenletExit =
+ PyErr_NewException("greenlet.GreenletExit", PyExc_BaseException, NULL);
+ if (PyExc_GreenletExit == NULL) {
+ INITERROR;
+ }
+
+ ts_empty_tuple = PyTuple_New(0);
+ if (ts_empty_tuple == NULL) {
+ INITERROR;
+ }
+
+ ts_empty_dict = PyDict_New();
+ if (ts_empty_dict == NULL) {
+ INITERROR;
+ }
+
+ ts_current = green_create_main();
+ if (ts_current == NULL) {
+ INITERROR;
+ }
+
+ Py_INCREF(&PyGreenlet_Type);
+ PyModule_AddObject(m, "greenlet", (PyObject*)&PyGreenlet_Type);
+ Py_INCREF(PyExc_GreenletError);
+ PyModule_AddObject(m, "error", PyExc_GreenletError);
+ Py_INCREF(PyExc_GreenletExit);
+ PyModule_AddObject(m, "GreenletExit", PyExc_GreenletExit);
+
+ PyModule_AddObject(m, "GREENLET_USE_GC", PyBool_FromLong(1));
+ PyModule_AddObject(m, "GREENLET_USE_TRACING", PyBool_FromLong(1));
+ PyModule_AddObject(
+ m, "GREENLET_USE_CONTEXT_VARS", PyBool_FromLong(GREENLET_PY37));
+
+ /* also publish module-level data as attributes of the greentype. */
+ /* XXX: Why? */
+ for (p = copy_on_greentype; *p; p++) {
+ PyObject* o = PyObject_GetAttrString(m, *p);
+ if (!o) {
+ continue;
+ }
+ PyDict_SetItemString(PyGreenlet_Type.tp_dict, *p, o);
+ Py_DECREF(o);
+ }
+
+ /*
+ * Expose C API
+ */
+
+ /* types */
+ _PyGreenlet_API[PyGreenlet_Type_NUM] = (void*)&PyGreenlet_Type;
+
+ /* exceptions */
+ _PyGreenlet_API[PyExc_GreenletError_NUM] = (void*)PyExc_GreenletError;
+ _PyGreenlet_API[PyExc_GreenletExit_NUM] = (void*)PyExc_GreenletExit;
+
+ /* methods */
+ _PyGreenlet_API[PyGreenlet_New_NUM] = (void*)PyGreenlet_New;
+ _PyGreenlet_API[PyGreenlet_GetCurrent_NUM] = (void*)PyGreenlet_GetCurrent;
+ _PyGreenlet_API[PyGreenlet_Throw_NUM] = (void*)PyGreenlet_Throw;
+ _PyGreenlet_API[PyGreenlet_Switch_NUM] = (void*)PyGreenlet_Switch;
+ _PyGreenlet_API[PyGreenlet_SetParent_NUM] = (void*)PyGreenlet_SetParent;
+
+ /* XXX: Note that our module name is ``greenlet._greenlet``, but for
+ backwards compatibility with existing C code, we need the _C_API to
+ be directly in greenlet.
+ */
+ c_api_object =
+ PyCapsule_New((void*)_PyGreenlet_API, "greenlet._C_API", NULL);
+ if (c_api_object != NULL) {
+ PyModule_AddObject(m, "_C_API", c_api_object);
+ }
#if PY_MAJOR_VERSION >= 3
- return m;
+ return m;
#endif
}
#ifdef __clang__
-#pragma clang diagnostic pop
+# pragma clang diagnostic pop
#endif
diff --git a/src/greenlet/greenlet.h b/src/greenlet/greenlet.h
index d31cae7..1397782 100644
--- a/src/greenlet/greenlet.h
+++ b/src/greenlet/greenlet.h
@@ -1,4 +1,4 @@
-/* vim:set noet ts=8 sw=8 : */
+/* -*- indent-tabs-mode: nil; tab-width: 4; -*- */
/* Greenlet object interface */
@@ -14,82 +14,82 @@ extern "C" {
/* This is deprecated and undocumented. It does not change. */
#define GREENLET_VERSION "1.0.0"
-
typedef struct _greenlet {
- PyObject_HEAD
- char* stack_start;
- char* stack_stop;
- char* stack_copy;
- intptr_t stack_saved;
- struct _greenlet* stack_prev;
- struct _greenlet* parent;
- PyObject* run_info;
- struct _frame* top_frame;
- int recursion_depth;
- PyObject* weakreflist;
+ PyObject_HEAD
+ char* stack_start;
+ char* stack_stop;
+ char* stack_copy;
+ intptr_t stack_saved;
+ struct _greenlet* stack_prev;
+ struct _greenlet* parent;
+ PyObject* run_info;
+ struct _frame* top_frame;
+ int recursion_depth;
+ PyObject* weakreflist;
#if PY_VERSION_HEX >= 0x030700A3
- _PyErr_StackItem* exc_info;
- _PyErr_StackItem exc_state;
+ _PyErr_StackItem* exc_info;
+ _PyErr_StackItem exc_state;
#else
- PyObject* exc_type;
- PyObject* exc_value;
- PyObject* exc_traceback;
+ PyObject* exc_type;
+ PyObject* exc_value;
+ PyObject* exc_traceback;
#endif
- PyObject* dict;
+ PyObject* dict;
#if PY_VERSION_HEX >= 0x030700A3
- PyObject* context;
+ PyObject* context;
#endif
} PyGreenlet;
-#define PyGreenlet_Check(op) PyObject_TypeCheck(op, &PyGreenlet_Type)
-#define PyGreenlet_MAIN(op) (((PyGreenlet*)(op))->stack_stop == (char*) -1)
-#define PyGreenlet_STARTED(op) (((PyGreenlet*)(op))->stack_stop != NULL)
-#define PyGreenlet_ACTIVE(op) (((PyGreenlet*)(op))->stack_start != NULL)
+#define PyGreenlet_Check(op) PyObject_TypeCheck(op, &PyGreenlet_Type)
+#define PyGreenlet_MAIN(op) (((PyGreenlet*)(op))->stack_stop == (char*)-1)
+#define PyGreenlet_STARTED(op) (((PyGreenlet*)(op))->stack_stop != NULL)
+#define PyGreenlet_ACTIVE(op) (((PyGreenlet*)(op))->stack_start != NULL)
#define PyGreenlet_GET_PARENT(op) (((PyGreenlet*)(op))->parent)
/* C API functions */
/* Total number of symbols that are exported */
-#define PyGreenlet_API_pointers 8
+#define PyGreenlet_API_pointers 8
-#define PyGreenlet_Type_NUM 0
-#define PyExc_GreenletError_NUM 1
-#define PyExc_GreenletExit_NUM 2
+#define PyGreenlet_Type_NUM 0
+#define PyExc_GreenletError_NUM 1
+#define PyExc_GreenletExit_NUM 2
-#define PyGreenlet_New_NUM 3
+#define PyGreenlet_New_NUM 3
#define PyGreenlet_GetCurrent_NUM 4
-#define PyGreenlet_Throw_NUM 5
-#define PyGreenlet_Switch_NUM 6
-#define PyGreenlet_SetParent_NUM 7
+#define PyGreenlet_Throw_NUM 5
+#define PyGreenlet_Switch_NUM 6
+#define PyGreenlet_SetParent_NUM 7
#ifndef GREENLET_MODULE
/* This section is used by modules that uses the greenlet C API */
-static void **_PyGreenlet_API = NULL;
+static void** _PyGreenlet_API = NULL;
-#define PyGreenlet_Type (*(PyTypeObject *) _PyGreenlet_API[PyGreenlet_Type_NUM])
+# define PyGreenlet_Type \
+ (*(PyTypeObject*)_PyGreenlet_API[PyGreenlet_Type_NUM])
-#define PyExc_GreenletError \
- ((PyObject *) _PyGreenlet_API[PyExc_GreenletError_NUM])
+# define PyExc_GreenletError \
+ ((PyObject*)_PyGreenlet_API[PyExc_GreenletError_NUM])
-#define PyExc_GreenletExit \
- ((PyObject *) _PyGreenlet_API[PyExc_GreenletExit_NUM])
+# define PyExc_GreenletExit \
+ ((PyObject*)_PyGreenlet_API[PyExc_GreenletExit_NUM])
/*
* PyGreenlet_New(PyObject *args)
*
* greenlet.greenlet(run, parent=None)
*/
-#define PyGreenlet_New \
- (* (PyGreenlet * (*)(PyObject *run, PyGreenlet *parent)) \
- _PyGreenlet_API[PyGreenlet_New_NUM])
+# define PyGreenlet_New \
+ (*(PyGreenlet * (*)(PyObject * run, PyGreenlet * parent)) \
+ _PyGreenlet_API[PyGreenlet_New_NUM])
/*
* PyGreenlet_GetCurrent(void)
*
* greenlet.getcurrent()
*/
-#define PyGreenlet_GetCurrent \
- (* (PyGreenlet * (*)(void)) _PyGreenlet_API[PyGreenlet_GetCurrent_NUM])
+# define PyGreenlet_GetCurrent \
+ (*(PyGreenlet * (*)(void)) _PyGreenlet_API[PyGreenlet_GetCurrent_NUM])
/*
* PyGreenlet_Throw(
@@ -100,37 +100,40 @@ static void **_PyGreenlet_API = NULL;
*
* g.throw(...)
*/
-#define PyGreenlet_Throw \
- (* (PyObject * (*) \
- (PyGreenlet *self, PyObject *typ, PyObject *val, PyObject *tb)) \
- _PyGreenlet_API[PyGreenlet_Throw_NUM])
+# define PyGreenlet_Throw \
+ (*(PyObject * (*)(PyGreenlet * self, \
+ PyObject * typ, \
+ PyObject * val, \
+ PyObject * tb)) \
+ _PyGreenlet_API[PyGreenlet_Throw_NUM])
/*
* PyGreenlet_Switch(PyGreenlet *greenlet, PyObject *args)
*
* g.switch(*args, **kwargs)
*/
-#define PyGreenlet_Switch \
- (* (PyObject * (*)(PyGreenlet *greenlet, PyObject *args, PyObject *kwargs)) \
- _PyGreenlet_API[PyGreenlet_Switch_NUM])
+# define PyGreenlet_Switch \
+ (*(PyObject * \
+ (*)(PyGreenlet * greenlet, PyObject * args, PyObject * kwargs)) \
+ _PyGreenlet_API[PyGreenlet_Switch_NUM])
/*
* PyGreenlet_SetParent(PyObject *greenlet, PyObject *new_parent)
*
* g.parent = new_parent
*/
-#define PyGreenlet_SetParent \
- (* (int (*)(PyGreenlet *greenlet, PyGreenlet *nparent)) \
- _PyGreenlet_API[PyGreenlet_SetParent_NUM])
+# define PyGreenlet_SetParent \
+ (*(int (*)(PyGreenlet * greenlet, PyGreenlet * nparent)) \
+ _PyGreenlet_API[PyGreenlet_SetParent_NUM])
/* Macro that imports greenlet and initializes C API */
/* NOTE: This has actually moved to ``greenlet._greenlet._C_API``, but we
keep the older definition to be sure older code that might have a copy of
the header still works. */
-#define PyGreenlet_Import() \
-{ \
- _PyGreenlet_API = (void**)PyCapsule_Import("greenlet._C_API", 0); \
-}
+# define PyGreenlet_Import() \
+ { \
+ _PyGreenlet_API = (void**)PyCapsule_Import("greenlet._C_API", 0); \
+ }
#endif /* GREENLET_MODULE */
diff --git a/src/greenlet/tests/_test_extension.c b/src/greenlet/tests/_test_extension.c
index ec8b70c..4fe087d 100644
--- a/src/greenlet/tests/_test_extension.c
+++ b/src/greenlet/tests/_test_extension.c
@@ -5,204 +5,212 @@
#include "../greenlet.h"
#ifndef Py_RETURN_NONE
-#define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
+# define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
#endif
#define TEST_MODULE_NAME "_test_extension"
-static PyObject *
-test_switch(PyObject *self, PyObject *greenlet)
+static PyObject*
+test_switch(PyObject* self, PyObject* greenlet)
{
- PyObject *result = NULL;
-
- if (greenlet == NULL || !PyGreenlet_Check(greenlet)) {
- PyErr_BadArgument();
- return NULL;
- }
-
- result = PyGreenlet_Switch((PyGreenlet *) greenlet, NULL, NULL);
- if (result == NULL) {
- if (!PyErr_Occurred()) {
- PyErr_SetString(
- PyExc_AssertionError,
- "greenlet.switch() failed for some reason.");
- }
- return NULL;
- }
- Py_INCREF(result);
- return result;
+ PyObject* result = NULL;
+
+ if (greenlet == NULL || !PyGreenlet_Check(greenlet)) {
+ PyErr_BadArgument();
+ return NULL;
+ }
+
+ result = PyGreenlet_Switch((PyGreenlet*)greenlet, NULL, NULL);
+ if (result == NULL) {
+ if (!PyErr_Occurred()) {
+ PyErr_SetString(PyExc_AssertionError,
+ "greenlet.switch() failed for some reason.");
+ }
+ return NULL;
+ }
+ Py_INCREF(result);
+ return result;
}
-static PyObject *
-test_switch_kwargs(PyObject *self, PyObject *args, PyObject *kwargs)
+static PyObject*
+test_switch_kwargs(PyObject* self, PyObject* args, PyObject* kwargs)
{
- PyGreenlet *g = NULL;
- PyObject *result = NULL;
-
- PyArg_ParseTuple(args, "O!", &PyGreenlet_Type, &g);
-
- if (g == NULL || !PyGreenlet_Check(g)) {
- PyErr_BadArgument();
- return NULL;
- }
-
- result = PyGreenlet_Switch(g, NULL, kwargs);
- if (result == NULL) {
- if (!PyErr_Occurred()) {
- PyErr_SetString(
- PyExc_AssertionError,
- "greenlet.switch() failed for some reason.");
- }
- return NULL;
- }
- Py_XINCREF(result);
- return result;
+ PyGreenlet* g = NULL;
+ PyObject* result = NULL;
+
+ PyArg_ParseTuple(args, "O!", &PyGreenlet_Type, &g);
+
+ if (g == NULL || !PyGreenlet_Check(g)) {
+ PyErr_BadArgument();
+ return NULL;
+ }
+
+ result = PyGreenlet_Switch(g, NULL, kwargs);
+ if (result == NULL) {
+ if (!PyErr_Occurred()) {
+ PyErr_SetString(PyExc_AssertionError,
+ "greenlet.switch() failed for some reason.");
+ }
+ return NULL;
+ }
+ Py_XINCREF(result);
+ return result;
}
-static PyObject *
-test_getcurrent(PyObject *self)
+static PyObject*
+test_getcurrent(PyObject* self)
{
- PyGreenlet *g = PyGreenlet_GetCurrent();
- if (g == NULL || !PyGreenlet_Check(g) || !PyGreenlet_ACTIVE(g)) {
- PyErr_SetString(
- PyExc_AssertionError,
- "getcurrent() returned an invalid greenlet");
- Py_XDECREF(g);
- return NULL;
- }
- Py_DECREF(g);
- Py_RETURN_NONE;
+ PyGreenlet* g = PyGreenlet_GetCurrent();
+ if (g == NULL || !PyGreenlet_Check(g) || !PyGreenlet_ACTIVE(g)) {
+ PyErr_SetString(PyExc_AssertionError,
+ "getcurrent() returned an invalid greenlet");
+ Py_XDECREF(g);
+ return NULL;
+ }
+ Py_DECREF(g);
+ Py_RETURN_NONE;
}
-static PyObject *
-test_setparent(PyObject *self, PyObject *arg)
+static PyObject*
+test_setparent(PyObject* self, PyObject* arg)
{
- PyGreenlet *current;
- PyGreenlet *greenlet = NULL;
-
- if (arg == NULL || !PyGreenlet_Check(arg))
- {
- PyErr_BadArgument();
- return NULL;
- }
- if ((current = PyGreenlet_GetCurrent()) == NULL) {
- return NULL;
- }
- greenlet = (PyGreenlet *) arg;
- if (PyGreenlet_SetParent(greenlet, current)) {
- Py_DECREF(current);
- return NULL;
- }
- Py_DECREF(current);
- if (PyGreenlet_Switch(greenlet, NULL, NULL) == NULL) {
- return NULL;
- }
- Py_RETURN_NONE;
+ PyGreenlet* current;
+ PyGreenlet* greenlet = NULL;
+
+ if (arg == NULL || !PyGreenlet_Check(arg)) {
+ PyErr_BadArgument();
+ return NULL;
+ }
+ if ((current = PyGreenlet_GetCurrent()) == NULL) {
+ return NULL;
+ }
+ greenlet = (PyGreenlet*)arg;
+ if (PyGreenlet_SetParent(greenlet, current)) {
+ Py_DECREF(current);
+ return NULL;
+ }
+ Py_DECREF(current);
+ if (PyGreenlet_Switch(greenlet, NULL, NULL) == NULL) {
+ return NULL;
+ }
+ Py_RETURN_NONE;
}
-static PyObject *
-test_new_greenlet(PyObject *self, PyObject *callable)
+static PyObject*
+test_new_greenlet(PyObject* self, PyObject* callable)
{
- PyObject *result = NULL;
- PyGreenlet *greenlet = PyGreenlet_New(callable, NULL);
+ PyObject* result = NULL;
+ PyGreenlet* greenlet = PyGreenlet_New(callable, NULL);
- if (!greenlet) {
- return NULL;
- }
+ if (!greenlet) {
+ return NULL;
+ }
- result = PyGreenlet_Switch(greenlet, NULL, NULL);
- if (result == NULL) {
- return NULL;
- }
+ result = PyGreenlet_Switch(greenlet, NULL, NULL);
+ if (result == NULL) {
+ return NULL;
+ }
- Py_INCREF(result);
- return result;
+ Py_INCREF(result);
+ return result;
}
-static PyObject *
-test_raise_dead_greenlet(PyObject *self)
+static PyObject*
+test_raise_dead_greenlet(PyObject* self)
{
- PyErr_SetString(PyExc_GreenletExit, "test GreenletExit exception.");
- return NULL;
+ PyErr_SetString(PyExc_GreenletExit, "test GreenletExit exception.");
+ return NULL;
}
-static PyObject *
-test_raise_greenlet_error(PyObject *self)
+static PyObject*
+test_raise_greenlet_error(PyObject* self)
{
- PyErr_SetString(PyExc_GreenletError, "test greenlet.error exception");
- return NULL;
+ PyErr_SetString(PyExc_GreenletError, "test greenlet.error exception");
+ return NULL;
}
-static PyObject *
-test_throw(PyObject *self, PyGreenlet *g)
+static PyObject*
+test_throw(PyObject* self, PyGreenlet* g)
{
- const char msg[] = "take that sucka!";
- PyObject *msg_obj = Py_BuildValue("s", msg);
- PyGreenlet_Throw(g, PyExc_ValueError, msg_obj, NULL);
- Py_DECREF(msg_obj);
- Py_RETURN_NONE;
+ const char msg[] = "take that sucka!";
+ PyObject* msg_obj = Py_BuildValue("s", msg);
+ PyGreenlet_Throw(g, PyExc_ValueError, msg_obj, NULL);
+ Py_DECREF(msg_obj);
+ Py_RETURN_NONE;
}
static PyMethodDef test_methods[] = {
- {"test_switch", (PyCFunction) test_switch, METH_O,
- "Switch to the provided greenlet sending provided arguments, and \n"
- "return the results."},
- {"test_switch_kwargs", (PyCFunction) test_switch_kwargs,
- METH_VARARGS | METH_KEYWORDS,
- "Switch to the provided greenlet sending the provided keyword args."},
- {"test_getcurrent", (PyCFunction) test_getcurrent, METH_NOARGS,
- "Test PyGreenlet_GetCurrent()"},
- {"test_setparent", (PyCFunction) test_setparent, METH_O,
- "Se the parent of the provided greenlet and switch to it."},
- {"test_new_greenlet", (PyCFunction) test_new_greenlet, METH_O,
- "Test PyGreenlet_New()"},
- {"test_raise_dead_greenlet", (PyCFunction) test_raise_dead_greenlet,
- METH_NOARGS, "Just raise greenlet.GreenletExit"},
- {"test_raise_greenlet_error", (PyCFunction) test_raise_greenlet_error,
- METH_NOARGS, "Just raise greenlet.error"},
- {"test_throw", (PyCFunction) test_throw, METH_O,
- "Throw a ValueError at the provided greenlet"},
- {NULL, NULL, 0, NULL}
-};
+ {"test_switch",
+ (PyCFunction)test_switch,
+ METH_O,
+ "Switch to the provided greenlet sending provided arguments, and \n"
+ "return the results."},
+ {"test_switch_kwargs",
+ (PyCFunction)test_switch_kwargs,
+ METH_VARARGS | METH_KEYWORDS,
+ "Switch to the provided greenlet sending the provided keyword args."},
+ {"test_getcurrent",
+ (PyCFunction)test_getcurrent,
+ METH_NOARGS,
+ "Test PyGreenlet_GetCurrent()"},
+ {"test_setparent",
+ (PyCFunction)test_setparent,
+ METH_O,
+ "Se the parent of the provided greenlet and switch to it."},
+ {"test_new_greenlet",
+ (PyCFunction)test_new_greenlet,
+ METH_O,
+ "Test PyGreenlet_New()"},
+ {"test_raise_dead_greenlet",
+ (PyCFunction)test_raise_dead_greenlet,
+ METH_NOARGS,
+ "Just raise greenlet.GreenletExit"},
+ {"test_raise_greenlet_error",
+ (PyCFunction)test_raise_greenlet_error,
+ METH_NOARGS,
+ "Just raise greenlet.error"},
+ {"test_throw",
+ (PyCFunction)test_throw,
+ METH_O,
+ "Throw a ValueError at the provided greenlet"},
+ {NULL, NULL, 0, NULL}};
#if PY_MAJOR_VERSION >= 3
-#define INITERROR return NULL
-
-static struct PyModuleDef moduledef = {
- PyModuleDef_HEAD_INIT,
- TEST_MODULE_NAME,
- NULL,
- 0,
- test_methods,
- NULL,
- NULL,
- NULL,
- NULL
-};
+# define INITERROR return NULL
+
+static struct PyModuleDef moduledef = {PyModuleDef_HEAD_INIT,
+ TEST_MODULE_NAME,
+ NULL,
+ 0,
+ test_methods,
+ NULL,
+ NULL,
+ NULL,
+ NULL};
PyMODINIT_FUNC
PyInit__test_extension(void)
#else
-#define INITERROR return
+# define INITERROR return
PyMODINIT_FUNC
init_test_extension(void)
#endif
{
- PyObject *module = NULL;
+ PyObject* module = NULL;
#if PY_MAJOR_VERSION >= 3
- module = PyModule_Create(&moduledef);
+ module = PyModule_Create(&moduledef);
#else
- module = Py_InitModule(TEST_MODULE_NAME, test_methods);
+ module = Py_InitModule(TEST_MODULE_NAME, test_methods);
#endif
- if (module == NULL) {
- INITERROR;
- }
+ if (module == NULL) {
+ INITERROR;
+ }
- PyGreenlet_Import();
+ PyGreenlet_Import();
#if PY_MAJOR_VERSION >= 3
- return module;
+ return module;
#endif
}
diff --git a/src/greenlet/tests/_test_extension_cpp.cpp b/src/greenlet/tests/_test_extension_cpp.cpp
index fc3adad..72e3d81 100644
--- a/src/greenlet/tests/_test_extension_cpp.cpp
+++ b/src/greenlet/tests/_test_extension_cpp.cpp
@@ -4,47 +4,51 @@
#include "../greenlet.h"
-struct exception_t
-{
- int depth;
- exception_t(int depth) : depth(depth) { }
+struct exception_t {
+ int depth;
+ exception_t(int depth) : depth(depth) {}
};
/* Functions are called via pointers to prevent inlining */
static void (*p_test_exception_throw)(int depth);
static PyObject* (*p_test_exception_switch_recurse)(int depth, int left);
-static void test_exception_throw(int depth)
+static void
+test_exception_throw(int depth)
{
- throw exception_t(depth);
+ throw exception_t(depth);
}
-static PyObject* test_exception_switch_recurse(int depth, int left)
+static PyObject*
+test_exception_switch_recurse(int depth, int left)
{
- if (left > 0) {
- return p_test_exception_switch_recurse(depth, left - 1);
- }
-
- PyObject* result = NULL;
- PyGreenlet* self = PyGreenlet_GetCurrent();
- if (self == NULL)
- return NULL;
-
- try {
- PyGreenlet_Switch(self->parent, NULL, NULL);
- p_test_exception_throw(depth);
- PyErr_SetString(PyExc_RuntimeError, "throwing C++ exception didn't work");
- } catch(exception_t& e) {
- if (e.depth != depth)
- PyErr_SetString(PyExc_AssertionError, "depth mismatch");
- else
- result = PyLong_FromLong(depth);
- } catch(...) {
- PyErr_SetString(PyExc_RuntimeError, "unexpected C++ exception");
- }
-
- Py_DECREF(self);
- return result;
+ if (left > 0) {
+ return p_test_exception_switch_recurse(depth, left - 1);
+ }
+
+ PyObject* result = NULL;
+ PyGreenlet* self = PyGreenlet_GetCurrent();
+ if (self == NULL)
+ return NULL;
+
+ try {
+ PyGreenlet_Switch(self->parent, NULL, NULL);
+ p_test_exception_throw(depth);
+ PyErr_SetString(PyExc_RuntimeError,
+ "throwing C++ exception didn't work");
+ }
+ catch (exception_t& e) {
+ if (e.depth != depth)
+ PyErr_SetString(PyExc_AssertionError, "depth mismatch");
+ else
+ result = PyLong_FromLong(depth);
+ }
+ catch (...) {
+ PyErr_SetString(PyExc_RuntimeError, "unexpected C++ exception");
+ }
+
+ Py_DECREF(self);
+ return result;
}
/* test_exception_switch(int depth)
@@ -53,66 +57,65 @@ static PyObject* test_exception_switch_recurse(int depth, int left)
* - throws an exception that (expected to be caught in the same function)
* - verifies depth matches (exceptions shouldn't be caught in other greenlets)
*/
-static PyObject *
-test_exception_switch(PyObject *self, PyObject *args)
+static PyObject*
+test_exception_switch(PyObject* self, PyObject* args)
{
- int depth;
- if (!PyArg_ParseTuple(args, "i", &depth))
- return NULL;
- return p_test_exception_switch_recurse(depth, depth);
+ int depth;
+ if (!PyArg_ParseTuple(args, "i", &depth))
+ return NULL;
+ return p_test_exception_switch_recurse(depth, depth);
}
static PyMethodDef test_methods[] = {
- {"test_exception_switch", (PyCFunction)&test_exception_switch, METH_VARARGS,
- "Switches to parent twice, to test exception handling and greenlet switching."},
- {NULL, NULL, 0, NULL}
-};
-
+ {"test_exception_switch",
+ (PyCFunction)&test_exception_switch,
+ METH_VARARGS,
+ "Switches to parent twice, to test exception handling and greenlet "
+ "switching."},
+ {NULL, NULL, 0, NULL}};
#if PY_MAJOR_VERSION >= 3
-#define INITERROR return NULL
-
-static struct PyModuleDef moduledef = {
- PyModuleDef_HEAD_INIT,
- "greenlet.tests._test_extension_cpp",
- NULL,
- 0,
- test_methods,
- NULL,
- NULL,
- NULL,
- NULL
-};
+# define INITERROR return NULL
+
+static struct PyModuleDef moduledef = {PyModuleDef_HEAD_INIT,
+ "greenlet.tests._test_extension_cpp",
+ NULL,
+ 0,
+ test_methods,
+ NULL,
+ NULL,
+ NULL,
+ NULL};
PyMODINIT_FUNC
PyInit__test_extension_cpp(void)
#else
-#define INITERROR return
+# define INITERROR return
PyMODINIT_FUNC
init_test_extension_cpp(void)
#endif
{
- PyObject *module = NULL;
+ PyObject* module = NULL;
#if PY_MAJOR_VERSION >= 3
- module = PyModule_Create(&moduledef);
+ module = PyModule_Create(&moduledef);
#else
- module = Py_InitModule("greenlet.tests._test_extension_cpp", test_methods);
+ module = Py_InitModule("greenlet.tests._test_extension_cpp", test_methods);
#endif
- if (module == NULL) {
- INITERROR;
- }
+ if (module == NULL) {
+ INITERROR;
+ }
- PyGreenlet_Import();
- if (_PyGreenlet_API == NULL) {
- INITERROR;
- }
+ PyGreenlet_Import();
+ if (_PyGreenlet_API == NULL) {
+ INITERROR;
+ }
- p_test_exception_throw = test_exception_throw;
- p_test_exception_switch_recurse = test_exception_switch_recurse;
+ p_test_exception_throw = test_exception_throw;
+ p_test_exception_switch_recurse = test_exception_switch_recurse;
#if PY_MAJOR_VERSION >= 3
- return module;
+ return module;
#endif
}