blob: 4facda48102d2afbbd2985cd4ed1eeb21b57e355 [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
Thomas Wouterse2176022007-09-20 17:35:10 +000031 section should work for GCC on any PowerPC
32 platform, irrespective of OS.
33 POWER? Who knows :-) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000034
Michael W. Hudson75eabd22005-01-18 15:56:11 +000035#define READ_TIMESTAMP(var) ppc_getcounter(&var)
Michael W. Hudson800ba232004-08-12 18:19:17 +000036
Fredrik Lundh7a830892006-05-27 10:39:48 +000037static void
Michael W. Hudson800ba232004-08-12 18:19:17 +000038ppc_getcounter(uint64 *v)
39{
40 register unsigned long tbu, tb, tbu2;
41
42 loop:
43 asm volatile ("mftbu %0" : "=r" (tbu) );
44 asm volatile ("mftb %0" : "=r" (tb) );
45 asm volatile ("mftbu %0" : "=r" (tbu2));
46 if (__builtin_expect(tbu != tbu2, 0)) goto loop;
47
Tim Peters7df5e7f2006-05-26 23:14:37 +000048 /* The slightly peculiar way of writing the next lines is
Michael W. Hudson800ba232004-08-12 18:19:17 +000049 compiled better by GCC than any other way I tried. */
50 ((long*)(v))[0] = tbu;
51 ((long*)(v))[1] = tb;
52}
53
Michael W. Hudson75eabd22005-01-18 15:56:11 +000054#else /* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000055
Michael W. Hudson75eabd22005-01-18 15:56:11 +000056#define READ_TIMESTAMP(val) \
57 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000058
59#endif
60
Tim Peters7df5e7f2006-05-26 23:14:37 +000061void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000062 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
63{
64 uint64 intr, inst, loop;
65 PyThreadState *tstate = PyThreadState_Get();
66 if (!tstate->interp->tscdump)
67 return;
68 intr = intr1 - intr0;
69 inst = inst1 - inst0 - intr;
70 loop = loop1 - loop0 - intr;
71 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
72 opcode, ticked, inst, loop);
73}
Michael W. Hudson800ba232004-08-12 18:19:17 +000074
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000075#endif
76
Guido van Rossum04691fc1992-08-12 15:35:34 +000077/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000078/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000079
Guido van Rossum408027e1996-12-30 16:17:54 +000080#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000081/* For debugging the interpreter: */
82#define LLTRACE 1 /* Low-level trace feature */
83#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +000084#endif
85
Jeremy Hylton52820442001-01-03 23:52:36 +000086typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +000087
Guido van Rossum374a9221991-04-04 10:40:29 +000088/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000089#ifdef WITH_TSC
Fredrik Lundh7a830892006-05-27 10:39:48 +000090static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000091#else
Fredrik Lundh7a830892006-05-27 10:39:48 +000092static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +000094static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
95static PyObject * do_call(PyObject *, PyObject ***, int, int);
96static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouterse2176022007-09-20 17:35:10 +000097static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
98 PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +000099static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
100static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000101#define CALL_FLAG_VAR 1
102#define CALL_FLAG_KW 2
103
Guido van Rossum0a066c01992-03-27 17:29:15 +0000104#ifdef LLTRACE
Fredrik Lundh1b949402006-05-26 12:01:49 +0000105static int lltrace;
Fredrik Lundh7a830892006-05-27 10:39:48 +0000106static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000107#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +0000108static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Fred Drake5755ce62001-06-27 19:19:46 +0000109 int, PyObject *);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000110static int call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000111 PyFrameObject *, int, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000112static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
113static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Armin Rigobf57a142004-03-22 19:24:58 +0000114 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000115
Fredrik Lundh7a830892006-05-27 10:39:48 +0000116static PyObject * apply_slice(PyObject *, PyObject *, PyObject *);
117static int assign_slice(PyObject *, PyObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000118 PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000119static PyObject * cmp_outcome(int, PyObject *, PyObject *);
120static PyObject * import_from(PyObject *, PyObject *);
121static int import_all_from(PyObject *, PyObject *);
122static PyObject * build_class(PyObject *, PyObject *, PyObject *);
123static int exec_statement(PyFrameObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000124 PyObject *, PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000125static void set_exc_info(PyThreadState *, PyObject *, PyObject *, PyObject *);
126static void reset_exc_info(PyThreadState *);
127static void format_exc_check_arg(PyObject *, char *, PyObject *);
128static PyObject * string_concatenate(PyObject *, PyObject *,
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000129 PyFrameObject *, unsigned char *);
Benjamin Petersone18ef192009-01-20 14:21:16 +0000130static PyObject * kwd_as_string(PyObject *);
Guido van Rossum374a9221991-04-04 10:40:29 +0000131
Paul Prescode68140d2000-08-30 20:25:01 +0000132#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000133 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000134#define GLOBAL_NAME_ERROR_MSG \
135 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000136#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000137 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000138#define UNBOUNDFREE_ERROR_MSG \
139 "free variable '%.200s' referenced before assignment" \
140 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000141
Guido van Rossum950361c1997-01-24 13:49:28 +0000142/* Dynamic execution profile */
143#ifdef DYNAMIC_EXECUTION_PROFILE
144#ifdef DXPAIRS
145static long dxpairs[257][256];
146#define dxp dxpairs[256]
147#else
148static long dxp[256];
149#endif
150#endif
151
Jeremy Hylton985eba52003-02-05 23:13:00 +0000152/* Function call profile */
153#ifdef CALL_PROFILE
154#define PCALL_NUM 11
155static int pcall[PCALL_NUM];
156
157#define PCALL_ALL 0
158#define PCALL_FUNCTION 1
159#define PCALL_FAST_FUNCTION 2
160#define PCALL_FASTER_FUNCTION 3
161#define PCALL_METHOD 4
162#define PCALL_BOUND_METHOD 5
163#define PCALL_CFUNCTION 6
164#define PCALL_TYPE 7
165#define PCALL_GENERATOR 8
166#define PCALL_OTHER 9
167#define PCALL_POP 10
168
169/* Notes about the statistics
170
171 PCALL_FAST stats
172
173 FAST_FUNCTION means no argument tuple needs to be created.
174 FASTER_FUNCTION means that the fast-path frame setup code is used.
175
176 If there is a method call where the call can be optimized by changing
177 the argument tuple and calling the function directly, it gets recorded
178 twice.
179
180 As a result, the relationship among the statistics appears to be
181 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
182 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
183 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
184 PCALL_METHOD > PCALL_BOUND_METHOD
185*/
186
187#define PCALL(POS) pcall[POS]++
188
189PyObject *
190PyEval_GetCallStats(PyObject *self)
191{
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000192 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000193 pcall[0], pcall[1], pcall[2], pcall[3],
194 pcall[4], pcall[5], pcall[6], pcall[7],
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000195 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000196}
197#else
198#define PCALL(O)
199
200PyObject *
201PyEval_GetCallStats(PyObject *self)
202{
203 Py_INCREF(Py_None);
204 return Py_None;
205}
206#endif
207
Tim Peters5ca576e2001-06-18 22:08:13 +0000208
Guido van Rossume59214e1994-08-30 08:01:59 +0000209#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000210
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000211#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000212#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000213#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000214#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000215
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000216static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000217static PyThread_type_lock pending_lock = 0; /* for pending calls */
Guido van Rossuma9672091994-09-14 13:31:22 +0000218static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000219
Tim Peters7f468f22004-10-11 02:40:51 +0000220int
221PyEval_ThreadsInitialized(void)
222{
223 return interpreter_lock != 0;
224}
225
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000226void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000227PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000228{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000229 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000230 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000231 interpreter_lock = PyThread_allocate_lock();
232 PyThread_acquire_lock(interpreter_lock, 1);
233 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000234}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000235
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000236void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000237PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000238{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000239 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000240}
241
242void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000243PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000244{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000245 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000246}
247
248void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000249PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000250{
251 if (tstate == NULL)
252 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000253 /* Check someone has called PyEval_InitThreads() to create the lock */
254 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000255 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000256 if (PyThreadState_Swap(tstate) != NULL)
257 Py_FatalError(
258 "PyEval_AcquireThread: non-NULL old thread state");
259}
260
261void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000262PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000263{
264 if (tstate == NULL)
265 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
266 if (PyThreadState_Swap(NULL) != tstate)
267 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000268 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000269}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000270
271/* This function is called from PyOS_AfterFork to ensure that newly
272 created child processes don't hold locks referring to threads which
273 are not running in the child process. (This could also be done using
274 pthread_atfork mechanism, at least for the pthreads implementation.) */
275
276void
277PyEval_ReInitThreads(void)
278{
Jesse Noller5e62ca42008-07-16 20:03:47 +0000279 PyObject *threading, *result;
280 PyThreadState *tstate;
281
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000282 if (!interpreter_lock)
283 return;
284 /*XXX Can't use PyThread_free_lock here because it does too
285 much error-checking. Doing this cleanly would require
286 adding a new function to each thread_*.h. Instead, just
287 create a new lock and waste a little bit of memory */
288 interpreter_lock = PyThread_allocate_lock();
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000289 pending_lock = PyThread_allocate_lock();
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000290 PyThread_acquire_lock(interpreter_lock, 1);
291 main_thread = PyThread_get_thread_ident();
Jesse Noller5e62ca42008-07-16 20:03:47 +0000292
293 /* Update the threading module with the new state.
294 */
295 tstate = PyThreadState_GET();
296 threading = PyMapping_GetItemString(tstate->interp->modules,
297 "threading");
298 if (threading == NULL) {
299 /* threading not imported */
300 PyErr_Clear();
301 return;
302 }
303 result = PyObject_CallMethod(threading, "_after_fork", NULL);
304 if (result == NULL)
305 PyErr_WriteUnraisable(threading);
306 else
307 Py_DECREF(result);
308 Py_DECREF(threading);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000309}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000310#endif
311
Guido van Rossumff4949e1992-08-05 19:58:53 +0000312/* Functions save_thread and restore_thread are always defined so
313 dynamically loaded modules needn't be compiled separately for use
314 with and without threads: */
315
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000316PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000317PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000318{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000319 PyThreadState *tstate = PyThreadState_Swap(NULL);
320 if (tstate == NULL)
321 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000322#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000323 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000324 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000325#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000326 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000327}
328
329void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000330PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000331{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000332 if (tstate == NULL)
333 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000334#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000335 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000336 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000337 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000338 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000339 }
340#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000341 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000342}
343
344
Guido van Rossuma9672091994-09-14 13:31:22 +0000345/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
346 signal handlers or Mac I/O completion routines) can schedule calls
347 to a function to be called synchronously.
348 The synchronous function is called with one void* argument.
349 It should return 0 for success or -1 for failure -- failure should
350 be accompanied by an exception.
351
352 If registry succeeds, the registry function returns 0; if it fails
353 (e.g. due to too many pending calls) it returns -1 (without setting
354 an exception condition).
355
356 Note that because registry may occur from within signal handlers,
357 or other asynchronous events, calling malloc() is unsafe!
358
359#ifdef WITH_THREAD
360 Any thread can schedule pending calls, but only the main thread
361 will execute them.
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000362 There is no facility to schedule calls to a particular thread, but
363 that should be easy to change, should that ever be required. In
364 that case, the static variables here should go into the python
365 threadstate.
Guido van Rossuma9672091994-09-14 13:31:22 +0000366#endif
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000367*/
Guido van Rossuma9672091994-09-14 13:31:22 +0000368
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000369#ifdef WITH_THREAD
370
371/* The WITH_THREAD implementation is thread-safe. It allows
372 scheduling to be made from any thread, and even from an executing
373 callback.
374 */
375
376#define NPENDINGCALLS 32
377static struct {
378 int (*func)(void *);
379 void *arg;
380} pendingcalls[NPENDINGCALLS];
381static int pendingfirst = 0;
382static int pendinglast = 0;
383static volatile int pendingcalls_to_do = 1; /* trigger initialization of lock */
384static char pendingbusy = 0;
385
386int
387Py_AddPendingCall(int (*func)(void *), void *arg)
388{
389 int i, j, result=0;
390 PyThread_type_lock lock = pending_lock;
391
392 /* try a few times for the lock. Since this mechanism is used
393 * for signal handling (on the main thread), there is a (slim)
394 * chance that a signal is delivered on the same thread while we
395 * hold the lock during the Py_MakePendingCalls() function.
396 * This avoids a deadlock in that case.
397 * Note that signals can be delivered on any thread. In particular,
398 * on Windows, a SIGINT is delivered on a system-created worker
399 * thread.
400 * We also check for lock being NULL, in the unlikely case that
401 * this function is called before any bytecode evaluation takes place.
402 */
403 if (lock != NULL) {
404 for (i = 0; i<100; i++) {
405 if (PyThread_acquire_lock(lock, NOWAIT_LOCK))
406 break;
407 }
408 if (i == 100)
409 return -1;
410 }
411
412 i = pendinglast;
413 j = (i + 1) % NPENDINGCALLS;
414 if (j == pendingfirst) {
415 result = -1; /* Queue full */
416 } else {
417 pendingcalls[i].func = func;
418 pendingcalls[i].arg = arg;
419 pendinglast = j;
420 }
421 /* signal main loop */
422 _Py_Ticker = 0;
423 pendingcalls_to_do = 1;
424 if (lock != NULL)
425 PyThread_release_lock(lock);
426 return result;
427}
428
429int
430Py_MakePendingCalls(void)
431{
432 int i;
433 int r = 0;
434
435 if (!pending_lock) {
436 /* initial allocation of the lock */
437 pending_lock = PyThread_allocate_lock();
438 if (pending_lock == NULL)
439 return -1;
440 }
441
442 /* only service pending calls on main thread */
443 if (main_thread && PyThread_get_thread_ident() != main_thread)
444 return 0;
445 /* don't perform recursive pending calls */
446 if (pendingbusy)
447 return 0;
448 pendingbusy = 1;
449 /* perform a bounded number of calls, in case of recursion */
450 for (i=0; i<NPENDINGCALLS; i++) {
451 int j;
452 int (*func)(void *);
453 void *arg;
454
455 /* pop one item off the queue while holding the lock */
456 PyThread_acquire_lock(pending_lock, WAIT_LOCK);
457 j = pendingfirst;
458 if (j == pendinglast) {
459 func = NULL; /* Queue empty */
460 } else {
461 func = pendingcalls[j].func;
462 arg = pendingcalls[j].arg;
463 pendingfirst = (j + 1) % NPENDINGCALLS;
464 }
465 pendingcalls_to_do = pendingfirst != pendinglast;
466 PyThread_release_lock(pending_lock);
467 /* having released the lock, perform the callback */
468 if (func == NULL)
469 break;
470 r = func(arg);
471 if (r)
472 break;
473 }
474 pendingbusy = 0;
475 return r;
476}
477
478#else /* if ! defined WITH_THREAD */
479
480/*
481 WARNING! ASYNCHRONOUSLY EXECUTING CODE!
482 This code is used for signal handling in python that isn't built
483 with WITH_THREAD.
484 Don't use this implementation when Py_AddPendingCalls() can happen
485 on a different thread!
486
Guido van Rossuma9672091994-09-14 13:31:22 +0000487 There are two possible race conditions:
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000488 (1) nested asynchronous calls to Py_AddPendingCall()
489 (2) AddPendingCall() calls made while pending calls are being processed.
490
491 (1) is very unlikely because typically signal delivery
492 is blocked during signal handling. So it should be impossible.
493 (2) is a real possibility.
Guido van Rossuma9672091994-09-14 13:31:22 +0000494 The current code is safe against (2), but not against (1).
495 The safety against (2) is derived from the fact that only one
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000496 thread is present, interrupted by signals, and that the critical
497 section is protected with the "busy" variable. On Windows, which
498 delivers SIGINT on a system thread, this does not hold and therefore
499 Windows really shouldn't use this version.
500 The two threads could theoretically wiggle around the "busy" variable.
Guido van Rossuma027efa1997-05-05 20:56:21 +0000501*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000502
Guido van Rossuma9672091994-09-14 13:31:22 +0000503#define NPENDINGCALLS 32
504static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000505 int (*func)(void *);
506 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000507} pendingcalls[NPENDINGCALLS];
508static volatile int pendingfirst = 0;
509static volatile int pendinglast = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000510static volatile int pendingcalls_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000511
512int
Thomas Wouters334fb892000-07-25 12:56:38 +0000513Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000514{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000515 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000516 int i, j;
517 /* XXX Begin critical section */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000518 if (busy)
519 return -1;
520 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000521 i = pendinglast;
522 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000523 if (j == pendingfirst) {
524 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000525 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000526 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000527 pendingcalls[i].func = func;
528 pendingcalls[i].arg = arg;
529 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000530
531 _Py_Ticker = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000532 pendingcalls_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000533 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000534 /* XXX End critical section */
535 return 0;
536}
537
Guido van Rossum180d7b41994-09-29 09:45:57 +0000538int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000539Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000540{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000541 static int busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000542 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000543 return 0;
544 busy = 1;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000545 pendingcalls_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000546 for (;;) {
547 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000548 int (*func)(void *);
549 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000550 i = pendingfirst;
551 if (i == pendinglast)
552 break; /* Queue empty */
553 func = pendingcalls[i].func;
554 arg = pendingcalls[i].arg;
555 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000556 if (func(arg) < 0) {
557 busy = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000558 pendingcalls_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000559 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000560 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000561 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000562 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000563 return 0;
564}
565
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000566#endif /* WITH_THREAD */
567
Guido van Rossuma9672091994-09-14 13:31:22 +0000568
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000569/* The interpreter's recursion limit */
570
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000571#ifndef Py_DEFAULT_RECURSION_LIMIT
572#define Py_DEFAULT_RECURSION_LIMIT 1000
573#endif
574static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
575int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000576
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000577int
578Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000579{
580 return recursion_limit;
581}
582
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000583void
584Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000585{
586 recursion_limit = new_limit;
Thomas Woutersae406c62007-09-19 17:27:43 +0000587 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000588}
589
Armin Rigo2b3eb402003-10-28 12:05:48 +0000590/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
591 if the recursion_depth reaches _Py_CheckRecursionLimit.
592 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
593 to guarantee that _Py_CheckRecursiveCall() is regularly called.
594 Without USE_STACKCHECK, there is no need for this. */
595int
596_Py_CheckRecursiveCall(char *where)
597{
598 PyThreadState *tstate = PyThreadState_GET();
599
600#ifdef USE_STACKCHECK
601 if (PyOS_CheckStack()) {
602 --tstate->recursion_depth;
603 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
604 return -1;
605 }
606#endif
607 if (tstate->recursion_depth > recursion_limit) {
608 --tstate->recursion_depth;
609 PyErr_Format(PyExc_RuntimeError,
610 "maximum recursion depth exceeded%s",
611 where);
612 return -1;
613 }
Thomas Woutersae406c62007-09-19 17:27:43 +0000614 _Py_CheckRecursionLimit = recursion_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000615 return 0;
616}
617
Guido van Rossum374a9221991-04-04 10:40:29 +0000618/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000619enum why_code {
620 WHY_NOT = 0x0001, /* No error */
621 WHY_EXCEPTION = 0x0002, /* Exception occurred */
622 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
623 WHY_RETURN = 0x0008, /* 'return' statement */
624 WHY_BREAK = 0x0010, /* 'break' statement */
625 WHY_CONTINUE = 0x0020, /* 'continue' statement */
626 WHY_YIELD = 0x0040 /* 'yield' operator */
627};
Guido van Rossum374a9221991-04-04 10:40:29 +0000628
Fredrik Lundh7a830892006-05-27 10:39:48 +0000629static enum why_code do_raise(PyObject *, PyObject *, PyObject *);
630static int unpack_iterable(PyObject *, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000631
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +0000632/* Records whether tracing is on for any thread. Counts the number of
633 threads for which tstate->c_tracefunc is non-NULL, so if the value
634 is 0, we know we don't have to check this thread's c_tracefunc.
635 This speeds up the if statement in PyEval_EvalFrameEx() after
636 fast_next_opcode*/
637static int _Py_TracingPossible = 0;
638
Skip Montanarod581d772002-09-03 20:10:45 +0000639/* for manipulating the thread switch and periodic "stuff" - used to be
640 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000641int _Py_CheckInterval = 100;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000642volatile int _Py_Ticker = 0; /* so that we hit a "tick" first thing */
Guido van Rossum374a9221991-04-04 10:40:29 +0000643
Guido van Rossumb209a111997-04-29 18:18:01 +0000644PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000645PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000646{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000647 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000648 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000649 (PyObject **)NULL, 0,
650 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000651 (PyObject **)NULL, 0,
652 NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000653}
654
655
656/* Interpreter main loop */
657
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000658PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000659PyEval_EvalFrame(PyFrameObject *f) {
660 /* This is for backward compatibility with extension modules that
Thomas Wouterse2176022007-09-20 17:35:10 +0000661 used this API; core interpreter code should call
662 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000663 return PyEval_EvalFrameEx(f, 0);
664}
665
666PyObject *
Anthony Baxtera863d332006-04-11 07:43:46 +0000667PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000668{
Guido van Rossum950361c1997-01-24 13:49:28 +0000669#ifdef DXPAIRS
670 int lastopcode = 0;
671#endif
Thomas Wouterse2176022007-09-20 17:35:10 +0000672 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000673 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000674 register int opcode; /* Current opcode */
675 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000676 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000677 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000678 register PyObject *x; /* Result object -- NULL if error */
679 register PyObject *v; /* Temporary objects popped off stack */
680 register PyObject *w;
681 register PyObject *u;
682 register PyObject *t;
Barry Warsaw23c9ec82000-08-21 15:44:01 +0000683 register PyObject *stream = NULL; /* for PRINT opcodes */
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000684 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000685 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000686 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000687 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000688
Tim Peters8a5c3c72004-04-05 19:36:21 +0000689 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000690
691 not (instr_lb <= current_bytecode_offset < instr_ub)
692
Tim Peters8a5c3c72004-04-05 19:36:21 +0000693 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000694 initial values are such as to make this false the first
695 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000696 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000697
Guido van Rossumd076c731998-10-07 19:42:25 +0000698 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000699 PyObject *names;
700 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000701#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000702 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000703 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000704#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000705
Neal Norwitza81d2202002-07-14 00:27:26 +0000706/* Tuple access macros */
707
708#ifndef Py_DEBUG
709#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
710#else
711#define GETITEM(v, i) PyTuple_GetItem((v), (i))
712#endif
713
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000714#ifdef WITH_TSC
715/* Use Pentium timestamp counter to mark certain events:
716 inst0 -- beginning of switch statement for opcode dispatch
717 inst1 -- end of switch statement (may be skipped)
718 loop0 -- the top of the mainloop
Tim Peters7df5e7f2006-05-26 23:14:37 +0000719 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000720 (may be skipped)
721 intr1 -- beginning of long interruption
722 intr2 -- end of long interruption
723
724 Many opcodes call out to helper C functions. In some cases, the
725 time in those functions should be counted towards the time for the
726 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
727 calls another Python function; there's no point in charge all the
728 bytecode executed by the called function to the caller.
729
730 It's hard to make a useful judgement statically. In the presence
731 of operator overloading, it's impossible to tell if a call will
732 execute new Python code or not.
733
734 It's a case-by-case judgement. I'll use intr1 for the following
735 cases:
736
737 EXEC_STMT
738 IMPORT_STAR
739 IMPORT_FROM
740 CALL_FUNCTION (and friends)
741
742 */
743 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
744 int ticked = 0;
745
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000746 READ_TIMESTAMP(inst0);
747 READ_TIMESTAMP(inst1);
748 READ_TIMESTAMP(loop0);
749 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000750
751 /* shut up the compiler */
752 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000753#endif
754
Guido van Rossum374a9221991-04-04 10:40:29 +0000755/* Code access macros */
756
Martin v. Löwis18e16552006-02-15 17:27:45 +0000757#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000758#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000759#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000760#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000761#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000762#define JUMPBY(x) (next_instr += (x))
763
Raymond Hettingerf606f872003-03-16 03:11:04 +0000764/* OpCode prediction macros
Thomas Wouterse2176022007-09-20 17:35:10 +0000765 Some opcodes tend to come in pairs thus making it possible to
766 predict the second code when the first is run. For example,
767 COMPARE_OP is often followed by JUMP_IF_FALSE or JUMP_IF_TRUE. And,
768 those opcodes are often followed by a POP_TOP.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000769
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000770 Verifying the prediction costs a single high-speed test of a register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000771 variable against a constant. If the pairing was good, then the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000772 processor's own internal branch predication has a high likelihood of
773 success, resulting in a nearly zero-overhead transition to the
774 next opcode. A successful prediction saves a trip through the eval-loop
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000775 including its two unpredictable branches, the HAS_ARG test and the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000776 switch-case. Combined with the processor's internal branch prediction,
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000777 a successful PREDICT has the effect of making the two opcodes run as if
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000778 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000779
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000780 If collecting opcode statistics, your choices are to either keep the
781 predictions turned-on and interpret the results as if some opcodes
782 had been combined or turn-off predictions so that the opcode frequency
783 counter updates for both opcodes.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000784*/
785
Raymond Hettingera7216982004-02-08 19:59:27 +0000786#ifdef DYNAMIC_EXECUTION_PROFILE
787#define PREDICT(op) if (0) goto PRED_##op
788#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000789#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000790#endif
791
Raymond Hettingerf606f872003-03-16 03:11:04 +0000792#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000793#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000794
Guido van Rossum374a9221991-04-04 10:40:29 +0000795/* Stack manipulation macros */
796
Martin v. Löwis18e16552006-02-15 17:27:45 +0000797/* The stack can grow at most MAXINT deep, as co_nlocals and
798 co_stacksize are ints. */
799#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000800#define EMPTY() (STACK_LEVEL() == 0)
801#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000802#define SECOND() (stack_pointer[-2])
803#define THIRD() (stack_pointer[-3])
804#define FOURTH() (stack_pointer[-4])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000805#define SET_TOP(v) (stack_pointer[-1] = (v))
806#define SET_SECOND(v) (stack_pointer[-2] = (v))
807#define SET_THIRD(v) (stack_pointer[-3] = (v))
808#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000809#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000810#define BASIC_PUSH(v) (*stack_pointer++ = (v))
811#define BASIC_POP() (*--stack_pointer)
812
Guido van Rossum96a42c81992-01-12 02:29:51 +0000813#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000814#define PUSH(v) { (void)(BASIC_PUSH(v), \
815 lltrace && prtrace(TOP(), "push")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000816 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouterse2176022007-09-20 17:35:10 +0000817#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
818 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000819#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
820 lltrace && prtrace(TOP(), "stackadj")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000821 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes52729ac2007-12-14 02:33:57 +0000822#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
823 prtrace((STACK_POINTER)[-1], "ext_pop")), \
824 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000825#else
826#define PUSH(v) BASIC_PUSH(v)
827#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000828#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000829#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000830#endif
831
Guido van Rossum681d79a1995-07-18 14:51:37 +0000832/* Local variable macros */
833
834#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000835
836/* The SETLOCAL() macro must not DECREF the local variable in-place and
837 then store the new value; it must copy the old value to a temporary
838 value, then store the new value, and then DECREF the temporary value.
839 This is because it is possible that during the DECREF the frame is
840 accessed by other code (e.g. a __del__ method or gc.collect()) and the
841 variable would be pointing to already-freed memory. */
842#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
843 GETLOCAL(i) = value; \
844 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000845
Guido van Rossuma027efa1997-05-05 20:56:21 +0000846/* Start of code */
847
Tim Peters5ca576e2001-06-18 22:08:13 +0000848 if (f == NULL)
849 return NULL;
850
Armin Rigo1d313ab2003-10-25 14:33:09 +0000851 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000852 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000853 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000854
Tim Peters5ca576e2001-06-18 22:08:13 +0000855 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000856
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000857 if (tstate->use_tracing) {
858 if (tstate->c_tracefunc != NULL) {
859 /* tstate->c_tracefunc, if defined, is a
860 function that will be called on *every* entry
861 to a code block. Its return value, if not
862 None, is a function that will be called at
863 the start of each executed line of code.
864 (Actually, the function must return itself
865 in order to continue tracing.) The trace
866 functions are called with three arguments:
867 a pointer to the current frame, a string
868 indicating why the function is called, and
869 an argument which depends on the situation.
870 The global trace function is also called
871 whenever an exception is detected. */
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000872 if (call_trace_protected(tstate->c_tracefunc,
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000873 tstate->c_traceobj,
874 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000875 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000876 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000877 }
878 }
879 if (tstate->c_profilefunc != NULL) {
880 /* Similar for c_profilefunc, except it needn't
881 return itself and isn't called for "line" events */
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000882 if (call_trace_protected(tstate->c_profilefunc,
883 tstate->c_profileobj,
884 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000885 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000886 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000887 }
888 }
889 }
890
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000891 co = f->f_code;
892 names = co->co_names;
893 consts = co->co_consts;
894 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000895 freevars = f->f_localsplus + co->co_nlocals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000896 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000897 /* An explanation is in order for the next line.
898
899 f->f_lasti now refers to the index of the last instruction
900 executed. You might think this was obvious from the name, but
901 this wasn't always true before 2.3! PyFrame_New now sets
902 f->f_lasti to -1 (i.e. the index *before* the first instruction)
903 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000904 does work. Promise.
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000905
906 When the PREDICT() macros are enabled, some opcode pairs follow in
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000907 direct succession without updating f->f_lasti. A successful
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000908 prediction effectively links the two codes together as if they
909 were a single new opcode; accordingly,f->f_lasti will point to
910 the first code in the pair (for instance, GET_ITER followed by
911 FOR_ITER is effectively a single opcode and f->f_lasti will point
912 at to the beginning of the combined pair.)
913 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000914 next_instr = first_instr + f->f_lasti + 1;
915 stack_pointer = f->f_stacktop;
916 assert(stack_pointer != NULL);
917 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
918
Tim Peters5ca576e2001-06-18 22:08:13 +0000919#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000920 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000921#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000922#if defined(Py_DEBUG) || defined(LLTRACE)
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000923 filename = PyString_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000924#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000925
Guido van Rossum374a9221991-04-04 10:40:29 +0000926 why = WHY_NOT;
927 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000928 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000929 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000930
Anthony Baxtera863d332006-04-11 07:43:46 +0000931 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000932 why = WHY_EXCEPTION;
933 goto on_error;
934 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000935
Guido van Rossum374a9221991-04-04 10:40:29 +0000936 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000937#ifdef WITH_TSC
938 if (inst1 == 0) {
939 /* Almost surely, the opcode executed a break
940 or a continue, preventing inst1 from being set
941 on the way out of the loop.
942 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000943 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000944 loop1 = inst1;
945 }
946 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
947 intr0, intr1);
948 ticked = 0;
949 inst1 = 0;
950 intr0 = 0;
951 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000952 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000953#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000954 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000955 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000956
Guido van Rossuma027efa1997-05-05 20:56:21 +0000957 /* Do periodic things. Doing this every time through
958 the loop would add too much overhead, so we do it
959 only every Nth instruction. We also do it if
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000960 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
Guido van Rossuma027efa1997-05-05 20:56:21 +0000961 event needs attention (e.g. a signal handler or
962 async I/O handler); see Py_AddPendingCall() and
963 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000964
Skip Montanarod581d772002-09-03 20:10:45 +0000965 if (--_Py_Ticker < 0) {
Thomas Woutersae406c62007-09-19 17:27:43 +0000966 if (*next_instr == SETUP_FINALLY) {
967 /* Make the last opcode before
968 a try: finally: block uninterruptable. */
969 goto fast_next_opcode;
970 }
Skip Montanarod581d772002-09-03 20:10:45 +0000971 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000972 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000973#ifdef WITH_TSC
974 ticked = 1;
975#endif
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000976 if (pendingcalls_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000977 if (Py_MakePendingCalls() < 0) {
978 why = WHY_EXCEPTION;
979 goto on_error;
980 }
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000981 if (pendingcalls_to_do)
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +0000982 /* MakePendingCalls() didn't succeed.
983 Force early re-execution of this
984 "periodic" code, possibly after
985 a thread switch */
986 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +0000987 }
Guido van Rossume59214e1994-08-30 08:01:59 +0000988#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000989 if (interpreter_lock) {
990 /* Give another thread a chance */
991
Guido van Rossum25ce5661997-08-02 03:10:38 +0000992 if (PyThreadState_Swap(NULL) != tstate)
993 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000994 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000995
996 /* Other threads may run now */
997
Guido van Rossum65d5b571998-12-21 19:32:43 +0000998 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000999 if (PyThreadState_Swap(tstate) != NULL)
1000 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +00001001
1002 /* Check for thread interrupts */
1003
1004 if (tstate->async_exc != NULL) {
1005 x = tstate->async_exc;
1006 tstate->async_exc = NULL;
1007 PyErr_SetNone(x);
1008 Py_DECREF(x);
1009 why = WHY_EXCEPTION;
1010 goto on_error;
1011 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001012 }
1013#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001014 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001015
Neil Schemenauer63543862002-02-17 19:10:14 +00001016 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +00001017 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001018
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001019 /* line-by-line tracing support */
1020
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +00001021 if (_Py_TracingPossible &&
1022 tstate->c_tracefunc != NULL && !tstate->tracing) {
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001023 /* see maybe_call_line_trace
1024 for expository comments */
1025 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001026
Michael W. Hudson58ee2af2003-04-29 16:18:47 +00001027 err = maybe_call_line_trace(tstate->c_tracefunc,
1028 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +00001029 f, &instr_lb, &instr_ub,
1030 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001031 /* Reload possibly changed frame fields */
1032 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +00001033 if (f->f_stacktop != NULL) {
1034 stack_pointer = f->f_stacktop;
1035 f->f_stacktop = NULL;
1036 }
1037 if (err) {
1038 /* trace function raised an exception */
1039 goto on_error;
1040 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001041 }
1042
1043 /* Extract opcode and argument */
1044
Guido van Rossum374a9221991-04-04 10:40:29 +00001045 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +00001046 oparg = 0; /* allows oparg to be stored in a register because
1047 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001048 if (HAS_ARG(opcode))
1049 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00001050 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001051#ifdef DYNAMIC_EXECUTION_PROFILE
1052#ifdef DXPAIRS
1053 dxpairs[lastopcode][opcode]++;
1054 lastopcode = opcode;
1055#endif
1056 dxp[opcode]++;
1057#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001058
Guido van Rossum96a42c81992-01-12 02:29:51 +00001059#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +00001060 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001061
Guido van Rossum96a42c81992-01-12 02:29:51 +00001062 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001063 if (HAS_ARG(opcode)) {
1064 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001065 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001066 }
1067 else {
1068 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001069 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +00001070 }
1071 }
1072#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001073
Guido van Rossum374a9221991-04-04 10:40:29 +00001074 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001075 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001076
Guido van Rossum374a9221991-04-04 10:40:29 +00001077 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001078
Guido van Rossum374a9221991-04-04 10:40:29 +00001079 /* BEWARE!
1080 It is essential that any operation that fails sets either
1081 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1082 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001083
Guido van Rossum374a9221991-04-04 10:40:29 +00001084 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001085
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001086 case NOP:
1087 goto fast_next_opcode;
1088
Neil Schemenauer63543862002-02-17 19:10:14 +00001089 case LOAD_FAST:
1090 x = GETLOCAL(oparg);
1091 if (x != NULL) {
1092 Py_INCREF(x);
1093 PUSH(x);
1094 goto fast_next_opcode;
1095 }
1096 format_exc_check_arg(PyExc_UnboundLocalError,
1097 UNBOUNDLOCAL_ERROR_MSG,
1098 PyTuple_GetItem(co->co_varnames, oparg));
1099 break;
1100
1101 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +00001102 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +00001103 Py_INCREF(x);
1104 PUSH(x);
1105 goto fast_next_opcode;
1106
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001107 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +00001108 case STORE_FAST:
1109 v = POP();
1110 SETLOCAL(oparg, v);
1111 goto fast_next_opcode;
1112
Raymond Hettingerf606f872003-03-16 03:11:04 +00001113 PREDICTED(POP_TOP);
Guido van Rossum374a9221991-04-04 10:40:29 +00001114 case POP_TOP:
1115 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001116 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +00001117 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001118
Guido van Rossum374a9221991-04-04 10:40:29 +00001119 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001120 v = TOP();
1121 w = SECOND();
1122 SET_TOP(w);
1123 SET_SECOND(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001124 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001125
Guido van Rossum374a9221991-04-04 10:40:29 +00001126 case ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001127 v = TOP();
1128 w = SECOND();
1129 x = THIRD();
1130 SET_TOP(w);
1131 SET_SECOND(x);
1132 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001133 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001134
Thomas Wouters434d0822000-08-24 20:11:32 +00001135 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001136 u = TOP();
1137 v = SECOND();
1138 w = THIRD();
1139 x = FOURTH();
1140 SET_TOP(v);
1141 SET_SECOND(w);
1142 SET_THIRD(x);
1143 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001144 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001145
Guido van Rossum374a9221991-04-04 10:40:29 +00001146 case DUP_TOP:
1147 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001148 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001149 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001150 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001151
Thomas Wouters434d0822000-08-24 20:11:32 +00001152 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001153 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001154 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001155 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001156 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001157 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001158 STACKADJ(2);
1159 SET_TOP(x);
1160 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001161 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001162 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001163 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001164 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001165 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001166 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001167 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001168 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001169 STACKADJ(3);
1170 SET_TOP(x);
1171 SET_SECOND(w);
1172 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001173 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001174 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001175 Py_FatalError("invalid argument to DUP_TOPX"
1176 " (bytecode corruption?)");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001177 /* Never returns, so don't bother to set why. */
Tim Peters35ba6892000-10-11 07:04:49 +00001178 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001179
Guido van Rossum374a9221991-04-04 10:40:29 +00001180 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001181 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001182 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001183 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001184 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001185 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001186 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001187
Guido van Rossum374a9221991-04-04 10:40:29 +00001188 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001189 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001190 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001191 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001192 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001193 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001194 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001195
Guido van Rossum374a9221991-04-04 10:40:29 +00001196 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001197 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001198 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001199 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001200 if (err == 0) {
1201 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001202 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001203 continue;
1204 }
1205 else if (err > 0) {
1206 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001207 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001208 err = 0;
1209 continue;
1210 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001211 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001212 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001213
Guido van Rossum374a9221991-04-04 10:40:29 +00001214 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001215 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001216 x = PyObject_Repr(v);
1217 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001218 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001219 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001220 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001221
Guido van Rossum7928cd71991-10-24 14:59:31 +00001222 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001223 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001224 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001225 Py_DECREF(v);
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 Rossum50564e81996-01-12 01:13:16 +00001230 case BINARY_POWER:
1231 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001232 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001233 x = PyNumber_Power(v, w, Py_None);
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 Rossum50564e81996-01-12 01:13:16 +00001238 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001239
Guido van Rossum374a9221991-04-04 10:40:29 +00001240 case BINARY_MULTIPLY:
1241 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001242 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001243 x = PyNumber_Multiply(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 Rossum374a9221991-04-04 10:40:29 +00001248 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001249
Guido van Rossum374a9221991-04-04 10:40:29 +00001250 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001251 if (!_Py_QnewFlag) {
1252 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001253 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001254 x = PyNumber_Divide(v, w);
1255 Py_DECREF(v);
1256 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001257 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001258 if (x != NULL) continue;
1259 break;
1260 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001261 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001262 BINARY_TRUE_DIVIDE */
1263 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001264 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001265 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001266 x = PyNumber_TrueDivide(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001267 Py_DECREF(v);
1268 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001269 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001270 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001271 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001272
Guido van Rossum4668b002001-08-08 05:00:18 +00001273 case BINARY_FLOOR_DIVIDE:
1274 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001275 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001276 x = PyNumber_FloorDivide(v, w);
1277 Py_DECREF(v);
1278 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001279 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001280 if (x != NULL) continue;
1281 break;
1282
Guido van Rossum374a9221991-04-04 10:40:29 +00001283 case BINARY_MODULO:
1284 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001285 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001286 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001287 Py_DECREF(v);
1288 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001289 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001290 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001291 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001292
Guido van Rossum374a9221991-04-04 10:40:29 +00001293 case BINARY_ADD:
1294 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001295 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001296 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001297 /* INLINE: int + int */
1298 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001299 a = PyInt_AS_LONG(v);
1300 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001301 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001302 if ((i^a) < 0 && (i^b) < 0)
1303 goto slow_add;
1304 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001305 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001306 else if (PyString_CheckExact(v) &&
1307 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001308 x = string_concatenate(v, w, f, next_instr);
1309 /* string_concatenate consumed the ref to v */
1310 goto skip_decref_vx;
1311 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001312 else {
1313 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001314 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001315 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001316 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001317 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001318 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001319 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001320 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001321 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001322
Guido van Rossum374a9221991-04-04 10:40:29 +00001323 case BINARY_SUBTRACT:
1324 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001325 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001326 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001327 /* INLINE: int - int */
1328 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001329 a = PyInt_AS_LONG(v);
1330 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001331 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001332 if ((i^a) < 0 && (i^~b) < 0)
1333 goto slow_sub;
1334 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001335 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001336 else {
1337 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001338 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001339 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001340 Py_DECREF(v);
1341 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001342 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001343 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001344 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001345
Guido van Rossum374a9221991-04-04 10:40:29 +00001346 case BINARY_SUBSCR:
1347 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001348 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001349 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001350 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001351 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001352 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001353 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001354 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001355 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001356 Py_INCREF(x);
1357 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001358 else
1359 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001360 }
1361 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001362 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001363 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001364 Py_DECREF(v);
1365 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001366 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001367 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001368 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001369
Guido van Rossum7928cd71991-10-24 14:59:31 +00001370 case BINARY_LSHIFT:
1371 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001372 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001373 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001374 Py_DECREF(v);
1375 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001376 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001377 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001378 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001379
Guido van Rossum7928cd71991-10-24 14:59:31 +00001380 case BINARY_RSHIFT:
1381 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001382 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001383 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001384 Py_DECREF(v);
1385 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001386 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001387 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001388 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001389
Guido van Rossum7928cd71991-10-24 14:59:31 +00001390 case BINARY_AND:
1391 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001392 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001393 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001394 Py_DECREF(v);
1395 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001396 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001397 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001398 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001399
Guido van Rossum7928cd71991-10-24 14:59:31 +00001400 case BINARY_XOR:
1401 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001402 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001403 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001404 Py_DECREF(v);
1405 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001406 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001407 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001408 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001409
Guido van Rossum7928cd71991-10-24 14:59:31 +00001410 case BINARY_OR:
1411 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001412 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001413 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001414 Py_DECREF(v);
1415 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001416 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001417 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001418 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001419
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001420 case LIST_APPEND:
1421 w = POP();
Antoine Pitroud0c35152008-12-17 00:38:28 +00001422 v = stack_pointer[-oparg];
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001423 err = PyList_Append(v, w);
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001424 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001425 if (err == 0) {
1426 PREDICT(JUMP_ABSOLUTE);
1427 continue;
1428 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001429 break;
1430
Thomas Wouters434d0822000-08-24 20:11:32 +00001431 case INPLACE_POWER:
1432 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001433 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001434 x = PyNumber_InPlacePower(v, w, Py_None);
1435 Py_DECREF(v);
1436 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001437 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001438 if (x != NULL) continue;
1439 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001440
Thomas Wouters434d0822000-08-24 20:11:32 +00001441 case INPLACE_MULTIPLY:
1442 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001443 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001444 x = PyNumber_InPlaceMultiply(v, w);
1445 Py_DECREF(v);
1446 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001447 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001448 if (x != NULL) continue;
1449 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001450
Thomas Wouters434d0822000-08-24 20:11:32 +00001451 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001452 if (!_Py_QnewFlag) {
1453 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001454 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001455 x = PyNumber_InPlaceDivide(v, w);
1456 Py_DECREF(v);
1457 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001458 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001459 if (x != NULL) continue;
1460 break;
1461 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001462 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001463 INPLACE_TRUE_DIVIDE */
1464 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001465 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001466 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001467 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001468 Py_DECREF(v);
1469 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001470 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001471 if (x != NULL) continue;
1472 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001473
Guido van Rossum4668b002001-08-08 05:00:18 +00001474 case INPLACE_FLOOR_DIVIDE:
1475 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001476 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001477 x = PyNumber_InPlaceFloorDivide(v, w);
1478 Py_DECREF(v);
1479 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001480 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001481 if (x != NULL) continue;
1482 break;
1483
Thomas Wouters434d0822000-08-24 20:11:32 +00001484 case INPLACE_MODULO:
1485 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001486 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001487 x = PyNumber_InPlaceRemainder(v, w);
1488 Py_DECREF(v);
1489 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001490 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001491 if (x != NULL) continue;
1492 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001493
Thomas Wouters434d0822000-08-24 20:11:32 +00001494 case INPLACE_ADD:
1495 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001496 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001497 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001498 /* INLINE: int + int */
1499 register long a, b, i;
1500 a = PyInt_AS_LONG(v);
1501 b = PyInt_AS_LONG(w);
1502 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001503 if ((i^a) < 0 && (i^b) < 0)
1504 goto slow_iadd;
1505 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001506 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001507 else if (PyString_CheckExact(v) &&
1508 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001509 x = string_concatenate(v, w, f, next_instr);
1510 /* string_concatenate consumed the ref to v */
1511 goto skip_decref_v;
1512 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001513 else {
1514 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001515 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001516 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001517 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001518 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001519 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001520 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001521 if (x != NULL) continue;
1522 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001523
Thomas Wouters434d0822000-08-24 20:11:32 +00001524 case INPLACE_SUBTRACT:
1525 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001526 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001527 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001528 /* INLINE: int - int */
1529 register long a, b, i;
1530 a = PyInt_AS_LONG(v);
1531 b = PyInt_AS_LONG(w);
1532 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001533 if ((i^a) < 0 && (i^~b) < 0)
1534 goto slow_isub;
1535 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001536 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001537 else {
1538 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001539 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001540 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001541 Py_DECREF(v);
1542 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001543 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001544 if (x != NULL) continue;
1545 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001546
Thomas Wouters434d0822000-08-24 20:11:32 +00001547 case INPLACE_LSHIFT:
1548 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001549 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001550 x = PyNumber_InPlaceLshift(v, w);
1551 Py_DECREF(v);
1552 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001553 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001554 if (x != NULL) continue;
1555 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001556
Thomas Wouters434d0822000-08-24 20:11:32 +00001557 case INPLACE_RSHIFT:
1558 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001559 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001560 x = PyNumber_InPlaceRshift(v, w);
1561 Py_DECREF(v);
1562 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001563 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001564 if (x != NULL) continue;
1565 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001566
Thomas Wouters434d0822000-08-24 20:11:32 +00001567 case INPLACE_AND:
1568 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001569 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001570 x = PyNumber_InPlaceAnd(v, w);
1571 Py_DECREF(v);
1572 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001573 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001574 if (x != NULL) continue;
1575 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001576
Thomas Wouters434d0822000-08-24 20:11:32 +00001577 case INPLACE_XOR:
1578 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001579 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001580 x = PyNumber_InPlaceXor(v, w);
1581 Py_DECREF(v);
1582 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001583 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001584 if (x != NULL) continue;
1585 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001586
Thomas Wouters434d0822000-08-24 20:11:32 +00001587 case INPLACE_OR:
1588 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001589 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001590 x = PyNumber_InPlaceOr(v, w);
1591 Py_DECREF(v);
1592 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001593 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001594 if (x != NULL) continue;
1595 break;
1596
Guido van Rossum374a9221991-04-04 10:40:29 +00001597 case SLICE+0:
1598 case SLICE+1:
1599 case SLICE+2:
1600 case SLICE+3:
1601 if ((opcode-SLICE) & 2)
1602 w = POP();
1603 else
1604 w = NULL;
1605 if ((opcode-SLICE) & 1)
1606 v = POP();
1607 else
1608 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001609 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001610 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001611 Py_DECREF(u);
1612 Py_XDECREF(v);
1613 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001614 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001615 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001616 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001617
Guido van Rossum374a9221991-04-04 10:40:29 +00001618 case STORE_SLICE+0:
1619 case STORE_SLICE+1:
1620 case STORE_SLICE+2:
1621 case STORE_SLICE+3:
1622 if ((opcode-STORE_SLICE) & 2)
1623 w = POP();
1624 else
1625 w = NULL;
1626 if ((opcode-STORE_SLICE) & 1)
1627 v = POP();
1628 else
1629 v = NULL;
1630 u = POP();
1631 t = POP();
1632 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001633 Py_DECREF(t);
1634 Py_DECREF(u);
1635 Py_XDECREF(v);
1636 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001637 if (err == 0) continue;
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 DELETE_SLICE+0:
1641 case DELETE_SLICE+1:
1642 case DELETE_SLICE+2:
1643 case DELETE_SLICE+3:
1644 if ((opcode-DELETE_SLICE) & 2)
1645 w = POP();
1646 else
1647 w = NULL;
1648 if ((opcode-DELETE_SLICE) & 1)
1649 v = POP();
1650 else
1651 v = NULL;
1652 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001653 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001654 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001655 Py_DECREF(u);
1656 Py_XDECREF(v);
1657 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001658 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001659 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001660
Guido van Rossum374a9221991-04-04 10:40:29 +00001661 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001662 w = TOP();
1663 v = SECOND();
1664 u = THIRD();
1665 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001666 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001667 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001668 Py_DECREF(u);
1669 Py_DECREF(v);
1670 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001671 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001672 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001673
Guido van Rossum374a9221991-04-04 10:40:29 +00001674 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001675 w = TOP();
1676 v = SECOND();
1677 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001678 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001679 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001680 Py_DECREF(v);
1681 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001682 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001683 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001684
Guido van Rossum374a9221991-04-04 10:40:29 +00001685 case PRINT_EXPR:
1686 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001687 w = PySys_GetObject("displayhook");
1688 if (w == NULL) {
1689 PyErr_SetString(PyExc_RuntimeError,
1690 "lost sys.displayhook");
1691 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001692 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001693 }
1694 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001695 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001696 if (x == NULL)
1697 err = -1;
1698 }
1699 if (err == 0) {
1700 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001701 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001702 if (w == NULL)
1703 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001704 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001705 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001706 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001707 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001708
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001709 case PRINT_ITEM_TO:
1710 w = stream = POP();
1711 /* fall through to PRINT_ITEM */
1712
Guido van Rossum374a9221991-04-04 10:40:29 +00001713 case PRINT_ITEM:
1714 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001715 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001716 w = PySys_GetObject("stdout");
1717 if (w == NULL) {
1718 PyErr_SetString(PyExc_RuntimeError,
1719 "lost sys.stdout");
1720 err = -1;
1721 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001722 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001723 /* PyFile_SoftSpace() can exececute arbitrary code
1724 if sys.stdout is an instance with a __getattr__.
1725 If __getattr__ raises an exception, w will
1726 be freed, so we need to prevent that temporarily. */
1727 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001728 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001729 err = PyFile_WriteString(" ", w);
1730 if (err == 0)
1731 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001732 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001733 /* XXX move into writeobject() ? */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001734 if (PyString_Check(v)) {
1735 char *s = PyString_AS_STRING(v);
1736 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001737 if (len == 0 ||
1738 !isspace(Py_CHARMASK(s[len-1])) ||
1739 s[len-1] == ' ')
1740 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001741 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001742#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001743 else if (PyUnicode_Check(v)) {
1744 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001745 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001746 if (len == 0 ||
1747 !Py_UNICODE_ISSPACE(s[len-1]) ||
1748 s[len-1] == ' ')
1749 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001750 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001751#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001752 else
1753 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001754 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001755 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001756 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001757 Py_XDECREF(stream);
1758 stream = NULL;
1759 if (err == 0)
1760 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001761 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001762
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001763 case PRINT_NEWLINE_TO:
1764 w = stream = POP();
1765 /* fall through to PRINT_NEWLINE */
1766
Guido van Rossum374a9221991-04-04 10:40:29 +00001767 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001768 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001769 w = PySys_GetObject("stdout");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001770 if (w == NULL) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001771 PyErr_SetString(PyExc_RuntimeError,
1772 "lost sys.stdout");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001773 why = WHY_EXCEPTION;
1774 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001775 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001776 if (w != NULL) {
Georg Brandlaa76d772008-07-01 20:56:03 +00001777 /* w.write() may replace sys.stdout, so we
1778 * have to keep our reference to it */
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001779 Py_INCREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001780 err = PyFile_WriteString("\n", w);
1781 if (err == 0)
1782 PyFile_SoftSpace(w, 0);
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001783 Py_DECREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001784 }
1785 Py_XDECREF(stream);
1786 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001787 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001788
Thomas Wouters434d0822000-08-24 20:11:32 +00001789
1790#ifdef CASE_TOO_BIG
1791 default: switch (opcode) {
1792#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001793 case RAISE_VARARGS:
1794 u = v = w = NULL;
1795 switch (oparg) {
1796 case 3:
1797 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001798 /* Fallthrough */
1799 case 2:
1800 v = POP(); /* value */
1801 /* Fallthrough */
1802 case 1:
1803 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001804 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001805 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001806 break;
1807 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001808 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001809 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001810 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001811 break;
1812 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001813 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001814
Guido van Rossum374a9221991-04-04 10:40:29 +00001815 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001816 if ((x = f->f_locals) != NULL) {
1817 Py_INCREF(x);
1818 PUSH(x);
1819 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001820 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001821 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001822 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001823
Guido van Rossum374a9221991-04-04 10:40:29 +00001824 case RETURN_VALUE:
1825 retval = POP();
1826 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001827 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001828
Tim Peters5ca576e2001-06-18 22:08:13 +00001829 case YIELD_VALUE:
1830 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001831 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001832 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001833 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001834
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001835 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001836 w = TOP();
1837 v = SECOND();
1838 u = THIRD();
1839 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001840 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001841 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001842 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001843 Py_DECREF(u);
1844 Py_DECREF(v);
1845 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001846 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001847
Guido van Rossum374a9221991-04-04 10:40:29 +00001848 case POP_BLOCK:
1849 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001850 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001851 while (STACK_LEVEL() > b->b_level) {
1852 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001853 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001854 }
1855 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001856 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001857
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00001858 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001859 case END_FINALLY:
1860 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001861 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001862 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001863 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001864 if (why == WHY_RETURN ||
1865 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001866 retval = POP();
1867 }
Thomas Wouterse2176022007-09-20 17:35:10 +00001868 else if (PyExceptionClass_Check(v) ||
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001869 PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001870 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001871 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001872 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001873 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001874 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001875 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001876 else if (v != Py_None) {
1877 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001878 "'finally' pops bad exception");
1879 why = WHY_EXCEPTION;
1880 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001881 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001882 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001883
Guido van Rossum374a9221991-04-04 10:40:29 +00001884 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001885 u = TOP();
1886 v = SECOND();
1887 w = THIRD();
1888 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001889 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001890 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001891 Py_DECREF(u);
1892 Py_DECREF(v);
1893 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001894 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001895
Guido van Rossum374a9221991-04-04 10:40:29 +00001896 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001897 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001898 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001899 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001900 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001901 err = PyDict_SetItem(x, w, v);
1902 else
1903 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001904 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001905 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001906 break;
1907 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001908 PyErr_Format(PyExc_SystemError,
1909 "no locals found when storing %s",
1910 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001911 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001912
Guido van Rossum374a9221991-04-04 10:40:29 +00001913 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001914 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001915 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001916 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001917 format_exc_check_arg(PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001918 NAME_ERROR_MSG,
1919 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001920 break;
1921 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001922 PyErr_Format(PyExc_SystemError,
1923 "no locals when deleting %s",
1924 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001925 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001926
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001927 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001928 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001929 v = POP();
Thomas Wouterse2176022007-09-20 17:35:10 +00001930 if (PyTuple_CheckExact(v) &&
1931 PyTuple_GET_SIZE(v) == oparg) {
1932 PyObject **items = \
1933 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001934 while (oparg--) {
1935 w = items[oparg];
1936 Py_INCREF(w);
1937 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001938 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001939 Py_DECREF(v);
1940 continue;
Thomas Wouterse2176022007-09-20 17:35:10 +00001941 } else if (PyList_CheckExact(v) &&
1942 PyList_GET_SIZE(v) == oparg) {
1943 PyObject **items = \
1944 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001945 while (oparg--) {
1946 w = items[oparg];
1947 Py_INCREF(w);
1948 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001949 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001950 } else if (unpack_iterable(v, oparg,
Thomas Wouterse2176022007-09-20 17:35:10 +00001951 stack_pointer + oparg)) {
Tim Petersd6d010b2001-06-21 02:49:55 +00001952 stack_pointer += oparg;
Georg Brandl5cb76c12007-03-21 09:00:39 +00001953 } else {
1954 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001955 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001956 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001957 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001958 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001959
Guido van Rossum374a9221991-04-04 10:40:29 +00001960 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001961 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001962 v = TOP();
1963 u = SECOND();
1964 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001965 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1966 Py_DECREF(v);
1967 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001968 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001969 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001970
Guido van Rossum374a9221991-04-04 10:40:29 +00001971 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001972 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001973 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00001974 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
1975 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00001976 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001977 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001978
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001979 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001980 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001981 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001982 err = PyDict_SetItem(f->f_globals, w, v);
1983 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001984 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001985 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001986
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001987 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00001988 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00001989 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00001990 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00001991 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00001992 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001993
Guido van Rossum374a9221991-04-04 10:40:29 +00001994 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001995 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001996 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00001997 PyErr_Format(PyExc_SystemError,
1998 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00001999 PyObject_REPR(w));
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002000 why = WHY_EXCEPTION;
Guido van Rossum681d79a1995-07-18 14:51:37 +00002001 break;
2002 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002003 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002004 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002005 Py_XINCREF(x);
2006 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002007 else {
2008 x = PyObject_GetItem(v, w);
2009 if (x == NULL && PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002010 if (!PyErr_ExceptionMatches(
2011 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002012 break;
2013 PyErr_Clear();
2014 }
2015 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002016 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002017 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002018 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002019 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002020 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00002021 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00002022 PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00002023 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002024 break;
2025 }
2026 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002027 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00002028 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002029 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002030 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002031
Guido van Rossum374a9221991-04-04 10:40:29 +00002032 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00002033 w = GETITEM(names, oparg);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002034 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00002035 /* Inline the PyDict_GetItem() calls.
2036 WARNING: this is an extreme speed hack.
2037 Do not try this at home. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002038 long hash = ((PyStringObject *)w)->ob_shash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002039 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002040 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00002041 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002042 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00002043 e = d->ma_lookup(d, w, hash);
2044 if (e == NULL) {
2045 x = NULL;
2046 break;
2047 }
2048 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002049 if (x != NULL) {
2050 Py_INCREF(x);
2051 PUSH(x);
2052 continue;
2053 }
2054 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00002055 e = d->ma_lookup(d, w, hash);
2056 if (e == NULL) {
2057 x = NULL;
2058 break;
2059 }
2060 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002061 if (x != NULL) {
2062 Py_INCREF(x);
2063 PUSH(x);
2064 continue;
2065 }
2066 goto load_global_error;
2067 }
2068 }
2069 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00002070 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002071 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002072 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002073 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002074 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00002075 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00002076 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002077 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002078 break;
2079 }
2080 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002081 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00002082 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00002083 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00002084
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00002085 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00002086 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002087 if (x != NULL) {
2088 SETLOCAL(oparg, NULL);
2089 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00002090 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00002091 format_exc_check_arg(
2092 PyExc_UnboundLocalError,
2093 UNBOUNDLOCAL_ERROR_MSG,
2094 PyTuple_GetItem(co->co_varnames, oparg)
2095 );
2096 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002097
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002098 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002099 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002100 Py_INCREF(x);
2101 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00002102 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002103 break;
2104
2105 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002106 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002107 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002108 if (w != NULL) {
2109 PUSH(w);
2110 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00002111 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00002112 err = -1;
2113 /* Don't stomp existing exception */
2114 if (PyErr_Occurred())
2115 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00002116 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
2117 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00002118 oparg);
2119 format_exc_check_arg(
2120 PyExc_UnboundLocalError,
2121 UNBOUNDLOCAL_ERROR_MSG,
2122 v);
2123 } else {
Thomas Wouterse2176022007-09-20 17:35:10 +00002124 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
2125 PyTuple_GET_SIZE(co->co_cellvars));
2126 format_exc_check_arg(PyExc_NameError,
2127 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002128 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002129 break;
2130
2131 case STORE_DEREF:
2132 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002133 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002134 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00002135 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002136 continue;
2137
Guido van Rossum374a9221991-04-04 10:40:29 +00002138 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00002139 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002140 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002141 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002142 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002143 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002144 }
2145 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002146 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002147 }
2148 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002149
Guido van Rossum374a9221991-04-04 10:40:29 +00002150 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00002151 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002152 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002153 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002154 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00002155 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002156 }
2157 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002158 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002159 }
2160 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002161
Guido van Rossum374a9221991-04-04 10:40:29 +00002162 case BUILD_MAP:
Raymond Hettingerfd7ed402007-12-18 21:24:09 +00002163 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002164 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002165 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002166 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002167
Raymond Hettingereffde122007-12-18 18:26:18 +00002168 case STORE_MAP:
2169 w = TOP(); /* key */
2170 u = SECOND(); /* value */
2171 v = THIRD(); /* dict */
2172 STACKADJ(-2);
2173 assert (PyDict_CheckExact(v));
2174 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2175 Py_DECREF(u);
2176 Py_DECREF(w);
2177 if (err == 0) continue;
2178 break;
2179
Guido van Rossum374a9221991-04-04 10:40:29 +00002180 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00002181 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002182 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002183 x = PyObject_GetAttr(v, w);
2184 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002185 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002186 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002187 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002188
Guido van Rossum374a9221991-04-04 10:40:29 +00002189 case COMPARE_OP:
2190 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002191 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00002192 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002193 /* INLINE: cmp(int, int) */
2194 register long a, b;
2195 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002196 a = PyInt_AS_LONG(v);
2197 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002198 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002199 case PyCmp_LT: res = a < b; break;
2200 case PyCmp_LE: res = a <= b; break;
2201 case PyCmp_EQ: res = a == b; break;
2202 case PyCmp_NE: res = a != b; break;
2203 case PyCmp_GT: res = a > b; break;
2204 case PyCmp_GE: res = a >= b; break;
2205 case PyCmp_IS: res = v == w; break;
2206 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002207 default: goto slow_compare;
2208 }
2209 x = res ? Py_True : Py_False;
2210 Py_INCREF(x);
2211 }
2212 else {
2213 slow_compare:
2214 x = cmp_outcome(oparg, v, w);
2215 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002216 Py_DECREF(v);
2217 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002218 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002219 if (x == NULL) break;
2220 PREDICT(JUMP_IF_FALSE);
2221 PREDICT(JUMP_IF_TRUE);
2222 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002223
Guido van Rossum374a9221991-04-04 10:40:29 +00002224 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002225 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002226 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002227 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002228 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002229 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002230 break;
2231 }
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002232 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002233 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002234 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002235 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2236 w = PyTuple_Pack(5,
2237 w,
2238 f->f_globals,
2239 f->f_locals == NULL ?
2240 Py_None : f->f_locals,
2241 v,
2242 u);
2243 else
2244 w = PyTuple_Pack(4,
2245 w,
2246 f->f_globals,
2247 f->f_locals == NULL ?
2248 Py_None : f->f_locals,
2249 v);
2250 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002251 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002252 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002253 u = POP();
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002254 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002255 x = NULL;
2256 break;
2257 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002258 READ_TIMESTAMP(intr0);
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002259 v = x;
2260 x = PyEval_CallObject(v, w);
2261 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002262 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002263 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002264 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002265 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002266 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002267
Thomas Wouters52152252000-08-17 22:55:00 +00002268 case IMPORT_STAR:
2269 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002270 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002271 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002272 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002273 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002274 break;
2275 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002276 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002277 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002278 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002279 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002280 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002281 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002282 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002283
Thomas Wouters52152252000-08-17 22:55:00 +00002284 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002285 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002286 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002287 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002288 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002289 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002290 PUSH(x);
2291 if (x != NULL) continue;
2292 break;
2293
Guido van Rossum374a9221991-04-04 10:40:29 +00002294 case JUMP_FORWARD:
2295 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002296 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002297
Raymond Hettingerf606f872003-03-16 03:11:04 +00002298 PREDICTED_WITH_ARG(JUMP_IF_FALSE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002299 case JUMP_IF_FALSE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002300 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002301 if (w == Py_True) {
2302 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002303 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002304 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002305 if (w == Py_False) {
2306 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002307 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002308 }
2309 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002310 if (err > 0)
2311 err = 0;
2312 else if (err == 0)
Guido van Rossum374a9221991-04-04 10:40:29 +00002313 JUMPBY(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002314 else
2315 break;
2316 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002317
Raymond Hettingerf606f872003-03-16 03:11:04 +00002318 PREDICTED_WITH_ARG(JUMP_IF_TRUE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002319 case JUMP_IF_TRUE:
Raymond Hettinger21012b82003-02-26 18:11:50 +00002320 w = TOP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002321 if (w == Py_False) {
2322 PREDICT(POP_TOP);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002323 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002324 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002325 if (w == Py_True) {
2326 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002327 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002328 }
2329 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002330 if (err > 0) {
2331 err = 0;
Guido van Rossum374a9221991-04-04 10:40:29 +00002332 JUMPBY(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002333 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002334 else if (err == 0)
2335 ;
2336 else
2337 break;
2338 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002339
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002340 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002341 case JUMP_ABSOLUTE:
2342 JUMPTO(oparg);
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002343#if FAST_LOOPS
2344 /* Enabling this path speeds-up all while and for-loops by bypassing
2345 the per-loop checks for signals. By default, this should be turned-off
2346 because it prevents detection of a control-break in tight loops like
2347 "while 1: pass". Compile with this option turned-on when you need
2348 the speed-up and do not need break checking inside tight loops (ones
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002349 that contain only instructions ending with goto fast_next_opcode).
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002350 */
2351 goto fast_next_opcode;
2352#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002353 continue;
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002354#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002355
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002356 case GET_ITER:
2357 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002358 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002359 x = PyObject_GetIter(v);
2360 Py_DECREF(v);
2361 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002362 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002363 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002364 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002365 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002366 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002367 break;
2368
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002369 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002370 case FOR_ITER:
2371 /* before: [iter]; after: [iter, iter()] *or* [] */
2372 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002373 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002374 if (x != NULL) {
2375 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002376 PREDICT(STORE_FAST);
2377 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002378 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002379 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002380 if (PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002381 if (!PyErr_ExceptionMatches(
2382 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002383 break;
2384 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002385 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002386 /* iterator ended normally */
2387 x = v = POP();
2388 Py_DECREF(v);
2389 JUMPBY(oparg);
2390 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002391
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002392 case BREAK_LOOP:
2393 why = WHY_BREAK;
2394 goto fast_block_end;
2395
2396 case CONTINUE_LOOP:
2397 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002398 if (!retval) {
2399 x = NULL;
2400 break;
2401 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002402 why = WHY_CONTINUE;
2403 goto fast_block_end;
2404
Guido van Rossum374a9221991-04-04 10:40:29 +00002405 case SETUP_LOOP:
2406 case SETUP_EXCEPT:
2407 case SETUP_FINALLY:
Thomas Wouterse2176022007-09-20 17:35:10 +00002408 /* NOTE: If you add any new block-setup opcodes that
2409 are not try/except/finally handlers, you may need
2410 to update the PyGen_NeedsFinalizing() function.
2411 */
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002412
Guido van Rossumb209a111997-04-29 18:18:01 +00002413 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002414 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002415 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002416
Guido van Rossumc2e20742006-02-27 22:32:47 +00002417 case WITH_CLEANUP:
2418 {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002419 /* At the top of the stack are 1-3 values indicating
2420 how/why we entered the finally clause:
2421 - TOP = None
2422 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2423 - TOP = WHY_*; no retval below it
2424 - (TOP, SECOND, THIRD) = exc_info()
2425 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002426 In the last case, we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002427 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002428 otherwise we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002429 EXIT(None, None, None)
2430
2431 In all cases, we remove EXIT from the stack, leaving
2432 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002433
2434 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002435 *and* the function call returns a 'true' value, we
2436 "zap" this information, to prevent END_FINALLY from
2437 re-raising the exception. (But non-local gotos
2438 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002439 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002440
Nick Coghlan7af53be2008-03-07 14:13:28 +00002441 PyObject *exit_func;
2442
2443 u = POP();
2444 if (u == Py_None) {
2445 exit_func = TOP();
2446 SET_TOP(u);
2447 v = w = Py_None;
2448 }
2449 else if (PyInt_Check(u)) {
2450 switch(PyInt_AS_LONG(u)) {
2451 case WHY_RETURN:
2452 case WHY_CONTINUE:
2453 /* Retval in TOP. */
2454 exit_func = SECOND();
2455 SET_SECOND(TOP());
2456 SET_TOP(u);
2457 break;
2458 default:
2459 exit_func = TOP();
2460 SET_TOP(u);
2461 break;
2462 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002463 u = v = w = Py_None;
2464 }
2465 else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002466 v = TOP();
2467 w = SECOND();
2468 exit_func = THIRD();
2469 SET_TOP(u);
2470 SET_SECOND(v);
2471 SET_THIRD(w);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002472 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002473 /* XXX Not the fastest way to call it... */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002474 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2475 NULL);
Amaury Forgeot d'Arcad9b5992008-12-10 23:22:49 +00002476 Py_DECREF(exit_func);
2477 if (x == NULL)
Guido van Rossumf6694362006-03-10 02:28:35 +00002478 break; /* Go to error exit */
Amaury Forgeot d'Arcad9b5992008-12-10 23:22:49 +00002479
2480 if (u != Py_None)
2481 err = PyObject_IsTrue(x);
2482 else
2483 err = 0;
2484 Py_DECREF(x);
2485
2486 if (err < 0)
2487 break; /* Go to error exit */
2488 else if (err > 0) {
2489 err = 0;
Guido van Rossumf6694362006-03-10 02:28:35 +00002490 /* There was an exception and a true return */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002491 STACKADJ(-2);
Guido van Rossumf6694362006-03-10 02:28:35 +00002492 Py_INCREF(Py_None);
2493 SET_TOP(Py_None);
Guido van Rossumf6694362006-03-10 02:28:35 +00002494 Py_DECREF(u);
2495 Py_DECREF(v);
2496 Py_DECREF(w);
2497 } else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002498 /* The stack was rearranged to remove EXIT
2499 above. Let END_FINALLY do its thing */
Guido van Rossumf6694362006-03-10 02:28:35 +00002500 }
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00002501 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002502 break;
2503 }
2504
Guido van Rossumf10570b1995-07-07 22:53:21 +00002505 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002506 {
2507 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002508 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002509 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002510#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002511 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002512#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002513 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002514#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002515 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002516 PUSH(x);
2517 if (x != NULL)
2518 continue;
2519 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002520 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002521
Jeremy Hylton76901512000-03-28 23:49:17 +00002522 case CALL_FUNCTION_VAR:
2523 case CALL_FUNCTION_KW:
2524 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002525 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002526 int na = oparg & 0xff;
2527 int nk = (oparg>>8) & 0xff;
2528 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002529 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002530 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002531 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002532 if (flags & CALL_FLAG_VAR)
2533 n++;
2534 if (flags & CALL_FLAG_KW)
2535 n++;
2536 pfunc = stack_pointer - n - 1;
2537 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002538
Guido van Rossumac7be682001-01-17 15:42:30 +00002539 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002540 && PyMethod_GET_SELF(func) != NULL) {
2541 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002542 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002543 func = PyMethod_GET_FUNCTION(func);
2544 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002545 Py_DECREF(*pfunc);
2546 *pfunc = self;
2547 na++;
2548 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002549 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002550 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002551 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002552 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002553 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002554 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002555 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002556 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002557
Jeremy Hylton76901512000-03-28 23:49:17 +00002558 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002559 w = POP();
2560 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002561 }
2562 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002563 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002564 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002565 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002566 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002567
Guido van Rossum681d79a1995-07-18 14:51:37 +00002568 case MAKE_FUNCTION:
2569 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002570 x = PyFunction_New(v, f->f_globals);
2571 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002572 /* XXX Maybe this should be a separate opcode? */
2573 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002574 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002575 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002576 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002577 x = NULL;
2578 break;
2579 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002580 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002581 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002582 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002583 }
2584 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002585 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002586 }
2587 PUSH(x);
2588 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002589
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002590 case MAKE_CLOSURE:
2591 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002592 v = POP(); /* code object */
2593 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002594 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002595 if (x != NULL) {
2596 v = POP();
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002597 if (PyFunction_SetClosure(x, v) != 0) {
2598 /* Can't happen unless bytecode is corrupt. */
2599 why = WHY_EXCEPTION;
2600 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002601 Py_DECREF(v);
2602 }
2603 if (x != NULL && oparg > 0) {
2604 v = PyTuple_New(oparg);
2605 if (v == NULL) {
2606 Py_DECREF(x);
2607 x = NULL;
2608 break;
2609 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002610 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002611 w = POP();
2612 PyTuple_SET_ITEM(v, oparg, w);
2613 }
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002614 if (PyFunction_SetDefaults(x, v) != 0) {
2615 /* Can't happen unless
2616 PyFunction_SetDefaults changes. */
2617 why = WHY_EXCEPTION;
2618 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002619 Py_DECREF(v);
2620 }
2621 PUSH(x);
2622 break;
2623 }
2624
Guido van Rossum8861b741996-07-30 16:49:37 +00002625 case BUILD_SLICE:
2626 if (oparg == 3)
2627 w = POP();
2628 else
2629 w = NULL;
2630 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002631 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002632 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002633 Py_DECREF(u);
2634 Py_DECREF(v);
2635 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002636 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002637 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002638 break;
2639
Fred Drakeef8ace32000-08-24 00:32:09 +00002640 case EXTENDED_ARG:
2641 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002642 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002643 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002644
Guido van Rossum374a9221991-04-04 10:40:29 +00002645 default:
2646 fprintf(stderr,
2647 "XXX lineno: %d, opcode: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002648 PyCode_Addr2Line(f->f_code, f->f_lasti),
2649 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002650 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002651 why = WHY_EXCEPTION;
2652 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002653
2654#ifdef CASE_TOO_BIG
2655 }
2656#endif
2657
Guido van Rossum374a9221991-04-04 10:40:29 +00002658 } /* switch */
2659
2660 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002661
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002662 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002663
Guido van Rossum374a9221991-04-04 10:40:29 +00002664 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002665
Guido van Rossum374a9221991-04-04 10:40:29 +00002666 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002667 if (err == 0 && x != NULL) {
2668#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002669 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002670 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002671 fprintf(stderr,
2672 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002673 else {
2674#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002675 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002676 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002677#ifdef CHECKEXC
2678 }
2679#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002680 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002681 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002682 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002683 err = 0;
2684 }
2685
Guido van Rossum374a9221991-04-04 10:40:29 +00002686 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002687
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002688 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002689 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002690 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002691 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002692 why = WHY_EXCEPTION;
2693 }
2694 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002695#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002696 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002697 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002698 if (PyErr_Occurred()) {
Neal Norwitz8250fbe2008-01-27 17:12:15 +00002699 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002700 sprintf(buf, "Stack unwind with exception "
2701 "set and why=%d", why);
2702 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002703 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002704 }
2705#endif
2706
2707 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002708
Guido van Rossum374a9221991-04-04 10:40:29 +00002709 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002710 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002711
Fred Drake8f51f542001-10-04 14:48:42 +00002712 if (tstate->c_tracefunc != NULL)
2713 call_exc_trace(tstate->c_tracefunc,
2714 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002715 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002716
Guido van Rossum374a9221991-04-04 10:40:29 +00002717 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002718
Guido van Rossum374a9221991-04-04 10:40:29 +00002719 if (why == WHY_RERAISE)
2720 why = WHY_EXCEPTION;
2721
2722 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002723
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002724fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002725 while (why != WHY_NOT && f->f_iblock > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002726 PyTryBlock *b = PyFrame_BlockPop(f);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002727
Tim Peters8a5c3c72004-04-05 19:36:21 +00002728 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002729 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
2730 /* For a continue inside a try block,
2731 don't pop the block for the loop. */
Thomas Wouters1ee64222001-09-24 19:32:01 +00002732 PyFrame_BlockSetup(f, b->b_type, b->b_handler,
2733 b->b_level);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002734 why = WHY_NOT;
2735 JUMPTO(PyInt_AS_LONG(retval));
2736 Py_DECREF(retval);
2737 break;
2738 }
2739
Guido van Rossum374a9221991-04-04 10:40:29 +00002740 while (STACK_LEVEL() > b->b_level) {
2741 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002742 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002743 }
2744 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2745 why = WHY_NOT;
2746 JUMPTO(b->b_handler);
2747 break;
2748 }
2749 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002750 (b->b_type == SETUP_EXCEPT &&
2751 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002752 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002753 PyObject *exc, *val, *tb;
2754 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002755 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002756 val = Py_None;
2757 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002758 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002759 /* Make the raw exception data
2760 available to the handler,
2761 so a program can emulate the
2762 Python main loop. Don't do
2763 this for 'finally'. */
2764 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002765 PyErr_NormalizeException(
2766 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002767 set_exc_info(tstate,
2768 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002769 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002770 if (tb == NULL) {
2771 Py_INCREF(Py_None);
2772 PUSH(Py_None);
2773 } else
2774 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002775 PUSH(val);
2776 PUSH(exc);
2777 }
2778 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002779 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002780 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002781 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002782 PUSH(v);
2783 }
2784 why = WHY_NOT;
2785 JUMPTO(b->b_handler);
2786 break;
2787 }
2788 } /* unwind stack */
2789
2790 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002791
Guido van Rossum374a9221991-04-04 10:40:29 +00002792 if (why != WHY_NOT)
2793 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002794 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002795
Guido van Rossum374a9221991-04-04 10:40:29 +00002796 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002797
Tim Peters8a5c3c72004-04-05 19:36:21 +00002798 assert(why != WHY_YIELD);
2799 /* Pop remaining stack entries. */
2800 while (!EMPTY()) {
2801 v = POP();
2802 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002803 }
2804
Tim Peters8a5c3c72004-04-05 19:36:21 +00002805 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002806 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002807
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002808fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002809 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002810 if (tstate->c_tracefunc) {
2811 if (why == WHY_RETURN || why == WHY_YIELD) {
2812 if (call_trace(tstate->c_tracefunc,
2813 tstate->c_traceobj, f,
2814 PyTrace_RETURN, retval)) {
2815 Py_XDECREF(retval);
2816 retval = NULL;
2817 why = WHY_EXCEPTION;
2818 }
2819 }
2820 else if (why == WHY_EXCEPTION) {
2821 call_trace_protected(tstate->c_tracefunc,
2822 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002823 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002824 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002825 }
Fred Drake8f51f542001-10-04 14:48:42 +00002826 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002827 if (why == WHY_EXCEPTION)
2828 call_trace_protected(tstate->c_profilefunc,
2829 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002830 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002831 else if (call_trace(tstate->c_profilefunc,
2832 tstate->c_profileobj, f,
2833 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002834 Py_XDECREF(retval);
2835 retval = NULL;
2836 why = WHY_EXCEPTION;
2837 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002838 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002839 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002840
Tim Peters7df5e7f2006-05-26 23:14:37 +00002841 if (tstate->frame->f_exc_type != NULL)
2842 reset_exc_info(tstate);
2843 else {
2844 assert(tstate->frame->f_exc_value == NULL);
2845 assert(tstate->frame->f_exc_traceback == NULL);
2846 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002847
Tim Peters5ca576e2001-06-18 22:08:13 +00002848 /* pop frame */
Thomas Woutersae406c62007-09-19 17:27:43 +00002849exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002850 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002851 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002852
Guido van Rossum96a42c81992-01-12 02:29:51 +00002853 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002854}
2855
Guido van Rossumc2e20742006-02-27 22:32:47 +00002856/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002857 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002858 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002859
Tim Peters6d6c1a32001-08-02 04:15:00 +00002860PyObject *
2861PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002862 PyObject **args, int argcount, PyObject **kws, int kwcount,
2863 PyObject **defs, int defcount, PyObject *closure)
2864{
2865 register PyFrameObject *f;
2866 register PyObject *retval = NULL;
2867 register PyObject **fastlocals, **freevars;
2868 PyThreadState *tstate = PyThreadState_GET();
2869 PyObject *x, *u;
2870
2871 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002872 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002873 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002874 return NULL;
2875 }
2876
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002877 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002878 assert(globals != NULL);
2879 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002880 if (f == NULL)
2881 return NULL;
2882
2883 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002884 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002885
2886 if (co->co_argcount > 0 ||
2887 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2888 int i;
2889 int n = argcount;
2890 PyObject *kwdict = NULL;
2891 if (co->co_flags & CO_VARKEYWORDS) {
2892 kwdict = PyDict_New();
2893 if (kwdict == NULL)
2894 goto fail;
2895 i = co->co_argcount;
2896 if (co->co_flags & CO_VARARGS)
2897 i++;
2898 SETLOCAL(i, kwdict);
2899 }
2900 if (argcount > co->co_argcount) {
2901 if (!(co->co_flags & CO_VARARGS)) {
2902 PyErr_Format(PyExc_TypeError,
2903 "%.200s() takes %s %d "
2904 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002905 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00002906 defcount ? "at most" : "exactly",
2907 co->co_argcount,
2908 kwcount ? "non-keyword " : "",
2909 co->co_argcount == 1 ? "" : "s",
2910 argcount);
2911 goto fail;
2912 }
2913 n = co->co_argcount;
2914 }
2915 for (i = 0; i < n; i++) {
2916 x = args[i];
2917 Py_INCREF(x);
2918 SETLOCAL(i, x);
2919 }
2920 if (co->co_flags & CO_VARARGS) {
2921 u = PyTuple_New(argcount - n);
2922 if (u == NULL)
2923 goto fail;
2924 SETLOCAL(co->co_argcount, u);
2925 for (i = n; i < argcount; i++) {
2926 x = args[i];
2927 Py_INCREF(x);
2928 PyTuple_SET_ITEM(u, i-n, x);
2929 }
2930 }
2931 for (i = 0; i < kwcount; i++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002932 PyObject **co_varnames;
Tim Peters5ca576e2001-06-18 22:08:13 +00002933 PyObject *keyword = kws[2*i];
2934 PyObject *value = kws[2*i + 1];
2935 int j;
Benjamin Peterson78821dd2009-01-25 17:15:10 +00002936 if (keyword == NULL || !(PyString_Check(keyword)
2937#ifdef Py_USING_UNICODE
2938 || PyUnicode_Check(keyword)
2939#endif
2940 )) {
Tim Peters5ca576e2001-06-18 22:08:13 +00002941 PyErr_Format(PyExc_TypeError,
2942 "%.200s() keywords must be strings",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002943 PyString_AsString(co->co_name));
Tim Peters5ca576e2001-06-18 22:08:13 +00002944 goto fail;
2945 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002946 /* Speed hack: do raw pointer compares. As names are
2947 normally interned this should almost always hit. */
2948 co_varnames = PySequence_Fast_ITEMS(co->co_varnames);
Tim Peters5ca576e2001-06-18 22:08:13 +00002949 for (j = 0; j < co->co_argcount; j++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002950 PyObject *nm = co_varnames[j];
2951 if (nm == keyword)
2952 goto kw_found;
2953 }
2954 /* Slow fallback, just in case */
2955 for (j = 0; j < co->co_argcount; j++) {
2956 PyObject *nm = co_varnames[j];
Tim Peters5ca576e2001-06-18 22:08:13 +00002957 int cmp = PyObject_RichCompareBool(
2958 keyword, nm, Py_EQ);
2959 if (cmp > 0)
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002960 goto kw_found;
Tim Peters5ca576e2001-06-18 22:08:13 +00002961 else if (cmp < 0)
2962 goto fail;
2963 }
2964 /* Check errors from Compare */
2965 if (PyErr_Occurred())
2966 goto fail;
2967 if (j >= co->co_argcount) {
2968 if (kwdict == NULL) {
Benjamin Petersone18ef192009-01-20 14:21:16 +00002969 PyObject *kwd_str = kwd_as_string(keyword);
2970 if (kwd_str) {
2971 PyErr_Format(PyExc_TypeError,
2972 "%.200s() got an unexpected "
2973 "keyword argument '%.400s'",
2974 PyString_AsString(co->co_name),
2975 PyString_AsString(kwd_str));
2976 Py_DECREF(kwd_str);
2977 }
Tim Peters5ca576e2001-06-18 22:08:13 +00002978 goto fail;
2979 }
2980 PyDict_SetItem(kwdict, keyword, value);
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002981 continue;
Tim Peters5ca576e2001-06-18 22:08:13 +00002982 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002983kw_found:
2984 if (GETLOCAL(j) != NULL) {
Benjamin Petersone18ef192009-01-20 14:21:16 +00002985 PyObject *kwd_str = kwd_as_string(keyword);
2986 if (kwd_str) {
2987 PyErr_Format(PyExc_TypeError,
2988 "%.200s() got multiple "
2989 "values for keyword "
2990 "argument '%.400s'",
2991 PyString_AsString(co->co_name),
2992 PyString_AsString(kwd_str));
2993 Py_DECREF(kwd_str);
2994 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002995 goto fail;
Tim Peters5ca576e2001-06-18 22:08:13 +00002996 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002997 Py_INCREF(value);
2998 SETLOCAL(j, value);
Tim Peters5ca576e2001-06-18 22:08:13 +00002999 }
3000 if (argcount < co->co_argcount) {
3001 int m = co->co_argcount - defcount;
3002 for (i = argcount; i < m; i++) {
3003 if (GETLOCAL(i) == NULL) {
3004 PyErr_Format(PyExc_TypeError,
3005 "%.200s() takes %s %d "
3006 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003007 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00003008 ((co->co_flags & CO_VARARGS) ||
3009 defcount) ? "at least"
3010 : "exactly",
3011 m, kwcount ? "non-keyword " : "",
3012 m == 1 ? "" : "s", i);
3013 goto fail;
3014 }
3015 }
3016 if (n > m)
3017 i = n - m;
3018 else
3019 i = 0;
3020 for (; i < defcount; i++) {
3021 if (GETLOCAL(m+i) == NULL) {
3022 PyObject *def = defs[i];
3023 Py_INCREF(def);
3024 SETLOCAL(m+i, def);
3025 }
3026 }
3027 }
3028 }
3029 else {
3030 if (argcount > 0 || kwcount > 0) {
3031 PyErr_Format(PyExc_TypeError,
3032 "%.200s() takes no arguments (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003033 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00003034 argcount + kwcount);
3035 goto fail;
3036 }
3037 }
3038 /* Allocate and initialize storage for cell vars, and copy free
3039 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00003040 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00003041 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00003042 char *cellname, *argname;
3043 PyObject *c;
3044
3045 nargs = co->co_argcount;
3046 if (co->co_flags & CO_VARARGS)
3047 nargs++;
3048 if (co->co_flags & CO_VARKEYWORDS)
3049 nargs++;
3050
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003051 /* Initialize each cell var, taking into account
3052 cell vars that are initialized from arguments.
3053
3054 Should arrange for the compiler to put cellvars
3055 that are arguments at the beginning of the cellvars
3056 list so that we can march over it more efficiently?
3057 */
Richard Jonescebbefc2006-05-23 18:28:17 +00003058 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003059 cellname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00003060 PyTuple_GET_ITEM(co->co_cellvars, i));
3061 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003062 for (j = 0; j < nargs; j++) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003063 argname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00003064 PyTuple_GET_ITEM(co->co_varnames, j));
3065 if (strcmp(cellname, argname) == 0) {
3066 c = PyCell_New(GETLOCAL(j));
3067 if (c == NULL)
3068 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00003069 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00003070 found = 1;
3071 break;
3072 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003073 }
3074 if (found == 0) {
3075 c = PyCell_New(NULL);
3076 if (c == NULL)
3077 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00003078 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00003079 }
3080 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003081 }
Richard Jonescebbefc2006-05-23 18:28:17 +00003082 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00003083 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00003084 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00003085 PyObject *o = PyTuple_GET_ITEM(closure, i);
3086 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00003087 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00003088 }
3089 }
3090
Tim Peters5ca576e2001-06-18 22:08:13 +00003091 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003092 /* Don't need to keep the reference to f_back, it will be set
3093 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00003094 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003095 f->f_back = NULL;
3096
Jeremy Hylton985eba52003-02-05 23:13:00 +00003097 PCALL(PCALL_GENERATOR);
3098
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003099 /* Create a new generator that owns the ready to run frame
3100 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00003101 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00003102 }
3103
Thomas Woutersae406c62007-09-19 17:27:43 +00003104 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003105
Thomas Woutersae406c62007-09-19 17:27:43 +00003106fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003107
Tim Petersb13680b2001-11-27 23:29:29 +00003108 /* decref'ing the frame can cause __del__ methods to get invoked,
3109 which can call back into Python. While we're done with the
3110 current Python frame (f), the associated C stack is still in use,
3111 so recursion_depth must be boosted for the duration.
3112 */
3113 assert(tstate != NULL);
3114 ++tstate->recursion_depth;
Thomas Woutersae406c62007-09-19 17:27:43 +00003115 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00003116 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00003117 return retval;
3118}
3119
3120
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003121
Benjamin Petersone18ef192009-01-20 14:21:16 +00003122static PyObject *
3123kwd_as_string(PyObject *kwd) {
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003124#ifdef Py_USING_UNICODE
Benjamin Petersone18ef192009-01-20 14:21:16 +00003125 if (PyString_Check(kwd)) {
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003126#else
3127 assert(PyString_Check(kwd));
3128#endif
Benjamin Petersone18ef192009-01-20 14:21:16 +00003129 Py_INCREF(kwd);
3130 return kwd;
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003131#ifdef Py_USING_UNICODE
Benjamin Petersone18ef192009-01-20 14:21:16 +00003132 }
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003133 return _PyUnicode_AsDefaultEncodedString(kwd, "replace");
3134#endif
Benjamin Petersone18ef192009-01-20 14:21:16 +00003135}
3136
3137
Guido van Rossumc9fbb722003-03-01 03:36:33 +00003138/* Implementation notes for set_exc_info() and reset_exc_info():
3139
3140- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
3141 'exc_traceback'. These always travel together.
3142
3143- tstate->curexc_ZZZ is the "hot" exception that is set by
3144 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
3145
3146- Once an exception is caught by an except clause, it is transferred
3147 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
3148 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00003149 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00003150
3151- Now let me explain the complicated dance with frame->f_exc_ZZZ.
3152
3153 Long ago, when none of this existed, there were just a few globals:
3154 one set corresponding to the "hot" exception, and one set
3155 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
3156 globals; they were simply stored as sys.exc_ZZZ. For backwards
3157 compatibility, they still are!) The problem was that in code like
3158 this:
3159
3160 try:
3161 "something that may fail"
3162 except "some exception":
3163 "do something else first"
3164 "print the exception from sys.exc_ZZZ."
3165
3166 if "do something else first" invoked something that raised and caught
3167 an exception, sys.exc_ZZZ were overwritten. That was a frequent
3168 cause of subtle bugs. I fixed this by changing the semantics as
3169 follows:
3170
3171 - Within one frame, sys.exc_ZZZ will hold the last exception caught
3172 *in that frame*.
3173
3174 - But initially, and as long as no exception is caught in a given
3175 frame, sys.exc_ZZZ will hold the last exception caught in the
3176 previous frame (or the frame before that, etc.).
3177
3178 The first bullet fixed the bug in the above example. The second
3179 bullet was for backwards compatibility: it was (and is) common to
3180 have a function that is called when an exception is caught, and to
3181 have that function access the caught exception via sys.exc_ZZZ.
3182 (Example: traceback.print_exc()).
3183
3184 At the same time I fixed the problem that sys.exc_ZZZ weren't
3185 thread-safe, by introducing sys.exc_info() which gets it from tstate;
3186 but that's really a separate improvement.
3187
3188 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
3189 variables to what they were before the current frame was called. The
3190 set_exc_info() function saves them on the frame so that
3191 reset_exc_info() can restore them. The invariant is that
3192 frame->f_exc_ZZZ is NULL iff the current frame never caught an
3193 exception (where "catching" an exception applies only to successful
3194 except clauses); and if the current frame ever caught an exception,
3195 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
3196 at the start of the current frame.
3197
3198*/
3199
Fredrik Lundh7a830892006-05-27 10:39:48 +00003200static void
Guido van Rossumac7be682001-01-17 15:42:30 +00003201set_exc_info(PyThreadState *tstate,
3202 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003203{
Tim Peters7df5e7f2006-05-26 23:14:37 +00003204 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003205 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00003206
Tim Peters7df5e7f2006-05-26 23:14:37 +00003207 assert(type != NULL);
3208 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003209 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003210 assert(frame->f_exc_value == NULL);
3211 assert(frame->f_exc_traceback == NULL);
3212 /* This frame didn't catch an exception before. */
3213 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003214 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003215 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003216 Py_INCREF(Py_None);
3217 tstate->exc_type = Py_None;
3218 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003219 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003220 Py_XINCREF(tstate->exc_value);
3221 Py_XINCREF(tstate->exc_traceback);
3222 frame->f_exc_type = tstate->exc_type;
3223 frame->f_exc_value = tstate->exc_value;
3224 frame->f_exc_traceback = tstate->exc_traceback;
3225 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003226 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003227 tmp_type = tstate->exc_type;
3228 tmp_value = tstate->exc_value;
3229 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003230 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003231 Py_XINCREF(value);
3232 Py_XINCREF(tb);
3233 tstate->exc_type = type;
3234 tstate->exc_value = value;
3235 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003236 Py_XDECREF(tmp_type);
3237 Py_XDECREF(tmp_value);
3238 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003239 /* For b/w compatibility */
3240 PySys_SetObject("exc_type", type);
3241 PySys_SetObject("exc_value", value);
3242 PySys_SetObject("exc_traceback", tb);
3243}
3244
Fredrik Lundh7a830892006-05-27 10:39:48 +00003245static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003246reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003247{
3248 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003249 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003250
3251 /* It's a precondition that the thread state's frame caught an
3252 * exception -- verify in a debug build.
3253 */
3254 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003255 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003256 assert(frame != NULL);
3257 assert(frame->f_exc_type != NULL);
3258
3259 /* Copy the frame's exception info back to the thread state. */
3260 tmp_type = tstate->exc_type;
3261 tmp_value = tstate->exc_value;
3262 tmp_tb = tstate->exc_traceback;
3263 Py_INCREF(frame->f_exc_type);
3264 Py_XINCREF(frame->f_exc_value);
3265 Py_XINCREF(frame->f_exc_traceback);
3266 tstate->exc_type = frame->f_exc_type;
3267 tstate->exc_value = frame->f_exc_value;
3268 tstate->exc_traceback = frame->f_exc_traceback;
3269 Py_XDECREF(tmp_type);
3270 Py_XDECREF(tmp_value);
3271 Py_XDECREF(tmp_tb);
3272
3273 /* For b/w compatibility */
3274 PySys_SetObject("exc_type", frame->f_exc_type);
3275 PySys_SetObject("exc_value", frame->f_exc_value);
3276 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
3277
3278 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003279 tmp_type = frame->f_exc_type;
3280 tmp_value = frame->f_exc_value;
3281 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003282 frame->f_exc_type = NULL;
3283 frame->f_exc_value = NULL;
3284 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003285 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003286 Py_XDECREF(tmp_value);
3287 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003288}
3289
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003290/* Logic for the raise statement (too complicated for inlining).
3291 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003292static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003293do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003294{
Guido van Rossumd295f121998-04-09 21:39:57 +00003295 if (type == NULL) {
3296 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003297 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003298 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3299 value = tstate->exc_value;
3300 tb = tstate->exc_traceback;
3301 Py_XINCREF(type);
3302 Py_XINCREF(value);
3303 Py_XINCREF(tb);
3304 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003305
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003306 /* We support the following forms of raise:
3307 raise <class>, <classinstance>
3308 raise <class>, <argument tuple>
3309 raise <class>, None
3310 raise <class>, <argument>
3311 raise <classinstance>, None
3312 raise <string>, <object>
3313 raise <string>, None
3314
3315 An omitted second argument is the same as None.
3316
3317 In addition, raise <tuple>, <anything> is the same as
3318 raising the tuple's first item (and it better have one!);
3319 this rule is applied recursively.
3320
3321 Finally, an optional third argument can be supplied, which
3322 gives the traceback to be substituted (useful when
3323 re-raising an exception after examining it). */
3324
3325 /* First, check the traceback argument, replacing None with
3326 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003327 if (tb == Py_None) {
3328 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003329 tb = NULL;
3330 }
3331 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003332 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003333 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003334 goto raise_error;
3335 }
3336
3337 /* Next, replace a missing value with None */
3338 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003339 value = Py_None;
3340 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003341 }
3342
3343 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003344 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3345 PyObject *tmp = type;
3346 type = PyTuple_GET_ITEM(type, 0);
3347 Py_INCREF(type);
3348 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003349 }
3350
Brett Cannon129bd522007-01-30 21:34:36 +00003351 if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003352 PyErr_NormalizeException(&type, &value, &tb);
3353
Brett Cannonbf364092006-03-01 04:25:17 +00003354 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003355 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003356 if (value != Py_None) {
3357 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003358 "instance exception may not have a separate value");
3359 goto raise_error;
3360 }
3361 else {
3362 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003363 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003364 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003365 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003366 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003367 }
3368 }
3369 else {
3370 /* Not something you can raise. You get an exception
3371 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003372 PyErr_Format(PyExc_TypeError,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003373 "exceptions must be classes or instances, not %s",
3374 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003375 goto raise_error;
3376 }
Guido van Rossum504153d2008-03-18 04:26:48 +00003377
3378 assert(PyExceptionClass_Check(type));
3379 if (Py_Py3kWarningFlag && PyClass_Check(type)) {
Benjamin Peterson9f4f4812008-04-27 03:01:45 +00003380 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003381 "exceptions must derive from BaseException "
3382 "in 3.x", 1) < 0)
Guido van Rossum504153d2008-03-18 04:26:48 +00003383 goto raise_error;
3384 }
3385
Guido van Rossumb209a111997-04-29 18:18:01 +00003386 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003387 if (tb == NULL)
3388 return WHY_EXCEPTION;
3389 else
3390 return WHY_RERAISE;
3391 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003392 Py_XDECREF(value);
3393 Py_XDECREF(type);
3394 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003395 return WHY_EXCEPTION;
3396}
3397
Tim Petersd6d010b2001-06-21 02:49:55 +00003398/* Iterate v argcnt times and store the results on the stack (via decreasing
3399 sp). Return 1 for success, 0 if error. */
3400
Fredrik Lundh7a830892006-05-27 10:39:48 +00003401static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003402unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003403{
Tim Petersd6d010b2001-06-21 02:49:55 +00003404 int i = 0;
3405 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003406 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003407
Tim Petersd6d010b2001-06-21 02:49:55 +00003408 assert(v != NULL);
3409
3410 it = PyObject_GetIter(v);
3411 if (it == NULL)
3412 goto Error;
3413
3414 for (; i < argcnt; i++) {
3415 w = PyIter_Next(it);
3416 if (w == NULL) {
3417 /* Iterator done, via error or exhaustion. */
3418 if (!PyErr_Occurred()) {
3419 PyErr_Format(PyExc_ValueError,
3420 "need more than %d value%s to unpack",
3421 i, i == 1 ? "" : "s");
3422 }
3423 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003424 }
3425 *--sp = w;
3426 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003427
3428 /* We better have exhausted the iterator now. */
3429 w = PyIter_Next(it);
3430 if (w == NULL) {
3431 if (PyErr_Occurred())
3432 goto Error;
3433 Py_DECREF(it);
3434 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003435 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003436 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003437 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003438 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003439Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003440 for (; i > 0; i--, sp++)
3441 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003442 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003443 return 0;
3444}
3445
3446
Guido van Rossum96a42c81992-01-12 02:29:51 +00003447#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003448static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003449prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003450{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003451 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003452 if (PyObject_Print(v, stdout, 0) != 0)
3453 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003454 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003455 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003456}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003457#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003458
Fredrik Lundh7a830892006-05-27 10:39:48 +00003459static void
Fred Drake5755ce62001-06-27 19:19:46 +00003460call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003461{
Guido van Rossumb209a111997-04-29 18:18:01 +00003462 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003463 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003464 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003465 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003466 value = Py_None;
3467 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003468 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003469 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003470 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003471 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003472 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003473 }
Fred Drake5755ce62001-06-27 19:19:46 +00003474 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003475 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003476 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003477 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003478 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003479 Py_XDECREF(type);
3480 Py_XDECREF(value);
3481 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003482 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003483}
3484
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003485static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003486call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003487 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003488{
3489 PyObject *type, *value, *traceback;
3490 int err;
3491 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003492 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003493 if (err == 0)
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003494 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003495 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003496 return 0;
3497 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003498 else {
3499 Py_XDECREF(type);
3500 Py_XDECREF(value);
3501 Py_XDECREF(traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003502 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003503 }
3504}
3505
Fredrik Lundh7a830892006-05-27 10:39:48 +00003506static int
Fred Drake5755ce62001-06-27 19:19:46 +00003507call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3508 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003509{
Fred Drake5755ce62001-06-27 19:19:46 +00003510 register PyThreadState *tstate = frame->f_tstate;
3511 int result;
3512 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003513 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003514 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003515 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003516 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003517 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3518 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003519 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003520 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003521}
3522
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003523PyObject *
3524_PyEval_CallTracing(PyObject *func, PyObject *args)
3525{
3526 PyFrameObject *frame = PyEval_GetFrame();
3527 PyThreadState *tstate = frame->f_tstate;
3528 int save_tracing = tstate->tracing;
3529 int save_use_tracing = tstate->use_tracing;
3530 PyObject *result;
3531
3532 tstate->tracing = 0;
3533 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3534 || (tstate->c_profilefunc != NULL));
3535 result = PyObject_Call(func, args, NULL);
3536 tstate->tracing = save_tracing;
3537 tstate->use_tracing = save_use_tracing;
3538 return result;
3539}
3540
Fredrik Lundh7a830892006-05-27 10:39:48 +00003541static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003542maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003543 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3544 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003545{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003546 int result = 0;
3547
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003548 /* If the last instruction executed isn't in the current
3549 instruction window, reset the window. If the last
3550 instruction happens to fall at the start of a line or if it
3551 represents a jump backwards, call the trace function.
3552 */
Michael W. Hudson53d58bb2002-08-30 13:09:51 +00003553 if ((frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub)) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003554 int line;
3555 PyAddrPair bounds;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003556
Thomas Woutersae406c62007-09-19 17:27:43 +00003557 line = PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3558 &bounds);
3559 if (line >= 0) {
Michael W. Hudson02ff6a92002-09-11 15:36:32 +00003560 frame->f_lineno = line;
Tim Peters8a5c3c72004-04-05 19:36:21 +00003561 result = call_trace(func, obj, frame,
Michael W. Hudson006c7522002-11-08 13:08:46 +00003562 PyTrace_LINE, Py_None);
Thomas Woutersae406c62007-09-19 17:27:43 +00003563 }
3564 *instr_lb = bounds.ap_lower;
3565 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003566 }
Armin Rigobf57a142004-03-22 19:24:58 +00003567 else if (frame->f_lasti <= *instr_prev) {
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003568 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003569 }
3570 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003571 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003572}
3573
Fred Drake5755ce62001-06-27 19:19:46 +00003574void
3575PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003576{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003577 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003578 PyObject *temp = tstate->c_profileobj;
3579 Py_XINCREF(arg);
3580 tstate->c_profilefunc = NULL;
3581 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003582 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003583 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003584 Py_XDECREF(temp);
3585 tstate->c_profilefunc = func;
3586 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003587 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003588 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003589}
3590
3591void
3592PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3593{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003594 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003595 PyObject *temp = tstate->c_traceobj;
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +00003596 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003597 Py_XINCREF(arg);
3598 tstate->c_tracefunc = NULL;
3599 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003600 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003601 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003602 Py_XDECREF(temp);
3603 tstate->c_tracefunc = func;
3604 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003605 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003606 tstate->use_tracing = ((func != NULL)
3607 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003608}
3609
Guido van Rossumb209a111997-04-29 18:18:01 +00003610PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003611PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003612{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003613 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003614 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003615 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003616 else
3617 return current_frame->f_builtins;
3618}
3619
Guido van Rossumb209a111997-04-29 18:18:01 +00003620PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003621PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003622{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003623 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003624 if (current_frame == NULL)
3625 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003626 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003627 return current_frame->f_locals;
3628}
3629
Guido van Rossumb209a111997-04-29 18:18:01 +00003630PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003631PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003632{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003633 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003634 if (current_frame == NULL)
3635 return NULL;
3636 else
3637 return current_frame->f_globals;
3638}
3639
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003640PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003641PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003642{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003643 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003644 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003645}
3646
Guido van Rossum6135a871995-01-09 17:53:26 +00003647int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003648PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003649{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003650 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003651 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003652}
3653
Guido van Rossumbe270261997-05-22 22:26:18 +00003654int
Tim Peters5ba58662001-07-16 02:29:45 +00003655PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003656{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003657 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003658 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003659
3660 if (current_frame != NULL) {
3661 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003662 const int compilerflags = codeflags & PyCF_MASK;
3663 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003664 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003665 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003666 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003667#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003668 if (codeflags & CO_GENERATOR_ALLOWED) {
3669 result = 1;
3670 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3671 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003672#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003673 }
3674 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003675}
3676
3677int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003678Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003679{
Guido van Rossumb209a111997-04-29 18:18:01 +00003680 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003681 if (f == NULL)
3682 return 0;
3683 if (!PyFile_SoftSpace(f, 0))
3684 return 0;
3685 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003686}
3687
Guido van Rossum3f5da241990-12-20 15:06:42 +00003688
Guido van Rossum681d79a1995-07-18 14:51:37 +00003689/* External interface to call any callable object.
3690 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003691
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003692#undef PyEval_CallObject
3693/* for backward compatibility: export this interface */
3694
Guido van Rossumb209a111997-04-29 18:18:01 +00003695PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003696PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003697{
Guido van Rossumb209a111997-04-29 18:18:01 +00003698 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003699}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003700#define PyEval_CallObject(func,arg) \
3701 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003702
Guido van Rossumb209a111997-04-29 18:18:01 +00003703PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003704PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003705{
Jeremy Hylton52820442001-01-03 23:52:36 +00003706 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003707
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003708 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003709 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003710 if (arg == NULL)
3711 return NULL;
3712 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003713 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003714 PyErr_SetString(PyExc_TypeError,
3715 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003716 return NULL;
3717 }
3718 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003719 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003720
Guido van Rossumb209a111997-04-29 18:18:01 +00003721 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003722 PyErr_SetString(PyExc_TypeError,
3723 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003724 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003725 return NULL;
3726 }
3727
Tim Peters6d6c1a32001-08-02 04:15:00 +00003728 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003729 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003730 return result;
3731}
3732
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003733const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003734PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003735{
3736 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003737 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003738 else if (PyFunction_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003739 return PyString_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003740 else if (PyCFunction_Check(func))
3741 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3742 else if (PyClass_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003743 return PyString_AsString(((PyClassObject*)func)->cl_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003744 else if (PyInstance_Check(func)) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003745 return PyString_AsString(
Jeremy Hylton512a2372001-04-11 13:52:29 +00003746 ((PyInstanceObject*)func)->in_class->cl_name);
3747 } else {
3748 return func->ob_type->tp_name;
3749 }
3750}
3751
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003752const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003753PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003754{
3755 if (PyMethod_Check(func))
3756 return "()";
3757 else if (PyFunction_Check(func))
3758 return "()";
3759 else if (PyCFunction_Check(func))
3760 return "()";
3761 else if (PyClass_Check(func))
3762 return " constructor";
3763 else if (PyInstance_Check(func)) {
3764 return " instance";
3765 } else {
3766 return " object";
3767 }
3768}
3769
Fredrik Lundh7a830892006-05-27 10:39:48 +00003770static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003771err_args(PyObject *func, int flags, int nargs)
3772{
3773 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003774 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003775 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003776 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003777 nargs);
3778 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003779 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003780 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003781 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003782 nargs);
3783}
3784
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003785#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003786if (tstate->use_tracing && tstate->c_profilefunc) { \
3787 if (call_trace(tstate->c_profilefunc, \
3788 tstate->c_profileobj, \
3789 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003790 func)) { \
3791 x = NULL; \
3792 } \
3793 else { \
3794 x = call; \
3795 if (tstate->c_profilefunc != NULL) { \
3796 if (x == NULL) { \
3797 call_trace_protected(tstate->c_profilefunc, \
3798 tstate->c_profileobj, \
3799 tstate->frame, PyTrace_C_EXCEPTION, \
3800 func); \
3801 /* XXX should pass (type, value, tb) */ \
3802 } else { \
3803 if (call_trace(tstate->c_profilefunc, \
3804 tstate->c_profileobj, \
3805 tstate->frame, PyTrace_C_RETURN, \
3806 func)) { \
3807 Py_DECREF(x); \
3808 x = NULL; \
3809 } \
3810 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003811 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003812 } \
3813} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003814 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003815 }
3816
Fredrik Lundh7a830892006-05-27 10:39:48 +00003817static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003818call_function(PyObject ***pp_stack, int oparg
3819#ifdef WITH_TSC
3820 , uint64* pintr0, uint64* pintr1
3821#endif
3822 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003823{
3824 int na = oparg & 0xff;
3825 int nk = (oparg>>8) & 0xff;
3826 int n = na + 2 * nk;
3827 PyObject **pfunc = (*pp_stack) - n - 1;
3828 PyObject *func = *pfunc;
3829 PyObject *x, *w;
3830
Jeremy Hylton985eba52003-02-05 23:13:00 +00003831 /* Always dispatch PyCFunction first, because these are
3832 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003833 */
3834 if (PyCFunction_Check(func) && nk == 0) {
3835 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003836 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003837
3838 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003839 if (flags & (METH_NOARGS | METH_O)) {
3840 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3841 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003842 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003843 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003844 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003845 else if (flags & METH_O && na == 1) {
3846 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003847 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003848 Py_DECREF(arg);
3849 }
3850 else {
3851 err_args(func, flags, na);
3852 x = NULL;
3853 }
3854 }
3855 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003856 PyObject *callargs;
3857 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003858 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003859 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003860 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003861 Py_XDECREF(callargs);
3862 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003863 } else {
3864 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3865 /* optimize access to bound methods */
3866 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003867 PCALL(PCALL_METHOD);
3868 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003869 Py_INCREF(self);
3870 func = PyMethod_GET_FUNCTION(func);
3871 Py_INCREF(func);
3872 Py_DECREF(*pfunc);
3873 *pfunc = self;
3874 na++;
3875 n++;
3876 } else
3877 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003878 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003879 if (PyFunction_Check(func))
3880 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003881 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003882 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003883 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003884 Py_DECREF(func);
3885 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00003886
Armin Rigod34fa522006-03-28 19:10:40 +00003887 /* Clear the stack of the function object. Also removes
3888 the arguments in case they weren't consumed already
3889 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00003890 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003891 while ((*pp_stack) > pfunc) {
3892 w = EXT_POP(*pp_stack);
3893 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003894 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003895 }
3896 return x;
3897}
3898
Jeremy Hylton192690e2002-08-16 18:36:11 +00003899/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00003900 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00003901 For the simplest case -- a function that takes only positional
3902 arguments and is called with only positional arguments -- it
3903 inlines the most primitive frame setup code from
3904 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
3905 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00003906*/
3907
Fredrik Lundh7a830892006-05-27 10:39:48 +00003908static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00003909fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00003910{
Jeremy Hylton985eba52003-02-05 23:13:00 +00003911 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00003912 PyObject *globals = PyFunction_GET_GLOBALS(func);
3913 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
3914 PyObject **d = NULL;
3915 int nd = 0;
3916
Jeremy Hylton985eba52003-02-05 23:13:00 +00003917 PCALL(PCALL_FUNCTION);
3918 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00003919 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00003920 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
3921 PyFrameObject *f;
3922 PyObject *retval = NULL;
3923 PyThreadState *tstate = PyThreadState_GET();
3924 PyObject **fastlocals, **stack;
3925 int i;
3926
3927 PCALL(PCALL_FASTER_FUNCTION);
3928 assert(globals != NULL);
3929 /* XXX Perhaps we should create a specialized
3930 PyFrame_New() that doesn't take locals, but does
3931 take builtins without sanity checking them.
3932 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00003933 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003934 f = PyFrame_New(tstate, co, globals, NULL);
3935 if (f == NULL)
3936 return NULL;
3937
3938 fastlocals = f->f_localsplus;
3939 stack = (*pp_stack) - n;
3940
3941 for (i = 0; i < n; i++) {
3942 Py_INCREF(*stack);
3943 fastlocals[i] = *stack++;
3944 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00003945 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003946 ++tstate->recursion_depth;
3947 Py_DECREF(f);
3948 --tstate->recursion_depth;
3949 return retval;
3950 }
Jeremy Hylton52820442001-01-03 23:52:36 +00003951 if (argdefs != NULL) {
3952 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimese93237d2007-12-19 02:37:44 +00003953 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00003954 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00003955 return PyEval_EvalCodeEx(co, globals,
3956 (PyObject *)NULL, (*pp_stack)-n, na,
3957 (*pp_stack)-2*nk, nk, d, nd,
3958 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00003959}
3960
Fredrik Lundh7a830892006-05-27 10:39:48 +00003961static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00003962update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
3963 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00003964{
3965 PyObject *kwdict = NULL;
3966 if (orig_kwdict == NULL)
3967 kwdict = PyDict_New();
3968 else {
3969 kwdict = PyDict_Copy(orig_kwdict);
3970 Py_DECREF(orig_kwdict);
3971 }
3972 if (kwdict == NULL)
3973 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00003974 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00003975 int err;
3976 PyObject *value = EXT_POP(*pp_stack);
3977 PyObject *key = EXT_POP(*pp_stack);
3978 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003979 PyErr_Format(PyExc_TypeError,
3980 "%.200s%s got multiple values "
3981 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00003982 PyEval_GetFuncName(func),
3983 PyEval_GetFuncDesc(func),
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003984 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00003985 Py_DECREF(key);
3986 Py_DECREF(value);
3987 Py_DECREF(kwdict);
3988 return NULL;
3989 }
3990 err = PyDict_SetItem(kwdict, key, value);
3991 Py_DECREF(key);
3992 Py_DECREF(value);
3993 if (err) {
3994 Py_DECREF(kwdict);
3995 return NULL;
3996 }
3997 }
3998 return kwdict;
3999}
4000
Fredrik Lundh7a830892006-05-27 10:39:48 +00004001static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004002update_star_args(int nstack, int nstar, PyObject *stararg,
4003 PyObject ***pp_stack)
4004{
4005 PyObject *callargs, *w;
4006
4007 callargs = PyTuple_New(nstack + nstar);
4008 if (callargs == NULL) {
4009 return NULL;
4010 }
4011 if (nstar) {
4012 int i;
4013 for (i = 0; i < nstar; i++) {
4014 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4015 Py_INCREF(a);
4016 PyTuple_SET_ITEM(callargs, nstack + i, a);
4017 }
4018 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00004019 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00004020 w = EXT_POP(*pp_stack);
4021 PyTuple_SET_ITEM(callargs, nstack, w);
4022 }
4023 return callargs;
4024}
4025
Fredrik Lundh7a830892006-05-27 10:39:48 +00004026static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004027load_args(PyObject ***pp_stack, int na)
4028{
4029 PyObject *args = PyTuple_New(na);
4030 PyObject *w;
4031
4032 if (args == NULL)
4033 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00004034 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00004035 w = EXT_POP(*pp_stack);
4036 PyTuple_SET_ITEM(args, na, w);
4037 }
4038 return args;
4039}
4040
Fredrik Lundh7a830892006-05-27 10:39:48 +00004041static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004042do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4043{
4044 PyObject *callargs = NULL;
4045 PyObject *kwdict = NULL;
4046 PyObject *result = NULL;
4047
4048 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00004049 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00004050 if (kwdict == NULL)
4051 goto call_fail;
4052 }
4053 callargs = load_args(pp_stack, na);
4054 if (callargs == NULL)
4055 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004056#ifdef CALL_PROFILE
4057 /* At this point, we have to look at the type of func to
4058 update the call stats properly. Do it here so as to avoid
4059 exposing the call stats machinery outside ceval.c
4060 */
4061 if (PyFunction_Check(func))
4062 PCALL(PCALL_FUNCTION);
4063 else if (PyMethod_Check(func))
4064 PCALL(PCALL_METHOD);
4065 else if (PyType_Check(func))
4066 PCALL(PCALL_TYPE);
4067 else
4068 PCALL(PCALL_OTHER);
4069#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00004070 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00004071 call_fail:
4072 Py_XDECREF(callargs);
4073 Py_XDECREF(kwdict);
4074 return result;
4075}
4076
Fredrik Lundh7a830892006-05-27 10:39:48 +00004077static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004078ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4079{
4080 int nstar = 0;
4081 PyObject *callargs = NULL;
4082 PyObject *stararg = NULL;
4083 PyObject *kwdict = NULL;
4084 PyObject *result = NULL;
4085
4086 if (flags & CALL_FLAG_KW) {
4087 kwdict = EXT_POP(*pp_stack);
Georg Brandl2134e752007-05-21 20:34:16 +00004088 if (!PyDict_Check(kwdict)) {
4089 PyObject *d;
4090 d = PyDict_New();
4091 if (d == NULL)
4092 goto ext_call_fail;
4093 if (PyDict_Update(d, kwdict) != 0) {
4094 Py_DECREF(d);
4095 /* PyDict_Update raises attribute
4096 * error (percolated from an attempt
4097 * to get 'keys' attribute) instead of
4098 * a type error if its second argument
4099 * is not a mapping.
4100 */
4101 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4102 PyErr_Format(PyExc_TypeError,
4103 "%.200s%.200s argument after ** "
4104 "must be a mapping, not %.200s",
4105 PyEval_GetFuncName(func),
4106 PyEval_GetFuncDesc(func),
4107 kwdict->ob_type->tp_name);
4108 }
4109 goto ext_call_fail;
4110 }
4111 Py_DECREF(kwdict);
4112 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00004113 }
4114 }
4115 if (flags & CALL_FLAG_VAR) {
4116 stararg = EXT_POP(*pp_stack);
4117 if (!PyTuple_Check(stararg)) {
4118 PyObject *t = NULL;
4119 t = PySequence_Tuple(stararg);
4120 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00004121 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4122 PyErr_Format(PyExc_TypeError,
Georg Brandl2134e752007-05-21 20:34:16 +00004123 "%.200s%.200s argument after * "
4124 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00004125 PyEval_GetFuncName(func),
Georg Brandl2134e752007-05-21 20:34:16 +00004126 PyEval_GetFuncDesc(func),
4127 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00004128 }
Jeremy Hylton52820442001-01-03 23:52:36 +00004129 goto ext_call_fail;
4130 }
4131 Py_DECREF(stararg);
4132 stararg = t;
4133 }
4134 nstar = PyTuple_GET_SIZE(stararg);
4135 }
4136 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00004137 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00004138 if (kwdict == NULL)
4139 goto ext_call_fail;
4140 }
4141 callargs = update_star_args(na, nstar, stararg, pp_stack);
4142 if (callargs == NULL)
4143 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004144#ifdef CALL_PROFILE
4145 /* At this point, we have to look at the type of func to
4146 update the call stats properly. Do it here so as to avoid
4147 exposing the call stats machinery outside ceval.c
4148 */
4149 if (PyFunction_Check(func))
4150 PCALL(PCALL_FUNCTION);
4151 else if (PyMethod_Check(func))
4152 PCALL(PCALL_METHOD);
4153 else if (PyType_Check(func))
4154 PCALL(PCALL_TYPE);
4155 else
4156 PCALL(PCALL_OTHER);
4157#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +00004158 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersae406c62007-09-19 17:27:43 +00004159ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00004160 Py_XDECREF(callargs);
4161 Py_XDECREF(kwdict);
4162 Py_XDECREF(stararg);
4163 return result;
4164}
4165
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004166/* Extract a slice index from a PyInt or PyLong or an object with the
4167 nb_index slot defined, and store in *pi.
4168 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4169 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 +00004170 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004171*/
Tim Petersb5196382001-12-16 19:44:20 +00004172/* Note: If v is NULL, return success without storing into *pi. This
4173 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4174 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004175*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004176int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004177_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004178{
Tim Petersb5196382001-12-16 19:44:20 +00004179 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004180 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00004181 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004182 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
4183 however, it looks like it should be AsSsize_t.
4184 There should be a comment here explaining why.
4185 */
4186 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00004187 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004188 else if (PyIndex_Check(v)) {
4189 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004190 if (x == -1 && PyErr_Occurred())
4191 return 0;
4192 }
4193 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00004194 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004195 "slice indices must be integers or "
4196 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00004197 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004198 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00004199 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004200 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00004201 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004202}
4203
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004204#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004205#define ISINDEX(x) ((x) == NULL || \
4206 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00004207
Fredrik Lundh7a830892006-05-27 10:39:48 +00004208static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004209apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004210{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004211 PyTypeObject *tp = u->ob_type;
4212 PySequenceMethods *sq = tp->tp_as_sequence;
4213
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004214 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004215 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004216 if (!_PyEval_SliceIndex(v, &ilow))
4217 return NULL;
4218 if (!_PyEval_SliceIndex(w, &ihigh))
4219 return NULL;
4220 return PySequence_GetSlice(u, ilow, ihigh);
4221 }
4222 else {
4223 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00004224 if (slice != NULL) {
4225 PyObject *res = PyObject_GetItem(u, slice);
4226 Py_DECREF(slice);
4227 return res;
4228 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00004229 else
4230 return NULL;
4231 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004232}
Guido van Rossum3f5da241990-12-20 15:06:42 +00004233
Fredrik Lundh7a830892006-05-27 10:39:48 +00004234static int
Guido van Rossumac7be682001-01-17 15:42:30 +00004235assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
4236 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004237{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004238 PyTypeObject *tp = u->ob_type;
4239 PySequenceMethods *sq = tp->tp_as_sequence;
4240
Georg Brandl0fca97a2007-03-05 22:28:08 +00004241 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004242 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004243 if (!_PyEval_SliceIndex(v, &ilow))
4244 return -1;
4245 if (!_PyEval_SliceIndex(w, &ihigh))
4246 return -1;
4247 if (x == NULL)
4248 return PySequence_DelSlice(u, ilow, ihigh);
4249 else
4250 return PySequence_SetSlice(u, ilow, ihigh, x);
4251 }
4252 else {
4253 PyObject *slice = PySlice_New(v, w, NULL);
4254 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00004255 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004256 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00004257 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00004258 else
Guido van Rossum354797c2001-12-03 19:45:06 +00004259 res = PyObject_DelItem(u, slice);
4260 Py_DECREF(slice);
4261 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004262 }
4263 else
4264 return -1;
4265 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004266}
4267
Guido van Rossum04edb522008-03-18 02:49:46 +00004268#define Py3kExceptionClass_Check(x) \
4269 (PyType_Check((x)) && \
4270 PyType_FastSubclass((PyTypeObject*)(x), Py_TPFLAGS_BASE_EXC_SUBCLASS))
4271
4272#define CANNOT_CATCH_MSG "catching classes that don't inherit from " \
Georg Brandld5b635f2008-03-25 08:29:14 +00004273 "BaseException is not allowed in 3.x"
Guido van Rossum04edb522008-03-18 02:49:46 +00004274
Fredrik Lundh7a830892006-05-27 10:39:48 +00004275static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004276cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004277{
Guido van Rossumac7be682001-01-17 15:42:30 +00004278 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004279 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00004280 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00004281 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004282 break;
4283 case PyCmp_IS_NOT:
4284 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004285 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004286 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004287 res = PySequence_Contains(w, v);
4288 if (res < 0)
4289 return NULL;
4290 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004291 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00004292 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004293 if (res < 0)
4294 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004295 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004296 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004297 case PyCmp_EXC_MATCH:
Brett Cannon129bd522007-01-30 21:34:36 +00004298 if (PyTuple_Check(w)) {
4299 Py_ssize_t i, length;
4300 length = PyTuple_Size(w);
4301 for (i = 0; i < length; i += 1) {
4302 PyObject *exc = PyTuple_GET_ITEM(w, i);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004303 if (PyString_Check(exc)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004304 int ret_val;
4305 ret_val = PyErr_WarnEx(
Thomas Wouterse2176022007-09-20 17:35:10 +00004306 PyExc_DeprecationWarning,
4307 "catching of string "
4308 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004309 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004310 return NULL;
4311 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004312 else if (Py_Py3kWarningFlag &&
4313 !PyTuple_Check(exc) &&
4314 !Py3kExceptionClass_Check(exc))
Guido van Rossum04edb522008-03-18 02:49:46 +00004315 {
4316 int ret_val;
4317 ret_val = PyErr_WarnEx(
4318 PyExc_DeprecationWarning,
4319 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004320 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004321 return NULL;
4322 }
Brett Cannon129bd522007-01-30 21:34:36 +00004323 }
4324 }
4325 else {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004326 if (PyString_Check(w)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004327 int ret_val;
4328 ret_val = PyErr_WarnEx(
4329 PyExc_DeprecationWarning,
4330 "catching of string "
Thomas Wouterse2176022007-09-20 17:35:10 +00004331 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004332 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004333 return NULL;
4334 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004335 else if (Py_Py3kWarningFlag &&
4336 !PyTuple_Check(w) &&
4337 !Py3kExceptionClass_Check(w))
Guido van Rossum04edb522008-03-18 02:49:46 +00004338 {
4339 int ret_val;
4340 ret_val = PyErr_WarnEx(
4341 PyExc_DeprecationWarning,
4342 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004343 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004344 return NULL;
4345 }
Brett Cannon129bd522007-01-30 21:34:36 +00004346 }
Barry Warsaw4249f541997-08-22 21:26:19 +00004347 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004348 break;
4349 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00004350 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004351 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004352 v = res ? Py_True : Py_False;
4353 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004354 return v;
4355}
4356
Fredrik Lundh7a830892006-05-27 10:39:48 +00004357static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00004358import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004359{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004360 PyObject *x;
4361
4362 x = PyObject_GetAttr(v, name);
4363 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004364 PyErr_Format(PyExc_ImportError,
4365 "cannot import name %.230s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004366 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004367 }
Thomas Wouters52152252000-08-17 22:55:00 +00004368 return x;
4369}
Guido van Rossumac7be682001-01-17 15:42:30 +00004370
Fredrik Lundh7a830892006-05-27 10:39:48 +00004371static int
Thomas Wouters52152252000-08-17 22:55:00 +00004372import_all_from(PyObject *locals, PyObject *v)
4373{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004374 PyObject *all = PyObject_GetAttrString(v, "__all__");
4375 PyObject *dict, *name, *value;
4376 int skip_leading_underscores = 0;
4377 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004378
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004379 if (all == NULL) {
4380 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4381 return -1; /* Unexpected error */
4382 PyErr_Clear();
4383 dict = PyObject_GetAttrString(v, "__dict__");
4384 if (dict == NULL) {
4385 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4386 return -1;
4387 PyErr_SetString(PyExc_ImportError,
4388 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004389 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004390 }
4391 all = PyMapping_Keys(dict);
4392 Py_DECREF(dict);
4393 if (all == NULL)
4394 return -1;
4395 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004396 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004397
4398 for (pos = 0, err = 0; ; pos++) {
4399 name = PySequence_GetItem(all, pos);
4400 if (name == NULL) {
4401 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4402 err = -1;
4403 else
4404 PyErr_Clear();
4405 break;
4406 }
4407 if (skip_leading_underscores &&
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004408 PyString_Check(name) &&
4409 PyString_AS_STRING(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004410 {
4411 Py_DECREF(name);
4412 continue;
4413 }
4414 value = PyObject_GetAttr(v, name);
4415 if (value == NULL)
4416 err = -1;
Armin Rigo70370852006-11-29 21:59:22 +00004417 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004418 err = PyDict_SetItem(locals, name, value);
Armin Rigo70370852006-11-29 21:59:22 +00004419 else
4420 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004421 Py_DECREF(name);
4422 Py_XDECREF(value);
4423 if (err != 0)
4424 break;
4425 }
4426 Py_DECREF(all);
4427 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004428}
4429
Fredrik Lundh7a830892006-05-27 10:39:48 +00004430static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004431build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004432{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004433 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004434
4435 if (PyDict_Check(methods))
4436 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004437 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004438 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004439 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4440 base = PyTuple_GET_ITEM(bases, 0);
4441 metaclass = PyObject_GetAttrString(base, "__class__");
4442 if (metaclass == NULL) {
4443 PyErr_Clear();
4444 metaclass = (PyObject *)base->ob_type;
4445 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004446 }
4447 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004448 else {
4449 PyObject *g = PyEval_GetGlobals();
4450 if (g != NULL && PyDict_Check(g))
4451 metaclass = PyDict_GetItemString(g, "__metaclass__");
4452 if (metaclass == NULL)
4453 metaclass = (PyObject *) &PyClass_Type;
4454 Py_INCREF(metaclass);
4455 }
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004456 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods,
Thomas Woutersae406c62007-09-19 17:27:43 +00004457 NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004458 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004459 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004460 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004461 in a base that was not a class (such the random module
4462 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004463 by augmenting the error message with more information.*/
4464
4465 PyObject *ptype, *pvalue, *ptraceback;
4466
4467 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004468 if (PyString_Check(pvalue)) {
Raymond Hettingercfc31922004-09-16 16:41:57 +00004469 PyObject *newmsg;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004470 newmsg = PyString_FromFormat(
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004471 "Error when calling the metaclass bases\n"
Thomas Woutersae406c62007-09-19 17:27:43 +00004472 " %s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004473 PyString_AS_STRING(pvalue));
Raymond Hettingercfc31922004-09-16 16:41:57 +00004474 if (newmsg != NULL) {
4475 Py_DECREF(pvalue);
4476 pvalue = newmsg;
4477 }
4478 }
4479 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004480 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004481 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004482}
4483
Fredrik Lundh7a830892006-05-27 10:39:48 +00004484static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004485exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4486 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004487{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004488 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004489 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004490 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004491
Guido van Rossumb209a111997-04-29 18:18:01 +00004492 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4493 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004494 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004495 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004496 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004497 locals = PyTuple_GetItem(prog, 2);
4498 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004499 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004500 if (globals == Py_None) {
4501 globals = PyEval_GetGlobals();
4502 if (locals == Py_None) {
4503 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004504 plain = 1;
4505 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004506 if (!globals || !locals) {
4507 PyErr_SetString(PyExc_SystemError,
4508 "globals and locals cannot be NULL");
4509 return -1;
4510 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004511 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004512 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004513 locals = globals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004514 if (!PyString_Check(prog) &&
Benjamin Peterson78821dd2009-01-25 17:15:10 +00004515#ifdef Py_USING_UNICODE
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004516 !PyUnicode_Check(prog) &&
Benjamin Peterson78821dd2009-01-25 17:15:10 +00004517#endif
Guido van Rossumb209a111997-04-29 18:18:01 +00004518 !PyCode_Check(prog) &&
4519 !PyFile_Check(prog)) {
4520 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004521 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004522 return -1;
4523 }
Fred Drake661ea262000-10-24 19:57:45 +00004524 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004525 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004526 "exec: arg 2 must be a dictionary or None");
4527 return -1;
4528 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004529 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004530 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004531 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004532 return -1;
4533 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004534 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004535 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004536 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004537 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4538 PyErr_SetString(PyExc_TypeError,
4539 "code object passed to exec may not contain free variables");
4540 return -1;
4541 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004542 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004543 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004544 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004545 FILE *fp = PyFile_AsFile(prog);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004546 char *name = PyString_AsString(PyFile_Name(prog));
Jeremy Hylton714b1122007-02-25 16:01:58 +00004547 PyCompilerFlags cf;
Thomas Woutersae406c62007-09-19 17:27:43 +00004548 if (name == NULL)
4549 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004550 cf.cf_flags = 0;
4551 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004552 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004553 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004554 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004555 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004556 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004557 }
4558 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004559 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004560 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004561 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004562 cf.cf_flags = 0;
4563#ifdef Py_USING_UNICODE
4564 if (PyUnicode_Check(prog)) {
4565 tmp = PyUnicode_AsUTF8String(prog);
4566 if (tmp == NULL)
4567 return -1;
4568 prog = tmp;
4569 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4570 }
4571#endif
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004572 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004573 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004574 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004575 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004576 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004577 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004578 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004579 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004580 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004581 if (plain)
4582 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004583 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004584 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004585 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004586 return 0;
4587}
Guido van Rossum24c13741995-02-14 09:42:43 +00004588
Fredrik Lundh7a830892006-05-27 10:39:48 +00004589static void
Paul Prescode68140d2000-08-30 20:25:01 +00004590format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4591{
4592 char *obj_str;
4593
4594 if (!obj)
4595 return;
4596
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004597 obj_str = PyString_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00004598 if (!obj_str)
4599 return;
4600
4601 PyErr_Format(exc, format_str, obj_str);
4602}
Guido van Rossum950361c1997-01-24 13:49:28 +00004603
Fredrik Lundh7a830892006-05-27 10:39:48 +00004604static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004605string_concatenate(PyObject *v, PyObject *w,
4606 PyFrameObject *f, unsigned char *next_instr)
4607{
4608 /* This function implements 'variable += expr' when both arguments
4609 are strings. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004610 Py_ssize_t v_len = PyString_GET_SIZE(v);
4611 Py_ssize_t w_len = PyString_GET_SIZE(w);
Armin Rigo97ff0472006-08-09 15:37:26 +00004612 Py_ssize_t new_len = v_len + w_len;
4613 if (new_len < 0) {
4614 PyErr_SetString(PyExc_OverflowError,
4615 "strings are too large to concat");
4616 return NULL;
4617 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004618
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004619 if (v->ob_refcnt == 2) {
4620 /* In the common case, there are 2 references to the value
4621 * stored in 'variable' when the += is performed: one on the
Thomas Wouterse2176022007-09-20 17:35:10 +00004622 * value stack (in 'v') and one still stored in the
4623 * 'variable'. We try to delete the variable now to reduce
4624 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004625 */
4626 switch (*next_instr) {
4627 case STORE_FAST:
4628 {
4629 int oparg = PEEKARG();
4630 PyObject **fastlocals = f->f_localsplus;
4631 if (GETLOCAL(oparg) == v)
4632 SETLOCAL(oparg, NULL);
4633 break;
4634 }
4635 case STORE_DEREF:
4636 {
Thomas Wouterse2176022007-09-20 17:35:10 +00004637 PyObject **freevars = (f->f_localsplus +
4638 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004639 PyObject *c = freevars[PEEKARG()];
4640 if (PyCell_GET(c) == v)
4641 PyCell_Set(c, NULL);
4642 break;
4643 }
4644 case STORE_NAME:
4645 {
4646 PyObject *names = f->f_code->co_names;
4647 PyObject *name = GETITEM(names, PEEKARG());
4648 PyObject *locals = f->f_locals;
4649 if (PyDict_CheckExact(locals) &&
4650 PyDict_GetItem(locals, name) == v) {
4651 if (PyDict_DelItem(locals, name) != 0) {
4652 PyErr_Clear();
4653 }
4654 }
4655 break;
4656 }
4657 }
4658 }
4659
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004660 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004661 /* Now we own the last reference to 'v', so we can resize it
4662 * in-place.
4663 */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004664 if (_PyString_Resize(&v, new_len) != 0) {
4665 /* XXX if _PyString_Resize() fails, 'v' has been
Thomas Wouterse2176022007-09-20 17:35:10 +00004666 * deallocated so it cannot be put back into
4667 * 'variable'. The MemoryError is raised when there
4668 * is no value in 'variable', which might (very
4669 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004670 */
4671 return NULL;
4672 }
4673 /* copy 'w' into the newly allocated area of 'v' */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004674 memcpy(PyString_AS_STRING(v) + v_len,
4675 PyString_AS_STRING(w), w_len);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004676 return v;
4677 }
4678 else {
4679 /* When in-place resizing is not an option. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004680 PyString_Concat(&v, w);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004681 return v;
4682 }
4683}
4684
Guido van Rossum950361c1997-01-24 13:49:28 +00004685#ifdef DYNAMIC_EXECUTION_PROFILE
4686
Fredrik Lundh7a830892006-05-27 10:39:48 +00004687static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004688getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004689{
4690 int i;
4691 PyObject *l = PyList_New(256);
4692 if (l == NULL) return NULL;
4693 for (i = 0; i < 256; i++) {
4694 PyObject *x = PyInt_FromLong(a[i]);
4695 if (x == NULL) {
4696 Py_DECREF(l);
4697 return NULL;
4698 }
4699 PyList_SetItem(l, i, x);
4700 }
4701 for (i = 0; i < 256; i++)
4702 a[i] = 0;
4703 return l;
4704}
4705
4706PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004707_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004708{
4709#ifndef DXPAIRS
4710 return getarray(dxp);
4711#else
4712 int i;
4713 PyObject *l = PyList_New(257);
4714 if (l == NULL) return NULL;
4715 for (i = 0; i < 257; i++) {
4716 PyObject *x = getarray(dxpairs[i]);
4717 if (x == NULL) {
4718 Py_DECREF(l);
4719 return NULL;
4720 }
4721 PyList_SetItem(l, i, x);
4722 }
4723 return l;
4724#endif
4725}
4726
4727#endif