mirror of
https://github.com/python/cpython.git
synced 2024-11-23 18:04:37 +08:00
33b790978d
Add PyConfig_Get(), PyConfig_GetInt(), PyConfig_Set() and PyConfig_Names() functions to get and set the current runtime Python configuration. Add visibility and "sys spec" to config and preconfig specifications. _PyConfig_AsDict() now converts PyConfig.xoptions as a dictionary. Co-authored-by: Bénédikt Tran <10796600+picnixz@users.noreply.github.com>
4182 lines
119 KiB
C
4182 lines
119 KiB
C
/*
|
|
* C Extension module to test Python interpreter C APIs.
|
|
*
|
|
* The 'test_*' functions exported by this module are run as part of the
|
|
* standard Python regression test, via Lib/test/test_capi.py.
|
|
*/
|
|
|
|
// Include parts.h first since it takes care of NDEBUG and Py_BUILD_CORE macros
|
|
// and including Python.h.
|
|
//
|
|
// Several parts of this module are broken out into files in _testcapi/.
|
|
// Include definitions from there.
|
|
#include "_testcapi/parts.h"
|
|
|
|
#include "frameobject.h" // PyFrame_New()
|
|
#include "marshal.h" // PyMarshal_WriteLongToFile()
|
|
|
|
#include <float.h> // FLT_MAX
|
|
#include <signal.h>
|
|
#include <stddef.h> // offsetof()
|
|
|
|
#ifdef HAVE_SYS_WAIT_H
|
|
# include <sys/wait.h> // W_STOPCODE
|
|
#endif
|
|
|
|
#ifdef bool
|
|
# error "The public headers should not include <stdbool.h>, see gh-48924"
|
|
#endif
|
|
|
|
#include "_testcapi/util.h"
|
|
|
|
|
|
// Forward declarations
|
|
static struct PyModuleDef _testcapimodule;
|
|
|
|
// Module state
|
|
typedef struct {
|
|
PyObject *error; // _testcapi.error object
|
|
} testcapistate_t;
|
|
|
|
static testcapistate_t*
|
|
get_testcapi_state(PyObject *module)
|
|
{
|
|
void *state = PyModule_GetState(module);
|
|
assert(state != NULL);
|
|
return (testcapistate_t *)state;
|
|
}
|
|
|
|
static PyObject *
|
|
get_testerror(PyObject *self) {
|
|
testcapistate_t *state = get_testcapi_state(self);
|
|
return state->error;
|
|
}
|
|
|
|
/* Raise _testcapi.error with test_name + ": " + msg, and return NULL. */
|
|
|
|
static PyObject *
|
|
raiseTestError(PyObject *self, const char* test_name, const char* msg)
|
|
{
|
|
PyErr_Format(get_testerror(self), "%s: %s", test_name, msg);
|
|
return NULL;
|
|
}
|
|
|
|
/* Test #defines from pyconfig.h (particularly the SIZEOF_* defines).
|
|
|
|
The ones derived from autoconf on the UNIX-like OSes can be relied
|
|
upon (in the absence of sloppy cross-compiling), but the Windows
|
|
platforms have these hardcoded. Better safe than sorry.
|
|
*/
|
|
static PyObject*
|
|
sizeof_error(PyObject *self, const char* fatname, const char* typname,
|
|
int expected, int got)
|
|
{
|
|
PyErr_Format(get_testerror(self),
|
|
"%s #define == %d but sizeof(%s) == %d",
|
|
fatname, expected, typname, got);
|
|
return (PyObject*)NULL;
|
|
}
|
|
|
|
static PyObject*
|
|
test_config(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
#define CHECK_SIZEOF(FATNAME, TYPE) \
|
|
do { \
|
|
if (FATNAME != sizeof(TYPE)) { \
|
|
return sizeof_error(self, #FATNAME, #TYPE, FATNAME, sizeof(TYPE)); \
|
|
} \
|
|
} while (0)
|
|
|
|
CHECK_SIZEOF(SIZEOF_SHORT, short);
|
|
CHECK_SIZEOF(SIZEOF_INT, int);
|
|
CHECK_SIZEOF(SIZEOF_LONG, long);
|
|
CHECK_SIZEOF(SIZEOF_VOID_P, void*);
|
|
CHECK_SIZEOF(SIZEOF_TIME_T, time_t);
|
|
CHECK_SIZEOF(SIZEOF_LONG_LONG, long long);
|
|
|
|
#undef CHECK_SIZEOF
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
test_sizeof_c_types(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
#if defined(__GNUC__) && ((__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ > 5)))
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wtype-limits"
|
|
#endif
|
|
#define CHECK_SIZEOF(TYPE, EXPECTED) \
|
|
do { \
|
|
if (EXPECTED != sizeof(TYPE)) { \
|
|
PyErr_Format(get_testerror(self), \
|
|
"sizeof(%s) = %u instead of %u", \
|
|
#TYPE, sizeof(TYPE), EXPECTED); \
|
|
return (PyObject*)NULL; \
|
|
} \
|
|
} while (0)
|
|
#define IS_SIGNED(TYPE) (((TYPE)-1) < (TYPE)0)
|
|
#define CHECK_SIGNNESS(TYPE, SIGNED) \
|
|
do { \
|
|
if (IS_SIGNED(TYPE) != SIGNED) { \
|
|
PyErr_Format(get_testerror(self), \
|
|
"%s signness is %i, instead of %i", \
|
|
#TYPE, IS_SIGNED(TYPE), SIGNED); \
|
|
return (PyObject*)NULL; \
|
|
} \
|
|
} while (0)
|
|
|
|
/* integer types */
|
|
CHECK_SIZEOF(Py_UCS1, 1);
|
|
CHECK_SIZEOF(Py_UCS2, 2);
|
|
CHECK_SIZEOF(Py_UCS4, 4);
|
|
CHECK_SIGNNESS(Py_UCS1, 0);
|
|
CHECK_SIGNNESS(Py_UCS2, 0);
|
|
CHECK_SIGNNESS(Py_UCS4, 0);
|
|
CHECK_SIZEOF(int32_t, 4);
|
|
CHECK_SIGNNESS(int32_t, 1);
|
|
CHECK_SIZEOF(uint32_t, 4);
|
|
CHECK_SIGNNESS(uint32_t, 0);
|
|
CHECK_SIZEOF(int64_t, 8);
|
|
CHECK_SIGNNESS(int64_t, 1);
|
|
CHECK_SIZEOF(uint64_t, 8);
|
|
CHECK_SIGNNESS(uint64_t, 0);
|
|
|
|
/* pointer/size types */
|
|
CHECK_SIZEOF(size_t, sizeof(void *));
|
|
CHECK_SIGNNESS(size_t, 0);
|
|
CHECK_SIZEOF(Py_ssize_t, sizeof(void *));
|
|
CHECK_SIGNNESS(Py_ssize_t, 1);
|
|
|
|
CHECK_SIZEOF(uintptr_t, sizeof(void *));
|
|
CHECK_SIGNNESS(uintptr_t, 0);
|
|
CHECK_SIZEOF(intptr_t, sizeof(void *));
|
|
CHECK_SIGNNESS(intptr_t, 1);
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
#undef IS_SIGNED
|
|
#undef CHECK_SIGNESS
|
|
#undef CHECK_SIZEOF
|
|
#if defined(__GNUC__) && ((__GNUC__ > 4) || ((__GNUC__ == 4) && (__GNUC_MINOR__ > 5)))
|
|
#pragma GCC diagnostic pop
|
|
#endif
|
|
}
|
|
|
|
static PyObject*
|
|
test_list_api(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject* list;
|
|
int i;
|
|
|
|
/* SF bug 132008: PyList_Reverse segfaults */
|
|
#define NLIST 30
|
|
list = PyList_New(NLIST);
|
|
if (list == (PyObject*)NULL)
|
|
return (PyObject*)NULL;
|
|
/* list = range(NLIST) */
|
|
for (i = 0; i < NLIST; ++i) {
|
|
PyObject* anint = PyLong_FromLong(i);
|
|
if (anint == (PyObject*)NULL) {
|
|
Py_DECREF(list);
|
|
return (PyObject*)NULL;
|
|
}
|
|
PyList_SET_ITEM(list, i, anint);
|
|
}
|
|
/* list.reverse(), via PyList_Reverse() */
|
|
i = PyList_Reverse(list); /* should not blow up! */
|
|
if (i != 0) {
|
|
Py_DECREF(list);
|
|
return (PyObject*)NULL;
|
|
}
|
|
/* Check that list == range(29, -1, -1) now */
|
|
for (i = 0; i < NLIST; ++i) {
|
|
PyObject* anint = PyList_GET_ITEM(list, i);
|
|
if (PyLong_AS_LONG(anint) != NLIST-1-i) {
|
|
PyErr_SetString(get_testerror(self),
|
|
"test_list_api: reverse screwed up");
|
|
Py_DECREF(list);
|
|
return (PyObject*)NULL;
|
|
}
|
|
}
|
|
Py_DECREF(list);
|
|
#undef NLIST
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static int
|
|
test_dict_inner(PyObject *self, int count)
|
|
{
|
|
Py_ssize_t pos = 0, iterations = 0;
|
|
int i;
|
|
PyObject *dict = PyDict_New();
|
|
PyObject *v, *k;
|
|
|
|
if (dict == NULL)
|
|
return -1;
|
|
|
|
for (i = 0; i < count; i++) {
|
|
v = PyLong_FromLong(i);
|
|
if (v == NULL) {
|
|
goto error;
|
|
}
|
|
if (PyDict_SetItem(dict, v, v) < 0) {
|
|
Py_DECREF(v);
|
|
goto error;
|
|
}
|
|
Py_DECREF(v);
|
|
}
|
|
|
|
k = v = UNINITIALIZED_PTR;
|
|
while (PyDict_Next(dict, &pos, &k, &v)) {
|
|
PyObject *o;
|
|
iterations++;
|
|
|
|
assert(k != UNINITIALIZED_PTR);
|
|
assert(v != UNINITIALIZED_PTR);
|
|
i = PyLong_AS_LONG(v) + 1;
|
|
o = PyLong_FromLong(i);
|
|
if (o == NULL) {
|
|
goto error;
|
|
}
|
|
if (PyDict_SetItem(dict, k, o) < 0) {
|
|
Py_DECREF(o);
|
|
goto error;
|
|
}
|
|
Py_DECREF(o);
|
|
k = v = UNINITIALIZED_PTR;
|
|
}
|
|
assert(k == UNINITIALIZED_PTR);
|
|
assert(v == UNINITIALIZED_PTR);
|
|
|
|
Py_DECREF(dict);
|
|
|
|
if (iterations != count) {
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_dict_iteration: dict iteration went wrong ");
|
|
return -1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
error:
|
|
Py_DECREF(dict);
|
|
return -1;
|
|
}
|
|
|
|
|
|
|
|
static PyObject*
|
|
test_dict_iteration(PyObject* self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < 200; i++) {
|
|
if (test_dict_inner(self, i) < 0) {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* Issue #4701: Check that PyObject_Hash implicitly calls
|
|
* PyType_Ready if it hasn't already been called
|
|
*/
|
|
static PyTypeObject _HashInheritanceTester_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"hashinheritancetester", /* Name of this type */
|
|
sizeof(PyObject), /* Basic object size */
|
|
0, /* Item size for varobject */
|
|
(destructor)PyObject_Free, /* tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_as_async */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT, /* tp_flags */
|
|
0, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
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 */
|
|
0, /* tp_alloc */
|
|
PyType_GenericNew, /* tp_new */
|
|
};
|
|
|
|
static PyObject*
|
|
pycompilestring(PyObject* self, PyObject *obj) {
|
|
if (PyBytes_CheckExact(obj) == 0) {
|
|
PyErr_SetString(PyExc_ValueError, "Argument must be a bytes object");
|
|
return NULL;
|
|
}
|
|
const char *the_string = PyBytes_AsString(obj);
|
|
if (the_string == NULL) {
|
|
return NULL;
|
|
}
|
|
return Py_CompileString(the_string, "<string>", Py_file_input);
|
|
}
|
|
|
|
static PyObject*
|
|
test_lazy_hash_inheritance(PyObject* self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyTypeObject *type;
|
|
PyObject *obj;
|
|
Py_hash_t hash;
|
|
|
|
type = &_HashInheritanceTester_Type;
|
|
|
|
if (type->tp_dict != NULL)
|
|
/* The type has already been initialized. This probably means
|
|
-R is being used. */
|
|
Py_RETURN_NONE;
|
|
|
|
|
|
obj = PyObject_New(PyObject, type);
|
|
if (obj == NULL) {
|
|
PyErr_Clear();
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_lazy_hash_inheritance: failed to create object");
|
|
return NULL;
|
|
}
|
|
|
|
if (type->tp_dict != NULL) {
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_lazy_hash_inheritance: type initialised too soon");
|
|
Py_DECREF(obj);
|
|
return NULL;
|
|
}
|
|
|
|
hash = PyObject_Hash(obj);
|
|
if ((hash == -1) && PyErr_Occurred()) {
|
|
PyErr_Clear();
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_lazy_hash_inheritance: could not hash object");
|
|
Py_DECREF(obj);
|
|
return NULL;
|
|
}
|
|
|
|
if (type->tp_dict == NULL) {
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_lazy_hash_inheritance: type not initialised by hash()");
|
|
Py_DECREF(obj);
|
|
return NULL;
|
|
}
|
|
|
|
if (type->tp_hash != PyType_Type.tp_hash) {
|
|
PyErr_SetString(
|
|
get_testerror(self),
|
|
"test_lazy_hash_inheritance: unexpected hash function");
|
|
Py_DECREF(obj);
|
|
return NULL;
|
|
}
|
|
|
|
Py_DECREF(obj);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
return_none(void *unused)
|
|
{
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
raise_error(void *unused)
|
|
{
|
|
PyErr_SetNone(PyExc_ValueError);
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject *
|
|
py_buildvalue(PyObject *self, PyObject *args)
|
|
{
|
|
const char *fmt;
|
|
PyObject *objs[10] = {NULL};
|
|
if (!PyArg_ParseTuple(args, "s|OOOOOOOOOO", &fmt,
|
|
&objs[0], &objs[1], &objs[2], &objs[3], &objs[4],
|
|
&objs[5], &objs[6], &objs[7], &objs[8], &objs[9]))
|
|
{
|
|
return NULL;
|
|
}
|
|
for(int i = 0; i < 10; i++) {
|
|
NULLABLE(objs[i]);
|
|
}
|
|
return Py_BuildValue(fmt,
|
|
objs[0], objs[1], objs[2], objs[3], objs[4],
|
|
objs[5], objs[6], objs[7], objs[8], objs[9]);
|
|
}
|
|
|
|
static PyObject *
|
|
py_buildvalue_ints(PyObject *self, PyObject *args)
|
|
{
|
|
const char *fmt;
|
|
unsigned int values[10] = {0};
|
|
if (!PyArg_ParseTuple(args, "s|IIIIIIIIII", &fmt,
|
|
&values[0], &values[1], &values[2], &values[3], &values[4],
|
|
&values[5], &values[6], &values[7], &values[8], &values[9]))
|
|
{
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue(fmt,
|
|
values[0], values[1], values[2], values[3], values[4],
|
|
values[5], values[6], values[7], values[8], values[9]);
|
|
}
|
|
|
|
static int
|
|
test_buildvalue_N_error(PyObject *self, const char *fmt)
|
|
{
|
|
PyObject *arg, *res;
|
|
|
|
arg = PyList_New(0);
|
|
if (arg == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
Py_INCREF(arg);
|
|
res = Py_BuildValue(fmt, return_none, NULL, arg);
|
|
if (res == NULL) {
|
|
return -1;
|
|
}
|
|
Py_DECREF(res);
|
|
if (Py_REFCNT(arg) != 1) {
|
|
PyErr_Format(get_testerror(self), "test_buildvalue_N: "
|
|
"arg was not decrefed in successful "
|
|
"Py_BuildValue(\"%s\")", fmt);
|
|
return -1;
|
|
}
|
|
|
|
Py_INCREF(arg);
|
|
res = Py_BuildValue(fmt, raise_error, NULL, arg);
|
|
if (res != NULL || !PyErr_Occurred()) {
|
|
PyErr_Format(get_testerror(self), "test_buildvalue_N: "
|
|
"Py_BuildValue(\"%s\") didn't complain", fmt);
|
|
return -1;
|
|
}
|
|
PyErr_Clear();
|
|
if (Py_REFCNT(arg) != 1) {
|
|
PyErr_Format(get_testerror(self), "test_buildvalue_N: "
|
|
"arg was not decrefed in failed "
|
|
"Py_BuildValue(\"%s\")", fmt);
|
|
return -1;
|
|
}
|
|
Py_DECREF(arg);
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
test_buildvalue_N(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *arg, *res;
|
|
|
|
arg = PyList_New(0);
|
|
if (arg == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_INCREF(arg);
|
|
res = Py_BuildValue("N", arg);
|
|
if (res == NULL) {
|
|
return NULL;
|
|
}
|
|
if (res != arg) {
|
|
return raiseTestError(self, "test_buildvalue_N",
|
|
"Py_BuildValue(\"N\") returned wrong result");
|
|
}
|
|
if (Py_REFCNT(arg) != 2) {
|
|
return raiseTestError(self, "test_buildvalue_N",
|
|
"arg was not decrefed in Py_BuildValue(\"N\")");
|
|
}
|
|
Py_DECREF(res);
|
|
Py_DECREF(arg);
|
|
|
|
if (test_buildvalue_N_error(self, "O&N") < 0)
|
|
return NULL;
|
|
if (test_buildvalue_N_error(self, "(O&N)") < 0)
|
|
return NULL;
|
|
if (test_buildvalue_N_error(self, "[O&N]") < 0)
|
|
return NULL;
|
|
if (test_buildvalue_N_error(self, "{O&N}") < 0)
|
|
return NULL;
|
|
if (test_buildvalue_N_error(self, "{()O&(())N}") < 0)
|
|
return NULL;
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
test_get_statictype_slots(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
newfunc tp_new = PyType_GetSlot(&PyLong_Type, Py_tp_new);
|
|
if (PyLong_Type.tp_new != tp_new) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: tp_new of long");
|
|
return NULL;
|
|
}
|
|
|
|
reprfunc tp_repr = PyType_GetSlot(&PyLong_Type, Py_tp_repr);
|
|
if (PyLong_Type.tp_repr != tp_repr) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: tp_repr of long");
|
|
return NULL;
|
|
}
|
|
|
|
ternaryfunc tp_call = PyType_GetSlot(&PyLong_Type, Py_tp_call);
|
|
if (tp_call != NULL) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: tp_call of long");
|
|
return NULL;
|
|
}
|
|
|
|
binaryfunc nb_add = PyType_GetSlot(&PyLong_Type, Py_nb_add);
|
|
if (PyLong_Type.tp_as_number->nb_add != nb_add) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: nb_add of long");
|
|
return NULL;
|
|
}
|
|
|
|
lenfunc mp_length = PyType_GetSlot(&PyLong_Type, Py_mp_length);
|
|
if (mp_length != NULL) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: mp_length of long");
|
|
return NULL;
|
|
}
|
|
|
|
void *over_value = PyType_GetSlot(&PyLong_Type, Py_bf_releasebuffer + 1);
|
|
if (over_value != NULL) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: max+1 of long");
|
|
return NULL;
|
|
}
|
|
|
|
tp_new = PyType_GetSlot(&PyLong_Type, 0);
|
|
if (tp_new != NULL) {
|
|
PyErr_SetString(PyExc_AssertionError, "mismatch: slot 0 of long");
|
|
return NULL;
|
|
}
|
|
if (PyErr_ExceptionMatches(PyExc_SystemError)) {
|
|
// This is the right exception
|
|
PyErr_Clear();
|
|
}
|
|
else {
|
|
return NULL;
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyType_Slot HeapTypeNameType_slots[] = {
|
|
{0},
|
|
};
|
|
|
|
static PyType_Spec HeapTypeNameType_Spec = {
|
|
.name = "_testcapi.HeapTypeNameType",
|
|
.basicsize = sizeof(PyObject),
|
|
.flags = Py_TPFLAGS_DEFAULT,
|
|
.slots = HeapTypeNameType_slots,
|
|
};
|
|
|
|
static PyObject *
|
|
get_heaptype_for_name(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyType_FromSpec(&HeapTypeNameType_Spec);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
get_type_name(PyObject *self, PyObject *type)
|
|
{
|
|
assert(PyType_Check(type));
|
|
return PyType_GetName((PyTypeObject *)type);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
get_type_qualname(PyObject *self, PyObject *type)
|
|
{
|
|
assert(PyType_Check(type));
|
|
return PyType_GetQualName((PyTypeObject *)type);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
get_type_fullyqualname(PyObject *self, PyObject *type)
|
|
{
|
|
assert(PyType_Check(type));
|
|
return PyType_GetFullyQualifiedName((PyTypeObject *)type);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
get_type_module_name(PyObject *self, PyObject *type)
|
|
{
|
|
assert(PyType_Check(type));
|
|
return PyType_GetModuleName((PyTypeObject *)type);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
test_get_type_dict(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
/* Test for PyType_GetDict */
|
|
|
|
// Assert ints have a `to_bytes` method
|
|
PyObject *long_dict = PyType_GetDict(&PyLong_Type);
|
|
assert(long_dict);
|
|
assert(PyDict_GetItemString(long_dict, "to_bytes")); // borrowed ref
|
|
Py_DECREF(long_dict);
|
|
|
|
// Make a new type, add an attribute to it and assert it's there
|
|
PyObject *HeapTypeNameType = PyType_FromSpec(&HeapTypeNameType_Spec);
|
|
assert(HeapTypeNameType);
|
|
assert(PyObject_SetAttrString(
|
|
HeapTypeNameType, "new_attr", Py_NewRef(Py_None)) >= 0);
|
|
PyObject *type_dict = PyType_GetDict((PyTypeObject*)HeapTypeNameType);
|
|
assert(type_dict);
|
|
assert(PyDict_GetItemString(type_dict, "new_attr")); // borrowed ref
|
|
Py_DECREF(HeapTypeNameType);
|
|
Py_DECREF(type_dict);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
pyobject_repr_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyObject_Repr(NULL);
|
|
}
|
|
|
|
static PyObject *
|
|
pyobject_str_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyObject_Str(NULL);
|
|
}
|
|
|
|
static PyObject *
|
|
pyobject_bytes_from_null(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
return PyObject_Bytes(NULL);
|
|
}
|
|
|
|
static PyObject *
|
|
set_errno(PyObject *self, PyObject *args)
|
|
{
|
|
int new_errno;
|
|
|
|
if (!PyArg_ParseTuple(args, "i:set_errno", &new_errno))
|
|
return NULL;
|
|
|
|
errno = new_errno;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* test_thread_state spawns a thread of its own, and that thread releases
|
|
* `thread_done` when it's finished. The driver code has to know when the
|
|
* thread finishes, because the thread uses a PyObject (the callable) that
|
|
* may go away when the driver finishes. The former lack of this explicit
|
|
* synchronization caused rare segfaults, so rare that they were seen only
|
|
* on a Mac buildbot (although they were possible on any box).
|
|
*/
|
|
static PyThread_type_lock thread_done = NULL;
|
|
|
|
static int
|
|
_make_call(void *callable)
|
|
{
|
|
PyObject *rc;
|
|
int success;
|
|
PyGILState_STATE s = PyGILState_Ensure();
|
|
rc = PyObject_CallNoArgs((PyObject *)callable);
|
|
success = (rc != NULL);
|
|
Py_XDECREF(rc);
|
|
PyGILState_Release(s);
|
|
return success;
|
|
}
|
|
|
|
/* Same thing, but releases `thread_done` when it returns. This variant
|
|
* should be called only from threads spawned by test_thread_state().
|
|
*/
|
|
static void
|
|
_make_call_from_thread(void *callable)
|
|
{
|
|
_make_call(callable);
|
|
PyThread_release_lock(thread_done);
|
|
}
|
|
|
|
static PyObject *
|
|
test_thread_state(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *fn;
|
|
int success = 1;
|
|
|
|
if (!PyArg_ParseTuple(args, "O:test_thread_state", &fn))
|
|
return NULL;
|
|
|
|
if (!PyCallable_Check(fn)) {
|
|
PyErr_Format(PyExc_TypeError, "'%s' object is not callable",
|
|
Py_TYPE(fn)->tp_name);
|
|
return NULL;
|
|
}
|
|
|
|
thread_done = PyThread_allocate_lock();
|
|
if (thread_done == NULL)
|
|
return PyErr_NoMemory();
|
|
PyThread_acquire_lock(thread_done, 1);
|
|
|
|
/* Start a new thread with our callback. */
|
|
PyThread_start_new_thread(_make_call_from_thread, fn);
|
|
/* Make the callback with the thread lock held by this thread */
|
|
success &= _make_call(fn);
|
|
/* Do it all again, but this time with the thread-lock released */
|
|
Py_BEGIN_ALLOW_THREADS
|
|
success &= _make_call(fn);
|
|
PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
|
|
Py_END_ALLOW_THREADS
|
|
|
|
/* And once more with and without a thread
|
|
XXX - should use a lock and work out exactly what we are trying
|
|
to test <wink>
|
|
*/
|
|
Py_BEGIN_ALLOW_THREADS
|
|
PyThread_start_new_thread(_make_call_from_thread, fn);
|
|
success &= _make_call(fn);
|
|
PyThread_acquire_lock(thread_done, 1); /* wait for thread to finish */
|
|
Py_END_ALLOW_THREADS
|
|
|
|
/* Release lock we acquired above. This is required on HP-UX. */
|
|
PyThread_release_lock(thread_done);
|
|
|
|
PyThread_free_lock(thread_done);
|
|
if (!success)
|
|
return NULL;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
gilstate_ensure_release(PyObject *module, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyGILState_STATE state = PyGILState_Ensure();
|
|
PyGILState_Release(state);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
#ifndef MS_WINDOWS
|
|
static PyThread_type_lock wait_done = NULL;
|
|
|
|
static void wait_for_lock(void *unused) {
|
|
PyThread_acquire_lock(wait_done, 1);
|
|
PyThread_release_lock(wait_done);
|
|
PyThread_free_lock(wait_done);
|
|
wait_done = NULL;
|
|
}
|
|
|
|
// These can be used to test things that care about the existence of another
|
|
// thread that the threading module doesn't know about.
|
|
|
|
static PyObject *
|
|
spawn_pthread_waiter(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
if (wait_done) {
|
|
PyErr_SetString(PyExc_RuntimeError, "thread already running");
|
|
return NULL;
|
|
}
|
|
wait_done = PyThread_allocate_lock();
|
|
if (wait_done == NULL)
|
|
return PyErr_NoMemory();
|
|
PyThread_acquire_lock(wait_done, 1);
|
|
PyThread_start_new_thread(wait_for_lock, NULL);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
end_spawned_pthread(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
if (!wait_done) {
|
|
PyErr_SetString(PyExc_RuntimeError, "call _spawn_pthread_waiter 1st");
|
|
return NULL;
|
|
}
|
|
PyThread_release_lock(wait_done);
|
|
Py_RETURN_NONE;
|
|
}
|
|
#endif // not MS_WINDOWS
|
|
|
|
/* test Py_AddPendingCalls using threads */
|
|
static int _pending_callback(void *arg)
|
|
{
|
|
/* we assume the argument is callable object to which we own a reference */
|
|
PyObject *callable = (PyObject *)arg;
|
|
PyObject *r = PyObject_CallNoArgs(callable);
|
|
Py_DECREF(callable);
|
|
Py_XDECREF(r);
|
|
return r != NULL ? 0 : -1;
|
|
}
|
|
|
|
/* The following requests n callbacks to _pending_callback. It can be
|
|
* run from any python thread.
|
|
*/
|
|
static PyObject *
|
|
pending_threadfunc(PyObject *self, PyObject *arg, PyObject *kwargs)
|
|
{
|
|
static char *kwlist[] = {"callback", "num",
|
|
"blocking", "ensure_added", NULL};
|
|
PyObject *callable;
|
|
unsigned int num = 1;
|
|
int blocking = 0;
|
|
int ensure_added = 0;
|
|
if (!PyArg_ParseTupleAndKeywords(arg, kwargs,
|
|
"O|I$pp:_pending_threadfunc", kwlist,
|
|
&callable, &num, &blocking, &ensure_added))
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/* create the reference for the callbackwhile we hold the lock */
|
|
for (unsigned int i = 0; i < num; i++) {
|
|
Py_INCREF(callable);
|
|
}
|
|
|
|
PyThreadState *save_tstate = NULL;
|
|
if (!blocking) {
|
|
save_tstate = PyEval_SaveThread();
|
|
}
|
|
|
|
unsigned int num_added = 0;
|
|
for (; num_added < num; num_added++) {
|
|
if (ensure_added) {
|
|
int r;
|
|
do {
|
|
r = Py_AddPendingCall(&_pending_callback, callable);
|
|
} while (r < 0);
|
|
}
|
|
else {
|
|
if (Py_AddPendingCall(&_pending_callback, callable) < 0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!blocking) {
|
|
PyEval_RestoreThread(save_tstate);
|
|
}
|
|
|
|
for (unsigned int i = num_added; i < num; i++) {
|
|
Py_DECREF(callable); /* unsuccessful add, destroy the extra reference */
|
|
}
|
|
/* The callable is decref'ed above in each added _pending_callback(). */
|
|
return PyLong_FromUnsignedLong((unsigned long)num_added);
|
|
}
|
|
|
|
/* Test PyOS_string_to_double. */
|
|
static PyObject *
|
|
test_string_to_double(PyObject *self, PyObject *Py_UNUSED(ignored)) {
|
|
double result;
|
|
const char *msg;
|
|
|
|
#define CHECK_STRING(STR, expected) \
|
|
do { \
|
|
result = PyOS_string_to_double(STR, NULL, NULL); \
|
|
if (result == -1.0 && PyErr_Occurred()) { \
|
|
return NULL; \
|
|
} \
|
|
if (result != (double)expected) { \
|
|
msg = "conversion of " STR " to float failed"; \
|
|
goto fail; \
|
|
} \
|
|
} while (0)
|
|
|
|
#define CHECK_INVALID(STR) \
|
|
do { \
|
|
result = PyOS_string_to_double(STR, NULL, NULL); \
|
|
if (result == -1.0 && PyErr_Occurred()) { \
|
|
if (PyErr_ExceptionMatches(PyExc_ValueError)) { \
|
|
PyErr_Clear(); \
|
|
} \
|
|
else { \
|
|
return NULL; \
|
|
} \
|
|
} \
|
|
else { \
|
|
msg = "conversion of " STR " didn't raise ValueError"; \
|
|
goto fail; \
|
|
} \
|
|
} while (0)
|
|
|
|
CHECK_STRING("0.1", 0.1);
|
|
CHECK_STRING("1.234", 1.234);
|
|
CHECK_STRING("-1.35", -1.35);
|
|
CHECK_STRING(".1e01", 1.0);
|
|
CHECK_STRING("2.e-2", 0.02);
|
|
|
|
CHECK_INVALID(" 0.1");
|
|
CHECK_INVALID("\t\n-3");
|
|
CHECK_INVALID(".123 ");
|
|
CHECK_INVALID("3\n");
|
|
CHECK_INVALID("123abc");
|
|
|
|
Py_RETURN_NONE;
|
|
fail:
|
|
return raiseTestError(self, "test_string_to_double", msg);
|
|
#undef CHECK_STRING
|
|
#undef CHECK_INVALID
|
|
}
|
|
|
|
|
|
/* Coverage testing of capsule objects. */
|
|
|
|
static const char *capsule_name = "capsule name";
|
|
static char *capsule_pointer = "capsule pointer";
|
|
static char *capsule_context = "capsule context";
|
|
static const char *capsule_error = NULL;
|
|
static int
|
|
capsule_destructor_call_count = 0;
|
|
|
|
static void
|
|
capsule_destructor(PyObject *o) {
|
|
capsule_destructor_call_count++;
|
|
if (PyCapsule_GetContext(o) != capsule_context) {
|
|
capsule_error = "context did not match in destructor!";
|
|
} else if (PyCapsule_GetDestructor(o) != capsule_destructor) {
|
|
capsule_error = "destructor did not match in destructor! (woah!)";
|
|
} else if (PyCapsule_GetName(o) != capsule_name) {
|
|
capsule_error = "name did not match in destructor!";
|
|
} else if (PyCapsule_GetPointer(o, capsule_name) != capsule_pointer) {
|
|
capsule_error = "pointer did not match in destructor!";
|
|
}
|
|
}
|
|
|
|
typedef struct {
|
|
char *name;
|
|
char *module;
|
|
char *attribute;
|
|
} known_capsule;
|
|
|
|
static PyObject *
|
|
test_capsule(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *object;
|
|
const char *error = NULL;
|
|
void *pointer;
|
|
void *pointer2;
|
|
known_capsule known_capsules[] = {
|
|
#define KNOWN_CAPSULE(module, name) { module "." name, module, name }
|
|
KNOWN_CAPSULE("_socket", "CAPI"),
|
|
KNOWN_CAPSULE("_curses", "_C_API"),
|
|
KNOWN_CAPSULE("datetime", "datetime_CAPI"),
|
|
{ NULL, NULL },
|
|
};
|
|
known_capsule *known = &known_capsules[0];
|
|
|
|
#define FAIL(x) \
|
|
do { \
|
|
error = (x); \
|
|
goto exit; \
|
|
} while (0)
|
|
|
|
#define CHECK_DESTRUCTOR \
|
|
do { \
|
|
if (capsule_error) { \
|
|
FAIL(capsule_error); \
|
|
} \
|
|
else if (!capsule_destructor_call_count) { \
|
|
FAIL("destructor not called!"); \
|
|
} \
|
|
capsule_destructor_call_count = 0; \
|
|
} while (0)
|
|
|
|
object = PyCapsule_New(capsule_pointer, capsule_name, capsule_destructor);
|
|
PyCapsule_SetContext(object, capsule_context);
|
|
capsule_destructor(object);
|
|
CHECK_DESTRUCTOR;
|
|
Py_DECREF(object);
|
|
CHECK_DESTRUCTOR;
|
|
|
|
object = PyCapsule_New(known, "ignored", NULL);
|
|
PyCapsule_SetPointer(object, capsule_pointer);
|
|
PyCapsule_SetName(object, capsule_name);
|
|
PyCapsule_SetDestructor(object, capsule_destructor);
|
|
PyCapsule_SetContext(object, capsule_context);
|
|
capsule_destructor(object);
|
|
CHECK_DESTRUCTOR;
|
|
/* intentionally access using the wrong name */
|
|
pointer2 = PyCapsule_GetPointer(object, "the wrong name");
|
|
if (!PyErr_Occurred()) {
|
|
FAIL("PyCapsule_GetPointer should have failed but did not!");
|
|
}
|
|
PyErr_Clear();
|
|
if (pointer2) {
|
|
if (pointer2 == capsule_pointer) {
|
|
FAIL("PyCapsule_GetPointer should not have"
|
|
" returned the internal pointer!");
|
|
} else {
|
|
FAIL("PyCapsule_GetPointer should have "
|
|
"returned NULL pointer but did not!");
|
|
}
|
|
}
|
|
PyCapsule_SetDestructor(object, NULL);
|
|
Py_DECREF(object);
|
|
if (capsule_destructor_call_count) {
|
|
FAIL("destructor called when it should not have been!");
|
|
}
|
|
|
|
for (known = &known_capsules[0]; known->module != NULL; known++) {
|
|
/* yeah, ordinarily I wouldn't do this either,
|
|
but it's fine for this test harness.
|
|
*/
|
|
static char buffer[256];
|
|
#undef FAIL
|
|
#define FAIL(x) \
|
|
do { \
|
|
sprintf(buffer, "%s module: \"%s\" attribute: \"%s\"", \
|
|
x, known->module, known->attribute); \
|
|
error = buffer; \
|
|
goto exit; \
|
|
} while (0)
|
|
|
|
PyObject *module = PyImport_ImportModule(known->module);
|
|
if (module) {
|
|
pointer = PyCapsule_Import(known->name, 0);
|
|
if (!pointer) {
|
|
Py_DECREF(module);
|
|
FAIL("PyCapsule_GetPointer returned NULL unexpectedly!");
|
|
}
|
|
object = PyObject_GetAttrString(module, known->attribute);
|
|
if (!object) {
|
|
Py_DECREF(module);
|
|
return NULL;
|
|
}
|
|
pointer2 = PyCapsule_GetPointer(object,
|
|
"weebles wobble but they don't fall down");
|
|
if (!PyErr_Occurred()) {
|
|
Py_DECREF(object);
|
|
Py_DECREF(module);
|
|
FAIL("PyCapsule_GetPointer should have failed but did not!");
|
|
}
|
|
PyErr_Clear();
|
|
if (pointer2) {
|
|
Py_DECREF(module);
|
|
Py_DECREF(object);
|
|
if (pointer2 == pointer) {
|
|
FAIL("PyCapsule_GetPointer should not have"
|
|
" returned its internal pointer!");
|
|
} else {
|
|
FAIL("PyCapsule_GetPointer should have"
|
|
" returned NULL pointer but did not!");
|
|
}
|
|
}
|
|
Py_DECREF(object);
|
|
Py_DECREF(module);
|
|
}
|
|
else
|
|
PyErr_Clear();
|
|
}
|
|
|
|
exit:
|
|
if (error) {
|
|
return raiseTestError(self, "test_capsule", error);
|
|
}
|
|
Py_RETURN_NONE;
|
|
#undef FAIL
|
|
}
|
|
|
|
#ifdef HAVE_GETTIMEOFDAY
|
|
/* Profiling of integer performance */
|
|
static void print_delta(int test, struct timeval *s, struct timeval *e)
|
|
{
|
|
e->tv_sec -= s->tv_sec;
|
|
e->tv_usec -= s->tv_usec;
|
|
if (e->tv_usec < 0) {
|
|
e->tv_sec -=1;
|
|
e->tv_usec += 1000000;
|
|
}
|
|
printf("Test %d: %d.%06ds\n", test, (int)e->tv_sec, (int)e->tv_usec);
|
|
}
|
|
|
|
static PyObject *
|
|
profile_int(PyObject *self, PyObject* args)
|
|
{
|
|
int i, k;
|
|
struct timeval start, stop;
|
|
PyObject *single, **multiple, *op1, *result;
|
|
|
|
/* Test 1: Allocate and immediately deallocate
|
|
many small integers */
|
|
gettimeofday(&start, NULL);
|
|
for(k=0; k < 20000; k++)
|
|
for(i=0; i < 1000; i++) {
|
|
single = PyLong_FromLong(i);
|
|
Py_DECREF(single);
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
print_delta(1, &start, &stop);
|
|
|
|
/* Test 2: Allocate and immediately deallocate
|
|
many large integers */
|
|
gettimeofday(&start, NULL);
|
|
for(k=0; k < 20000; k++)
|
|
for(i=0; i < 1000; i++) {
|
|
single = PyLong_FromLong(i+1000000);
|
|
Py_DECREF(single);
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
print_delta(2, &start, &stop);
|
|
|
|
/* Test 3: Allocate a few integers, then release
|
|
them all simultaneously. */
|
|
multiple = malloc(sizeof(PyObject*) * 1000);
|
|
if (multiple == NULL)
|
|
return PyErr_NoMemory();
|
|
gettimeofday(&start, NULL);
|
|
for(k=0; k < 20000; k++) {
|
|
for(i=0; i < 1000; i++) {
|
|
multiple[i] = PyLong_FromLong(i+1000000);
|
|
}
|
|
for(i=0; i < 1000; i++) {
|
|
Py_DECREF(multiple[i]);
|
|
}
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
print_delta(3, &start, &stop);
|
|
free(multiple);
|
|
|
|
/* Test 4: Allocate many integers, then release
|
|
them all simultaneously. */
|
|
multiple = malloc(sizeof(PyObject*) * 1000000);
|
|
if (multiple == NULL)
|
|
return PyErr_NoMemory();
|
|
gettimeofday(&start, NULL);
|
|
for(k=0; k < 20; k++) {
|
|
for(i=0; i < 1000000; i++) {
|
|
multiple[i] = PyLong_FromLong(i+1000000);
|
|
}
|
|
for(i=0; i < 1000000; i++) {
|
|
Py_DECREF(multiple[i]);
|
|
}
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
print_delta(4, &start, &stop);
|
|
free(multiple);
|
|
|
|
/* Test 5: Allocate many integers < 32000 */
|
|
multiple = malloc(sizeof(PyObject*) * 1000000);
|
|
if (multiple == NULL)
|
|
return PyErr_NoMemory();
|
|
gettimeofday(&start, NULL);
|
|
for(k=0; k < 10; k++) {
|
|
for(i=0; i < 1000000; i++) {
|
|
multiple[i] = PyLong_FromLong(i+1000);
|
|
}
|
|
for(i=0; i < 1000000; i++) {
|
|
Py_DECREF(multiple[i]);
|
|
}
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
print_delta(5, &start, &stop);
|
|
free(multiple);
|
|
|
|
/* Test 6: Perform small int addition */
|
|
op1 = PyLong_FromLong(1);
|
|
gettimeofday(&start, NULL);
|
|
for(i=0; i < 10000000; i++) {
|
|
result = PyNumber_Add(op1, op1);
|
|
Py_DECREF(result);
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
Py_DECREF(op1);
|
|
print_delta(6, &start, &stop);
|
|
|
|
/* Test 7: Perform medium int addition */
|
|
op1 = PyLong_FromLong(1000);
|
|
if (op1 == NULL)
|
|
return NULL;
|
|
gettimeofday(&start, NULL);
|
|
for(i=0; i < 10000000; i++) {
|
|
result = PyNumber_Add(op1, op1);
|
|
Py_XDECREF(result);
|
|
}
|
|
gettimeofday(&stop, NULL);
|
|
Py_DECREF(op1);
|
|
print_delta(7, &start, &stop);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
#endif
|
|
|
|
/* Issue 6012 */
|
|
static PyObject *str1, *str2;
|
|
static int
|
|
failing_converter(PyObject *obj, void *arg)
|
|
{
|
|
/* Clone str1, then let the conversion fail. */
|
|
assert(str1);
|
|
str2 = Py_NewRef(str1);
|
|
return 0;
|
|
}
|
|
static PyObject*
|
|
argparsing(PyObject *o, PyObject *args)
|
|
{
|
|
PyObject *res;
|
|
str1 = str2 = NULL;
|
|
if (!PyArg_ParseTuple(args, "O&O&",
|
|
PyUnicode_FSConverter, &str1,
|
|
failing_converter, &str2)) {
|
|
if (!str2)
|
|
/* argument converter not called? */
|
|
return NULL;
|
|
/* Should be 1 */
|
|
res = PyLong_FromSsize_t(Py_REFCNT(str2));
|
|
Py_DECREF(str2);
|
|
PyErr_Clear();
|
|
return res;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* To test that the result of PyCode_NewEmpty has the right members. */
|
|
static PyObject *
|
|
code_newempty(PyObject *self, PyObject *args)
|
|
{
|
|
const char *filename;
|
|
const char *funcname;
|
|
int firstlineno;
|
|
|
|
if (!PyArg_ParseTuple(args, "ssi:code_newempty",
|
|
&filename, &funcname, &firstlineno))
|
|
return NULL;
|
|
|
|
return (PyObject *)PyCode_NewEmpty(filename, funcname, firstlineno);
|
|
}
|
|
|
|
static PyObject *
|
|
make_memoryview_from_NULL_pointer(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
Py_buffer info;
|
|
if (PyBuffer_FillInfo(&info, NULL, NULL, 1, 1, PyBUF_FULL_RO) < 0)
|
|
return NULL;
|
|
return PyMemoryView_FromBuffer(&info);
|
|
}
|
|
|
|
static PyObject *
|
|
buffer_fill_info(PyObject *self, PyObject *args)
|
|
{
|
|
Py_buffer info;
|
|
const char *data;
|
|
Py_ssize_t size;
|
|
int readonly;
|
|
int flags;
|
|
|
|
if (!PyArg_ParseTuple(args, "s#ii:buffer_fill_info",
|
|
&data, &size, &readonly, &flags)) {
|
|
return NULL;
|
|
}
|
|
|
|
if (PyBuffer_FillInfo(&info, NULL, (void *)data, size, readonly, flags) < 0) {
|
|
return NULL;
|
|
}
|
|
return PyMemoryView_FromBuffer(&info);
|
|
}
|
|
|
|
static PyObject *
|
|
test_from_contiguous(PyObject* self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
int data[9] = {-1,-1,-1,-1,-1,-1,-1,-1,-1};
|
|
int init[5] = {0, 1, 2, 3, 4};
|
|
Py_ssize_t itemsize = sizeof(int);
|
|
Py_ssize_t shape = 5;
|
|
Py_ssize_t strides = 2 * itemsize;
|
|
Py_buffer view = {
|
|
data,
|
|
NULL,
|
|
5 * itemsize,
|
|
itemsize,
|
|
1,
|
|
1,
|
|
NULL,
|
|
&shape,
|
|
&strides,
|
|
NULL,
|
|
NULL
|
|
};
|
|
int *ptr;
|
|
int i;
|
|
|
|
PyBuffer_FromContiguous(&view, init, view.len, 'C');
|
|
ptr = view.buf;
|
|
for (i = 0; i < 5; i++) {
|
|
if (ptr[2*i] != i) {
|
|
PyErr_SetString(get_testerror(self),
|
|
"test_from_contiguous: incorrect result");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
view.buf = &data[8];
|
|
view.strides[0] = -2 * itemsize;
|
|
|
|
PyBuffer_FromContiguous(&view, init, view.len, 'C');
|
|
ptr = view.buf;
|
|
for (i = 0; i < 5; i++) {
|
|
if (*(ptr-2*i) != i) {
|
|
PyErr_SetString(get_testerror(self),
|
|
"test_from_contiguous: incorrect result");
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
#if (defined(__linux__) || defined(__FreeBSD__)) && defined(__GNUC__)
|
|
|
|
static PyObject *
|
|
test_pep3118_obsolete_write_locks(PyObject* self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *b;
|
|
char *dummy[1];
|
|
int ret, match;
|
|
|
|
/* PyBuffer_FillInfo() */
|
|
ret = PyBuffer_FillInfo(NULL, NULL, dummy, 1, 0, PyBUF_SIMPLE);
|
|
match = PyErr_Occurred() && PyErr_ExceptionMatches(PyExc_BufferError);
|
|
PyErr_Clear();
|
|
if (ret != -1 || match == 0)
|
|
goto error;
|
|
|
|
PyObject *mod_io = PyImport_ImportModule("_io");
|
|
if (mod_io == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
/* bytesiobuf_getbuffer() */
|
|
PyTypeObject *type = (PyTypeObject *)PyObject_GetAttrString(
|
|
mod_io, "_BytesIOBuffer");
|
|
Py_DECREF(mod_io);
|
|
if (type == NULL) {
|
|
return NULL;
|
|
}
|
|
b = type->tp_alloc(type, 0);
|
|
Py_DECREF(type);
|
|
if (b == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
ret = PyObject_GetBuffer(b, NULL, PyBUF_SIMPLE);
|
|
Py_DECREF(b);
|
|
match = PyErr_Occurred() && PyErr_ExceptionMatches(PyExc_BufferError);
|
|
PyErr_Clear();
|
|
if (ret != -1 || match == 0)
|
|
goto error;
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
error:
|
|
PyErr_SetString(get_testerror(self),
|
|
"test_pep3118_obsolete_write_locks: failure");
|
|
return NULL;
|
|
}
|
|
#endif
|
|
|
|
/* This tests functions that historically supported write locks. It is
|
|
wrong to call getbuffer() with view==NULL and a compliant getbufferproc
|
|
is entitled to segfault in that case. */
|
|
static PyObject *
|
|
getbuffer_with_null_view(PyObject* self, PyObject *obj)
|
|
{
|
|
if (PyObject_GetBuffer(obj, NULL, PyBUF_SIMPLE) < 0)
|
|
return NULL;
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* PyBuffer_SizeFromFormat() */
|
|
static PyObject *
|
|
test_PyBuffer_SizeFromFormat(PyObject *self, PyObject *args)
|
|
{
|
|
const char *format;
|
|
|
|
if (!PyArg_ParseTuple(args, "s:test_PyBuffer_SizeFromFormat",
|
|
&format)) {
|
|
return NULL;
|
|
}
|
|
|
|
RETURN_SIZE(PyBuffer_SizeFromFormat(format));
|
|
}
|
|
|
|
/* Test that the fatal error from not having a current thread doesn't
|
|
cause an infinite loop. Run via Lib/test/test_capi.py */
|
|
static PyObject *
|
|
crash_no_current_thread(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
Py_BEGIN_ALLOW_THREADS
|
|
/* Using PyThreadState_Get() directly allows the test to pass in
|
|
!pydebug mode. However, the test only actually tests anything
|
|
in pydebug mode, since that's where the infinite loop was in
|
|
the first place. */
|
|
PyThreadState_Get();
|
|
Py_END_ALLOW_THREADS
|
|
return NULL;
|
|
}
|
|
|
|
/* Test that the GILState thread and the "current" thread match. */
|
|
static PyObject *
|
|
test_current_tstate_matches(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyThreadState *orig_tstate = PyThreadState_Get();
|
|
|
|
if (orig_tstate != PyGILState_GetThisThreadState()) {
|
|
PyErr_SetString(PyExc_RuntimeError,
|
|
"current thread state doesn't match GILState");
|
|
return NULL;
|
|
}
|
|
|
|
const char *err = NULL;
|
|
PyThreadState_Swap(NULL);
|
|
PyThreadState *substate = Py_NewInterpreter();
|
|
|
|
if (substate != PyThreadState_Get()) {
|
|
err = "subinterpreter thread state not current";
|
|
goto finally;
|
|
}
|
|
if (substate != PyGILState_GetThisThreadState()) {
|
|
err = "subinterpreter thread state doesn't match GILState";
|
|
goto finally;
|
|
}
|
|
|
|
finally:
|
|
Py_EndInterpreter(substate);
|
|
PyThreadState_Swap(orig_tstate);
|
|
|
|
if (err != NULL) {
|
|
PyErr_SetString(PyExc_RuntimeError, err);
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* To run some code in a sub-interpreter. */
|
|
static PyObject *
|
|
run_in_subinterp(PyObject *self, PyObject *args)
|
|
{
|
|
const char *code;
|
|
int r;
|
|
PyThreadState *substate, *mainstate;
|
|
/* only initialise 'cflags.cf_flags' to test backwards compatibility */
|
|
PyCompilerFlags cflags = {0};
|
|
|
|
if (!PyArg_ParseTuple(args, "s:run_in_subinterp",
|
|
&code))
|
|
return NULL;
|
|
|
|
mainstate = PyThreadState_Get();
|
|
|
|
PyThreadState_Swap(NULL);
|
|
|
|
substate = Py_NewInterpreter();
|
|
if (substate == NULL) {
|
|
/* Since no new thread state was created, there is no exception to
|
|
propagate; raise a fresh one after swapping in the old thread
|
|
state. */
|
|
PyThreadState_Swap(mainstate);
|
|
PyErr_SetString(PyExc_RuntimeError, "sub-interpreter creation failed");
|
|
return NULL;
|
|
}
|
|
r = PyRun_SimpleStringFlags(code, &cflags);
|
|
Py_EndInterpreter(substate);
|
|
|
|
PyThreadState_Swap(mainstate);
|
|
|
|
return PyLong_FromLong(r);
|
|
}
|
|
|
|
static PyMethodDef ml;
|
|
|
|
static PyObject *
|
|
create_cfunction(PyObject *self, PyObject *args)
|
|
{
|
|
return PyCFunction_NewEx(&ml, self, NULL);
|
|
}
|
|
|
|
static PyMethodDef ml = {
|
|
"create_cfunction",
|
|
create_cfunction,
|
|
METH_NOARGS,
|
|
NULL
|
|
};
|
|
|
|
static PyObject *
|
|
_test_incref(PyObject *ob)
|
|
{
|
|
return Py_NewRef(ob);
|
|
}
|
|
|
|
static PyObject *
|
|
test_xincref_doesnt_leak(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *obj = PyLong_FromLong(0);
|
|
Py_XINCREF(_test_incref(obj));
|
|
Py_DECREF(obj);
|
|
Py_DECREF(obj);
|
|
Py_DECREF(obj);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_incref_doesnt_leak(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *obj = PyLong_FromLong(0);
|
|
Py_INCREF(_test_incref(obj));
|
|
Py_DECREF(obj);
|
|
Py_DECREF(obj);
|
|
Py_DECREF(obj);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_xdecref_doesnt_leak(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
Py_XDECREF(PyLong_FromLong(0));
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_decref_doesnt_leak(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
Py_DECREF(PyLong_FromLong(0));
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_structseq_newtype_doesnt_leak(PyObject *Py_UNUSED(self),
|
|
PyObject *Py_UNUSED(args))
|
|
{
|
|
PyStructSequence_Desc descr;
|
|
PyStructSequence_Field descr_fields[3];
|
|
|
|
descr_fields[0] = (PyStructSequence_Field){"foo", "foo value"};
|
|
descr_fields[1] = (PyStructSequence_Field){NULL, "some hidden value"};
|
|
descr_fields[2] = (PyStructSequence_Field){0, NULL};
|
|
|
|
descr.name = "_testcapi.test_descr";
|
|
descr.doc = "This is used to test for memory leaks in NewType";
|
|
descr.fields = descr_fields;
|
|
descr.n_in_sequence = 1;
|
|
|
|
PyTypeObject* structseq_type = PyStructSequence_NewType(&descr);
|
|
if (structseq_type == NULL) {
|
|
return NULL;
|
|
}
|
|
assert(PyType_Check(structseq_type));
|
|
assert(PyType_FastSubclass(structseq_type, Py_TPFLAGS_TUPLE_SUBCLASS));
|
|
Py_DECREF(structseq_type);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_structseq_newtype_null_descr_doc(PyObject *Py_UNUSED(self),
|
|
PyObject *Py_UNUSED(args))
|
|
{
|
|
PyStructSequence_Field descr_fields[1] = {
|
|
(PyStructSequence_Field){NULL, NULL}
|
|
};
|
|
// Test specifically for NULL .doc field.
|
|
PyStructSequence_Desc descr = {"_testcapi.test_descr", NULL, &descr_fields[0], 0};
|
|
|
|
PyTypeObject* structseq_type = PyStructSequence_NewType(&descr);
|
|
assert(structseq_type != NULL);
|
|
assert(PyType_Check(structseq_type));
|
|
assert(PyType_FastSubclass(structseq_type, Py_TPFLAGS_TUPLE_SUBCLASS));
|
|
Py_DECREF(structseq_type);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_incref_decref_API(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *obj = PyLong_FromLong(0);
|
|
Py_IncRef(obj);
|
|
Py_DecRef(obj);
|
|
Py_DecRef(obj);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
typedef struct {
|
|
PyThread_type_lock start_event;
|
|
PyThread_type_lock exit_event;
|
|
PyObject *callback;
|
|
} test_c_thread_t;
|
|
|
|
static void
|
|
temporary_c_thread(void *data)
|
|
{
|
|
test_c_thread_t *test_c_thread = data;
|
|
PyGILState_STATE state;
|
|
PyObject *res;
|
|
|
|
PyThread_release_lock(test_c_thread->start_event);
|
|
|
|
/* Allocate a Python thread state for this thread */
|
|
state = PyGILState_Ensure();
|
|
|
|
res = PyObject_CallNoArgs(test_c_thread->callback);
|
|
Py_CLEAR(test_c_thread->callback);
|
|
|
|
if (res == NULL) {
|
|
PyErr_Print();
|
|
}
|
|
else {
|
|
Py_DECREF(res);
|
|
}
|
|
|
|
/* Destroy the Python thread state for this thread */
|
|
PyGILState_Release(state);
|
|
|
|
PyThread_release_lock(test_c_thread->exit_event);
|
|
}
|
|
|
|
static test_c_thread_t test_c_thread;
|
|
|
|
static PyObject *
|
|
call_in_temporary_c_thread(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *res = NULL;
|
|
PyObject *callback = NULL;
|
|
long thread;
|
|
int wait = 1;
|
|
if (!PyArg_ParseTuple(args, "O|i", &callback, &wait))
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
test_c_thread.start_event = PyThread_allocate_lock();
|
|
test_c_thread.exit_event = PyThread_allocate_lock();
|
|
test_c_thread.callback = NULL;
|
|
if (!test_c_thread.start_event || !test_c_thread.exit_event) {
|
|
PyErr_SetString(PyExc_RuntimeError, "could not allocate lock");
|
|
goto exit;
|
|
}
|
|
|
|
test_c_thread.callback = Py_NewRef(callback);
|
|
|
|
PyThread_acquire_lock(test_c_thread.start_event, 1);
|
|
PyThread_acquire_lock(test_c_thread.exit_event, 1);
|
|
|
|
thread = PyThread_start_new_thread(temporary_c_thread, &test_c_thread);
|
|
if (thread == -1) {
|
|
PyErr_SetString(PyExc_RuntimeError, "unable to start the thread");
|
|
PyThread_release_lock(test_c_thread.start_event);
|
|
PyThread_release_lock(test_c_thread.exit_event);
|
|
goto exit;
|
|
}
|
|
|
|
PyThread_acquire_lock(test_c_thread.start_event, 1);
|
|
PyThread_release_lock(test_c_thread.start_event);
|
|
|
|
if (!wait) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
Py_BEGIN_ALLOW_THREADS
|
|
PyThread_acquire_lock(test_c_thread.exit_event, 1);
|
|
PyThread_release_lock(test_c_thread.exit_event);
|
|
Py_END_ALLOW_THREADS
|
|
|
|
res = Py_NewRef(Py_None);
|
|
|
|
exit:
|
|
Py_CLEAR(test_c_thread.callback);
|
|
if (test_c_thread.start_event) {
|
|
PyThread_free_lock(test_c_thread.start_event);
|
|
test_c_thread.start_event = NULL;
|
|
}
|
|
if (test_c_thread.exit_event) {
|
|
PyThread_free_lock(test_c_thread.exit_event);
|
|
test_c_thread.exit_event = NULL;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
static PyObject *
|
|
join_temporary_c_thread(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
Py_BEGIN_ALLOW_THREADS
|
|
PyThread_acquire_lock(test_c_thread.exit_event, 1);
|
|
PyThread_release_lock(test_c_thread.exit_event);
|
|
Py_END_ALLOW_THREADS
|
|
Py_CLEAR(test_c_thread.callback);
|
|
PyThread_free_lock(test_c_thread.start_event);
|
|
test_c_thread.start_event = NULL;
|
|
PyThread_free_lock(test_c_thread.exit_event);
|
|
test_c_thread.exit_event = NULL;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/* marshal */
|
|
|
|
static PyObject*
|
|
pymarshal_write_long_to_file(PyObject* self, PyObject *args)
|
|
{
|
|
long value;
|
|
PyObject *filename;
|
|
int version;
|
|
FILE *fp;
|
|
|
|
if (!PyArg_ParseTuple(args, "lOi:pymarshal_write_long_to_file",
|
|
&value, &filename, &version))
|
|
return NULL;
|
|
|
|
fp = _Py_fopen_obj(filename, "wb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
PyMarshal_WriteLongToFile(value, fp, version);
|
|
assert(!PyErr_Occurred());
|
|
|
|
fclose(fp);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
pymarshal_write_object_to_file(PyObject* self, PyObject *args)
|
|
{
|
|
PyObject *obj;
|
|
PyObject *filename;
|
|
int version;
|
|
FILE *fp;
|
|
|
|
if (!PyArg_ParseTuple(args, "OOi:pymarshal_write_object_to_file",
|
|
&obj, &filename, &version))
|
|
return NULL;
|
|
|
|
fp = _Py_fopen_obj(filename, "wb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
PyMarshal_WriteObjectToFile(obj, fp, version);
|
|
assert(!PyErr_Occurred());
|
|
|
|
fclose(fp);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
pymarshal_read_short_from_file(PyObject* self, PyObject *args)
|
|
{
|
|
int value;
|
|
long pos;
|
|
PyObject *filename;
|
|
FILE *fp;
|
|
|
|
if (!PyArg_ParseTuple(args, "O:pymarshal_read_short_from_file", &filename))
|
|
return NULL;
|
|
|
|
fp = _Py_fopen_obj(filename, "rb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
value = PyMarshal_ReadShortFromFile(fp);
|
|
pos = ftell(fp);
|
|
|
|
fclose(fp);
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
return Py_BuildValue("il", value, pos);
|
|
}
|
|
|
|
static PyObject*
|
|
pymarshal_read_long_from_file(PyObject* self, PyObject *args)
|
|
{
|
|
long value, pos;
|
|
PyObject *filename;
|
|
FILE *fp;
|
|
|
|
if (!PyArg_ParseTuple(args, "O:pymarshal_read_long_from_file", &filename))
|
|
return NULL;
|
|
|
|
fp = _Py_fopen_obj(filename, "rb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
value = PyMarshal_ReadLongFromFile(fp);
|
|
pos = ftell(fp);
|
|
|
|
fclose(fp);
|
|
if (PyErr_Occurred())
|
|
return NULL;
|
|
return Py_BuildValue("ll", value, pos);
|
|
}
|
|
|
|
static PyObject*
|
|
pymarshal_read_last_object_from_file(PyObject* self, PyObject *args)
|
|
{
|
|
PyObject *filename;
|
|
if (!PyArg_ParseTuple(args, "O:pymarshal_read_last_object_from_file", &filename))
|
|
return NULL;
|
|
|
|
FILE *fp = _Py_fopen_obj(filename, "rb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *obj = PyMarshal_ReadLastObjectFromFile(fp);
|
|
long pos = ftell(fp);
|
|
|
|
fclose(fp);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue("Nl", obj, pos);
|
|
}
|
|
|
|
static PyObject*
|
|
pymarshal_read_object_from_file(PyObject* self, PyObject *args)
|
|
{
|
|
PyObject *filename;
|
|
if (!PyArg_ParseTuple(args, "O:pymarshal_read_object_from_file", &filename))
|
|
return NULL;
|
|
|
|
FILE *fp = _Py_fopen_obj(filename, "rb");
|
|
if (fp == NULL) {
|
|
PyErr_SetFromErrno(PyExc_OSError);
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *obj = PyMarshal_ReadObjectFromFile(fp);
|
|
long pos = ftell(fp);
|
|
|
|
fclose(fp);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
return Py_BuildValue("Nl", obj, pos);
|
|
}
|
|
|
|
static PyObject*
|
|
return_null_without_error(PyObject *self, PyObject *args)
|
|
{
|
|
/* invalid call: return NULL without setting an error,
|
|
* _Py_CheckFunctionResult() must detect such bug at runtime. */
|
|
PyErr_Clear();
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject*
|
|
return_result_with_error(PyObject *self, PyObject *args)
|
|
{
|
|
/* invalid call: return a result with an error set,
|
|
* _Py_CheckFunctionResult() must detect such bug at runtime. */
|
|
PyErr_SetNone(PyExc_ValueError);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
getitem_with_error(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *map, *key;
|
|
if (!PyArg_ParseTuple(args, "OO", &map, &key)) {
|
|
return NULL;
|
|
}
|
|
|
|
PyErr_SetString(PyExc_ValueError, "bug");
|
|
return PyObject_GetItem(map, key);
|
|
}
|
|
|
|
static PyObject *
|
|
dict_get_version(PyObject *self, PyObject *args)
|
|
{
|
|
PyDictObject *dict;
|
|
uint64_t version;
|
|
|
|
if (!PyArg_ParseTuple(args, "O!", &PyDict_Type, &dict))
|
|
return NULL;
|
|
|
|
_Py_COMP_DIAG_PUSH
|
|
_Py_COMP_DIAG_IGNORE_DEPR_DECLS
|
|
version = dict->ma_version_tag;
|
|
_Py_COMP_DIAG_POP
|
|
|
|
static_assert(sizeof(unsigned long long) >= sizeof(version),
|
|
"version is larger than unsigned long long");
|
|
return PyLong_FromUnsignedLongLong((unsigned long long)version);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
raise_SIGINT_then_send_None(PyObject *self, PyObject *args)
|
|
{
|
|
PyGenObject *gen;
|
|
|
|
if (!PyArg_ParseTuple(args, "O!", &PyGen_Type, &gen))
|
|
return NULL;
|
|
|
|
/* This is used in a test to check what happens if a signal arrives just
|
|
as we're in the process of entering a yield from chain (see
|
|
bpo-30039).
|
|
|
|
Needs to be done in C, because:
|
|
- we don't have a Python wrapper for raise()
|
|
- we need to make sure that the Python-level signal handler doesn't run
|
|
*before* we enter the generator frame, which is impossible in Python
|
|
because we check for signals before every bytecode operation.
|
|
*/
|
|
raise(SIGINT);
|
|
return PyObject_CallMethod((PyObject *)gen, "send", "O", Py_None);
|
|
}
|
|
|
|
|
|
static PyObject*
|
|
stack_pointer(PyObject *self, PyObject *args)
|
|
{
|
|
int v = 5;
|
|
return PyLong_FromVoidPtr(&v);
|
|
}
|
|
|
|
|
|
#ifdef W_STOPCODE
|
|
static PyObject*
|
|
py_w_stopcode(PyObject *self, PyObject *args)
|
|
{
|
|
int sig, status;
|
|
if (!PyArg_ParseTuple(args, "i", &sig)) {
|
|
return NULL;
|
|
}
|
|
status = W_STOPCODE(sig);
|
|
return PyLong_FromLong(status);
|
|
}
|
|
#endif
|
|
|
|
|
|
static PyObject *
|
|
test_pythread_tss_key_state(PyObject *self, PyObject *args)
|
|
{
|
|
Py_tss_t tss_key = Py_tss_NEEDS_INIT;
|
|
if (PyThread_tss_is_created(&tss_key)) {
|
|
return raiseTestError(self, "test_pythread_tss_key_state",
|
|
"TSS key not in an uninitialized state at "
|
|
"creation time");
|
|
}
|
|
if (PyThread_tss_create(&tss_key) != 0) {
|
|
PyErr_SetString(PyExc_RuntimeError, "PyThread_tss_create failed");
|
|
return NULL;
|
|
}
|
|
if (!PyThread_tss_is_created(&tss_key)) {
|
|
return raiseTestError(self, "test_pythread_tss_key_state",
|
|
"PyThread_tss_create succeeded, "
|
|
"but with TSS key in an uninitialized state");
|
|
}
|
|
if (PyThread_tss_create(&tss_key) != 0) {
|
|
return raiseTestError(self, "test_pythread_tss_key_state",
|
|
"PyThread_tss_create unsuccessful with "
|
|
"an already initialized key");
|
|
}
|
|
#define CHECK_TSS_API(expr) \
|
|
do { \
|
|
(void)(expr); \
|
|
if (!PyThread_tss_is_created(&tss_key)) { \
|
|
return raiseTestError(self, "test_pythread_tss_key_state", \
|
|
"TSS key initialization state was not " \
|
|
"preserved after calling " #expr); \
|
|
} \
|
|
} while (0)
|
|
|
|
CHECK_TSS_API(PyThread_tss_set(&tss_key, NULL));
|
|
CHECK_TSS_API(PyThread_tss_get(&tss_key));
|
|
#undef CHECK_TSS_API
|
|
PyThread_tss_delete(&tss_key);
|
|
if (PyThread_tss_is_created(&tss_key)) {
|
|
return raiseTestError(self, "test_pythread_tss_key_state",
|
|
"PyThread_tss_delete called, but did not "
|
|
"set the key state to uninitialized");
|
|
}
|
|
|
|
Py_tss_t *ptr_key = PyThread_tss_alloc();
|
|
if (ptr_key == NULL) {
|
|
PyErr_SetString(PyExc_RuntimeError, "PyThread_tss_alloc failed");
|
|
return NULL;
|
|
}
|
|
if (PyThread_tss_is_created(ptr_key)) {
|
|
return raiseTestError(self, "test_pythread_tss_key_state",
|
|
"TSS key not in an uninitialized state at "
|
|
"allocation time");
|
|
}
|
|
PyThread_tss_free(ptr_key);
|
|
ptr_key = NULL;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
/* def bad_get(self, obj, cls):
|
|
cls()
|
|
return repr(self)
|
|
*/
|
|
static PyObject*
|
|
bad_get(PyObject *module, PyObject *args)
|
|
{
|
|
PyObject *self, *obj, *cls;
|
|
if (!PyArg_ParseTuple(args, "OOO", &self, &obj, &cls)) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject *res = PyObject_CallNoArgs(cls);
|
|
if (res == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_DECREF(res);
|
|
|
|
return PyObject_Repr(self);
|
|
}
|
|
|
|
|
|
#ifdef Py_REF_DEBUG
|
|
static PyObject *
|
|
negative_refcount(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
PyObject *obj = PyUnicode_FromString("negative_refcount");
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
assert(Py_REFCNT(obj) == 1);
|
|
|
|
Py_SET_REFCNT(obj, 0);
|
|
/* Py_DECREF() must call _Py_NegativeRefcount() and abort Python */
|
|
Py_DECREF(obj);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
decref_freed_object(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
PyObject *obj = PyUnicode_FromString("decref_freed_object");
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
assert(Py_REFCNT(obj) == 1);
|
|
|
|
// Deallocate the memory
|
|
Py_DECREF(obj);
|
|
// obj is a now a dangling pointer
|
|
|
|
// gh-109496: If Python is built in debug mode, Py_DECREF() must call
|
|
// _Py_NegativeRefcount() and abort Python.
|
|
Py_DECREF(obj);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
#endif
|
|
|
|
|
|
/* Functions for testing C calling conventions (METH_*) are named meth_*,
|
|
* e.g. "meth_varargs" for METH_VARARGS.
|
|
*
|
|
* They all return a tuple of their C-level arguments, with None instead
|
|
* of NULL and Python tuples instead of C arrays.
|
|
*/
|
|
|
|
|
|
static PyObject*
|
|
_null_to_none(PyObject* obj)
|
|
{
|
|
if (obj == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
return Py_NewRef(obj);
|
|
}
|
|
|
|
static PyObject*
|
|
meth_varargs(PyObject* self, PyObject* args)
|
|
{
|
|
return Py_BuildValue("NO", _null_to_none(self), args);
|
|
}
|
|
|
|
static PyObject*
|
|
meth_varargs_keywords(PyObject* self, PyObject* args, PyObject* kwargs)
|
|
{
|
|
return Py_BuildValue("NON", _null_to_none(self), args, _null_to_none(kwargs));
|
|
}
|
|
|
|
static PyObject*
|
|
meth_o(PyObject* self, PyObject* obj)
|
|
{
|
|
return Py_BuildValue("NO", _null_to_none(self), obj);
|
|
}
|
|
|
|
static PyObject*
|
|
meth_noargs(PyObject* self, PyObject* ignored)
|
|
{
|
|
return _null_to_none(self);
|
|
}
|
|
|
|
static PyObject*
|
|
_fastcall_to_tuple(PyObject* const* args, Py_ssize_t nargs)
|
|
{
|
|
PyObject *tuple = PyTuple_New(nargs);
|
|
if (tuple == NULL) {
|
|
return NULL;
|
|
}
|
|
for (Py_ssize_t i=0; i < nargs; i++) {
|
|
Py_INCREF(args[i]);
|
|
PyTuple_SET_ITEM(tuple, i, args[i]);
|
|
}
|
|
return tuple;
|
|
}
|
|
|
|
static PyObject*
|
|
meth_fastcall(PyObject* self, PyObject* const* args, Py_ssize_t nargs)
|
|
{
|
|
return Py_BuildValue(
|
|
"NN", _null_to_none(self), _fastcall_to_tuple(args, nargs)
|
|
);
|
|
}
|
|
|
|
static PyObject*
|
|
meth_fastcall_keywords(PyObject* self, PyObject* const* args,
|
|
Py_ssize_t nargs, PyObject* kwargs)
|
|
{
|
|
PyObject *pyargs = _fastcall_to_tuple(args, nargs);
|
|
if (pyargs == NULL) {
|
|
return NULL;
|
|
}
|
|
assert(args != NULL || nargs == 0);
|
|
PyObject* const* args_offset = args == NULL ? NULL : args + nargs;
|
|
PyObject *pykwargs = PyObject_Vectorcall((PyObject*)&PyDict_Type,
|
|
args_offset, 0, kwargs);
|
|
return Py_BuildValue("NNN", _null_to_none(self), pyargs, pykwargs);
|
|
}
|
|
|
|
static PyObject*
|
|
test_pycfunction_call(PyObject *module, PyObject *args)
|
|
{
|
|
// Function removed in the Python 3.13 API but was kept in the stable ABI.
|
|
extern PyObject* PyCFunction_Call(PyObject *callable, PyObject *args, PyObject *kwargs);
|
|
|
|
PyObject *func, *pos_args, *kwargs = NULL;
|
|
if (!PyArg_ParseTuple(args, "OO!|O!", &func, &PyTuple_Type, &pos_args, &PyDict_Type, &kwargs)) {
|
|
return NULL;
|
|
}
|
|
return PyCFunction_Call(func, pos_args, kwargs);
|
|
}
|
|
|
|
static PyObject*
|
|
pynumber_tobase(PyObject *module, PyObject *args)
|
|
{
|
|
PyObject *obj;
|
|
int base;
|
|
if (!PyArg_ParseTuple(args, "Oi:pynumber_tobase",
|
|
&obj, &base)) {
|
|
return NULL;
|
|
}
|
|
return PyNumber_ToBase(obj, base);
|
|
}
|
|
|
|
static PyObject*
|
|
test_set_type_size(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
PyObject *obj = PyList_New(0);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
// Ensure that following tests don't modify the object,
|
|
// to ensure that Py_DECREF() will not crash.
|
|
assert(Py_TYPE(obj) == &PyList_Type);
|
|
assert(Py_SIZE(obj) == 0);
|
|
|
|
// bpo-39573: Test Py_SET_TYPE() and Py_SET_SIZE() functions.
|
|
Py_SET_TYPE(obj, &PyList_Type);
|
|
Py_SET_SIZE(obj, 0);
|
|
|
|
Py_DECREF(obj);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
// Test Py_CLEAR() macro
|
|
static PyObject*
|
|
test_py_clear(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
// simple case with a variable
|
|
PyObject *obj = PyList_New(0);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_CLEAR(obj);
|
|
assert(obj == NULL);
|
|
|
|
// gh-98724: complex case, Py_CLEAR() argument has a side effect
|
|
PyObject* array[1];
|
|
array[0] = PyList_New(0);
|
|
if (array[0] == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject **p = array;
|
|
Py_CLEAR(*p++);
|
|
assert(array[0] == NULL);
|
|
assert(p == array + 1);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
// Test Py_SETREF() and Py_XSETREF() macros, similar to test_py_clear()
|
|
static PyObject*
|
|
test_py_setref(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
// Py_SETREF() simple case with a variable
|
|
PyObject *obj = PyList_New(0);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_SETREF(obj, NULL);
|
|
assert(obj == NULL);
|
|
|
|
// Py_XSETREF() simple case with a variable
|
|
PyObject *obj2 = PyList_New(0);
|
|
if (obj2 == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_XSETREF(obj2, NULL);
|
|
assert(obj2 == NULL);
|
|
// test Py_XSETREF() when the argument is NULL
|
|
Py_XSETREF(obj2, NULL);
|
|
assert(obj2 == NULL);
|
|
|
|
// gh-98724: complex case, Py_SETREF() argument has a side effect
|
|
PyObject* array[1];
|
|
array[0] = PyList_New(0);
|
|
if (array[0] == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject **p = array;
|
|
Py_SETREF(*p++, NULL);
|
|
assert(array[0] == NULL);
|
|
assert(p == array + 1);
|
|
|
|
// gh-98724: complex case, Py_XSETREF() argument has a side effect
|
|
PyObject* array2[1];
|
|
array2[0] = PyList_New(0);
|
|
if (array2[0] == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
PyObject **p2 = array2;
|
|
Py_XSETREF(*p2++, NULL);
|
|
assert(array2[0] == NULL);
|
|
assert(p2 == array2 + 1);
|
|
|
|
// test Py_XSETREF() when the argument is NULL
|
|
p2 = array2;
|
|
Py_XSETREF(*p2++, NULL);
|
|
assert(array2[0] == NULL);
|
|
assert(p2 == array2 + 1);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
#define TEST_REFCOUNT() \
|
|
do { \
|
|
PyObject *obj = PyList_New(0); \
|
|
if (obj == NULL) { \
|
|
return NULL; \
|
|
} \
|
|
assert(Py_REFCNT(obj) == 1); \
|
|
\
|
|
/* test Py_NewRef() */ \
|
|
PyObject *ref = Py_NewRef(obj); \
|
|
assert(ref == obj); \
|
|
assert(Py_REFCNT(obj) == 2); \
|
|
Py_DECREF(ref); \
|
|
\
|
|
/* test Py_XNewRef() */ \
|
|
PyObject *xref = Py_XNewRef(obj); \
|
|
assert(xref == obj); \
|
|
assert(Py_REFCNT(obj) == 2); \
|
|
Py_DECREF(xref); \
|
|
\
|
|
assert(Py_XNewRef(NULL) == NULL); \
|
|
\
|
|
Py_DECREF(obj); \
|
|
Py_RETURN_NONE; \
|
|
} while (0)
|
|
|
|
|
|
// Test Py_NewRef() and Py_XNewRef() macros
|
|
static PyObject*
|
|
test_refcount_macros(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
TEST_REFCOUNT();
|
|
}
|
|
|
|
#undef Py_NewRef
|
|
#undef Py_XNewRef
|
|
|
|
// Test Py_NewRef() and Py_XNewRef() functions, after undefining macros.
|
|
static PyObject*
|
|
test_refcount_funcs(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
TEST_REFCOUNT();
|
|
}
|
|
|
|
|
|
// Test Py_Is() function
|
|
#define TEST_PY_IS() \
|
|
do { \
|
|
PyObject *o_none = Py_None; \
|
|
PyObject *o_true = Py_True; \
|
|
PyObject *o_false = Py_False; \
|
|
PyObject *obj = PyList_New(0); \
|
|
if (obj == NULL) { \
|
|
return NULL; \
|
|
} \
|
|
\
|
|
/* test Py_Is() */ \
|
|
assert(Py_Is(obj, obj)); \
|
|
assert(!Py_Is(obj, o_none)); \
|
|
\
|
|
/* test Py_None */ \
|
|
assert(Py_Is(o_none, o_none)); \
|
|
assert(!Py_Is(obj, o_none)); \
|
|
\
|
|
/* test Py_True */ \
|
|
assert(Py_Is(o_true, o_true)); \
|
|
assert(!Py_Is(o_false, o_true)); \
|
|
assert(!Py_Is(obj, o_true)); \
|
|
\
|
|
/* test Py_False */ \
|
|
assert(Py_Is(o_false, o_false)); \
|
|
assert(!Py_Is(o_true, o_false)); \
|
|
assert(!Py_Is(obj, o_false)); \
|
|
\
|
|
Py_DECREF(obj); \
|
|
Py_RETURN_NONE; \
|
|
} while (0)
|
|
|
|
// Test Py_Is() macro
|
|
static PyObject*
|
|
test_py_is_macros(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
TEST_PY_IS();
|
|
}
|
|
|
|
#undef Py_Is
|
|
|
|
// Test Py_Is() function, after undefining its macro.
|
|
static PyObject*
|
|
test_py_is_funcs(PyObject *self, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
TEST_PY_IS();
|
|
}
|
|
|
|
|
|
// type->tp_version_tag
|
|
static PyObject *
|
|
type_get_version(PyObject *self, PyObject *type)
|
|
{
|
|
if (!PyType_Check(type)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a type");
|
|
return NULL;
|
|
}
|
|
PyObject *res = PyLong_FromUnsignedLong(
|
|
((PyTypeObject *)type)->tp_version_tag);
|
|
if (res == NULL) {
|
|
assert(PyErr_Occurred());
|
|
return NULL;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
static PyObject *
|
|
type_modified(PyObject *self, PyObject *type)
|
|
{
|
|
if (!PyType_Check(type)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a type");
|
|
return NULL;
|
|
}
|
|
PyType_Modified((PyTypeObject *)type);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
type_assign_version(PyObject *self, PyObject *type)
|
|
{
|
|
if (!PyType_Check(type)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a type");
|
|
return NULL;
|
|
}
|
|
int res = PyUnstable_Type_AssignVersionTag((PyTypeObject *)type);
|
|
return PyLong_FromLong(res);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
type_get_tp_bases(PyObject *self, PyObject *type)
|
|
{
|
|
PyObject *bases = ((PyTypeObject *)type)->tp_bases;
|
|
if (bases == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
return Py_NewRef(bases);
|
|
}
|
|
|
|
static PyObject *
|
|
type_get_tp_mro(PyObject *self, PyObject *type)
|
|
{
|
|
PyObject *mro = ((PyTypeObject *)type)->tp_mro;
|
|
if (mro == NULL) {
|
|
Py_RETURN_NONE;
|
|
}
|
|
return Py_NewRef(mro);
|
|
}
|
|
|
|
|
|
/* We only use 2 in test_capi/test_misc.py. */
|
|
#define NUM_BASIC_STATIC_TYPES 2
|
|
static PyTypeObject BasicStaticTypes[NUM_BASIC_STATIC_TYPES] = {
|
|
#define INIT_BASIC_STATIC_TYPE \
|
|
{ \
|
|
PyVarObject_HEAD_INIT(NULL, 0) \
|
|
.tp_name = "BasicStaticType", \
|
|
.tp_basicsize = sizeof(PyObject), \
|
|
}
|
|
INIT_BASIC_STATIC_TYPE,
|
|
INIT_BASIC_STATIC_TYPE,
|
|
#undef INIT_BASIC_STATIC_TYPE
|
|
};
|
|
static int num_basic_static_types_used = 0;
|
|
|
|
static PyObject *
|
|
get_basic_static_type(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *base = NULL;
|
|
if (!PyArg_ParseTuple(args, "|O", &base)) {
|
|
return NULL;
|
|
}
|
|
assert(base == NULL || PyType_Check(base));
|
|
|
|
if(num_basic_static_types_used >= NUM_BASIC_STATIC_TYPES) {
|
|
PyErr_SetString(PyExc_RuntimeError, "no more available basic static types");
|
|
return NULL;
|
|
}
|
|
PyTypeObject *cls = &BasicStaticTypes[num_basic_static_types_used++];
|
|
|
|
if (base != NULL) {
|
|
cls->tp_bases = PyTuple_Pack(1, base);
|
|
if (cls->tp_bases == NULL) {
|
|
return NULL;
|
|
}
|
|
cls->tp_base = (PyTypeObject *)Py_NewRef(base);
|
|
}
|
|
if (PyType_Ready(cls) < 0) {
|
|
Py_DECREF(cls->tp_bases);
|
|
Py_DECREF(cls->tp_base);
|
|
return NULL;
|
|
}
|
|
return (PyObject *)cls;
|
|
}
|
|
|
|
|
|
// Test PyThreadState C API
|
|
static PyObject *
|
|
test_tstate_capi(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
// PyThreadState_Get()
|
|
PyThreadState *tstate = PyThreadState_Get();
|
|
assert(tstate != NULL);
|
|
|
|
// PyThreadState_GET()
|
|
PyThreadState *tstate2 = PyThreadState_Get();
|
|
assert(tstate2 == tstate);
|
|
|
|
// PyThreadState_GetUnchecked()
|
|
PyThreadState *tstate3 = PyThreadState_GetUnchecked();
|
|
assert(tstate3 == tstate);
|
|
|
|
// PyThreadState_EnterTracing(), PyThreadState_LeaveTracing()
|
|
PyThreadState_EnterTracing(tstate);
|
|
PyThreadState_LeaveTracing(tstate);
|
|
|
|
// PyThreadState_GetDict(): no tstate argument
|
|
PyObject *dict = PyThreadState_GetDict();
|
|
// PyThreadState_GetDict() API can return NULL if PyDict_New() fails,
|
|
// but it should not occur in practice.
|
|
assert(dict != NULL);
|
|
assert(PyDict_Check(dict));
|
|
// dict is a borrowed reference
|
|
|
|
// PyThreadState_GetInterpreter()
|
|
PyInterpreterState *interp = PyThreadState_GetInterpreter(tstate);
|
|
assert(interp != NULL);
|
|
|
|
// PyThreadState_GetFrame()
|
|
PyFrameObject*frame = PyThreadState_GetFrame(tstate);
|
|
assert(frame != NULL);
|
|
assert(PyFrame_Check(frame));
|
|
Py_DECREF(frame);
|
|
|
|
// PyThreadState_GetID()
|
|
uint64_t id = PyThreadState_GetID(tstate);
|
|
assert(id >= 1);
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
frame_getlocals(PyObject *self, PyObject *frame)
|
|
{
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
return PyFrame_GetLocals((PyFrameObject *)frame);
|
|
}
|
|
|
|
static PyObject *
|
|
frame_getglobals(PyObject *self, PyObject *frame)
|
|
{
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
return PyFrame_GetGlobals((PyFrameObject *)frame);
|
|
}
|
|
|
|
static PyObject *
|
|
frame_getgenerator(PyObject *self, PyObject *frame)
|
|
{
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
return PyFrame_GetGenerator((PyFrameObject *)frame);
|
|
}
|
|
|
|
static PyObject *
|
|
frame_getbuiltins(PyObject *self, PyObject *frame)
|
|
{
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
return PyFrame_GetBuiltins((PyFrameObject *)frame);
|
|
}
|
|
|
|
static PyObject *
|
|
frame_getlasti(PyObject *self, PyObject *frame)
|
|
{
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
int lasti = PyFrame_GetLasti((PyFrameObject *)frame);
|
|
if (lasti < 0) {
|
|
assert(lasti == -1);
|
|
Py_RETURN_NONE;
|
|
}
|
|
return PyLong_FromLong(lasti);
|
|
}
|
|
|
|
static PyObject *
|
|
frame_new(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *code, *globals, *locals;
|
|
if (!PyArg_ParseTuple(args, "OOO", &code, &globals, &locals)) {
|
|
return NULL;
|
|
}
|
|
if (!PyCode_Check(code)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a code object");
|
|
return NULL;
|
|
}
|
|
PyThreadState *tstate = PyThreadState_Get();
|
|
|
|
return (PyObject *)PyFrame_New(tstate, (PyCodeObject *)code, globals, locals);
|
|
}
|
|
|
|
static PyObject *
|
|
test_frame_getvar(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *frame, *name;
|
|
if (!PyArg_ParseTuple(args, "OO", &frame, &name)) {
|
|
return NULL;
|
|
}
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
|
|
return PyFrame_GetVar((PyFrameObject *)frame, name);
|
|
}
|
|
|
|
static PyObject *
|
|
test_frame_getvarstring(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *frame;
|
|
const char *name;
|
|
if (!PyArg_ParseTuple(args, "Oy", &frame, &name)) {
|
|
return NULL;
|
|
}
|
|
if (!PyFrame_Check(frame)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a frame");
|
|
return NULL;
|
|
}
|
|
|
|
return PyFrame_GetVarString((PyFrameObject *)frame, name);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
gen_get_code(PyObject *self, PyObject *gen)
|
|
{
|
|
if (!PyGen_Check(gen)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a generator object");
|
|
return NULL;
|
|
}
|
|
return (PyObject *)PyGen_GetCode((PyGenObject *)gen);
|
|
}
|
|
|
|
static PyObject *
|
|
eval_eval_code_ex(PyObject *mod, PyObject *pos_args)
|
|
{
|
|
PyObject *result = NULL;
|
|
PyObject *code;
|
|
PyObject *globals;
|
|
PyObject *locals = NULL;
|
|
PyObject *args = NULL;
|
|
PyObject *kwargs = NULL;
|
|
PyObject *defaults = NULL;
|
|
PyObject *kw_defaults = NULL;
|
|
PyObject *closure = NULL;
|
|
|
|
PyObject **c_kwargs = NULL;
|
|
|
|
if (!PyArg_ParseTuple(pos_args,
|
|
"OO|OO!O!O!OO:eval_code_ex",
|
|
&code,
|
|
&globals,
|
|
&locals,
|
|
&PyTuple_Type, &args,
|
|
&PyDict_Type, &kwargs,
|
|
&PyTuple_Type, &defaults,
|
|
&kw_defaults,
|
|
&closure))
|
|
{
|
|
goto exit;
|
|
}
|
|
|
|
NULLABLE(code);
|
|
NULLABLE(globals);
|
|
NULLABLE(locals);
|
|
NULLABLE(kw_defaults);
|
|
NULLABLE(closure);
|
|
|
|
PyObject **c_args = NULL;
|
|
Py_ssize_t c_args_len = 0;
|
|
if (args) {
|
|
c_args = &PyTuple_GET_ITEM(args, 0);
|
|
c_args_len = PyTuple_Size(args);
|
|
}
|
|
|
|
Py_ssize_t c_kwargs_len = 0;
|
|
if (kwargs) {
|
|
c_kwargs_len = PyDict_Size(kwargs);
|
|
if (c_kwargs_len > 0) {
|
|
c_kwargs = PyMem_NEW(PyObject*, 2 * c_kwargs_len);
|
|
if (!c_kwargs) {
|
|
PyErr_NoMemory();
|
|
goto exit;
|
|
}
|
|
|
|
Py_ssize_t i = 0;
|
|
Py_ssize_t pos = 0;
|
|
while (PyDict_Next(kwargs, &pos, &c_kwargs[i], &c_kwargs[i + 1])) {
|
|
i += 2;
|
|
}
|
|
c_kwargs_len = i / 2;
|
|
/* XXX This is broken if the caller deletes dict items! */
|
|
}
|
|
}
|
|
|
|
PyObject **c_defaults = NULL;
|
|
Py_ssize_t c_defaults_len = 0;
|
|
if (defaults) {
|
|
c_defaults = &PyTuple_GET_ITEM(defaults, 0);
|
|
c_defaults_len = PyTuple_Size(defaults);
|
|
}
|
|
|
|
result = PyEval_EvalCodeEx(
|
|
code,
|
|
globals,
|
|
locals,
|
|
c_args,
|
|
(int)c_args_len,
|
|
c_kwargs,
|
|
(int)c_kwargs_len,
|
|
c_defaults,
|
|
(int)c_defaults_len,
|
|
kw_defaults,
|
|
closure
|
|
);
|
|
|
|
exit:
|
|
if (c_kwargs) {
|
|
PyMem_DEL(c_kwargs);
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
get_feature_macros(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
PyObject *result = PyDict_New();
|
|
if (!result) {
|
|
return NULL;
|
|
}
|
|
int res;
|
|
#include "_testcapi_feature_macros.inc"
|
|
return result;
|
|
}
|
|
|
|
static PyObject *
|
|
test_code_api(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
PyCodeObject *co = PyCode_NewEmpty("_testcapi", "dummy", 1);
|
|
if (co == NULL) {
|
|
return NULL;
|
|
}
|
|
/* co_code */
|
|
{
|
|
PyObject *co_code = PyCode_GetCode(co);
|
|
if (co_code == NULL) {
|
|
goto fail;
|
|
}
|
|
assert(PyBytes_CheckExact(co_code));
|
|
if (PyObject_Length(co_code) == 0) {
|
|
PyErr_SetString(PyExc_ValueError, "empty co_code");
|
|
Py_DECREF(co_code);
|
|
goto fail;
|
|
}
|
|
Py_DECREF(co_code);
|
|
}
|
|
/* co_varnames */
|
|
{
|
|
PyObject *co_varnames = PyCode_GetVarnames(co);
|
|
if (co_varnames == NULL) {
|
|
goto fail;
|
|
}
|
|
if (!PyTuple_CheckExact(co_varnames)) {
|
|
PyErr_SetString(PyExc_TypeError, "co_varnames not tuple");
|
|
Py_DECREF(co_varnames);
|
|
goto fail;
|
|
}
|
|
if (PyTuple_GET_SIZE(co_varnames) != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "non-empty co_varnames");
|
|
Py_DECREF(co_varnames);
|
|
goto fail;
|
|
}
|
|
Py_DECREF(co_varnames);
|
|
}
|
|
/* co_cellvars */
|
|
{
|
|
PyObject *co_cellvars = PyCode_GetCellvars(co);
|
|
if (co_cellvars == NULL) {
|
|
goto fail;
|
|
}
|
|
if (!PyTuple_CheckExact(co_cellvars)) {
|
|
PyErr_SetString(PyExc_TypeError, "co_cellvars not tuple");
|
|
Py_DECREF(co_cellvars);
|
|
goto fail;
|
|
}
|
|
if (PyTuple_GET_SIZE(co_cellvars) != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "non-empty co_cellvars");
|
|
Py_DECREF(co_cellvars);
|
|
goto fail;
|
|
}
|
|
Py_DECREF(co_cellvars);
|
|
}
|
|
/* co_freevars */
|
|
{
|
|
PyObject *co_freevars = PyCode_GetFreevars(co);
|
|
if (co_freevars == NULL) {
|
|
goto fail;
|
|
}
|
|
if (!PyTuple_CheckExact(co_freevars)) {
|
|
PyErr_SetString(PyExc_TypeError, "co_freevars not tuple");
|
|
Py_DECREF(co_freevars);
|
|
goto fail;
|
|
}
|
|
if (PyTuple_GET_SIZE(co_freevars) != 0) {
|
|
PyErr_SetString(PyExc_ValueError, "non-empty co_freevars");
|
|
Py_DECREF(co_freevars);
|
|
goto fail;
|
|
}
|
|
Py_DECREF(co_freevars);
|
|
}
|
|
Py_DECREF(co);
|
|
Py_RETURN_NONE;
|
|
fail:
|
|
Py_DECREF(co);
|
|
return NULL;
|
|
}
|
|
|
|
static int
|
|
record_func(PyObject *obj, PyFrameObject *f, int what, PyObject *arg)
|
|
{
|
|
assert(PyList_Check(obj));
|
|
PyObject *what_obj = NULL;
|
|
PyObject *line_obj = NULL;
|
|
PyObject *tuple = NULL;
|
|
int res = -1;
|
|
what_obj = PyLong_FromLong(what);
|
|
if (what_obj == NULL) {
|
|
goto error;
|
|
}
|
|
int line = PyFrame_GetLineNumber(f);
|
|
line_obj = PyLong_FromLong(line);
|
|
if (line_obj == NULL) {
|
|
goto error;
|
|
}
|
|
tuple = PyTuple_Pack(3, what_obj, line_obj, arg);
|
|
if (tuple == NULL) {
|
|
goto error;
|
|
}
|
|
PyTuple_SET_ITEM(tuple, 0, what_obj);
|
|
if (PyList_Append(obj, tuple)) {
|
|
goto error;
|
|
}
|
|
res = 0;
|
|
error:
|
|
Py_XDECREF(what_obj);
|
|
Py_XDECREF(line_obj);
|
|
Py_XDECREF(tuple);
|
|
return res;
|
|
}
|
|
|
|
static PyObject *
|
|
settrace_to_record(PyObject *self, PyObject *list)
|
|
{
|
|
|
|
if (!PyList_Check(list)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a list");
|
|
return NULL;
|
|
}
|
|
PyEval_SetTrace(record_func, list);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static int
|
|
error_func(PyObject *obj, PyFrameObject *f, int what, PyObject *arg)
|
|
{
|
|
assert(PyList_Check(obj));
|
|
/* Only raise if list is empty, otherwise append None
|
|
* This ensures that we only raise once */
|
|
if (PyList_GET_SIZE(obj)) {
|
|
return 0;
|
|
}
|
|
if (PyList_Append(obj, Py_None)) {
|
|
return -1;
|
|
}
|
|
PyErr_SetString(PyExc_Exception, "an exception");
|
|
return -1;
|
|
}
|
|
|
|
static PyObject *
|
|
settrace_to_error(PyObject *self, PyObject *list)
|
|
{
|
|
if (!PyList_Check(list)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument must be a list");
|
|
return NULL;
|
|
}
|
|
PyEval_SetTrace(error_func, list);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
clear_managed_dict(PyObject *self, PyObject *obj)
|
|
{
|
|
PyObject_ClearManagedDict(obj);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
test_macros(PyObject *self, PyObject *Py_UNUSED(args))
|
|
{
|
|
struct MyStruct {
|
|
int x;
|
|
};
|
|
wchar_t array[3];
|
|
|
|
// static_assert(), Py_BUILD_ASSERT()
|
|
static_assert(1 == 1, "bug");
|
|
Py_BUILD_ASSERT(1 == 1);
|
|
|
|
|
|
// Py_MIN(), Py_MAX(), Py_ABS()
|
|
assert(Py_MIN(5, 11) == 5);
|
|
assert(Py_MAX(5, 11) == 11);
|
|
assert(Py_ABS(-5) == 5);
|
|
|
|
// Py_STRINGIFY()
|
|
assert(strcmp(Py_STRINGIFY(123), "123") == 0);
|
|
|
|
// Py_MEMBER_SIZE(), Py_ARRAY_LENGTH()
|
|
assert(Py_MEMBER_SIZE(struct MyStruct, x) == sizeof(int));
|
|
assert(Py_ARRAY_LENGTH(array) == 3);
|
|
|
|
// Py_CHARMASK()
|
|
int c = 0xab00 | 7;
|
|
assert(Py_CHARMASK(c) == 7);
|
|
|
|
// _Py_IS_TYPE_SIGNED()
|
|
assert(_Py_IS_TYPE_SIGNED(int));
|
|
assert(!_Py_IS_TYPE_SIGNED(unsigned int));
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_code(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *code = PyFunction_GetCode(func);
|
|
if (code != NULL) {
|
|
return Py_NewRef(code);
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_globals(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *globals = PyFunction_GetGlobals(func);
|
|
if (globals != NULL) {
|
|
return Py_NewRef(globals);
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_module(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *module = PyFunction_GetModule(func);
|
|
if (module != NULL) {
|
|
return Py_NewRef(module);
|
|
} else {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_defaults(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *defaults = PyFunction_GetDefaults(func);
|
|
if (defaults != NULL) {
|
|
return Py_NewRef(defaults);
|
|
} else if (PyErr_Occurred()) {
|
|
return NULL;
|
|
} else {
|
|
Py_RETURN_NONE; // This can happen when `defaults` are set to `None`
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_set_defaults(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *func = NULL, *defaults = NULL;
|
|
if (!PyArg_ParseTuple(args, "OO", &func, &defaults)) {
|
|
return NULL;
|
|
}
|
|
int result = PyFunction_SetDefaults(func, defaults);
|
|
if (result == -1)
|
|
return NULL;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_kw_defaults(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *defaults = PyFunction_GetKwDefaults(func);
|
|
if (defaults != NULL) {
|
|
return Py_NewRef(defaults);
|
|
} else if (PyErr_Occurred()) {
|
|
return NULL;
|
|
} else {
|
|
Py_RETURN_NONE; // This can happen when `kwdefaults` are set to `None`
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_set_kw_defaults(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *func = NULL, *defaults = NULL;
|
|
if (!PyArg_ParseTuple(args, "OO", &func, &defaults)) {
|
|
return NULL;
|
|
}
|
|
int result = PyFunction_SetKwDefaults(func, defaults);
|
|
if (result == -1)
|
|
return NULL;
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
function_get_closure(PyObject *self, PyObject *func)
|
|
{
|
|
PyObject *closure = PyFunction_GetClosure(func);
|
|
if (closure != NULL) {
|
|
return Py_NewRef(closure);
|
|
} else if (PyErr_Occurred()) {
|
|
return NULL;
|
|
} else {
|
|
Py_RETURN_NONE; // This can happen when `closure` is set to `None`
|
|
}
|
|
}
|
|
|
|
static PyObject *
|
|
function_set_closure(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *func = NULL, *closure = NULL;
|
|
if (!PyArg_ParseTuple(args, "OO", &func, &closure)) {
|
|
return NULL;
|
|
}
|
|
int result = PyFunction_SetClosure(func, closure);
|
|
if (result == -1) {
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
check_pyimport_addmodule(PyObject *self, PyObject *args)
|
|
{
|
|
const char *name;
|
|
if (!PyArg_ParseTuple(args, "s", &name)) {
|
|
return NULL;
|
|
}
|
|
|
|
// test PyImport_AddModuleRef()
|
|
PyObject *module = PyImport_AddModuleRef(name);
|
|
if (module == NULL) {
|
|
return NULL;
|
|
}
|
|
assert(PyModule_Check(module));
|
|
// module is a strong reference
|
|
|
|
// test PyImport_AddModule()
|
|
PyObject *module2 = PyImport_AddModule(name);
|
|
if (module2 == NULL) {
|
|
goto error;
|
|
}
|
|
assert(PyModule_Check(module2));
|
|
assert(module2 == module);
|
|
// module2 is a borrowed ref
|
|
|
|
// test PyImport_AddModuleObject()
|
|
PyObject *name_obj = PyUnicode_FromString(name);
|
|
if (name_obj == NULL) {
|
|
goto error;
|
|
}
|
|
PyObject *module3 = PyImport_AddModuleObject(name_obj);
|
|
Py_DECREF(name_obj);
|
|
if (module3 == NULL) {
|
|
goto error;
|
|
}
|
|
assert(PyModule_Check(module3));
|
|
assert(module3 == module);
|
|
// module3 is a borrowed ref
|
|
|
|
return module;
|
|
|
|
error:
|
|
Py_DECREF(module);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
test_weakref_capi(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
|
|
{
|
|
// Ignore PyWeakref_GetObject() deprecation, we test it on purpose
|
|
_Py_COMP_DIAG_PUSH
|
|
_Py_COMP_DIAG_IGNORE_DEPR_DECLS
|
|
|
|
// Create a new heap type, create an instance of this type, and delete the
|
|
// type. This object supports weak references.
|
|
PyObject *new_type = PyObject_CallFunction((PyObject*)&PyType_Type,
|
|
"s(){}", "TypeName");
|
|
if (new_type == NULL) {
|
|
return NULL;
|
|
}
|
|
PyObject *obj = PyObject_CallNoArgs(new_type);
|
|
Py_DECREF(new_type);
|
|
if (obj == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_ssize_t refcnt = Py_REFCNT(obj);
|
|
|
|
// test PyWeakref_NewRef(), reference is alive
|
|
PyObject *weakref = PyWeakref_NewRef(obj, NULL);
|
|
if (weakref == NULL) {
|
|
Py_DECREF(obj);
|
|
return NULL;
|
|
}
|
|
|
|
// test PyWeakref_Check(), valid weakref object
|
|
assert(PyWeakref_Check(weakref));
|
|
assert(PyWeakref_CheckRefExact(weakref));
|
|
assert(PyWeakref_CheckRefExact(weakref));
|
|
assert(Py_REFCNT(obj) == refcnt);
|
|
|
|
// test PyWeakref_GetRef(), reference is alive
|
|
PyObject *ref = UNINITIALIZED_PTR;
|
|
assert(PyWeakref_GetRef(weakref, &ref) == 1);
|
|
assert(ref == obj);
|
|
assert(Py_REFCNT(obj) == (refcnt + 1));
|
|
Py_DECREF(ref);
|
|
|
|
// test PyWeakref_GetObject(), reference is alive
|
|
ref = PyWeakref_GetObject(weakref); // borrowed ref
|
|
assert(ref == obj);
|
|
|
|
// test PyWeakref_GET_OBJECT(), reference is alive
|
|
ref = PyWeakref_GET_OBJECT(weakref); // borrowed ref
|
|
assert(ref == obj);
|
|
|
|
// delete the referenced object: clear the weakref
|
|
assert(Py_REFCNT(obj) == 1);
|
|
Py_DECREF(obj);
|
|
|
|
// test PyWeakref_GET_OBJECT(), reference is dead
|
|
assert(PyWeakref_GET_OBJECT(weakref) == Py_None);
|
|
|
|
// test PyWeakref_GetRef(), reference is dead
|
|
ref = UNINITIALIZED_PTR;
|
|
assert(PyWeakref_GetRef(weakref, &ref) == 0);
|
|
assert(ref == NULL);
|
|
|
|
// test PyWeakref_Check(), not a weakref object
|
|
PyObject *invalid_weakref = Py_None;
|
|
assert(!PyWeakref_Check(invalid_weakref));
|
|
assert(!PyWeakref_CheckRefExact(invalid_weakref));
|
|
assert(!PyWeakref_CheckRefExact(invalid_weakref));
|
|
|
|
// test PyWeakref_GetRef(), invalid type
|
|
assert(!PyErr_Occurred());
|
|
ref = UNINITIALIZED_PTR;
|
|
assert(PyWeakref_GetRef(invalid_weakref, &ref) == -1);
|
|
assert(PyErr_ExceptionMatches(PyExc_TypeError));
|
|
PyErr_Clear();
|
|
assert(ref == NULL);
|
|
|
|
// test PyWeakref_GetObject(), invalid type
|
|
assert(PyWeakref_GetObject(invalid_weakref) == NULL);
|
|
assert(PyErr_ExceptionMatches(PyExc_SystemError));
|
|
PyErr_Clear();
|
|
|
|
// test PyWeakref_GetRef(NULL)
|
|
ref = UNINITIALIZED_PTR;
|
|
assert(PyWeakref_GetRef(NULL, &ref) == -1);
|
|
assert(PyErr_ExceptionMatches(PyExc_SystemError));
|
|
assert(ref == NULL);
|
|
PyErr_Clear();
|
|
|
|
// test PyWeakref_GetObject(NULL)
|
|
assert(PyWeakref_GetObject(NULL) == NULL);
|
|
assert(PyErr_ExceptionMatches(PyExc_SystemError));
|
|
PyErr_Clear();
|
|
|
|
Py_DECREF(weakref);
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
_Py_COMP_DIAG_POP
|
|
}
|
|
|
|
struct simpletracer_data {
|
|
int create_count;
|
|
int destroy_count;
|
|
void* addresses[10];
|
|
};
|
|
|
|
static int _simpletracer(PyObject *obj, PyRefTracerEvent event, void* data) {
|
|
struct simpletracer_data* the_data = (struct simpletracer_data*)data;
|
|
assert(the_data->create_count + the_data->destroy_count < (int)Py_ARRAY_LENGTH(the_data->addresses));
|
|
the_data->addresses[the_data->create_count + the_data->destroy_count] = obj;
|
|
if (event == PyRefTracer_CREATE) {
|
|
the_data->create_count++;
|
|
} else {
|
|
the_data->destroy_count++;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *
|
|
test_reftracer(PyObject *ob, PyObject *Py_UNUSED(ignored))
|
|
{
|
|
// Save the current tracer and data to restore it later
|
|
void* current_data;
|
|
PyRefTracer current_tracer = PyRefTracer_GetTracer(¤t_data);
|
|
|
|
struct simpletracer_data tracer_data = {0};
|
|
void* the_data = &tracer_data;
|
|
// Install a simple tracer function
|
|
if (PyRefTracer_SetTracer(_simpletracer, the_data) != 0) {
|
|
goto failed;
|
|
}
|
|
|
|
// Check that the tracer was correctly installed
|
|
void* data;
|
|
if (PyRefTracer_GetTracer(&data) != _simpletracer || data != the_data) {
|
|
PyErr_SetString(PyExc_AssertionError, "The reftracer not correctly installed");
|
|
(void)PyRefTracer_SetTracer(NULL, NULL);
|
|
goto failed;
|
|
}
|
|
|
|
// Create a bunch of objects
|
|
PyObject* obj = PyList_New(0);
|
|
if (obj == NULL) {
|
|
goto failed;
|
|
}
|
|
PyObject* obj2 = PyDict_New();
|
|
if (obj2 == NULL) {
|
|
Py_DECREF(obj);
|
|
goto failed;
|
|
}
|
|
|
|
// Kill all objects
|
|
Py_DECREF(obj);
|
|
Py_DECREF(obj2);
|
|
|
|
// Remove the tracer
|
|
(void)PyRefTracer_SetTracer(NULL, NULL);
|
|
|
|
// Check that the tracer was removed
|
|
if (PyRefTracer_GetTracer(&data) != NULL || data != NULL) {
|
|
PyErr_SetString(PyExc_ValueError, "The reftracer was not correctly removed");
|
|
goto failed;
|
|
}
|
|
|
|
if (tracer_data.create_count != 2 ||
|
|
tracer_data.addresses[0] != obj ||
|
|
tracer_data.addresses[1] != obj2) {
|
|
PyErr_SetString(PyExc_ValueError, "The object creation was not correctly traced");
|
|
goto failed;
|
|
}
|
|
|
|
if (tracer_data.destroy_count != 2 ||
|
|
tracer_data.addresses[2] != obj ||
|
|
tracer_data.addresses[3] != obj2) {
|
|
PyErr_SetString(PyExc_ValueError, "The object destruction was not correctly traced");
|
|
goto failed;
|
|
}
|
|
PyRefTracer_SetTracer(current_tracer, current_data);
|
|
Py_RETURN_NONE;
|
|
failed:
|
|
PyRefTracer_SetTracer(current_tracer, current_data);
|
|
return NULL;
|
|
}
|
|
|
|
static PyObject *
|
|
function_set_warning(PyObject *Py_UNUSED(module), PyObject *Py_UNUSED(args))
|
|
{
|
|
if (PyErr_WarnEx(PyExc_RuntimeWarning, "Testing PyErr_WarnEx", 2)) {
|
|
return NULL;
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject *
|
|
test_critical_sections(PyObject *module, PyObject *Py_UNUSED(args))
|
|
{
|
|
Py_BEGIN_CRITICAL_SECTION(module);
|
|
Py_END_CRITICAL_SECTION();
|
|
|
|
Py_BEGIN_CRITICAL_SECTION2(module, module);
|
|
Py_END_CRITICAL_SECTION2();
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyMethodDef TestMethods[] = {
|
|
{"set_errno", set_errno, METH_VARARGS},
|
|
{"test_config", test_config, METH_NOARGS},
|
|
{"test_sizeof_c_types", test_sizeof_c_types, METH_NOARGS},
|
|
{"test_list_api", test_list_api, METH_NOARGS},
|
|
{"test_dict_iteration", test_dict_iteration, METH_NOARGS},
|
|
{"test_lazy_hash_inheritance", test_lazy_hash_inheritance,METH_NOARGS},
|
|
{"test_xincref_doesnt_leak",test_xincref_doesnt_leak, METH_NOARGS},
|
|
{"test_incref_doesnt_leak", test_incref_doesnt_leak, METH_NOARGS},
|
|
{"test_xdecref_doesnt_leak",test_xdecref_doesnt_leak, METH_NOARGS},
|
|
{"test_decref_doesnt_leak", test_decref_doesnt_leak, METH_NOARGS},
|
|
{"test_structseq_newtype_doesnt_leak",
|
|
test_structseq_newtype_doesnt_leak, METH_NOARGS},
|
|
{"test_structseq_newtype_null_descr_doc",
|
|
test_structseq_newtype_null_descr_doc, METH_NOARGS},
|
|
{"test_incref_decref_API", test_incref_decref_API, METH_NOARGS},
|
|
{"pyobject_repr_from_null", pyobject_repr_from_null, METH_NOARGS},
|
|
{"pyobject_str_from_null", pyobject_str_from_null, METH_NOARGS},
|
|
{"pyobject_bytes_from_null", pyobject_bytes_from_null, METH_NOARGS},
|
|
{"test_string_to_double", test_string_to_double, METH_NOARGS},
|
|
{"test_capsule", (PyCFunction)test_capsule, METH_NOARGS},
|
|
{"test_from_contiguous", (PyCFunction)test_from_contiguous, METH_NOARGS},
|
|
#if (defined(__linux__) || defined(__FreeBSD__)) && defined(__GNUC__)
|
|
{"test_pep3118_obsolete_write_locks", (PyCFunction)test_pep3118_obsolete_write_locks, METH_NOARGS},
|
|
#endif
|
|
{"getbuffer_with_null_view", getbuffer_with_null_view, METH_O},
|
|
{"PyBuffer_SizeFromFormat", test_PyBuffer_SizeFromFormat, METH_VARARGS},
|
|
{"py_buildvalue", py_buildvalue, METH_VARARGS},
|
|
{"py_buildvalue_ints", py_buildvalue_ints, METH_VARARGS},
|
|
{"test_buildvalue_N", test_buildvalue_N, METH_NOARGS},
|
|
{"test_get_statictype_slots", test_get_statictype_slots, METH_NOARGS},
|
|
{"get_heaptype_for_name", get_heaptype_for_name, METH_NOARGS},
|
|
{"get_type_name", get_type_name, METH_O},
|
|
{"get_type_qualname", get_type_qualname, METH_O},
|
|
{"get_type_fullyqualname", get_type_fullyqualname, METH_O},
|
|
{"get_type_module_name", get_type_module_name, METH_O},
|
|
{"test_get_type_dict", test_get_type_dict, METH_NOARGS},
|
|
{"test_reftracer", test_reftracer, METH_NOARGS},
|
|
{"_test_thread_state", test_thread_state, METH_VARARGS},
|
|
{"gilstate_ensure_release", gilstate_ensure_release, METH_NOARGS},
|
|
#ifndef MS_WINDOWS
|
|
{"_spawn_pthread_waiter", spawn_pthread_waiter, METH_NOARGS},
|
|
{"_end_spawned_pthread", end_spawned_pthread, METH_NOARGS},
|
|
#endif
|
|
{"_pending_threadfunc", _PyCFunction_CAST(pending_threadfunc),
|
|
METH_VARARGS|METH_KEYWORDS},
|
|
#ifdef HAVE_GETTIMEOFDAY
|
|
{"profile_int", profile_int, METH_NOARGS},
|
|
#endif
|
|
{"argparsing", argparsing, METH_VARARGS},
|
|
{"code_newempty", code_newempty, METH_VARARGS},
|
|
{"eval_code_ex", eval_eval_code_ex, METH_VARARGS},
|
|
{"make_memoryview_from_NULL_pointer", make_memoryview_from_NULL_pointer,
|
|
METH_NOARGS},
|
|
{"buffer_fill_info", buffer_fill_info, METH_VARARGS},
|
|
{"crash_no_current_thread", crash_no_current_thread, METH_NOARGS},
|
|
{"test_current_tstate_matches", test_current_tstate_matches, METH_NOARGS},
|
|
{"run_in_subinterp", run_in_subinterp, METH_VARARGS},
|
|
{"create_cfunction", create_cfunction, METH_NOARGS},
|
|
{"call_in_temporary_c_thread", call_in_temporary_c_thread, METH_VARARGS,
|
|
PyDoc_STR("set_error_class(error_class) -> None")},
|
|
{"join_temporary_c_thread", join_temporary_c_thread, METH_NOARGS},
|
|
{"pymarshal_write_long_to_file",
|
|
pymarshal_write_long_to_file, METH_VARARGS},
|
|
{"pymarshal_write_object_to_file",
|
|
pymarshal_write_object_to_file, METH_VARARGS},
|
|
{"pymarshal_read_short_from_file",
|
|
pymarshal_read_short_from_file, METH_VARARGS},
|
|
{"pymarshal_read_long_from_file",
|
|
pymarshal_read_long_from_file, METH_VARARGS},
|
|
{"pymarshal_read_last_object_from_file",
|
|
pymarshal_read_last_object_from_file, METH_VARARGS},
|
|
{"pymarshal_read_object_from_file",
|
|
pymarshal_read_object_from_file, METH_VARARGS},
|
|
{"return_null_without_error", return_null_without_error, METH_NOARGS},
|
|
{"return_result_with_error", return_result_with_error, METH_NOARGS},
|
|
{"getitem_with_error", getitem_with_error, METH_VARARGS},
|
|
{"Py_CompileString", pycompilestring, METH_O},
|
|
{"dict_get_version", dict_get_version, METH_VARARGS},
|
|
{"raise_SIGINT_then_send_None", raise_SIGINT_then_send_None, METH_VARARGS},
|
|
{"stack_pointer", stack_pointer, METH_NOARGS},
|
|
#ifdef W_STOPCODE
|
|
{"W_STOPCODE", py_w_stopcode, METH_VARARGS},
|
|
#endif
|
|
{"test_pythread_tss_key_state", test_pythread_tss_key_state, METH_VARARGS},
|
|
{"bad_get", bad_get, METH_VARARGS},
|
|
#ifdef Py_REF_DEBUG
|
|
{"negative_refcount", negative_refcount, METH_NOARGS},
|
|
{"decref_freed_object", decref_freed_object, METH_NOARGS},
|
|
#endif
|
|
{"meth_varargs", meth_varargs, METH_VARARGS},
|
|
{"meth_varargs_keywords", _PyCFunction_CAST(meth_varargs_keywords), METH_VARARGS|METH_KEYWORDS},
|
|
{"meth_o", meth_o, METH_O},
|
|
{"meth_noargs", meth_noargs, METH_NOARGS},
|
|
{"meth_fastcall", _PyCFunction_CAST(meth_fastcall), METH_FASTCALL},
|
|
{"meth_fastcall_keywords", _PyCFunction_CAST(meth_fastcall_keywords), METH_FASTCALL|METH_KEYWORDS},
|
|
{"pycfunction_call", test_pycfunction_call, METH_VARARGS},
|
|
{"pynumber_tobase", pynumber_tobase, METH_VARARGS},
|
|
{"test_set_type_size", test_set_type_size, METH_NOARGS},
|
|
{"test_py_clear", test_py_clear, METH_NOARGS},
|
|
{"test_py_setref", test_py_setref, METH_NOARGS},
|
|
{"test_refcount_macros", test_refcount_macros, METH_NOARGS},
|
|
{"test_refcount_funcs", test_refcount_funcs, METH_NOARGS},
|
|
{"test_py_is_macros", test_py_is_macros, METH_NOARGS},
|
|
{"test_py_is_funcs", test_py_is_funcs, METH_NOARGS},
|
|
{"type_get_version", type_get_version, METH_O, PyDoc_STR("type->tp_version_tag")},
|
|
{"type_modified", type_modified, METH_O, PyDoc_STR("PyType_Modified")},
|
|
{"type_assign_version", type_assign_version, METH_O, PyDoc_STR("PyUnstable_Type_AssignVersionTag")},
|
|
{"type_get_tp_bases", type_get_tp_bases, METH_O},
|
|
{"type_get_tp_mro", type_get_tp_mro, METH_O},
|
|
{"get_basic_static_type", get_basic_static_type, METH_VARARGS, NULL},
|
|
{"test_tstate_capi", test_tstate_capi, METH_NOARGS, NULL},
|
|
{"frame_getlocals", frame_getlocals, METH_O, NULL},
|
|
{"frame_getglobals", frame_getglobals, METH_O, NULL},
|
|
{"frame_getgenerator", frame_getgenerator, METH_O, NULL},
|
|
{"frame_getbuiltins", frame_getbuiltins, METH_O, NULL},
|
|
{"frame_getlasti", frame_getlasti, METH_O, NULL},
|
|
{"frame_new", frame_new, METH_VARARGS, NULL},
|
|
{"frame_getvar", test_frame_getvar, METH_VARARGS, NULL},
|
|
{"frame_getvarstring", test_frame_getvarstring, METH_VARARGS, NULL},
|
|
{"gen_get_code", gen_get_code, METH_O, NULL},
|
|
{"get_feature_macros", get_feature_macros, METH_NOARGS, NULL},
|
|
{"test_code_api", test_code_api, METH_NOARGS, NULL},
|
|
{"settrace_to_error", settrace_to_error, METH_O, NULL},
|
|
{"settrace_to_record", settrace_to_record, METH_O, NULL},
|
|
{"test_macros", test_macros, METH_NOARGS, NULL},
|
|
{"clear_managed_dict", clear_managed_dict, METH_O, NULL},
|
|
{"function_get_code", function_get_code, METH_O, NULL},
|
|
{"function_get_globals", function_get_globals, METH_O, NULL},
|
|
{"function_get_module", function_get_module, METH_O, NULL},
|
|
{"function_get_defaults", function_get_defaults, METH_O, NULL},
|
|
{"function_set_defaults", function_set_defaults, METH_VARARGS, NULL},
|
|
{"function_get_kw_defaults", function_get_kw_defaults, METH_O, NULL},
|
|
{"function_set_kw_defaults", function_set_kw_defaults, METH_VARARGS, NULL},
|
|
{"function_get_closure", function_get_closure, METH_O, NULL},
|
|
{"function_set_closure", function_set_closure, METH_VARARGS, NULL},
|
|
{"check_pyimport_addmodule", check_pyimport_addmodule, METH_VARARGS},
|
|
{"test_weakref_capi", test_weakref_capi, METH_NOARGS},
|
|
{"function_set_warning", function_set_warning, METH_NOARGS},
|
|
{"test_critical_sections", test_critical_sections, METH_NOARGS},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
} matmulObject;
|
|
|
|
static PyObject *
|
|
matmulType_matmul(PyObject *self, PyObject *other)
|
|
{
|
|
return Py_BuildValue("(sOO)", "matmul", self, other);
|
|
}
|
|
|
|
static PyObject *
|
|
matmulType_imatmul(PyObject *self, PyObject *other)
|
|
{
|
|
return Py_BuildValue("(sOO)", "imatmul", self, other);
|
|
}
|
|
|
|
static void
|
|
matmulType_dealloc(PyObject *self)
|
|
{
|
|
Py_TYPE(self)->tp_free(self);
|
|
}
|
|
|
|
static PyNumberMethods matmulType_as_number = {
|
|
0, /* nb_add */
|
|
0, /* nb_subtract */
|
|
0, /* nb_multiply */
|
|
0, /* nb_remainde r*/
|
|
0, /* nb_divmod */
|
|
0, /* nb_power */
|
|
0, /* nb_negative */
|
|
0, /* tp_positive */
|
|
0, /* tp_absolute */
|
|
0, /* tp_bool */
|
|
0, /* nb_invert */
|
|
0, /* nb_lshift */
|
|
0, /* nb_rshift */
|
|
0, /* nb_and */
|
|
0, /* nb_xor */
|
|
0, /* nb_or */
|
|
0, /* nb_int */
|
|
0, /* nb_reserved */
|
|
0, /* nb_float */
|
|
0, /* nb_inplace_add */
|
|
0, /* nb_inplace_subtract */
|
|
0, /* nb_inplace_multiply */
|
|
0, /* nb_inplace_remainder */
|
|
0, /* nb_inplace_power */
|
|
0, /* nb_inplace_lshift */
|
|
0, /* nb_inplace_rshift */
|
|
0, /* nb_inplace_and */
|
|
0, /* nb_inplace_xor */
|
|
0, /* nb_inplace_or */
|
|
0, /* nb_floor_divide */
|
|
0, /* nb_true_divide */
|
|
0, /* nb_inplace_floor_divide */
|
|
0, /* nb_inplace_true_divide */
|
|
0, /* nb_index */
|
|
matmulType_matmul, /* nb_matrix_multiply */
|
|
matmulType_imatmul /* nb_matrix_inplace_multiply */
|
|
};
|
|
|
|
static PyTypeObject matmulType = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"matmulType",
|
|
sizeof(matmulObject), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
matmulType_dealloc, /* destructor tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_as_async */
|
|
0, /* tp_repr */
|
|
&matmulType_as_number, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
PyObject_GenericSetAttr, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
0, /* tp_flags */
|
|
"C level type with matrix operations defined",
|
|
0, /* traverseproc tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
0, /* tp_methods */
|
|
0, /* tp_members */
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
PyType_GenericNew, /* tp_new */
|
|
PyObject_Free, /* tp_free */
|
|
};
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
} ipowObject;
|
|
|
|
static PyObject *
|
|
ipowType_ipow(PyObject *self, PyObject *other, PyObject *mod)
|
|
{
|
|
return PyTuple_Pack(2, other, mod);
|
|
}
|
|
|
|
static PyNumberMethods ipowType_as_number = {
|
|
.nb_inplace_power = ipowType_ipow
|
|
};
|
|
|
|
static PyTypeObject ipowType = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
.tp_name = "ipowType",
|
|
.tp_basicsize = sizeof(ipowObject),
|
|
.tp_as_number = &ipowType_as_number,
|
|
.tp_new = PyType_GenericNew
|
|
};
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *ao_iterator;
|
|
} awaitObject;
|
|
|
|
|
|
static PyObject *
|
|
awaitObject_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject *v;
|
|
awaitObject *ao;
|
|
|
|
if (!PyArg_UnpackTuple(args, "awaitObject", 1, 1, &v))
|
|
return NULL;
|
|
|
|
ao = (awaitObject *)type->tp_alloc(type, 0);
|
|
if (ao == NULL) {
|
|
return NULL;
|
|
}
|
|
|
|
ao->ao_iterator = Py_NewRef(v);
|
|
|
|
return (PyObject *)ao;
|
|
}
|
|
|
|
|
|
static void
|
|
awaitObject_dealloc(awaitObject *ao)
|
|
{
|
|
Py_CLEAR(ao->ao_iterator);
|
|
Py_TYPE(ao)->tp_free(ao);
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
awaitObject_await(awaitObject *ao)
|
|
{
|
|
return Py_NewRef(ao->ao_iterator);
|
|
}
|
|
|
|
static PyAsyncMethods awaitType_as_async = {
|
|
(unaryfunc)awaitObject_await, /* am_await */
|
|
0, /* am_aiter */
|
|
0, /* am_anext */
|
|
0, /* am_send */
|
|
};
|
|
|
|
|
|
static PyTypeObject awaitType = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"awaitType",
|
|
sizeof(awaitObject), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
(destructor)awaitObject_dealloc, /* destructor tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
&awaitType_as_async, /* tp_as_async */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
PyObject_GenericGetAttr, /* tp_getattro */
|
|
PyObject_GenericSetAttr, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
0, /* tp_flags */
|
|
"C level type with tp_as_async",
|
|
0, /* traverseproc tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
0, /* tp_methods */
|
|
0, /* tp_members */
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
awaitObject_new, /* tp_new */
|
|
PyObject_Free, /* tp_free */
|
|
};
|
|
|
|
|
|
/* Test bpo-35983: create a subclass of "list" which checks that instances
|
|
* are not deallocated twice */
|
|
|
|
typedef struct {
|
|
PyListObject list;
|
|
int deallocated;
|
|
} MyListObject;
|
|
|
|
static PyObject *
|
|
MyList_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
|
|
{
|
|
PyObject* op = PyList_Type.tp_new(type, args, kwds);
|
|
((MyListObject*)op)->deallocated = 0;
|
|
return op;
|
|
}
|
|
|
|
void
|
|
MyList_dealloc(MyListObject* op)
|
|
{
|
|
if (op->deallocated) {
|
|
/* We cannot raise exceptions here but we still want the testsuite
|
|
* to fail when we hit this */
|
|
Py_FatalError("MyList instance deallocated twice");
|
|
}
|
|
op->deallocated = 1;
|
|
PyList_Type.tp_dealloc((PyObject *)op);
|
|
}
|
|
|
|
static PyTypeObject MyList_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"MyList",
|
|
sizeof(MyListObject),
|
|
0,
|
|
(destructor)MyList_dealloc, /* tp_dealloc */
|
|
0, /* tp_vectorcall_offset */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_as_async */
|
|
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 */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
0, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
0, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
0, /* &PyList_Type */ /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
0, /* tp_init */
|
|
0, /* tp_alloc */
|
|
MyList_new, /* tp_new */
|
|
};
|
|
|
|
/* Test PEP 560 */
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *item;
|
|
} PyGenericAliasObject;
|
|
|
|
static void
|
|
generic_alias_dealloc(PyGenericAliasObject *self)
|
|
{
|
|
Py_CLEAR(self->item);
|
|
Py_TYPE(self)->tp_free((PyObject *)self);
|
|
}
|
|
|
|
static PyObject *
|
|
generic_alias_mro_entries(PyGenericAliasObject *self, PyObject *bases)
|
|
{
|
|
return PyTuple_Pack(1, self->item);
|
|
}
|
|
|
|
static PyMethodDef generic_alias_methods[] = {
|
|
{"__mro_entries__", _PyCFunction_CAST(generic_alias_mro_entries), METH_O, NULL},
|
|
{NULL} /* sentinel */
|
|
};
|
|
|
|
static PyTypeObject GenericAlias_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"GenericAlias",
|
|
sizeof(PyGenericAliasObject),
|
|
0,
|
|
.tp_dealloc = (destructor)generic_alias_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
.tp_methods = generic_alias_methods,
|
|
};
|
|
|
|
static PyObject *
|
|
generic_alias_new(PyObject *item)
|
|
{
|
|
PyGenericAliasObject *o = PyObject_New(PyGenericAliasObject, &GenericAlias_Type);
|
|
if (o == NULL) {
|
|
return NULL;
|
|
}
|
|
o->item = Py_NewRef(item);
|
|
return (PyObject*) o;
|
|
}
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
} PyGenericObject;
|
|
|
|
static PyObject *
|
|
generic_class_getitem(PyObject *type, PyObject *item)
|
|
{
|
|
return generic_alias_new(item);
|
|
}
|
|
|
|
static PyMethodDef generic_methods[] = {
|
|
{"__class_getitem__", generic_class_getitem, METH_O|METH_CLASS, NULL},
|
|
{NULL} /* sentinel */
|
|
};
|
|
|
|
static PyTypeObject Generic_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"Generic",
|
|
sizeof(PyGenericObject),
|
|
0,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
.tp_methods = generic_methods,
|
|
};
|
|
|
|
static PyMethodDef meth_instance_methods[] = {
|
|
{"meth_varargs", meth_varargs, METH_VARARGS},
|
|
{"meth_varargs_keywords", _PyCFunction_CAST(meth_varargs_keywords), METH_VARARGS|METH_KEYWORDS},
|
|
{"meth_o", meth_o, METH_O},
|
|
{"meth_noargs", meth_noargs, METH_NOARGS},
|
|
{"meth_fastcall", _PyCFunction_CAST(meth_fastcall), METH_FASTCALL},
|
|
{"meth_fastcall_keywords", _PyCFunction_CAST(meth_fastcall_keywords), METH_FASTCALL|METH_KEYWORDS},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
static PyTypeObject MethInstance_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"MethInstance",
|
|
sizeof(PyObject),
|
|
.tp_new = PyType_GenericNew,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_methods = meth_instance_methods,
|
|
.tp_doc = (char*)PyDoc_STR(
|
|
"Class with normal (instance) methods to test calling conventions"),
|
|
};
|
|
|
|
static PyMethodDef meth_class_methods[] = {
|
|
{"meth_varargs", meth_varargs, METH_VARARGS|METH_CLASS},
|
|
{"meth_varargs_keywords", _PyCFunction_CAST(meth_varargs_keywords), METH_VARARGS|METH_KEYWORDS|METH_CLASS},
|
|
{"meth_o", meth_o, METH_O|METH_CLASS},
|
|
{"meth_noargs", meth_noargs, METH_NOARGS|METH_CLASS},
|
|
{"meth_fastcall", _PyCFunction_CAST(meth_fastcall), METH_FASTCALL|METH_CLASS},
|
|
{"meth_fastcall_keywords", _PyCFunction_CAST(meth_fastcall_keywords), METH_FASTCALL|METH_KEYWORDS|METH_CLASS},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
static PyTypeObject MethClass_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"MethClass",
|
|
sizeof(PyObject),
|
|
.tp_new = PyType_GenericNew,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_methods = meth_class_methods,
|
|
.tp_doc = PyDoc_STR(
|
|
"Class with class methods to test calling conventions"),
|
|
};
|
|
|
|
static PyMethodDef meth_static_methods[] = {
|
|
{"meth_varargs", meth_varargs, METH_VARARGS|METH_STATIC},
|
|
{"meth_varargs_keywords", _PyCFunction_CAST(meth_varargs_keywords), METH_VARARGS|METH_KEYWORDS|METH_STATIC},
|
|
{"meth_o", meth_o, METH_O|METH_STATIC},
|
|
{"meth_noargs", meth_noargs, METH_NOARGS|METH_STATIC},
|
|
{"meth_fastcall", _PyCFunction_CAST(meth_fastcall), METH_FASTCALL|METH_STATIC},
|
|
{"meth_fastcall_keywords", _PyCFunction_CAST(meth_fastcall_keywords), METH_FASTCALL|METH_KEYWORDS|METH_STATIC},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
|
|
static PyTypeObject MethStatic_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"MethStatic",
|
|
sizeof(PyObject),
|
|
.tp_new = PyType_GenericNew,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT,
|
|
.tp_methods = meth_static_methods,
|
|
.tp_doc = PyDoc_STR(
|
|
"Class with static methods to test calling conventions"),
|
|
};
|
|
|
|
/* ContainerNoGC -- a simple container without GC methods */
|
|
|
|
typedef struct {
|
|
PyObject_HEAD
|
|
PyObject *value;
|
|
} ContainerNoGCobject;
|
|
|
|
static PyObject *
|
|
ContainerNoGC_new(PyTypeObject *type, PyObject *args, PyObject *kwargs)
|
|
{
|
|
PyObject *value;
|
|
char *names[] = {"value", NULL};
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O", names, &value)) {
|
|
return NULL;
|
|
}
|
|
PyObject *self = type->tp_alloc(type, 0);
|
|
if (self == NULL) {
|
|
return NULL;
|
|
}
|
|
Py_INCREF(value);
|
|
((ContainerNoGCobject *)self)->value = value;
|
|
return self;
|
|
}
|
|
|
|
static void
|
|
ContainerNoGC_dealloc(ContainerNoGCobject *self)
|
|
{
|
|
Py_DECREF(self->value);
|
|
Py_TYPE(self)->tp_free((PyObject *)self);
|
|
}
|
|
|
|
static PyMemberDef ContainerNoGC_members[] = {
|
|
{"value", _Py_T_OBJECT, offsetof(ContainerNoGCobject, value), Py_READONLY,
|
|
PyDoc_STR("a container value for test purposes")},
|
|
{0}
|
|
};
|
|
|
|
static PyTypeObject ContainerNoGC_type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"_testcapi.ContainerNoGC",
|
|
sizeof(ContainerNoGCobject),
|
|
.tp_dealloc = (destructor)ContainerNoGC_dealloc,
|
|
.tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
|
|
.tp_members = ContainerNoGC_members,
|
|
.tp_new = ContainerNoGC_new,
|
|
};
|
|
|
|
|
|
static struct PyModuleDef _testcapimodule = {
|
|
PyModuleDef_HEAD_INIT,
|
|
.m_name = "_testcapi",
|
|
.m_size = sizeof(testcapistate_t),
|
|
.m_methods = TestMethods,
|
|
};
|
|
|
|
/* Per PEP 489, this module will not be converted to multi-phase initialization
|
|
*/
|
|
|
|
PyMODINIT_FUNC
|
|
PyInit__testcapi(void)
|
|
{
|
|
PyObject *m;
|
|
|
|
m = PyModule_Create(&_testcapimodule);
|
|
if (m == NULL)
|
|
return NULL;
|
|
#ifdef Py_GIL_DISABLED
|
|
PyUnstable_Module_SetGIL(m, Py_MOD_GIL_NOT_USED);
|
|
#endif
|
|
|
|
Py_SET_TYPE(&_HashInheritanceTester_Type, &PyType_Type);
|
|
if (PyType_Ready(&_HashInheritanceTester_Type) < 0) {
|
|
return NULL;
|
|
}
|
|
if (PyType_Ready(&matmulType) < 0)
|
|
return NULL;
|
|
Py_INCREF(&matmulType);
|
|
PyModule_AddObject(m, "matmulType", (PyObject *)&matmulType);
|
|
if (PyType_Ready(&ipowType) < 0) {
|
|
return NULL;
|
|
}
|
|
Py_INCREF(&ipowType);
|
|
PyModule_AddObject(m, "ipowType", (PyObject *)&ipowType);
|
|
|
|
if (PyType_Ready(&awaitType) < 0)
|
|
return NULL;
|
|
Py_INCREF(&awaitType);
|
|
PyModule_AddObject(m, "awaitType", (PyObject *)&awaitType);
|
|
|
|
MyList_Type.tp_base = &PyList_Type;
|
|
if (PyType_Ready(&MyList_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&MyList_Type);
|
|
PyModule_AddObject(m, "MyList", (PyObject *)&MyList_Type);
|
|
|
|
if (PyType_Ready(&GenericAlias_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&GenericAlias_Type);
|
|
PyModule_AddObject(m, "GenericAlias", (PyObject *)&GenericAlias_Type);
|
|
|
|
if (PyType_Ready(&Generic_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&Generic_Type);
|
|
PyModule_AddObject(m, "Generic", (PyObject *)&Generic_Type);
|
|
|
|
if (PyType_Ready(&MethInstance_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&MethInstance_Type);
|
|
PyModule_AddObject(m, "MethInstance", (PyObject *)&MethInstance_Type);
|
|
|
|
if (PyType_Ready(&MethClass_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&MethClass_Type);
|
|
PyModule_AddObject(m, "MethClass", (PyObject *)&MethClass_Type);
|
|
|
|
if (PyType_Ready(&MethStatic_Type) < 0)
|
|
return NULL;
|
|
Py_INCREF(&MethStatic_Type);
|
|
PyModule_AddObject(m, "MethStatic", (PyObject *)&MethStatic_Type);
|
|
|
|
PyModule_AddObject(m, "CHAR_MAX", PyLong_FromLong(CHAR_MAX));
|
|
PyModule_AddObject(m, "CHAR_MIN", PyLong_FromLong(CHAR_MIN));
|
|
PyModule_AddObject(m, "UCHAR_MAX", PyLong_FromLong(UCHAR_MAX));
|
|
PyModule_AddObject(m, "SHRT_MAX", PyLong_FromLong(SHRT_MAX));
|
|
PyModule_AddObject(m, "SHRT_MIN", PyLong_FromLong(SHRT_MIN));
|
|
PyModule_AddObject(m, "USHRT_MAX", PyLong_FromLong(USHRT_MAX));
|
|
PyModule_AddObject(m, "INT_MAX", PyLong_FromLong(INT_MAX));
|
|
PyModule_AddObject(m, "INT_MIN", PyLong_FromLong(INT_MIN));
|
|
PyModule_AddObject(m, "UINT_MAX", PyLong_FromUnsignedLong(UINT_MAX));
|
|
PyModule_AddObject(m, "LONG_MAX", PyLong_FromLong(LONG_MAX));
|
|
PyModule_AddObject(m, "LONG_MIN", PyLong_FromLong(LONG_MIN));
|
|
PyModule_AddObject(m, "ULONG_MAX", PyLong_FromUnsignedLong(ULONG_MAX));
|
|
PyModule_AddObject(m, "FLT_MAX", PyFloat_FromDouble(FLT_MAX));
|
|
PyModule_AddObject(m, "FLT_MIN", PyFloat_FromDouble(FLT_MIN));
|
|
PyModule_AddObject(m, "DBL_MAX", PyFloat_FromDouble(DBL_MAX));
|
|
PyModule_AddObject(m, "DBL_MIN", PyFloat_FromDouble(DBL_MIN));
|
|
PyModule_AddObject(m, "LLONG_MAX", PyLong_FromLongLong(LLONG_MAX));
|
|
PyModule_AddObject(m, "LLONG_MIN", PyLong_FromLongLong(LLONG_MIN));
|
|
PyModule_AddObject(m, "ULLONG_MAX", PyLong_FromUnsignedLongLong(ULLONG_MAX));
|
|
PyModule_AddObject(m, "PY_SSIZE_T_MAX", PyLong_FromSsize_t(PY_SSIZE_T_MAX));
|
|
PyModule_AddObject(m, "PY_SSIZE_T_MIN", PyLong_FromSsize_t(PY_SSIZE_T_MIN));
|
|
PyModule_AddObject(m, "SIZE_MAX", PyLong_FromSize_t(SIZE_MAX));
|
|
PyModule_AddObject(m, "SIZEOF_WCHAR_T", PyLong_FromSsize_t(sizeof(wchar_t)));
|
|
PyModule_AddObject(m, "SIZEOF_VOID_P", PyLong_FromSsize_t(sizeof(void*)));
|
|
PyModule_AddObject(m, "SIZEOF_TIME_T", PyLong_FromSsize_t(sizeof(time_t)));
|
|
PyModule_AddObject(m, "SIZEOF_PID_T", PyLong_FromSsize_t(sizeof(pid_t)));
|
|
PyModule_AddObject(m, "Py_Version", PyLong_FromUnsignedLong(Py_Version));
|
|
Py_INCREF(&PyInstanceMethod_Type);
|
|
PyModule_AddObject(m, "instancemethod", (PyObject *)&PyInstanceMethod_Type);
|
|
|
|
PyModule_AddIntConstant(m, "the_number_three", 3);
|
|
PyModule_AddIntMacro(m, Py_C_RECURSION_LIMIT);
|
|
PyModule_AddObject(m, "INT32_MIN", PyLong_FromInt32(INT32_MIN));
|
|
PyModule_AddObject(m, "INT32_MAX", PyLong_FromInt32(INT32_MAX));
|
|
PyModule_AddObject(m, "UINT32_MAX", PyLong_FromUInt32(UINT32_MAX));
|
|
PyModule_AddObject(m, "INT64_MIN", PyLong_FromInt64(INT64_MIN));
|
|
PyModule_AddObject(m, "INT64_MAX", PyLong_FromInt64(INT64_MAX));
|
|
PyModule_AddObject(m, "UINT64_MAX", PyLong_FromUInt64(UINT64_MAX));
|
|
|
|
if (PyModule_AddIntMacro(m, Py_single_input)) {
|
|
return NULL;
|
|
}
|
|
if (PyModule_AddIntMacro(m, Py_file_input)) {
|
|
return NULL;
|
|
}
|
|
if (PyModule_AddIntMacro(m, Py_eval_input)) {
|
|
return NULL;
|
|
}
|
|
|
|
testcapistate_t *state = get_testcapi_state(m);
|
|
state->error = PyErr_NewException("_testcapi.error", NULL, NULL);
|
|
PyModule_AddObject(m, "error", state->error);
|
|
|
|
if (PyType_Ready(&ContainerNoGC_type) < 0) {
|
|
return NULL;
|
|
}
|
|
Py_INCREF(&ContainerNoGC_type);
|
|
if (PyModule_AddObject(m, "ContainerNoGC",
|
|
(PyObject *) &ContainerNoGC_type) < 0)
|
|
return NULL;
|
|
|
|
/* Include tests from the _testcapi/ directory */
|
|
if (_PyTestCapi_Init_Vectorcall(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Heaptype(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Abstract(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Bytes(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Unicode(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_GetArgs(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_DateTime(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Docstring(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Mem(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Watchers(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Long(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Float(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Complex(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Numbers(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Dict(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Set(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_List(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Tuple(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Structmember(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Exceptions(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Code(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Buffer(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_File(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Codec(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Immortal(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_GC(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_PyAtomic(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Run(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Hash(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Time(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Monitoring(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Object(m) < 0) {
|
|
return NULL;
|
|
}
|
|
if (_PyTestCapi_Init_Config(m) < 0) {
|
|
return NULL;
|
|
}
|
|
|
|
PyState_AddModule(m, &_testcapimodule);
|
|
return m;
|
|
}
|