JezK

Edit File: greenlet_thread_state_dict_cleanup.hpp

#ifndef GREENLET_THREAD_STATE_DICT_CLEANUP_HPP
#define GREENLET_THREAD_STATE_DICT_CLEANUP_HPP

#include "greenlet_internal.hpp"
#include "greenlet_thread_state.hpp"

#ifdef __clang__
#    pragma clang diagnostic push
#    pragma clang diagnostic ignored "-Wmissing-field-initializers"
#endif

#ifndef G_THREAD_STATE_DICT_CLEANUP_TYPE
// shut the compiler up if it looks at this file in isolation
#define ThreadStateCreator int
#endif

// Define a Python object that goes in the Python thread state dict
// when the greenlet thread state is created, and which owns the
// reference to the greenlet thread local state.
// When the thread state dict is cleaned up, so too is the thread
// state. This works best if we make sure there are no circular
// references to the thread state.
typedef struct _PyGreenletCleanup {
    PyObject_HEAD
    ThreadStateCreator* thread_state_creator;
} PyGreenletCleanup;

static void
cleanup_do_dealloc(PyGreenletCleanup* self)
{
    ThreadStateCreator* tmp = self->thread_state_creator;
    self->thread_state_creator = nullptr;
    if (tmp) {
        delete tmp;
    }
}

static void
cleanup_dealloc(PyGreenletCleanup* self)
{
    PyObject_GC_UnTrack(self);
    cleanup_do_dealloc(self);
}

static int
cleanup_clear(PyGreenletCleanup* self)
{
    // This method is never called by our test cases.
    cleanup_do_dealloc(self);
    return 0;
}

static int
cleanup_traverse(PyGreenletCleanup* self, visitproc visit, void* arg)
{
    if (self->thread_state_creator) {
        return self->thread_state_creator->tp_traverse(visit, arg);
    }
    return 0;
}

static int
cleanup_is_gc(PyGreenlet* UNUSED(self))
{
    return 1;
}

static PyTypeObject PyGreenletCleanup_Type = {
    PyVarObject_HEAD_INIT(NULL, 0)
    "greenlet._greenlet.ThreadStateCleanup",
    sizeof(struct _PyGreenletCleanup),
    0,                   /* tp_itemsize */
    /* methods */
    (destructor)cleanup_dealloc, /* tp_dealloc */
    0,                         /* tp_print */
    0,                         /* tp_getattr */
    0,                         /* tp_setattr */
    0,                         /* tp_compare */
    0,                         /* tp_repr */
    0,                         /* 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*/
    G_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
    "Internal use only",                        /* tp_doc */
    (traverseproc)cleanup_traverse, /* tp_traverse */
    (inquiry)cleanup_clear,         /* tp_clear */
    0,                                  /* tp_richcompare */
    // XXX: Don't our flags promise a weakref?
    0,                           /* tp_weaklistoffset */
    0,                                  /* tp_iter */
    0,                                  /* tp_iternext */
    0,                      /* tp_methods */
    0,                      /* tp_members */
    0,                      /* tp_getset */
    0,                                  /* tp_base */
    0,                                  /* tp_dict */
    0,                                  /* tp_descr_get */
    0,                                  /* tp_descr_set */
    0,         /* tp_dictoffset */
    0,               /* tp_init */
    PyType_GenericAlloc,                  /* tp_alloc */
    PyType_GenericNew,                          /* tp_new */
    PyObject_GC_Del,                   /* tp_free */
    (inquiry)cleanup_is_gc,         /* tp_is_gc */
};

#ifdef __clang__
#    pragma clang diagnostic pop
#endif

#endif

Anmeldung Newsletter | Fairnetzt-Hochrhein