1 #include "Python.h"
2 #include "pycore_call.h"          // _PyObject_CallNoArgsTstate()
3 #include "pycore_ceval.h"         // _PyEval_EvalFrame()
4 #include "pycore_object.h"        // _PyObject_GC_TRACK()
5 #include "pycore_pyerrors.h"      // _PyErr_Occurred()
6 #include "pycore_pystate.h"       // _PyThreadState_GET()
7 #include "pycore_tuple.h"         // _PyTuple_ITEMS()
8 #include "frameobject.h"          // _PyFrame_New_NoTrack()
9 
10 
11 static PyObject *const *
12 _PyStack_UnpackDict(PyThreadState *tstate,
13                     PyObject *const *args, Py_ssize_t nargs,
14                     PyObject *kwargs, PyObject **p_kwnames);
15 
16 static void
17 _PyStack_UnpackDict_Free(PyObject *const *stack, Py_ssize_t nargs,
18                          PyObject *kwnames);
19 
20 
21 static PyObject *
null_error(PyThreadState * tstate)22 null_error(PyThreadState *tstate)
23 {
24     if (!_PyErr_Occurred(tstate)) {
25         _PyErr_SetString(tstate, PyExc_SystemError,
26                          "null argument to internal routine");
27     }
28     return NULL;
29 }
30 
31 
32 PyObject*
_Py_CheckFunctionResult(PyThreadState * tstate,PyObject * callable,PyObject * result,const char * where)33 _Py_CheckFunctionResult(PyThreadState *tstate, PyObject *callable,
34                         PyObject *result, const char *where)
35 {
36     assert((callable != NULL) ^ (where != NULL));
37 
38     if (result == NULL) {
39         if (!_PyErr_Occurred(tstate)) {
40             if (callable)
41                 _PyErr_Format(tstate, PyExc_SystemError,
42                               "%R returned NULL without setting an exception",
43                               callable);
44             else
45                 _PyErr_Format(tstate, PyExc_SystemError,
46                               "%s returned NULL without setting an exception",
47                               where);
48 #ifdef Py_DEBUG
49             /* Ensure that the bug is caught in debug mode.
50                Py_FatalError() logs the SystemError exception raised above. */
51             Py_FatalError("a function returned NULL without setting an exception");
52 #endif
53             return NULL;
54         }
55     }
56     else {
57         if (_PyErr_Occurred(tstate)) {
58             Py_DECREF(result);
59 
60             if (callable) {
61                 _PyErr_FormatFromCauseTstate(
62                     tstate, PyExc_SystemError,
63                     "%R returned a result with an exception set", callable);
64             }
65             else {
66                 _PyErr_FormatFromCauseTstate(
67                     tstate, PyExc_SystemError,
68                     "%s returned a result with an exception set", where);
69             }
70 #ifdef Py_DEBUG
71             /* Ensure that the bug is caught in debug mode.
72                Py_FatalError() logs the SystemError exception raised above. */
73             Py_FatalError("a function returned a result with an exception set");
74 #endif
75             return NULL;
76         }
77     }
78     return result;
79 }
80 
81 
82 int
_Py_CheckSlotResult(PyObject * obj,const char * slot_name,int success)83 _Py_CheckSlotResult(PyObject *obj, const char *slot_name, int success)
84 {
85     PyThreadState *tstate = _PyThreadState_GET();
86     if (!success) {
87         if (!_PyErr_Occurred(tstate)) {
88             _Py_FatalErrorFormat(__func__,
89                                  "Slot %s of type %s failed "
90                                  "without setting an exception",
91                                  slot_name, Py_TYPE(obj)->tp_name);
92         }
93     }
94     else {
95         if (_PyErr_Occurred(tstate)) {
96             _Py_FatalErrorFormat(__func__,
97                                  "Slot %s of type %s succeeded "
98                                  "with an exception set",
99                                  slot_name, Py_TYPE(obj)->tp_name);
100         }
101     }
102     return 1;
103 }
104 
105 
106 /* --- Core PyObject call functions ------------------------------- */
107 
108 /* Call a callable Python object without any arguments */
109 PyObject *
PyObject_CallNoArgs(PyObject * func)110 PyObject_CallNoArgs(PyObject *func)
111 {
112     return _PyObject_CallNoArgs(func);
113 }
114 
115 
116 PyObject *
_PyObject_FastCallDictTstate(PyThreadState * tstate,PyObject * callable,PyObject * const * args,size_t nargsf,PyObject * kwargs)117 _PyObject_FastCallDictTstate(PyThreadState *tstate, PyObject *callable,
118                              PyObject *const *args, size_t nargsf,
119                              PyObject *kwargs)
120 {
121     assert(callable != NULL);
122 
123     /* PyObject_VectorcallDict() must not be called with an exception set,
124        because it can clear it (directly or indirectly) and so the
125        caller loses its exception */
126     assert(!_PyErr_Occurred(tstate));
127 
128     Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
129     assert(nargs >= 0);
130     assert(nargs == 0 || args != NULL);
131     assert(kwargs == NULL || PyDict_Check(kwargs));
132 
133     vectorcallfunc func = _PyVectorcall_Function(callable);
134     if (func == NULL) {
135         /* Use tp_call instead */
136         return _PyObject_MakeTpCall(tstate, callable, args, nargs, kwargs);
137     }
138 
139     PyObject *res;
140     if (kwargs == NULL || PyDict_GET_SIZE(kwargs) == 0) {
141         res = func(callable, args, nargsf, NULL);
142     }
143     else {
144         PyObject *kwnames;
145         PyObject *const *newargs;
146         newargs = _PyStack_UnpackDict(tstate,
147                                       args, nargs,
148                                       kwargs, &kwnames);
149         if (newargs == NULL) {
150             return NULL;
151         }
152         res = func(callable, newargs,
153                    nargs | PY_VECTORCALL_ARGUMENTS_OFFSET, kwnames);
154         _PyStack_UnpackDict_Free(newargs, nargs, kwnames);
155     }
156     return _Py_CheckFunctionResult(tstate, callable, res, NULL);
157 }
158 
159 
160 PyObject *
PyObject_VectorcallDict(PyObject * callable,PyObject * const * args,size_t nargsf,PyObject * kwargs)161 PyObject_VectorcallDict(PyObject *callable, PyObject *const *args,
162                        size_t nargsf, PyObject *kwargs)
163 {
164     PyThreadState *tstate = _PyThreadState_GET();
165     return _PyObject_FastCallDictTstate(tstate, callable, args, nargsf, kwargs);
166 }
167 
168 
169 PyObject *
_PyObject_MakeTpCall(PyThreadState * tstate,PyObject * callable,PyObject * const * args,Py_ssize_t nargs,PyObject * keywords)170 _PyObject_MakeTpCall(PyThreadState *tstate, PyObject *callable,
171                      PyObject *const *args, Py_ssize_t nargs,
172                      PyObject *keywords)
173 {
174     assert(nargs >= 0);
175     assert(nargs == 0 || args != NULL);
176     assert(keywords == NULL || PyTuple_Check(keywords) || PyDict_Check(keywords));
177 
178     /* Slow path: build a temporary tuple for positional arguments and a
179      * temporary dictionary for keyword arguments (if any) */
180     ternaryfunc call = Py_TYPE(callable)->tp_call;
181     if (call == NULL) {
182         _PyErr_Format(tstate, PyExc_TypeError,
183                       "'%.200s' object is not callable",
184                       Py_TYPE(callable)->tp_name);
185         return NULL;
186     }
187 
188     PyObject *argstuple = _PyTuple_FromArray(args, nargs);
189     if (argstuple == NULL) {
190         return NULL;
191     }
192 
193     PyObject *kwdict;
194     if (keywords == NULL || PyDict_Check(keywords)) {
195         kwdict = keywords;
196     }
197     else {
198         if (PyTuple_GET_SIZE(keywords)) {
199             assert(args != NULL);
200             kwdict = _PyStack_AsDict(args + nargs, keywords);
201             if (kwdict == NULL) {
202                 Py_DECREF(argstuple);
203                 return NULL;
204             }
205         }
206         else {
207             keywords = kwdict = NULL;
208         }
209     }
210 
211     PyObject *result = NULL;
212     if (_Py_EnterRecursiveCallTstate(tstate, " while calling a Python object") == 0)
213     {
214         result = _PyCFunctionWithKeywords_TrampolineCall(
215             (PyCFunctionWithKeywords)call, callable, argstuple, kwdict);
216         _Py_LeaveRecursiveCallTstate(tstate);
217     }
218 
219     Py_DECREF(argstuple);
220     if (kwdict != keywords) {
221         Py_DECREF(kwdict);
222     }
223 
224     return _Py_CheckFunctionResult(tstate, callable, result, NULL);
225 }
226 
227 
228 vectorcallfunc
PyVectorcall_Function(PyObject * callable)229 PyVectorcall_Function(PyObject *callable)
230 {
231     return _PyVectorcall_FunctionInline(callable);
232 }
233 
234 
235 static PyObject *
_PyVectorcall_Call(PyThreadState * tstate,vectorcallfunc func,PyObject * callable,PyObject * tuple,PyObject * kwargs)236 _PyVectorcall_Call(PyThreadState *tstate, vectorcallfunc func,
237                    PyObject *callable, PyObject *tuple, PyObject *kwargs)
238 {
239     assert(func != NULL);
240 
241     Py_ssize_t nargs = PyTuple_GET_SIZE(tuple);
242 
243     /* Fast path for no keywords */
244     if (kwargs == NULL || PyDict_GET_SIZE(kwargs) == 0) {
245         return func(callable, _PyTuple_ITEMS(tuple), nargs, NULL);
246     }
247 
248     /* Convert arguments & call */
249     PyObject *const *args;
250     PyObject *kwnames;
251     args = _PyStack_UnpackDict(tstate,
252                                _PyTuple_ITEMS(tuple), nargs,
253                                kwargs, &kwnames);
254     if (args == NULL) {
255         return NULL;
256     }
257     PyObject *result = func(callable, args,
258                             nargs | PY_VECTORCALL_ARGUMENTS_OFFSET, kwnames);
259     _PyStack_UnpackDict_Free(args, nargs, kwnames);
260 
261     return _Py_CheckFunctionResult(tstate, callable, result, NULL);
262 }
263 
264 
265 PyObject *
PyVectorcall_Call(PyObject * callable,PyObject * tuple,PyObject * kwargs)266 PyVectorcall_Call(PyObject *callable, PyObject *tuple, PyObject *kwargs)
267 {
268     PyThreadState *tstate = _PyThreadState_GET();
269 
270     /* get vectorcallfunc as in _PyVectorcall_Function, but without
271      * the Py_TPFLAGS_HAVE_VECTORCALL check */
272     Py_ssize_t offset = Py_TYPE(callable)->tp_vectorcall_offset;
273     if (offset <= 0) {
274         _PyErr_Format(tstate, PyExc_TypeError,
275                       "'%.200s' object does not support vectorcall",
276                       Py_TYPE(callable)->tp_name);
277         return NULL;
278     }
279     assert(PyCallable_Check(callable));
280 
281     vectorcallfunc func;
282     memcpy(&func, (char *) callable + offset, sizeof(func));
283     if (func == NULL) {
284         _PyErr_Format(tstate, PyExc_TypeError,
285                       "'%.200s' object does not support vectorcall",
286                       Py_TYPE(callable)->tp_name);
287         return NULL;
288     }
289 
290     return _PyVectorcall_Call(tstate, func, callable, tuple, kwargs);
291 }
292 
293 
294 PyObject *
PyObject_Vectorcall(PyObject * callable,PyObject * const * args,size_t nargsf,PyObject * kwnames)295 PyObject_Vectorcall(PyObject *callable, PyObject *const *args,
296                      size_t nargsf, PyObject *kwnames)
297 {
298     PyThreadState *tstate = _PyThreadState_GET();
299     return _PyObject_VectorcallTstate(tstate, callable,
300                                       args, nargsf, kwnames);
301 }
302 
303 
304 PyObject *
_PyObject_FastCall(PyObject * func,PyObject * const * args,Py_ssize_t nargs)305 _PyObject_FastCall(PyObject *func, PyObject *const *args, Py_ssize_t nargs)
306 {
307     PyThreadState *tstate = _PyThreadState_GET();
308     return _PyObject_FastCallTstate(tstate, func, args, nargs);
309 }
310 
311 
312 PyObject *
_PyObject_Call(PyThreadState * tstate,PyObject * callable,PyObject * args,PyObject * kwargs)313 _PyObject_Call(PyThreadState *tstate, PyObject *callable,
314                PyObject *args, PyObject *kwargs)
315 {
316     ternaryfunc call;
317     PyObject *result;
318 
319     /* PyObject_Call() must not be called with an exception set,
320        because it can clear it (directly or indirectly) and so the
321        caller loses its exception */
322     assert(!_PyErr_Occurred(tstate));
323     assert(PyTuple_Check(args));
324     assert(kwargs == NULL || PyDict_Check(kwargs));
325 
326     vectorcallfunc vector_func = _PyVectorcall_Function(callable);
327     if (vector_func != NULL) {
328         return _PyVectorcall_Call(tstate, vector_func, callable, args, kwargs);
329     }
330     else {
331         call = Py_TYPE(callable)->tp_call;
332         if (call == NULL) {
333             _PyErr_Format(tstate, PyExc_TypeError,
334                           "'%.200s' object is not callable",
335                           Py_TYPE(callable)->tp_name);
336             return NULL;
337         }
338 
339         if (_Py_EnterRecursiveCallTstate(tstate, " while calling a Python object")) {
340             return NULL;
341         }
342 
343         result = (*call)(callable, args, kwargs);
344 
345         _Py_LeaveRecursiveCallTstate(tstate);
346 
347         return _Py_CheckFunctionResult(tstate, callable, result, NULL);
348     }
349 }
350 
351 PyObject *
PyObject_Call(PyObject * callable,PyObject * args,PyObject * kwargs)352 PyObject_Call(PyObject *callable, PyObject *args, PyObject *kwargs)
353 {
354     PyThreadState *tstate = _PyThreadState_GET();
355     return _PyObject_Call(tstate, callable, args, kwargs);
356 }
357 
358 
359 PyObject *
PyCFunction_Call(PyObject * callable,PyObject * args,PyObject * kwargs)360 PyCFunction_Call(PyObject *callable, PyObject *args, PyObject *kwargs)
361 {
362     PyThreadState *tstate = _PyThreadState_GET();
363     return _PyObject_Call(tstate, callable, args, kwargs);
364 }
365 
366 
367 PyObject *
PyObject_CallOneArg(PyObject * func,PyObject * arg)368 PyObject_CallOneArg(PyObject *func, PyObject *arg)
369 {
370     assert(arg != NULL);
371     PyObject *_args[2];
372     PyObject **args = _args + 1;  // For PY_VECTORCALL_ARGUMENTS_OFFSET
373     args[0] = arg;
374     PyThreadState *tstate = _PyThreadState_GET();
375     size_t nargsf = 1 | PY_VECTORCALL_ARGUMENTS_OFFSET;
376     return _PyObject_VectorcallTstate(tstate, func, args, nargsf, NULL);
377 }
378 
379 
380 /* --- PyFunction call functions ---------------------------------- */
381 
382 PyObject *
_PyFunction_Vectorcall(PyObject * func,PyObject * const * stack,size_t nargsf,PyObject * kwnames)383 _PyFunction_Vectorcall(PyObject *func, PyObject* const* stack,
384                        size_t nargsf, PyObject *kwnames)
385 {
386     assert(PyFunction_Check(func));
387     PyFunctionObject *f = (PyFunctionObject *)func;
388     Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
389     assert(nargs >= 0);
390     PyThreadState *tstate = _PyThreadState_GET();
391     assert(nargs == 0 || stack != NULL);
392     if (((PyCodeObject *)f->func_code)->co_flags & CO_OPTIMIZED) {
393         return _PyEval_Vector(tstate, f, NULL, stack, nargs, kwnames);
394     }
395     else {
396         return _PyEval_Vector(tstate, f, f->func_globals, stack, nargs, kwnames);
397     }
398 }
399 
400 /* --- More complex call functions -------------------------------- */
401 
402 /* External interface to call any callable object.
403    The args must be a tuple or NULL.  The kwargs must be a dict or NULL. */
404 PyObject *
PyEval_CallObjectWithKeywords(PyObject * callable,PyObject * args,PyObject * kwargs)405 PyEval_CallObjectWithKeywords(PyObject *callable,
406                               PyObject *args, PyObject *kwargs)
407 {
408     PyThreadState *tstate = _PyThreadState_GET();
409 #ifdef Py_DEBUG
410     /* PyEval_CallObjectWithKeywords() must not be called with an exception
411        set. It raises a new exception if parameters are invalid or if
412        PyTuple_New() fails, and so the original exception is lost. */
413     assert(!_PyErr_Occurred(tstate));
414 #endif
415 
416     if (args != NULL && !PyTuple_Check(args)) {
417         _PyErr_SetString(tstate, PyExc_TypeError,
418                          "argument list must be a tuple");
419         return NULL;
420     }
421 
422     if (kwargs != NULL && !PyDict_Check(kwargs)) {
423         _PyErr_SetString(tstate, PyExc_TypeError,
424                          "keyword list must be a dictionary");
425         return NULL;
426     }
427 
428     if (args == NULL) {
429         return _PyObject_FastCallDictTstate(tstate, callable, NULL, 0, kwargs);
430     }
431     else {
432         return _PyObject_Call(tstate, callable, args, kwargs);
433     }
434 }
435 
436 
437 PyObject *
PyObject_CallObject(PyObject * callable,PyObject * args)438 PyObject_CallObject(PyObject *callable, PyObject *args)
439 {
440     PyThreadState *tstate = _PyThreadState_GET();
441     assert(!_PyErr_Occurred(tstate));
442     if (args == NULL) {
443         return _PyObject_CallNoArgsTstate(tstate, callable);
444     }
445     if (!PyTuple_Check(args)) {
446         _PyErr_SetString(tstate, PyExc_TypeError,
447                          "argument list must be a tuple");
448         return NULL;
449     }
450     return _PyObject_Call(tstate, callable, args, NULL);
451 }
452 
453 
454 /* Call callable(obj, *args, **kwargs). */
455 PyObject *
_PyObject_Call_Prepend(PyThreadState * tstate,PyObject * callable,PyObject * obj,PyObject * args,PyObject * kwargs)456 _PyObject_Call_Prepend(PyThreadState *tstate, PyObject *callable,
457                        PyObject *obj, PyObject *args, PyObject *kwargs)
458 {
459     assert(PyTuple_Check(args));
460 
461     PyObject *small_stack[_PY_FASTCALL_SMALL_STACK];
462     PyObject **stack;
463 
464     Py_ssize_t argcount = PyTuple_GET_SIZE(args);
465     if (argcount + 1 <= (Py_ssize_t)Py_ARRAY_LENGTH(small_stack)) {
466         stack = small_stack;
467     }
468     else {
469         stack = PyMem_Malloc((argcount + 1) * sizeof(PyObject *));
470         if (stack == NULL) {
471             PyErr_NoMemory();
472             return NULL;
473         }
474     }
475 
476     /* use borrowed references */
477     stack[0] = obj;
478     memcpy(&stack[1],
479            _PyTuple_ITEMS(args),
480            argcount * sizeof(PyObject *));
481 
482     PyObject *result = _PyObject_FastCallDictTstate(tstate, callable,
483                                                     stack, argcount + 1,
484                                                     kwargs);
485     if (stack != small_stack) {
486         PyMem_Free(stack);
487     }
488     return result;
489 }
490 
491 
492 /* --- Call with a format string ---------------------------------- */
493 
494 static PyObject *
_PyObject_CallFunctionVa(PyThreadState * tstate,PyObject * callable,const char * format,va_list va,int is_size_t)495 _PyObject_CallFunctionVa(PyThreadState *tstate, PyObject *callable,
496                          const char *format, va_list va, int is_size_t)
497 {
498     PyObject* small_stack[_PY_FASTCALL_SMALL_STACK];
499     const Py_ssize_t small_stack_len = Py_ARRAY_LENGTH(small_stack);
500     PyObject **stack;
501     Py_ssize_t nargs, i;
502     PyObject *result;
503 
504     if (callable == NULL) {
505         return null_error(tstate);
506     }
507 
508     if (!format || !*format) {
509         return _PyObject_CallNoArgsTstate(tstate, callable);
510     }
511 
512     if (is_size_t) {
513         stack = _Py_VaBuildStack_SizeT(small_stack, small_stack_len,
514                                        format, va, &nargs);
515     }
516     else {
517         stack = _Py_VaBuildStack(small_stack, small_stack_len,
518                                  format, va, &nargs);
519     }
520     if (stack == NULL) {
521         return NULL;
522     }
523 
524     if (nargs == 1 && PyTuple_Check(stack[0])) {
525         /* Special cases for backward compatibility:
526            - PyObject_CallFunction(func, "O", tuple) calls func(*tuple)
527            - PyObject_CallFunction(func, "(OOO)", arg1, arg2, arg3) calls
528              func(*(arg1, arg2, arg3)): func(arg1, arg2, arg3) */
529         PyObject *args = stack[0];
530         result = _PyObject_VectorcallTstate(tstate, callable,
531                                             _PyTuple_ITEMS(args),
532                                             PyTuple_GET_SIZE(args),
533                                             NULL);
534     }
535     else {
536         result = _PyObject_VectorcallTstate(tstate, callable,
537                                             stack, nargs, NULL);
538     }
539 
540     for (i = 0; i < nargs; ++i) {
541         Py_DECREF(stack[i]);
542     }
543     if (stack != small_stack) {
544         PyMem_Free(stack);
545     }
546     return result;
547 }
548 
549 
550 PyObject *
PyObject_CallFunction(PyObject * callable,const char * format,...)551 PyObject_CallFunction(PyObject *callable, const char *format, ...)
552 {
553     va_list va;
554     PyObject *result;
555     PyThreadState *tstate = _PyThreadState_GET();
556 
557     va_start(va, format);
558     result = _PyObject_CallFunctionVa(tstate, callable, format, va, 0);
559     va_end(va);
560 
561     return result;
562 }
563 
564 
565 /* PyEval_CallFunction is exact copy of PyObject_CallFunction.
566  * This function is kept for backward compatibility.
567  */
568 PyObject *
PyEval_CallFunction(PyObject * callable,const char * format,...)569 PyEval_CallFunction(PyObject *callable, const char *format, ...)
570 {
571     va_list va;
572     PyObject *result;
573     PyThreadState *tstate = _PyThreadState_GET();
574 
575     va_start(va, format);
576     result = _PyObject_CallFunctionVa(tstate, callable, format, va, 0);
577     va_end(va);
578 
579     return result;
580 }
581 
582 
583 PyObject *
_PyObject_CallFunction_SizeT(PyObject * callable,const char * format,...)584 _PyObject_CallFunction_SizeT(PyObject *callable, const char *format, ...)
585 {
586     PyThreadState *tstate = _PyThreadState_GET();
587 
588     va_list va;
589     va_start(va, format);
590     PyObject *result = _PyObject_CallFunctionVa(tstate, callable, format, va, 1);
591     va_end(va);
592 
593     return result;
594 }
595 
596 
597 static PyObject*
callmethod(PyThreadState * tstate,PyObject * callable,const char * format,va_list va,int is_size_t)598 callmethod(PyThreadState *tstate, PyObject* callable, const char *format, va_list va, int is_size_t)
599 {
600     assert(callable != NULL);
601     if (!PyCallable_Check(callable)) {
602         _PyErr_Format(tstate, PyExc_TypeError,
603                       "attribute of type '%.200s' is not callable",
604                       Py_TYPE(callable)->tp_name);
605         return NULL;
606     }
607 
608     return _PyObject_CallFunctionVa(tstate, callable, format, va, is_size_t);
609 }
610 
611 PyObject *
PyObject_CallMethod(PyObject * obj,const char * name,const char * format,...)612 PyObject_CallMethod(PyObject *obj, const char *name, const char *format, ...)
613 {
614     PyThreadState *tstate = _PyThreadState_GET();
615 
616     if (obj == NULL || name == NULL) {
617         return null_error(tstate);
618     }
619 
620     PyObject *callable = PyObject_GetAttrString(obj, name);
621     if (callable == NULL) {
622         return NULL;
623     }
624 
625     va_list va;
626     va_start(va, format);
627     PyObject *retval = callmethod(tstate, callable, format, va, 0);
628     va_end(va);
629 
630     Py_DECREF(callable);
631     return retval;
632 }
633 
634 
635 /* PyEval_CallMethod is exact copy of PyObject_CallMethod.
636  * This function is kept for backward compatibility.
637  */
638 PyObject *
PyEval_CallMethod(PyObject * obj,const char * name,const char * format,...)639 PyEval_CallMethod(PyObject *obj, const char *name, const char *format, ...)
640 {
641     PyThreadState *tstate = _PyThreadState_GET();
642     if (obj == NULL || name == NULL) {
643         return null_error(tstate);
644     }
645 
646     PyObject *callable = PyObject_GetAttrString(obj, name);
647     if (callable == NULL) {
648         return NULL;
649     }
650 
651     va_list va;
652     va_start(va, format);
653     PyObject *retval = callmethod(tstate, callable, format, va, 0);
654     va_end(va);
655 
656     Py_DECREF(callable);
657     return retval;
658 }
659 
660 
661 PyObject *
_PyObject_CallMethod(PyObject * obj,PyObject * name,const char * format,...)662 _PyObject_CallMethod(PyObject *obj, PyObject *name,
663                      const char *format, ...)
664 {
665     PyThreadState *tstate = _PyThreadState_GET();
666     if (obj == NULL || name == NULL) {
667         return null_error(tstate);
668     }
669 
670     PyObject *callable = PyObject_GetAttr(obj, name);
671     if (callable == NULL) {
672         return NULL;
673     }
674 
675     va_list va;
676     va_start(va, format);
677     PyObject *retval = callmethod(tstate, callable, format, va, 1);
678     va_end(va);
679 
680     Py_DECREF(callable);
681     return retval;
682 }
683 
684 
685 PyObject *
_PyObject_CallMethodId(PyObject * obj,_Py_Identifier * name,const char * format,...)686 _PyObject_CallMethodId(PyObject *obj, _Py_Identifier *name,
687                        const char *format, ...)
688 {
689     PyThreadState *tstate = _PyThreadState_GET();
690     if (obj == NULL || name == NULL) {
691         return null_error(tstate);
692     }
693 
694     PyObject *callable = _PyObject_GetAttrId(obj, name);
695     if (callable == NULL) {
696         return NULL;
697     }
698 
699     va_list va;
700     va_start(va, format);
701     PyObject *retval = callmethod(tstate, callable, format, va, 0);
702     va_end(va);
703 
704     Py_DECREF(callable);
705     return retval;
706 }
707 
708 
_PyObject_CallMethodFormat(PyThreadState * tstate,PyObject * callable,const char * format,...)709 PyObject * _PyObject_CallMethodFormat(PyThreadState *tstate, PyObject *callable,
710                                       const char *format, ...)
711 {
712     va_list va;
713     va_start(va, format);
714     PyObject *retval = callmethod(tstate, callable, format, va, 0);
715     va_end(va);
716     return retval;
717 }
718 
719 
720 PyObject *
_PyObject_CallMethod_SizeT(PyObject * obj,const char * name,const char * format,...)721 _PyObject_CallMethod_SizeT(PyObject *obj, const char *name,
722                            const char *format, ...)
723 {
724     PyThreadState *tstate = _PyThreadState_GET();
725     if (obj == NULL || name == NULL) {
726         return null_error(tstate);
727     }
728 
729     PyObject *callable = PyObject_GetAttrString(obj, name);
730     if (callable == NULL) {
731         return NULL;
732     }
733 
734     va_list va;
735     va_start(va, format);
736     PyObject *retval = callmethod(tstate, callable, format, va, 1);
737     va_end(va);
738 
739     Py_DECREF(callable);
740     return retval;
741 }
742 
743 
744 PyObject *
_PyObject_CallMethodId_SizeT(PyObject * obj,_Py_Identifier * name,const char * format,...)745 _PyObject_CallMethodId_SizeT(PyObject *obj, _Py_Identifier *name,
746                              const char *format, ...)
747 {
748     PyThreadState *tstate = _PyThreadState_GET();
749     if (obj == NULL || name == NULL) {
750         return null_error(tstate);
751     }
752 
753     PyObject *callable = _PyObject_GetAttrId(obj, name);
754     if (callable == NULL) {
755         return NULL;
756     }
757 
758     va_list va;
759     va_start(va, format);
760     PyObject *retval = callmethod(tstate, callable, format, va, 1);
761     va_end(va);
762 
763     Py_DECREF(callable);
764     return retval;
765 }
766 
767 
768 /* --- Call with "..." arguments ---------------------------------- */
769 
770 static PyObject *
object_vacall(PyThreadState * tstate,PyObject * base,PyObject * callable,va_list vargs)771 object_vacall(PyThreadState *tstate, PyObject *base,
772               PyObject *callable, va_list vargs)
773 {
774     PyObject *small_stack[_PY_FASTCALL_SMALL_STACK];
775     PyObject **stack;
776     Py_ssize_t nargs;
777     PyObject *result;
778     Py_ssize_t i;
779     va_list countva;
780 
781     if (callable == NULL) {
782         return null_error(tstate);
783     }
784 
785     /* Count the number of arguments */
786     va_copy(countva, vargs);
787     nargs = base ? 1 : 0;
788     while (1) {
789         PyObject *arg = va_arg(countva, PyObject *);
790         if (arg == NULL) {
791             break;
792         }
793         nargs++;
794     }
795     va_end(countva);
796 
797     /* Copy arguments */
798     if (nargs <= (Py_ssize_t)Py_ARRAY_LENGTH(small_stack)) {
799         stack = small_stack;
800     }
801     else {
802         stack = PyMem_Malloc(nargs * sizeof(stack[0]));
803         if (stack == NULL) {
804             PyErr_NoMemory();
805             return NULL;
806         }
807     }
808 
809     i = 0;
810     if (base) {
811         stack[i++] = base;
812     }
813 
814     for (; i < nargs; ++i) {
815         stack[i] = va_arg(vargs, PyObject *);
816     }
817 
818     /* Call the function */
819     result = _PyObject_VectorcallTstate(tstate, callable, stack, nargs, NULL);
820 
821     if (stack != small_stack) {
822         PyMem_Free(stack);
823     }
824     return result;
825 }
826 
827 
828 PyObject *
PyObject_VectorcallMethod(PyObject * name,PyObject * const * args,size_t nargsf,PyObject * kwnames)829 PyObject_VectorcallMethod(PyObject *name, PyObject *const *args,
830                            size_t nargsf, PyObject *kwnames)
831 {
832     assert(name != NULL);
833     assert(args != NULL);
834     assert(PyVectorcall_NARGS(nargsf) >= 1);
835 
836     PyThreadState *tstate = _PyThreadState_GET();
837     PyObject *callable = NULL;
838     /* Use args[0] as "self" argument */
839     int unbound = _PyObject_GetMethod(args[0], name, &callable);
840     if (callable == NULL) {
841         return NULL;
842     }
843 
844     if (unbound) {
845         /* We must remove PY_VECTORCALL_ARGUMENTS_OFFSET since
846          * that would be interpreted as allowing to change args[-1] */
847         nargsf &= ~PY_VECTORCALL_ARGUMENTS_OFFSET;
848     }
849     else {
850         /* Skip "self". We can keep PY_VECTORCALL_ARGUMENTS_OFFSET since
851          * args[-1] in the onward call is args[0] here. */
852         args++;
853         nargsf--;
854     }
855     PyObject *result = _PyObject_VectorcallTstate(tstate, callable,
856                                                   args, nargsf, kwnames);
857     Py_DECREF(callable);
858     return result;
859 }
860 
861 
862 PyObject *
PyObject_CallMethodObjArgs(PyObject * obj,PyObject * name,...)863 PyObject_CallMethodObjArgs(PyObject *obj, PyObject *name, ...)
864 {
865     PyThreadState *tstate = _PyThreadState_GET();
866     if (obj == NULL || name == NULL) {
867         return null_error(tstate);
868     }
869 
870     PyObject *callable = NULL;
871     int is_method = _PyObject_GetMethod(obj, name, &callable);
872     if (callable == NULL) {
873         return NULL;
874     }
875     obj = is_method ? obj : NULL;
876 
877     va_list vargs;
878     va_start(vargs, name);
879     PyObject *result = object_vacall(tstate, obj, callable, vargs);
880     va_end(vargs);
881 
882     Py_DECREF(callable);
883     return result;
884 }
885 
886 
887 PyObject *
_PyObject_CallMethodIdObjArgs(PyObject * obj,_Py_Identifier * name,...)888 _PyObject_CallMethodIdObjArgs(PyObject *obj, _Py_Identifier *name, ...)
889 {
890     PyThreadState *tstate = _PyThreadState_GET();
891     if (obj == NULL || name == NULL) {
892         return null_error(tstate);
893     }
894 
895     PyObject *oname = _PyUnicode_FromId(name); /* borrowed */
896     if (!oname) {
897         return NULL;
898     }
899 
900     PyObject *callable = NULL;
901     int is_method = _PyObject_GetMethod(obj, oname, &callable);
902     if (callable == NULL) {
903         return NULL;
904     }
905     obj = is_method ? obj : NULL;
906 
907     va_list vargs;
908     va_start(vargs, name);
909     PyObject *result = object_vacall(tstate, obj, callable, vargs);
910     va_end(vargs);
911 
912     Py_DECREF(callable);
913     return result;
914 }
915 
916 
917 PyObject *
PyObject_CallFunctionObjArgs(PyObject * callable,...)918 PyObject_CallFunctionObjArgs(PyObject *callable, ...)
919 {
920     PyThreadState *tstate = _PyThreadState_GET();
921     va_list vargs;
922     PyObject *result;
923 
924     va_start(vargs, callable);
925     result = object_vacall(tstate, NULL, callable, vargs);
926     va_end(vargs);
927 
928     return result;
929 }
930 
931 
932 /* --- PyStack functions ------------------------------------------ */
933 
934 PyObject *
_PyStack_AsDict(PyObject * const * values,PyObject * kwnames)935 _PyStack_AsDict(PyObject *const *values, PyObject *kwnames)
936 {
937     Py_ssize_t nkwargs;
938 
939     assert(kwnames != NULL);
940     nkwargs = PyTuple_GET_SIZE(kwnames);
941     return _PyDict_FromItems(&PyTuple_GET_ITEM(kwnames, 0), 1,
942                              values, 1, nkwargs);
943 }
944 
945 
946 /* Convert (args, nargs, kwargs: dict) into a (stack, nargs, kwnames: tuple).
947 
948    Allocate a new argument vector and keyword names tuple. Return the argument
949    vector; return NULL with exception set on error. Return the keyword names
950    tuple in *p_kwnames.
951 
952    This also checks that all keyword names are strings. If not, a TypeError is
953    raised.
954 
955    The newly allocated argument vector supports PY_VECTORCALL_ARGUMENTS_OFFSET.
956 
957    When done, you must call _PyStack_UnpackDict_Free(stack, nargs, kwnames) */
958 static PyObject *const *
_PyStack_UnpackDict(PyThreadState * tstate,PyObject * const * args,Py_ssize_t nargs,PyObject * kwargs,PyObject ** p_kwnames)959 _PyStack_UnpackDict(PyThreadState *tstate,
960                     PyObject *const *args, Py_ssize_t nargs,
961                     PyObject *kwargs, PyObject **p_kwnames)
962 {
963     assert(nargs >= 0);
964     assert(kwargs != NULL);
965     assert(PyDict_Check(kwargs));
966 
967     Py_ssize_t nkwargs = PyDict_GET_SIZE(kwargs);
968     /* Check for overflow in the PyMem_Malloc() call below. The subtraction
969      * in this check cannot overflow: both maxnargs and nkwargs are
970      * non-negative signed integers, so their difference fits in the type. */
971     Py_ssize_t maxnargs = PY_SSIZE_T_MAX / sizeof(args[0]) - 1;
972     if (nargs > maxnargs - nkwargs) {
973         _PyErr_NoMemory(tstate);
974         return NULL;
975     }
976 
977     /* Add 1 to support PY_VECTORCALL_ARGUMENTS_OFFSET */
978     PyObject **stack = PyMem_Malloc((1 + nargs + nkwargs) * sizeof(args[0]));
979     if (stack == NULL) {
980         _PyErr_NoMemory(tstate);
981         return NULL;
982     }
983 
984     PyObject *kwnames = PyTuple_New(nkwargs);
985     if (kwnames == NULL) {
986         PyMem_Free(stack);
987         return NULL;
988     }
989 
990     stack++;  /* For PY_VECTORCALL_ARGUMENTS_OFFSET */
991 
992     /* Copy positional arguments */
993     for (Py_ssize_t i = 0; i < nargs; i++) {
994         Py_INCREF(args[i]);
995         stack[i] = args[i];
996     }
997 
998     PyObject **kwstack = stack + nargs;
999     /* This loop doesn't support lookup function mutating the dictionary
1000        to change its size. It's a deliberate choice for speed, this function is
1001        called in the performance critical hot code. */
1002     Py_ssize_t pos = 0, i = 0;
1003     PyObject *key, *value;
1004     unsigned long keys_are_strings = Py_TPFLAGS_UNICODE_SUBCLASS;
1005     while (PyDict_Next(kwargs, &pos, &key, &value)) {
1006         keys_are_strings &= Py_TYPE(key)->tp_flags;
1007         Py_INCREF(key);
1008         Py_INCREF(value);
1009         PyTuple_SET_ITEM(kwnames, i, key);
1010         kwstack[i] = value;
1011         i++;
1012     }
1013 
1014     /* keys_are_strings has the value Py_TPFLAGS_UNICODE_SUBCLASS if that
1015      * flag is set for all keys. Otherwise, keys_are_strings equals 0.
1016      * We do this check once at the end instead of inside the loop above
1017      * because it simplifies the deallocation in the failing case.
1018      * It happens to also make the loop above slightly more efficient. */
1019     if (!keys_are_strings) {
1020         _PyErr_SetString(tstate, PyExc_TypeError,
1021                          "keywords must be strings");
1022         _PyStack_UnpackDict_Free(stack, nargs, kwnames);
1023         return NULL;
1024     }
1025 
1026     *p_kwnames = kwnames;
1027     return stack;
1028 }
1029 
1030 static void
_PyStack_UnpackDict_Free(PyObject * const * stack,Py_ssize_t nargs,PyObject * kwnames)1031 _PyStack_UnpackDict_Free(PyObject *const *stack, Py_ssize_t nargs,
1032                          PyObject *kwnames)
1033 {
1034     Py_ssize_t n = PyTuple_GET_SIZE(kwnames) + nargs;
1035     for (Py_ssize_t i = 0; i < n; i++) {
1036         Py_DECREF(stack[i]);
1037     }
1038     PyMem_Free((PyObject **)stack - 1);
1039     Py_DECREF(kwnames);
1040 }
1041