blob: 67eefba07ad80ab249bf3d1d199846ec3e213584 [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
Fredrik Lundh7a830892006-05-27 10:39:48 +00009/* enable more aggressive intra-module optimizations, where available */
Fredrik Lundh57640f52006-05-26 11:54:04 +000010#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
Tim Peters7df5e7f2006-05-26 23:14:37 +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
31 section should work for GCC on any PowerPC platform,
32 irrespective of OS. POWER? Who knows :-) */
33
Michael W. Hudson75eabd22005-01-18 15:56:11 +000034#define READ_TIMESTAMP(var) ppc_getcounter(&var)
Michael W. Hudson800ba232004-08-12 18:19:17 +000035
Fredrik Lundh7a830892006-05-27 10:39:48 +000036static void
Michael W. Hudson800ba232004-08-12 18:19:17 +000037ppc_getcounter(uint64 *v)
38{
39 register unsigned long tbu, tb, tbu2;
40
41 loop:
42 asm volatile ("mftbu %0" : "=r" (tbu) );
43 asm volatile ("mftb %0" : "=r" (tb) );
44 asm volatile ("mftbu %0" : "=r" (tbu2));
45 if (__builtin_expect(tbu != tbu2, 0)) goto loop;
46
Tim Peters7df5e7f2006-05-26 23:14:37 +000047 /* The slightly peculiar way of writing the next lines is
Michael W. Hudson800ba232004-08-12 18:19:17 +000048 compiled better by GCC than any other way I tried. */
49 ((long*)(v))[0] = tbu;
50 ((long*)(v))[1] = tb;
51}
52
Michael W. Hudson75eabd22005-01-18 15:56:11 +000053#else /* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000054
Michael W. Hudson75eabd22005-01-18 15:56:11 +000055#define READ_TIMESTAMP(val) \
56 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000057
58#endif
59
Tim Peters7df5e7f2006-05-26 23:14:37 +000060void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000061 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
62{
63 uint64 intr, inst, loop;
64 PyThreadState *tstate = PyThreadState_Get();
65 if (!tstate->interp->tscdump)
66 return;
67 intr = intr1 - intr0;
68 inst = inst1 - inst0 - intr;
69 loop = loop1 - loop0 - intr;
70 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
71 opcode, ticked, inst, loop);
72}
Michael W. Hudson800ba232004-08-12 18:19:17 +000073
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000074#endif
75
Guido van Rossum04691fc1992-08-12 15:35:34 +000076/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000077/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000078
Guido van Rossum408027e1996-12-30 16:17:54 +000079#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000080/* For debugging the interpreter: */
81#define LLTRACE 1 /* Low-level trace feature */
82#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +000083#endif
84
Jeremy Hylton52820442001-01-03 23:52:36 +000085typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +000086
Guido van Rossum374a9221991-04-04 10:40:29 +000087/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000088#ifdef WITH_TSC
Fredrik Lundh7a830892006-05-27 10:39:48 +000089static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000090#else
Fredrik Lundh7a830892006-05-27 10:39:48 +000091static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000092#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +000093static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
94static PyObject * do_call(PyObject *, PyObject ***, int, int);
95static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
96static PyObject * update_keyword_args(PyObject *, int, PyObject ***,PyObject *);
97static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
98static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +000099#define CALL_FLAG_VAR 1
100#define CALL_FLAG_KW 2
101
Guido van Rossum0a066c01992-03-27 17:29:15 +0000102#ifdef LLTRACE
Fredrik Lundh1b949402006-05-26 12:01:49 +0000103static int lltrace;
Fredrik Lundh7a830892006-05-27 10:39:48 +0000104static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000105#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +0000106static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Fred Drake5755ce62001-06-27 19:19:46 +0000107 int, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000108static void call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000109 PyFrameObject *, int, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000110static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
111static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Armin Rigobf57a142004-03-22 19:24:58 +0000112 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000113
Fredrik Lundh7a830892006-05-27 10:39:48 +0000114static PyObject * apply_slice(PyObject *, PyObject *, PyObject *);
115static int assign_slice(PyObject *, PyObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000116 PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000117static PyObject * cmp_outcome(int, PyObject *, PyObject *);
118static PyObject * import_from(PyObject *, PyObject *);
119static int import_all_from(PyObject *, PyObject *);
120static PyObject * build_class(PyObject *, PyObject *, PyObject *);
121static int exec_statement(PyFrameObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000122 PyObject *, PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000123static void set_exc_info(PyThreadState *, PyObject *, PyObject *, PyObject *);
124static void reset_exc_info(PyThreadState *);
125static void format_exc_check_arg(PyObject *, char *, PyObject *);
126static PyObject * string_concatenate(PyObject *, PyObject *,
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000127 PyFrameObject *, unsigned char *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000128
Paul Prescode68140d2000-08-30 20:25:01 +0000129#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000130 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000131#define GLOBAL_NAME_ERROR_MSG \
132 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000133#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000134 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000135#define UNBOUNDFREE_ERROR_MSG \
136 "free variable '%.200s' referenced before assignment" \
137 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000138
Guido van Rossum950361c1997-01-24 13:49:28 +0000139/* Dynamic execution profile */
140#ifdef DYNAMIC_EXECUTION_PROFILE
141#ifdef DXPAIRS
142static long dxpairs[257][256];
143#define dxp dxpairs[256]
144#else
145static long dxp[256];
146#endif
147#endif
148
Jeremy Hylton985eba52003-02-05 23:13:00 +0000149/* Function call profile */
150#ifdef CALL_PROFILE
151#define PCALL_NUM 11
152static int pcall[PCALL_NUM];
153
154#define PCALL_ALL 0
155#define PCALL_FUNCTION 1
156#define PCALL_FAST_FUNCTION 2
157#define PCALL_FASTER_FUNCTION 3
158#define PCALL_METHOD 4
159#define PCALL_BOUND_METHOD 5
160#define PCALL_CFUNCTION 6
161#define PCALL_TYPE 7
162#define PCALL_GENERATOR 8
163#define PCALL_OTHER 9
164#define PCALL_POP 10
165
166/* Notes about the statistics
167
168 PCALL_FAST stats
169
170 FAST_FUNCTION means no argument tuple needs to be created.
171 FASTER_FUNCTION means that the fast-path frame setup code is used.
172
173 If there is a method call where the call can be optimized by changing
174 the argument tuple and calling the function directly, it gets recorded
175 twice.
176
177 As a result, the relationship among the statistics appears to be
178 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
179 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
180 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
181 PCALL_METHOD > PCALL_BOUND_METHOD
182*/
183
184#define PCALL(POS) pcall[POS]++
185
186PyObject *
187PyEval_GetCallStats(PyObject *self)
188{
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000189 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000190 pcall[0], pcall[1], pcall[2], pcall[3],
191 pcall[4], pcall[5], pcall[6], pcall[7],
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000192 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000193}
194#else
195#define PCALL(O)
196
197PyObject *
198PyEval_GetCallStats(PyObject *self)
199{
200 Py_INCREF(Py_None);
201 return Py_None;
202}
203#endif
204
Tim Peters5ca576e2001-06-18 22:08:13 +0000205
Guido van Rossume59214e1994-08-30 08:01:59 +0000206#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000207
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000208#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000209#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000210#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000211#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000212
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000213static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Guido van Rossuma9672091994-09-14 13:31:22 +0000214static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000215
Tim Peters7f468f22004-10-11 02:40:51 +0000216int
217PyEval_ThreadsInitialized(void)
218{
219 return interpreter_lock != 0;
220}
221
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000222void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000223PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000224{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000225 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000226 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000227 interpreter_lock = PyThread_allocate_lock();
228 PyThread_acquire_lock(interpreter_lock, 1);
229 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000230}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000231
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000232void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000233PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000234{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000235 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000236}
237
238void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000239PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000240{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000241 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000242}
243
244void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000245PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000246{
247 if (tstate == NULL)
248 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000249 /* Check someone has called PyEval_InitThreads() to create the lock */
250 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000251 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000252 if (PyThreadState_Swap(tstate) != NULL)
253 Py_FatalError(
254 "PyEval_AcquireThread: non-NULL old thread state");
255}
256
257void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000258PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000259{
260 if (tstate == NULL)
261 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
262 if (PyThreadState_Swap(NULL) != tstate)
263 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000264 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000265}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000266
267/* This function is called from PyOS_AfterFork to ensure that newly
268 created child processes don't hold locks referring to threads which
269 are not running in the child process. (This could also be done using
270 pthread_atfork mechanism, at least for the pthreads implementation.) */
271
272void
273PyEval_ReInitThreads(void)
274{
275 if (!interpreter_lock)
276 return;
277 /*XXX Can't use PyThread_free_lock here because it does too
278 much error-checking. Doing this cleanly would require
279 adding a new function to each thread_*.h. Instead, just
280 create a new lock and waste a little bit of memory */
281 interpreter_lock = PyThread_allocate_lock();
282 PyThread_acquire_lock(interpreter_lock, 1);
283 main_thread = PyThread_get_thread_ident();
284}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000285#endif
286
Guido van Rossumff4949e1992-08-05 19:58:53 +0000287/* Functions save_thread and restore_thread are always defined so
288 dynamically loaded modules needn't be compiled separately for use
289 with and without threads: */
290
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000291PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000292PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000293{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000294 PyThreadState *tstate = PyThreadState_Swap(NULL);
295 if (tstate == NULL)
296 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000297#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000298 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000299 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000300#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000301 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000302}
303
304void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000305PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000306{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000307 if (tstate == NULL)
308 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000309#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000310 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000311 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000312 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000313 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000314 }
315#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000316 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000317}
318
319
Guido van Rossuma9672091994-09-14 13:31:22 +0000320/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
321 signal handlers or Mac I/O completion routines) can schedule calls
322 to a function to be called synchronously.
323 The synchronous function is called with one void* argument.
324 It should return 0 for success or -1 for failure -- failure should
325 be accompanied by an exception.
326
327 If registry succeeds, the registry function returns 0; if it fails
328 (e.g. due to too many pending calls) it returns -1 (without setting
329 an exception condition).
330
331 Note that because registry may occur from within signal handlers,
332 or other asynchronous events, calling malloc() is unsafe!
333
334#ifdef WITH_THREAD
335 Any thread can schedule pending calls, but only the main thread
336 will execute them.
337#endif
338
339 XXX WARNING! ASYNCHRONOUSLY EXECUTING CODE!
340 There are two possible race conditions:
341 (1) nested asynchronous registry calls;
342 (2) registry calls made while pending calls are being processed.
343 While (1) is very unlikely, (2) is a real possibility.
344 The current code is safe against (2), but not against (1).
345 The safety against (2) is derived from the fact that only one
346 thread (the main thread) ever takes things out of the queue.
Guido van Rossuma9672091994-09-14 13:31:22 +0000347
Guido van Rossuma027efa1997-05-05 20:56:21 +0000348 XXX Darn! With the advent of thread state, we should have an array
349 of pending calls per thread in the thread state! Later...
350*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000351
Guido van Rossuma9672091994-09-14 13:31:22 +0000352#define NPENDINGCALLS 32
353static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000354 int (*func)(void *);
355 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000356} pendingcalls[NPENDINGCALLS];
357static volatile int pendingfirst = 0;
358static volatile int pendinglast = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000359static volatile int things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000360
361int
Thomas Wouters334fb892000-07-25 12:56:38 +0000362Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000363{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000364 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000365 int i, j;
366 /* XXX Begin critical section */
367 /* XXX If you want this to be safe against nested
368 XXX asynchronous calls, you'll have to work harder! */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000369 if (busy)
370 return -1;
371 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000372 i = pendinglast;
373 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000374 if (j == pendingfirst) {
375 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000376 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000377 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000378 pendingcalls[i].func = func;
379 pendingcalls[i].arg = arg;
380 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000381
382 _Py_Ticker = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000383 things_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000384 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000385 /* XXX End critical section */
386 return 0;
387}
388
Guido van Rossum180d7b41994-09-29 09:45:57 +0000389int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000390Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000391{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000392 static int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000393#ifdef WITH_THREAD
Guido van Rossum65d5b571998-12-21 19:32:43 +0000394 if (main_thread && PyThread_get_thread_ident() != main_thread)
Guido van Rossuma9672091994-09-14 13:31:22 +0000395 return 0;
396#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000397 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000398 return 0;
399 busy = 1;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000400 things_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000401 for (;;) {
402 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000403 int (*func)(void *);
404 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000405 i = pendingfirst;
406 if (i == pendinglast)
407 break; /* Queue empty */
408 func = pendingcalls[i].func;
409 arg = pendingcalls[i].arg;
410 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000411 if (func(arg) < 0) {
412 busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000413 things_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000414 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000415 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000416 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000417 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000418 return 0;
419}
420
421
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000422/* The interpreter's recursion limit */
423
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000424#ifndef Py_DEFAULT_RECURSION_LIMIT
425#define Py_DEFAULT_RECURSION_LIMIT 1000
426#endif
427static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
428int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000429
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000430int
431Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000432{
433 return recursion_limit;
434}
435
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000436void
437Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000438{
439 recursion_limit = new_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000440 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000441}
442
Armin Rigo2b3eb402003-10-28 12:05:48 +0000443/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
444 if the recursion_depth reaches _Py_CheckRecursionLimit.
445 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
446 to guarantee that _Py_CheckRecursiveCall() is regularly called.
447 Without USE_STACKCHECK, there is no need for this. */
448int
449_Py_CheckRecursiveCall(char *where)
450{
451 PyThreadState *tstate = PyThreadState_GET();
452
453#ifdef USE_STACKCHECK
454 if (PyOS_CheckStack()) {
455 --tstate->recursion_depth;
456 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
457 return -1;
458 }
459#endif
460 if (tstate->recursion_depth > recursion_limit) {
461 --tstate->recursion_depth;
462 PyErr_Format(PyExc_RuntimeError,
463 "maximum recursion depth exceeded%s",
464 where);
465 return -1;
466 }
467 _Py_CheckRecursionLimit = recursion_limit;
468 return 0;
469}
470
Guido van Rossum374a9221991-04-04 10:40:29 +0000471/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000472enum why_code {
473 WHY_NOT = 0x0001, /* No error */
474 WHY_EXCEPTION = 0x0002, /* Exception occurred */
475 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
476 WHY_RETURN = 0x0008, /* 'return' statement */
477 WHY_BREAK = 0x0010, /* 'break' statement */
478 WHY_CONTINUE = 0x0020, /* 'continue' statement */
479 WHY_YIELD = 0x0040 /* 'yield' operator */
480};
Guido van Rossum374a9221991-04-04 10:40:29 +0000481
Fredrik Lundh7a830892006-05-27 10:39:48 +0000482static enum why_code do_raise(PyObject *, PyObject *, PyObject *);
483static int unpack_iterable(PyObject *, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000484
Skip Montanarod581d772002-09-03 20:10:45 +0000485/* for manipulating the thread switch and periodic "stuff" - used to be
486 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000487int _Py_CheckInterval = 100;
488volatile int _Py_Ticker = 100;
Guido van Rossum374a9221991-04-04 10:40:29 +0000489
Guido van Rossumb209a111997-04-29 18:18:01 +0000490PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000491PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000492{
Jeremy Hylton985eba52003-02-05 23:13:00 +0000493 /* XXX raise SystemError if globals is NULL */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000494 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000495 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000496 (PyObject **)NULL, 0,
497 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000498 (PyObject **)NULL, 0,
499 NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000500}
501
502
503/* Interpreter main loop */
504
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000505PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000506PyEval_EvalFrame(PyFrameObject *f) {
507 /* This is for backward compatibility with extension modules that
508 used this API; core interpreter code should call PyEval_EvalFrameEx() */
509 return PyEval_EvalFrameEx(f, 0);
510}
511
512PyObject *
Anthony Baxtera863d332006-04-11 07:43:46 +0000513PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000514{
Guido van Rossum950361c1997-01-24 13:49:28 +0000515#ifdef DXPAIRS
516 int lastopcode = 0;
517#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +0000518 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000519 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000520 register int opcode; /* Current opcode */
521 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000522 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000523 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000524 register PyObject *x; /* Result object -- NULL if error */
525 register PyObject *v; /* Temporary objects popped off stack */
526 register PyObject *w;
527 register PyObject *u;
528 register PyObject *t;
Barry Warsaw23c9ec82000-08-21 15:44:01 +0000529 register PyObject *stream = NULL; /* for PRINT opcodes */
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000530 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000531 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000532 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000533 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000534
Tim Peters8a5c3c72004-04-05 19:36:21 +0000535 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000536
537 not (instr_lb <= current_bytecode_offset < instr_ub)
538
Tim Peters8a5c3c72004-04-05 19:36:21 +0000539 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000540 initial values are such as to make this false the first
541 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000542 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000543
Guido van Rossumd076c731998-10-07 19:42:25 +0000544 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000545 PyObject *names;
546 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000547#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000548 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000549 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000550#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000551
Neal Norwitza81d2202002-07-14 00:27:26 +0000552/* Tuple access macros */
553
554#ifndef Py_DEBUG
555#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
556#else
557#define GETITEM(v, i) PyTuple_GetItem((v), (i))
558#endif
559
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000560#ifdef WITH_TSC
561/* Use Pentium timestamp counter to mark certain events:
562 inst0 -- beginning of switch statement for opcode dispatch
563 inst1 -- end of switch statement (may be skipped)
564 loop0 -- the top of the mainloop
Tim Peters7df5e7f2006-05-26 23:14:37 +0000565 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000566 (may be skipped)
567 intr1 -- beginning of long interruption
568 intr2 -- end of long interruption
569
570 Many opcodes call out to helper C functions. In some cases, the
571 time in those functions should be counted towards the time for the
572 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
573 calls another Python function; there's no point in charge all the
574 bytecode executed by the called function to the caller.
575
576 It's hard to make a useful judgement statically. In the presence
577 of operator overloading, it's impossible to tell if a call will
578 execute new Python code or not.
579
580 It's a case-by-case judgement. I'll use intr1 for the following
581 cases:
582
583 EXEC_STMT
584 IMPORT_STAR
585 IMPORT_FROM
586 CALL_FUNCTION (and friends)
587
588 */
589 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
590 int ticked = 0;
591
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000592 READ_TIMESTAMP(inst0);
593 READ_TIMESTAMP(inst1);
594 READ_TIMESTAMP(loop0);
595 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000596
597 /* shut up the compiler */
598 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000599#endif
600
Guido van Rossum374a9221991-04-04 10:40:29 +0000601/* Code access macros */
602
Martin v. Löwis18e16552006-02-15 17:27:45 +0000603#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000604#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000605#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000606#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000607#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000608#define JUMPBY(x) (next_instr += (x))
609
Raymond Hettingerf606f872003-03-16 03:11:04 +0000610/* OpCode prediction macros
611 Some opcodes tend to come in pairs thus making it possible to predict
612 the second code when the first is run. For example, COMPARE_OP is often
613 followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And, those opcodes are often
614 followed by a POP_TOP.
615
616 Verifying the prediction costs a single high-speed test of register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000617 variable against a constant. If the pairing was good, then the
Raymond Hettingerf606f872003-03-16 03:11:04 +0000618 processor has a high likelihood of making its own successful branch
619 prediction which results in a nearly zero overhead transition to the
620 next opcode.
621
622 A successful prediction saves a trip through the eval-loop including
623 its two unpredictable branches, the HASARG test and the switch-case.
Raymond Hettingera7216982004-02-08 19:59:27 +0000624
Tim Peters8a5c3c72004-04-05 19:36:21 +0000625 If collecting opcode statistics, turn off prediction so that
626 statistics are accurately maintained (the predictions bypass
Raymond Hettingera7216982004-02-08 19:59:27 +0000627 the opcode frequency counter updates).
Raymond Hettingerf606f872003-03-16 03:11:04 +0000628*/
629
Raymond Hettingera7216982004-02-08 19:59:27 +0000630#ifdef DYNAMIC_EXECUTION_PROFILE
631#define PREDICT(op) if (0) goto PRED_##op
632#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000633#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000634#endif
635
Raymond Hettingerf606f872003-03-16 03:11:04 +0000636#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000637#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000638
Guido van Rossum374a9221991-04-04 10:40:29 +0000639/* Stack manipulation macros */
640
Martin v. Löwis18e16552006-02-15 17:27:45 +0000641/* The stack can grow at most MAXINT deep, as co_nlocals and
642 co_stacksize are ints. */
643#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000644#define EMPTY() (STACK_LEVEL() == 0)
645#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000646#define SECOND() (stack_pointer[-2])
647#define THIRD() (stack_pointer[-3])
648#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000649#define SET_TOP(v) (stack_pointer[-1] = (v))
650#define SET_SECOND(v) (stack_pointer[-2] = (v))
651#define SET_THIRD(v) (stack_pointer[-3] = (v))
652#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000653#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000654#define BASIC_PUSH(v) (*stack_pointer++ = (v))
655#define BASIC_POP() (*--stack_pointer)
656
Guido van Rossum96a42c81992-01-12 02:29:51 +0000657#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000658#define PUSH(v) { (void)(BASIC_PUSH(v), \
659 lltrace && prtrace(TOP(), "push")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000660 assert(STACK_LEVEL() <= co->co_stacksize); }
Fred Drakede26cfc2001-10-13 06:11:28 +0000661#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000662#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
663 lltrace && prtrace(TOP(), "stackadj")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000664 assert(STACK_LEVEL() <= co->co_stacksize); }
Guido van Rossumc2e20742006-02-27 22:32:47 +0000665#define EXT_POP(STACK_POINTER) (lltrace && prtrace(*(STACK_POINTER), "ext_pop"), *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000666#else
667#define PUSH(v) BASIC_PUSH(v)
668#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000669#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000670#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000671#endif
672
Guido van Rossum681d79a1995-07-18 14:51:37 +0000673/* Local variable macros */
674
675#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000676
677/* The SETLOCAL() macro must not DECREF the local variable in-place and
678 then store the new value; it must copy the old value to a temporary
679 value, then store the new value, and then DECREF the temporary value.
680 This is because it is possible that during the DECREF the frame is
681 accessed by other code (e.g. a __del__ method or gc.collect()) and the
682 variable would be pointing to already-freed memory. */
683#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
684 GETLOCAL(i) = value; \
685 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000686
Guido van Rossuma027efa1997-05-05 20:56:21 +0000687/* Start of code */
688
Tim Peters5ca576e2001-06-18 22:08:13 +0000689 if (f == NULL)
690 return NULL;
691
Armin Rigo1d313ab2003-10-25 14:33:09 +0000692 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000693 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000694 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000695
Tim Peters5ca576e2001-06-18 22:08:13 +0000696 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000697
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000698 if (tstate->use_tracing) {
699 if (tstate->c_tracefunc != NULL) {
700 /* tstate->c_tracefunc, if defined, is a
701 function that will be called on *every* entry
702 to a code block. Its return value, if not
703 None, is a function that will be called at
704 the start of each executed line of code.
705 (Actually, the function must return itself
706 in order to continue tracing.) The trace
707 functions are called with three arguments:
708 a pointer to the current frame, a string
709 indicating why the function is called, and
710 an argument which depends on the situation.
711 The global trace function is also called
712 whenever an exception is detected. */
713 if (call_trace(tstate->c_tracefunc, tstate->c_traceobj,
714 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000715 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000716 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000717 }
718 }
719 if (tstate->c_profilefunc != NULL) {
720 /* Similar for c_profilefunc, except it needn't
721 return itself and isn't called for "line" events */
722 if (call_trace(tstate->c_profilefunc,
723 tstate->c_profileobj,
724 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000725 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000726 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000727 }
728 }
729 }
730
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000731 co = f->f_code;
732 names = co->co_names;
733 consts = co->co_consts;
734 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000735 freevars = f->f_localsplus + co->co_nlocals;
Brett Cannonc9371d42005-06-25 08:23:41 +0000736 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000737 /* An explanation is in order for the next line.
738
739 f->f_lasti now refers to the index of the last instruction
740 executed. You might think this was obvious from the name, but
741 this wasn't always true before 2.3! PyFrame_New now sets
742 f->f_lasti to -1 (i.e. the index *before* the first instruction)
743 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000744 does work. Promise.
745
746 When the PREDICT() macros are enabled, some opcode pairs follow in
747 direct succession without updating f->f_lasti. A successful
748 prediction effectively links the two codes together as if they
749 were a single new opcode; accordingly,f->f_lasti will point to
750 the first code in the pair (for instance, GET_ITER followed by
751 FOR_ITER is effectively a single opcode and f->f_lasti will point
752 at to the beginning of the combined pair.)
753 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000754 next_instr = first_instr + f->f_lasti + 1;
755 stack_pointer = f->f_stacktop;
756 assert(stack_pointer != NULL);
757 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
758
Tim Peters5ca576e2001-06-18 22:08:13 +0000759#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000760 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000761#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000762#if defined(Py_DEBUG) || defined(LLTRACE)
Tim Peters5ca576e2001-06-18 22:08:13 +0000763 filename = PyString_AsString(co->co_filename);
764#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000765
Guido van Rossum374a9221991-04-04 10:40:29 +0000766 why = WHY_NOT;
767 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000768 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000769 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000770
Anthony Baxtera863d332006-04-11 07:43:46 +0000771 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000772 why = WHY_EXCEPTION;
773 goto on_error;
774 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000775
Guido van Rossum374a9221991-04-04 10:40:29 +0000776 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000777#ifdef WITH_TSC
778 if (inst1 == 0) {
779 /* Almost surely, the opcode executed a break
780 or a continue, preventing inst1 from being set
781 on the way out of the loop.
782 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000783 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000784 loop1 = inst1;
785 }
786 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
787 intr0, intr1);
788 ticked = 0;
789 inst1 = 0;
790 intr0 = 0;
791 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000792 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000793#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000794 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000795 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000796
Guido van Rossuma027efa1997-05-05 20:56:21 +0000797 /* Do periodic things. Doing this every time through
798 the loop would add too much overhead, so we do it
799 only every Nth instruction. We also do it if
800 ``things_to_do'' is set, i.e. when an asynchronous
801 event needs attention (e.g. a signal handler or
802 async I/O handler); see Py_AddPendingCall() and
803 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000804
Skip Montanarod581d772002-09-03 20:10:45 +0000805 if (--_Py_Ticker < 0) {
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000806 if (*next_instr == SETUP_FINALLY) {
807 /* Make the last opcode before
808 a try: finally: block uninterruptable. */
809 goto fast_next_opcode;
810 }
Skip Montanarod581d772002-09-03 20:10:45 +0000811 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000812 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000813#ifdef WITH_TSC
814 ticked = 1;
815#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000816 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000817 if (Py_MakePendingCalls() < 0) {
818 why = WHY_EXCEPTION;
819 goto on_error;
820 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000821 if (things_to_do)
822 /* MakePendingCalls() didn't succeed.
823 Force early re-execution of this
824 "periodic" code, possibly after
825 a thread switch */
826 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000827 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000828#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000829 if (interpreter_lock) {
830 /* Give another thread a chance */
831
Guido van Rossum25ce5661997-08-02 03:10:38 +0000832 if (PyThreadState_Swap(NULL) != tstate)
833 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000834 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000835
836 /* Other threads may run now */
837
Guido van Rossum65d5b571998-12-21 19:32:43 +0000838 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000839 if (PyThreadState_Swap(tstate) != NULL)
840 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000841
842 /* Check for thread interrupts */
843
844 if (tstate->async_exc != NULL) {
845 x = tstate->async_exc;
846 tstate->async_exc = NULL;
847 PyErr_SetNone(x);
848 Py_DECREF(x);
849 why = WHY_EXCEPTION;
850 goto on_error;
851 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000852 }
853#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000854 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000855
Neil Schemenauer63543862002-02-17 19:10:14 +0000856 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000857 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000858
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000859 /* line-by-line tracing support */
860
861 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
862 /* see maybe_call_line_trace
863 for expository comments */
864 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000865
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000866 err = maybe_call_line_trace(tstate->c_tracefunc,
867 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000868 f, &instr_lb, &instr_ub,
869 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000870 /* Reload possibly changed frame fields */
871 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000872 if (f->f_stacktop != NULL) {
873 stack_pointer = f->f_stacktop;
874 f->f_stacktop = NULL;
875 }
876 if (err) {
877 /* trace function raised an exception */
878 goto on_error;
879 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000880 }
881
882 /* Extract opcode and argument */
883
Guido van Rossum374a9221991-04-04 10:40:29 +0000884 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000885 oparg = 0; /* allows oparg to be stored in a register because
886 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000887 if (HAS_ARG(opcode))
888 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000889 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000890#ifdef DYNAMIC_EXECUTION_PROFILE
891#ifdef DXPAIRS
892 dxpairs[lastopcode][opcode]++;
893 lastopcode = opcode;
894#endif
895 dxp[opcode]++;
896#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000897
Guido van Rossum96a42c81992-01-12 02:29:51 +0000898#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000899 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000900
Guido van Rossum96a42c81992-01-12 02:29:51 +0000901 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000902 if (HAS_ARG(opcode)) {
903 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000904 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000905 }
906 else {
907 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000908 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000909 }
910 }
911#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000912
Guido van Rossum374a9221991-04-04 10:40:29 +0000913 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000914 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000915
Guido van Rossum374a9221991-04-04 10:40:29 +0000916 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000917
Guido van Rossum374a9221991-04-04 10:40:29 +0000918 /* BEWARE!
919 It is essential that any operation that fails sets either
920 x to NULL, err to nonzero, or why to anything but WHY_NOT,
921 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000922
Guido van Rossum374a9221991-04-04 10:40:29 +0000923 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000924
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000925 case NOP:
926 goto fast_next_opcode;
927
Neil Schemenauer63543862002-02-17 19:10:14 +0000928 case LOAD_FAST:
929 x = GETLOCAL(oparg);
930 if (x != NULL) {
931 Py_INCREF(x);
932 PUSH(x);
933 goto fast_next_opcode;
934 }
935 format_exc_check_arg(PyExc_UnboundLocalError,
936 UNBOUNDLOCAL_ERROR_MSG,
937 PyTuple_GetItem(co->co_varnames, oparg));
938 break;
939
940 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000941 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000942 Py_INCREF(x);
943 PUSH(x);
944 goto fast_next_opcode;
945
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000946 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000947 case STORE_FAST:
948 v = POP();
949 SETLOCAL(oparg, v);
950 goto fast_next_opcode;
951
Raymond Hettingerf606f872003-03-16 03:11:04 +0000952 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000953 case POP_TOP:
954 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000955 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000956 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000957
Guido van Rossum374a9221991-04-04 10:40:29 +0000958 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000959 v = TOP();
960 w = SECOND();
961 SET_TOP(w);
962 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000963 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000964
Guido van Rossum374a9221991-04-04 10:40:29 +0000965 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000966 v = TOP();
967 w = SECOND();
968 x = THIRD();
969 SET_TOP(w);
970 SET_SECOND(x);
971 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000972 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000973
Thomas Wouters434d0822000-08-24 20:11:32 +0000974 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000975 u = TOP();
976 v = SECOND();
977 w = THIRD();
978 x = FOURTH();
979 SET_TOP(v);
980 SET_SECOND(w);
981 SET_THIRD(x);
982 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000983 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000984
Guido van Rossum374a9221991-04-04 10:40:29 +0000985 case DUP_TOP:
986 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000987 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000988 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000989 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000990
Thomas Wouters434d0822000-08-24 20:11:32 +0000991 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000992 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +0000993 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +0000994 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000995 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +0000996 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000997 STACKADJ(2);
998 SET_TOP(x);
999 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001000 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001001 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001002 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001003 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001004 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001005 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001006 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001007 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001008 STACKADJ(3);
1009 SET_TOP(x);
1010 SET_SECOND(w);
1011 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001012 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001013 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001014 Py_FatalError("invalid argument to DUP_TOPX"
1015 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001016 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001017
Guido van Rossum374a9221991-04-04 10:40:29 +00001018 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001019 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001020 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001021 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001022 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001023 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001024 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001025
Guido van Rossum374a9221991-04-04 10:40:29 +00001026 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001027 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001028 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001029 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001030 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001031 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001032 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001033
Guido van Rossum374a9221991-04-04 10:40:29 +00001034 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001035 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001036 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001037 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001038 if (err == 0) {
1039 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001040 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001041 continue;
1042 }
1043 else if (err > 0) {
1044 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001045 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001046 err = 0;
1047 continue;
1048 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001049 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001050 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001051
Guido van Rossum374a9221991-04-04 10:40:29 +00001052 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001053 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001054 x = PyObject_Repr(v);
1055 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001056 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001057 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001058 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001059
Guido van Rossum7928cd71991-10-24 14:59:31 +00001060 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001061 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001062 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001063 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001064 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001065 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001066 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001067
Guido van Rossum50564e81996-01-12 01:13:16 +00001068 case BINARY_POWER:
1069 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001070 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001071 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001072 Py_DECREF(v);
1073 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001074 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001075 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001076 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001077
Guido van Rossum374a9221991-04-04 10:40:29 +00001078 case BINARY_MULTIPLY:
1079 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001080 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001081 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001082 Py_DECREF(v);
1083 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001084 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001085 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001086 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001087
Guido van Rossum374a9221991-04-04 10:40:29 +00001088 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001089 if (!_Py_QnewFlag) {
1090 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001091 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001092 x = PyNumber_Divide(v, w);
1093 Py_DECREF(v);
1094 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001095 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001096 if (x != NULL) continue;
1097 break;
1098 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001099 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001100 BINARY_TRUE_DIVIDE */
1101 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001102 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001103 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001104 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001105 Py_DECREF(v);
1106 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001107 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001108 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001109 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001110
Guido van Rossum4668b002001-08-08 05:00:18 +00001111 case BINARY_FLOOR_DIVIDE:
1112 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001113 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001114 x = PyNumber_FloorDivide(v, w);
1115 Py_DECREF(v);
1116 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001117 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001118 if (x != NULL) continue;
1119 break;
1120
Guido van Rossum374a9221991-04-04 10:40:29 +00001121 case BINARY_MODULO:
1122 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001123 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001124 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001125 Py_DECREF(v);
1126 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001127 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001128 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001129 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001130
Guido van Rossum374a9221991-04-04 10:40:29 +00001131 case BINARY_ADD:
1132 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001133 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001134 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001135 /* INLINE: int + int */
1136 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001137 a = PyInt_AS_LONG(v);
1138 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001139 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001140 if ((i^a) < 0 && (i^b) < 0)
1141 goto slow_add;
1142 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001143 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001144 else if (PyString_CheckExact(v) &&
1145 PyString_CheckExact(w)) {
1146 x = string_concatenate(v, w, f, next_instr);
1147 /* string_concatenate consumed the ref to v */
1148 goto skip_decref_vx;
1149 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001150 else {
1151 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001152 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001153 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001154 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001155 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001156 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001157 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001158 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001159 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001160
Guido van Rossum374a9221991-04-04 10:40:29 +00001161 case BINARY_SUBTRACT:
1162 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001163 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001164 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001165 /* INLINE: int - int */
1166 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001167 a = PyInt_AS_LONG(v);
1168 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001169 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001170 if ((i^a) < 0 && (i^~b) < 0)
1171 goto slow_sub;
1172 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001173 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001174 else {
1175 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001176 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001177 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001178 Py_DECREF(v);
1179 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001180 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001181 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001182 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001183
Guido van Rossum374a9221991-04-04 10:40:29 +00001184 case BINARY_SUBSCR:
1185 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001186 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001187 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001188 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001189 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001190 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001191 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001192 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001193 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001194 Py_INCREF(x);
1195 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001196 else
1197 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001198 }
1199 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001200 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001201 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001202 Py_DECREF(v);
1203 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001204 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001205 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001206 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001207
Guido van Rossum7928cd71991-10-24 14:59:31 +00001208 case BINARY_LSHIFT:
1209 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001210 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001211 x = PyNumber_Lshift(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 Rossum7928cd71991-10-24 14:59:31 +00001216 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001217
Guido van Rossum7928cd71991-10-24 14:59:31 +00001218 case BINARY_RSHIFT:
1219 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001220 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001221 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001222 Py_DECREF(v);
1223 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001224 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001225 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001226 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001227
Guido van Rossum7928cd71991-10-24 14:59:31 +00001228 case BINARY_AND:
1229 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001230 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001231 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001232 Py_DECREF(v);
1233 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001234 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001235 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001236 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001237
Guido van Rossum7928cd71991-10-24 14:59:31 +00001238 case BINARY_XOR:
1239 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001240 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001241 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001242 Py_DECREF(v);
1243 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001244 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001245 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001246 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001247
Guido van Rossum7928cd71991-10-24 14:59:31 +00001248 case BINARY_OR:
1249 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001250 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001251 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001252 Py_DECREF(v);
1253 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001254 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001255 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001256 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001257
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001258 case LIST_APPEND:
1259 w = POP();
1260 v = POP();
1261 err = PyList_Append(v, w);
1262 Py_DECREF(v);
1263 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001264 if (err == 0) {
1265 PREDICT(JUMP_ABSOLUTE);
1266 continue;
1267 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001268 break;
1269
Thomas Wouters434d0822000-08-24 20:11:32 +00001270 case INPLACE_POWER:
1271 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001272 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001273 x = PyNumber_InPlacePower(v, w, Py_None);
1274 Py_DECREF(v);
1275 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001276 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001277 if (x != NULL) continue;
1278 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001279
Thomas Wouters434d0822000-08-24 20:11:32 +00001280 case INPLACE_MULTIPLY:
1281 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001282 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001283 x = PyNumber_InPlaceMultiply(v, w);
1284 Py_DECREF(v);
1285 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001286 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001287 if (x != NULL) continue;
1288 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001289
Thomas Wouters434d0822000-08-24 20:11:32 +00001290 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001291 if (!_Py_QnewFlag) {
1292 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001293 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001294 x = PyNumber_InPlaceDivide(v, w);
1295 Py_DECREF(v);
1296 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001297 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001298 if (x != NULL) continue;
1299 break;
1300 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001301 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001302 INPLACE_TRUE_DIVIDE */
1303 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001304 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001305 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001306 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001307 Py_DECREF(v);
1308 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001309 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001310 if (x != NULL) continue;
1311 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001312
Guido van Rossum4668b002001-08-08 05:00:18 +00001313 case INPLACE_FLOOR_DIVIDE:
1314 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001315 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001316 x = PyNumber_InPlaceFloorDivide(v, w);
1317 Py_DECREF(v);
1318 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001319 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001320 if (x != NULL) continue;
1321 break;
1322
Thomas Wouters434d0822000-08-24 20:11:32 +00001323 case INPLACE_MODULO:
1324 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001325 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001326 x = PyNumber_InPlaceRemainder(v, w);
1327 Py_DECREF(v);
1328 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001329 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001330 if (x != NULL) continue;
1331 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001332
Thomas Wouters434d0822000-08-24 20:11:32 +00001333 case INPLACE_ADD:
1334 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001335 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001336 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001337 /* INLINE: int + int */
1338 register long a, b, i;
1339 a = PyInt_AS_LONG(v);
1340 b = PyInt_AS_LONG(w);
1341 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001342 if ((i^a) < 0 && (i^b) < 0)
1343 goto slow_iadd;
1344 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001345 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001346 else if (PyString_CheckExact(v) &&
1347 PyString_CheckExact(w)) {
1348 x = string_concatenate(v, w, f, next_instr);
1349 /* string_concatenate consumed the ref to v */
1350 goto skip_decref_v;
1351 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001352 else {
1353 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001354 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001355 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001356 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001357 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001358 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001359 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001360 if (x != NULL) continue;
1361 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001362
Thomas Wouters434d0822000-08-24 20:11:32 +00001363 case INPLACE_SUBTRACT:
1364 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001365 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001366 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001367 /* INLINE: int - int */
1368 register long a, b, i;
1369 a = PyInt_AS_LONG(v);
1370 b = PyInt_AS_LONG(w);
1371 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001372 if ((i^a) < 0 && (i^~b) < 0)
1373 goto slow_isub;
1374 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001375 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001376 else {
1377 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001378 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001379 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001380 Py_DECREF(v);
1381 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001382 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001383 if (x != NULL) continue;
1384 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001385
Thomas Wouters434d0822000-08-24 20:11:32 +00001386 case INPLACE_LSHIFT:
1387 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001388 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001389 x = PyNumber_InPlaceLshift(v, w);
1390 Py_DECREF(v);
1391 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001392 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001393 if (x != NULL) continue;
1394 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001395
Thomas Wouters434d0822000-08-24 20:11:32 +00001396 case INPLACE_RSHIFT:
1397 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001398 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001399 x = PyNumber_InPlaceRshift(v, w);
1400 Py_DECREF(v);
1401 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001402 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001403 if (x != NULL) continue;
1404 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001405
Thomas Wouters434d0822000-08-24 20:11:32 +00001406 case INPLACE_AND:
1407 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001408 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001409 x = PyNumber_InPlaceAnd(v, w);
1410 Py_DECREF(v);
1411 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001412 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001413 if (x != NULL) continue;
1414 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001415
Thomas Wouters434d0822000-08-24 20:11:32 +00001416 case INPLACE_XOR:
1417 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001418 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001419 x = PyNumber_InPlaceXor(v, w);
1420 Py_DECREF(v);
1421 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001422 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001423 if (x != NULL) continue;
1424 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001425
Thomas Wouters434d0822000-08-24 20:11:32 +00001426 case INPLACE_OR:
1427 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001428 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001429 x = PyNumber_InPlaceOr(v, w);
1430 Py_DECREF(v);
1431 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001432 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001433 if (x != NULL) continue;
1434 break;
1435
Guido van Rossum374a9221991-04-04 10:40:29 +00001436 case SLICE+0:
1437 case SLICE+1:
1438 case SLICE+2:
1439 case SLICE+3:
1440 if ((opcode-SLICE) & 2)
1441 w = POP();
1442 else
1443 w = NULL;
1444 if ((opcode-SLICE) & 1)
1445 v = POP();
1446 else
1447 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001448 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001449 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001450 Py_DECREF(u);
1451 Py_XDECREF(v);
1452 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001453 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001454 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001455 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001456
Guido van Rossum374a9221991-04-04 10:40:29 +00001457 case STORE_SLICE+0:
1458 case STORE_SLICE+1:
1459 case STORE_SLICE+2:
1460 case STORE_SLICE+3:
1461 if ((opcode-STORE_SLICE) & 2)
1462 w = POP();
1463 else
1464 w = NULL;
1465 if ((opcode-STORE_SLICE) & 1)
1466 v = POP();
1467 else
1468 v = NULL;
1469 u = POP();
1470 t = POP();
1471 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001472 Py_DECREF(t);
1473 Py_DECREF(u);
1474 Py_XDECREF(v);
1475 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001476 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001477 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001478
Guido van Rossum374a9221991-04-04 10:40:29 +00001479 case DELETE_SLICE+0:
1480 case DELETE_SLICE+1:
1481 case DELETE_SLICE+2:
1482 case DELETE_SLICE+3:
1483 if ((opcode-DELETE_SLICE) & 2)
1484 w = POP();
1485 else
1486 w = NULL;
1487 if ((opcode-DELETE_SLICE) & 1)
1488 v = POP();
1489 else
1490 v = NULL;
1491 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001492 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001493 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001494 Py_DECREF(u);
1495 Py_XDECREF(v);
1496 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001497 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001498 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001499
Guido van Rossum374a9221991-04-04 10:40:29 +00001500 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001501 w = TOP();
1502 v = SECOND();
1503 u = THIRD();
1504 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001505 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001506 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001507 Py_DECREF(u);
1508 Py_DECREF(v);
1509 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001510 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001511 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001512
Guido van Rossum374a9221991-04-04 10:40:29 +00001513 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001514 w = TOP();
1515 v = SECOND();
1516 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001517 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001518 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001519 Py_DECREF(v);
1520 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001521 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001522 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001523
Guido van Rossum374a9221991-04-04 10:40:29 +00001524 case PRINT_EXPR:
1525 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001526 w = PySys_GetObject("displayhook");
1527 if (w == NULL) {
1528 PyErr_SetString(PyExc_RuntimeError,
1529 "lost sys.displayhook");
1530 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001531 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001532 }
1533 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001534 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001535 if (x == NULL)
1536 err = -1;
1537 }
1538 if (err == 0) {
1539 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001540 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001541 if (w == NULL)
1542 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001543 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001544 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001545 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001546 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001547
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001548 case PRINT_ITEM_TO:
1549 w = stream = POP();
1550 /* fall through to PRINT_ITEM */
1551
Guido van Rossum374a9221991-04-04 10:40:29 +00001552 case PRINT_ITEM:
1553 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001554 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001555 w = PySys_GetObject("stdout");
1556 if (w == NULL) {
1557 PyErr_SetString(PyExc_RuntimeError,
1558 "lost sys.stdout");
1559 err = -1;
1560 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001561 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001562 /* PyFile_SoftSpace() can exececute arbitrary code
1563 if sys.stdout is an instance with a __getattr__.
1564 If __getattr__ raises an exception, w will
1565 be freed, so we need to prevent that temporarily. */
1566 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001567 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001568 err = PyFile_WriteString(" ", w);
1569 if (err == 0)
1570 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001571 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001572 /* XXX move into writeobject() ? */
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001573 if (PyString_Check(v)) {
1574 char *s = PyString_AS_STRING(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001575 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001576 if (len == 0 ||
1577 !isspace(Py_CHARMASK(s[len-1])) ||
1578 s[len-1] == ' ')
1579 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001580 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001581#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001582 else if (PyUnicode_Check(v)) {
1583 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001584 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001585 if (len == 0 ||
1586 !Py_UNICODE_ISSPACE(s[len-1]) ||
1587 s[len-1] == ' ')
1588 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001589 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001590#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001591 else
1592 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001593 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001594 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001595 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001596 Py_XDECREF(stream);
1597 stream = NULL;
1598 if (err == 0)
1599 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001600 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001601
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001602 case PRINT_NEWLINE_TO:
1603 w = stream = POP();
1604 /* fall through to PRINT_NEWLINE */
1605
Guido van Rossum374a9221991-04-04 10:40:29 +00001606 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001607 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001608 w = PySys_GetObject("stdout");
1609 if (w == NULL)
1610 PyErr_SetString(PyExc_RuntimeError,
1611 "lost sys.stdout");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001612 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001613 if (w != NULL) {
1614 err = PyFile_WriteString("\n", w);
1615 if (err == 0)
1616 PyFile_SoftSpace(w, 0);
1617 }
1618 Py_XDECREF(stream);
1619 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001620 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001621
Thomas Wouters434d0822000-08-24 20:11:32 +00001622
1623#ifdef CASE_TOO_BIG
1624 default: switch (opcode) {
1625#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001626 case RAISE_VARARGS:
1627 u = v = w = NULL;
1628 switch (oparg) {
1629 case 3:
1630 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001631 /* Fallthrough */
1632 case 2:
1633 v = POP(); /* value */
1634 /* Fallthrough */
1635 case 1:
1636 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001637 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001638 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001639 break;
1640 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001641 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001642 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001643 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001644 break;
1645 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001646 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001647
Guido van Rossum374a9221991-04-04 10:40:29 +00001648 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001649 if ((x = f->f_locals) != NULL) {
1650 Py_INCREF(x);
1651 PUSH(x);
1652 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001653 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001654 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001655 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001656
Guido van Rossum374a9221991-04-04 10:40:29 +00001657 case RETURN_VALUE:
1658 retval = POP();
1659 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001660 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001661
Tim Peters5ca576e2001-06-18 22:08:13 +00001662 case YIELD_VALUE:
1663 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001664 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001665 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001666 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001667
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001668 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001669 w = TOP();
1670 v = SECOND();
1671 u = THIRD();
1672 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001673 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001674 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001675 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001676 Py_DECREF(u);
1677 Py_DECREF(v);
1678 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001679 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001680
Guido van Rossum374a9221991-04-04 10:40:29 +00001681 case POP_BLOCK:
1682 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001683 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001684 while (STACK_LEVEL() > b->b_level) {
1685 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001686 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001687 }
1688 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001689 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001690
Guido van Rossum374a9221991-04-04 10:40:29 +00001691 case END_FINALLY:
1692 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001693 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001694 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001695 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001696 if (why == WHY_RETURN ||
1697 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001698 retval = POP();
1699 }
Brett Cannonbf364092006-03-01 04:25:17 +00001700 else if (PyExceptionClass_Check(v) || PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001701 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001702 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001703 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001704 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001705 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001706 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001707 else if (v != Py_None) {
1708 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001709 "'finally' pops bad exception");
1710 why = WHY_EXCEPTION;
1711 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001712 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001713 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001714
Guido van Rossum374a9221991-04-04 10:40:29 +00001715 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001716 u = TOP();
1717 v = SECOND();
1718 w = THIRD();
1719 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001720 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001721 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001722 Py_DECREF(u);
1723 Py_DECREF(v);
1724 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001725 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001726
Guido van Rossum374a9221991-04-04 10:40:29 +00001727 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001728 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001729 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001730 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001731 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001732 err = PyDict_SetItem(x, w, v);
1733 else
1734 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001735 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001736 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001737 break;
1738 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001739 PyErr_Format(PyExc_SystemError,
1740 "no locals found when storing %s",
1741 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001742 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001743
Guido van Rossum374a9221991-04-04 10:40:29 +00001744 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001745 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001746 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001747 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001748 format_exc_check_arg(PyExc_NameError,
1749 NAME_ERROR_MSG ,w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001750 break;
1751 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001752 PyErr_Format(PyExc_SystemError,
1753 "no locals when deleting %s",
1754 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001755 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001756
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001757 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001758 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001759 v = POP();
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001760 if (PyTuple_CheckExact(v) && PyTuple_GET_SIZE(v) == oparg) {
1761 PyObject **items = ((PyTupleObject *)v)->ob_item;
1762 while (oparg--) {
1763 w = items[oparg];
1764 Py_INCREF(w);
1765 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001766 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001767 Py_DECREF(v);
1768 continue;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001769 } else if (PyList_CheckExact(v) && PyList_GET_SIZE(v) == oparg) {
1770 PyObject **items = ((PyListObject *)v)->ob_item;
1771 while (oparg--) {
1772 w = items[oparg];
1773 Py_INCREF(w);
1774 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001775 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001776 } else if (unpack_iterable(v, oparg,
Tim Petersd6d010b2001-06-21 02:49:55 +00001777 stack_pointer + oparg))
1778 stack_pointer += oparg;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001779 else {
1780 if (PyErr_ExceptionMatches(PyExc_TypeError))
1781 PyErr_SetString(PyExc_TypeError,
1782 "unpack non-sequence");
Barry Warsawe42b18f1997-08-25 22:13:04 +00001783 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001784 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001785 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001786 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001787
Guido van Rossum374a9221991-04-04 10:40:29 +00001788 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001789 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001790 v = TOP();
1791 u = SECOND();
1792 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001793 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1794 Py_DECREF(v);
1795 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001796 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001797 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001798
Guido van Rossum374a9221991-04-04 10:40:29 +00001799 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001800 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001801 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001802 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1803 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001804 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001805 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001806
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001807 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001808 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001809 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001810 err = PyDict_SetItem(f->f_globals, w, v);
1811 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001812 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001813 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001814
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001815 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001816 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001817 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001818 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001819 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001820 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001821
Guido van Rossum374a9221991-04-04 10:40:29 +00001822 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001823 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001824 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001825 PyErr_Format(PyExc_SystemError,
1826 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001827 PyObject_REPR(w));
Guido van Rossum681d79a1995-07-18 14:51:37 +00001828 break;
1829 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001830 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001831 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001832 Py_XINCREF(x);
1833 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001834 else {
1835 x = PyObject_GetItem(v, w);
1836 if (x == NULL && PyErr_Occurred()) {
1837 if (!PyErr_ExceptionMatches(PyExc_KeyError))
1838 break;
1839 PyErr_Clear();
1840 }
1841 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001842 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001843 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001844 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001845 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001846 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001847 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001848 PyExc_NameError,
Paul Prescode68140d2000-08-30 20:25:01 +00001849 NAME_ERROR_MSG ,w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001850 break;
1851 }
1852 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001853 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001854 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001855 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001856 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001857
Guido van Rossum374a9221991-04-04 10:40:29 +00001858 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001859 w = GETITEM(names, oparg);
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001860 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001861 /* Inline the PyDict_GetItem() calls.
1862 WARNING: this is an extreme speed hack.
1863 Do not try this at home. */
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001864 long hash = ((PyStringObject *)w)->ob_shash;
1865 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001866 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00001867 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001868 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00001869 e = d->ma_lookup(d, w, hash);
1870 if (e == NULL) {
1871 x = NULL;
1872 break;
1873 }
1874 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001875 if (x != NULL) {
1876 Py_INCREF(x);
1877 PUSH(x);
1878 continue;
1879 }
1880 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00001881 e = d->ma_lookup(d, w, hash);
1882 if (e == NULL) {
1883 x = NULL;
1884 break;
1885 }
1886 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001887 if (x != NULL) {
1888 Py_INCREF(x);
1889 PUSH(x);
1890 continue;
1891 }
1892 goto load_global_error;
1893 }
1894 }
1895 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001896 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001897 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001898 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001899 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001900 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001901 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001902 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001903 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001904 break;
1905 }
1906 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001907 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001908 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001909 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001910
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001911 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001912 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001913 if (x != NULL) {
1914 SETLOCAL(oparg, NULL);
1915 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001916 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001917 format_exc_check_arg(
1918 PyExc_UnboundLocalError,
1919 UNBOUNDLOCAL_ERROR_MSG,
1920 PyTuple_GetItem(co->co_varnames, oparg)
1921 );
1922 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001923
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001924 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001925 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001926 Py_INCREF(x);
1927 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001928 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001929 break;
1930
1931 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001932 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001933 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001934 if (w != NULL) {
1935 PUSH(w);
1936 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001937 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001938 err = -1;
1939 /* Don't stomp existing exception */
1940 if (PyErr_Occurred())
1941 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00001942 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1943 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001944 oparg);
1945 format_exc_check_arg(
1946 PyExc_UnboundLocalError,
1947 UNBOUNDLOCAL_ERROR_MSG,
1948 v);
1949 } else {
Richard Jonescebbefc2006-05-23 18:28:17 +00001950 v = PyTuple_GET_ITEM(
Raymond Hettinger467a6982004-04-07 11:39:21 +00001951 co->co_freevars,
Richard Jonescebbefc2006-05-23 18:28:17 +00001952 oparg - PyTuple_GET_SIZE(co->co_cellvars));
Raymond Hettinger467a6982004-04-07 11:39:21 +00001953 format_exc_check_arg(
1954 PyExc_NameError,
1955 UNBOUNDFREE_ERROR_MSG,
1956 v);
1957 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001958 break;
1959
1960 case STORE_DEREF:
1961 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001962 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001963 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001964 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001965 continue;
1966
Guido van Rossum374a9221991-04-04 10:40:29 +00001967 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001968 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001969 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001970 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001971 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001972 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001973 }
1974 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001975 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001976 }
1977 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001978
Guido van Rossum374a9221991-04-04 10:40:29 +00001979 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001980 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001981 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001982 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001983 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001984 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001985 }
1986 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001987 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001988 }
1989 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001990
Guido van Rossum374a9221991-04-04 10:40:29 +00001991 case BUILD_MAP:
Guido van Rossumb209a111997-04-29 18:18:01 +00001992 x = PyDict_New();
Guido van Rossum374a9221991-04-04 10:40:29 +00001993 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001994 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001995 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001996
Guido van Rossum374a9221991-04-04 10:40:29 +00001997 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001998 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001999 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002000 x = PyObject_GetAttr(v, w);
2001 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002002 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002003 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002004 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002005
Guido van Rossum374a9221991-04-04 10:40:29 +00002006 case COMPARE_OP:
2007 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002008 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00002009 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002010 /* INLINE: cmp(int, int) */
2011 register long a, b;
2012 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002013 a = PyInt_AS_LONG(v);
2014 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002015 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002016 case PyCmp_LT: res = a < b; break;
2017 case PyCmp_LE: res = a <= b; break;
2018 case PyCmp_EQ: res = a == b; break;
2019 case PyCmp_NE: res = a != b; break;
2020 case PyCmp_GT: res = a > b; break;
2021 case PyCmp_GE: res = a >= b; break;
2022 case PyCmp_IS: res = v == w; break;
2023 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002024 default: goto slow_compare;
2025 }
2026 x = res ? Py_True : Py_False;
2027 Py_INCREF(x);
2028 }
2029 else {
2030 slow_compare:
2031 x = cmp_outcome(oparg, v, w);
2032 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002033 Py_DECREF(v);
2034 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002035 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002036 if (x == NULL) break;
2037 PREDICT(JUMP_IF_FALSE);
2038 PREDICT(JUMP_IF_TRUE);
2039 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002040
Guido van Rossum374a9221991-04-04 10:40:29 +00002041 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002042 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002043 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002044 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002045 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002046 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002047 break;
2048 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002049 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002050 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002051 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2052 w = PyTuple_Pack(5,
2053 w,
2054 f->f_globals,
2055 f->f_locals == NULL ?
2056 Py_None : f->f_locals,
2057 v,
2058 u);
2059 else
2060 w = PyTuple_Pack(4,
2061 w,
2062 f->f_globals,
2063 f->f_locals == NULL ?
2064 Py_None : f->f_locals,
2065 v);
2066 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002067 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002068 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002069 u = POP();
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002070 x = NULL;
2071 break;
2072 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002073 READ_TIMESTAMP(intr0);
Guido van Rossumb209a111997-04-29 18:18:01 +00002074 x = PyEval_CallObject(x, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002075 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002076 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002077 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002078 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002079 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002080
Thomas Wouters52152252000-08-17 22:55:00 +00002081 case IMPORT_STAR:
2082 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002083 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002084 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002085 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002086 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002087 break;
2088 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002089 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002090 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002091 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002092 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002093 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002094 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002095 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002096
Thomas Wouters52152252000-08-17 22:55:00 +00002097 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002098 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002099 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002100 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002101 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002102 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002103 PUSH(x);
2104 if (x != NULL) continue;
2105 break;
2106
Guido van Rossum374a9221991-04-04 10:40:29 +00002107 case JUMP_FORWARD:
2108 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002109 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002110
Raymond Hettingerf606f872003-03-16 03:11:04 +00002111 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002112 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002113 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002114 if (w == Py_True) {
2115 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002116 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002117 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002118 if (w == Py_False) {
2119 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002120 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002121 }
2122 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002123 if (err > 0)
2124 err = 0;
2125 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002126 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002127 else
2128 break;
2129 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002130
Raymond Hettingerf606f872003-03-16 03:11:04 +00002131 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002132 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002133 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002134 if (w == Py_False) {
2135 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002136 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002137 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002138 if (w == Py_True) {
2139 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002140 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002141 }
2142 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002143 if (err > 0) {
2144 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002145 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002146 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002147 else if (err == 0)
2148 ;
2149 else
2150 break;
2151 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002152
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002153 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002154 case JUMP_ABSOLUTE:
2155 JUMPTO(oparg);
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002156 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002157
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002158 case GET_ITER:
2159 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002160 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002161 x = PyObject_GetIter(v);
2162 Py_DECREF(v);
2163 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002164 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002165 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002166 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002167 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002168 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002169 break;
2170
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002171 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002172 case FOR_ITER:
2173 /* before: [iter]; after: [iter, iter()] *or* [] */
2174 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002175 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002176 if (x != NULL) {
2177 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002178 PREDICT(STORE_FAST);
2179 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002180 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002181 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002182 if (PyErr_Occurred()) {
2183 if (!PyErr_ExceptionMatches(PyExc_StopIteration))
2184 break;
2185 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002186 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002187 /* iterator ended normally */
2188 x = v = POP();
2189 Py_DECREF(v);
2190 JUMPBY(oparg);
2191 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002192
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002193 case BREAK_LOOP:
2194 why = WHY_BREAK;
2195 goto fast_block_end;
2196
2197 case CONTINUE_LOOP:
2198 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002199 if (!retval) {
2200 x = NULL;
2201 break;
2202 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002203 why = WHY_CONTINUE;
2204 goto fast_block_end;
2205
Guido van Rossum374a9221991-04-04 10:40:29 +00002206 case SETUP_LOOP:
2207 case SETUP_EXCEPT:
2208 case SETUP_FINALLY:
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002209 /* NOTE: If you add any new block-setup opcodes that are not try/except/finally
2210 handlers, you may need to update the PyGen_NeedsFinalizing() function. */
2211
Guido van Rossumb209a111997-04-29 18:18:01 +00002212 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002213 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002214 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002215
Guido van Rossumc2e20742006-02-27 22:32:47 +00002216 case WITH_CLEANUP:
2217 {
2218 /* TOP is the context.__exit__ bound method.
2219 Below that are 1-3 values indicating how/why
2220 we entered the finally clause:
2221 - SECOND = None
Guido van Rossumf6694362006-03-10 02:28:35 +00002222 - (SECOND, THIRD) = (WHY_{RETURN,CONTINUE}), retval
Guido van Rossumc2e20742006-02-27 22:32:47 +00002223 - SECOND = WHY_*; no retval below it
2224 - (SECOND, THIRD, FOURTH) = exc_info()
2225 In the last case, we must call
2226 TOP(SECOND, THIRD, FOURTH)
2227 otherwise we must call
2228 TOP(None, None, None)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002229
2230 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002231 *and* the function call returns a 'true' value, we
2232 "zap" this information, to prevent END_FINALLY from
2233 re-raising the exception. (But non-local gotos
2234 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002235 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002236
Guido van Rossumc2e20742006-02-27 22:32:47 +00002237 x = TOP();
2238 u = SECOND();
2239 if (PyInt_Check(u) || u == Py_None) {
2240 u = v = w = Py_None;
2241 }
2242 else {
2243 v = THIRD();
2244 w = FOURTH();
2245 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002246 /* XXX Not the fastest way to call it... */
2247 x = PyObject_CallFunctionObjArgs(x, u, v, w, NULL);
2248 if (x == NULL)
2249 break; /* Go to error exit */
2250 if (u != Py_None && PyObject_IsTrue(x)) {
2251 /* There was an exception and a true return */
2252 Py_DECREF(x);
2253 x = TOP(); /* Again */
2254 STACKADJ(-3);
2255 Py_INCREF(Py_None);
2256 SET_TOP(Py_None);
2257 Py_DECREF(x);
2258 Py_DECREF(u);
2259 Py_DECREF(v);
2260 Py_DECREF(w);
2261 } else {
2262 /* Let END_FINALLY do its thing */
2263 Py_DECREF(x);
2264 x = POP();
2265 Py_DECREF(x);
2266 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002267 break;
2268 }
2269
Guido van Rossumf10570b1995-07-07 22:53:21 +00002270 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002271 {
2272 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002273 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002274 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002275#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002276 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002277#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002278 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002279#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002280 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002281 PUSH(x);
2282 if (x != NULL)
2283 continue;
2284 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002285 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002286
Jeremy Hylton76901512000-03-28 23:49:17 +00002287 case CALL_FUNCTION_VAR:
2288 case CALL_FUNCTION_KW:
2289 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002290 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002291 int na = oparg & 0xff;
2292 int nk = (oparg>>8) & 0xff;
2293 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002294 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002295 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002296 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002297 if (flags & CALL_FLAG_VAR)
2298 n++;
2299 if (flags & CALL_FLAG_KW)
2300 n++;
2301 pfunc = stack_pointer - n - 1;
2302 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002303
Guido van Rossumac7be682001-01-17 15:42:30 +00002304 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002305 && PyMethod_GET_SELF(func) != NULL) {
2306 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002307 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002308 func = PyMethod_GET_FUNCTION(func);
2309 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002310 Py_DECREF(*pfunc);
2311 *pfunc = self;
2312 na++;
2313 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002314 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002315 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002316 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002317 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002318 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002319 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002320 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002321 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002322
Jeremy Hylton76901512000-03-28 23:49:17 +00002323 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002324 w = POP();
2325 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002326 }
2327 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002328 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002329 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002330 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002331 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002332
Guido van Rossum681d79a1995-07-18 14:51:37 +00002333 case MAKE_FUNCTION:
2334 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002335 x = PyFunction_New(v, f->f_globals);
2336 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002337 /* XXX Maybe this should be a separate opcode? */
2338 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002339 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002340 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002341 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002342 x = NULL;
2343 break;
2344 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002345 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002346 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002347 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002348 }
2349 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002350 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002351 }
2352 PUSH(x);
2353 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002354
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002355 case MAKE_CLOSURE:
2356 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002357 v = POP(); /* code object */
2358 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002359 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002360 if (x != NULL) {
2361 v = POP();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002362 err = PyFunction_SetClosure(x, v);
2363 Py_DECREF(v);
2364 }
2365 if (x != NULL && oparg > 0) {
2366 v = PyTuple_New(oparg);
2367 if (v == NULL) {
2368 Py_DECREF(x);
2369 x = NULL;
2370 break;
2371 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002372 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002373 w = POP();
2374 PyTuple_SET_ITEM(v, oparg, w);
2375 }
2376 err = PyFunction_SetDefaults(x, v);
2377 Py_DECREF(v);
2378 }
2379 PUSH(x);
2380 break;
2381 }
2382
Guido van Rossum8861b741996-07-30 16:49:37 +00002383 case BUILD_SLICE:
2384 if (oparg == 3)
2385 w = POP();
2386 else
2387 w = NULL;
2388 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002389 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002390 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002391 Py_DECREF(u);
2392 Py_DECREF(v);
2393 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002394 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002395 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002396 break;
2397
Fred Drakeef8ace32000-08-24 00:32:09 +00002398 case EXTENDED_ARG:
2399 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002400 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002401 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002402
Guido van Rossum374a9221991-04-04 10:40:29 +00002403 default:
2404 fprintf(stderr,
2405 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002406 PyCode_Addr2Line(f->f_code, f->f_lasti),
2407 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002408 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002409 why = WHY_EXCEPTION;
2410 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002411
2412#ifdef CASE_TOO_BIG
2413 }
2414#endif
2415
Guido van Rossum374a9221991-04-04 10:40:29 +00002416 } /* switch */
2417
2418 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002419
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002420 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002421
Guido van Rossum374a9221991-04-04 10:40:29 +00002422 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002423
Guido van Rossum374a9221991-04-04 10:40:29 +00002424 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002425 if (err == 0 && x != NULL) {
2426#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002427 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002428 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002429 fprintf(stderr,
2430 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002431 else {
2432#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002433 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002434 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002435#ifdef CHECKEXC
2436 }
2437#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002438 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002439 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002440 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002441 err = 0;
2442 }
2443
Guido van Rossum374a9221991-04-04 10:40:29 +00002444 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002445
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002446 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002447 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002448 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002449 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002450 why = WHY_EXCEPTION;
2451 }
2452 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002453#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002454 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002455 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002456 if (PyErr_Occurred()) {
Jeremy Hylton904ed862003-11-05 17:29:35 +00002457 char buf[1024];
2458 sprintf(buf, "Stack unwind with exception "
2459 "set and why=%d", why);
2460 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002461 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002462 }
2463#endif
2464
2465 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002466
Guido van Rossum374a9221991-04-04 10:40:29 +00002467 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002468 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002469
Fred Drake8f51f542001-10-04 14:48:42 +00002470 if (tstate->c_tracefunc != NULL)
2471 call_exc_trace(tstate->c_tracefunc,
2472 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002473 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002474
Guido van Rossum374a9221991-04-04 10:40:29 +00002475 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002476
Guido van Rossum374a9221991-04-04 10:40:29 +00002477 if (why == WHY_RERAISE)
2478 why = WHY_EXCEPTION;
2479
2480 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002481
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002482fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002483 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002484 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002485
Tim Peters8a5c3c72004-04-05 19:36:21 +00002486 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002487 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2488 /* For a continue inside a try block,
2489 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002490 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2491 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002492 why = WHY_NOT;
2493 JUMPTO(PyInt_AS_LONG(retval));
2494 Py_DECREF(retval);
2495 break;
2496 }
2497
Guido van Rossum374a9221991-04-04 10:40:29 +00002498 while (STACK_LEVEL() > b->b_level) {
2499 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002500 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002501 }
2502 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2503 why = WHY_NOT;
2504 JUMPTO(b->b_handler);
2505 break;
2506 }
2507 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002508 (b->b_type == SETUP_EXCEPT &&
2509 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002510 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002511 PyObject *exc, *val, *tb;
2512 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002513 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002514 val = Py_None;
2515 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002516 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002517 /* Make the raw exception data
2518 available to the handler,
2519 so a program can emulate the
2520 Python main loop. Don't do
2521 this for 'finally'. */
2522 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002523 PyErr_NormalizeException(
2524 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002525 set_exc_info(tstate,
2526 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002527 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002528 if (tb == NULL) {
2529 Py_INCREF(Py_None);
2530 PUSH(Py_None);
2531 } else
2532 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002533 PUSH(val);
2534 PUSH(exc);
2535 }
2536 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002537 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002538 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002539 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002540 PUSH(v);
2541 }
2542 why = WHY_NOT;
2543 JUMPTO(b->b_handler);
2544 break;
2545 }
2546 } /* unwind stack */
2547
2548 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002549
Guido van Rossum374a9221991-04-04 10:40:29 +00002550 if (why != WHY_NOT)
2551 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002552 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002553
Guido van Rossum374a9221991-04-04 10:40:29 +00002554 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002555
Tim Peters8a5c3c72004-04-05 19:36:21 +00002556 assert(why != WHY_YIELD);
2557 /* Pop remaining stack entries. */
2558 while (!EMPTY()) {
2559 v = POP();
2560 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002561 }
2562
Tim Peters8a5c3c72004-04-05 19:36:21 +00002563 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002564 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002565
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002566fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002567 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002568 if (tstate->c_tracefunc) {
2569 if (why == WHY_RETURN || why == WHY_YIELD) {
2570 if (call_trace(tstate->c_tracefunc,
2571 tstate->c_traceobj, f,
2572 PyTrace_RETURN, retval)) {
2573 Py_XDECREF(retval);
2574 retval = NULL;
2575 why = WHY_EXCEPTION;
2576 }
2577 }
2578 else if (why == WHY_EXCEPTION) {
2579 call_trace_protected(tstate->c_tracefunc,
2580 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002581 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002582 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002583 }
Fred Drake8f51f542001-10-04 14:48:42 +00002584 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002585 if (why == WHY_EXCEPTION)
2586 call_trace_protected(tstate->c_profilefunc,
2587 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002588 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002589 else if (call_trace(tstate->c_profilefunc,
2590 tstate->c_profileobj, f,
2591 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002592 Py_XDECREF(retval);
2593 retval = NULL;
2594 why = WHY_EXCEPTION;
2595 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002596 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002597 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002598
Tim Peters7df5e7f2006-05-26 23:14:37 +00002599 if (tstate->frame->f_exc_type != NULL)
2600 reset_exc_info(tstate);
2601 else {
2602 assert(tstate->frame->f_exc_value == NULL);
2603 assert(tstate->frame->f_exc_traceback == NULL);
2604 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002605
Tim Peters5ca576e2001-06-18 22:08:13 +00002606 /* pop frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +00002607 exit_eval_frame:
2608 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002609 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002610
Guido van Rossum96a42c81992-01-12 02:29:51 +00002611 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002612}
2613
Guido van Rossumc2e20742006-02-27 22:32:47 +00002614/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002615 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002616 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002617
Tim Peters6d6c1a32001-08-02 04:15:00 +00002618PyObject *
2619PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002620 PyObject **args, int argcount, PyObject **kws, int kwcount,
2621 PyObject **defs, int defcount, PyObject *closure)
2622{
2623 register PyFrameObject *f;
2624 register PyObject *retval = NULL;
2625 register PyObject **fastlocals, **freevars;
2626 PyThreadState *tstate = PyThreadState_GET();
2627 PyObject *x, *u;
2628
2629 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002630 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002631 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002632 return NULL;
2633 }
2634
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002635 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002636 assert(globals != NULL);
2637 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002638 if (f == NULL)
2639 return NULL;
2640
2641 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002642 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002643
2644 if (co->co_argcount > 0 ||
2645 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2646 int i;
2647 int n = argcount;
2648 PyObject *kwdict = NULL;
2649 if (co->co_flags & CO_VARKEYWORDS) {
2650 kwdict = PyDict_New();
2651 if (kwdict == NULL)
2652 goto fail;
2653 i = co->co_argcount;
2654 if (co->co_flags & CO_VARARGS)
2655 i++;
2656 SETLOCAL(i, kwdict);
2657 }
2658 if (argcount > co->co_argcount) {
2659 if (!(co->co_flags & CO_VARARGS)) {
2660 PyErr_Format(PyExc_TypeError,
2661 "%.200s() takes %s %d "
2662 "%sargument%s (%d given)",
2663 PyString_AsString(co->co_name),
2664 defcount ? "at most" : "exactly",
2665 co->co_argcount,
2666 kwcount ? "non-keyword " : "",
2667 co->co_argcount == 1 ? "" : "s",
2668 argcount);
2669 goto fail;
2670 }
2671 n = co->co_argcount;
2672 }
2673 for (i = 0; i < n; i++) {
2674 x = args[i];
2675 Py_INCREF(x);
2676 SETLOCAL(i, x);
2677 }
2678 if (co->co_flags & CO_VARARGS) {
2679 u = PyTuple_New(argcount - n);
2680 if (u == NULL)
2681 goto fail;
2682 SETLOCAL(co->co_argcount, u);
2683 for (i = n; i < argcount; i++) {
2684 x = args[i];
2685 Py_INCREF(x);
2686 PyTuple_SET_ITEM(u, i-n, x);
2687 }
2688 }
2689 for (i = 0; i < kwcount; i++) {
2690 PyObject *keyword = kws[2*i];
2691 PyObject *value = kws[2*i + 1];
2692 int j;
2693 if (keyword == NULL || !PyString_Check(keyword)) {
2694 PyErr_Format(PyExc_TypeError,
2695 "%.200s() keywords must be strings",
2696 PyString_AsString(co->co_name));
2697 goto fail;
2698 }
2699 /* XXX slow -- speed up using dictionary? */
2700 for (j = 0; j < co->co_argcount; j++) {
2701 PyObject *nm = PyTuple_GET_ITEM(
2702 co->co_varnames, j);
2703 int cmp = PyObject_RichCompareBool(
2704 keyword, nm, Py_EQ);
2705 if (cmp > 0)
2706 break;
2707 else if (cmp < 0)
2708 goto fail;
2709 }
2710 /* Check errors from Compare */
2711 if (PyErr_Occurred())
2712 goto fail;
2713 if (j >= co->co_argcount) {
2714 if (kwdict == NULL) {
2715 PyErr_Format(PyExc_TypeError,
2716 "%.200s() got an unexpected "
2717 "keyword argument '%.400s'",
2718 PyString_AsString(co->co_name),
2719 PyString_AsString(keyword));
2720 goto fail;
2721 }
2722 PyDict_SetItem(kwdict, keyword, value);
2723 }
2724 else {
2725 if (GETLOCAL(j) != NULL) {
2726 PyErr_Format(PyExc_TypeError,
2727 "%.200s() got multiple "
2728 "values for keyword "
2729 "argument '%.400s'",
2730 PyString_AsString(co->co_name),
2731 PyString_AsString(keyword));
2732 goto fail;
2733 }
2734 Py_INCREF(value);
2735 SETLOCAL(j, value);
2736 }
2737 }
2738 if (argcount < co->co_argcount) {
2739 int m = co->co_argcount - defcount;
2740 for (i = argcount; i < m; i++) {
2741 if (GETLOCAL(i) == NULL) {
2742 PyErr_Format(PyExc_TypeError,
2743 "%.200s() takes %s %d "
2744 "%sargument%s (%d given)",
2745 PyString_AsString(co->co_name),
2746 ((co->co_flags & CO_VARARGS) ||
2747 defcount) ? "at least"
2748 : "exactly",
2749 m, kwcount ? "non-keyword " : "",
2750 m == 1 ? "" : "s", i);
2751 goto fail;
2752 }
2753 }
2754 if (n > m)
2755 i = n - m;
2756 else
2757 i = 0;
2758 for (; i < defcount; i++) {
2759 if (GETLOCAL(m+i) == NULL) {
2760 PyObject *def = defs[i];
2761 Py_INCREF(def);
2762 SETLOCAL(m+i, def);
2763 }
2764 }
2765 }
2766 }
2767 else {
2768 if (argcount > 0 || kwcount > 0) {
2769 PyErr_Format(PyExc_TypeError,
2770 "%.200s() takes no arguments (%d given)",
2771 PyString_AsString(co->co_name),
2772 argcount + kwcount);
2773 goto fail;
2774 }
2775 }
2776 /* Allocate and initialize storage for cell vars, and copy free
2777 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00002778 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00002779 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002780 char *cellname, *argname;
2781 PyObject *c;
2782
2783 nargs = co->co_argcount;
2784 if (co->co_flags & CO_VARARGS)
2785 nargs++;
2786 if (co->co_flags & CO_VARKEYWORDS)
2787 nargs++;
2788
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002789 /* Initialize each cell var, taking into account
2790 cell vars that are initialized from arguments.
2791
2792 Should arrange for the compiler to put cellvars
2793 that are arguments at the beginning of the cellvars
2794 list so that we can march over it more efficiently?
2795 */
Richard Jonescebbefc2006-05-23 18:28:17 +00002796 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002797 cellname = PyString_AS_STRING(
2798 PyTuple_GET_ITEM(co->co_cellvars, i));
2799 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002800 for (j = 0; j < nargs; j++) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002801 argname = PyString_AS_STRING(
2802 PyTuple_GET_ITEM(co->co_varnames, j));
2803 if (strcmp(cellname, argname) == 0) {
2804 c = PyCell_New(GETLOCAL(j));
2805 if (c == NULL)
2806 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002807 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002808 found = 1;
2809 break;
2810 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002811 }
2812 if (found == 0) {
2813 c = PyCell_New(NULL);
2814 if (c == NULL)
2815 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002816 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002817 }
2818 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002819 }
Richard Jonescebbefc2006-05-23 18:28:17 +00002820 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002821 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00002822 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002823 PyObject *o = PyTuple_GET_ITEM(closure, i);
2824 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00002825 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002826 }
2827 }
2828
Tim Peters5ca576e2001-06-18 22:08:13 +00002829 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002830 /* Don't need to keep the reference to f_back, it will be set
2831 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002832 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002833 f->f_back = NULL;
2834
Jeremy Hylton985eba52003-02-05 23:13:00 +00002835 PCALL(PCALL_GENERATOR);
2836
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002837 /* Create a new generator that owns the ready to run frame
2838 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002839 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002840 }
2841
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00002842 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002843
2844 fail: /* Jump here from prelude on failure */
2845
Tim Petersb13680b2001-11-27 23:29:29 +00002846 /* decref'ing the frame can cause __del__ methods to get invoked,
2847 which can call back into Python. While we're done with the
2848 current Python frame (f), the associated C stack is still in use,
2849 so recursion_depth must be boosted for the duration.
2850 */
2851 assert(tstate != NULL);
2852 ++tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002853 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002854 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002855 return retval;
2856}
2857
2858
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002859/* Implementation notes for set_exc_info() and reset_exc_info():
2860
2861- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2862 'exc_traceback'. These always travel together.
2863
2864- tstate->curexc_ZZZ is the "hot" exception that is set by
2865 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2866
2867- Once an exception is caught by an except clause, it is transferred
2868 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2869 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00002870 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002871
2872- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2873
2874 Long ago, when none of this existed, there were just a few globals:
2875 one set corresponding to the "hot" exception, and one set
2876 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2877 globals; they were simply stored as sys.exc_ZZZ. For backwards
2878 compatibility, they still are!) The problem was that in code like
2879 this:
2880
2881 try:
2882 "something that may fail"
2883 except "some exception":
2884 "do something else first"
2885 "print the exception from sys.exc_ZZZ."
2886
2887 if "do something else first" invoked something that raised and caught
2888 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2889 cause of subtle bugs. I fixed this by changing the semantics as
2890 follows:
2891
2892 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2893 *in that frame*.
2894
2895 - But initially, and as long as no exception is caught in a given
2896 frame, sys.exc_ZZZ will hold the last exception caught in the
2897 previous frame (or the frame before that, etc.).
2898
2899 The first bullet fixed the bug in the above example. The second
2900 bullet was for backwards compatibility: it was (and is) common to
2901 have a function that is called when an exception is caught, and to
2902 have that function access the caught exception via sys.exc_ZZZ.
2903 (Example: traceback.print_exc()).
2904
2905 At the same time I fixed the problem that sys.exc_ZZZ weren't
2906 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2907 but that's really a separate improvement.
2908
2909 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2910 variables to what they were before the current frame was called. The
2911 set_exc_info() function saves them on the frame so that
2912 reset_exc_info() can restore them. The invariant is that
2913 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2914 exception (where "catching" an exception applies only to successful
2915 except clauses); and if the current frame ever caught an exception,
2916 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2917 at the start of the current frame.
2918
2919*/
2920
Fredrik Lundh7a830892006-05-27 10:39:48 +00002921static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002922set_exc_info(PyThreadState *tstate,
2923 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002924{
Tim Peters7df5e7f2006-05-26 23:14:37 +00002925 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002926 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002927
Tim Peters7df5e7f2006-05-26 23:14:37 +00002928 assert(type != NULL);
2929 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002930 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002931 assert(frame->f_exc_value == NULL);
2932 assert(frame->f_exc_traceback == NULL);
2933 /* This frame didn't catch an exception before. */
2934 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002935 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002936 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002937 Py_INCREF(Py_None);
2938 tstate->exc_type = Py_None;
2939 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002940 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002941 Py_XINCREF(tstate->exc_value);
2942 Py_XINCREF(tstate->exc_traceback);
2943 frame->f_exc_type = tstate->exc_type;
2944 frame->f_exc_value = tstate->exc_value;
2945 frame->f_exc_traceback = tstate->exc_traceback;
2946 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002947 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002948 tmp_type = tstate->exc_type;
2949 tmp_value = tstate->exc_value;
2950 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002951 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002952 Py_XINCREF(value);
2953 Py_XINCREF(tb);
2954 tstate->exc_type = type;
2955 tstate->exc_value = value;
2956 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002957 Py_XDECREF(tmp_type);
2958 Py_XDECREF(tmp_value);
2959 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002960 /* For b/w compatibility */
2961 PySys_SetObject("exc_type", type);
2962 PySys_SetObject("exc_value", value);
2963 PySys_SetObject("exc_traceback", tb);
2964}
2965
Fredrik Lundh7a830892006-05-27 10:39:48 +00002966static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002967reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002968{
2969 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002970 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002971
2972 /* It's a precondition that the thread state's frame caught an
2973 * exception -- verify in a debug build.
2974 */
2975 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002976 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002977 assert(frame != NULL);
2978 assert(frame->f_exc_type != NULL);
2979
2980 /* Copy the frame's exception info back to the thread state. */
2981 tmp_type = tstate->exc_type;
2982 tmp_value = tstate->exc_value;
2983 tmp_tb = tstate->exc_traceback;
2984 Py_INCREF(frame->f_exc_type);
2985 Py_XINCREF(frame->f_exc_value);
2986 Py_XINCREF(frame->f_exc_traceback);
2987 tstate->exc_type = frame->f_exc_type;
2988 tstate->exc_value = frame->f_exc_value;
2989 tstate->exc_traceback = frame->f_exc_traceback;
2990 Py_XDECREF(tmp_type);
2991 Py_XDECREF(tmp_value);
2992 Py_XDECREF(tmp_tb);
2993
2994 /* For b/w compatibility */
2995 PySys_SetObject("exc_type", frame->f_exc_type);
2996 PySys_SetObject("exc_value", frame->f_exc_value);
2997 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
2998
2999 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003000 tmp_type = frame->f_exc_type;
3001 tmp_value = frame->f_exc_value;
3002 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003003 frame->f_exc_type = NULL;
3004 frame->f_exc_value = NULL;
3005 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003006 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003007 Py_XDECREF(tmp_value);
3008 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003009}
3010
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003011/* Logic for the raise statement (too complicated for inlining).
3012 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003013static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003014do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003015{
Guido van Rossumd295f121998-04-09 21:39:57 +00003016 if (type == NULL) {
3017 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003018 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003019 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3020 value = tstate->exc_value;
3021 tb = tstate->exc_traceback;
3022 Py_XINCREF(type);
3023 Py_XINCREF(value);
3024 Py_XINCREF(tb);
3025 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003026
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003027 /* We support the following forms of raise:
3028 raise <class>, <classinstance>
3029 raise <class>, <argument tuple>
3030 raise <class>, None
3031 raise <class>, <argument>
3032 raise <classinstance>, None
3033 raise <string>, <object>
3034 raise <string>, None
3035
3036 An omitted second argument is the same as None.
3037
3038 In addition, raise <tuple>, <anything> is the same as
3039 raising the tuple's first item (and it better have one!);
3040 this rule is applied recursively.
3041
3042 Finally, an optional third argument can be supplied, which
3043 gives the traceback to be substituted (useful when
3044 re-raising an exception after examining it). */
3045
3046 /* First, check the traceback argument, replacing None with
3047 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003048 if (tb == Py_None) {
3049 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003050 tb = NULL;
3051 }
3052 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003053 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003054 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003055 goto raise_error;
3056 }
3057
3058 /* Next, replace a missing value with None */
3059 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003060 value = Py_None;
3061 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003062 }
3063
3064 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003065 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3066 PyObject *tmp = type;
3067 type = PyTuple_GET_ITEM(type, 0);
3068 Py_INCREF(type);
3069 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003070 }
3071
Brett Cannona7446e32006-02-27 23:39:10 +00003072 if (PyString_CheckExact(type)) {
Tim Petersafb2c802002-04-18 18:06:20 +00003073 /* Raising builtin string is deprecated but still allowed --
3074 * do nothing. Raising an instance of a new-style str
3075 * subclass is right out. */
Brett Cannonbf364092006-03-01 04:25:17 +00003076 if (PyErr_Warn(PyExc_DeprecationWarning,
Brett Cannona7446e32006-02-27 23:39:10 +00003077 "raising a string exception is deprecated"))
3078 goto raise_error;
3079 }
Brett Cannonbf364092006-03-01 04:25:17 +00003080 else if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003081 PyErr_NormalizeException(&type, &value, &tb);
3082
Brett Cannonbf364092006-03-01 04:25:17 +00003083 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003084 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003085 if (value != Py_None) {
3086 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003087 "instance exception may not have a separate value");
3088 goto raise_error;
3089 }
3090 else {
3091 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003092 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003093 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003094 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003095 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003096 }
3097 }
3098 else {
3099 /* Not something you can raise. You get an exception
3100 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003101 PyErr_Format(PyExc_TypeError,
Neal Norwitz37aa0662003-01-10 15:31:15 +00003102 "exceptions must be classes, instances, or "
3103 "strings (deprecated), not %s",
3104 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003105 goto raise_error;
3106 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003107 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003108 if (tb == NULL)
3109 return WHY_EXCEPTION;
3110 else
3111 return WHY_RERAISE;
3112 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003113 Py_XDECREF(value);
3114 Py_XDECREF(type);
3115 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003116 return WHY_EXCEPTION;
3117}
3118
Tim Petersd6d010b2001-06-21 02:49:55 +00003119/* Iterate v argcnt times and store the results on the stack (via decreasing
3120 sp). Return 1 for success, 0 if error. */
3121
Fredrik Lundh7a830892006-05-27 10:39:48 +00003122static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003123unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003124{
Tim Petersd6d010b2001-06-21 02:49:55 +00003125 int i = 0;
3126 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003127 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003128
Tim Petersd6d010b2001-06-21 02:49:55 +00003129 assert(v != NULL);
3130
3131 it = PyObject_GetIter(v);
3132 if (it == NULL)
3133 goto Error;
3134
3135 for (; i < argcnt; i++) {
3136 w = PyIter_Next(it);
3137 if (w == NULL) {
3138 /* Iterator done, via error or exhaustion. */
3139 if (!PyErr_Occurred()) {
3140 PyErr_Format(PyExc_ValueError,
3141 "need more than %d value%s to unpack",
3142 i, i == 1 ? "" : "s");
3143 }
3144 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003145 }
3146 *--sp = w;
3147 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003148
3149 /* We better have exhausted the iterator now. */
3150 w = PyIter_Next(it);
3151 if (w == NULL) {
3152 if (PyErr_Occurred())
3153 goto Error;
3154 Py_DECREF(it);
3155 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003156 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003157 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003158 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003159 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003160Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003161 for (; i > 0; i--, sp++)
3162 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003163 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003164 return 0;
3165}
3166
3167
Guido van Rossum96a42c81992-01-12 02:29:51 +00003168#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003169static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003170prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003171{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003172 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003173 if (PyObject_Print(v, stdout, 0) != 0)
3174 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003175 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003176 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003177}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003178#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003179
Fredrik Lundh7a830892006-05-27 10:39:48 +00003180static void
Fred Drake5755ce62001-06-27 19:19:46 +00003181call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003182{
Guido van Rossumb209a111997-04-29 18:18:01 +00003183 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003184 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003185 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003186 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003187 value = Py_None;
3188 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003189 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003190 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003191 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003192 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003193 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003194 }
Fred Drake5755ce62001-06-27 19:19:46 +00003195 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003196 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003197 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003198 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003199 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003200 Py_XDECREF(type);
3201 Py_XDECREF(value);
3202 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003203 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003204}
3205
Fredrik Lundh7a830892006-05-27 10:39:48 +00003206static void
Fred Drake4ec5d562001-10-04 19:26:43 +00003207call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003208 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003209{
3210 PyObject *type, *value, *traceback;
3211 int err;
3212 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003213 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003214 if (err == 0)
3215 PyErr_Restore(type, value, traceback);
3216 else {
3217 Py_XDECREF(type);
3218 Py_XDECREF(value);
3219 Py_XDECREF(traceback);
3220 }
3221}
3222
Fredrik Lundh7a830892006-05-27 10:39:48 +00003223static int
Fred Drake5755ce62001-06-27 19:19:46 +00003224call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3225 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003226{
Fred Drake5755ce62001-06-27 19:19:46 +00003227 register PyThreadState *tstate = frame->f_tstate;
3228 int result;
3229 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003230 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003231 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003232 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003233 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003234 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3235 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003236 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003237 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003238}
3239
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003240PyObject *
3241_PyEval_CallTracing(PyObject *func, PyObject *args)
3242{
3243 PyFrameObject *frame = PyEval_GetFrame();
3244 PyThreadState *tstate = frame->f_tstate;
3245 int save_tracing = tstate->tracing;
3246 int save_use_tracing = tstate->use_tracing;
3247 PyObject *result;
3248
3249 tstate->tracing = 0;
3250 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3251 || (tstate->c_profilefunc != NULL));
3252 result = PyObject_Call(func, args, NULL);
3253 tstate->tracing = save_tracing;
3254 tstate->use_tracing = save_use_tracing;
3255 return result;
3256}
3257
Fredrik Lundh7a830892006-05-27 10:39:48 +00003258static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003259maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003260 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3261 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003262{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003263 int result = 0;
3264
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003265 /* If the last instruction executed isn't in the current
3266 instruction window, reset the window. If the last
3267 instruction happens to fall at the start of a line or if it
3268 represents a jump backwards, call the trace function.
3269 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003270 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003271 int line;
3272 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003273
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003274 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3275 &bounds);
3276 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003277 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003278 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003279 PyTrace_LINE, Py_None);
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003280 }
3281 *instr_lb = bounds.ap_lower;
3282 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003283 }
Armin Rigobf57a142004-03-22 19:24:58 +00003284 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003285 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003286 }
3287 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003288 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003289}
3290
Fred Drake5755ce62001-06-27 19:19:46 +00003291void
3292PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003293{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003294 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003295 PyObject *temp = tstate->c_profileobj;
3296 Py_XINCREF(arg);
3297 tstate->c_profilefunc = NULL;
3298 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003299 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003300 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003301 Py_XDECREF(temp);
3302 tstate->c_profilefunc = func;
3303 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003304 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003305 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003306}
3307
3308void
3309PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3310{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003311 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003312 PyObject *temp = tstate->c_traceobj;
3313 Py_XINCREF(arg);
3314 tstate->c_tracefunc = NULL;
3315 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003316 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003317 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003318 Py_XDECREF(temp);
3319 tstate->c_tracefunc = func;
3320 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003321 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003322 tstate->use_tracing = ((func != NULL)
3323 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003324}
3325
Guido van Rossumb209a111997-04-29 18:18:01 +00003326PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003327PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003328{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003329 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003330 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003331 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003332 else
3333 return current_frame->f_builtins;
3334}
3335
Guido van Rossumb209a111997-04-29 18:18:01 +00003336PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003337PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003338{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003339 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003340 if (current_frame == NULL)
3341 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003342 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003343 return current_frame->f_locals;
3344}
3345
Guido van Rossumb209a111997-04-29 18:18:01 +00003346PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003347PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003348{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003349 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003350 if (current_frame == NULL)
3351 return NULL;
3352 else
3353 return current_frame->f_globals;
3354}
3355
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003356PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003357PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003358{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003359 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003360 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003361}
3362
Guido van Rossum6135a871995-01-09 17:53:26 +00003363int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003364PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003365{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003366 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003367 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003368}
3369
Guido van Rossumbe270261997-05-22 22:26:18 +00003370int
Tim Peters5ba58662001-07-16 02:29:45 +00003371PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003372{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003373 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003374 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003375
3376 if (current_frame != NULL) {
3377 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003378 const int compilerflags = codeflags & PyCF_MASK;
3379 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003380 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003381 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003382 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003383#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003384 if (codeflags & CO_GENERATOR_ALLOWED) {
3385 result = 1;
3386 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3387 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003388#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003389 }
3390 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003391}
3392
3393int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003394Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003395{
Guido van Rossumb209a111997-04-29 18:18:01 +00003396 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003397 if (f == NULL)
3398 return 0;
3399 if (!PyFile_SoftSpace(f, 0))
3400 return 0;
3401 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003402}
3403
Guido van Rossum3f5da241990-12-20 15:06:42 +00003404
Guido van Rossum681d79a1995-07-18 14:51:37 +00003405/* External interface to call any callable object.
3406 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003407
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003408#undef PyEval_CallObject
3409/* for backward compatibility: export this interface */
3410
Guido van Rossumb209a111997-04-29 18:18:01 +00003411PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003412PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003413{
Guido van Rossumb209a111997-04-29 18:18:01 +00003414 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003415}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003416#define PyEval_CallObject(func,arg) \
3417 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003418
Guido van Rossumb209a111997-04-29 18:18:01 +00003419PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003420PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003421{
Jeremy Hylton52820442001-01-03 23:52:36 +00003422 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003423
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003424 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003425 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003426 if (arg == NULL)
3427 return NULL;
3428 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003429 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003430 PyErr_SetString(PyExc_TypeError,
3431 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003432 return NULL;
3433 }
3434 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003435 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003436
Guido van Rossumb209a111997-04-29 18:18:01 +00003437 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003438 PyErr_SetString(PyExc_TypeError,
3439 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003440 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003441 return NULL;
3442 }
3443
Tim Peters6d6c1a32001-08-02 04:15:00 +00003444 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003445 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003446 return result;
3447}
3448
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003449const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003450PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003451{
3452 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003453 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003454 else if (PyFunction_Check(func))
3455 return PyString_AsString(((PyFunctionObject*)func)->func_name);
3456 else if (PyCFunction_Check(func))
3457 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3458 else if (PyClass_Check(func))
3459 return PyString_AsString(((PyClassObject*)func)->cl_name);
3460 else if (PyInstance_Check(func)) {
3461 return PyString_AsString(
3462 ((PyInstanceObject*)func)->in_class->cl_name);
3463 } else {
3464 return func->ob_type->tp_name;
3465 }
3466}
3467
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003468const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003469PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003470{
3471 if (PyMethod_Check(func))
3472 return "()";
3473 else if (PyFunction_Check(func))
3474 return "()";
3475 else if (PyCFunction_Check(func))
3476 return "()";
3477 else if (PyClass_Check(func))
3478 return " constructor";
3479 else if (PyInstance_Check(func)) {
3480 return " instance";
3481 } else {
3482 return " object";
3483 }
3484}
3485
Fredrik Lundh7a830892006-05-27 10:39:48 +00003486static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003487err_args(PyObject *func, int flags, int nargs)
3488{
3489 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003490 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003491 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003492 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003493 nargs);
3494 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003495 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003496 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003497 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003498 nargs);
3499}
3500
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003501#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003502if (tstate->use_tracing && tstate->c_profilefunc) { \
3503 if (call_trace(tstate->c_profilefunc, \
3504 tstate->c_profileobj, \
3505 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003506 func)) { \
3507 x = NULL; \
3508 } \
3509 else { \
3510 x = call; \
3511 if (tstate->c_profilefunc != NULL) { \
3512 if (x == NULL) { \
3513 call_trace_protected(tstate->c_profilefunc, \
3514 tstate->c_profileobj, \
3515 tstate->frame, PyTrace_C_EXCEPTION, \
3516 func); \
3517 /* XXX should pass (type, value, tb) */ \
3518 } else { \
3519 if (call_trace(tstate->c_profilefunc, \
3520 tstate->c_profileobj, \
3521 tstate->frame, PyTrace_C_RETURN, \
3522 func)) { \
3523 Py_DECREF(x); \
3524 x = NULL; \
3525 } \
3526 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003527 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003528 } \
3529} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003530 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003531 }
3532
Fredrik Lundh7a830892006-05-27 10:39:48 +00003533static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003534call_function(PyObject ***pp_stack, int oparg
3535#ifdef WITH_TSC
3536 , uint64* pintr0, uint64* pintr1
3537#endif
3538 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003539{
3540 int na = oparg & 0xff;
3541 int nk = (oparg>>8) & 0xff;
3542 int n = na + 2 * nk;
3543 PyObject **pfunc = (*pp_stack) - n - 1;
3544 PyObject *func = *pfunc;
3545 PyObject *x, *w;
3546
Jeremy Hylton985eba52003-02-05 23:13:00 +00003547 /* Always dispatch PyCFunction first, because these are
3548 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003549 */
3550 if (PyCFunction_Check(func) && nk == 0) {
3551 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003552 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003553
3554 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003555 if (flags & (METH_NOARGS | METH_O)) {
3556 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3557 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003558 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003559 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003560 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003561 else if (flags & METH_O && na == 1) {
3562 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003563 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003564 Py_DECREF(arg);
3565 }
3566 else {
3567 err_args(func, flags, na);
3568 x = NULL;
3569 }
3570 }
3571 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003572 PyObject *callargs;
3573 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003574 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003575 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003576 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003577 Py_XDECREF(callargs);
3578 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003579 } else {
3580 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3581 /* optimize access to bound methods */
3582 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003583 PCALL(PCALL_METHOD);
3584 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003585 Py_INCREF(self);
3586 func = PyMethod_GET_FUNCTION(func);
3587 Py_INCREF(func);
3588 Py_DECREF(*pfunc);
3589 *pfunc = self;
3590 na++;
3591 n++;
3592 } else
3593 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003594 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003595 if (PyFunction_Check(func))
3596 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003597 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003598 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003599 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003600 Py_DECREF(func);
3601 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003602
Armin Rigod34fa522006-03-28 19:10:40 +00003603 /* Clear the stack of the function object. Also removes
3604 the arguments in case they weren't consumed already
3605 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003606 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003607 while ((*pp_stack) > pfunc) {
3608 w = EXT_POP(*pp_stack);
3609 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003610 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003611 }
3612 return x;
3613}
3614
Jeremy Hylton192690e2002-08-16 18:36:11 +00003615/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003616 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003617 For the simplest case -- a function that takes only positional
3618 arguments and is called with only positional arguments -- it
3619 inlines the most primitive frame setup code from
3620 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3621 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003622*/
3623
Fredrik Lundh7a830892006-05-27 10:39:48 +00003624static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003625fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003626{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003627 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003628 PyObject *globals = PyFunction_GET_GLOBALS(func);
3629 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3630 PyObject **d = NULL;
3631 int nd = 0;
3632
Jeremy Hylton985eba52003-02-05 23:13:00 +00003633 PCALL(PCALL_FUNCTION);
3634 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003635 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003636 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3637 PyFrameObject *f;
3638 PyObject *retval = NULL;
3639 PyThreadState *tstate = PyThreadState_GET();
3640 PyObject **fastlocals, **stack;
3641 int i;
3642
3643 PCALL(PCALL_FASTER_FUNCTION);
3644 assert(globals != NULL);
3645 /* XXX Perhaps we should create a specialized
3646 PyFrame_New() that doesn't take locals, but does
3647 take builtins without sanity checking them.
3648 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003649 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003650 f = PyFrame_New(tstate, co, globals, NULL);
3651 if (f == NULL)
3652 return NULL;
3653
3654 fastlocals = f->f_localsplus;
3655 stack = (*pp_stack) - n;
3656
3657 for (i = 0; i < n; i++) {
3658 Py_INCREF(*stack);
3659 fastlocals[i] = *stack++;
3660 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003661 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003662 ++tstate->recursion_depth;
3663 Py_DECREF(f);
3664 --tstate->recursion_depth;
3665 return retval;
3666 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003667 if (argdefs != NULL) {
3668 d = &PyTuple_GET_ITEM(argdefs, 0);
3669 nd = ((PyTupleObject *)argdefs)->ob_size;
3670 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003671 return PyEval_EvalCodeEx(co, globals,
3672 (PyObject *)NULL, (*pp_stack)-n, na,
3673 (*pp_stack)-2*nk, nk, d, nd,
3674 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003675}
3676
Fredrik Lundh7a830892006-05-27 10:39:48 +00003677static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003678update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3679 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003680{
3681 PyObject *kwdict = NULL;
3682 if (orig_kwdict == NULL)
3683 kwdict = PyDict_New();
3684 else {
3685 kwdict = PyDict_Copy(orig_kwdict);
3686 Py_DECREF(orig_kwdict);
3687 }
3688 if (kwdict == NULL)
3689 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003690 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003691 int err;
3692 PyObject *value = EXT_POP(*pp_stack);
3693 PyObject *key = EXT_POP(*pp_stack);
3694 if (PyDict_GetItem(kwdict, key) != NULL) {
Guido van Rossumac7be682001-01-17 15:42:30 +00003695 PyErr_Format(PyExc_TypeError,
Ka-Ping Yee20579702001-01-15 22:14:16 +00003696 "%.200s%s got multiple values "
Jeremy Hylton512a2372001-04-11 13:52:29 +00003697 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003698 PyEval_GetFuncName(func),
3699 PyEval_GetFuncDesc(func),
Jeremy Hylton512a2372001-04-11 13:52:29 +00003700 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003701 Py_DECREF(key);
3702 Py_DECREF(value);
3703 Py_DECREF(kwdict);
3704 return NULL;
3705 }
3706 err = PyDict_SetItem(kwdict, key, value);
3707 Py_DECREF(key);
3708 Py_DECREF(value);
3709 if (err) {
3710 Py_DECREF(kwdict);
3711 return NULL;
3712 }
3713 }
3714 return kwdict;
3715}
3716
Fredrik Lundh7a830892006-05-27 10:39:48 +00003717static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003718update_star_args(int nstack, int nstar, PyObject *stararg,
3719 PyObject ***pp_stack)
3720{
3721 PyObject *callargs, *w;
3722
3723 callargs = PyTuple_New(nstack + nstar);
3724 if (callargs == NULL) {
3725 return NULL;
3726 }
3727 if (nstar) {
3728 int i;
3729 for (i = 0; i < nstar; i++) {
3730 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3731 Py_INCREF(a);
3732 PyTuple_SET_ITEM(callargs, nstack + i, a);
3733 }
3734 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003735 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003736 w = EXT_POP(*pp_stack);
3737 PyTuple_SET_ITEM(callargs, nstack, w);
3738 }
3739 return callargs;
3740}
3741
Fredrik Lundh7a830892006-05-27 10:39:48 +00003742static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003743load_args(PyObject ***pp_stack, int na)
3744{
3745 PyObject *args = PyTuple_New(na);
3746 PyObject *w;
3747
3748 if (args == NULL)
3749 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003750 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003751 w = EXT_POP(*pp_stack);
3752 PyTuple_SET_ITEM(args, na, w);
3753 }
3754 return args;
3755}
3756
Fredrik Lundh7a830892006-05-27 10:39:48 +00003757static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003758do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3759{
3760 PyObject *callargs = NULL;
3761 PyObject *kwdict = NULL;
3762 PyObject *result = NULL;
3763
3764 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003765 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003766 if (kwdict == NULL)
3767 goto call_fail;
3768 }
3769 callargs = load_args(pp_stack, na);
3770 if (callargs == NULL)
3771 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003772#ifdef CALL_PROFILE
3773 /* At this point, we have to look at the type of func to
3774 update the call stats properly. Do it here so as to avoid
3775 exposing the call stats machinery outside ceval.c
3776 */
3777 if (PyFunction_Check(func))
3778 PCALL(PCALL_FUNCTION);
3779 else if (PyMethod_Check(func))
3780 PCALL(PCALL_METHOD);
3781 else if (PyType_Check(func))
3782 PCALL(PCALL_TYPE);
3783 else
3784 PCALL(PCALL_OTHER);
3785#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003786 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003787 call_fail:
3788 Py_XDECREF(callargs);
3789 Py_XDECREF(kwdict);
3790 return result;
3791}
3792
Fredrik Lundh7a830892006-05-27 10:39:48 +00003793static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003794ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3795{
3796 int nstar = 0;
3797 PyObject *callargs = NULL;
3798 PyObject *stararg = NULL;
3799 PyObject *kwdict = NULL;
3800 PyObject *result = NULL;
3801
3802 if (flags & CALL_FLAG_KW) {
3803 kwdict = EXT_POP(*pp_stack);
3804 if (!(kwdict && PyDict_Check(kwdict))) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003805 PyErr_Format(PyExc_TypeError,
Jeremy Hylton512a2372001-04-11 13:52:29 +00003806 "%s%s argument after ** "
3807 "must be a dictionary",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003808 PyEval_GetFuncName(func),
3809 PyEval_GetFuncDesc(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003810 goto ext_call_fail;
3811 }
3812 }
3813 if (flags & CALL_FLAG_VAR) {
3814 stararg = EXT_POP(*pp_stack);
3815 if (!PyTuple_Check(stararg)) {
3816 PyObject *t = NULL;
3817 t = PySequence_Tuple(stararg);
3818 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003819 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3820 PyErr_Format(PyExc_TypeError,
3821 "%s%s argument after * "
3822 "must be a sequence",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003823 PyEval_GetFuncName(func),
3824 PyEval_GetFuncDesc(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003825 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003826 goto ext_call_fail;
3827 }
3828 Py_DECREF(stararg);
3829 stararg = t;
3830 }
3831 nstar = PyTuple_GET_SIZE(stararg);
3832 }
3833 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003834 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003835 if (kwdict == NULL)
3836 goto ext_call_fail;
3837 }
3838 callargs = update_star_args(na, nstar, stararg, pp_stack);
3839 if (callargs == NULL)
3840 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003841#ifdef CALL_PROFILE
3842 /* At this point, we have to look at the type of func to
3843 update the call stats properly. Do it here so as to avoid
3844 exposing the call stats machinery outside ceval.c
3845 */
3846 if (PyFunction_Check(func))
3847 PCALL(PCALL_FUNCTION);
3848 else if (PyMethod_Check(func))
3849 PCALL(PCALL_METHOD);
3850 else if (PyType_Check(func))
3851 PCALL(PCALL_TYPE);
3852 else
3853 PCALL(PCALL_OTHER);
3854#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003855 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003856 ext_call_fail:
3857 Py_XDECREF(callargs);
3858 Py_XDECREF(kwdict);
3859 Py_XDECREF(stararg);
3860 return result;
3861}
3862
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003863/* Extract a slice index from a PyInt or PyLong or an object with the
3864 nb_index slot defined, and store in *pi.
3865 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3866 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 +00003867 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003868*/
Tim Petersb5196382001-12-16 19:44:20 +00003869/* Note: If v is NULL, return success without storing into *pi. This
3870 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3871 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003872*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003873int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003874_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003875{
Tim Petersb5196382001-12-16 19:44:20 +00003876 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003877 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00003878 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003879 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
3880 however, it looks like it should be AsSsize_t.
3881 There should be a comment here explaining why.
3882 */
3883 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00003884 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003885 else if (PyIndex_Check(v)) {
3886 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003887 if (x == -1 && PyErr_Occurred())
3888 return 0;
3889 }
3890 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003891 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003892 "slice indices must be integers or "
3893 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003894 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003895 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003896 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003897 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003898 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003899}
3900
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003901#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003902#define ISINDEX(x) ((x) == NULL || \
3903 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00003904
Fredrik Lundh7a830892006-05-27 10:39:48 +00003905static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003906apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003907{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003908 PyTypeObject *tp = u->ob_type;
3909 PySequenceMethods *sq = tp->tp_as_sequence;
3910
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003911 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003912 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003913 if (!_PyEval_SliceIndex(v, &ilow))
3914 return NULL;
3915 if (!_PyEval_SliceIndex(w, &ihigh))
3916 return NULL;
3917 return PySequence_GetSlice(u, ilow, ihigh);
3918 }
3919 else {
3920 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00003921 if (slice != NULL) {
3922 PyObject *res = PyObject_GetItem(u, slice);
3923 Py_DECREF(slice);
3924 return res;
3925 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00003926 else
3927 return NULL;
3928 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003929}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003930
Fredrik Lundh7a830892006-05-27 10:39:48 +00003931static int
Guido van Rossumac7be682001-01-17 15:42:30 +00003932assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
3933 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003934{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003935 PyTypeObject *tp = u->ob_type;
3936 PySequenceMethods *sq = tp->tp_as_sequence;
3937
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003938 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003939 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003940 if (!_PyEval_SliceIndex(v, &ilow))
3941 return -1;
3942 if (!_PyEval_SliceIndex(w, &ihigh))
3943 return -1;
3944 if (x == NULL)
3945 return PySequence_DelSlice(u, ilow, ihigh);
3946 else
3947 return PySequence_SetSlice(u, ilow, ihigh, x);
3948 }
3949 else {
3950 PyObject *slice = PySlice_New(v, w, NULL);
3951 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00003952 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003953 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00003954 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00003955 else
Guido van Rossum354797c2001-12-03 19:45:06 +00003956 res = PyObject_DelItem(u, slice);
3957 Py_DECREF(slice);
3958 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003959 }
3960 else
3961 return -1;
3962 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003963}
3964
Fredrik Lundh7a830892006-05-27 10:39:48 +00003965static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003966cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003967{
Guido van Rossumac7be682001-01-17 15:42:30 +00003968 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003969 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003970 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003971 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003972 break;
3973 case PyCmp_IS_NOT:
3974 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003975 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003976 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003977 res = PySequence_Contains(w, v);
3978 if (res < 0)
3979 return NULL;
3980 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003981 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003982 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003983 if (res < 0)
3984 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003985 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003986 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003987 case PyCmp_EXC_MATCH:
Barry Warsaw4249f541997-08-22 21:26:19 +00003988 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003989 break;
3990 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003991 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003992 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003993 v = res ? Py_True : Py_False;
3994 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003995 return v;
3996}
3997
Fredrik Lundh7a830892006-05-27 10:39:48 +00003998static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00003999import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004000{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004001 PyObject *x;
4002
4003 x = PyObject_GetAttr(v, name);
4004 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004005 PyErr_Format(PyExc_ImportError,
4006 "cannot import name %.230s",
4007 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004008 }
Thomas Wouters52152252000-08-17 22:55:00 +00004009 return x;
4010}
Guido van Rossumac7be682001-01-17 15:42:30 +00004011
Fredrik Lundh7a830892006-05-27 10:39:48 +00004012static int
Thomas Wouters52152252000-08-17 22:55:00 +00004013import_all_from(PyObject *locals, PyObject *v)
4014{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004015 PyObject *all = PyObject_GetAttrString(v, "__all__");
4016 PyObject *dict, *name, *value;
4017 int skip_leading_underscores = 0;
4018 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004019
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004020 if (all == NULL) {
4021 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4022 return -1; /* Unexpected error */
4023 PyErr_Clear();
4024 dict = PyObject_GetAttrString(v, "__dict__");
4025 if (dict == NULL) {
4026 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4027 return -1;
4028 PyErr_SetString(PyExc_ImportError,
4029 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004030 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004031 }
4032 all = PyMapping_Keys(dict);
4033 Py_DECREF(dict);
4034 if (all == NULL)
4035 return -1;
4036 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004037 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004038
4039 for (pos = 0, err = 0; ; pos++) {
4040 name = PySequence_GetItem(all, pos);
4041 if (name == NULL) {
4042 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4043 err = -1;
4044 else
4045 PyErr_Clear();
4046 break;
4047 }
4048 if (skip_leading_underscores &&
4049 PyString_Check(name) &&
4050 PyString_AS_STRING(name)[0] == '_')
4051 {
4052 Py_DECREF(name);
4053 continue;
4054 }
4055 value = PyObject_GetAttr(v, name);
4056 if (value == NULL)
4057 err = -1;
Armin Rigo70370852006-11-29 21:59:22 +00004058 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004059 err = PyDict_SetItem(locals, name, value);
Armin Rigo70370852006-11-29 21:59:22 +00004060 else
4061 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004062 Py_DECREF(name);
4063 Py_XDECREF(value);
4064 if (err != 0)
4065 break;
4066 }
4067 Py_DECREF(all);
4068 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004069}
4070
Fredrik Lundh7a830892006-05-27 10:39:48 +00004071static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004072build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004073{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004074 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004075
4076 if (PyDict_Check(methods))
4077 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004078 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004079 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004080 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4081 base = PyTuple_GET_ITEM(bases, 0);
4082 metaclass = PyObject_GetAttrString(base, "__class__");
4083 if (metaclass == NULL) {
4084 PyErr_Clear();
4085 metaclass = (PyObject *)base->ob_type;
4086 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004087 }
4088 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004089 else {
4090 PyObject *g = PyEval_GetGlobals();
4091 if (g != NULL && PyDict_Check(g))
4092 metaclass = PyDict_GetItemString(g, "__metaclass__");
4093 if (metaclass == NULL)
4094 metaclass = (PyObject *) &PyClass_Type;
4095 Py_INCREF(metaclass);
4096 }
Georg Brandl684fd0c2006-05-25 19:15:31 +00004097 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods, NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004098 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004099 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004100 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004101 in a base that was not a class (such the random module
4102 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004103 by augmenting the error message with more information.*/
4104
4105 PyObject *ptype, *pvalue, *ptraceback;
4106
4107 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
4108 if (PyString_Check(pvalue)) {
4109 PyObject *newmsg;
4110 newmsg = PyString_FromFormat(
4111 "Error when calling the metaclass bases\n %s",
4112 PyString_AS_STRING(pvalue));
4113 if (newmsg != NULL) {
4114 Py_DECREF(pvalue);
4115 pvalue = newmsg;
4116 }
4117 }
4118 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004119 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004120 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004121}
4122
Fredrik Lundh7a830892006-05-27 10:39:48 +00004123static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004124exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4125 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004126{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004127 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004128 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004129 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004130
Guido van Rossumb209a111997-04-29 18:18:01 +00004131 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4132 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004133 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004134 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004135 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004136 locals = PyTuple_GetItem(prog, 2);
4137 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004138 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004139 if (globals == Py_None) {
4140 globals = PyEval_GetGlobals();
4141 if (locals == Py_None) {
4142 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004143 plain = 1;
4144 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004145 if (!globals || !locals) {
4146 PyErr_SetString(PyExc_SystemError,
4147 "globals and locals cannot be NULL");
4148 return -1;
4149 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004150 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004151 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004152 locals = globals;
Guido van Rossumb209a111997-04-29 18:18:01 +00004153 if (!PyString_Check(prog) &&
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004154 !PyUnicode_Check(prog) &&
Guido van Rossumb209a111997-04-29 18:18:01 +00004155 !PyCode_Check(prog) &&
4156 !PyFile_Check(prog)) {
4157 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004158 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004159 return -1;
4160 }
Fred Drake661ea262000-10-24 19:57:45 +00004161 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004162 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004163 "exec: arg 2 must be a dictionary or None");
4164 return -1;
4165 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004166 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004167 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004168 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004169 return -1;
4170 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004171 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004172 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004173 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004174 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4175 PyErr_SetString(PyExc_TypeError,
4176 "code object passed to exec may not contain free variables");
4177 return -1;
4178 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004179 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004180 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004181 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004182 FILE *fp = PyFile_AsFile(prog);
4183 char *name = PyString_AsString(PyFile_Name(prog));
Tim Peters5ba58662001-07-16 02:29:45 +00004184 PyCompilerFlags cf;
4185 cf.cf_flags = 0;
4186 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004187 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004188 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004189 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004190 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004191 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004192 }
4193 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004194 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004195 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004196 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004197 cf.cf_flags = 0;
4198#ifdef Py_USING_UNICODE
4199 if (PyUnicode_Check(prog)) {
4200 tmp = PyUnicode_AsUTF8String(prog);
4201 if (tmp == NULL)
4202 return -1;
4203 prog = tmp;
4204 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4205 }
4206#endif
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004207 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004208 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004209 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004210 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004211 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004212 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004213 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004214 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004215 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004216 if (plain)
4217 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004218 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004219 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004220 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004221 return 0;
4222}
Guido van Rossum24c13741995-02-14 09:42:43 +00004223
Fredrik Lundh7a830892006-05-27 10:39:48 +00004224static void
Paul Prescode68140d2000-08-30 20:25:01 +00004225format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4226{
4227 char *obj_str;
4228
4229 if (!obj)
4230 return;
4231
4232 obj_str = PyString_AsString(obj);
4233 if (!obj_str)
4234 return;
4235
4236 PyErr_Format(exc, format_str, obj_str);
4237}
Guido van Rossum950361c1997-01-24 13:49:28 +00004238
Fredrik Lundh7a830892006-05-27 10:39:48 +00004239static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004240string_concatenate(PyObject *v, PyObject *w,
4241 PyFrameObject *f, unsigned char *next_instr)
4242{
4243 /* This function implements 'variable += expr' when both arguments
4244 are strings. */
Armin Rigo97ff0472006-08-09 15:37:26 +00004245 Py_ssize_t v_len = PyString_GET_SIZE(v);
4246 Py_ssize_t w_len = PyString_GET_SIZE(w);
4247 Py_ssize_t new_len = v_len + w_len;
4248 if (new_len < 0) {
4249 PyErr_SetString(PyExc_OverflowError,
4250 "strings are too large to concat");
4251 return NULL;
4252 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004253
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004254 if (v->ob_refcnt == 2) {
4255 /* In the common case, there are 2 references to the value
4256 * stored in 'variable' when the += is performed: one on the
4257 * value stack (in 'v') and one still stored in the 'variable'.
4258 * We try to delete the variable now to reduce the refcnt to 1.
4259 */
4260 switch (*next_instr) {
4261 case STORE_FAST:
4262 {
4263 int oparg = PEEKARG();
4264 PyObject **fastlocals = f->f_localsplus;
4265 if (GETLOCAL(oparg) == v)
4266 SETLOCAL(oparg, NULL);
4267 break;
4268 }
4269 case STORE_DEREF:
4270 {
Richard Jonescebbefc2006-05-23 18:28:17 +00004271 PyObject **freevars = f->f_localsplus + f->f_code->co_nlocals;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004272 PyObject *c = freevars[PEEKARG()];
4273 if (PyCell_GET(c) == v)
4274 PyCell_Set(c, NULL);
4275 break;
4276 }
4277 case STORE_NAME:
4278 {
4279 PyObject *names = f->f_code->co_names;
4280 PyObject *name = GETITEM(names, PEEKARG());
4281 PyObject *locals = f->f_locals;
4282 if (PyDict_CheckExact(locals) &&
4283 PyDict_GetItem(locals, name) == v) {
4284 if (PyDict_DelItem(locals, name) != 0) {
4285 PyErr_Clear();
4286 }
4287 }
4288 break;
4289 }
4290 }
4291 }
4292
Armin Rigo618fbf52004-08-07 20:58:32 +00004293 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004294 /* Now we own the last reference to 'v', so we can resize it
4295 * in-place.
4296 */
Armin Rigo97ff0472006-08-09 15:37:26 +00004297 if (_PyString_Resize(&v, new_len) != 0) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004298 /* XXX if _PyString_Resize() fails, 'v' has been
4299 * deallocated so it cannot be put back into 'variable'.
4300 * The MemoryError is raised when there is no value in
4301 * 'variable', which might (very remotely) be a cause
4302 * of incompatibilities.
4303 */
4304 return NULL;
4305 }
4306 /* copy 'w' into the newly allocated area of 'v' */
4307 memcpy(PyString_AS_STRING(v) + v_len,
4308 PyString_AS_STRING(w), w_len);
4309 return v;
4310 }
4311 else {
4312 /* When in-place resizing is not an option. */
4313 PyString_Concat(&v, w);
4314 return v;
4315 }
4316}
4317
Guido van Rossum950361c1997-01-24 13:49:28 +00004318#ifdef DYNAMIC_EXECUTION_PROFILE
4319
Fredrik Lundh7a830892006-05-27 10:39:48 +00004320static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004321getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004322{
4323 int i;
4324 PyObject *l = PyList_New(256);
4325 if (l == NULL) return NULL;
4326 for (i = 0; i < 256; i++) {
4327 PyObject *x = PyInt_FromLong(a[i]);
4328 if (x == NULL) {
4329 Py_DECREF(l);
4330 return NULL;
4331 }
4332 PyList_SetItem(l, i, x);
4333 }
4334 for (i = 0; i < 256; i++)
4335 a[i] = 0;
4336 return l;
4337}
4338
4339PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004340_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004341{
4342#ifndef DXPAIRS
4343 return getarray(dxp);
4344#else
4345 int i;
4346 PyObject *l = PyList_New(257);
4347 if (l == NULL) return NULL;
4348 for (i = 0; i < 257; i++) {
4349 PyObject *x = getarray(dxpairs[i]);
4350 if (x == NULL) {
4351 Py_DECREF(l);
4352 return NULL;
4353 }
4354 PyList_SetItem(l, i, x);
4355 }
4356 return l;
4357#endif
4358}
4359
4360#endif