cpython/Modules/_testlimitedcapi/vectorcall_limited.c

202 lines
4.7 KiB
C
Raw Normal View History

/* Test Vectorcall in the limited API */
// Need limited C API version 3.12 for PyObject_Vectorcall()
#include "pyconfig.h" // Py_GIL_DISABLED
#if !defined(Py_GIL_DISABLED) && !defined(Py_LIMITED_API)
# define Py_LIMITED_API 0x030c0000
#endif
#include "parts.h"
#include "clinic/vectorcall_limited.c.h"
/*[clinic input]
module _testlimitedcapi
[clinic start generated code]*/
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=2700057f9c1135ba]*/
static PyObject *
LimitedVectorCallClass_tpcall(PyObject *self, PyObject *args, PyObject *kwargs) {
return PyUnicode_FromString("tp_call called");
}
static PyObject *
LimitedVectorCallClass_vectorcall(PyObject *callable,
PyObject *const *args,
size_t nargsf,
PyObject *kwnames) {
return PyUnicode_FromString("vectorcall called");
}
static PyObject *
LimitedVectorCallClass_new(PyTypeObject *tp, PyTypeObject *a, PyTypeObject *kw)
{
PyObject *self = ((allocfunc)PyType_GetSlot(tp, Py_tp_alloc))(tp, 0);
if (!self) {
return NULL;
}
*(vectorcallfunc*)((char*)self + sizeof(PyObject)) = (
LimitedVectorCallClass_vectorcall);
return self;
}
/*[clinic input]
_testlimitedcapi.call_vectorcall
callable: object
/
[clinic start generated code]*/
static PyObject *
_testlimitedcapi_call_vectorcall(PyObject *module, PyObject *callable)
/*[clinic end generated code: output=9cbb7832263a8eef input=0743636c12dccb28]*/
{
PyObject *args[3] = { NULL, NULL, NULL };
PyObject *kwname = NULL, *kwnames = NULL, *result = NULL;
args[1] = PyUnicode_FromString("foo");
if (!args[1]) {
goto leave;
}
args[2] = PyUnicode_FromString("bar");
if (!args[2]) {
goto leave;
}
kwname = PyUnicode_InternFromString("baz");
if (!kwname) {
goto leave;
}
kwnames = PyTuple_New(1);
if (!kwnames) {
goto leave;
}
if (PyTuple_SetItem(kwnames, 0, kwname)) {
goto leave;
}
result = PyObject_Vectorcall(
callable,
args + 1,
1 | PY_VECTORCALL_ARGUMENTS_OFFSET,
kwnames
);
leave:
Py_XDECREF(args[1]);
Py_XDECREF(args[2]);
Py_XDECREF(kwnames);
return result;
}
/*[clinic input]
_testlimitedcapi.call_vectorcall_method
callable: object
/
[clinic start generated code]*/
static PyObject *
_testlimitedcapi_call_vectorcall_method(PyObject *module, PyObject *callable)
/*[clinic end generated code: output=4558323a46cc09eb input=a736f7dbf15f1be5]*/
{
PyObject *args[3] = { NULL, NULL, NULL };
PyObject *name = NULL, *kwname = NULL,
*kwnames = NULL, *result = NULL;
name = PyUnicode_FromString("f");
if (!name) {
goto leave;
}
args[0] = callable;
args[1] = PyUnicode_FromString("foo");
if (!args[1]) {
goto leave;
}
args[2] = PyUnicode_FromString("bar");
if (!args[2]) {
goto leave;
}
kwname = PyUnicode_InternFromString("baz");
if (!kwname) {
goto leave;
}
kwnames = PyTuple_New(1);
if (!kwnames) {
goto leave;
}
if (PyTuple_SetItem(kwnames, 0, kwname)) {
goto leave;
}
result = PyObject_VectorcallMethod(
name,
args,
2 | PY_VECTORCALL_ARGUMENTS_OFFSET,
kwnames
);
leave:
Py_XDECREF(name);
Py_XDECREF(args[1]);
Py_XDECREF(args[2]);
Py_XDECREF(kwnames);
return result;
}
static PyMemberDef LimitedVectorCallClass_members[] = {
{"__vectorcalloffset__", Py_T_PYSSIZET, sizeof(PyObject), Py_READONLY},
{NULL}
};
static PyType_Slot LimitedVectorallClass_slots[] = {
{Py_tp_new, LimitedVectorCallClass_new},
{Py_tp_call, LimitedVectorCallClass_tpcall},
{Py_tp_members, LimitedVectorCallClass_members},
{0},
};
static PyType_Spec LimitedVectorCallClass_spec = {
.name = "_testlimitedcapi.LimitedVectorCallClass",
.basicsize = (int)(sizeof(PyObject) + sizeof(vectorcallfunc)),
.flags = Py_TPFLAGS_DEFAULT
| Py_TPFLAGS_HAVE_VECTORCALL
| Py_TPFLAGS_BASETYPE,
.slots = LimitedVectorallClass_slots,
};
static PyMethodDef TestMethods[] = {
_TESTLIMITEDCAPI_CALL_VECTORCALL_METHODDEF
_TESTLIMITEDCAPI_CALL_VECTORCALL_METHOD_METHODDEF
{NULL},
};
int
_PyTestLimitedCAPI_Init_VectorcallLimited(PyObject *m)
{
if (PyModule_AddFunctions(m, TestMethods) < 0) {
return -1;
}
PyObject *LimitedVectorCallClass = PyType_FromModuleAndSpec(
m, &LimitedVectorCallClass_spec, NULL);
if (!LimitedVectorCallClass) {
return -1;
}
if (PyModule_AddType(m, (PyTypeObject *)LimitedVectorCallClass) < 0) {
return -1;
}
Py_DECREF(LimitedVectorCallClass);
return 0;
}