blob: 266cbd2caa135a7583d5251a1a7ea93900b0e391 [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
404 Later, MAXFREELIST was added to bound the # of frames saved on
405 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 */
411#define MAXFREELIST 200 /* max value for numfree */
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000412
Guido van Rossum3f5da241990-12-20 15:06:42 +0000413static void
Fred Drake1b190b42000-07-09 05:40:56 +0000414frame_dealloc(PyFrameObject *f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000415{
Thomas Wouters477c8d52006-05-27 19:21:47 +0000416 PyObject **p, **valuestack;
417 PyCodeObject *co;
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000418
Guido van Rossumd8faa362007-04-27 19:54:29 +0000419 PyObject_GC_UnTrack(f);
Guido van Rossumd724b232000-03-13 16:01:29 +0000420 Py_TRASHCAN_SAFE_BEGIN(f)
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000421 /* Kill all local variables */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000422 valuestack = f->f_valuestack;
423 for (p = f->f_localsplus; p < valuestack; p++)
424 Py_CLEAR(*p);
Guido van Rossum7582bfb1997-02-14 16:27:29 +0000425
Tim Peters5ca576e2001-06-18 22:08:13 +0000426 /* Free stack */
Tim Peters8c963692001-06-23 05:26:56 +0000427 if (f->f_stacktop != NULL) {
Thomas Wouters477c8d52006-05-27 19:21:47 +0000428 for (p = valuestack; p < f->f_stacktop; p++)
Tim Peters8c963692001-06-23 05:26:56 +0000429 Py_XDECREF(*p);
Tim Peters5ca576e2001-06-18 22:08:13 +0000430 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000431
Guido van Rossum18752471997-04-29 14:49:28 +0000432 Py_XDECREF(f->f_back);
Neal Norwitzc91ed402002-12-30 22:29:22 +0000433 Py_DECREF(f->f_builtins);
434 Py_DECREF(f->f_globals);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000435 Py_CLEAR(f->f_locals);
436 Py_CLEAR(f->f_trace);
437 Py_CLEAR(f->f_exc_type);
438 Py_CLEAR(f->f_exc_value);
439 Py_CLEAR(f->f_exc_traceback);
440
Guido van Rossumd8faa362007-04-27 19:54:29 +0000441 co = f->f_code;
442 if (co->co_zombieframe == NULL)
443 co->co_zombieframe = f;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000444 else if (numfree < MAXFREELIST) {
Tim Petersb7ba7432002-04-13 05:21:47 +0000445 ++numfree;
446 f->f_back = free_list;
447 free_list = f;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000448 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000449 else
Tim Petersb7ba7432002-04-13 05:21:47 +0000450 PyObject_GC_Del(f);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000451
Guido van Rossumd8faa362007-04-27 19:54:29 +0000452 Py_DECREF(co);
Guido van Rossumd724b232000-03-13 16:01:29 +0000453 Py_TRASHCAN_SAFE_END(f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000454}
455
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000456static int
457frame_traverse(PyFrameObject *f, visitproc visit, void *arg)
458{
459 PyObject **fastlocals, **p;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000460 int i, slots;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000461
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000462 Py_VISIT(f->f_back);
463 Py_VISIT(f->f_code);
464 Py_VISIT(f->f_builtins);
465 Py_VISIT(f->f_globals);
466 Py_VISIT(f->f_locals);
467 Py_VISIT(f->f_trace);
468 Py_VISIT(f->f_exc_type);
469 Py_VISIT(f->f_exc_value);
470 Py_VISIT(f->f_exc_traceback);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000471
472 /* locals */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000473 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 +0000474 fastlocals = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000475 for (i = slots; --i >= 0; ++fastlocals)
476 Py_VISIT(*fastlocals);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000477
478 /* stack */
479 if (f->f_stacktop != NULL) {
480 for (p = f->f_valuestack; p < f->f_stacktop; p++)
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000481 Py_VISIT(*p);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000482 }
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000483 return 0;
484}
485
486static void
487frame_clear(PyFrameObject *f)
488{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000489 PyObject **fastlocals, **p, **oldtop;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000490 int i, slots;
491
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000492 /* Before anything else, make sure that this frame is clearly marked
Guido van Rossumd8faa362007-04-27 19:54:29 +0000493 * as being defunct! Else, e.g., a generator reachable from this
494 * frame may also point to this frame, believe itself to still be
495 * active, and try cleaning up this frame again.
496 */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000497 oldtop = f->f_stacktop;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000498 f->f_stacktop = NULL;
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000499
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000500 Py_CLEAR(f->f_exc_type);
501 Py_CLEAR(f->f_exc_value);
502 Py_CLEAR(f->f_exc_traceback);
503 Py_CLEAR(f->f_trace);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000504
505 /* locals */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000506 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 +0000507 fastlocals = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000508 for (i = slots; --i >= 0; ++fastlocals)
509 Py_CLEAR(*fastlocals);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000510
511 /* stack */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000512 if (oldtop != NULL) {
513 for (p = f->f_valuestack; p < oldtop; p++)
514 Py_CLEAR(*p);
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000515 }
516}
517
518
Guido van Rossum18752471997-04-29 14:49:28 +0000519PyTypeObject PyFrame_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000520 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000521 "frame",
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000522 sizeof(PyFrameObject),
523 sizeof(PyObject *),
Guido van Rossumd8faa362007-04-27 19:54:29 +0000524 (destructor)frame_dealloc, /* tp_dealloc */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000525 0, /* tp_print */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000526 0, /* tp_getattr */
527 0, /* tp_setattr */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000528 0, /* tp_compare */
529 0, /* tp_repr */
530 0, /* tp_as_number */
531 0, /* tp_as_sequence */
532 0, /* tp_as_mapping */
533 0, /* tp_hash */
534 0, /* tp_call */
535 0, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000536 PyObject_GenericGetAttr, /* tp_getattro */
537 PyObject_GenericSetAttr, /* tp_setattro */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000538 0, /* tp_as_buffer */
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000539 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
Guido van Rossumd8faa362007-04-27 19:54:29 +0000540 0, /* tp_doc */
541 (traverseproc)frame_traverse, /* tp_traverse */
Neil Schemenauer19cd2922001-07-12 13:27:11 +0000542 (inquiry)frame_clear, /* tp_clear */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000543 0, /* tp_richcompare */
544 0, /* tp_weaklistoffset */
545 0, /* tp_iter */
546 0, /* tp_iternext */
547 0, /* tp_methods */
548 frame_memberlist, /* tp_members */
549 frame_getsetlist, /* tp_getset */
550 0, /* tp_base */
551 0, /* tp_dict */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000552};
553
Neal Norwitzc91ed402002-12-30 22:29:22 +0000554static PyObject *builtin_object;
555
Neal Norwitzb2501f42002-12-31 03:42:13 +0000556int _PyFrame_Init()
Neal Norwitzc91ed402002-12-30 22:29:22 +0000557{
Martin v. Löwis5b222132007-06-10 09:51:05 +0000558 builtin_object = PyUnicode_InternFromString("__builtins__");
Neal Norwitzc91ed402002-12-30 22:29:22 +0000559 return (builtin_object != NULL);
560}
561
Guido van Rossum18752471997-04-29 14:49:28 +0000562PyFrameObject *
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000563PyFrame_New(PyThreadState *tstate, PyCodeObject *code, PyObject *globals,
Jeremy Hylton30c9f392001-03-13 01:58:22 +0000564 PyObject *locals)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000565{
Guido van Rossuma027efa1997-05-05 20:56:21 +0000566 PyFrameObject *back = tstate->frame;
Guido van Rossum18752471997-04-29 14:49:28 +0000567 PyFrameObject *f;
568 PyObject *builtins;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000569 Py_ssize_t i;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000570
Michael W. Hudson69734a52002-08-19 16:54:08 +0000571#ifdef Py_DEBUG
572 if (code == NULL || globals == NULL || !PyDict_Check(globals) ||
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000573 (locals != NULL && !PyMapping_Check(locals))) {
Guido van Rossum18752471997-04-29 14:49:28 +0000574 PyErr_BadInternalCall();
Guido van Rossum3f5da241990-12-20 15:06:42 +0000575 return NULL;
576 }
Michael W. Hudson69734a52002-08-19 16:54:08 +0000577#endif
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000578 if (back == NULL || back->f_globals != globals) {
579 builtins = PyDict_GetItem(globals, builtin_object);
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000580 if (builtins) {
581 if (PyModule_Check(builtins)) {
582 builtins = PyModule_GetDict(builtins);
583 assert(!builtins || PyDict_Check(builtins));
584 }
585 else if (!PyDict_Check(builtins))
586 builtins = NULL;
587 }
588 if (builtins == NULL) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000589 /* No builtins! Make up a minimal one
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000590 Give them 'None', at least. */
591 builtins = PyDict_New();
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000592 if (builtins == NULL ||
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000593 PyDict_SetItemString(
594 builtins, "None", Py_None) < 0)
595 return NULL;
596 }
597 else
598 Py_INCREF(builtins);
599
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000600 }
601 else {
602 /* If we share the globals, we share the builtins.
603 Save a lookup and a call. */
604 builtins = back->f_builtins;
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000605 assert(builtins != NULL && PyDict_Check(builtins));
606 Py_INCREF(builtins);
Guido van Rossumbde6ff71998-02-19 20:48:26 +0000607 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000608 if (code->co_zombieframe != NULL) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000609 f = code->co_zombieframe;
610 code->co_zombieframe = NULL;
611 _Py_NewReference((PyObject *)f);
612 assert(f->f_code == code);
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000613 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000614 else {
615 Py_ssize_t extras, ncells, nfrees;
616 ncells = PyTuple_GET_SIZE(code->co_cellvars);
617 nfrees = PyTuple_GET_SIZE(code->co_freevars);
618 extras = code->co_stacksize + code->co_nlocals + ncells +
619 nfrees;
620 if (free_list == NULL) {
621 f = PyObject_GC_NewVar(PyFrameObject, &PyFrame_Type,
622 extras);
623 if (f == NULL) {
624 Py_DECREF(builtins);
625 return NULL;
626 }
627 }
628 else {
629 assert(numfree > 0);
630 --numfree;
631 f = free_list;
632 free_list = free_list->f_back;
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000633 if (Py_Size(f) < extras) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000634 f = PyObject_GC_Resize(PyFrameObject, f, extras);
635 if (f == NULL) {
636 Py_DECREF(builtins);
637 return NULL;
638 }
639 }
640 _Py_NewReference((PyObject *)f);
641 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000642
643 f->f_code = code;
644 extras = code->co_nlocals + ncells + nfrees;
645 f->f_valuestack = f->f_localsplus + extras;
646 for (i=0; i<extras; i++)
647 f->f_localsplus[i] = NULL;
648 f->f_locals = NULL;
649 f->f_trace = NULL;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000650 f->f_exc_type = f->f_exc_value = f->f_exc_traceback = NULL;
Guido van Rossuma9e7dc11992-10-18 18:53:57 +0000651 }
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000652 f->f_stacktop = f->f_valuestack;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000653 f->f_builtins = builtins;
Guido van Rossum18752471997-04-29 14:49:28 +0000654 Py_XINCREF(back);
Guido van Rossum2271bf71995-07-18 14:30:34 +0000655 f->f_back = back;
Guido van Rossum18752471997-04-29 14:49:28 +0000656 Py_INCREF(code);
Guido van Rossum18752471997-04-29 14:49:28 +0000657 Py_INCREF(globals);
Guido van Rossum2271bf71995-07-18 14:30:34 +0000658 f->f_globals = globals;
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000659 /* Most functions have CO_NEWLOCALS and CO_OPTIMIZED set. */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000660 if ((code->co_flags & (CO_NEWLOCALS | CO_OPTIMIZED)) ==
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000661 (CO_NEWLOCALS | CO_OPTIMIZED))
Thomas Wouters477c8d52006-05-27 19:21:47 +0000662 ; /* f_locals = NULL; will be set by PyFrame_FastToLocals() */
Jeremy Hyltonbd5cbf82003-02-05 22:39:29 +0000663 else if (code->co_flags & CO_NEWLOCALS) {
664 locals = PyDict_New();
665 if (locals == NULL) {
666 Py_DECREF(f);
667 return NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000668 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000669 f->f_locals = locals;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000670 }
Guido van Rossum2271bf71995-07-18 14:30:34 +0000671 else {
672 if (locals == NULL)
673 locals = globals;
Guido van Rossum18752471997-04-29 14:49:28 +0000674 Py_INCREF(locals);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000675 f->f_locals = locals;
Guido van Rossum2271bf71995-07-18 14:30:34 +0000676 }
Guido van Rossumeb46d671997-08-02 02:59:08 +0000677 f->f_tstate = tstate;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000678
Michael W. Hudsondd32a912002-08-15 14:59:02 +0000679 f->f_lasti = -1;
Guido van Rossum747596a1997-01-24 04:00:21 +0000680 f->f_lineno = code->co_firstlineno;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000681 f->f_iblock = 0;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000682
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000683 _PyObject_GC_TRACK(f);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000684 return f;
685}
686
Guido van Rossum3f5da241990-12-20 15:06:42 +0000687/* Block management */
688
689void
Fred Drake1b190b42000-07-09 05:40:56 +0000690PyFrame_BlockSetup(PyFrameObject *f, int type, int handler, int level)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000691{
Guido van Rossum18752471997-04-29 14:49:28 +0000692 PyTryBlock *b;
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000693 if (f->f_iblock >= CO_MAXBLOCKS)
Guido van Rossum18752471997-04-29 14:49:28 +0000694 Py_FatalError("XXX block stack overflow");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000695 b = &f->f_blockstack[f->f_iblock++];
696 b->b_type = type;
697 b->b_level = level;
698 b->b_handler = handler;
699}
700
Guido van Rossum18752471997-04-29 14:49:28 +0000701PyTryBlock *
Fred Drake1b190b42000-07-09 05:40:56 +0000702PyFrame_BlockPop(PyFrameObject *f)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000703{
Guido van Rossum18752471997-04-29 14:49:28 +0000704 PyTryBlock *b;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000705 if (f->f_iblock <= 0)
Guido van Rossum18752471997-04-29 14:49:28 +0000706 Py_FatalError("XXX block stack underflow");
Guido van Rossum3f5da241990-12-20 15:06:42 +0000707 b = &f->f_blockstack[--f->f_iblock];
708 return b;
709}
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000710
Guido van Rossumd8faa362007-04-27 19:54:29 +0000711/* Convert between "fast" version of locals and dictionary version.
712
713 map and values are input arguments. map is a tuple of strings.
714 values is an array of PyObject*. At index i, map[i] is the name of
715 the variable with value values[i]. The function copies the first
716 nmap variable from map/values into dict. If values[i] is NULL,
717 the variable is deleted from dict.
718
719 If deref is true, then the values being copied are cell variables
720 and the value is extracted from the cell variable before being put
721 in dict.
722
723 Exceptions raised while modifying the dict are silently ignored,
724 because there is no good way to report them.
725 */
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000726
Guido van Rossumf68d8e52001-04-14 17:55:09 +0000727static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000728map_to_dict(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000729 int deref)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000730{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000731 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000732 assert(PyTuple_Check(map));
733 assert(PyDict_Check(dict));
734 assert(PyTuple_Size(map) >= nmap);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000735 for (j = nmap; --j >= 0; ) {
Jeremy Hylton1a48ca82001-12-06 15:48:16 +0000736 PyObject *key = PyTuple_GET_ITEM(map, j);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000737 PyObject *value = values[j];
Neal Norwitzbed67842007-10-27 04:00:45 +0000738 assert(PyUnicode_Check(key));
Guido van Rossumd8faa362007-04-27 19:54:29 +0000739 if (deref) {
740 assert(PyCell_Check(value));
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000741 value = PyCell_GET(value);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000742 }
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000743 if (value == NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000744 if (PyObject_DelItem(dict, key) != 0)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000745 PyErr_Clear();
746 }
747 else {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000748 if (PyObject_SetItem(dict, key, value) != 0)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000749 PyErr_Clear();
750 }
751 }
752}
753
Guido van Rossumd8faa362007-04-27 19:54:29 +0000754/* Copy values from the "locals" dict into the fast locals.
755
756 dict is an input argument containing string keys representing
757 variables names and arbitrary PyObject* as values.
758
759 map and values are input arguments. map is a tuple of strings.
760 values is an array of PyObject*. At index i, map[i] is the name of
761 the variable with value values[i]. The function copies the first
762 nmap variable from map/values into dict. If values[i] is NULL,
763 the variable is deleted from dict.
764
765 If deref is true, then the values being copied are cell variables
766 and the value is extracted from the cell variable before being put
767 in dict. If clear is true, then variables in map but not in dict
768 are set to NULL in map; if clear is false, variables missing in
769 dict are ignored.
770
771 Exceptions raised while modifying the dict are silently ignored,
772 because there is no good way to report them.
773*/
774
Guido van Rossum6b356e72001-04-14 17:55:41 +0000775static void
Martin v. Löwis18e16552006-02-15 17:27:45 +0000776dict_to_map(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000777 int deref, int clear)
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000778{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000779 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000780 assert(PyTuple_Check(map));
781 assert(PyDict_Check(dict));
782 assert(PyTuple_Size(map) >= nmap);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000783 for (j = nmap; --j >= 0; ) {
Jeremy Hylton1a48ca82001-12-06 15:48:16 +0000784 PyObject *key = PyTuple_GET_ITEM(map, j);
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000785 PyObject *value = PyObject_GetItem(dict, key);
Martin v. Löwis5b222132007-06-10 09:51:05 +0000786 assert(PyUnicode_Check(key));
Guido van Rossumd8faa362007-04-27 19:54:29 +0000787 /* We only care about NULLs if clear is true. */
788 if (value == NULL) {
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000789 PyErr_Clear();
Guido van Rossumd8faa362007-04-27 19:54:29 +0000790 if (!clear)
791 continue;
792 }
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000793 if (deref) {
Guido van Rossumd8faa362007-04-27 19:54:29 +0000794 assert(PyCell_Check(values[j]));
795 if (PyCell_GET(values[j]) != value) {
796 if (PyCell_Set(values[j], value) < 0)
797 PyErr_Clear();
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000798 }
Guido van Rossumd8faa362007-04-27 19:54:29 +0000799 } else if (values[j] != value) {
800 Py_XINCREF(value);
801 Py_XDECREF(values[j]);
802 values[j] = value;
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000803 }
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000804 Py_XDECREF(value);
Jeremy Hylton220ae7c2001-03-21 16:43:47 +0000805 }
806}
Jeremy Hylton2b724da2001-01-29 22:51:52 +0000807
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000808void
Fred Drake1b190b42000-07-09 05:40:56 +0000809PyFrame_FastToLocals(PyFrameObject *f)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000810{
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000811 /* Merge fast locals into f->f_locals */
Guido van Rossum18752471997-04-29 14:49:28 +0000812 PyObject *locals, *map;
813 PyObject **fast;
814 PyObject *error_type, *error_value, *error_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000815 PyCodeObject *co;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000816 Py_ssize_t j;
Guido van Rossumd8faa362007-04-27 19:54:29 +0000817 int ncells, nfreevars;
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000818 if (f == NULL)
819 return;
Guido van Rossum2271bf71995-07-18 14:30:34 +0000820 locals = f->f_locals;
821 if (locals == NULL) {
Guido van Rossum18752471997-04-29 14:49:28 +0000822 locals = f->f_locals = PyDict_New();
Guido van Rossum2271bf71995-07-18 14:30:34 +0000823 if (locals == NULL) {
Guido van Rossum18752471997-04-29 14:49:28 +0000824 PyErr_Clear(); /* Can't report it :-( */
Guido van Rossum2271bf71995-07-18 14:30:34 +0000825 return;
826 }
827 }
Thomas Wouters477c8d52006-05-27 19:21:47 +0000828 co = f->f_code;
829 map = co->co_varnames;
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000830 if (!PyTuple_Check(map))
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000831 return;
Guido van Rossum18752471997-04-29 14:49:28 +0000832 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000833 fast = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000834 j = PyTuple_GET_SIZE(map);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000835 if (j > co->co_nlocals)
836 j = co->co_nlocals;
837 if (co->co_nlocals)
Jeremy Hylton174d2762003-10-21 18:10:28 +0000838 map_to_dict(map, j, locals, fast, 0);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000839 ncells = PyTuple_GET_SIZE(co->co_cellvars);
840 nfreevars = PyTuple_GET_SIZE(co->co_freevars);
841 if (ncells || nfreevars) {
842 map_to_dict(co->co_cellvars, ncells,
843 locals, fast + co->co_nlocals, 1);
Guido van Rossumd8faa362007-04-27 19:54:29 +0000844 /* If the namespace is unoptimized, then one of the
845 following cases applies:
846 1. It does not contain free variables, because it
847 uses import * or is a top-level namespace.
848 2. It is a class namespace.
849 We don't want to accidentally copy free variables
850 into the locals dict used by the class.
851 */
852 if (co->co_flags & CO_OPTIMIZED) {
853 map_to_dict(co->co_freevars, nfreevars,
854 locals, fast + co->co_nlocals + ncells, 1);
855 }
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000856 }
Guido van Rossum18752471997-04-29 14:49:28 +0000857 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000858}
859
860void
Fred Drake1b190b42000-07-09 05:40:56 +0000861PyFrame_LocalsToFast(PyFrameObject *f, int clear)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000862{
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000863 /* Merge f->f_locals into fast locals */
Guido van Rossum18752471997-04-29 14:49:28 +0000864 PyObject *locals, *map;
865 PyObject **fast;
866 PyObject *error_type, *error_value, *error_traceback;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000867 PyCodeObject *co;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000868 Py_ssize_t j;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000869 int ncells, nfreevars;
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000870 if (f == NULL)
871 return;
872 locals = f->f_locals;
Thomas Wouters477c8d52006-05-27 19:21:47 +0000873 co = f->f_code;
874 map = co->co_varnames;
Jeremy Hylton24ea8d32002-04-20 04:46:55 +0000875 if (locals == NULL)
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000876 return;
Raymond Hettinger214b1c32004-07-02 06:41:07 +0000877 if (!PyTuple_Check(map))
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000878 return;
Guido van Rossum18752471997-04-29 14:49:28 +0000879 PyErr_Fetch(&error_type, &error_value, &error_traceback);
Guido van Rossumf3e85a01997-01-20 04:20:52 +0000880 fast = f->f_localsplus;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000881 j = PyTuple_GET_SIZE(map);
Thomas Wouters477c8d52006-05-27 19:21:47 +0000882 if (j > co->co_nlocals)
883 j = co->co_nlocals;
884 if (co->co_nlocals)
885 dict_to_map(co->co_varnames, j, locals, fast, 0, clear);
886 ncells = PyTuple_GET_SIZE(co->co_cellvars);
887 nfreevars = PyTuple_GET_SIZE(co->co_freevars);
888 if (ncells || nfreevars) {
889 dict_to_map(co->co_cellvars, ncells,
890 locals, fast + co->co_nlocals, 1, clear);
891 dict_to_map(co->co_freevars, nfreevars,
892 locals, fast + co->co_nlocals + ncells, 1,
Guido van Rossumd8faa362007-04-27 19:54:29 +0000893 clear);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000894 }
Guido van Rossum18752471997-04-29 14:49:28 +0000895 PyErr_Restore(error_type, error_value, error_traceback);
Guido van Rossum1d5735e1994-08-30 08:27:36 +0000896}
Guido van Rossum404b95d1997-08-05 02:09:46 +0000897
898/* Clear out the free list */
899
900void
Fred Drake1b190b42000-07-09 05:40:56 +0000901PyFrame_Fini(void)
Guido van Rossum404b95d1997-08-05 02:09:46 +0000902{
903 while (free_list != NULL) {
904 PyFrameObject *f = free_list;
905 free_list = free_list->f_back;
Neil Schemenauer4f4817f2001-08-29 23:52:17 +0000906 PyObject_GC_Del(f);
Tim Petersb7ba7432002-04-13 05:21:47 +0000907 --numfree;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000908 }
Tim Petersb7ba7432002-04-13 05:21:47 +0000909 assert(numfree == 0);
Neal Norwitzc91ed402002-12-30 22:29:22 +0000910 Py_XDECREF(builtin_object);
911 builtin_object = NULL;
Guido van Rossum404b95d1997-08-05 02:09:46 +0000912}