2023-09-25 08:24:19 -03:00
|
|
|
/* Test Vectorcall in the limited API */
|
|
|
|
|
2023-11-20 09:52:00 -04:00
|
|
|
#include "pyconfig.h" // Py_GIL_DISABLED
|
2023-10-30 13:06:09 -03:00
|
|
|
|
2023-11-20 09:52:00 -04:00
|
|
|
#ifndef Py_GIL_DISABLED
|
2022-08-17 08:48:43 -03:00
|
|
|
#define Py_LIMITED_API 0x030c0000 // 3.12
|
2023-10-30 13:06:09 -03:00
|
|
|
#endif
|
|
|
|
|
2022-08-09 04:03:11 -03:00
|
|
|
#include "parts.h"
|
2023-09-25 08:24:19 -03:00
|
|
|
#include "clinic/vectorcall_limited.c.h"
|
2022-08-09 04:03:11 -03:00
|
|
|
|
2023-09-25 08:24:19 -03:00
|
|
|
/*[clinic input]
|
|
|
|
module _testcapi
|
|
|
|
[clinic start generated code]*/
|
|
|
|
/*[clinic end generated code: output=da39a3ee5e6b4b0d input=6361033e795369fc]*/
|
2022-08-08 09:12:05 -03:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2023-09-25 08:24:19 -03:00
|
|
|
/*[clinic input]
|
|
|
|
_testcapi.call_vectorcall
|
|
|
|
|
|
|
|
callable: object
|
|
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
|
2022-10-27 06:45:42 -03:00
|
|
|
static PyObject *
|
2023-09-25 08:24:19 -03:00
|
|
|
_testcapi_call_vectorcall(PyObject *module, PyObject *callable)
|
|
|
|
/*[clinic end generated code: output=bae81eec97fcaad7 input=55d88f92240957ee]*/
|
2022-10-27 06:45:42 -03:00
|
|
|
{
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2023-09-25 08:24:19 -03:00
|
|
|
/*[clinic input]
|
|
|
|
_testcapi.call_vectorcall_method
|
|
|
|
|
|
|
|
callable: object
|
|
|
|
/
|
|
|
|
[clinic start generated code]*/
|
|
|
|
|
2022-10-27 06:45:42 -03:00
|
|
|
static PyObject *
|
2023-09-25 08:24:19 -03:00
|
|
|
_testcapi_call_vectorcall_method(PyObject *module, PyObject *callable)
|
|
|
|
/*[clinic end generated code: output=e661f48dda08b6fb input=5ba81c27511395b6]*/
|
2022-10-27 06:45:42 -03:00
|
|
|
{
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2022-08-08 09:12:05 -03:00
|
|
|
static PyMemberDef LimitedVectorCallClass_members[] = {
|
2023-07-25 10:28:30 -03:00
|
|
|
{"__vectorcalloffset__", Py_T_PYSSIZET, sizeof(PyObject), Py_READONLY},
|
2022-08-08 09:12:05 -03:00
|
|
|
{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 = "_testcapi.LimitedVectorCallClass",
|
|
|
|
.basicsize = (int)(sizeof(PyObject) + sizeof(vectorcallfunc)),
|
|
|
|
.flags = Py_TPFLAGS_DEFAULT
|
|
|
|
| Py_TPFLAGS_HAVE_VECTORCALL
|
|
|
|
| Py_TPFLAGS_BASETYPE,
|
|
|
|
.slots = LimitedVectorallClass_slots,
|
|
|
|
};
|
|
|
|
|
|
|
|
static PyMethodDef TestMethods[] = {
|
2023-09-25 08:24:19 -03:00
|
|
|
_TESTCAPI_CALL_VECTORCALL_METHODDEF
|
|
|
|
_TESTCAPI_CALL_VECTORCALL_METHOD_METHODDEF
|
2022-08-08 09:12:05 -03:00
|
|
|
{NULL},
|
|
|
|
};
|
|
|
|
|
|
|
|
int
|
|
|
|
_PyTestCapi_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;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|