blob: 9dddd2ff2fb2bfc8f0daa920b227e910bb2e56d7 [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. Kuchling5f958702006-10-27 13:29:41 +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. Kuchling5f958702006-10-27 13:29:41 +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); }
Neal Norwitz03c566a2007-04-16 06:19:52 +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
744 does work. Promise. */
745 next_instr = first_instr + f->f_lasti + 1;
746 stack_pointer = f->f_stacktop;
747 assert(stack_pointer != NULL);
748 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
749
Tim Peters5ca576e2001-06-18 22:08:13 +0000750#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000751 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000752#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000753#if defined(Py_DEBUG) || defined(LLTRACE)
Tim Peters5ca576e2001-06-18 22:08:13 +0000754 filename = PyString_AsString(co->co_filename);
755#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000756
Guido van Rossum374a9221991-04-04 10:40:29 +0000757 why = WHY_NOT;
758 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000759 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000760 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000761
Anthony Baxtera863d332006-04-11 07:43:46 +0000762 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000763 why = WHY_EXCEPTION;
764 goto on_error;
765 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000766
Guido van Rossum374a9221991-04-04 10:40:29 +0000767 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000768#ifdef WITH_TSC
769 if (inst1 == 0) {
770 /* Almost surely, the opcode executed a break
771 or a continue, preventing inst1 from being set
772 on the way out of the loop.
773 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000774 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000775 loop1 = inst1;
776 }
777 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
778 intr0, intr1);
779 ticked = 0;
780 inst1 = 0;
781 intr0 = 0;
782 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000783 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000784#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000785 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000786 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000787
Guido van Rossuma027efa1997-05-05 20:56:21 +0000788 /* Do periodic things. Doing this every time through
789 the loop would add too much overhead, so we do it
790 only every Nth instruction. We also do it if
791 ``things_to_do'' is set, i.e. when an asynchronous
792 event needs attention (e.g. a signal handler or
793 async I/O handler); see Py_AddPendingCall() and
794 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000795
Skip Montanarod581d772002-09-03 20:10:45 +0000796 if (--_Py_Ticker < 0) {
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000797 if (*next_instr == SETUP_FINALLY) {
798 /* Make the last opcode before
799 a try: finally: block uninterruptable. */
800 goto fast_next_opcode;
801 }
Skip Montanarod581d772002-09-03 20:10:45 +0000802 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000803 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000804#ifdef WITH_TSC
805 ticked = 1;
806#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000807 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000808 if (Py_MakePendingCalls() < 0) {
809 why = WHY_EXCEPTION;
810 goto on_error;
811 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000812 if (things_to_do)
813 /* MakePendingCalls() didn't succeed.
814 Force early re-execution of this
815 "periodic" code, possibly after
816 a thread switch */
817 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000818 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000819#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000820 if (interpreter_lock) {
821 /* Give another thread a chance */
822
Guido van Rossum25ce5661997-08-02 03:10:38 +0000823 if (PyThreadState_Swap(NULL) != tstate)
824 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000825 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000826
827 /* Other threads may run now */
828
Guido van Rossum65d5b571998-12-21 19:32:43 +0000829 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000830 if (PyThreadState_Swap(tstate) != NULL)
831 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000832
833 /* Check for thread interrupts */
834
835 if (tstate->async_exc != NULL) {
836 x = tstate->async_exc;
837 tstate->async_exc = NULL;
838 PyErr_SetNone(x);
839 Py_DECREF(x);
840 why = WHY_EXCEPTION;
841 goto on_error;
842 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000843 }
844#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000845 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000846
Neil Schemenauer63543862002-02-17 19:10:14 +0000847 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000848 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000849
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000850 /* line-by-line tracing support */
851
852 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
853 /* see maybe_call_line_trace
854 for expository comments */
855 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000856
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000857 err = maybe_call_line_trace(tstate->c_tracefunc,
858 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000859 f, &instr_lb, &instr_ub,
860 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000861 /* Reload possibly changed frame fields */
862 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000863 if (f->f_stacktop != NULL) {
864 stack_pointer = f->f_stacktop;
865 f->f_stacktop = NULL;
866 }
867 if (err) {
868 /* trace function raised an exception */
869 goto on_error;
870 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000871 }
872
873 /* Extract opcode and argument */
874
Guido van Rossum374a9221991-04-04 10:40:29 +0000875 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000876 oparg = 0; /* allows oparg to be stored in a register because
877 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000878 if (HAS_ARG(opcode))
879 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000880 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000881#ifdef DYNAMIC_EXECUTION_PROFILE
882#ifdef DXPAIRS
883 dxpairs[lastopcode][opcode]++;
884 lastopcode = opcode;
885#endif
886 dxp[opcode]++;
887#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000888
Guido van Rossum96a42c81992-01-12 02:29:51 +0000889#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000890 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000891
Guido van Rossum96a42c81992-01-12 02:29:51 +0000892 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000893 if (HAS_ARG(opcode)) {
894 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000895 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000896 }
897 else {
898 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000899 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000900 }
901 }
902#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000903
Guido van Rossum374a9221991-04-04 10:40:29 +0000904 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000905 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000906
Guido van Rossum374a9221991-04-04 10:40:29 +0000907 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000908
Guido van Rossum374a9221991-04-04 10:40:29 +0000909 /* BEWARE!
910 It is essential that any operation that fails sets either
911 x to NULL, err to nonzero, or why to anything but WHY_NOT,
912 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000913
Guido van Rossum374a9221991-04-04 10:40:29 +0000914 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000915
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000916 case NOP:
917 goto fast_next_opcode;
918
Neil Schemenauer63543862002-02-17 19:10:14 +0000919 case LOAD_FAST:
920 x = GETLOCAL(oparg);
921 if (x != NULL) {
922 Py_INCREF(x);
923 PUSH(x);
924 goto fast_next_opcode;
925 }
926 format_exc_check_arg(PyExc_UnboundLocalError,
927 UNBOUNDLOCAL_ERROR_MSG,
928 PyTuple_GetItem(co->co_varnames, oparg));
929 break;
930
931 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000932 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000933 Py_INCREF(x);
934 PUSH(x);
935 goto fast_next_opcode;
936
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000937 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000938 case STORE_FAST:
939 v = POP();
940 SETLOCAL(oparg, v);
941 goto fast_next_opcode;
942
Raymond Hettingerf606f872003-03-16 03:11:04 +0000943 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000944 case POP_TOP:
945 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000946 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000947 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000948
Guido van Rossum374a9221991-04-04 10:40:29 +0000949 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000950 v = TOP();
951 w = SECOND();
952 SET_TOP(w);
953 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000954 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000955
Guido van Rossum374a9221991-04-04 10:40:29 +0000956 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000957 v = TOP();
958 w = SECOND();
959 x = THIRD();
960 SET_TOP(w);
961 SET_SECOND(x);
962 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000963 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000964
Thomas Wouters434d0822000-08-24 20:11:32 +0000965 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000966 u = TOP();
967 v = SECOND();
968 w = THIRD();
969 x = FOURTH();
970 SET_TOP(v);
971 SET_SECOND(w);
972 SET_THIRD(x);
973 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000974 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000975
Guido van Rossum374a9221991-04-04 10:40:29 +0000976 case DUP_TOP:
977 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000978 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000979 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000980 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000981
Thomas Wouters434d0822000-08-24 20:11:32 +0000982 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000983 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +0000984 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +0000985 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000986 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +0000987 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000988 STACKADJ(2);
989 SET_TOP(x);
990 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +0000991 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000992 } else if (oparg == 3) {
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 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +0000998 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000999 STACKADJ(3);
1000 SET_TOP(x);
1001 SET_SECOND(w);
1002 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001003 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001004 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001005 Py_FatalError("invalid argument to DUP_TOPX"
1006 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001007 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001008
Guido van Rossum374a9221991-04-04 10:40:29 +00001009 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001010 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001011 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001012 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001013 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001014 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001015 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001016
Guido van Rossum374a9221991-04-04 10:40:29 +00001017 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001018 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001019 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001020 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001021 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001022 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001023 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001024
Guido van Rossum374a9221991-04-04 10:40:29 +00001025 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001026 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001027 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001028 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001029 if (err == 0) {
1030 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001031 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001032 continue;
1033 }
1034 else if (err > 0) {
1035 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001036 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001037 err = 0;
1038 continue;
1039 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001040 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001041 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001042
Guido van Rossum374a9221991-04-04 10:40:29 +00001043 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001044 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001045 x = PyObject_Repr(v);
1046 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001047 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001048 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001049 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001050
Guido van Rossum7928cd71991-10-24 14:59:31 +00001051 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001052 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001053 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001054 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001055 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001056 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001057 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001058
Guido van Rossum50564e81996-01-12 01:13:16 +00001059 case BINARY_POWER:
1060 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001061 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001062 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001063 Py_DECREF(v);
1064 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001065 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001066 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001067 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001068
Guido van Rossum374a9221991-04-04 10:40:29 +00001069 case BINARY_MULTIPLY:
1070 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001071 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001072 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001073 Py_DECREF(v);
1074 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001075 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001076 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001077 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001078
Guido van Rossum374a9221991-04-04 10:40:29 +00001079 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001080 if (!_Py_QnewFlag) {
1081 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001082 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001083 x = PyNumber_Divide(v, w);
1084 Py_DECREF(v);
1085 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001086 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001087 if (x != NULL) continue;
1088 break;
1089 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001090 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001091 BINARY_TRUE_DIVIDE */
1092 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001093 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001094 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001095 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001096 Py_DECREF(v);
1097 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001098 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001099 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001100 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001101
Guido van Rossum4668b002001-08-08 05:00:18 +00001102 case BINARY_FLOOR_DIVIDE:
1103 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001104 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001105 x = PyNumber_FloorDivide(v, w);
1106 Py_DECREF(v);
1107 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001108 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001109 if (x != NULL) continue;
1110 break;
1111
Guido van Rossum374a9221991-04-04 10:40:29 +00001112 case BINARY_MODULO:
1113 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001114 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001115 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001116 Py_DECREF(v);
1117 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001118 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001119 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001120 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001121
Guido van Rossum374a9221991-04-04 10:40:29 +00001122 case BINARY_ADD:
1123 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001124 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001125 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001126 /* INLINE: int + int */
1127 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001128 a = PyInt_AS_LONG(v);
1129 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001130 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001131 if ((i^a) < 0 && (i^b) < 0)
1132 goto slow_add;
1133 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001134 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001135 else if (PyString_CheckExact(v) &&
1136 PyString_CheckExact(w)) {
1137 x = string_concatenate(v, w, f, next_instr);
1138 /* string_concatenate consumed the ref to v */
1139 goto skip_decref_vx;
1140 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001141 else {
1142 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001143 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001144 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001145 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001146 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001147 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001148 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001149 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001150 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001151
Guido van Rossum374a9221991-04-04 10:40:29 +00001152 case BINARY_SUBTRACT:
1153 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001154 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001155 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001156 /* INLINE: int - int */
1157 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001158 a = PyInt_AS_LONG(v);
1159 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001160 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001161 if ((i^a) < 0 && (i^~b) < 0)
1162 goto slow_sub;
1163 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001164 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001165 else {
1166 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001167 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001168 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001169 Py_DECREF(v);
1170 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001171 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001172 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001173 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001174
Guido van Rossum374a9221991-04-04 10:40:29 +00001175 case BINARY_SUBSCR:
1176 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001177 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001178 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001179 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001180 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001181 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001182 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001183 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001184 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001185 Py_INCREF(x);
1186 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001187 else
1188 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001189 }
1190 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001191 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001192 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001193 Py_DECREF(v);
1194 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001195 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001196 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001197 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001198
Guido van Rossum7928cd71991-10-24 14:59:31 +00001199 case BINARY_LSHIFT:
1200 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001201 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001202 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001203 Py_DECREF(v);
1204 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001205 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001206 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001207 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001208
Guido van Rossum7928cd71991-10-24 14:59:31 +00001209 case BINARY_RSHIFT:
1210 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001211 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001212 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001213 Py_DECREF(v);
1214 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001215 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001216 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001217 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001218
Guido van Rossum7928cd71991-10-24 14:59:31 +00001219 case BINARY_AND:
1220 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001221 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001222 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001223 Py_DECREF(v);
1224 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001225 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001226 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001227 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001228
Guido van Rossum7928cd71991-10-24 14:59:31 +00001229 case BINARY_XOR:
1230 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001231 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001232 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001233 Py_DECREF(v);
1234 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001235 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001236 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001237 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001238
Guido van Rossum7928cd71991-10-24 14:59:31 +00001239 case BINARY_OR:
1240 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001241 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001242 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001243 Py_DECREF(v);
1244 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001245 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001246 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001247 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001248
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001249 case LIST_APPEND:
1250 w = POP();
1251 v = POP();
1252 err = PyList_Append(v, w);
1253 Py_DECREF(v);
1254 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001255 if (err == 0) {
1256 PREDICT(JUMP_ABSOLUTE);
1257 continue;
1258 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001259 break;
1260
Thomas Wouters434d0822000-08-24 20:11:32 +00001261 case INPLACE_POWER:
1262 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001263 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001264 x = PyNumber_InPlacePower(v, w, Py_None);
1265 Py_DECREF(v);
1266 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001267 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001268 if (x != NULL) continue;
1269 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001270
Thomas Wouters434d0822000-08-24 20:11:32 +00001271 case INPLACE_MULTIPLY:
1272 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001273 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001274 x = PyNumber_InPlaceMultiply(v, w);
1275 Py_DECREF(v);
1276 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001277 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001278 if (x != NULL) continue;
1279 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001280
Thomas Wouters434d0822000-08-24 20:11:32 +00001281 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001282 if (!_Py_QnewFlag) {
1283 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001284 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001285 x = PyNumber_InPlaceDivide(v, w);
1286 Py_DECREF(v);
1287 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001288 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001289 if (x != NULL) continue;
1290 break;
1291 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001292 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001293 INPLACE_TRUE_DIVIDE */
1294 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001295 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001296 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001297 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001298 Py_DECREF(v);
1299 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001300 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001301 if (x != NULL) continue;
1302 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001303
Guido van Rossum4668b002001-08-08 05:00:18 +00001304 case INPLACE_FLOOR_DIVIDE:
1305 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001306 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001307 x = PyNumber_InPlaceFloorDivide(v, w);
1308 Py_DECREF(v);
1309 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001310 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001311 if (x != NULL) continue;
1312 break;
1313
Thomas Wouters434d0822000-08-24 20:11:32 +00001314 case INPLACE_MODULO:
1315 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001316 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001317 x = PyNumber_InPlaceRemainder(v, w);
1318 Py_DECREF(v);
1319 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001320 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001321 if (x != NULL) continue;
1322 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001323
Thomas Wouters434d0822000-08-24 20:11:32 +00001324 case INPLACE_ADD:
1325 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001326 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001327 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001328 /* INLINE: int + int */
1329 register long a, b, i;
1330 a = PyInt_AS_LONG(v);
1331 b = PyInt_AS_LONG(w);
1332 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001333 if ((i^a) < 0 && (i^b) < 0)
1334 goto slow_iadd;
1335 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001336 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001337 else if (PyString_CheckExact(v) &&
1338 PyString_CheckExact(w)) {
1339 x = string_concatenate(v, w, f, next_instr);
1340 /* string_concatenate consumed the ref to v */
1341 goto skip_decref_v;
1342 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001343 else {
1344 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001345 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001346 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001347 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001348 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001349 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001350 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001351 if (x != NULL) continue;
1352 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001353
Thomas Wouters434d0822000-08-24 20:11:32 +00001354 case INPLACE_SUBTRACT:
1355 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001356 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001357 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001358 /* INLINE: int - int */
1359 register long a, b, i;
1360 a = PyInt_AS_LONG(v);
1361 b = PyInt_AS_LONG(w);
1362 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001363 if ((i^a) < 0 && (i^~b) < 0)
1364 goto slow_isub;
1365 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001366 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001367 else {
1368 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001369 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001370 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001371 Py_DECREF(v);
1372 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001373 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001374 if (x != NULL) continue;
1375 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001376
Thomas Wouters434d0822000-08-24 20:11:32 +00001377 case INPLACE_LSHIFT:
1378 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001379 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001380 x = PyNumber_InPlaceLshift(v, w);
1381 Py_DECREF(v);
1382 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001383 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001384 if (x != NULL) continue;
1385 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001386
Thomas Wouters434d0822000-08-24 20:11:32 +00001387 case INPLACE_RSHIFT:
1388 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001389 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001390 x = PyNumber_InPlaceRshift(v, w);
1391 Py_DECREF(v);
1392 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001393 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001394 if (x != NULL) continue;
1395 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001396
Thomas Wouters434d0822000-08-24 20:11:32 +00001397 case INPLACE_AND:
1398 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001399 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001400 x = PyNumber_InPlaceAnd(v, w);
1401 Py_DECREF(v);
1402 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001403 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001404 if (x != NULL) continue;
1405 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001406
Thomas Wouters434d0822000-08-24 20:11:32 +00001407 case INPLACE_XOR:
1408 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001409 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001410 x = PyNumber_InPlaceXor(v, w);
1411 Py_DECREF(v);
1412 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001413 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001414 if (x != NULL) continue;
1415 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001416
Thomas Wouters434d0822000-08-24 20:11:32 +00001417 case INPLACE_OR:
1418 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001419 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001420 x = PyNumber_InPlaceOr(v, w);
1421 Py_DECREF(v);
1422 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001423 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001424 if (x != NULL) continue;
1425 break;
1426
Guido van Rossum374a9221991-04-04 10:40:29 +00001427 case SLICE+0:
1428 case SLICE+1:
1429 case SLICE+2:
1430 case SLICE+3:
1431 if ((opcode-SLICE) & 2)
1432 w = POP();
1433 else
1434 w = NULL;
1435 if ((opcode-SLICE) & 1)
1436 v = POP();
1437 else
1438 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001439 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001440 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001441 Py_DECREF(u);
1442 Py_XDECREF(v);
1443 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001444 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001445 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001446 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001447
Guido van Rossum374a9221991-04-04 10:40:29 +00001448 case STORE_SLICE+0:
1449 case STORE_SLICE+1:
1450 case STORE_SLICE+2:
1451 case STORE_SLICE+3:
1452 if ((opcode-STORE_SLICE) & 2)
1453 w = POP();
1454 else
1455 w = NULL;
1456 if ((opcode-STORE_SLICE) & 1)
1457 v = POP();
1458 else
1459 v = NULL;
1460 u = POP();
1461 t = POP();
1462 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001463 Py_DECREF(t);
1464 Py_DECREF(u);
1465 Py_XDECREF(v);
1466 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001467 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001468 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001469
Guido van Rossum374a9221991-04-04 10:40:29 +00001470 case DELETE_SLICE+0:
1471 case DELETE_SLICE+1:
1472 case DELETE_SLICE+2:
1473 case DELETE_SLICE+3:
1474 if ((opcode-DELETE_SLICE) & 2)
1475 w = POP();
1476 else
1477 w = NULL;
1478 if ((opcode-DELETE_SLICE) & 1)
1479 v = POP();
1480 else
1481 v = NULL;
1482 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001483 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001484 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001485 Py_DECREF(u);
1486 Py_XDECREF(v);
1487 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001488 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001489 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001490
Guido van Rossum374a9221991-04-04 10:40:29 +00001491 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001492 w = TOP();
1493 v = SECOND();
1494 u = THIRD();
1495 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001496 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001497 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001498 Py_DECREF(u);
1499 Py_DECREF(v);
1500 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001501 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001502 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001503
Guido van Rossum374a9221991-04-04 10:40:29 +00001504 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001505 w = TOP();
1506 v = SECOND();
1507 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001508 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001509 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001510 Py_DECREF(v);
1511 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001512 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001513 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001514
Guido van Rossum374a9221991-04-04 10:40:29 +00001515 case PRINT_EXPR:
1516 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001517 w = PySys_GetObject("displayhook");
1518 if (w == NULL) {
1519 PyErr_SetString(PyExc_RuntimeError,
1520 "lost sys.displayhook");
1521 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001522 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001523 }
1524 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001525 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001526 if (x == NULL)
1527 err = -1;
1528 }
1529 if (err == 0) {
1530 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001531 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001532 if (w == NULL)
1533 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001534 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001535 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001536 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001537 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001538
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001539 case PRINT_ITEM_TO:
1540 w = stream = POP();
1541 /* fall through to PRINT_ITEM */
1542
Guido van Rossum374a9221991-04-04 10:40:29 +00001543 case PRINT_ITEM:
1544 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001545 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001546 w = PySys_GetObject("stdout");
1547 if (w == NULL) {
1548 PyErr_SetString(PyExc_RuntimeError,
1549 "lost sys.stdout");
1550 err = -1;
1551 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001552 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001553 /* PyFile_SoftSpace() can exececute arbitrary code
1554 if sys.stdout is an instance with a __getattr__.
1555 If __getattr__ raises an exception, w will
1556 be freed, so we need to prevent that temporarily. */
1557 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001558 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001559 err = PyFile_WriteString(" ", w);
1560 if (err == 0)
1561 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001562 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001563 /* XXX move into writeobject() ? */
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001564 if (PyString_Check(v)) {
1565 char *s = PyString_AS_STRING(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001566 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001567 if (len == 0 ||
1568 !isspace(Py_CHARMASK(s[len-1])) ||
1569 s[len-1] == ' ')
1570 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001571 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001572#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001573 else if (PyUnicode_Check(v)) {
1574 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001575 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001576 if (len == 0 ||
1577 !Py_UNICODE_ISSPACE(s[len-1]) ||
1578 s[len-1] == ' ')
1579 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001580 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001581#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001582 else
1583 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001584 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001585 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001586 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001587 Py_XDECREF(stream);
1588 stream = NULL;
1589 if (err == 0)
1590 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001591 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001592
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001593 case PRINT_NEWLINE_TO:
1594 w = stream = POP();
1595 /* fall through to PRINT_NEWLINE */
1596
Guido van Rossum374a9221991-04-04 10:40:29 +00001597 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001598 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001599 w = PySys_GetObject("stdout");
1600 if (w == NULL)
1601 PyErr_SetString(PyExc_RuntimeError,
1602 "lost sys.stdout");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001603 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001604 if (w != NULL) {
1605 err = PyFile_WriteString("\n", w);
1606 if (err == 0)
1607 PyFile_SoftSpace(w, 0);
1608 }
1609 Py_XDECREF(stream);
1610 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001611 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001612
Thomas Wouters434d0822000-08-24 20:11:32 +00001613
1614#ifdef CASE_TOO_BIG
1615 default: switch (opcode) {
1616#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001617 case RAISE_VARARGS:
1618 u = v = w = NULL;
1619 switch (oparg) {
1620 case 3:
1621 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001622 /* Fallthrough */
1623 case 2:
1624 v = POP(); /* value */
1625 /* Fallthrough */
1626 case 1:
1627 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001628 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001629 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001630 break;
1631 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001632 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001633 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001634 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001635 break;
1636 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001637 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001638
Guido van Rossum374a9221991-04-04 10:40:29 +00001639 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001640 if ((x = f->f_locals) != NULL) {
1641 Py_INCREF(x);
1642 PUSH(x);
1643 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001644 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001645 PyErr_SetString(PyExc_SystemError, "no locals");
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 RETURN_VALUE:
1649 retval = POP();
1650 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001651 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001652
Tim Peters5ca576e2001-06-18 22:08:13 +00001653 case YIELD_VALUE:
1654 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001655 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001656 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001657 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001658
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001659 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001660 w = TOP();
1661 v = SECOND();
1662 u = THIRD();
1663 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001664 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001665 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001666 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001667 Py_DECREF(u);
1668 Py_DECREF(v);
1669 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001670 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001671
Guido van Rossum374a9221991-04-04 10:40:29 +00001672 case POP_BLOCK:
1673 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001674 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001675 while (STACK_LEVEL() > b->b_level) {
1676 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001677 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001678 }
1679 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001680 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001681
Guido van Rossum374a9221991-04-04 10:40:29 +00001682 case END_FINALLY:
1683 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001684 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001685 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001686 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001687 if (why == WHY_RETURN ||
1688 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001689 retval = POP();
1690 }
Brett Cannonbf364092006-03-01 04:25:17 +00001691 else if (PyExceptionClass_Check(v) || PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001692 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001693 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001694 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001695 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001696 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001697 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001698 else if (v != Py_None) {
1699 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001700 "'finally' pops bad exception");
1701 why = WHY_EXCEPTION;
1702 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001703 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001704 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001705
Guido van Rossum374a9221991-04-04 10:40:29 +00001706 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001707 u = TOP();
1708 v = SECOND();
1709 w = THIRD();
1710 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001711 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001712 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001713 Py_DECREF(u);
1714 Py_DECREF(v);
1715 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001716 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001717
Guido van Rossum374a9221991-04-04 10:40:29 +00001718 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001719 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001720 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001721 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001722 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001723 err = PyDict_SetItem(x, w, v);
1724 else
1725 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001726 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001727 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001728 break;
1729 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001730 PyErr_Format(PyExc_SystemError,
1731 "no locals found when storing %s",
1732 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001733 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001734
Guido van Rossum374a9221991-04-04 10:40:29 +00001735 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001736 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001737 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001738 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001739 format_exc_check_arg(PyExc_NameError,
1740 NAME_ERROR_MSG ,w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001741 break;
1742 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001743 PyErr_Format(PyExc_SystemError,
1744 "no locals when deleting %s",
1745 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001746 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001747
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001748 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001749 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001750 v = POP();
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001751 if (PyTuple_CheckExact(v) && PyTuple_GET_SIZE(v) == oparg) {
1752 PyObject **items = ((PyTupleObject *)v)->ob_item;
1753 while (oparg--) {
1754 w = items[oparg];
1755 Py_INCREF(w);
1756 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001757 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001758 Py_DECREF(v);
1759 continue;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001760 } else if (PyList_CheckExact(v) && PyList_GET_SIZE(v) == oparg) {
1761 PyObject **items = ((PyListObject *)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 Hettingerf114a3a2004-03-08 23:25:30 +00001767 } else if (unpack_iterable(v, oparg,
Georg Brandl8a10ea42007-03-21 09:00:55 +00001768 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001769 stack_pointer += oparg;
Georg Brandl8a10ea42007-03-21 09:00:55 +00001770 } else {
1771 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001772 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001773 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001774 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001775 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001776
Guido van Rossum374a9221991-04-04 10:40:29 +00001777 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001778 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001779 v = TOP();
1780 u = SECOND();
1781 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001782 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1783 Py_DECREF(v);
1784 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001785 if (err == 0) continue;
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 DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001789 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001790 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001791 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1792 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001793 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001794 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001795
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001796 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001797 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001798 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001799 err = PyDict_SetItem(f->f_globals, w, v);
1800 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001801 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001802 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001803
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001804 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001805 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001806 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001807 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001808 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001809 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001810
Guido van Rossum374a9221991-04-04 10:40:29 +00001811 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001812 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001813 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001814 PyErr_Format(PyExc_SystemError,
1815 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001816 PyObject_REPR(w));
Guido van Rossum681d79a1995-07-18 14:51:37 +00001817 break;
1818 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001819 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001820 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001821 Py_XINCREF(x);
1822 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001823 else {
1824 x = PyObject_GetItem(v, w);
1825 if (x == NULL && PyErr_Occurred()) {
1826 if (!PyErr_ExceptionMatches(PyExc_KeyError))
1827 break;
1828 PyErr_Clear();
1829 }
1830 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001831 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001832 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001833 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001834 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001835 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001836 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001837 PyExc_NameError,
Paul Prescode68140d2000-08-30 20:25:01 +00001838 NAME_ERROR_MSG ,w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001839 break;
1840 }
1841 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001842 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001843 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001844 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001845 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001846
Guido van Rossum374a9221991-04-04 10:40:29 +00001847 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001848 w = GETITEM(names, oparg);
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001849 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001850 /* Inline the PyDict_GetItem() calls.
1851 WARNING: this is an extreme speed hack.
1852 Do not try this at home. */
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001853 long hash = ((PyStringObject *)w)->ob_shash;
1854 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001855 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00001856 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001857 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00001858 e = d->ma_lookup(d, w, hash);
1859 if (e == NULL) {
1860 x = NULL;
1861 break;
1862 }
1863 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001864 if (x != NULL) {
1865 Py_INCREF(x);
1866 PUSH(x);
1867 continue;
1868 }
1869 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00001870 e = d->ma_lookup(d, w, hash);
1871 if (e == NULL) {
1872 x = NULL;
1873 break;
1874 }
1875 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001876 if (x != NULL) {
1877 Py_INCREF(x);
1878 PUSH(x);
1879 continue;
1880 }
1881 goto load_global_error;
1882 }
1883 }
1884 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001885 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001886 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001887 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001888 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001889 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001890 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001891 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001892 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001893 break;
1894 }
1895 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001896 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001897 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001898 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001899
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001900 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001901 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001902 if (x != NULL) {
1903 SETLOCAL(oparg, NULL);
1904 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001905 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001906 format_exc_check_arg(
1907 PyExc_UnboundLocalError,
1908 UNBOUNDLOCAL_ERROR_MSG,
1909 PyTuple_GetItem(co->co_varnames, oparg)
1910 );
1911 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001912
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001913 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001914 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001915 Py_INCREF(x);
1916 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001917 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001918 break;
1919
1920 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001921 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001922 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001923 if (w != NULL) {
1924 PUSH(w);
1925 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001926 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001927 err = -1;
1928 /* Don't stomp existing exception */
1929 if (PyErr_Occurred())
1930 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00001931 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1932 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001933 oparg);
1934 format_exc_check_arg(
1935 PyExc_UnboundLocalError,
1936 UNBOUNDLOCAL_ERROR_MSG,
1937 v);
1938 } else {
Richard Jonescebbefc2006-05-23 18:28:17 +00001939 v = PyTuple_GET_ITEM(
Raymond Hettinger467a6982004-04-07 11:39:21 +00001940 co->co_freevars,
Richard Jonescebbefc2006-05-23 18:28:17 +00001941 oparg - PyTuple_GET_SIZE(co->co_cellvars));
Raymond Hettinger467a6982004-04-07 11:39:21 +00001942 format_exc_check_arg(
1943 PyExc_NameError,
1944 UNBOUNDFREE_ERROR_MSG,
1945 v);
1946 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001947 break;
1948
1949 case STORE_DEREF:
1950 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001951 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001952 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001953 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001954 continue;
1955
Guido van Rossum374a9221991-04-04 10:40:29 +00001956 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001957 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001958 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001959 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001960 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001961 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001962 }
1963 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001964 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001965 }
1966 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001967
Guido van Rossum374a9221991-04-04 10:40:29 +00001968 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001969 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001970 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001971 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001972 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001973 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001974 }
1975 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001976 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001977 }
1978 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001979
Guido van Rossum374a9221991-04-04 10:40:29 +00001980 case BUILD_MAP:
Guido van Rossumb209a111997-04-29 18:18:01 +00001981 x = PyDict_New();
Guido van Rossum374a9221991-04-04 10:40:29 +00001982 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001983 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001984 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001985
Guido van Rossum374a9221991-04-04 10:40:29 +00001986 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001987 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001988 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001989 x = PyObject_GetAttr(v, w);
1990 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001991 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001992 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001993 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001994
Guido van Rossum374a9221991-04-04 10:40:29 +00001995 case COMPARE_OP:
1996 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001997 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001998 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001999 /* INLINE: cmp(int, int) */
2000 register long a, b;
2001 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002002 a = PyInt_AS_LONG(v);
2003 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002004 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002005 case PyCmp_LT: res = a < b; break;
2006 case PyCmp_LE: res = a <= b; break;
2007 case PyCmp_EQ: res = a == b; break;
2008 case PyCmp_NE: res = a != b; break;
2009 case PyCmp_GT: res = a > b; break;
2010 case PyCmp_GE: res = a >= b; break;
2011 case PyCmp_IS: res = v == w; break;
2012 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002013 default: goto slow_compare;
2014 }
2015 x = res ? Py_True : Py_False;
2016 Py_INCREF(x);
2017 }
2018 else {
2019 slow_compare:
2020 x = cmp_outcome(oparg, v, w);
2021 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002022 Py_DECREF(v);
2023 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002024 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002025 if (x == NULL) break;
2026 PREDICT(JUMP_IF_FALSE);
2027 PREDICT(JUMP_IF_TRUE);
2028 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002029
Guido van Rossum374a9221991-04-04 10:40:29 +00002030 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002031 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002032 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002033 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002034 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002035 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002036 break;
2037 }
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002038 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002039 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002040 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2041 w = PyTuple_Pack(5,
2042 w,
2043 f->f_globals,
2044 f->f_locals == NULL ?
2045 Py_None : f->f_locals,
2046 v,
2047 u);
2048 else
2049 w = PyTuple_Pack(4,
2050 w,
2051 f->f_globals,
2052 f->f_locals == NULL ?
2053 Py_None : f->f_locals,
2054 v);
2055 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002056 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002057 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002058 u = POP();
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002059 x = NULL;
2060 break;
2061 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002062 READ_TIMESTAMP(intr0);
Guido van Rossumb209a111997-04-29 18:18:01 +00002063 x = PyEval_CallObject(x, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002064 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002065 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002066 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002067 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002068 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002069
Thomas Wouters52152252000-08-17 22:55:00 +00002070 case IMPORT_STAR:
2071 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002072 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002073 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002074 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002075 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002076 break;
2077 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002078 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002079 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002080 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002081 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002082 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002083 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002084 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002085
Thomas Wouters52152252000-08-17 22:55:00 +00002086 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002087 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002088 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002089 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002090 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002091 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002092 PUSH(x);
2093 if (x != NULL) continue;
2094 break;
2095
Guido van Rossum374a9221991-04-04 10:40:29 +00002096 case JUMP_FORWARD:
2097 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002098 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002099
Raymond Hettingerf606f872003-03-16 03:11:04 +00002100 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002101 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002102 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002103 if (w == Py_True) {
2104 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002105 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002106 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002107 if (w == Py_False) {
2108 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002109 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002110 }
2111 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002112 if (err > 0)
2113 err = 0;
2114 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002115 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002116 else
2117 break;
2118 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002119
Raymond Hettingerf606f872003-03-16 03:11:04 +00002120 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002121 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002122 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002123 if (w == Py_False) {
2124 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002125 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002126 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002127 if (w == Py_True) {
2128 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002129 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002130 }
2131 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002132 if (err > 0) {
2133 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002134 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002135 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002136 else if (err == 0)
2137 ;
2138 else
2139 break;
2140 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002141
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002142 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002143 case JUMP_ABSOLUTE:
2144 JUMPTO(oparg);
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002145 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002146
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002147 case GET_ITER:
2148 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002149 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002150 x = PyObject_GetIter(v);
2151 Py_DECREF(v);
2152 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002153 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002154 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002155 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002156 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002157 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002158 break;
2159
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002160 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002161 case FOR_ITER:
2162 /* before: [iter]; after: [iter, iter()] *or* [] */
2163 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002164 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002165 if (x != NULL) {
2166 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002167 PREDICT(STORE_FAST);
2168 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002169 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002170 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002171 if (PyErr_Occurred()) {
2172 if (!PyErr_ExceptionMatches(PyExc_StopIteration))
2173 break;
2174 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002175 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002176 /* iterator ended normally */
2177 x = v = POP();
2178 Py_DECREF(v);
2179 JUMPBY(oparg);
2180 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002181
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002182 case BREAK_LOOP:
2183 why = WHY_BREAK;
2184 goto fast_block_end;
2185
2186 case CONTINUE_LOOP:
2187 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002188 if (!retval) {
2189 x = NULL;
2190 break;
2191 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002192 why = WHY_CONTINUE;
2193 goto fast_block_end;
2194
Guido van Rossum374a9221991-04-04 10:40:29 +00002195 case SETUP_LOOP:
2196 case SETUP_EXCEPT:
2197 case SETUP_FINALLY:
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002198 /* NOTE: If you add any new block-setup opcodes that are not try/except/finally
2199 handlers, you may need to update the PyGen_NeedsFinalizing() function. */
2200
Guido van Rossumb209a111997-04-29 18:18:01 +00002201 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002202 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002203 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002204
Guido van Rossumc2e20742006-02-27 22:32:47 +00002205 case WITH_CLEANUP:
2206 {
2207 /* TOP is the context.__exit__ bound method.
2208 Below that are 1-3 values indicating how/why
2209 we entered the finally clause:
2210 - SECOND = None
Guido van Rossumf6694362006-03-10 02:28:35 +00002211 - (SECOND, THIRD) = (WHY_{RETURN,CONTINUE}), retval
Guido van Rossumc2e20742006-02-27 22:32:47 +00002212 - SECOND = WHY_*; no retval below it
2213 - (SECOND, THIRD, FOURTH) = exc_info()
2214 In the last case, we must call
2215 TOP(SECOND, THIRD, FOURTH)
2216 otherwise we must call
2217 TOP(None, None, None)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002218
2219 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002220 *and* the function call returns a 'true' value, we
2221 "zap" this information, to prevent END_FINALLY from
2222 re-raising the exception. (But non-local gotos
2223 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002224 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002225
Guido van Rossumc2e20742006-02-27 22:32:47 +00002226 x = TOP();
2227 u = SECOND();
2228 if (PyInt_Check(u) || u == Py_None) {
2229 u = v = w = Py_None;
2230 }
2231 else {
2232 v = THIRD();
2233 w = FOURTH();
2234 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002235 /* XXX Not the fastest way to call it... */
2236 x = PyObject_CallFunctionObjArgs(x, u, v, w, NULL);
2237 if (x == NULL)
2238 break; /* Go to error exit */
2239 if (u != Py_None && PyObject_IsTrue(x)) {
2240 /* There was an exception and a true return */
2241 Py_DECREF(x);
2242 x = TOP(); /* Again */
2243 STACKADJ(-3);
2244 Py_INCREF(Py_None);
2245 SET_TOP(Py_None);
2246 Py_DECREF(x);
2247 Py_DECREF(u);
2248 Py_DECREF(v);
2249 Py_DECREF(w);
2250 } else {
2251 /* Let END_FINALLY do its thing */
2252 Py_DECREF(x);
2253 x = POP();
2254 Py_DECREF(x);
2255 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002256 break;
2257 }
2258
Guido van Rossumf10570b1995-07-07 22:53:21 +00002259 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002260 {
2261 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002262 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002263 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002264#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002265 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002266#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002267 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002268#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002269 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002270 PUSH(x);
2271 if (x != NULL)
2272 continue;
2273 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002274 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002275
Jeremy Hylton76901512000-03-28 23:49:17 +00002276 case CALL_FUNCTION_VAR:
2277 case CALL_FUNCTION_KW:
2278 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002279 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002280 int na = oparg & 0xff;
2281 int nk = (oparg>>8) & 0xff;
2282 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002283 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002284 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002285 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002286 if (flags & CALL_FLAG_VAR)
2287 n++;
2288 if (flags & CALL_FLAG_KW)
2289 n++;
2290 pfunc = stack_pointer - n - 1;
2291 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002292
Guido van Rossumac7be682001-01-17 15:42:30 +00002293 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002294 && PyMethod_GET_SELF(func) != NULL) {
2295 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002296 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002297 func = PyMethod_GET_FUNCTION(func);
2298 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002299 Py_DECREF(*pfunc);
2300 *pfunc = self;
2301 na++;
2302 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002303 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002304 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002305 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002306 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002307 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002308 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002309 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002310 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002311
Jeremy Hylton76901512000-03-28 23:49:17 +00002312 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002313 w = POP();
2314 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002315 }
2316 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002317 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002318 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002319 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002320 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002321
Guido van Rossum681d79a1995-07-18 14:51:37 +00002322 case MAKE_FUNCTION:
2323 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002324 x = PyFunction_New(v, f->f_globals);
2325 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002326 /* XXX Maybe this should be a separate opcode? */
2327 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002328 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002329 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002330 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002331 x = NULL;
2332 break;
2333 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002334 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002335 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002336 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002337 }
2338 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002339 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002340 }
2341 PUSH(x);
2342 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002343
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002344 case MAKE_CLOSURE:
2345 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002346 v = POP(); /* code object */
2347 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002348 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002349 if (x != NULL) {
2350 v = POP();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002351 err = PyFunction_SetClosure(x, v);
2352 Py_DECREF(v);
2353 }
2354 if (x != NULL && oparg > 0) {
2355 v = PyTuple_New(oparg);
2356 if (v == NULL) {
2357 Py_DECREF(x);
2358 x = NULL;
2359 break;
2360 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002361 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002362 w = POP();
2363 PyTuple_SET_ITEM(v, oparg, w);
2364 }
2365 err = PyFunction_SetDefaults(x, v);
2366 Py_DECREF(v);
2367 }
2368 PUSH(x);
2369 break;
2370 }
2371
Guido van Rossum8861b741996-07-30 16:49:37 +00002372 case BUILD_SLICE:
2373 if (oparg == 3)
2374 w = POP();
2375 else
2376 w = NULL;
2377 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002378 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002379 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002380 Py_DECREF(u);
2381 Py_DECREF(v);
2382 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002383 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002384 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002385 break;
2386
Fred Drakeef8ace32000-08-24 00:32:09 +00002387 case EXTENDED_ARG:
2388 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002389 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002390 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002391
Guido van Rossum374a9221991-04-04 10:40:29 +00002392 default:
2393 fprintf(stderr,
2394 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002395 PyCode_Addr2Line(f->f_code, f->f_lasti),
2396 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002397 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002398 why = WHY_EXCEPTION;
2399 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002400
2401#ifdef CASE_TOO_BIG
2402 }
2403#endif
2404
Guido van Rossum374a9221991-04-04 10:40:29 +00002405 } /* switch */
2406
2407 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002408
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002409 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002410
Guido van Rossum374a9221991-04-04 10:40:29 +00002411 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002412
Guido van Rossum374a9221991-04-04 10:40:29 +00002413 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002414 if (err == 0 && x != NULL) {
2415#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002416 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002417 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002418 fprintf(stderr,
2419 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002420 else {
2421#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002422 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002423 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002424#ifdef CHECKEXC
2425 }
2426#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002427 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002428 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002429 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002430 err = 0;
2431 }
2432
Guido van Rossum374a9221991-04-04 10:40:29 +00002433 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002434
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002435 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002436 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002437 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002438 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002439 why = WHY_EXCEPTION;
2440 }
2441 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002442#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002443 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002444 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002445 if (PyErr_Occurred()) {
Jeremy Hylton904ed862003-11-05 17:29:35 +00002446 char buf[1024];
2447 sprintf(buf, "Stack unwind with exception "
2448 "set and why=%d", why);
2449 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002450 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002451 }
2452#endif
2453
2454 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002455
Guido van Rossum374a9221991-04-04 10:40:29 +00002456 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002457 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002458
Fred Drake8f51f542001-10-04 14:48:42 +00002459 if (tstate->c_tracefunc != NULL)
2460 call_exc_trace(tstate->c_tracefunc,
2461 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002462 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002463
Guido van Rossum374a9221991-04-04 10:40:29 +00002464 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002465
Guido van Rossum374a9221991-04-04 10:40:29 +00002466 if (why == WHY_RERAISE)
2467 why = WHY_EXCEPTION;
2468
2469 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002470
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002471fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002472 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002473 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002474
Tim Peters8a5c3c72004-04-05 19:36:21 +00002475 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002476 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2477 /* For a continue inside a try block,
2478 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002479 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2480 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002481 why = WHY_NOT;
2482 JUMPTO(PyInt_AS_LONG(retval));
2483 Py_DECREF(retval);
2484 break;
2485 }
2486
Guido van Rossum374a9221991-04-04 10:40:29 +00002487 while (STACK_LEVEL() > b->b_level) {
2488 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002489 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002490 }
2491 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2492 why = WHY_NOT;
2493 JUMPTO(b->b_handler);
2494 break;
2495 }
2496 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002497 (b->b_type == SETUP_EXCEPT &&
2498 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002499 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002500 PyObject *exc, *val, *tb;
2501 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002502 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002503 val = Py_None;
2504 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002505 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002506 /* Make the raw exception data
2507 available to the handler,
2508 so a program can emulate the
2509 Python main loop. Don't do
2510 this for 'finally'. */
2511 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002512 PyErr_NormalizeException(
2513 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002514 set_exc_info(tstate,
2515 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002516 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002517 if (tb == NULL) {
2518 Py_INCREF(Py_None);
2519 PUSH(Py_None);
2520 } else
2521 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002522 PUSH(val);
2523 PUSH(exc);
2524 }
2525 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002526 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002527 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002528 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002529 PUSH(v);
2530 }
2531 why = WHY_NOT;
2532 JUMPTO(b->b_handler);
2533 break;
2534 }
2535 } /* unwind stack */
2536
2537 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002538
Guido van Rossum374a9221991-04-04 10:40:29 +00002539 if (why != WHY_NOT)
2540 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002541 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002542
Guido van Rossum374a9221991-04-04 10:40:29 +00002543 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002544
Tim Peters8a5c3c72004-04-05 19:36:21 +00002545 assert(why != WHY_YIELD);
2546 /* Pop remaining stack entries. */
2547 while (!EMPTY()) {
2548 v = POP();
2549 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002550 }
2551
Tim Peters8a5c3c72004-04-05 19:36:21 +00002552 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002553 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002554
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002555fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002556 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002557 if (tstate->c_tracefunc) {
2558 if (why == WHY_RETURN || why == WHY_YIELD) {
2559 if (call_trace(tstate->c_tracefunc,
2560 tstate->c_traceobj, f,
2561 PyTrace_RETURN, retval)) {
2562 Py_XDECREF(retval);
2563 retval = NULL;
2564 why = WHY_EXCEPTION;
2565 }
2566 }
2567 else if (why == WHY_EXCEPTION) {
2568 call_trace_protected(tstate->c_tracefunc,
2569 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002570 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002571 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002572 }
Fred Drake8f51f542001-10-04 14:48:42 +00002573 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002574 if (why == WHY_EXCEPTION)
2575 call_trace_protected(tstate->c_profilefunc,
2576 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002577 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002578 else if (call_trace(tstate->c_profilefunc,
2579 tstate->c_profileobj, f,
2580 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002581 Py_XDECREF(retval);
2582 retval = NULL;
2583 why = WHY_EXCEPTION;
2584 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002585 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002586 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002587
Tim Peters7df5e7f2006-05-26 23:14:37 +00002588 if (tstate->frame->f_exc_type != NULL)
2589 reset_exc_info(tstate);
2590 else {
2591 assert(tstate->frame->f_exc_value == NULL);
2592 assert(tstate->frame->f_exc_traceback == NULL);
2593 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002594
Tim Peters5ca576e2001-06-18 22:08:13 +00002595 /* pop frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +00002596 exit_eval_frame:
2597 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002598 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002599
Guido van Rossum96a42c81992-01-12 02:29:51 +00002600 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002601}
2602
Guido van Rossumc2e20742006-02-27 22:32:47 +00002603/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002604 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002605 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002606
Tim Peters6d6c1a32001-08-02 04:15:00 +00002607PyObject *
2608PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002609 PyObject **args, int argcount, PyObject **kws, int kwcount,
2610 PyObject **defs, int defcount, PyObject *closure)
2611{
2612 register PyFrameObject *f;
2613 register PyObject *retval = NULL;
2614 register PyObject **fastlocals, **freevars;
2615 PyThreadState *tstate = PyThreadState_GET();
2616 PyObject *x, *u;
2617
2618 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002619 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002620 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002621 return NULL;
2622 }
2623
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002624 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002625 assert(globals != NULL);
2626 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002627 if (f == NULL)
2628 return NULL;
2629
2630 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002631 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002632
2633 if (co->co_argcount > 0 ||
2634 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2635 int i;
2636 int n = argcount;
2637 PyObject *kwdict = NULL;
2638 if (co->co_flags & CO_VARKEYWORDS) {
2639 kwdict = PyDict_New();
2640 if (kwdict == NULL)
2641 goto fail;
2642 i = co->co_argcount;
2643 if (co->co_flags & CO_VARARGS)
2644 i++;
2645 SETLOCAL(i, kwdict);
2646 }
2647 if (argcount > co->co_argcount) {
2648 if (!(co->co_flags & CO_VARARGS)) {
2649 PyErr_Format(PyExc_TypeError,
2650 "%.200s() takes %s %d "
2651 "%sargument%s (%d given)",
2652 PyString_AsString(co->co_name),
2653 defcount ? "at most" : "exactly",
2654 co->co_argcount,
2655 kwcount ? "non-keyword " : "",
2656 co->co_argcount == 1 ? "" : "s",
2657 argcount);
2658 goto fail;
2659 }
2660 n = co->co_argcount;
2661 }
2662 for (i = 0; i < n; i++) {
2663 x = args[i];
2664 Py_INCREF(x);
2665 SETLOCAL(i, x);
2666 }
2667 if (co->co_flags & CO_VARARGS) {
2668 u = PyTuple_New(argcount - n);
2669 if (u == NULL)
2670 goto fail;
2671 SETLOCAL(co->co_argcount, u);
2672 for (i = n; i < argcount; i++) {
2673 x = args[i];
2674 Py_INCREF(x);
2675 PyTuple_SET_ITEM(u, i-n, x);
2676 }
2677 }
2678 for (i = 0; i < kwcount; i++) {
2679 PyObject *keyword = kws[2*i];
2680 PyObject *value = kws[2*i + 1];
2681 int j;
2682 if (keyword == NULL || !PyString_Check(keyword)) {
2683 PyErr_Format(PyExc_TypeError,
2684 "%.200s() keywords must be strings",
2685 PyString_AsString(co->co_name));
2686 goto fail;
2687 }
2688 /* XXX slow -- speed up using dictionary? */
2689 for (j = 0; j < co->co_argcount; j++) {
2690 PyObject *nm = PyTuple_GET_ITEM(
2691 co->co_varnames, j);
2692 int cmp = PyObject_RichCompareBool(
2693 keyword, nm, Py_EQ);
2694 if (cmp > 0)
2695 break;
2696 else if (cmp < 0)
2697 goto fail;
2698 }
2699 /* Check errors from Compare */
2700 if (PyErr_Occurred())
2701 goto fail;
2702 if (j >= co->co_argcount) {
2703 if (kwdict == NULL) {
2704 PyErr_Format(PyExc_TypeError,
2705 "%.200s() got an unexpected "
2706 "keyword argument '%.400s'",
2707 PyString_AsString(co->co_name),
2708 PyString_AsString(keyword));
2709 goto fail;
2710 }
2711 PyDict_SetItem(kwdict, keyword, value);
2712 }
2713 else {
2714 if (GETLOCAL(j) != NULL) {
2715 PyErr_Format(PyExc_TypeError,
2716 "%.200s() got multiple "
2717 "values for keyword "
2718 "argument '%.400s'",
2719 PyString_AsString(co->co_name),
2720 PyString_AsString(keyword));
2721 goto fail;
2722 }
2723 Py_INCREF(value);
2724 SETLOCAL(j, value);
2725 }
2726 }
2727 if (argcount < co->co_argcount) {
2728 int m = co->co_argcount - defcount;
2729 for (i = argcount; i < m; i++) {
2730 if (GETLOCAL(i) == NULL) {
2731 PyErr_Format(PyExc_TypeError,
2732 "%.200s() takes %s %d "
2733 "%sargument%s (%d given)",
2734 PyString_AsString(co->co_name),
2735 ((co->co_flags & CO_VARARGS) ||
2736 defcount) ? "at least"
2737 : "exactly",
2738 m, kwcount ? "non-keyword " : "",
2739 m == 1 ? "" : "s", i);
2740 goto fail;
2741 }
2742 }
2743 if (n > m)
2744 i = n - m;
2745 else
2746 i = 0;
2747 for (; i < defcount; i++) {
2748 if (GETLOCAL(m+i) == NULL) {
2749 PyObject *def = defs[i];
2750 Py_INCREF(def);
2751 SETLOCAL(m+i, def);
2752 }
2753 }
2754 }
2755 }
2756 else {
2757 if (argcount > 0 || kwcount > 0) {
2758 PyErr_Format(PyExc_TypeError,
2759 "%.200s() takes no arguments (%d given)",
2760 PyString_AsString(co->co_name),
2761 argcount + kwcount);
2762 goto fail;
2763 }
2764 }
2765 /* Allocate and initialize storage for cell vars, and copy free
2766 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00002767 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00002768 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002769 char *cellname, *argname;
2770 PyObject *c;
2771
2772 nargs = co->co_argcount;
2773 if (co->co_flags & CO_VARARGS)
2774 nargs++;
2775 if (co->co_flags & CO_VARKEYWORDS)
2776 nargs++;
2777
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002778 /* Initialize each cell var, taking into account
2779 cell vars that are initialized from arguments.
2780
2781 Should arrange for the compiler to put cellvars
2782 that are arguments at the beginning of the cellvars
2783 list so that we can march over it more efficiently?
2784 */
Richard Jonescebbefc2006-05-23 18:28:17 +00002785 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002786 cellname = PyString_AS_STRING(
2787 PyTuple_GET_ITEM(co->co_cellvars, i));
2788 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002789 for (j = 0; j < nargs; j++) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002790 argname = PyString_AS_STRING(
2791 PyTuple_GET_ITEM(co->co_varnames, j));
2792 if (strcmp(cellname, argname) == 0) {
2793 c = PyCell_New(GETLOCAL(j));
2794 if (c == NULL)
2795 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002796 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002797 found = 1;
2798 break;
2799 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002800 }
2801 if (found == 0) {
2802 c = PyCell_New(NULL);
2803 if (c == NULL)
2804 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002805 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002806 }
2807 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002808 }
Richard Jonescebbefc2006-05-23 18:28:17 +00002809 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002810 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00002811 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002812 PyObject *o = PyTuple_GET_ITEM(closure, i);
2813 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00002814 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002815 }
2816 }
2817
Tim Peters5ca576e2001-06-18 22:08:13 +00002818 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002819 /* Don't need to keep the reference to f_back, it will be set
2820 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002821 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002822 f->f_back = NULL;
2823
Jeremy Hylton985eba52003-02-05 23:13:00 +00002824 PCALL(PCALL_GENERATOR);
2825
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002826 /* Create a new generator that owns the ready to run frame
2827 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002828 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002829 }
2830
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00002831 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002832
2833 fail: /* Jump here from prelude on failure */
2834
Tim Petersb13680b2001-11-27 23:29:29 +00002835 /* decref'ing the frame can cause __del__ methods to get invoked,
2836 which can call back into Python. While we're done with the
2837 current Python frame (f), the associated C stack is still in use,
2838 so recursion_depth must be boosted for the duration.
2839 */
2840 assert(tstate != NULL);
2841 ++tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002842 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002843 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002844 return retval;
2845}
2846
2847
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002848/* Implementation notes for set_exc_info() and reset_exc_info():
2849
2850- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2851 'exc_traceback'. These always travel together.
2852
2853- tstate->curexc_ZZZ is the "hot" exception that is set by
2854 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2855
2856- Once an exception is caught by an except clause, it is transferred
2857 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2858 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00002859 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002860
2861- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2862
2863 Long ago, when none of this existed, there were just a few globals:
2864 one set corresponding to the "hot" exception, and one set
2865 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2866 globals; they were simply stored as sys.exc_ZZZ. For backwards
2867 compatibility, they still are!) The problem was that in code like
2868 this:
2869
2870 try:
2871 "something that may fail"
2872 except "some exception":
2873 "do something else first"
2874 "print the exception from sys.exc_ZZZ."
2875
2876 if "do something else first" invoked something that raised and caught
2877 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2878 cause of subtle bugs. I fixed this by changing the semantics as
2879 follows:
2880
2881 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2882 *in that frame*.
2883
2884 - But initially, and as long as no exception is caught in a given
2885 frame, sys.exc_ZZZ will hold the last exception caught in the
2886 previous frame (or the frame before that, etc.).
2887
2888 The first bullet fixed the bug in the above example. The second
2889 bullet was for backwards compatibility: it was (and is) common to
2890 have a function that is called when an exception is caught, and to
2891 have that function access the caught exception via sys.exc_ZZZ.
2892 (Example: traceback.print_exc()).
2893
2894 At the same time I fixed the problem that sys.exc_ZZZ weren't
2895 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2896 but that's really a separate improvement.
2897
2898 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2899 variables to what they were before the current frame was called. The
2900 set_exc_info() function saves them on the frame so that
2901 reset_exc_info() can restore them. The invariant is that
2902 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2903 exception (where "catching" an exception applies only to successful
2904 except clauses); and if the current frame ever caught an exception,
2905 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2906 at the start of the current frame.
2907
2908*/
2909
Fredrik Lundh7a830892006-05-27 10:39:48 +00002910static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002911set_exc_info(PyThreadState *tstate,
2912 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002913{
Tim Peters7df5e7f2006-05-26 23:14:37 +00002914 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002915 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002916
Tim Peters7df5e7f2006-05-26 23:14:37 +00002917 assert(type != NULL);
2918 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002919 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002920 assert(frame->f_exc_value == NULL);
2921 assert(frame->f_exc_traceback == NULL);
2922 /* This frame didn't catch an exception before. */
2923 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002924 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002925 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002926 Py_INCREF(Py_None);
2927 tstate->exc_type = Py_None;
2928 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002929 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002930 Py_XINCREF(tstate->exc_value);
2931 Py_XINCREF(tstate->exc_traceback);
2932 frame->f_exc_type = tstate->exc_type;
2933 frame->f_exc_value = tstate->exc_value;
2934 frame->f_exc_traceback = tstate->exc_traceback;
2935 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002936 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002937 tmp_type = tstate->exc_type;
2938 tmp_value = tstate->exc_value;
2939 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002940 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002941 Py_XINCREF(value);
2942 Py_XINCREF(tb);
2943 tstate->exc_type = type;
2944 tstate->exc_value = value;
2945 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002946 Py_XDECREF(tmp_type);
2947 Py_XDECREF(tmp_value);
2948 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002949 /* For b/w compatibility */
2950 PySys_SetObject("exc_type", type);
2951 PySys_SetObject("exc_value", value);
2952 PySys_SetObject("exc_traceback", tb);
2953}
2954
Fredrik Lundh7a830892006-05-27 10:39:48 +00002955static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002956reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002957{
2958 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002959 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002960
2961 /* It's a precondition that the thread state's frame caught an
2962 * exception -- verify in a debug build.
2963 */
2964 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002965 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002966 assert(frame != NULL);
2967 assert(frame->f_exc_type != NULL);
2968
2969 /* Copy the frame's exception info back to the thread state. */
2970 tmp_type = tstate->exc_type;
2971 tmp_value = tstate->exc_value;
2972 tmp_tb = tstate->exc_traceback;
2973 Py_INCREF(frame->f_exc_type);
2974 Py_XINCREF(frame->f_exc_value);
2975 Py_XINCREF(frame->f_exc_traceback);
2976 tstate->exc_type = frame->f_exc_type;
2977 tstate->exc_value = frame->f_exc_value;
2978 tstate->exc_traceback = frame->f_exc_traceback;
2979 Py_XDECREF(tmp_type);
2980 Py_XDECREF(tmp_value);
2981 Py_XDECREF(tmp_tb);
2982
2983 /* For b/w compatibility */
2984 PySys_SetObject("exc_type", frame->f_exc_type);
2985 PySys_SetObject("exc_value", frame->f_exc_value);
2986 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
2987
2988 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002989 tmp_type = frame->f_exc_type;
2990 tmp_value = frame->f_exc_value;
2991 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00002992 frame->f_exc_type = NULL;
2993 frame->f_exc_value = NULL;
2994 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002995 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002996 Py_XDECREF(tmp_value);
2997 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002998}
2999
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003000/* Logic for the raise statement (too complicated for inlining).
3001 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003002static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003003do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003004{
Guido van Rossumd295f121998-04-09 21:39:57 +00003005 if (type == NULL) {
3006 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003007 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003008 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3009 value = tstate->exc_value;
3010 tb = tstate->exc_traceback;
3011 Py_XINCREF(type);
3012 Py_XINCREF(value);
3013 Py_XINCREF(tb);
3014 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003015
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003016 /* We support the following forms of raise:
3017 raise <class>, <classinstance>
3018 raise <class>, <argument tuple>
3019 raise <class>, None
3020 raise <class>, <argument>
3021 raise <classinstance>, None
3022 raise <string>, <object>
3023 raise <string>, None
3024
3025 An omitted second argument is the same as None.
3026
3027 In addition, raise <tuple>, <anything> is the same as
3028 raising the tuple's first item (and it better have one!);
3029 this rule is applied recursively.
3030
3031 Finally, an optional third argument can be supplied, which
3032 gives the traceback to be substituted (useful when
3033 re-raising an exception after examining it). */
3034
3035 /* First, check the traceback argument, replacing None with
3036 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003037 if (tb == Py_None) {
3038 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003039 tb = NULL;
3040 }
3041 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003042 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003043 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003044 goto raise_error;
3045 }
3046
3047 /* Next, replace a missing value with None */
3048 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003049 value = Py_None;
3050 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003051 }
3052
3053 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003054 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3055 PyObject *tmp = type;
3056 type = PyTuple_GET_ITEM(type, 0);
3057 Py_INCREF(type);
3058 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003059 }
3060
Brett Cannona7446e32006-02-27 23:39:10 +00003061 if (PyString_CheckExact(type)) {
Tim Petersafb2c802002-04-18 18:06:20 +00003062 /* Raising builtin string is deprecated but still allowed --
3063 * do nothing. Raising an instance of a new-style str
3064 * subclass is right out. */
Brett Cannonbf364092006-03-01 04:25:17 +00003065 if (PyErr_Warn(PyExc_DeprecationWarning,
Brett Cannona7446e32006-02-27 23:39:10 +00003066 "raising a string exception is deprecated"))
3067 goto raise_error;
3068 }
Brett Cannonbf364092006-03-01 04:25:17 +00003069 else if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003070 PyErr_NormalizeException(&type, &value, &tb);
3071
Brett Cannonbf364092006-03-01 04:25:17 +00003072 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003073 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003074 if (value != Py_None) {
3075 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003076 "instance exception may not have a separate value");
3077 goto raise_error;
3078 }
3079 else {
3080 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003081 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003082 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003083 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003084 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003085 }
3086 }
3087 else {
3088 /* Not something you can raise. You get an exception
3089 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003090 PyErr_Format(PyExc_TypeError,
Neal Norwitz37aa0662003-01-10 15:31:15 +00003091 "exceptions must be classes, instances, or "
3092 "strings (deprecated), not %s",
3093 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003094 goto raise_error;
3095 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003096 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003097 if (tb == NULL)
3098 return WHY_EXCEPTION;
3099 else
3100 return WHY_RERAISE;
3101 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003102 Py_XDECREF(value);
3103 Py_XDECREF(type);
3104 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003105 return WHY_EXCEPTION;
3106}
3107
Tim Petersd6d010b2001-06-21 02:49:55 +00003108/* Iterate v argcnt times and store the results on the stack (via decreasing
3109 sp). Return 1 for success, 0 if error. */
3110
Fredrik Lundh7a830892006-05-27 10:39:48 +00003111static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003112unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003113{
Tim Petersd6d010b2001-06-21 02:49:55 +00003114 int i = 0;
3115 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003116 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003117
Tim Petersd6d010b2001-06-21 02:49:55 +00003118 assert(v != NULL);
3119
3120 it = PyObject_GetIter(v);
3121 if (it == NULL)
3122 goto Error;
3123
3124 for (; i < argcnt; i++) {
3125 w = PyIter_Next(it);
3126 if (w == NULL) {
3127 /* Iterator done, via error or exhaustion. */
3128 if (!PyErr_Occurred()) {
3129 PyErr_Format(PyExc_ValueError,
3130 "need more than %d value%s to unpack",
3131 i, i == 1 ? "" : "s");
3132 }
3133 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003134 }
3135 *--sp = w;
3136 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003137
3138 /* We better have exhausted the iterator now. */
3139 w = PyIter_Next(it);
3140 if (w == NULL) {
3141 if (PyErr_Occurred())
3142 goto Error;
3143 Py_DECREF(it);
3144 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003145 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003146 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003147 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003148 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003149Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003150 for (; i > 0; i--, sp++)
3151 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003152 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003153 return 0;
3154}
3155
3156
Guido van Rossum96a42c81992-01-12 02:29:51 +00003157#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003158static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003159prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003160{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003161 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003162 if (PyObject_Print(v, stdout, 0) != 0)
3163 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003164 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003165 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003166}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003167#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003168
Fredrik Lundh7a830892006-05-27 10:39:48 +00003169static void
Fred Drake5755ce62001-06-27 19:19:46 +00003170call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003171{
Guido van Rossumb209a111997-04-29 18:18:01 +00003172 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003173 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003174 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003175 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003176 value = Py_None;
3177 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003178 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003179 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003180 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003181 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003182 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003183 }
Fred Drake5755ce62001-06-27 19:19:46 +00003184 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003185 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003186 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003187 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003188 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003189 Py_XDECREF(type);
3190 Py_XDECREF(value);
3191 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003192 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003193}
3194
Fredrik Lundh7a830892006-05-27 10:39:48 +00003195static void
Fred Drake4ec5d562001-10-04 19:26:43 +00003196call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003197 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003198{
3199 PyObject *type, *value, *traceback;
3200 int err;
3201 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003202 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003203 if (err == 0)
3204 PyErr_Restore(type, value, traceback);
3205 else {
3206 Py_XDECREF(type);
3207 Py_XDECREF(value);
3208 Py_XDECREF(traceback);
3209 }
3210}
3211
Fredrik Lundh7a830892006-05-27 10:39:48 +00003212static int
Fred Drake5755ce62001-06-27 19:19:46 +00003213call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3214 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003215{
Fred Drake5755ce62001-06-27 19:19:46 +00003216 register PyThreadState *tstate = frame->f_tstate;
3217 int result;
3218 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003219 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003220 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003221 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003222 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003223 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3224 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003225 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003226 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003227}
3228
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003229PyObject *
3230_PyEval_CallTracing(PyObject *func, PyObject *args)
3231{
3232 PyFrameObject *frame = PyEval_GetFrame();
3233 PyThreadState *tstate = frame->f_tstate;
3234 int save_tracing = tstate->tracing;
3235 int save_use_tracing = tstate->use_tracing;
3236 PyObject *result;
3237
3238 tstate->tracing = 0;
3239 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3240 || (tstate->c_profilefunc != NULL));
3241 result = PyObject_Call(func, args, NULL);
3242 tstate->tracing = save_tracing;
3243 tstate->use_tracing = save_use_tracing;
3244 return result;
3245}
3246
Fredrik Lundh7a830892006-05-27 10:39:48 +00003247static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003248maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003249 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3250 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003251{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003252 int result = 0;
3253
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003254 /* If the last instruction executed isn't in the current
3255 instruction window, reset the window. If the last
3256 instruction happens to fall at the start of a line or if it
3257 represents a jump backwards, call the trace function.
3258 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003259 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003260 int line;
3261 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003262
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003263 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3264 &bounds);
3265 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003266 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003267 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003268 PyTrace_LINE, Py_None);
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003269 }
3270 *instr_lb = bounds.ap_lower;
3271 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003272 }
Armin Rigobf57a142004-03-22 19:24:58 +00003273 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003274 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003275 }
3276 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003277 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003278}
3279
Fred Drake5755ce62001-06-27 19:19:46 +00003280void
3281PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003282{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003283 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003284 PyObject *temp = tstate->c_profileobj;
3285 Py_XINCREF(arg);
3286 tstate->c_profilefunc = NULL;
3287 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003288 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003289 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003290 Py_XDECREF(temp);
3291 tstate->c_profilefunc = func;
3292 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003293 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003294 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003295}
3296
3297void
3298PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3299{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003300 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003301 PyObject *temp = tstate->c_traceobj;
3302 Py_XINCREF(arg);
3303 tstate->c_tracefunc = NULL;
3304 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003305 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003306 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003307 Py_XDECREF(temp);
3308 tstate->c_tracefunc = func;
3309 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003310 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003311 tstate->use_tracing = ((func != NULL)
3312 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003313}
3314
Guido van Rossumb209a111997-04-29 18:18:01 +00003315PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003316PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003317{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003318 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003319 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003320 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003321 else
3322 return current_frame->f_builtins;
3323}
3324
Guido van Rossumb209a111997-04-29 18:18:01 +00003325PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003326PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003327{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003328 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003329 if (current_frame == NULL)
3330 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003331 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003332 return current_frame->f_locals;
3333}
3334
Guido van Rossumb209a111997-04-29 18:18:01 +00003335PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003336PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003337{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003338 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003339 if (current_frame == NULL)
3340 return NULL;
3341 else
3342 return current_frame->f_globals;
3343}
3344
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003345PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003346PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003347{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003348 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003349 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003350}
3351
Guido van Rossum6135a871995-01-09 17:53:26 +00003352int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003353PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003354{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003355 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003356 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003357}
3358
Guido van Rossumbe270261997-05-22 22:26:18 +00003359int
Tim Peters5ba58662001-07-16 02:29:45 +00003360PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003361{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003362 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003363 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003364
3365 if (current_frame != NULL) {
3366 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003367 const int compilerflags = codeflags & PyCF_MASK;
3368 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003369 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003370 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003371 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003372#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003373 if (codeflags & CO_GENERATOR_ALLOWED) {
3374 result = 1;
3375 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3376 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003377#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003378 }
3379 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003380}
3381
3382int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003383Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003384{
Guido van Rossumb209a111997-04-29 18:18:01 +00003385 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003386 if (f == NULL)
3387 return 0;
3388 if (!PyFile_SoftSpace(f, 0))
3389 return 0;
3390 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003391}
3392
Guido van Rossum3f5da241990-12-20 15:06:42 +00003393
Guido van Rossum681d79a1995-07-18 14:51:37 +00003394/* External interface to call any callable object.
3395 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003396
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003397#undef PyEval_CallObject
3398/* for backward compatibility: export this interface */
3399
Guido van Rossumb209a111997-04-29 18:18:01 +00003400PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003401PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003402{
Guido van Rossumb209a111997-04-29 18:18:01 +00003403 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003404}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003405#define PyEval_CallObject(func,arg) \
3406 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003407
Guido van Rossumb209a111997-04-29 18:18:01 +00003408PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003409PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003410{
Jeremy Hylton52820442001-01-03 23:52:36 +00003411 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003412
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003413 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003414 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003415 if (arg == NULL)
3416 return NULL;
3417 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003418 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003419 PyErr_SetString(PyExc_TypeError,
3420 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003421 return NULL;
3422 }
3423 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003424 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003425
Guido van Rossumb209a111997-04-29 18:18:01 +00003426 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003427 PyErr_SetString(PyExc_TypeError,
3428 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003429 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003430 return NULL;
3431 }
3432
Tim Peters6d6c1a32001-08-02 04:15:00 +00003433 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003434 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003435 return result;
3436}
3437
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003438const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003439PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003440{
3441 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003442 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003443 else if (PyFunction_Check(func))
3444 return PyString_AsString(((PyFunctionObject*)func)->func_name);
3445 else if (PyCFunction_Check(func))
3446 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3447 else if (PyClass_Check(func))
3448 return PyString_AsString(((PyClassObject*)func)->cl_name);
3449 else if (PyInstance_Check(func)) {
3450 return PyString_AsString(
3451 ((PyInstanceObject*)func)->in_class->cl_name);
3452 } else {
3453 return func->ob_type->tp_name;
3454 }
3455}
3456
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003457const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003458PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003459{
3460 if (PyMethod_Check(func))
3461 return "()";
3462 else if (PyFunction_Check(func))
3463 return "()";
3464 else if (PyCFunction_Check(func))
3465 return "()";
3466 else if (PyClass_Check(func))
3467 return " constructor";
3468 else if (PyInstance_Check(func)) {
3469 return " instance";
3470 } else {
3471 return " object";
3472 }
3473}
3474
Fredrik Lundh7a830892006-05-27 10:39:48 +00003475static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003476err_args(PyObject *func, int flags, int nargs)
3477{
3478 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003479 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003480 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003481 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003482 nargs);
3483 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003484 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003485 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003486 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003487 nargs);
3488}
3489
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003490#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003491if (tstate->use_tracing && tstate->c_profilefunc) { \
3492 if (call_trace(tstate->c_profilefunc, \
3493 tstate->c_profileobj, \
3494 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003495 func)) { \
3496 x = NULL; \
3497 } \
3498 else { \
3499 x = call; \
3500 if (tstate->c_profilefunc != NULL) { \
3501 if (x == NULL) { \
3502 call_trace_protected(tstate->c_profilefunc, \
3503 tstate->c_profileobj, \
3504 tstate->frame, PyTrace_C_EXCEPTION, \
3505 func); \
3506 /* XXX should pass (type, value, tb) */ \
3507 } else { \
3508 if (call_trace(tstate->c_profilefunc, \
3509 tstate->c_profileobj, \
3510 tstate->frame, PyTrace_C_RETURN, \
3511 func)) { \
3512 Py_DECREF(x); \
3513 x = NULL; \
3514 } \
3515 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003516 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003517 } \
3518} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003519 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003520 }
3521
Fredrik Lundh7a830892006-05-27 10:39:48 +00003522static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003523call_function(PyObject ***pp_stack, int oparg
3524#ifdef WITH_TSC
3525 , uint64* pintr0, uint64* pintr1
3526#endif
3527 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003528{
3529 int na = oparg & 0xff;
3530 int nk = (oparg>>8) & 0xff;
3531 int n = na + 2 * nk;
3532 PyObject **pfunc = (*pp_stack) - n - 1;
3533 PyObject *func = *pfunc;
3534 PyObject *x, *w;
3535
Jeremy Hylton985eba52003-02-05 23:13:00 +00003536 /* Always dispatch PyCFunction first, because these are
3537 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003538 */
3539 if (PyCFunction_Check(func) && nk == 0) {
3540 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003541 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003542
3543 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003544 if (flags & (METH_NOARGS | METH_O)) {
3545 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3546 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003547 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003548 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003549 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003550 else if (flags & METH_O && na == 1) {
3551 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003552 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003553 Py_DECREF(arg);
3554 }
3555 else {
3556 err_args(func, flags, na);
3557 x = NULL;
3558 }
3559 }
3560 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003561 PyObject *callargs;
3562 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003563 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003564 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003565 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003566 Py_XDECREF(callargs);
3567 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003568 } else {
3569 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3570 /* optimize access to bound methods */
3571 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003572 PCALL(PCALL_METHOD);
3573 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003574 Py_INCREF(self);
3575 func = PyMethod_GET_FUNCTION(func);
3576 Py_INCREF(func);
3577 Py_DECREF(*pfunc);
3578 *pfunc = self;
3579 na++;
3580 n++;
3581 } else
3582 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003583 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003584 if (PyFunction_Check(func))
3585 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003586 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003587 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003588 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003589 Py_DECREF(func);
3590 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003591
Armin Rigod34fa522006-03-28 19:10:40 +00003592 /* Clear the stack of the function object. Also removes
3593 the arguments in case they weren't consumed already
3594 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003595 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003596 while ((*pp_stack) > pfunc) {
3597 w = EXT_POP(*pp_stack);
3598 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003599 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003600 }
3601 return x;
3602}
3603
Jeremy Hylton192690e2002-08-16 18:36:11 +00003604/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003605 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003606 For the simplest case -- a function that takes only positional
3607 arguments and is called with only positional arguments -- it
3608 inlines the most primitive frame setup code from
3609 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3610 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003611*/
3612
Fredrik Lundh7a830892006-05-27 10:39:48 +00003613static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003614fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003615{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003616 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003617 PyObject *globals = PyFunction_GET_GLOBALS(func);
3618 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3619 PyObject **d = NULL;
3620 int nd = 0;
3621
Jeremy Hylton985eba52003-02-05 23:13:00 +00003622 PCALL(PCALL_FUNCTION);
3623 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003624 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003625 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3626 PyFrameObject *f;
3627 PyObject *retval = NULL;
3628 PyThreadState *tstate = PyThreadState_GET();
3629 PyObject **fastlocals, **stack;
3630 int i;
3631
3632 PCALL(PCALL_FASTER_FUNCTION);
3633 assert(globals != NULL);
3634 /* XXX Perhaps we should create a specialized
3635 PyFrame_New() that doesn't take locals, but does
3636 take builtins without sanity checking them.
3637 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003638 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003639 f = PyFrame_New(tstate, co, globals, NULL);
3640 if (f == NULL)
3641 return NULL;
3642
3643 fastlocals = f->f_localsplus;
3644 stack = (*pp_stack) - n;
3645
3646 for (i = 0; i < n; i++) {
3647 Py_INCREF(*stack);
3648 fastlocals[i] = *stack++;
3649 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003650 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003651 ++tstate->recursion_depth;
3652 Py_DECREF(f);
3653 --tstate->recursion_depth;
3654 return retval;
3655 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003656 if (argdefs != NULL) {
3657 d = &PyTuple_GET_ITEM(argdefs, 0);
3658 nd = ((PyTupleObject *)argdefs)->ob_size;
3659 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003660 return PyEval_EvalCodeEx(co, globals,
3661 (PyObject *)NULL, (*pp_stack)-n, na,
3662 (*pp_stack)-2*nk, nk, d, nd,
3663 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003664}
3665
Fredrik Lundh7a830892006-05-27 10:39:48 +00003666static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003667update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3668 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003669{
3670 PyObject *kwdict = NULL;
3671 if (orig_kwdict == NULL)
3672 kwdict = PyDict_New();
3673 else {
3674 kwdict = PyDict_Copy(orig_kwdict);
3675 Py_DECREF(orig_kwdict);
3676 }
3677 if (kwdict == NULL)
3678 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003679 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003680 int err;
3681 PyObject *value = EXT_POP(*pp_stack);
3682 PyObject *key = EXT_POP(*pp_stack);
3683 if (PyDict_GetItem(kwdict, key) != NULL) {
Guido van Rossumac7be682001-01-17 15:42:30 +00003684 PyErr_Format(PyExc_TypeError,
Ka-Ping Yee20579702001-01-15 22:14:16 +00003685 "%.200s%s got multiple values "
Jeremy Hylton512a2372001-04-11 13:52:29 +00003686 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003687 PyEval_GetFuncName(func),
3688 PyEval_GetFuncDesc(func),
Jeremy Hylton512a2372001-04-11 13:52:29 +00003689 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003690 Py_DECREF(key);
3691 Py_DECREF(value);
3692 Py_DECREF(kwdict);
3693 return NULL;
3694 }
3695 err = PyDict_SetItem(kwdict, key, value);
3696 Py_DECREF(key);
3697 Py_DECREF(value);
3698 if (err) {
3699 Py_DECREF(kwdict);
3700 return NULL;
3701 }
3702 }
3703 return kwdict;
3704}
3705
Fredrik Lundh7a830892006-05-27 10:39:48 +00003706static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003707update_star_args(int nstack, int nstar, PyObject *stararg,
3708 PyObject ***pp_stack)
3709{
3710 PyObject *callargs, *w;
3711
3712 callargs = PyTuple_New(nstack + nstar);
3713 if (callargs == NULL) {
3714 return NULL;
3715 }
3716 if (nstar) {
3717 int i;
3718 for (i = 0; i < nstar; i++) {
3719 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3720 Py_INCREF(a);
3721 PyTuple_SET_ITEM(callargs, nstack + i, a);
3722 }
3723 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003724 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003725 w = EXT_POP(*pp_stack);
3726 PyTuple_SET_ITEM(callargs, nstack, w);
3727 }
3728 return callargs;
3729}
3730
Fredrik Lundh7a830892006-05-27 10:39:48 +00003731static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003732load_args(PyObject ***pp_stack, int na)
3733{
3734 PyObject *args = PyTuple_New(na);
3735 PyObject *w;
3736
3737 if (args == NULL)
3738 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003739 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003740 w = EXT_POP(*pp_stack);
3741 PyTuple_SET_ITEM(args, na, w);
3742 }
3743 return args;
3744}
3745
Fredrik Lundh7a830892006-05-27 10:39:48 +00003746static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003747do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3748{
3749 PyObject *callargs = NULL;
3750 PyObject *kwdict = NULL;
3751 PyObject *result = NULL;
3752
3753 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003754 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003755 if (kwdict == NULL)
3756 goto call_fail;
3757 }
3758 callargs = load_args(pp_stack, na);
3759 if (callargs == NULL)
3760 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003761#ifdef CALL_PROFILE
3762 /* At this point, we have to look at the type of func to
3763 update the call stats properly. Do it here so as to avoid
3764 exposing the call stats machinery outside ceval.c
3765 */
3766 if (PyFunction_Check(func))
3767 PCALL(PCALL_FUNCTION);
3768 else if (PyMethod_Check(func))
3769 PCALL(PCALL_METHOD);
3770 else if (PyType_Check(func))
3771 PCALL(PCALL_TYPE);
3772 else
3773 PCALL(PCALL_OTHER);
3774#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003775 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003776 call_fail:
3777 Py_XDECREF(callargs);
3778 Py_XDECREF(kwdict);
3779 return result;
3780}
3781
Fredrik Lundh7a830892006-05-27 10:39:48 +00003782static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003783ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3784{
3785 int nstar = 0;
3786 PyObject *callargs = NULL;
3787 PyObject *stararg = NULL;
3788 PyObject *kwdict = NULL;
3789 PyObject *result = NULL;
3790
3791 if (flags & CALL_FLAG_KW) {
3792 kwdict = EXT_POP(*pp_stack);
3793 if (!(kwdict && PyDict_Check(kwdict))) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003794 PyErr_Format(PyExc_TypeError,
Jeremy Hylton512a2372001-04-11 13:52:29 +00003795 "%s%s argument after ** "
3796 "must be a dictionary",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003797 PyEval_GetFuncName(func),
3798 PyEval_GetFuncDesc(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003799 goto ext_call_fail;
3800 }
3801 }
3802 if (flags & CALL_FLAG_VAR) {
3803 stararg = EXT_POP(*pp_stack);
3804 if (!PyTuple_Check(stararg)) {
3805 PyObject *t = NULL;
3806 t = PySequence_Tuple(stararg);
3807 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003808 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3809 PyErr_Format(PyExc_TypeError,
3810 "%s%s argument after * "
3811 "must be a sequence",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003812 PyEval_GetFuncName(func),
3813 PyEval_GetFuncDesc(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003814 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003815 goto ext_call_fail;
3816 }
3817 Py_DECREF(stararg);
3818 stararg = t;
3819 }
3820 nstar = PyTuple_GET_SIZE(stararg);
3821 }
3822 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003823 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003824 if (kwdict == NULL)
3825 goto ext_call_fail;
3826 }
3827 callargs = update_star_args(na, nstar, stararg, pp_stack);
3828 if (callargs == NULL)
3829 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003830#ifdef CALL_PROFILE
3831 /* At this point, we have to look at the type of func to
3832 update the call stats properly. Do it here so as to avoid
3833 exposing the call stats machinery outside ceval.c
3834 */
3835 if (PyFunction_Check(func))
3836 PCALL(PCALL_FUNCTION);
3837 else if (PyMethod_Check(func))
3838 PCALL(PCALL_METHOD);
3839 else if (PyType_Check(func))
3840 PCALL(PCALL_TYPE);
3841 else
3842 PCALL(PCALL_OTHER);
3843#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003844 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003845 ext_call_fail:
3846 Py_XDECREF(callargs);
3847 Py_XDECREF(kwdict);
3848 Py_XDECREF(stararg);
3849 return result;
3850}
3851
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003852/* Extract a slice index from a PyInt or PyLong or an object with the
3853 nb_index slot defined, and store in *pi.
3854 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3855 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 +00003856 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003857*/
Tim Petersb5196382001-12-16 19:44:20 +00003858/* Note: If v is NULL, return success without storing into *pi. This
3859 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3860 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003861*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003862int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003863_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003864{
Tim Petersb5196382001-12-16 19:44:20 +00003865 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003866 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00003867 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003868 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
3869 however, it looks like it should be AsSsize_t.
3870 There should be a comment here explaining why.
3871 */
3872 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00003873 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003874 else if (PyIndex_Check(v)) {
3875 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003876 if (x == -1 && PyErr_Occurred())
3877 return 0;
3878 }
3879 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003880 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003881 "slice indices must be integers or "
3882 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003883 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003884 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003885 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003886 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003887 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003888}
3889
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003890#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003891#define ISINDEX(x) ((x) == NULL || \
3892 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00003893
Fredrik Lundh7a830892006-05-27 10:39:48 +00003894static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003895apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003896{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003897 PyTypeObject *tp = u->ob_type;
3898 PySequenceMethods *sq = tp->tp_as_sequence;
3899
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003900 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003901 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003902 if (!_PyEval_SliceIndex(v, &ilow))
3903 return NULL;
3904 if (!_PyEval_SliceIndex(w, &ihigh))
3905 return NULL;
3906 return PySequence_GetSlice(u, ilow, ihigh);
3907 }
3908 else {
3909 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00003910 if (slice != NULL) {
3911 PyObject *res = PyObject_GetItem(u, slice);
3912 Py_DECREF(slice);
3913 return res;
3914 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00003915 else
3916 return NULL;
3917 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003918}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003919
Fredrik Lundh7a830892006-05-27 10:39:48 +00003920static int
Guido van Rossumac7be682001-01-17 15:42:30 +00003921assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
3922 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003923{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003924 PyTypeObject *tp = u->ob_type;
3925 PySequenceMethods *sq = tp->tp_as_sequence;
3926
Georg Brandl0ea89162007-03-05 22:28:13 +00003927 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003928 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003929 if (!_PyEval_SliceIndex(v, &ilow))
3930 return -1;
3931 if (!_PyEval_SliceIndex(w, &ihigh))
3932 return -1;
3933 if (x == NULL)
3934 return PySequence_DelSlice(u, ilow, ihigh);
3935 else
3936 return PySequence_SetSlice(u, ilow, ihigh, x);
3937 }
3938 else {
3939 PyObject *slice = PySlice_New(v, w, NULL);
3940 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00003941 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003942 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00003943 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00003944 else
Guido van Rossum354797c2001-12-03 19:45:06 +00003945 res = PyObject_DelItem(u, slice);
3946 Py_DECREF(slice);
3947 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003948 }
3949 else
3950 return -1;
3951 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003952}
3953
Fredrik Lundh7a830892006-05-27 10:39:48 +00003954static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003955cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003956{
Guido van Rossumac7be682001-01-17 15:42:30 +00003957 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003958 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003959 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003960 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003961 break;
3962 case PyCmp_IS_NOT:
3963 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003964 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003965 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003966 res = PySequence_Contains(w, v);
3967 if (res < 0)
3968 return NULL;
3969 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003970 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003971 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003972 if (res < 0)
3973 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003974 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003975 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003976 case PyCmp_EXC_MATCH:
Barry Warsaw4249f541997-08-22 21:26:19 +00003977 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003978 break;
3979 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003980 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003981 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003982 v = res ? Py_True : Py_False;
3983 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003984 return v;
3985}
3986
Fredrik Lundh7a830892006-05-27 10:39:48 +00003987static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00003988import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00003989{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003990 PyObject *x;
3991
3992 x = PyObject_GetAttr(v, name);
3993 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00003994 PyErr_Format(PyExc_ImportError,
3995 "cannot import name %.230s",
3996 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003997 }
Thomas Wouters52152252000-08-17 22:55:00 +00003998 return x;
3999}
Guido van Rossumac7be682001-01-17 15:42:30 +00004000
Fredrik Lundh7a830892006-05-27 10:39:48 +00004001static int
Thomas Wouters52152252000-08-17 22:55:00 +00004002import_all_from(PyObject *locals, PyObject *v)
4003{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004004 PyObject *all = PyObject_GetAttrString(v, "__all__");
4005 PyObject *dict, *name, *value;
4006 int skip_leading_underscores = 0;
4007 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004008
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004009 if (all == NULL) {
4010 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4011 return -1; /* Unexpected error */
4012 PyErr_Clear();
4013 dict = PyObject_GetAttrString(v, "__dict__");
4014 if (dict == NULL) {
4015 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4016 return -1;
4017 PyErr_SetString(PyExc_ImportError,
4018 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004019 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004020 }
4021 all = PyMapping_Keys(dict);
4022 Py_DECREF(dict);
4023 if (all == NULL)
4024 return -1;
4025 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004026 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004027
4028 for (pos = 0, err = 0; ; pos++) {
4029 name = PySequence_GetItem(all, pos);
4030 if (name == NULL) {
4031 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4032 err = -1;
4033 else
4034 PyErr_Clear();
4035 break;
4036 }
4037 if (skip_leading_underscores &&
4038 PyString_Check(name) &&
4039 PyString_AS_STRING(name)[0] == '_')
4040 {
4041 Py_DECREF(name);
4042 continue;
4043 }
4044 value = PyObject_GetAttr(v, name);
4045 if (value == NULL)
4046 err = -1;
Armin Rigo1bc1ab22006-11-29 22:07:38 +00004047 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004048 err = PyDict_SetItem(locals, name, value);
Armin Rigo1bc1ab22006-11-29 22:07:38 +00004049 else
4050 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004051 Py_DECREF(name);
4052 Py_XDECREF(value);
4053 if (err != 0)
4054 break;
4055 }
4056 Py_DECREF(all);
4057 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004058}
4059
Fredrik Lundh7a830892006-05-27 10:39:48 +00004060static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004061build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004062{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004063 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004064
4065 if (PyDict_Check(methods))
4066 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004067 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004068 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004069 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4070 base = PyTuple_GET_ITEM(bases, 0);
4071 metaclass = PyObject_GetAttrString(base, "__class__");
4072 if (metaclass == NULL) {
4073 PyErr_Clear();
4074 metaclass = (PyObject *)base->ob_type;
4075 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004076 }
4077 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004078 else {
4079 PyObject *g = PyEval_GetGlobals();
4080 if (g != NULL && PyDict_Check(g))
4081 metaclass = PyDict_GetItemString(g, "__metaclass__");
4082 if (metaclass == NULL)
4083 metaclass = (PyObject *) &PyClass_Type;
4084 Py_INCREF(metaclass);
4085 }
Georg Brandl684fd0c2006-05-25 19:15:31 +00004086 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods, NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004087 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004088 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004089 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004090 in a base that was not a class (such the random module
4091 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004092 by augmenting the error message with more information.*/
4093
4094 PyObject *ptype, *pvalue, *ptraceback;
4095
4096 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
4097 if (PyString_Check(pvalue)) {
4098 PyObject *newmsg;
4099 newmsg = PyString_FromFormat(
4100 "Error when calling the metaclass bases\n %s",
4101 PyString_AS_STRING(pvalue));
4102 if (newmsg != NULL) {
4103 Py_DECREF(pvalue);
4104 pvalue = newmsg;
4105 }
4106 }
4107 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004108 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004109 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004110}
4111
Fredrik Lundh7a830892006-05-27 10:39:48 +00004112static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004113exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4114 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004115{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004116 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004117 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004118 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004119
Guido van Rossumb209a111997-04-29 18:18:01 +00004120 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4121 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004122 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004123 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004124 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004125 locals = PyTuple_GetItem(prog, 2);
4126 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004127 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004128 if (globals == Py_None) {
4129 globals = PyEval_GetGlobals();
4130 if (locals == Py_None) {
4131 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004132 plain = 1;
4133 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004134 if (!globals || !locals) {
4135 PyErr_SetString(PyExc_SystemError,
4136 "globals and locals cannot be NULL");
4137 return -1;
4138 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004139 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004140 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004141 locals = globals;
Guido van Rossumb209a111997-04-29 18:18:01 +00004142 if (!PyString_Check(prog) &&
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004143 !PyUnicode_Check(prog) &&
Guido van Rossumb209a111997-04-29 18:18:01 +00004144 !PyCode_Check(prog) &&
4145 !PyFile_Check(prog)) {
4146 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004147 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004148 return -1;
4149 }
Fred Drake661ea262000-10-24 19:57:45 +00004150 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004151 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004152 "exec: arg 2 must be a dictionary or None");
4153 return -1;
4154 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004155 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004156 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004157 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004158 return -1;
4159 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004160 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004161 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004162 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004163 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4164 PyErr_SetString(PyExc_TypeError,
4165 "code object passed to exec may not contain free variables");
4166 return -1;
4167 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004168 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004169 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004170 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004171 FILE *fp = PyFile_AsFile(prog);
4172 char *name = PyString_AsString(PyFile_Name(prog));
Tim Peters5ba58662001-07-16 02:29:45 +00004173 PyCompilerFlags cf;
Neal Norwitza5f5f142007-02-25 16:19:21 +00004174 if (name == NULL)
4175 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004176 cf.cf_flags = 0;
4177 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004178 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004179 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004180 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004181 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004182 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004183 }
4184 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004185 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004186 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004187 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004188 cf.cf_flags = 0;
4189#ifdef Py_USING_UNICODE
4190 if (PyUnicode_Check(prog)) {
4191 tmp = PyUnicode_AsUTF8String(prog);
4192 if (tmp == NULL)
4193 return -1;
4194 prog = tmp;
4195 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4196 }
4197#endif
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004198 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004199 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004200 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004201 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004202 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004203 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004204 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004205 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004206 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004207 if (plain)
4208 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004209 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004210 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004211 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004212 return 0;
4213}
Guido van Rossum24c13741995-02-14 09:42:43 +00004214
Fredrik Lundh7a830892006-05-27 10:39:48 +00004215static void
Paul Prescode68140d2000-08-30 20:25:01 +00004216format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4217{
4218 char *obj_str;
4219
4220 if (!obj)
4221 return;
4222
4223 obj_str = PyString_AsString(obj);
4224 if (!obj_str)
4225 return;
4226
4227 PyErr_Format(exc, format_str, obj_str);
4228}
Guido van Rossum950361c1997-01-24 13:49:28 +00004229
Fredrik Lundh7a830892006-05-27 10:39:48 +00004230static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004231string_concatenate(PyObject *v, PyObject *w,
4232 PyFrameObject *f, unsigned char *next_instr)
4233{
4234 /* This function implements 'variable += expr' when both arguments
4235 are strings. */
Armin Rigo97ff0472006-08-09 15:37:26 +00004236 Py_ssize_t v_len = PyString_GET_SIZE(v);
4237 Py_ssize_t w_len = PyString_GET_SIZE(w);
4238 Py_ssize_t new_len = v_len + w_len;
4239 if (new_len < 0) {
4240 PyErr_SetString(PyExc_OverflowError,
4241 "strings are too large to concat");
4242 return NULL;
4243 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004244
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004245 if (v->ob_refcnt == 2) {
4246 /* In the common case, there are 2 references to the value
4247 * stored in 'variable' when the += is performed: one on the
4248 * value stack (in 'v') and one still stored in the 'variable'.
4249 * We try to delete the variable now to reduce the refcnt to 1.
4250 */
4251 switch (*next_instr) {
4252 case STORE_FAST:
4253 {
4254 int oparg = PEEKARG();
4255 PyObject **fastlocals = f->f_localsplus;
4256 if (GETLOCAL(oparg) == v)
4257 SETLOCAL(oparg, NULL);
4258 break;
4259 }
4260 case STORE_DEREF:
4261 {
Richard Jonescebbefc2006-05-23 18:28:17 +00004262 PyObject **freevars = f->f_localsplus + f->f_code->co_nlocals;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004263 PyObject *c = freevars[PEEKARG()];
4264 if (PyCell_GET(c) == v)
4265 PyCell_Set(c, NULL);
4266 break;
4267 }
4268 case STORE_NAME:
4269 {
4270 PyObject *names = f->f_code->co_names;
4271 PyObject *name = GETITEM(names, PEEKARG());
4272 PyObject *locals = f->f_locals;
4273 if (PyDict_CheckExact(locals) &&
4274 PyDict_GetItem(locals, name) == v) {
4275 if (PyDict_DelItem(locals, name) != 0) {
4276 PyErr_Clear();
4277 }
4278 }
4279 break;
4280 }
4281 }
4282 }
4283
Armin Rigo618fbf52004-08-07 20:58:32 +00004284 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004285 /* Now we own the last reference to 'v', so we can resize it
4286 * in-place.
4287 */
Armin Rigo97ff0472006-08-09 15:37:26 +00004288 if (_PyString_Resize(&v, new_len) != 0) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004289 /* XXX if _PyString_Resize() fails, 'v' has been
4290 * deallocated so it cannot be put back into 'variable'.
4291 * The MemoryError is raised when there is no value in
4292 * 'variable', which might (very remotely) be a cause
4293 * of incompatibilities.
4294 */
4295 return NULL;
4296 }
4297 /* copy 'w' into the newly allocated area of 'v' */
4298 memcpy(PyString_AS_STRING(v) + v_len,
4299 PyString_AS_STRING(w), w_len);
4300 return v;
4301 }
4302 else {
4303 /* When in-place resizing is not an option. */
4304 PyString_Concat(&v, w);
4305 return v;
4306 }
4307}
4308
Guido van Rossum950361c1997-01-24 13:49:28 +00004309#ifdef DYNAMIC_EXECUTION_PROFILE
4310
Fredrik Lundh7a830892006-05-27 10:39:48 +00004311static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004312getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004313{
4314 int i;
4315 PyObject *l = PyList_New(256);
4316 if (l == NULL) return NULL;
4317 for (i = 0; i < 256; i++) {
4318 PyObject *x = PyInt_FromLong(a[i]);
4319 if (x == NULL) {
4320 Py_DECREF(l);
4321 return NULL;
4322 }
4323 PyList_SetItem(l, i, x);
4324 }
4325 for (i = 0; i < 256; i++)
4326 a[i] = 0;
4327 return l;
4328}
4329
4330PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004331_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004332{
4333#ifndef DXPAIRS
4334 return getarray(dxp);
4335#else
4336 int i;
4337 PyObject *l = PyList_New(257);
4338 if (l == NULL) return NULL;
4339 for (i = 0; i < 257; i++) {
4340 PyObject *x = getarray(dxpairs[i]);
4341 if (x == NULL) {
4342 Py_DECREF(l);
4343 return NULL;
4344 }
4345 PyList_SetItem(l, i, x);
4346 }
4347 return l;
4348#endif
4349}
4350
4351#endif