blob: 2ce3ec9c10858aa47cd2d1501a2b5f34adeb059d [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 Rossumff4949e1992-08-05 19:58:53 +000016#include "eval.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000017#include "opcode.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000018#include "structmember.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000019
Guido van Rossumc6004111993-11-05 10:22:19 +000020#include <ctype.h>
21
Thomas Wouters477c8d52006-05-27 19:21:47 +000022#ifndef WITH_TSC
Michael W. Hudson75eabd22005-01-18 15:56:11 +000023
24#define READ_TIMESTAMP(var)
25
26#else
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000027
28typedef unsigned long long uint64;
29
Michael W. Hudson800ba232004-08-12 18:19:17 +000030#if defined(__ppc__) /* <- Don't know if this is the correct symbol; this
Thomas Wouters8ce81f72007-09-20 18:22:40 +000031 section should work for GCC on any PowerPC
32 platform, irrespective of OS.
33 POWER? Who knows :-) */
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{
40 register unsigned long tbu, tb, tbu2;
41
42 loop:
43 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;
47
Thomas Wouters477c8d52006-05-27 19:21:47 +000048 /* The slightly peculiar way of writing the next lines is
Michael W. Hudson800ba232004-08-12 18:19:17 +000049 compiled better by GCC than any other way I tried. */
50 ((long*)(v))[0] = tbu;
51 ((long*)(v))[1] = tb;
52}
53
Michael W. Hudson75eabd22005-01-18 15:56:11 +000054#else /* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000055
Michael W. Hudson75eabd22005-01-18 15:56:11 +000056#define READ_TIMESTAMP(val) \
57 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000058
59#endif
60
Thomas Wouters477c8d52006-05-27 19:21:47 +000061void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000062 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
63{
64 uint64 intr, inst, loop;
65 PyThreadState *tstate = PyThreadState_Get();
66 if (!tstate->interp->tscdump)
67 return;
68 intr = intr1 - intr0;
69 inst = inst1 - inst0 - intr;
70 loop = loop1 - loop0 - intr;
71 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
72 opcode, ticked, inst, loop);
73}
Michael W. Hudson800ba232004-08-12 18:19:17 +000074
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000075#endif
76
Guido van Rossum04691fc1992-08-12 15:35:34 +000077/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000078/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000079
Guido van Rossum408027e1996-12-30 16:17:54 +000080#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000081/* For debugging the interpreter: */
82#define LLTRACE 1 /* Low-level trace feature */
83#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +000084#endif
85
Jeremy Hylton52820442001-01-03 23:52:36 +000086typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +000087
Guido van Rossum374a9221991-04-04 10:40:29 +000088/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000089#ifdef WITH_TSC
Thomas Wouters477c8d52006-05-27 19:21:47 +000090static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091#else
Thomas Wouters477c8d52006-05-27 19:21:47 +000092static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
Thomas Wouters477c8d52006-05-27 19:21:47 +000094static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
95static PyObject * do_call(PyObject *, PyObject ***, int, int);
96static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouters8ce81f72007-09-20 18:22:40 +000097static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
98 PyObject *);
Thomas Wouters477c8d52006-05-27 19:21:47 +000099static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
100static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000101#define CALL_FLAG_VAR 1
102#define CALL_FLAG_KW 2
103
Guido van Rossum0a066c01992-03-27 17:29:15 +0000104#ifdef LLTRACE
Guido van Rossumc2e20742006-02-27 22:32:47 +0000105static int lltrace;
Tim Petersdbd9ba62000-07-09 03:09:57 +0000106static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000107#endif
Fred Drake5755ce62001-06-27 19:19:46 +0000108static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
109 int, PyObject *);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000110static int call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000111 PyFrameObject *, int, PyObject *);
Fred Drake5755ce62001-06-27 19:19:46 +0000112static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
Tim Peters8a5c3c72004-04-05 19:36:21 +0000113static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Armin Rigobf57a142004-03-22 19:24:58 +0000114 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000115
Thomas Wouters477c8d52006-05-27 19:21:47 +0000116static PyObject * cmp_outcome(int, PyObject *, PyObject *);
117static PyObject * import_from(PyObject *, PyObject *);
Thomas Wouters52152252000-08-17 22:55:00 +0000118static int import_all_from(PyObject *, PyObject *);
Neal Norwitzda059e32007-08-26 05:33:45 +0000119static void format_exc_check_arg(PyObject *, const char *, PyObject *);
Guido van Rossum98297ee2007-11-06 21:34:58 +0000120static PyObject * unicode_concatenate(PyObject *, PyObject *,
121 PyFrameObject *, unsigned char *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000122
Paul Prescode68140d2000-08-30 20:25:01 +0000123#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000124 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000125#define GLOBAL_NAME_ERROR_MSG \
126 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000127#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000128 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000129#define UNBOUNDFREE_ERROR_MSG \
130 "free variable '%.200s' referenced before assignment" \
131 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000132
Guido van Rossum950361c1997-01-24 13:49:28 +0000133/* Dynamic execution profile */
134#ifdef DYNAMIC_EXECUTION_PROFILE
135#ifdef DXPAIRS
136static long dxpairs[257][256];
137#define dxp dxpairs[256]
138#else
139static long dxp[256];
140#endif
141#endif
142
Jeremy Hylton985eba52003-02-05 23:13:00 +0000143/* Function call profile */
144#ifdef CALL_PROFILE
145#define PCALL_NUM 11
146static int pcall[PCALL_NUM];
147
148#define PCALL_ALL 0
149#define PCALL_FUNCTION 1
150#define PCALL_FAST_FUNCTION 2
151#define PCALL_FASTER_FUNCTION 3
152#define PCALL_METHOD 4
153#define PCALL_BOUND_METHOD 5
154#define PCALL_CFUNCTION 6
155#define PCALL_TYPE 7
156#define PCALL_GENERATOR 8
157#define PCALL_OTHER 9
158#define PCALL_POP 10
159
160/* Notes about the statistics
161
162 PCALL_FAST stats
163
164 FAST_FUNCTION means no argument tuple needs to be created.
165 FASTER_FUNCTION means that the fast-path frame setup code is used.
166
167 If there is a method call where the call can be optimized by changing
168 the argument tuple and calling the function directly, it gets recorded
169 twice.
170
171 As a result, the relationship among the statistics appears to be
172 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
173 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
174 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
175 PCALL_METHOD > PCALL_BOUND_METHOD
176*/
177
178#define PCALL(POS) pcall[POS]++
179
180PyObject *
181PyEval_GetCallStats(PyObject *self)
182{
Thomas Wouters89f507f2006-12-13 04:49:30 +0000183 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000184 pcall[0], pcall[1], pcall[2], pcall[3],
185 pcall[4], pcall[5], pcall[6], pcall[7],
Thomas Wouters89f507f2006-12-13 04:49:30 +0000186 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000187}
188#else
189#define PCALL(O)
190
191PyObject *
192PyEval_GetCallStats(PyObject *self)
193{
194 Py_INCREF(Py_None);
195 return Py_None;
196}
197#endif
198
Tim Peters5ca576e2001-06-18 22:08:13 +0000199
Guido van Rossume59214e1994-08-30 08:01:59 +0000200#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000201
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000202#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000203#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000204#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000205#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000206
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000207static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Guido van Rossuma9672091994-09-14 13:31:22 +0000208static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000209
Tim Peters7f468f22004-10-11 02:40:51 +0000210int
211PyEval_ThreadsInitialized(void)
212{
213 return interpreter_lock != 0;
214}
215
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000216void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000217PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000218{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000219 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000220 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000221 interpreter_lock = PyThread_allocate_lock();
222 PyThread_acquire_lock(interpreter_lock, 1);
223 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000224}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000225
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000226void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000227PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000228{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000229 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000230}
231
232void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000233PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000234{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000235 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236}
237
238void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000239PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000240{
241 if (tstate == NULL)
242 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000243 /* Check someone has called PyEval_InitThreads() to create the lock */
244 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000245 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000246 if (PyThreadState_Swap(tstate) != NULL)
247 Py_FatalError(
248 "PyEval_AcquireThread: non-NULL old thread state");
249}
250
251void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000252PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000253{
254 if (tstate == NULL)
255 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
256 if (PyThreadState_Swap(NULL) != tstate)
257 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000258 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000259}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000260
261/* This function is called from PyOS_AfterFork to ensure that newly
262 created child processes don't hold locks referring to threads which
263 are not running in the child process. (This could also be done using
264 pthread_atfork mechanism, at least for the pthreads implementation.) */
265
266void
267PyEval_ReInitThreads(void)
268{
Jesse Nollera8513972008-07-17 16:49:17 +0000269 PyObject *threading, *result;
270 PyThreadState *tstate;
271
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000272 if (!interpreter_lock)
273 return;
274 /*XXX Can't use PyThread_free_lock here because it does too
275 much error-checking. Doing this cleanly would require
276 adding a new function to each thread_*.h. Instead, just
277 create a new lock and waste a little bit of memory */
278 interpreter_lock = PyThread_allocate_lock();
279 PyThread_acquire_lock(interpreter_lock, 1);
280 main_thread = PyThread_get_thread_ident();
Jesse Nollera8513972008-07-17 16:49:17 +0000281
282 /* Update the threading module with the new state.
283 */
284 tstate = PyThreadState_GET();
285 threading = PyMapping_GetItemString(tstate->interp->modules,
286 "threading");
287 if (threading == NULL) {
288 /* threading not imported */
289 PyErr_Clear();
290 return;
291 }
292 result = PyObject_CallMethod(threading, "_after_fork", NULL);
293 if (result == NULL)
294 PyErr_WriteUnraisable(threading);
295 else
296 Py_DECREF(result);
297 Py_DECREF(threading);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000298}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000299#endif
300
Guido van Rossumff4949e1992-08-05 19:58:53 +0000301/* Functions save_thread and restore_thread are always defined so
302 dynamically loaded modules needn't be compiled separately for use
303 with and without threads: */
304
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000305PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000306PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000307{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000308 PyThreadState *tstate = PyThreadState_Swap(NULL);
309 if (tstate == NULL)
310 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000311#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000312 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000313 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000314#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000315 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000316}
317
318void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000319PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000320{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000321 if (tstate == NULL)
322 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000323#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000324 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000325 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000326 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000327 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000328 }
329#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000330 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000331}
332
333
Guido van Rossuma9672091994-09-14 13:31:22 +0000334/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
335 signal handlers or Mac I/O completion routines) can schedule calls
336 to a function to be called synchronously.
337 The synchronous function is called with one void* argument.
338 It should return 0 for success or -1 for failure -- failure should
339 be accompanied by an exception.
340
341 If registry succeeds, the registry function returns 0; if it fails
342 (e.g. due to too many pending calls) it returns -1 (without setting
343 an exception condition).
344
345 Note that because registry may occur from within signal handlers,
346 or other asynchronous events, calling malloc() is unsafe!
347
348#ifdef WITH_THREAD
349 Any thread can schedule pending calls, but only the main thread
350 will execute them.
351#endif
352
353 XXX WARNING! ASYNCHRONOUSLY EXECUTING CODE!
354 There are two possible race conditions:
355 (1) nested asynchronous registry calls;
356 (2) registry calls made while pending calls are being processed.
357 While (1) is very unlikely, (2) is a real possibility.
358 The current code is safe against (2), but not against (1).
359 The safety against (2) is derived from the fact that only one
360 thread (the main thread) ever takes things out of the queue.
Guido van Rossuma9672091994-09-14 13:31:22 +0000361
Guido van Rossuma027efa1997-05-05 20:56:21 +0000362 XXX Darn! With the advent of thread state, we should have an array
363 of pending calls per thread in the thread state! Later...
364*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000365
Guido van Rossuma9672091994-09-14 13:31:22 +0000366#define NPENDINGCALLS 32
367static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000368 int (*func)(void *);
369 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000370} pendingcalls[NPENDINGCALLS];
371static volatile int pendingfirst = 0;
372static volatile int pendinglast = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000373static volatile int things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000374
375int
Thomas Wouters334fb892000-07-25 12:56:38 +0000376Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000377{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000378 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000379 int i, j;
380 /* XXX Begin critical section */
381 /* XXX If you want this to be safe against nested
382 XXX asynchronous calls, you'll have to work harder! */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000383 if (busy)
384 return -1;
385 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000386 i = pendinglast;
387 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000388 if (j == pendingfirst) {
389 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000390 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000391 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000392 pendingcalls[i].func = func;
393 pendingcalls[i].arg = arg;
394 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000395
396 _Py_Ticker = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000397 things_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000398 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000399 /* XXX End critical section */
400 return 0;
401}
402
Guido van Rossum180d7b41994-09-29 09:45:57 +0000403int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000404Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000405{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000406 static int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000407#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000408 if (main_thread && PyThread_get_thread_ident() != main_thread)
Guido van Rossuma9672091994-09-14 13:31:22 +0000409 return 0;
410#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000411 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000412 return 0;
413 busy = 1;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000414 things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000415 for (;;) {
416 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000417 int (*func)(void *);
418 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000419 i = pendingfirst;
420 if (i == pendinglast)
421 break; /* Queue empty */
422 func = pendingcalls[i].func;
423 arg = pendingcalls[i].arg;
424 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000425 if (func(arg) < 0) {
426 busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000427 things_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000428 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000429 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000430 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000431 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000432 return 0;
433}
434
435
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000436/* The interpreter's recursion limit */
437
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000438#ifndef Py_DEFAULT_RECURSION_LIMIT
439#define Py_DEFAULT_RECURSION_LIMIT 1000
440#endif
441static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
442int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000443
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000444int
445Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000446{
447 return recursion_limit;
448}
449
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000450void
451Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000452{
453 recursion_limit = new_limit;
Thomas Woutersce272b62007-09-19 21:19:28 +0000454 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000455}
456
Armin Rigo2b3eb402003-10-28 12:05:48 +0000457/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
458 if the recursion_depth reaches _Py_CheckRecursionLimit.
459 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
460 to guarantee that _Py_CheckRecursiveCall() is regularly called.
461 Without USE_STACKCHECK, there is no need for this. */
462int
463_Py_CheckRecursiveCall(char *where)
464{
465 PyThreadState *tstate = PyThreadState_GET();
466
467#ifdef USE_STACKCHECK
468 if (PyOS_CheckStack()) {
469 --tstate->recursion_depth;
470 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
471 return -1;
472 }
473#endif
Antoine Pitrou658fad82008-09-03 18:34:34 +0000474 _Py_CheckRecursionLimit = recursion_limit;
Martin v. Löwis5b222132007-06-10 09:51:05 +0000475 if (tstate->recursion_critical)
476 /* Somebody asked that we don't check for recursion. */
477 return 0;
478 if (tstate->overflowed) {
479 if (tstate->recursion_depth > recursion_limit + 50) {
480 /* Overflowing while handling an overflow. Give up. */
481 Py_FatalError("Cannot recover from stack overflow.");
482 }
483 return 0;
484 }
Armin Rigo2b3eb402003-10-28 12:05:48 +0000485 if (tstate->recursion_depth > recursion_limit) {
486 --tstate->recursion_depth;
Martin v. Löwis5b222132007-06-10 09:51:05 +0000487 tstate->overflowed = 1;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000488 PyErr_Format(PyExc_RuntimeError,
489 "maximum recursion depth exceeded%s",
490 where);
491 return -1;
492 }
Armin Rigo2b3eb402003-10-28 12:05:48 +0000493 return 0;
494}
495
Guido van Rossum374a9221991-04-04 10:40:29 +0000496/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000497enum why_code {
498 WHY_NOT = 0x0001, /* No error */
499 WHY_EXCEPTION = 0x0002, /* Exception occurred */
500 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
501 WHY_RETURN = 0x0008, /* 'return' statement */
502 WHY_BREAK = 0x0010, /* 'break' statement */
503 WHY_CONTINUE = 0x0020, /* 'continue' statement */
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000504 WHY_YIELD = 0x0040, /* 'yield' operator */
505 WHY_SILENCED = 0x0080 /* Exception silenced by 'with' */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000506};
Guido van Rossum374a9221991-04-04 10:40:29 +0000507
Collin Winter828f04a2007-08-31 00:04:24 +0000508static enum why_code do_raise(PyObject *, PyObject *);
Guido van Rossum0368b722007-05-11 16:50:42 +0000509static int unpack_iterable(PyObject *, int, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000510
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +0000511/* Records whether tracing is on for any thread. Counts the number of
512 threads for which tstate->c_tracefunc is non-NULL, so if the value
513 is 0, we know we don't have to check this thread's c_tracefunc.
514 This speeds up the if statement in PyEval_EvalFrameEx() after
515 fast_next_opcode*/
516static int _Py_TracingPossible = 0;
517
Skip Montanarod581d772002-09-03 20:10:45 +0000518/* for manipulating the thread switch and periodic "stuff" - used to be
519 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000520int _Py_CheckInterval = 100;
521volatile int _Py_Ticker = 100;
Guido van Rossum374a9221991-04-04 10:40:29 +0000522
Guido van Rossumb209a111997-04-29 18:18:01 +0000523PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000524PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000525{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000526 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000527 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000528 (PyObject **)NULL, 0,
529 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000530 (PyObject **)NULL, 0,
Guido van Rossum4f72a782006-10-27 23:31:49 +0000531 NULL, NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000532}
533
534
535/* Interpreter main loop */
536
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000537PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000538PyEval_EvalFrame(PyFrameObject *f) {
539 /* This is for backward compatibility with extension modules that
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000540 used this API; core interpreter code should call
541 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000542 return PyEval_EvalFrameEx(f, 0);
543}
544
545PyObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000546PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000547{
Guido van Rossum950361c1997-01-24 13:49:28 +0000548#ifdef DXPAIRS
549 int lastopcode = 0;
550#endif
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000551 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000552 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000553 register int opcode; /* Current opcode */
554 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000555 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000556 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000557 register PyObject *x; /* Result object -- NULL if error */
558 register PyObject *v; /* Temporary objects popped off stack */
559 register PyObject *w;
560 register PyObject *u;
561 register PyObject *t;
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000562 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000563 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000564 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000565 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000566
Tim Peters8a5c3c72004-04-05 19:36:21 +0000567 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000568
569 not (instr_lb <= current_bytecode_offset < instr_ub)
570
Tim Peters8a5c3c72004-04-05 19:36:21 +0000571 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000572 initial values are such as to make this false the first
573 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000574 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000575
Guido van Rossumd076c731998-10-07 19:42:25 +0000576 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000577 PyObject *names;
578 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000579#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000580 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000581 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000582#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000583
Neal Norwitza81d2202002-07-14 00:27:26 +0000584/* Tuple access macros */
585
586#ifndef Py_DEBUG
587#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
588#else
589#define GETITEM(v, i) PyTuple_GetItem((v), (i))
590#endif
591
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000592#ifdef WITH_TSC
593/* Use Pentium timestamp counter to mark certain events:
594 inst0 -- beginning of switch statement for opcode dispatch
595 inst1 -- end of switch statement (may be skipped)
596 loop0 -- the top of the mainloop
Thomas Wouters477c8d52006-05-27 19:21:47 +0000597 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000598 (may be skipped)
599 intr1 -- beginning of long interruption
600 intr2 -- end of long interruption
601
602 Many opcodes call out to helper C functions. In some cases, the
603 time in those functions should be counted towards the time for the
604 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
605 calls another Python function; there's no point in charge all the
606 bytecode executed by the called function to the caller.
607
608 It's hard to make a useful judgement statically. In the presence
609 of operator overloading, it's impossible to tell if a call will
610 execute new Python code or not.
611
612 It's a case-by-case judgement. I'll use intr1 for the following
613 cases:
614
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000615 IMPORT_STAR
616 IMPORT_FROM
617 CALL_FUNCTION (and friends)
618
619 */
620 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
621 int ticked = 0;
622
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000623 READ_TIMESTAMP(inst0);
624 READ_TIMESTAMP(inst1);
625 READ_TIMESTAMP(loop0);
626 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000627
628 /* shut up the compiler */
629 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000630#endif
631
Guido van Rossum374a9221991-04-04 10:40:29 +0000632/* Code access macros */
633
Martin v. Löwis18e16552006-02-15 17:27:45 +0000634#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000635#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000636#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000637#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000638#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000639#define JUMPBY(x) (next_instr += (x))
640
Raymond Hettingerf606f872003-03-16 03:11:04 +0000641/* OpCode prediction macros
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000642 Some opcodes tend to come in pairs thus making it possible to
643 predict the second code when the first is run. For example,
644 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
645 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000646
Georg Brandl86b2fb92008-07-16 03:43:04 +0000647 Verifying the prediction costs a single high-speed test of a register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000648 variable against a constant. If the pairing was good, then the
Georg Brandl86b2fb92008-07-16 03:43:04 +0000649 processor's own internal branch predication has a high likelihood of
650 success, resulting in a nearly zero-overhead transition to the
651 next opcode. A successful prediction saves a trip through the eval-loop
Antoine Pitrou9a2310d2008-07-25 22:39:39 +0000652 including its two unpredictable branches, the HAS_ARG test and the
Georg Brandl86b2fb92008-07-16 03:43:04 +0000653 switch-case. Combined with the processor's internal branch prediction,
Antoine Pitrou9a2310d2008-07-25 22:39:39 +0000654 a successful PREDICT has the effect of making the two opcodes run as if
Georg Brandl86b2fb92008-07-16 03:43:04 +0000655 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000656
Georg Brandl86b2fb92008-07-16 03:43:04 +0000657 If collecting opcode statistics, your choices are to either keep the
658 predictions turned-on and interpret the results as if some opcodes
659 had been combined or turn-off predictions so that the opcode frequency
660 counter updates for both opcodes.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000661*/
662
Raymond Hettingera7216982004-02-08 19:59:27 +0000663#ifdef DYNAMIC_EXECUTION_PROFILE
664#define PREDICT(op) if (0) goto PRED_##op
665#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000666#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000667#endif
668
Raymond Hettingerf606f872003-03-16 03:11:04 +0000669#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000670#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000671
Guido van Rossum374a9221991-04-04 10:40:29 +0000672/* Stack manipulation macros */
673
Martin v. Löwis18e16552006-02-15 17:27:45 +0000674/* The stack can grow at most MAXINT deep, as co_nlocals and
675 co_stacksize are ints. */
676#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000677#define EMPTY() (STACK_LEVEL() == 0)
678#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000679#define SECOND() (stack_pointer[-2])
680#define THIRD() (stack_pointer[-3])
681#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000682#define SET_TOP(v) (stack_pointer[-1] = (v))
683#define SET_SECOND(v) (stack_pointer[-2] = (v))
684#define SET_THIRD(v) (stack_pointer[-3] = (v))
685#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000686#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000687#define BASIC_PUSH(v) (*stack_pointer++ = (v))
688#define BASIC_POP() (*--stack_pointer)
689
Guido van Rossum96a42c81992-01-12 02:29:51 +0000690#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000691#define PUSH(v) { (void)(BASIC_PUSH(v), \
692 lltrace && prtrace(TOP(), "push")); \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000693 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouters8ce81f72007-09-20 18:22:40 +0000694#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
695 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000696#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
697 lltrace && prtrace(TOP(), "stackadj")); \
Thomas Wouters477c8d52006-05-27 19:21:47 +0000698 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes0449f632007-12-15 01:27:15 +0000699#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
700 prtrace((STACK_POINTER)[-1], "ext_pop")), \
701 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000702#else
703#define PUSH(v) BASIC_PUSH(v)
704#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000705#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000706#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000707#endif
708
Guido van Rossum681d79a1995-07-18 14:51:37 +0000709/* Local variable macros */
710
711#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000712
713/* The SETLOCAL() macro must not DECREF the local variable in-place and
714 then store the new value; it must copy the old value to a temporary
715 value, then store the new value, and then DECREF the temporary value.
716 This is because it is possible that during the DECREF the frame is
717 accessed by other code (e.g. a __del__ method or gc.collect()) and the
718 variable would be pointing to already-freed memory. */
719#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
720 GETLOCAL(i) = value; \
721 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000722
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000723
724#define UNWIND_BLOCK(b) \
725 while (STACK_LEVEL() > (b)->b_level) { \
726 PyObject *v = POP(); \
727 Py_XDECREF(v); \
728 }
729
730#define UNWIND_EXCEPT_HANDLER(b) \
Benjamin Peterson27d63672008-06-15 20:09:12 +0000731 { \
732 PyObject *type, *value, *traceback; \
733 assert(STACK_LEVEL() >= (b)->b_level + 3); \
734 while (STACK_LEVEL() > (b)->b_level + 3) { \
735 value = POP(); \
736 Py_XDECREF(value); \
737 } \
738 type = tstate->exc_type; \
739 value = tstate->exc_value; \
740 traceback = tstate->exc_traceback; \
741 tstate->exc_type = POP(); \
742 tstate->exc_value = POP(); \
743 tstate->exc_traceback = POP(); \
744 Py_XDECREF(type); \
745 Py_XDECREF(value); \
746 Py_XDECREF(traceback); \
747 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000748
749#define SAVE_EXC_STATE() \
750 { \
Benjamin Peterson27d63672008-06-15 20:09:12 +0000751 PyObject *type, *value, *traceback; \
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000752 Py_XINCREF(tstate->exc_type); \
753 Py_XINCREF(tstate->exc_value); \
754 Py_XINCREF(tstate->exc_traceback); \
Benjamin Peterson27d63672008-06-15 20:09:12 +0000755 type = f->f_exc_type; \
756 value = f->f_exc_value; \
757 traceback = f->f_exc_traceback; \
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000758 f->f_exc_type = tstate->exc_type; \
759 f->f_exc_value = tstate->exc_value; \
760 f->f_exc_traceback = tstate->exc_traceback; \
Benjamin Peterson27d63672008-06-15 20:09:12 +0000761 Py_XDECREF(type); \
762 Py_XDECREF(value); \
763 Py_XDECREF(traceback); \
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000764 }
765
766#define SWAP_EXC_STATE() \
767 { \
768 PyObject *tmp; \
769 tmp = tstate->exc_type; \
770 tstate->exc_type = f->f_exc_type; \
771 f->f_exc_type = tmp; \
772 tmp = tstate->exc_value; \
773 tstate->exc_value = f->f_exc_value; \
774 f->f_exc_value = tmp; \
775 tmp = tstate->exc_traceback; \
776 tstate->exc_traceback = f->f_exc_traceback; \
777 f->f_exc_traceback = tmp; \
778 }
779
Guido van Rossuma027efa1997-05-05 20:56:21 +0000780/* Start of code */
781
Tim Peters5ca576e2001-06-18 22:08:13 +0000782 if (f == NULL)
783 return NULL;
784
Armin Rigo1d313ab2003-10-25 14:33:09 +0000785 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000786 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000787 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000788
Tim Peters5ca576e2001-06-18 22:08:13 +0000789 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000790
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000791 if (tstate->use_tracing) {
792 if (tstate->c_tracefunc != NULL) {
793 /* tstate->c_tracefunc, if defined, is a
794 function that will be called on *every* entry
795 to a code block. Its return value, if not
796 None, is a function that will be called at
797 the start of each executed line of code.
798 (Actually, the function must return itself
799 in order to continue tracing.) The trace
800 functions are called with three arguments:
801 a pointer to the current frame, a string
802 indicating why the function is called, and
803 an argument which depends on the situation.
804 The global trace function is also called
805 whenever an exception is detected. */
Antoine Pitrou9a2310d2008-07-25 22:39:39 +0000806 if (call_trace_protected(tstate->c_tracefunc,
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000807 tstate->c_traceobj,
808 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000809 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000810 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000811 }
812 }
813 if (tstate->c_profilefunc != NULL) {
814 /* Similar for c_profilefunc, except it needn't
815 return itself and isn't called for "line" events */
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +0000816 if (call_trace_protected(tstate->c_profilefunc,
817 tstate->c_profileobj,
818 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000819 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000820 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000821 }
822 }
823 }
824
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000825 co = f->f_code;
826 names = co->co_names;
827 consts = co->co_consts;
828 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000829 freevars = f->f_localsplus + co->co_nlocals;
Christian Heimes72b710a2008-05-26 13:28:38 +0000830 first_instr = (unsigned char*) PyBytes_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000831 /* An explanation is in order for the next line.
832
833 f->f_lasti now refers to the index of the last instruction
834 executed. You might think this was obvious from the name, but
835 this wasn't always true before 2.3! PyFrame_New now sets
836 f->f_lasti to -1 (i.e. the index *before* the first instruction)
837 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Antoine Pitrou9a2310d2008-07-25 22:39:39 +0000838 does work. Promise.
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000839
840 When the PREDICT() macros are enabled, some opcode pairs follow in
Antoine Pitrou9a2310d2008-07-25 22:39:39 +0000841 direct succession without updating f->f_lasti. A successful
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000842 prediction effectively links the two codes together as if they
843 were a single new opcode; accordingly,f->f_lasti will point to
844 the first code in the pair (for instance, GET_ITER followed by
845 FOR_ITER is effectively a single opcode and f->f_lasti will point
846 at to the beginning of the combined pair.)
847 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000848 next_instr = first_instr + f->f_lasti + 1;
849 stack_pointer = f->f_stacktop;
850 assert(stack_pointer != NULL);
851 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
852
Benjamin Petersoneec3d712008-06-11 15:59:43 +0000853 if (f->f_code->co_flags & CO_GENERATOR) {
854 if (f->f_exc_type != NULL && f->f_exc_type != Py_None) {
855 /* We were in an except handler when we left,
856 restore the exception state which was put aside
857 (see YIELD_VALUE). */
858 SWAP_EXC_STATE();
859 }
860 else {
861 SAVE_EXC_STATE();
862 }
863 }
864
Tim Peters5ca576e2001-06-18 22:08:13 +0000865#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000866 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000867#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000868#if defined(Py_DEBUG) || defined(LLTRACE)
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +0000869 filename = _PyUnicode_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000870#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000871
Guido van Rossum374a9221991-04-04 10:40:29 +0000872 why = WHY_NOT;
873 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000874 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000875 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000876
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000877 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000878 why = WHY_EXCEPTION;
879 goto on_error;
880 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000881
Guido van Rossum374a9221991-04-04 10:40:29 +0000882 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000883#ifdef WITH_TSC
884 if (inst1 == 0) {
885 /* Almost surely, the opcode executed a break
886 or a continue, preventing inst1 from being set
887 on the way out of the loop.
888 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000889 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000890 loop1 = inst1;
891 }
892 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
893 intr0, intr1);
894 ticked = 0;
895 inst1 = 0;
896 intr0 = 0;
897 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000898 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000899#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000900 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000901 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000902
Guido van Rossuma027efa1997-05-05 20:56:21 +0000903 /* Do periodic things. Doing this every time through
904 the loop would add too much overhead, so we do it
905 only every Nth instruction. We also do it if
906 ``things_to_do'' is set, i.e. when an asynchronous
907 event needs attention (e.g. a signal handler or
908 async I/O handler); see Py_AddPendingCall() and
909 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000910
Skip Montanarod581d772002-09-03 20:10:45 +0000911 if (--_Py_Ticker < 0) {
Thomas Woutersce272b62007-09-19 21:19:28 +0000912 if (*next_instr == SETUP_FINALLY) {
913 /* Make the last opcode before
914 a try: finally: block uninterruptable. */
915 goto fast_next_opcode;
916 }
Skip Montanarod581d772002-09-03 20:10:45 +0000917 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000918 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000919#ifdef WITH_TSC
920 ticked = 1;
921#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000922 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000923 if (Py_MakePendingCalls() < 0) {
924 why = WHY_EXCEPTION;
925 goto on_error;
926 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000927 if (things_to_do)
928 /* MakePendingCalls() didn't succeed.
929 Force early re-execution of this
930 "periodic" code, possibly after
931 a thread switch */
932 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000933 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000934#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000935 if (interpreter_lock) {
936 /* Give another thread a chance */
937
Guido van Rossum25ce5661997-08-02 03:10:38 +0000938 if (PyThreadState_Swap(NULL) != tstate)
939 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000940 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000941
942 /* Other threads may run now */
943
Guido van Rossum65d5b571998-12-21 19:32:43 +0000944 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000945 if (PyThreadState_Swap(tstate) != NULL)
946 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000947
948 /* Check for thread interrupts */
949
950 if (tstate->async_exc != NULL) {
951 x = tstate->async_exc;
952 tstate->async_exc = NULL;
953 PyErr_SetNone(x);
954 Py_DECREF(x);
955 why = WHY_EXCEPTION;
956 goto on_error;
957 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000958 }
959#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000960 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000961
Neil Schemenauer63543862002-02-17 19:10:14 +0000962 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000963 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000964
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000965 /* line-by-line tracing support */
966
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +0000967 if (_Py_TracingPossible &&
968 tstate->c_tracefunc != NULL && !tstate->tracing) {
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000969 /* see maybe_call_line_trace
970 for expository comments */
971 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000972
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000973 err = maybe_call_line_trace(tstate->c_tracefunc,
974 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000975 f, &instr_lb, &instr_ub,
976 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000977 /* Reload possibly changed frame fields */
978 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000979 if (f->f_stacktop != NULL) {
980 stack_pointer = f->f_stacktop;
981 f->f_stacktop = NULL;
982 }
983 if (err) {
984 /* trace function raised an exception */
985 goto on_error;
986 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000987 }
988
989 /* Extract opcode and argument */
990
Guido van Rossum374a9221991-04-04 10:40:29 +0000991 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000992 oparg = 0; /* allows oparg to be stored in a register because
993 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000994 if (HAS_ARG(opcode))
995 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000996 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000997#ifdef DYNAMIC_EXECUTION_PROFILE
998#ifdef DXPAIRS
999 dxpairs[lastopcode][opcode]++;
1000 lastopcode = opcode;
1001#endif
1002 dxp[opcode]++;
1003#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001004
Guido van Rossum96a42c81992-01-12 02:29:51 +00001005#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +00001006 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001007
Guido van Rossum96a42c81992-01-12 02:29:51 +00001008 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001009 if (HAS_ARG(opcode)) {
1010 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001011 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001012 }
1013 else {
1014 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001015 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +00001016 }
1017 }
1018#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001019
Guido van Rossum374a9221991-04-04 10:40:29 +00001020 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001021 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001022
Guido van Rossum374a9221991-04-04 10:40:29 +00001023 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001024
Guido van Rossum374a9221991-04-04 10:40:29 +00001025 /* BEWARE!
1026 It is essential that any operation that fails sets either
1027 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1028 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001029
Guido van Rossum374a9221991-04-04 10:40:29 +00001030 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001031
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001032 case NOP:
1033 goto fast_next_opcode;
1034
Neil Schemenauer63543862002-02-17 19:10:14 +00001035 case LOAD_FAST:
1036 x = GETLOCAL(oparg);
1037 if (x != NULL) {
1038 Py_INCREF(x);
1039 PUSH(x);
1040 goto fast_next_opcode;
1041 }
1042 format_exc_check_arg(PyExc_UnboundLocalError,
1043 UNBOUNDLOCAL_ERROR_MSG,
1044 PyTuple_GetItem(co->co_varnames, oparg));
1045 break;
1046
1047 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +00001048 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +00001049 Py_INCREF(x);
1050 PUSH(x);
1051 goto fast_next_opcode;
1052
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001053 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +00001054 case STORE_FAST:
1055 v = POP();
1056 SETLOCAL(oparg, v);
1057 goto fast_next_opcode;
1058
Raymond Hettingerf606f872003-03-16 03:11:04 +00001059 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +00001060 case POP_TOP:
1061 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001062 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +00001063 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001064
Guido van Rossum374a9221991-04-04 10:40:29 +00001065 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001066 v = TOP();
1067 w = SECOND();
1068 SET_TOP(w);
1069 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001070 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001071
Guido van Rossum374a9221991-04-04 10:40:29 +00001072 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001073 v = TOP();
1074 w = SECOND();
1075 x = THIRD();
1076 SET_TOP(w);
1077 SET_SECOND(x);
1078 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001079 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001080
Thomas Wouters434d0822000-08-24 20:11:32 +00001081 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001082 u = TOP();
1083 v = SECOND();
1084 w = THIRD();
1085 x = FOURTH();
1086 SET_TOP(v);
1087 SET_SECOND(w);
1088 SET_THIRD(x);
1089 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001090 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001091
Guido van Rossum374a9221991-04-04 10:40:29 +00001092 case DUP_TOP:
1093 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001094 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001095 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001096 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001097
Thomas Wouters434d0822000-08-24 20:11:32 +00001098 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001099 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001100 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001101 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001102 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001103 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001104 STACKADJ(2);
1105 SET_TOP(x);
1106 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001107 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001108 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001109 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001110 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001111 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001112 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001113 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001114 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001115 STACKADJ(3);
1116 SET_TOP(x);
1117 SET_SECOND(w);
1118 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001119 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001120 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001121 Py_FatalError("invalid argument to DUP_TOPX"
1122 " (bytecode corruption?)");
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00001123 /* Never returns, so don't bother to set why. */
Tim Peters35ba6892000-10-11 07:04:49 +00001124 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001125
Guido van Rossum374a9221991-04-04 10:40:29 +00001126 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001127 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001128 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001129 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001130 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001131 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001132 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001133
Guido van Rossum374a9221991-04-04 10:40:29 +00001134 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001135 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001136 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001137 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001138 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001139 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001140 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001141
Guido van Rossum374a9221991-04-04 10:40:29 +00001142 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001143 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001144 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001145 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001146 if (err == 0) {
1147 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001148 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001149 continue;
1150 }
1151 else if (err > 0) {
1152 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001153 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001154 err = 0;
1155 continue;
1156 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001157 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001158 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001159
Guido van Rossum7928cd71991-10-24 14:59:31 +00001160 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001161 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001162 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001163 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001164 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001165 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001166 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001167
Guido van Rossum50564e81996-01-12 01:13:16 +00001168 case BINARY_POWER:
1169 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001170 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001171 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001172 Py_DECREF(v);
1173 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001174 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001175 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001176 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001177
Guido van Rossum374a9221991-04-04 10:40:29 +00001178 case BINARY_MULTIPLY:
1179 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001180 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001181 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001182 Py_DECREF(v);
1183 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001184 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001185 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001186 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001187
Tim Peters3caca232001-12-06 06:23:26 +00001188 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001189 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001190 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001191 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001192 Py_DECREF(v);
1193 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001194 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001195 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001196 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001197
Guido van Rossum4668b002001-08-08 05:00:18 +00001198 case BINARY_FLOOR_DIVIDE:
1199 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001200 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001201 x = PyNumber_FloorDivide(v, w);
1202 Py_DECREF(v);
1203 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001204 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001205 if (x != NULL) continue;
1206 break;
1207
Guido van Rossum374a9221991-04-04 10:40:29 +00001208 case BINARY_MODULO:
1209 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001210 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001211 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001212 Py_DECREF(v);
1213 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001214 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001215 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001216 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001217
Guido van Rossum374a9221991-04-04 10:40:29 +00001218 case BINARY_ADD:
1219 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001220 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001221 if (PyUnicode_CheckExact(v) &&
Guido van Rossum98297ee2007-11-06 21:34:58 +00001222 PyUnicode_CheckExact(w)) {
1223 x = unicode_concatenate(v, w, f, next_instr);
1224 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001225 goto skip_decref_vx;
1226 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001227 else {
Guido van Rossumc12da691997-07-17 23:12:42 +00001228 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001229 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001230 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001231 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001232 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001233 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001234 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001235 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001236
Guido van Rossum374a9221991-04-04 10:40:29 +00001237 case BINARY_SUBTRACT:
1238 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001239 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001240 x = PyNumber_Subtract(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001241 Py_DECREF(v);
1242 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001243 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001244 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001245 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001246
Guido van Rossum374a9221991-04-04 10:40:29 +00001247 case BINARY_SUBSCR:
1248 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001249 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001250 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001251 Py_DECREF(v);
1252 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001253 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001254 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001255 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001256
Guido van Rossum7928cd71991-10-24 14:59:31 +00001257 case BINARY_LSHIFT:
1258 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001259 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001260 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001261 Py_DECREF(v);
1262 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001263 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001264 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001265 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001266
Guido van Rossum7928cd71991-10-24 14:59:31 +00001267 case BINARY_RSHIFT:
1268 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001269 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001270 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001271 Py_DECREF(v);
1272 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001273 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001274 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001275 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001276
Guido van Rossum7928cd71991-10-24 14:59:31 +00001277 case BINARY_AND:
1278 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001279 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001280 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001281 Py_DECREF(v);
1282 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001283 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001284 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001285 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001286
Guido van Rossum7928cd71991-10-24 14:59:31 +00001287 case BINARY_XOR:
1288 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001289 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001290 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001291 Py_DECREF(v);
1292 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001293 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001294 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001295 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001296
Guido van Rossum7928cd71991-10-24 14:59:31 +00001297 case BINARY_OR:
1298 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001299 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001300 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001301 Py_DECREF(v);
1302 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001303 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001304 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001305 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001306
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001307 case LIST_APPEND:
1308 w = POP();
Antoine Pitrouf289ae62008-12-18 11:06:25 +00001309 v = stack_pointer[-oparg];
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001310 err = PyList_Append(v, w);
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001311 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001312 if (err == 0) {
1313 PREDICT(JUMP_ABSOLUTE);
1314 continue;
1315 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001316 break;
1317
Nick Coghlan650f0d02007-04-15 12:05:43 +00001318 case SET_ADD:
1319 w = POP();
Antoine Pitrouf289ae62008-12-18 11:06:25 +00001320 v = stack_pointer[-oparg];
Nick Coghlan650f0d02007-04-15 12:05:43 +00001321 err = PySet_Add(v, w);
Nick Coghlan650f0d02007-04-15 12:05:43 +00001322 Py_DECREF(w);
1323 if (err == 0) {
1324 PREDICT(JUMP_ABSOLUTE);
1325 continue;
1326 }
1327 break;
1328
Thomas Wouters434d0822000-08-24 20:11:32 +00001329 case INPLACE_POWER:
1330 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001331 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001332 x = PyNumber_InPlacePower(v, w, Py_None);
1333 Py_DECREF(v);
1334 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001335 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001336 if (x != NULL) continue;
1337 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001338
Thomas Wouters434d0822000-08-24 20:11:32 +00001339 case INPLACE_MULTIPLY:
1340 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001341 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001342 x = PyNumber_InPlaceMultiply(v, w);
1343 Py_DECREF(v);
1344 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001345 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001346 if (x != NULL) continue;
1347 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001348
Tim Peters54b11912001-12-25 18:49:11 +00001349 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001350 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001351 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001352 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001353 Py_DECREF(v);
1354 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001355 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001356 if (x != NULL) continue;
1357 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001358
Guido van Rossum4668b002001-08-08 05:00:18 +00001359 case INPLACE_FLOOR_DIVIDE:
1360 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001361 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001362 x = PyNumber_InPlaceFloorDivide(v, w);
1363 Py_DECREF(v);
1364 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001365 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001366 if (x != NULL) continue;
1367 break;
1368
Thomas Wouters434d0822000-08-24 20:11:32 +00001369 case INPLACE_MODULO:
1370 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001371 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001372 x = PyNumber_InPlaceRemainder(v, w);
1373 Py_DECREF(v);
1374 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001375 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001376 if (x != NULL) continue;
1377 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001378
Thomas Wouters434d0822000-08-24 20:11:32 +00001379 case INPLACE_ADD:
1380 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001381 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001382 if (PyUnicode_CheckExact(v) &&
Guido van Rossum98297ee2007-11-06 21:34:58 +00001383 PyUnicode_CheckExact(w)) {
1384 x = unicode_concatenate(v, w, f, next_instr);
1385 /* unicode_concatenate consumed the ref to v */
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001386 goto skip_decref_v;
1387 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001388 else {
Thomas Wouters434d0822000-08-24 20:11:32 +00001389 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001390 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001391 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001392 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001393 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001394 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001395 if (x != NULL) continue;
1396 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001397
Thomas Wouters434d0822000-08-24 20:11:32 +00001398 case INPLACE_SUBTRACT:
1399 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001400 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001401 x = PyNumber_InPlaceSubtract(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001402 Py_DECREF(v);
1403 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001404 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001405 if (x != NULL) continue;
1406 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001407
Thomas Wouters434d0822000-08-24 20:11:32 +00001408 case INPLACE_LSHIFT:
1409 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001410 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001411 x = PyNumber_InPlaceLshift(v, w);
1412 Py_DECREF(v);
1413 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001414 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001415 if (x != NULL) continue;
1416 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001417
Thomas Wouters434d0822000-08-24 20:11:32 +00001418 case INPLACE_RSHIFT:
1419 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001420 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001421 x = PyNumber_InPlaceRshift(v, w);
1422 Py_DECREF(v);
1423 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001424 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001425 if (x != NULL) continue;
1426 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001427
Thomas Wouters434d0822000-08-24 20:11:32 +00001428 case INPLACE_AND:
1429 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001430 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001431 x = PyNumber_InPlaceAnd(v, w);
1432 Py_DECREF(v);
1433 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001434 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001435 if (x != NULL) continue;
1436 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001437
Thomas Wouters434d0822000-08-24 20:11:32 +00001438 case INPLACE_XOR:
1439 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001440 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001441 x = PyNumber_InPlaceXor(v, w);
1442 Py_DECREF(v);
1443 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001444 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001445 if (x != NULL) continue;
1446 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001447
Thomas Wouters434d0822000-08-24 20:11:32 +00001448 case INPLACE_OR:
1449 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001450 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001451 x = PyNumber_InPlaceOr(v, w);
1452 Py_DECREF(v);
1453 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001454 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001455 if (x != NULL) continue;
1456 break;
1457
Guido van Rossum374a9221991-04-04 10:40:29 +00001458 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001459 w = TOP();
1460 v = SECOND();
1461 u = THIRD();
1462 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001463 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001464 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001465 Py_DECREF(u);
1466 Py_DECREF(v);
1467 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001468 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001469 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001470
Guido van Rossum374a9221991-04-04 10:40:29 +00001471 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001472 w = TOP();
1473 v = SECOND();
1474 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001475 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001476 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001477 Py_DECREF(v);
1478 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001479 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001480 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001481
Guido van Rossum374a9221991-04-04 10:40:29 +00001482 case PRINT_EXPR:
1483 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001484 w = PySys_GetObject("displayhook");
1485 if (w == NULL) {
1486 PyErr_SetString(PyExc_RuntimeError,
1487 "lost sys.displayhook");
1488 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001489 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001490 }
1491 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001492 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001493 if (x == NULL)
1494 err = -1;
1495 }
1496 if (err == 0) {
1497 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001498 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001499 if (w == NULL)
1500 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001501 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001502 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001503 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001504 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001505
Thomas Wouters434d0822000-08-24 20:11:32 +00001506#ifdef CASE_TOO_BIG
1507 default: switch (opcode) {
1508#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001509 case RAISE_VARARGS:
Collin Winter828f04a2007-08-31 00:04:24 +00001510 v = w = NULL;
Guido van Rossumf10570b1995-07-07 22:53:21 +00001511 switch (oparg) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00001512 case 2:
1513 v = POP(); /* cause */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001514 case 1:
1515 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001516 case 0: /* Fallthrough */
Collin Winter828f04a2007-08-31 00:04:24 +00001517 why = do_raise(w, v);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001518 break;
1519 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001520 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001521 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001522 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001523 break;
1524 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001525 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001526
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001527 case STORE_LOCALS:
1528 x = POP();
1529 v = f->f_locals;
1530 Py_XDECREF(v);
1531 f->f_locals = x;
1532 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001533
Guido van Rossum374a9221991-04-04 10:40:29 +00001534 case RETURN_VALUE:
1535 retval = POP();
1536 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001537 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001538
Tim Peters5ca576e2001-06-18 22:08:13 +00001539 case YIELD_VALUE:
1540 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001541 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001542 why = WHY_YIELD;
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001543 /* Put aside the current exception state and restore
1544 that of the calling frame. This only serves when
1545 "yield" is used inside an except handler. */
1546 SWAP_EXC_STATE();
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001547 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001548
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001549 case POP_EXCEPT:
1550 {
1551 PyTryBlock *b = PyFrame_BlockPop(f);
1552 if (b->b_type != EXCEPT_HANDLER) {
1553 PyErr_SetString(PyExc_SystemError,
1554 "popped block is not an except handler");
1555 why = WHY_EXCEPTION;
1556 break;
1557 }
1558 UNWIND_EXCEPT_HANDLER(b);
1559 }
1560 continue;
1561
Guido van Rossum374a9221991-04-04 10:40:29 +00001562 case POP_BLOCK:
1563 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001564 PyTryBlock *b = PyFrame_BlockPop(f);
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001565 UNWIND_BLOCK(b);
Guido van Rossum374a9221991-04-04 10:40:29 +00001566 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001567 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001568
Christian Heimes180510d2008-03-03 19:15:45 +00001569 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001570 case END_FINALLY:
1571 v = POP();
Christian Heimes217cfd12007-12-02 14:31:20 +00001572 if (PyLong_Check(v)) {
1573 why = (enum why_code) PyLong_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001574 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001575 if (why == WHY_RETURN ||
1576 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001577 retval = POP();
Benjamin Petersoneec3d712008-06-11 15:59:43 +00001578 if (why == WHY_SILENCED) {
1579 /* An exception was silenced by 'with', we must
1580 manually unwind the EXCEPT_HANDLER block which was
1581 created when the exception was caught, otherwise
1582 the stack will be in an inconsistent state. */
1583 PyTryBlock *b = PyFrame_BlockPop(f);
1584 if (b->b_type != EXCEPT_HANDLER) {
1585 PyErr_SetString(PyExc_SystemError,
1586 "popped block is not an except handler");
1587 why = WHY_EXCEPTION;
1588 }
1589 else {
1590 UNWIND_EXCEPT_HANDLER(b);
1591 why = WHY_NOT;
1592 }
1593 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001594 }
Brett Cannonf74225d2007-02-26 21:10:16 +00001595 else if (PyExceptionClass_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001596 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001597 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001598 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001599 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001600 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001601 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001602 else if (v != Py_None) {
1603 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001604 "'finally' pops bad exception");
1605 why = WHY_EXCEPTION;
1606 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001607 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001608 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001609
Guido van Rossum52cc1d82007-03-18 15:41:51 +00001610 case LOAD_BUILD_CLASS:
1611 x = PyDict_GetItemString(f->f_builtins,
1612 "__build_class__");
1613 if (x == NULL) {
1614 PyErr_SetString(PyExc_ImportError,
1615 "__build_class__ not found");
1616 break;
1617 }
1618 Py_INCREF(x);
1619 PUSH(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001620 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001621
Guido van Rossum374a9221991-04-04 10:40:29 +00001622 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001623 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001624 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001625 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001626 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001627 err = PyDict_SetItem(x, w, v);
1628 else
1629 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001630 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001631 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001632 break;
1633 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001634 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001635 "no locals found when storing %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001636 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001637
Guido van Rossum374a9221991-04-04 10:40:29 +00001638 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001639 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001640 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001641 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001642 format_exc_check_arg(PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001643 NAME_ERROR_MSG,
1644 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001645 break;
1646 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001647 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001648 "no locals when deleting %R", w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001649 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001650
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001651 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001652 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001653 v = POP();
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001654 if (PyTuple_CheckExact(v) &&
1655 PyTuple_GET_SIZE(v) == oparg) {
1656 PyObject **items = \
1657 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001658 while (oparg--) {
1659 w = items[oparg];
1660 Py_INCREF(w);
1661 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001662 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001663 Py_DECREF(v);
1664 continue;
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001665 } else if (PyList_CheckExact(v) &&
1666 PyList_GET_SIZE(v) == oparg) {
1667 PyObject **items = \
1668 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001669 while (oparg--) {
1670 w = items[oparg];
1671 Py_INCREF(w);
1672 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001673 }
Guido van Rossum0368b722007-05-11 16:50:42 +00001674 } else if (unpack_iterable(v, oparg, -1,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001675 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001676 stack_pointer += oparg;
Guido van Rossumd8faa362007-04-27 19:54:29 +00001677 } else {
1678 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001679 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001680 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001681 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001682 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001683
Guido van Rossum0368b722007-05-11 16:50:42 +00001684 case UNPACK_EX:
1685 {
1686 int totalargs = 1 + (oparg & 0xFF) + (oparg >> 8);
1687 v = POP();
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00001688
Guido van Rossum0368b722007-05-11 16:50:42 +00001689 if (unpack_iterable(v, oparg & 0xFF, oparg >> 8,
1690 stack_pointer + totalargs)) {
1691 stack_pointer += totalargs;
1692 } else {
1693 why = WHY_EXCEPTION;
1694 }
1695 Py_DECREF(v);
1696 break;
1697 }
1698
Guido van Rossum374a9221991-04-04 10:40:29 +00001699 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001700 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001701 v = TOP();
1702 u = SECOND();
1703 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001704 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1705 Py_DECREF(v);
1706 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001707 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001708 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001709
Guido van Rossum374a9221991-04-04 10:40:29 +00001710 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001711 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001712 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001713 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1714 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001715 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001716 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001717
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001718 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001719 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001720 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001721 err = PyDict_SetItem(f->f_globals, w, v);
1722 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001723 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001724 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001725
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001726 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001727 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001728 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001729 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001730 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001731 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001732
Guido van Rossum374a9221991-04-04 10:40:29 +00001733 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001734 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001735 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001736 PyErr_Format(PyExc_SystemError,
Guido van Rossum98297ee2007-11-06 21:34:58 +00001737 "no locals when loading %R", w);
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00001738 why = WHY_EXCEPTION;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001739 break;
1740 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001741 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001742 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001743 Py_XINCREF(x);
1744 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001745 else {
1746 x = PyObject_GetItem(v, w);
1747 if (x == NULL && PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001748 if (!PyErr_ExceptionMatches(
1749 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001750 break;
1751 PyErr_Clear();
1752 }
1753 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001754 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001755 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001756 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001757 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001758 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001759 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001760 PyExc_NameError,
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001761 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001762 break;
1763 }
1764 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001765 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001766 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001767 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001768 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001769
Guido van Rossum374a9221991-04-04 10:40:29 +00001770 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001771 w = GETITEM(names, oparg);
Neal Norwitzda059e32007-08-26 05:33:45 +00001772 if (PyUnicode_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001773 /* Inline the PyDict_GetItem() calls.
1774 WARNING: this is an extreme speed hack.
1775 Do not try this at home. */
Neal Norwitzda059e32007-08-26 05:33:45 +00001776 long hash = ((PyUnicodeObject *)w)->hash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001777 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001778 PyDictObject *d;
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001779 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001780 d = (PyDictObject *)(f->f_globals);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001781 e = d->ma_lookup(d, w, hash);
1782 if (e == NULL) {
1783 x = NULL;
1784 break;
1785 }
1786 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001787 if (x != NULL) {
1788 Py_INCREF(x);
1789 PUSH(x);
1790 continue;
1791 }
1792 d = (PyDictObject *)(f->f_builtins);
Thomas Wouters4d70c3d2006-06-08 14:42:34 +00001793 e = d->ma_lookup(d, w, hash);
1794 if (e == NULL) {
1795 x = NULL;
1796 break;
1797 }
1798 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001799 if (x != NULL) {
1800 Py_INCREF(x);
1801 PUSH(x);
1802 continue;
1803 }
1804 goto load_global_error;
1805 }
1806 }
1807 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001808 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001809 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001810 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001811 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001812 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001813 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001814 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001815 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001816 break;
1817 }
1818 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001819 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001820 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001821 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001822
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001823 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001824 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001825 if (x != NULL) {
1826 SETLOCAL(oparg, NULL);
1827 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001828 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001829 format_exc_check_arg(
1830 PyExc_UnboundLocalError,
1831 UNBOUNDLOCAL_ERROR_MSG,
1832 PyTuple_GetItem(co->co_varnames, oparg)
1833 );
1834 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001835
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001836 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001837 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001838 Py_INCREF(x);
1839 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001840 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001841 break;
1842
1843 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001844 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001845 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001846 if (w != NULL) {
1847 PUSH(w);
1848 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001849 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001850 err = -1;
1851 /* Don't stomp existing exception */
1852 if (PyErr_Occurred())
1853 break;
Thomas Wouters477c8d52006-05-27 19:21:47 +00001854 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1855 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001856 oparg);
1857 format_exc_check_arg(
1858 PyExc_UnboundLocalError,
1859 UNBOUNDLOCAL_ERROR_MSG,
1860 v);
1861 } else {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00001862 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
1863 PyTuple_GET_SIZE(co->co_cellvars));
1864 format_exc_check_arg(PyExc_NameError,
1865 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001866 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001867 break;
1868
1869 case STORE_DEREF:
1870 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001871 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001872 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001873 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001874 continue;
1875
Guido van Rossum374a9221991-04-04 10:40:29 +00001876 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001877 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001878 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001879 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001880 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001881 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001882 }
1883 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001884 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001885 }
1886 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001887
Guido van Rossum374a9221991-04-04 10:40:29 +00001888 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001889 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001890 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001891 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001892 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001893 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001894 }
1895 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001896 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001897 }
1898 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001899
Guido van Rossum86e58e22006-08-28 15:27:34 +00001900 case BUILD_SET:
1901 x = PySet_New(NULL);
1902 if (x != NULL) {
1903 for (; --oparg >= 0;) {
1904 w = POP();
1905 if (err == 0)
1906 err = PySet_Add(x, w);
1907 Py_DECREF(w);
1908 }
1909 if (err != 0) {
1910 Py_DECREF(x);
1911 break;
1912 }
1913 PUSH(x);
1914 continue;
1915 }
1916 break;
1917
Guido van Rossum374a9221991-04-04 10:40:29 +00001918 case BUILD_MAP:
Christian Heimes99170a52007-12-19 02:07:34 +00001919 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001920 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001921 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001922 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001923
Christian Heimes99170a52007-12-19 02:07:34 +00001924 case STORE_MAP:
1925 w = TOP(); /* key */
1926 u = SECOND(); /* value */
1927 v = THIRD(); /* dict */
1928 STACKADJ(-2);
1929 assert (PyDict_CheckExact(v));
1930 err = PyDict_SetItem(v, w, u); /* v[w] = u */
1931 Py_DECREF(u);
1932 Py_DECREF(w);
1933 if (err == 0) continue;
1934 break;
1935
Antoine Pitrouf289ae62008-12-18 11:06:25 +00001936 case MAP_ADD:
1937 w = TOP(); /* key */
1938 u = SECOND(); /* value */
1939 STACKADJ(-2);
1940 v = stack_pointer[-oparg]; /* dict */
1941 assert (PyDict_CheckExact(v));
1942 err = PyDict_SetItem(v, w, u); /* v[w] = u */
1943 Py_DECREF(u);
1944 Py_DECREF(w);
1945 if (err == 0) {
1946 PREDICT(JUMP_ABSOLUTE);
1947 continue;
1948 }
1949 break;
1950
Guido van Rossum374a9221991-04-04 10:40:29 +00001951 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001952 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001953 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001954 x = PyObject_GetAttr(v, w);
1955 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001956 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001957 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001958 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001959
Guido van Rossum374a9221991-04-04 10:40:29 +00001960 case COMPARE_OP:
1961 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001962 v = TOP();
Martin v. Löwis3267f872007-12-04 20:34:12 +00001963 x = cmp_outcome(oparg, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001964 Py_DECREF(v);
1965 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001966 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001967 if (x == NULL) break;
1968 PREDICT(JUMP_IF_FALSE);
1969 PREDICT(JUMP_IF_TRUE);
1970 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001971
Guido van Rossum374a9221991-04-04 10:40:29 +00001972 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001973 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001974 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001975 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001976 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00001977 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00001978 break;
1979 }
Christian Heimesbbffeb62008-01-24 09:42:52 +00001980 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001981 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001982 u = TOP();
Christian Heimes217cfd12007-12-02 14:31:20 +00001983 if (PyLong_AsLong(u) != -1 || PyErr_Occurred())
Thomas Woutersf7f438b2006-02-28 16:09:29 +00001984 w = PyTuple_Pack(5,
1985 w,
1986 f->f_globals,
1987 f->f_locals == NULL ?
1988 Py_None : f->f_locals,
1989 v,
1990 u);
1991 else
1992 w = PyTuple_Pack(4,
1993 w,
1994 f->f_globals,
1995 f->f_locals == NULL ?
1996 Py_None : f->f_locals,
1997 v);
1998 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001999 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002000 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002001 u = POP();
Christian Heimesbbffeb62008-01-24 09:42:52 +00002002 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002003 x = NULL;
2004 break;
2005 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002006 READ_TIMESTAMP(intr0);
Christian Heimesbbffeb62008-01-24 09:42:52 +00002007 v = x;
2008 x = PyEval_CallObject(v, w);
2009 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002010 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002011 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002012 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002013 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002014 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002015
Thomas Wouters52152252000-08-17 22:55:00 +00002016 case IMPORT_STAR:
2017 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002018 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002019 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002020 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002021 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002022 break;
2023 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002024 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002025 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002026 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002027 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002028 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002029 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002030 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002031
Thomas Wouters52152252000-08-17 22:55:00 +00002032 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002033 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002034 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002035 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002036 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002037 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002038 PUSH(x);
2039 if (x != NULL) continue;
2040 break;
2041
Guido van Rossum374a9221991-04-04 10:40:29 +00002042 case JUMP_FORWARD:
2043 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002044 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002045
Raymond Hettingerf606f872003-03-16 03:11:04 +00002046 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002047 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002048 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002049 if (w == Py_True) {
2050 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002051 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002052 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002053 if (w == Py_False) {
2054 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002055 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002056 }
2057 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002058 if (err > 0)
2059 err = 0;
2060 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002061 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002062 else
2063 break;
2064 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002065
Raymond Hettingerf606f872003-03-16 03:11:04 +00002066 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002067 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002068 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002069 if (w == Py_False) {
2070 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002071 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002072 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002073 if (w == Py_True) {
2074 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002075 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002076 }
2077 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002078 if (err > 0) {
2079 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002080 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002081 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002082 else if (err == 0)
2083 ;
2084 else
2085 break;
2086 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002087
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002088 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002089 case JUMP_ABSOLUTE:
2090 JUMPTO(oparg);
Guido van Rossum58da9312007-11-10 23:39:45 +00002091#if FAST_LOOPS
2092 /* Enabling this path speeds-up all while and for-loops by bypassing
2093 the per-loop checks for signals. By default, this should be turned-off
2094 because it prevents detection of a control-break in tight loops like
2095 "while 1: pass". Compile with this option turned-on when you need
2096 the speed-up and do not need break checking inside tight loops (ones
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002097 that contain only instructions ending with goto fast_next_opcode).
Guido van Rossum58da9312007-11-10 23:39:45 +00002098 */
2099 goto fast_next_opcode;
2100#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002101 continue;
Guido van Rossum58da9312007-11-10 23:39:45 +00002102#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002103
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002104 case GET_ITER:
2105 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002106 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002107 x = PyObject_GetIter(v);
2108 Py_DECREF(v);
2109 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002110 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002111 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002112 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002113 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002114 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002115 break;
2116
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002117 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002118 case FOR_ITER:
2119 /* before: [iter]; after: [iter, iter()] *or* [] */
2120 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002121 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002122 if (x != NULL) {
2123 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002124 PREDICT(STORE_FAST);
2125 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002126 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002127 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002128 if (PyErr_Occurred()) {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00002129 if (!PyErr_ExceptionMatches(
2130 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002131 break;
2132 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002133 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002134 /* iterator ended normally */
2135 x = v = POP();
2136 Py_DECREF(v);
2137 JUMPBY(oparg);
2138 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002139
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002140 case BREAK_LOOP:
2141 why = WHY_BREAK;
2142 goto fast_block_end;
2143
2144 case CONTINUE_LOOP:
Christian Heimes217cfd12007-12-02 14:31:20 +00002145 retval = PyLong_FromLong(oparg);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002146 if (!retval) {
2147 x = NULL;
2148 break;
2149 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002150 why = WHY_CONTINUE;
2151 goto fast_block_end;
2152
Guido van Rossum374a9221991-04-04 10:40:29 +00002153 case SETUP_LOOP:
2154 case SETUP_EXCEPT:
2155 case SETUP_FINALLY:
Thomas Wouters8ce81f72007-09-20 18:22:40 +00002156 /* NOTE: If you add any new block-setup opcodes that
2157 are not try/except/finally handlers, you may need
2158 to update the PyGen_NeedsFinalizing() function.
2159 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002160
Guido van Rossumb209a111997-04-29 18:18:01 +00002161 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002162 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002163 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002164
Guido van Rossumc2e20742006-02-27 22:32:47 +00002165 case WITH_CLEANUP:
2166 {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002167 /* At the top of the stack are 1-3 values indicating
2168 how/why we entered the finally clause:
2169 - TOP = None
2170 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2171 - TOP = WHY_*; no retval below it
2172 - (TOP, SECOND, THIRD) = exc_info()
2173 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002174 In the last case, we must call
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002175 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002176 otherwise we must call
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002177 EXIT(None, None, None)
2178
2179 In all cases, we remove EXIT from the stack, leaving
2180 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002181
2182 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002183 *and* the function call returns a 'true' value, we
2184 "zap" this information, to prevent END_FINALLY from
2185 re-raising the exception. (But non-local gotos
2186 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002187 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002188
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002189 PyObject *exit_func = POP();
2190 u = TOP();
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002191 if (u == Py_None) {
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002192 v = w = Py_None;
2193 }
2194 else if (PyLong_Check(u)) {
Guido van Rossumc2e20742006-02-27 22:32:47 +00002195 u = v = w = Py_None;
2196 }
2197 else {
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002198 v = SECOND();
2199 w = THIRD();
Guido van Rossumc2e20742006-02-27 22:32:47 +00002200 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002201 /* XXX Not the fastest way to call it... */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002202 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2203 NULL);
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002204 Py_DECREF(exit_func);
2205 if (x == NULL)
Guido van Rossumf6694362006-03-10 02:28:35 +00002206 break; /* Go to error exit */
Amaury Forgeot d'Arc10b24e82008-12-10 23:49:33 +00002207
2208 if (u != Py_None)
2209 err = PyObject_IsTrue(x);
2210 else
2211 err = 0;
2212 Py_DECREF(x);
2213
2214 if (err < 0)
2215 break; /* Go to error exit */
2216 else if (err > 0) {
2217 err = 0;
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002218 /* There was an exception and a True return */
Christian Heimesdd15f6c2008-03-16 00:07:10 +00002219 STACKADJ(-2);
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002220 SET_TOP(PyLong_FromLong((long) WHY_SILENCED));
Guido van Rossumf6694362006-03-10 02:28:35 +00002221 Py_DECREF(u);
2222 Py_DECREF(v);
2223 Py_DECREF(w);
Guido van Rossumf6694362006-03-10 02:28:35 +00002224 }
Christian Heimes180510d2008-03-03 19:15:45 +00002225 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002226 break;
2227 }
2228
Guido van Rossumf10570b1995-07-07 22:53:21 +00002229 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002230 {
2231 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002232 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002233 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002234#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002235 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002236#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002237 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002238#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002239 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002240 PUSH(x);
2241 if (x != NULL)
2242 continue;
2243 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002244 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002245
Jeremy Hylton76901512000-03-28 23:49:17 +00002246 case CALL_FUNCTION_VAR:
2247 case CALL_FUNCTION_KW:
2248 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002249 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002250 int na = oparg & 0xff;
2251 int nk = (oparg>>8) & 0xff;
2252 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002253 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002254 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002255 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002256 if (flags & CALL_FLAG_VAR)
2257 n++;
2258 if (flags & CALL_FLAG_KW)
2259 n++;
2260 pfunc = stack_pointer - n - 1;
2261 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002262
Guido van Rossumac7be682001-01-17 15:42:30 +00002263 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002264 && PyMethod_GET_SELF(func) != NULL) {
2265 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002266 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002267 func = PyMethod_GET_FUNCTION(func);
2268 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002269 Py_DECREF(*pfunc);
2270 *pfunc = self;
2271 na++;
2272 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002273 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002274 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002275 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002276 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002277 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002278 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002279 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002280 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002281
Jeremy Hylton76901512000-03-28 23:49:17 +00002282 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002283 w = POP();
2284 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002285 }
2286 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002287 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002288 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002289 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002290 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002291
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002292 case MAKE_CLOSURE:
Guido van Rossum681d79a1995-07-18 14:51:37 +00002293 case MAKE_FUNCTION:
Guido van Rossum4f72a782006-10-27 23:31:49 +00002294 {
2295 int posdefaults = oparg & 0xff;
2296 int kwdefaults = (oparg>>8) & 0xff;
Neal Norwitzc1505362006-12-28 06:47:50 +00002297 int num_annotations = (oparg >> 16) & 0x7fff;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002298
Guido van Rossum681d79a1995-07-18 14:51:37 +00002299 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002300 x = PyFunction_New(v, f->f_globals);
2301 Py_DECREF(v);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002302
Guido van Rossum0240b922007-02-26 21:23:50 +00002303 if (x != NULL && opcode == MAKE_CLOSURE) {
2304 v = POP();
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00002305 if (PyFunction_SetClosure(x, v) != 0) {
2306 /* Can't happen unless bytecode is corrupt. */
2307 why = WHY_EXCEPTION;
2308 }
Guido van Rossum0240b922007-02-26 21:23:50 +00002309 Py_DECREF(v);
2310 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002311
2312 if (x != NULL && num_annotations > 0) {
2313 Py_ssize_t name_ix;
2314 u = POP(); /* names of args with annotations */
2315 v = PyDict_New();
2316 if (v == NULL) {
2317 Py_DECREF(x);
2318 x = NULL;
2319 break;
2320 }
2321 name_ix = PyTuple_Size(u);
2322 assert(num_annotations == name_ix+1);
2323 while (name_ix > 0) {
2324 --name_ix;
2325 t = PyTuple_GET_ITEM(u, name_ix);
2326 w = POP();
2327 /* XXX(nnorwitz): check for errors */
2328 PyDict_SetItem(v, t, w);
2329 Py_DECREF(w);
2330 }
2331
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00002332 if (PyFunction_SetAnnotations(x, v) != 0) {
2333 /* Can't happen unless
2334 PyFunction_SetAnnotations changes. */
2335 why = WHY_EXCEPTION;
2336 }
Neal Norwitzc1505362006-12-28 06:47:50 +00002337 Py_DECREF(v);
2338 Py_DECREF(u);
2339 }
2340
Guido van Rossum681d79a1995-07-18 14:51:37 +00002341 /* XXX Maybe this should be a separate opcode? */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002342 if (x != NULL && posdefaults > 0) {
2343 v = PyTuple_New(posdefaults);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002344 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002345 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002346 x = NULL;
2347 break;
2348 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002349 while (--posdefaults >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002350 w = POP();
Guido van Rossum4f72a782006-10-27 23:31:49 +00002351 PyTuple_SET_ITEM(v, posdefaults, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002352 }
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00002353 if (PyFunction_SetDefaults(x, v) != 0) {
2354 /* Can't happen unless
2355 PyFunction_SetDefaults changes. */
2356 why = WHY_EXCEPTION;
2357 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002358 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002359 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002360 if (x != NULL && kwdefaults > 0) {
2361 v = PyDict_New();
2362 if (v == NULL) {
2363 Py_DECREF(x);
2364 x = NULL;
2365 break;
2366 }
2367 while (--kwdefaults >= 0) {
2368 w = POP(); /* default value */
2369 u = POP(); /* kw only arg name */
Neal Norwitzc1505362006-12-28 06:47:50 +00002370 /* XXX(nnorwitz): check for errors */
Guido van Rossum4f72a782006-10-27 23:31:49 +00002371 PyDict_SetItem(v, u, w);
Georg Brandl94ab0002007-02-26 13:58:18 +00002372 Py_DECREF(w);
2373 Py_DECREF(u);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002374 }
Jeffrey Yasskinb9083f32008-12-11 06:18:33 +00002375 if (PyFunction_SetKwDefaults(x, v) != 0) {
2376 /* Can't happen unless
2377 PyFunction_SetKwDefaults changes. */
2378 why = WHY_EXCEPTION;
2379 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002380 Py_DECREF(v);
2381 }
Guido van Rossum681d79a1995-07-18 14:51:37 +00002382 PUSH(x);
2383 break;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002384 }
Guido van Rossum8861b741996-07-30 16:49:37 +00002385
2386 case BUILD_SLICE:
2387 if (oparg == 3)
2388 w = POP();
2389 else
2390 w = NULL;
2391 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002392 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002393 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002394 Py_DECREF(u);
2395 Py_DECREF(v);
2396 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002397 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002398 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002399 break;
2400
Fred Drakeef8ace32000-08-24 00:32:09 +00002401 case EXTENDED_ARG:
2402 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002403 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002404 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002405
Guido van Rossum374a9221991-04-04 10:40:29 +00002406 default:
2407 fprintf(stderr,
2408 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002409 PyCode_Addr2Line(f->f_code, f->f_lasti),
2410 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002411 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002412 why = WHY_EXCEPTION;
2413 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002414
2415#ifdef CASE_TOO_BIG
2416 }
2417#endif
2418
Guido van Rossum374a9221991-04-04 10:40:29 +00002419 } /* switch */
2420
2421 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002422
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002423 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002424
Guido van Rossum374a9221991-04-04 10:40:29 +00002425 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002426
Guido van Rossum374a9221991-04-04 10:40:29 +00002427 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002428 if (err == 0 && x != NULL) {
2429#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002430 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002431 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002432 fprintf(stderr,
2433 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002434 else {
2435#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002436 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002437 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002438#ifdef CHECKEXC
2439 }
2440#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002441 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002442 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002443 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002444 err = 0;
2445 }
2446
Guido van Rossum374a9221991-04-04 10:40:29 +00002447 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002448
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002449 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002450 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002451 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002452 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002453 why = WHY_EXCEPTION;
2454 }
2455 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002456#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002457 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002458 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002459 if (PyErr_Occurred()) {
Christian Heimes412dc9c2008-01-27 18:55:54 +00002460 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002461 sprintf(buf, "Stack unwind with exception "
2462 "set and why=%d", why);
2463 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002464 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002465 }
2466#endif
2467
2468 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002469
Guido van Rossum374a9221991-04-04 10:40:29 +00002470 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002471 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002472
Fred Drake8f51f542001-10-04 14:48:42 +00002473 if (tstate->c_tracefunc != NULL)
2474 call_exc_trace(tstate->c_tracefunc,
2475 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002476 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002477
Guido van Rossum374a9221991-04-04 10:40:29 +00002478 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002479
Guido van Rossum374a9221991-04-04 10:40:29 +00002480 if (why == WHY_RERAISE)
2481 why = WHY_EXCEPTION;
2482
2483 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002484
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002485fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002486 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002487 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002488
Tim Peters8a5c3c72004-04-05 19:36:21 +00002489 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002490 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2491 /* For a continue inside a try block,
2492 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002493 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2494 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002495 why = WHY_NOT;
Christian Heimes217cfd12007-12-02 14:31:20 +00002496 JUMPTO(PyLong_AS_LONG(retval));
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002497 Py_DECREF(retval);
2498 break;
2499 }
2500
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002501 if (b->b_type == EXCEPT_HANDLER) {
2502 UNWIND_EXCEPT_HANDLER(b);
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002503 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002504 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002505 UNWIND_BLOCK(b);
Guido van Rossum374a9221991-04-04 10:40:29 +00002506 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2507 why = WHY_NOT;
2508 JUMPTO(b->b_handler);
2509 break;
2510 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002511 if (why == WHY_EXCEPTION && (b->b_type == SETUP_EXCEPT
2512 || b->b_type == SETUP_FINALLY)) {
2513 PyObject *exc, *val, *tb;
2514 int handler = b->b_handler;
2515 /* Beware, this invalidates all b->b_* fields */
2516 PyFrame_BlockSetup(f, EXCEPT_HANDLER, -1, STACK_LEVEL());
2517 PUSH(tstate->exc_traceback);
2518 PUSH(tstate->exc_value);
2519 if (tstate->exc_type != NULL) {
2520 PUSH(tstate->exc_type);
Guido van Rossum374a9221991-04-04 10:40:29 +00002521 }
2522 else {
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002523 Py_INCREF(Py_None);
2524 PUSH(Py_None);
Guido van Rossum374a9221991-04-04 10:40:29 +00002525 }
Benjamin Petersoneec3d712008-06-11 15:59:43 +00002526 PyErr_Fetch(&exc, &val, &tb);
2527 /* Make the raw exception data
2528 available to the handler,
2529 so a program can emulate the
2530 Python main loop. */
2531 PyErr_NormalizeException(
2532 &exc, &val, &tb);
2533 PyException_SetTraceback(val, tb);
2534 Py_INCREF(exc);
2535 tstate->exc_type = exc;
2536 Py_INCREF(val);
2537 tstate->exc_value = val;
2538 tstate->exc_traceback = tb;
2539 if (tb == NULL)
2540 tb = Py_None;
2541 Py_INCREF(tb);
2542 PUSH(tb);
2543 PUSH(val);
2544 PUSH(exc);
2545 why = WHY_NOT;
2546 JUMPTO(handler);
2547 break;
2548 }
2549 if (b->b_type == SETUP_FINALLY) {
2550 if (why & (WHY_RETURN | WHY_CONTINUE))
2551 PUSH(retval);
2552 PUSH(PyLong_FromLong((long)why));
Guido van Rossum374a9221991-04-04 10:40:29 +00002553 why = WHY_NOT;
2554 JUMPTO(b->b_handler);
2555 break;
2556 }
2557 } /* unwind stack */
2558
2559 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002560
Guido van Rossum374a9221991-04-04 10:40:29 +00002561 if (why != WHY_NOT)
2562 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002563 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002564
Guido van Rossum374a9221991-04-04 10:40:29 +00002565 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002566
Tim Peters8a5c3c72004-04-05 19:36:21 +00002567 assert(why != WHY_YIELD);
2568 /* Pop remaining stack entries. */
2569 while (!EMPTY()) {
2570 v = POP();
2571 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002572 }
2573
Tim Peters8a5c3c72004-04-05 19:36:21 +00002574 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002575 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002576
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002577fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002578 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002579 if (tstate->c_tracefunc) {
2580 if (why == WHY_RETURN || why == WHY_YIELD) {
2581 if (call_trace(tstate->c_tracefunc,
2582 tstate->c_traceobj, f,
2583 PyTrace_RETURN, retval)) {
2584 Py_XDECREF(retval);
2585 retval = NULL;
2586 why = WHY_EXCEPTION;
2587 }
2588 }
2589 else if (why == WHY_EXCEPTION) {
2590 call_trace_protected(tstate->c_tracefunc,
2591 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002592 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002593 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002594 }
Fred Drake8f51f542001-10-04 14:48:42 +00002595 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002596 if (why == WHY_EXCEPTION)
2597 call_trace_protected(tstate->c_profilefunc,
2598 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002599 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002600 else if (call_trace(tstate->c_profilefunc,
2601 tstate->c_profileobj, f,
2602 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002603 Py_XDECREF(retval);
2604 retval = NULL;
2605 why = WHY_EXCEPTION;
2606 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002607 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002608 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002609
Tim Peters5ca576e2001-06-18 22:08:13 +00002610 /* pop frame */
Thomas Woutersce272b62007-09-19 21:19:28 +00002611exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002612 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002613 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002614
Guido van Rossum96a42c81992-01-12 02:29:51 +00002615 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002616}
2617
Guido van Rossumc2e20742006-02-27 22:32:47 +00002618/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002619 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002620 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002621
Tim Peters6d6c1a32001-08-02 04:15:00 +00002622PyObject *
2623PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002624 PyObject **args, int argcount, PyObject **kws, int kwcount,
Guido van Rossum4f72a782006-10-27 23:31:49 +00002625 PyObject **defs, int defcount, PyObject *kwdefs, PyObject *closure)
Tim Peters5ca576e2001-06-18 22:08:13 +00002626{
2627 register PyFrameObject *f;
2628 register PyObject *retval = NULL;
2629 register PyObject **fastlocals, **freevars;
2630 PyThreadState *tstate = PyThreadState_GET();
2631 PyObject *x, *u;
2632
2633 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002634 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002635 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002636 return NULL;
2637 }
2638
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00002639 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002640 assert(globals != NULL);
2641 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002642 if (f == NULL)
2643 return NULL;
2644
2645 fastlocals = f->f_localsplus;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002646 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002647
2648 if (co->co_argcount > 0 ||
Guido van Rossum4f72a782006-10-27 23:31:49 +00002649 co->co_kwonlyargcount > 0 ||
Tim Peters5ca576e2001-06-18 22:08:13 +00002650 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2651 int i;
2652 int n = argcount;
2653 PyObject *kwdict = NULL;
2654 if (co->co_flags & CO_VARKEYWORDS) {
2655 kwdict = PyDict_New();
2656 if (kwdict == NULL)
2657 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002658 i = co->co_argcount + co->co_kwonlyargcount;
Tim Peters5ca576e2001-06-18 22:08:13 +00002659 if (co->co_flags & CO_VARARGS)
2660 i++;
2661 SETLOCAL(i, kwdict);
2662 }
2663 if (argcount > co->co_argcount) {
2664 if (!(co->co_flags & CO_VARARGS)) {
2665 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002666 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002667 "%spositional argument%s (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002668 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002669 defcount ? "at most" : "exactly",
2670 co->co_argcount,
2671 kwcount ? "non-keyword " : "",
2672 co->co_argcount == 1 ? "" : "s",
2673 argcount);
2674 goto fail;
2675 }
2676 n = co->co_argcount;
2677 }
2678 for (i = 0; i < n; i++) {
2679 x = args[i];
2680 Py_INCREF(x);
2681 SETLOCAL(i, x);
2682 }
2683 if (co->co_flags & CO_VARARGS) {
2684 u = PyTuple_New(argcount - n);
2685 if (u == NULL)
2686 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002687 SETLOCAL(co->co_argcount + co->co_kwonlyargcount, u);
Tim Peters5ca576e2001-06-18 22:08:13 +00002688 for (i = n; i < argcount; i++) {
2689 x = args[i];
2690 Py_INCREF(x);
2691 PyTuple_SET_ITEM(u, i-n, x);
2692 }
2693 }
2694 for (i = 0; i < kwcount; i++) {
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002695 PyObject **co_varnames;
Tim Peters5ca576e2001-06-18 22:08:13 +00002696 PyObject *keyword = kws[2*i];
2697 PyObject *value = kws[2*i + 1];
2698 int j;
Neal Norwitzda059e32007-08-26 05:33:45 +00002699 if (keyword == NULL || !PyUnicode_Check(keyword)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002700 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002701 "%U() keywords must be strings",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002702 co->co_name);
Tim Peters5ca576e2001-06-18 22:08:13 +00002703 goto fail;
2704 }
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002705 /* Speed hack: do raw pointer compares. As names are
2706 normally interned this should almost always hit. */
2707 co_varnames = PySequence_Fast_ITEMS(co->co_varnames);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002708 for (j = 0;
2709 j < co->co_argcount + co->co_kwonlyargcount;
2710 j++) {
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002711 PyObject *nm = co_varnames[j];
2712 if (nm == keyword)
2713 goto kw_found;
2714 }
2715 /* Slow fallback, just in case */
2716 for (j = 0;
2717 j < co->co_argcount + co->co_kwonlyargcount;
2718 j++) {
2719 PyObject *nm = co_varnames[j];
Tim Peters5ca576e2001-06-18 22:08:13 +00002720 int cmp = PyObject_RichCompareBool(
2721 keyword, nm, Py_EQ);
2722 if (cmp > 0)
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002723 goto kw_found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002724 else if (cmp < 0)
2725 goto fail;
2726 }
2727 /* Check errors from Compare */
2728 if (PyErr_Occurred())
2729 goto fail;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002730 if (j >= co->co_argcount + co->co_kwonlyargcount) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002731 if (kwdict == NULL) {
2732 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002733 "%U() got an unexpected "
Walter Dörwald573c08c2007-05-25 15:46:59 +00002734 "keyword argument '%S'",
2735 co->co_name,
2736 keyword);
Tim Peters5ca576e2001-06-18 22:08:13 +00002737 goto fail;
2738 }
2739 PyDict_SetItem(kwdict, keyword, value);
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002740 continue;
Tim Peters5ca576e2001-06-18 22:08:13 +00002741 }
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002742kw_found:
2743 if (GETLOCAL(j) != NULL) {
2744 PyErr_Format(PyExc_TypeError,
2745 "%U() got multiple "
2746 "values for keyword "
2747 "argument '%S'",
2748 co->co_name,
2749 keyword);
2750 goto fail;
Tim Peters5ca576e2001-06-18 22:08:13 +00002751 }
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002752 Py_INCREF(value);
2753 SETLOCAL(j, value);
Tim Peters5ca576e2001-06-18 22:08:13 +00002754 }
Guido van Rossum4f72a782006-10-27 23:31:49 +00002755 if (co->co_kwonlyargcount > 0) {
2756 for (i = co->co_argcount;
2757 i < co->co_argcount + co->co_kwonlyargcount;
2758 i++) {
Guido van Rossum29602e42006-11-22 04:45:33 +00002759 PyObject *name, *def;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002760 if (GETLOCAL(i) != NULL)
2761 continue;
Guido van Rossum29602e42006-11-22 04:45:33 +00002762 name = PyTuple_GET_ITEM(co->co_varnames, i);
2763 def = NULL;
Guido van Rossum4f72a782006-10-27 23:31:49 +00002764 if (kwdefs != NULL)
2765 def = PyDict_GetItem(kwdefs, name);
2766 if (def != NULL) {
2767 Py_INCREF(def);
2768 SETLOCAL(i, def);
2769 continue;
2770 }
2771 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002772 "%U() needs keyword-only argument %S",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002773 co->co_name, name);
Guido van Rossum4f72a782006-10-27 23:31:49 +00002774 goto fail;
2775 }
2776 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002777 if (argcount < co->co_argcount) {
2778 int m = co->co_argcount - defcount;
2779 for (i = argcount; i < m; i++) {
2780 if (GETLOCAL(i) == NULL) {
2781 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002782 "%U() takes %s %d "
Guido van Rossum4f72a782006-10-27 23:31:49 +00002783 "%spositional argument%s "
2784 "(%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002785 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002786 ((co->co_flags & CO_VARARGS) ||
2787 defcount) ? "at least"
2788 : "exactly",
2789 m, kwcount ? "non-keyword " : "",
2790 m == 1 ? "" : "s", i);
2791 goto fail;
2792 }
2793 }
2794 if (n > m)
2795 i = n - m;
2796 else
2797 i = 0;
2798 for (; i < defcount; i++) {
2799 if (GETLOCAL(m+i) == NULL) {
2800 PyObject *def = defs[i];
2801 Py_INCREF(def);
2802 SETLOCAL(m+i, def);
2803 }
2804 }
2805 }
2806 }
2807 else {
2808 if (argcount > 0 || kwcount > 0) {
2809 PyErr_Format(PyExc_TypeError,
Guido van Rossum00bc0e02007-10-15 02:52:41 +00002810 "%U() takes no arguments (%d given)",
Walter Dörwald573c08c2007-05-25 15:46:59 +00002811 co->co_name,
Tim Peters5ca576e2001-06-18 22:08:13 +00002812 argcount + kwcount);
2813 goto fail;
2814 }
2815 }
2816 /* Allocate and initialize storage for cell vars, and copy free
2817 vars into frame. This isn't too efficient right now. */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002818 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002819 int i, j, nargs, found;
Martin v. Löwis5b222132007-06-10 09:51:05 +00002820 Py_UNICODE *cellname, *argname;
Tim Peters5ca576e2001-06-18 22:08:13 +00002821 PyObject *c;
2822
Amaury Forgeot d'Arce670bd42007-11-24 00:29:24 +00002823 nargs = co->co_argcount + co->co_kwonlyargcount;
Tim Peters5ca576e2001-06-18 22:08:13 +00002824 if (co->co_flags & CO_VARARGS)
2825 nargs++;
2826 if (co->co_flags & CO_VARKEYWORDS)
2827 nargs++;
2828
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002829 /* Initialize each cell var, taking into account
2830 cell vars that are initialized from arguments.
2831
2832 Should arrange for the compiler to put cellvars
2833 that are arguments at the beginning of the cellvars
2834 list so that we can march over it more efficiently?
2835 */
Thomas Wouters477c8d52006-05-27 19:21:47 +00002836 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002837 cellname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002838 PyTuple_GET_ITEM(co->co_cellvars, i));
2839 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002840 for (j = 0; j < nargs; j++) {
Martin v. Löwis5b222132007-06-10 09:51:05 +00002841 argname = PyUnicode_AS_UNICODE(
Tim Peters5ca576e2001-06-18 22:08:13 +00002842 PyTuple_GET_ITEM(co->co_varnames, j));
Martin v. Löwis5b222132007-06-10 09:51:05 +00002843 if (Py_UNICODE_strcmp(cellname, argname) == 0) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002844 c = PyCell_New(GETLOCAL(j));
2845 if (c == NULL)
2846 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002847 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002848 found = 1;
2849 break;
2850 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002851 }
2852 if (found == 0) {
2853 c = PyCell_New(NULL);
2854 if (c == NULL)
2855 goto fail;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002856 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002857 }
2858 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002859 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00002860 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002861 int i;
Thomas Wouters477c8d52006-05-27 19:21:47 +00002862 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002863 PyObject *o = PyTuple_GET_ITEM(closure, i);
2864 Py_INCREF(o);
Thomas Wouters477c8d52006-05-27 19:21:47 +00002865 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002866 }
2867 }
2868
Tim Peters5ca576e2001-06-18 22:08:13 +00002869 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002870 /* Don't need to keep the reference to f_back, it will be set
2871 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002872 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002873 f->f_back = NULL;
2874
Jeremy Hylton985eba52003-02-05 23:13:00 +00002875 PCALL(PCALL_GENERATOR);
2876
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002877 /* Create a new generator that owns the ready to run frame
2878 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002879 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002880 }
2881
Thomas Woutersce272b62007-09-19 21:19:28 +00002882 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002883
Thomas Woutersce272b62007-09-19 21:19:28 +00002884fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00002885
Tim Petersb13680b2001-11-27 23:29:29 +00002886 /* decref'ing the frame can cause __del__ methods to get invoked,
2887 which can call back into Python. While we're done with the
2888 current Python frame (f), the associated C stack is still in use,
2889 so recursion_depth must be boosted for the duration.
2890 */
2891 assert(tstate != NULL);
2892 ++tstate->recursion_depth;
Thomas Woutersce272b62007-09-19 21:19:28 +00002893 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002894 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002895 return retval;
2896}
2897
2898
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002899/* Logic for the raise statement (too complicated for inlining).
2900 This *consumes* a reference count to each of its arguments. */
Raymond Hettinger7c958652004-04-06 10:11:10 +00002901static enum why_code
Collin Winter828f04a2007-08-31 00:04:24 +00002902do_raise(PyObject *exc, PyObject *cause)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002903{
Guido van Rossumb4fb6e42008-06-14 20:20:24 +00002904 PyObject *type = NULL, *value = NULL;
Collin Winter828f04a2007-08-31 00:04:24 +00002905
2906 if (exc == NULL) {
Guido van Rossumd295f121998-04-09 21:39:57 +00002907 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00002908 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumb4fb6e42008-06-14 20:20:24 +00002909 PyObject *tb;
Collin Winter828f04a2007-08-31 00:04:24 +00002910 type = tstate->exc_type;
Guido van Rossumd295f121998-04-09 21:39:57 +00002911 value = tstate->exc_value;
2912 tb = tstate->exc_traceback;
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002913 if (type == Py_None) {
2914 PyErr_SetString(PyExc_RuntimeError,
2915 "No active exception to reraise");
2916 return WHY_EXCEPTION;
2917 }
2918 Py_XINCREF(type);
Guido van Rossumd295f121998-04-09 21:39:57 +00002919 Py_XINCREF(value);
2920 Py_XINCREF(tb);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002921 PyErr_Restore(type, value, tb);
2922 return WHY_RERAISE;
Guido van Rossumd295f121998-04-09 21:39:57 +00002923 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002924
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002925 /* We support the following forms of raise:
Collin Winter828f04a2007-08-31 00:04:24 +00002926 raise
2927 raise <instance>
2928 raise <type> */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002929
Collin Winter828f04a2007-08-31 00:04:24 +00002930 if (PyExceptionClass_Check(exc)) {
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002931 type = exc;
2932 value = PyObject_CallObject(exc, NULL);
Collin Winter828f04a2007-08-31 00:04:24 +00002933 if (value == NULL)
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002934 goto raise_error;
2935 }
Collin Winter828f04a2007-08-31 00:04:24 +00002936 else if (PyExceptionInstance_Check(exc)) {
2937 value = exc;
2938 type = PyExceptionInstance_Class(exc);
Guido van Rossumb209a111997-04-29 18:18:01 +00002939 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002940 }
2941 else {
2942 /* Not something you can raise. You get an exception
2943 anyway, just not what you specified :-) */
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002944 Py_DECREF(exc);
Guido van Rossum45aecf42006-03-15 04:58:47 +00002945 PyErr_SetString(PyExc_TypeError,
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002946 "exceptions must derive from BaseException");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002947 goto raise_error;
2948 }
Collin Winter828f04a2007-08-31 00:04:24 +00002949
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00002950 if (cause) {
2951 PyObject *fixed_cause;
2952 if (PyExceptionClass_Check(cause)) {
2953 fixed_cause = PyObject_CallObject(cause, NULL);
2954 if (fixed_cause == NULL)
2955 goto raise_error;
2956 Py_DECREF(cause);
2957 }
2958 else if (PyExceptionInstance_Check(cause)) {
2959 fixed_cause = cause;
2960 }
2961 else {
2962 PyErr_SetString(PyExc_TypeError,
2963 "exception causes must derive from "
2964 "BaseException");
2965 goto raise_error;
2966 }
2967 PyException_SetCause(value, fixed_cause);
2968 }
Collin Winter828f04a2007-08-31 00:04:24 +00002969
Guido van Rossumb4fb6e42008-06-14 20:20:24 +00002970 PyErr_SetObject(type, value);
2971 /* PyErr_SetObject incref's its arguments */
2972 Py_XDECREF(value);
2973 Py_XDECREF(type);
Collin Winter828f04a2007-08-31 00:04:24 +00002974 return WHY_EXCEPTION;
2975
2976raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00002977 Py_XDECREF(value);
2978 Py_XDECREF(type);
Collin Winter1966f1c2007-09-01 20:26:44 +00002979 Py_XDECREF(cause);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00002980 return WHY_EXCEPTION;
2981}
2982
Tim Petersd6d010b2001-06-21 02:49:55 +00002983/* Iterate v argcnt times and store the results on the stack (via decreasing
Guido van Rossum0368b722007-05-11 16:50:42 +00002984 sp). Return 1 for success, 0 if error.
Antoine Pitrou9a2310d2008-07-25 22:39:39 +00002985
Guido van Rossum0368b722007-05-11 16:50:42 +00002986 If argcntafter == -1, do a simple unpack. If it is >= 0, do an unpack
2987 with a variable target.
2988*/
Tim Petersd6d010b2001-06-21 02:49:55 +00002989
Barry Warsawe42b18f1997-08-25 22:13:04 +00002990static int
Guido van Rossum0368b722007-05-11 16:50:42 +00002991unpack_iterable(PyObject *v, int argcnt, int argcntafter, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00002992{
Guido van Rossum0368b722007-05-11 16:50:42 +00002993 int i = 0, j = 0;
2994 Py_ssize_t ll = 0;
Tim Petersd6d010b2001-06-21 02:49:55 +00002995 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00002996 PyObject *w;
Guido van Rossum0368b722007-05-11 16:50:42 +00002997 PyObject *l = NULL; /* variable list */
Guido van Rossumac7be682001-01-17 15:42:30 +00002998
Tim Petersd6d010b2001-06-21 02:49:55 +00002999 assert(v != NULL);
3000
3001 it = PyObject_GetIter(v);
3002 if (it == NULL)
3003 goto Error;
3004
3005 for (; i < argcnt; i++) {
3006 w = PyIter_Next(it);
3007 if (w == NULL) {
3008 /* Iterator done, via error or exhaustion. */
3009 if (!PyErr_Occurred()) {
3010 PyErr_Format(PyExc_ValueError,
3011 "need more than %d value%s to unpack",
3012 i, i == 1 ? "" : "s");
3013 }
3014 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003015 }
3016 *--sp = w;
3017 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003018
Guido van Rossum0368b722007-05-11 16:50:42 +00003019 if (argcntafter == -1) {
3020 /* We better have exhausted the iterator now. */
3021 w = PyIter_Next(it);
3022 if (w == NULL) {
3023 if (PyErr_Occurred())
3024 goto Error;
3025 Py_DECREF(it);
3026 return 1;
3027 }
3028 Py_DECREF(w);
3029 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
3030 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003031 }
Guido van Rossum0368b722007-05-11 16:50:42 +00003032
3033 l = PySequence_List(it);
3034 if (l == NULL)
3035 goto Error;
3036 *--sp = l;
3037 i++;
3038
3039 ll = PyList_GET_SIZE(l);
3040 if (ll < argcntafter) {
3041 PyErr_Format(PyExc_ValueError, "need more than %zd values to unpack",
3042 argcnt + ll);
3043 goto Error;
3044 }
3045
3046 /* Pop the "after-variable" args off the list. */
3047 for (j = argcntafter; j > 0; j--, i++) {
3048 *--sp = PyList_GET_ITEM(l, ll - j);
3049 }
3050 /* Resize the list. */
Christian Heimes90aa7642007-12-19 02:45:37 +00003051 Py_SIZE(l) = ll - argcntafter;
Guido van Rossum0368b722007-05-11 16:50:42 +00003052 Py_DECREF(it);
3053 return 1;
3054
Tim Petersd6d010b2001-06-21 02:49:55 +00003055Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003056 for (; i > 0; i--, sp++)
3057 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003058 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003059 return 0;
3060}
3061
3062
Guido van Rossum96a42c81992-01-12 02:29:51 +00003063#ifdef LLTRACE
Guido van Rossum3f5da241990-12-20 15:06:42 +00003064static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003065prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003066{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003067 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003068 if (PyObject_Print(v, stdout, 0) != 0)
3069 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003070 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003071 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003072}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003073#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003074
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003075static void
Fred Drake5755ce62001-06-27 19:19:46 +00003076call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003077{
Guido van Rossumb209a111997-04-29 18:18:01 +00003078 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003079 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003080 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003081 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003082 value = Py_None;
3083 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003084 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003085 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003086 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003087 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003088 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003089 }
Fred Drake5755ce62001-06-27 19:19:46 +00003090 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003091 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003092 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003093 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003094 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003095 Py_XDECREF(type);
3096 Py_XDECREF(value);
3097 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003098 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003099}
3100
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003101static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003102call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003103 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003104{
3105 PyObject *type, *value, *traceback;
3106 int err;
3107 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003108 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003109 if (err == 0)
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003110 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003111 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003112 return 0;
3113 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003114 else {
3115 Py_XDECREF(type);
3116 Py_XDECREF(value);
3117 Py_XDECREF(traceback);
Amaury Forgeot d'Arcf05149a2007-11-13 01:05:30 +00003118 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003119 }
3120}
3121
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003122static int
Fred Drake5755ce62001-06-27 19:19:46 +00003123call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3124 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003125{
Fred Drake5755ce62001-06-27 19:19:46 +00003126 register PyThreadState *tstate = frame->f_tstate;
3127 int result;
3128 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003129 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003130 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003131 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003132 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003133 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3134 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003135 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003136 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003137}
3138
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003139PyObject *
3140_PyEval_CallTracing(PyObject *func, PyObject *args)
3141{
3142 PyFrameObject *frame = PyEval_GetFrame();
3143 PyThreadState *tstate = frame->f_tstate;
3144 int save_tracing = tstate->tracing;
3145 int save_use_tracing = tstate->use_tracing;
3146 PyObject *result;
3147
3148 tstate->tracing = 0;
3149 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3150 || (tstate->c_profilefunc != NULL));
3151 result = PyObject_Call(func, args, NULL);
3152 tstate->tracing = save_tracing;
3153 tstate->use_tracing = save_use_tracing;
3154 return result;
3155}
3156
Michael W. Hudson006c7522002-11-08 13:08:46 +00003157static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003158maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003159 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3160 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003161{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003162 int result = 0;
3163
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003164 /* If the last instruction executed isn't in the current
3165 instruction window, reset the window. If the last
3166 instruction happens to fall at the start of a line or if it
3167 represents a jump backwards, call the trace function.
3168 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003169 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003170 int line;
3171 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003172
Thomas Woutersce272b62007-09-19 21:19:28 +00003173 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3174 &bounds);
3175 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003176 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003177 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003178 PyTrace_LINE, Py_None);
Thomas Woutersce272b62007-09-19 21:19:28 +00003179 }
3180 *instr_lb = bounds.ap_lower;
3181 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003182 }
Armin Rigobf57a142004-03-22 19:24:58 +00003183 else if (frame->f_lasti <= *instr_prev) {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003184 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003185 }
3186 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003187 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003188}
3189
Fred Drake5755ce62001-06-27 19:19:46 +00003190void
3191PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003192{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003193 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003194 PyObject *temp = tstate->c_profileobj;
3195 Py_XINCREF(arg);
3196 tstate->c_profilefunc = NULL;
3197 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003198 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003199 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003200 Py_XDECREF(temp);
3201 tstate->c_profilefunc = func;
3202 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003203 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003204 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003205}
3206
3207void
3208PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3209{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003210 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003211 PyObject *temp = tstate->c_traceobj;
Jeffrey Yasskin008d8ef2008-12-06 17:09:27 +00003212 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003213 Py_XINCREF(arg);
3214 tstate->c_tracefunc = NULL;
3215 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003216 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003217 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003218 Py_XDECREF(temp);
3219 tstate->c_tracefunc = func;
3220 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003221 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003222 tstate->use_tracing = ((func != NULL)
3223 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003224}
3225
Guido van Rossumb209a111997-04-29 18:18:01 +00003226PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003227PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003228{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003229 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003230 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003231 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003232 else
3233 return current_frame->f_builtins;
3234}
3235
Guido van Rossumb209a111997-04-29 18:18:01 +00003236PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003237PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003238{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003239 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003240 if (current_frame == NULL)
3241 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003242 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003243 return current_frame->f_locals;
3244}
3245
Guido van Rossumb209a111997-04-29 18:18:01 +00003246PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003247PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003248{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003249 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003250 if (current_frame == NULL)
3251 return NULL;
3252 else
3253 return current_frame->f_globals;
3254}
3255
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003256PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003257PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003258{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003259 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003260 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003261}
3262
Guido van Rossum6135a871995-01-09 17:53:26 +00003263int
Tim Peters5ba58662001-07-16 02:29:45 +00003264PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003265{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003266 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003267 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003268
3269 if (current_frame != NULL) {
3270 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003271 const int compilerflags = codeflags & PyCF_MASK;
3272 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003273 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003274 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003275 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003276#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003277 if (codeflags & CO_GENERATOR_ALLOWED) {
3278 result = 1;
3279 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3280 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003281#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003282 }
3283 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003284}
3285
Guido van Rossum3f5da241990-12-20 15:06:42 +00003286
Guido van Rossum681d79a1995-07-18 14:51:37 +00003287/* External interface to call any callable object.
3288 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003289
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003290#undef PyEval_CallObject
3291/* for backward compatibility: export this interface */
3292
Guido van Rossumb209a111997-04-29 18:18:01 +00003293PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003294PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003295{
Guido van Rossumb209a111997-04-29 18:18:01 +00003296 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003297}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003298#define PyEval_CallObject(func,arg) \
3299 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003300
Guido van Rossumb209a111997-04-29 18:18:01 +00003301PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003302PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003303{
Jeremy Hylton52820442001-01-03 23:52:36 +00003304 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003305
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003306 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003307 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003308 if (arg == NULL)
3309 return NULL;
3310 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003311 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003312 PyErr_SetString(PyExc_TypeError,
3313 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003314 return NULL;
3315 }
3316 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003317 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003318
Guido van Rossumb209a111997-04-29 18:18:01 +00003319 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003320 PyErr_SetString(PyExc_TypeError,
3321 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003322 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003323 return NULL;
3324 }
3325
Tim Peters6d6c1a32001-08-02 04:15:00 +00003326 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003327 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003328 return result;
3329}
3330
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003331const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003332PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003333{
3334 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003335 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003336 else if (PyFunction_Check(func))
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003337 return _PyUnicode_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003338 else if (PyCFunction_Check(func))
3339 return ((PyCFunctionObject*)func)->m_ml->ml_name;
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003340 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003341 return func->ob_type->tp_name;
Jeremy Hylton512a2372001-04-11 13:52:29 +00003342}
3343
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003344const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003345PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003346{
3347 if (PyMethod_Check(func))
3348 return "()";
3349 else if (PyFunction_Check(func))
3350 return "()";
3351 else if (PyCFunction_Check(func))
3352 return "()";
Guido van Rossum50e9fb92006-08-17 05:42:55 +00003353 else
Jeremy Hylton512a2372001-04-11 13:52:29 +00003354 return " object";
Jeremy Hylton512a2372001-04-11 13:52:29 +00003355}
3356
Neal Norwitzaddfe0c2002-11-10 14:33:26 +00003357static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003358err_args(PyObject *func, int flags, int nargs)
3359{
3360 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003361 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003362 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003363 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003364 nargs);
3365 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003366 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003367 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003368 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003369 nargs);
3370}
3371
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003372#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003373if (tstate->use_tracing && tstate->c_profilefunc) { \
3374 if (call_trace(tstate->c_profilefunc, \
3375 tstate->c_profileobj, \
3376 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003377 func)) { \
3378 x = NULL; \
3379 } \
3380 else { \
3381 x = call; \
3382 if (tstate->c_profilefunc != NULL) { \
3383 if (x == NULL) { \
3384 call_trace_protected(tstate->c_profilefunc, \
3385 tstate->c_profileobj, \
3386 tstate->frame, PyTrace_C_EXCEPTION, \
3387 func); \
3388 /* XXX should pass (type, value, tb) */ \
3389 } else { \
3390 if (call_trace(tstate->c_profilefunc, \
3391 tstate->c_profileobj, \
3392 tstate->frame, PyTrace_C_RETURN, \
3393 func)) { \
3394 Py_DECREF(x); \
3395 x = NULL; \
3396 } \
3397 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003398 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003399 } \
3400} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003401 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003402 }
3403
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003404static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003405call_function(PyObject ***pp_stack, int oparg
3406#ifdef WITH_TSC
3407 , uint64* pintr0, uint64* pintr1
3408#endif
3409 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003410{
3411 int na = oparg & 0xff;
3412 int nk = (oparg>>8) & 0xff;
3413 int n = na + 2 * nk;
3414 PyObject **pfunc = (*pp_stack) - n - 1;
3415 PyObject *func = *pfunc;
3416 PyObject *x, *w;
3417
Jeremy Hylton985eba52003-02-05 23:13:00 +00003418 /* Always dispatch PyCFunction first, because these are
3419 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003420 */
3421 if (PyCFunction_Check(func) && nk == 0) {
3422 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003423 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003424
3425 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003426 if (flags & (METH_NOARGS | METH_O)) {
3427 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3428 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003429 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003430 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003431 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003432 else if (flags & METH_O && na == 1) {
3433 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003434 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003435 Py_DECREF(arg);
3436 }
3437 else {
3438 err_args(func, flags, na);
3439 x = NULL;
3440 }
3441 }
3442 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003443 PyObject *callargs;
3444 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003445 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003446 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003447 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003448 Py_XDECREF(callargs);
3449 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003450 } else {
3451 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3452 /* optimize access to bound methods */
3453 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003454 PCALL(PCALL_METHOD);
3455 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003456 Py_INCREF(self);
3457 func = PyMethod_GET_FUNCTION(func);
3458 Py_INCREF(func);
3459 Py_DECREF(*pfunc);
3460 *pfunc = self;
3461 na++;
3462 n++;
3463 } else
3464 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003465 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003466 if (PyFunction_Check(func))
3467 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003468 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003469 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003470 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003471 Py_DECREF(func);
3472 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003473
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00003474 /* Clear the stack of the function object. Also removes
3475 the arguments in case they weren't consumed already
3476 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003477 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003478 while ((*pp_stack) > pfunc) {
3479 w = EXT_POP(*pp_stack);
3480 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003481 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003482 }
3483 return x;
3484}
3485
Jeremy Hylton192690e2002-08-16 18:36:11 +00003486/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003487 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003488 For the simplest case -- a function that takes only positional
3489 arguments and is called with only positional arguments -- it
3490 inlines the most primitive frame setup code from
3491 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3492 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003493*/
3494
3495static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003496fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003497{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003498 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003499 PyObject *globals = PyFunction_GET_GLOBALS(func);
3500 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003501 PyObject *kwdefs = PyFunction_GET_KW_DEFAULTS(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003502 PyObject **d = NULL;
3503 int nd = 0;
3504
Jeremy Hylton985eba52003-02-05 23:13:00 +00003505 PCALL(PCALL_FUNCTION);
3506 PCALL(PCALL_FAST_FUNCTION);
Guido van Rossum4f72a782006-10-27 23:31:49 +00003507 if (argdefs == NULL && co->co_argcount == n &&
3508 co->co_kwonlyargcount == 0 && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003509 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3510 PyFrameObject *f;
3511 PyObject *retval = NULL;
3512 PyThreadState *tstate = PyThreadState_GET();
3513 PyObject **fastlocals, **stack;
3514 int i;
3515
3516 PCALL(PCALL_FASTER_FUNCTION);
3517 assert(globals != NULL);
3518 /* XXX Perhaps we should create a specialized
3519 PyFrame_New() that doesn't take locals, but does
3520 take builtins without sanity checking them.
3521 */
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003522 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003523 f = PyFrame_New(tstate, co, globals, NULL);
3524 if (f == NULL)
3525 return NULL;
3526
3527 fastlocals = f->f_localsplus;
3528 stack = (*pp_stack) - n;
3529
3530 for (i = 0; i < n; i++) {
3531 Py_INCREF(*stack);
3532 fastlocals[i] = *stack++;
3533 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003534 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003535 ++tstate->recursion_depth;
3536 Py_DECREF(f);
3537 --tstate->recursion_depth;
3538 return retval;
3539 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003540 if (argdefs != NULL) {
3541 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimes90aa7642007-12-19 02:45:37 +00003542 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003543 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003544 return PyEval_EvalCodeEx(co, globals,
3545 (PyObject *)NULL, (*pp_stack)-n, na,
Guido van Rossum4f72a782006-10-27 23:31:49 +00003546 (*pp_stack)-2*nk, nk, d, nd, kwdefs,
Jeremy Hylton985eba52003-02-05 23:13:00 +00003547 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003548}
3549
3550static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003551update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3552 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003553{
3554 PyObject *kwdict = NULL;
3555 if (orig_kwdict == NULL)
3556 kwdict = PyDict_New();
3557 else {
3558 kwdict = PyDict_Copy(orig_kwdict);
3559 Py_DECREF(orig_kwdict);
3560 }
3561 if (kwdict == NULL)
3562 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003563 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003564 int err;
3565 PyObject *value = EXT_POP(*pp_stack);
3566 PyObject *key = EXT_POP(*pp_stack);
3567 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersce272b62007-09-19 21:19:28 +00003568 PyErr_Format(PyExc_TypeError,
3569 "%.200s%s got multiple values "
3570 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003571 PyEval_GetFuncName(func),
3572 PyEval_GetFuncDesc(func),
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003573 _PyUnicode_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003574 Py_DECREF(key);
3575 Py_DECREF(value);
3576 Py_DECREF(kwdict);
3577 return NULL;
3578 }
3579 err = PyDict_SetItem(kwdict, key, value);
3580 Py_DECREF(key);
3581 Py_DECREF(value);
3582 if (err) {
3583 Py_DECREF(kwdict);
3584 return NULL;
3585 }
3586 }
3587 return kwdict;
3588}
3589
3590static PyObject *
3591update_star_args(int nstack, int nstar, PyObject *stararg,
3592 PyObject ***pp_stack)
3593{
3594 PyObject *callargs, *w;
3595
3596 callargs = PyTuple_New(nstack + nstar);
3597 if (callargs == NULL) {
3598 return NULL;
3599 }
3600 if (nstar) {
3601 int i;
3602 for (i = 0; i < nstar; i++) {
3603 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3604 Py_INCREF(a);
3605 PyTuple_SET_ITEM(callargs, nstack + i, a);
3606 }
3607 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003608 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003609 w = EXT_POP(*pp_stack);
3610 PyTuple_SET_ITEM(callargs, nstack, w);
3611 }
3612 return callargs;
3613}
3614
3615static PyObject *
3616load_args(PyObject ***pp_stack, int na)
3617{
3618 PyObject *args = PyTuple_New(na);
3619 PyObject *w;
3620
3621 if (args == NULL)
3622 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003623 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003624 w = EXT_POP(*pp_stack);
3625 PyTuple_SET_ITEM(args, na, w);
3626 }
3627 return args;
3628}
3629
3630static PyObject *
3631do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3632{
3633 PyObject *callargs = NULL;
3634 PyObject *kwdict = NULL;
3635 PyObject *result = NULL;
3636
3637 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003638 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003639 if (kwdict == NULL)
3640 goto call_fail;
3641 }
3642 callargs = load_args(pp_stack, na);
3643 if (callargs == NULL)
3644 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003645#ifdef CALL_PROFILE
3646 /* At this point, we have to look at the type of func to
3647 update the call stats properly. Do it here so as to avoid
3648 exposing the call stats machinery outside ceval.c
3649 */
3650 if (PyFunction_Check(func))
3651 PCALL(PCALL_FUNCTION);
3652 else if (PyMethod_Check(func))
3653 PCALL(PCALL_METHOD);
3654 else if (PyType_Check(func))
3655 PCALL(PCALL_TYPE);
3656 else
3657 PCALL(PCALL_OTHER);
3658#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003659 result = PyObject_Call(func, callargs, kwdict);
Thomas Wouters7ce29ca2007-09-19 21:56:32 +00003660call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003661 Py_XDECREF(callargs);
3662 Py_XDECREF(kwdict);
3663 return result;
3664}
3665
3666static PyObject *
3667ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3668{
3669 int nstar = 0;
3670 PyObject *callargs = NULL;
3671 PyObject *stararg = NULL;
3672 PyObject *kwdict = NULL;
3673 PyObject *result = NULL;
3674
3675 if (flags & CALL_FLAG_KW) {
3676 kwdict = EXT_POP(*pp_stack);
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003677 if (!PyDict_Check(kwdict)) {
3678 PyObject *d;
3679 d = PyDict_New();
3680 if (d == NULL)
3681 goto ext_call_fail;
3682 if (PyDict_Update(d, kwdict) != 0) {
3683 Py_DECREF(d);
3684 /* PyDict_Update raises attribute
3685 * error (percolated from an attempt
3686 * to get 'keys' attribute) instead of
3687 * a type error if its second argument
3688 * is not a mapping.
3689 */
3690 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
3691 PyErr_Format(PyExc_TypeError,
3692 "%.200s%.200s argument after ** "
3693 "must be a mapping, not %.200s",
3694 PyEval_GetFuncName(func),
3695 PyEval_GetFuncDesc(func),
3696 kwdict->ob_type->tp_name);
3697 }
3698 goto ext_call_fail;
3699 }
3700 Py_DECREF(kwdict);
3701 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00003702 }
3703 }
3704 if (flags & CALL_FLAG_VAR) {
3705 stararg = EXT_POP(*pp_stack);
3706 if (!PyTuple_Check(stararg)) {
3707 PyObject *t = NULL;
3708 t = PySequence_Tuple(stararg);
3709 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003710 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3711 PyErr_Format(PyExc_TypeError,
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003712 "%.200s%.200s argument after * "
3713 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003714 PyEval_GetFuncName(func),
Guido van Rossumd59da4b2007-05-22 18:11:13 +00003715 PyEval_GetFuncDesc(func),
3716 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003717 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003718 goto ext_call_fail;
3719 }
3720 Py_DECREF(stararg);
3721 stararg = t;
3722 }
3723 nstar = PyTuple_GET_SIZE(stararg);
3724 }
3725 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003726 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003727 if (kwdict == NULL)
3728 goto ext_call_fail;
3729 }
3730 callargs = update_star_args(na, nstar, stararg, pp_stack);
3731 if (callargs == NULL)
3732 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003733#ifdef CALL_PROFILE
3734 /* At this point, we have to look at the type of func to
3735 update the call stats properly. Do it here so as to avoid
3736 exposing the call stats machinery outside ceval.c
3737 */
3738 if (PyFunction_Check(func))
3739 PCALL(PCALL_FUNCTION);
3740 else if (PyMethod_Check(func))
3741 PCALL(PCALL_METHOD);
3742 else if (PyType_Check(func))
3743 PCALL(PCALL_TYPE);
3744 else
3745 PCALL(PCALL_OTHER);
3746#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003747 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersce272b62007-09-19 21:19:28 +00003748ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00003749 Py_XDECREF(callargs);
3750 Py_XDECREF(kwdict);
3751 Py_XDECREF(stararg);
3752 return result;
3753}
3754
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003755/* Extract a slice index from a PyInt or PyLong or an object with the
3756 nb_index slot defined, and store in *pi.
3757 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3758 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 +00003759 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003760*/
Tim Petersb5196382001-12-16 19:44:20 +00003761/* Note: If v is NULL, return success without storing into *pi. This
3762 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3763 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003764*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003765int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003766_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003767{
Tim Petersb5196382001-12-16 19:44:20 +00003768 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003769 Py_ssize_t x;
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +00003770 if (PyIndex_Check(v)) {
Thomas Wouters00ee7ba2006-08-21 19:07:27 +00003771 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003772 if (x == -1 && PyErr_Occurred())
3773 return 0;
3774 }
3775 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003776 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003777 "slice indices must be integers or "
3778 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003779 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003780 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003781 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003782 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003783 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003784}
3785
Guido van Rossum486364b2007-06-30 05:01:58 +00003786#define CANNOT_CATCH_MSG "catching classes that do not inherit from "\
Brett Cannon39590462007-02-26 22:01:14 +00003787 "BaseException is not allowed"
Brett Cannonf74225d2007-02-26 21:10:16 +00003788
Guido van Rossumb209a111997-04-29 18:18:01 +00003789static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003790cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003791{
Guido van Rossumac7be682001-01-17 15:42:30 +00003792 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003793 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003794 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003795 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003796 break;
3797 case PyCmp_IS_NOT:
3798 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003799 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003800 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003801 res = PySequence_Contains(w, v);
3802 if (res < 0)
3803 return NULL;
3804 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003805 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003806 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003807 if (res < 0)
3808 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003809 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003810 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003811 case PyCmp_EXC_MATCH:
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003812 if (PyTuple_Check(w)) {
3813 Py_ssize_t i, length;
3814 length = PyTuple_Size(w);
3815 for (i = 0; i < length; i += 1) {
3816 PyObject *exc = PyTuple_GET_ITEM(w, i);
Brett Cannon39590462007-02-26 22:01:14 +00003817 if (!PyExceptionClass_Check(exc)) {
3818 PyErr_SetString(PyExc_TypeError,
3819 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003820 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003821 }
3822 }
3823 }
3824 else {
Brett Cannon39590462007-02-26 22:01:14 +00003825 if (!PyExceptionClass_Check(w)) {
3826 PyErr_SetString(PyExc_TypeError,
3827 CANNOT_CATCH_MSG);
Brett Cannonf74225d2007-02-26 21:10:16 +00003828 return NULL;
Thomas Wouters9fe394c2007-02-05 01:24:16 +00003829 }
3830 }
Barry Warsaw4249f541997-08-22 21:26:19 +00003831 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003832 break;
3833 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003834 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003835 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003836 v = res ? Py_True : Py_False;
3837 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003838 return v;
3839}
3840
Thomas Wouters52152252000-08-17 22:55:00 +00003841static PyObject *
3842import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00003843{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003844 PyObject *x;
3845
3846 x = PyObject_GetAttr(v, name);
3847 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Walter Dörwald573c08c2007-05-25 15:46:59 +00003848 PyErr_Format(PyExc_ImportError, "cannot import name %S", name);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003849 }
Thomas Wouters52152252000-08-17 22:55:00 +00003850 return x;
3851}
Guido van Rossumac7be682001-01-17 15:42:30 +00003852
Thomas Wouters52152252000-08-17 22:55:00 +00003853static int
3854import_all_from(PyObject *locals, PyObject *v)
3855{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003856 PyObject *all = PyObject_GetAttrString(v, "__all__");
3857 PyObject *dict, *name, *value;
3858 int skip_leading_underscores = 0;
3859 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00003860
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003861 if (all == NULL) {
3862 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3863 return -1; /* Unexpected error */
3864 PyErr_Clear();
3865 dict = PyObject_GetAttrString(v, "__dict__");
3866 if (dict == NULL) {
3867 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
3868 return -1;
3869 PyErr_SetString(PyExc_ImportError,
3870 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00003871 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003872 }
3873 all = PyMapping_Keys(dict);
3874 Py_DECREF(dict);
3875 if (all == NULL)
3876 return -1;
3877 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003878 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003879
3880 for (pos = 0, err = 0; ; pos++) {
3881 name = PySequence_GetItem(all, pos);
3882 if (name == NULL) {
3883 if (!PyErr_ExceptionMatches(PyExc_IndexError))
3884 err = -1;
3885 else
3886 PyErr_Clear();
3887 break;
3888 }
3889 if (skip_leading_underscores &&
Martin v. Löwis5b222132007-06-10 09:51:05 +00003890 PyUnicode_Check(name) &&
3891 PyUnicode_AS_UNICODE(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003892 {
3893 Py_DECREF(name);
3894 continue;
3895 }
3896 value = PyObject_GetAttr(v, name);
3897 if (value == NULL)
3898 err = -1;
Thomas Wouters89f507f2006-12-13 04:49:30 +00003899 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003900 err = PyDict_SetItem(locals, name, value);
Thomas Wouters89f507f2006-12-13 04:49:30 +00003901 else
3902 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003903 Py_DECREF(name);
3904 Py_XDECREF(value);
3905 if (err != 0)
3906 break;
3907 }
3908 Py_DECREF(all);
3909 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00003910}
3911
Guido van Rossumac7be682001-01-17 15:42:30 +00003912static void
Neal Norwitzda059e32007-08-26 05:33:45 +00003913format_exc_check_arg(PyObject *exc, const char *format_str, PyObject *obj)
Paul Prescode68140d2000-08-30 20:25:01 +00003914{
Neal Norwitzda059e32007-08-26 05:33:45 +00003915 const char *obj_str;
Paul Prescode68140d2000-08-30 20:25:01 +00003916
3917 if (!obj)
3918 return;
3919
Marc-André Lemburg4cc0f242008-08-07 18:54:33 +00003920 obj_str = _PyUnicode_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00003921 if (!obj_str)
3922 return;
3923
3924 PyErr_Format(exc, format_str, obj_str);
3925}
Guido van Rossum950361c1997-01-24 13:49:28 +00003926
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003927static PyObject *
Guido van Rossum98297ee2007-11-06 21:34:58 +00003928unicode_concatenate(PyObject *v, PyObject *w,
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003929 PyFrameObject *f, unsigned char *next_instr)
3930{
3931 /* This function implements 'variable += expr' when both arguments
Guido van Rossum98297ee2007-11-06 21:34:58 +00003932 are (Unicode) strings. */
3933 Py_ssize_t v_len = PyUnicode_GET_SIZE(v);
3934 Py_ssize_t w_len = PyUnicode_GET_SIZE(w);
Thomas Wouters0e3f5912006-08-11 14:57:12 +00003935 Py_ssize_t new_len = v_len + w_len;
3936 if (new_len < 0) {
3937 PyErr_SetString(PyExc_OverflowError,
3938 "strings are too large to concat");
3939 return NULL;
3940 }
Thomas Wouters477c8d52006-05-27 19:21:47 +00003941
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003942 if (v->ob_refcnt == 2) {
3943 /* In the common case, there are 2 references to the value
3944 * stored in 'variable' when the += is performed: one on the
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003945 * value stack (in 'v') and one still stored in the
3946 * 'variable'. We try to delete the variable now to reduce
3947 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003948 */
3949 switch (*next_instr) {
3950 case STORE_FAST:
3951 {
3952 int oparg = PEEKARG();
3953 PyObject **fastlocals = f->f_localsplus;
3954 if (GETLOCAL(oparg) == v)
3955 SETLOCAL(oparg, NULL);
3956 break;
3957 }
3958 case STORE_DEREF:
3959 {
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003960 PyObject **freevars = (f->f_localsplus +
3961 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003962 PyObject *c = freevars[PEEKARG()];
3963 if (PyCell_GET(c) == v)
3964 PyCell_Set(c, NULL);
3965 break;
3966 }
3967 case STORE_NAME:
3968 {
3969 PyObject *names = f->f_code->co_names;
3970 PyObject *name = GETITEM(names, PEEKARG());
3971 PyObject *locals = f->f_locals;
3972 if (PyDict_CheckExact(locals) &&
3973 PyDict_GetItem(locals, name) == v) {
3974 if (PyDict_DelItem(locals, name) != 0) {
3975 PyErr_Clear();
3976 }
3977 }
3978 break;
3979 }
3980 }
3981 }
3982
Guido van Rossum98297ee2007-11-06 21:34:58 +00003983 if (v->ob_refcnt == 1 && !PyUnicode_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003984 /* Now we own the last reference to 'v', so we can resize it
3985 * in-place.
3986 */
Guido van Rossum98297ee2007-11-06 21:34:58 +00003987 if (PyUnicode_Resize(&v, new_len) != 0) {
3988 /* XXX if PyUnicode_Resize() fails, 'v' has been
Thomas Wouters8ce81f72007-09-20 18:22:40 +00003989 * deallocated so it cannot be put back into
3990 * 'variable'. The MemoryError is raised when there
3991 * is no value in 'variable', which might (very
3992 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003993 */
3994 return NULL;
3995 }
3996 /* copy 'w' into the newly allocated area of 'v' */
Guido van Rossum98297ee2007-11-06 21:34:58 +00003997 memcpy(PyUnicode_AS_UNICODE(v) + v_len,
3998 PyUnicode_AS_UNICODE(w), w_len*sizeof(Py_UNICODE));
Raymond Hettinger52a21b82004-08-06 18:43:09 +00003999 return v;
4000 }
4001 else {
4002 /* When in-place resizing is not an option. */
Guido van Rossum98297ee2007-11-06 21:34:58 +00004003 w = PyUnicode_Concat(v, w);
4004 Py_DECREF(v);
4005 return w;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004006 }
4007}
4008
Guido van Rossum950361c1997-01-24 13:49:28 +00004009#ifdef DYNAMIC_EXECUTION_PROFILE
4010
Skip Montanarof118cb12001-10-15 20:51:38 +00004011static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004012getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004013{
4014 int i;
4015 PyObject *l = PyList_New(256);
4016 if (l == NULL) return NULL;
4017 for (i = 0; i < 256; i++) {
Christian Heimes217cfd12007-12-02 14:31:20 +00004018 PyObject *x = PyLong_FromLong(a[i]);
Guido van Rossum950361c1997-01-24 13:49:28 +00004019 if (x == NULL) {
4020 Py_DECREF(l);
4021 return NULL;
4022 }
4023 PyList_SetItem(l, i, x);
4024 }
4025 for (i = 0; i < 256; i++)
4026 a[i] = 0;
4027 return l;
4028}
4029
4030PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004031_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004032{
4033#ifndef DXPAIRS
4034 return getarray(dxp);
4035#else
4036 int i;
4037 PyObject *l = PyList_New(257);
4038 if (l == NULL) return NULL;
4039 for (i = 0; i < 257; i++) {
4040 PyObject *x = getarray(dxpairs[i]);
4041 if (x == NULL) {
4042 Py_DECREF(l);
4043 return NULL;
4044 }
4045 PyList_SetItem(l, i, x);
4046 }
4047 return l;
4048#endif
4049}
4050
4051#endif