Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 1 | /* Frame object implementation */ |
| 2 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 3 | #include "Python.h" |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 4 | |
Jeremy Hylton | 3e0055f | 2005-10-20 19:59:25 +0000 | [diff] [blame] | 5 | #include "code.h" |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 6 | #include "frameobject.h" |
| 7 | #include "opcode.h" |
| 8 | #include "structmember.h" |
| 9 | |
Neal Norwitz | 91787cb | 2002-12-18 23:33:35 +0000 | [diff] [blame] | 10 | #undef MIN |
| 11 | #undef MAX |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 12 | #define MIN(a, b) ((a) < (b) ? (a) : (b)) |
| 13 | #define MAX(a, b) ((a) > (b) ? (a) : (b)) |
| 14 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 15 | #define OFF(x) offsetof(PyFrameObject, x) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 16 | |
Guido van Rossum | 6f79937 | 2001-09-20 20:46:19 +0000 | [diff] [blame] | 17 | static PyMemberDef frame_memberlist[] = { |
Guido van Rossum | 33d2689 | 2007-08-05 15:29:28 +0000 | [diff] [blame] | 18 | {"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 Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 23 | {NULL} /* Sentinel */ |
| 24 | }; |
| 25 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 26 | static PyObject * |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 27 | frame_getlocals(PyFrameObject *f, void *closure) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 28 | { |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 29 | PyFrame_FastToLocals(f); |
| 30 | Py_INCREF(f->f_locals); |
| 31 | return f->f_locals; |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 32 | } |
| 33 | |
Alexandre Vassalotti | 7b82b40 | 2009-07-21 04:30:03 +0000 | [diff] [blame] | 34 | int |
| 35 | PyFrame_GetLineNumber(PyFrameObject *f) |
| 36 | { |
| 37 | if (f->f_trace) |
| 38 | return f->f_lineno; |
| 39 | else |
| 40 | return PyCode_Addr2Line(f->f_code, f->f_lasti); |
| 41 | } |
| 42 | |
Michael W. Hudson | dd32a91 | 2002-08-15 14:59:02 +0000 | [diff] [blame] | 43 | static PyObject * |
| 44 | frame_getlineno(PyFrameObject *f, void *closure) |
| 45 | { |
Alexandre Vassalotti | 7b82b40 | 2009-07-21 04:30:03 +0000 | [diff] [blame] | 46 | return PyLong_FromLong(PyFrame_GetLineNumber(f)); |
Michael W. Hudson | dd32a91 | 2002-08-15 14:59:02 +0000 | [diff] [blame] | 47 | } |
| 48 | |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 49 | /* Setter for f_lineno - you can set f_lineno from within a trace function in |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 50 | * order to jump to a given line of code, subject to some restrictions. Most |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 51 | * lines are OK to jump to because they don't make any assumptions about the |
| 52 | * state of the stack (obvious because you could remove the line and the code |
| 53 | * would still work without any stack errors), but there are some constructs |
| 54 | * that limit jumping: |
| 55 | * |
| 56 | * o Lines with an 'except' statement on them can't be jumped to, because |
| 57 | * they expect an exception to be on the top of the stack. |
| 58 | * o Lines that live in a 'finally' block can't be jumped from or to, since |
| 59 | * the END_FINALLY expects to clean up the stack after the 'try' block. |
| 60 | * o 'try'/'for'/'while' blocks can't be jumped into because the blockstack |
| 61 | * needs to be set up before their code runs, and for 'for' loops the |
| 62 | * iterator needs to be on the stack. |
| 63 | */ |
| 64 | static int |
| 65 | frame_setlineno(PyFrameObject *f, PyObject* p_new_lineno) |
| 66 | { |
| 67 | int new_lineno = 0; /* The new value of f_lineno */ |
Martin v. Löwis | d1a1d1e | 2007-12-04 22:10:37 +0000 | [diff] [blame] | 68 | long l_new_lineno; |
| 69 | int overflow; |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 70 | int new_lasti = 0; /* The new value of f_lasti */ |
| 71 | int new_iblock = 0; /* The new value of f_iblock */ |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 72 | unsigned char *code = NULL; /* The bytecode for the frame... */ |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 73 | Py_ssize_t code_len = 0; /* ...and its length */ |
Amaury Forgeot d'Arc | 24e581f | 2009-06-01 21:28:37 +0000 | [diff] [blame] | 74 | unsigned char *lnotab = NULL; /* Iterating over co_lnotab */ |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 75 | Py_ssize_t lnotab_len = 0; /* (ditto) */ |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 76 | int offset = 0; /* (ditto) */ |
| 77 | int line = 0; /* (ditto) */ |
| 78 | int addr = 0; /* (ditto) */ |
| 79 | int min_addr = 0; /* Scanning the SETUPs and POPs */ |
| 80 | int max_addr = 0; /* (ditto) */ |
| 81 | int delta_iblock = 0; /* (ditto) */ |
| 82 | int min_delta_iblock = 0; /* (ditto) */ |
| 83 | int min_iblock = 0; /* (ditto) */ |
| 84 | int f_lasti_setup_addr = 0; /* Policing no-jump-into-finally */ |
| 85 | int new_lasti_setup_addr = 0; /* (ditto) */ |
| 86 | int blockstack[CO_MAXBLOCKS]; /* Walking the 'finally' blocks */ |
| 87 | int in_finally[CO_MAXBLOCKS]; /* (ditto) */ |
| 88 | int blockstack_top = 0; /* (ditto) */ |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 89 | unsigned char setup_op = 0; /* (ditto) */ |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 90 | |
| 91 | /* f_lineno must be an integer. */ |
Martin v. Löwis | d1a1d1e | 2007-12-04 22:10:37 +0000 | [diff] [blame] | 92 | if (!PyLong_CheckExact(p_new_lineno)) { |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 93 | PyErr_SetString(PyExc_ValueError, |
| 94 | "lineno must be an integer"); |
| 95 | return -1; |
| 96 | } |
| 97 | |
| 98 | /* You can only do this from within a trace function, not via |
| 99 | * _getframe or similar hackery. */ |
| 100 | if (!f->f_trace) |
| 101 | { |
| 102 | PyErr_Format(PyExc_ValueError, |
Jeffrey Yasskin | 4db6ff6 | 2009-05-20 19:09:05 +0000 | [diff] [blame] | 103 | "f_lineno can only be set by a" |
| 104 | " line trace function"); |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 105 | return -1; |
| 106 | } |
| 107 | |
| 108 | /* Fail if the line comes before the start of the code block. */ |
Martin v. Löwis | d1a1d1e | 2007-12-04 22:10:37 +0000 | [diff] [blame] | 109 | 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. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 122 | 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 | } |
Jeffrey Yasskin | 4db6ff6 | 2009-05-20 19:09:05 +0000 | [diff] [blame] | 128 | else if (new_lineno == f->f_code->co_firstlineno) { |
| 129 | new_lasti = 0; |
| 130 | new_lineno = f->f_code->co_firstlineno; |
| 131 | } |
| 132 | else { |
| 133 | /* Find the bytecode offset for the start of the given |
| 134 | * line, or the first code-owning line after it. */ |
Antoine Pitrou | b7ffad5 | 2009-06-01 23:29:07 +0000 | [diff] [blame] | 135 | char *tmp; |
Jeffrey Yasskin | 4db6ff6 | 2009-05-20 19:09:05 +0000 | [diff] [blame] | 136 | PyBytes_AsStringAndSize(f->f_code->co_lnotab, |
Antoine Pitrou | b7ffad5 | 2009-06-01 23:29:07 +0000 | [diff] [blame] | 137 | &tmp, &lnotab_len); |
| 138 | lnotab = (unsigned char *) tmp; |
Jeffrey Yasskin | 4db6ff6 | 2009-05-20 19:09:05 +0000 | [diff] [blame] | 139 | addr = 0; |
| 140 | line = f->f_code->co_firstlineno; |
| 141 | new_lasti = -1; |
| 142 | for (offset = 0; offset < lnotab_len; offset += 2) { |
| 143 | addr += lnotab[offset]; |
| 144 | line += lnotab[offset+1]; |
| 145 | if (line >= new_lineno) { |
| 146 | new_lasti = addr; |
| 147 | new_lineno = line; |
| 148 | break; |
| 149 | } |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 150 | } |
| 151 | } |
| 152 | |
| 153 | /* If we didn't reach the requested line, return an error. */ |
| 154 | if (new_lasti == -1) { |
| 155 | PyErr_Format(PyExc_ValueError, |
| 156 | "line %d comes after the current code block", |
| 157 | new_lineno); |
| 158 | return -1; |
| 159 | } |
| 160 | |
| 161 | /* We're now ready to look at the bytecode. */ |
Christian Heimes | 72b710a | 2008-05-26 13:28:38 +0000 | [diff] [blame] | 162 | PyBytes_AsStringAndSize(f->f_code->co_code, (char **)&code, &code_len); |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 163 | min_addr = MIN(new_lasti, f->f_lasti); |
| 164 | max_addr = MAX(new_lasti, f->f_lasti); |
| 165 | |
| 166 | /* You can't jump onto a line with an 'except' statement on it - |
| 167 | * they expect to have an exception on the top of the stack, which |
| 168 | * won't be true if you jump to them. They always start with code |
| 169 | * that either pops the exception using POP_TOP (plain 'except:' |
| 170 | * lines do this) or duplicates the exception on the stack using |
| 171 | * DUP_TOP (if there's an exception type specified). See compile.c, |
| 172 | * 'com_try_except' for the full details. There aren't any other |
| 173 | * cases (AFAIK) where a line's code can start with DUP_TOP or |
| 174 | * POP_TOP, but if any ever appear, they'll be subject to the same |
| 175 | * restriction (but with a different error message). */ |
| 176 | if (code[new_lasti] == DUP_TOP || code[new_lasti] == POP_TOP) { |
| 177 | PyErr_SetString(PyExc_ValueError, |
| 178 | "can't jump to 'except' line as there's no exception"); |
| 179 | return -1; |
| 180 | } |
| 181 | |
| 182 | /* You can't jump into or out of a 'finally' block because the 'try' |
| 183 | * block leaves something on the stack for the END_FINALLY to clean |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 184 | * up. So we walk the bytecode, maintaining a simulated blockstack. |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 185 | * When we reach the old or new address and it's in a 'finally' block |
| 186 | * we note the address of the corresponding SETUP_FINALLY. The jump |
| 187 | * is only legal if neither address is in a 'finally' block or |
| 188 | * they're both in the same one. 'blockstack' is a stack of the |
| 189 | * bytecode addresses of the SETUP_X opcodes, and 'in_finally' tracks |
| 190 | * whether we're in a 'finally' block at each blockstack level. */ |
| 191 | f_lasti_setup_addr = -1; |
| 192 | new_lasti_setup_addr = -1; |
| 193 | memset(blockstack, '\0', sizeof(blockstack)); |
| 194 | memset(in_finally, '\0', sizeof(in_finally)); |
| 195 | blockstack_top = 0; |
| 196 | for (addr = 0; addr < code_len; addr++) { |
| 197 | unsigned char op = code[addr]; |
| 198 | switch (op) { |
| 199 | case SETUP_LOOP: |
| 200 | case SETUP_EXCEPT: |
| 201 | case SETUP_FINALLY: |
| 202 | blockstack[blockstack_top++] = addr; |
| 203 | in_finally[blockstack_top-1] = 0; |
| 204 | break; |
| 205 | |
| 206 | case POP_BLOCK: |
Neal Norwitz | ee65e22 | 2002-12-19 18:16:57 +0000 | [diff] [blame] | 207 | assert(blockstack_top > 0); |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 208 | setup_op = code[blockstack[blockstack_top-1]]; |
| 209 | if (setup_op == SETUP_FINALLY) { |
| 210 | in_finally[blockstack_top-1] = 1; |
| 211 | } |
| 212 | else { |
| 213 | blockstack_top--; |
| 214 | } |
| 215 | break; |
| 216 | |
| 217 | case END_FINALLY: |
| 218 | /* Ignore END_FINALLYs for SETUP_EXCEPTs - they exist |
| 219 | * in the bytecode but don't correspond to an actual |
Neal Norwitz | ee65e22 | 2002-12-19 18:16:57 +0000 | [diff] [blame] | 220 | * 'finally' block. (If blockstack_top is 0, we must |
| 221 | * be seeing such an END_FINALLY.) */ |
| 222 | if (blockstack_top > 0) { |
| 223 | setup_op = code[blockstack[blockstack_top-1]]; |
| 224 | if (setup_op == SETUP_FINALLY) { |
| 225 | blockstack_top--; |
| 226 | } |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 227 | } |
| 228 | break; |
| 229 | } |
| 230 | |
| 231 | /* For the addresses we're interested in, see whether they're |
| 232 | * within a 'finally' block and if so, remember the address |
| 233 | * of the SETUP_FINALLY. */ |
| 234 | if (addr == new_lasti || addr == f->f_lasti) { |
| 235 | int i = 0; |
| 236 | int setup_addr = -1; |
| 237 | for (i = blockstack_top-1; i >= 0; i--) { |
| 238 | if (in_finally[i]) { |
| 239 | setup_addr = blockstack[i]; |
| 240 | break; |
| 241 | } |
| 242 | } |
| 243 | |
| 244 | if (setup_addr != -1) { |
| 245 | if (addr == new_lasti) { |
| 246 | new_lasti_setup_addr = setup_addr; |
| 247 | } |
| 248 | |
| 249 | if (addr == f->f_lasti) { |
| 250 | f_lasti_setup_addr = setup_addr; |
| 251 | } |
| 252 | } |
| 253 | } |
| 254 | |
| 255 | if (op >= HAVE_ARGUMENT) { |
| 256 | addr += 2; |
| 257 | } |
| 258 | } |
| 259 | |
Neal Norwitz | ee65e22 | 2002-12-19 18:16:57 +0000 | [diff] [blame] | 260 | /* Verify that the blockstack tracking code didn't get lost. */ |
| 261 | assert(blockstack_top == 0); |
| 262 | |
| 263 | /* After all that, are we jumping into / out of a 'finally' block? */ |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 264 | if (new_lasti_setup_addr != f_lasti_setup_addr) { |
| 265 | PyErr_SetString(PyExc_ValueError, |
| 266 | "can't jump into or out of a 'finally' block"); |
| 267 | return -1; |
| 268 | } |
| 269 | |
| 270 | |
| 271 | /* Police block-jumping (you can't jump into the middle of a block) |
| 272 | * and ensure that the blockstack finishes up in a sensible state (by |
| 273 | * popping any blocks we're jumping out of). We look at all the |
| 274 | * blockstack operations between the current position and the new |
| 275 | * one, and keep track of how many blocks we drop out of on the way. |
| 276 | * By also keeping track of the lowest blockstack position we see, we |
| 277 | * can tell whether the jump goes into any blocks without coming out |
| 278 | * again - in that case we raise an exception below. */ |
| 279 | delta_iblock = 0; |
| 280 | for (addr = min_addr; addr < max_addr; addr++) { |
| 281 | unsigned char op = code[addr]; |
| 282 | switch (op) { |
| 283 | case SETUP_LOOP: |
| 284 | case SETUP_EXCEPT: |
| 285 | case SETUP_FINALLY: |
| 286 | delta_iblock++; |
| 287 | break; |
| 288 | |
| 289 | case POP_BLOCK: |
| 290 | delta_iblock--; |
| 291 | break; |
| 292 | } |
| 293 | |
| 294 | min_delta_iblock = MIN(min_delta_iblock, delta_iblock); |
| 295 | |
| 296 | if (op >= HAVE_ARGUMENT) { |
| 297 | addr += 2; |
| 298 | } |
| 299 | } |
| 300 | |
| 301 | /* Derive the absolute iblock values from the deltas. */ |
| 302 | min_iblock = f->f_iblock + min_delta_iblock; |
| 303 | if (new_lasti > f->f_lasti) { |
| 304 | /* Forwards jump. */ |
| 305 | new_iblock = f->f_iblock + delta_iblock; |
| 306 | } |
| 307 | else { |
| 308 | /* Backwards jump. */ |
| 309 | new_iblock = f->f_iblock - delta_iblock; |
| 310 | } |
| 311 | |
| 312 | /* Are we jumping into a block? */ |
| 313 | if (new_iblock > min_iblock) { |
| 314 | PyErr_SetString(PyExc_ValueError, |
| 315 | "can't jump into the middle of a block"); |
| 316 | return -1; |
| 317 | } |
| 318 | |
| 319 | /* Pop any blocks that we're jumping out of. */ |
| 320 | while (f->f_iblock > new_iblock) { |
| 321 | PyTryBlock *b = &f->f_blockstack[--f->f_iblock]; |
| 322 | while ((f->f_stacktop - f->f_valuestack) > b->b_level) { |
| 323 | PyObject *v = (*--f->f_stacktop); |
| 324 | Py_DECREF(v); |
| 325 | } |
| 326 | } |
| 327 | |
| 328 | /* Finally set the new f_lineno and f_lasti and return OK. */ |
| 329 | f->f_lineno = new_lineno; |
| 330 | f->f_lasti = new_lasti; |
| 331 | return 0; |
| 332 | } |
| 333 | |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 334 | static PyObject * |
| 335 | frame_gettrace(PyFrameObject *f, void *closure) |
| 336 | { |
| 337 | PyObject* trace = f->f_trace; |
| 338 | |
| 339 | if (trace == NULL) |
| 340 | trace = Py_None; |
| 341 | |
| 342 | Py_INCREF(trace); |
| 343 | |
| 344 | return trace; |
| 345 | } |
| 346 | |
| 347 | static int |
| 348 | frame_settrace(PyFrameObject *f, PyObject* v, void *closure) |
| 349 | { |
Alexandre Vassalotti | 7b82b40 | 2009-07-21 04:30:03 +0000 | [diff] [blame] | 350 | PyObject* old_value; |
| 351 | |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 352 | /* We rely on f_lineno being accurate when f_trace is set. */ |
Alexandre Vassalotti | 7b82b40 | 2009-07-21 04:30:03 +0000 | [diff] [blame] | 353 | f->f_lineno = PyFrame_GetLineNumber(f); |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 354 | |
Alexandre Vassalotti | 7b82b40 | 2009-07-21 04:30:03 +0000 | [diff] [blame] | 355 | old_value = f->f_trace; |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 356 | Py_XINCREF(v); |
| 357 | f->f_trace = v; |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 358 | Py_XDECREF(old_value); |
| 359 | |
| 360 | return 0; |
| 361 | } |
| 362 | |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 363 | |
Guido van Rossum | 32d34c8 | 2001-09-20 21:45:26 +0000 | [diff] [blame] | 364 | static PyGetSetDef frame_getsetlist[] = { |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 365 | {"f_locals", (getter)frame_getlocals, NULL, NULL}, |
Michael W. Hudson | cfd3884 | 2002-12-17 16:15:34 +0000 | [diff] [blame] | 366 | {"f_lineno", (getter)frame_getlineno, |
| 367 | (setter)frame_setlineno, NULL}, |
Michael W. Hudson | 02ff6a9 | 2002-09-11 15:36:32 +0000 | [diff] [blame] | 368 | {"f_trace", (getter)frame_gettrace, (setter)frame_settrace, NULL}, |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 369 | {0} |
| 370 | }; |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 371 | |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 372 | /* Stack frames are allocated and deallocated at a considerable rate. |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 373 | In an attempt to improve the speed of function calls, we: |
| 374 | |
| 375 | 1. Hold a single "zombie" frame on each code object. This retains |
| 376 | the allocated and initialised frame object from an invocation of |
| 377 | the code object. The zombie is reanimated the next time we need a |
| 378 | frame object for that code object. Doing this saves the malloc/ |
| 379 | realloc required when using a free_list frame that isn't the |
| 380 | correct size. It also saves some field initialisation. |
| 381 | |
| 382 | In zombie mode, no field of PyFrameObject holds a reference, but |
| 383 | the following fields are still valid: |
| 384 | |
| 385 | * ob_type, ob_size, f_code, f_valuestack; |
| 386 | |
| 387 | * f_locals, f_trace, |
| 388 | f_exc_type, f_exc_value, f_exc_traceback are NULL; |
| 389 | |
| 390 | * f_localsplus does not require re-allocation and |
| 391 | the local variables in f_localsplus are NULL. |
| 392 | |
| 393 | 2. We also maintain a separate free list of stack frames (just like |
| 394 | integers are allocated in a special way -- see intobject.c). When |
| 395 | a stack frame is on the free list, only the following members have |
| 396 | a meaning: |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 397 | ob_type == &Frametype |
| 398 | f_back next item on free list, or NULL |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 399 | f_stacksize size of value stack |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 400 | ob_size size of localsplus |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 401 | Note that the value and block stacks are preserved -- this can save |
| 402 | another malloc() call or two (and two free() calls as well!). |
| 403 | Also note that, unlike for integers, each frame object is a |
| 404 | malloc'ed object in its own right -- it is only the actual calls to |
| 405 | malloc() that we are trying to save here, not the administration. |
| 406 | After all, while a typical program may make millions of calls, a |
| 407 | call depth of more than 20 or 30 is probably already exceptional |
| 408 | unless the program contains run-away recursion. I hope. |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 409 | |
Christian Heimes | 2202f87 | 2008-02-06 14:31:34 +0000 | [diff] [blame] | 410 | Later, PyFrame_MAXFREELIST was added to bound the # of frames saved on |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 411 | free_list. Else programs creating lots of cyclic trash involving |
| 412 | frames could provoke free_list into growing without bound. |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 413 | */ |
| 414 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 415 | static PyFrameObject *free_list = NULL; |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 416 | static int numfree = 0; /* number of frames currently in free_list */ |
Christian Heimes | 2202f87 | 2008-02-06 14:31:34 +0000 | [diff] [blame] | 417 | /* max value for numfree */ |
| 418 | #define PyFrame_MAXFREELIST 200 |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 419 | |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 420 | static void |
Fred Drake | 1b190b4 | 2000-07-09 05:40:56 +0000 | [diff] [blame] | 421 | frame_dealloc(PyFrameObject *f) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 422 | { |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 423 | PyObject **p, **valuestack; |
| 424 | PyCodeObject *co; |
Guido van Rossum | 7582bfb | 1997-02-14 16:27:29 +0000 | [diff] [blame] | 425 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 426 | PyObject_GC_UnTrack(f); |
Guido van Rossum | d724b23 | 2000-03-13 16:01:29 +0000 | [diff] [blame] | 427 | Py_TRASHCAN_SAFE_BEGIN(f) |
Guido van Rossum | 7582bfb | 1997-02-14 16:27:29 +0000 | [diff] [blame] | 428 | /* Kill all local variables */ |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 429 | valuestack = f->f_valuestack; |
| 430 | for (p = f->f_localsplus; p < valuestack; p++) |
| 431 | Py_CLEAR(*p); |
Guido van Rossum | 7582bfb | 1997-02-14 16:27:29 +0000 | [diff] [blame] | 432 | |
Tim Peters | 5ca576e | 2001-06-18 22:08:13 +0000 | [diff] [blame] | 433 | /* Free stack */ |
Tim Peters | 8c96369 | 2001-06-23 05:26:56 +0000 | [diff] [blame] | 434 | if (f->f_stacktop != NULL) { |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 435 | for (p = valuestack; p < f->f_stacktop; p++) |
Tim Peters | 8c96369 | 2001-06-23 05:26:56 +0000 | [diff] [blame] | 436 | Py_XDECREF(*p); |
Tim Peters | 5ca576e | 2001-06-18 22:08:13 +0000 | [diff] [blame] | 437 | } |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 438 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 439 | Py_XDECREF(f->f_back); |
Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 440 | Py_DECREF(f->f_builtins); |
| 441 | Py_DECREF(f->f_globals); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 442 | Py_CLEAR(f->f_locals); |
| 443 | Py_CLEAR(f->f_trace); |
| 444 | Py_CLEAR(f->f_exc_type); |
| 445 | Py_CLEAR(f->f_exc_value); |
| 446 | Py_CLEAR(f->f_exc_traceback); |
| 447 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 448 | co = f->f_code; |
| 449 | if (co->co_zombieframe == NULL) |
| 450 | co->co_zombieframe = f; |
Christian Heimes | 2202f87 | 2008-02-06 14:31:34 +0000 | [diff] [blame] | 451 | else if (numfree < PyFrame_MAXFREELIST) { |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 452 | ++numfree; |
| 453 | f->f_back = free_list; |
| 454 | free_list = f; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 455 | } |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 456 | else |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 457 | PyObject_GC_Del(f); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 458 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 459 | Py_DECREF(co); |
Guido van Rossum | d724b23 | 2000-03-13 16:01:29 +0000 | [diff] [blame] | 460 | Py_TRASHCAN_SAFE_END(f) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 461 | } |
| 462 | |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 463 | static int |
| 464 | frame_traverse(PyFrameObject *f, visitproc visit, void *arg) |
| 465 | { |
| 466 | PyObject **fastlocals, **p; |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 467 | int i, slots; |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 468 | |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 469 | Py_VISIT(f->f_back); |
| 470 | Py_VISIT(f->f_code); |
| 471 | Py_VISIT(f->f_builtins); |
| 472 | Py_VISIT(f->f_globals); |
| 473 | Py_VISIT(f->f_locals); |
| 474 | Py_VISIT(f->f_trace); |
| 475 | Py_VISIT(f->f_exc_type); |
| 476 | Py_VISIT(f->f_exc_value); |
| 477 | Py_VISIT(f->f_exc_traceback); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 478 | |
| 479 | /* locals */ |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 480 | slots = f->f_code->co_nlocals + PyTuple_GET_SIZE(f->f_code->co_cellvars) + PyTuple_GET_SIZE(f->f_code->co_freevars); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 481 | fastlocals = f->f_localsplus; |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 482 | for (i = slots; --i >= 0; ++fastlocals) |
| 483 | Py_VISIT(*fastlocals); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 484 | |
| 485 | /* stack */ |
| 486 | if (f->f_stacktop != NULL) { |
| 487 | for (p = f->f_valuestack; p < f->f_stacktop; p++) |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 488 | Py_VISIT(*p); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 489 | } |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 490 | return 0; |
| 491 | } |
| 492 | |
| 493 | static void |
| 494 | frame_clear(PyFrameObject *f) |
| 495 | { |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 496 | PyObject **fastlocals, **p, **oldtop; |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 497 | int i, slots; |
| 498 | |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 499 | /* Before anything else, make sure that this frame is clearly marked |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 500 | * as being defunct! Else, e.g., a generator reachable from this |
| 501 | * frame may also point to this frame, believe itself to still be |
| 502 | * active, and try cleaning up this frame again. |
| 503 | */ |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 504 | oldtop = f->f_stacktop; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 505 | f->f_stacktop = NULL; |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 506 | |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 507 | Py_CLEAR(f->f_exc_type); |
| 508 | Py_CLEAR(f->f_exc_value); |
| 509 | Py_CLEAR(f->f_exc_traceback); |
| 510 | Py_CLEAR(f->f_trace); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 511 | |
| 512 | /* locals */ |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 513 | slots = f->f_code->co_nlocals + PyTuple_GET_SIZE(f->f_code->co_cellvars) + PyTuple_GET_SIZE(f->f_code->co_freevars); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 514 | fastlocals = f->f_localsplus; |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 515 | for (i = slots; --i >= 0; ++fastlocals) |
| 516 | Py_CLEAR(*fastlocals); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 517 | |
| 518 | /* stack */ |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 519 | if (oldtop != NULL) { |
| 520 | for (p = f->f_valuestack; p < oldtop; p++) |
| 521 | Py_CLEAR(*p); |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 522 | } |
| 523 | } |
| 524 | |
Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 525 | static PyObject * |
| 526 | frame_sizeof(PyFrameObject *f) |
| 527 | { |
| 528 | Py_ssize_t res, extras, ncells, nfrees; |
| 529 | |
| 530 | ncells = PyTuple_GET_SIZE(f->f_code->co_cellvars); |
| 531 | nfrees = PyTuple_GET_SIZE(f->f_code->co_freevars); |
| 532 | extras = f->f_code->co_stacksize + f->f_code->co_nlocals + |
| 533 | ncells + nfrees; |
Christian Heimes | fefeca5 | 2008-10-02 18:38:11 +0000 | [diff] [blame] | 534 | /* subtract one as it is already included in PyFrameObject */ |
Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 535 | res = sizeof(PyFrameObject) + (extras-1) * sizeof(PyObject *); |
| 536 | |
| 537 | return PyLong_FromSsize_t(res); |
| 538 | } |
| 539 | |
| 540 | PyDoc_STRVAR(sizeof__doc__, |
| 541 | "F.__sizeof__() -> size of F in memory, in bytes"); |
| 542 | |
| 543 | static PyMethodDef frame_methods[] = { |
| 544 | {"__sizeof__", (PyCFunction)frame_sizeof, METH_NOARGS, |
| 545 | sizeof__doc__}, |
| 546 | {NULL, NULL} /* sentinel */ |
| 547 | }; |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 548 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 549 | PyTypeObject PyFrame_Type = { |
Martin v. Löwis | 9f2e346 | 2007-07-21 17:22:18 +0000 | [diff] [blame] | 550 | PyVarObject_HEAD_INIT(&PyType_Type, 0) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 551 | "frame", |
Neil Schemenauer | 4f4817f | 2001-08-29 23:52:17 +0000 | [diff] [blame] | 552 | sizeof(PyFrameObject), |
| 553 | sizeof(PyObject *), |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 554 | (destructor)frame_dealloc, /* tp_dealloc */ |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 555 | 0, /* tp_print */ |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 556 | 0, /* tp_getattr */ |
| 557 | 0, /* tp_setattr */ |
Mark Dickinson | e94c679 | 2009-02-02 20:36:42 +0000 | [diff] [blame] | 558 | 0, /* tp_reserved */ |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 559 | 0, /* tp_repr */ |
| 560 | 0, /* tp_as_number */ |
| 561 | 0, /* tp_as_sequence */ |
| 562 | 0, /* tp_as_mapping */ |
| 563 | 0, /* tp_hash */ |
| 564 | 0, /* tp_call */ |
| 565 | 0, /* tp_str */ |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 566 | PyObject_GenericGetAttr, /* tp_getattro */ |
| 567 | PyObject_GenericSetAttr, /* tp_setattro */ |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 568 | 0, /* tp_as_buffer */ |
Neil Schemenauer | 4f4817f | 2001-08-29 23:52:17 +0000 | [diff] [blame] | 569 | Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */ |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 570 | 0, /* tp_doc */ |
| 571 | (traverseproc)frame_traverse, /* tp_traverse */ |
Neil Schemenauer | 19cd292 | 2001-07-12 13:27:11 +0000 | [diff] [blame] | 572 | (inquiry)frame_clear, /* tp_clear */ |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 573 | 0, /* tp_richcompare */ |
| 574 | 0, /* tp_weaklistoffset */ |
| 575 | 0, /* tp_iter */ |
| 576 | 0, /* tp_iternext */ |
Robert Schuppenies | fbe94c5 | 2008-07-14 10:13:31 +0000 | [diff] [blame] | 577 | frame_methods, /* tp_methods */ |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 578 | frame_memberlist, /* tp_members */ |
| 579 | frame_getsetlist, /* tp_getset */ |
| 580 | 0, /* tp_base */ |
| 581 | 0, /* tp_dict */ |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 582 | }; |
| 583 | |
Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 584 | static PyObject *builtin_object; |
| 585 | |
Neal Norwitz | b2501f4 | 2002-12-31 03:42:13 +0000 | [diff] [blame] | 586 | int _PyFrame_Init() |
Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 587 | { |
Martin v. Löwis | 5b22213 | 2007-06-10 09:51:05 +0000 | [diff] [blame] | 588 | builtin_object = PyUnicode_InternFromString("__builtins__"); |
Benjamin Peterson | ae937c0 | 2009-04-18 20:54:08 +0000 | [diff] [blame] | 589 | if (builtin_object == NULL) |
| 590 | return 0; |
| 591 | return 1; |
Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 592 | } |
| 593 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 594 | PyFrameObject * |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 595 | PyFrame_New(PyThreadState *tstate, PyCodeObject *code, PyObject *globals, |
Jeremy Hylton | 30c9f39 | 2001-03-13 01:58:22 +0000 | [diff] [blame] | 596 | PyObject *locals) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 597 | { |
Guido van Rossum | a027efa | 1997-05-05 20:56:21 +0000 | [diff] [blame] | 598 | PyFrameObject *back = tstate->frame; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 599 | PyFrameObject *f; |
| 600 | PyObject *builtins; |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 601 | Py_ssize_t i; |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 602 | |
Michael W. Hudson | 69734a5 | 2002-08-19 16:54:08 +0000 | [diff] [blame] | 603 | #ifdef Py_DEBUG |
| 604 | if (code == NULL || globals == NULL || !PyDict_Check(globals) || |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 605 | (locals != NULL && !PyMapping_Check(locals))) { |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 606 | PyErr_BadInternalCall(); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 607 | return NULL; |
| 608 | } |
Michael W. Hudson | 69734a5 | 2002-08-19 16:54:08 +0000 | [diff] [blame] | 609 | #endif |
Guido van Rossum | bde6ff7 | 1998-02-19 20:48:26 +0000 | [diff] [blame] | 610 | if (back == NULL || back->f_globals != globals) { |
| 611 | builtins = PyDict_GetItem(globals, builtin_object); |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 612 | if (builtins) { |
| 613 | if (PyModule_Check(builtins)) { |
| 614 | builtins = PyModule_GetDict(builtins); |
| 615 | assert(!builtins || PyDict_Check(builtins)); |
| 616 | } |
| 617 | else if (!PyDict_Check(builtins)) |
| 618 | builtins = NULL; |
| 619 | } |
| 620 | if (builtins == NULL) { |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 621 | /* No builtins! Make up a minimal one |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 622 | Give them 'None', at least. */ |
| 623 | builtins = PyDict_New(); |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 624 | if (builtins == NULL || |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 625 | PyDict_SetItemString( |
| 626 | builtins, "None", Py_None) < 0) |
| 627 | return NULL; |
| 628 | } |
| 629 | else |
| 630 | Py_INCREF(builtins); |
| 631 | |
Guido van Rossum | bde6ff7 | 1998-02-19 20:48:26 +0000 | [diff] [blame] | 632 | } |
| 633 | else { |
| 634 | /* If we share the globals, we share the builtins. |
| 635 | Save a lookup and a call. */ |
| 636 | builtins = back->f_builtins; |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 637 | assert(builtins != NULL && PyDict_Check(builtins)); |
| 638 | Py_INCREF(builtins); |
Guido van Rossum | bde6ff7 | 1998-02-19 20:48:26 +0000 | [diff] [blame] | 639 | } |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 640 | if (code->co_zombieframe != NULL) { |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 641 | f = code->co_zombieframe; |
| 642 | code->co_zombieframe = NULL; |
| 643 | _Py_NewReference((PyObject *)f); |
| 644 | assert(f->f_code == code); |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 645 | } |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 646 | else { |
| 647 | Py_ssize_t extras, ncells, nfrees; |
| 648 | ncells = PyTuple_GET_SIZE(code->co_cellvars); |
| 649 | nfrees = PyTuple_GET_SIZE(code->co_freevars); |
| 650 | extras = code->co_stacksize + code->co_nlocals + ncells + |
| 651 | nfrees; |
| 652 | if (free_list == NULL) { |
| 653 | f = PyObject_GC_NewVar(PyFrameObject, &PyFrame_Type, |
| 654 | extras); |
| 655 | if (f == NULL) { |
| 656 | Py_DECREF(builtins); |
| 657 | return NULL; |
| 658 | } |
| 659 | } |
| 660 | else { |
| 661 | assert(numfree > 0); |
| 662 | --numfree; |
| 663 | f = free_list; |
| 664 | free_list = free_list->f_back; |
Christian Heimes | 90aa764 | 2007-12-19 02:45:37 +0000 | [diff] [blame] | 665 | if (Py_SIZE(f) < extras) { |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 666 | f = PyObject_GC_Resize(PyFrameObject, f, extras); |
| 667 | if (f == NULL) { |
| 668 | Py_DECREF(builtins); |
| 669 | return NULL; |
| 670 | } |
| 671 | } |
| 672 | _Py_NewReference((PyObject *)f); |
| 673 | } |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 674 | |
| 675 | f->f_code = code; |
| 676 | extras = code->co_nlocals + ncells + nfrees; |
| 677 | f->f_valuestack = f->f_localsplus + extras; |
| 678 | for (i=0; i<extras; i++) |
| 679 | f->f_localsplus[i] = NULL; |
| 680 | f->f_locals = NULL; |
| 681 | f->f_trace = NULL; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 682 | f->f_exc_type = f->f_exc_value = f->f_exc_traceback = NULL; |
Guido van Rossum | a9e7dc1 | 1992-10-18 18:53:57 +0000 | [diff] [blame] | 683 | } |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 684 | f->f_stacktop = f->f_valuestack; |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 685 | f->f_builtins = builtins; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 686 | Py_XINCREF(back); |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 687 | f->f_back = back; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 688 | Py_INCREF(code); |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 689 | Py_INCREF(globals); |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 690 | f->f_globals = globals; |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 691 | /* Most functions have CO_NEWLOCALS and CO_OPTIMIZED set. */ |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 692 | if ((code->co_flags & (CO_NEWLOCALS | CO_OPTIMIZED)) == |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 693 | (CO_NEWLOCALS | CO_OPTIMIZED)) |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 694 | ; /* f_locals = NULL; will be set by PyFrame_FastToLocals() */ |
Jeremy Hylton | bd5cbf8 | 2003-02-05 22:39:29 +0000 | [diff] [blame] | 695 | else if (code->co_flags & CO_NEWLOCALS) { |
| 696 | locals = PyDict_New(); |
| 697 | if (locals == NULL) { |
| 698 | Py_DECREF(f); |
| 699 | return NULL; |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 700 | } |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 701 | f->f_locals = locals; |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 702 | } |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 703 | else { |
| 704 | if (locals == NULL) |
| 705 | locals = globals; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 706 | Py_INCREF(locals); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 707 | f->f_locals = locals; |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 708 | } |
Guido van Rossum | eb46d67 | 1997-08-02 02:59:08 +0000 | [diff] [blame] | 709 | f->f_tstate = tstate; |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 710 | |
Michael W. Hudson | dd32a91 | 2002-08-15 14:59:02 +0000 | [diff] [blame] | 711 | f->f_lasti = -1; |
Guido van Rossum | 747596a | 1997-01-24 04:00:21 +0000 | [diff] [blame] | 712 | f->f_lineno = code->co_firstlineno; |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 713 | f->f_iblock = 0; |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 714 | |
Neil Schemenauer | 4f4817f | 2001-08-29 23:52:17 +0000 | [diff] [blame] | 715 | _PyObject_GC_TRACK(f); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 716 | return f; |
| 717 | } |
| 718 | |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 719 | /* Block management */ |
| 720 | |
| 721 | void |
Fred Drake | 1b190b4 | 2000-07-09 05:40:56 +0000 | [diff] [blame] | 722 | PyFrame_BlockSetup(PyFrameObject *f, int type, int handler, int level) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 723 | { |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 724 | PyTryBlock *b; |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 725 | if (f->f_iblock >= CO_MAXBLOCKS) |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 726 | Py_FatalError("XXX block stack overflow"); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 727 | b = &f->f_blockstack[f->f_iblock++]; |
| 728 | b->b_type = type; |
| 729 | b->b_level = level; |
| 730 | b->b_handler = handler; |
| 731 | } |
| 732 | |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 733 | PyTryBlock * |
Fred Drake | 1b190b4 | 2000-07-09 05:40:56 +0000 | [diff] [blame] | 734 | PyFrame_BlockPop(PyFrameObject *f) |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 735 | { |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 736 | PyTryBlock *b; |
Guido van Rossum | d7047b3 | 1995-01-02 19:07:15 +0000 | [diff] [blame] | 737 | if (f->f_iblock <= 0) |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 738 | Py_FatalError("XXX block stack underflow"); |
Guido van Rossum | 3f5da24 | 1990-12-20 15:06:42 +0000 | [diff] [blame] | 739 | b = &f->f_blockstack[--f->f_iblock]; |
| 740 | return b; |
| 741 | } |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 742 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 743 | /* Convert between "fast" version of locals and dictionary version. |
| 744 | |
| 745 | map and values are input arguments. map is a tuple of strings. |
| 746 | values is an array of PyObject*. At index i, map[i] is the name of |
| 747 | the variable with value values[i]. The function copies the first |
| 748 | nmap variable from map/values into dict. If values[i] is NULL, |
| 749 | the variable is deleted from dict. |
| 750 | |
| 751 | If deref is true, then the values being copied are cell variables |
| 752 | and the value is extracted from the cell variable before being put |
| 753 | in dict. |
| 754 | |
| 755 | Exceptions raised while modifying the dict are silently ignored, |
| 756 | because there is no good way to report them. |
| 757 | */ |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 758 | |
Guido van Rossum | f68d8e5 | 2001-04-14 17:55:09 +0000 | [diff] [blame] | 759 | static void |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 760 | map_to_dict(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values, |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 761 | int deref) |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 762 | { |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 763 | Py_ssize_t j; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 764 | assert(PyTuple_Check(map)); |
| 765 | assert(PyDict_Check(dict)); |
| 766 | assert(PyTuple_Size(map) >= nmap); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 767 | for (j = nmap; --j >= 0; ) { |
Jeremy Hylton | 1a48ca8 | 2001-12-06 15:48:16 +0000 | [diff] [blame] | 768 | PyObject *key = PyTuple_GET_ITEM(map, j); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 769 | PyObject *value = values[j]; |
Neal Norwitz | bed6784 | 2007-10-27 04:00:45 +0000 | [diff] [blame] | 770 | assert(PyUnicode_Check(key)); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 771 | if (deref) { |
| 772 | assert(PyCell_Check(value)); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 773 | value = PyCell_GET(value); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 774 | } |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 775 | if (value == NULL) { |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 776 | if (PyObject_DelItem(dict, key) != 0) |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 777 | PyErr_Clear(); |
| 778 | } |
| 779 | else { |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 780 | if (PyObject_SetItem(dict, key, value) != 0) |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 781 | PyErr_Clear(); |
| 782 | } |
| 783 | } |
| 784 | } |
| 785 | |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 786 | /* Copy values from the "locals" dict into the fast locals. |
| 787 | |
| 788 | dict is an input argument containing string keys representing |
| 789 | variables names and arbitrary PyObject* as values. |
| 790 | |
| 791 | map and values are input arguments. map is a tuple of strings. |
| 792 | values is an array of PyObject*. At index i, map[i] is the name of |
| 793 | the variable with value values[i]. The function copies the first |
| 794 | nmap variable from map/values into dict. If values[i] is NULL, |
| 795 | the variable is deleted from dict. |
| 796 | |
| 797 | If deref is true, then the values being copied are cell variables |
| 798 | and the value is extracted from the cell variable before being put |
| 799 | in dict. If clear is true, then variables in map but not in dict |
| 800 | are set to NULL in map; if clear is false, variables missing in |
| 801 | dict are ignored. |
| 802 | |
| 803 | Exceptions raised while modifying the dict are silently ignored, |
| 804 | because there is no good way to report them. |
| 805 | */ |
| 806 | |
Guido van Rossum | 6b356e7 | 2001-04-14 17:55:41 +0000 | [diff] [blame] | 807 | static void |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 808 | dict_to_map(PyObject *map, Py_ssize_t nmap, PyObject *dict, PyObject **values, |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 809 | int deref, int clear) |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 810 | { |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 811 | Py_ssize_t j; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 812 | assert(PyTuple_Check(map)); |
| 813 | assert(PyDict_Check(dict)); |
| 814 | assert(PyTuple_Size(map) >= nmap); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 815 | for (j = nmap; --j >= 0; ) { |
Jeremy Hylton | 1a48ca8 | 2001-12-06 15:48:16 +0000 | [diff] [blame] | 816 | PyObject *key = PyTuple_GET_ITEM(map, j); |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 817 | PyObject *value = PyObject_GetItem(dict, key); |
Martin v. Löwis | 5b22213 | 2007-06-10 09:51:05 +0000 | [diff] [blame] | 818 | assert(PyUnicode_Check(key)); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 819 | /* We only care about NULLs if clear is true. */ |
| 820 | if (value == NULL) { |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 821 | PyErr_Clear(); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 822 | if (!clear) |
| 823 | continue; |
| 824 | } |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 825 | if (deref) { |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 826 | assert(PyCell_Check(values[j])); |
| 827 | if (PyCell_GET(values[j]) != value) { |
| 828 | if (PyCell_Set(values[j], value) < 0) |
| 829 | PyErr_Clear(); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 830 | } |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 831 | } else if (values[j] != value) { |
| 832 | Py_XINCREF(value); |
| 833 | Py_XDECREF(values[j]); |
| 834 | values[j] = value; |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 835 | } |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 836 | Py_XDECREF(value); |
Jeremy Hylton | 220ae7c | 2001-03-21 16:43:47 +0000 | [diff] [blame] | 837 | } |
| 838 | } |
Jeremy Hylton | 2b724da | 2001-01-29 22:51:52 +0000 | [diff] [blame] | 839 | |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 840 | void |
Fred Drake | 1b190b4 | 2000-07-09 05:40:56 +0000 | [diff] [blame] | 841 | PyFrame_FastToLocals(PyFrameObject *f) |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 842 | { |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 843 | /* Merge fast locals into f->f_locals */ |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 844 | PyObject *locals, *map; |
| 845 | PyObject **fast; |
| 846 | PyObject *error_type, *error_value, *error_traceback; |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 847 | PyCodeObject *co; |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 848 | Py_ssize_t j; |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 849 | int ncells, nfreevars; |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 850 | if (f == NULL) |
| 851 | return; |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 852 | locals = f->f_locals; |
| 853 | if (locals == NULL) { |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 854 | locals = f->f_locals = PyDict_New(); |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 855 | if (locals == NULL) { |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 856 | PyErr_Clear(); /* Can't report it :-( */ |
Guido van Rossum | 2271bf7 | 1995-07-18 14:30:34 +0000 | [diff] [blame] | 857 | return; |
| 858 | } |
| 859 | } |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 860 | co = f->f_code; |
| 861 | map = co->co_varnames; |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 862 | if (!PyTuple_Check(map)) |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 863 | return; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 864 | PyErr_Fetch(&error_type, &error_value, &error_traceback); |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 865 | fast = f->f_localsplus; |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 866 | j = PyTuple_GET_SIZE(map); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 867 | if (j > co->co_nlocals) |
| 868 | j = co->co_nlocals; |
| 869 | if (co->co_nlocals) |
Jeremy Hylton | 174d276 | 2003-10-21 18:10:28 +0000 | [diff] [blame] | 870 | map_to_dict(map, j, locals, fast, 0); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 871 | ncells = PyTuple_GET_SIZE(co->co_cellvars); |
| 872 | nfreevars = PyTuple_GET_SIZE(co->co_freevars); |
| 873 | if (ncells || nfreevars) { |
| 874 | map_to_dict(co->co_cellvars, ncells, |
| 875 | locals, fast + co->co_nlocals, 1); |
Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 876 | /* If the namespace is unoptimized, then one of the |
| 877 | following cases applies: |
| 878 | 1. It does not contain free variables, because it |
| 879 | uses import * or is a top-level namespace. |
| 880 | 2. It is a class namespace. |
| 881 | We don't want to accidentally copy free variables |
| 882 | into the locals dict used by the class. |
| 883 | */ |
| 884 | if (co->co_flags & CO_OPTIMIZED) { |
| 885 | map_to_dict(co->co_freevars, nfreevars, |
| 886 | locals, fast + co->co_nlocals + ncells, 1); |
| 887 | } |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 888 | } |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 889 | PyErr_Restore(error_type, error_value, error_traceback); |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 890 | } |
| 891 | |
| 892 | void |
Fred Drake | 1b190b4 | 2000-07-09 05:40:56 +0000 | [diff] [blame] | 893 | PyFrame_LocalsToFast(PyFrameObject *f, int clear) |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 894 | { |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 895 | /* Merge f->f_locals into fast locals */ |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 896 | PyObject *locals, *map; |
| 897 | PyObject **fast; |
| 898 | PyObject *error_type, *error_value, *error_traceback; |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 899 | PyCodeObject *co; |
Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 900 | Py_ssize_t j; |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 901 | int ncells, nfreevars; |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 902 | if (f == NULL) |
| 903 | return; |
| 904 | locals = f->f_locals; |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 905 | co = f->f_code; |
| 906 | map = co->co_varnames; |
Jeremy Hylton | 24ea8d3 | 2002-04-20 04:46:55 +0000 | [diff] [blame] | 907 | if (locals == NULL) |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 908 | return; |
Raymond Hettinger | 214b1c3 | 2004-07-02 06:41:07 +0000 | [diff] [blame] | 909 | if (!PyTuple_Check(map)) |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 910 | return; |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 911 | PyErr_Fetch(&error_type, &error_value, &error_traceback); |
Guido van Rossum | f3e85a0 | 1997-01-20 04:20:52 +0000 | [diff] [blame] | 912 | fast = f->f_localsplus; |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 913 | j = PyTuple_GET_SIZE(map); |
Thomas Wouters | 477c8d5 | 2006-05-27 19:21:47 +0000 | [diff] [blame] | 914 | if (j > co->co_nlocals) |
| 915 | j = co->co_nlocals; |
| 916 | if (co->co_nlocals) |
| 917 | dict_to_map(co->co_varnames, j, locals, fast, 0, clear); |
| 918 | ncells = PyTuple_GET_SIZE(co->co_cellvars); |
| 919 | nfreevars = PyTuple_GET_SIZE(co->co_freevars); |
| 920 | if (ncells || nfreevars) { |
| 921 | dict_to_map(co->co_cellvars, ncells, |
| 922 | locals, fast + co->co_nlocals, 1, clear); |
Georg Brandl | 3dbca81 | 2008-07-23 16:10:53 +0000 | [diff] [blame] | 923 | /* Same test as in PyFrame_FastToLocals() above. */ |
| 924 | if (co->co_flags & CO_OPTIMIZED) { |
| 925 | dict_to_map(co->co_freevars, nfreevars, |
| 926 | locals, fast + co->co_nlocals + ncells, 1, |
| 927 | clear); |
| 928 | } |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 929 | } |
Guido van Rossum | 1875247 | 1997-04-29 14:49:28 +0000 | [diff] [blame] | 930 | PyErr_Restore(error_type, error_value, error_traceback); |
Guido van Rossum | 1d5735e | 1994-08-30 08:27:36 +0000 | [diff] [blame] | 931 | } |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 932 | |
| 933 | /* Clear out the free list */ |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 934 | int |
| 935 | PyFrame_ClearFreeList(void) |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 936 | { |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 937 | int freelist_size = numfree; |
| 938 | |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 939 | while (free_list != NULL) { |
| 940 | PyFrameObject *f = free_list; |
| 941 | free_list = free_list->f_back; |
Neil Schemenauer | 4f4817f | 2001-08-29 23:52:17 +0000 | [diff] [blame] | 942 | PyObject_GC_Del(f); |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 943 | --numfree; |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 944 | } |
Tim Peters | b7ba743 | 2002-04-13 05:21:47 +0000 | [diff] [blame] | 945 | assert(numfree == 0); |
Christian Heimes | a156e09 | 2008-02-16 07:38:31 +0000 | [diff] [blame] | 946 | return freelist_size; |
| 947 | } |
| 948 | |
| 949 | void |
| 950 | PyFrame_Fini(void) |
| 951 | { |
| 952 | (void)PyFrame_ClearFreeList(); |
Neal Norwitz | c91ed40 | 2002-12-30 22:29:22 +0000 | [diff] [blame] | 953 | Py_XDECREF(builtin_object); |
| 954 | builtin_object = NULL; |
Guido van Rossum | 404b95d | 1997-08-05 02:09:46 +0000 | [diff] [blame] | 955 | } |