blob: 06d524b275d118a863e618f3597fb0bd31fdf27f [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 *);
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +0000108static int 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. */
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +0000713 if (call_trace_protected(tstate->c_tracefunc,
714 tstate->c_traceobj,
715 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000716 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000717 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000718 }
719 }
720 if (tstate->c_profilefunc != NULL) {
721 /* Similar for c_profilefunc, except it needn't
722 return itself and isn't called for "line" events */
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +0000723 if (call_trace_protected(tstate->c_profilefunc,
724 tstate->c_profileobj,
725 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000726 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000727 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000728 }
729 }
730 }
731
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000732 co = f->f_code;
733 names = co->co_names;
734 consts = co->co_consts;
735 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000736 freevars = f->f_localsplus + co->co_nlocals;
Brett Cannonc9371d42005-06-25 08:23:41 +0000737 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000738 /* An explanation is in order for the next line.
739
740 f->f_lasti now refers to the index of the last instruction
741 executed. You might think this was obvious from the name, but
742 this wasn't always true before 2.3! PyFrame_New now sets
743 f->f_lasti to -1 (i.e. the index *before* the first instruction)
744 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
745 does work. Promise. */
746 next_instr = first_instr + f->f_lasti + 1;
747 stack_pointer = f->f_stacktop;
748 assert(stack_pointer != NULL);
749 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
750
Tim Peters5ca576e2001-06-18 22:08:13 +0000751#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000752 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000753#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000754#if defined(Py_DEBUG) || defined(LLTRACE)
Tim Peters5ca576e2001-06-18 22:08:13 +0000755 filename = PyString_AsString(co->co_filename);
756#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000757
Guido van Rossum374a9221991-04-04 10:40:29 +0000758 why = WHY_NOT;
759 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000760 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000761 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000762
Anthony Baxtera863d332006-04-11 07:43:46 +0000763 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000764 why = WHY_EXCEPTION;
765 goto on_error;
766 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000767
Guido van Rossum374a9221991-04-04 10:40:29 +0000768 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000769#ifdef WITH_TSC
770 if (inst1 == 0) {
771 /* Almost surely, the opcode executed a break
772 or a continue, preventing inst1 from being set
773 on the way out of the loop.
774 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000775 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000776 loop1 = inst1;
777 }
778 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
779 intr0, intr1);
780 ticked = 0;
781 inst1 = 0;
782 intr0 = 0;
783 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000784 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000785#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000786 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000787 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000788
Guido van Rossuma027efa1997-05-05 20:56:21 +0000789 /* Do periodic things. Doing this every time through
790 the loop would add too much overhead, so we do it
791 only every Nth instruction. We also do it if
792 ``things_to_do'' is set, i.e. when an asynchronous
793 event needs attention (e.g. a signal handler or
794 async I/O handler); see Py_AddPendingCall() and
795 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000796
Skip Montanarod581d772002-09-03 20:10:45 +0000797 if (--_Py_Ticker < 0) {
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000798 if (*next_instr == SETUP_FINALLY) {
799 /* Make the last opcode before
800 a try: finally: block uninterruptable. */
801 goto fast_next_opcode;
802 }
Skip Montanarod581d772002-09-03 20:10:45 +0000803 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000804 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000805#ifdef WITH_TSC
806 ticked = 1;
807#endif
Guido van Rossuma027efa1997-05-05 20:56:21 +0000808 if (things_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000809 if (Py_MakePendingCalls() < 0) {
810 why = WHY_EXCEPTION;
811 goto on_error;
812 }
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000813 if (things_to_do)
814 /* MakePendingCalls() didn't succeed.
815 Force early re-execution of this
816 "periodic" code, possibly after
817 a thread switch */
818 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000819 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000820#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000821 if (interpreter_lock) {
822 /* Give another thread a chance */
823
Guido van Rossum25ce5661997-08-02 03:10:38 +0000824 if (PyThreadState_Swap(NULL) != tstate)
825 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000826 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000827
828 /* Other threads may run now */
829
Guido van Rossum65d5b571998-12-21 19:32:43 +0000830 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000831 if (PyThreadState_Swap(tstate) != NULL)
832 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000833
834 /* Check for thread interrupts */
835
836 if (tstate->async_exc != NULL) {
837 x = tstate->async_exc;
838 tstate->async_exc = NULL;
839 PyErr_SetNone(x);
840 Py_DECREF(x);
841 why = WHY_EXCEPTION;
842 goto on_error;
843 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000844 }
845#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000846 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000847
Neil Schemenauer63543862002-02-17 19:10:14 +0000848 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +0000849 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +0000850
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000851 /* line-by-line tracing support */
852
853 if (tstate->c_tracefunc != NULL && !tstate->tracing) {
854 /* see maybe_call_line_trace
855 for expository comments */
856 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +0000857
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000858 err = maybe_call_line_trace(tstate->c_tracefunc,
859 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +0000860 f, &instr_lb, &instr_ub,
861 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000862 /* Reload possibly changed frame fields */
863 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +0000864 if (f->f_stacktop != NULL) {
865 stack_pointer = f->f_stacktop;
866 f->f_stacktop = NULL;
867 }
868 if (err) {
869 /* trace function raised an exception */
870 goto on_error;
871 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000872 }
873
874 /* Extract opcode and argument */
875
Guido van Rossum374a9221991-04-04 10:40:29 +0000876 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +0000877 oparg = 0; /* allows oparg to be stored in a register because
878 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000879 if (HAS_ARG(opcode))
880 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +0000881 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +0000882#ifdef DYNAMIC_EXECUTION_PROFILE
883#ifdef DXPAIRS
884 dxpairs[lastopcode][opcode]++;
885 lastopcode = opcode;
886#endif
887 dxp[opcode]++;
888#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000889
Guido van Rossum96a42c81992-01-12 02:29:51 +0000890#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +0000891 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +0000892
Guido van Rossum96a42c81992-01-12 02:29:51 +0000893 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +0000894 if (HAS_ARG(opcode)) {
895 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000896 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +0000897 }
898 else {
899 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000900 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +0000901 }
902 }
903#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000904
Guido van Rossum374a9221991-04-04 10:40:29 +0000905 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000906 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +0000907
Guido van Rossum374a9221991-04-04 10:40:29 +0000908 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +0000909
Guido van Rossum374a9221991-04-04 10:40:29 +0000910 /* BEWARE!
911 It is essential that any operation that fails sets either
912 x to NULL, err to nonzero, or why to anything but WHY_NOT,
913 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000914
Guido van Rossum374a9221991-04-04 10:40:29 +0000915 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +0000916
Raymond Hettinger9c18e812004-06-21 16:31:15 +0000917 case NOP:
918 goto fast_next_opcode;
919
Neil Schemenauer63543862002-02-17 19:10:14 +0000920 case LOAD_FAST:
921 x = GETLOCAL(oparg);
922 if (x != NULL) {
923 Py_INCREF(x);
924 PUSH(x);
925 goto fast_next_opcode;
926 }
927 format_exc_check_arg(PyExc_UnboundLocalError,
928 UNBOUNDLOCAL_ERROR_MSG,
929 PyTuple_GetItem(co->co_varnames, oparg));
930 break;
931
932 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +0000933 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +0000934 Py_INCREF(x);
935 PUSH(x);
936 goto fast_next_opcode;
937
Raymond Hettinger7dc52212003-03-16 20:14:44 +0000938 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +0000939 case STORE_FAST:
940 v = POP();
941 SETLOCAL(oparg, v);
942 goto fast_next_opcode;
943
Raymond Hettingerf606f872003-03-16 03:11:04 +0000944 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +0000945 case POP_TOP:
946 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000947 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +0000948 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000949
Guido van Rossum374a9221991-04-04 10:40:29 +0000950 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000951 v = TOP();
952 w = SECOND();
953 SET_TOP(w);
954 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000955 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000956
Guido van Rossum374a9221991-04-04 10:40:29 +0000957 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000958 v = TOP();
959 w = SECOND();
960 x = THIRD();
961 SET_TOP(w);
962 SET_SECOND(x);
963 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000964 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000965
Thomas Wouters434d0822000-08-24 20:11:32 +0000966 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +0000967 u = TOP();
968 v = SECOND();
969 w = THIRD();
970 x = FOURTH();
971 SET_TOP(v);
972 SET_SECOND(w);
973 SET_THIRD(x);
974 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000975 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000976
Guido van Rossum374a9221991-04-04 10:40:29 +0000977 case DUP_TOP:
978 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +0000979 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +0000980 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +0000981 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +0000982
Thomas Wouters434d0822000-08-24 20:11:32 +0000983 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000984 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +0000985 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +0000986 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000987 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +0000988 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000989 STACKADJ(2);
990 SET_TOP(x);
991 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +0000992 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +0000993 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +0000994 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +0000995 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000996 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +0000997 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +0000998 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +0000999 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001000 STACKADJ(3);
1001 SET_TOP(x);
1002 SET_SECOND(w);
1003 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001004 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001005 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001006 Py_FatalError("invalid argument to DUP_TOPX"
1007 " (bytecode corruption?)");
Tim Peters35ba6892000-10-11 07:04:49 +00001008 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001009
Guido van Rossum374a9221991-04-04 10:40:29 +00001010 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001011 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001012 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001013 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001014 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001015 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001016 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001017
Guido van Rossum374a9221991-04-04 10:40:29 +00001018 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001019 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001020 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001021 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001022 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001023 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001024 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001025
Guido van Rossum374a9221991-04-04 10:40:29 +00001026 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001027 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001028 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001029 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001030 if (err == 0) {
1031 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001032 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001033 continue;
1034 }
1035 else if (err > 0) {
1036 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001037 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001038 err = 0;
1039 continue;
1040 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001041 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001042 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001043
Guido van Rossum374a9221991-04-04 10:40:29 +00001044 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001045 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001046 x = PyObject_Repr(v);
1047 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001048 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001049 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001050 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001051
Guido van Rossum7928cd71991-10-24 14:59:31 +00001052 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001053 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001054 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001055 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001056 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001057 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001058 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001059
Guido van Rossum50564e81996-01-12 01:13:16 +00001060 case BINARY_POWER:
1061 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001062 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001063 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001064 Py_DECREF(v);
1065 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001066 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001067 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001068 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001069
Guido van Rossum374a9221991-04-04 10:40:29 +00001070 case BINARY_MULTIPLY:
1071 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001072 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001073 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001074 Py_DECREF(v);
1075 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001076 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001077 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001078 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001079
Guido van Rossum374a9221991-04-04 10:40:29 +00001080 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001081 if (!_Py_QnewFlag) {
1082 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001083 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001084 x = PyNumber_Divide(v, w);
1085 Py_DECREF(v);
1086 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001087 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001088 if (x != NULL) continue;
1089 break;
1090 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001091 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001092 BINARY_TRUE_DIVIDE */
1093 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001094 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001095 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001096 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001097 Py_DECREF(v);
1098 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001099 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001100 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001101 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001102
Guido van Rossum4668b002001-08-08 05:00:18 +00001103 case BINARY_FLOOR_DIVIDE:
1104 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001105 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001106 x = PyNumber_FloorDivide(v, w);
1107 Py_DECREF(v);
1108 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001109 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001110 if (x != NULL) continue;
1111 break;
1112
Guido van Rossum374a9221991-04-04 10:40:29 +00001113 case BINARY_MODULO:
1114 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001115 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001116 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001117 Py_DECREF(v);
1118 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001119 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001120 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001121 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001122
Guido van Rossum374a9221991-04-04 10:40:29 +00001123 case BINARY_ADD:
1124 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001125 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001126 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001127 /* INLINE: int + int */
1128 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001129 a = PyInt_AS_LONG(v);
1130 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001131 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001132 if ((i^a) < 0 && (i^b) < 0)
1133 goto slow_add;
1134 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001135 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001136 else if (PyString_CheckExact(v) &&
1137 PyString_CheckExact(w)) {
1138 x = string_concatenate(v, w, f, next_instr);
1139 /* string_concatenate consumed the ref to v */
1140 goto skip_decref_vx;
1141 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001142 else {
1143 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001144 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001145 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001146 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001147 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001148 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001149 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001150 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001151 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001152
Guido van Rossum374a9221991-04-04 10:40:29 +00001153 case BINARY_SUBTRACT:
1154 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001155 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001156 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001157 /* INLINE: int - int */
1158 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001159 a = PyInt_AS_LONG(v);
1160 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001161 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001162 if ((i^a) < 0 && (i^~b) < 0)
1163 goto slow_sub;
1164 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001165 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001166 else {
1167 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001168 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001169 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001170 Py_DECREF(v);
1171 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001172 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001173 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001174 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001175
Guido van Rossum374a9221991-04-04 10:40:29 +00001176 case BINARY_SUBSCR:
1177 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001178 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001179 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001180 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001181 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001182 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001183 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001184 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001185 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001186 Py_INCREF(x);
1187 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001188 else
1189 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001190 }
1191 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001192 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001193 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001194 Py_DECREF(v);
1195 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001196 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001197 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001198 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001199
Guido van Rossum7928cd71991-10-24 14:59:31 +00001200 case BINARY_LSHIFT:
1201 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001202 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001203 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001204 Py_DECREF(v);
1205 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001206 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001207 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001208 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001209
Guido van Rossum7928cd71991-10-24 14:59:31 +00001210 case BINARY_RSHIFT:
1211 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001212 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001213 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001214 Py_DECREF(v);
1215 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001216 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001217 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001218 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001219
Guido van Rossum7928cd71991-10-24 14:59:31 +00001220 case BINARY_AND:
1221 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001222 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001223 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001224 Py_DECREF(v);
1225 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001226 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001227 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001228 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001229
Guido van Rossum7928cd71991-10-24 14:59:31 +00001230 case BINARY_XOR:
1231 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001232 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001233 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001234 Py_DECREF(v);
1235 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001236 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001237 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001238 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001239
Guido van Rossum7928cd71991-10-24 14:59:31 +00001240 case BINARY_OR:
1241 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001242 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001243 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001244 Py_DECREF(v);
1245 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001246 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001247 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001248 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001249
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001250 case LIST_APPEND:
1251 w = POP();
1252 v = POP();
1253 err = PyList_Append(v, w);
1254 Py_DECREF(v);
1255 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001256 if (err == 0) {
1257 PREDICT(JUMP_ABSOLUTE);
1258 continue;
1259 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001260 break;
1261
Thomas Wouters434d0822000-08-24 20:11:32 +00001262 case INPLACE_POWER:
1263 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001264 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001265 x = PyNumber_InPlacePower(v, w, Py_None);
1266 Py_DECREF(v);
1267 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001268 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001269 if (x != NULL) continue;
1270 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001271
Thomas Wouters434d0822000-08-24 20:11:32 +00001272 case INPLACE_MULTIPLY:
1273 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001274 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001275 x = PyNumber_InPlaceMultiply(v, w);
1276 Py_DECREF(v);
1277 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001278 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001279 if (x != NULL) continue;
1280 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001281
Thomas Wouters434d0822000-08-24 20:11:32 +00001282 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001283 if (!_Py_QnewFlag) {
1284 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001285 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001286 x = PyNumber_InPlaceDivide(v, w);
1287 Py_DECREF(v);
1288 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001289 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001290 if (x != NULL) continue;
1291 break;
1292 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001293 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001294 INPLACE_TRUE_DIVIDE */
1295 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001296 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001297 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001298 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001299 Py_DECREF(v);
1300 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001301 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001302 if (x != NULL) continue;
1303 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001304
Guido van Rossum4668b002001-08-08 05:00:18 +00001305 case INPLACE_FLOOR_DIVIDE:
1306 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001307 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001308 x = PyNumber_InPlaceFloorDivide(v, w);
1309 Py_DECREF(v);
1310 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001311 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001312 if (x != NULL) continue;
1313 break;
1314
Thomas Wouters434d0822000-08-24 20:11:32 +00001315 case INPLACE_MODULO:
1316 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001317 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001318 x = PyNumber_InPlaceRemainder(v, w);
1319 Py_DECREF(v);
1320 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001321 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001322 if (x != NULL) continue;
1323 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001324
Thomas Wouters434d0822000-08-24 20:11:32 +00001325 case INPLACE_ADD:
1326 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001327 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001328 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001329 /* INLINE: int + int */
1330 register long a, b, i;
1331 a = PyInt_AS_LONG(v);
1332 b = PyInt_AS_LONG(w);
1333 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001334 if ((i^a) < 0 && (i^b) < 0)
1335 goto slow_iadd;
1336 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001337 }
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001338 else if (PyString_CheckExact(v) &&
1339 PyString_CheckExact(w)) {
1340 x = string_concatenate(v, w, f, next_instr);
1341 /* string_concatenate consumed the ref to v */
1342 goto skip_decref_v;
1343 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001344 else {
1345 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001346 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001347 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001348 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001349 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001350 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001351 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001352 if (x != NULL) continue;
1353 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001354
Thomas Wouters434d0822000-08-24 20:11:32 +00001355 case INPLACE_SUBTRACT:
1356 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001357 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001358 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001359 /* INLINE: int - int */
1360 register long a, b, i;
1361 a = PyInt_AS_LONG(v);
1362 b = PyInt_AS_LONG(w);
1363 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001364 if ((i^a) < 0 && (i^~b) < 0)
1365 goto slow_isub;
1366 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001367 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001368 else {
1369 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001370 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001371 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001372 Py_DECREF(v);
1373 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001374 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001375 if (x != NULL) continue;
1376 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001377
Thomas Wouters434d0822000-08-24 20:11:32 +00001378 case INPLACE_LSHIFT:
1379 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001380 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001381 x = PyNumber_InPlaceLshift(v, w);
1382 Py_DECREF(v);
1383 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001384 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001385 if (x != NULL) continue;
1386 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001387
Thomas Wouters434d0822000-08-24 20:11:32 +00001388 case INPLACE_RSHIFT:
1389 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001390 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001391 x = PyNumber_InPlaceRshift(v, w);
1392 Py_DECREF(v);
1393 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001394 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001395 if (x != NULL) continue;
1396 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001397
Thomas Wouters434d0822000-08-24 20:11:32 +00001398 case INPLACE_AND:
1399 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001400 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001401 x = PyNumber_InPlaceAnd(v, w);
1402 Py_DECREF(v);
1403 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001404 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001405 if (x != NULL) continue;
1406 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001407
Thomas Wouters434d0822000-08-24 20:11:32 +00001408 case INPLACE_XOR:
1409 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001410 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001411 x = PyNumber_InPlaceXor(v, w);
1412 Py_DECREF(v);
1413 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001414 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001415 if (x != NULL) continue;
1416 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001417
Thomas Wouters434d0822000-08-24 20:11:32 +00001418 case INPLACE_OR:
1419 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001420 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001421 x = PyNumber_InPlaceOr(v, w);
1422 Py_DECREF(v);
1423 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001424 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001425 if (x != NULL) continue;
1426 break;
1427
Guido van Rossum374a9221991-04-04 10:40:29 +00001428 case SLICE+0:
1429 case SLICE+1:
1430 case SLICE+2:
1431 case SLICE+3:
1432 if ((opcode-SLICE) & 2)
1433 w = POP();
1434 else
1435 w = NULL;
1436 if ((opcode-SLICE) & 1)
1437 v = POP();
1438 else
1439 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001440 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001441 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001442 Py_DECREF(u);
1443 Py_XDECREF(v);
1444 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001445 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001446 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001447 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001448
Guido van Rossum374a9221991-04-04 10:40:29 +00001449 case STORE_SLICE+0:
1450 case STORE_SLICE+1:
1451 case STORE_SLICE+2:
1452 case STORE_SLICE+3:
1453 if ((opcode-STORE_SLICE) & 2)
1454 w = POP();
1455 else
1456 w = NULL;
1457 if ((opcode-STORE_SLICE) & 1)
1458 v = POP();
1459 else
1460 v = NULL;
1461 u = POP();
1462 t = POP();
1463 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001464 Py_DECREF(t);
1465 Py_DECREF(u);
1466 Py_XDECREF(v);
1467 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001468 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001469 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001470
Guido van Rossum374a9221991-04-04 10:40:29 +00001471 case DELETE_SLICE+0:
1472 case DELETE_SLICE+1:
1473 case DELETE_SLICE+2:
1474 case DELETE_SLICE+3:
1475 if ((opcode-DELETE_SLICE) & 2)
1476 w = POP();
1477 else
1478 w = NULL;
1479 if ((opcode-DELETE_SLICE) & 1)
1480 v = POP();
1481 else
1482 v = NULL;
1483 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001484 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001485 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001486 Py_DECREF(u);
1487 Py_XDECREF(v);
1488 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001489 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001490 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001491
Guido van Rossum374a9221991-04-04 10:40:29 +00001492 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001493 w = TOP();
1494 v = SECOND();
1495 u = THIRD();
1496 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001497 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001498 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001499 Py_DECREF(u);
1500 Py_DECREF(v);
1501 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001502 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001503 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001504
Guido van Rossum374a9221991-04-04 10:40:29 +00001505 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001506 w = TOP();
1507 v = SECOND();
1508 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001509 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001510 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001511 Py_DECREF(v);
1512 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001513 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001514 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001515
Guido van Rossum374a9221991-04-04 10:40:29 +00001516 case PRINT_EXPR:
1517 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001518 w = PySys_GetObject("displayhook");
1519 if (w == NULL) {
1520 PyErr_SetString(PyExc_RuntimeError,
1521 "lost sys.displayhook");
1522 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001523 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001524 }
1525 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001526 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001527 if (x == NULL)
1528 err = -1;
1529 }
1530 if (err == 0) {
1531 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001532 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001533 if (w == NULL)
1534 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001535 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001536 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001537 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001538 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001539
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001540 case PRINT_ITEM_TO:
1541 w = stream = POP();
1542 /* fall through to PRINT_ITEM */
1543
Guido van Rossum374a9221991-04-04 10:40:29 +00001544 case PRINT_ITEM:
1545 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001546 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001547 w = PySys_GetObject("stdout");
1548 if (w == NULL) {
1549 PyErr_SetString(PyExc_RuntimeError,
1550 "lost sys.stdout");
1551 err = -1;
1552 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001553 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001554 /* PyFile_SoftSpace() can exececute arbitrary code
1555 if sys.stdout is an instance with a __getattr__.
1556 If __getattr__ raises an exception, w will
1557 be freed, so we need to prevent that temporarily. */
1558 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001559 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001560 err = PyFile_WriteString(" ", w);
1561 if (err == 0)
1562 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001563 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001564 /* XXX move into writeobject() ? */
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001565 if (PyString_Check(v)) {
1566 char *s = PyString_AS_STRING(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001567 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001568 if (len == 0 ||
1569 !isspace(Py_CHARMASK(s[len-1])) ||
1570 s[len-1] == ' ')
1571 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001572 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001573#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001574 else if (PyUnicode_Check(v)) {
1575 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001576 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001577 if (len == 0 ||
1578 !Py_UNICODE_ISSPACE(s[len-1]) ||
1579 s[len-1] == ' ')
1580 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001581 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001582#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001583 else
1584 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001585 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001586 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001587 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001588 Py_XDECREF(stream);
1589 stream = NULL;
1590 if (err == 0)
1591 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001592 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001593
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001594 case PRINT_NEWLINE_TO:
1595 w = stream = POP();
1596 /* fall through to PRINT_NEWLINE */
1597
Guido van Rossum374a9221991-04-04 10:40:29 +00001598 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001599 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001600 w = PySys_GetObject("stdout");
1601 if (w == NULL)
1602 PyErr_SetString(PyExc_RuntimeError,
1603 "lost sys.stdout");
Guido van Rossum3165fe61992-09-25 21:59:05 +00001604 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001605 if (w != NULL) {
1606 err = PyFile_WriteString("\n", w);
1607 if (err == 0)
1608 PyFile_SoftSpace(w, 0);
1609 }
1610 Py_XDECREF(stream);
1611 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001612 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001613
Thomas Wouters434d0822000-08-24 20:11:32 +00001614
1615#ifdef CASE_TOO_BIG
1616 default: switch (opcode) {
1617#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001618 case RAISE_VARARGS:
1619 u = v = w = NULL;
1620 switch (oparg) {
1621 case 3:
1622 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001623 /* Fallthrough */
1624 case 2:
1625 v = POP(); /* value */
1626 /* Fallthrough */
1627 case 1:
1628 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001629 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001630 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001631 break;
1632 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001633 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001634 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001635 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001636 break;
1637 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001638 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001639
Guido van Rossum374a9221991-04-04 10:40:29 +00001640 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001641 if ((x = f->f_locals) != NULL) {
1642 Py_INCREF(x);
1643 PUSH(x);
1644 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001645 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001646 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001647 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001648
Guido van Rossum374a9221991-04-04 10:40:29 +00001649 case RETURN_VALUE:
1650 retval = POP();
1651 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001652 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001653
Tim Peters5ca576e2001-06-18 22:08:13 +00001654 case YIELD_VALUE:
1655 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001656 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001657 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001658 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001659
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001660 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001661 w = TOP();
1662 v = SECOND();
1663 u = THIRD();
1664 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001665 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001666 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001667 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001668 Py_DECREF(u);
1669 Py_DECREF(v);
1670 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001671 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001672
Guido van Rossum374a9221991-04-04 10:40:29 +00001673 case POP_BLOCK:
1674 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001675 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001676 while (STACK_LEVEL() > b->b_level) {
1677 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001678 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001679 }
1680 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001681 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001682
Guido van Rossum374a9221991-04-04 10:40:29 +00001683 case END_FINALLY:
1684 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001685 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001686 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001687 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001688 if (why == WHY_RETURN ||
1689 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001690 retval = POP();
1691 }
Brett Cannonbf364092006-03-01 04:25:17 +00001692 else if (PyExceptionClass_Check(v) || PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001693 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001694 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001695 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001696 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001697 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001698 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001699 else if (v != Py_None) {
1700 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001701 "'finally' pops bad exception");
1702 why = WHY_EXCEPTION;
1703 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001704 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001705 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001706
Guido van Rossum374a9221991-04-04 10:40:29 +00001707 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001708 u = TOP();
1709 v = SECOND();
1710 w = THIRD();
1711 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001712 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001713 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001714 Py_DECREF(u);
1715 Py_DECREF(v);
1716 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001717 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001718
Guido van Rossum374a9221991-04-04 10:40:29 +00001719 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001720 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001721 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001722 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001723 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001724 err = PyDict_SetItem(x, w, v);
1725 else
1726 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001727 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001728 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001729 break;
1730 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001731 PyErr_Format(PyExc_SystemError,
1732 "no locals found when storing %s",
1733 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001734 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001735
Guido van Rossum374a9221991-04-04 10:40:29 +00001736 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001737 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001738 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001739 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001740 format_exc_check_arg(PyExc_NameError,
1741 NAME_ERROR_MSG ,w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001742 break;
1743 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001744 PyErr_Format(PyExc_SystemError,
1745 "no locals when deleting %s",
1746 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001747 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001748
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001749 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001750 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001751 v = POP();
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001752 if (PyTuple_CheckExact(v) && PyTuple_GET_SIZE(v) == oparg) {
1753 PyObject **items = ((PyTupleObject *)v)->ob_item;
1754 while (oparg--) {
1755 w = items[oparg];
1756 Py_INCREF(w);
1757 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001758 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001759 Py_DECREF(v);
1760 continue;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001761 } else if (PyList_CheckExact(v) && PyList_GET_SIZE(v) == oparg) {
1762 PyObject **items = ((PyListObject *)v)->ob_item;
1763 while (oparg--) {
1764 w = items[oparg];
1765 Py_INCREF(w);
1766 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001767 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001768 } else if (unpack_iterable(v, oparg,
Georg Brandl8a10ea42007-03-21 09:00:55 +00001769 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001770 stack_pointer += oparg;
Georg Brandl8a10ea42007-03-21 09:00:55 +00001771 } else {
1772 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001773 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001774 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001775 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001776 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001777
Guido van Rossum374a9221991-04-04 10:40:29 +00001778 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001779 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001780 v = TOP();
1781 u = SECOND();
1782 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001783 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1784 Py_DECREF(v);
1785 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001786 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001787 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001788
Guido van Rossum374a9221991-04-04 10:40:29 +00001789 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001790 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001791 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001792 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1793 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001794 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001795 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001796
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001797 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001798 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001799 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001800 err = PyDict_SetItem(f->f_globals, w, v);
1801 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001802 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001803 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001804
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001805 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001806 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001807 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001808 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001809 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001810 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001811
Guido van Rossum374a9221991-04-04 10:40:29 +00001812 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001813 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001814 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001815 PyErr_Format(PyExc_SystemError,
1816 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001817 PyObject_REPR(w));
Guido van Rossum681d79a1995-07-18 14:51:37 +00001818 break;
1819 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001820 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001821 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001822 Py_XINCREF(x);
1823 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001824 else {
1825 x = PyObject_GetItem(v, w);
1826 if (x == NULL && PyErr_Occurred()) {
1827 if (!PyErr_ExceptionMatches(PyExc_KeyError))
1828 break;
1829 PyErr_Clear();
1830 }
1831 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001832 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001833 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001834 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001835 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001836 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00001837 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001838 PyExc_NameError,
Paul Prescode68140d2000-08-30 20:25:01 +00001839 NAME_ERROR_MSG ,w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001840 break;
1841 }
1842 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00001843 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001844 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001845 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001846 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001847
Guido van Rossum374a9221991-04-04 10:40:29 +00001848 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001849 w = GETITEM(names, oparg);
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001850 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00001851 /* Inline the PyDict_GetItem() calls.
1852 WARNING: this is an extreme speed hack.
1853 Do not try this at home. */
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001854 long hash = ((PyStringObject *)w)->ob_shash;
1855 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001856 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00001857 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001858 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00001859 e = d->ma_lookup(d, w, hash);
1860 if (e == NULL) {
1861 x = NULL;
1862 break;
1863 }
1864 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001865 if (x != NULL) {
1866 Py_INCREF(x);
1867 PUSH(x);
1868 continue;
1869 }
1870 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00001871 e = d->ma_lookup(d, w, hash);
1872 if (e == NULL) {
1873 x = NULL;
1874 break;
1875 }
1876 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001877 if (x != NULL) {
1878 Py_INCREF(x);
1879 PUSH(x);
1880 continue;
1881 }
1882 goto load_global_error;
1883 }
1884 }
1885 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00001886 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001887 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00001888 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001889 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001890 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00001891 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00001892 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00001893 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001894 break;
1895 }
1896 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001897 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001898 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001899 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001900
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00001901 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001902 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001903 if (x != NULL) {
1904 SETLOCAL(oparg, NULL);
1905 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00001906 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001907 format_exc_check_arg(
1908 PyExc_UnboundLocalError,
1909 UNBOUNDLOCAL_ERROR_MSG,
1910 PyTuple_GetItem(co->co_varnames, oparg)
1911 );
1912 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001913
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001914 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001915 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001916 Py_INCREF(x);
1917 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001918 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001919 break;
1920
1921 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001922 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001923 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001924 if (w != NULL) {
1925 PUSH(w);
1926 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00001927 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001928 err = -1;
1929 /* Don't stomp existing exception */
1930 if (PyErr_Occurred())
1931 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00001932 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
1933 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00001934 oparg);
1935 format_exc_check_arg(
1936 PyExc_UnboundLocalError,
1937 UNBOUNDLOCAL_ERROR_MSG,
1938 v);
1939 } else {
Richard Jonescebbefc2006-05-23 18:28:17 +00001940 v = PyTuple_GET_ITEM(
Raymond Hettinger467a6982004-04-07 11:39:21 +00001941 co->co_freevars,
Richard Jonescebbefc2006-05-23 18:28:17 +00001942 oparg - PyTuple_GET_SIZE(co->co_cellvars));
Raymond Hettinger467a6982004-04-07 11:39:21 +00001943 format_exc_check_arg(
1944 PyExc_NameError,
1945 UNBOUNDFREE_ERROR_MSG,
1946 v);
1947 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001948 break;
1949
1950 case STORE_DEREF:
1951 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00001952 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001953 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00001954 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001955 continue;
1956
Guido van Rossum374a9221991-04-04 10:40:29 +00001957 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00001958 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001959 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001960 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001961 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001962 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001963 }
1964 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001965 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001966 }
1967 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001968
Guido van Rossum374a9221991-04-04 10:40:29 +00001969 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00001970 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001971 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001972 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001973 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00001974 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001975 }
1976 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001977 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001978 }
1979 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001980
Guido van Rossum374a9221991-04-04 10:40:29 +00001981 case BUILD_MAP:
Guido van Rossumb209a111997-04-29 18:18:01 +00001982 x = PyDict_New();
Guido van Rossum374a9221991-04-04 10:40:29 +00001983 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001984 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001985 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001986
Guido van Rossum374a9221991-04-04 10:40:29 +00001987 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001988 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001989 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001990 x = PyObject_GetAttr(v, w);
1991 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001992 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001993 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001994 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001995
Guido van Rossum374a9221991-04-04 10:40:29 +00001996 case COMPARE_OP:
1997 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001998 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001999 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002000 /* INLINE: cmp(int, int) */
2001 register long a, b;
2002 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002003 a = PyInt_AS_LONG(v);
2004 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002005 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002006 case PyCmp_LT: res = a < b; break;
2007 case PyCmp_LE: res = a <= b; break;
2008 case PyCmp_EQ: res = a == b; break;
2009 case PyCmp_NE: res = a != b; break;
2010 case PyCmp_GT: res = a > b; break;
2011 case PyCmp_GE: res = a >= b; break;
2012 case PyCmp_IS: res = v == w; break;
2013 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002014 default: goto slow_compare;
2015 }
2016 x = res ? Py_True : Py_False;
2017 Py_INCREF(x);
2018 }
2019 else {
2020 slow_compare:
2021 x = cmp_outcome(oparg, v, w);
2022 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002023 Py_DECREF(v);
2024 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002025 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002026 if (x == NULL) break;
2027 PREDICT(JUMP_IF_FALSE);
2028 PREDICT(JUMP_IF_TRUE);
2029 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002030
Guido van Rossum374a9221991-04-04 10:40:29 +00002031 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002032 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002033 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002034 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002035 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002036 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002037 break;
2038 }
Guido van Rossume105f982008-01-23 20:09:39 +00002039 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002040 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002041 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002042 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2043 w = PyTuple_Pack(5,
2044 w,
2045 f->f_globals,
2046 f->f_locals == NULL ?
2047 Py_None : f->f_locals,
2048 v,
2049 u);
2050 else
2051 w = PyTuple_Pack(4,
2052 w,
2053 f->f_globals,
2054 f->f_locals == NULL ?
2055 Py_None : f->f_locals,
2056 v);
2057 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002058 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002059 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002060 u = POP();
Guido van Rossume105f982008-01-23 20:09:39 +00002061 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002062 x = NULL;
2063 break;
2064 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002065 READ_TIMESTAMP(intr0);
Guido van Rossume105f982008-01-23 20:09:39 +00002066 v = x;
2067 x = PyEval_CallObject(v, w);
2068 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002069 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002070 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002071 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002072 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002073 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002074
Thomas Wouters52152252000-08-17 22:55:00 +00002075 case IMPORT_STAR:
2076 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002077 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002078 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002079 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002080 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002081 break;
2082 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002083 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002084 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002085 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002086 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002087 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002088 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002089 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002090
Thomas Wouters52152252000-08-17 22:55:00 +00002091 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002092 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002093 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002094 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002095 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002096 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002097 PUSH(x);
2098 if (x != NULL) continue;
2099 break;
2100
Guido van Rossum374a9221991-04-04 10:40:29 +00002101 case JUMP_FORWARD:
2102 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002103 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002104
Raymond Hettingerf606f872003-03-16 03:11:04 +00002105 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002106 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002107 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002108 if (w == Py_True) {
2109 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002110 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002111 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002112 if (w == Py_False) {
2113 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002114 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002115 }
2116 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002117 if (err > 0)
2118 err = 0;
2119 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002120 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002121 else
2122 break;
2123 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002124
Raymond Hettingerf606f872003-03-16 03:11:04 +00002125 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002126 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002127 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002128 if (w == Py_False) {
2129 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002130 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002131 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002132 if (w == Py_True) {
2133 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002134 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002135 }
2136 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002137 if (err > 0) {
2138 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002139 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002140 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002141 else if (err == 0)
2142 ;
2143 else
2144 break;
2145 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002146
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002147 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002148 case JUMP_ABSOLUTE:
2149 JUMPTO(oparg);
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002150 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002151
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002152 case GET_ITER:
2153 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002154 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002155 x = PyObject_GetIter(v);
2156 Py_DECREF(v);
2157 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002158 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002159 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002160 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002161 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002162 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002163 break;
2164
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002165 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002166 case FOR_ITER:
2167 /* before: [iter]; after: [iter, iter()] *or* [] */
2168 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002169 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002170 if (x != NULL) {
2171 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002172 PREDICT(STORE_FAST);
2173 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002174 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002175 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002176 if (PyErr_Occurred()) {
2177 if (!PyErr_ExceptionMatches(PyExc_StopIteration))
2178 break;
2179 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002180 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002181 /* iterator ended normally */
2182 x = v = POP();
2183 Py_DECREF(v);
2184 JUMPBY(oparg);
2185 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002186
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002187 case BREAK_LOOP:
2188 why = WHY_BREAK;
2189 goto fast_block_end;
2190
2191 case CONTINUE_LOOP:
2192 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002193 if (!retval) {
2194 x = NULL;
2195 break;
2196 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002197 why = WHY_CONTINUE;
2198 goto fast_block_end;
2199
Guido van Rossum374a9221991-04-04 10:40:29 +00002200 case SETUP_LOOP:
2201 case SETUP_EXCEPT:
2202 case SETUP_FINALLY:
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002203 /* NOTE: If you add any new block-setup opcodes that are not try/except/finally
2204 handlers, you may need to update the PyGen_NeedsFinalizing() function. */
2205
Guido van Rossumb209a111997-04-29 18:18:01 +00002206 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002207 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002208 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002209
Guido van Rossumc2e20742006-02-27 22:32:47 +00002210 case WITH_CLEANUP:
2211 {
2212 /* TOP is the context.__exit__ bound method.
2213 Below that are 1-3 values indicating how/why
2214 we entered the finally clause:
2215 - SECOND = None
Guido van Rossumf6694362006-03-10 02:28:35 +00002216 - (SECOND, THIRD) = (WHY_{RETURN,CONTINUE}), retval
Guido van Rossumc2e20742006-02-27 22:32:47 +00002217 - SECOND = WHY_*; no retval below it
2218 - (SECOND, THIRD, FOURTH) = exc_info()
2219 In the last case, we must call
2220 TOP(SECOND, THIRD, FOURTH)
2221 otherwise we must call
2222 TOP(None, None, None)
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002223
2224 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002225 *and* the function call returns a 'true' value, we
2226 "zap" this information, to prevent END_FINALLY from
2227 re-raising the exception. (But non-local gotos
2228 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002229 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002230
Guido van Rossumc2e20742006-02-27 22:32:47 +00002231 x = TOP();
2232 u = SECOND();
2233 if (PyInt_Check(u) || u == Py_None) {
2234 u = v = w = Py_None;
2235 }
2236 else {
2237 v = THIRD();
2238 w = FOURTH();
2239 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002240 /* XXX Not the fastest way to call it... */
2241 x = PyObject_CallFunctionObjArgs(x, u, v, w, NULL);
2242 if (x == NULL)
2243 break; /* Go to error exit */
2244 if (u != Py_None && PyObject_IsTrue(x)) {
2245 /* There was an exception and a true return */
2246 Py_DECREF(x);
2247 x = TOP(); /* Again */
2248 STACKADJ(-3);
2249 Py_INCREF(Py_None);
2250 SET_TOP(Py_None);
2251 Py_DECREF(x);
2252 Py_DECREF(u);
2253 Py_DECREF(v);
2254 Py_DECREF(w);
2255 } else {
2256 /* Let END_FINALLY do its thing */
2257 Py_DECREF(x);
2258 x = POP();
2259 Py_DECREF(x);
2260 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002261 break;
2262 }
2263
Guido van Rossumf10570b1995-07-07 22:53:21 +00002264 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002265 {
2266 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002267 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002268 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002269#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002270 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002271#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002272 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002273#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002274 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002275 PUSH(x);
2276 if (x != NULL)
2277 continue;
2278 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002279 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002280
Jeremy Hylton76901512000-03-28 23:49:17 +00002281 case CALL_FUNCTION_VAR:
2282 case CALL_FUNCTION_KW:
2283 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002284 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002285 int na = oparg & 0xff;
2286 int nk = (oparg>>8) & 0xff;
2287 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002288 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002289 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002290 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002291 if (flags & CALL_FLAG_VAR)
2292 n++;
2293 if (flags & CALL_FLAG_KW)
2294 n++;
2295 pfunc = stack_pointer - n - 1;
2296 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002297
Guido van Rossumac7be682001-01-17 15:42:30 +00002298 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002299 && PyMethod_GET_SELF(func) != NULL) {
2300 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002301 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002302 func = PyMethod_GET_FUNCTION(func);
2303 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002304 Py_DECREF(*pfunc);
2305 *pfunc = self;
2306 na++;
2307 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002308 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002309 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002310 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002311 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002312 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002313 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002314 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002315 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002316
Jeremy Hylton76901512000-03-28 23:49:17 +00002317 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002318 w = POP();
2319 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002320 }
2321 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002322 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002323 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002324 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002325 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002326
Guido van Rossum681d79a1995-07-18 14:51:37 +00002327 case MAKE_FUNCTION:
2328 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002329 x = PyFunction_New(v, f->f_globals);
2330 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002331 /* XXX Maybe this should be a separate opcode? */
2332 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002333 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002334 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002335 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002336 x = NULL;
2337 break;
2338 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002339 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002340 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002341 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002342 }
2343 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002344 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002345 }
2346 PUSH(x);
2347 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002348
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002349 case MAKE_CLOSURE:
2350 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002351 v = POP(); /* code object */
2352 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002353 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002354 if (x != NULL) {
2355 v = POP();
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002356 err = PyFunction_SetClosure(x, v);
2357 Py_DECREF(v);
2358 }
2359 if (x != NULL && oparg > 0) {
2360 v = PyTuple_New(oparg);
2361 if (v == NULL) {
2362 Py_DECREF(x);
2363 x = NULL;
2364 break;
2365 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002366 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002367 w = POP();
2368 PyTuple_SET_ITEM(v, oparg, w);
2369 }
2370 err = PyFunction_SetDefaults(x, v);
2371 Py_DECREF(v);
2372 }
2373 PUSH(x);
2374 break;
2375 }
2376
Guido van Rossum8861b741996-07-30 16:49:37 +00002377 case BUILD_SLICE:
2378 if (oparg == 3)
2379 w = POP();
2380 else
2381 w = NULL;
2382 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002383 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002384 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002385 Py_DECREF(u);
2386 Py_DECREF(v);
2387 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002388 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002389 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002390 break;
2391
Fred Drakeef8ace32000-08-24 00:32:09 +00002392 case EXTENDED_ARG:
2393 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002394 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002395 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002396
Guido van Rossum374a9221991-04-04 10:40:29 +00002397 default:
2398 fprintf(stderr,
2399 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002400 PyCode_Addr2Line(f->f_code, f->f_lasti),
2401 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002402 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002403 why = WHY_EXCEPTION;
2404 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002405
2406#ifdef CASE_TOO_BIG
2407 }
2408#endif
2409
Guido van Rossum374a9221991-04-04 10:40:29 +00002410 } /* switch */
2411
2412 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002413
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002414 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002415
Guido van Rossum374a9221991-04-04 10:40:29 +00002416 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002417
Guido van Rossum374a9221991-04-04 10:40:29 +00002418 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002419 if (err == 0 && x != NULL) {
2420#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002421 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002422 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002423 fprintf(stderr,
2424 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002425 else {
2426#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002427 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002428 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002429#ifdef CHECKEXC
2430 }
2431#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002432 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002433 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002434 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002435 err = 0;
2436 }
2437
Guido van Rossum374a9221991-04-04 10:40:29 +00002438 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002439
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002440 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002441 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002442 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002443 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002444 why = WHY_EXCEPTION;
2445 }
2446 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002447#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002448 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002449 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002450 if (PyErr_Occurred()) {
Jeremy Hylton904ed862003-11-05 17:29:35 +00002451 char buf[1024];
2452 sprintf(buf, "Stack unwind with exception "
2453 "set and why=%d", why);
2454 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002455 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002456 }
2457#endif
2458
2459 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002460
Guido van Rossum374a9221991-04-04 10:40:29 +00002461 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002462 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002463
Fred Drake8f51f542001-10-04 14:48:42 +00002464 if (tstate->c_tracefunc != NULL)
2465 call_exc_trace(tstate->c_tracefunc,
2466 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002467 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002468
Guido van Rossum374a9221991-04-04 10:40:29 +00002469 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002470
Guido van Rossum374a9221991-04-04 10:40:29 +00002471 if (why == WHY_RERAISE)
2472 why = WHY_EXCEPTION;
2473
2474 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002475
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002476fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002477 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002478 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002479
Tim Peters8a5c3c72004-04-05 19:36:21 +00002480 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002481 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2482 /* For a continue inside a try block,
2483 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002484 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2485 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002486 why = WHY_NOT;
2487 JUMPTO(PyInt_AS_LONG(retval));
2488 Py_DECREF(retval);
2489 break;
2490 }
2491
Guido van Rossum374a9221991-04-04 10:40:29 +00002492 while (STACK_LEVEL() > b->b_level) {
2493 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002494 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002495 }
2496 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2497 why = WHY_NOT;
2498 JUMPTO(b->b_handler);
2499 break;
2500 }
2501 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002502 (b->b_type == SETUP_EXCEPT &&
2503 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002504 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002505 PyObject *exc, *val, *tb;
2506 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002507 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002508 val = Py_None;
2509 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002510 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002511 /* Make the raw exception data
2512 available to the handler,
2513 so a program can emulate the
2514 Python main loop. Don't do
2515 this for 'finally'. */
2516 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002517 PyErr_NormalizeException(
2518 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002519 set_exc_info(tstate,
2520 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002521 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002522 if (tb == NULL) {
2523 Py_INCREF(Py_None);
2524 PUSH(Py_None);
2525 } else
2526 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002527 PUSH(val);
2528 PUSH(exc);
2529 }
2530 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002531 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002532 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002533 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002534 PUSH(v);
2535 }
2536 why = WHY_NOT;
2537 JUMPTO(b->b_handler);
2538 break;
2539 }
2540 } /* unwind stack */
2541
2542 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002543
Guido van Rossum374a9221991-04-04 10:40:29 +00002544 if (why != WHY_NOT)
2545 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002546 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002547
Guido van Rossum374a9221991-04-04 10:40:29 +00002548 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002549
Tim Peters8a5c3c72004-04-05 19:36:21 +00002550 assert(why != WHY_YIELD);
2551 /* Pop remaining stack entries. */
2552 while (!EMPTY()) {
2553 v = POP();
2554 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002555 }
2556
Tim Peters8a5c3c72004-04-05 19:36:21 +00002557 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002558 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002559
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002560fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002561 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002562 if (tstate->c_tracefunc) {
2563 if (why == WHY_RETURN || why == WHY_YIELD) {
2564 if (call_trace(tstate->c_tracefunc,
2565 tstate->c_traceobj, f,
2566 PyTrace_RETURN, retval)) {
2567 Py_XDECREF(retval);
2568 retval = NULL;
2569 why = WHY_EXCEPTION;
2570 }
2571 }
2572 else if (why == WHY_EXCEPTION) {
2573 call_trace_protected(tstate->c_tracefunc,
2574 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002575 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002576 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002577 }
Fred Drake8f51f542001-10-04 14:48:42 +00002578 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002579 if (why == WHY_EXCEPTION)
2580 call_trace_protected(tstate->c_profilefunc,
2581 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002582 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002583 else if (call_trace(tstate->c_profilefunc,
2584 tstate->c_profileobj, f,
2585 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002586 Py_XDECREF(retval);
2587 retval = NULL;
2588 why = WHY_EXCEPTION;
2589 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002590 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002591 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002592
Tim Peters7df5e7f2006-05-26 23:14:37 +00002593 if (tstate->frame->f_exc_type != NULL)
2594 reset_exc_info(tstate);
2595 else {
2596 assert(tstate->frame->f_exc_value == NULL);
2597 assert(tstate->frame->f_exc_traceback == NULL);
2598 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002599
Tim Peters5ca576e2001-06-18 22:08:13 +00002600 /* pop frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +00002601 exit_eval_frame:
2602 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002603 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002604
Guido van Rossum96a42c81992-01-12 02:29:51 +00002605 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002606}
2607
Guido van Rossumc2e20742006-02-27 22:32:47 +00002608/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002609 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002610 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002611
Tim Peters6d6c1a32001-08-02 04:15:00 +00002612PyObject *
2613PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002614 PyObject **args, int argcount, PyObject **kws, int kwcount,
2615 PyObject **defs, int defcount, PyObject *closure)
2616{
2617 register PyFrameObject *f;
2618 register PyObject *retval = NULL;
2619 register PyObject **fastlocals, **freevars;
2620 PyThreadState *tstate = PyThreadState_GET();
2621 PyObject *x, *u;
2622
2623 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002624 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002625 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002626 return NULL;
2627 }
2628
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002629 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002630 assert(globals != NULL);
2631 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002632 if (f == NULL)
2633 return NULL;
2634
2635 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002636 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002637
2638 if (co->co_argcount > 0 ||
2639 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2640 int i;
2641 int n = argcount;
2642 PyObject *kwdict = NULL;
2643 if (co->co_flags & CO_VARKEYWORDS) {
2644 kwdict = PyDict_New();
2645 if (kwdict == NULL)
2646 goto fail;
2647 i = co->co_argcount;
2648 if (co->co_flags & CO_VARARGS)
2649 i++;
2650 SETLOCAL(i, kwdict);
2651 }
2652 if (argcount > co->co_argcount) {
2653 if (!(co->co_flags & CO_VARARGS)) {
2654 PyErr_Format(PyExc_TypeError,
2655 "%.200s() takes %s %d "
2656 "%sargument%s (%d given)",
2657 PyString_AsString(co->co_name),
2658 defcount ? "at most" : "exactly",
2659 co->co_argcount,
2660 kwcount ? "non-keyword " : "",
2661 co->co_argcount == 1 ? "" : "s",
2662 argcount);
2663 goto fail;
2664 }
2665 n = co->co_argcount;
2666 }
2667 for (i = 0; i < n; i++) {
2668 x = args[i];
2669 Py_INCREF(x);
2670 SETLOCAL(i, x);
2671 }
2672 if (co->co_flags & CO_VARARGS) {
2673 u = PyTuple_New(argcount - n);
2674 if (u == NULL)
2675 goto fail;
2676 SETLOCAL(co->co_argcount, u);
2677 for (i = n; i < argcount; i++) {
2678 x = args[i];
2679 Py_INCREF(x);
2680 PyTuple_SET_ITEM(u, i-n, x);
2681 }
2682 }
2683 for (i = 0; i < kwcount; i++) {
2684 PyObject *keyword = kws[2*i];
2685 PyObject *value = kws[2*i + 1];
2686 int j;
2687 if (keyword == NULL || !PyString_Check(keyword)) {
2688 PyErr_Format(PyExc_TypeError,
2689 "%.200s() keywords must be strings",
2690 PyString_AsString(co->co_name));
2691 goto fail;
2692 }
2693 /* XXX slow -- speed up using dictionary? */
2694 for (j = 0; j < co->co_argcount; j++) {
2695 PyObject *nm = PyTuple_GET_ITEM(
2696 co->co_varnames, j);
2697 int cmp = PyObject_RichCompareBool(
2698 keyword, nm, Py_EQ);
2699 if (cmp > 0)
2700 break;
2701 else if (cmp < 0)
2702 goto fail;
2703 }
2704 /* Check errors from Compare */
2705 if (PyErr_Occurred())
2706 goto fail;
2707 if (j >= co->co_argcount) {
2708 if (kwdict == NULL) {
2709 PyErr_Format(PyExc_TypeError,
2710 "%.200s() got an unexpected "
2711 "keyword argument '%.400s'",
2712 PyString_AsString(co->co_name),
2713 PyString_AsString(keyword));
2714 goto fail;
2715 }
2716 PyDict_SetItem(kwdict, keyword, value);
2717 }
2718 else {
2719 if (GETLOCAL(j) != NULL) {
2720 PyErr_Format(PyExc_TypeError,
2721 "%.200s() got multiple "
2722 "values for keyword "
2723 "argument '%.400s'",
2724 PyString_AsString(co->co_name),
2725 PyString_AsString(keyword));
2726 goto fail;
2727 }
2728 Py_INCREF(value);
2729 SETLOCAL(j, value);
2730 }
2731 }
2732 if (argcount < co->co_argcount) {
2733 int m = co->co_argcount - defcount;
2734 for (i = argcount; i < m; i++) {
2735 if (GETLOCAL(i) == NULL) {
2736 PyErr_Format(PyExc_TypeError,
2737 "%.200s() takes %s %d "
2738 "%sargument%s (%d given)",
2739 PyString_AsString(co->co_name),
2740 ((co->co_flags & CO_VARARGS) ||
2741 defcount) ? "at least"
2742 : "exactly",
2743 m, kwcount ? "non-keyword " : "",
2744 m == 1 ? "" : "s", i);
2745 goto fail;
2746 }
2747 }
2748 if (n > m)
2749 i = n - m;
2750 else
2751 i = 0;
2752 for (; i < defcount; i++) {
2753 if (GETLOCAL(m+i) == NULL) {
2754 PyObject *def = defs[i];
2755 Py_INCREF(def);
2756 SETLOCAL(m+i, def);
2757 }
2758 }
2759 }
2760 }
2761 else {
2762 if (argcount > 0 || kwcount > 0) {
2763 PyErr_Format(PyExc_TypeError,
2764 "%.200s() takes no arguments (%d given)",
2765 PyString_AsString(co->co_name),
2766 argcount + kwcount);
2767 goto fail;
2768 }
2769 }
2770 /* Allocate and initialize storage for cell vars, and copy free
2771 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00002772 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00002773 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002774 char *cellname, *argname;
2775 PyObject *c;
2776
2777 nargs = co->co_argcount;
2778 if (co->co_flags & CO_VARARGS)
2779 nargs++;
2780 if (co->co_flags & CO_VARKEYWORDS)
2781 nargs++;
2782
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002783 /* Initialize each cell var, taking into account
2784 cell vars that are initialized from arguments.
2785
2786 Should arrange for the compiler to put cellvars
2787 that are arguments at the beginning of the cellvars
2788 list so that we can march over it more efficiently?
2789 */
Richard Jonescebbefc2006-05-23 18:28:17 +00002790 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002791 cellname = PyString_AS_STRING(
2792 PyTuple_GET_ITEM(co->co_cellvars, i));
2793 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002794 for (j = 0; j < nargs; j++) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002795 argname = PyString_AS_STRING(
2796 PyTuple_GET_ITEM(co->co_varnames, j));
2797 if (strcmp(cellname, argname) == 0) {
2798 c = PyCell_New(GETLOCAL(j));
2799 if (c == NULL)
2800 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002801 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00002802 found = 1;
2803 break;
2804 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002805 }
2806 if (found == 0) {
2807 c = PyCell_New(NULL);
2808 if (c == NULL)
2809 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00002810 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00002811 }
2812 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002813 }
Richard Jonescebbefc2006-05-23 18:28:17 +00002814 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002815 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00002816 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002817 PyObject *o = PyTuple_GET_ITEM(closure, i);
2818 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00002819 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00002820 }
2821 }
2822
Tim Peters5ca576e2001-06-18 22:08:13 +00002823 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002824 /* Don't need to keep the reference to f_back, it will be set
2825 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00002826 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002827 f->f_back = NULL;
2828
Jeremy Hylton985eba52003-02-05 23:13:00 +00002829 PCALL(PCALL_GENERATOR);
2830
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00002831 /* Create a new generator that owns the ready to run frame
2832 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00002833 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00002834 }
2835
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00002836 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00002837
2838 fail: /* Jump here from prelude on failure */
2839
Tim Petersb13680b2001-11-27 23:29:29 +00002840 /* decref'ing the frame can cause __del__ methods to get invoked,
2841 which can call back into Python. While we're done with the
2842 current Python frame (f), the associated C stack is still in use,
2843 so recursion_depth must be boosted for the duration.
2844 */
2845 assert(tstate != NULL);
2846 ++tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002847 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00002848 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00002849 return retval;
2850}
2851
2852
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002853/* Implementation notes for set_exc_info() and reset_exc_info():
2854
2855- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
2856 'exc_traceback'. These always travel together.
2857
2858- tstate->curexc_ZZZ is the "hot" exception that is set by
2859 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
2860
2861- Once an exception is caught by an except clause, it is transferred
2862 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
2863 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00002864 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00002865
2866- Now let me explain the complicated dance with frame->f_exc_ZZZ.
2867
2868 Long ago, when none of this existed, there were just a few globals:
2869 one set corresponding to the "hot" exception, and one set
2870 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
2871 globals; they were simply stored as sys.exc_ZZZ. For backwards
2872 compatibility, they still are!) The problem was that in code like
2873 this:
2874
2875 try:
2876 "something that may fail"
2877 except "some exception":
2878 "do something else first"
2879 "print the exception from sys.exc_ZZZ."
2880
2881 if "do something else first" invoked something that raised and caught
2882 an exception, sys.exc_ZZZ were overwritten. That was a frequent
2883 cause of subtle bugs. I fixed this by changing the semantics as
2884 follows:
2885
2886 - Within one frame, sys.exc_ZZZ will hold the last exception caught
2887 *in that frame*.
2888
2889 - But initially, and as long as no exception is caught in a given
2890 frame, sys.exc_ZZZ will hold the last exception caught in the
2891 previous frame (or the frame before that, etc.).
2892
2893 The first bullet fixed the bug in the above example. The second
2894 bullet was for backwards compatibility: it was (and is) common to
2895 have a function that is called when an exception is caught, and to
2896 have that function access the caught exception via sys.exc_ZZZ.
2897 (Example: traceback.print_exc()).
2898
2899 At the same time I fixed the problem that sys.exc_ZZZ weren't
2900 thread-safe, by introducing sys.exc_info() which gets it from tstate;
2901 but that's really a separate improvement.
2902
2903 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
2904 variables to what they were before the current frame was called. The
2905 set_exc_info() function saves them on the frame so that
2906 reset_exc_info() can restore them. The invariant is that
2907 frame->f_exc_ZZZ is NULL iff the current frame never caught an
2908 exception (where "catching" an exception applies only to successful
2909 except clauses); and if the current frame ever caught an exception,
2910 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
2911 at the start of the current frame.
2912
2913*/
2914
Fredrik Lundh7a830892006-05-27 10:39:48 +00002915static void
Guido van Rossumac7be682001-01-17 15:42:30 +00002916set_exc_info(PyThreadState *tstate,
2917 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002918{
Tim Peters7df5e7f2006-05-26 23:14:37 +00002919 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002920 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00002921
Tim Peters7df5e7f2006-05-26 23:14:37 +00002922 assert(type != NULL);
2923 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002924 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002925 assert(frame->f_exc_value == NULL);
2926 assert(frame->f_exc_traceback == NULL);
2927 /* This frame didn't catch an exception before. */
2928 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002929 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00002930 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00002931 Py_INCREF(Py_None);
2932 tstate->exc_type = Py_None;
2933 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002934 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002935 Py_XINCREF(tstate->exc_value);
2936 Py_XINCREF(tstate->exc_traceback);
2937 frame->f_exc_type = tstate->exc_type;
2938 frame->f_exc_value = tstate->exc_value;
2939 frame->f_exc_traceback = tstate->exc_traceback;
2940 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00002941 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002942 tmp_type = tstate->exc_type;
2943 tmp_value = tstate->exc_value;
2944 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002945 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002946 Py_XINCREF(value);
2947 Py_XINCREF(tb);
2948 tstate->exc_type = type;
2949 tstate->exc_value = value;
2950 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002951 Py_XDECREF(tmp_type);
2952 Py_XDECREF(tmp_value);
2953 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002954 /* For b/w compatibility */
2955 PySys_SetObject("exc_type", type);
2956 PySys_SetObject("exc_value", value);
2957 PySys_SetObject("exc_traceback", tb);
2958}
2959
Fredrik Lundh7a830892006-05-27 10:39:48 +00002960static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00002961reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00002962{
2963 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002964 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002965
2966 /* It's a precondition that the thread state's frame caught an
2967 * exception -- verify in a debug build.
2968 */
2969 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002970 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00002971 assert(frame != NULL);
2972 assert(frame->f_exc_type != NULL);
2973
2974 /* Copy the frame's exception info back to the thread state. */
2975 tmp_type = tstate->exc_type;
2976 tmp_value = tstate->exc_value;
2977 tmp_tb = tstate->exc_traceback;
2978 Py_INCREF(frame->f_exc_type);
2979 Py_XINCREF(frame->f_exc_value);
2980 Py_XINCREF(frame->f_exc_traceback);
2981 tstate->exc_type = frame->f_exc_type;
2982 tstate->exc_value = frame->f_exc_value;
2983 tstate->exc_traceback = frame->f_exc_traceback;
2984 Py_XDECREF(tmp_type);
2985 Py_XDECREF(tmp_value);
2986 Py_XDECREF(tmp_tb);
2987
2988 /* For b/w compatibility */
2989 PySys_SetObject("exc_type", frame->f_exc_type);
2990 PySys_SetObject("exc_value", frame->f_exc_value);
2991 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
2992
2993 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00002994 tmp_type = frame->f_exc_type;
2995 tmp_value = frame->f_exc_value;
2996 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00002997 frame->f_exc_type = NULL;
2998 frame->f_exc_value = NULL;
2999 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003000 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003001 Py_XDECREF(tmp_value);
3002 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003003}
3004
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003005/* Logic for the raise statement (too complicated for inlining).
3006 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003007static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003008do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003009{
Guido van Rossumd295f121998-04-09 21:39:57 +00003010 if (type == NULL) {
3011 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003012 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003013 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3014 value = tstate->exc_value;
3015 tb = tstate->exc_traceback;
3016 Py_XINCREF(type);
3017 Py_XINCREF(value);
3018 Py_XINCREF(tb);
3019 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003020
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003021 /* We support the following forms of raise:
3022 raise <class>, <classinstance>
3023 raise <class>, <argument tuple>
3024 raise <class>, None
3025 raise <class>, <argument>
3026 raise <classinstance>, None
3027 raise <string>, <object>
3028 raise <string>, None
3029
3030 An omitted second argument is the same as None.
3031
3032 In addition, raise <tuple>, <anything> is the same as
3033 raising the tuple's first item (and it better have one!);
3034 this rule is applied recursively.
3035
3036 Finally, an optional third argument can be supplied, which
3037 gives the traceback to be substituted (useful when
3038 re-raising an exception after examining it). */
3039
3040 /* First, check the traceback argument, replacing None with
3041 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003042 if (tb == Py_None) {
3043 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003044 tb = NULL;
3045 }
3046 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003047 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003048 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003049 goto raise_error;
3050 }
3051
3052 /* Next, replace a missing value with None */
3053 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003054 value = Py_None;
3055 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003056 }
3057
3058 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003059 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3060 PyObject *tmp = type;
3061 type = PyTuple_GET_ITEM(type, 0);
3062 Py_INCREF(type);
3063 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003064 }
3065
Brett Cannona7446e32006-02-27 23:39:10 +00003066 if (PyString_CheckExact(type)) {
Tim Petersafb2c802002-04-18 18:06:20 +00003067 /* Raising builtin string is deprecated but still allowed --
3068 * do nothing. Raising an instance of a new-style str
3069 * subclass is right out. */
Brett Cannonbf364092006-03-01 04:25:17 +00003070 if (PyErr_Warn(PyExc_DeprecationWarning,
Brett Cannona7446e32006-02-27 23:39:10 +00003071 "raising a string exception is deprecated"))
3072 goto raise_error;
3073 }
Brett Cannonbf364092006-03-01 04:25:17 +00003074 else if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003075 PyErr_NormalizeException(&type, &value, &tb);
3076
Brett Cannonbf364092006-03-01 04:25:17 +00003077 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003078 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003079 if (value != Py_None) {
3080 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003081 "instance exception may not have a separate value");
3082 goto raise_error;
3083 }
3084 else {
3085 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003086 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003087 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003088 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003089 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003090 }
3091 }
3092 else {
3093 /* Not something you can raise. You get an exception
3094 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003095 PyErr_Format(PyExc_TypeError,
Neal Norwitz37aa0662003-01-10 15:31:15 +00003096 "exceptions must be classes, instances, or "
3097 "strings (deprecated), not %s",
3098 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003099 goto raise_error;
3100 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003101 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003102 if (tb == NULL)
3103 return WHY_EXCEPTION;
3104 else
3105 return WHY_RERAISE;
3106 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003107 Py_XDECREF(value);
3108 Py_XDECREF(type);
3109 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003110 return WHY_EXCEPTION;
3111}
3112
Tim Petersd6d010b2001-06-21 02:49:55 +00003113/* Iterate v argcnt times and store the results on the stack (via decreasing
3114 sp). Return 1 for success, 0 if error. */
3115
Fredrik Lundh7a830892006-05-27 10:39:48 +00003116static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003117unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003118{
Tim Petersd6d010b2001-06-21 02:49:55 +00003119 int i = 0;
3120 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003121 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003122
Tim Petersd6d010b2001-06-21 02:49:55 +00003123 assert(v != NULL);
3124
3125 it = PyObject_GetIter(v);
3126 if (it == NULL)
3127 goto Error;
3128
3129 for (; i < argcnt; i++) {
3130 w = PyIter_Next(it);
3131 if (w == NULL) {
3132 /* Iterator done, via error or exhaustion. */
3133 if (!PyErr_Occurred()) {
3134 PyErr_Format(PyExc_ValueError,
3135 "need more than %d value%s to unpack",
3136 i, i == 1 ? "" : "s");
3137 }
3138 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003139 }
3140 *--sp = w;
3141 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003142
3143 /* We better have exhausted the iterator now. */
3144 w = PyIter_Next(it);
3145 if (w == NULL) {
3146 if (PyErr_Occurred())
3147 goto Error;
3148 Py_DECREF(it);
3149 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003150 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003151 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003152 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003153 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003154Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003155 for (; i > 0; i--, sp++)
3156 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003157 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003158 return 0;
3159}
3160
3161
Guido van Rossum96a42c81992-01-12 02:29:51 +00003162#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003163static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003164prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003165{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003166 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003167 if (PyObject_Print(v, stdout, 0) != 0)
3168 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003169 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003170 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003171}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003172#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003173
Fredrik Lundh7a830892006-05-27 10:39:48 +00003174static void
Fred Drake5755ce62001-06-27 19:19:46 +00003175call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003176{
Guido van Rossumb209a111997-04-29 18:18:01 +00003177 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003178 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003179 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003180 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003181 value = Py_None;
3182 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003183 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003184 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003185 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003186 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003187 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003188 }
Fred Drake5755ce62001-06-27 19:19:46 +00003189 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003190 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003191 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003192 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003193 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003194 Py_XDECREF(type);
3195 Py_XDECREF(value);
3196 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003197 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003198}
3199
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +00003200static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003201call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003202 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003203{
3204 PyObject *type, *value, *traceback;
3205 int err;
3206 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003207 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003208 if (err == 0)
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +00003209 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003210 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +00003211 return 0;
3212 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003213 else {
3214 Py_XDECREF(type);
3215 Py_XDECREF(value);
3216 Py_XDECREF(traceback);
Amaury Forgeot d'Arcc572dc32007-11-13 22:43:05 +00003217 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003218 }
3219}
3220
Fredrik Lundh7a830892006-05-27 10:39:48 +00003221static int
Fred Drake5755ce62001-06-27 19:19:46 +00003222call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3223 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003224{
Fred Drake5755ce62001-06-27 19:19:46 +00003225 register PyThreadState *tstate = frame->f_tstate;
3226 int result;
3227 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003228 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003229 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003230 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003231 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003232 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3233 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003234 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003235 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003236}
3237
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003238PyObject *
3239_PyEval_CallTracing(PyObject *func, PyObject *args)
3240{
3241 PyFrameObject *frame = PyEval_GetFrame();
3242 PyThreadState *tstate = frame->f_tstate;
3243 int save_tracing = tstate->tracing;
3244 int save_use_tracing = tstate->use_tracing;
3245 PyObject *result;
3246
3247 tstate->tracing = 0;
3248 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3249 || (tstate->c_profilefunc != NULL));
3250 result = PyObject_Call(func, args, NULL);
3251 tstate->tracing = save_tracing;
3252 tstate->use_tracing = save_use_tracing;
3253 return result;
3254}
3255
Fredrik Lundh7a830892006-05-27 10:39:48 +00003256static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003257maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003258 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3259 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003260{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003261 int result = 0;
3262
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003263 /* If the last instruction executed isn't in the current
3264 instruction window, reset the window. If the last
3265 instruction happens to fall at the start of a line or if it
3266 represents a jump backwards, call the trace function.
3267 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003268 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003269 int line;
3270 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003271
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003272 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3273 &bounds);
3274 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003275 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003276 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003277 PyTrace_LINE, Py_None);
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003278 }
3279 *instr_lb = bounds.ap_lower;
3280 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003281 }
Armin Rigobf57a142004-03-22 19:24:58 +00003282 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003283 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003284 }
3285 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003286 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003287}
3288
Fred Drake5755ce62001-06-27 19:19:46 +00003289void
3290PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003291{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003292 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003293 PyObject *temp = tstate->c_profileobj;
3294 Py_XINCREF(arg);
3295 tstate->c_profilefunc = NULL;
3296 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003297 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003298 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003299 Py_XDECREF(temp);
3300 tstate->c_profilefunc = func;
3301 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003302 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003303 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003304}
3305
3306void
3307PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3308{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003309 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003310 PyObject *temp = tstate->c_traceobj;
3311 Py_XINCREF(arg);
3312 tstate->c_tracefunc = NULL;
3313 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003314 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003315 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003316 Py_XDECREF(temp);
3317 tstate->c_tracefunc = func;
3318 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003319 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003320 tstate->use_tracing = ((func != NULL)
3321 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003322}
3323
Guido van Rossumb209a111997-04-29 18:18:01 +00003324PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003325PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003326{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003327 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003328 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003329 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003330 else
3331 return current_frame->f_builtins;
3332}
3333
Guido van Rossumb209a111997-04-29 18:18:01 +00003334PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003335PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003336{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003337 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003338 if (current_frame == NULL)
3339 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003340 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003341 return current_frame->f_locals;
3342}
3343
Guido van Rossumb209a111997-04-29 18:18:01 +00003344PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003345PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003346{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003347 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003348 if (current_frame == NULL)
3349 return NULL;
3350 else
3351 return current_frame->f_globals;
3352}
3353
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003354PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003355PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003356{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003357 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003358 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003359}
3360
Guido van Rossum6135a871995-01-09 17:53:26 +00003361int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003362PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003363{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003364 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003365 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003366}
3367
Guido van Rossumbe270261997-05-22 22:26:18 +00003368int
Tim Peters5ba58662001-07-16 02:29:45 +00003369PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003370{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003371 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003372 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003373
3374 if (current_frame != NULL) {
3375 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003376 const int compilerflags = codeflags & PyCF_MASK;
3377 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003378 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003379 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003380 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003381#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003382 if (codeflags & CO_GENERATOR_ALLOWED) {
3383 result = 1;
3384 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3385 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003386#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003387 }
3388 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003389}
3390
3391int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003392Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003393{
Guido van Rossumb209a111997-04-29 18:18:01 +00003394 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003395 if (f == NULL)
3396 return 0;
3397 if (!PyFile_SoftSpace(f, 0))
3398 return 0;
3399 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003400}
3401
Guido van Rossum3f5da241990-12-20 15:06:42 +00003402
Guido van Rossum681d79a1995-07-18 14:51:37 +00003403/* External interface to call any callable object.
3404 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003405
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003406#undef PyEval_CallObject
3407/* for backward compatibility: export this interface */
3408
Guido van Rossumb209a111997-04-29 18:18:01 +00003409PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003410PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003411{
Guido van Rossumb209a111997-04-29 18:18:01 +00003412 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003413}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003414#define PyEval_CallObject(func,arg) \
3415 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003416
Guido van Rossumb209a111997-04-29 18:18:01 +00003417PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003418PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003419{
Jeremy Hylton52820442001-01-03 23:52:36 +00003420 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003421
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003422 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003423 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003424 if (arg == NULL)
3425 return NULL;
3426 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003427 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003428 PyErr_SetString(PyExc_TypeError,
3429 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003430 return NULL;
3431 }
3432 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003433 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003434
Guido van Rossumb209a111997-04-29 18:18:01 +00003435 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003436 PyErr_SetString(PyExc_TypeError,
3437 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003438 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003439 return NULL;
3440 }
3441
Tim Peters6d6c1a32001-08-02 04:15:00 +00003442 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003443 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003444 return result;
3445}
3446
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003447const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003448PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003449{
3450 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003451 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003452 else if (PyFunction_Check(func))
3453 return PyString_AsString(((PyFunctionObject*)func)->func_name);
3454 else if (PyCFunction_Check(func))
3455 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3456 else if (PyClass_Check(func))
3457 return PyString_AsString(((PyClassObject*)func)->cl_name);
3458 else if (PyInstance_Check(func)) {
3459 return PyString_AsString(
3460 ((PyInstanceObject*)func)->in_class->cl_name);
3461 } else {
3462 return func->ob_type->tp_name;
3463 }
3464}
3465
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003466const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003467PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003468{
3469 if (PyMethod_Check(func))
3470 return "()";
3471 else if (PyFunction_Check(func))
3472 return "()";
3473 else if (PyCFunction_Check(func))
3474 return "()";
3475 else if (PyClass_Check(func))
3476 return " constructor";
3477 else if (PyInstance_Check(func)) {
3478 return " instance";
3479 } else {
3480 return " object";
3481 }
3482}
3483
Fredrik Lundh7a830892006-05-27 10:39:48 +00003484static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003485err_args(PyObject *func, int flags, int nargs)
3486{
3487 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003488 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003489 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003490 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003491 nargs);
3492 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003493 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003494 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003495 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003496 nargs);
3497}
3498
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003499#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003500if (tstate->use_tracing && tstate->c_profilefunc) { \
3501 if (call_trace(tstate->c_profilefunc, \
3502 tstate->c_profileobj, \
3503 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003504 func)) { \
3505 x = NULL; \
3506 } \
3507 else { \
3508 x = call; \
3509 if (tstate->c_profilefunc != NULL) { \
3510 if (x == NULL) { \
3511 call_trace_protected(tstate->c_profilefunc, \
3512 tstate->c_profileobj, \
3513 tstate->frame, PyTrace_C_EXCEPTION, \
3514 func); \
3515 /* XXX should pass (type, value, tb) */ \
3516 } else { \
3517 if (call_trace(tstate->c_profilefunc, \
3518 tstate->c_profileobj, \
3519 tstate->frame, PyTrace_C_RETURN, \
3520 func)) { \
3521 Py_DECREF(x); \
3522 x = NULL; \
3523 } \
3524 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003525 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003526 } \
3527} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003528 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003529 }
3530
Fredrik Lundh7a830892006-05-27 10:39:48 +00003531static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003532call_function(PyObject ***pp_stack, int oparg
3533#ifdef WITH_TSC
3534 , uint64* pintr0, uint64* pintr1
3535#endif
3536 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003537{
3538 int na = oparg & 0xff;
3539 int nk = (oparg>>8) & 0xff;
3540 int n = na + 2 * nk;
3541 PyObject **pfunc = (*pp_stack) - n - 1;
3542 PyObject *func = *pfunc;
3543 PyObject *x, *w;
3544
Jeremy Hylton985eba52003-02-05 23:13:00 +00003545 /* Always dispatch PyCFunction first, because these are
3546 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003547 */
3548 if (PyCFunction_Check(func) && nk == 0) {
3549 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003550 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003551
3552 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003553 if (flags & (METH_NOARGS | METH_O)) {
3554 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3555 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003556 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003557 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003558 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003559 else if (flags & METH_O && na == 1) {
3560 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003561 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003562 Py_DECREF(arg);
3563 }
3564 else {
3565 err_args(func, flags, na);
3566 x = NULL;
3567 }
3568 }
3569 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003570 PyObject *callargs;
3571 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003572 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003573 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003574 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003575 Py_XDECREF(callargs);
3576 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003577 } else {
3578 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3579 /* optimize access to bound methods */
3580 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003581 PCALL(PCALL_METHOD);
3582 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003583 Py_INCREF(self);
3584 func = PyMethod_GET_FUNCTION(func);
3585 Py_INCREF(func);
3586 Py_DECREF(*pfunc);
3587 *pfunc = self;
3588 na++;
3589 n++;
3590 } else
3591 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003592 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003593 if (PyFunction_Check(func))
3594 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003595 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003596 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003597 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003598 Py_DECREF(func);
3599 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003600
Armin Rigod34fa522006-03-28 19:10:40 +00003601 /* Clear the stack of the function object. Also removes
3602 the arguments in case they weren't consumed already
3603 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003604 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003605 while ((*pp_stack) > pfunc) {
3606 w = EXT_POP(*pp_stack);
3607 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003608 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003609 }
3610 return x;
3611}
3612
Jeremy Hylton192690e2002-08-16 18:36:11 +00003613/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003614 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003615 For the simplest case -- a function that takes only positional
3616 arguments and is called with only positional arguments -- it
3617 inlines the most primitive frame setup code from
3618 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3619 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003620*/
3621
Fredrik Lundh7a830892006-05-27 10:39:48 +00003622static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003623fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003624{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003625 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003626 PyObject *globals = PyFunction_GET_GLOBALS(func);
3627 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3628 PyObject **d = NULL;
3629 int nd = 0;
3630
Jeremy Hylton985eba52003-02-05 23:13:00 +00003631 PCALL(PCALL_FUNCTION);
3632 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003633 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003634 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3635 PyFrameObject *f;
3636 PyObject *retval = NULL;
3637 PyThreadState *tstate = PyThreadState_GET();
3638 PyObject **fastlocals, **stack;
3639 int i;
3640
3641 PCALL(PCALL_FASTER_FUNCTION);
3642 assert(globals != NULL);
3643 /* XXX Perhaps we should create a specialized
3644 PyFrame_New() that doesn't take locals, but does
3645 take builtins without sanity checking them.
3646 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003647 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003648 f = PyFrame_New(tstate, co, globals, NULL);
3649 if (f == NULL)
3650 return NULL;
3651
3652 fastlocals = f->f_localsplus;
3653 stack = (*pp_stack) - n;
3654
3655 for (i = 0; i < n; i++) {
3656 Py_INCREF(*stack);
3657 fastlocals[i] = *stack++;
3658 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003659 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003660 ++tstate->recursion_depth;
3661 Py_DECREF(f);
3662 --tstate->recursion_depth;
3663 return retval;
3664 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003665 if (argdefs != NULL) {
3666 d = &PyTuple_GET_ITEM(argdefs, 0);
3667 nd = ((PyTupleObject *)argdefs)->ob_size;
3668 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003669 return PyEval_EvalCodeEx(co, globals,
3670 (PyObject *)NULL, (*pp_stack)-n, na,
3671 (*pp_stack)-2*nk, nk, d, nd,
3672 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003673}
3674
Fredrik Lundh7a830892006-05-27 10:39:48 +00003675static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003676update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3677 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003678{
3679 PyObject *kwdict = NULL;
3680 if (orig_kwdict == NULL)
3681 kwdict = PyDict_New();
3682 else {
3683 kwdict = PyDict_Copy(orig_kwdict);
3684 Py_DECREF(orig_kwdict);
3685 }
3686 if (kwdict == NULL)
3687 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003688 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003689 int err;
3690 PyObject *value = EXT_POP(*pp_stack);
3691 PyObject *key = EXT_POP(*pp_stack);
3692 if (PyDict_GetItem(kwdict, key) != NULL) {
Guido van Rossumac7be682001-01-17 15:42:30 +00003693 PyErr_Format(PyExc_TypeError,
Ka-Ping Yee20579702001-01-15 22:14:16 +00003694 "%.200s%s got multiple values "
Jeremy Hylton512a2372001-04-11 13:52:29 +00003695 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003696 PyEval_GetFuncName(func),
3697 PyEval_GetFuncDesc(func),
Jeremy Hylton512a2372001-04-11 13:52:29 +00003698 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003699 Py_DECREF(key);
3700 Py_DECREF(value);
3701 Py_DECREF(kwdict);
3702 return NULL;
3703 }
3704 err = PyDict_SetItem(kwdict, key, value);
3705 Py_DECREF(key);
3706 Py_DECREF(value);
3707 if (err) {
3708 Py_DECREF(kwdict);
3709 return NULL;
3710 }
3711 }
3712 return kwdict;
3713}
3714
Fredrik Lundh7a830892006-05-27 10:39:48 +00003715static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003716update_star_args(int nstack, int nstar, PyObject *stararg,
3717 PyObject ***pp_stack)
3718{
3719 PyObject *callargs, *w;
3720
3721 callargs = PyTuple_New(nstack + nstar);
3722 if (callargs == NULL) {
3723 return NULL;
3724 }
3725 if (nstar) {
3726 int i;
3727 for (i = 0; i < nstar; i++) {
3728 PyObject *a = PyTuple_GET_ITEM(stararg, i);
3729 Py_INCREF(a);
3730 PyTuple_SET_ITEM(callargs, nstack + i, a);
3731 }
3732 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003733 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003734 w = EXT_POP(*pp_stack);
3735 PyTuple_SET_ITEM(callargs, nstack, w);
3736 }
3737 return callargs;
3738}
3739
Fredrik Lundh7a830892006-05-27 10:39:48 +00003740static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003741load_args(PyObject ***pp_stack, int na)
3742{
3743 PyObject *args = PyTuple_New(na);
3744 PyObject *w;
3745
3746 if (args == NULL)
3747 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003748 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003749 w = EXT_POP(*pp_stack);
3750 PyTuple_SET_ITEM(args, na, w);
3751 }
3752 return args;
3753}
3754
Fredrik Lundh7a830892006-05-27 10:39:48 +00003755static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003756do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
3757{
3758 PyObject *callargs = NULL;
3759 PyObject *kwdict = NULL;
3760 PyObject *result = NULL;
3761
3762 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003763 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003764 if (kwdict == NULL)
3765 goto call_fail;
3766 }
3767 callargs = load_args(pp_stack, na);
3768 if (callargs == NULL)
3769 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003770#ifdef CALL_PROFILE
3771 /* At this point, we have to look at the type of func to
3772 update the call stats properly. Do it here so as to avoid
3773 exposing the call stats machinery outside ceval.c
3774 */
3775 if (PyFunction_Check(func))
3776 PCALL(PCALL_FUNCTION);
3777 else if (PyMethod_Check(func))
3778 PCALL(PCALL_METHOD);
3779 else if (PyType_Check(func))
3780 PCALL(PCALL_TYPE);
3781 else
3782 PCALL(PCALL_OTHER);
3783#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003784 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003785 call_fail:
3786 Py_XDECREF(callargs);
3787 Py_XDECREF(kwdict);
3788 return result;
3789}
3790
Fredrik Lundh7a830892006-05-27 10:39:48 +00003791static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00003792ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
3793{
3794 int nstar = 0;
3795 PyObject *callargs = NULL;
3796 PyObject *stararg = NULL;
3797 PyObject *kwdict = NULL;
3798 PyObject *result = NULL;
3799
3800 if (flags & CALL_FLAG_KW) {
3801 kwdict = EXT_POP(*pp_stack);
3802 if (!(kwdict && PyDict_Check(kwdict))) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003803 PyErr_Format(PyExc_TypeError,
Jeremy Hylton512a2372001-04-11 13:52:29 +00003804 "%s%s argument after ** "
3805 "must be a dictionary",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003806 PyEval_GetFuncName(func),
3807 PyEval_GetFuncDesc(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003808 goto ext_call_fail;
3809 }
3810 }
3811 if (flags & CALL_FLAG_VAR) {
3812 stararg = EXT_POP(*pp_stack);
3813 if (!PyTuple_Check(stararg)) {
3814 PyObject *t = NULL;
3815 t = PySequence_Tuple(stararg);
3816 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00003817 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
3818 PyErr_Format(PyExc_TypeError,
3819 "%s%s argument after * "
3820 "must be a sequence",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003821 PyEval_GetFuncName(func),
3822 PyEval_GetFuncDesc(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003823 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003824 goto ext_call_fail;
3825 }
3826 Py_DECREF(stararg);
3827 stararg = t;
3828 }
3829 nstar = PyTuple_GET_SIZE(stararg);
3830 }
3831 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00003832 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003833 if (kwdict == NULL)
3834 goto ext_call_fail;
3835 }
3836 callargs = update_star_args(na, nstar, stararg, pp_stack);
3837 if (callargs == NULL)
3838 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00003839#ifdef CALL_PROFILE
3840 /* At this point, we have to look at the type of func to
3841 update the call stats properly. Do it here so as to avoid
3842 exposing the call stats machinery outside ceval.c
3843 */
3844 if (PyFunction_Check(func))
3845 PCALL(PCALL_FUNCTION);
3846 else if (PyMethod_Check(func))
3847 PCALL(PCALL_METHOD);
3848 else if (PyType_Check(func))
3849 PCALL(PCALL_TYPE);
3850 else
3851 PCALL(PCALL_OTHER);
3852#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00003853 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00003854 ext_call_fail:
3855 Py_XDECREF(callargs);
3856 Py_XDECREF(kwdict);
3857 Py_XDECREF(stararg);
3858 return result;
3859}
3860
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003861/* Extract a slice index from a PyInt or PyLong or an object with the
3862 nb_index slot defined, and store in *pi.
3863 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
3864 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 +00003865 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00003866*/
Tim Petersb5196382001-12-16 19:44:20 +00003867/* Note: If v is NULL, return success without storing into *pi. This
3868 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
3869 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00003870*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00003871int
Martin v. Löwis18e16552006-02-15 17:27:45 +00003872_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003873{
Tim Petersb5196382001-12-16 19:44:20 +00003874 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003875 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00003876 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003877 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
3878 however, it looks like it should be AsSsize_t.
3879 There should be a comment here explaining why.
3880 */
3881 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00003882 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003883 else if (PyIndex_Check(v)) {
3884 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003885 if (x == -1 && PyErr_Occurred())
3886 return 0;
3887 }
3888 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003889 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003890 "slice indices must be integers or "
3891 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00003892 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003893 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00003894 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003895 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00003896 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003897}
3898
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003899#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00003900#define ISINDEX(x) ((x) == NULL || \
3901 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00003902
Fredrik Lundh7a830892006-05-27 10:39:48 +00003903static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003904apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003905{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003906 PyTypeObject *tp = u->ob_type;
3907 PySequenceMethods *sq = tp->tp_as_sequence;
3908
Guido van Rossum38fff8c2006-03-07 18:50:55 +00003909 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003910 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003911 if (!_PyEval_SliceIndex(v, &ilow))
3912 return NULL;
3913 if (!_PyEval_SliceIndex(w, &ihigh))
3914 return NULL;
3915 return PySequence_GetSlice(u, ilow, ihigh);
3916 }
3917 else {
3918 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00003919 if (slice != NULL) {
3920 PyObject *res = PyObject_GetItem(u, slice);
3921 Py_DECREF(slice);
3922 return res;
3923 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00003924 else
3925 return NULL;
3926 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003927}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003928
Fredrik Lundh7a830892006-05-27 10:39:48 +00003929static int
Guido van Rossumac7be682001-01-17 15:42:30 +00003930assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
3931 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003932{
Guido van Rossum50d756e2001-08-18 17:43:36 +00003933 PyTypeObject *tp = u->ob_type;
3934 PySequenceMethods *sq = tp->tp_as_sequence;
3935
Georg Brandl0ea89162007-03-05 22:28:13 +00003936 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00003937 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003938 if (!_PyEval_SliceIndex(v, &ilow))
3939 return -1;
3940 if (!_PyEval_SliceIndex(w, &ihigh))
3941 return -1;
3942 if (x == NULL)
3943 return PySequence_DelSlice(u, ilow, ihigh);
3944 else
3945 return PySequence_SetSlice(u, ilow, ihigh, x);
3946 }
3947 else {
3948 PyObject *slice = PySlice_New(v, w, NULL);
3949 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00003950 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003951 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00003952 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00003953 else
Guido van Rossum354797c2001-12-03 19:45:06 +00003954 res = PyObject_DelItem(u, slice);
3955 Py_DECREF(slice);
3956 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00003957 }
3958 else
3959 return -1;
3960 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003961}
3962
Fredrik Lundh7a830892006-05-27 10:39:48 +00003963static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003964cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003965{
Guido van Rossumac7be682001-01-17 15:42:30 +00003966 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003967 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00003968 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00003969 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003970 break;
3971 case PyCmp_IS_NOT:
3972 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003973 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003974 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003975 res = PySequence_Contains(w, v);
3976 if (res < 0)
3977 return NULL;
3978 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003979 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00003980 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00003981 if (res < 0)
3982 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00003983 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003984 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00003985 case PyCmp_EXC_MATCH:
Barry Warsaw4249f541997-08-22 21:26:19 +00003986 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003987 break;
3988 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00003989 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003990 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003991 v = res ? Py_True : Py_False;
3992 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003993 return v;
3994}
3995
Fredrik Lundh7a830892006-05-27 10:39:48 +00003996static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00003997import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00003998{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00003999 PyObject *x;
4000
4001 x = PyObject_GetAttr(v, name);
4002 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004003 PyErr_Format(PyExc_ImportError,
4004 "cannot import name %.230s",
4005 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004006 }
Thomas Wouters52152252000-08-17 22:55:00 +00004007 return x;
4008}
Guido van Rossumac7be682001-01-17 15:42:30 +00004009
Fredrik Lundh7a830892006-05-27 10:39:48 +00004010static int
Thomas Wouters52152252000-08-17 22:55:00 +00004011import_all_from(PyObject *locals, PyObject *v)
4012{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004013 PyObject *all = PyObject_GetAttrString(v, "__all__");
4014 PyObject *dict, *name, *value;
4015 int skip_leading_underscores = 0;
4016 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004017
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004018 if (all == NULL) {
4019 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4020 return -1; /* Unexpected error */
4021 PyErr_Clear();
4022 dict = PyObject_GetAttrString(v, "__dict__");
4023 if (dict == NULL) {
4024 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4025 return -1;
4026 PyErr_SetString(PyExc_ImportError,
4027 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004028 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004029 }
4030 all = PyMapping_Keys(dict);
4031 Py_DECREF(dict);
4032 if (all == NULL)
4033 return -1;
4034 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004035 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004036
4037 for (pos = 0, err = 0; ; pos++) {
4038 name = PySequence_GetItem(all, pos);
4039 if (name == NULL) {
4040 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4041 err = -1;
4042 else
4043 PyErr_Clear();
4044 break;
4045 }
4046 if (skip_leading_underscores &&
4047 PyString_Check(name) &&
4048 PyString_AS_STRING(name)[0] == '_')
4049 {
4050 Py_DECREF(name);
4051 continue;
4052 }
4053 value = PyObject_GetAttr(v, name);
4054 if (value == NULL)
4055 err = -1;
Armin Rigo1bc1ab22006-11-29 22:07:38 +00004056 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004057 err = PyDict_SetItem(locals, name, value);
Armin Rigo1bc1ab22006-11-29 22:07:38 +00004058 else
4059 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004060 Py_DECREF(name);
4061 Py_XDECREF(value);
4062 if (err != 0)
4063 break;
4064 }
4065 Py_DECREF(all);
4066 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004067}
4068
Fredrik Lundh7a830892006-05-27 10:39:48 +00004069static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004070build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004071{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004072 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004073
4074 if (PyDict_Check(methods))
4075 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004076 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004077 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004078 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4079 base = PyTuple_GET_ITEM(bases, 0);
4080 metaclass = PyObject_GetAttrString(base, "__class__");
4081 if (metaclass == NULL) {
4082 PyErr_Clear();
4083 metaclass = (PyObject *)base->ob_type;
4084 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004085 }
4086 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004087 else {
4088 PyObject *g = PyEval_GetGlobals();
4089 if (g != NULL && PyDict_Check(g))
4090 metaclass = PyDict_GetItemString(g, "__metaclass__");
4091 if (metaclass == NULL)
4092 metaclass = (PyObject *) &PyClass_Type;
4093 Py_INCREF(metaclass);
4094 }
Georg Brandl684fd0c2006-05-25 19:15:31 +00004095 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods, NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004096 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004097 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004098 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004099 in a base that was not a class (such the random module
4100 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004101 by augmenting the error message with more information.*/
4102
4103 PyObject *ptype, *pvalue, *ptraceback;
4104
4105 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
4106 if (PyString_Check(pvalue)) {
4107 PyObject *newmsg;
4108 newmsg = PyString_FromFormat(
4109 "Error when calling the metaclass bases\n %s",
4110 PyString_AS_STRING(pvalue));
4111 if (newmsg != NULL) {
4112 Py_DECREF(pvalue);
4113 pvalue = newmsg;
4114 }
4115 }
4116 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004117 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004118 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004119}
4120
Fredrik Lundh7a830892006-05-27 10:39:48 +00004121static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004122exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4123 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004124{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004125 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004126 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004127 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004128
Guido van Rossumb209a111997-04-29 18:18:01 +00004129 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4130 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004131 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004132 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004133 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004134 locals = PyTuple_GetItem(prog, 2);
4135 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004136 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004137 if (globals == Py_None) {
4138 globals = PyEval_GetGlobals();
4139 if (locals == Py_None) {
4140 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004141 plain = 1;
4142 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004143 if (!globals || !locals) {
4144 PyErr_SetString(PyExc_SystemError,
4145 "globals and locals cannot be NULL");
4146 return -1;
4147 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004148 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004149 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004150 locals = globals;
Guido van Rossumb209a111997-04-29 18:18:01 +00004151 if (!PyString_Check(prog) &&
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004152 !PyUnicode_Check(prog) &&
Guido van Rossumb209a111997-04-29 18:18:01 +00004153 !PyCode_Check(prog) &&
4154 !PyFile_Check(prog)) {
4155 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004156 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004157 return -1;
4158 }
Fred Drake661ea262000-10-24 19:57:45 +00004159 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004160 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004161 "exec: arg 2 must be a dictionary or None");
4162 return -1;
4163 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004164 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004165 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004166 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004167 return -1;
4168 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004169 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004170 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004171 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004172 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4173 PyErr_SetString(PyExc_TypeError,
4174 "code object passed to exec may not contain free variables");
4175 return -1;
4176 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004177 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004178 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004179 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004180 FILE *fp = PyFile_AsFile(prog);
4181 char *name = PyString_AsString(PyFile_Name(prog));
Tim Peters5ba58662001-07-16 02:29:45 +00004182 PyCompilerFlags cf;
Neal Norwitza5f5f142007-02-25 16:19:21 +00004183 if (name == NULL)
4184 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004185 cf.cf_flags = 0;
4186 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004187 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004188 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004189 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004190 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004191 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004192 }
4193 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004194 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004195 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004196 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004197 cf.cf_flags = 0;
4198#ifdef Py_USING_UNICODE
4199 if (PyUnicode_Check(prog)) {
4200 tmp = PyUnicode_AsUTF8String(prog);
4201 if (tmp == NULL)
4202 return -1;
4203 prog = tmp;
4204 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4205 }
4206#endif
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004207 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004208 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004209 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004210 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004211 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004212 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004213 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004214 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004215 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004216 if (plain)
4217 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004218 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004219 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004220 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004221 return 0;
4222}
Guido van Rossum24c13741995-02-14 09:42:43 +00004223
Fredrik Lundh7a830892006-05-27 10:39:48 +00004224static void
Paul Prescode68140d2000-08-30 20:25:01 +00004225format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4226{
4227 char *obj_str;
4228
4229 if (!obj)
4230 return;
4231
4232 obj_str = PyString_AsString(obj);
4233 if (!obj_str)
4234 return;
4235
4236 PyErr_Format(exc, format_str, obj_str);
4237}
Guido van Rossum950361c1997-01-24 13:49:28 +00004238
Fredrik Lundh7a830892006-05-27 10:39:48 +00004239static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004240string_concatenate(PyObject *v, PyObject *w,
4241 PyFrameObject *f, unsigned char *next_instr)
4242{
4243 /* This function implements 'variable += expr' when both arguments
4244 are strings. */
Armin Rigo97ff0472006-08-09 15:37:26 +00004245 Py_ssize_t v_len = PyString_GET_SIZE(v);
4246 Py_ssize_t w_len = PyString_GET_SIZE(w);
4247 Py_ssize_t new_len = v_len + w_len;
4248 if (new_len < 0) {
4249 PyErr_SetString(PyExc_OverflowError,
4250 "strings are too large to concat");
4251 return NULL;
4252 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004253
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004254 if (v->ob_refcnt == 2) {
4255 /* In the common case, there are 2 references to the value
4256 * stored in 'variable' when the += is performed: one on the
4257 * value stack (in 'v') and one still stored in the 'variable'.
4258 * We try to delete the variable now to reduce the refcnt to 1.
4259 */
4260 switch (*next_instr) {
4261 case STORE_FAST:
4262 {
4263 int oparg = PEEKARG();
4264 PyObject **fastlocals = f->f_localsplus;
4265 if (GETLOCAL(oparg) == v)
4266 SETLOCAL(oparg, NULL);
4267 break;
4268 }
4269 case STORE_DEREF:
4270 {
Richard Jonescebbefc2006-05-23 18:28:17 +00004271 PyObject **freevars = f->f_localsplus + f->f_code->co_nlocals;
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004272 PyObject *c = freevars[PEEKARG()];
4273 if (PyCell_GET(c) == v)
4274 PyCell_Set(c, NULL);
4275 break;
4276 }
4277 case STORE_NAME:
4278 {
4279 PyObject *names = f->f_code->co_names;
4280 PyObject *name = GETITEM(names, PEEKARG());
4281 PyObject *locals = f->f_locals;
4282 if (PyDict_CheckExact(locals) &&
4283 PyDict_GetItem(locals, name) == v) {
4284 if (PyDict_DelItem(locals, name) != 0) {
4285 PyErr_Clear();
4286 }
4287 }
4288 break;
4289 }
4290 }
4291 }
4292
Armin Rigo618fbf52004-08-07 20:58:32 +00004293 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004294 /* Now we own the last reference to 'v', so we can resize it
4295 * in-place.
4296 */
Armin Rigo97ff0472006-08-09 15:37:26 +00004297 if (_PyString_Resize(&v, new_len) != 0) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004298 /* XXX if _PyString_Resize() fails, 'v' has been
4299 * deallocated so it cannot be put back into 'variable'.
4300 * The MemoryError is raised when there is no value in
4301 * 'variable', which might (very remotely) be a cause
4302 * of incompatibilities.
4303 */
4304 return NULL;
4305 }
4306 /* copy 'w' into the newly allocated area of 'v' */
4307 memcpy(PyString_AS_STRING(v) + v_len,
4308 PyString_AS_STRING(w), w_len);
4309 return v;
4310 }
4311 else {
4312 /* When in-place resizing is not an option. */
4313 PyString_Concat(&v, w);
4314 return v;
4315 }
4316}
4317
Guido van Rossum950361c1997-01-24 13:49:28 +00004318#ifdef DYNAMIC_EXECUTION_PROFILE
4319
Fredrik Lundh7a830892006-05-27 10:39:48 +00004320static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004321getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004322{
4323 int i;
4324 PyObject *l = PyList_New(256);
4325 if (l == NULL) return NULL;
4326 for (i = 0; i < 256; i++) {
4327 PyObject *x = PyInt_FromLong(a[i]);
4328 if (x == NULL) {
4329 Py_DECREF(l);
4330 return NULL;
4331 }
4332 PyList_SetItem(l, i, x);
4333 }
4334 for (i = 0; i < 256; i++)
4335 a[i] = 0;
4336 return l;
4337}
4338
4339PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004340_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004341{
4342#ifndef DXPAIRS
4343 return getarray(dxp);
4344#else
4345 int i;
4346 PyObject *l = PyList_New(257);
4347 if (l == NULL) return NULL;
4348 for (i = 0; i < 257; i++) {
4349 PyObject *x = getarray(dxpairs[i]);
4350 if (x == NULL) {
4351 Py_DECREF(l);
4352 return NULL;
4353 }
4354 PyList_SetItem(l, i, x);
4355 }
4356 return l;
4357#endif
4358}
4359
4360#endif