blob: 658ce1da52402f9c286538b7326fb63e99162b4e [file] [log] [blame]
Guido van Rossum3f5da241990-12-20 15:06:42 +00001/* Frame object implementation */
2
Guido van Rossum18752471997-04-29 14:49:28 +00003#include "Python.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00004
Jeremy Hylton3e0055f2005-10-20 19:59:25 +00005#include "code.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +00006#include "frameobject.h"
7#include "opcode.h"
8#include "structmember.h"
9
Neal Norwitz91787cb2002-12-18 23:33:35 +000010#undef MIN
11#undef MAX
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000012#define MIN(a, b) ((a) < (b) ? (a) : (b))
13#define MAX(a, b) ((a) > (b) ? (a) : (b))
14
Guido van Rossum18752471997-04-29 14:49:28 +000015#define OFF(x) offsetof(PyFrameObject, x)
Guido van Rossum3f5da241990-12-20 15:06:42 +000016
Guido van Rossum6f799372001-09-20 20:46:19 +000017static PyMemberDef frame_memberlist[] = {
Guido van Rossum33d26892007-08-05 15:29:28 +000018 {"f_back", T_OBJECT, OFF(f_back), READONLY},
19 {"f_code", T_OBJECT, OFF(f_code), READONLY},
20 {"f_builtins", T_OBJECT, OFF(f_builtins),READONLY},
21 {"f_globals", T_OBJECT, OFF(f_globals), READONLY},
22 {"f_lasti", T_INT, OFF(f_lasti), READONLY},
Guido van Rossuma027efa1997-05-05 20:56:21 +000023 {"f_exc_type", T_OBJECT, OFF(f_exc_type)},
24 {"f_exc_value", T_OBJECT, OFF(f_exc_value)},
25 {"f_exc_traceback", T_OBJECT, OFF(f_exc_traceback)},
Guido van Rossum3f5da241990-12-20 15:06:42 +000026 {NULL} /* Sentinel */
27};
28
Guido van Rossum18752471997-04-29 14:49:28 +000029static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +000030frame_getlocals(PyFrameObject *f, void *closure)
Guido van Rossum3f5da241990-12-20 15:06:42 +000031{
Tim Peters6d6c1a32001-08-02 04:15:00 +000032 PyFrame_FastToLocals(f);
33 Py_INCREF(f->f_locals);
34 return f->f_locals;
Guido van Rossum3f5da241990-12-20 15:06:42 +000035}
36
Michael W. Hudsondd32a912002-08-15 14:59:02 +000037static PyObject *
38frame_getlineno(PyFrameObject *f, void *closure)
39{
40 int lineno;
41
Michael W. Hudson02ff6a92002-09-11 15:36:32 +000042 if (f->f_trace)
43 lineno = f->f_lineno;
44 else
45 lineno = PyCode_Addr2Line(f->f_code, f->f_lasti);
Michael W. Hudsondd32a912002-08-15 14:59:02 +000046
Christian Heimes217cfd12007-12-02 14:31:20 +000047 return PyLong_FromLong(lineno);
Michael W. Hudsondd32a912002-08-15 14:59:02 +000048}
49
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000050/* Setter for f_lineno - you can set f_lineno from within a trace function in
Guido van Rossumd8faa362007-04-27 19:54:29 +000051 * order to jump to a given line of code, subject to some restrictions. Most
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000052 * lines are OK to jump to because they don't make any assumptions about the
53 * state of the stack (obvious because you could remove the line and the code
54 * would still work without any stack errors), but there are some constructs
55 * that limit jumping:
56 *
57 * o Lines with an 'except' statement on them can't be jumped to, because
58 * they expect an exception to be on the top of the stack.
59 * o Lines that live in a 'finally' block can't be jumped from or to, since
60 * the END_FINALLY expects to clean up the stack after the 'try' block.
61 * o 'try'/'for'/'while' blocks can't be jumped into because the blockstack
62 * needs to be set up before their code runs, and for 'for' loops the
63 * iterator needs to be on the stack.
64 */
65static int
66frame_setlineno(PyFrameObject *f, PyObject* p_new_lineno)
67{
68 int new_lineno = 0; /* The new value of f_lineno */
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +000069 long l_new_lineno;
70 int overflow;
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000071 int new_lasti = 0; /* The new value of f_lasti */
72 int new_iblock = 0; /* The new value of f_iblock */
Guido van Rossumd8faa362007-04-27 19:54:29 +000073 unsigned char *code = NULL; /* The bytecode for the frame... */
Martin v. Löwis18e16552006-02-15 17:27:45 +000074 Py_ssize_t code_len = 0; /* ...and its length */
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000075 char *lnotab = NULL; /* Iterating over co_lnotab */
Martin v. Löwis18e16552006-02-15 17:27:45 +000076 Py_ssize_t lnotab_len = 0; /* (ditto) */
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000077 int offset = 0; /* (ditto) */
78 int line = 0; /* (ditto) */
79 int addr = 0; /* (ditto) */
80 int min_addr = 0; /* Scanning the SETUPs and POPs */
81 int max_addr = 0; /* (ditto) */
82 int delta_iblock = 0; /* (ditto) */
83 int min_delta_iblock = 0; /* (ditto) */
84 int min_iblock = 0; /* (ditto) */
85 int f_lasti_setup_addr = 0; /* Policing no-jump-into-finally */
86 int new_lasti_setup_addr = 0; /* (ditto) */
87 int blockstack[CO_MAXBLOCKS]; /* Walking the 'finally' blocks */
88 int in_finally[CO_MAXBLOCKS]; /* (ditto) */
89 int blockstack_top = 0; /* (ditto) */
Guido van Rossumd8faa362007-04-27 19:54:29 +000090 unsigned char setup_op = 0; /* (ditto) */
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000091
92 /* f_lineno must be an integer. */
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +000093 if (!PyLong_CheckExact(p_new_lineno)) {
Michael W. Hudsoncfd38842002-12-17 16:15:34 +000094 PyErr_SetString(PyExc_ValueError,
95 "lineno must be an integer");
96 return -1;
97 }
98
99 /* You can only do this from within a trace function, not via
100 * _getframe or similar hackery. */
101 if (!f->f_trace)
102 {
103 PyErr_Format(PyExc_ValueError,
104 "f_lineno can only be set by a trace function");
105 return -1;
106 }
107
108 /* Fail if the line comes before the start of the code block. */
Martin v. Löwisd1a1d1e2007-12-04 22:10:37 +0000109 l_new_lineno = PyLong_AsLongAndOverflow(p_new_lineno, &overflow);
110 if (overflow
111#if SIZEOF_LONG > SIZEOF_INT
112 || l_new_lineno > INT_MAX
113 || l_new_lineno < INT_MIN
114#endif
115 ) {
116 PyErr_SetString(PyExc_ValueError,
117 "lineno out of range");
118 return -1;
119 }
120 new_lineno = (int)l_new_lineno;
121
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000122 if (new_lineno < f->f_code->co_firstlineno) {
123 PyErr_Format(PyExc_ValueError,
124 "line %d comes before the current code block",
125 new_lineno);
126 return -1;
127 }
128
129 /* Find the bytecode offset for the start of the given line, or the
130 * first code-owning line after it. */
131 PyString_AsStringAndSize(f->f_code->co_lnotab, &lnotab, &lnotab_len);
132 addr = 0;
133 line = f->f_code->co_firstlineno;
134 new_lasti = -1;
135 for (offset = 0; offset < lnotab_len; offset += 2) {
136 addr += lnotab[offset];
137 line += lnotab[offset+1];
138 if (line >= new_lineno) {
139 new_lasti = addr;
140 new_lineno = line;
141 break;
142 }
143 }
144
145 /* If we didn't reach the requested line, return an error. */
146 if (new_lasti == -1) {
147 PyErr_Format(PyExc_ValueError,
148 "line %d comes after the current code block",
149 new_lineno);
150 return -1;
151 }
152
153 /* We're now ready to look at the bytecode. */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000154 PyString_AsStringAndSize(f->f_code->co_code, (char **)&code, &code_len);
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000155 min_addr = MIN(new_lasti, f->f_lasti);
156 max_addr = MAX(new_lasti, f->f_lasti);
157
158 /* You can't jump onto a line with an 'except' statement on it -
159 * they expect to have an exception on the top of the stack, which
160 * won't be true if you jump to them. They always start with code
161 * that either pops the exception using POP_TOP (plain 'except:'
162 * lines do this) or duplicates the exception on the stack using
163 * DUP_TOP (if there's an exception type specified). See compile.c,
164 * 'com_try_except' for the full details. There aren't any other
165 * cases (AFAIK) where a line's code can start with DUP_TOP or
166 * POP_TOP, but if any ever appear, they'll be subject to the same
167 * restriction (but with a different error message). */
168 if (code[new_lasti] == DUP_TOP || code[new_lasti] == POP_TOP) {
169 PyErr_SetString(PyExc_ValueError,
170 "can't jump to 'except' line as there's no exception");
171 return -1;
172 }
173
174 /* You can't jump into or out of a 'finally' block because the 'try'
175 * block leaves something on the stack for the END_FINALLY to clean
Guido van Rossumd8faa362007-04-27 19:54:29 +0000176 * up. So we walk the bytecode, maintaining a simulated blockstack.
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000177 * When we reach the old or new address and it's in a 'finally' block
178 * we note the address of the corresponding SETUP_FINALLY. The jump
179 * is only legal if neither address is in a 'finally' block or
180 * they're both in the same one. 'blockstack' is a stack of the
181 * bytecode addresses of the SETUP_X opcodes, and 'in_finally' tracks
182 * whether we're in a 'finally' block at each blockstack level. */
183 f_lasti_setup_addr = -1;
184 new_lasti_setup_addr = -1;
185 memset(blockstack, '\0', sizeof(blockstack));
186 memset(in_finally, '\0', sizeof(in_finally));
187 blockstack_top = 0;
188 for (addr = 0; addr < code_len; addr++) {
189 unsigned char op = code[addr];
190 switch (op) {
191 case SETUP_LOOP:
192 case SETUP_EXCEPT:
193 case SETUP_FINALLY:
194 blockstack[blockstack_top++] = addr;
195 in_finally[blockstack_top-1] = 0;
196 break;
197
198 case POP_BLOCK:
Neal Norwitzee65e222002-12-19 18:16:57 +0000199 assert(blockstack_top > 0);
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000200 setup_op = code[blockstack[blockstack_top-1]];
201 if (setup_op == SETUP_FINALLY) {
202 in_finally[blockstack_top-1] = 1;
203 }
204 else {
205 blockstack_top--;
206 }
207 break;
208
209 case END_FINALLY:
210 /* Ignore END_FINALLYs for SETUP_EXCEPTs - they exist
211 * in the bytecode but don't correspond to an actual
Neal Norwitzee65e222002-12-19 18:16:57 +0000212 * 'finally' block. (If blockstack_top is 0, we must
213 * be seeing such an END_FINALLY.) */
214 if (blockstack_top > 0) {
215 setup_op = code[blockstack[blockstack_top-1]];
216 if (setup_op == SETUP_FINALLY) {
217 blockstack_top--;
218 }
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000219 }
220 break;
221 }
222
223 /* For the addresses we're interested in, see whether they're
224 * within a 'finally' block and if so, remember the address
225 * of the SETUP_FINALLY. */
226 if (addr == new_lasti || addr == f->f_lasti) {
227 int i = 0;
228 int setup_addr = -1;
229 for (i = blockstack_top-1; i >= 0; i--) {
230 if (in_finally[i]) {
231 setup_addr = blockstack[i];
232 break;
233 }
234 }
235
236 if (setup_addr != -1) {
237 if (addr == new_lasti) {
238 new_lasti_setup_addr = setup_addr;
239 }
240
241 if (addr == f->f_lasti) {
242 f_lasti_setup_addr = setup_addr;
243 }
244 }
245 }
246
247 if (op >= HAVE_ARGUMENT) {
248 addr += 2;
249 }
250 }
251
Neal Norwitzee65e222002-12-19 18:16:57 +0000252 /* Verify that the blockstack tracking code didn't get lost. */
253 assert(blockstack_top == 0);
254
255 /* After all that, are we jumping into / out of a 'finally' block? */
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000256 if (new_lasti_setup_addr != f_lasti_setup_addr) {
257 PyErr_SetString(PyExc_ValueError,
258 "can't jump into or out of a 'finally' block");
259 return -1;
260 }
261
262
263 /* Police block-jumping (you can't jump into the middle of a block)
264 * and ensure that the blockstack finishes up in a sensible state (by
265 * popping any blocks we're jumping out of). We look at all the
266 * blockstack operations between the current position and the new
267 * one, and keep track of how many blocks we drop out of on the way.
268 * By also keeping track of the lowest blockstack position we see, we
269 * can tell whether the jump goes into any blocks without coming out
270 * again - in that case we raise an exception below. */
271 delta_iblock = 0;
272 for (addr = min_addr; addr < max_addr; addr++) {
273 unsigned char op = code[addr];
274 switch (op) {
275 case SETUP_LOOP:
276 case SETUP_EXCEPT:
277 case SETUP_FINALLY:
278 delta_iblock++;
279 break;
280
281 case POP_BLOCK:
282 delta_iblock--;
283 break;
284 }
285
286 min_delta_iblock = MIN(min_delta_iblock, delta_iblock);
287
288 if (op >= HAVE_ARGUMENT) {
289 addr += 2;
290 }
291 }
292
293 /* Derive the absolute iblock values from the deltas. */
294 min_iblock = f->f_iblock + min_delta_iblock;
295 if (new_lasti > f->f_lasti) {
296 /* Forwards jump. */
297 new_iblock = f->f_iblock + delta_iblock;
298 }
299 else {
300 /* Backwards jump. */
301 new_iblock = f->f_iblock - delta_iblock;
302 }
303
304 /* Are we jumping into a block? */
305 if (new_iblock > min_iblock) {
306 PyErr_SetString(PyExc_ValueError,
307 "can't jump into the middle of a block");
308 return -1;
309 }
310
311 /* Pop any blocks that we're jumping out of. */
312 while (f->f_iblock > new_iblock) {
313 PyTryBlock *b = &f->f_blockstack[--f->f_iblock];
314 while ((f->f_stacktop - f->f_valuestack) > b->b_level) {
315 PyObject *v = (*--f->f_stacktop);
316 Py_DECREF(v);
317 }
318 }
319
320 /* Finally set the new f_lineno and f_lasti and return OK. */
321 f->f_lineno = new_lineno;
322 f->f_lasti = new_lasti;
323 return 0;
324}
325
Michael W. Hudson02ff6a92002-09-11 15:36:32 +0000326static PyObject *
327frame_gettrace(PyFrameObject *f, void *closure)
328{
329 PyObject* trace = f->f_trace;
330
331 if (trace == NULL)
332 trace = Py_None;
333
334 Py_INCREF(trace);
335
336 return trace;
337}
338
339static int
340frame_settrace(PyFrameObject *f, PyObject* v, void *closure)
341{
342 /* We rely on f_lineno being accurate when f_trace is set. */
343
344 PyObject* old_value = f->f_trace;
345
346 Py_XINCREF(v);
347 f->f_trace = v;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000348
Michael W. Hudson02ff6a92002-09-11 15:36:32 +0000349 if (v != NULL)
350 f->f_lineno = PyCode_Addr2Line(f->f_code, f->f_lasti);
351
352 Py_XDECREF(old_value);
353
354 return 0;
355}
356
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000357
Guido van Rossum32d34c82001-09-20 21:45:26 +0000358static PyGetSetDef frame_getsetlist[] = {
Tim Peters6d6c1a32001-08-02 04:15:00 +0000359 {"f_locals", (getter)frame_getlocals, NULL, NULL},
Michael W. Hudsoncfd38842002-12-17 16:15:34 +0000360 {"f_lineno", (getter)frame_getlineno,
361 (setter)frame_setlineno, NULL},
Michael W. Hudson02ff6a92002-09-11 15:36:32 +0000362 {"f_trace", (getter)frame_gettrace, (setter)frame_settrace, NULL},
Tim Peters6d6c1a32001-08-02 04:15:00 +0000363 {0}
364};
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000365
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000366/* Stack frames are allocated and deallocated at a considerable rate.
Thomas Wouters477c8d52006-05-27 19:21:47 +0000367 In an attempt to improve the speed of function calls, we:
368
369 1. Hold a single "zombie" frame on each code object. This retains
370 the allocated and initialised frame object from an invocation of
371 the code object. The zombie is reanimated the next time we need a
372 frame object for that code object. Doing this saves the malloc/
373 realloc required when using a free_list frame that isn't the
374 correct size. It also saves some field initialisation.
375
376 In zombie mode, no field of PyFrameObject holds a reference, but
377 the following fields are still valid:
378
379 * ob_type, ob_size, f_code, f_valuestack;
380
381 * f_locals, f_trace,
382 f_exc_type, f_exc_value, f_exc_traceback are NULL;
383
384 * f_localsplus does not require re-allocation and
385 the local variables in f_localsplus are NULL.
386
387 2. We also maintain a separate free list of stack frames (just like
388 integers are allocated in a special way -- see intobject.c). When
389 a stack frame is on the free list, only the following members have
390 a meaning:
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000391 ob_type == &Frametype
392 f_back next item on free list, or NULL
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000393 f_stacksize size of value stack
Guido van Rossumd8faa362007-04-27 19:54:29 +0000394 ob_size size of localsplus
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000395 Note that the value and block stacks are preserved -- this can save
396 another malloc() call or two (and two free() calls as well!).
397 Also note that, unlike for integers, each frame object is a
398 malloc'ed object in its own right -- it is only the actual calls to
399 malloc() that we are trying to save here, not the administration.
400 After all, while a typical program may make millions of calls, a
401 call depth of more than 20 or 30 is probably already exceptional
402 unless the program contains run-away recursion. I hope.
Tim Petersb7ba7432002-04-13 05:21:47 +0000403
Christian Heimes2202f872008-02-06 14:31:34 +0000404 Later, PyFrame_MAXFREELIST was added to bound the # of frames saved on
Tim Petersb7ba7432002-04-13 05:21:47 +0000405 free_list. Else programs creating lots of cyclic trash involving
406 frames could provoke free_list into growing without bound.
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000407*/
408
Guido van Rossum18752471997-04-29 14:49:28 +0000409static PyFrameObject *free_list = NULL;
Tim Petersb7ba7432002-04-13 05:21:47 +0000410static int numfree = 0; /* number of frames currently in free_list */
Christian Heimes2202f872008-02-06 14:31:34 +0000411/* max value for numfree */
412#define PyFrame_MAXFREELIST 200
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000413
Guido van Rossum3f5da241990-12-20 15:06:42 +0000414static void
Fred Drake1b190b42000-07-09 05:40:56 +0000415frame_dealloc(PyFrameObject *f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000416{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000417 PyObject **p, **valuestack;
418 PyCodeObject *co;
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000419
Guido van Rossumd8faa362007-04-27 19:54:29 +0000420 PyObject_GC_UnTrack(f);
Guido van Rossumd724b232000-03-13 16:01:29 +0000421 Py_TRASHCAN_SAFE_BEGIN(f)
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000422 /* Kill all local variables */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000423 valuestack = f->f_valuestack;
424 for (p = f->f_localsplus; p < valuestack; p++)
425 Py_CLEAR(*p);
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000426
Tim Peters5ca576e2001-06-18 22:08:13 +0000427 /* Free stack */
Tim Peters8c963692001-06-23 05:26:56 +0000428 if (f->f_stacktop != NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +0000429 for (p = valuestack; p < f->f_stacktop; p++)
Tim Peters8c963692001-06-23 05:26:56 +0000430 Py_XDECREF(*p);
Tim Peters5ca576e2001-06-18 22:08:13 +0000431 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000432
Guido van Rossum18752471997-04-29 14:49:28 +0000433 Py_XDECREF(f->f_back);
Neal Norwitzc91ed402002-12-30 22:29:22 +0000434 Py_DECREF(f->f_builtins);
435 Py_DECREF(f->f_globals);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000436 Py_CLEAR(f->f_locals);
437 Py_CLEAR(f->f_trace);
438 Py_CLEAR(f->f_exc_type);
439 Py_CLEAR(f->f_exc_value);
440 Py_CLEAR(f->f_exc_traceback);
441
Guido van Rossumd8faa362007-04-27 19:54:29 +0000442 co = f->f_code;
443 if (co->co_zombieframe == NULL)
444 co->co_zombieframe = f;
Christian Heimes2202f872008-02-06 14:31:34 +0000445 else if (numfree < PyFrame_MAXFREELIST) {
Tim Petersb7ba7432002-04-13 05:21:47 +0000446 ++numfree;
447 f->f_back = free_list;
448 free_list = f;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000449 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000450 else
Tim Petersb7ba7432002-04-13 05:21:47 +0000451 PyObject_GC_Del(f);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000452
Guido van Rossumd8faa362007-04-27 19:54:29 +0000453 Py_DECREF(co);
Guido van Rossumd724b232000-03-13 16:01:29 +0000454 Py_TRASHCAN_SAFE_END(f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000455}
456
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000457static int
458frame_traverse(PyFrameObject *f, visitproc visit, void *arg)
459{
460 PyObject **fastlocals, **p;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000461 int i, slots;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000462
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000463 Py_VISIT(f->f_back);
464 Py_VISIT(f->f_code);
465 Py_VISIT(f->f_builtins);
466 Py_VISIT(f->f_globals);
467 Py_VISIT(f->f_locals);
468 Py_VISIT(f->f_trace);
469 Py_VISIT(f->f_exc_type);
470 Py_VISIT(f->f_exc_value);
471 Py_VISIT(f->f_exc_traceback);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000472
473 /* locals */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000474 slots = f->f_code->co_nlocals + PyTuple_GET_SIZE(f->f_code->co_cellvars) + PyTuple_GET_SIZE(f->f_code->co_freevars);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000475 fastlocals = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000476 for (i = slots; --i >= 0; ++fastlocals)
477 Py_VISIT(*fastlocals);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000478
479 /* stack */
480 if (f->f_stacktop != NULL) {
481 for (p = f->f_valuestack; p < f->f_stacktop; p++)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000482 Py_VISIT(*p);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000483 }
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000484 return 0;
485}
486
487static void
488frame_clear(PyFrameObject *f)
489{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000490 PyObject **fastlocals, **p, **oldtop;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000491 int i, slots;
492
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000493 /* Before anything else, make sure that this frame is clearly marked
Guido van Rossumd8faa362007-04-27 19:54:29 +0000494 * as being defunct! Else, e.g., a generator reachable from this
495 * frame may also point to this frame, believe itself to still be
496 * active, and try cleaning up this frame again.
497 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000498 oldtop = f->f_stacktop;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000499 f->f_stacktop = NULL;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000500
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000501 Py_CLEAR(f->f_exc_type);
502 Py_CLEAR(f->f_exc_value);
503 Py_CLEAR(f->f_exc_traceback);
504 Py_CLEAR(f->f_trace);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000505
506 /* locals */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000507 slots = f->f_code->co_nlocals + PyTuple_GET_SIZE(f->f_code->co_cellvars) + PyTuple_GET_SIZE(f->f_code->co_freevars);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000508 fastlocals = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000509 for (i = slots; --i >= 0; ++fastlocals)
510 Py_CLEAR(*fastlocals);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000511
512 /* stack */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000513 if (oldtop != NULL) {
514 for (p = f->f_valuestack; p < oldtop; p++)
515 Py_CLEAR(*p);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000516 }
517}
518
519
Guido van Rossum18752471997-04-29 14:49:28 +0000520PyTypeObject PyFrame_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000521 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000522 "frame",
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000523 sizeof(PyFrameObject),
524 sizeof(PyObject *),
Guido van Rossumd8faa362007-04-27 19:54:29 +0000525 (destructor)frame_dealloc, /* tp_dealloc */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000526 0, /* tp_print */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000527 0, /* tp_getattr */
528 0, /* tp_setattr */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000529 0, /* tp_compare */
530 0, /* tp_repr */
531 0, /* tp_as_number */
532 0, /* tp_as_sequence */
533 0, /* tp_as_mapping */
534 0, /* tp_hash */
535 0, /* tp_call */
536 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000537 PyObject_GenericGetAttr, /* tp_getattro */
538 PyObject_GenericSetAttr, /* tp_setattro */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000539 0, /* tp_as_buffer */
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000540 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000541 0, /* tp_doc */
542 (traverseproc)frame_traverse, /* tp_traverse */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000543 (inquiry)frame_clear, /* tp_clear */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000544 0, /* tp_richcompare */
545 0, /* tp_weaklistoffset */
546 0, /* tp_iter */
547 0, /* tp_iternext */
548 0, /* tp_methods */
549 frame_memberlist, /* tp_members */
550 frame_getsetlist, /* tp_getset */
551 0, /* tp_base */
552 0, /* tp_dict */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000553};
554
Neal Norwitzc91ed402002-12-30 22:29:22 +0000555static PyObject *builtin_object;
556
Neal Norwitzb2501f42002-12-31 03:42:13 +0000557int _PyFrame_Init()
Neal Norwitzc91ed402002-12-30 22:29:22 +0000558{
Martin v. Löwis5b222132007-06-10 09:51:05 +0000559 builtin_object = PyUnicode_InternFromString("__builtins__");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000560 return (builtin_object != NULL);
561}
562
Guido van Rossum18752471997-04-29 14:49:28 +0000563PyFrameObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000564PyFrame_New(PyThreadState *tstate, PyCodeObject *code, PyObject *globals,
Jeremy Hylton30c9f392001-03-13 01:58:22 +0000565 PyObject *locals)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000566{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000567 PyFrameObject *back = tstate->frame;
Guido van Rossum18752471997-04-29 14:49:28 +0000568 PyFrameObject *f;
569 PyObject *builtins;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000570 Py_ssize_t i;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000571
Michael W. Hudson69734a52002-08-19 16:54:08 +0000572#ifdef Py_DEBUG
573 if (code == NULL || globals == NULL || !PyDict_Check(globals) ||
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000574 (locals != NULL && !PyMapping_Check(locals))) {
Guido van Rossum18752471997-04-29 14:49:28 +0000575 PyErr_BadInternalCall();
Guido van Rossum3f5da241990-12-20 15:06:42 +0000576 return NULL;
577 }
Michael W. Hudson69734a52002-08-19 16:54:08 +0000578#endif
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000579 if (back == NULL || back->f_globals != globals) {
580 builtins = PyDict_GetItem(globals, builtin_object);
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000581 if (builtins) {
582 if (PyModule_Check(builtins)) {
583 builtins = PyModule_GetDict(builtins);
584 assert(!builtins || PyDict_Check(builtins));
585 }
586 else if (!PyDict_Check(builtins))
587 builtins = NULL;
588 }
589 if (builtins == NULL) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000590 /* No builtins! Make up a minimal one
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000591 Give them 'None', at least. */
592 builtins = PyDict_New();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000593 if (builtins == NULL ||
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000594 PyDict_SetItemString(
595 builtins, "None", Py_None) < 0)
596 return NULL;
597 }
598 else
599 Py_INCREF(builtins);
600
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000601 }
602 else {
603 /* If we share the globals, we share the builtins.
604 Save a lookup and a call. */
605 builtins = back->f_builtins;
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000606 assert(builtins != NULL && PyDict_Check(builtins));
607 Py_INCREF(builtins);
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000608 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000609 if (code->co_zombieframe != NULL) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000610 f = code->co_zombieframe;
611 code->co_zombieframe = NULL;
612 _Py_NewReference((PyObject *)f);
613 assert(f->f_code == code);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000614 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000615 else {
616 Py_ssize_t extras, ncells, nfrees;
617 ncells = PyTuple_GET_SIZE(code->co_cellvars);
618 nfrees = PyTuple_GET_SIZE(code->co_freevars);
619 extras = code->co_stacksize + code->co_nlocals + ncells +
620 nfrees;
621 if (free_list == NULL) {
622 f = PyObject_GC_NewVar(PyFrameObject, &PyFrame_Type,
623 extras);
624 if (f == NULL) {
625 Py_DECREF(builtins);
626 return NULL;
627 }
628 }
629 else {
630 assert(numfree > 0);
631 --numfree;
632 f = free_list;
633 free_list = free_list->f_back;
Christian Heimes90aa7642007-12-19 02:45:37 +0000634 if (Py_SIZE(f) < extras) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000635 f = PyObject_GC_Resize(PyFrameObject, f, extras);
636 if (f == NULL) {
637 Py_DECREF(builtins);
638 return NULL;
639 }
640 }
641 _Py_NewReference((PyObject *)f);
642 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000643
644 f->f_code = code;
645 extras = code->co_nlocals + ncells + nfrees;
646 f->f_valuestack = f->f_localsplus + extras;
647 for (i=0; i<extras; i++)
648 f->f_localsplus[i] = NULL;
649 f->f_locals = NULL;
650 f->f_trace = NULL;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000651 f->f_exc_type = f->f_exc_value = f->f_exc_traceback = NULL;
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000652 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000653 f->f_stacktop = f->f_valuestack;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000654 f->f_builtins = builtins;
Guido van Rossum18752471997-04-29 14:49:28 +0000655 Py_XINCREF(back);
Guido van Rossum2271bf71995-07-18 14:30:34 +0000656 f->f_back = back;
Guido van Rossum18752471997-04-29 14:49:28 +0000657 Py_INCREF(code);
Guido van Rossum18752471997-04-29 14:49:28 +0000658 Py_INCREF(globals);
Guido van Rossum2271bf71995-07-18 14:30:34 +0000659 f->f_globals = globals;
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000660 /* Most functions have CO_NEWLOCALS and CO_OPTIMIZED set. */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000661 if ((code->co_flags & (CO_NEWLOCALS | CO_OPTIMIZED)) ==
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000662 (CO_NEWLOCALS | CO_OPTIMIZED))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000663 ; /* f_locals = NULL; will be set by PyFrame_FastToLocals() */
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000664 else if (code->co_flags & CO_NEWLOCALS) {
665 locals = PyDict_New();
666 if (locals == NULL) {
667 Py_DECREF(f);
668 return NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000669 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000670 f->f_locals = locals;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000671 }
Guido van Rossum2271bf71995-07-18 14:30:34 +0000672 else {
673 if (locals == NULL)
674 locals = globals;
Guido van Rossum18752471997-04-29 14:49:28 +0000675 Py_INCREF(locals);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000676 f->f_locals = locals;
Guido van Rossum2271bf71995-07-18 14:30:34 +0000677 }
Guido van Rossumeb46d671997-08-02 02:59:08 +0000678 f->f_tstate = tstate;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000679
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000680 f->f_lasti = -1;
Guido van Rossum747596a1997-01-24 04:00:21 +0000681 f->f_lineno = code->co_firstlineno;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000682 f->f_iblock = 0;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000683
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000684 _PyObject_GC_TRACK(f);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000685 return f;
686}
687
Guido van Rossum3f5da241990-12-20 15:06:42 +0000688/* Block management */
689
690void
Fred Drake1b190b42000-07-09 05:40:56 +0000691PyFrame_BlockSetup(PyFrameObject *f, int type, int handler, int level)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000692{
Guido van Rossum18752471997-04-29 14:49:28 +0000693 PyTryBlock *b;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000694 if (f->f_iblock >= CO_MAXBLOCKS)
Guido van Rossum18752471997-04-29 14:49:28 +0000695 Py_FatalError("XXX block stack overflow");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000696 b = &f->f_blockstack[f->f_iblock++];
697 b->b_type = type;
698 b->b_level = level;
699 b->b_handler = handler;
700}
701
Guido van Rossum18752471997-04-29 14:49:28 +0000702PyTryBlock *
Fred Drake1b190b42000-07-09 05:40:56 +0000703PyFrame_BlockPop(PyFrameObject *f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000704{
Guido van Rossum18752471997-04-29 14:49:28 +0000705 PyTryBlock *b;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000706 if (f->f_iblock <= 0)
Guido van Rossum18752471997-04-29 14:49:28 +0000707 Py_FatalError("XXX block stack underflow");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000708 b = &f->f_blockstack[--f->f_iblock];
709 return b;
710}
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000711
Guido van Rossumd8faa362007-04-27 19:54:29 +0000712/* Convert between "fast" version of locals and dictionary version.
713
714 map and values are input arguments. map is a tuple of strings.
715 values is an array of PyObject*. At index i, map[i] is the name of
716 the variable with value values[i]. The function copies the first
717 nmap variable from map/values into dict. If values[i] is NULL,
718 the variable is deleted from dict.
719
720 If deref is true, then the values being copied are cell variables
721 and the value is extracted from the cell variable before being put
722 in dict.
723
724 Exceptions raised while modifying the dict are silently ignored,
725 because there is no good way to report them.
726 */
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000727
Guido van Rossumf68d8e52001-04-14 17:55:09 +0000728static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000729map_to_dict(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000730 int deref)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000731{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000732 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000733 assert(PyTuple_Check(map));
734 assert(PyDict_Check(dict));
735 assert(PyTuple_Size(map) >= nmap);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000736 for (j = nmap; --j >= 0; ) {
Jeremy Hylton1a48ca82001-12-06 15:48:16 +0000737 PyObject *key = PyTuple_GET_ITEM(map, j);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000738 PyObject *value = values[j];
Neal Norwitzbed67842007-10-27 04:00:45 +0000739 assert(PyUnicode_Check(key));
Guido van Rossumd8faa362007-04-27 19:54:29 +0000740 if (deref) {
741 assert(PyCell_Check(value));
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000742 value = PyCell_GET(value);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000743 }
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000744 if (value == NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000745 if (PyObject_DelItem(dict, key) != 0)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000746 PyErr_Clear();
747 }
748 else {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000749 if (PyObject_SetItem(dict, key, value) != 0)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000750 PyErr_Clear();
751 }
752 }
753}
754
Guido van Rossumd8faa362007-04-27 19:54:29 +0000755/* Copy values from the "locals" dict into the fast locals.
756
757 dict is an input argument containing string keys representing
758 variables names and arbitrary PyObject* as values.
759
760 map and values are input arguments. map is a tuple of strings.
761 values is an array of PyObject*. At index i, map[i] is the name of
762 the variable with value values[i]. The function copies the first
763 nmap variable from map/values into dict. If values[i] is NULL,
764 the variable is deleted from dict.
765
766 If deref is true, then the values being copied are cell variables
767 and the value is extracted from the cell variable before being put
768 in dict. If clear is true, then variables in map but not in dict
769 are set to NULL in map; if clear is false, variables missing in
770 dict are ignored.
771
772 Exceptions raised while modifying the dict are silently ignored,
773 because there is no good way to report them.
774*/
775
Guido van Rossum6b356e72001-04-14 17:55:41 +0000776static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000777dict_to_map(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000778 int deref, int clear)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000779{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000780 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000781 assert(PyTuple_Check(map));
782 assert(PyDict_Check(dict));
783 assert(PyTuple_Size(map) >= nmap);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000784 for (j = nmap; --j >= 0; ) {
Jeremy Hylton1a48ca82001-12-06 15:48:16 +0000785 PyObject *key = PyTuple_GET_ITEM(map, j);
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000786 PyObject *value = PyObject_GetItem(dict, key);
Martin v. Löwis5b222132007-06-10 09:51:05 +0000787 assert(PyUnicode_Check(key));
Guido van Rossumd8faa362007-04-27 19:54:29 +0000788 /* We only care about NULLs if clear is true. */
789 if (value == NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000790 PyErr_Clear();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000791 if (!clear)
792 continue;
793 }
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000794 if (deref) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000795 assert(PyCell_Check(values[j]));
796 if (PyCell_GET(values[j]) != value) {
797 if (PyCell_Set(values[j], value) < 0)
798 PyErr_Clear();
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000799 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000800 } else if (values[j] != value) {
801 Py_XINCREF(value);
802 Py_XDECREF(values[j]);
803 values[j] = value;
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000804 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000805 Py_XDECREF(value);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000806 }
807}
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000808
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000809void
Fred Drake1b190b42000-07-09 05:40:56 +0000810PyFrame_FastToLocals(PyFrameObject *f)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000811{
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000812 /* Merge fast locals into f->f_locals */
Guido van Rossum18752471997-04-29 14:49:28 +0000813 PyObject *locals, *map;
814 PyObject **fast;
815 PyObject *error_type, *error_value, *error_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000816 PyCodeObject *co;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000817 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000818 int ncells, nfreevars;
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000819 if (f == NULL)
820 return;
Guido van Rossum2271bf71995-07-18 14:30:34 +0000821 locals = f->f_locals;
822 if (locals == NULL) {
Guido van Rossum18752471997-04-29 14:49:28 +0000823 locals = f->f_locals = PyDict_New();
Guido van Rossum2271bf71995-07-18 14:30:34 +0000824 if (locals == NULL) {
Guido van Rossum18752471997-04-29 14:49:28 +0000825 PyErr_Clear(); /* Can't report it :-( */
Guido van Rossum2271bf71995-07-18 14:30:34 +0000826 return;
827 }
828 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000829 co = f->f_code;
830 map = co->co_varnames;
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000831 if (!PyTuple_Check(map))
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000832 return;
Guido van Rossum18752471997-04-29 14:49:28 +0000833 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000834 fast = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000835 j = PyTuple_GET_SIZE(map);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000836 if (j > co->co_nlocals)
837 j = co->co_nlocals;
838 if (co->co_nlocals)
Jeremy Hylton174d2762003-10-21 18:10:28 +0000839 map_to_dict(map, j, locals, fast, 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000840 ncells = PyTuple_GET_SIZE(co->co_cellvars);
841 nfreevars = PyTuple_GET_SIZE(co->co_freevars);
842 if (ncells || nfreevars) {
843 map_to_dict(co->co_cellvars, ncells,
844 locals, fast + co->co_nlocals, 1);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000845 /* If the namespace is unoptimized, then one of the
846 following cases applies:
847 1. It does not contain free variables, because it
848 uses import * or is a top-level namespace.
849 2. It is a class namespace.
850 We don't want to accidentally copy free variables
851 into the locals dict used by the class.
852 */
853 if (co->co_flags & CO_OPTIMIZED) {
854 map_to_dict(co->co_freevars, nfreevars,
855 locals, fast + co->co_nlocals + ncells, 1);
856 }
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000857 }
Guido van Rossum18752471997-04-29 14:49:28 +0000858 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000859}
860
861void
Fred Drake1b190b42000-07-09 05:40:56 +0000862PyFrame_LocalsToFast(PyFrameObject *f, int clear)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000863{
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000864 /* Merge f->f_locals into fast locals */
Guido van Rossum18752471997-04-29 14:49:28 +0000865 PyObject *locals, *map;
866 PyObject **fast;
867 PyObject *error_type, *error_value, *error_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000868 PyCodeObject *co;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000869 Py_ssize_t j;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000870 int ncells, nfreevars;
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000871 if (f == NULL)
872 return;
873 locals = f->f_locals;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000874 co = f->f_code;
875 map = co->co_varnames;
Jeremy Hylton24ea8d32002-04-20 04:46:55 +0000876 if (locals == NULL)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000877 return;
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000878 if (!PyTuple_Check(map))
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000879 return;
Guido van Rossum18752471997-04-29 14:49:28 +0000880 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000881 fast = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000882 j = PyTuple_GET_SIZE(map);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000883 if (j > co->co_nlocals)
884 j = co->co_nlocals;
885 if (co->co_nlocals)
886 dict_to_map(co->co_varnames, j, locals, fast, 0, clear);
887 ncells = PyTuple_GET_SIZE(co->co_cellvars);
888 nfreevars = PyTuple_GET_SIZE(co->co_freevars);
889 if (ncells || nfreevars) {
890 dict_to_map(co->co_cellvars, ncells,
891 locals, fast + co->co_nlocals, 1, clear);
892 dict_to_map(co->co_freevars, nfreevars,
893 locals, fast + co->co_nlocals + ncells, 1,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000894 clear);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000895 }
Guido van Rossum18752471997-04-29 14:49:28 +0000896 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000897}
Guido van Rossum404b95d1997-08-05 02:09:46 +0000898
899/* Clear out the free list */
900
901void
Fred Drake1b190b42000-07-09 05:40:56 +0000902PyFrame_Fini(void)
Guido van Rossum404b95d1997-08-05 02:09:46 +0000903{
904 while (free_list != NULL) {
905 PyFrameObject *f = free_list;
906 free_list = free_list->f_back;
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000907 PyObject_GC_Del(f);
Tim Petersb7ba7432002-04-13 05:21:47 +0000908 --numfree;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000909 }
Tim Petersb7ba7432002-04-13 05:21:47 +0000910 assert(numfree == 0);
Neal Norwitzc91ed402002-12-30 22:29:22 +0000911 Py_XDECREF(builtin_object);
912 builtin_object = NULL;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000913}