blob: 5bef9f566e8fe36ae8c18a2bba722f4f62966aef [file] [log] [blame]
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001#include "Python.h"
Victor Stinnerbcda8f12018-11-21 22:27:47 +01002#include "pycore_object.h"
Victor Stinner621cebe2018-11-12 16:53:38 +01003#include "pycore_pystate.h"
Victor Stinnerec13b932018-11-25 23:56:17 +01004#include "pycore_tupleobject.h"
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01005#include "frameobject.h"
6
7
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +02008static PyObject *
9cfunction_call_varargs(PyObject *func, PyObject *args, PyObject *kwargs);
10
11
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010012static PyObject *
13null_error(void)
14{
15 if (!PyErr_Occurred())
16 PyErr_SetString(PyExc_SystemError,
17 "null argument to internal routine");
18 return NULL;
19}
20
21
22PyObject*
23_Py_CheckFunctionResult(PyObject *callable, PyObject *result, const char *where)
24{
25 int err_occurred = (PyErr_Occurred() != NULL);
26
27 assert((callable != NULL) ^ (where != NULL));
28
29 if (result == NULL) {
30 if (!err_occurred) {
31 if (callable)
32 PyErr_Format(PyExc_SystemError,
33 "%R returned NULL without setting an error",
34 callable);
35 else
36 PyErr_Format(PyExc_SystemError,
37 "%s returned NULL without setting an error",
38 where);
39#ifdef Py_DEBUG
40 /* Ensure that the bug is caught in debug mode */
41 Py_FatalError("a function returned NULL without setting an error");
42#endif
43 return NULL;
44 }
45 }
46 else {
47 if (err_occurred) {
48 Py_DECREF(result);
49
50 if (callable) {
51 _PyErr_FormatFromCause(PyExc_SystemError,
52 "%R returned a result with an error set",
53 callable);
54 }
55 else {
56 _PyErr_FormatFromCause(PyExc_SystemError,
57 "%s returned a result with an error set",
58 where);
59 }
60#ifdef Py_DEBUG
61 /* Ensure that the bug is caught in debug mode */
62 Py_FatalError("a function returned a result with an error set");
63#endif
64 return NULL;
65 }
66 }
67 return result;
68}
69
70
71/* --- Core PyObject call functions ------------------------------- */
72
73PyObject *
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +020074_PyObject_FastCallDict(PyObject *callable, PyObject *const *args,
75 size_t nargsf, PyObject *kwargs)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010076{
77 /* _PyObject_FastCallDict() must not be called with an exception set,
78 because it can clear it (directly or indirectly) and so the
79 caller loses its exception */
80 assert(!PyErr_Occurred());
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010081 assert(callable != NULL);
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +020082
83 Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010084 assert(nargs >= 0);
85 assert(nargs == 0 || args != NULL);
86 assert(kwargs == NULL || PyDict_Check(kwargs));
87
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +020088 vectorcallfunc func = _PyVectorcall_Function(callable);
89 if (func == NULL) {
90 /* Use tp_call instead */
91 return _PyObject_MakeTpCall(callable, args, nargs, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010092 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +020093
94 PyObject *res;
95 if (kwargs == NULL) {
96 res = func(callable, args, nargsf, NULL);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +010097 }
98 else {
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +020099 PyObject *kwnames;
100 PyObject *const *newargs;
101 if (_PyStack_UnpackDict(args, nargs, kwargs, &newargs, &kwnames) < 0) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100102 return NULL;
103 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200104 res = func(callable, newargs, nargs, kwnames);
105 if (kwnames != NULL) {
106 Py_ssize_t i, n = PyTuple_GET_SIZE(kwnames) + nargs;
107 for (i = 0; i < n; i++) {
108 Py_DECREF(newargs[i]);
109 }
110 PyMem_Free((PyObject **)newargs);
111 Py_DECREF(kwnames);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100112 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100113 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200114 return _Py_CheckFunctionResult(callable, res, NULL);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100115}
116
117
118PyObject *
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200119_PyObject_MakeTpCall(PyObject *callable, PyObject *const *args, Py_ssize_t nargs, PyObject *keywords)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100120{
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200121 /* Slow path: build a temporary tuple for positional arguments and a
122 * temporary dictionary for keyword arguments (if any) */
123 ternaryfunc call = Py_TYPE(callable)->tp_call;
124 if (call == NULL) {
125 PyErr_Format(PyExc_TypeError, "'%.200s' object is not callable",
126 Py_TYPE(callable)->tp_name);
127 return NULL;
128 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100129
130 assert(nargs >= 0);
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200131 assert(nargs == 0 || args != NULL);
132 assert(keywords == NULL || PyTuple_Check(keywords) || PyDict_Check(keywords));
133 PyObject *argstuple = _PyTuple_FromArray(args, nargs);
134 if (argstuple == NULL) {
135 return NULL;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100136 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200137
138 PyObject *kwdict;
139 if (keywords == NULL || PyDict_Check(keywords)) {
140 kwdict = keywords;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100141 }
142 else {
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200143 if (PyTuple_GET_SIZE(keywords)) {
144 assert(args != NULL);
145 kwdict = _PyStack_AsDict(args + nargs, keywords);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100146 if (kwdict == NULL) {
147 Py_DECREF(argstuple);
148 return NULL;
149 }
150 }
151 else {
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200152 keywords = kwdict = NULL;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100153 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100154 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200155
156 PyObject *result = NULL;
157 if (Py_EnterRecursiveCall(" while calling a Python object") == 0)
158 {
159 result = call(callable, argstuple, kwdict);
160 Py_LeaveRecursiveCall();
161 }
162
163 Py_DECREF(argstuple);
164 if (kwdict != keywords) {
165 Py_DECREF(kwdict);
166 }
167
168 result = _Py_CheckFunctionResult(callable, result, NULL);
169 return result;
170}
171
172
173PyObject *
174PyVectorcall_Call(PyObject *callable, PyObject *tuple, PyObject *kwargs)
175{
Petr Viktorinfb9423f2019-06-02 23:52:20 +0200176 /* get vectorcallfunc as in _PyVectorcall_Function, but without
177 * the _Py_TPFLAGS_HAVE_VECTORCALL check */
178 Py_ssize_t offset = Py_TYPE(callable)->tp_vectorcall_offset;
179 if ((offset <= 0) || (!Py_TYPE(callable)->tp_call)) {
180 PyErr_Format(PyExc_TypeError, "'%.200s' object does not support vectorcall",
181 Py_TYPE(callable)->tp_name);
182 return NULL;
183 }
184 vectorcallfunc func = *(vectorcallfunc *)(((char *)callable) + offset);
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200185 if (func == NULL) {
186 PyErr_Format(PyExc_TypeError, "'%.200s' object does not support vectorcall",
187 Py_TYPE(callable)->tp_name);
188 return NULL;
189 }
Petr Viktorinfb9423f2019-06-02 23:52:20 +0200190
191 /* Convert arguments & call */
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200192 PyObject *const *args;
193 Py_ssize_t nargs = PyTuple_GET_SIZE(tuple);
194 PyObject *kwnames;
195 if (_PyStack_UnpackDict(_PyTuple_ITEMS(tuple), nargs,
196 kwargs, &args, &kwnames) < 0) {
197 return NULL;
198 }
199 PyObject *result = func(callable, args, nargs, kwnames);
200 if (kwnames != NULL) {
201 Py_ssize_t i, n = PyTuple_GET_SIZE(kwnames) + nargs;
202 for (i = 0; i < n; i++) {
203 Py_DECREF(args[i]);
204 }
205 PyMem_Free((PyObject **)args);
206 Py_DECREF(kwnames);
207 }
208
209 return result;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100210}
211
212
213PyObject *
214PyObject_Call(PyObject *callable, PyObject *args, PyObject *kwargs)
215{
216 ternaryfunc call;
217 PyObject *result;
218
219 /* PyObject_Call() must not be called with an exception set,
220 because it can clear it (directly or indirectly) and so the
221 caller loses its exception */
222 assert(!PyErr_Occurred());
223 assert(PyTuple_Check(args));
224 assert(kwargs == NULL || PyDict_Check(kwargs));
225
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200226 if (_PyVectorcall_Function(callable) != NULL) {
227 return PyVectorcall_Call(callable, args, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100228 }
229 else if (PyCFunction_Check(callable)) {
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200230 /* This must be a METH_VARARGS function, otherwise we would be
231 * in the previous case */
232 return cfunction_call_varargs(callable, args, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100233 }
234 else {
235 call = callable->ob_type->tp_call;
236 if (call == NULL) {
237 PyErr_Format(PyExc_TypeError, "'%.200s' object is not callable",
238 callable->ob_type->tp_name);
239 return NULL;
240 }
241
242 if (Py_EnterRecursiveCall(" while calling a Python object"))
243 return NULL;
244
245 result = (*call)(callable, args, kwargs);
246
247 Py_LeaveRecursiveCall();
248
249 return _Py_CheckFunctionResult(callable, result, NULL);
250 }
251}
252
253
254/* --- PyFunction call functions ---------------------------------- */
255
256static PyObject* _Py_HOT_FUNCTION
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200257function_code_fastcall(PyCodeObject *co, PyObject *const *args, Py_ssize_t nargs,
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100258 PyObject *globals)
259{
260 PyFrameObject *f;
Victor Stinner50b48572018-11-01 01:51:40 +0100261 PyThreadState *tstate = _PyThreadState_GET();
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100262 PyObject **fastlocals;
263 Py_ssize_t i;
264 PyObject *result;
265
266 assert(globals != NULL);
267 /* XXX Perhaps we should create a specialized
268 _PyFrame_New_NoTrack() that doesn't take locals, but does
269 take builtins without sanity checking them.
270 */
271 assert(tstate != NULL);
272 f = _PyFrame_New_NoTrack(tstate, co, globals, NULL);
273 if (f == NULL) {
274 return NULL;
275 }
276
277 fastlocals = f->f_localsplus;
278
279 for (i = 0; i < nargs; i++) {
280 Py_INCREF(*args);
281 fastlocals[i] = *args++;
282 }
283 result = PyEval_EvalFrameEx(f,0);
284
285 if (Py_REFCNT(f) > 1) {
286 Py_DECREF(f);
287 _PyObject_GC_TRACK(f);
288 }
289 else {
290 ++tstate->recursion_depth;
291 Py_DECREF(f);
292 --tstate->recursion_depth;
293 }
294 return result;
295}
296
297
298PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200299_PyFunction_FastCallDict(PyObject *func, PyObject *const *args, Py_ssize_t nargs,
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100300 PyObject *kwargs)
301{
302 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
303 PyObject *globals = PyFunction_GET_GLOBALS(func);
304 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
305 PyObject *kwdefs, *closure, *name, *qualname;
306 PyObject *kwtuple, **k;
307 PyObject **d;
308 Py_ssize_t nd, nk;
309 PyObject *result;
310
311 assert(func != NULL);
312 assert(nargs >= 0);
313 assert(nargs == 0 || args != NULL);
314 assert(kwargs == NULL || PyDict_Check(kwargs));
315
316 if (co->co_kwonlyargcount == 0 &&
317 (kwargs == NULL || PyDict_GET_SIZE(kwargs) == 0) &&
Victor Stinner086c3ae2017-10-25 05:26:17 -0700318 (co->co_flags & ~PyCF_MASK) == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE))
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100319 {
320 /* Fast paths */
Pablo Galindocd74e662019-06-01 18:08:04 +0100321 if (argdefs == NULL && co->co_argcount == nargs) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100322 return function_code_fastcall(co, args, nargs, globals);
323 }
324 else if (nargs == 0 && argdefs != NULL
Pablo Galindocd74e662019-06-01 18:08:04 +0100325 && co->co_argcount == PyTuple_GET_SIZE(argdefs)) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100326 /* function called with no arguments, but all parameters have
327 a default value: use default values as arguments .*/
Victor Stinnerd17a6932018-11-09 16:56:48 +0100328 args = _PyTuple_ITEMS(argdefs);
Serhiy Storchakafff9a312017-03-21 08:53:25 +0200329 return function_code_fastcall(co, args, PyTuple_GET_SIZE(argdefs),
330 globals);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100331 }
332 }
333
334 nk = (kwargs != NULL) ? PyDict_GET_SIZE(kwargs) : 0;
335 if (nk != 0) {
336 Py_ssize_t pos, i;
337
Victor Stinner561ca802017-02-23 18:26:43 +0100338 /* bpo-29318, bpo-27840: Caller and callee functions must not share
339 the dictionary: kwargs must be copied. */
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100340 kwtuple = PyTuple_New(2 * nk);
341 if (kwtuple == NULL) {
342 return NULL;
343 }
344
Victor Stinnerd17a6932018-11-09 16:56:48 +0100345 k = _PyTuple_ITEMS(kwtuple);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100346 pos = i = 0;
347 while (PyDict_Next(kwargs, &pos, &k[i], &k[i+1])) {
348 /* We must hold strong references because keyword arguments can be
349 indirectly modified while the function is called:
350 see issue #2016 and test_extcall */
351 Py_INCREF(k[i]);
352 Py_INCREF(k[i+1]);
353 i += 2;
354 }
Eric Lippert5a95ba22018-10-22 08:52:46 -0700355 assert(i / 2 == nk);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100356 }
357 else {
358 kwtuple = NULL;
359 k = NULL;
360 }
361
362 kwdefs = PyFunction_GET_KW_DEFAULTS(func);
363 closure = PyFunction_GET_CLOSURE(func);
364 name = ((PyFunctionObject *)func) -> func_name;
365 qualname = ((PyFunctionObject *)func) -> func_qualname;
366
367 if (argdefs != NULL) {
Victor Stinnerd17a6932018-11-09 16:56:48 +0100368 d = _PyTuple_ITEMS(argdefs);
Serhiy Storchakafff9a312017-03-21 08:53:25 +0200369 nd = PyTuple_GET_SIZE(argdefs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100370 }
371 else {
372 d = NULL;
373 nd = 0;
374 }
375
376 result = _PyEval_EvalCodeWithName((PyObject*)co, globals, (PyObject *)NULL,
377 args, nargs,
Zackery Spytzc6ea8972017-07-31 08:24:37 -0600378 k, k != NULL ? k + 1 : NULL, nk, 2,
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100379 d, nd, kwdefs,
380 closure, name, qualname);
381 Py_XDECREF(kwtuple);
382 return result;
383}
384
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200385
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100386PyObject *
Jeroen Demeyer37788bc2019-05-30 15:11:22 +0200387_PyFunction_Vectorcall(PyObject *func, PyObject* const* stack,
388 size_t nargsf, PyObject *kwnames)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100389{
390 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
391 PyObject *globals = PyFunction_GET_GLOBALS(func);
392 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
393 PyObject *kwdefs, *closure, *name, *qualname;
394 PyObject **d;
395 Py_ssize_t nkwargs = (kwnames == NULL) ? 0 : PyTuple_GET_SIZE(kwnames);
396 Py_ssize_t nd;
397
398 assert(PyFunction_Check(func));
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200399 Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100400 assert(nargs >= 0);
401 assert(kwnames == NULL || PyTuple_CheckExact(kwnames));
402 assert((nargs == 0 && nkwargs == 0) || stack != NULL);
403 /* kwnames must only contains str strings, no subclass, and all keys must
404 be unique */
405
406 if (co->co_kwonlyargcount == 0 && nkwargs == 0 &&
Victor Stinner086c3ae2017-10-25 05:26:17 -0700407 (co->co_flags & ~PyCF_MASK) == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE))
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100408 {
Pablo Galindocd74e662019-06-01 18:08:04 +0100409 if (argdefs == NULL && co->co_argcount == nargs) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100410 return function_code_fastcall(co, stack, nargs, globals);
411 }
412 else if (nargs == 0 && argdefs != NULL
Pablo Galindocd74e662019-06-01 18:08:04 +0100413 && co->co_argcount == PyTuple_GET_SIZE(argdefs)) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100414 /* function called with no arguments, but all parameters have
415 a default value: use default values as arguments .*/
Victor Stinnerd17a6932018-11-09 16:56:48 +0100416 stack = _PyTuple_ITEMS(argdefs);
Serhiy Storchakafff9a312017-03-21 08:53:25 +0200417 return function_code_fastcall(co, stack, PyTuple_GET_SIZE(argdefs),
418 globals);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100419 }
420 }
421
422 kwdefs = PyFunction_GET_KW_DEFAULTS(func);
423 closure = PyFunction_GET_CLOSURE(func);
424 name = ((PyFunctionObject *)func) -> func_name;
425 qualname = ((PyFunctionObject *)func) -> func_qualname;
426
427 if (argdefs != NULL) {
Victor Stinnerd17a6932018-11-09 16:56:48 +0100428 d = _PyTuple_ITEMS(argdefs);
Serhiy Storchakafff9a312017-03-21 08:53:25 +0200429 nd = PyTuple_GET_SIZE(argdefs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100430 }
431 else {
432 d = NULL;
433 nd = 0;
434 }
435 return _PyEval_EvalCodeWithName((PyObject*)co, globals, (PyObject *)NULL,
436 stack, nargs,
Victor Stinnerd17a6932018-11-09 16:56:48 +0100437 nkwargs ? _PyTuple_ITEMS(kwnames) : NULL,
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100438 stack + nargs,
439 nkwargs, 1,
440 d, (int)nd, kwdefs,
441 closure, name, qualname);
442}
443
444
445/* --- PyCFunction call functions --------------------------------- */
446
447PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200448_PyMethodDef_RawFastCallDict(PyMethodDef *method, PyObject *self,
449 PyObject *const *args, Py_ssize_t nargs,
450 PyObject *kwargs)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100451{
452 /* _PyMethodDef_RawFastCallDict() must not be called with an exception set,
453 because it can clear it (directly or indirectly) and so the
454 caller loses its exception */
455 assert(!PyErr_Occurred());
456
457 assert(method != NULL);
458 assert(nargs >= 0);
459 assert(nargs == 0 || args != NULL);
460 assert(kwargs == NULL || PyDict_Check(kwargs));
461
462 PyCFunction meth = method->ml_meth;
463 int flags = method->ml_flags & ~(METH_CLASS | METH_STATIC | METH_COEXIST);
464 PyObject *result = NULL;
465
466 if (Py_EnterRecursiveCall(" while calling a Python object")) {
467 return NULL;
468 }
469
470 switch (flags)
471 {
472 case METH_NOARGS:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300473 if (kwargs != NULL && PyDict_GET_SIZE(kwargs) != 0) {
474 goto no_keyword_error;
475 }
476
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100477 if (nargs != 0) {
478 PyErr_Format(PyExc_TypeError,
479 "%.200s() takes no arguments (%zd given)",
480 method->ml_name, nargs);
481 goto exit;
482 }
483
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100484 result = (*meth) (self, NULL);
485 break;
486
487 case METH_O:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300488 if (kwargs != NULL && PyDict_GET_SIZE(kwargs) != 0) {
489 goto no_keyword_error;
490 }
491
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100492 if (nargs != 1) {
493 PyErr_Format(PyExc_TypeError,
494 "%.200s() takes exactly one argument (%zd given)",
495 method->ml_name, nargs);
496 goto exit;
497 }
498
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100499 result = (*meth) (self, args[0]);
500 break;
501
502 case METH_VARARGS:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300503 if (kwargs != NULL && PyDict_GET_SIZE(kwargs) != 0) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100504 goto no_keyword_error;
505 }
Stefan Krahf432a322017-08-21 13:09:59 +0200506 /* fall through */
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100507
508 case METH_VARARGS | METH_KEYWORDS:
509 {
510 /* Slow-path: create a temporary tuple for positional arguments */
Sergey Fedoseevf1b9abe2019-02-26 02:37:26 +0500511 PyObject *argstuple = _PyTuple_FromArray(args, nargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100512 if (argstuple == NULL) {
513 goto exit;
514 }
515
516 if (flags & METH_KEYWORDS) {
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200517 result = (*(PyCFunctionWithKeywords)(void(*)(void))meth) (self, argstuple, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100518 }
519 else {
520 result = (*meth) (self, argstuple);
521 }
522 Py_DECREF(argstuple);
523 break;
524 }
525
526 case METH_FASTCALL:
527 {
Serhiy Storchaka6969eaf2017-07-03 21:20:15 +0300528 if (kwargs != NULL && PyDict_GET_SIZE(kwargs) != 0) {
529 goto no_keyword_error;
530 }
531
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200532 result = (*(_PyCFunctionFast)(void(*)(void))meth) (self, args, nargs);
Serhiy Storchaka6969eaf2017-07-03 21:20:15 +0300533 break;
534 }
535
536 case METH_FASTCALL | METH_KEYWORDS:
537 {
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200538 PyObject *const *stack;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100539 PyObject *kwnames;
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200540 _PyCFunctionFastWithKeywords fastmeth = (_PyCFunctionFastWithKeywords)(void(*)(void))meth;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100541
542 if (_PyStack_UnpackDict(args, nargs, kwargs, &stack, &kwnames) < 0) {
543 goto exit;
544 }
545
546 result = (*fastmeth) (self, stack, nargs, kwnames);
Jeroen Demeyer77aa3962019-05-22 13:09:35 +0200547 if (kwnames != NULL) {
548 Py_ssize_t i, n = nargs + PyTuple_GET_SIZE(kwnames);
549 for (i = 0; i < n; i++) {
550 Py_DECREF(stack[i]);
551 }
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200552 PyMem_Free((PyObject **)stack);
Jeroen Demeyer77aa3962019-05-22 13:09:35 +0200553 Py_DECREF(kwnames);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100554 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100555 break;
556 }
557
558 default:
559 PyErr_SetString(PyExc_SystemError,
560 "Bad call flags in _PyMethodDef_RawFastCallDict. "
561 "METH_OLDARGS is no longer supported!");
562 goto exit;
563 }
564
565 goto exit;
566
567no_keyword_error:
568 PyErr_Format(PyExc_TypeError,
569 "%.200s() takes no keyword arguments",
Sylvaind67a1032017-03-27 23:36:08 +0200570 method->ml_name);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100571
572exit:
573 Py_LeaveRecursiveCall();
574 return result;
575}
576
577
578PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200579_PyCFunction_FastCallDict(PyObject *func,
580 PyObject *const *args, Py_ssize_t nargs,
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100581 PyObject *kwargs)
582{
583 PyObject *result;
584
585 assert(func != NULL);
586 assert(PyCFunction_Check(func));
587
588 result = _PyMethodDef_RawFastCallDict(((PyCFunctionObject*)func)->m_ml,
589 PyCFunction_GET_SELF(func),
590 args, nargs, kwargs);
591 result = _Py_CheckFunctionResult(func, result, NULL);
592 return result;
593}
594
595
596PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +0200597_PyMethodDef_RawFastCallKeywords(PyMethodDef *method, PyObject *self,
598 PyObject *const *args, Py_ssize_t nargs,
599 PyObject *kwnames)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100600{
601 /* _PyMethodDef_RawFastCallKeywords() must not be called with an exception set,
602 because it can clear it (directly or indirectly) and so the
603 caller loses its exception */
604 assert(!PyErr_Occurred());
605
606 assert(method != NULL);
607 assert(nargs >= 0);
608 assert(kwnames == NULL || PyTuple_CheckExact(kwnames));
609 /* kwnames must only contains str strings, no subclass, and all keys must
610 be unique */
611
612 PyCFunction meth = method->ml_meth;
613 int flags = method->ml_flags & ~(METH_CLASS | METH_STATIC | METH_COEXIST);
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300614 Py_ssize_t nkwargs = kwnames == NULL ? 0 : PyTuple_GET_SIZE(kwnames);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100615 PyObject *result = NULL;
616
617 if (Py_EnterRecursiveCall(" while calling a Python object")) {
618 return NULL;
619 }
620
621 switch (flags)
622 {
623 case METH_NOARGS:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300624 if (nkwargs) {
625 goto no_keyword_error;
626 }
627
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100628 if (nargs != 0) {
629 PyErr_Format(PyExc_TypeError,
630 "%.200s() takes no arguments (%zd given)",
631 method->ml_name, nargs);
632 goto exit;
633 }
634
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100635 result = (*meth) (self, NULL);
636 break;
637
638 case METH_O:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300639 if (nkwargs) {
640 goto no_keyword_error;
641 }
642
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100643 if (nargs != 1) {
644 PyErr_Format(PyExc_TypeError,
645 "%.200s() takes exactly one argument (%zd given)",
646 method->ml_name, nargs);
647 goto exit;
648 }
649
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100650 result = (*meth) (self, args[0]);
651 break;
652
653 case METH_FASTCALL:
Serhiy Storchaka6969eaf2017-07-03 21:20:15 +0300654 if (nkwargs) {
655 goto no_keyword_error;
656 }
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200657 result = ((_PyCFunctionFast)(void(*)(void))meth) (self, args, nargs);
Serhiy Storchaka6969eaf2017-07-03 21:20:15 +0300658 break;
659
660 case METH_FASTCALL | METH_KEYWORDS:
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100661 /* Fast-path: avoid temporary dict to pass keyword arguments */
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200662 result = ((_PyCFunctionFastWithKeywords)(void(*)(void))meth) (self, args, nargs, kwnames);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100663 break;
664
665 case METH_VARARGS:
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300666 if (nkwargs) {
667 goto no_keyword_error;
668 }
Stefan Krahf432a322017-08-21 13:09:59 +0200669 /* fall through */
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300670
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100671 case METH_VARARGS | METH_KEYWORDS:
672 {
673 /* Slow-path: create a temporary tuple for positional arguments
674 and a temporary dict for keyword arguments */
675 PyObject *argtuple;
676
Sergey Fedoseevf1b9abe2019-02-26 02:37:26 +0500677 argtuple = _PyTuple_FromArray(args, nargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100678 if (argtuple == NULL) {
679 goto exit;
680 }
681
682 if (flags & METH_KEYWORDS) {
683 PyObject *kwdict;
684
685 if (nkwargs > 0) {
686 kwdict = _PyStack_AsDict(args + nargs, kwnames);
687 if (kwdict == NULL) {
688 Py_DECREF(argtuple);
689 goto exit;
690 }
691 }
692 else {
693 kwdict = NULL;
694 }
695
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200696 result = (*(PyCFunctionWithKeywords)(void(*)(void))meth) (self, argtuple, kwdict);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100697 Py_XDECREF(kwdict);
698 }
699 else {
700 result = (*meth) (self, argtuple);
701 }
702 Py_DECREF(argtuple);
703 break;
704 }
705
706 default:
707 PyErr_SetString(PyExc_SystemError,
Jeroen Demeyera8b46942019-05-17 12:21:35 +0200708 "Bad call flags in _PyMethodDef_RawFastCallKeywords. "
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100709 "METH_OLDARGS is no longer supported!");
710 goto exit;
711 }
712
713 goto exit;
714
715no_keyword_error:
716 PyErr_Format(PyExc_TypeError,
717 "%.200s() takes no keyword arguments",
718 method->ml_name);
719
720exit:
721 Py_LeaveRecursiveCall();
722 return result;
723}
724
725
726PyObject *
Jeroen Demeyer37788bc2019-05-30 15:11:22 +0200727_PyCFunction_Vectorcall(PyObject *func,
728 PyObject *const *args, size_t nargsf,
729 PyObject *kwnames)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100730{
731 PyObject *result;
732
733 assert(func != NULL);
734 assert(PyCFunction_Check(func));
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200735 Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100736
737 result = _PyMethodDef_RawFastCallKeywords(((PyCFunctionObject*)func)->m_ml,
738 PyCFunction_GET_SELF(func),
739 args, nargs, kwnames);
740 result = _Py_CheckFunctionResult(func, result, NULL);
741 return result;
742}
743
744
745static PyObject *
746cfunction_call_varargs(PyObject *func, PyObject *args, PyObject *kwargs)
747{
748 assert(!PyErr_Occurred());
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300749 assert(kwargs == NULL || PyDict_Check(kwargs));
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100750
751 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
752 PyObject *self = PyCFunction_GET_SELF(func);
753 PyObject *result;
754
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200755 assert(PyCFunction_GET_FLAGS(func) & METH_VARARGS);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100756 if (PyCFunction_GET_FLAGS(func) & METH_KEYWORDS) {
757 if (Py_EnterRecursiveCall(" while calling a Python object")) {
758 return NULL;
759 }
760
Serhiy Storchaka62be7422018-11-27 13:27:31 +0200761 result = (*(PyCFunctionWithKeywords)(void(*)(void))meth)(self, args, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100762
763 Py_LeaveRecursiveCall();
764 }
765 else {
Serhiy Storchaka5eb788b2017-06-06 18:45:22 +0300766 if (kwargs != NULL && PyDict_GET_SIZE(kwargs) != 0) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100767 PyErr_Format(PyExc_TypeError, "%.200s() takes no keyword arguments",
768 ((PyCFunctionObject*)func)->m_ml->ml_name);
769 return NULL;
770 }
771
772 if (Py_EnterRecursiveCall(" while calling a Python object")) {
773 return NULL;
774 }
775
776 result = (*meth)(self, args);
777
778 Py_LeaveRecursiveCall();
779 }
780
781 return _Py_CheckFunctionResult(func, result, NULL);
782}
783
784
785PyObject *
786PyCFunction_Call(PyObject *func, PyObject *args, PyObject *kwargs)
787{
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200788 /* For METH_VARARGS, we cannot use vectorcall as the vectorcall pointer
789 * is NULL. This is intentional, since vectorcall would be slower. */
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100790 if (PyCFunction_GET_FLAGS(func) & METH_VARARGS) {
791 return cfunction_call_varargs(func, args, kwargs);
792 }
Jeroen Demeyeraacc77f2019-05-29 20:31:52 +0200793 return PyVectorcall_Call(func, args, kwargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100794}
795
796
797/* --- More complex call functions -------------------------------- */
798
799/* External interface to call any callable object.
800 The args must be a tuple or NULL. The kwargs must be a dict or NULL. */
801PyObject *
802PyEval_CallObjectWithKeywords(PyObject *callable,
803 PyObject *args, PyObject *kwargs)
804{
805#ifdef Py_DEBUG
806 /* PyEval_CallObjectWithKeywords() must not be called with an exception
807 set. It raises a new exception if parameters are invalid or if
808 PyTuple_New() fails, and so the original exception is lost. */
809 assert(!PyErr_Occurred());
810#endif
811
INADA Naoki3824cd82017-03-01 20:41:03 +0900812 if (args != NULL && !PyTuple_Check(args)) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100813 PyErr_SetString(PyExc_TypeError,
814 "argument list must be a tuple");
815 return NULL;
816 }
817
818 if (kwargs != NULL && !PyDict_Check(kwargs)) {
819 PyErr_SetString(PyExc_TypeError,
820 "keyword list must be a dictionary");
821 return NULL;
822 }
823
INADA Naoki3824cd82017-03-01 20:41:03 +0900824 if (args == NULL) {
825 return _PyObject_FastCallDict(callable, NULL, 0, kwargs);
826 }
827 else {
828 return PyObject_Call(callable, args, kwargs);
829 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100830}
831
832
833PyObject *
834PyObject_CallObject(PyObject *callable, PyObject *args)
835{
836 return PyEval_CallObjectWithKeywords(callable, args, NULL);
837}
838
839
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100840/* Call callable(obj, *args, **kwargs). */
841PyObject *
842_PyObject_Call_Prepend(PyObject *callable,
843 PyObject *obj, PyObject *args, PyObject *kwargs)
844{
845 PyObject *small_stack[_PY_FASTCALL_SMALL_STACK];
846 PyObject **stack;
847 Py_ssize_t argcount;
848 PyObject *result;
849
850 assert(PyTuple_Check(args));
851
852 argcount = PyTuple_GET_SIZE(args);
853 if (argcount + 1 <= (Py_ssize_t)Py_ARRAY_LENGTH(small_stack)) {
854 stack = small_stack;
855 }
856 else {
857 stack = PyMem_Malloc((argcount + 1) * sizeof(PyObject *));
858 if (stack == NULL) {
859 PyErr_NoMemory();
860 return NULL;
861 }
862 }
863
864 /* use borrowed references */
865 stack[0] = obj;
866 memcpy(&stack[1],
Victor Stinnerd17a6932018-11-09 16:56:48 +0100867 _PyTuple_ITEMS(args),
868 argcount * sizeof(PyObject *));
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100869
870 result = _PyObject_FastCallDict(callable,
871 stack, argcount + 1,
872 kwargs);
873 if (stack != small_stack) {
874 PyMem_Free(stack);
875 }
876 return result;
877}
878
879
880/* --- Call with a format string ---------------------------------- */
881
882static PyObject *
883_PyObject_CallFunctionVa(PyObject *callable, const char *format,
884 va_list va, int is_size_t)
885{
886 PyObject* small_stack[_PY_FASTCALL_SMALL_STACK];
887 const Py_ssize_t small_stack_len = Py_ARRAY_LENGTH(small_stack);
888 PyObject **stack;
889 Py_ssize_t nargs, i;
890 PyObject *result;
891
892 if (callable == NULL) {
893 return null_error();
894 }
895
896 if (!format || !*format) {
897 return _PyObject_CallNoArg(callable);
898 }
899
900 if (is_size_t) {
901 stack = _Py_VaBuildStack_SizeT(small_stack, small_stack_len,
902 format, va, &nargs);
903 }
904 else {
905 stack = _Py_VaBuildStack(small_stack, small_stack_len,
906 format, va, &nargs);
907 }
908 if (stack == NULL) {
909 return NULL;
910 }
911
912 if (nargs == 1 && PyTuple_Check(stack[0])) {
913 /* Special cases for backward compatibility:
914 - PyObject_CallFunction(func, "O", tuple) calls func(*tuple)
915 - PyObject_CallFunction(func, "(OOO)", arg1, arg2, arg3) calls
916 func(*(arg1, arg2, arg3)): func(arg1, arg2, arg3) */
917 PyObject *args = stack[0];
918 result = _PyObject_FastCall(callable,
Victor Stinnerd17a6932018-11-09 16:56:48 +0100919 _PyTuple_ITEMS(args),
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100920 PyTuple_GET_SIZE(args));
921 }
922 else {
923 result = _PyObject_FastCall(callable, stack, nargs);
924 }
925
926 for (i = 0; i < nargs; ++i) {
927 Py_DECREF(stack[i]);
928 }
929 if (stack != small_stack) {
930 PyMem_Free(stack);
931 }
932 return result;
933}
934
935
936PyObject *
937PyObject_CallFunction(PyObject *callable, const char *format, ...)
938{
939 va_list va;
940 PyObject *result;
941
942 va_start(va, format);
943 result = _PyObject_CallFunctionVa(callable, format, va, 0);
944 va_end(va);
945
946 return result;
947}
948
949
INADA Naokiaa289a52017-03-14 18:00:59 +0900950/* PyEval_CallFunction is exact copy of PyObject_CallFunction.
951 * This function is kept for backward compatibility.
952 */
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100953PyObject *
954PyEval_CallFunction(PyObject *callable, const char *format, ...)
955{
INADA Naokiaa289a52017-03-14 18:00:59 +0900956 va_list va;
957 PyObject *result;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100958
INADA Naokiaa289a52017-03-14 18:00:59 +0900959 va_start(va, format);
960 result = _PyObject_CallFunctionVa(callable, format, va, 0);
961 va_end(va);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100962
INADA Naokiaa289a52017-03-14 18:00:59 +0900963 return result;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +0100964}
965
966
967PyObject *
968_PyObject_CallFunction_SizeT(PyObject *callable, const char *format, ...)
969{
970 va_list va;
971 PyObject *result;
972
973 va_start(va, format);
974 result = _PyObject_CallFunctionVa(callable, format, va, 1);
975 va_end(va);
976
977 return result;
978}
979
980
981static PyObject*
982callmethod(PyObject* callable, const char *format, va_list va, int is_size_t)
983{
984 assert(callable != NULL);
985
986 if (!PyCallable_Check(callable)) {
987 PyErr_Format(PyExc_TypeError,
988 "attribute of type '%.200s' is not callable",
989 Py_TYPE(callable)->tp_name);
990 return NULL;
991 }
992
993 return _PyObject_CallFunctionVa(callable, format, va, is_size_t);
994}
995
996
997PyObject *
998PyObject_CallMethod(PyObject *obj, const char *name, const char *format, ...)
999{
1000 va_list va;
1001 PyObject *callable, *retval;
1002
1003 if (obj == NULL || name == NULL) {
1004 return null_error();
1005 }
1006
1007 callable = PyObject_GetAttrString(obj, name);
1008 if (callable == NULL)
1009 return NULL;
1010
1011 va_start(va, format);
1012 retval = callmethod(callable, format, va, 0);
1013 va_end(va);
1014
1015 Py_DECREF(callable);
1016 return retval;
1017}
1018
1019
INADA Naokiaa289a52017-03-14 18:00:59 +09001020/* PyEval_CallMethod is exact copy of PyObject_CallMethod.
1021 * This function is kept for backward compatibility.
1022 */
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001023PyObject *
1024PyEval_CallMethod(PyObject *obj, const char *name, const char *format, ...)
1025{
INADA Naokiaa289a52017-03-14 18:00:59 +09001026 va_list va;
1027 PyObject *callable, *retval;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001028
INADA Naokiaa289a52017-03-14 18:00:59 +09001029 if (obj == NULL || name == NULL) {
1030 return null_error();
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001031 }
1032
INADA Naokiaa289a52017-03-14 18:00:59 +09001033 callable = PyObject_GetAttrString(obj, name);
1034 if (callable == NULL)
1035 return NULL;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001036
INADA Naokiaa289a52017-03-14 18:00:59 +09001037 va_start(va, format);
1038 retval = callmethod(callable, format, va, 0);
1039 va_end(va);
1040
1041 Py_DECREF(callable);
1042 return retval;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001043}
1044
1045
1046PyObject *
1047_PyObject_CallMethodId(PyObject *obj, _Py_Identifier *name,
1048 const char *format, ...)
1049{
1050 va_list va;
1051 PyObject *callable, *retval;
1052
1053 if (obj == NULL || name == NULL) {
1054 return null_error();
1055 }
1056
1057 callable = _PyObject_GetAttrId(obj, name);
1058 if (callable == NULL)
1059 return NULL;
1060
1061 va_start(va, format);
1062 retval = callmethod(callable, format, va, 0);
1063 va_end(va);
1064
1065 Py_DECREF(callable);
1066 return retval;
1067}
1068
1069
1070PyObject *
1071_PyObject_CallMethod_SizeT(PyObject *obj, const char *name,
1072 const char *format, ...)
1073{
1074 va_list va;
1075 PyObject *callable, *retval;
1076
1077 if (obj == NULL || name == NULL) {
1078 return null_error();
1079 }
1080
1081 callable = PyObject_GetAttrString(obj, name);
1082 if (callable == NULL)
1083 return NULL;
1084
1085 va_start(va, format);
1086 retval = callmethod(callable, format, va, 1);
1087 va_end(va);
1088
1089 Py_DECREF(callable);
1090 return retval;
1091}
1092
1093
1094PyObject *
1095_PyObject_CallMethodId_SizeT(PyObject *obj, _Py_Identifier *name,
1096 const char *format, ...)
1097{
1098 va_list va;
1099 PyObject *callable, *retval;
1100
1101 if (obj == NULL || name == NULL) {
1102 return null_error();
1103 }
1104
1105 callable = _PyObject_GetAttrId(obj, name);
1106 if (callable == NULL) {
1107 return NULL;
1108 }
1109
1110 va_start(va, format);
1111 retval = callmethod(callable, format, va, 1);
1112 va_end(va);
1113
1114 Py_DECREF(callable);
1115 return retval;
1116}
1117
1118
1119/* --- Call with "..." arguments ---------------------------------- */
1120
1121static PyObject *
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001122object_vacall(PyObject *base, PyObject *callable, va_list vargs)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001123{
1124 PyObject *small_stack[_PY_FASTCALL_SMALL_STACK];
1125 PyObject **stack;
1126 Py_ssize_t nargs;
1127 PyObject *result;
1128 Py_ssize_t i;
1129 va_list countva;
1130
1131 if (callable == NULL) {
1132 return null_error();
1133 }
1134
1135 /* Count the number of arguments */
1136 va_copy(countva, vargs);
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001137 nargs = base ? 1 : 0;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001138 while (1) {
1139 PyObject *arg = va_arg(countva, PyObject *);
1140 if (arg == NULL) {
1141 break;
1142 }
1143 nargs++;
1144 }
1145 va_end(countva);
1146
1147 /* Copy arguments */
1148 if (nargs <= (Py_ssize_t)Py_ARRAY_LENGTH(small_stack)) {
1149 stack = small_stack;
1150 }
1151 else {
1152 stack = PyMem_Malloc(nargs * sizeof(stack[0]));
1153 if (stack == NULL) {
1154 PyErr_NoMemory();
1155 return NULL;
1156 }
1157 }
1158
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001159 i = 0;
1160 if (base) {
1161 stack[i++] = base;
1162 }
1163
1164 for (; i < nargs; ++i) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001165 stack[i] = va_arg(vargs, PyObject *);
1166 }
1167
1168 /* Call the function */
1169 result = _PyObject_FastCall(callable, stack, nargs);
1170
1171 if (stack != small_stack) {
1172 PyMem_Free(stack);
1173 }
1174 return result;
1175}
1176
1177
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001178PyObject *
1179PyObject_CallMethodObjArgs(PyObject *obj, PyObject *name, ...)
1180{
1181 if (obj == NULL || name == NULL) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001182 return null_error();
1183 }
1184
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001185 PyObject *callable = NULL;
1186 int is_method = _PyObject_GetMethod(obj, name, &callable);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001187 if (callable == NULL) {
1188 return NULL;
1189 }
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001190 obj = is_method ? obj : NULL;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001191
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001192 va_list vargs;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001193 va_start(vargs, name);
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001194 PyObject *result = object_vacall(obj, callable, vargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001195 va_end(vargs);
1196
1197 Py_DECREF(callable);
1198 return result;
1199}
1200
1201
1202PyObject *
1203_PyObject_CallMethodIdObjArgs(PyObject *obj,
1204 struct _Py_Identifier *name, ...)
1205{
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001206 if (obj == NULL || name == NULL) {
1207 return null_error();
1208 }
1209
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001210 PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
1211 if (!oname) {
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001212 return NULL;
1213 }
1214
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001215 PyObject *callable = NULL;
1216 int is_method = _PyObject_GetMethod(obj, oname, &callable);
1217 if (callable == NULL) {
1218 return NULL;
1219 }
1220 obj = is_method ? obj : NULL;
1221
1222 va_list vargs;
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001223 va_start(vargs, name);
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001224 PyObject *result = object_vacall(obj, callable, vargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001225 va_end(vargs);
1226
1227 Py_DECREF(callable);
1228 return result;
1229}
1230
1231
1232PyObject *
1233PyObject_CallFunctionObjArgs(PyObject *callable, ...)
1234{
1235 va_list vargs;
1236 PyObject *result;
1237
1238 va_start(vargs, callable);
Michael J. Sullivan47dd2f92019-05-26 00:23:34 -07001239 result = object_vacall(NULL, callable, vargs);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001240 va_end(vargs);
1241
1242 return result;
1243}
1244
1245
1246/* --- PyStack functions ------------------------------------------ */
1247
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001248PyObject *
Serhiy Storchakaa5552f02017-12-15 13:11:11 +02001249_PyStack_AsDict(PyObject *const *values, PyObject *kwnames)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001250{
1251 Py_ssize_t nkwargs;
1252 PyObject *kwdict;
1253 Py_ssize_t i;
1254
1255 assert(kwnames != NULL);
1256 nkwargs = PyTuple_GET_SIZE(kwnames);
1257 kwdict = _PyDict_NewPresized(nkwargs);
1258 if (kwdict == NULL) {
1259 return NULL;
1260 }
1261
1262 for (i = 0; i < nkwargs; i++) {
1263 PyObject *key = PyTuple_GET_ITEM(kwnames, i);
1264 PyObject *value = *values++;
1265 /* If key already exists, replace it with the new value */
1266 if (PyDict_SetItem(kwdict, key, value)) {
1267 Py_DECREF(kwdict);
1268 return NULL;
1269 }
1270 }
1271 return kwdict;
1272}
1273
1274
1275int
Serhiy Storchakaa5552f02017-12-15 13:11:11 +02001276_PyStack_UnpackDict(PyObject *const *args, Py_ssize_t nargs, PyObject *kwargs,
1277 PyObject *const **p_stack, PyObject **p_kwnames)
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001278{
1279 PyObject **stack, **kwstack;
1280 Py_ssize_t nkwargs;
1281 Py_ssize_t pos, i;
1282 PyObject *key, *value;
1283 PyObject *kwnames;
1284
1285 assert(nargs >= 0);
1286 assert(kwargs == NULL || PyDict_CheckExact(kwargs));
1287
1288 if (kwargs == NULL || (nkwargs = PyDict_GET_SIZE(kwargs)) == 0) {
1289 *p_stack = args;
1290 *p_kwnames = NULL;
1291 return 0;
1292 }
1293
1294 if ((size_t)nargs > PY_SSIZE_T_MAX / sizeof(stack[0]) - (size_t)nkwargs) {
1295 PyErr_NoMemory();
1296 return -1;
1297 }
1298
1299 stack = PyMem_Malloc((nargs + nkwargs) * sizeof(stack[0]));
1300 if (stack == NULL) {
1301 PyErr_NoMemory();
1302 return -1;
1303 }
1304
1305 kwnames = PyTuple_New(nkwargs);
1306 if (kwnames == NULL) {
1307 PyMem_Free(stack);
1308 return -1;
1309 }
1310
Jeroen Demeyer77aa3962019-05-22 13:09:35 +02001311 /* Copy positional arguments */
1312 for (i = 0; i < nargs; i++) {
1313 Py_INCREF(args[i]);
1314 stack[i] = args[i];
1315 }
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001316
1317 kwstack = stack + nargs;
1318 pos = i = 0;
1319 /* This loop doesn't support lookup function mutating the dictionary
1320 to change its size. It's a deliberate choice for speed, this function is
1321 called in the performance critical hot code. */
1322 while (PyDict_Next(kwargs, &pos, &key, &value)) {
1323 Py_INCREF(key);
Jeroen Demeyer77aa3962019-05-22 13:09:35 +02001324 Py_INCREF(value);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001325 PyTuple_SET_ITEM(kwnames, i, key);
Victor Stinnerc22bfaa2017-02-12 19:27:05 +01001326 kwstack[i] = value;
1327 i++;
1328 }
1329
1330 *p_stack = stack;
1331 *p_kwnames = kwnames;
1332 return 0;
1333}