blob: 8d980fdc93905583ea48e90c8622b8fa6d02b164 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum3f5da241990-12-20 15:06:42 +00002/* Execute compiled code */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003
Guido van Rossum681d79a1995-07-18 14:51:37 +00004/* XXX TO DO:
Guido van Rossum681d79a1995-07-18 14:51:37 +00005 XXX speed up searching for keywords by using a dictionary
Guido van Rossum681d79a1995-07-18 14:51:37 +00006 XXX document it!
7 */
8
Thomas Wouters477c8d52006-05-27 19:21:47 +00009/* enable more aggressive intra-module optimizations, where available */
10#define PY_LOCAL_AGGRESSIVE
11
Guido van Rossumb209a111997-04-29 18:18:01 +000012#include "Python.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000013
Jeremy Hylton3e0055f2005-10-20 19:59:25 +000014#include "code.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000015#include "frameobject.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000016#include "opcode.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000017#include "structmember.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000018
Guido van Rossumc6004111993-11-05 10:22:19 +000019#include <ctype.h>
20
Thomas Wouters477c8d52006-05-27 19:21:47 +000021#ifndef WITH_TSC
Michael W. Hudson75eabd22005-01-18 15:56:11 +000022
23#define READ_TIMESTAMP(var)
24
25#else
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000026
27typedef unsigned long long uint64;
28
Ezio Melotti13925002011-03-16 11:05:33 +020029/* PowerPC support.
David Malcolmf1397ad2011-01-06 17:01:36 +000030 "__ppc__" appears to be the preprocessor definition to detect on OS X, whereas
31 "__powerpc__" appears to be the correct one for Linux with GCC
32*/
33#if defined(__ppc__) || defined (__powerpc__)
Michael W. Hudson800ba232004-08-12 18:19:17 +000034
Michael W. Hudson75eabd22005-01-18 15:56:11 +000035#define READ_TIMESTAMP(var) ppc_getcounter(&var)
Michael W. Hudson800ba232004-08-12 18:19:17 +000036
37static void
38ppc_getcounter(uint64 *v)
39{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000040 register unsigned long tbu, tb, tbu2;
Michael W. Hudson800ba232004-08-12 18:19:17 +000041
42 loop:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000043 asm volatile ("mftbu %0" : "=r" (tbu) );
44 asm volatile ("mftb %0" : "=r" (tb) );
45 asm volatile ("mftbu %0" : "=r" (tbu2));
46 if (__builtin_expect(tbu != tbu2, 0)) goto loop;
Michael W. Hudson800ba232004-08-12 18:19:17 +000047
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000048 /* The slightly peculiar way of writing the next lines is
49 compiled better by GCC than any other way I tried. */
50 ((long*)(v))[0] = tbu;
51 ((long*)(v))[1] = tb;
Michael W. Hudson800ba232004-08-12 18:19:17 +000052}
53
Mark Dickinsona25b1312009-10-31 10:18:44 +000054#elif defined(__i386__)
55
56/* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000057
Michael W. Hudson75eabd22005-01-18 15:56:11 +000058#define READ_TIMESTAMP(val) \
59 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000060
Mark Dickinsona25b1312009-10-31 10:18:44 +000061#elif defined(__x86_64__)
62
63/* for gcc/x86_64, the "A" constraint in DI mode means *either* rax *or* rdx;
64 not edx:eax as it does for i386. Since rdtsc puts its result in edx:eax
65 even in 64-bit mode, we need to use "a" and "d" for the lower and upper
66 32-bit pieces of the result. */
67
68#define READ_TIMESTAMP(val) \
69 __asm__ __volatile__("rdtsc" : \
70 "=a" (((int*)&(val))[0]), "=d" (((int*)&(val))[1]));
71
72
73#else
74
75#error "Don't know how to implement timestamp counter for this architecture"
76
Michael W. Hudson800ba232004-08-12 18:19:17 +000077#endif
78
Thomas Wouters477c8d52006-05-27 19:21:47 +000079void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000080 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000081{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082 uint64 intr, inst, loop;
83 PyThreadState *tstate = PyThreadState_Get();
84 if (!tstate->interp->tscdump)
85 return;
86 intr = intr1 - intr0;
87 inst = inst1 - inst0 - intr;
88 loop = loop1 - loop0 - intr;
89 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
Stefan Krahb7e10102010-06-23 18:42:39 +000090 opcode, ticked, inst, loop);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091}
Michael W. Hudson800ba232004-08-12 18:19:17 +000092
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
94
Guido van Rossum04691fc1992-08-12 15:35:34 +000095/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000096/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000097
Guido van Rossum408027e1996-12-30 16:17:54 +000098#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000099/* For debugging the interpreter: */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000100#define LLTRACE 1 /* Low-level trace feature */
101#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000102#endif
103
Jeremy Hylton52820442001-01-03 23:52:36 +0000104typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +0000105
Guido van Rossum374a9221991-04-04 10:40:29 +0000106/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000107#ifdef WITH_TSC
Thomas Wouters477c8d52006-05-27 19:21:47 +0000108static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000109#else
Thomas Wouters477c8d52006-05-27 19:21:47 +0000110static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000111#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +0000112static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
113static PyObject * do_call(PyObject *, PyObject ***, int, int);
114static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000115static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000116 PyObject *);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000117static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
118static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000119#define CALL_FLAG_VAR 1
120#define CALL_FLAG_KW 2
121
Guido van Rossum0a066c01992-03-27 17:29:15 +0000122#ifdef LLTRACE
Guido van Rossumc2e20742006-02-27 22:32:47 +0000123static int lltrace;
Tim Petersdbd9ba62000-07-09 03:09:57 +0000124static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000125#endif
Fred Drake5755ce62001-06-27 19:19:46 +0000126static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000127 int, PyObject *);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000128static int call_trace_protected(Py_tracefunc, PyObject *,
Stefan Krahb7e10102010-06-23 18:42:39 +0000129 PyFrameObject *, int, PyObject *);
Fred Drake5755ce62001-06-27 19:19:46 +0000130static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
Tim Peters8a5c3c72004-04-05 19:36:21 +0000131static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Stefan Krahb7e10102010-06-23 18:42:39 +0000132 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000133
Thomas Wouters477c8d52006-05-27 19:21:47 +0000134static PyObject * cmp_outcome(int, PyObject *, PyObject *);
135static PyObject * import_from(PyObject *, PyObject *);
Thomas Wouters52152252000-08-17 22:55:00 +0000136static int import_all_from(PyObject *, PyObject *);
Neal Norwitzda059e32007-08-26 05:33:45 +0000137static void format_exc_check_arg(PyObject *, const char *, PyObject *);
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +0000138static void format_exc_unbound(PyCodeObject *co, int oparg);
Victor Stinnerd2a915d2011-10-02 20:34:20 +0200139static PyObject * unicode_concatenate(PyObject *, PyObject *,
140 PyFrameObject *, unsigned char *);
Benjamin Peterson876b2f22009-06-28 03:18:59 +0000141static PyObject * special_lookup(PyObject *, char *, PyObject **);
Guido van Rossum374a9221991-04-04 10:40:29 +0000142
Paul Prescode68140d2000-08-30 20:25:01 +0000143#define NAME_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000144 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000145#define GLOBAL_NAME_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000146 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000147#define UNBOUNDLOCAL_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000148 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000149#define UNBOUNDFREE_ERROR_MSG \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000150 "free variable '%.200s' referenced before assignment" \
151 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000152
Guido van Rossum950361c1997-01-24 13:49:28 +0000153/* Dynamic execution profile */
154#ifdef DYNAMIC_EXECUTION_PROFILE
155#ifdef DXPAIRS
156static long dxpairs[257][256];
157#define dxp dxpairs[256]
158#else
159static long dxp[256];
160#endif
161#endif
162
Jeremy Hylton985eba52003-02-05 23:13:00 +0000163/* Function call profile */
164#ifdef CALL_PROFILE
165#define PCALL_NUM 11
166static int pcall[PCALL_NUM];
167
168#define PCALL_ALL 0
169#define PCALL_FUNCTION 1
170#define PCALL_FAST_FUNCTION 2
171#define PCALL_FASTER_FUNCTION 3
172#define PCALL_METHOD 4
173#define PCALL_BOUND_METHOD 5
174#define PCALL_CFUNCTION 6
175#define PCALL_TYPE 7
176#define PCALL_GENERATOR 8
177#define PCALL_OTHER 9
178#define PCALL_POP 10
179
180/* Notes about the statistics
181
182 PCALL_FAST stats
183
184 FAST_FUNCTION means no argument tuple needs to be created.
185 FASTER_FUNCTION means that the fast-path frame setup code is used.
186
187 If there is a method call where the call can be optimized by changing
188 the argument tuple and calling the function directly, it gets recorded
189 twice.
190
191 As a result, the relationship among the statistics appears to be
192 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
193 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
194 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
195 PCALL_METHOD > PCALL_BOUND_METHOD
196*/
197
198#define PCALL(POS) pcall[POS]++
199
200PyObject *
201PyEval_GetCallStats(PyObject *self)
202{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000203 return Py_BuildValue("iiiiiiiiiii",
204 pcall[0], pcall[1], pcall[2], pcall[3],
205 pcall[4], pcall[5], pcall[6], pcall[7],
206 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000207}
208#else
209#define PCALL(O)
210
211PyObject *
212PyEval_GetCallStats(PyObject *self)
213{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000214 Py_INCREF(Py_None);
215 return Py_None;
Jeremy Hylton985eba52003-02-05 23:13:00 +0000216}
217#endif
218
Tim Peters5ca576e2001-06-18 22:08:13 +0000219
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000220#ifdef WITH_THREAD
221#define GIL_REQUEST _Py_atomic_load_relaxed(&gil_drop_request)
222#else
223#define GIL_REQUEST 0
224#endif
225
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000226/* This can set eval_breaker to 0 even though gil_drop_request became
227 1. We believe this is all right because the eval loop will release
228 the GIL eventually anyway. */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000229#define COMPUTE_EVAL_BREAKER() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000230 _Py_atomic_store_relaxed( \
231 &eval_breaker, \
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000232 GIL_REQUEST | \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000233 _Py_atomic_load_relaxed(&pendingcalls_to_do) | \
234 pending_async_exc)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000235
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000236#ifdef WITH_THREAD
237
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000238#define SET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000239 do { \
240 _Py_atomic_store_relaxed(&gil_drop_request, 1); \
241 _Py_atomic_store_relaxed(&eval_breaker, 1); \
242 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000243
244#define RESET_GIL_DROP_REQUEST() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000245 do { \
246 _Py_atomic_store_relaxed(&gil_drop_request, 0); \
247 COMPUTE_EVAL_BREAKER(); \
248 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000249
Benjamin Petersond2be5b42010-09-10 22:47:02 +0000250#endif
251
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000252/* Pending calls are only modified under pending_lock */
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000253#define SIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000254 do { \
255 _Py_atomic_store_relaxed(&pendingcalls_to_do, 1); \
256 _Py_atomic_store_relaxed(&eval_breaker, 1); \
257 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000258
259#define UNSIGNAL_PENDING_CALLS() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000260 do { \
261 _Py_atomic_store_relaxed(&pendingcalls_to_do, 0); \
262 COMPUTE_EVAL_BREAKER(); \
263 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000264
265#define SIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000266 do { \
267 pending_async_exc = 1; \
268 _Py_atomic_store_relaxed(&eval_breaker, 1); \
269 } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000270
271#define UNSIGNAL_ASYNC_EXC() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000272 do { pending_async_exc = 0; COMPUTE_EVAL_BREAKER(); } while (0)
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000273
274
Guido van Rossume59214e1994-08-30 08:01:59 +0000275#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000276
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000278#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000279#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000280#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000281
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000282static PyThread_type_lock pending_lock = 0; /* for pending calls */
Guido van Rossuma9672091994-09-14 13:31:22 +0000283static long main_thread = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000284/* This single variable consolidates all requests to break out of the fast path
285 in the eval loop. */
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000286static _Py_atomic_int eval_breaker = {0};
287/* Request for dropping the GIL */
288static _Py_atomic_int gil_drop_request = {0};
289/* Request for running pending calls. */
290static _Py_atomic_int pendingcalls_to_do = {0};
291/* Request for looking at the `async_exc` field of the current thread state.
292 Guarded by the GIL. */
293static int pending_async_exc = 0;
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000294
295#include "ceval_gil.h"
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000296
Tim Peters7f468f22004-10-11 02:40:51 +0000297int
298PyEval_ThreadsInitialized(void)
299{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000300 return gil_created();
Tim Peters7f468f22004-10-11 02:40:51 +0000301}
302
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000303void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000304PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000305{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000306 if (gil_created())
307 return;
308 create_gil();
309 take_gil(PyThreadState_GET());
310 main_thread = PyThread_get_thread_ident();
311 if (!pending_lock)
312 pending_lock = PyThread_allocate_lock();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000313}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000314
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000315void
Antoine Pitrou1df15362010-09-13 14:16:46 +0000316_PyEval_FiniThreads(void)
317{
318 if (!gil_created())
319 return;
320 destroy_gil();
321 assert(!gil_created());
322}
323
324void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000325PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000326{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000327 PyThreadState *tstate = PyThreadState_GET();
328 if (tstate == NULL)
329 Py_FatalError("PyEval_AcquireLock: current thread state is NULL");
330 take_gil(tstate);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000331}
332
333void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000334PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000335{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000336 /* This function must succeed when the current thread state is NULL.
337 We therefore avoid PyThreadState_GET() which dumps a fatal error
338 in debug mode.
339 */
340 drop_gil((PyThreadState*)_Py_atomic_load_relaxed(
341 &_PyThreadState_Current));
Guido van Rossum25ce5661997-08-02 03:10:38 +0000342}
343
344void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000345PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000346{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000347 if (tstate == NULL)
348 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
349 /* Check someone has called PyEval_InitThreads() to create the lock */
350 assert(gil_created());
351 take_gil(tstate);
352 if (PyThreadState_Swap(tstate) != NULL)
353 Py_FatalError(
354 "PyEval_AcquireThread: non-NULL old thread state");
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000355}
356
357void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000358PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000359{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000360 if (tstate == NULL)
361 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
362 if (PyThreadState_Swap(NULL) != tstate)
363 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
364 drop_gil(tstate);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000365}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000366
367/* This function is called from PyOS_AfterFork to ensure that newly
368 created child processes don't hold locks referring to threads which
369 are not running in the child process. (This could also be done using
370 pthread_atfork mechanism, at least for the pthreads implementation.) */
371
372void
373PyEval_ReInitThreads(void)
374{
Martin v. Löwisbd928fe2011-10-14 10:20:37 +0200375 _Py_IDENTIFIER(_after_fork);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000376 PyObject *threading, *result;
377 PyThreadState *tstate = PyThreadState_GET();
Jesse Nollera8513972008-07-17 16:49:17 +0000378
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000379 if (!gil_created())
380 return;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000381 recreate_gil();
382 pending_lock = PyThread_allocate_lock();
383 take_gil(tstate);
384 main_thread = PyThread_get_thread_ident();
Jesse Nollera8513972008-07-17 16:49:17 +0000385
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000386 /* Update the threading module with the new state.
387 */
388 tstate = PyThreadState_GET();
389 threading = PyMapping_GetItemString(tstate->interp->modules,
390 "threading");
391 if (threading == NULL) {
392 /* threading not imported */
393 PyErr_Clear();
394 return;
395 }
Martin v. Löwisafe55bb2011-10-09 10:38:36 +0200396 result = _PyObject_CallMethodId(threading, &PyId__after_fork, NULL);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000397 if (result == NULL)
398 PyErr_WriteUnraisable(threading);
399 else
400 Py_DECREF(result);
401 Py_DECREF(threading);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000402}
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000403
404#else
Jeffrey Yasskin39370832010-05-03 19:29:34 +0000405static _Py_atomic_int eval_breaker = {0};
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000406static int pending_async_exc = 0;
407#endif /* WITH_THREAD */
408
409/* This function is used to signal that async exceptions are waiting to be
410 raised, therefore it is also useful in non-threaded builds. */
411
412void
413_PyEval_SignalAsyncExc(void)
414{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 SIGNAL_ASYNC_EXC();
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000416}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000417
Guido van Rossumff4949e1992-08-05 19:58:53 +0000418/* Functions save_thread and restore_thread are always defined so
419 dynamically loaded modules needn't be compiled separately for use
420 with and without threads: */
421
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000422PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000423PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000424{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000425 PyThreadState *tstate = PyThreadState_Swap(NULL);
426 if (tstate == NULL)
427 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000428#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000429 if (gil_created())
430 drop_gil(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000431#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000432 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000433}
434
435void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000436PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000437{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000438 if (tstate == NULL)
439 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000440#ifdef WITH_THREAD
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000441 if (gil_created()) {
442 int err = errno;
443 take_gil(tstate);
Antoine Pitrou0d5e52d2011-05-04 20:02:30 +0200444 /* _Py_Finalizing is protected by the GIL */
445 if (_Py_Finalizing && tstate != _Py_Finalizing) {
446 drop_gil(tstate);
447 PyThread_exit_thread();
448 assert(0); /* unreachable */
449 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000450 errno = err;
451 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000452#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000453 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000454}
455
456
Guido van Rossuma9672091994-09-14 13:31:22 +0000457/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
458 signal handlers or Mac I/O completion routines) can schedule calls
459 to a function to be called synchronously.
460 The synchronous function is called with one void* argument.
461 It should return 0 for success or -1 for failure -- failure should
462 be accompanied by an exception.
463
464 If registry succeeds, the registry function returns 0; if it fails
465 (e.g. due to too many pending calls) it returns -1 (without setting
466 an exception condition).
467
468 Note that because registry may occur from within signal handlers,
469 or other asynchronous events, calling malloc() is unsafe!
470
471#ifdef WITH_THREAD
472 Any thread can schedule pending calls, but only the main thread
473 will execute them.
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000474 There is no facility to schedule calls to a particular thread, but
475 that should be easy to change, should that ever be required. In
476 that case, the static variables here should go into the python
477 threadstate.
Guido van Rossuma9672091994-09-14 13:31:22 +0000478#endif
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000479*/
Guido van Rossuma9672091994-09-14 13:31:22 +0000480
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000481#ifdef WITH_THREAD
482
483/* The WITH_THREAD implementation is thread-safe. It allows
484 scheduling to be made from any thread, and even from an executing
485 callback.
486 */
487
488#define NPENDINGCALLS 32
489static struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000490 int (*func)(void *);
491 void *arg;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000492} pendingcalls[NPENDINGCALLS];
493static int pendingfirst = 0;
494static int pendinglast = 0;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000495
496int
497Py_AddPendingCall(int (*func)(void *), void *arg)
498{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000499 int i, j, result=0;
500 PyThread_type_lock lock = pending_lock;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000501
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000502 /* try a few times for the lock. Since this mechanism is used
503 * for signal handling (on the main thread), there is a (slim)
504 * chance that a signal is delivered on the same thread while we
505 * hold the lock during the Py_MakePendingCalls() function.
506 * This avoids a deadlock in that case.
507 * Note that signals can be delivered on any thread. In particular,
508 * on Windows, a SIGINT is delivered on a system-created worker
509 * thread.
510 * We also check for lock being NULL, in the unlikely case that
511 * this function is called before any bytecode evaluation takes place.
512 */
513 if (lock != NULL) {
514 for (i = 0; i<100; i++) {
515 if (PyThread_acquire_lock(lock, NOWAIT_LOCK))
516 break;
517 }
518 if (i == 100)
519 return -1;
520 }
521
522 i = pendinglast;
523 j = (i + 1) % NPENDINGCALLS;
524 if (j == pendingfirst) {
525 result = -1; /* Queue full */
526 } else {
527 pendingcalls[i].func = func;
528 pendingcalls[i].arg = arg;
529 pendinglast = j;
530 }
531 /* signal main loop */
532 SIGNAL_PENDING_CALLS();
533 if (lock != NULL)
534 PyThread_release_lock(lock);
535 return result;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000536}
537
538int
539Py_MakePendingCalls(void)
540{
Charles-François Natalif23339a2011-07-23 18:15:43 +0200541 static int busy = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000542 int i;
543 int r = 0;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000544
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 if (!pending_lock) {
546 /* initial allocation of the lock */
547 pending_lock = PyThread_allocate_lock();
548 if (pending_lock == NULL)
549 return -1;
550 }
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000551
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000552 /* only service pending calls on main thread */
553 if (main_thread && PyThread_get_thread_ident() != main_thread)
554 return 0;
555 /* don't perform recursive pending calls */
Charles-François Natalif23339a2011-07-23 18:15:43 +0200556 if (busy)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 return 0;
Charles-François Natalif23339a2011-07-23 18:15:43 +0200558 busy = 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000559 /* perform a bounded number of calls, in case of recursion */
560 for (i=0; i<NPENDINGCALLS; i++) {
561 int j;
562 int (*func)(void *);
563 void *arg = NULL;
564
565 /* pop one item off the queue while holding the lock */
566 PyThread_acquire_lock(pending_lock, WAIT_LOCK);
567 j = pendingfirst;
568 if (j == pendinglast) {
569 func = NULL; /* Queue empty */
570 } else {
571 func = pendingcalls[j].func;
572 arg = pendingcalls[j].arg;
573 pendingfirst = (j + 1) % NPENDINGCALLS;
574 }
575 if (pendingfirst != pendinglast)
576 SIGNAL_PENDING_CALLS();
577 else
578 UNSIGNAL_PENDING_CALLS();
579 PyThread_release_lock(pending_lock);
580 /* having released the lock, perform the callback */
581 if (func == NULL)
582 break;
583 r = func(arg);
584 if (r)
585 break;
586 }
Charles-François Natalif23339a2011-07-23 18:15:43 +0200587 busy = 0;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000588 return r;
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000589}
590
591#else /* if ! defined WITH_THREAD */
592
593/*
594 WARNING! ASYNCHRONOUSLY EXECUTING CODE!
595 This code is used for signal handling in python that isn't built
596 with WITH_THREAD.
597 Don't use this implementation when Py_AddPendingCalls() can happen
598 on a different thread!
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000599
Guido van Rossuma9672091994-09-14 13:31:22 +0000600 There are two possible race conditions:
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000601 (1) nested asynchronous calls to Py_AddPendingCall()
602 (2) AddPendingCall() calls made while pending calls are being processed.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000603
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000604 (1) is very unlikely because typically signal delivery
605 is blocked during signal handling. So it should be impossible.
606 (2) is a real possibility.
Guido van Rossuma9672091994-09-14 13:31:22 +0000607 The current code is safe against (2), but not against (1).
608 The safety against (2) is derived from the fact that only one
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000609 thread is present, interrupted by signals, and that the critical
610 section is protected with the "busy" variable. On Windows, which
611 delivers SIGINT on a system thread, this does not hold and therefore
612 Windows really shouldn't use this version.
613 The two threads could theoretically wiggle around the "busy" variable.
Guido van Rossuma027efa1997-05-05 20:56:21 +0000614*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000615
Guido van Rossuma9672091994-09-14 13:31:22 +0000616#define NPENDINGCALLS 32
617static struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000618 int (*func)(void *);
619 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000620} pendingcalls[NPENDINGCALLS];
621static volatile int pendingfirst = 0;
622static volatile int pendinglast = 0;
Benjamin Peterson08ec84c2010-05-30 14:49:32 +0000623static _Py_atomic_int pendingcalls_to_do = {0};
Guido van Rossuma9672091994-09-14 13:31:22 +0000624
625int
Thomas Wouters334fb892000-07-25 12:56:38 +0000626Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000627{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000628 static volatile int busy = 0;
629 int i, j;
630 /* XXX Begin critical section */
631 if (busy)
632 return -1;
633 busy = 1;
634 i = pendinglast;
635 j = (i + 1) % NPENDINGCALLS;
636 if (j == pendingfirst) {
637 busy = 0;
638 return -1; /* Queue full */
639 }
640 pendingcalls[i].func = func;
641 pendingcalls[i].arg = arg;
642 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000643
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000644 SIGNAL_PENDING_CALLS();
645 busy = 0;
646 /* XXX End critical section */
647 return 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000648}
649
Guido van Rossum180d7b41994-09-29 09:45:57 +0000650int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000651Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000652{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000653 static int busy = 0;
654 if (busy)
655 return 0;
656 busy = 1;
657 UNSIGNAL_PENDING_CALLS();
658 for (;;) {
659 int i;
660 int (*func)(void *);
661 void *arg;
662 i = pendingfirst;
663 if (i == pendinglast)
664 break; /* Queue empty */
665 func = pendingcalls[i].func;
666 arg = pendingcalls[i].arg;
667 pendingfirst = (i + 1) % NPENDINGCALLS;
668 if (func(arg) < 0) {
669 busy = 0;
670 SIGNAL_PENDING_CALLS(); /* We're not done yet */
671 return -1;
672 }
673 }
674 busy = 0;
675 return 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000676}
677
Benjamin Petersone5bf3832009-01-17 23:43:58 +0000678#endif /* WITH_THREAD */
679
Guido van Rossuma9672091994-09-14 13:31:22 +0000680
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000681/* The interpreter's recursion limit */
682
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000683#ifndef Py_DEFAULT_RECURSION_LIMIT
684#define Py_DEFAULT_RECURSION_LIMIT 1000
685#endif
686static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
687int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000688
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000689int
690Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000691{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000692 return recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000693}
694
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000695void
696Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000697{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000698 recursion_limit = new_limit;
699 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000700}
701
Armin Rigo2b3eb402003-10-28 12:05:48 +0000702/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
703 if the recursion_depth reaches _Py_CheckRecursionLimit.
704 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
705 to guarantee that _Py_CheckRecursiveCall() is regularly called.
706 Without USE_STACKCHECK, there is no need for this. */
707int
708_Py_CheckRecursiveCall(char *where)
709{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000710 PyThreadState *tstate = PyThreadState_GET();
Armin Rigo2b3eb402003-10-28 12:05:48 +0000711
712#ifdef USE_STACKCHECK
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000713 if (PyOS_CheckStack()) {
714 --tstate->recursion_depth;
715 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
716 return -1;
717 }
Armin Rigo2b3eb402003-10-28 12:05:48 +0000718#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719 _Py_CheckRecursionLimit = recursion_limit;
720 if (tstate->recursion_critical)
721 /* Somebody asked that we don't check for recursion. */
722 return 0;
723 if (tstate->overflowed) {
724 if (tstate->recursion_depth > recursion_limit + 50) {
725 /* Overflowing while handling an overflow. Give up. */
726 Py_FatalError("Cannot recover from stack overflow.");
727 }
728 return 0;
729 }
730 if (tstate->recursion_depth > recursion_limit) {
731 --tstate->recursion_depth;
732 tstate->overflowed = 1;
733 PyErr_Format(PyExc_RuntimeError,
734 "maximum recursion depth exceeded%s",
735 where);
736 return -1;
737 }
738 return 0;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000739}
740
Guido van Rossum374a9221991-04-04 10:40:29 +0000741/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000742enum why_code {
Stefan Krahb7e10102010-06-23 18:42:39 +0000743 WHY_NOT = 0x0001, /* No error */
744 WHY_EXCEPTION = 0x0002, /* Exception occurred */
745 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
746 WHY_RETURN = 0x0008, /* 'return' statement */
747 WHY_BREAK = 0x0010, /* 'break' statement */
748 WHY_CONTINUE = 0x0020, /* 'continue' statement */
749 WHY_YIELD = 0x0040, /* 'yield' operator */
750 WHY_SILENCED = 0x0080 /* Exception silenced by 'with' */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000751};
Guido van Rossum374a9221991-04-04 10:40:29 +0000752
Benjamin Peterson87880242011-07-03 16:48:31 -0500753static void save_exc_state(PyThreadState *, PyFrameObject *);
754static void swap_exc_state(PyThreadState *, PyFrameObject *);
755static void restore_and_clear_exc_state(PyThreadState *, PyFrameObject *);
Collin Winter828f04a2007-08-31 00:04:24 +0000756static enum why_code do_raise(PyObject *, PyObject *);
Guido van Rossum0368b722007-05-11 16:50:42 +0000757static int unpack_iterable(PyObject *, int, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000758
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +0000759/* Records whether tracing is on for any thread. Counts the number of
760 threads for which tstate->c_tracefunc is non-NULL, so if the value
761 is 0, we know we don't have to check this thread's c_tracefunc.
762 This speeds up the if statement in PyEval_EvalFrameEx() after
763 fast_next_opcode*/
764static int _Py_TracingPossible = 0;
765
Antoine Pitrou074e5ed2009-11-10 19:50:40 +0000766
Guido van Rossum374a9221991-04-04 10:40:29 +0000767
Guido van Rossumb209a111997-04-29 18:18:01 +0000768PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +0000769PyEval_EvalCode(PyObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000770{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000771 return PyEval_EvalCodeEx(co,
772 globals, locals,
773 (PyObject **)NULL, 0,
774 (PyObject **)NULL, 0,
775 (PyObject **)NULL, 0,
776 NULL, NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000777}
778
779
780/* Interpreter main loop */
781
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000782PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000783PyEval_EvalFrame(PyFrameObject *f) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000784 /* This is for backward compatibility with extension modules that
785 used this API; core interpreter code should call
786 PyEval_EvalFrameEx() */
787 return PyEval_EvalFrameEx(f, 0);
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000788}
789
790PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000791PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000792{
Guido van Rossum950361c1997-01-24 13:49:28 +0000793#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000794 int lastopcode = 0;
Guido van Rossum950361c1997-01-24 13:49:28 +0000795#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000796 register PyObject **stack_pointer; /* Next free slot in value stack */
797 register unsigned char *next_instr;
798 register int opcode; /* Current opcode */
799 register int oparg; /* Current opcode argument, if any */
800 register enum why_code why; /* Reason for block stack unwind */
801 register int err; /* Error status -- nonzero if error */
802 register PyObject *x; /* Result object -- NULL if error */
803 register PyObject *v; /* Temporary objects popped off stack */
804 register PyObject *w;
805 register PyObject *u;
806 register PyObject *t;
807 register PyObject **fastlocals, **freevars;
808 PyObject *retval = NULL; /* Return value */
809 PyThreadState *tstate = PyThreadState_GET();
810 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000811
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000812 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000813
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000814 not (instr_lb <= current_bytecode_offset < instr_ub)
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000815
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 is true when the line being executed has changed. The
817 initial values are such as to make this false the first
818 time it is tested. */
819 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000821 unsigned char *first_instr;
822 PyObject *names;
823 PyObject *consts;
Guido van Rossum374a9221991-04-04 10:40:29 +0000824
Antoine Pitroub52ec782009-01-25 16:34:23 +0000825/* Computed GOTOs, or
826 the-optimization-commonly-but-improperly-known-as-"threaded code"
827 using gcc's labels-as-values extension
828 (http://gcc.gnu.org/onlinedocs/gcc/Labels-as-Values.html).
829
830 The traditional bytecode evaluation loop uses a "switch" statement, which
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000831 decent compilers will optimize as a single indirect branch instruction
Antoine Pitroub52ec782009-01-25 16:34:23 +0000832 combined with a lookup table of jump addresses. However, since the
833 indirect jump instruction is shared by all opcodes, the CPU will have a
834 hard time making the right prediction for where to jump next (actually,
835 it will be always wrong except in the uncommon case of a sequence of
836 several identical opcodes).
837
838 "Threaded code" in contrast, uses an explicit jump table and an explicit
839 indirect jump instruction at the end of each opcode. Since the jump
840 instruction is at a different address for each opcode, the CPU will make a
841 separate prediction for each of these instructions, which is equivalent to
842 predicting the second opcode of each opcode pair. These predictions have
843 a much better chance to turn out valid, especially in small bytecode loops.
844
845 A mispredicted branch on a modern CPU flushes the whole pipeline and
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000846 can cost several CPU cycles (depending on the pipeline depth),
Antoine Pitroub52ec782009-01-25 16:34:23 +0000847 and potentially many more instructions (depending on the pipeline width).
848 A correctly predicted branch, however, is nearly free.
849
850 At the time of this writing, the "threaded code" version is up to 15-20%
851 faster than the normal "switch" version, depending on the compiler and the
852 CPU architecture.
853
854 We disable the optimization if DYNAMIC_EXECUTION_PROFILE is defined,
855 because it would render the measurements invalid.
856
857
858 NOTE: care must be taken that the compiler doesn't try to "optimize" the
859 indirect jumps by sharing them between all opcodes. Such optimizations
860 can be disabled on gcc by using the -fno-gcse flag (or possibly
861 -fno-crossjumping).
862*/
863
Antoine Pitrou042b1282010-08-13 21:15:58 +0000864#ifdef DYNAMIC_EXECUTION_PROFILE
Antoine Pitroub52ec782009-01-25 16:34:23 +0000865#undef USE_COMPUTED_GOTOS
Antoine Pitrou042b1282010-08-13 21:15:58 +0000866#define USE_COMPUTED_GOTOS 0
Antoine Pitroub52ec782009-01-25 16:34:23 +0000867#endif
868
Antoine Pitrou042b1282010-08-13 21:15:58 +0000869#ifdef HAVE_COMPUTED_GOTOS
870 #ifndef USE_COMPUTED_GOTOS
871 #define USE_COMPUTED_GOTOS 1
872 #endif
873#else
874 #if defined(USE_COMPUTED_GOTOS) && USE_COMPUTED_GOTOS
875 #error "Computed gotos are not supported on this compiler."
876 #endif
877 #undef USE_COMPUTED_GOTOS
878 #define USE_COMPUTED_GOTOS 0
879#endif
880
881#if USE_COMPUTED_GOTOS
Antoine Pitroub52ec782009-01-25 16:34:23 +0000882/* Import the static jump table */
883#include "opcode_targets.h"
884
885/* This macro is used when several opcodes defer to the same implementation
886 (e.g. SETUP_LOOP, SETUP_FINALLY) */
887#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000888 TARGET_##op: \
889 opcode = op; \
890 if (HAS_ARG(op)) \
891 oparg = NEXTARG(); \
892 case op: \
893 goto impl; \
Antoine Pitroub52ec782009-01-25 16:34:23 +0000894
895#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000896 TARGET_##op: \
897 opcode = op; \
898 if (HAS_ARG(op)) \
899 oparg = NEXTARG(); \
900 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000901
902
903#define DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000904 { \
905 if (!_Py_atomic_load_relaxed(&eval_breaker)) { \
906 FAST_DISPATCH(); \
907 } \
908 continue; \
909 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000910
911#ifdef LLTRACE
912#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000913 { \
914 if (!lltrace && !_Py_TracingPossible) { \
915 f->f_lasti = INSTR_OFFSET(); \
916 goto *opcode_targets[*next_instr++]; \
917 } \
918 goto fast_next_opcode; \
919 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000920#else
921#define FAST_DISPATCH() \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000922 { \
923 if (!_Py_TracingPossible) { \
924 f->f_lasti = INSTR_OFFSET(); \
925 goto *opcode_targets[*next_instr++]; \
926 } \
927 goto fast_next_opcode; \
928 }
Antoine Pitroub52ec782009-01-25 16:34:23 +0000929#endif
930
931#else
932#define TARGET(op) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000933 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000934#define TARGET_WITH_IMPL(op, impl) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 /* silence compiler warnings about `impl` unused */ \
936 if (0) goto impl; \
937 case op:
Antoine Pitroub52ec782009-01-25 16:34:23 +0000938#define DISPATCH() continue
939#define FAST_DISPATCH() goto fast_next_opcode
940#endif
941
942
Neal Norwitza81d2202002-07-14 00:27:26 +0000943/* Tuple access macros */
944
945#ifndef Py_DEBUG
946#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
947#else
948#define GETITEM(v, i) PyTuple_GetItem((v), (i))
949#endif
950
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000951#ifdef WITH_TSC
952/* Use Pentium timestamp counter to mark certain events:
953 inst0 -- beginning of switch statement for opcode dispatch
954 inst1 -- end of switch statement (may be skipped)
955 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000956 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000957 (may be skipped)
958 intr1 -- beginning of long interruption
959 intr2 -- end of long interruption
960
961 Many opcodes call out to helper C functions. In some cases, the
962 time in those functions should be counted towards the time for the
963 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
964 calls another Python function; there's no point in charge all the
965 bytecode executed by the called function to the caller.
966
967 It's hard to make a useful judgement statically. In the presence
968 of operator overloading, it's impossible to tell if a call will
969 execute new Python code or not.
970
971 It's a case-by-case judgement. I'll use intr1 for the following
972 cases:
973
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000974 IMPORT_STAR
975 IMPORT_FROM
976 CALL_FUNCTION (and friends)
977
978 */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
980 int ticked = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000981
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000982 READ_TIMESTAMP(inst0);
983 READ_TIMESTAMP(inst1);
984 READ_TIMESTAMP(loop0);
985 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000986
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000987 /* shut up the compiler */
988 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000989#endif
990
Guido van Rossum374a9221991-04-04 10:40:29 +0000991/* Code access macros */
992
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000993#define INSTR_OFFSET() ((int)(next_instr - first_instr))
994#define NEXTOP() (*next_instr++)
995#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
996#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
997#define JUMPTO(x) (next_instr = first_instr + (x))
998#define JUMPBY(x) (next_instr += (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000999
Raymond Hettingerf606f872003-03-16 03:11:04 +00001000/* OpCode prediction macros
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001001 Some opcodes tend to come in pairs thus making it possible to
1002 predict the second code when the first is run. For example,
1003 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
1004 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001005
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001006 Verifying the prediction costs a single high-speed test of a register
1007 variable against a constant. If the pairing was good, then the
1008 processor's own internal branch predication has a high likelihood of
1009 success, resulting in a nearly zero-overhead transition to the
1010 next opcode. A successful prediction saves a trip through the eval-loop
1011 including its two unpredictable branches, the HAS_ARG test and the
1012 switch-case. Combined with the processor's internal branch prediction,
1013 a successful PREDICT has the effect of making the two opcodes run as if
1014 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +00001015
Georg Brandl86b2fb92008-07-16 03:43:04 +00001016 If collecting opcode statistics, your choices are to either keep the
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001017 predictions turned-on and interpret the results as if some opcodes
1018 had been combined or turn-off predictions so that the opcode frequency
1019 counter updates for both opcodes.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001020
1021 Opcode prediction is disabled with threaded code, since the latter allows
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001022 the CPU to record separate branch prediction information for each
1023 opcode.
Antoine Pitroub52ec782009-01-25 16:34:23 +00001024
Raymond Hettingerf606f872003-03-16 03:11:04 +00001025*/
1026
Antoine Pitrou042b1282010-08-13 21:15:58 +00001027#if defined(DYNAMIC_EXECUTION_PROFILE) || USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001028#define PREDICT(op) if (0) goto PRED_##op
1029#define PREDICTED(op) PRED_##op:
1030#define PREDICTED_WITH_ARG(op) PRED_##op:
Raymond Hettingera7216982004-02-08 19:59:27 +00001031#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001032#define PREDICT(op) if (*next_instr == op) goto PRED_##op
1033#define PREDICTED(op) PRED_##op: next_instr++
1034#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Antoine Pitroub52ec782009-01-25 16:34:23 +00001035#endif
1036
Raymond Hettingerf606f872003-03-16 03:11:04 +00001037
Guido van Rossum374a9221991-04-04 10:40:29 +00001038/* Stack manipulation macros */
1039
Martin v. Löwis18e16552006-02-15 17:27:45 +00001040/* The stack can grow at most MAXINT deep, as co_nlocals and
1041 co_stacksize are ints. */
Stefan Krahb7e10102010-06-23 18:42:39 +00001042#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
1043#define EMPTY() (STACK_LEVEL() == 0)
1044#define TOP() (stack_pointer[-1])
1045#define SECOND() (stack_pointer[-2])
1046#define THIRD() (stack_pointer[-3])
1047#define FOURTH() (stack_pointer[-4])
1048#define PEEK(n) (stack_pointer[-(n)])
1049#define SET_TOP(v) (stack_pointer[-1] = (v))
1050#define SET_SECOND(v) (stack_pointer[-2] = (v))
1051#define SET_THIRD(v) (stack_pointer[-3] = (v))
1052#define SET_FOURTH(v) (stack_pointer[-4] = (v))
1053#define SET_VALUE(n, v) (stack_pointer[-(n)] = (v))
1054#define BASIC_STACKADJ(n) (stack_pointer += n)
1055#define BASIC_PUSH(v) (*stack_pointer++ = (v))
1056#define BASIC_POP() (*--stack_pointer)
Guido van Rossum374a9221991-04-04 10:40:29 +00001057
Guido van Rossum96a42c81992-01-12 02:29:51 +00001058#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001059#define PUSH(v) { (void)(BASIC_PUSH(v), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001060 lltrace && prtrace(TOP(), "push")); \
1061 assert(STACK_LEVEL() <= co->co_stacksize); }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001062#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001063 BASIC_POP())
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001064#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
Stefan Krahb7e10102010-06-23 18:42:39 +00001065 lltrace && prtrace(TOP(), "stackadj")); \
1066 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes0449f632007-12-15 01:27:15 +00001067#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
Stefan Krahb7e10102010-06-23 18:42:39 +00001068 prtrace((STACK_POINTER)[-1], "ext_pop")), \
1069 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001070#else
Stefan Krahb7e10102010-06-23 18:42:39 +00001071#define PUSH(v) BASIC_PUSH(v)
1072#define POP() BASIC_POP()
1073#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +00001074#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +00001075#endif
1076
Guido van Rossum681d79a1995-07-18 14:51:37 +00001077/* Local variable macros */
1078
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001079#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +00001080
1081/* The SETLOCAL() macro must not DECREF the local variable in-place and
1082 then store the new value; it must copy the old value to a temporary
1083 value, then store the new value, and then DECREF the temporary value.
1084 This is because it is possible that during the DECREF the frame is
1085 accessed by other code (e.g. a __del__ method or gc.collect()) and the
1086 variable would be pointing to already-freed memory. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001087#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
Stefan Krahb7e10102010-06-23 18:42:39 +00001088 GETLOCAL(i) = value; \
1089 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +00001090
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001091
1092#define UNWIND_BLOCK(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001093 while (STACK_LEVEL() > (b)->b_level) { \
1094 PyObject *v = POP(); \
1095 Py_XDECREF(v); \
1096 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001097
1098#define UNWIND_EXCEPT_HANDLER(b) \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001099 { \
1100 PyObject *type, *value, *traceback; \
1101 assert(STACK_LEVEL() >= (b)->b_level + 3); \
1102 while (STACK_LEVEL() > (b)->b_level + 3) { \
1103 value = POP(); \
1104 Py_XDECREF(value); \
1105 } \
1106 type = tstate->exc_type; \
1107 value = tstate->exc_value; \
1108 traceback = tstate->exc_traceback; \
1109 tstate->exc_type = POP(); \
1110 tstate->exc_value = POP(); \
1111 tstate->exc_traceback = POP(); \
1112 Py_XDECREF(type); \
1113 Py_XDECREF(value); \
1114 Py_XDECREF(traceback); \
1115 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001116
Guido van Rossuma027efa1997-05-05 20:56:21 +00001117/* Start of code */
1118
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001119 /* push frame */
1120 if (Py_EnterRecursiveCall(""))
1121 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +00001122
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +00001124
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001125 if (tstate->use_tracing) {
1126 if (tstate->c_tracefunc != NULL) {
1127 /* tstate->c_tracefunc, if defined, is a
1128 function that will be called on *every* entry
1129 to a code block. Its return value, if not
1130 None, is a function that will be called at
1131 the start of each executed line of code.
1132 (Actually, the function must return itself
1133 in order to continue tracing.) The trace
1134 functions are called with three arguments:
1135 a pointer to the current frame, a string
1136 indicating why the function is called, and
1137 an argument which depends on the situation.
1138 The global trace function is also called
1139 whenever an exception is detected. */
1140 if (call_trace_protected(tstate->c_tracefunc,
1141 tstate->c_traceobj,
1142 f, PyTrace_CALL, Py_None)) {
1143 /* Trace function raised an error */
1144 goto exit_eval_frame;
1145 }
1146 }
1147 if (tstate->c_profilefunc != NULL) {
1148 /* Similar for c_profilefunc, except it needn't
1149 return itself and isn't called for "line" events */
1150 if (call_trace_protected(tstate->c_profilefunc,
1151 tstate->c_profileobj,
1152 f, PyTrace_CALL, Py_None)) {
1153 /* Profile function raised an error */
1154 goto exit_eval_frame;
1155 }
1156 }
1157 }
Neil Schemenauer6c0f2002001-09-04 19:03:35 +00001158
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001159 co = f->f_code;
1160 names = co->co_names;
1161 consts = co->co_consts;
1162 fastlocals = f->f_localsplus;
1163 freevars = f->f_localsplus + co->co_nlocals;
1164 first_instr = (unsigned char*) PyBytes_AS_STRING(co->co_code);
1165 /* An explanation is in order for the next line.
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001166
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001167 f->f_lasti now refers to the index of the last instruction
1168 executed. You might think this was obvious from the name, but
1169 this wasn't always true before 2.3! PyFrame_New now sets
1170 f->f_lasti to -1 (i.e. the index *before* the first instruction)
1171 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
1172 does work. Promise.
Thomas Wouters902d6eb2007-01-09 23:18:33 +00001173
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001174 When the PREDICT() macros are enabled, some opcode pairs follow in
1175 direct succession without updating f->f_lasti. A successful
1176 prediction effectively links the two codes together as if they
1177 were a single new opcode; accordingly,f->f_lasti will point to
1178 the first code in the pair (for instance, GET_ITER followed by
1179 FOR_ITER is effectively a single opcode and f->f_lasti will point
1180 at to the beginning of the combined pair.)
1181 */
1182 next_instr = first_instr + f->f_lasti + 1;
1183 stack_pointer = f->f_stacktop;
1184 assert(stack_pointer != NULL);
1185 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001186
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001187 if (co->co_flags & CO_GENERATOR && !throwflag) {
1188 if (f->f_exc_type != NULL && f->f_exc_type != Py_None) {
1189 /* We were in an except handler when we left,
1190 restore the exception state which was put aside
1191 (see YIELD_VALUE). */
Benjamin Peterson87880242011-07-03 16:48:31 -05001192 swap_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 }
Benjamin Peterson87880242011-07-03 16:48:31 -05001194 else
1195 save_exc_state(tstate, f);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001196 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001197
Tim Peters5ca576e2001-06-18 22:08:13 +00001198#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001199 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00001200#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00001201
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001202 why = WHY_NOT;
1203 err = 0;
1204 x = Py_None; /* Not a reference, just anything non-NULL */
1205 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00001206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001207 if (throwflag) { /* support for generator.throw() */
1208 why = WHY_EXCEPTION;
1209 goto on_error;
1210 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00001211
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001212 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001213#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001214 if (inst1 == 0) {
1215 /* Almost surely, the opcode executed a break
1216 or a continue, preventing inst1 from being set
1217 on the way out of the loop.
1218 */
1219 READ_TIMESTAMP(inst1);
1220 loop1 = inst1;
1221 }
1222 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
1223 intr0, intr1);
1224 ticked = 0;
1225 inst1 = 0;
1226 intr0 = 0;
1227 intr1 = 0;
1228 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001229#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001230 assert(stack_pointer >= f->f_valuestack); /* else underflow */
1231 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001232
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001233 /* Do periodic things. Doing this every time through
1234 the loop would add too much overhead, so we do it
1235 only every Nth instruction. We also do it if
1236 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
1237 event needs attention (e.g. a signal handler or
1238 async I/O handler); see Py_AddPendingCall() and
1239 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +00001240
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001241 if (_Py_atomic_load_relaxed(&eval_breaker)) {
1242 if (*next_instr == SETUP_FINALLY) {
1243 /* Make the last opcode before
Ezio Melotti13925002011-03-16 11:05:33 +02001244 a try: finally: block uninterruptible. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001245 goto fast_next_opcode;
1246 }
1247 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001248#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001249 ticked = 1;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00001250#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 if (_Py_atomic_load_relaxed(&pendingcalls_to_do)) {
1252 if (Py_MakePendingCalls() < 0) {
1253 why = WHY_EXCEPTION;
1254 goto on_error;
1255 }
1256 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001257#ifdef WITH_THREAD
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001258 if (_Py_atomic_load_relaxed(&gil_drop_request)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 /* Give another thread a chance */
1260 if (PyThreadState_Swap(NULL) != tstate)
1261 Py_FatalError("ceval: tstate mix-up");
1262 drop_gil(tstate);
1263
1264 /* Other threads may run now */
1265
1266 take_gil(tstate);
1267 if (PyThreadState_Swap(tstate) != NULL)
1268 Py_FatalError("ceval: orphan tstate");
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001269 }
Benjamin Petersond2be5b42010-09-10 22:47:02 +00001270#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001271 /* Check for asynchronous exceptions. */
1272 if (tstate->async_exc != NULL) {
1273 x = tstate->async_exc;
1274 tstate->async_exc = NULL;
1275 UNSIGNAL_ASYNC_EXC();
1276 PyErr_SetNone(x);
1277 Py_DECREF(x);
1278 why = WHY_EXCEPTION;
1279 goto on_error;
1280 }
1281 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001282
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001283 fast_next_opcode:
1284 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001285
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001286 /* line-by-line tracing support */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001288 if (_Py_TracingPossible &&
1289 tstate->c_tracefunc != NULL && !tstate->tracing) {
1290 /* see maybe_call_line_trace
1291 for expository comments */
1292 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001293
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001294 err = maybe_call_line_trace(tstate->c_tracefunc,
1295 tstate->c_traceobj,
1296 f, &instr_lb, &instr_ub,
1297 &instr_prev);
1298 /* Reload possibly changed frame fields */
1299 JUMPTO(f->f_lasti);
1300 if (f->f_stacktop != NULL) {
1301 stack_pointer = f->f_stacktop;
1302 f->f_stacktop = NULL;
1303 }
1304 if (err) {
1305 /* trace function raised an exception */
1306 goto on_error;
1307 }
1308 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001309
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001310 /* Extract opcode and argument */
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001311
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001312 opcode = NEXTOP();
1313 oparg = 0; /* allows oparg to be stored in a register because
1314 it doesn't have to be remembered across a full loop */
1315 if (HAS_ARG(opcode))
1316 oparg = NEXTARG();
Stefan Krahb7e10102010-06-23 18:42:39 +00001317 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001318#ifdef DYNAMIC_EXECUTION_PROFILE
1319#ifdef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001320 dxpairs[lastopcode][opcode]++;
1321 lastopcode = opcode;
Guido van Rossum950361c1997-01-24 13:49:28 +00001322#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001323 dxp[opcode]++;
Guido van Rossum950361c1997-01-24 13:49:28 +00001324#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001325
Guido van Rossum96a42c81992-01-12 02:29:51 +00001326#ifdef LLTRACE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001327 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001329 if (lltrace) {
1330 if (HAS_ARG(opcode)) {
1331 printf("%d: %d, %d\n",
1332 f->f_lasti, opcode, oparg);
1333 }
1334 else {
1335 printf("%d: %d\n",
1336 f->f_lasti, opcode);
1337 }
1338 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001339#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001340
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001341 /* Main switch on opcode */
1342 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001343
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001344 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001345
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001346 /* BEWARE!
1347 It is essential that any operation that fails sets either
1348 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1349 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001351 TARGET(NOP)
1352 FAST_DISPATCH();
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001354 TARGET(LOAD_FAST)
1355 x = GETLOCAL(oparg);
1356 if (x != NULL) {
1357 Py_INCREF(x);
1358 PUSH(x);
1359 FAST_DISPATCH();
1360 }
1361 format_exc_check_arg(PyExc_UnboundLocalError,
1362 UNBOUNDLOCAL_ERROR_MSG,
1363 PyTuple_GetItem(co->co_varnames, oparg));
1364 break;
Neil Schemenauer63543862002-02-17 19:10:14 +00001365
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001366 TARGET(LOAD_CONST)
1367 x = GETITEM(consts, oparg);
1368 Py_INCREF(x);
1369 PUSH(x);
1370 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001371
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001372 PREDICTED_WITH_ARG(STORE_FAST);
1373 TARGET(STORE_FAST)
1374 v = POP();
1375 SETLOCAL(oparg, v);
1376 FAST_DISPATCH();
Neil Schemenauer63543862002-02-17 19:10:14 +00001377
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001378 TARGET(POP_TOP)
1379 v = POP();
1380 Py_DECREF(v);
1381 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001382
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001383 TARGET(ROT_TWO)
1384 v = TOP();
1385 w = SECOND();
1386 SET_TOP(w);
1387 SET_SECOND(v);
1388 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001389
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001390 TARGET(ROT_THREE)
1391 v = TOP();
1392 w = SECOND();
1393 x = THIRD();
1394 SET_TOP(w);
1395 SET_SECOND(x);
1396 SET_THIRD(v);
1397 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001398
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001399 TARGET(DUP_TOP)
1400 v = TOP();
1401 Py_INCREF(v);
1402 PUSH(v);
1403 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001404
Antoine Pitrou74a69fa2010-09-04 18:43:52 +00001405 TARGET(DUP_TOP_TWO)
1406 x = TOP();
1407 Py_INCREF(x);
1408 w = SECOND();
1409 Py_INCREF(w);
1410 STACKADJ(2);
1411 SET_TOP(x);
1412 SET_SECOND(w);
1413 FAST_DISPATCH();
Thomas Wouters434d0822000-08-24 20:11:32 +00001414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001415 TARGET(UNARY_POSITIVE)
1416 v = TOP();
1417 x = PyNumber_Positive(v);
1418 Py_DECREF(v);
1419 SET_TOP(x);
1420 if (x != NULL) DISPATCH();
1421 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001422
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001423 TARGET(UNARY_NEGATIVE)
1424 v = TOP();
1425 x = PyNumber_Negative(v);
1426 Py_DECREF(v);
1427 SET_TOP(x);
1428 if (x != NULL) DISPATCH();
1429 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001430
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001431 TARGET(UNARY_NOT)
1432 v = TOP();
1433 err = PyObject_IsTrue(v);
1434 Py_DECREF(v);
1435 if (err == 0) {
1436 Py_INCREF(Py_True);
1437 SET_TOP(Py_True);
1438 DISPATCH();
1439 }
1440 else if (err > 0) {
1441 Py_INCREF(Py_False);
1442 SET_TOP(Py_False);
1443 err = 0;
1444 DISPATCH();
1445 }
1446 STACKADJ(-1);
1447 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001448
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001449 TARGET(UNARY_INVERT)
1450 v = TOP();
1451 x = PyNumber_Invert(v);
1452 Py_DECREF(v);
1453 SET_TOP(x);
1454 if (x != NULL) DISPATCH();
1455 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001456
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001457 TARGET(BINARY_POWER)
1458 w = POP();
1459 v = TOP();
1460 x = PyNumber_Power(v, w, Py_None);
1461 Py_DECREF(v);
1462 Py_DECREF(w);
1463 SET_TOP(x);
1464 if (x != NULL) DISPATCH();
1465 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001467 TARGET(BINARY_MULTIPLY)
1468 w = POP();
1469 v = TOP();
1470 x = PyNumber_Multiply(v, w);
1471 Py_DECREF(v);
1472 Py_DECREF(w);
1473 SET_TOP(x);
1474 if (x != NULL) DISPATCH();
1475 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001476
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001477 TARGET(BINARY_TRUE_DIVIDE)
1478 w = POP();
1479 v = TOP();
1480 x = PyNumber_TrueDivide(v, w);
1481 Py_DECREF(v);
1482 Py_DECREF(w);
1483 SET_TOP(x);
1484 if (x != NULL) DISPATCH();
1485 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001486
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001487 TARGET(BINARY_FLOOR_DIVIDE)
1488 w = POP();
1489 v = TOP();
1490 x = PyNumber_FloorDivide(v, w);
1491 Py_DECREF(v);
1492 Py_DECREF(w);
1493 SET_TOP(x);
1494 if (x != NULL) DISPATCH();
1495 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001496
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001497 TARGET(BINARY_MODULO)
1498 w = POP();
1499 v = TOP();
1500 if (PyUnicode_CheckExact(v))
1501 x = PyUnicode_Format(v, w);
1502 else
1503 x = PyNumber_Remainder(v, w);
1504 Py_DECREF(v);
1505 Py_DECREF(w);
1506 SET_TOP(x);
1507 if (x != NULL) DISPATCH();
1508 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001509
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001510 TARGET(BINARY_ADD)
1511 w = POP();
1512 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001513 if (PyUnicode_CheckExact(v) &&
1514 PyUnicode_CheckExact(w)) {
1515 x = unicode_concatenate(v, w, f, next_instr);
1516 /* unicode_concatenate consumed the ref to v */
1517 goto skip_decref_vx;
1518 }
1519 else {
1520 x = PyNumber_Add(v, w);
1521 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001522 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001523 skip_decref_vx:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001524 Py_DECREF(w);
1525 SET_TOP(x);
1526 if (x != NULL) DISPATCH();
1527 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001528
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001529 TARGET(BINARY_SUBTRACT)
1530 w = POP();
1531 v = TOP();
1532 x = PyNumber_Subtract(v, w);
1533 Py_DECREF(v);
1534 Py_DECREF(w);
1535 SET_TOP(x);
1536 if (x != NULL) DISPATCH();
1537 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001539 TARGET(BINARY_SUBSCR)
1540 w = POP();
1541 v = TOP();
1542 x = PyObject_GetItem(v, w);
1543 Py_DECREF(v);
1544 Py_DECREF(w);
1545 SET_TOP(x);
1546 if (x != NULL) DISPATCH();
1547 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001548
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001549 TARGET(BINARY_LSHIFT)
1550 w = POP();
1551 v = TOP();
1552 x = PyNumber_Lshift(v, w);
1553 Py_DECREF(v);
1554 Py_DECREF(w);
1555 SET_TOP(x);
1556 if (x != NULL) DISPATCH();
1557 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001558
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001559 TARGET(BINARY_RSHIFT)
1560 w = POP();
1561 v = TOP();
1562 x = PyNumber_Rshift(v, w);
1563 Py_DECREF(v);
1564 Py_DECREF(w);
1565 SET_TOP(x);
1566 if (x != NULL) DISPATCH();
1567 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001569 TARGET(BINARY_AND)
1570 w = POP();
1571 v = TOP();
1572 x = PyNumber_And(v, w);
1573 Py_DECREF(v);
1574 Py_DECREF(w);
1575 SET_TOP(x);
1576 if (x != NULL) DISPATCH();
1577 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001578
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001579 TARGET(BINARY_XOR)
1580 w = POP();
1581 v = TOP();
1582 x = PyNumber_Xor(v, w);
1583 Py_DECREF(v);
1584 Py_DECREF(w);
1585 SET_TOP(x);
1586 if (x != NULL) DISPATCH();
1587 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001588
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001589 TARGET(BINARY_OR)
1590 w = POP();
1591 v = TOP();
1592 x = PyNumber_Or(v, w);
1593 Py_DECREF(v);
1594 Py_DECREF(w);
1595 SET_TOP(x);
1596 if (x != NULL) DISPATCH();
1597 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001598
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001599 TARGET(LIST_APPEND)
1600 w = POP();
1601 v = PEEK(oparg);
1602 err = PyList_Append(v, w);
1603 Py_DECREF(w);
1604 if (err == 0) {
1605 PREDICT(JUMP_ABSOLUTE);
1606 DISPATCH();
1607 }
1608 break;
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001609
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001610 TARGET(SET_ADD)
1611 w = POP();
1612 v = stack_pointer[-oparg];
1613 err = PySet_Add(v, w);
1614 Py_DECREF(w);
1615 if (err == 0) {
1616 PREDICT(JUMP_ABSOLUTE);
1617 DISPATCH();
1618 }
1619 break;
Nick Coghlan650f0d02007-04-15 12:05:43 +00001620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001621 TARGET(INPLACE_POWER)
1622 w = POP();
1623 v = TOP();
1624 x = PyNumber_InPlacePower(v, w, Py_None);
1625 Py_DECREF(v);
1626 Py_DECREF(w);
1627 SET_TOP(x);
1628 if (x != NULL) DISPATCH();
1629 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001630
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001631 TARGET(INPLACE_MULTIPLY)
1632 w = POP();
1633 v = TOP();
1634 x = PyNumber_InPlaceMultiply(v, w);
1635 Py_DECREF(v);
1636 Py_DECREF(w);
1637 SET_TOP(x);
1638 if (x != NULL) DISPATCH();
1639 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001640
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001641 TARGET(INPLACE_TRUE_DIVIDE)
1642 w = POP();
1643 v = TOP();
1644 x = PyNumber_InPlaceTrueDivide(v, w);
1645 Py_DECREF(v);
1646 Py_DECREF(w);
1647 SET_TOP(x);
1648 if (x != NULL) DISPATCH();
1649 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001650
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001651 TARGET(INPLACE_FLOOR_DIVIDE)
1652 w = POP();
1653 v = TOP();
1654 x = PyNumber_InPlaceFloorDivide(v, w);
1655 Py_DECREF(v);
1656 Py_DECREF(w);
1657 SET_TOP(x);
1658 if (x != NULL) DISPATCH();
1659 break;
Guido van Rossum4668b002001-08-08 05:00:18 +00001660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001661 TARGET(INPLACE_MODULO)
1662 w = POP();
1663 v = TOP();
1664 x = PyNumber_InPlaceRemainder(v, w);
1665 Py_DECREF(v);
1666 Py_DECREF(w);
1667 SET_TOP(x);
1668 if (x != NULL) DISPATCH();
1669 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001670
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001671 TARGET(INPLACE_ADD)
1672 w = POP();
1673 v = TOP();
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001674 if (PyUnicode_CheckExact(v) &&
1675 PyUnicode_CheckExact(w)) {
1676 x = unicode_concatenate(v, w, f, next_instr);
1677 /* unicode_concatenate consumed the ref to v */
1678 goto skip_decref_v;
1679 }
1680 else {
1681 x = PyNumber_InPlaceAdd(v, w);
1682 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001683 Py_DECREF(v);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02001684 skip_decref_v:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001685 Py_DECREF(w);
1686 SET_TOP(x);
1687 if (x != NULL) DISPATCH();
1688 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001690 TARGET(INPLACE_SUBTRACT)
1691 w = POP();
1692 v = TOP();
1693 x = PyNumber_InPlaceSubtract(v, w);
1694 Py_DECREF(v);
1695 Py_DECREF(w);
1696 SET_TOP(x);
1697 if (x != NULL) DISPATCH();
1698 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001699
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001700 TARGET(INPLACE_LSHIFT)
1701 w = POP();
1702 v = TOP();
1703 x = PyNumber_InPlaceLshift(v, w);
1704 Py_DECREF(v);
1705 Py_DECREF(w);
1706 SET_TOP(x);
1707 if (x != NULL) DISPATCH();
1708 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001709
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001710 TARGET(INPLACE_RSHIFT)
1711 w = POP();
1712 v = TOP();
1713 x = PyNumber_InPlaceRshift(v, w);
1714 Py_DECREF(v);
1715 Py_DECREF(w);
1716 SET_TOP(x);
1717 if (x != NULL) DISPATCH();
1718 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001719
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001720 TARGET(INPLACE_AND)
1721 w = POP();
1722 v = TOP();
1723 x = PyNumber_InPlaceAnd(v, w);
1724 Py_DECREF(v);
1725 Py_DECREF(w);
1726 SET_TOP(x);
1727 if (x != NULL) DISPATCH();
1728 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001729
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001730 TARGET(INPLACE_XOR)
1731 w = POP();
1732 v = TOP();
1733 x = PyNumber_InPlaceXor(v, w);
1734 Py_DECREF(v);
1735 Py_DECREF(w);
1736 SET_TOP(x);
1737 if (x != NULL) DISPATCH();
1738 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001739
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001740 TARGET(INPLACE_OR)
1741 w = POP();
1742 v = TOP();
1743 x = PyNumber_InPlaceOr(v, w);
1744 Py_DECREF(v);
1745 Py_DECREF(w);
1746 SET_TOP(x);
1747 if (x != NULL) DISPATCH();
1748 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001749
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001750 TARGET(STORE_SUBSCR)
1751 w = TOP();
1752 v = SECOND();
1753 u = THIRD();
1754 STACKADJ(-3);
1755 /* v[w] = u */
1756 err = PyObject_SetItem(v, w, u);
1757 Py_DECREF(u);
1758 Py_DECREF(v);
1759 Py_DECREF(w);
1760 if (err == 0) DISPATCH();
1761 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001762
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001763 TARGET(DELETE_SUBSCR)
1764 w = TOP();
1765 v = SECOND();
1766 STACKADJ(-2);
1767 /* del v[w] */
1768 err = PyObject_DelItem(v, w);
1769 Py_DECREF(v);
1770 Py_DECREF(w);
1771 if (err == 0) DISPATCH();
1772 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001773
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001774 TARGET(PRINT_EXPR)
1775 v = POP();
1776 w = PySys_GetObject("displayhook");
1777 if (w == NULL) {
1778 PyErr_SetString(PyExc_RuntimeError,
1779 "lost sys.displayhook");
1780 err = -1;
1781 x = NULL;
1782 }
1783 if (err == 0) {
1784 x = PyTuple_Pack(1, v);
1785 if (x == NULL)
1786 err = -1;
1787 }
1788 if (err == 0) {
1789 w = PyEval_CallObject(w, x);
1790 Py_XDECREF(w);
1791 if (w == NULL)
1792 err = -1;
1793 }
1794 Py_DECREF(v);
1795 Py_XDECREF(x);
1796 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001797
Thomas Wouters434d0822000-08-24 20:11:32 +00001798#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001799 default: switch (opcode) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001800#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001801 TARGET(RAISE_VARARGS)
1802 v = w = NULL;
1803 switch (oparg) {
1804 case 2:
1805 v = POP(); /* cause */
1806 case 1:
1807 w = POP(); /* exc */
1808 case 0: /* Fallthrough */
1809 why = do_raise(w, v);
1810 break;
1811 default:
1812 PyErr_SetString(PyExc_SystemError,
1813 "bad RAISE_VARARGS oparg");
1814 why = WHY_EXCEPTION;
1815 break;
1816 }
1817 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001819 TARGET(STORE_LOCALS)
1820 x = POP();
1821 v = f->f_locals;
1822 Py_XDECREF(v);
1823 f->f_locals = x;
1824 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001825
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001826 TARGET(RETURN_VALUE)
1827 retval = POP();
1828 why = WHY_RETURN;
1829 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001830
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001831 TARGET(YIELD_VALUE)
1832 retval = POP();
1833 f->f_stacktop = stack_pointer;
1834 why = WHY_YIELD;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001835 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001836
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001837 TARGET(POP_EXCEPT)
1838 {
1839 PyTryBlock *b = PyFrame_BlockPop(f);
1840 if (b->b_type != EXCEPT_HANDLER) {
1841 PyErr_SetString(PyExc_SystemError,
1842 "popped block is not an except handler");
1843 why = WHY_EXCEPTION;
1844 break;
1845 }
1846 UNWIND_EXCEPT_HANDLER(b);
1847 }
1848 DISPATCH();
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001849
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001850 TARGET(POP_BLOCK)
1851 {
1852 PyTryBlock *b = PyFrame_BlockPop(f);
1853 UNWIND_BLOCK(b);
1854 }
1855 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00001856
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001857 PREDICTED(END_FINALLY);
1858 TARGET(END_FINALLY)
1859 v = POP();
1860 if (PyLong_Check(v)) {
1861 why = (enum why_code) PyLong_AS_LONG(v);
1862 assert(why != WHY_YIELD);
1863 if (why == WHY_RETURN ||
1864 why == WHY_CONTINUE)
1865 retval = POP();
1866 if (why == WHY_SILENCED) {
1867 /* An exception was silenced by 'with', we must
1868 manually unwind the EXCEPT_HANDLER block which was
1869 created when the exception was caught, otherwise
1870 the stack will be in an inconsistent state. */
1871 PyTryBlock *b = PyFrame_BlockPop(f);
1872 assert(b->b_type == EXCEPT_HANDLER);
1873 UNWIND_EXCEPT_HANDLER(b);
1874 why = WHY_NOT;
1875 }
1876 }
1877 else if (PyExceptionClass_Check(v)) {
1878 w = POP();
1879 u = POP();
1880 PyErr_Restore(v, w, u);
1881 why = WHY_RERAISE;
1882 break;
1883 }
1884 else if (v != Py_None) {
1885 PyErr_SetString(PyExc_SystemError,
1886 "'finally' pops bad exception");
1887 why = WHY_EXCEPTION;
1888 }
1889 Py_DECREF(v);
1890 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001891
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001892 TARGET(LOAD_BUILD_CLASS)
1893 x = PyDict_GetItemString(f->f_builtins,
1894 "__build_class__");
1895 if (x == NULL) {
1896 PyErr_SetString(PyExc_ImportError,
1897 "__build_class__ not found");
1898 break;
1899 }
1900 Py_INCREF(x);
1901 PUSH(x);
1902 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001903
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001904 TARGET(STORE_NAME)
1905 w = GETITEM(names, oparg);
1906 v = POP();
1907 if ((x = f->f_locals) != NULL) {
1908 if (PyDict_CheckExact(x))
1909 err = PyDict_SetItem(x, w, v);
1910 else
1911 err = PyObject_SetItem(x, w, v);
1912 Py_DECREF(v);
1913 if (err == 0) DISPATCH();
1914 break;
1915 }
1916 PyErr_Format(PyExc_SystemError,
1917 "no locals found when storing %R", w);
1918 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001919
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001920 TARGET(DELETE_NAME)
1921 w = GETITEM(names, oparg);
1922 if ((x = f->f_locals) != NULL) {
1923 if ((err = PyObject_DelItem(x, w)) != 0)
1924 format_exc_check_arg(PyExc_NameError,
1925 NAME_ERROR_MSG,
1926 w);
1927 break;
1928 }
1929 PyErr_Format(PyExc_SystemError,
1930 "no locals when deleting %R", w);
1931 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001932
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001933 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
1934 TARGET(UNPACK_SEQUENCE)
1935 v = POP();
1936 if (PyTuple_CheckExact(v) &&
1937 PyTuple_GET_SIZE(v) == oparg) {
1938 PyObject **items = \
1939 ((PyTupleObject *)v)->ob_item;
1940 while (oparg--) {
1941 w = items[oparg];
1942 Py_INCREF(w);
1943 PUSH(w);
1944 }
1945 Py_DECREF(v);
1946 DISPATCH();
1947 } else if (PyList_CheckExact(v) &&
1948 PyList_GET_SIZE(v) == oparg) {
1949 PyObject **items = \
1950 ((PyListObject *)v)->ob_item;
1951 while (oparg--) {
1952 w = items[oparg];
1953 Py_INCREF(w);
1954 PUSH(w);
1955 }
1956 } else if (unpack_iterable(v, oparg, -1,
1957 stack_pointer + oparg)) {
1958 STACKADJ(oparg);
1959 } else {
1960 /* unpack_iterable() raised an exception */
1961 why = WHY_EXCEPTION;
1962 }
1963 Py_DECREF(v);
1964 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001965
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001966 TARGET(UNPACK_EX)
1967 {
1968 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
1969 v = POP();
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00001970
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001971 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
1972 stack_pointer + totalargs)) {
1973 stack_pointer += totalargs;
1974 } else {
1975 why = WHY_EXCEPTION;
1976 }
1977 Py_DECREF(v);
1978 break;
1979 }
Guido van Rossum0368b722007-05-11 16:50:42 +00001980
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001981 TARGET(STORE_ATTR)
1982 w = GETITEM(names, oparg);
1983 v = TOP();
1984 u = SECOND();
1985 STACKADJ(-2);
1986 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1987 Py_DECREF(v);
1988 Py_DECREF(u);
1989 if (err == 0) DISPATCH();
1990 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001991
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001992 TARGET(DELETE_ATTR)
1993 w = GETITEM(names, oparg);
1994 v = POP();
1995 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1996 /* del v.w */
1997 Py_DECREF(v);
1998 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001999
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002000 TARGET(STORE_GLOBAL)
2001 w = GETITEM(names, oparg);
2002 v = POP();
2003 err = PyDict_SetItem(f->f_globals, w, v);
2004 Py_DECREF(v);
2005 if (err == 0) DISPATCH();
2006 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002007
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002008 TARGET(DELETE_GLOBAL)
2009 w = GETITEM(names, oparg);
2010 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
2011 format_exc_check_arg(
2012 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
2013 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002014
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002015 TARGET(LOAD_NAME)
2016 w = GETITEM(names, oparg);
2017 if ((v = f->f_locals) == NULL) {
2018 PyErr_Format(PyExc_SystemError,
2019 "no locals when loading %R", w);
2020 why = WHY_EXCEPTION;
2021 break;
2022 }
2023 if (PyDict_CheckExact(v)) {
2024 x = PyDict_GetItem(v, w);
2025 Py_XINCREF(x);
2026 }
2027 else {
2028 x = PyObject_GetItem(v, w);
2029 if (x == NULL && PyErr_Occurred()) {
2030 if (!PyErr_ExceptionMatches(
2031 PyExc_KeyError))
2032 break;
2033 PyErr_Clear();
2034 }
2035 }
2036 if (x == NULL) {
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002037 x = PyDict_GetItem(f->f_globals, w);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002038 if (x == NULL) {
Benjamin Peterson20f9c3c2010-07-20 22:39:34 +00002039 x = PyDict_GetItem(f->f_builtins, w);
2040 if (x == NULL) {
2041 format_exc_check_arg(
2042 PyExc_NameError,
2043 NAME_ERROR_MSG, w);
2044 break;
2045 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002046 }
2047 Py_INCREF(x);
2048 }
2049 PUSH(x);
2050 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002051
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002052 TARGET(LOAD_GLOBAL)
2053 w = GETITEM(names, oparg);
2054 if (PyUnicode_CheckExact(w)) {
2055 /* Inline the PyDict_GetItem() calls.
2056 WARNING: this is an extreme speed hack.
2057 Do not try this at home. */
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02002058 Py_hash_t hash = ((PyASCIIObject *)w)->hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002059 if (hash != -1) {
2060 PyDictObject *d;
2061 PyDictEntry *e;
2062 d = (PyDictObject *)(f->f_globals);
2063 e = d->ma_lookup(d, w, hash);
2064 if (e == NULL) {
2065 x = NULL;
2066 break;
2067 }
2068 x = e->me_value;
2069 if (x != NULL) {
2070 Py_INCREF(x);
2071 PUSH(x);
2072 DISPATCH();
2073 }
2074 d = (PyDictObject *)(f->f_builtins);
2075 e = d->ma_lookup(d, w, hash);
2076 if (e == NULL) {
2077 x = NULL;
2078 break;
2079 }
2080 x = e->me_value;
2081 if (x != NULL) {
2082 Py_INCREF(x);
2083 PUSH(x);
2084 DISPATCH();
2085 }
2086 goto load_global_error;
2087 }
2088 }
2089 /* This is the un-inlined version of the code above */
2090 x = PyDict_GetItem(f->f_globals, w);
2091 if (x == NULL) {
2092 x = PyDict_GetItem(f->f_builtins, w);
2093 if (x == NULL) {
2094 load_global_error:
2095 format_exc_check_arg(
2096 PyExc_NameError,
2097 GLOBAL_NAME_ERROR_MSG, w);
2098 break;
2099 }
2100 }
2101 Py_INCREF(x);
2102 PUSH(x);
2103 DISPATCH();
Guido van Rossum681d79a1995-07-18 14:51:37 +00002104
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002105 TARGET(DELETE_FAST)
2106 x = GETLOCAL(oparg);
2107 if (x != NULL) {
2108 SETLOCAL(oparg, NULL);
2109 DISPATCH();
2110 }
2111 format_exc_check_arg(
2112 PyExc_UnboundLocalError,
2113 UNBOUNDLOCAL_ERROR_MSG,
2114 PyTuple_GetItem(co->co_varnames, oparg)
2115 );
2116 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002117
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002118 TARGET(DELETE_DEREF)
2119 x = freevars[oparg];
2120 if (PyCell_GET(x) != NULL) {
2121 PyCell_Set(x, NULL);
Benjamin Peterson00ebe2c2010-09-10 22:02:31 +00002122 DISPATCH();
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002123 }
2124 err = -1;
2125 format_exc_unbound(co, oparg);
2126 break;
2127
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002128 TARGET(LOAD_CLOSURE)
2129 x = freevars[oparg];
2130 Py_INCREF(x);
2131 PUSH(x);
2132 if (x != NULL) DISPATCH();
2133 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002134
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002135 TARGET(LOAD_DEREF)
2136 x = freevars[oparg];
2137 w = PyCell_Get(x);
2138 if (w != NULL) {
2139 PUSH(w);
2140 DISPATCH();
2141 }
2142 err = -1;
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00002143 format_exc_unbound(co, oparg);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002144 break;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002145
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002146 TARGET(STORE_DEREF)
2147 w = POP();
2148 x = freevars[oparg];
2149 PyCell_Set(x, w);
2150 Py_DECREF(w);
2151 DISPATCH();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002152
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002153 TARGET(BUILD_TUPLE)
2154 x = PyTuple_New(oparg);
2155 if (x != NULL) {
2156 for (; --oparg >= 0;) {
2157 w = POP();
2158 PyTuple_SET_ITEM(x, oparg, w);
2159 }
2160 PUSH(x);
2161 DISPATCH();
2162 }
2163 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002164
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002165 TARGET(BUILD_LIST)
2166 x = PyList_New(oparg);
2167 if (x != NULL) {
2168 for (; --oparg >= 0;) {
2169 w = POP();
2170 PyList_SET_ITEM(x, oparg, w);
2171 }
2172 PUSH(x);
2173 DISPATCH();
2174 }
2175 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002176
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002177 TARGET(BUILD_SET)
2178 x = PySet_New(NULL);
2179 if (x != NULL) {
2180 for (; --oparg >= 0;) {
2181 w = POP();
2182 if (err == 0)
2183 err = PySet_Add(x, w);
2184 Py_DECREF(w);
2185 }
2186 if (err != 0) {
2187 Py_DECREF(x);
2188 break;
2189 }
2190 PUSH(x);
2191 DISPATCH();
2192 }
2193 break;
Guido van Rossum86e58e22006-08-28 15:27:34 +00002194
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002195 TARGET(BUILD_MAP)
2196 x = _PyDict_NewPresized((Py_ssize_t)oparg);
2197 PUSH(x);
2198 if (x != NULL) DISPATCH();
2199 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002200
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002201 TARGET(STORE_MAP)
2202 w = TOP(); /* key */
2203 u = SECOND(); /* value */
2204 v = THIRD(); /* dict */
2205 STACKADJ(-2);
2206 assert (PyDict_CheckExact(v));
2207 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2208 Py_DECREF(u);
2209 Py_DECREF(w);
2210 if (err == 0) DISPATCH();
2211 break;
Christian Heimes99170a52007-12-19 02:07:34 +00002212
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002213 TARGET(MAP_ADD)
2214 w = TOP(); /* key */
2215 u = SECOND(); /* value */
2216 STACKADJ(-2);
2217 v = stack_pointer[-oparg]; /* dict */
2218 assert (PyDict_CheckExact(v));
2219 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2220 Py_DECREF(u);
2221 Py_DECREF(w);
2222 if (err == 0) {
2223 PREDICT(JUMP_ABSOLUTE);
2224 DISPATCH();
2225 }
2226 break;
Antoine Pitrouf289ae62008-12-18 11:06:25 +00002227
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002228 TARGET(LOAD_ATTR)
2229 w = GETITEM(names, oparg);
2230 v = TOP();
2231 x = PyObject_GetAttr(v, w);
2232 Py_DECREF(v);
2233 SET_TOP(x);
2234 if (x != NULL) DISPATCH();
2235 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002236
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002237 TARGET(COMPARE_OP)
2238 w = POP();
2239 v = TOP();
2240 x = cmp_outcome(oparg, v, w);
2241 Py_DECREF(v);
2242 Py_DECREF(w);
2243 SET_TOP(x);
2244 if (x == NULL) break;
2245 PREDICT(POP_JUMP_IF_FALSE);
2246 PREDICT(POP_JUMP_IF_TRUE);
2247 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002248
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002249 TARGET(IMPORT_NAME)
2250 w = GETITEM(names, oparg);
2251 x = PyDict_GetItemString(f->f_builtins, "__import__");
2252 if (x == NULL) {
2253 PyErr_SetString(PyExc_ImportError,
2254 "__import__ not found");
2255 break;
2256 }
2257 Py_INCREF(x);
2258 v = POP();
2259 u = TOP();
2260 if (PyLong_AsLong(u) != -1 || PyErr_Occurred())
2261 w = PyTuple_Pack(5,
2262 w,
2263 f->f_globals,
2264 f->f_locals == NULL ?
2265 Py_None : f->f_locals,
2266 v,
2267 u);
2268 else
2269 w = PyTuple_Pack(4,
2270 w,
2271 f->f_globals,
2272 f->f_locals == NULL ?
2273 Py_None : f->f_locals,
2274 v);
2275 Py_DECREF(v);
2276 Py_DECREF(u);
2277 if (w == NULL) {
2278 u = POP();
2279 Py_DECREF(x);
2280 x = NULL;
2281 break;
2282 }
2283 READ_TIMESTAMP(intr0);
2284 v = x;
2285 x = PyEval_CallObject(v, w);
2286 Py_DECREF(v);
2287 READ_TIMESTAMP(intr1);
2288 Py_DECREF(w);
2289 SET_TOP(x);
2290 if (x != NULL) DISPATCH();
2291 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002292
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002293 TARGET(IMPORT_STAR)
2294 v = POP();
2295 PyFrame_FastToLocals(f);
2296 if ((x = f->f_locals) == NULL) {
2297 PyErr_SetString(PyExc_SystemError,
2298 "no locals found during 'import *'");
2299 break;
2300 }
2301 READ_TIMESTAMP(intr0);
2302 err = import_all_from(x, v);
2303 READ_TIMESTAMP(intr1);
2304 PyFrame_LocalsToFast(f, 0);
2305 Py_DECREF(v);
2306 if (err == 0) DISPATCH();
2307 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002308
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002309 TARGET(IMPORT_FROM)
2310 w = GETITEM(names, oparg);
2311 v = TOP();
2312 READ_TIMESTAMP(intr0);
2313 x = import_from(v, w);
2314 READ_TIMESTAMP(intr1);
2315 PUSH(x);
2316 if (x != NULL) DISPATCH();
2317 break;
Thomas Wouters52152252000-08-17 22:55:00 +00002318
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002319 TARGET(JUMP_FORWARD)
2320 JUMPBY(oparg);
2321 FAST_DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002322
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002323 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
2324 TARGET(POP_JUMP_IF_FALSE)
2325 w = POP();
2326 if (w == Py_True) {
2327 Py_DECREF(w);
2328 FAST_DISPATCH();
2329 }
2330 if (w == Py_False) {
2331 Py_DECREF(w);
2332 JUMPTO(oparg);
2333 FAST_DISPATCH();
2334 }
2335 err = PyObject_IsTrue(w);
2336 Py_DECREF(w);
2337 if (err > 0)
2338 err = 0;
2339 else if (err == 0)
2340 JUMPTO(oparg);
2341 else
2342 break;
2343 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002344
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002345 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
2346 TARGET(POP_JUMP_IF_TRUE)
2347 w = POP();
2348 if (w == Py_False) {
2349 Py_DECREF(w);
2350 FAST_DISPATCH();
2351 }
2352 if (w == Py_True) {
2353 Py_DECREF(w);
2354 JUMPTO(oparg);
2355 FAST_DISPATCH();
2356 }
2357 err = PyObject_IsTrue(w);
2358 Py_DECREF(w);
2359 if (err > 0) {
2360 err = 0;
2361 JUMPTO(oparg);
2362 }
2363 else if (err == 0)
2364 ;
2365 else
2366 break;
2367 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002368
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002369 TARGET(JUMP_IF_FALSE_OR_POP)
2370 w = TOP();
2371 if (w == Py_True) {
2372 STACKADJ(-1);
2373 Py_DECREF(w);
2374 FAST_DISPATCH();
2375 }
2376 if (w == Py_False) {
2377 JUMPTO(oparg);
2378 FAST_DISPATCH();
2379 }
2380 err = PyObject_IsTrue(w);
2381 if (err > 0) {
2382 STACKADJ(-1);
2383 Py_DECREF(w);
2384 err = 0;
2385 }
2386 else if (err == 0)
2387 JUMPTO(oparg);
2388 else
2389 break;
2390 DISPATCH();
Jeffrey Yasskin9de7ec72009-02-25 02:25:04 +00002391
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002392 TARGET(JUMP_IF_TRUE_OR_POP)
2393 w = TOP();
2394 if (w == Py_False) {
2395 STACKADJ(-1);
2396 Py_DECREF(w);
2397 FAST_DISPATCH();
2398 }
2399 if (w == Py_True) {
2400 JUMPTO(oparg);
2401 FAST_DISPATCH();
2402 }
2403 err = PyObject_IsTrue(w);
2404 if (err > 0) {
2405 err = 0;
2406 JUMPTO(oparg);
2407 }
2408 else if (err == 0) {
2409 STACKADJ(-1);
2410 Py_DECREF(w);
2411 }
2412 else
2413 break;
2414 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002415
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002416 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
2417 TARGET(JUMP_ABSOLUTE)
2418 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002419#if FAST_LOOPS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002420 /* Enabling this path speeds-up all while and for-loops by bypassing
2421 the per-loop checks for signals. By default, this should be turned-off
2422 because it prevents detection of a control-break in tight loops like
2423 "while 1: pass". Compile with this option turned-on when you need
2424 the speed-up and do not need break checking inside tight loops (ones
2425 that contain only instructions ending with FAST_DISPATCH).
2426 */
2427 FAST_DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002428#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002429 DISPATCH();
Guido van Rossum58da9312007-11-10 23:39:45 +00002430#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002431
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002432 TARGET(GET_ITER)
2433 /* before: [obj]; after [getiter(obj)] */
2434 v = TOP();
2435 x = PyObject_GetIter(v);
2436 Py_DECREF(v);
2437 if (x != NULL) {
2438 SET_TOP(x);
2439 PREDICT(FOR_ITER);
2440 DISPATCH();
2441 }
2442 STACKADJ(-1);
2443 break;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002444
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002445 PREDICTED_WITH_ARG(FOR_ITER);
2446 TARGET(FOR_ITER)
2447 /* before: [iter]; after: [iter, iter()] *or* [] */
2448 v = TOP();
2449 x = (*v->ob_type->tp_iternext)(v);
2450 if (x != NULL) {
2451 PUSH(x);
2452 PREDICT(STORE_FAST);
2453 PREDICT(UNPACK_SEQUENCE);
2454 DISPATCH();
2455 }
2456 if (PyErr_Occurred()) {
2457 if (!PyErr_ExceptionMatches(
2458 PyExc_StopIteration))
2459 break;
2460 PyErr_Clear();
2461 }
2462 /* iterator ended normally */
2463 x = v = POP();
2464 Py_DECREF(v);
2465 JUMPBY(oparg);
2466 DISPATCH();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002467
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002468 TARGET(BREAK_LOOP)
2469 why = WHY_BREAK;
2470 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002471
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002472 TARGET(CONTINUE_LOOP)
2473 retval = PyLong_FromLong(oparg);
2474 if (!retval) {
2475 x = NULL;
2476 break;
2477 }
2478 why = WHY_CONTINUE;
2479 goto fast_block_end;
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002480
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002481 TARGET_WITH_IMPL(SETUP_LOOP, _setup_finally)
2482 TARGET_WITH_IMPL(SETUP_EXCEPT, _setup_finally)
2483 TARGET(SETUP_FINALLY)
2484 _setup_finally:
2485 /* NOTE: If you add any new block-setup opcodes that
2486 are not try/except/finally handlers, you may need
2487 to update the PyGen_NeedsFinalizing() function.
2488 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002489
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002490 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
2491 STACK_LEVEL());
2492 DISPATCH();
Guido van Rossumac7be682001-01-17 15:42:30 +00002493
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002494 TARGET(SETUP_WITH)
2495 {
2496 static PyObject *exit, *enter;
2497 w = TOP();
2498 x = special_lookup(w, "__exit__", &exit);
2499 if (!x)
2500 break;
2501 SET_TOP(x);
2502 u = special_lookup(w, "__enter__", &enter);
2503 Py_DECREF(w);
2504 if (!u) {
2505 x = NULL;
2506 break;
2507 }
2508 x = PyObject_CallFunctionObjArgs(u, NULL);
2509 Py_DECREF(u);
2510 if (!x)
2511 break;
2512 /* Setup the finally block before pushing the result
2513 of __enter__ on the stack. */
2514 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2515 STACK_LEVEL());
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002516
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002517 PUSH(x);
2518 DISPATCH();
2519 }
Benjamin Peterson876b2f22009-06-28 03:18:59 +00002520
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002521 TARGET(WITH_CLEANUP)
2522 {
2523 /* At the top of the stack are 1-3 values indicating
2524 how/why we entered the finally clause:
2525 - TOP = None
2526 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2527 - TOP = WHY_*; no retval below it
2528 - (TOP, SECOND, THIRD) = exc_info()
2529 (FOURTH, FITH, SIXTH) = previous exception for EXCEPT_HANDLER
2530 Below them is EXIT, the context.__exit__ bound method.
2531 In the last case, we must call
2532 EXIT(TOP, SECOND, THIRD)
2533 otherwise we must call
2534 EXIT(None, None, None)
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002535
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002536 In the first two cases, we remove EXIT from the
2537 stack, leaving the rest in the same order. In the
2538 third case, we shift the bottom 3 values of the
2539 stack down, and replace the empty spot with NULL.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002540
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002541 In addition, if the stack represents an exception,
2542 *and* the function call returns a 'true' value, we
2543 push WHY_SILENCED onto the stack. END_FINALLY will
2544 then not re-raise the exception. (But non-local
2545 gotos should still be resumed.)
2546 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002547
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002548 PyObject *exit_func;
2549 u = TOP();
2550 if (u == Py_None) {
2551 (void)POP();
2552 exit_func = TOP();
2553 SET_TOP(u);
2554 v = w = Py_None;
2555 }
2556 else if (PyLong_Check(u)) {
2557 (void)POP();
2558 switch(PyLong_AsLong(u)) {
2559 case WHY_RETURN:
2560 case WHY_CONTINUE:
2561 /* Retval in TOP. */
2562 exit_func = SECOND();
2563 SET_SECOND(TOP());
2564 SET_TOP(u);
2565 break;
2566 default:
2567 exit_func = TOP();
2568 SET_TOP(u);
2569 break;
2570 }
2571 u = v = w = Py_None;
2572 }
2573 else {
2574 PyObject *tp, *exc, *tb;
2575 PyTryBlock *block;
2576 v = SECOND();
2577 w = THIRD();
2578 tp = FOURTH();
2579 exc = PEEK(5);
2580 tb = PEEK(6);
2581 exit_func = PEEK(7);
2582 SET_VALUE(7, tb);
2583 SET_VALUE(6, exc);
2584 SET_VALUE(5, tp);
2585 /* UNWIND_EXCEPT_HANDLER will pop this off. */
2586 SET_FOURTH(NULL);
2587 /* We just shifted the stack down, so we have
2588 to tell the except handler block that the
2589 values are lower than it expects. */
2590 block = &f->f_blockstack[f->f_iblock - 1];
2591 assert(block->b_type == EXCEPT_HANDLER);
2592 block->b_level--;
2593 }
2594 /* XXX Not the fastest way to call it... */
2595 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2596 NULL);
2597 Py_DECREF(exit_func);
2598 if (x == NULL)
2599 break; /* Go to error exit */
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002601 if (u != Py_None)
2602 err = PyObject_IsTrue(x);
2603 else
2604 err = 0;
2605 Py_DECREF(x);
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002606
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002607 if (err < 0)
2608 break; /* Go to error exit */
2609 else if (err > 0) {
2610 err = 0;
2611 /* There was an exception and a True return */
2612 PUSH(PyLong_FromLong((long) WHY_SILENCED));
2613 }
2614 PREDICT(END_FINALLY);
2615 break;
2616 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002617
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002618 TARGET(CALL_FUNCTION)
2619 {
2620 PyObject **sp;
2621 PCALL(PCALL_ALL);
2622 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002623#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002624 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002625#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002626 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002627#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002628 stack_pointer = sp;
2629 PUSH(x);
2630 if (x != NULL)
2631 DISPATCH();
2632 break;
2633 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002634
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002635 TARGET_WITH_IMPL(CALL_FUNCTION_VAR, _call_function_var_kw)
2636 TARGET_WITH_IMPL(CALL_FUNCTION_KW, _call_function_var_kw)
2637 TARGET(CALL_FUNCTION_VAR_KW)
2638 _call_function_var_kw:
2639 {
2640 int na = oparg & 0xff;
2641 int nk = (oparg>>8) & 0xff;
2642 int flags = (opcode - CALL_FUNCTION) & 3;
2643 int n = na + 2 * nk;
2644 PyObject **pfunc, *func, **sp;
2645 PCALL(PCALL_ALL);
2646 if (flags & CALL_FLAG_VAR)
2647 n++;
2648 if (flags & CALL_FLAG_KW)
2649 n++;
2650 pfunc = stack_pointer - n - 1;
2651 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002652
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002653 if (PyMethod_Check(func)
Stefan Krahb7e10102010-06-23 18:42:39 +00002654 && PyMethod_GET_SELF(func) != NULL) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002655 PyObject *self = PyMethod_GET_SELF(func);
2656 Py_INCREF(self);
2657 func = PyMethod_GET_FUNCTION(func);
2658 Py_INCREF(func);
2659 Py_DECREF(*pfunc);
2660 *pfunc = self;
2661 na++;
Brett Cannonb94767f2011-02-22 20:15:44 +00002662 /* n++; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002663 } else
2664 Py_INCREF(func);
2665 sp = stack_pointer;
2666 READ_TIMESTAMP(intr0);
2667 x = ext_do_call(func, &sp, flags, na, nk);
2668 READ_TIMESTAMP(intr1);
2669 stack_pointer = sp;
2670 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002671
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002672 while (stack_pointer > pfunc) {
2673 w = POP();
2674 Py_DECREF(w);
2675 }
2676 PUSH(x);
2677 if (x != NULL)
2678 DISPATCH();
2679 break;
2680 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002681
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002682 TARGET_WITH_IMPL(MAKE_CLOSURE, _make_function)
2683 TARGET(MAKE_FUNCTION)
2684 _make_function:
2685 {
2686 int posdefaults = oparg & 0xff;
2687 int kwdefaults = (oparg>>8) & 0xff;
2688 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002689
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002690 v = POP(); /* code object */
2691 x = PyFunction_New(v, f->f_globals);
2692 Py_DECREF(v);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002693
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002694 if (x != NULL && opcode == MAKE_CLOSURE) {
2695 v = POP();
2696 if (PyFunction_SetClosure(x, v) != 0) {
2697 /* Can't happen unless bytecode is corrupt. */
2698 why = WHY_EXCEPTION;
2699 }
2700 Py_DECREF(v);
2701 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002702
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002703 if (x != NULL && num_annotations > 0) {
2704 Py_ssize_t name_ix;
2705 u = POP(); /* names of args with annotations */
2706 v = PyDict_New();
2707 if (v == NULL) {
2708 Py_DECREF(x);
2709 x = NULL;
2710 break;
2711 }
2712 name_ix = PyTuple_Size(u);
2713 assert(num_annotations == name_ix+1);
2714 while (name_ix > 0) {
2715 --name_ix;
2716 t = PyTuple_GET_ITEM(u, name_ix);
2717 w = POP();
2718 /* XXX(nnorwitz): check for errors */
2719 PyDict_SetItem(v, t, w);
2720 Py_DECREF(w);
2721 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002722
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002723 if (PyFunction_SetAnnotations(x, v) != 0) {
2724 /* Can't happen unless
2725 PyFunction_SetAnnotations changes. */
2726 why = WHY_EXCEPTION;
2727 }
2728 Py_DECREF(v);
2729 Py_DECREF(u);
2730 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002731
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002732 /* XXX Maybe this should be a separate opcode? */
2733 if (x != NULL && posdefaults > 0) {
2734 v = PyTuple_New(posdefaults);
2735 if (v == NULL) {
2736 Py_DECREF(x);
2737 x = NULL;
2738 break;
2739 }
2740 while (--posdefaults >= 0) {
2741 w = POP();
2742 PyTuple_SET_ITEM(v, posdefaults, w);
2743 }
2744 if (PyFunction_SetDefaults(x, v) != 0) {
2745 /* Can't happen unless
2746 PyFunction_SetDefaults changes. */
2747 why = WHY_EXCEPTION;
2748 }
2749 Py_DECREF(v);
2750 }
2751 if (x != NULL && kwdefaults > 0) {
2752 v = PyDict_New();
2753 if (v == NULL) {
2754 Py_DECREF(x);
2755 x = NULL;
2756 break;
2757 }
2758 while (--kwdefaults >= 0) {
2759 w = POP(); /* default value */
2760 u = POP(); /* kw only arg name */
2761 /* XXX(nnorwitz): check for errors */
2762 PyDict_SetItem(v, u, w);
2763 Py_DECREF(w);
2764 Py_DECREF(u);
2765 }
2766 if (PyFunction_SetKwDefaults(x, v) != 0) {
2767 /* Can't happen unless
2768 PyFunction_SetKwDefaults changes. */
2769 why = WHY_EXCEPTION;
2770 }
2771 Py_DECREF(v);
2772 }
2773 PUSH(x);
2774 break;
2775 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002776
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002777 TARGET(BUILD_SLICE)
2778 if (oparg == 3)
2779 w = POP();
2780 else
2781 w = NULL;
2782 v = POP();
2783 u = TOP();
2784 x = PySlice_New(u, v, w);
2785 Py_DECREF(u);
2786 Py_DECREF(v);
2787 Py_XDECREF(w);
2788 SET_TOP(x);
2789 if (x != NULL) DISPATCH();
2790 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002791
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002792 TARGET(EXTENDED_ARG)
2793 opcode = NEXTOP();
2794 oparg = oparg<<16 | NEXTARG();
2795 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002796
Antoine Pitrou042b1282010-08-13 21:15:58 +00002797#if USE_COMPUTED_GOTOS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002798 _unknown_opcode:
Antoine Pitroub52ec782009-01-25 16:34:23 +00002799#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002800 default:
2801 fprintf(stderr,
2802 "XXX lineno: %d, opcode: %d\n",
2803 PyFrame_GetLineNumber(f),
2804 opcode);
2805 PyErr_SetString(PyExc_SystemError, "unknown opcode");
2806 why = WHY_EXCEPTION;
2807 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002808
2809#ifdef CASE_TOO_BIG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002810 }
Guido van Rossum04691fc1992-08-12 15:35:34 +00002811#endif
2812
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002813 } /* switch */
Guido van Rossum374a9221991-04-04 10:40:29 +00002814
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002815 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002816
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002817 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002818
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002819 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002820
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002821 if (why == WHY_NOT) {
2822 if (err == 0 && x != NULL) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002823#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002824 /* This check is expensive! */
2825 if (PyErr_Occurred())
2826 fprintf(stderr,
2827 "XXX undetected error\n");
2828 else {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002829#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002830 READ_TIMESTAMP(loop1);
2831 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002832#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002833 }
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002834#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002835 }
2836 why = WHY_EXCEPTION;
2837 x = Py_None;
2838 err = 0;
2839 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002840
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002841 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002842
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002843 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
2844 if (!PyErr_Occurred()) {
2845 PyErr_SetString(PyExc_SystemError,
2846 "error return without exception set");
2847 why = WHY_EXCEPTION;
2848 }
2849 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002850#ifdef CHECKEXC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002851 else {
2852 /* This check is expensive! */
2853 if (PyErr_Occurred()) {
2854 char buf[128];
2855 sprintf(buf, "Stack unwind with exception "
2856 "set and why=%d", why);
2857 Py_FatalError(buf);
2858 }
2859 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002860#endif
2861
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002862 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002863
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002864 if (why == WHY_EXCEPTION) {
2865 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002866
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002867 if (tstate->c_tracefunc != NULL)
2868 call_exc_trace(tstate->c_tracefunc,
2869 tstate->c_traceobj, f);
2870 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002871
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002872 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002874 if (why == WHY_RERAISE)
2875 why = WHY_EXCEPTION;
Guido van Rossum374a9221991-04-04 10:40:29 +00002876
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002877 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002878
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002879fast_block_end:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002880 while (why != WHY_NOT && f->f_iblock > 0) {
2881 /* Peek at the current block. */
2882 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002883
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002884 assert(why != WHY_YIELD);
2885 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2886 why = WHY_NOT;
2887 JUMPTO(PyLong_AS_LONG(retval));
2888 Py_DECREF(retval);
2889 break;
2890 }
2891 /* Now we have to pop the block. */
2892 f->f_iblock--;
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002893
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002894 if (b->b_type == EXCEPT_HANDLER) {
2895 UNWIND_EXCEPT_HANDLER(b);
2896 continue;
2897 }
2898 UNWIND_BLOCK(b);
2899 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2900 why = WHY_NOT;
2901 JUMPTO(b->b_handler);
2902 break;
2903 }
2904 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
2905 || b->b_type == SETUP_FINALLY)) {
2906 PyObject *exc, *val, *tb;
2907 int handler = b->b_handler;
2908 /* Beware, this invalidates all b->b_* fields */
2909 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
2910 PUSH(tstate->exc_traceback);
2911 PUSH(tstate->exc_value);
2912 if (tstate->exc_type != NULL) {
2913 PUSH(tstate->exc_type);
2914 }
2915 else {
2916 Py_INCREF(Py_None);
2917 PUSH(Py_None);
2918 }
2919 PyErr_Fetch(&exc, &val, &tb);
2920 /* Make the raw exception data
2921 available to the handler,
2922 so a program can emulate the
2923 Python main loop. */
2924 PyErr_NormalizeException(
2925 &exc, &val, &tb);
2926 PyException_SetTraceback(val, tb);
2927 Py_INCREF(exc);
2928 tstate->exc_type = exc;
2929 Py_INCREF(val);
2930 tstate->exc_value = val;
2931 tstate->exc_traceback = tb;
2932 if (tb == NULL)
2933 tb = Py_None;
2934 Py_INCREF(tb);
2935 PUSH(tb);
2936 PUSH(val);
2937 PUSH(exc);
2938 why = WHY_NOT;
2939 JUMPTO(handler);
2940 break;
2941 }
2942 if (b->b_type == SETUP_FINALLY) {
2943 if (why & (WHY_RETURN | WHY_CONTINUE))
2944 PUSH(retval);
2945 PUSH(PyLong_FromLong((long)why));
2946 why = WHY_NOT;
2947 JUMPTO(b->b_handler);
2948 break;
2949 }
2950 } /* unwind stack */
Guido van Rossum374a9221991-04-04 10:40:29 +00002951
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002952 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002953
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002954 if (why != WHY_NOT)
2955 break;
2956 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002957
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002958 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002959
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002960 assert(why != WHY_YIELD);
2961 /* Pop remaining stack entries. */
2962 while (!EMPTY()) {
2963 v = POP();
2964 Py_XDECREF(v);
2965 }
Guido van Rossum35974fb2001-12-06 21:28:18 +00002966
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002967 if (why != WHY_RETURN)
2968 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002969
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002970fast_yield:
Benjamin Petersonac913412011-07-03 16:25:11 -05002971 if (co->co_flags & CO_GENERATOR && (why == WHY_YIELD || why == WHY_RETURN)) {
2972 /* The purpose of this block is to put aside the generator's exception
2973 state and restore that of the calling frame. If the current
2974 exception state is from the caller, we clear the exception values
2975 on the generator frame, so they are not swapped back in latter. The
2976 origin of the current exception state is determined by checking for
2977 except handler blocks, which we must be in iff a new exception
2978 state came into existence in this frame. (An uncaught exception
2979 would have why == WHY_EXCEPTION, and we wouldn't be here). */
2980 int i;
2981 for (i = 0; i < f->f_iblock; i++)
2982 if (f->f_blockstack[i].b_type == EXCEPT_HANDLER)
2983 break;
2984 if (i == f->f_iblock)
2985 /* We did not create this exception. */
Benjamin Peterson87880242011-07-03 16:48:31 -05002986 restore_and_clear_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05002987 else
Benjamin Peterson87880242011-07-03 16:48:31 -05002988 swap_exc_state(tstate, f);
Benjamin Petersonac913412011-07-03 16:25:11 -05002989 }
Benjamin Peterson83195c32011-07-03 13:44:00 -05002990
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002991 if (tstate->use_tracing) {
2992 if (tstate->c_tracefunc) {
2993 if (why == WHY_RETURN || why == WHY_YIELD) {
2994 if (call_trace(tstate->c_tracefunc,
2995 tstate->c_traceobj, f,
2996 PyTrace_RETURN, retval)) {
2997 Py_XDECREF(retval);
2998 retval = NULL;
2999 why = WHY_EXCEPTION;
3000 }
3001 }
3002 else if (why == WHY_EXCEPTION) {
3003 call_trace_protected(tstate->c_tracefunc,
3004 tstate->c_traceobj, f,
3005 PyTrace_RETURN, NULL);
3006 }
3007 }
3008 if (tstate->c_profilefunc) {
3009 if (why == WHY_EXCEPTION)
3010 call_trace_protected(tstate->c_profilefunc,
3011 tstate->c_profileobj, f,
3012 PyTrace_RETURN, NULL);
3013 else if (call_trace(tstate->c_profilefunc,
3014 tstate->c_profileobj, f,
3015 PyTrace_RETURN, retval)) {
3016 Py_XDECREF(retval);
3017 retval = NULL;
Brett Cannonb94767f2011-02-22 20:15:44 +00003018 /* why = WHY_EXCEPTION; */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003019 }
3020 }
3021 }
Guido van Rossuma4240131997-01-21 21:18:36 +00003022
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003023 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00003024exit_eval_frame:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003025 Py_LeaveRecursiveCall();
3026 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00003027
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003028 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00003029}
3030
Benjamin Petersonb204a422011-06-05 22:04:07 -05003031static void
Benjamin Petersone109c702011-06-24 09:37:26 -05003032format_missing(const char *kind, PyCodeObject *co, PyObject *names)
3033{
3034 int err;
3035 Py_ssize_t len = PyList_GET_SIZE(names);
3036 PyObject *name_str, *comma, *tail, *tmp;
3037
3038 assert(PyList_CheckExact(names));
3039 assert(len >= 1);
3040 /* Deal with the joys of natural language. */
3041 switch (len) {
3042 case 1:
3043 name_str = PyList_GET_ITEM(names, 0);
3044 Py_INCREF(name_str);
3045 break;
3046 case 2:
3047 name_str = PyUnicode_FromFormat("%U and %U",
3048 PyList_GET_ITEM(names, len - 2),
3049 PyList_GET_ITEM(names, len - 1));
3050 break;
3051 default:
3052 tail = PyUnicode_FromFormat(", %U, and %U",
3053 PyList_GET_ITEM(names, len - 2),
3054 PyList_GET_ITEM(names, len - 1));
3055 /* Chop off the last two objects in the list. This shouldn't actually
3056 fail, but we can't be too careful. */
3057 err = PyList_SetSlice(names, len - 2, len, NULL);
3058 if (err == -1) {
3059 Py_DECREF(tail);
3060 return;
3061 }
3062 /* Stitch everything up into a nice comma-separated list. */
3063 comma = PyUnicode_FromString(", ");
3064 if (comma == NULL) {
3065 Py_DECREF(tail);
3066 return;
3067 }
3068 tmp = PyUnicode_Join(comma, names);
3069 Py_DECREF(comma);
3070 if (tmp == NULL) {
3071 Py_DECREF(tail);
3072 return;
3073 }
3074 name_str = PyUnicode_Concat(tmp, tail);
3075 Py_DECREF(tmp);
3076 Py_DECREF(tail);
3077 break;
3078 }
3079 if (name_str == NULL)
3080 return;
3081 PyErr_Format(PyExc_TypeError,
3082 "%U() missing %i required %s argument%s: %U",
3083 co->co_name,
3084 len,
3085 kind,
3086 len == 1 ? "" : "s",
3087 name_str);
3088 Py_DECREF(name_str);
3089}
3090
3091static void
3092missing_arguments(PyCodeObject *co, int missing, int defcount,
3093 PyObject **fastlocals)
3094{
3095 int i, j = 0;
3096 int start, end;
3097 int positional = defcount != -1;
3098 const char *kind = positional ? "positional" : "keyword-only";
3099 PyObject *missing_names;
3100
3101 /* Compute the names of the arguments that are missing. */
3102 missing_names = PyList_New(missing);
3103 if (missing_names == NULL)
3104 return;
3105 if (positional) {
3106 start = 0;
3107 end = co->co_argcount - defcount;
3108 }
3109 else {
3110 start = co->co_argcount;
3111 end = start + co->co_kwonlyargcount;
3112 }
3113 for (i = start; i < end; i++) {
3114 if (GETLOCAL(i) == NULL) {
3115 PyObject *raw = PyTuple_GET_ITEM(co->co_varnames, i);
3116 PyObject *name = PyObject_Repr(raw);
3117 if (name == NULL) {
3118 Py_DECREF(missing_names);
3119 return;
3120 }
3121 PyList_SET_ITEM(missing_names, j++, name);
3122 }
3123 }
3124 assert(j == missing);
3125 format_missing(kind, co, missing_names);
3126 Py_DECREF(missing_names);
3127}
3128
3129static void
3130too_many_positional(PyCodeObject *co, int given, int defcount, PyObject **fastlocals)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003131{
3132 int plural;
3133 int kwonly_given = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003134 int i;
3135 PyObject *sig, *kwonly_sig;
3136
Benjamin Petersone109c702011-06-24 09:37:26 -05003137 assert((co->co_flags & CO_VARARGS) == 0);
3138 /* Count missing keyword-only args. */
Benjamin Petersonb204a422011-06-05 22:04:07 -05003139 for (i = co->co_argcount; i < co->co_argcount + co->co_kwonlyargcount; i++)
Benjamin Petersone109c702011-06-24 09:37:26 -05003140 if (GETLOCAL(i) != NULL)
Benjamin Petersonb204a422011-06-05 22:04:07 -05003141 kwonly_given++;
Benjamin Petersone109c702011-06-24 09:37:26 -05003142 if (defcount) {
3143 int atleast = co->co_argcount - defcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003144 plural = 1;
3145 sig = PyUnicode_FromFormat("from %d to %d", atleast, co->co_argcount);
3146 }
3147 else {
3148 plural = co->co_argcount != 1;
3149 sig = PyUnicode_FromFormat("%d", co->co_argcount);
3150 }
3151 if (sig == NULL)
3152 return;
3153 if (kwonly_given) {
3154 const char *format = " positional argument%s (and %d keyword-only argument%s)";
3155 kwonly_sig = PyUnicode_FromFormat(format, given != 1 ? "s" : "", kwonly_given,
3156 kwonly_given != 1 ? "s" : "");
3157 if (kwonly_sig == NULL) {
3158 Py_DECREF(sig);
3159 return;
3160 }
3161 }
3162 else {
3163 /* This will not fail. */
3164 kwonly_sig = PyUnicode_FromString("");
Benjamin Petersone109c702011-06-24 09:37:26 -05003165 assert(kwonly_sig != NULL);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003166 }
3167 PyErr_Format(PyExc_TypeError,
3168 "%U() takes %U positional argument%s but %d%U %s given",
3169 co->co_name,
3170 sig,
3171 plural ? "s" : "",
3172 given,
3173 kwonly_sig,
3174 given == 1 && !kwonly_given ? "was" : "were");
3175 Py_DECREF(sig);
3176 Py_DECREF(kwonly_sig);
3177}
3178
Guido van Rossumc2e20742006-02-27 22:32:47 +00003179/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00003180 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00003181 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00003182
Tim Peters6d6c1a32001-08-02 04:15:00 +00003183PyObject *
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003184PyEval_EvalCodeEx(PyObject *_co, PyObject *globals, PyObject *locals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003185 PyObject **args, int argcount, PyObject **kws, int kwcount,
3186 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00003187{
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00003188 PyCodeObject* co = (PyCodeObject*)_co;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003189 register PyFrameObject *f;
3190 register PyObject *retval = NULL;
3191 register PyObject **fastlocals, **freevars;
3192 PyThreadState *tstate = PyThreadState_GET();
3193 PyObject *x, *u;
3194 int total_args = co->co_argcount + co->co_kwonlyargcount;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003195 int i;
3196 int n = argcount;
3197 PyObject *kwdict = NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003198
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003199 if (globals == NULL) {
3200 PyErr_SetString(PyExc_SystemError,
3201 "PyEval_EvalCodeEx: NULL globals");
3202 return NULL;
3203 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003204
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003205 assert(tstate != NULL);
3206 assert(globals != NULL);
3207 f = PyFrame_New(tstate, co, globals, locals);
3208 if (f == NULL)
3209 return NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +00003210
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003211 fastlocals = f->f_localsplus;
3212 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00003213
Benjamin Petersonb204a422011-06-05 22:04:07 -05003214 /* Parse arguments. */
3215 if (co->co_flags & CO_VARKEYWORDS) {
3216 kwdict = PyDict_New();
3217 if (kwdict == NULL)
3218 goto fail;
3219 i = total_args;
3220 if (co->co_flags & CO_VARARGS)
3221 i++;
3222 SETLOCAL(i, kwdict);
3223 }
3224 if (argcount > co->co_argcount)
3225 n = co->co_argcount;
3226 for (i = 0; i < n; i++) {
3227 x = args[i];
3228 Py_INCREF(x);
3229 SETLOCAL(i, x);
3230 }
3231 if (co->co_flags & CO_VARARGS) {
3232 u = PyTuple_New(argcount - n);
3233 if (u == NULL)
3234 goto fail;
3235 SETLOCAL(total_args, u);
3236 for (i = n; i < argcount; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003237 x = args[i];
3238 Py_INCREF(x);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003239 PyTuple_SET_ITEM(u, i-n, x);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003240 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003241 }
3242 for (i = 0; i < kwcount; i++) {
3243 PyObject **co_varnames;
3244 PyObject *keyword = kws[2*i];
3245 PyObject *value = kws[2*i + 1];
3246 int j;
3247 if (keyword == NULL || !PyUnicode_Check(keyword)) {
3248 PyErr_Format(PyExc_TypeError,
3249 "%U() keywords must be strings",
3250 co->co_name);
3251 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003252 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003253 /* Speed hack: do raw pointer compares. As names are
3254 normally interned this should almost always hit. */
3255 co_varnames = ((PyTupleObject *)(co->co_varnames))->ob_item;
3256 for (j = 0; j < total_args; j++) {
3257 PyObject *nm = co_varnames[j];
3258 if (nm == keyword)
3259 goto kw_found;
3260 }
3261 /* Slow fallback, just in case */
3262 for (j = 0; j < total_args; j++) {
3263 PyObject *nm = co_varnames[j];
3264 int cmp = PyObject_RichCompareBool(
3265 keyword, nm, Py_EQ);
3266 if (cmp > 0)
3267 goto kw_found;
3268 else if (cmp < 0)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003269 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003270 }
3271 if (j >= total_args && kwdict == NULL) {
3272 PyErr_Format(PyExc_TypeError,
3273 "%U() got an unexpected "
3274 "keyword argument '%S'",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003275 co->co_name,
3276 keyword);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003277 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003278 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003279 PyDict_SetItem(kwdict, keyword, value);
3280 continue;
3281 kw_found:
3282 if (GETLOCAL(j) != NULL) {
3283 PyErr_Format(PyExc_TypeError,
3284 "%U() got multiple "
3285 "values for argument '%S'",
3286 co->co_name,
3287 keyword);
3288 goto fail;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003289 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003290 Py_INCREF(value);
3291 SETLOCAL(j, value);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003292 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003293 if (argcount > co->co_argcount && !(co->co_flags & CO_VARARGS)) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003294 too_many_positional(co, argcount, defcount, fastlocals);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003295 goto fail;
3296 }
Benjamin Petersonb204a422011-06-05 22:04:07 -05003297 if (argcount < co->co_argcount) {
3298 int m = co->co_argcount - defcount;
Benjamin Petersone109c702011-06-24 09:37:26 -05003299 int missing = 0;
3300 for (i = argcount; i < m; i++)
3301 if (GETLOCAL(i) == NULL)
3302 missing++;
3303 if (missing) {
3304 missing_arguments(co, missing, defcount, fastlocals);
3305 goto fail;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003306 }
3307 if (n > m)
3308 i = n - m;
3309 else
3310 i = 0;
3311 for (; i < defcount; i++) {
3312 if (GETLOCAL(m+i) == NULL) {
3313 PyObject *def = defs[i];
3314 Py_INCREF(def);
3315 SETLOCAL(m+i, def);
3316 }
3317 }
3318 }
3319 if (co->co_kwonlyargcount > 0) {
Benjamin Petersone109c702011-06-24 09:37:26 -05003320 int missing = 0;
Benjamin Petersonb204a422011-06-05 22:04:07 -05003321 for (i = co->co_argcount; i < total_args; i++) {
3322 PyObject *name;
3323 if (GETLOCAL(i) != NULL)
3324 continue;
3325 name = PyTuple_GET_ITEM(co->co_varnames, i);
3326 if (kwdefs != NULL) {
3327 PyObject *def = PyDict_GetItem(kwdefs, name);
3328 if (def) {
3329 Py_INCREF(def);
3330 SETLOCAL(i, def);
3331 continue;
3332 }
3333 }
Benjamin Petersone109c702011-06-24 09:37:26 -05003334 missing++;
3335 }
3336 if (missing) {
3337 missing_arguments(co, missing, -1, fastlocals);
Benjamin Petersonb204a422011-06-05 22:04:07 -05003338 goto fail;
3339 }
3340 }
3341
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003342 /* Allocate and initialize storage for cell vars, and copy free
Benjamin Peterson90037602011-06-25 22:54:45 -05003343 vars into frame. */
3344 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003345 PyObject *c;
Benjamin Peterson90037602011-06-25 22:54:45 -05003346 int arg;
3347 /* Possibly account for the cell variable being an argument. */
3348 if (co->co_cell2arg != NULL &&
3349 (arg = co->co_cell2arg[i]) != CO_CELL_NOT_AN_ARG)
3350 c = PyCell_New(GETLOCAL(arg));
3351 else
3352 c = PyCell_New(NULL);
3353 if (c == NULL)
3354 goto fail;
3355 SETLOCAL(co->co_nlocals + i, c);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003356 }
Benjamin Peterson90037602011-06-25 22:54:45 -05003357 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
3358 PyObject *o = PyTuple_GET_ITEM(closure, i);
3359 Py_INCREF(o);
3360 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003361 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003362
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003363 if (co->co_flags & CO_GENERATOR) {
3364 /* Don't need to keep the reference to f_back, it will be set
3365 * when the generator is resumed. */
3366 Py_XDECREF(f->f_back);
3367 f->f_back = NULL;
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003368
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003369 PCALL(PCALL_GENERATOR);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003370
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003371 /* Create a new generator that owns the ready to run frame
3372 * and return that as the value. */
3373 return PyGen_New(f);
3374 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003375
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003376 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003377
Thomas Woutersce272b62007-09-19 21:19:28 +00003378fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003379
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003380 /* decref'ing the frame can cause __del__ methods to get invoked,
3381 which can call back into Python. While we're done with the
3382 current Python frame (f), the associated C stack is still in use,
3383 so recursion_depth must be boosted for the duration.
3384 */
3385 assert(tstate != NULL);
3386 ++tstate->recursion_depth;
3387 Py_DECREF(f);
3388 --tstate->recursion_depth;
3389 return retval;
Tim Peters5ca576e2001-06-18 22:08:13 +00003390}
3391
3392
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003393static PyObject *
3394special_lookup(PyObject *o, char *meth, PyObject **cache)
3395{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003396 PyObject *res;
3397 res = _PyObject_LookupSpecial(o, meth, cache);
3398 if (res == NULL && !PyErr_Occurred()) {
3399 PyErr_SetObject(PyExc_AttributeError, *cache);
3400 return NULL;
3401 }
3402 return res;
Benjamin Peterson876b2f22009-06-28 03:18:59 +00003403}
3404
3405
Benjamin Peterson87880242011-07-03 16:48:31 -05003406/* These 3 functions deal with the exception state of generators. */
3407
3408static void
3409save_exc_state(PyThreadState *tstate, PyFrameObject *f)
3410{
3411 PyObject *type, *value, *traceback;
3412 Py_XINCREF(tstate->exc_type);
3413 Py_XINCREF(tstate->exc_value);
3414 Py_XINCREF(tstate->exc_traceback);
3415 type = f->f_exc_type;
3416 value = f->f_exc_value;
3417 traceback = f->f_exc_traceback;
3418 f->f_exc_type = tstate->exc_type;
3419 f->f_exc_value = tstate->exc_value;
3420 f->f_exc_traceback = tstate->exc_traceback;
3421 Py_XDECREF(type);
3422 Py_XDECREF(value);
Victor Stinnerd2a915d2011-10-02 20:34:20 +02003423 Py_XDECREF(traceback);
Benjamin Peterson87880242011-07-03 16:48:31 -05003424}
3425
3426static void
3427swap_exc_state(PyThreadState *tstate, PyFrameObject *f)
3428{
3429 PyObject *tmp;
3430 tmp = tstate->exc_type;
3431 tstate->exc_type = f->f_exc_type;
3432 f->f_exc_type = tmp;
3433 tmp = tstate->exc_value;
3434 tstate->exc_value = f->f_exc_value;
3435 f->f_exc_value = tmp;
3436 tmp = tstate->exc_traceback;
3437 tstate->exc_traceback = f->f_exc_traceback;
3438 f->f_exc_traceback = tmp;
3439}
3440
3441static void
3442restore_and_clear_exc_state(PyThreadState *tstate, PyFrameObject *f)
3443{
3444 PyObject *type, *value, *tb;
3445 type = tstate->exc_type;
3446 value = tstate->exc_value;
3447 tb = tstate->exc_traceback;
3448 tstate->exc_type = f->f_exc_type;
3449 tstate->exc_value = f->f_exc_value;
3450 tstate->exc_traceback = f->f_exc_traceback;
3451 f->f_exc_type = NULL;
3452 f->f_exc_value = NULL;
3453 f->f_exc_traceback = NULL;
3454 Py_XDECREF(type);
3455 Py_XDECREF(value);
3456 Py_XDECREF(tb);
3457}
3458
3459
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003460/* Logic for the raise statement (too complicated for inlining).
3461 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00003462static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00003463do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003464{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003465 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00003466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003467 if (exc == NULL) {
3468 /* Reraise */
3469 PyThreadState *tstate = PyThreadState_GET();
3470 PyObject *tb;
3471 type = tstate->exc_type;
3472 value = tstate->exc_value;
3473 tb = tstate->exc_traceback;
3474 if (type == Py_None) {
3475 PyErr_SetString(PyExc_RuntimeError,
3476 "No active exception to reraise");
3477 return WHY_EXCEPTION;
3478 }
3479 Py_XINCREF(type);
3480 Py_XINCREF(value);
3481 Py_XINCREF(tb);
3482 PyErr_Restore(type, value, tb);
3483 return WHY_RERAISE;
3484 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003485
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003486 /* We support the following forms of raise:
3487 raise
Collin Winter828f04a2007-08-31 00:04:24 +00003488 raise <instance>
3489 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003490
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003491 if (PyExceptionClass_Check(exc)) {
3492 type = exc;
3493 value = PyObject_CallObject(exc, NULL);
3494 if (value == NULL)
3495 goto raise_error;
Benjamin Peterson5afa03a2011-07-15 14:09:26 -05003496 if (!PyExceptionInstance_Check(value)) {
3497 PyErr_Format(PyExc_TypeError,
3498 "calling %R should have returned an instance of "
3499 "BaseException, not %R",
3500 type, Py_TYPE(value));
3501 goto raise_error;
3502 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003503 }
3504 else if (PyExceptionInstance_Check(exc)) {
3505 value = exc;
3506 type = PyExceptionInstance_Class(exc);
3507 Py_INCREF(type);
3508 }
3509 else {
3510 /* Not something you can raise. You get an exception
3511 anyway, just not what you specified :-) */
3512 Py_DECREF(exc);
3513 PyErr_SetString(PyExc_TypeError,
3514 "exceptions must derive from BaseException");
3515 goto raise_error;
3516 }
Collin Winter828f04a2007-08-31 00:04:24 +00003517
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003518 if (cause) {
3519 PyObject *fixed_cause;
3520 if (PyExceptionClass_Check(cause)) {
3521 fixed_cause = PyObject_CallObject(cause, NULL);
3522 if (fixed_cause == NULL)
3523 goto raise_error;
3524 Py_DECREF(cause);
3525 }
3526 else if (PyExceptionInstance_Check(cause)) {
3527 fixed_cause = cause;
3528 }
3529 else {
3530 PyErr_SetString(PyExc_TypeError,
3531 "exception causes must derive from "
3532 "BaseException");
3533 goto raise_error;
3534 }
3535 PyException_SetCause(value, fixed_cause);
3536 }
Collin Winter828f04a2007-08-31 00:04:24 +00003537
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003538 PyErr_SetObject(type, value);
3539 /* PyErr_SetObject incref's its arguments */
3540 Py_XDECREF(value);
3541 Py_XDECREF(type);
3542 return WHY_EXCEPTION;
Collin Winter828f04a2007-08-31 00:04:24 +00003543
3544raise_error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003545 Py_XDECREF(value);
3546 Py_XDECREF(type);
3547 Py_XDECREF(cause);
3548 return WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003549}
3550
Tim Petersd6d010b2001-06-21 02:49:55 +00003551/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00003552 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00003553
Guido van Rossum0368b722007-05-11 16:50:42 +00003554 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
3555 with a variable target.
3556*/
Tim Petersd6d010b2001-06-21 02:49:55 +00003557
Barry Warsawe42b18f1997-08-25 22:13:04 +00003558static int
Guido van Rossum0368b722007-05-11 16:50:42 +00003559unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003560{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003561 int i = 0, j = 0;
3562 Py_ssize_t ll = 0;
3563 PyObject *it; /* iter(v) */
3564 PyObject *w;
3565 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00003566
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003567 assert(v != NULL);
Tim Petersd6d010b2001-06-21 02:49:55 +00003568
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003569 it = PyObject_GetIter(v);
3570 if (it == NULL)
3571 goto Error;
Tim Petersd6d010b2001-06-21 02:49:55 +00003572
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003573 for (; i < argcnt; i++) {
3574 w = PyIter_Next(it);
3575 if (w == NULL) {
3576 /* Iterator done, via error or exhaustion. */
3577 if (!PyErr_Occurred()) {
3578 PyErr_Format(PyExc_ValueError,
3579 "need more than %d value%s to unpack",
3580 i, i == 1 ? "" : "s");
3581 }
3582 goto Error;
3583 }
3584 *--sp = w;
3585 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003586
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003587 if (argcntafter == -1) {
3588 /* We better have exhausted the iterator now. */
3589 w = PyIter_Next(it);
3590 if (w == NULL) {
3591 if (PyErr_Occurred())
3592 goto Error;
3593 Py_DECREF(it);
3594 return 1;
3595 }
3596 Py_DECREF(w);
Georg Brandl0310a832010-07-10 10:32:36 +00003597 PyErr_Format(PyExc_ValueError, "too many values to unpack "
3598 "(expected %d)", argcnt);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003599 goto Error;
3600 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003601
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003602 l = PySequence_List(it);
3603 if (l == NULL)
3604 goto Error;
3605 *--sp = l;
3606 i++;
Guido van Rossum0368b722007-05-11 16:50:42 +00003607
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003608 ll = PyList_GET_SIZE(l);
3609 if (ll < argcntafter) {
3610 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3611 argcnt + ll);
3612 goto Error;
3613 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003614
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003615 /* Pop the "after-variable" args off the list. */
3616 for (j = argcntafter; j > 0; j--, i++) {
3617 *--sp = PyList_GET_ITEM(l, ll - j);
3618 }
3619 /* Resize the list. */
3620 Py_SIZE(l) = ll - argcntafter;
3621 Py_DECREF(it);
3622 return 1;
Guido van Rossum0368b722007-05-11 16:50:42 +00003623
Tim Petersd6d010b2001-06-21 02:49:55 +00003624Error:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003625 for (; i > 0; i--, sp++)
3626 Py_DECREF(*sp);
3627 Py_XDECREF(it);
3628 return 0;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003629}
3630
3631
Guido van Rossum96a42c81992-01-12 02:29:51 +00003632#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003633static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003634prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003635{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003636 printf("%s ", str);
3637 if (PyObject_Print(v, stdout, 0) != 0)
3638 PyErr_Clear(); /* Don't know what else to do */
3639 printf("\n");
3640 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003641}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003642#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003643
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003644static void
Fred Drake5755ce62001-06-27 19:19:46 +00003645call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003646{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003647 PyObject *type, *value, *traceback, *arg;
3648 int err;
3649 PyErr_Fetch(&type, &value, &traceback);
3650 if (value == NULL) {
3651 value = Py_None;
3652 Py_INCREF(value);
3653 }
3654 arg = PyTuple_Pack(3, type, value, traceback);
3655 if (arg == NULL) {
3656 PyErr_Restore(type, value, traceback);
3657 return;
3658 }
3659 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
3660 Py_DECREF(arg);
3661 if (err == 0)
3662 PyErr_Restore(type, value, traceback);
3663 else {
3664 Py_XDECREF(type);
3665 Py_XDECREF(value);
3666 Py_XDECREF(traceback);
3667 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003668}
3669
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003670static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003671call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003672 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003673{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003674 PyObject *type, *value, *traceback;
3675 int err;
3676 PyErr_Fetch(&type, &value, &traceback);
3677 err = call_trace(func, obj, frame, what, arg);
3678 if (err == 0)
3679 {
3680 PyErr_Restore(type, value, traceback);
3681 return 0;
3682 }
3683 else {
3684 Py_XDECREF(type);
3685 Py_XDECREF(value);
3686 Py_XDECREF(traceback);
3687 return -1;
3688 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003689}
3690
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003691static int
Fred Drake5755ce62001-06-27 19:19:46 +00003692call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003693 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003694{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003695 register PyThreadState *tstate = frame->f_tstate;
3696 int result;
3697 if (tstate->tracing)
3698 return 0;
3699 tstate->tracing++;
3700 tstate->use_tracing = 0;
3701 result = func(obj, frame, what, arg);
3702 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3703 || (tstate->c_profilefunc != NULL));
3704 tstate->tracing--;
3705 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003706}
3707
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003708PyObject *
3709_PyEval_CallTracing(PyObject *func, PyObject *args)
3710{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003711 PyFrameObject *frame = PyEval_GetFrame();
3712 PyThreadState *tstate = frame->f_tstate;
3713 int save_tracing = tstate->tracing;
3714 int save_use_tracing = tstate->use_tracing;
3715 PyObject *result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003716
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003717 tstate->tracing = 0;
3718 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3719 || (tstate->c_profilefunc != NULL));
3720 result = PyObject_Call(func, args, NULL);
3721 tstate->tracing = save_tracing;
3722 tstate->use_tracing = save_use_tracing;
3723 return result;
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003724}
3725
Alexandre Vassalotti7b82b402009-07-21 04:30:03 +00003726/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Michael W. Hudson006c7522002-11-08 13:08:46 +00003727static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003728maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003729 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3730 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003731{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003732 int result = 0;
3733 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003734
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003735 /* If the last instruction executed isn't in the current
3736 instruction window, reset the window.
3737 */
3738 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
3739 PyAddrPair bounds;
3740 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3741 &bounds);
3742 *instr_lb = bounds.ap_lower;
3743 *instr_ub = bounds.ap_upper;
3744 }
3745 /* If the last instruction falls at the start of a line or if
3746 it represents a jump backwards, update the frame's line
3747 number and call the trace function. */
3748 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3749 frame->f_lineno = line;
3750 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
3751 }
3752 *instr_prev = frame->f_lasti;
3753 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003754}
3755
Fred Drake5755ce62001-06-27 19:19:46 +00003756void
3757PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003758{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003759 PyThreadState *tstate = PyThreadState_GET();
3760 PyObject *temp = tstate->c_profileobj;
3761 Py_XINCREF(arg);
3762 tstate->c_profilefunc = NULL;
3763 tstate->c_profileobj = NULL;
3764 /* Must make sure that tracing is not ignored if 'temp' is freed */
3765 tstate->use_tracing = tstate->c_tracefunc != NULL;
3766 Py_XDECREF(temp);
3767 tstate->c_profilefunc = func;
3768 tstate->c_profileobj = arg;
3769 /* Flag that tracing or profiling is turned on */
3770 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003771}
3772
3773void
3774PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3775{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003776 PyThreadState *tstate = PyThreadState_GET();
3777 PyObject *temp = tstate->c_traceobj;
3778 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
3779 Py_XINCREF(arg);
3780 tstate->c_tracefunc = NULL;
3781 tstate->c_traceobj = NULL;
3782 /* Must make sure that profiling is not ignored if 'temp' is freed */
3783 tstate->use_tracing = tstate->c_profilefunc != NULL;
3784 Py_XDECREF(temp);
3785 tstate->c_tracefunc = func;
3786 tstate->c_traceobj = arg;
3787 /* Flag that tracing or profiling is turned on */
3788 tstate->use_tracing = ((func != NULL)
3789 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003790}
3791
Guido van Rossumb209a111997-04-29 18:18:01 +00003792PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003793PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003794{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003795 PyFrameObject *current_frame = PyEval_GetFrame();
3796 if (current_frame == NULL)
3797 return PyThreadState_GET()->interp->builtins;
3798 else
3799 return current_frame->f_builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003800}
3801
Guido van Rossumb209a111997-04-29 18:18:01 +00003802PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003803PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003804{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003805 PyFrameObject *current_frame = PyEval_GetFrame();
3806 if (current_frame == NULL)
3807 return NULL;
3808 PyFrame_FastToLocals(current_frame);
3809 return current_frame->f_locals;
Guido van Rossum5b722181993-03-30 17:46:03 +00003810}
3811
Guido van Rossumb209a111997-04-29 18:18:01 +00003812PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003813PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003814{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003815 PyFrameObject *current_frame = PyEval_GetFrame();
3816 if (current_frame == NULL)
3817 return NULL;
3818 else
3819 return current_frame->f_globals;
Guido van Rossum3f5da241990-12-20 15:06:42 +00003820}
3821
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003822PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003823PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003824{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003825 PyThreadState *tstate = PyThreadState_GET();
3826 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003827}
3828
Guido van Rossum6135a871995-01-09 17:53:26 +00003829int
Tim Peters5ba58662001-07-16 02:29:45 +00003830PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003831{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003832 PyFrameObject *current_frame = PyEval_GetFrame();
3833 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003834
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003835 if (current_frame != NULL) {
3836 const int codeflags = current_frame->f_code->co_flags;
3837 const int compilerflags = codeflags & PyCF_MASK;
3838 if (compilerflags) {
3839 result = 1;
3840 cf->cf_flags |= compilerflags;
3841 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003842#if 0 /* future keyword */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003843 if (codeflags & CO_GENERATOR_ALLOWED) {
3844 result = 1;
3845 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3846 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003847#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003848 }
3849 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003850}
3851
Guido van Rossum3f5da241990-12-20 15:06:42 +00003852
Guido van Rossum681d79a1995-07-18 14:51:37 +00003853/* External interface to call any callable object.
Antoine Pitrou8689a102010-04-01 16:53:15 +00003854 The arg must be a tuple or NULL. The kw must be a dict or NULL. */
Guido van Rossume59214e1994-08-30 08:01:59 +00003855
Guido van Rossumb209a111997-04-29 18:18:01 +00003856PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003857PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003858{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003859 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003860
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003861 if (arg == NULL) {
3862 arg = PyTuple_New(0);
3863 if (arg == NULL)
3864 return NULL;
3865 }
3866 else if (!PyTuple_Check(arg)) {
3867 PyErr_SetString(PyExc_TypeError,
3868 "argument list must be a tuple");
3869 return NULL;
3870 }
3871 else
3872 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003873
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003874 if (kw != NULL && !PyDict_Check(kw)) {
3875 PyErr_SetString(PyExc_TypeError,
3876 "keyword list must be a dictionary");
3877 Py_DECREF(arg);
3878 return NULL;
3879 }
Guido van Rossume3e61c11995-08-04 04:14:47 +00003880
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003881 result = PyObject_Call(func, arg, kw);
3882 Py_DECREF(arg);
3883 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00003884}
3885
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003886const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003887PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003888{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003889 if (PyMethod_Check(func))
3890 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
3891 else if (PyFunction_Check(func))
3892 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
3893 else if (PyCFunction_Check(func))
3894 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3895 else
3896 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003897}
3898
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003899const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003900PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003901{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003902 if (PyMethod_Check(func))
3903 return "()";
3904 else if (PyFunction_Check(func))
3905 return "()";
3906 else if (PyCFunction_Check(func))
3907 return "()";
3908 else
3909 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00003910}
3911
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00003912static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003913err_args(PyObject *func, int flags, int nargs)
3914{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003915 if (flags & METH_NOARGS)
3916 PyErr_Format(PyExc_TypeError,
3917 "%.200s() takes no arguments (%d given)",
3918 ((PyCFunctionObject *)func)->m_ml->ml_name,
3919 nargs);
3920 else
3921 PyErr_Format(PyExc_TypeError,
3922 "%.200s() takes exactly one argument (%d given)",
3923 ((PyCFunctionObject *)func)->m_ml->ml_name,
3924 nargs);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003925}
3926
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003927#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003928if (tstate->use_tracing && tstate->c_profilefunc) { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003929 if (call_trace(tstate->c_profilefunc, \
3930 tstate->c_profileobj, \
3931 tstate->frame, PyTrace_C_CALL, \
3932 func)) { \
3933 x = NULL; \
3934 } \
3935 else { \
3936 x = call; \
3937 if (tstate->c_profilefunc != NULL) { \
3938 if (x == NULL) { \
3939 call_trace_protected(tstate->c_profilefunc, \
3940 tstate->c_profileobj, \
3941 tstate->frame, PyTrace_C_EXCEPTION, \
3942 func); \
3943 /* XXX should pass (type, value, tb) */ \
3944 } else { \
3945 if (call_trace(tstate->c_profilefunc, \
3946 tstate->c_profileobj, \
3947 tstate->frame, PyTrace_C_RETURN, \
3948 func)) { \
3949 Py_DECREF(x); \
3950 x = NULL; \
3951 } \
3952 } \
3953 } \
3954 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003955} else { \
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003956 x = call; \
3957 }
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003958
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003959static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003960call_function(PyObject ***pp_stack, int oparg
3961#ifdef WITH_TSC
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003962 , uint64* pintr0, uint64* pintr1
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003963#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003964 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003965{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003966 int na = oparg & 0xff;
3967 int nk = (oparg>>8) & 0xff;
3968 int n = na + 2 * nk;
3969 PyObject **pfunc = (*pp_stack) - n - 1;
3970 PyObject *func = *pfunc;
3971 PyObject *x, *w;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003972
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003973 /* Always dispatch PyCFunction first, because these are
3974 presumed to be the most frequent callable object.
3975 */
3976 if (PyCFunction_Check(func) && nk == 0) {
3977 int flags = PyCFunction_GET_FLAGS(func);
3978 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003979
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00003980 PCALL(PCALL_CFUNCTION);
3981 if (flags & (METH_NOARGS | METH_O)) {
3982 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3983 PyObject *self = PyCFunction_GET_SELF(func);
3984 if (flags & METH_NOARGS && na == 0) {
3985 C_TRACE(x, (*meth)(self,NULL));
3986 }
3987 else if (flags & METH_O && na == 1) {
3988 PyObject *arg = EXT_POP(*pp_stack);
3989 C_TRACE(x, (*meth)(self,arg));
3990 Py_DECREF(arg);
3991 }
3992 else {
3993 err_args(func, flags, na);
3994 x = NULL;
3995 }
3996 }
3997 else {
3998 PyObject *callargs;
3999 callargs = load_args(pp_stack, na);
4000 READ_TIMESTAMP(*pintr0);
4001 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
4002 READ_TIMESTAMP(*pintr1);
4003 Py_XDECREF(callargs);
4004 }
4005 } else {
4006 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
4007 /* optimize access to bound methods */
4008 PyObject *self = PyMethod_GET_SELF(func);
4009 PCALL(PCALL_METHOD);
4010 PCALL(PCALL_BOUND_METHOD);
4011 Py_INCREF(self);
4012 func = PyMethod_GET_FUNCTION(func);
4013 Py_INCREF(func);
4014 Py_DECREF(*pfunc);
4015 *pfunc = self;
4016 na++;
4017 n++;
4018 } else
4019 Py_INCREF(func);
4020 READ_TIMESTAMP(*pintr0);
4021 if (PyFunction_Check(func))
4022 x = fast_function(func, pp_stack, n, na, nk);
4023 else
4024 x = do_call(func, pp_stack, na, nk);
4025 READ_TIMESTAMP(*pintr1);
4026 Py_DECREF(func);
4027 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00004028
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004029 /* Clear the stack of the function object. Also removes
4030 the arguments in case they weren't consumed already
4031 (fast_function() and err_args() leave them on the stack).
4032 */
4033 while ((*pp_stack) > pfunc) {
4034 w = EXT_POP(*pp_stack);
4035 Py_DECREF(w);
4036 PCALL(PCALL_POP);
4037 }
4038 return x;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004039}
4040
Jeremy Hylton192690e2002-08-16 18:36:11 +00004041/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004042 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004043 For the simplest case -- a function that takes only positional
4044 arguments and is called with only positional arguments -- it
4045 inlines the most primitive frame setup code from
4046 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4047 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004048*/
4049
4050static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004051fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004052{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004053 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
4054 PyObject *globals = PyFunction_GET_GLOBALS(func);
4055 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4056 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
4057 PyObject **d = NULL;
4058 int nd = 0;
Jeremy Hylton52820442001-01-03 23:52:36 +00004059
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004060 PCALL(PCALL_FUNCTION);
4061 PCALL(PCALL_FAST_FUNCTION);
4062 if (argdefs == NULL && co->co_argcount == n &&
4063 co->co_kwonlyargcount == 0 && nk==0 &&
4064 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4065 PyFrameObject *f;
4066 PyObject *retval = NULL;
4067 PyThreadState *tstate = PyThreadState_GET();
4068 PyObject **fastlocals, **stack;
4069 int i;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004070
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004071 PCALL(PCALL_FASTER_FUNCTION);
4072 assert(globals != NULL);
4073 /* XXX Perhaps we should create a specialized
4074 PyFrame_New() that doesn't take locals, but does
4075 take builtins without sanity checking them.
4076 */
4077 assert(tstate != NULL);
4078 f = PyFrame_New(tstate, co, globals, NULL);
4079 if (f == NULL)
4080 return NULL;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004081
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004082 fastlocals = f->f_localsplus;
4083 stack = (*pp_stack) - n;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004084
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004085 for (i = 0; i < n; i++) {
4086 Py_INCREF(*stack);
4087 fastlocals[i] = *stack++;
4088 }
4089 retval = PyEval_EvalFrameEx(f,0);
4090 ++tstate->recursion_depth;
4091 Py_DECREF(f);
4092 --tstate->recursion_depth;
4093 return retval;
4094 }
4095 if (argdefs != NULL) {
4096 d = &PyTuple_GET_ITEM(argdefs, 0);
4097 nd = Py_SIZE(argdefs);
4098 }
Martin v. Löwis4d0d4712010-12-03 20:14:31 +00004099 return PyEval_EvalCodeEx((PyObject*)co, globals,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004100 (PyObject *)NULL, (*pp_stack)-n, na,
4101 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
4102 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004103}
4104
4105static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004106update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4107 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004108{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004109 PyObject *kwdict = NULL;
4110 if (orig_kwdict == NULL)
4111 kwdict = PyDict_New();
4112 else {
4113 kwdict = PyDict_Copy(orig_kwdict);
4114 Py_DECREF(orig_kwdict);
4115 }
4116 if (kwdict == NULL)
4117 return NULL;
4118 while (--nk >= 0) {
4119 int err;
4120 PyObject *value = EXT_POP(*pp_stack);
4121 PyObject *key = EXT_POP(*pp_stack);
4122 if (PyDict_GetItem(kwdict, key) != NULL) {
4123 PyErr_Format(PyExc_TypeError,
4124 "%.200s%s got multiple values "
4125 "for keyword argument '%U'",
4126 PyEval_GetFuncName(func),
4127 PyEval_GetFuncDesc(func),
4128 key);
4129 Py_DECREF(key);
4130 Py_DECREF(value);
4131 Py_DECREF(kwdict);
4132 return NULL;
4133 }
4134 err = PyDict_SetItem(kwdict, key, value);
4135 Py_DECREF(key);
4136 Py_DECREF(value);
4137 if (err) {
4138 Py_DECREF(kwdict);
4139 return NULL;
4140 }
4141 }
4142 return kwdict;
Jeremy Hylton52820442001-01-03 23:52:36 +00004143}
4144
4145static PyObject *
4146update_star_args(int nstack, int nstar, PyObject *stararg,
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004147 PyObject ***pp_stack)
Jeremy Hylton52820442001-01-03 23:52:36 +00004148{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004149 PyObject *callargs, *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004150
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004151 callargs = PyTuple_New(nstack + nstar);
4152 if (callargs == NULL) {
4153 return NULL;
4154 }
4155 if (nstar) {
4156 int i;
4157 for (i = 0; i < nstar; i++) {
4158 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4159 Py_INCREF(a);
4160 PyTuple_SET_ITEM(callargs, nstack + i, a);
4161 }
4162 }
4163 while (--nstack >= 0) {
4164 w = EXT_POP(*pp_stack);
4165 PyTuple_SET_ITEM(callargs, nstack, w);
4166 }
4167 return callargs;
Jeremy Hylton52820442001-01-03 23:52:36 +00004168}
4169
4170static PyObject *
4171load_args(PyObject ***pp_stack, int na)
4172{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004173 PyObject *args = PyTuple_New(na);
4174 PyObject *w;
Jeremy Hylton52820442001-01-03 23:52:36 +00004175
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004176 if (args == NULL)
4177 return NULL;
4178 while (--na >= 0) {
4179 w = EXT_POP(*pp_stack);
4180 PyTuple_SET_ITEM(args, na, w);
4181 }
4182 return args;
Jeremy Hylton52820442001-01-03 23:52:36 +00004183}
4184
4185static PyObject *
4186do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4187{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004188 PyObject *callargs = NULL;
4189 PyObject *kwdict = NULL;
4190 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004191
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004192 if (nk > 0) {
4193 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
4194 if (kwdict == NULL)
4195 goto call_fail;
4196 }
4197 callargs = load_args(pp_stack, na);
4198 if (callargs == NULL)
4199 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004200#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004201 /* At this point, we have to look at the type of func to
4202 update the call stats properly. Do it here so as to avoid
4203 exposing the call stats machinery outside ceval.c
4204 */
4205 if (PyFunction_Check(func))
4206 PCALL(PCALL_FUNCTION);
4207 else if (PyMethod_Check(func))
4208 PCALL(PCALL_METHOD);
4209 else if (PyType_Check(func))
4210 PCALL(PCALL_TYPE);
4211 else if (PyCFunction_Check(func))
4212 PCALL(PCALL_CFUNCTION);
4213 else
4214 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004215#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004216 if (PyCFunction_Check(func)) {
4217 PyThreadState *tstate = PyThreadState_GET();
4218 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4219 }
4220 else
4221 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00004222call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004223 Py_XDECREF(callargs);
4224 Py_XDECREF(kwdict);
4225 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004226}
4227
4228static PyObject *
4229ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4230{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004231 int nstar = 0;
4232 PyObject *callargs = NULL;
4233 PyObject *stararg = NULL;
4234 PyObject *kwdict = NULL;
4235 PyObject *result = NULL;
Jeremy Hylton52820442001-01-03 23:52:36 +00004236
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004237 if (flags & CALL_FLAG_KW) {
4238 kwdict = EXT_POP(*pp_stack);
4239 if (!PyDict_Check(kwdict)) {
4240 PyObject *d;
4241 d = PyDict_New();
4242 if (d == NULL)
4243 goto ext_call_fail;
4244 if (PyDict_Update(d, kwdict) != 0) {
4245 Py_DECREF(d);
4246 /* PyDict_Update raises attribute
4247 * error (percolated from an attempt
4248 * to get 'keys' attribute) instead of
4249 * a type error if its second argument
4250 * is not a mapping.
4251 */
4252 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4253 PyErr_Format(PyExc_TypeError,
4254 "%.200s%.200s argument after ** "
4255 "must be a mapping, not %.200s",
4256 PyEval_GetFuncName(func),
4257 PyEval_GetFuncDesc(func),
4258 kwdict->ob_type->tp_name);
4259 }
4260 goto ext_call_fail;
4261 }
4262 Py_DECREF(kwdict);
4263 kwdict = d;
4264 }
4265 }
4266 if (flags & CALL_FLAG_VAR) {
4267 stararg = EXT_POP(*pp_stack);
4268 if (!PyTuple_Check(stararg)) {
4269 PyObject *t = NULL;
4270 t = PySequence_Tuple(stararg);
4271 if (t == NULL) {
4272 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4273 PyErr_Format(PyExc_TypeError,
4274 "%.200s%.200s argument after * "
Victor Stinner0a5f65a2011-03-22 01:09:21 +01004275 "must be a sequence, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004276 PyEval_GetFuncName(func),
4277 PyEval_GetFuncDesc(func),
4278 stararg->ob_type->tp_name);
4279 }
4280 goto ext_call_fail;
4281 }
4282 Py_DECREF(stararg);
4283 stararg = t;
4284 }
4285 nstar = PyTuple_GET_SIZE(stararg);
4286 }
4287 if (nk > 0) {
4288 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
4289 if (kwdict == NULL)
4290 goto ext_call_fail;
4291 }
4292 callargs = update_star_args(na, nstar, stararg, pp_stack);
4293 if (callargs == NULL)
4294 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004295#ifdef CALL_PROFILE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004296 /* At this point, we have to look at the type of func to
4297 update the call stats properly. Do it here so as to avoid
4298 exposing the call stats machinery outside ceval.c
4299 */
4300 if (PyFunction_Check(func))
4301 PCALL(PCALL_FUNCTION);
4302 else if (PyMethod_Check(func))
4303 PCALL(PCALL_METHOD);
4304 else if (PyType_Check(func))
4305 PCALL(PCALL_TYPE);
4306 else if (PyCFunction_Check(func))
4307 PCALL(PCALL_CFUNCTION);
4308 else
4309 PCALL(PCALL_OTHER);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004310#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004311 if (PyCFunction_Check(func)) {
4312 PyThreadState *tstate = PyThreadState_GET();
4313 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4314 }
4315 else
4316 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00004317ext_call_fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004318 Py_XDECREF(callargs);
4319 Py_XDECREF(kwdict);
4320 Py_XDECREF(stararg);
4321 return result;
Jeremy Hylton52820442001-01-03 23:52:36 +00004322}
4323
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004324/* Extract a slice index from a PyInt or PyLong or an object with the
4325 nb_index slot defined, and store in *pi.
4326 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4327 and silently boost values less than -PY_SSIZE_T_MAX-1 to -PY_SSIZE_T_MAX-1.
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004328 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004329*/
Tim Petersb5196382001-12-16 19:44:20 +00004330/* Note: If v is NULL, return success without storing into *pi. This
4331 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4332 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004333*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004334int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004335_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004336{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004337 if (v != NULL) {
4338 Py_ssize_t x;
4339 if (PyIndex_Check(v)) {
4340 x = PyNumber_AsSsize_t(v, NULL);
4341 if (x == -1 && PyErr_Occurred())
4342 return 0;
4343 }
4344 else {
4345 PyErr_SetString(PyExc_TypeError,
4346 "slice indices must be integers or "
4347 "None or have an __index__ method");
4348 return 0;
4349 }
4350 *pi = x;
4351 }
4352 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004353}
4354
Guido van Rossum486364b2007-06-30 05:01:58 +00004355#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004356 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00004357
Guido van Rossumb209a111997-04-29 18:18:01 +00004358static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004359cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004360{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004361 int res = 0;
4362 switch (op) {
4363 case PyCmp_IS:
4364 res = (v == w);
4365 break;
4366 case PyCmp_IS_NOT:
4367 res = (v != w);
4368 break;
4369 case PyCmp_IN:
4370 res = PySequence_Contains(w, v);
4371 if (res < 0)
4372 return NULL;
4373 break;
4374 case PyCmp_NOT_IN:
4375 res = PySequence_Contains(w, v);
4376 if (res < 0)
4377 return NULL;
4378 res = !res;
4379 break;
4380 case PyCmp_EXC_MATCH:
4381 if (PyTuple_Check(w)) {
4382 Py_ssize_t i, length;
4383 length = PyTuple_Size(w);
4384 for (i = 0; i < length; i += 1) {
4385 PyObject *exc = PyTuple_GET_ITEM(w, i);
4386 if (!PyExceptionClass_Check(exc)) {
4387 PyErr_SetString(PyExc_TypeError,
4388 CANNOT_CATCH_MSG);
4389 return NULL;
4390 }
4391 }
4392 }
4393 else {
4394 if (!PyExceptionClass_Check(w)) {
4395 PyErr_SetString(PyExc_TypeError,
4396 CANNOT_CATCH_MSG);
4397 return NULL;
4398 }
4399 }
4400 res = PyErr_GivenExceptionMatches(v, w);
4401 break;
4402 default:
4403 return PyObject_RichCompare(v, w, op);
4404 }
4405 v = res ? Py_True : Py_False;
4406 Py_INCREF(v);
4407 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004408}
4409
Thomas Wouters52152252000-08-17 22:55:00 +00004410static PyObject *
4411import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004412{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004413 PyObject *x;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004414
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004415 x = PyObject_GetAttr(v, name);
4416 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
4417 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
4418 }
4419 return x;
Thomas Wouters52152252000-08-17 22:55:00 +00004420}
Guido van Rossumac7be682001-01-17 15:42:30 +00004421
Thomas Wouters52152252000-08-17 22:55:00 +00004422static int
4423import_all_from(PyObject *locals, PyObject *v)
4424{
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004425 _Py_IDENTIFIER(__all__);
4426 _Py_IDENTIFIER(__dict__);
4427 PyObject *all = _PyObject_GetAttrId(v, &PyId___all__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004428 PyObject *dict, *name, *value;
4429 int skip_leading_underscores = 0;
4430 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004431
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004432 if (all == NULL) {
4433 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4434 return -1; /* Unexpected error */
4435 PyErr_Clear();
Martin v. Löwis1c67dd92011-10-14 15:16:45 +02004436 dict = _PyObject_GetAttrId(v, &PyId___dict__);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004437 if (dict == NULL) {
4438 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4439 return -1;
4440 PyErr_SetString(PyExc_ImportError,
4441 "from-import-* object has no __dict__ and no __all__");
4442 return -1;
4443 }
4444 all = PyMapping_Keys(dict);
4445 Py_DECREF(dict);
4446 if (all == NULL)
4447 return -1;
4448 skip_leading_underscores = 1;
4449 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004450
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004451 for (pos = 0, err = 0; ; pos++) {
4452 name = PySequence_GetItem(all, pos);
4453 if (name == NULL) {
4454 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4455 err = -1;
4456 else
4457 PyErr_Clear();
4458 break;
4459 }
4460 if (skip_leading_underscores &&
4461 PyUnicode_Check(name) &&
Martin v. Löwisd63a3b82011-09-28 07:41:54 +02004462 PyUnicode_READY(name) != -1 &&
4463 PyUnicode_READ_CHAR(name, 0) == '_')
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004464 {
4465 Py_DECREF(name);
4466 continue;
4467 }
4468 value = PyObject_GetAttr(v, name);
4469 if (value == NULL)
4470 err = -1;
4471 else if (PyDict_CheckExact(locals))
4472 err = PyDict_SetItem(locals, name, value);
4473 else
4474 err = PyObject_SetItem(locals, name, value);
4475 Py_DECREF(name);
4476 Py_XDECREF(value);
4477 if (err != 0)
4478 break;
4479 }
4480 Py_DECREF(all);
4481 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004482}
4483
Guido van Rossumac7be682001-01-17 15:42:30 +00004484static void
Neal Norwitzda059e32007-08-26 05:33:45 +00004485format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00004486{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004487 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00004488
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004489 if (!obj)
4490 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004491
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004492 obj_str = _PyUnicode_AsString(obj);
4493 if (!obj_str)
4494 return;
Paul Prescode68140d2000-08-30 20:25:01 +00004495
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004496 PyErr_Format(exc, format_str, obj_str);
Paul Prescode68140d2000-08-30 20:25:01 +00004497}
Guido van Rossum950361c1997-01-24 13:49:28 +00004498
Amaury Forgeot d'Arcba117ef2010-09-10 21:39:53 +00004499static void
4500format_exc_unbound(PyCodeObject *co, int oparg)
4501{
4502 PyObject *name;
4503 /* Don't stomp existing exception */
4504 if (PyErr_Occurred())
4505 return;
4506 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
4507 name = PyTuple_GET_ITEM(co->co_cellvars,
4508 oparg);
4509 format_exc_check_arg(
4510 PyExc_UnboundLocalError,
4511 UNBOUNDLOCAL_ERROR_MSG,
4512 name);
4513 } else {
4514 name = PyTuple_GET_ITEM(co->co_freevars, oparg -
4515 PyTuple_GET_SIZE(co->co_cellvars));
4516 format_exc_check_arg(PyExc_NameError,
4517 UNBOUNDFREE_ERROR_MSG, name);
4518 }
4519}
4520
Victor Stinnerd2a915d2011-10-02 20:34:20 +02004521static PyObject *
4522unicode_concatenate(PyObject *v, PyObject *w,
4523 PyFrameObject *f, unsigned char *next_instr)
4524{
4525 PyObject *res;
4526 if (Py_REFCNT(v) == 2) {
4527 /* In the common case, there are 2 references to the value
4528 * stored in 'variable' when the += is performed: one on the
4529 * value stack (in 'v') and one still stored in the
4530 * 'variable'. We try to delete the variable now to reduce
4531 * the refcnt to 1.
4532 */
4533 switch (*next_instr) {
4534 case STORE_FAST:
4535 {
4536 int oparg = PEEKARG();
4537 PyObject **fastlocals = f->f_localsplus;
4538 if (GETLOCAL(oparg) == v)
4539 SETLOCAL(oparg, NULL);
4540 break;
4541 }
4542 case STORE_DEREF:
4543 {
4544 PyObject **freevars = (f->f_localsplus +
4545 f->f_code->co_nlocals);
4546 PyObject *c = freevars[PEEKARG()];
4547 if (PyCell_GET(c) == v)
4548 PyCell_Set(c, NULL);
4549 break;
4550 }
4551 case STORE_NAME:
4552 {
4553 PyObject *names = f->f_code->co_names;
4554 PyObject *name = GETITEM(names, PEEKARG());
4555 PyObject *locals = f->f_locals;
4556 if (PyDict_CheckExact(locals) &&
4557 PyDict_GetItem(locals, name) == v) {
4558 if (PyDict_DelItem(locals, name) != 0) {
4559 PyErr_Clear();
4560 }
4561 }
4562 break;
4563 }
4564 }
4565 }
4566 res = v;
4567 PyUnicode_Append(&res, w);
4568 return res;
4569}
4570
Guido van Rossum950361c1997-01-24 13:49:28 +00004571#ifdef DYNAMIC_EXECUTION_PROFILE
4572
Skip Montanarof118cb12001-10-15 20:51:38 +00004573static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004574getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004575{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004576 int i;
4577 PyObject *l = PyList_New(256);
4578 if (l == NULL) return NULL;
4579 for (i = 0; i < 256; i++) {
4580 PyObject *x = PyLong_FromLong(a[i]);
4581 if (x == NULL) {
4582 Py_DECREF(l);
4583 return NULL;
4584 }
4585 PyList_SetItem(l, i, x);
4586 }
4587 for (i = 0; i < 256; i++)
4588 a[i] = 0;
4589 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004590}
4591
4592PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004593_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004594{
4595#ifndef DXPAIRS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004596 return getarray(dxp);
Guido van Rossum950361c1997-01-24 13:49:28 +00004597#else
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00004598 int i;
4599 PyObject *l = PyList_New(257);
4600 if (l == NULL) return NULL;
4601 for (i = 0; i < 257; i++) {
4602 PyObject *x = getarray(dxpairs[i]);
4603 if (x == NULL) {
4604 Py_DECREF(l);
4605 return NULL;
4606 }
4607 PyList_SetItem(l, i, x);
4608 }
4609 return l;
Guido van Rossum950361c1997-01-24 13:49:28 +00004610#endif
4611}
4612
4613#endif