blob: e5e70463f74dc369547245fe8ed6f4ee108a9689 [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
Mark Dickinson504a1512009-10-31 10:11:28 +000054#elif defined(__i386__)
55
56/* this is for linux/x86 (and probably any other GCC/x86 combo) */
Michael W. Hudson800ba232004-08-12 18:19:17 +000057
Michael W. Hudson75eabd22005-01-18 15:56:11 +000058#define READ_TIMESTAMP(val) \
59 __asm__ __volatile__("rdtsc" : "=A" (val))
Michael W. Hudson800ba232004-08-12 18:19:17 +000060
Mark Dickinson504a1512009-10-31 10:11:28 +000061#elif defined(__x86_64__)
62
63/* for gcc/x86_64, the "A" constraint in DI mode means *either* rax *or* rdx;
64 not edx:eax as it does for i386. Since rdtsc puts its result in edx:eax
65 even in 64-bit mode, we need to use "a" and "d" for the lower and upper
66 32-bit pieces of the result. */
67
68#define READ_TIMESTAMP(val) \
69 __asm__ __volatile__("rdtsc" : \
70 "=a" (((int*)&(val))[0]), "=d" (((int*)&(val))[1]));
71
72
73#else
74
75#error "Don't know how to implement timestamp counter for this architecture"
76
Michael W. Hudson800ba232004-08-12 18:19:17 +000077#endif
78
Tim Peters7df5e7f2006-05-26 23:14:37 +000079void dump_tsc(int opcode, int ticked, uint64 inst0, uint64 inst1,
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000080 uint64 loop0, uint64 loop1, uint64 intr0, uint64 intr1)
81{
82 uint64 intr, inst, loop;
83 PyThreadState *tstate = PyThreadState_Get();
84 if (!tstate->interp->tscdump)
85 return;
86 intr = intr1 - intr0;
87 inst = inst1 - inst0 - intr;
88 loop = loop1 - loop0 - intr;
89 fprintf(stderr, "opcode=%03d t=%d inst=%06lld loop=%06lld\n",
90 opcode, ticked, inst, loop);
91}
Michael W. Hudson800ba232004-08-12 18:19:17 +000092
Martin v. Löwisf30d60e2004-06-08 08:17:44 +000093#endif
94
Guido van Rossum04691fc1992-08-12 15:35:34 +000095/* Turn this on if your compiler chokes on the big switch: */
Guido van Rossum1ae940a1995-01-02 19:04:15 +000096/* #define CASE_TOO_BIG 1 */
Guido van Rossum04691fc1992-08-12 15:35:34 +000097
Guido van Rossum408027e1996-12-30 16:17:54 +000098#ifdef Py_DEBUG
Guido van Rossum96a42c81992-01-12 02:29:51 +000099/* For debugging the interpreter: */
100#define LLTRACE 1 /* Low-level trace feature */
101#define CHECKEXC 1 /* Double-check exception checking */
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000102#endif
103
Jeremy Hylton52820442001-01-03 23:52:36 +0000104typedef PyObject *(*callproc)(PyObject *, PyObject *, PyObject *);
Guido van Rossum5b722181993-03-30 17:46:03 +0000105
Guido van Rossum374a9221991-04-04 10:40:29 +0000106/* Forward declarations */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000107#ifdef WITH_TSC
Fredrik Lundh7a830892006-05-27 10:39:48 +0000108static PyObject * call_function(PyObject ***, int, uint64*, uint64*);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000109#else
Fredrik Lundh7a830892006-05-27 10:39:48 +0000110static PyObject * call_function(PyObject ***, int);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000111#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +0000112static PyObject * fast_function(PyObject *, PyObject ***, int, int, int);
113static PyObject * do_call(PyObject *, PyObject ***, int, int);
114static PyObject * ext_do_call(PyObject *, PyObject ***, int, int, int);
Thomas Wouterse2176022007-09-20 17:35:10 +0000115static PyObject * update_keyword_args(PyObject *, int, PyObject ***,
116 PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000117static PyObject * update_star_args(int, int, PyObject *, PyObject ***);
118static PyObject * load_args(PyObject ***, int);
Jeremy Hylton52820442001-01-03 23:52:36 +0000119#define CALL_FLAG_VAR 1
120#define CALL_FLAG_KW 2
121
Guido van Rossum0a066c01992-03-27 17:29:15 +0000122#ifdef LLTRACE
Fredrik Lundh1b949402006-05-26 12:01:49 +0000123static int lltrace;
Fredrik Lundh7a830892006-05-27 10:39:48 +0000124static int prtrace(PyObject *, char *);
Guido van Rossum0a066c01992-03-27 17:29:15 +0000125#endif
Fredrik Lundh7a830892006-05-27 10:39:48 +0000126static int call_trace(Py_tracefunc, PyObject *, PyFrameObject *,
Fred Drake5755ce62001-06-27 19:19:46 +0000127 int, PyObject *);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000128static int call_trace_protected(Py_tracefunc, PyObject *,
Armin Rigo1c2d7e52005-09-20 18:34:01 +0000129 PyFrameObject *, int, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000130static void call_exc_trace(Py_tracefunc, PyObject *, PyFrameObject *);
131static int maybe_call_line_trace(Py_tracefunc, PyObject *,
Armin Rigobf57a142004-03-22 19:24:58 +0000132 PyFrameObject *, int *, int *, int *);
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000133
Fredrik Lundh7a830892006-05-27 10:39:48 +0000134static PyObject * apply_slice(PyObject *, PyObject *, PyObject *);
135static int assign_slice(PyObject *, PyObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000136 PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000137static PyObject * cmp_outcome(int, PyObject *, PyObject *);
138static PyObject * import_from(PyObject *, PyObject *);
139static int import_all_from(PyObject *, PyObject *);
140static PyObject * build_class(PyObject *, PyObject *, PyObject *);
141static int exec_statement(PyFrameObject *,
Tim Petersdbd9ba62000-07-09 03:09:57 +0000142 PyObject *, PyObject *, PyObject *);
Fredrik Lundh7a830892006-05-27 10:39:48 +0000143static void set_exc_info(PyThreadState *, PyObject *, PyObject *, PyObject *);
144static void reset_exc_info(PyThreadState *);
145static void format_exc_check_arg(PyObject *, char *, PyObject *);
146static PyObject * string_concatenate(PyObject *, PyObject *,
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000147 PyFrameObject *, unsigned char *);
Benjamin Petersone18ef192009-01-20 14:21:16 +0000148static PyObject * kwd_as_string(PyObject *);
Benjamin Peterson1880d8b2009-05-25 13:13:44 +0000149static PyObject * special_lookup(PyObject *, char *, PyObject **);
Guido van Rossum374a9221991-04-04 10:40:29 +0000150
Paul Prescode68140d2000-08-30 20:25:01 +0000151#define NAME_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000152 "name '%.200s' is not defined"
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000153#define GLOBAL_NAME_ERROR_MSG \
154 "global name '%.200s' is not defined"
Paul Prescode68140d2000-08-30 20:25:01 +0000155#define UNBOUNDLOCAL_ERROR_MSG \
Fred Drake661ea262000-10-24 19:57:45 +0000156 "local variable '%.200s' referenced before assignment"
Jeremy Hyltonc76770c2001-04-13 16:51:46 +0000157#define UNBOUNDFREE_ERROR_MSG \
158 "free variable '%.200s' referenced before assignment" \
159 " in enclosing scope"
Guido van Rossum374a9221991-04-04 10:40:29 +0000160
Guido van Rossum950361c1997-01-24 13:49:28 +0000161/* Dynamic execution profile */
162#ifdef DYNAMIC_EXECUTION_PROFILE
163#ifdef DXPAIRS
164static long dxpairs[257][256];
165#define dxp dxpairs[256]
166#else
167static long dxp[256];
168#endif
169#endif
170
Jeremy Hylton985eba52003-02-05 23:13:00 +0000171/* Function call profile */
172#ifdef CALL_PROFILE
173#define PCALL_NUM 11
174static int pcall[PCALL_NUM];
175
176#define PCALL_ALL 0
177#define PCALL_FUNCTION 1
178#define PCALL_FAST_FUNCTION 2
179#define PCALL_FASTER_FUNCTION 3
180#define PCALL_METHOD 4
181#define PCALL_BOUND_METHOD 5
182#define PCALL_CFUNCTION 6
183#define PCALL_TYPE 7
184#define PCALL_GENERATOR 8
185#define PCALL_OTHER 9
186#define PCALL_POP 10
187
188/* Notes about the statistics
189
190 PCALL_FAST stats
191
192 FAST_FUNCTION means no argument tuple needs to be created.
193 FASTER_FUNCTION means that the fast-path frame setup code is used.
194
195 If there is a method call where the call can be optimized by changing
196 the argument tuple and calling the function directly, it gets recorded
197 twice.
198
199 As a result, the relationship among the statistics appears to be
200 PCALL_ALL == PCALL_FUNCTION + PCALL_METHOD - PCALL_BOUND_METHOD +
201 PCALL_CFUNCTION + PCALL_TYPE + PCALL_GENERATOR + PCALL_OTHER
202 PCALL_FUNCTION > PCALL_FAST_FUNCTION > PCALL_FASTER_FUNCTION
203 PCALL_METHOD > PCALL_BOUND_METHOD
204*/
205
206#define PCALL(POS) pcall[POS]++
207
208PyObject *
209PyEval_GetCallStats(PyObject *self)
210{
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000211 return Py_BuildValue("iiiiiiiiiii",
Jeremy Hylton985eba52003-02-05 23:13:00 +0000212 pcall[0], pcall[1], pcall[2], pcall[3],
213 pcall[4], pcall[5], pcall[6], pcall[7],
Andrew M. Kuchling1f3ebe02006-10-27 13:22:46 +0000214 pcall[8], pcall[9], pcall[10]);
Jeremy Hylton985eba52003-02-05 23:13:00 +0000215}
216#else
217#define PCALL(O)
218
219PyObject *
220PyEval_GetCallStats(PyObject *self)
221{
222 Py_INCREF(Py_None);
223 return Py_None;
224}
225#endif
226
Tim Peters5ca576e2001-06-18 22:08:13 +0000227
Guido van Rossume59214e1994-08-30 08:01:59 +0000228#ifdef WITH_THREAD
Guido van Rossumff4949e1992-08-05 19:58:53 +0000229
Martin v. Löwis0e8bd7e2006-06-10 12:23:46 +0000230#ifdef HAVE_ERRNO_H
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000231#include <errno.h>
Guido van Rossum2571cc81999-04-07 16:07:23 +0000232#endif
Guido van Rossum49b56061998-10-01 20:42:43 +0000233#include "pythread.h"
Guido van Rossumff4949e1992-08-05 19:58:53 +0000234
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +0000235static PyThread_type_lock interpreter_lock = 0; /* This is the GIL */
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000236static PyThread_type_lock pending_lock = 0; /* for pending calls */
Guido van Rossuma9672091994-09-14 13:31:22 +0000237static long main_thread = 0;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000238
Tim Peters7f468f22004-10-11 02:40:51 +0000239int
240PyEval_ThreadsInitialized(void)
241{
242 return interpreter_lock != 0;
243}
244
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000245void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000246PyEval_InitThreads(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000247{
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000248 if (interpreter_lock)
Sjoerd Mullendered59d201993-01-06 13:36:38 +0000249 return;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000250 interpreter_lock = PyThread_allocate_lock();
251 PyThread_acquire_lock(interpreter_lock, 1);
252 main_thread = PyThread_get_thread_ident();
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000253}
Guido van Rossumff4949e1992-08-05 19:58:53 +0000254
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000255void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000256PyEval_AcquireLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000257{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000258 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000259}
260
261void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000262PyEval_ReleaseLock(void)
Guido van Rossum25ce5661997-08-02 03:10:38 +0000263{
Guido van Rossum65d5b571998-12-21 19:32:43 +0000264 PyThread_release_lock(interpreter_lock);
Guido van Rossum25ce5661997-08-02 03:10:38 +0000265}
266
267void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000268PyEval_AcquireThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000269{
270 if (tstate == NULL)
271 Py_FatalError("PyEval_AcquireThread: NULL new thread state");
Mark Hammond8d98d2c2003-04-19 15:41:53 +0000272 /* Check someone has called PyEval_InitThreads() to create the lock */
273 assert(interpreter_lock);
Guido van Rossum65d5b571998-12-21 19:32:43 +0000274 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000275 if (PyThreadState_Swap(tstate) != NULL)
276 Py_FatalError(
277 "PyEval_AcquireThread: non-NULL old thread state");
278}
279
280void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000281PyEval_ReleaseThread(PyThreadState *tstate)
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000282{
283 if (tstate == NULL)
284 Py_FatalError("PyEval_ReleaseThread: NULL thread state");
285 if (PyThreadState_Swap(NULL) != tstate)
286 Py_FatalError("PyEval_ReleaseThread: wrong thread state");
Guido van Rossum65d5b571998-12-21 19:32:43 +0000287 PyThread_release_lock(interpreter_lock);
Guido van Rossum9cc8a201997-07-19 19:55:50 +0000288}
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000289
290/* This function is called from PyOS_AfterFork to ensure that newly
291 created child processes don't hold locks referring to threads which
292 are not running in the child process. (This could also be done using
293 pthread_atfork mechanism, at least for the pthreads implementation.) */
294
295void
296PyEval_ReInitThreads(void)
297{
Jesse Noller5e62ca42008-07-16 20:03:47 +0000298 PyObject *threading, *result;
299 PyThreadState *tstate;
300
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000301 if (!interpreter_lock)
302 return;
303 /*XXX Can't use PyThread_free_lock here because it does too
304 much error-checking. Doing this cleanly would require
305 adding a new function to each thread_*.h. Instead, just
306 create a new lock and waste a little bit of memory */
307 interpreter_lock = PyThread_allocate_lock();
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000308 pending_lock = PyThread_allocate_lock();
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000309 PyThread_acquire_lock(interpreter_lock, 1);
310 main_thread = PyThread_get_thread_ident();
Jesse Noller5e62ca42008-07-16 20:03:47 +0000311
312 /* Update the threading module with the new state.
313 */
314 tstate = PyThreadState_GET();
315 threading = PyMapping_GetItemString(tstate->interp->modules,
316 "threading");
317 if (threading == NULL) {
318 /* threading not imported */
319 PyErr_Clear();
320 return;
321 }
322 result = PyObject_CallMethod(threading, "_after_fork", NULL);
323 if (result == NULL)
324 PyErr_WriteUnraisable(threading);
325 else
326 Py_DECREF(result);
327 Py_DECREF(threading);
Guido van Rossumfee3a2d2000-08-27 17:34:07 +0000328}
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000329#endif
330
Guido van Rossumff4949e1992-08-05 19:58:53 +0000331/* Functions save_thread and restore_thread are always defined so
332 dynamically loaded modules needn't be compiled separately for use
333 with and without threads: */
334
Guido van Rossum2fca21f71997-07-18 23:56:58 +0000335PyThreadState *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000336PyEval_SaveThread(void)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000337{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000338 PyThreadState *tstate = PyThreadState_Swap(NULL);
339 if (tstate == NULL)
340 Py_FatalError("PyEval_SaveThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000341#ifdef WITH_THREAD
Guido van Rossumb74eca91997-09-30 22:03:16 +0000342 if (interpreter_lock)
Guido van Rossum65d5b571998-12-21 19:32:43 +0000343 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000344#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000345 return tstate;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000346}
347
348void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000349PyEval_RestoreThread(PyThreadState *tstate)
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000350{
Guido van Rossumb74eca91997-09-30 22:03:16 +0000351 if (tstate == NULL)
352 Py_FatalError("PyEval_RestoreThread: NULL tstate");
Guido van Rossume59214e1994-08-30 08:01:59 +0000353#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000354 if (interpreter_lock) {
Guido van Rossumb74eca91997-09-30 22:03:16 +0000355 int err = errno;
Guido van Rossum65d5b571998-12-21 19:32:43 +0000356 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000357 errno = err;
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000358 }
359#endif
Guido van Rossumb74eca91997-09-30 22:03:16 +0000360 PyThreadState_Swap(tstate);
Guido van Rossum1984f1e1992-08-04 12:41:02 +0000361}
362
363
Guido van Rossuma9672091994-09-14 13:31:22 +0000364/* Mechanism whereby asynchronously executing callbacks (e.g. UNIX
365 signal handlers or Mac I/O completion routines) can schedule calls
366 to a function to be called synchronously.
367 The synchronous function is called with one void* argument.
368 It should return 0 for success or -1 for failure -- failure should
369 be accompanied by an exception.
370
371 If registry succeeds, the registry function returns 0; if it fails
372 (e.g. due to too many pending calls) it returns -1 (without setting
373 an exception condition).
374
375 Note that because registry may occur from within signal handlers,
376 or other asynchronous events, calling malloc() is unsafe!
377
378#ifdef WITH_THREAD
379 Any thread can schedule pending calls, but only the main thread
380 will execute them.
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000381 There is no facility to schedule calls to a particular thread, but
382 that should be easy to change, should that ever be required. In
383 that case, the static variables here should go into the python
384 threadstate.
Guido van Rossuma9672091994-09-14 13:31:22 +0000385#endif
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000386*/
Guido van Rossuma9672091994-09-14 13:31:22 +0000387
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000388#ifdef WITH_THREAD
389
390/* The WITH_THREAD implementation is thread-safe. It allows
391 scheduling to be made from any thread, and even from an executing
392 callback.
393 */
394
395#define NPENDINGCALLS 32
396static struct {
397 int (*func)(void *);
398 void *arg;
399} pendingcalls[NPENDINGCALLS];
400static int pendingfirst = 0;
401static int pendinglast = 0;
402static volatile int pendingcalls_to_do = 1; /* trigger initialization of lock */
403static char pendingbusy = 0;
404
405int
406Py_AddPendingCall(int (*func)(void *), void *arg)
407{
408 int i, j, result=0;
409 PyThread_type_lock lock = pending_lock;
410
411 /* try a few times for the lock. Since this mechanism is used
412 * for signal handling (on the main thread), there is a (slim)
413 * chance that a signal is delivered on the same thread while we
414 * hold the lock during the Py_MakePendingCalls() function.
415 * This avoids a deadlock in that case.
416 * Note that signals can be delivered on any thread. In particular,
417 * on Windows, a SIGINT is delivered on a system-created worker
418 * thread.
419 * We also check for lock being NULL, in the unlikely case that
420 * this function is called before any bytecode evaluation takes place.
421 */
422 if (lock != NULL) {
423 for (i = 0; i<100; i++) {
424 if (PyThread_acquire_lock(lock, NOWAIT_LOCK))
425 break;
426 }
427 if (i == 100)
428 return -1;
429 }
430
431 i = pendinglast;
432 j = (i + 1) % NPENDINGCALLS;
433 if (j == pendingfirst) {
434 result = -1; /* Queue full */
435 } else {
436 pendingcalls[i].func = func;
437 pendingcalls[i].arg = arg;
438 pendinglast = j;
439 }
440 /* signal main loop */
441 _Py_Ticker = 0;
442 pendingcalls_to_do = 1;
443 if (lock != NULL)
444 PyThread_release_lock(lock);
445 return result;
446}
447
448int
449Py_MakePendingCalls(void)
450{
451 int i;
452 int r = 0;
453
454 if (!pending_lock) {
455 /* initial allocation of the lock */
456 pending_lock = PyThread_allocate_lock();
457 if (pending_lock == NULL)
458 return -1;
459 }
460
461 /* only service pending calls on main thread */
462 if (main_thread && PyThread_get_thread_ident() != main_thread)
463 return 0;
464 /* don't perform recursive pending calls */
465 if (pendingbusy)
466 return 0;
467 pendingbusy = 1;
468 /* perform a bounded number of calls, in case of recursion */
469 for (i=0; i<NPENDINGCALLS; i++) {
470 int j;
471 int (*func)(void *);
Mark Dickinson9e58a372009-02-08 17:33:11 +0000472 void *arg = NULL;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000473
474 /* pop one item off the queue while holding the lock */
475 PyThread_acquire_lock(pending_lock, WAIT_LOCK);
476 j = pendingfirst;
477 if (j == pendinglast) {
478 func = NULL; /* Queue empty */
479 } else {
480 func = pendingcalls[j].func;
481 arg = pendingcalls[j].arg;
482 pendingfirst = (j + 1) % NPENDINGCALLS;
483 }
484 pendingcalls_to_do = pendingfirst != pendinglast;
485 PyThread_release_lock(pending_lock);
486 /* having released the lock, perform the callback */
487 if (func == NULL)
488 break;
489 r = func(arg);
490 if (r)
491 break;
492 }
493 pendingbusy = 0;
494 return r;
495}
496
497#else /* if ! defined WITH_THREAD */
498
499/*
500 WARNING! ASYNCHRONOUSLY EXECUTING CODE!
501 This code is used for signal handling in python that isn't built
502 with WITH_THREAD.
503 Don't use this implementation when Py_AddPendingCalls() can happen
504 on a different thread!
505
Guido van Rossuma9672091994-09-14 13:31:22 +0000506 There are two possible race conditions:
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000507 (1) nested asynchronous calls to Py_AddPendingCall()
508 (2) AddPendingCall() calls made while pending calls are being processed.
509
510 (1) is very unlikely because typically signal delivery
511 is blocked during signal handling. So it should be impossible.
512 (2) is a real possibility.
Guido van Rossuma9672091994-09-14 13:31:22 +0000513 The current code is safe against (2), but not against (1).
514 The safety against (2) is derived from the fact that only one
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000515 thread is present, interrupted by signals, and that the critical
516 section is protected with the "busy" variable. On Windows, which
517 delivers SIGINT on a system thread, this does not hold and therefore
518 Windows really shouldn't use this version.
519 The two threads could theoretically wiggle around the "busy" variable.
Guido van Rossuma027efa1997-05-05 20:56:21 +0000520*/
Guido van Rossum8861b741996-07-30 16:49:37 +0000521
Guido van Rossuma9672091994-09-14 13:31:22 +0000522#define NPENDINGCALLS 32
523static struct {
Thomas Wouters334fb892000-07-25 12:56:38 +0000524 int (*func)(void *);
525 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000526} pendingcalls[NPENDINGCALLS];
527static volatile int pendingfirst = 0;
528static volatile int pendinglast = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000529static volatile int pendingcalls_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000530
531int
Thomas Wouters334fb892000-07-25 12:56:38 +0000532Py_AddPendingCall(int (*func)(void *), void *arg)
Guido van Rossuma9672091994-09-14 13:31:22 +0000533{
Michael W. Hudson30ea2f22004-07-07 17:44:12 +0000534 static volatile int busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000535 int i, j;
536 /* XXX Begin critical section */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000537 if (busy)
538 return -1;
539 busy = 1;
Guido van Rossuma9672091994-09-14 13:31:22 +0000540 i = pendinglast;
541 j = (i + 1) % NPENDINGCALLS;
Guido van Rossum04e70322002-07-17 16:57:13 +0000542 if (j == pendingfirst) {
543 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000544 return -1; /* Queue full */
Guido van Rossum04e70322002-07-17 16:57:13 +0000545 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000546 pendingcalls[i].func = func;
547 pendingcalls[i].arg = arg;
548 pendinglast = j;
Skip Montanarod581d772002-09-03 20:10:45 +0000549
550 _Py_Ticker = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000551 pendingcalls_to_do = 1; /* Signal main loop */
Guido van Rossum180d7b41994-09-29 09:45:57 +0000552 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000553 /* XXX End critical section */
554 return 0;
555}
556
Guido van Rossum180d7b41994-09-29 09:45:57 +0000557int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000558Py_MakePendingCalls(void)
Guido van Rossuma9672091994-09-14 13:31:22 +0000559{
Guido van Rossum180d7b41994-09-29 09:45:57 +0000560 static int busy = 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +0000561 if (busy)
Guido van Rossum180d7b41994-09-29 09:45:57 +0000562 return 0;
563 busy = 1;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000564 pendingcalls_to_do = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000565 for (;;) {
566 int i;
Thomas Wouters334fb892000-07-25 12:56:38 +0000567 int (*func)(void *);
568 void *arg;
Guido van Rossuma9672091994-09-14 13:31:22 +0000569 i = pendingfirst;
570 if (i == pendinglast)
571 break; /* Queue empty */
572 func = pendingcalls[i].func;
573 arg = pendingcalls[i].arg;
574 pendingfirst = (i + 1) % NPENDINGCALLS;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000575 if (func(arg) < 0) {
576 busy = 0;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000577 pendingcalls_to_do = 1; /* We're not done yet */
Guido van Rossuma9672091994-09-14 13:31:22 +0000578 return -1;
Guido van Rossum180d7b41994-09-29 09:45:57 +0000579 }
Guido van Rossuma9672091994-09-14 13:31:22 +0000580 }
Guido van Rossum180d7b41994-09-29 09:45:57 +0000581 busy = 0;
Guido van Rossuma9672091994-09-14 13:31:22 +0000582 return 0;
583}
584
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000585#endif /* WITH_THREAD */
586
Guido van Rossuma9672091994-09-14 13:31:22 +0000587
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000588/* The interpreter's recursion limit */
589
Hye-Shik Changb6fa2812005-04-04 15:49:02 +0000590#ifndef Py_DEFAULT_RECURSION_LIMIT
591#define Py_DEFAULT_RECURSION_LIMIT 1000
592#endif
593static int recursion_limit = Py_DEFAULT_RECURSION_LIMIT;
594int _Py_CheckRecursionLimit = Py_DEFAULT_RECURSION_LIMIT;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000595
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000596int
597Py_GetRecursionLimit(void)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000598{
599 return recursion_limit;
600}
601
Vladimir Marangozov7bd25be2000-09-01 11:07:19 +0000602void
603Py_SetRecursionLimit(int new_limit)
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000604{
605 recursion_limit = new_limit;
Thomas Woutersae406c62007-09-19 17:27:43 +0000606 _Py_CheckRecursionLimit = recursion_limit;
Jeremy Hyltonee5adfb2000-08-31 19:23:01 +0000607}
608
Armin Rigo2b3eb402003-10-28 12:05:48 +0000609/* the macro Py_EnterRecursiveCall() only calls _Py_CheckRecursiveCall()
610 if the recursion_depth reaches _Py_CheckRecursionLimit.
611 If USE_STACKCHECK, the macro decrements _Py_CheckRecursionLimit
612 to guarantee that _Py_CheckRecursiveCall() is regularly called.
613 Without USE_STACKCHECK, there is no need for this. */
614int
615_Py_CheckRecursiveCall(char *where)
616{
617 PyThreadState *tstate = PyThreadState_GET();
618
619#ifdef USE_STACKCHECK
620 if (PyOS_CheckStack()) {
621 --tstate->recursion_depth;
622 PyErr_SetString(PyExc_MemoryError, "Stack overflow");
623 return -1;
624 }
625#endif
626 if (tstate->recursion_depth > recursion_limit) {
627 --tstate->recursion_depth;
628 PyErr_Format(PyExc_RuntimeError,
629 "maximum recursion depth exceeded%s",
630 where);
631 return -1;
632 }
Thomas Woutersae406c62007-09-19 17:27:43 +0000633 _Py_CheckRecursionLimit = recursion_limit;
Armin Rigo2b3eb402003-10-28 12:05:48 +0000634 return 0;
635}
636
Guido van Rossum374a9221991-04-04 10:40:29 +0000637/* Status code for main loop (reason for stack unwind) */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000638enum why_code {
639 WHY_NOT = 0x0001, /* No error */
640 WHY_EXCEPTION = 0x0002, /* Exception occurred */
641 WHY_RERAISE = 0x0004, /* Exception re-raised by 'finally' */
642 WHY_RETURN = 0x0008, /* 'return' statement */
643 WHY_BREAK = 0x0010, /* 'break' statement */
644 WHY_CONTINUE = 0x0020, /* 'continue' statement */
645 WHY_YIELD = 0x0040 /* 'yield' operator */
646};
Guido van Rossum374a9221991-04-04 10:40:29 +0000647
Fredrik Lundh7a830892006-05-27 10:39:48 +0000648static enum why_code do_raise(PyObject *, PyObject *, PyObject *);
649static int unpack_iterable(PyObject *, int, PyObject **);
Guido van Rossum1aa14831997-01-21 05:34:20 +0000650
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +0000651/* Records whether tracing is on for any thread. Counts the number of
652 threads for which tstate->c_tracefunc is non-NULL, so if the value
653 is 0, we know we don't have to check this thread's c_tracefunc.
654 This speeds up the if statement in PyEval_EvalFrameEx() after
655 fast_next_opcode*/
656static int _Py_TracingPossible = 0;
657
Skip Montanarod581d772002-09-03 20:10:45 +0000658/* for manipulating the thread switch and periodic "stuff" - used to be
659 per thread, now just a pair o' globals */
Skip Montanaro99dba272002-09-03 20:19:06 +0000660int _Py_CheckInterval = 100;
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000661volatile int _Py_Ticker = 0; /* so that we hit a "tick" first thing */
Guido van Rossum374a9221991-04-04 10:40:29 +0000662
Guido van Rossumb209a111997-04-29 18:18:01 +0000663PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000664PyEval_EvalCode(PyCodeObject *co, PyObject *globals, PyObject *locals)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000665{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000666 return PyEval_EvalCodeEx(co,
Guido van Rossum681d79a1995-07-18 14:51:37 +0000667 globals, locals,
Guido van Rossumb209a111997-04-29 18:18:01 +0000668 (PyObject **)NULL, 0,
669 (PyObject **)NULL, 0,
Jeremy Hylton64949cb2001-01-25 20:06:59 +0000670 (PyObject **)NULL, 0,
671 NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +0000672}
673
674
675/* Interpreter main loop */
676
Martin v. Löwis8d97e332004-06-27 15:43:12 +0000677PyObject *
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000678PyEval_EvalFrame(PyFrameObject *f) {
679 /* This is for backward compatibility with extension modules that
Thomas Wouterse2176022007-09-20 17:35:10 +0000680 used this API; core interpreter code should call
681 PyEval_EvalFrameEx() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000682 return PyEval_EvalFrameEx(f, 0);
683}
684
685PyObject *
Anthony Baxtera863d332006-04-11 07:43:46 +0000686PyEval_EvalFrameEx(PyFrameObject *f, int throwflag)
Guido van Rossum374a9221991-04-04 10:40:29 +0000687{
Guido van Rossum950361c1997-01-24 13:49:28 +0000688#ifdef DXPAIRS
689 int lastopcode = 0;
690#endif
Thomas Wouterse2176022007-09-20 17:35:10 +0000691 register PyObject **stack_pointer; /* Next free slot in value stack */
Guido van Rossum374a9221991-04-04 10:40:29 +0000692 register unsigned char *next_instr;
Armin Rigo8817fcd2004-06-17 10:22:40 +0000693 register int opcode; /* Current opcode */
694 register int oparg; /* Current opcode argument, if any */
Raymond Hettinger7c958652004-04-06 10:11:10 +0000695 register enum why_code why; /* Reason for block stack unwind */
Guido van Rossum374a9221991-04-04 10:40:29 +0000696 register int err; /* Error status -- nonzero if error */
Guido van Rossumb209a111997-04-29 18:18:01 +0000697 register PyObject *x; /* Result object -- NULL if error */
698 register PyObject *v; /* Temporary objects popped off stack */
699 register PyObject *w;
700 register PyObject *u;
701 register PyObject *t;
Barry Warsaw23c9ec82000-08-21 15:44:01 +0000702 register PyObject *stream = NULL; /* for PRINT opcodes */
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000703 register PyObject **fastlocals, **freevars;
Guido van Rossum014518f1998-11-23 21:09:51 +0000704 PyObject *retval = NULL; /* Return value */
Guido van Rossum885553e1998-12-21 18:33:30 +0000705 PyThreadState *tstate = PyThreadState_GET();
Tim Peters5ca576e2001-06-18 22:08:13 +0000706 PyCodeObject *co;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000707
Tim Peters8a5c3c72004-04-05 19:36:21 +0000708 /* when tracing we set things up so that
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000709
710 not (instr_lb <= current_bytecode_offset < instr_ub)
711
Tim Peters8a5c3c72004-04-05 19:36:21 +0000712 is true when the line being executed has changed. The
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000713 initial values are such as to make this false the first
714 time it is tested. */
Armin Rigobf57a142004-03-22 19:24:58 +0000715 int instr_ub = -1, instr_lb = 0, instr_prev = -1;
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000716
Guido van Rossumd076c731998-10-07 19:42:25 +0000717 unsigned char *first_instr;
Skip Montanaro04d80f82002-08-04 21:03:35 +0000718 PyObject *names;
719 PyObject *consts;
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000720#if defined(Py_DEBUG) || defined(LLTRACE)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000721 /* Make it easier to find out where we are with a debugger */
Tim Peters5ca576e2001-06-18 22:08:13 +0000722 char *filename;
Guido van Rossum99bec951992-09-03 20:29:45 +0000723#endif
Guido van Rossum374a9221991-04-04 10:40:29 +0000724
Neal Norwitza81d2202002-07-14 00:27:26 +0000725/* Tuple access macros */
726
727#ifndef Py_DEBUG
728#define GETITEM(v, i) PyTuple_GET_ITEM((PyTupleObject *)(v), (i))
729#else
730#define GETITEM(v, i) PyTuple_GetItem((v), (i))
731#endif
732
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000733#ifdef WITH_TSC
734/* Use Pentium timestamp counter to mark certain events:
735 inst0 -- beginning of switch statement for opcode dispatch
736 inst1 -- end of switch statement (may be skipped)
737 loop0 -- the top of the mainloop
Tim Peters7df5e7f2006-05-26 23:14:37 +0000738 loop1 -- place where control returns again to top of mainloop
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000739 (may be skipped)
740 intr1 -- beginning of long interruption
741 intr2 -- end of long interruption
742
743 Many opcodes call out to helper C functions. In some cases, the
744 time in those functions should be counted towards the time for the
745 opcode, but not in all cases. For example, a CALL_FUNCTION opcode
746 calls another Python function; there's no point in charge all the
747 bytecode executed by the called function to the caller.
748
749 It's hard to make a useful judgement statically. In the presence
750 of operator overloading, it's impossible to tell if a call will
751 execute new Python code or not.
752
753 It's a case-by-case judgement. I'll use intr1 for the following
754 cases:
755
756 EXEC_STMT
757 IMPORT_STAR
758 IMPORT_FROM
759 CALL_FUNCTION (and friends)
760
761 */
762 uint64 inst0, inst1, loop0, loop1, intr0 = 0, intr1 = 0;
763 int ticked = 0;
764
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000765 READ_TIMESTAMP(inst0);
766 READ_TIMESTAMP(inst1);
767 READ_TIMESTAMP(loop0);
768 READ_TIMESTAMP(loop1);
Michael W. Hudson800ba232004-08-12 18:19:17 +0000769
770 /* shut up the compiler */
771 opcode = 0;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000772#endif
773
Guido van Rossum374a9221991-04-04 10:40:29 +0000774/* Code access macros */
775
Martin v. Löwis18e16552006-02-15 17:27:45 +0000776#define INSTR_OFFSET() ((int)(next_instr - first_instr))
Guido van Rossum374a9221991-04-04 10:40:29 +0000777#define NEXTOP() (*next_instr++)
Raymond Hettinger5bed4562004-04-10 23:34:17 +0000778#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000779#define PEEKARG() ((next_instr[2]<<8) + next_instr[1])
Guido van Rossumd076c731998-10-07 19:42:25 +0000780#define JUMPTO(x) (next_instr = first_instr + (x))
Guido van Rossum374a9221991-04-04 10:40:29 +0000781#define JUMPBY(x) (next_instr += (x))
782
Raymond Hettingerf606f872003-03-16 03:11:04 +0000783/* OpCode prediction macros
Thomas Wouterse2176022007-09-20 17:35:10 +0000784 Some opcodes tend to come in pairs thus making it possible to
785 predict the second code when the first is run. For example,
Jeffrey Yasskin68d68522009-02-28 19:03:21 +0000786 GET_ITER is often followed by FOR_ITER. And FOR_ITER is often
787 followed by STORE_FAST or UNPACK_SEQUENCE.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000788
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000789 Verifying the prediction costs a single high-speed test of a register
Raymond Hettingerac2072922003-03-16 15:41:11 +0000790 variable against a constant. If the pairing was good, then the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000791 processor's own internal branch predication has a high likelihood of
792 success, resulting in a nearly zero-overhead transition to the
793 next opcode. A successful prediction saves a trip through the eval-loop
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000794 including its two unpredictable branches, the HAS_ARG test and the
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000795 switch-case. Combined with the processor's internal branch prediction,
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000796 a successful PREDICT has the effect of making the two opcodes run as if
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000797 they were a single new opcode with the bodies combined.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000798
Raymond Hettingerafae11e2008-07-05 02:11:55 +0000799 If collecting opcode statistics, your choices are to either keep the
800 predictions turned-on and interpret the results as if some opcodes
801 had been combined or turn-off predictions so that the opcode frequency
802 counter updates for both opcodes.
Raymond Hettingerf606f872003-03-16 03:11:04 +0000803*/
804
Raymond Hettingera7216982004-02-08 19:59:27 +0000805#ifdef DYNAMIC_EXECUTION_PROFILE
806#define PREDICT(op) if (0) goto PRED_##op
807#else
Raymond Hettingerac2072922003-03-16 15:41:11 +0000808#define PREDICT(op) if (*next_instr == op) goto PRED_##op
Raymond Hettingera7216982004-02-08 19:59:27 +0000809#endif
810
Raymond Hettingerf606f872003-03-16 03:11:04 +0000811#define PREDICTED(op) PRED_##op: next_instr++
Raymond Hettinger52a21b82004-08-06 18:43:09 +0000812#define PREDICTED_WITH_ARG(op) PRED_##op: oparg = PEEKARG(); next_instr += 3
Raymond Hettingerf606f872003-03-16 03:11:04 +0000813
Guido van Rossum374a9221991-04-04 10:40:29 +0000814/* Stack manipulation macros */
815
Martin v. Löwis18e16552006-02-15 17:27:45 +0000816/* The stack can grow at most MAXINT deep, as co_nlocals and
817 co_stacksize are ints. */
818#define STACK_LEVEL() ((int)(stack_pointer - f->f_valuestack))
Guido van Rossum374a9221991-04-04 10:40:29 +0000819#define EMPTY() (STACK_LEVEL() == 0)
820#define TOP() (stack_pointer[-1])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000821#define SECOND() (stack_pointer[-2])
822#define THIRD() (stack_pointer[-3])
823#define FOURTH() (stack_pointer[-4])
Benjamin Petersonb8338ab2009-06-28 16:08:02 +0000824#define PEEK(n) (stack_pointer[-(n)])
Raymond Hettinger663004b2003-01-09 15:24:30 +0000825#define SET_TOP(v) (stack_pointer[-1] = (v))
826#define SET_SECOND(v) (stack_pointer[-2] = (v))
827#define SET_THIRD(v) (stack_pointer[-3] = (v))
828#define SET_FOURTH(v) (stack_pointer[-4] = (v))
Benjamin Petersonb8338ab2009-06-28 16:08:02 +0000829#define SET_VALUE(n, v) (stack_pointer[-(n)] = (v))
Raymond Hettinger663004b2003-01-09 15:24:30 +0000830#define BASIC_STACKADJ(n) (stack_pointer += n)
Guido van Rossum374a9221991-04-04 10:40:29 +0000831#define BASIC_PUSH(v) (*stack_pointer++ = (v))
832#define BASIC_POP() (*--stack_pointer)
833
Guido van Rossum96a42c81992-01-12 02:29:51 +0000834#ifdef LLTRACE
Jeremy Hylton14368152001-10-17 13:29:30 +0000835#define PUSH(v) { (void)(BASIC_PUSH(v), \
836 lltrace && prtrace(TOP(), "push")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000837 assert(STACK_LEVEL() <= co->co_stacksize); }
Thomas Wouterse2176022007-09-20 17:35:10 +0000838#define POP() ((void)(lltrace && prtrace(TOP(), "pop")), \
839 BASIC_POP())
Raymond Hettinger663004b2003-01-09 15:24:30 +0000840#define STACKADJ(n) { (void)(BASIC_STACKADJ(n), \
841 lltrace && prtrace(TOP(), "stackadj")); \
Richard Jonescebbefc2006-05-23 18:28:17 +0000842 assert(STACK_LEVEL() <= co->co_stacksize); }
Christian Heimes52729ac2007-12-14 02:33:57 +0000843#define EXT_POP(STACK_POINTER) ((void)(lltrace && \
844 prtrace((STACK_POINTER)[-1], "ext_pop")), \
845 *--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000846#else
847#define PUSH(v) BASIC_PUSH(v)
848#define POP() BASIC_POP()
Raymond Hettinger663004b2003-01-09 15:24:30 +0000849#define STACKADJ(n) BASIC_STACKADJ(n)
Guido van Rossumc2e20742006-02-27 22:32:47 +0000850#define EXT_POP(STACK_POINTER) (*--(STACK_POINTER))
Guido van Rossum374a9221991-04-04 10:40:29 +0000851#endif
852
Guido van Rossum681d79a1995-07-18 14:51:37 +0000853/* Local variable macros */
854
855#define GETLOCAL(i) (fastlocals[i])
Guido van Rossumcfbf1a32002-03-28 20:17:52 +0000856
857/* The SETLOCAL() macro must not DECREF the local variable in-place and
858 then store the new value; it must copy the old value to a temporary
859 value, then store the new value, and then DECREF the temporary value.
860 This is because it is possible that during the DECREF the frame is
861 accessed by other code (e.g. a __del__ method or gc.collect()) and the
862 variable would be pointing to already-freed memory. */
863#define SETLOCAL(i, value) do { PyObject *tmp = GETLOCAL(i); \
864 GETLOCAL(i) = value; \
865 Py_XDECREF(tmp); } while (0)
Guido van Rossum681d79a1995-07-18 14:51:37 +0000866
Guido van Rossuma027efa1997-05-05 20:56:21 +0000867/* Start of code */
868
Tim Peters5ca576e2001-06-18 22:08:13 +0000869 if (f == NULL)
870 return NULL;
871
Armin Rigo1d313ab2003-10-25 14:33:09 +0000872 /* push frame */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000873 if (Py_EnterRecursiveCall(""))
Armin Rigo1d313ab2003-10-25 14:33:09 +0000874 return NULL;
Guido van Rossum8861b741996-07-30 16:49:37 +0000875
Tim Peters5ca576e2001-06-18 22:08:13 +0000876 tstate->frame = f;
Tim Peters5ca576e2001-06-18 22:08:13 +0000877
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000878 if (tstate->use_tracing) {
879 if (tstate->c_tracefunc != NULL) {
880 /* tstate->c_tracefunc, if defined, is a
881 function that will be called on *every* entry
882 to a code block. Its return value, if not
883 None, is a function that will be called at
884 the start of each executed line of code.
885 (Actually, the function must return itself
886 in order to continue tracing.) The trace
887 functions are called with three arguments:
888 a pointer to the current frame, a string
889 indicating why the function is called, and
890 an argument which depends on the situation.
891 The global trace function is also called
892 whenever an exception is detected. */
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000893 if (call_trace_protected(tstate->c_tracefunc,
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000894 tstate->c_traceobj,
895 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000896 /* Trace function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000897 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000898 }
899 }
900 if (tstate->c_profilefunc != NULL) {
901 /* Similar for c_profilefunc, except it needn't
902 return itself and isn't called for "line" events */
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +0000903 if (call_trace_protected(tstate->c_profilefunc,
904 tstate->c_profileobj,
905 f, PyTrace_CALL, Py_None)) {
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000906 /* Profile function raised an error */
Armin Rigo2b3eb402003-10-28 12:05:48 +0000907 goto exit_eval_frame;
Neil Schemenauer6c0f2002001-09-04 19:03:35 +0000908 }
909 }
910 }
911
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000912 co = f->f_code;
913 names = co->co_names;
914 consts = co->co_consts;
915 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +0000916 freevars = f->f_localsplus + co->co_nlocals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000917 first_instr = (unsigned char*) PyString_AS_STRING(co->co_code);
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000918 /* An explanation is in order for the next line.
919
920 f->f_lasti now refers to the index of the last instruction
921 executed. You might think this was obvious from the name, but
922 this wasn't always true before 2.3! PyFrame_New now sets
923 f->f_lasti to -1 (i.e. the index *before* the first instruction)
924 and YIELD_VALUE doesn't fiddle with f_lasti any more. So this
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000925 does work. Promise.
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000926
927 When the PREDICT() macros are enabled, some opcode pairs follow in
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +0000928 direct succession without updating f->f_lasti. A successful
Raymond Hettinger4bd97d42007-01-06 01:14:41 +0000929 prediction effectively links the two codes together as if they
930 were a single new opcode; accordingly,f->f_lasti will point to
931 the first code in the pair (for instance, GET_ITER followed by
932 FOR_ITER is effectively a single opcode and f->f_lasti will point
933 at to the beginning of the combined pair.)
934 */
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000935 next_instr = first_instr + f->f_lasti + 1;
936 stack_pointer = f->f_stacktop;
937 assert(stack_pointer != NULL);
938 f->f_stacktop = NULL; /* remains NULL unless yield suspends frame */
939
Tim Peters5ca576e2001-06-18 22:08:13 +0000940#ifdef LLTRACE
Jeremy Hylton3e0055f2005-10-20 19:59:25 +0000941 lltrace = PyDict_GetItemString(f->f_globals, "__lltrace__") != NULL;
Tim Peters5ca576e2001-06-18 22:08:13 +0000942#endif
Neal Norwitz5f5153e2005-10-21 04:28:38 +0000943#if defined(Py_DEBUG) || defined(LLTRACE)
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000944 filename = PyString_AsString(co->co_filename);
Tim Peters5ca576e2001-06-18 22:08:13 +0000945#endif
Guido van Rossumac7be682001-01-17 15:42:30 +0000946
Guido van Rossum374a9221991-04-04 10:40:29 +0000947 why = WHY_NOT;
948 err = 0;
Guido van Rossumb209a111997-04-29 18:18:01 +0000949 x = Py_None; /* Not a reference, just anything non-NULL */
Fred Drake48fba732000-10-11 13:54:07 +0000950 w = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +0000951
Anthony Baxtera863d332006-04-11 07:43:46 +0000952 if (throwflag) { /* support for generator.throw() */
Phillip J. Eby0d6615f2005-08-02 00:46:46 +0000953 why = WHY_EXCEPTION;
954 goto on_error;
955 }
Tim Peters7df5e7f2006-05-26 23:14:37 +0000956
Guido van Rossum374a9221991-04-04 10:40:29 +0000957 for (;;) {
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000958#ifdef WITH_TSC
959 if (inst1 == 0) {
960 /* Almost surely, the opcode executed a break
961 or a continue, preventing inst1 from being set
962 on the way out of the loop.
963 */
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000964 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000965 loop1 = inst1;
966 }
967 dump_tsc(opcode, ticked, inst0, inst1, loop0, loop1,
968 intr0, intr1);
969 ticked = 0;
970 inst1 = 0;
971 intr0 = 0;
972 intr1 = 0;
Michael W. Hudson75eabd22005-01-18 15:56:11 +0000973 READ_TIMESTAMP(loop0);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000974#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000975 assert(stack_pointer >= f->f_valuestack); /* else underflow */
Richard Jonescebbefc2006-05-23 18:28:17 +0000976 assert(STACK_LEVEL() <= co->co_stacksize); /* else overflow */
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000977
Guido van Rossuma027efa1997-05-05 20:56:21 +0000978 /* Do periodic things. Doing this every time through
979 the loop would add too much overhead, so we do it
980 only every Nth instruction. We also do it if
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000981 ``pendingcalls_to_do'' is set, i.e. when an asynchronous
Guido van Rossuma027efa1997-05-05 20:56:21 +0000982 event needs attention (e.g. a signal handler or
983 async I/O handler); see Py_AddPendingCall() and
984 Py_MakePendingCalls() above. */
Guido van Rossumac7be682001-01-17 15:42:30 +0000985
Skip Montanarod581d772002-09-03 20:10:45 +0000986 if (--_Py_Ticker < 0) {
Thomas Woutersae406c62007-09-19 17:27:43 +0000987 if (*next_instr == SETUP_FINALLY) {
988 /* Make the last opcode before
989 a try: finally: block uninterruptable. */
990 goto fast_next_opcode;
991 }
Skip Montanarod581d772002-09-03 20:10:45 +0000992 _Py_Ticker = _Py_CheckInterval;
Michael W. Hudson019a78e2002-11-08 12:53:11 +0000993 tstate->tick_counter++;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +0000994#ifdef WITH_TSC
995 ticked = 1;
996#endif
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +0000997 if (pendingcalls_to_do) {
Guido van Rossum8861b741996-07-30 16:49:37 +0000998 if (Py_MakePendingCalls() < 0) {
999 why = WHY_EXCEPTION;
1000 goto on_error;
1001 }
Kristján Valur Jónsson0e919382009-01-09 20:31:26 +00001002 if (pendingcalls_to_do)
Kurt B. Kaiser4c79a832004-11-23 18:06:08 +00001003 /* MakePendingCalls() didn't succeed.
1004 Force early re-execution of this
1005 "periodic" code, possibly after
1006 a thread switch */
1007 _Py_Ticker = 0;
Guido van Rossum8861b741996-07-30 16:49:37 +00001008 }
Guido van Rossume59214e1994-08-30 08:01:59 +00001009#ifdef WITH_THREAD
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001010 if (interpreter_lock) {
1011 /* Give another thread a chance */
1012
Guido van Rossum25ce5661997-08-02 03:10:38 +00001013 if (PyThreadState_Swap(NULL) != tstate)
1014 Py_FatalError("ceval: tstate mix-up");
Guido van Rossum65d5b571998-12-21 19:32:43 +00001015 PyThread_release_lock(interpreter_lock);
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001016
1017 /* Other threads may run now */
1018
Guido van Rossum65d5b571998-12-21 19:32:43 +00001019 PyThread_acquire_lock(interpreter_lock, 1);
Guido van Rossum25ce5661997-08-02 03:10:38 +00001020 if (PyThreadState_Swap(tstate) != NULL)
1021 Py_FatalError("ceval: orphan tstate");
Guido van Rossumb8b6d0c2003-06-28 21:53:52 +00001022
1023 /* Check for thread interrupts */
1024
1025 if (tstate->async_exc != NULL) {
1026 x = tstate->async_exc;
1027 tstate->async_exc = NULL;
1028 PyErr_SetNone(x);
1029 Py_DECREF(x);
1030 why = WHY_EXCEPTION;
1031 goto on_error;
1032 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001033 }
1034#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001035 }
Guido van Rossum1984f1e1992-08-04 12:41:02 +00001036
Neil Schemenauer63543862002-02-17 19:10:14 +00001037 fast_next_opcode:
Guido van Rossum99bec951992-09-03 20:29:45 +00001038 f->f_lasti = INSTR_OFFSET();
Guido van Rossumac7be682001-01-17 15:42:30 +00001039
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001040 /* line-by-line tracing support */
1041
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +00001042 if (_Py_TracingPossible &&
1043 tstate->c_tracefunc != NULL && !tstate->tracing) {
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001044 /* see maybe_call_line_trace
1045 for expository comments */
1046 f->f_stacktop = stack_pointer;
Tim Peters8a5c3c72004-04-05 19:36:21 +00001047
Michael W. Hudson58ee2af2003-04-29 16:18:47 +00001048 err = maybe_call_line_trace(tstate->c_tracefunc,
1049 tstate->c_traceobj,
Armin Rigobf57a142004-03-22 19:24:58 +00001050 f, &instr_lb, &instr_ub,
1051 &instr_prev);
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001052 /* Reload possibly changed frame fields */
1053 JUMPTO(f->f_lasti);
Michael W. Hudson58ee2af2003-04-29 16:18:47 +00001054 if (f->f_stacktop != NULL) {
1055 stack_pointer = f->f_stacktop;
1056 f->f_stacktop = NULL;
1057 }
1058 if (err) {
1059 /* trace function raised an exception */
1060 goto on_error;
1061 }
Michael W. Hudson019a78e2002-11-08 12:53:11 +00001062 }
1063
1064 /* Extract opcode and argument */
1065
Guido van Rossum374a9221991-04-04 10:40:29 +00001066 opcode = NEXTOP();
Armin Rigo8817fcd2004-06-17 10:22:40 +00001067 oparg = 0; /* allows oparg to be stored in a register because
1068 it doesn't have to be remembered across a full loop */
Raymond Hettinger5bed4562004-04-10 23:34:17 +00001069 if (HAS_ARG(opcode))
1070 oparg = NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00001071 dispatch_opcode:
Guido van Rossum950361c1997-01-24 13:49:28 +00001072#ifdef DYNAMIC_EXECUTION_PROFILE
1073#ifdef DXPAIRS
1074 dxpairs[lastopcode][opcode]++;
1075 lastopcode = opcode;
1076#endif
1077 dxp[opcode]++;
1078#endif
Guido van Rossum374a9221991-04-04 10:40:29 +00001079
Guido van Rossum96a42c81992-01-12 02:29:51 +00001080#ifdef LLTRACE
Guido van Rossum374a9221991-04-04 10:40:29 +00001081 /* Instruction tracing */
Guido van Rossumac7be682001-01-17 15:42:30 +00001082
Guido van Rossum96a42c81992-01-12 02:29:51 +00001083 if (lltrace) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001084 if (HAS_ARG(opcode)) {
1085 printf("%d: %d, %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001086 f->f_lasti, opcode, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001087 }
1088 else {
1089 printf("%d: %d\n",
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001090 f->f_lasti, opcode);
Guido van Rossum374a9221991-04-04 10:40:29 +00001091 }
1092 }
1093#endif
Michael W. Hudsondd32a912002-08-15 14:59:02 +00001094
Guido van Rossum374a9221991-04-04 10:40:29 +00001095 /* Main switch on opcode */
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001096 READ_TIMESTAMP(inst0);
Jeremy Hylton52820442001-01-03 23:52:36 +00001097
Guido van Rossum374a9221991-04-04 10:40:29 +00001098 switch (opcode) {
Guido van Rossumac7be682001-01-17 15:42:30 +00001099
Guido van Rossum374a9221991-04-04 10:40:29 +00001100 /* BEWARE!
1101 It is essential that any operation that fails sets either
1102 x to NULL, err to nonzero, or why to anything but WHY_NOT,
1103 and that no operation that succeeds does this! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001104
Guido van Rossum374a9221991-04-04 10:40:29 +00001105 /* case STOP_CODE: this is an error! */
Guido van Rossumac7be682001-01-17 15:42:30 +00001106
Raymond Hettinger9c18e812004-06-21 16:31:15 +00001107 case NOP:
1108 goto fast_next_opcode;
1109
Neil Schemenauer63543862002-02-17 19:10:14 +00001110 case LOAD_FAST:
1111 x = GETLOCAL(oparg);
1112 if (x != NULL) {
1113 Py_INCREF(x);
1114 PUSH(x);
1115 goto fast_next_opcode;
1116 }
1117 format_exc_check_arg(PyExc_UnboundLocalError,
1118 UNBOUNDLOCAL_ERROR_MSG,
1119 PyTuple_GetItem(co->co_varnames, oparg));
1120 break;
1121
1122 case LOAD_CONST:
Skip Montanaro04d80f82002-08-04 21:03:35 +00001123 x = GETITEM(consts, oparg);
Neil Schemenauer63543862002-02-17 19:10:14 +00001124 Py_INCREF(x);
1125 PUSH(x);
1126 goto fast_next_opcode;
1127
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001128 PREDICTED_WITH_ARG(STORE_FAST);
Neil Schemenauer63543862002-02-17 19:10:14 +00001129 case STORE_FAST:
1130 v = POP();
1131 SETLOCAL(oparg, v);
1132 goto fast_next_opcode;
1133
Guido van Rossum374a9221991-04-04 10:40:29 +00001134 case POP_TOP:
1135 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001136 Py_DECREF(v);
Neil Schemenauer63543862002-02-17 19:10:14 +00001137 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001138
Guido van Rossum374a9221991-04-04 10:40:29 +00001139 case ROT_TWO:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001140 v = TOP();
1141 w = SECOND();
1142 SET_TOP(w);
1143 SET_SECOND(v);
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 ROT_THREE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001147 v = TOP();
1148 w = SECOND();
1149 x = THIRD();
1150 SET_TOP(w);
1151 SET_SECOND(x);
1152 SET_THIRD(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001153 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001154
Thomas Wouters434d0822000-08-24 20:11:32 +00001155 case ROT_FOUR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001156 u = TOP();
1157 v = SECOND();
1158 w = THIRD();
1159 x = FOURTH();
1160 SET_TOP(v);
1161 SET_SECOND(w);
1162 SET_THIRD(x);
1163 SET_FOURTH(u);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001164 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001165
Guido van Rossum374a9221991-04-04 10:40:29 +00001166 case DUP_TOP:
1167 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001168 Py_INCREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001169 PUSH(v);
Raymond Hettinger080cb322003-03-14 01:37:42 +00001170 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00001171
Thomas Wouters434d0822000-08-24 20:11:32 +00001172 case DUP_TOPX:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001173 if (oparg == 2) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001174 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001175 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001176 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001177 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001178 STACKADJ(2);
1179 SET_TOP(x);
1180 SET_SECOND(w);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001181 goto fast_next_opcode;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001182 } else if (oparg == 3) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00001183 x = TOP();
Tim Peters35ba6892000-10-11 07:04:49 +00001184 Py_INCREF(x);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001185 w = SECOND();
Tim Peters35ba6892000-10-11 07:04:49 +00001186 Py_INCREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001187 v = THIRD();
Tim Peters35ba6892000-10-11 07:04:49 +00001188 Py_INCREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001189 STACKADJ(3);
1190 SET_TOP(x);
1191 SET_SECOND(w);
1192 SET_THIRD(v);
Raymond Hettingerf606f872003-03-16 03:11:04 +00001193 goto fast_next_opcode;
Thomas Wouters434d0822000-08-24 20:11:32 +00001194 }
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00001195 Py_FatalError("invalid argument to DUP_TOPX"
1196 " (bytecode corruption?)");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001197 /* Never returns, so don't bother to set why. */
Tim Peters35ba6892000-10-11 07:04:49 +00001198 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001199
Guido van Rossum374a9221991-04-04 10:40:29 +00001200 case UNARY_POSITIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001201 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001202 x = PyNumber_Positive(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001203 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001204 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001205 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001206 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001207
Guido van Rossum374a9221991-04-04 10:40:29 +00001208 case UNARY_NEGATIVE:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001209 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001210 x = PyNumber_Negative(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001211 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001212 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001213 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001214 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001215
Guido van Rossum374a9221991-04-04 10:40:29 +00001216 case UNARY_NOT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001217 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001218 err = PyObject_IsTrue(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001219 Py_DECREF(v);
Guido van Rossumfc490731997-05-06 15:06:49 +00001220 if (err == 0) {
1221 Py_INCREF(Py_True);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001222 SET_TOP(Py_True);
Guido van Rossumfc490731997-05-06 15:06:49 +00001223 continue;
1224 }
1225 else if (err > 0) {
1226 Py_INCREF(Py_False);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001227 SET_TOP(Py_False);
Guido van Rossumfc490731997-05-06 15:06:49 +00001228 err = 0;
1229 continue;
1230 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00001231 STACKADJ(-1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001232 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001233
Guido van Rossum374a9221991-04-04 10:40:29 +00001234 case UNARY_CONVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001235 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001236 x = PyObject_Repr(v);
1237 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001238 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001239 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001240 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001241
Guido van Rossum7928cd71991-10-24 14:59:31 +00001242 case UNARY_INVERT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001243 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001244 x = PyNumber_Invert(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00001245 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001246 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001247 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001248 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001249
Guido van Rossum50564e81996-01-12 01:13:16 +00001250 case BINARY_POWER:
1251 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001252 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001253 x = PyNumber_Power(v, w, Py_None);
Guido van Rossumb209a111997-04-29 18:18:01 +00001254 Py_DECREF(v);
1255 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001256 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001257 if (x != NULL) continue;
Guido van Rossum50564e81996-01-12 01:13:16 +00001258 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001259
Guido van Rossum374a9221991-04-04 10:40:29 +00001260 case BINARY_MULTIPLY:
1261 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001262 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001263 x = PyNumber_Multiply(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001264 Py_DECREF(v);
1265 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001266 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001267 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001268 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001269
Guido van Rossum374a9221991-04-04 10:40:29 +00001270 case BINARY_DIVIDE:
Tim Peters3caca232001-12-06 06:23:26 +00001271 if (!_Py_QnewFlag) {
1272 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001273 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001274 x = PyNumber_Divide(v, w);
1275 Py_DECREF(v);
1276 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001277 SET_TOP(x);
Tim Peters3caca232001-12-06 06:23:26 +00001278 if (x != NULL) continue;
1279 break;
1280 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001281 /* -Qnew is in effect: fall through to
Tim Peters3caca232001-12-06 06:23:26 +00001282 BINARY_TRUE_DIVIDE */
1283 case BINARY_TRUE_DIVIDE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001284 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001285 v = TOP();
Tim Peters3caca232001-12-06 06:23:26 +00001286 x = PyNumber_TrueDivide(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 Rossum4668b002001-08-08 05:00:18 +00001293 case BINARY_FLOOR_DIVIDE:
1294 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001295 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001296 x = PyNumber_FloorDivide(v, w);
1297 Py_DECREF(v);
1298 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001299 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001300 if (x != NULL) continue;
1301 break;
1302
Guido van Rossum374a9221991-04-04 10:40:29 +00001303 case BINARY_MODULO:
1304 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001305 v = TOP();
Collin Winter8725dce2009-02-20 19:30:41 +00001306 if (PyString_CheckExact(v))
1307 x = PyString_Format(v, w);
1308 else
1309 x = PyNumber_Remainder(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001310 Py_DECREF(v);
1311 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001312 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001313 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001314 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001315
Guido van Rossum374a9221991-04-04 10:40:29 +00001316 case BINARY_ADD:
1317 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001318 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001319 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001320 /* INLINE: int + int */
1321 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001322 a = PyInt_AS_LONG(v);
1323 b = PyInt_AS_LONG(w);
Mark Dickinson34398182009-12-02 17:33:41 +00001324 /* cast to avoid undefined behaviour
1325 on overflow */
1326 i = (long)((unsigned long)a + b);
Guido van Rossum87780df2001-08-23 02:58:07 +00001327 if ((i^a) < 0 && (i^b) < 0)
1328 goto slow_add;
1329 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001330 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001331 else if (PyString_CheckExact(v) &&
1332 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001333 x = string_concatenate(v, w, f, next_instr);
1334 /* string_concatenate consumed the ref to v */
1335 goto skip_decref_vx;
1336 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001337 else {
1338 slow_add:
Guido van Rossumc12da691997-07-17 23:12:42 +00001339 x = PyNumber_Add(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001340 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001341 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001342 skip_decref_vx:
Guido van Rossumb209a111997-04-29 18:18:01 +00001343 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001344 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001345 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001346 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001347
Guido van Rossum374a9221991-04-04 10:40:29 +00001348 case BINARY_SUBTRACT:
1349 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001350 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001351 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001352 /* INLINE: int - int */
1353 register long a, b, i;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00001354 a = PyInt_AS_LONG(v);
1355 b = PyInt_AS_LONG(w);
Mark Dickinson34398182009-12-02 17:33:41 +00001356 /* cast to avoid undefined behaviour
1357 on overflow */
1358 i = (long)((unsigned long)a - b);
Guido van Rossum87780df2001-08-23 02:58:07 +00001359 if ((i^a) < 0 && (i^~b) < 0)
1360 goto slow_sub;
1361 x = PyInt_FromLong(i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001362 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001363 else {
1364 slow_sub:
Guido van Rossumc12da691997-07-17 23:12:42 +00001365 x = PyNumber_Subtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001366 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001367 Py_DECREF(v);
1368 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001369 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001370 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001371 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001372
Guido van Rossum374a9221991-04-04 10:40:29 +00001373 case BINARY_SUBSCR:
1374 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001375 v = TOP();
Tim Petersb1c46982001-10-05 20:41:38 +00001376 if (PyList_CheckExact(v) && PyInt_CheckExact(w)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00001377 /* INLINE: list[int] */
Neal Norwitz814e9382006-03-02 07:54:28 +00001378 Py_ssize_t i = PyInt_AsSsize_t(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00001379 if (i < 0)
Guido van Rossumfa00e951998-07-08 15:02:37 +00001380 i += PyList_GET_SIZE(v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001381 if (i >= 0 && i < PyList_GET_SIZE(v)) {
Guido van Rossumfa00e951998-07-08 15:02:37 +00001382 x = PyList_GET_ITEM(v, i);
Guido van Rossumc12da691997-07-17 23:12:42 +00001383 Py_INCREF(x);
1384 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001385 else
1386 goto slow_get;
Guido van Rossumc12da691997-07-17 23:12:42 +00001387 }
1388 else
Raymond Hettinger467a6982004-04-07 11:39:21 +00001389 slow_get:
Guido van Rossumc12da691997-07-17 23:12:42 +00001390 x = PyObject_GetItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001391 Py_DECREF(v);
1392 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001393 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001394 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001395 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001396
Guido van Rossum7928cd71991-10-24 14:59:31 +00001397 case BINARY_LSHIFT:
1398 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001399 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001400 x = PyNumber_Lshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001401 Py_DECREF(v);
1402 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001403 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001404 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001405 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001406
Guido van Rossum7928cd71991-10-24 14:59:31 +00001407 case BINARY_RSHIFT:
1408 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001409 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001410 x = PyNumber_Rshift(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001411 Py_DECREF(v);
1412 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001413 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001414 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001415 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001416
Guido van Rossum7928cd71991-10-24 14:59:31 +00001417 case BINARY_AND:
1418 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001419 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001420 x = PyNumber_And(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001421 Py_DECREF(v);
1422 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001423 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001424 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001425 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001426
Guido van Rossum7928cd71991-10-24 14:59:31 +00001427 case BINARY_XOR:
1428 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001429 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001430 x = PyNumber_Xor(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001431 Py_DECREF(v);
1432 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001433 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001434 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001435 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001436
Guido van Rossum7928cd71991-10-24 14:59:31 +00001437 case BINARY_OR:
1438 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001439 v = TOP();
Guido van Rossumfc490731997-05-06 15:06:49 +00001440 x = PyNumber_Or(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001441 Py_DECREF(v);
1442 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001443 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001444 if (x != NULL) continue;
Guido van Rossum7928cd71991-10-24 14:59:31 +00001445 break;
Thomas Wouters434d0822000-08-24 20:11:32 +00001446
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001447 case LIST_APPEND:
1448 w = POP();
Benjamin Peterson8f7b94e2009-06-28 16:14:07 +00001449 v = PEEK(oparg);
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001450 err = PyList_Append(v, w);
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001451 Py_DECREF(w);
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00001452 if (err == 0) {
1453 PREDICT(JUMP_ABSOLUTE);
1454 continue;
1455 }
Raymond Hettingerdd80f762004-03-07 07:31:06 +00001456 break;
1457
Thomas Wouters434d0822000-08-24 20:11:32 +00001458 case INPLACE_POWER:
1459 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001460 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001461 x = PyNumber_InPlacePower(v, w, Py_None);
1462 Py_DECREF(v);
1463 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001464 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001465 if (x != NULL) continue;
1466 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001467
Thomas Wouters434d0822000-08-24 20:11:32 +00001468 case INPLACE_MULTIPLY:
1469 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001470 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001471 x = PyNumber_InPlaceMultiply(v, w);
1472 Py_DECREF(v);
1473 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001474 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001475 if (x != NULL) continue;
1476 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001477
Thomas Wouters434d0822000-08-24 20:11:32 +00001478 case INPLACE_DIVIDE:
Tim Peters54b11912001-12-25 18:49:11 +00001479 if (!_Py_QnewFlag) {
1480 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001481 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001482 x = PyNumber_InPlaceDivide(v, w);
1483 Py_DECREF(v);
1484 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001485 SET_TOP(x);
Tim Peters54b11912001-12-25 18:49:11 +00001486 if (x != NULL) continue;
1487 break;
1488 }
Raymond Hettinger663004b2003-01-09 15:24:30 +00001489 /* -Qnew is in effect: fall through to
Tim Peters54b11912001-12-25 18:49:11 +00001490 INPLACE_TRUE_DIVIDE */
1491 case INPLACE_TRUE_DIVIDE:
Thomas Wouters434d0822000-08-24 20:11:32 +00001492 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001493 v = TOP();
Tim Peters54b11912001-12-25 18:49:11 +00001494 x = PyNumber_InPlaceTrueDivide(v, w);
Thomas Wouters434d0822000-08-24 20:11:32 +00001495 Py_DECREF(v);
1496 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001497 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001498 if (x != NULL) continue;
1499 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001500
Guido van Rossum4668b002001-08-08 05:00:18 +00001501 case INPLACE_FLOOR_DIVIDE:
1502 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001503 v = TOP();
Guido van Rossum4668b002001-08-08 05:00:18 +00001504 x = PyNumber_InPlaceFloorDivide(v, w);
1505 Py_DECREF(v);
1506 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001507 SET_TOP(x);
Guido van Rossum4668b002001-08-08 05:00:18 +00001508 if (x != NULL) continue;
1509 break;
1510
Thomas Wouters434d0822000-08-24 20:11:32 +00001511 case INPLACE_MODULO:
1512 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001513 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001514 x = PyNumber_InPlaceRemainder(v, w);
1515 Py_DECREF(v);
1516 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001517 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001518 if (x != NULL) continue;
1519 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001520
Thomas Wouters434d0822000-08-24 20:11:32 +00001521 case INPLACE_ADD:
1522 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001523 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001524 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001525 /* INLINE: int + int */
1526 register long a, b, i;
1527 a = PyInt_AS_LONG(v);
1528 b = PyInt_AS_LONG(w);
1529 i = a + b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001530 if ((i^a) < 0 && (i^b) < 0)
1531 goto slow_iadd;
1532 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001533 }
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001534 else if (PyString_CheckExact(v) &&
1535 PyString_CheckExact(w)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001536 x = string_concatenate(v, w, f, next_instr);
1537 /* string_concatenate consumed the ref to v */
1538 goto skip_decref_v;
1539 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001540 else {
1541 slow_iadd:
Thomas Wouters434d0822000-08-24 20:11:32 +00001542 x = PyNumber_InPlaceAdd(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001543 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001544 Py_DECREF(v);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00001545 skip_decref_v:
Thomas Wouters434d0822000-08-24 20:11:32 +00001546 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001547 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001548 if (x != NULL) continue;
1549 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001550
Thomas Wouters434d0822000-08-24 20:11:32 +00001551 case INPLACE_SUBTRACT:
1552 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001553 v = TOP();
Tim Petersc1e6d962001-10-05 20:21:03 +00001554 if (PyInt_CheckExact(v) && PyInt_CheckExact(w)) {
Thomas Wouters434d0822000-08-24 20:11:32 +00001555 /* INLINE: int - int */
1556 register long a, b, i;
1557 a = PyInt_AS_LONG(v);
1558 b = PyInt_AS_LONG(w);
1559 i = a - b;
Guido van Rossum87780df2001-08-23 02:58:07 +00001560 if ((i^a) < 0 && (i^~b) < 0)
1561 goto slow_isub;
1562 x = PyInt_FromLong(i);
Thomas Wouters434d0822000-08-24 20:11:32 +00001563 }
Guido van Rossum87780df2001-08-23 02:58:07 +00001564 else {
1565 slow_isub:
Thomas Wouters434d0822000-08-24 20:11:32 +00001566 x = PyNumber_InPlaceSubtract(v, w);
Guido van Rossum87780df2001-08-23 02:58:07 +00001567 }
Thomas Wouters434d0822000-08-24 20:11:32 +00001568 Py_DECREF(v);
1569 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001570 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001571 if (x != NULL) continue;
1572 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001573
Thomas Wouters434d0822000-08-24 20:11:32 +00001574 case INPLACE_LSHIFT:
1575 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001576 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001577 x = PyNumber_InPlaceLshift(v, w);
1578 Py_DECREF(v);
1579 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001580 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001581 if (x != NULL) continue;
1582 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001583
Thomas Wouters434d0822000-08-24 20:11:32 +00001584 case INPLACE_RSHIFT:
1585 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001586 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001587 x = PyNumber_InPlaceRshift(v, w);
1588 Py_DECREF(v);
1589 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001590 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001591 if (x != NULL) continue;
1592 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001593
Thomas Wouters434d0822000-08-24 20:11:32 +00001594 case INPLACE_AND:
1595 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001596 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001597 x = PyNumber_InPlaceAnd(v, w);
1598 Py_DECREF(v);
1599 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001600 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001601 if (x != NULL) continue;
1602 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001603
Thomas Wouters434d0822000-08-24 20:11:32 +00001604 case INPLACE_XOR:
1605 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001606 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001607 x = PyNumber_InPlaceXor(v, w);
1608 Py_DECREF(v);
1609 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001610 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001611 if (x != NULL) continue;
1612 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001613
Thomas Wouters434d0822000-08-24 20:11:32 +00001614 case INPLACE_OR:
1615 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00001616 v = TOP();
Thomas Wouters434d0822000-08-24 20:11:32 +00001617 x = PyNumber_InPlaceOr(v, w);
1618 Py_DECREF(v);
1619 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001620 SET_TOP(x);
Thomas Wouters434d0822000-08-24 20:11:32 +00001621 if (x != NULL) continue;
1622 break;
1623
Guido van Rossum374a9221991-04-04 10:40:29 +00001624 case SLICE+0:
1625 case SLICE+1:
1626 case SLICE+2:
1627 case SLICE+3:
1628 if ((opcode-SLICE) & 2)
1629 w = POP();
1630 else
1631 w = NULL;
1632 if ((opcode-SLICE) & 1)
1633 v = POP();
1634 else
1635 v = NULL;
Raymond Hettinger663004b2003-01-09 15:24:30 +00001636 u = TOP();
Guido van Rossum374a9221991-04-04 10:40:29 +00001637 x = apply_slice(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001638 Py_DECREF(u);
1639 Py_XDECREF(v);
1640 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001641 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001642 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001643 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001644
Guido van Rossum374a9221991-04-04 10:40:29 +00001645 case STORE_SLICE+0:
1646 case STORE_SLICE+1:
1647 case STORE_SLICE+2:
1648 case STORE_SLICE+3:
1649 if ((opcode-STORE_SLICE) & 2)
1650 w = POP();
1651 else
1652 w = NULL;
1653 if ((opcode-STORE_SLICE) & 1)
1654 v = POP();
1655 else
1656 v = NULL;
1657 u = POP();
1658 t = POP();
1659 err = assign_slice(u, v, w, t); /* u[v:w] = t */
Guido van Rossumb209a111997-04-29 18:18:01 +00001660 Py_DECREF(t);
1661 Py_DECREF(u);
1662 Py_XDECREF(v);
1663 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001664 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001665 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001666
Guido van Rossum374a9221991-04-04 10:40:29 +00001667 case DELETE_SLICE+0:
1668 case DELETE_SLICE+1:
1669 case DELETE_SLICE+2:
1670 case DELETE_SLICE+3:
1671 if ((opcode-DELETE_SLICE) & 2)
1672 w = POP();
1673 else
1674 w = NULL;
1675 if ((opcode-DELETE_SLICE) & 1)
1676 v = POP();
1677 else
1678 v = NULL;
1679 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001680 err = assign_slice(u, v, w, (PyObject *)NULL);
Guido van Rossum374a9221991-04-04 10:40:29 +00001681 /* del u[v:w] */
Guido van Rossumb209a111997-04-29 18:18:01 +00001682 Py_DECREF(u);
1683 Py_XDECREF(v);
1684 Py_XDECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001685 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001686 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001687
Guido van Rossum374a9221991-04-04 10:40:29 +00001688 case STORE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001689 w = TOP();
1690 v = SECOND();
1691 u = THIRD();
1692 STACKADJ(-3);
Guido van Rossum374a9221991-04-04 10:40:29 +00001693 /* v[w] = u */
Guido van Rossumfc490731997-05-06 15:06:49 +00001694 err = PyObject_SetItem(v, w, u);
Guido van Rossumb209a111997-04-29 18:18:01 +00001695 Py_DECREF(u);
1696 Py_DECREF(v);
1697 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001698 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001699 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001700
Guido van Rossum374a9221991-04-04 10:40:29 +00001701 case DELETE_SUBSCR:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001702 w = TOP();
1703 v = SECOND();
1704 STACKADJ(-2);
Guido van Rossum374a9221991-04-04 10:40:29 +00001705 /* del v[w] */
Guido van Rossumfc490731997-05-06 15:06:49 +00001706 err = PyObject_DelItem(v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001707 Py_DECREF(v);
1708 Py_DECREF(w);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00001709 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001710 break;
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001711
Guido van Rossum374a9221991-04-04 10:40:29 +00001712 case PRINT_EXPR:
1713 v = POP();
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001714 w = PySys_GetObject("displayhook");
1715 if (w == NULL) {
1716 PyErr_SetString(PyExc_RuntimeError,
1717 "lost sys.displayhook");
1718 err = -1;
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001719 x = NULL;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001720 }
1721 if (err == 0) {
Raymond Hettinger8ae46892003-10-12 19:09:37 +00001722 x = PyTuple_Pack(1, v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001723 if (x == NULL)
1724 err = -1;
1725 }
1726 if (err == 0) {
1727 w = PyEval_CallObject(w, x);
Moshe Zadkaf5df3832001-01-11 11:55:37 +00001728 Py_XDECREF(w);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001729 if (w == NULL)
1730 err = -1;
Guido van Rossum374a9221991-04-04 10:40:29 +00001731 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001732 Py_DECREF(v);
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001733 Py_XDECREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00001734 break;
Moshe Zadkaf68f2fe2001-01-11 05:41:27 +00001735
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001736 case PRINT_ITEM_TO:
1737 w = stream = POP();
1738 /* fall through to PRINT_ITEM */
1739
Guido van Rossum374a9221991-04-04 10:40:29 +00001740 case PRINT_ITEM:
1741 v = POP();
Barry Warsaw093abe02000-08-29 04:56:13 +00001742 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001743 w = PySys_GetObject("stdout");
1744 if (w == NULL) {
1745 PyErr_SetString(PyExc_RuntimeError,
1746 "lost sys.stdout");
1747 err = -1;
1748 }
Guido van Rossum8f183201997-12-31 05:53:15 +00001749 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001750 /* PyFile_SoftSpace() can exececute arbitrary code
1751 if sys.stdout is an instance with a __getattr__.
1752 If __getattr__ raises an exception, w will
1753 be freed, so we need to prevent that temporarily. */
1754 Py_XINCREF(w);
Tim Peters8e5fd532002-03-24 19:25:00 +00001755 if (w != NULL && PyFile_SoftSpace(w, 0))
Guido van Rossumbe270261997-05-22 22:26:18 +00001756 err = PyFile_WriteString(" ", w);
1757 if (err == 0)
1758 err = PyFile_WriteObject(v, w, Py_PRINT_RAW);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001759 if (err == 0) {
Tim Peters8e5fd532002-03-24 19:25:00 +00001760 /* XXX move into writeobject() ? */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001761 if (PyString_Check(v)) {
1762 char *s = PyString_AS_STRING(v);
1763 Py_ssize_t len = PyString_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001764 if (len == 0 ||
1765 !isspace(Py_CHARMASK(s[len-1])) ||
1766 s[len-1] == ' ')
1767 PyFile_SoftSpace(w, 1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001768 }
Martin v. Löwis8d3ce5a2001-12-18 22:36:40 +00001769#ifdef Py_USING_UNICODE
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001770 else if (PyUnicode_Check(v)) {
1771 Py_UNICODE *s = PyUnicode_AS_UNICODE(v);
Martin v. Löwis66851282006-04-22 11:40:03 +00001772 Py_ssize_t len = PyUnicode_GET_SIZE(v);
Tim Peters8e5fd532002-03-24 19:25:00 +00001773 if (len == 0 ||
1774 !Py_UNICODE_ISSPACE(s[len-1]) ||
1775 s[len-1] == ' ')
1776 PyFile_SoftSpace(w, 1);
Marc-André Lemburg0c4d8d02001-11-20 15:17:25 +00001777 }
Michael W. Hudsond95c8282002-05-20 13:56:11 +00001778#endif
Tim Peters8e5fd532002-03-24 19:25:00 +00001779 else
1780 PyFile_SoftSpace(w, 1);
Guido van Rossum374a9221991-04-04 10:40:29 +00001781 }
Neal Norwitzc5131bc2003-06-29 14:48:32 +00001782 Py_XDECREF(w);
Guido van Rossumb209a111997-04-29 18:18:01 +00001783 Py_DECREF(v);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001784 Py_XDECREF(stream);
1785 stream = NULL;
1786 if (err == 0)
1787 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001788 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001789
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001790 case PRINT_NEWLINE_TO:
1791 w = stream = POP();
1792 /* fall through to PRINT_NEWLINE */
1793
Guido van Rossum374a9221991-04-04 10:40:29 +00001794 case PRINT_NEWLINE:
Barry Warsaw093abe02000-08-29 04:56:13 +00001795 if (stream == NULL || stream == Py_None) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001796 w = PySys_GetObject("stdout");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001797 if (w == NULL) {
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001798 PyErr_SetString(PyExc_RuntimeError,
1799 "lost sys.stdout");
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00001800 why = WHY_EXCEPTION;
1801 }
Guido van Rossum3165fe61992-09-25 21:59:05 +00001802 }
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001803 if (w != NULL) {
Georg Brandlaa76d772008-07-01 20:56:03 +00001804 /* w.write() may replace sys.stdout, so we
1805 * have to keep our reference to it */
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001806 Py_INCREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001807 err = PyFile_WriteString("\n", w);
1808 if (err == 0)
1809 PyFile_SoftSpace(w, 0);
Amaury Forgeot d'Arcbdd941f2008-07-01 20:38:04 +00001810 Py_DECREF(w);
Barry Warsaw23c9ec82000-08-21 15:44:01 +00001811 }
1812 Py_XDECREF(stream);
1813 stream = NULL;
Guido van Rossum374a9221991-04-04 10:40:29 +00001814 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001815
Thomas Wouters434d0822000-08-24 20:11:32 +00001816
1817#ifdef CASE_TOO_BIG
1818 default: switch (opcode) {
1819#endif
Guido van Rossumf10570b1995-07-07 22:53:21 +00001820 case RAISE_VARARGS:
1821 u = v = w = NULL;
1822 switch (oparg) {
1823 case 3:
1824 u = POP(); /* traceback */
Guido van Rossumf10570b1995-07-07 22:53:21 +00001825 /* Fallthrough */
1826 case 2:
1827 v = POP(); /* value */
1828 /* Fallthrough */
1829 case 1:
1830 w = POP(); /* exc */
Guido van Rossumd295f121998-04-09 21:39:57 +00001831 case 0: /* Fallthrough */
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001832 why = do_raise(w, v, u);
Guido van Rossumf10570b1995-07-07 22:53:21 +00001833 break;
1834 default:
Guido van Rossumb209a111997-04-29 18:18:01 +00001835 PyErr_SetString(PyExc_SystemError,
Guido van Rossumf10570b1995-07-07 22:53:21 +00001836 "bad RAISE_VARARGS oparg");
Guido van Rossumf10570b1995-07-07 22:53:21 +00001837 why = WHY_EXCEPTION;
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00001838 break;
1839 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001840 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001841
Guido van Rossum374a9221991-04-04 10:40:29 +00001842 case LOAD_LOCALS:
Raymond Hettinger467a6982004-04-07 11:39:21 +00001843 if ((x = f->f_locals) != NULL) {
1844 Py_INCREF(x);
1845 PUSH(x);
1846 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001847 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001848 PyErr_SetString(PyExc_SystemError, "no locals");
Guido van Rossum374a9221991-04-04 10:40:29 +00001849 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001850
Guido van Rossum374a9221991-04-04 10:40:29 +00001851 case RETURN_VALUE:
1852 retval = POP();
1853 why = WHY_RETURN;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001854 goto fast_block_end;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001855
Tim Peters5ca576e2001-06-18 22:08:13 +00001856 case YIELD_VALUE:
1857 retval = POP();
Tim Peters8c963692001-06-23 05:26:56 +00001858 f->f_stacktop = stack_pointer;
Tim Peters5ca576e2001-06-18 22:08:13 +00001859 why = WHY_YIELD;
Raymond Hettinger1dd83092004-02-06 18:32:33 +00001860 goto fast_yield;
Tim Peters5ca576e2001-06-18 22:08:13 +00001861
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001862 case EXEC_STMT:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001863 w = TOP();
1864 v = SECOND();
1865 u = THIRD();
1866 STACKADJ(-3);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001867 READ_TIMESTAMP(intr0);
Guido van Rossuma027efa1997-05-05 20:56:21 +00001868 err = exec_statement(f, u, v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00001869 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00001870 Py_DECREF(u);
1871 Py_DECREF(v);
1872 Py_DECREF(w);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00001873 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001874
Guido van Rossum374a9221991-04-04 10:40:29 +00001875 case POP_BLOCK:
1876 {
Guido van Rossumb209a111997-04-29 18:18:01 +00001877 PyTryBlock *b = PyFrame_BlockPop(f);
Guido van Rossum374a9221991-04-04 10:40:29 +00001878 while (STACK_LEVEL() > b->b_level) {
1879 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001880 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001881 }
1882 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001883 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00001884
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00001885 PREDICTED(END_FINALLY);
Guido van Rossum374a9221991-04-04 10:40:29 +00001886 case END_FINALLY:
1887 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001888 if (PyInt_Check(v)) {
Raymond Hettinger7c958652004-04-06 10:11:10 +00001889 why = (enum why_code) PyInt_AS_LONG(v);
Tim Peters8a5c3c72004-04-05 19:36:21 +00001890 assert(why != WHY_YIELD);
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00001891 if (why == WHY_RETURN ||
1892 why == WHY_CONTINUE)
Guido van Rossum374a9221991-04-04 10:40:29 +00001893 retval = POP();
1894 }
Thomas Wouterse2176022007-09-20 17:35:10 +00001895 else if (PyExceptionClass_Check(v) ||
Gregory P. Smithdd96db62008-06-09 04:58:54 +00001896 PyString_Check(v)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00001897 w = POP();
Guido van Rossumf10570b1995-07-07 22:53:21 +00001898 u = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00001899 PyErr_Restore(v, w, u);
Guido van Rossum374a9221991-04-04 10:40:29 +00001900 why = WHY_RERAISE;
Guido van Rossum0db1ef91995-07-28 23:06:00 +00001901 break;
Guido van Rossum374a9221991-04-04 10:40:29 +00001902 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001903 else if (v != Py_None) {
1904 PyErr_SetString(PyExc_SystemError,
Guido van Rossum374a9221991-04-04 10:40:29 +00001905 "'finally' pops bad exception");
1906 why = WHY_EXCEPTION;
1907 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001908 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001909 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001910
Guido van Rossum374a9221991-04-04 10:40:29 +00001911 case BUILD_CLASS:
Raymond Hettinger663004b2003-01-09 15:24:30 +00001912 u = TOP();
1913 v = SECOND();
1914 w = THIRD();
1915 STACKADJ(-2);
Guido van Rossum25831651993-05-19 14:50:45 +00001916 x = build_class(u, v, w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001917 SET_TOP(x);
Guido van Rossumb209a111997-04-29 18:18:01 +00001918 Py_DECREF(u);
1919 Py_DECREF(v);
1920 Py_DECREF(w);
Guido van Rossum374a9221991-04-04 10:40:29 +00001921 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001922
Guido van Rossum374a9221991-04-04 10:40:29 +00001923 case STORE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001924 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00001925 v = POP();
Raymond Hettinger467a6982004-04-07 11:39:21 +00001926 if ((x = f->f_locals) != NULL) {
Raymond Hettinger66bd2332004-08-02 08:30:07 +00001927 if (PyDict_CheckExact(x))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001928 err = PyDict_SetItem(x, w, v);
1929 else
1930 err = PyObject_SetItem(x, w, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001931 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001932 if (err == 0) continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00001933 break;
1934 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001935 PyErr_Format(PyExc_SystemError,
1936 "no locals found when storing %s",
1937 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001938 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001939
Guido van Rossum374a9221991-04-04 10:40:29 +00001940 case DELETE_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00001941 w = GETITEM(names, oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00001942 if ((x = f->f_locals) != NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00001943 if ((err = PyObject_DelItem(x, w)) != 0)
Raymond Hettinger467a6982004-04-07 11:39:21 +00001944 format_exc_check_arg(PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00001945 NAME_ERROR_MSG,
1946 w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00001947 break;
1948 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00001949 PyErr_Format(PyExc_SystemError,
1950 "no locals when deleting %s",
1951 PyObject_REPR(w));
Guido van Rossum374a9221991-04-04 10:40:29 +00001952 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00001953
Raymond Hettinger7dc52212003-03-16 20:14:44 +00001954 PREDICTED_WITH_ARG(UNPACK_SEQUENCE);
Thomas Wouters0be5aab2000-08-11 22:15:52 +00001955 case UNPACK_SEQUENCE:
Guido van Rossum374a9221991-04-04 10:40:29 +00001956 v = POP();
Thomas Wouterse2176022007-09-20 17:35:10 +00001957 if (PyTuple_CheckExact(v) &&
1958 PyTuple_GET_SIZE(v) == oparg) {
1959 PyObject **items = \
1960 ((PyTupleObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001961 while (oparg--) {
1962 w = items[oparg];
1963 Py_INCREF(w);
1964 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001965 }
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001966 Py_DECREF(v);
1967 continue;
Thomas Wouterse2176022007-09-20 17:35:10 +00001968 } else if (PyList_CheckExact(v) &&
1969 PyList_GET_SIZE(v) == oparg) {
1970 PyObject **items = \
1971 ((PyListObject *)v)->ob_item;
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001972 while (oparg--) {
1973 w = items[oparg];
1974 Py_INCREF(w);
1975 PUSH(w);
Barry Warsawe42b18f1997-08-25 22:13:04 +00001976 }
Raymond Hettingerf114a3a2004-03-08 23:25:30 +00001977 } else if (unpack_iterable(v, oparg,
Thomas Wouterse2176022007-09-20 17:35:10 +00001978 stack_pointer + oparg)) {
Benjamin Peterson8f7b94e2009-06-28 16:14:07 +00001979 STACKADJ(oparg);
Georg Brandl5cb76c12007-03-21 09:00:39 +00001980 } else {
1981 /* unpack_iterable() raised an exception */
Barry Warsawe42b18f1997-08-25 22:13:04 +00001982 why = WHY_EXCEPTION;
Tim Peters8b13b3e2001-09-30 05:58:42 +00001983 }
Guido van Rossumb209a111997-04-29 18:18:01 +00001984 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00001985 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001986
Guido van Rossum374a9221991-04-04 10:40:29 +00001987 case STORE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001988 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00001989 v = TOP();
1990 u = SECOND();
1991 STACKADJ(-2);
Guido van Rossumb209a111997-04-29 18:18:01 +00001992 err = PyObject_SetAttr(v, w, u); /* v.w = u */
1993 Py_DECREF(v);
1994 Py_DECREF(u);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00001995 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00001996 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00001997
Guido van Rossum374a9221991-04-04 10:40:29 +00001998 case DELETE_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00001999 w = GETITEM(names, oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002000 v = POP();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002001 err = PyObject_SetAttr(v, w, (PyObject *)NULL);
2002 /* del v.w */
Guido van Rossumb209a111997-04-29 18:18:01 +00002003 Py_DECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002004 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002005
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00002006 case STORE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00002007 w = GETITEM(names, oparg);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00002008 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002009 err = PyDict_SetItem(f->f_globals, w, v);
2010 Py_DECREF(v);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00002011 if (err == 0) continue;
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00002012 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002013
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00002014 case DELETE_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00002015 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002016 if ((err = PyDict_DelItem(f->f_globals, w)) != 0)
Paul Prescode68140d2000-08-30 20:25:01 +00002017 format_exc_check_arg(
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002018 PyExc_NameError, GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum32c6cdf1991-12-10 13:52:46 +00002019 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002020
Guido van Rossum374a9221991-04-04 10:40:29 +00002021 case LOAD_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002022 w = GETITEM(names, oparg);
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002023 if ((v = f->f_locals) == NULL) {
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002024 PyErr_Format(PyExc_SystemError,
2025 "no locals when loading %s",
Jeremy Hylton483638c2001-02-01 20:20:45 +00002026 PyObject_REPR(w));
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002027 why = WHY_EXCEPTION;
Guido van Rossum681d79a1995-07-18 14:51:37 +00002028 break;
2029 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002030 if (PyDict_CheckExact(v)) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002031 x = PyDict_GetItem(v, w);
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002032 Py_XINCREF(x);
2033 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002034 else {
2035 x = PyObject_GetItem(v, w);
2036 if (x == NULL && PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002037 if (!PyErr_ExceptionMatches(
2038 PyExc_KeyError))
Raymond Hettinger214b1c32004-07-02 06:41:07 +00002039 break;
2040 PyErr_Clear();
2041 }
2042 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002043 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002044 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002045 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002046 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002047 if (x == NULL) {
Paul Prescode68140d2000-08-30 20:25:01 +00002048 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00002049 PyExc_NameError,
Thomas Wouterse2176022007-09-20 17:35:10 +00002050 NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002051 break;
2052 }
2053 }
Michael W. Hudsona3711f72004-08-02 14:50:43 +00002054 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00002055 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002056 PUSH(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002057 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002058
Guido van Rossum374a9221991-04-04 10:40:29 +00002059 case LOAD_GLOBAL:
Skip Montanaro496e6582002-08-06 17:47:40 +00002060 w = GETITEM(names, oparg);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002061 if (PyString_CheckExact(w)) {
Guido van Rossumd8dbf842002-08-19 21:17:53 +00002062 /* Inline the PyDict_GetItem() calls.
2063 WARNING: this is an extreme speed hack.
2064 Do not try this at home. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00002065 long hash = ((PyStringObject *)w)->ob_shash;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002066 if (hash != -1) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002067 PyDictObject *d;
Armin Rigo35f6d362006-06-01 13:19:12 +00002068 PyDictEntry *e;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002069 d = (PyDictObject *)(f->f_globals);
Armin Rigo35f6d362006-06-01 13:19:12 +00002070 e = d->ma_lookup(d, w, hash);
2071 if (e == NULL) {
2072 x = NULL;
2073 break;
2074 }
2075 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002076 if (x != NULL) {
2077 Py_INCREF(x);
2078 PUSH(x);
2079 continue;
2080 }
2081 d = (PyDictObject *)(f->f_builtins);
Armin Rigo35f6d362006-06-01 13:19:12 +00002082 e = d->ma_lookup(d, w, hash);
2083 if (e == NULL) {
2084 x = NULL;
2085 break;
2086 }
2087 x = e->me_value;
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002088 if (x != NULL) {
2089 Py_INCREF(x);
2090 PUSH(x);
2091 continue;
2092 }
2093 goto load_global_error;
2094 }
2095 }
2096 /* This is the un-inlined version of the code above */
Guido van Rossumb209a111997-04-29 18:18:01 +00002097 x = PyDict_GetItem(f->f_globals, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002098 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002099 x = PyDict_GetItem(f->f_builtins, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002100 if (x == NULL) {
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002101 load_global_error:
Paul Prescode68140d2000-08-30 20:25:01 +00002102 format_exc_check_arg(
Guido van Rossumac7be682001-01-17 15:42:30 +00002103 PyExc_NameError,
Guido van Rossum3a4dfc82002-08-19 20:24:07 +00002104 GLOBAL_NAME_ERROR_MSG, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002105 break;
2106 }
2107 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002108 Py_INCREF(x);
Guido van Rossum374a9221991-04-04 10:40:29 +00002109 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00002110 continue;
Guido van Rossum681d79a1995-07-18 14:51:37 +00002111
Guido van Rossum8b17d6b1993-03-30 13:18:41 +00002112 case DELETE_FAST:
Guido van Rossum2e4c8991998-05-12 20:27:36 +00002113 x = GETLOCAL(oparg);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002114 if (x != NULL) {
2115 SETLOCAL(oparg, NULL);
2116 continue;
Guido van Rossum2e4c8991998-05-12 20:27:36 +00002117 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00002118 format_exc_check_arg(
2119 PyExc_UnboundLocalError,
2120 UNBOUNDLOCAL_ERROR_MSG,
2121 PyTuple_GetItem(co->co_varnames, oparg)
2122 );
2123 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002124
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002125 case LOAD_CLOSURE:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002126 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002127 Py_INCREF(x);
2128 PUSH(x);
Raymond Hettinger7eddd782004-04-07 14:38:08 +00002129 if (x != NULL) continue;
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002130 break;
2131
2132 case LOAD_DEREF:
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002133 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002134 w = PyCell_Get(x);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002135 if (w != NULL) {
2136 PUSH(w);
2137 continue;
Jeremy Hylton2524d692001-02-05 17:23:16 +00002138 }
Raymond Hettinger467a6982004-04-07 11:39:21 +00002139 err = -1;
2140 /* Don't stomp existing exception */
2141 if (PyErr_Occurred())
2142 break;
Richard Jonescebbefc2006-05-23 18:28:17 +00002143 if (oparg < PyTuple_GET_SIZE(co->co_cellvars)) {
2144 v = PyTuple_GET_ITEM(co->co_cellvars,
Raymond Hettinger467a6982004-04-07 11:39:21 +00002145 oparg);
2146 format_exc_check_arg(
2147 PyExc_UnboundLocalError,
2148 UNBOUNDLOCAL_ERROR_MSG,
2149 v);
2150 } else {
Thomas Wouterse2176022007-09-20 17:35:10 +00002151 v = PyTuple_GET_ITEM(co->co_freevars, oparg -
2152 PyTuple_GET_SIZE(co->co_cellvars));
2153 format_exc_check_arg(PyExc_NameError,
2154 UNBOUNDFREE_ERROR_MSG, v);
Raymond Hettinger467a6982004-04-07 11:39:21 +00002155 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002156 break;
2157
2158 case STORE_DEREF:
2159 w = POP();
Jeremy Hylton2b724da2001-01-29 22:51:52 +00002160 x = freevars[oparg];
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002161 PyCell_Set(x, w);
Jeremy Hylton30c9f392001-03-13 01:58:22 +00002162 Py_DECREF(w);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002163 continue;
2164
Guido van Rossum374a9221991-04-04 10:40:29 +00002165 case BUILD_TUPLE:
Guido van Rossumb209a111997-04-29 18:18:01 +00002166 x = PyTuple_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002167 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002168 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002169 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002170 PyTuple_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002171 }
2172 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002173 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002174 }
2175 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002176
Guido van Rossum374a9221991-04-04 10:40:29 +00002177 case BUILD_LIST:
Guido van Rossumb209a111997-04-29 18:18:01 +00002178 x = PyList_New(oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002179 if (x != NULL) {
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002180 for (; --oparg >= 0;) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002181 w = POP();
Guido van Rossum5053efc1998-08-04 15:27:50 +00002182 PyList_SET_ITEM(x, oparg, w);
Guido van Rossum374a9221991-04-04 10:40:29 +00002183 }
2184 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002185 continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002186 }
2187 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002188
Guido van Rossum374a9221991-04-04 10:40:29 +00002189 case BUILD_MAP:
Raymond Hettingerfd7ed402007-12-18 21:24:09 +00002190 x = _PyDict_NewPresized((Py_ssize_t)oparg);
Guido van Rossum374a9221991-04-04 10:40:29 +00002191 PUSH(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002192 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002193 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002194
Raymond Hettingereffde122007-12-18 18:26:18 +00002195 case STORE_MAP:
2196 w = TOP(); /* key */
2197 u = SECOND(); /* value */
2198 v = THIRD(); /* dict */
2199 STACKADJ(-2);
2200 assert (PyDict_CheckExact(v));
2201 err = PyDict_SetItem(v, w, u); /* v[w] = u */
2202 Py_DECREF(u);
2203 Py_DECREF(w);
2204 if (err == 0) continue;
2205 break;
2206
Guido van Rossum374a9221991-04-04 10:40:29 +00002207 case LOAD_ATTR:
Skip Montanaro496e6582002-08-06 17:47:40 +00002208 w = GETITEM(names, oparg);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002209 v = TOP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002210 x = PyObject_GetAttr(v, w);
2211 Py_DECREF(v);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002212 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002213 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002214 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002215
Guido van Rossum374a9221991-04-04 10:40:29 +00002216 case COMPARE_OP:
2217 w = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002218 v = TOP();
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00002219 if (PyInt_CheckExact(w) && PyInt_CheckExact(v)) {
Guido van Rossumc12da691997-07-17 23:12:42 +00002220 /* INLINE: cmp(int, int) */
2221 register long a, b;
2222 register int res;
Guido van Rossumcf183ac1998-12-04 18:51:36 +00002223 a = PyInt_AS_LONG(v);
2224 b = PyInt_AS_LONG(w);
Guido van Rossumc12da691997-07-17 23:12:42 +00002225 switch (oparg) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00002226 case PyCmp_LT: res = a < b; break;
2227 case PyCmp_LE: res = a <= b; break;
2228 case PyCmp_EQ: res = a == b; break;
2229 case PyCmp_NE: res = a != b; break;
2230 case PyCmp_GT: res = a > b; break;
2231 case PyCmp_GE: res = a >= b; break;
2232 case PyCmp_IS: res = v == w; break;
2233 case PyCmp_IS_NOT: res = v != w; break;
Guido van Rossumc12da691997-07-17 23:12:42 +00002234 default: goto slow_compare;
2235 }
2236 x = res ? Py_True : Py_False;
2237 Py_INCREF(x);
2238 }
2239 else {
2240 slow_compare:
2241 x = cmp_outcome(oparg, v, w);
2242 }
Guido van Rossumb209a111997-04-29 18:18:01 +00002243 Py_DECREF(v);
2244 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002245 SET_TOP(x);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002246 if (x == NULL) break;
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002247 PREDICT(POP_JUMP_IF_FALSE);
2248 PREDICT(POP_JUMP_IF_TRUE);
Raymond Hettingerf606f872003-03-16 03:11:04 +00002249 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002250
Guido van Rossum374a9221991-04-04 10:40:29 +00002251 case IMPORT_NAME:
Skip Montanaro496e6582002-08-06 17:47:40 +00002252 w = GETITEM(names, oparg);
Guido van Rossumb209a111997-04-29 18:18:01 +00002253 x = PyDict_GetItemString(f->f_builtins, "__import__");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002254 if (x == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002255 PyErr_SetString(PyExc_ImportError,
Guido van Rossumfc490731997-05-06 15:06:49 +00002256 "__import__ not found");
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002257 break;
2258 }
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002259 Py_INCREF(x);
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002260 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002261 u = TOP();
Thomas Woutersf7f438b2006-02-28 16:09:29 +00002262 if (PyInt_AsLong(u) != -1 || PyErr_Occurred())
2263 w = PyTuple_Pack(5,
2264 w,
2265 f->f_globals,
2266 f->f_locals == NULL ?
2267 Py_None : f->f_locals,
2268 v,
2269 u);
2270 else
2271 w = PyTuple_Pack(4,
2272 w,
2273 f->f_globals,
2274 f->f_locals == NULL ?
2275 Py_None : f->f_locals,
2276 v);
2277 Py_DECREF(v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002278 Py_DECREF(u);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002279 if (w == NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002280 u = POP();
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002281 Py_DECREF(x);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00002282 x = NULL;
2283 break;
2284 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002285 READ_TIMESTAMP(intr0);
Guido van Rossum1d9a9ea2008-01-23 20:19:01 +00002286 v = x;
2287 x = PyEval_CallObject(v, w);
2288 Py_DECREF(v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002289 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002290 Py_DECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002291 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002292 if (x != NULL) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002293 break;
Guido van Rossumac7be682001-01-17 15:42:30 +00002294
Thomas Wouters52152252000-08-17 22:55:00 +00002295 case IMPORT_STAR:
2296 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002297 PyFrame_FastToLocals(f);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002298 if ((x = f->f_locals) == NULL) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002299 PyErr_SetString(PyExc_SystemError,
Jeremy Hyltonc862cf42001-01-19 03:25:05 +00002300 "no locals found during 'import *'");
Guido van Rossum681d79a1995-07-18 14:51:37 +00002301 break;
2302 }
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002303 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002304 err = import_all_from(x, v);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002305 READ_TIMESTAMP(intr1);
Guido van Rossumb209a111997-04-29 18:18:01 +00002306 PyFrame_LocalsToFast(f, 0);
Thomas Wouters52152252000-08-17 22:55:00 +00002307 Py_DECREF(v);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002308 if (err == 0) continue;
Guido van Rossum374a9221991-04-04 10:40:29 +00002309 break;
Guido van Rossum25831651993-05-19 14:50:45 +00002310
Thomas Wouters52152252000-08-17 22:55:00 +00002311 case IMPORT_FROM:
Skip Montanaro496e6582002-08-06 17:47:40 +00002312 w = GETITEM(names, oparg);
Thomas Wouters52152252000-08-17 22:55:00 +00002313 v = TOP();
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002314 READ_TIMESTAMP(intr0);
Thomas Wouters52152252000-08-17 22:55:00 +00002315 x = import_from(v, w);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002316 READ_TIMESTAMP(intr1);
Thomas Wouters52152252000-08-17 22:55:00 +00002317 PUSH(x);
2318 if (x != NULL) continue;
2319 break;
2320
Guido van Rossum374a9221991-04-04 10:40:29 +00002321 case JUMP_FORWARD:
2322 JUMPBY(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002323 goto fast_next_opcode;
Guido van Rossumac7be682001-01-17 15:42:30 +00002324
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002325 PREDICTED_WITH_ARG(POP_JUMP_IF_FALSE);
2326 case POP_JUMP_IF_FALSE:
2327 w = POP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002328 if (w == Py_True) {
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002329 Py_DECREF(w);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002330 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002331 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002332 if (w == Py_False) {
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002333 Py_DECREF(w);
2334 JUMPTO(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002335 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002336 }
2337 err = PyObject_IsTrue(w);
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002338 Py_DECREF(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002339 if (err > 0)
2340 err = 0;
2341 else if (err == 0)
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002342 JUMPTO(oparg);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002343 else
2344 break;
2345 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002346
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002347 PREDICTED_WITH_ARG(POP_JUMP_IF_TRUE);
2348 case POP_JUMP_IF_TRUE:
2349 w = POP();
Raymond Hettingerf606f872003-03-16 03:11:04 +00002350 if (w == Py_False) {
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002351 Py_DECREF(w);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002352 goto fast_next_opcode;
Raymond Hettingerf606f872003-03-16 03:11:04 +00002353 }
Raymond Hettinger21012b82003-02-26 18:11:50 +00002354 if (w == Py_True) {
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002355 Py_DECREF(w);
2356 JUMPTO(oparg);
2357 goto fast_next_opcode;
2358 }
2359 err = PyObject_IsTrue(w);
2360 Py_DECREF(w);
2361 if (err > 0) {
2362 err = 0;
2363 JUMPTO(oparg);
2364 }
2365 else if (err == 0)
2366 ;
2367 else
2368 break;
2369 continue;
2370
2371 case JUMP_IF_FALSE_OR_POP:
2372 w = TOP();
2373 if (w == Py_True) {
2374 STACKADJ(-1);
2375 Py_DECREF(w);
2376 goto fast_next_opcode;
2377 }
2378 if (w == Py_False) {
2379 JUMPTO(oparg);
2380 goto fast_next_opcode;
2381 }
2382 err = PyObject_IsTrue(w);
2383 if (err > 0) {
2384 STACKADJ(-1);
2385 Py_DECREF(w);
2386 err = 0;
2387 }
2388 else if (err == 0)
2389 JUMPTO(oparg);
2390 else
2391 break;
2392 continue;
2393
2394 case JUMP_IF_TRUE_OR_POP:
2395 w = TOP();
2396 if (w == Py_False) {
2397 STACKADJ(-1);
2398 Py_DECREF(w);
2399 goto fast_next_opcode;
2400 }
2401 if (w == Py_True) {
2402 JUMPTO(oparg);
Neil Schemenauerc4b570f2003-06-01 19:21:12 +00002403 goto fast_next_opcode;
Raymond Hettinger21012b82003-02-26 18:11:50 +00002404 }
2405 err = PyObject_IsTrue(w);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002406 if (err > 0) {
2407 err = 0;
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002408 JUMPTO(oparg);
Guido van Rossum04691fc1992-08-12 15:35:34 +00002409 }
Jeffrey Yasskin68d68522009-02-28 19:03:21 +00002410 else if (err == 0) {
2411 STACKADJ(-1);
2412 Py_DECREF(w);
2413 }
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002414 else
2415 break;
2416 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002417
Raymond Hettingerfba1cfc2004-03-12 16:33:17 +00002418 PREDICTED_WITH_ARG(JUMP_ABSOLUTE);
Guido van Rossum374a9221991-04-04 10:40:29 +00002419 case JUMP_ABSOLUTE:
2420 JUMPTO(oparg);
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002421#if FAST_LOOPS
2422 /* Enabling this path speeds-up all while and for-loops by bypassing
2423 the per-loop checks for signals. By default, this should be turned-off
2424 because it prevents detection of a control-break in tight loops like
2425 "while 1: pass". Compile with this option turned-on when you need
2426 the speed-up and do not need break checking inside tight loops (ones
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00002427 that contain only instructions ending with goto fast_next_opcode).
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002428 */
2429 goto fast_next_opcode;
2430#else
Neil Schemenauerca2a2f12003-05-30 23:59:44 +00002431 continue;
Raymond Hettingerdc1d1ba2007-11-07 02:45:46 +00002432#endif
Guido van Rossumac7be682001-01-17 15:42:30 +00002433
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002434 case GET_ITER:
2435 /* before: [obj]; after [getiter(obj)] */
Raymond Hettinger663004b2003-01-09 15:24:30 +00002436 v = TOP();
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002437 x = PyObject_GetIter(v);
2438 Py_DECREF(v);
2439 if (x != NULL) {
Raymond Hettinger663004b2003-01-09 15:24:30 +00002440 SET_TOP(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002441 PREDICT(FOR_ITER);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002442 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002443 }
Raymond Hettinger8bb90a52003-01-14 12:43:10 +00002444 STACKADJ(-1);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002445 break;
2446
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002447 PREDICTED_WITH_ARG(FOR_ITER);
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002448 case FOR_ITER:
2449 /* before: [iter]; after: [iter, iter()] *or* [] */
2450 v = TOP();
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002451 x = (*v->ob_type->tp_iternext)(v);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002452 if (x != NULL) {
2453 PUSH(x);
Raymond Hettinger7dc52212003-03-16 20:14:44 +00002454 PREDICT(STORE_FAST);
2455 PREDICT(UNPACK_SEQUENCE);
Guido van Rossum213c7a62001-04-23 14:08:49 +00002456 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002457 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002458 if (PyErr_Occurred()) {
Thomas Wouterse2176022007-09-20 17:35:10 +00002459 if (!PyErr_ExceptionMatches(
2460 PyExc_StopIteration))
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002461 break;
2462 PyErr_Clear();
Guido van Rossum213c7a62001-04-23 14:08:49 +00002463 }
Raymond Hettingerdb0de9e2004-03-12 08:41:36 +00002464 /* iterator ended normally */
2465 x = v = POP();
2466 Py_DECREF(v);
2467 JUMPBY(oparg);
2468 continue;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002469
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002470 case BREAK_LOOP:
2471 why = WHY_BREAK;
2472 goto fast_block_end;
2473
2474 case CONTINUE_LOOP:
2475 retval = PyInt_FromLong(oparg);
Neal Norwitz02104df2006-05-19 06:31:23 +00002476 if (!retval) {
2477 x = NULL;
2478 break;
2479 }
Raymond Hettinger2d783e92004-03-12 09:12:22 +00002480 why = WHY_CONTINUE;
2481 goto fast_block_end;
2482
Guido van Rossum374a9221991-04-04 10:40:29 +00002483 case SETUP_LOOP:
2484 case SETUP_EXCEPT:
2485 case SETUP_FINALLY:
Thomas Wouterse2176022007-09-20 17:35:10 +00002486 /* NOTE: If you add any new block-setup opcodes that
2487 are not try/except/finally handlers, you may need
2488 to update the PyGen_NeedsFinalizing() function.
2489 */
Phillip J. Eby2ba96612006-04-10 17:51:05 +00002490
Guido van Rossumb209a111997-04-29 18:18:01 +00002491 PyFrame_BlockSetup(f, opcode, INSTR_OFFSET() + oparg,
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002492 STACK_LEVEL());
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002493 continue;
Guido van Rossumac7be682001-01-17 15:42:30 +00002494
Benjamin Peterson1880d8b2009-05-25 13:13:44 +00002495 case SETUP_WITH:
2496 {
2497 static PyObject *exit, *enter;
2498 w = TOP();
2499 x = special_lookup(w, "__exit__", &exit);
2500 if (!x)
2501 break;
2502 SET_TOP(x);
2503 u = special_lookup(w, "__enter__", &enter);
2504 Py_DECREF(w);
2505 if (!u) {
2506 x = NULL;
2507 break;
2508 }
2509 x = PyObject_CallFunctionObjArgs(u, NULL);
2510 Py_DECREF(u);
2511 if (!x)
2512 break;
2513 /* Setup the finally block before pushing the result
2514 of __enter__ on the stack. */
2515 PyFrame_BlockSetup(f, SETUP_FINALLY, INSTR_OFFSET() + oparg,
2516 STACK_LEVEL());
2517
2518 PUSH(x);
2519 continue;
2520 }
2521
Guido van Rossumc2e20742006-02-27 22:32:47 +00002522 case WITH_CLEANUP:
2523 {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002524 /* At the top of the stack are 1-3 values indicating
2525 how/why we entered the finally clause:
2526 - TOP = None
2527 - (TOP, SECOND) = (WHY_{RETURN,CONTINUE}), retval
2528 - TOP = WHY_*; no retval below it
2529 - (TOP, SECOND, THIRD) = exc_info()
2530 Below them is EXIT, the context.__exit__ bound method.
Guido van Rossumc2e20742006-02-27 22:32:47 +00002531 In the last case, we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002532 EXIT(TOP, SECOND, THIRD)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002533 otherwise we must call
Nick Coghlan7af53be2008-03-07 14:13:28 +00002534 EXIT(None, None, None)
2535
2536 In all cases, we remove EXIT from the stack, leaving
2537 the rest in the same order.
Guido van Rossum1a5e21e2006-02-28 21:57:43 +00002538
2539 In addition, if the stack represents an exception,
Guido van Rossumf6694362006-03-10 02:28:35 +00002540 *and* the function call returns a 'true' value, we
2541 "zap" this information, to prevent END_FINALLY from
2542 re-raising the exception. (But non-local gotos
2543 should still be resumed.)
Guido van Rossumc2e20742006-02-27 22:32:47 +00002544 */
Tim Peters7df5e7f2006-05-26 23:14:37 +00002545
Nick Coghlan7af53be2008-03-07 14:13:28 +00002546 PyObject *exit_func;
2547
2548 u = POP();
2549 if (u == Py_None) {
2550 exit_func = TOP();
2551 SET_TOP(u);
2552 v = w = Py_None;
2553 }
2554 else if (PyInt_Check(u)) {
2555 switch(PyInt_AS_LONG(u)) {
2556 case WHY_RETURN:
2557 case WHY_CONTINUE:
2558 /* Retval in TOP. */
2559 exit_func = SECOND();
2560 SET_SECOND(TOP());
2561 SET_TOP(u);
2562 break;
2563 default:
2564 exit_func = TOP();
2565 SET_TOP(u);
2566 break;
2567 }
Guido van Rossumc2e20742006-02-27 22:32:47 +00002568 u = v = w = Py_None;
2569 }
2570 else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002571 v = TOP();
2572 w = SECOND();
2573 exit_func = THIRD();
2574 SET_TOP(u);
2575 SET_SECOND(v);
2576 SET_THIRD(w);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002577 }
Guido van Rossumf6694362006-03-10 02:28:35 +00002578 /* XXX Not the fastest way to call it... */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002579 x = PyObject_CallFunctionObjArgs(exit_func, u, v, w,
2580 NULL);
Amaury Forgeot d'Arcad9b5992008-12-10 23:22:49 +00002581 Py_DECREF(exit_func);
2582 if (x == NULL)
Guido van Rossumf6694362006-03-10 02:28:35 +00002583 break; /* Go to error exit */
Amaury Forgeot d'Arcad9b5992008-12-10 23:22:49 +00002584
2585 if (u != Py_None)
2586 err = PyObject_IsTrue(x);
2587 else
2588 err = 0;
2589 Py_DECREF(x);
2590
2591 if (err < 0)
2592 break; /* Go to error exit */
2593 else if (err > 0) {
2594 err = 0;
Guido van Rossumf6694362006-03-10 02:28:35 +00002595 /* There was an exception and a true return */
Nick Coghlan7af53be2008-03-07 14:13:28 +00002596 STACKADJ(-2);
Guido van Rossumf6694362006-03-10 02:28:35 +00002597 Py_INCREF(Py_None);
2598 SET_TOP(Py_None);
Guido van Rossumf6694362006-03-10 02:28:35 +00002599 Py_DECREF(u);
2600 Py_DECREF(v);
2601 Py_DECREF(w);
2602 } else {
Nick Coghlan7af53be2008-03-07 14:13:28 +00002603 /* The stack was rearranged to remove EXIT
2604 above. Let END_FINALLY do its thing */
Guido van Rossumf6694362006-03-10 02:28:35 +00002605 }
Jeffrey Yasskin9063a992008-03-03 01:27:03 +00002606 PREDICT(END_FINALLY);
Guido van Rossumc2e20742006-02-27 22:32:47 +00002607 break;
2608 }
2609
Guido van Rossumf10570b1995-07-07 22:53:21 +00002610 case CALL_FUNCTION:
Armin Rigo8817fcd2004-06-17 10:22:40 +00002611 {
2612 PyObject **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002613 PCALL(PCALL_ALL);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002614 sp = stack_pointer;
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002615#ifdef WITH_TSC
Armin Rigo8817fcd2004-06-17 10:22:40 +00002616 x = call_function(&sp, oparg, &intr0, &intr1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002617#else
Armin Rigo8817fcd2004-06-17 10:22:40 +00002618 x = call_function(&sp, oparg);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002619#endif
Armin Rigo8817fcd2004-06-17 10:22:40 +00002620 stack_pointer = sp;
Jeremy Hyltone8c04322002-08-16 17:47:26 +00002621 PUSH(x);
2622 if (x != NULL)
2623 continue;
2624 break;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002625 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002626
Jeremy Hylton76901512000-03-28 23:49:17 +00002627 case CALL_FUNCTION_VAR:
2628 case CALL_FUNCTION_KW:
2629 case CALL_FUNCTION_VAR_KW:
Guido van Rossumf10570b1995-07-07 22:53:21 +00002630 {
Jeremy Hylton76901512000-03-28 23:49:17 +00002631 int na = oparg & 0xff;
2632 int nk = (oparg>>8) & 0xff;
2633 int flags = (opcode - CALL_FUNCTION) & 3;
Jeremy Hylton52820442001-01-03 23:52:36 +00002634 int n = na + 2 * nk;
Armin Rigo8817fcd2004-06-17 10:22:40 +00002635 PyObject **pfunc, *func, **sp;
Jeremy Hylton985eba52003-02-05 23:13:00 +00002636 PCALL(PCALL_ALL);
Jeremy Hylton52820442001-01-03 23:52:36 +00002637 if (flags & CALL_FLAG_VAR)
2638 n++;
2639 if (flags & CALL_FLAG_KW)
2640 n++;
2641 pfunc = stack_pointer - n - 1;
2642 func = *pfunc;
Jeremy Hylton52820442001-01-03 23:52:36 +00002643
Guido van Rossumac7be682001-01-17 15:42:30 +00002644 if (PyMethod_Check(func)
Jeremy Hylton52820442001-01-03 23:52:36 +00002645 && PyMethod_GET_SELF(func) != NULL) {
2646 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002647 Py_INCREF(self);
Jeremy Hylton52820442001-01-03 23:52:36 +00002648 func = PyMethod_GET_FUNCTION(func);
2649 Py_INCREF(func);
Jeremy Hylton76901512000-03-28 23:49:17 +00002650 Py_DECREF(*pfunc);
2651 *pfunc = self;
2652 na++;
2653 n++;
Guido van Rossumac7be682001-01-17 15:42:30 +00002654 } else
Jeremy Hylton52820442001-01-03 23:52:36 +00002655 Py_INCREF(func);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002656 sp = stack_pointer;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002657 READ_TIMESTAMP(intr0);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002658 x = ext_do_call(func, &sp, flags, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002659 READ_TIMESTAMP(intr1);
Armin Rigo8817fcd2004-06-17 10:22:40 +00002660 stack_pointer = sp;
Jeremy Hylton76901512000-03-28 23:49:17 +00002661 Py_DECREF(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00002662
Jeremy Hylton76901512000-03-28 23:49:17 +00002663 while (stack_pointer > pfunc) {
Jeremy Hylton52820442001-01-03 23:52:36 +00002664 w = POP();
2665 Py_DECREF(w);
Jeremy Hylton76901512000-03-28 23:49:17 +00002666 }
2667 PUSH(x);
Guido van Rossumac7be682001-01-17 15:42:30 +00002668 if (x != NULL)
Jeremy Hylton52820442001-01-03 23:52:36 +00002669 continue;
Jeremy Hylton76901512000-03-28 23:49:17 +00002670 break;
Guido van Rossumf10570b1995-07-07 22:53:21 +00002671 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002672
Guido van Rossum681d79a1995-07-18 14:51:37 +00002673 case MAKE_FUNCTION:
2674 v = POP(); /* code object */
Guido van Rossumb209a111997-04-29 18:18:01 +00002675 x = PyFunction_New(v, f->f_globals);
2676 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002677 /* XXX Maybe this should be a separate opcode? */
2678 if (x != NULL && oparg > 0) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002679 v = PyTuple_New(oparg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002680 if (v == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002681 Py_DECREF(x);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002682 x = NULL;
2683 break;
2684 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002685 while (--oparg >= 0) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002686 w = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002687 PyTuple_SET_ITEM(v, oparg, w);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002688 }
2689 err = PyFunction_SetDefaults(x, v);
Guido van Rossumb209a111997-04-29 18:18:01 +00002690 Py_DECREF(v);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002691 }
2692 PUSH(x);
2693 break;
Guido van Rossum8861b741996-07-30 16:49:37 +00002694
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002695 case MAKE_CLOSURE:
2696 {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002697 v = POP(); /* code object */
2698 x = PyFunction_New(v, f->f_globals);
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002699 Py_DECREF(v);
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00002700 if (x != NULL) {
2701 v = POP();
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002702 if (PyFunction_SetClosure(x, v) != 0) {
2703 /* Can't happen unless bytecode is corrupt. */
2704 why = WHY_EXCEPTION;
2705 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002706 Py_DECREF(v);
2707 }
2708 if (x != NULL && oparg > 0) {
2709 v = PyTuple_New(oparg);
2710 if (v == NULL) {
2711 Py_DECREF(x);
2712 x = NULL;
2713 break;
2714 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002715 while (--oparg >= 0) {
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002716 w = POP();
2717 PyTuple_SET_ITEM(v, oparg, w);
2718 }
Jeffrey Yasskin2d873bd2008-12-08 18:55:24 +00002719 if (PyFunction_SetDefaults(x, v) != 0) {
2720 /* Can't happen unless
2721 PyFunction_SetDefaults changes. */
2722 why = WHY_EXCEPTION;
2723 }
Jeremy Hylton64949cb2001-01-25 20:06:59 +00002724 Py_DECREF(v);
2725 }
2726 PUSH(x);
2727 break;
2728 }
2729
Guido van Rossum8861b741996-07-30 16:49:37 +00002730 case BUILD_SLICE:
2731 if (oparg == 3)
2732 w = POP();
2733 else
2734 w = NULL;
2735 v = POP();
Raymond Hettinger663004b2003-01-09 15:24:30 +00002736 u = TOP();
Guido van Rossum1aa14831997-01-21 05:34:20 +00002737 x = PySlice_New(u, v, w);
Guido van Rossumb209a111997-04-29 18:18:01 +00002738 Py_DECREF(u);
2739 Py_DECREF(v);
2740 Py_XDECREF(w);
Raymond Hettinger663004b2003-01-09 15:24:30 +00002741 SET_TOP(x);
Guido van Rossum3dfd53b1997-01-18 02:46:13 +00002742 if (x != NULL) continue;
Guido van Rossum8861b741996-07-30 16:49:37 +00002743 break;
2744
Fred Drakeef8ace32000-08-24 00:32:09 +00002745 case EXTENDED_ARG:
2746 opcode = NEXTOP();
Raymond Hettinger5bed4562004-04-10 23:34:17 +00002747 oparg = oparg<<16 | NEXTARG();
Fred Drakeef8ace32000-08-24 00:32:09 +00002748 goto dispatch_opcode;
Guido van Rossum8861b741996-07-30 16:49:37 +00002749
Guido van Rossum374a9221991-04-04 10:40:29 +00002750 default:
2751 fprintf(stderr,
2752 "XXX lineno: %d, opcode: %d\n",
Jeffrey Yasskinf7f858d2009-05-08 22:23:21 +00002753 PyFrame_GetLineNumber(f),
Michael W. Hudsondd32a912002-08-15 14:59:02 +00002754 opcode);
Guido van Rossumb209a111997-04-29 18:18:01 +00002755 PyErr_SetString(PyExc_SystemError, "unknown opcode");
Guido van Rossum374a9221991-04-04 10:40:29 +00002756 why = WHY_EXCEPTION;
2757 break;
Guido van Rossum04691fc1992-08-12 15:35:34 +00002758
2759#ifdef CASE_TOO_BIG
2760 }
2761#endif
2762
Guido van Rossum374a9221991-04-04 10:40:29 +00002763 } /* switch */
2764
2765 on_error:
Guido van Rossumac7be682001-01-17 15:42:30 +00002766
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002767 READ_TIMESTAMP(inst1);
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002768
Guido van Rossum374a9221991-04-04 10:40:29 +00002769 /* Quickly continue if no error occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002770
Guido van Rossum374a9221991-04-04 10:40:29 +00002771 if (why == WHY_NOT) {
Guido van Rossum681d79a1995-07-18 14:51:37 +00002772 if (err == 0 && x != NULL) {
2773#ifdef CHECKEXC
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002774 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002775 if (PyErr_Occurred())
Guido van Rossum681d79a1995-07-18 14:51:37 +00002776 fprintf(stderr,
2777 "XXX undetected error\n");
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002778 else {
2779#endif
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002780 READ_TIMESTAMP(loop1);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002781 continue; /* Normal, fast path */
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00002782#ifdef CHECKEXC
2783 }
2784#endif
Guido van Rossum681d79a1995-07-18 14:51:37 +00002785 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002786 why = WHY_EXCEPTION;
Guido van Rossumb209a111997-04-29 18:18:01 +00002787 x = Py_None;
Guido van Rossum374a9221991-04-04 10:40:29 +00002788 err = 0;
2789 }
2790
Guido van Rossum374a9221991-04-04 10:40:29 +00002791 /* Double-check exception status */
Guido van Rossumac7be682001-01-17 15:42:30 +00002792
Raymond Hettingerc8aa08b2004-04-11 14:59:33 +00002793 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002794 if (!PyErr_Occurred()) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00002795 PyErr_SetString(PyExc_SystemError,
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002796 "error return without exception set");
Guido van Rossum374a9221991-04-04 10:40:29 +00002797 why = WHY_EXCEPTION;
2798 }
2799 }
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002800#ifdef CHECKEXC
Guido van Rossum374a9221991-04-04 10:40:29 +00002801 else {
Guido van Rossumeb894eb1999-03-09 16:16:45 +00002802 /* This check is expensive! */
Guido van Rossumb209a111997-04-29 18:18:01 +00002803 if (PyErr_Occurred()) {
Neal Norwitz8250fbe2008-01-27 17:12:15 +00002804 char buf[128];
Jeremy Hylton904ed862003-11-05 17:29:35 +00002805 sprintf(buf, "Stack unwind with exception "
2806 "set and why=%d", why);
2807 Py_FatalError(buf);
Guido van Rossum681d79a1995-07-18 14:51:37 +00002808 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002809 }
2810#endif
2811
2812 /* Log traceback info if this is a real exception */
Guido van Rossumac7be682001-01-17 15:42:30 +00002813
Guido van Rossum374a9221991-04-04 10:40:29 +00002814 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002815 PyTraceBack_Here(f);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002816
Fred Drake8f51f542001-10-04 14:48:42 +00002817 if (tstate->c_tracefunc != NULL)
2818 call_exc_trace(tstate->c_tracefunc,
2819 tstate->c_traceobj, f);
Guido van Rossum014518f1998-11-23 21:09:51 +00002820 }
Guido van Rossumac7be682001-01-17 15:42:30 +00002821
Guido van Rossum374a9221991-04-04 10:40:29 +00002822 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
Guido van Rossumac7be682001-01-17 15:42:30 +00002823
Guido van Rossum374a9221991-04-04 10:40:29 +00002824 if (why == WHY_RERAISE)
2825 why = WHY_EXCEPTION;
2826
2827 /* Unwind stacks if a (pseudo) exception occurred */
Guido van Rossumac7be682001-01-17 15:42:30 +00002828
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002829fast_block_end:
Tim Peters8a5c3c72004-04-05 19:36:21 +00002830 while (why != WHY_NOT && f->f_iblock > 0) {
Benjamin Peterson4a3cf192009-07-01 23:45:19 +00002831 /* Peek at the current block. */
2832 PyTryBlock *b = &f->f_blockstack[f->f_iblock - 1];
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002833
Tim Peters8a5c3c72004-04-05 19:36:21 +00002834 assert(why != WHY_YIELD);
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002835 if (b->b_type == SETUP_LOOP && why == WHY_CONTINUE) {
Jeremy Hylton3faa52e2001-02-01 22:48:12 +00002836 why = WHY_NOT;
2837 JUMPTO(PyInt_AS_LONG(retval));
2838 Py_DECREF(retval);
2839 break;
2840 }
2841
Benjamin Peterson4a3cf192009-07-01 23:45:19 +00002842 /* Now we have to pop the block. */
2843 f->f_iblock--;
2844
Guido van Rossum374a9221991-04-04 10:40:29 +00002845 while (STACK_LEVEL() > b->b_level) {
2846 v = POP();
Guido van Rossumb209a111997-04-29 18:18:01 +00002847 Py_XDECREF(v);
Guido van Rossum374a9221991-04-04 10:40:29 +00002848 }
2849 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
2850 why = WHY_NOT;
2851 JUMPTO(b->b_handler);
2852 break;
2853 }
2854 if (b->b_type == SETUP_FINALLY ||
Guido van Rossum150b2df1996-12-05 23:17:11 +00002855 (b->b_type == SETUP_EXCEPT &&
2856 why == WHY_EXCEPTION)) {
Guido van Rossum374a9221991-04-04 10:40:29 +00002857 if (why == WHY_EXCEPTION) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002858 PyObject *exc, *val, *tb;
2859 PyErr_Fetch(&exc, &val, &tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002860 if (val == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00002861 val = Py_None;
2862 Py_INCREF(val);
Guido van Rossum374a9221991-04-04 10:40:29 +00002863 }
Guido van Rossum374a9221991-04-04 10:40:29 +00002864 /* Make the raw exception data
2865 available to the handler,
2866 so a program can emulate the
2867 Python main loop. Don't do
2868 this for 'finally'. */
2869 if (b->b_type == SETUP_EXCEPT) {
Barry Warsaweaedc7c1997-08-28 22:36:40 +00002870 PyErr_NormalizeException(
2871 &exc, &val, &tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00002872 set_exc_info(tstate,
2873 exc, val, tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002874 }
Jeremy Hyltonc6314892001-09-26 19:24:45 +00002875 if (tb == NULL) {
2876 Py_INCREF(Py_None);
2877 PUSH(Py_None);
2878 } else
2879 PUSH(tb);
Guido van Rossum374a9221991-04-04 10:40:29 +00002880 PUSH(val);
2881 PUSH(exc);
2882 }
2883 else {
Raymond Hettinger06032cb2004-04-06 09:37:35 +00002884 if (why & (WHY_RETURN | WHY_CONTINUE))
Guido van Rossum374a9221991-04-04 10:40:29 +00002885 PUSH(retval);
Guido van Rossumb209a111997-04-29 18:18:01 +00002886 v = PyInt_FromLong((long)why);
Guido van Rossum374a9221991-04-04 10:40:29 +00002887 PUSH(v);
2888 }
2889 why = WHY_NOT;
2890 JUMPTO(b->b_handler);
2891 break;
2892 }
2893 } /* unwind stack */
2894
2895 /* End the loop if we still have an error (or return) */
Guido van Rossumac7be682001-01-17 15:42:30 +00002896
Guido van Rossum374a9221991-04-04 10:40:29 +00002897 if (why != WHY_NOT)
2898 break;
Michael W. Hudson75eabd22005-01-18 15:56:11 +00002899 READ_TIMESTAMP(loop1);
Guido van Rossumac7be682001-01-17 15:42:30 +00002900
Guido van Rossum374a9221991-04-04 10:40:29 +00002901 } /* main loop */
Guido van Rossumac7be682001-01-17 15:42:30 +00002902
Tim Peters8a5c3c72004-04-05 19:36:21 +00002903 assert(why != WHY_YIELD);
2904 /* Pop remaining stack entries. */
2905 while (!EMPTY()) {
2906 v = POP();
2907 Py_XDECREF(v);
Guido van Rossum35974fb2001-12-06 21:28:18 +00002908 }
2909
Tim Peters8a5c3c72004-04-05 19:36:21 +00002910 if (why != WHY_RETURN)
Guido van Rossum96a42c81992-01-12 02:29:51 +00002911 retval = NULL;
Guido van Rossumac7be682001-01-17 15:42:30 +00002912
Raymond Hettinger1dd83092004-02-06 18:32:33 +00002913fast_yield:
Fred Drake9e3ad782001-07-03 23:39:52 +00002914 if (tstate->use_tracing) {
Barry Warsawe2eca0b2005-08-15 18:14:19 +00002915 if (tstate->c_tracefunc) {
2916 if (why == WHY_RETURN || why == WHY_YIELD) {
2917 if (call_trace(tstate->c_tracefunc,
2918 tstate->c_traceobj, f,
2919 PyTrace_RETURN, retval)) {
2920 Py_XDECREF(retval);
2921 retval = NULL;
2922 why = WHY_EXCEPTION;
2923 }
2924 }
2925 else if (why == WHY_EXCEPTION) {
2926 call_trace_protected(tstate->c_tracefunc,
2927 tstate->c_traceobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002928 PyTrace_RETURN, NULL);
Guido van Rossum96a42c81992-01-12 02:29:51 +00002929 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002930 }
Fred Drake8f51f542001-10-04 14:48:42 +00002931 if (tstate->c_profilefunc) {
Fred Drake4ec5d562001-10-04 19:26:43 +00002932 if (why == WHY_EXCEPTION)
2933 call_trace_protected(tstate->c_profilefunc,
2934 tstate->c_profileobj, f,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00002935 PyTrace_RETURN, NULL);
Fred Drake4ec5d562001-10-04 19:26:43 +00002936 else if (call_trace(tstate->c_profilefunc,
2937 tstate->c_profileobj, f,
2938 PyTrace_RETURN, retval)) {
Fred Drake9e3ad782001-07-03 23:39:52 +00002939 Py_XDECREF(retval);
2940 retval = NULL;
2941 why = WHY_EXCEPTION;
2942 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00002943 }
Guido van Rossum96a42c81992-01-12 02:29:51 +00002944 }
Guido van Rossuma4240131997-01-21 21:18:36 +00002945
Tim Peters7df5e7f2006-05-26 23:14:37 +00002946 if (tstate->frame->f_exc_type != NULL)
2947 reset_exc_info(tstate);
2948 else {
2949 assert(tstate->frame->f_exc_value == NULL);
2950 assert(tstate->frame->f_exc_traceback == NULL);
2951 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00002952
Tim Peters5ca576e2001-06-18 22:08:13 +00002953 /* pop frame */
Thomas Woutersae406c62007-09-19 17:27:43 +00002954exit_eval_frame:
Armin Rigo2b3eb402003-10-28 12:05:48 +00002955 Py_LeaveRecursiveCall();
Guido van Rossuma027efa1997-05-05 20:56:21 +00002956 tstate->frame = f->f_back;
Guido van Rossumac7be682001-01-17 15:42:30 +00002957
Guido van Rossum96a42c81992-01-12 02:29:51 +00002958 return retval;
Guido van Rossum374a9221991-04-04 10:40:29 +00002959}
2960
Guido van Rossumc2e20742006-02-27 22:32:47 +00002961/* This is gonna seem *real weird*, but if you put some other code between
Martin v. Löwis8d97e332004-06-27 15:43:12 +00002962 PyEval_EvalFrame() and PyEval_EvalCodeEx() you will need to adjust
Guido van Rossumc2e20742006-02-27 22:32:47 +00002963 the test in the if statements in Misc/gdbinit (pystack and pystackv). */
Skip Montanaro786ea6b2004-03-01 15:44:05 +00002964
Tim Peters6d6c1a32001-08-02 04:15:00 +00002965PyObject *
2966PyEval_EvalCodeEx(PyCodeObject *co, PyObject *globals, PyObject *locals,
Tim Peters5ca576e2001-06-18 22:08:13 +00002967 PyObject **args, int argcount, PyObject **kws, int kwcount,
2968 PyObject **defs, int defcount, PyObject *closure)
2969{
2970 register PyFrameObject *f;
2971 register PyObject *retval = NULL;
2972 register PyObject **fastlocals, **freevars;
2973 PyThreadState *tstate = PyThreadState_GET();
2974 PyObject *x, *u;
2975
2976 if (globals == NULL) {
Tim Peters8a5c3c72004-04-05 19:36:21 +00002977 PyErr_SetString(PyExc_SystemError,
Jeremy Hylton910d7d42001-08-12 21:52:24 +00002978 "PyEval_EvalCodeEx: NULL globals");
Tim Peters5ca576e2001-06-18 22:08:13 +00002979 return NULL;
2980 }
2981
Neal Norwitzdf6a6492006-08-13 18:10:10 +00002982 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00002983 assert(globals != NULL);
2984 f = PyFrame_New(tstate, co, globals, locals);
Tim Peters5ca576e2001-06-18 22:08:13 +00002985 if (f == NULL)
2986 return NULL;
2987
2988 fastlocals = f->f_localsplus;
Richard Jonescebbefc2006-05-23 18:28:17 +00002989 freevars = f->f_localsplus + co->co_nlocals;
Tim Peters5ca576e2001-06-18 22:08:13 +00002990
2991 if (co->co_argcount > 0 ||
2992 co->co_flags & (CO_VARARGS | CO_VARKEYWORDS)) {
2993 int i;
2994 int n = argcount;
2995 PyObject *kwdict = NULL;
2996 if (co->co_flags & CO_VARKEYWORDS) {
2997 kwdict = PyDict_New();
2998 if (kwdict == NULL)
2999 goto fail;
3000 i = co->co_argcount;
3001 if (co->co_flags & CO_VARARGS)
3002 i++;
3003 SETLOCAL(i, kwdict);
3004 }
3005 if (argcount > co->co_argcount) {
3006 if (!(co->co_flags & CO_VARARGS)) {
3007 PyErr_Format(PyExc_TypeError,
3008 "%.200s() takes %s %d "
3009 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003010 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00003011 defcount ? "at most" : "exactly",
3012 co->co_argcount,
3013 kwcount ? "non-keyword " : "",
3014 co->co_argcount == 1 ? "" : "s",
3015 argcount);
3016 goto fail;
3017 }
3018 n = co->co_argcount;
3019 }
3020 for (i = 0; i < n; i++) {
3021 x = args[i];
3022 Py_INCREF(x);
3023 SETLOCAL(i, x);
3024 }
3025 if (co->co_flags & CO_VARARGS) {
3026 u = PyTuple_New(argcount - n);
3027 if (u == NULL)
3028 goto fail;
3029 SETLOCAL(co->co_argcount, u);
3030 for (i = n; i < argcount; i++) {
3031 x = args[i];
3032 Py_INCREF(x);
3033 PyTuple_SET_ITEM(u, i-n, x);
3034 }
3035 }
3036 for (i = 0; i < kwcount; i++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003037 PyObject **co_varnames;
Tim Peters5ca576e2001-06-18 22:08:13 +00003038 PyObject *keyword = kws[2*i];
3039 PyObject *value = kws[2*i + 1];
3040 int j;
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003041 if (keyword == NULL || !(PyString_Check(keyword)
3042#ifdef Py_USING_UNICODE
3043 || PyUnicode_Check(keyword)
3044#endif
3045 )) {
Tim Peters5ca576e2001-06-18 22:08:13 +00003046 PyErr_Format(PyExc_TypeError,
3047 "%.200s() keywords must be strings",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003048 PyString_AsString(co->co_name));
Tim Peters5ca576e2001-06-18 22:08:13 +00003049 goto fail;
3050 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003051 /* Speed hack: do raw pointer compares. As names are
3052 normally interned this should almost always hit. */
3053 co_varnames = PySequence_Fast_ITEMS(co->co_varnames);
Tim Peters5ca576e2001-06-18 22:08:13 +00003054 for (j = 0; j < co->co_argcount; j++) {
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003055 PyObject *nm = co_varnames[j];
3056 if (nm == keyword)
3057 goto kw_found;
3058 }
3059 /* Slow fallback, just in case */
3060 for (j = 0; j < co->co_argcount; j++) {
3061 PyObject *nm = co_varnames[j];
Tim Peters5ca576e2001-06-18 22:08:13 +00003062 int cmp = PyObject_RichCompareBool(
3063 keyword, nm, Py_EQ);
3064 if (cmp > 0)
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003065 goto kw_found;
Tim Peters5ca576e2001-06-18 22:08:13 +00003066 else if (cmp < 0)
3067 goto fail;
3068 }
3069 /* Check errors from Compare */
3070 if (PyErr_Occurred())
3071 goto fail;
3072 if (j >= co->co_argcount) {
3073 if (kwdict == NULL) {
Benjamin Petersone18ef192009-01-20 14:21:16 +00003074 PyObject *kwd_str = kwd_as_string(keyword);
3075 if (kwd_str) {
3076 PyErr_Format(PyExc_TypeError,
3077 "%.200s() got an unexpected "
3078 "keyword argument '%.400s'",
3079 PyString_AsString(co->co_name),
3080 PyString_AsString(kwd_str));
3081 Py_DECREF(kwd_str);
3082 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003083 goto fail;
3084 }
3085 PyDict_SetItem(kwdict, keyword, value);
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003086 continue;
Tim Peters5ca576e2001-06-18 22:08:13 +00003087 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003088kw_found:
3089 if (GETLOCAL(j) != NULL) {
Benjamin Petersone18ef192009-01-20 14:21:16 +00003090 PyObject *kwd_str = kwd_as_string(keyword);
3091 if (kwd_str) {
3092 PyErr_Format(PyExc_TypeError,
3093 "%.200s() got multiple "
3094 "values for keyword "
3095 "argument '%.400s'",
3096 PyString_AsString(co->co_name),
3097 PyString_AsString(kwd_str));
3098 Py_DECREF(kwd_str);
3099 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003100 goto fail;
Tim Peters5ca576e2001-06-18 22:08:13 +00003101 }
Antoine Pitrouc2cc80c2008-07-25 22:13:52 +00003102 Py_INCREF(value);
3103 SETLOCAL(j, value);
Tim Peters5ca576e2001-06-18 22:08:13 +00003104 }
3105 if (argcount < co->co_argcount) {
3106 int m = co->co_argcount - defcount;
3107 for (i = argcount; i < m; i++) {
3108 if (GETLOCAL(i) == NULL) {
3109 PyErr_Format(PyExc_TypeError,
3110 "%.200s() takes %s %d "
3111 "%sargument%s (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003112 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00003113 ((co->co_flags & CO_VARARGS) ||
3114 defcount) ? "at least"
3115 : "exactly",
3116 m, kwcount ? "non-keyword " : "",
3117 m == 1 ? "" : "s", i);
3118 goto fail;
3119 }
3120 }
3121 if (n > m)
3122 i = n - m;
3123 else
3124 i = 0;
3125 for (; i < defcount; i++) {
3126 if (GETLOCAL(m+i) == NULL) {
3127 PyObject *def = defs[i];
3128 Py_INCREF(def);
3129 SETLOCAL(m+i, def);
3130 }
3131 }
3132 }
3133 }
3134 else {
3135 if (argcount > 0 || kwcount > 0) {
3136 PyErr_Format(PyExc_TypeError,
3137 "%.200s() takes no arguments (%d given)",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003138 PyString_AsString(co->co_name),
Tim Peters5ca576e2001-06-18 22:08:13 +00003139 argcount + kwcount);
3140 goto fail;
3141 }
3142 }
3143 /* Allocate and initialize storage for cell vars, and copy free
3144 vars into frame. This isn't too efficient right now. */
Richard Jonescebbefc2006-05-23 18:28:17 +00003145 if (PyTuple_GET_SIZE(co->co_cellvars)) {
Neal Norwitz245ce8d2006-06-12 02:16:10 +00003146 int i, j, nargs, found;
Tim Peters5ca576e2001-06-18 22:08:13 +00003147 char *cellname, *argname;
3148 PyObject *c;
3149
3150 nargs = co->co_argcount;
3151 if (co->co_flags & CO_VARARGS)
3152 nargs++;
3153 if (co->co_flags & CO_VARKEYWORDS)
3154 nargs++;
3155
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003156 /* Initialize each cell var, taking into account
3157 cell vars that are initialized from arguments.
3158
3159 Should arrange for the compiler to put cellvars
3160 that are arguments at the beginning of the cellvars
3161 list so that we can march over it more efficiently?
3162 */
Richard Jonescebbefc2006-05-23 18:28:17 +00003163 for (i = 0; i < PyTuple_GET_SIZE(co->co_cellvars); ++i) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003164 cellname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00003165 PyTuple_GET_ITEM(co->co_cellvars, i));
3166 found = 0;
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00003167 for (j = 0; j < nargs; j++) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003168 argname = PyString_AS_STRING(
Tim Peters5ca576e2001-06-18 22:08:13 +00003169 PyTuple_GET_ITEM(co->co_varnames, j));
3170 if (strcmp(cellname, argname) == 0) {
3171 c = PyCell_New(GETLOCAL(j));
3172 if (c == NULL)
3173 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00003174 GETLOCAL(co->co_nlocals + i) = c;
Tim Peters5ca576e2001-06-18 22:08:13 +00003175 found = 1;
3176 break;
3177 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003178 }
3179 if (found == 0) {
3180 c = PyCell_New(NULL);
3181 if (c == NULL)
3182 goto fail;
Richard Jonescebbefc2006-05-23 18:28:17 +00003183 SETLOCAL(co->co_nlocals + i, c);
Tim Peters5ca576e2001-06-18 22:08:13 +00003184 }
3185 }
Tim Peters5ca576e2001-06-18 22:08:13 +00003186 }
Richard Jonescebbefc2006-05-23 18:28:17 +00003187 if (PyTuple_GET_SIZE(co->co_freevars)) {
Tim Peters5ca576e2001-06-18 22:08:13 +00003188 int i;
Richard Jonescebbefc2006-05-23 18:28:17 +00003189 for (i = 0; i < PyTuple_GET_SIZE(co->co_freevars); ++i) {
Tim Peters5ca576e2001-06-18 22:08:13 +00003190 PyObject *o = PyTuple_GET_ITEM(closure, i);
3191 Py_INCREF(o);
Richard Jonescebbefc2006-05-23 18:28:17 +00003192 freevars[PyTuple_GET_SIZE(co->co_cellvars) + i] = o;
Tim Peters5ca576e2001-06-18 22:08:13 +00003193 }
3194 }
3195
Tim Peters5ca576e2001-06-18 22:08:13 +00003196 if (co->co_flags & CO_GENERATOR) {
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003197 /* Don't need to keep the reference to f_back, it will be set
3198 * when the generator is resumed. */
Tim Peters5ba58662001-07-16 02:29:45 +00003199 Py_XDECREF(f->f_back);
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003200 f->f_back = NULL;
3201
Jeremy Hylton985eba52003-02-05 23:13:00 +00003202 PCALL(PCALL_GENERATOR);
3203
Neil Schemenauer2b13ce82001-06-21 02:41:10 +00003204 /* Create a new generator that owns the ready to run frame
3205 * and return that as the value. */
Martin v. Löwise440e472004-06-01 15:22:42 +00003206 return PyGen_New(f);
Tim Peters5ca576e2001-06-18 22:08:13 +00003207 }
3208
Thomas Woutersae406c62007-09-19 17:27:43 +00003209 retval = PyEval_EvalFrameEx(f,0);
Tim Peters5ca576e2001-06-18 22:08:13 +00003210
Thomas Woutersae406c62007-09-19 17:27:43 +00003211fail: /* Jump here from prelude on failure */
Tim Peters5ca576e2001-06-18 22:08:13 +00003212
Tim Petersb13680b2001-11-27 23:29:29 +00003213 /* decref'ing the frame can cause __del__ methods to get invoked,
3214 which can call back into Python. While we're done with the
3215 current Python frame (f), the associated C stack is still in use,
3216 so recursion_depth must be boosted for the duration.
3217 */
3218 assert(tstate != NULL);
3219 ++tstate->recursion_depth;
Thomas Woutersae406c62007-09-19 17:27:43 +00003220 Py_DECREF(f);
Tim Petersb13680b2001-11-27 23:29:29 +00003221 --tstate->recursion_depth;
Tim Peters5ca576e2001-06-18 22:08:13 +00003222 return retval;
3223}
3224
3225
Benjamin Peterson1880d8b2009-05-25 13:13:44 +00003226static PyObject *
3227special_lookup(PyObject *o, char *meth, PyObject **cache)
3228{
3229 PyObject *res;
3230 if (PyInstance_Check(o)) {
3231 if (!*cache)
3232 return PyObject_GetAttrString(o, meth);
3233 else
3234 return PyObject_GetAttr(o, *cache);
3235 }
3236 res = _PyObject_LookupSpecial(o, meth, cache);
3237 if (res == NULL && !PyErr_Occurred()) {
3238 PyErr_SetObject(PyExc_AttributeError, *cache);
3239 return NULL;
3240 }
3241 return res;
3242}
3243
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003244
Benjamin Petersone18ef192009-01-20 14:21:16 +00003245static PyObject *
3246kwd_as_string(PyObject *kwd) {
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003247#ifdef Py_USING_UNICODE
Benjamin Petersone18ef192009-01-20 14:21:16 +00003248 if (PyString_Check(kwd)) {
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003249#else
3250 assert(PyString_Check(kwd));
3251#endif
Benjamin Petersone18ef192009-01-20 14:21:16 +00003252 Py_INCREF(kwd);
3253 return kwd;
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003254#ifdef Py_USING_UNICODE
Benjamin Petersone18ef192009-01-20 14:21:16 +00003255 }
Benjamin Peterson78821dd2009-01-25 17:15:10 +00003256 return _PyUnicode_AsDefaultEncodedString(kwd, "replace");
3257#endif
Benjamin Petersone18ef192009-01-20 14:21:16 +00003258}
3259
3260
Guido van Rossumc9fbb722003-03-01 03:36:33 +00003261/* Implementation notes for set_exc_info() and reset_exc_info():
3262
3263- Below, 'exc_ZZZ' stands for 'exc_type', 'exc_value' and
3264 'exc_traceback'. These always travel together.
3265
3266- tstate->curexc_ZZZ is the "hot" exception that is set by
3267 PyErr_SetString(), cleared by PyErr_Clear(), and so on.
3268
3269- Once an exception is caught by an except clause, it is transferred
3270 from tstate->curexc_ZZZ to tstate->exc_ZZZ, from which sys.exc_info()
3271 can pick it up. This is the primary task of set_exc_info().
Tim Peters7df5e7f2006-05-26 23:14:37 +00003272 XXX That can't be right: set_exc_info() doesn't look at tstate->curexc_ZZZ.
Guido van Rossumc9fbb722003-03-01 03:36:33 +00003273
3274- Now let me explain the complicated dance with frame->f_exc_ZZZ.
3275
3276 Long ago, when none of this existed, there were just a few globals:
3277 one set corresponding to the "hot" exception, and one set
3278 corresponding to sys.exc_ZZZ. (Actually, the latter weren't C
3279 globals; they were simply stored as sys.exc_ZZZ. For backwards
3280 compatibility, they still are!) The problem was that in code like
3281 this:
3282
3283 try:
3284 "something that may fail"
3285 except "some exception":
3286 "do something else first"
3287 "print the exception from sys.exc_ZZZ."
3288
3289 if "do something else first" invoked something that raised and caught
3290 an exception, sys.exc_ZZZ were overwritten. That was a frequent
3291 cause of subtle bugs. I fixed this by changing the semantics as
3292 follows:
3293
3294 - Within one frame, sys.exc_ZZZ will hold the last exception caught
3295 *in that frame*.
3296
3297 - But initially, and as long as no exception is caught in a given
3298 frame, sys.exc_ZZZ will hold the last exception caught in the
3299 previous frame (or the frame before that, etc.).
3300
3301 The first bullet fixed the bug in the above example. The second
3302 bullet was for backwards compatibility: it was (and is) common to
3303 have a function that is called when an exception is caught, and to
3304 have that function access the caught exception via sys.exc_ZZZ.
3305 (Example: traceback.print_exc()).
3306
3307 At the same time I fixed the problem that sys.exc_ZZZ weren't
3308 thread-safe, by introducing sys.exc_info() which gets it from tstate;
3309 but that's really a separate improvement.
3310
3311 The reset_exc_info() function in ceval.c restores the tstate->exc_ZZZ
3312 variables to what they were before the current frame was called. The
3313 set_exc_info() function saves them on the frame so that
3314 reset_exc_info() can restore them. The invariant is that
3315 frame->f_exc_ZZZ is NULL iff the current frame never caught an
3316 exception (where "catching" an exception applies only to successful
3317 except clauses); and if the current frame ever caught an exception,
3318 frame->f_exc_ZZZ is the exception that was stored in tstate->exc_ZZZ
3319 at the start of the current frame.
3320
3321*/
3322
Fredrik Lundh7a830892006-05-27 10:39:48 +00003323static void
Guido van Rossumac7be682001-01-17 15:42:30 +00003324set_exc_info(PyThreadState *tstate,
3325 PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003326{
Tim Peters7df5e7f2006-05-26 23:14:37 +00003327 PyFrameObject *frame = tstate->frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003328 PyObject *tmp_type, *tmp_value, *tmp_tb;
Barry Warsaw4249f541997-08-22 21:26:19 +00003329
Tim Peters7df5e7f2006-05-26 23:14:37 +00003330 assert(type != NULL);
3331 assert(frame != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003332 if (frame->f_exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003333 assert(frame->f_exc_value == NULL);
3334 assert(frame->f_exc_traceback == NULL);
3335 /* This frame didn't catch an exception before. */
3336 /* Save previous exception of this thread in this frame. */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003337 if (tstate->exc_type == NULL) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00003338 /* XXX Why is this set to Py_None? */
Guido van Rossuma027efa1997-05-05 20:56:21 +00003339 Py_INCREF(Py_None);
3340 tstate->exc_type = Py_None;
3341 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003342 Py_INCREF(tstate->exc_type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003343 Py_XINCREF(tstate->exc_value);
3344 Py_XINCREF(tstate->exc_traceback);
3345 frame->f_exc_type = tstate->exc_type;
3346 frame->f_exc_value = tstate->exc_value;
3347 frame->f_exc_traceback = tstate->exc_traceback;
3348 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00003349 /* Set new exception for this thread. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003350 tmp_type = tstate->exc_type;
3351 tmp_value = tstate->exc_value;
3352 tmp_tb = tstate->exc_traceback;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003353 Py_INCREF(type);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003354 Py_XINCREF(value);
3355 Py_XINCREF(tb);
3356 tstate->exc_type = type;
3357 tstate->exc_value = value;
3358 tstate->exc_traceback = tb;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003359 Py_XDECREF(tmp_type);
3360 Py_XDECREF(tmp_value);
3361 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003362 /* For b/w compatibility */
3363 PySys_SetObject("exc_type", type);
3364 PySys_SetObject("exc_value", value);
3365 PySys_SetObject("exc_traceback", tb);
3366}
3367
Fredrik Lundh7a830892006-05-27 10:39:48 +00003368static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003369reset_exc_info(PyThreadState *tstate)
Guido van Rossuma027efa1997-05-05 20:56:21 +00003370{
3371 PyFrameObject *frame;
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003372 PyObject *tmp_type, *tmp_value, *tmp_tb;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003373
3374 /* It's a precondition that the thread state's frame caught an
3375 * exception -- verify in a debug build.
3376 */
3377 assert(tstate != NULL);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003378 frame = tstate->frame;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003379 assert(frame != NULL);
3380 assert(frame->f_exc_type != NULL);
3381
3382 /* Copy the frame's exception info back to the thread state. */
3383 tmp_type = tstate->exc_type;
3384 tmp_value = tstate->exc_value;
3385 tmp_tb = tstate->exc_traceback;
3386 Py_INCREF(frame->f_exc_type);
3387 Py_XINCREF(frame->f_exc_value);
3388 Py_XINCREF(frame->f_exc_traceback);
3389 tstate->exc_type = frame->f_exc_type;
3390 tstate->exc_value = frame->f_exc_value;
3391 tstate->exc_traceback = frame->f_exc_traceback;
3392 Py_XDECREF(tmp_type);
3393 Py_XDECREF(tmp_value);
3394 Py_XDECREF(tmp_tb);
3395
3396 /* For b/w compatibility */
3397 PySys_SetObject("exc_type", frame->f_exc_type);
3398 PySys_SetObject("exc_value", frame->f_exc_value);
3399 PySys_SetObject("exc_traceback", frame->f_exc_traceback);
3400
3401 /* Clear the frame's exception info. */
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003402 tmp_type = frame->f_exc_type;
3403 tmp_value = frame->f_exc_value;
3404 tmp_tb = frame->f_exc_traceback;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003405 frame->f_exc_type = NULL;
3406 frame->f_exc_value = NULL;
3407 frame->f_exc_traceback = NULL;
Tim Peters7df5e7f2006-05-26 23:14:37 +00003408 Py_DECREF(tmp_type);
Guido van Rossumdf4c3081997-05-20 17:06:11 +00003409 Py_XDECREF(tmp_value);
3410 Py_XDECREF(tmp_tb);
Guido van Rossuma027efa1997-05-05 20:56:21 +00003411}
3412
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003413/* Logic for the raise statement (too complicated for inlining).
3414 This *consumes* a reference count to each of its arguments. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003415static enum why_code
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003416do_raise(PyObject *type, PyObject *value, PyObject *tb)
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003417{
Guido van Rossumd295f121998-04-09 21:39:57 +00003418 if (type == NULL) {
3419 /* Reraise */
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003420 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossumd295f121998-04-09 21:39:57 +00003421 type = tstate->exc_type == NULL ? Py_None : tstate->exc_type;
3422 value = tstate->exc_value;
3423 tb = tstate->exc_traceback;
3424 Py_XINCREF(type);
3425 Py_XINCREF(value);
3426 Py_XINCREF(tb);
3427 }
Guido van Rossumac7be682001-01-17 15:42:30 +00003428
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003429 /* We support the following forms of raise:
3430 raise <class>, <classinstance>
3431 raise <class>, <argument tuple>
3432 raise <class>, None
3433 raise <class>, <argument>
3434 raise <classinstance>, None
3435 raise <string>, <object>
3436 raise <string>, None
3437
3438 An omitted second argument is the same as None.
3439
3440 In addition, raise <tuple>, <anything> is the same as
3441 raising the tuple's first item (and it better have one!);
3442 this rule is applied recursively.
3443
3444 Finally, an optional third argument can be supplied, which
3445 gives the traceback to be substituted (useful when
3446 re-raising an exception after examining it). */
3447
3448 /* First, check the traceback argument, replacing None with
3449 NULL. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003450 if (tb == Py_None) {
3451 Py_DECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003452 tb = NULL;
3453 }
3454 else if (tb != NULL && !PyTraceBack_Check(tb)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003455 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00003456 "raise: arg 3 must be a traceback or None");
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003457 goto raise_error;
3458 }
3459
3460 /* Next, replace a missing value with None */
3461 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003462 value = Py_None;
3463 Py_INCREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003464 }
3465
3466 /* Next, repeatedly, replace a tuple exception with its first item */
Guido van Rossumb209a111997-04-29 18:18:01 +00003467 while (PyTuple_Check(type) && PyTuple_Size(type) > 0) {
3468 PyObject *tmp = type;
3469 type = PyTuple_GET_ITEM(type, 0);
3470 Py_INCREF(type);
3471 Py_DECREF(tmp);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003472 }
3473
Brett Cannon129bd522007-01-30 21:34:36 +00003474 if (PyExceptionClass_Check(type))
Barry Warsaw4249f541997-08-22 21:26:19 +00003475 PyErr_NormalizeException(&type, &value, &tb);
3476
Brett Cannonbf364092006-03-01 04:25:17 +00003477 else if (PyExceptionInstance_Check(type)) {
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003478 /* Raising an instance. The value should be a dummy. */
Guido van Rossumb209a111997-04-29 18:18:01 +00003479 if (value != Py_None) {
3480 PyErr_SetString(PyExc_TypeError,
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003481 "instance exception may not have a separate value");
3482 goto raise_error;
3483 }
3484 else {
3485 /* Normalize to raise <class>, <instance> */
Guido van Rossumb209a111997-04-29 18:18:01 +00003486 Py_DECREF(value);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003487 value = type;
Brett Cannonbf364092006-03-01 04:25:17 +00003488 type = PyExceptionInstance_Class(type);
Guido van Rossumb209a111997-04-29 18:18:01 +00003489 Py_INCREF(type);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003490 }
3491 }
3492 else {
3493 /* Not something you can raise. You get an exception
3494 anyway, just not what you specified :-) */
Jeremy Hylton960d9482001-04-27 02:25:33 +00003495 PyErr_Format(PyExc_TypeError,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003496 "exceptions must be classes or instances, not %s",
3497 type->ob_type->tp_name);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003498 goto raise_error;
3499 }
Guido van Rossum504153d2008-03-18 04:26:48 +00003500
3501 assert(PyExceptionClass_Check(type));
3502 if (Py_Py3kWarningFlag && PyClass_Check(type)) {
Benjamin Peterson9f4f4812008-04-27 03:01:45 +00003503 if (PyErr_WarnEx(PyExc_DeprecationWarning,
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00003504 "exceptions must derive from BaseException "
3505 "in 3.x", 1) < 0)
Guido van Rossum504153d2008-03-18 04:26:48 +00003506 goto raise_error;
3507 }
3508
Guido van Rossumb209a111997-04-29 18:18:01 +00003509 PyErr_Restore(type, value, tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003510 if (tb == NULL)
3511 return WHY_EXCEPTION;
3512 else
3513 return WHY_RERAISE;
3514 raise_error:
Guido van Rossumb209a111997-04-29 18:18:01 +00003515 Py_XDECREF(value);
3516 Py_XDECREF(type);
3517 Py_XDECREF(tb);
Guido van Rossum0aa9ee61996-12-10 18:07:35 +00003518 return WHY_EXCEPTION;
3519}
3520
Tim Petersd6d010b2001-06-21 02:49:55 +00003521/* Iterate v argcnt times and store the results on the stack (via decreasing
3522 sp). Return 1 for success, 0 if error. */
3523
Fredrik Lundh7a830892006-05-27 10:39:48 +00003524static int
Tim Petersd6d010b2001-06-21 02:49:55 +00003525unpack_iterable(PyObject *v, int argcnt, PyObject **sp)
Barry Warsawe42b18f1997-08-25 22:13:04 +00003526{
Tim Petersd6d010b2001-06-21 02:49:55 +00003527 int i = 0;
3528 PyObject *it; /* iter(v) */
Barry Warsawe42b18f1997-08-25 22:13:04 +00003529 PyObject *w;
Guido van Rossumac7be682001-01-17 15:42:30 +00003530
Tim Petersd6d010b2001-06-21 02:49:55 +00003531 assert(v != NULL);
3532
3533 it = PyObject_GetIter(v);
3534 if (it == NULL)
3535 goto Error;
3536
3537 for (; i < argcnt; i++) {
3538 w = PyIter_Next(it);
3539 if (w == NULL) {
3540 /* Iterator done, via error or exhaustion. */
3541 if (!PyErr_Occurred()) {
3542 PyErr_Format(PyExc_ValueError,
3543 "need more than %d value%s to unpack",
3544 i, i == 1 ? "" : "s");
3545 }
3546 goto Error;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003547 }
3548 *--sp = w;
3549 }
Tim Petersd6d010b2001-06-21 02:49:55 +00003550
3551 /* We better have exhausted the iterator now. */
3552 w = PyIter_Next(it);
3553 if (w == NULL) {
3554 if (PyErr_Occurred())
3555 goto Error;
3556 Py_DECREF(it);
3557 return 1;
Barry Warsawe42b18f1997-08-25 22:13:04 +00003558 }
Guido van Rossumbb8f59a2001-12-03 19:33:25 +00003559 Py_DECREF(w);
Tim Petersd6d010b2001-06-21 02:49:55 +00003560 PyErr_SetString(PyExc_ValueError, "too many values to unpack");
Barry Warsawe42b18f1997-08-25 22:13:04 +00003561 /* fall through */
Tim Petersd6d010b2001-06-21 02:49:55 +00003562Error:
Barry Warsaw91010551997-08-25 22:30:51 +00003563 for (; i > 0; i--, sp++)
3564 Py_DECREF(*sp);
Tim Petersd6d010b2001-06-21 02:49:55 +00003565 Py_XDECREF(it);
Barry Warsawe42b18f1997-08-25 22:13:04 +00003566 return 0;
3567}
3568
3569
Guido van Rossum96a42c81992-01-12 02:29:51 +00003570#ifdef LLTRACE
Fredrik Lundh7a830892006-05-27 10:39:48 +00003571static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003572prtrace(PyObject *v, char *str)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003573{
Guido van Rossum3f5da241990-12-20 15:06:42 +00003574 printf("%s ", str);
Guido van Rossumb209a111997-04-29 18:18:01 +00003575 if (PyObject_Print(v, stdout, 0) != 0)
3576 PyErr_Clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +00003577 printf("\n");
Guido van Rossumcc229ea2000-05-04 00:55:17 +00003578 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003579}
Guido van Rossum3f5da241990-12-20 15:06:42 +00003580#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003581
Fredrik Lundh7a830892006-05-27 10:39:48 +00003582static void
Fred Drake5755ce62001-06-27 19:19:46 +00003583call_exc_trace(Py_tracefunc func, PyObject *self, PyFrameObject *f)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003584{
Guido van Rossumb209a111997-04-29 18:18:01 +00003585 PyObject *type, *value, *traceback, *arg;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003586 int err;
Guido van Rossumb209a111997-04-29 18:18:01 +00003587 PyErr_Fetch(&type, &value, &traceback);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003588 if (value == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003589 value = Py_None;
3590 Py_INCREF(value);
Guido van Rossumbd9ccca1992-04-09 14:58:08 +00003591 }
Raymond Hettinger8ae46892003-10-12 19:09:37 +00003592 arg = PyTuple_Pack(3, type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003593 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003594 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003595 return;
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003596 }
Fred Drake5755ce62001-06-27 19:19:46 +00003597 err = call_trace(func, self, f, PyTrace_EXCEPTION, arg);
Guido van Rossumb209a111997-04-29 18:18:01 +00003598 Py_DECREF(arg);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003599 if (err == 0)
Guido van Rossumb209a111997-04-29 18:18:01 +00003600 PyErr_Restore(type, value, traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003601 else {
Guido van Rossumb209a111997-04-29 18:18:01 +00003602 Py_XDECREF(type);
3603 Py_XDECREF(value);
3604 Py_XDECREF(traceback);
Guido van Rossum1ae940a1995-01-02 19:04:15 +00003605 }
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003606}
3607
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003608static int
Fred Drake4ec5d562001-10-04 19:26:43 +00003609call_trace_protected(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003610 int what, PyObject *arg)
Fred Drake4ec5d562001-10-04 19:26:43 +00003611{
3612 PyObject *type, *value, *traceback;
3613 int err;
3614 PyErr_Fetch(&type, &value, &traceback);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003615 err = call_trace(func, obj, frame, what, arg);
Fred Drake4ec5d562001-10-04 19:26:43 +00003616 if (err == 0)
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003617 {
Fred Drake4ec5d562001-10-04 19:26:43 +00003618 PyErr_Restore(type, value, traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003619 return 0;
3620 }
Fred Drake4ec5d562001-10-04 19:26:43 +00003621 else {
3622 Py_XDECREF(type);
3623 Py_XDECREF(value);
3624 Py_XDECREF(traceback);
Amaury Forgeot d'Arc0d75f092007-11-13 21:54:28 +00003625 return -1;
Fred Drake4ec5d562001-10-04 19:26:43 +00003626 }
3627}
3628
Fredrik Lundh7a830892006-05-27 10:39:48 +00003629static int
Fred Drake5755ce62001-06-27 19:19:46 +00003630call_trace(Py_tracefunc func, PyObject *obj, PyFrameObject *frame,
3631 int what, PyObject *arg)
Guido van Rossum96a42c81992-01-12 02:29:51 +00003632{
Fred Drake5755ce62001-06-27 19:19:46 +00003633 register PyThreadState *tstate = frame->f_tstate;
3634 int result;
3635 if (tstate->tracing)
Guido van Rossum9c8d70d1992-03-23 18:19:28 +00003636 return 0;
Guido van Rossuma027efa1997-05-05 20:56:21 +00003637 tstate->tracing++;
Fred Drake9e3ad782001-07-03 23:39:52 +00003638 tstate->use_tracing = 0;
Fred Drake5755ce62001-06-27 19:19:46 +00003639 result = func(obj, frame, what, arg);
Fred Drake9e3ad782001-07-03 23:39:52 +00003640 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3641 || (tstate->c_profilefunc != NULL));
Guido van Rossuma027efa1997-05-05 20:56:21 +00003642 tstate->tracing--;
Fred Drake5755ce62001-06-27 19:19:46 +00003643 return result;
Guido van Rossum96a42c81992-01-12 02:29:51 +00003644}
3645
Guido van Rossuma12fe4e2003-04-09 19:06:21 +00003646PyObject *
3647_PyEval_CallTracing(PyObject *func, PyObject *args)
3648{
3649 PyFrameObject *frame = PyEval_GetFrame();
3650 PyThreadState *tstate = frame->f_tstate;
3651 int save_tracing = tstate->tracing;
3652 int save_use_tracing = tstate->use_tracing;
3653 PyObject *result;
3654
3655 tstate->tracing = 0;
3656 tstate->use_tracing = ((tstate->c_tracefunc != NULL)
3657 || (tstate->c_profilefunc != NULL));
3658 result = PyObject_Call(func, args, NULL);
3659 tstate->tracing = save_tracing;
3660 tstate->use_tracing = save_use_tracing;
3661 return result;
3662}
3663
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003664/* See Objects/lnotab_notes.txt for a description of how tracing works. */
Fredrik Lundh7a830892006-05-27 10:39:48 +00003665static int
Tim Peters8a5c3c72004-04-05 19:36:21 +00003666maybe_call_line_trace(Py_tracefunc func, PyObject *obj,
Armin Rigobf57a142004-03-22 19:24:58 +00003667 PyFrameObject *frame, int *instr_lb, int *instr_ub,
3668 int *instr_prev)
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003669{
Michael W. Hudson006c7522002-11-08 13:08:46 +00003670 int result = 0;
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003671 int line = frame->f_lineno;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003672
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003673 /* If the last instruction executed isn't in the current
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003674 instruction window, reset the window.
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003675 */
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003676 if (frame->f_lasti < *instr_lb || frame->f_lasti >= *instr_ub) {
Thomas Woutersae406c62007-09-19 17:27:43 +00003677 PyAddrPair bounds;
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003678 line = _PyCode_CheckLineNumber(frame->f_code, frame->f_lasti,
3679 &bounds);
Thomas Woutersae406c62007-09-19 17:27:43 +00003680 *instr_lb = bounds.ap_lower;
3681 *instr_ub = bounds.ap_upper;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003682 }
Jeffrey Yasskin655d8352009-05-23 23:23:01 +00003683 /* If the last instruction falls at the start of a line or if
3684 it represents a jump backwards, update the frame's line
3685 number and call the trace function. */
3686 if (frame->f_lasti == *instr_lb || frame->f_lasti < *instr_prev) {
3687 frame->f_lineno = line;
Jeremy Hyltona4ebc132006-04-18 14:47:00 +00003688 result = call_trace(func, obj, frame, PyTrace_LINE, Py_None);
Armin Rigobf57a142004-03-22 19:24:58 +00003689 }
3690 *instr_prev = frame->f_lasti;
Michael W. Hudson006c7522002-11-08 13:08:46 +00003691 return result;
Michael W. Hudsondd32a912002-08-15 14:59:02 +00003692}
3693
Fred Drake5755ce62001-06-27 19:19:46 +00003694void
3695PyEval_SetProfile(Py_tracefunc func, PyObject *arg)
Fred Draked0838392001-06-16 21:02:31 +00003696{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003697 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003698 PyObject *temp = tstate->c_profileobj;
3699 Py_XINCREF(arg);
3700 tstate->c_profilefunc = NULL;
3701 tstate->c_profileobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003702 /* Must make sure that tracing is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003703 tstate->use_tracing = tstate->c_tracefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003704 Py_XDECREF(temp);
3705 tstate->c_profilefunc = func;
3706 tstate->c_profileobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003707 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003708 tstate->use_tracing = (func != NULL) || (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003709}
3710
3711void
3712PyEval_SetTrace(Py_tracefunc func, PyObject *arg)
3713{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003714 PyThreadState *tstate = PyThreadState_GET();
Fred Drake5755ce62001-06-27 19:19:46 +00003715 PyObject *temp = tstate->c_traceobj;
Jeffrey Yasskinfd8a1ec2008-12-03 06:46:45 +00003716 _Py_TracingPossible += (func != NULL) - (tstate->c_tracefunc != NULL);
Fred Drake5755ce62001-06-27 19:19:46 +00003717 Py_XINCREF(arg);
3718 tstate->c_tracefunc = NULL;
3719 tstate->c_traceobj = NULL;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003720 /* Must make sure that profiling is not ignored if 'temp' is freed */
Fred Drake9e3ad782001-07-03 23:39:52 +00003721 tstate->use_tracing = tstate->c_profilefunc != NULL;
Fred Drake5755ce62001-06-27 19:19:46 +00003722 Py_XDECREF(temp);
3723 tstate->c_tracefunc = func;
3724 tstate->c_traceobj = arg;
Brett Cannon55fa66d2005-06-25 07:07:35 +00003725 /* Flag that tracing or profiling is turned on */
Fred Drake9e3ad782001-07-03 23:39:52 +00003726 tstate->use_tracing = ((func != NULL)
3727 || (tstate->c_profilefunc != NULL));
Fred Draked0838392001-06-16 21:02:31 +00003728}
3729
Guido van Rossumb209a111997-04-29 18:18:01 +00003730PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003731PyEval_GetBuiltins(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003732{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003733 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum6135a871995-01-09 17:53:26 +00003734 if (current_frame == NULL)
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003735 return PyThreadState_GET()->interp->builtins;
Guido van Rossum6135a871995-01-09 17:53:26 +00003736 else
3737 return current_frame->f_builtins;
3738}
3739
Guido van Rossumb209a111997-04-29 18:18:01 +00003740PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003741PyEval_GetLocals(void)
Guido van Rossum5b722181993-03-30 17:46:03 +00003742{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003743 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum5b722181993-03-30 17:46:03 +00003744 if (current_frame == NULL)
3745 return NULL;
Guido van Rossumb209a111997-04-29 18:18:01 +00003746 PyFrame_FastToLocals(current_frame);
Guido van Rossum5b722181993-03-30 17:46:03 +00003747 return current_frame->f_locals;
3748}
3749
Guido van Rossumb209a111997-04-29 18:18:01 +00003750PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003751PyEval_GetGlobals(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +00003752{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003753 PyFrameObject *current_frame = PyEval_GetFrame();
Guido van Rossum3f5da241990-12-20 15:06:42 +00003754 if (current_frame == NULL)
3755 return NULL;
3756 else
3757 return current_frame->f_globals;
3758}
3759
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003760PyFrameObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003761PyEval_GetFrame(void)
Guido van Rossume59214e1994-08-30 08:01:59 +00003762{
Nicholas Bastine5662ae2004-03-24 22:22:12 +00003763 PyThreadState *tstate = PyThreadState_GET();
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003764 return _PyThreadState_GetFrame(tstate);
Guido van Rossume59214e1994-08-30 08:01:59 +00003765}
3766
Guido van Rossum6135a871995-01-09 17:53:26 +00003767int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003768PyEval_GetRestricted(void)
Guido van Rossum6135a871995-01-09 17:53:26 +00003769{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003770 PyFrameObject *current_frame = PyEval_GetFrame();
Neal Norwitzb9845e72006-06-12 02:11:18 +00003771 return current_frame == NULL ? 0 : PyFrame_IsRestricted(current_frame);
Guido van Rossum6135a871995-01-09 17:53:26 +00003772}
3773
Guido van Rossumbe270261997-05-22 22:26:18 +00003774int
Tim Peters5ba58662001-07-16 02:29:45 +00003775PyEval_MergeCompilerFlags(PyCompilerFlags *cf)
Jeremy Hylton061d1062001-03-22 02:32:48 +00003776{
Guido van Rossum6297a7a2003-02-19 15:53:17 +00003777 PyFrameObject *current_frame = PyEval_GetFrame();
Just van Rossum3aaf42c2003-02-10 08:21:10 +00003778 int result = cf->cf_flags != 0;
Tim Peters5ba58662001-07-16 02:29:45 +00003779
3780 if (current_frame != NULL) {
3781 const int codeflags = current_frame->f_code->co_flags;
Tim Peterse2c18e92001-08-17 20:47:47 +00003782 const int compilerflags = codeflags & PyCF_MASK;
3783 if (compilerflags) {
Tim Peters5ba58662001-07-16 02:29:45 +00003784 result = 1;
Tim Peterse2c18e92001-08-17 20:47:47 +00003785 cf->cf_flags |= compilerflags;
Tim Peters5ba58662001-07-16 02:29:45 +00003786 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003787#if 0 /* future keyword */
Martin v. Löwis7198a522002-01-01 19:59:11 +00003788 if (codeflags & CO_GENERATOR_ALLOWED) {
3789 result = 1;
3790 cf->cf_flags |= CO_GENERATOR_ALLOWED;
3791 }
Neil Schemenauerc24ea082002-03-22 23:53:36 +00003792#endif
Tim Peters5ba58662001-07-16 02:29:45 +00003793 }
3794 return result;
Jeremy Hylton061d1062001-03-22 02:32:48 +00003795}
3796
3797int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003798Py_FlushLine(void)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003799{
Guido van Rossumb209a111997-04-29 18:18:01 +00003800 PyObject *f = PySys_GetObject("stdout");
Guido van Rossumbe270261997-05-22 22:26:18 +00003801 if (f == NULL)
3802 return 0;
3803 if (!PyFile_SoftSpace(f, 0))
3804 return 0;
3805 return PyFile_WriteString("\n", f);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00003806}
3807
Guido van Rossum3f5da241990-12-20 15:06:42 +00003808
Guido van Rossum681d79a1995-07-18 14:51:37 +00003809/* External interface to call any callable object.
3810 The arg must be a tuple or NULL. */
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003811
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003812#undef PyEval_CallObject
3813/* for backward compatibility: export this interface */
3814
Guido van Rossumb209a111997-04-29 18:18:01 +00003815PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003816PyEval_CallObject(PyObject *func, PyObject *arg)
Guido van Rossum83bf35c1991-07-27 21:32:34 +00003817{
Guido van Rossumb209a111997-04-29 18:18:01 +00003818 return PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003819}
Guido van Rossumd7ed6831997-08-30 15:02:50 +00003820#define PyEval_CallObject(func,arg) \
3821 PyEval_CallObjectWithKeywords(func, arg, (PyObject *)NULL)
Guido van Rossume59214e1994-08-30 08:01:59 +00003822
Guido van Rossumb209a111997-04-29 18:18:01 +00003823PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00003824PyEval_CallObjectWithKeywords(PyObject *func, PyObject *arg, PyObject *kw)
Guido van Rossum681d79a1995-07-18 14:51:37 +00003825{
Jeremy Hylton52820442001-01-03 23:52:36 +00003826 PyObject *result;
Guido van Rossum681d79a1995-07-18 14:51:37 +00003827
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003828 if (arg == NULL) {
Guido van Rossumb209a111997-04-29 18:18:01 +00003829 arg = PyTuple_New(0);
Hye-Shik Chang4af5c8c2006-03-07 15:39:21 +00003830 if (arg == NULL)
3831 return NULL;
3832 }
Guido van Rossumb209a111997-04-29 18:18:01 +00003833 else if (!PyTuple_Check(arg)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003834 PyErr_SetString(PyExc_TypeError,
3835 "argument list must be a tuple");
Guido van Rossum681d79a1995-07-18 14:51:37 +00003836 return NULL;
3837 }
3838 else
Guido van Rossumb209a111997-04-29 18:18:01 +00003839 Py_INCREF(arg);
Guido van Rossum681d79a1995-07-18 14:51:37 +00003840
Guido van Rossumb209a111997-04-29 18:18:01 +00003841 if (kw != NULL && !PyDict_Check(kw)) {
Guido van Rossuma027efa1997-05-05 20:56:21 +00003842 PyErr_SetString(PyExc_TypeError,
3843 "keyword list must be a dictionary");
Guido van Rossum25826c92000-04-21 21:17:39 +00003844 Py_DECREF(arg);
Guido van Rossume3e61c11995-08-04 04:14:47 +00003845 return NULL;
3846 }
3847
Tim Peters6d6c1a32001-08-02 04:15:00 +00003848 result = PyObject_Call(func, arg, kw);
Guido van Rossumb209a111997-04-29 18:18:01 +00003849 Py_DECREF(arg);
Jeremy Hylton52820442001-01-03 23:52:36 +00003850 return result;
3851}
3852
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003853const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003854PyEval_GetFuncName(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003855{
3856 if (PyMethod_Check(func))
Tim Peters6d6c1a32001-08-02 04:15:00 +00003857 return PyEval_GetFuncName(PyMethod_GET_FUNCTION(func));
Jeremy Hylton512a2372001-04-11 13:52:29 +00003858 else if (PyFunction_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003859 return PyString_AsString(((PyFunctionObject*)func)->func_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003860 else if (PyCFunction_Check(func))
3861 return ((PyCFunctionObject*)func)->m_ml->ml_name;
3862 else if (PyClass_Check(func))
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003863 return PyString_AsString(((PyClassObject*)func)->cl_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00003864 else if (PyInstance_Check(func)) {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00003865 return PyString_AsString(
Jeremy Hylton512a2372001-04-11 13:52:29 +00003866 ((PyInstanceObject*)func)->in_class->cl_name);
3867 } else {
3868 return func->ob_type->tp_name;
3869 }
3870}
3871
Jeremy Hyltonaf68c872005-12-10 18:50:16 +00003872const char *
Tim Peters6d6c1a32001-08-02 04:15:00 +00003873PyEval_GetFuncDesc(PyObject *func)
Jeremy Hylton512a2372001-04-11 13:52:29 +00003874{
3875 if (PyMethod_Check(func))
3876 return "()";
3877 else if (PyFunction_Check(func))
3878 return "()";
3879 else if (PyCFunction_Check(func))
3880 return "()";
3881 else if (PyClass_Check(func))
3882 return " constructor";
3883 else if (PyInstance_Check(func)) {
3884 return " instance";
3885 } else {
3886 return " object";
3887 }
3888}
3889
Fredrik Lundh7a830892006-05-27 10:39:48 +00003890static void
Jeremy Hylton192690e2002-08-16 18:36:11 +00003891err_args(PyObject *func, int flags, int nargs)
3892{
3893 if (flags & METH_NOARGS)
Tim Peters8a5c3c72004-04-05 19:36:21 +00003894 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003895 "%.200s() takes no arguments (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003896 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003897 nargs);
3898 else
Tim Peters8a5c3c72004-04-05 19:36:21 +00003899 PyErr_Format(PyExc_TypeError,
Guido van Rossum86c659a2002-08-23 14:11:35 +00003900 "%.200s() takes exactly one argument (%d given)",
Tim Peters8a5c3c72004-04-05 19:36:21 +00003901 ((PyCFunctionObject *)func)->m_ml->ml_name,
Jeremy Hylton192690e2002-08-16 18:36:11 +00003902 nargs);
3903}
3904
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003905#define C_TRACE(x, call) \
Nicholas Bastind858a772004-06-25 23:31:06 +00003906if (tstate->use_tracing && tstate->c_profilefunc) { \
3907 if (call_trace(tstate->c_profilefunc, \
3908 tstate->c_profileobj, \
3909 tstate->frame, PyTrace_C_CALL, \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003910 func)) { \
3911 x = NULL; \
3912 } \
3913 else { \
3914 x = call; \
3915 if (tstate->c_profilefunc != NULL) { \
3916 if (x == NULL) { \
3917 call_trace_protected(tstate->c_profilefunc, \
3918 tstate->c_profileobj, \
3919 tstate->frame, PyTrace_C_EXCEPTION, \
3920 func); \
3921 /* XXX should pass (type, value, tb) */ \
3922 } else { \
3923 if (call_trace(tstate->c_profilefunc, \
3924 tstate->c_profileobj, \
3925 tstate->frame, PyTrace_C_RETURN, \
3926 func)) { \
3927 Py_DECREF(x); \
3928 x = NULL; \
3929 } \
3930 } \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003931 } \
Nicholas Bastind858a772004-06-25 23:31:06 +00003932 } \
3933} else { \
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003934 x = call; \
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003935 }
3936
Fredrik Lundh7a830892006-05-27 10:39:48 +00003937static PyObject *
Martin v. Löwisf30d60e2004-06-08 08:17:44 +00003938call_function(PyObject ***pp_stack, int oparg
3939#ifdef WITH_TSC
3940 , uint64* pintr0, uint64* pintr1
3941#endif
3942 )
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003943{
3944 int na = oparg & 0xff;
3945 int nk = (oparg>>8) & 0xff;
3946 int n = na + 2 * nk;
3947 PyObject **pfunc = (*pp_stack) - n - 1;
3948 PyObject *func = *pfunc;
3949 PyObject *x, *w;
3950
Jeremy Hylton985eba52003-02-05 23:13:00 +00003951 /* Always dispatch PyCFunction first, because these are
3952 presumed to be the most frequent callable object.
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003953 */
3954 if (PyCFunction_Check(func) && nk == 0) {
3955 int flags = PyCFunction_GET_FLAGS(func);
Nicholas Bastind858a772004-06-25 23:31:06 +00003956 PyThreadState *tstate = PyThreadState_GET();
Raymond Hettingera7f56bc2004-06-26 04:34:33 +00003957
3958 PCALL(PCALL_CFUNCTION);
Jeremy Hylton192690e2002-08-16 18:36:11 +00003959 if (flags & (METH_NOARGS | METH_O)) {
3960 PyCFunction meth = PyCFunction_GET_FUNCTION(func);
3961 PyObject *self = PyCFunction_GET_SELF(func);
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003962 if (flags & METH_NOARGS && na == 0) {
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003963 C_TRACE(x, (*meth)(self,NULL));
Nicholas Bastinc69ebe82004-03-24 21:57:10 +00003964 }
Jeremy Hylton192690e2002-08-16 18:36:11 +00003965 else if (flags & METH_O && na == 1) {
3966 PyObject *arg = EXT_POP(*pp_stack);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003967 C_TRACE(x, (*meth)(self,arg));
Jeremy Hylton192690e2002-08-16 18:36:11 +00003968 Py_DECREF(arg);
3969 }
3970 else {
3971 err_args(func, flags, na);
3972 x = NULL;
3973 }
3974 }
3975 else {
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003976 PyObject *callargs;
3977 callargs = load_args(pp_stack, na);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003978 READ_TIMESTAMP(*pintr0);
Armin Rigo1c2d7e52005-09-20 18:34:01 +00003979 C_TRACE(x, PyCFunction_Call(func,callargs,NULL));
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003980 READ_TIMESTAMP(*pintr1);
Tim Peters8a5c3c72004-04-05 19:36:21 +00003981 Py_XDECREF(callargs);
3982 }
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003983 } else {
3984 if (PyMethod_Check(func) && PyMethod_GET_SELF(func) != NULL) {
3985 /* optimize access to bound methods */
3986 PyObject *self = PyMethod_GET_SELF(func);
Jeremy Hylton985eba52003-02-05 23:13:00 +00003987 PCALL(PCALL_METHOD);
3988 PCALL(PCALL_BOUND_METHOD);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003989 Py_INCREF(self);
3990 func = PyMethod_GET_FUNCTION(func);
3991 Py_INCREF(func);
3992 Py_DECREF(*pfunc);
3993 *pfunc = self;
3994 na++;
3995 n++;
3996 } else
3997 Py_INCREF(func);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00003998 READ_TIMESTAMP(*pintr0);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00003999 if (PyFunction_Check(func))
4000 x = fast_function(func, pp_stack, n, na, nk);
Tim Peters8a5c3c72004-04-05 19:36:21 +00004001 else
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004002 x = do_call(func, pp_stack, na, nk);
Michael W. Hudson75eabd22005-01-18 15:56:11 +00004003 READ_TIMESTAMP(*pintr1);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004004 Py_DECREF(func);
4005 }
Tim Peters8a5c3c72004-04-05 19:36:21 +00004006
Armin Rigod34fa522006-03-28 19:10:40 +00004007 /* Clear the stack of the function object. Also removes
4008 the arguments in case they weren't consumed already
4009 (fast_function() and err_args() leave them on the stack).
Thomas Wouters7f597322006-03-01 05:32:33 +00004010 */
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004011 while ((*pp_stack) > pfunc) {
4012 w = EXT_POP(*pp_stack);
4013 Py_DECREF(w);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004014 PCALL(PCALL_POP);
Jeremy Hyltone8c04322002-08-16 17:47:26 +00004015 }
4016 return x;
4017}
4018
Jeremy Hylton192690e2002-08-16 18:36:11 +00004019/* The fast_function() function optimize calls for which no argument
Jeremy Hylton52820442001-01-03 23:52:36 +00004020 tuple is necessary; the objects are passed directly from the stack.
Jeremy Hylton985eba52003-02-05 23:13:00 +00004021 For the simplest case -- a function that takes only positional
4022 arguments and is called with only positional arguments -- it
4023 inlines the most primitive frame setup code from
4024 PyEval_EvalCodeEx(), which vastly reduces the checks that must be
4025 done before evaluating the frame.
Jeremy Hylton52820442001-01-03 23:52:36 +00004026*/
4027
Fredrik Lundh7a830892006-05-27 10:39:48 +00004028static PyObject *
Guido van Rossumac7be682001-01-17 15:42:30 +00004029fast_function(PyObject *func, PyObject ***pp_stack, int n, int na, int nk)
Jeremy Hylton52820442001-01-03 23:52:36 +00004030{
Jeremy Hylton985eba52003-02-05 23:13:00 +00004031 PyCodeObject *co = (PyCodeObject *)PyFunction_GET_CODE(func);
Jeremy Hylton52820442001-01-03 23:52:36 +00004032 PyObject *globals = PyFunction_GET_GLOBALS(func);
4033 PyObject *argdefs = PyFunction_GET_DEFAULTS(func);
4034 PyObject **d = NULL;
4035 int nd = 0;
4036
Jeremy Hylton985eba52003-02-05 23:13:00 +00004037 PCALL(PCALL_FUNCTION);
4038 PCALL(PCALL_FAST_FUNCTION);
Raymond Hettinger40174c32003-05-31 07:04:16 +00004039 if (argdefs == NULL && co->co_argcount == n && nk==0 &&
Jeremy Hylton985eba52003-02-05 23:13:00 +00004040 co->co_flags == (CO_OPTIMIZED | CO_NEWLOCALS | CO_NOFREE)) {
4041 PyFrameObject *f;
4042 PyObject *retval = NULL;
4043 PyThreadState *tstate = PyThreadState_GET();
4044 PyObject **fastlocals, **stack;
4045 int i;
4046
4047 PCALL(PCALL_FASTER_FUNCTION);
4048 assert(globals != NULL);
4049 /* XXX Perhaps we should create a specialized
4050 PyFrame_New() that doesn't take locals, but does
4051 take builtins without sanity checking them.
4052 */
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004053 assert(tstate != NULL);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004054 f = PyFrame_New(tstate, co, globals, NULL);
4055 if (f == NULL)
4056 return NULL;
4057
4058 fastlocals = f->f_localsplus;
4059 stack = (*pp_stack) - n;
4060
4061 for (i = 0; i < n; i++) {
4062 Py_INCREF(*stack);
4063 fastlocals[i] = *stack++;
4064 }
Phillip J. Eby0d6615f2005-08-02 00:46:46 +00004065 retval = PyEval_EvalFrameEx(f,0);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004066 ++tstate->recursion_depth;
4067 Py_DECREF(f);
4068 --tstate->recursion_depth;
4069 return retval;
4070 }
Jeremy Hylton52820442001-01-03 23:52:36 +00004071 if (argdefs != NULL) {
4072 d = &PyTuple_GET_ITEM(argdefs, 0);
Christian Heimese93237d2007-12-19 02:37:44 +00004073 nd = Py_SIZE(argdefs);
Jeremy Hylton52820442001-01-03 23:52:36 +00004074 }
Jeremy Hylton985eba52003-02-05 23:13:00 +00004075 return PyEval_EvalCodeEx(co, globals,
4076 (PyObject *)NULL, (*pp_stack)-n, na,
4077 (*pp_stack)-2*nk, nk, d, nd,
4078 PyFunction_GET_CLOSURE(func));
Jeremy Hylton52820442001-01-03 23:52:36 +00004079}
4080
Fredrik Lundh7a830892006-05-27 10:39:48 +00004081static PyObject *
Ka-Ping Yee20579702001-01-15 22:14:16 +00004082update_keyword_args(PyObject *orig_kwdict, int nk, PyObject ***pp_stack,
4083 PyObject *func)
Jeremy Hylton52820442001-01-03 23:52:36 +00004084{
4085 PyObject *kwdict = NULL;
4086 if (orig_kwdict == NULL)
4087 kwdict = PyDict_New();
4088 else {
4089 kwdict = PyDict_Copy(orig_kwdict);
4090 Py_DECREF(orig_kwdict);
4091 }
4092 if (kwdict == NULL)
4093 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00004094 while (--nk >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00004095 int err;
4096 PyObject *value = EXT_POP(*pp_stack);
4097 PyObject *key = EXT_POP(*pp_stack);
4098 if (PyDict_GetItem(kwdict, key) != NULL) {
Thomas Woutersae406c62007-09-19 17:27:43 +00004099 PyErr_Format(PyExc_TypeError,
4100 "%.200s%s got multiple values "
4101 "for keyword argument '%.200s'",
Tim Peters6d6c1a32001-08-02 04:15:00 +00004102 PyEval_GetFuncName(func),
4103 PyEval_GetFuncDesc(func),
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004104 PyString_AsString(key));
Jeremy Hylton52820442001-01-03 23:52:36 +00004105 Py_DECREF(key);
4106 Py_DECREF(value);
4107 Py_DECREF(kwdict);
4108 return NULL;
4109 }
4110 err = PyDict_SetItem(kwdict, key, value);
4111 Py_DECREF(key);
4112 Py_DECREF(value);
4113 if (err) {
4114 Py_DECREF(kwdict);
4115 return NULL;
4116 }
4117 }
4118 return kwdict;
4119}
4120
Fredrik Lundh7a830892006-05-27 10:39:48 +00004121static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004122update_star_args(int nstack, int nstar, PyObject *stararg,
4123 PyObject ***pp_stack)
4124{
4125 PyObject *callargs, *w;
4126
4127 callargs = PyTuple_New(nstack + nstar);
4128 if (callargs == NULL) {
4129 return NULL;
4130 }
4131 if (nstar) {
4132 int i;
4133 for (i = 0; i < nstar; i++) {
4134 PyObject *a = PyTuple_GET_ITEM(stararg, i);
4135 Py_INCREF(a);
4136 PyTuple_SET_ITEM(callargs, nstack + i, a);
4137 }
4138 }
Raymond Hettinger5bed4562004-04-10 23:34:17 +00004139 while (--nstack >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00004140 w = EXT_POP(*pp_stack);
4141 PyTuple_SET_ITEM(callargs, nstack, w);
4142 }
4143 return callargs;
4144}
4145
Fredrik Lundh7a830892006-05-27 10:39:48 +00004146static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004147load_args(PyObject ***pp_stack, int na)
4148{
4149 PyObject *args = PyTuple_New(na);
4150 PyObject *w;
4151
4152 if (args == NULL)
4153 return NULL;
Raymond Hettinger5bed4562004-04-10 23:34:17 +00004154 while (--na >= 0) {
Jeremy Hylton52820442001-01-03 23:52:36 +00004155 w = EXT_POP(*pp_stack);
4156 PyTuple_SET_ITEM(args, na, w);
4157 }
4158 return args;
4159}
4160
Fredrik Lundh7a830892006-05-27 10:39:48 +00004161static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004162do_call(PyObject *func, PyObject ***pp_stack, int na, int nk)
4163{
4164 PyObject *callargs = NULL;
4165 PyObject *kwdict = NULL;
4166 PyObject *result = NULL;
4167
4168 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00004169 kwdict = update_keyword_args(NULL, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00004170 if (kwdict == NULL)
4171 goto call_fail;
4172 }
4173 callargs = load_args(pp_stack, na);
4174 if (callargs == NULL)
4175 goto call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004176#ifdef CALL_PROFILE
4177 /* At this point, we have to look at the type of func to
4178 update the call stats properly. Do it here so as to avoid
4179 exposing the call stats machinery outside ceval.c
4180 */
4181 if (PyFunction_Check(func))
4182 PCALL(PCALL_FUNCTION);
4183 else if (PyMethod_Check(func))
4184 PCALL(PCALL_METHOD);
4185 else if (PyType_Check(func))
4186 PCALL(PCALL_TYPE);
Antoine Pitrou46dbe272009-05-30 21:27:00 +00004187 else if (PyCFunction_Check(func))
4188 PCALL(PCALL_CFUNCTION);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004189 else
4190 PCALL(PCALL_OTHER);
4191#endif
Antoine Pitrou46dbe272009-05-30 21:27:00 +00004192 if (PyCFunction_Check(func)) {
4193 PyThreadState *tstate = PyThreadState_GET();
4194 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4195 }
4196 else
4197 result = PyObject_Call(func, callargs, kwdict);
Jeremy Hylton52820442001-01-03 23:52:36 +00004198 call_fail:
4199 Py_XDECREF(callargs);
4200 Py_XDECREF(kwdict);
4201 return result;
4202}
4203
Fredrik Lundh7a830892006-05-27 10:39:48 +00004204static PyObject *
Jeremy Hylton52820442001-01-03 23:52:36 +00004205ext_do_call(PyObject *func, PyObject ***pp_stack, int flags, int na, int nk)
4206{
4207 int nstar = 0;
4208 PyObject *callargs = NULL;
4209 PyObject *stararg = NULL;
4210 PyObject *kwdict = NULL;
4211 PyObject *result = NULL;
4212
4213 if (flags & CALL_FLAG_KW) {
4214 kwdict = EXT_POP(*pp_stack);
Georg Brandl2134e752007-05-21 20:34:16 +00004215 if (!PyDict_Check(kwdict)) {
4216 PyObject *d;
4217 d = PyDict_New();
4218 if (d == NULL)
4219 goto ext_call_fail;
4220 if (PyDict_Update(d, kwdict) != 0) {
4221 Py_DECREF(d);
4222 /* PyDict_Update raises attribute
4223 * error (percolated from an attempt
4224 * to get 'keys' attribute) instead of
4225 * a type error if its second argument
4226 * is not a mapping.
4227 */
4228 if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
4229 PyErr_Format(PyExc_TypeError,
4230 "%.200s%.200s argument after ** "
4231 "must be a mapping, not %.200s",
4232 PyEval_GetFuncName(func),
4233 PyEval_GetFuncDesc(func),
4234 kwdict->ob_type->tp_name);
4235 }
4236 goto ext_call_fail;
4237 }
4238 Py_DECREF(kwdict);
4239 kwdict = d;
Jeremy Hylton52820442001-01-03 23:52:36 +00004240 }
4241 }
4242 if (flags & CALL_FLAG_VAR) {
4243 stararg = EXT_POP(*pp_stack);
4244 if (!PyTuple_Check(stararg)) {
4245 PyObject *t = NULL;
4246 t = PySequence_Tuple(stararg);
4247 if (t == NULL) {
Jeremy Hylton512a2372001-04-11 13:52:29 +00004248 if (PyErr_ExceptionMatches(PyExc_TypeError)) {
4249 PyErr_Format(PyExc_TypeError,
Georg Brandl2134e752007-05-21 20:34:16 +00004250 "%.200s%.200s argument after * "
4251 "must be a sequence, not %200s",
Tim Peters6d6c1a32001-08-02 04:15:00 +00004252 PyEval_GetFuncName(func),
Georg Brandl2134e752007-05-21 20:34:16 +00004253 PyEval_GetFuncDesc(func),
4254 stararg->ob_type->tp_name);
Jeremy Hylton512a2372001-04-11 13:52:29 +00004255 }
Jeremy Hylton52820442001-01-03 23:52:36 +00004256 goto ext_call_fail;
4257 }
4258 Py_DECREF(stararg);
4259 stararg = t;
4260 }
4261 nstar = PyTuple_GET_SIZE(stararg);
4262 }
4263 if (nk > 0) {
Ka-Ping Yee20579702001-01-15 22:14:16 +00004264 kwdict = update_keyword_args(kwdict, nk, pp_stack, func);
Jeremy Hylton52820442001-01-03 23:52:36 +00004265 if (kwdict == NULL)
4266 goto ext_call_fail;
4267 }
4268 callargs = update_star_args(na, nstar, stararg, pp_stack);
4269 if (callargs == NULL)
4270 goto ext_call_fail;
Jeremy Hylton985eba52003-02-05 23:13:00 +00004271#ifdef CALL_PROFILE
4272 /* At this point, we have to look at the type of func to
4273 update the call stats properly. Do it here so as to avoid
4274 exposing the call stats machinery outside ceval.c
4275 */
4276 if (PyFunction_Check(func))
4277 PCALL(PCALL_FUNCTION);
4278 else if (PyMethod_Check(func))
4279 PCALL(PCALL_METHOD);
4280 else if (PyType_Check(func))
4281 PCALL(PCALL_TYPE);
Antoine Pitrou46dbe272009-05-30 21:27:00 +00004282 else if (PyCFunction_Check(func))
4283 PCALL(PCALL_CFUNCTION);
Jeremy Hylton985eba52003-02-05 23:13:00 +00004284 else
4285 PCALL(PCALL_OTHER);
4286#endif
Antoine Pitrou46dbe272009-05-30 21:27:00 +00004287 if (PyCFunction_Check(func)) {
4288 PyThreadState *tstate = PyThreadState_GET();
4289 C_TRACE(result, PyCFunction_Call(func, callargs, kwdict));
4290 }
4291 else
4292 result = PyObject_Call(func, callargs, kwdict);
Thomas Woutersae406c62007-09-19 17:27:43 +00004293ext_call_fail:
Jeremy Hylton52820442001-01-03 23:52:36 +00004294 Py_XDECREF(callargs);
4295 Py_XDECREF(kwdict);
4296 Py_XDECREF(stararg);
4297 return result;
4298}
4299
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004300/* Extract a slice index from a PyInt or PyLong or an object with the
4301 nb_index slot defined, and store in *pi.
4302 Silently reduce values larger than PY_SSIZE_T_MAX to PY_SSIZE_T_MAX,
4303 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 +00004304 Return 0 on error, 1 on success.
Tim Peterscb479e72001-12-16 19:11:44 +00004305*/
Tim Petersb5196382001-12-16 19:44:20 +00004306/* Note: If v is NULL, return success without storing into *pi. This
4307 is because_PyEval_SliceIndex() is called by apply_slice(), which can be
4308 called by the SLICE opcode with v and/or w equal to NULL.
Tim Peterscb479e72001-12-16 19:11:44 +00004309*/
Guido van Rossum20c6add2000-05-08 14:06:50 +00004310int
Martin v. Löwis18e16552006-02-15 17:27:45 +00004311_PyEval_SliceIndex(PyObject *v, Py_ssize_t *pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004312{
Tim Petersb5196382001-12-16 19:44:20 +00004313 if (v != NULL) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004314 Py_ssize_t x;
Andrew M. Kuchling2194b162000-02-23 22:18:48 +00004315 if (PyInt_Check(v)) {
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004316 /* XXX(nnorwitz): I think PyInt_AS_LONG is correct,
4317 however, it looks like it should be AsSsize_t.
4318 There should be a comment here explaining why.
4319 */
4320 x = PyInt_AS_LONG(v);
Tim Peters7df5e7f2006-05-26 23:14:37 +00004321 }
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004322 else if (PyIndex_Check(v)) {
4323 x = PyNumber_AsSsize_t(v, NULL);
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004324 if (x == -1 && PyErr_Occurred())
4325 return 0;
4326 }
4327 else {
Guido van Rossuma027efa1997-05-05 20:56:21 +00004328 PyErr_SetString(PyExc_TypeError,
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004329 "slice indices must be integers or "
4330 "None or have an __index__ method");
Guido van Rossum20c6add2000-05-08 14:06:50 +00004331 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004332 }
Guido van Rossuma027efa1997-05-05 20:56:21 +00004333 *pi = x;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004334 }
Guido van Rossum20c6add2000-05-08 14:06:50 +00004335 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004336}
4337
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004338#undef ISINDEX
Neal Norwitz8a87f5d2006-08-12 17:03:09 +00004339#define ISINDEX(x) ((x) == NULL || \
4340 PyInt_Check(x) || PyLong_Check(x) || PyIndex_Check(x))
Guido van Rossum50d756e2001-08-18 17:43:36 +00004341
Fredrik Lundh7a830892006-05-27 10:39:48 +00004342static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004343apply_slice(PyObject *u, PyObject *v, PyObject *w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004344{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004345 PyTypeObject *tp = u->ob_type;
4346 PySequenceMethods *sq = tp->tp_as_sequence;
4347
Guido van Rossum38fff8c2006-03-07 18:50:55 +00004348 if (sq && sq->sq_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004349 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004350 if (!_PyEval_SliceIndex(v, &ilow))
4351 return NULL;
4352 if (!_PyEval_SliceIndex(w, &ihigh))
4353 return NULL;
4354 return PySequence_GetSlice(u, ilow, ihigh);
4355 }
4356 else {
4357 PyObject *slice = PySlice_New(v, w, NULL);
Guido van Rossum354797c2001-12-03 19:45:06 +00004358 if (slice != NULL) {
4359 PyObject *res = PyObject_GetItem(u, slice);
4360 Py_DECREF(slice);
4361 return res;
4362 }
Guido van Rossum50d756e2001-08-18 17:43:36 +00004363 else
4364 return NULL;
4365 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004366}
Guido van Rossum3f5da241990-12-20 15:06:42 +00004367
Fredrik Lundh7a830892006-05-27 10:39:48 +00004368static int
Guido van Rossumac7be682001-01-17 15:42:30 +00004369assign_slice(PyObject *u, PyObject *v, PyObject *w, PyObject *x)
4370 /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004371{
Guido van Rossum50d756e2001-08-18 17:43:36 +00004372 PyTypeObject *tp = u->ob_type;
4373 PySequenceMethods *sq = tp->tp_as_sequence;
4374
Georg Brandl0fca97a2007-03-05 22:28:08 +00004375 if (sq && sq->sq_ass_slice && ISINDEX(v) && ISINDEX(w)) {
Martin v. Löwisdde99d22006-02-17 15:57:41 +00004376 Py_ssize_t ilow = 0, ihigh = PY_SSIZE_T_MAX;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004377 if (!_PyEval_SliceIndex(v, &ilow))
4378 return -1;
4379 if (!_PyEval_SliceIndex(w, &ihigh))
4380 return -1;
4381 if (x == NULL)
4382 return PySequence_DelSlice(u, ilow, ihigh);
4383 else
4384 return PySequence_SetSlice(u, ilow, ihigh, x);
4385 }
4386 else {
4387 PyObject *slice = PySlice_New(v, w, NULL);
4388 if (slice != NULL) {
Guido van Rossum354797c2001-12-03 19:45:06 +00004389 int res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004390 if (x != NULL)
Guido van Rossum354797c2001-12-03 19:45:06 +00004391 res = PyObject_SetItem(u, slice, x);
Guido van Rossum50d756e2001-08-18 17:43:36 +00004392 else
Guido van Rossum354797c2001-12-03 19:45:06 +00004393 res = PyObject_DelItem(u, slice);
4394 Py_DECREF(slice);
4395 return res;
Guido van Rossum50d756e2001-08-18 17:43:36 +00004396 }
4397 else
4398 return -1;
4399 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004400}
4401
Guido van Rossum04edb522008-03-18 02:49:46 +00004402#define Py3kExceptionClass_Check(x) \
4403 (PyType_Check((x)) && \
4404 PyType_FastSubclass((PyTypeObject*)(x), Py_TPFLAGS_BASE_EXC_SUBCLASS))
4405
4406#define CANNOT_CATCH_MSG "catching classes that don't inherit from " \
Georg Brandld5b635f2008-03-25 08:29:14 +00004407 "BaseException is not allowed in 3.x"
Guido van Rossum04edb522008-03-18 02:49:46 +00004408
Fredrik Lundh7a830892006-05-27 10:39:48 +00004409static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004410cmp_outcome(int op, register PyObject *v, register PyObject *w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004411{
Guido van Rossumac7be682001-01-17 15:42:30 +00004412 int res = 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004413 switch (op) {
Martin v. Löwis7198a522002-01-01 19:59:11 +00004414 case PyCmp_IS:
Guido van Rossum3f5da241990-12-20 15:06:42 +00004415 res = (v == w);
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004416 break;
4417 case PyCmp_IS_NOT:
4418 res = (v != w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004419 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004420 case PyCmp_IN:
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004421 res = PySequence_Contains(w, v);
4422 if (res < 0)
4423 return NULL;
4424 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004425 case PyCmp_NOT_IN:
Guido van Rossum7e33c6e1998-05-22 00:52:29 +00004426 res = PySequence_Contains(w, v);
Guido van Rossum3f5da241990-12-20 15:06:42 +00004427 if (res < 0)
4428 return NULL;
Raymond Hettinger4bad9ba2003-01-19 05:08:13 +00004429 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004430 break;
Martin v. Löwis7198a522002-01-01 19:59:11 +00004431 case PyCmp_EXC_MATCH:
Brett Cannon129bd522007-01-30 21:34:36 +00004432 if (PyTuple_Check(w)) {
4433 Py_ssize_t i, length;
4434 length = PyTuple_Size(w);
4435 for (i = 0; i < length; i += 1) {
4436 PyObject *exc = PyTuple_GET_ITEM(w, i);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004437 if (PyString_Check(exc)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004438 int ret_val;
4439 ret_val = PyErr_WarnEx(
Thomas Wouterse2176022007-09-20 17:35:10 +00004440 PyExc_DeprecationWarning,
4441 "catching of string "
4442 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004443 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004444 return NULL;
4445 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004446 else if (Py_Py3kWarningFlag &&
4447 !PyTuple_Check(exc) &&
4448 !Py3kExceptionClass_Check(exc))
Guido van Rossum04edb522008-03-18 02:49:46 +00004449 {
4450 int ret_val;
4451 ret_val = PyErr_WarnEx(
4452 PyExc_DeprecationWarning,
4453 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004454 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004455 return NULL;
4456 }
Brett Cannon129bd522007-01-30 21:34:36 +00004457 }
4458 }
4459 else {
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004460 if (PyString_Check(w)) {
Brett Cannon129bd522007-01-30 21:34:36 +00004461 int ret_val;
4462 ret_val = PyErr_WarnEx(
4463 PyExc_DeprecationWarning,
4464 "catching of string "
Thomas Wouterse2176022007-09-20 17:35:10 +00004465 "exceptions is deprecated", 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004466 if (ret_val < 0)
Brett Cannon129bd522007-01-30 21:34:36 +00004467 return NULL;
4468 }
Guido van Rossum20bda582008-03-18 03:15:05 +00004469 else if (Py_Py3kWarningFlag &&
4470 !PyTuple_Check(w) &&
4471 !Py3kExceptionClass_Check(w))
Guido van Rossum04edb522008-03-18 02:49:46 +00004472 {
4473 int ret_val;
4474 ret_val = PyErr_WarnEx(
4475 PyExc_DeprecationWarning,
4476 CANNOT_CATCH_MSG, 1);
Benjamin Petersonf19a7b92008-04-27 18:40:21 +00004477 if (ret_val < 0)
Guido van Rossum04edb522008-03-18 02:49:46 +00004478 return NULL;
4479 }
Brett Cannon129bd522007-01-30 21:34:36 +00004480 }
Barry Warsaw4249f541997-08-22 21:26:19 +00004481 res = PyErr_GivenExceptionMatches(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004482 break;
4483 default:
Guido van Rossumac7be682001-01-17 15:42:30 +00004484 return PyObject_RichCompare(v, w, op);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004485 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004486 v = res ? Py_True : Py_False;
4487 Py_INCREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00004488 return v;
4489}
4490
Fredrik Lundh7a830892006-05-27 10:39:48 +00004491static PyObject *
Thomas Wouters52152252000-08-17 22:55:00 +00004492import_from(PyObject *v, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004493{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004494 PyObject *x;
4495
4496 x = PyObject_GetAttr(v, name);
4497 if (x == NULL && PyErr_ExceptionMatches(PyExc_AttributeError)) {
Thomas Wouters52152252000-08-17 22:55:00 +00004498 PyErr_Format(PyExc_ImportError,
4499 "cannot import name %.230s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004500 PyString_AsString(name));
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004501 }
Thomas Wouters52152252000-08-17 22:55:00 +00004502 return x;
4503}
Guido van Rossumac7be682001-01-17 15:42:30 +00004504
Fredrik Lundh7a830892006-05-27 10:39:48 +00004505static int
Thomas Wouters52152252000-08-17 22:55:00 +00004506import_all_from(PyObject *locals, PyObject *v)
4507{
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004508 PyObject *all = PyObject_GetAttrString(v, "__all__");
4509 PyObject *dict, *name, *value;
4510 int skip_leading_underscores = 0;
4511 int pos, err;
Thomas Wouters52152252000-08-17 22:55:00 +00004512
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004513 if (all == NULL) {
4514 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4515 return -1; /* Unexpected error */
4516 PyErr_Clear();
4517 dict = PyObject_GetAttrString(v, "__dict__");
4518 if (dict == NULL) {
4519 if (!PyErr_ExceptionMatches(PyExc_AttributeError))
4520 return -1;
4521 PyErr_SetString(PyExc_ImportError,
4522 "from-import-* object has no __dict__ and no __all__");
Guido van Rossum3f5da241990-12-20 15:06:42 +00004523 return -1;
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004524 }
4525 all = PyMapping_Keys(dict);
4526 Py_DECREF(dict);
4527 if (all == NULL)
4528 return -1;
4529 skip_leading_underscores = 1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004530 }
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004531
4532 for (pos = 0, err = 0; ; pos++) {
4533 name = PySequence_GetItem(all, pos);
4534 if (name == NULL) {
4535 if (!PyErr_ExceptionMatches(PyExc_IndexError))
4536 err = -1;
4537 else
4538 PyErr_Clear();
4539 break;
4540 }
4541 if (skip_leading_underscores &&
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004542 PyString_Check(name) &&
4543 PyString_AS_STRING(name)[0] == '_')
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004544 {
4545 Py_DECREF(name);
4546 continue;
4547 }
4548 value = PyObject_GetAttr(v, name);
4549 if (value == NULL)
4550 err = -1;
Armin Rigo70370852006-11-29 21:59:22 +00004551 else if (PyDict_CheckExact(locals))
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004552 err = PyDict_SetItem(locals, name, value);
Armin Rigo70370852006-11-29 21:59:22 +00004553 else
4554 err = PyObject_SetItem(locals, name, value);
Guido van Rossum18d4d8f2001-01-12 16:24:03 +00004555 Py_DECREF(name);
4556 Py_XDECREF(value);
4557 if (err != 0)
4558 break;
4559 }
4560 Py_DECREF(all);
4561 return err;
Guido van Rossume9736fc1990-11-18 17:33:06 +00004562}
4563
Fredrik Lundh7a830892006-05-27 10:39:48 +00004564static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004565build_class(PyObject *methods, PyObject *bases, PyObject *name)
Guido van Rossume9736fc1990-11-18 17:33:06 +00004566{
Guido van Rossum7851eea2001-09-12 19:19:18 +00004567 PyObject *metaclass = NULL, *result, *base;
Tim Peters6d6c1a32001-08-02 04:15:00 +00004568
4569 if (PyDict_Check(methods))
4570 metaclass = PyDict_GetItemString(methods, "__metaclass__");
Guido van Rossum7851eea2001-09-12 19:19:18 +00004571 if (metaclass != NULL)
Guido van Rossum2556f2e2001-12-06 14:09:56 +00004572 Py_INCREF(metaclass);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004573 else if (PyTuple_Check(bases) && PyTuple_GET_SIZE(bases) > 0) {
4574 base = PyTuple_GET_ITEM(bases, 0);
4575 metaclass = PyObject_GetAttrString(base, "__class__");
4576 if (metaclass == NULL) {
4577 PyErr_Clear();
4578 metaclass = (PyObject *)base->ob_type;
4579 Py_INCREF(metaclass);
Guido van Rossum25831651993-05-19 14:50:45 +00004580 }
4581 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004582 else {
4583 PyObject *g = PyEval_GetGlobals();
4584 if (g != NULL && PyDict_Check(g))
4585 metaclass = PyDict_GetItemString(g, "__metaclass__");
4586 if (metaclass == NULL)
4587 metaclass = (PyObject *) &PyClass_Type;
4588 Py_INCREF(metaclass);
4589 }
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004590 result = PyObject_CallFunctionObjArgs(metaclass, name, bases, methods,
Thomas Woutersae406c62007-09-19 17:27:43 +00004591 NULL);
Guido van Rossum7851eea2001-09-12 19:19:18 +00004592 Py_DECREF(metaclass);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004593 if (result == NULL && PyErr_ExceptionMatches(PyExc_TypeError)) {
Tim Peters7df5e7f2006-05-26 23:14:37 +00004594 /* A type error here likely means that the user passed
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004595 in a base that was not a class (such the random module
4596 instead of the random.random type). Help them out with
Raymond Hettingercfc31922004-09-16 16:41:57 +00004597 by augmenting the error message with more information.*/
4598
4599 PyObject *ptype, *pvalue, *ptraceback;
4600
4601 PyErr_Fetch(&ptype, &pvalue, &ptraceback);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004602 if (PyString_Check(pvalue)) {
Raymond Hettingercfc31922004-09-16 16:41:57 +00004603 PyObject *newmsg;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004604 newmsg = PyString_FromFormat(
Jeremy Hylton7c1e3472007-02-26 16:14:51 +00004605 "Error when calling the metaclass bases\n"
Thomas Woutersae406c62007-09-19 17:27:43 +00004606 " %s",
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004607 PyString_AS_STRING(pvalue));
Raymond Hettingercfc31922004-09-16 16:41:57 +00004608 if (newmsg != NULL) {
4609 Py_DECREF(pvalue);
4610 pvalue = newmsg;
4611 }
4612 }
4613 PyErr_Restore(ptype, pvalue, ptraceback);
Raymond Hettingerf2c08302004-06-05 06:16:22 +00004614 }
Guido van Rossum7851eea2001-09-12 19:19:18 +00004615 return result;
Guido van Rossum25831651993-05-19 14:50:45 +00004616}
4617
Fredrik Lundh7a830892006-05-27 10:39:48 +00004618static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004619exec_statement(PyFrameObject *f, PyObject *prog, PyObject *globals,
4620 PyObject *locals)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004621{
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004622 int n;
Guido van Rossumb209a111997-04-29 18:18:01 +00004623 PyObject *v;
Guido van Rossum681d79a1995-07-18 14:51:37 +00004624 int plain = 0;
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004625
Guido van Rossumb209a111997-04-29 18:18:01 +00004626 if (PyTuple_Check(prog) && globals == Py_None && locals == Py_None &&
4627 ((n = PyTuple_Size(prog)) == 2 || n == 3)) {
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004628 /* Backward compatibility hack */
Guido van Rossumb209a111997-04-29 18:18:01 +00004629 globals = PyTuple_GetItem(prog, 1);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004630 if (n == 3)
Guido van Rossumb209a111997-04-29 18:18:01 +00004631 locals = PyTuple_GetItem(prog, 2);
4632 prog = PyTuple_GetItem(prog, 0);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004633 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004634 if (globals == Py_None) {
4635 globals = PyEval_GetGlobals();
4636 if (locals == Py_None) {
4637 locals = PyEval_GetLocals();
Guido van Rossum681d79a1995-07-18 14:51:37 +00004638 plain = 1;
4639 }
Neal Norwitzdf6a6492006-08-13 18:10:10 +00004640 if (!globals || !locals) {
4641 PyErr_SetString(PyExc_SystemError,
4642 "globals and locals cannot be NULL");
4643 return -1;
4644 }
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004645 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004646 else if (locals == Py_None)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004647 locals = globals;
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004648 if (!PyString_Check(prog) &&
Benjamin Peterson78821dd2009-01-25 17:15:10 +00004649#ifdef Py_USING_UNICODE
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004650 !PyUnicode_Check(prog) &&
Benjamin Peterson78821dd2009-01-25 17:15:10 +00004651#endif
Guido van Rossumb209a111997-04-29 18:18:01 +00004652 !PyCode_Check(prog) &&
4653 !PyFile_Check(prog)) {
4654 PyErr_SetString(PyExc_TypeError,
Guido van Rossumac7be682001-01-17 15:42:30 +00004655 "exec: arg 1 must be a string, file, or code object");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004656 return -1;
4657 }
Fred Drake661ea262000-10-24 19:57:45 +00004658 if (!PyDict_Check(globals)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004659 PyErr_SetString(PyExc_TypeError,
Fred Drake661ea262000-10-24 19:57:45 +00004660 "exec: arg 2 must be a dictionary or None");
4661 return -1;
4662 }
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004663 if (!PyMapping_Check(locals)) {
Fred Drake661ea262000-10-24 19:57:45 +00004664 PyErr_SetString(PyExc_TypeError,
Raymond Hettinger66bd2332004-08-02 08:30:07 +00004665 "exec: arg 3 must be a mapping or None");
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004666 return -1;
4667 }
Guido van Rossumb209a111997-04-29 18:18:01 +00004668 if (PyDict_GetItemString(globals, "__builtins__") == NULL)
Guido van Rossuma027efa1997-05-05 20:56:21 +00004669 PyDict_SetItemString(globals, "__builtins__", f->f_builtins);
Guido van Rossumb209a111997-04-29 18:18:01 +00004670 if (PyCode_Check(prog)) {
Jeremy Hylton733c8932001-12-13 19:51:56 +00004671 if (PyCode_GetNumFree((PyCodeObject *)prog) > 0) {
4672 PyErr_SetString(PyExc_TypeError,
4673 "code object passed to exec may not contain free variables");
4674 return -1;
4675 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004676 v = PyEval_EvalCode((PyCodeObject *) prog, globals, locals);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004677 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004678 else if (PyFile_Check(prog)) {
Guido van Rossumb209a111997-04-29 18:18:01 +00004679 FILE *fp = PyFile_AsFile(prog);
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004680 char *name = PyString_AsString(PyFile_Name(prog));
Jeremy Hylton714b1122007-02-25 16:01:58 +00004681 PyCompilerFlags cf;
Thomas Woutersae406c62007-09-19 17:27:43 +00004682 if (name == NULL)
4683 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004684 cf.cf_flags = 0;
4685 if (PyEval_MergeCompilerFlags(&cf))
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004686 v = PyRun_FileFlags(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004687 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004688 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004689 v = PyRun_File(fp, name, Py_file_input, globals,
Tim Peters8a5c3c72004-04-05 19:36:21 +00004690 locals);
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004691 }
4692 else {
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004693 PyObject *tmp = NULL;
Marc-André Lemburgd1ba4432000-09-19 21:04:18 +00004694 char *str;
Tim Peters5ba58662001-07-16 02:29:45 +00004695 PyCompilerFlags cf;
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004696 cf.cf_flags = 0;
4697#ifdef Py_USING_UNICODE
4698 if (PyUnicode_Check(prog)) {
4699 tmp = PyUnicode_AsUTF8String(prog);
4700 if (tmp == NULL)
4701 return -1;
4702 prog = tmp;
4703 cf.cf_flags |= PyCF_SOURCE_IS_UTF8;
4704 }
4705#endif
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004706 if (PyString_AsStringAndSize(prog, &str, NULL))
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004707 return -1;
Tim Peters5ba58662001-07-16 02:29:45 +00004708 if (PyEval_MergeCompilerFlags(&cf))
Tim Peters8a5c3c72004-04-05 19:36:21 +00004709 v = PyRun_StringFlags(str, Py_file_input, globals,
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004710 locals, &cf);
Tim Peters5ba58662001-07-16 02:29:45 +00004711 else
Jeremy Hyltonbc320242001-03-22 02:47:58 +00004712 v = PyRun_String(str, Py_file_input, globals, locals);
Just van Rossum3aaf42c2003-02-10 08:21:10 +00004713 Py_XDECREF(tmp);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004714 }
Guido van Rossuma400d8a2000-01-12 22:45:54 +00004715 if (plain)
4716 PyFrame_LocalsToFast(f, 0);
Guido van Rossum681d79a1995-07-18 14:51:37 +00004717 if (v == NULL)
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004718 return -1;
Guido van Rossumb209a111997-04-29 18:18:01 +00004719 Py_DECREF(v);
Guido van Rossumdb3165e1993-10-18 17:06:59 +00004720 return 0;
4721}
Guido van Rossum24c13741995-02-14 09:42:43 +00004722
Fredrik Lundh7a830892006-05-27 10:39:48 +00004723static void
Paul Prescode68140d2000-08-30 20:25:01 +00004724format_exc_check_arg(PyObject *exc, char *format_str, PyObject *obj)
4725{
4726 char *obj_str;
4727
4728 if (!obj)
4729 return;
4730
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004731 obj_str = PyString_AsString(obj);
Paul Prescode68140d2000-08-30 20:25:01 +00004732 if (!obj_str)
4733 return;
4734
4735 PyErr_Format(exc, format_str, obj_str);
4736}
Guido van Rossum950361c1997-01-24 13:49:28 +00004737
Fredrik Lundh7a830892006-05-27 10:39:48 +00004738static PyObject *
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004739string_concatenate(PyObject *v, PyObject *w,
4740 PyFrameObject *f, unsigned char *next_instr)
4741{
4742 /* This function implements 'variable += expr' when both arguments
4743 are strings. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004744 Py_ssize_t v_len = PyString_GET_SIZE(v);
4745 Py_ssize_t w_len = PyString_GET_SIZE(w);
Armin Rigo97ff0472006-08-09 15:37:26 +00004746 Py_ssize_t new_len = v_len + w_len;
4747 if (new_len < 0) {
4748 PyErr_SetString(PyExc_OverflowError,
4749 "strings are too large to concat");
4750 return NULL;
4751 }
Tim Peters7df5e7f2006-05-26 23:14:37 +00004752
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004753 if (v->ob_refcnt == 2) {
4754 /* In the common case, there are 2 references to the value
4755 * stored in 'variable' when the += is performed: one on the
Thomas Wouterse2176022007-09-20 17:35:10 +00004756 * value stack (in 'v') and one still stored in the
4757 * 'variable'. We try to delete the variable now to reduce
4758 * the refcnt to 1.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004759 */
4760 switch (*next_instr) {
4761 case STORE_FAST:
4762 {
4763 int oparg = PEEKARG();
4764 PyObject **fastlocals = f->f_localsplus;
4765 if (GETLOCAL(oparg) == v)
4766 SETLOCAL(oparg, NULL);
4767 break;
4768 }
4769 case STORE_DEREF:
4770 {
Thomas Wouterse2176022007-09-20 17:35:10 +00004771 PyObject **freevars = (f->f_localsplus +
4772 f->f_code->co_nlocals);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004773 PyObject *c = freevars[PEEKARG()];
4774 if (PyCell_GET(c) == v)
4775 PyCell_Set(c, NULL);
4776 break;
4777 }
4778 case STORE_NAME:
4779 {
4780 PyObject *names = f->f_code->co_names;
4781 PyObject *name = GETITEM(names, PEEKARG());
4782 PyObject *locals = f->f_locals;
4783 if (PyDict_CheckExact(locals) &&
4784 PyDict_GetItem(locals, name) == v) {
4785 if (PyDict_DelItem(locals, name) != 0) {
4786 PyErr_Clear();
4787 }
4788 }
4789 break;
4790 }
4791 }
4792 }
4793
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004794 if (v->ob_refcnt == 1 && !PyString_CHECK_INTERNED(v)) {
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004795 /* Now we own the last reference to 'v', so we can resize it
4796 * in-place.
4797 */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004798 if (_PyString_Resize(&v, new_len) != 0) {
4799 /* XXX if _PyString_Resize() fails, 'v' has been
Thomas Wouterse2176022007-09-20 17:35:10 +00004800 * deallocated so it cannot be put back into
4801 * 'variable'. The MemoryError is raised when there
4802 * is no value in 'variable', which might (very
4803 * remotely) be a cause of incompatibilities.
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004804 */
4805 return NULL;
4806 }
4807 /* copy 'w' into the newly allocated area of 'v' */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004808 memcpy(PyString_AS_STRING(v) + v_len,
4809 PyString_AS_STRING(w), w_len);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004810 return v;
4811 }
4812 else {
4813 /* When in-place resizing is not an option. */
Gregory P. Smithdd96db62008-06-09 04:58:54 +00004814 PyString_Concat(&v, w);
Raymond Hettinger52a21b82004-08-06 18:43:09 +00004815 return v;
4816 }
4817}
4818
Guido van Rossum950361c1997-01-24 13:49:28 +00004819#ifdef DYNAMIC_EXECUTION_PROFILE
4820
Fredrik Lundh7a830892006-05-27 10:39:48 +00004821static PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004822getarray(long a[256])
Guido van Rossum950361c1997-01-24 13:49:28 +00004823{
4824 int i;
4825 PyObject *l = PyList_New(256);
4826 if (l == NULL) return NULL;
4827 for (i = 0; i < 256; i++) {
4828 PyObject *x = PyInt_FromLong(a[i]);
4829 if (x == NULL) {
4830 Py_DECREF(l);
4831 return NULL;
4832 }
4833 PyList_SetItem(l, i, x);
4834 }
4835 for (i = 0; i < 256; i++)
4836 a[i] = 0;
4837 return l;
4838}
4839
4840PyObject *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00004841_Py_GetDXProfile(PyObject *self, PyObject *args)
Guido van Rossum950361c1997-01-24 13:49:28 +00004842{
4843#ifndef DXPAIRS
4844 return getarray(dxp);
4845#else
4846 int i;
4847 PyObject *l = PyList_New(257);
4848 if (l == NULL) return NULL;
4849 for (i = 0; i < 257; i++) {
4850 PyObject *x = getarray(dxpairs[i]);
4851 if (x == NULL) {
4852 Py_DECREF(l);
4853 return NULL;
4854 }
4855 PyList_SetItem(l, i, x);
4856 }
4857 return l;
4858#endif
4859}
4860
4861#endif