]> git.proxmox.com Git - mirror_edk2.git/blame - AppPkg/Applications/Python/Python-2.7.10/Objects/genobject.c
EmbeddedPkg: Extend NvVarStoreFormattedLib LIBRARY_CLASS
[mirror_edk2.git] / AppPkg / Applications / Python / Python-2.7.10 / Objects / genobject.c
CommitLineData
53b2ba57
DM
1/* Generator object implementation */\r
2\r
3#include "Python.h"\r
4#include "frameobject.h"\r
5#include "genobject.h"\r
6#include "ceval.h"\r
7#include "structmember.h"\r
8#include "opcode.h"\r
9\r
10static int\r
11gen_traverse(PyGenObject *gen, visitproc visit, void *arg)\r
12{\r
13 Py_VISIT((PyObject *)gen->gi_frame);\r
14 Py_VISIT(gen->gi_code);\r
15 return 0;\r
16}\r
17\r
18static void\r
19gen_dealloc(PyGenObject *gen)\r
20{\r
21 PyObject *self = (PyObject *) gen;\r
22\r
23 _PyObject_GC_UNTRACK(gen);\r
24\r
25 if (gen->gi_weakreflist != NULL)\r
26 PyObject_ClearWeakRefs(self);\r
27\r
28 _PyObject_GC_TRACK(self);\r
29\r
30 if (gen->gi_frame != NULL && gen->gi_frame->f_stacktop != NULL) {\r
31 /* Generator is paused, so we need to close */\r
32 Py_TYPE(gen)->tp_del(self);\r
33 if (self->ob_refcnt > 0)\r
34 return; /* resurrected. :( */\r
35 }\r
36\r
37 _PyObject_GC_UNTRACK(self);\r
38 Py_CLEAR(gen->gi_frame);\r
39 Py_CLEAR(gen->gi_code);\r
40 PyObject_GC_Del(gen);\r
41}\r
42\r
43\r
44static PyObject *\r
45gen_send_ex(PyGenObject *gen, PyObject *arg, int exc)\r
46{\r
47 PyThreadState *tstate = PyThreadState_GET();\r
48 PyFrameObject *f = gen->gi_frame;\r
49 PyObject *result;\r
50\r
51 if (gen->gi_running) {\r
52 PyErr_SetString(PyExc_ValueError,\r
53 "generator already executing");\r
54 return NULL;\r
55 }\r
56 if (f==NULL || f->f_stacktop == NULL) {\r
57 /* Only set exception if called from send() */\r
58 if (arg && !exc)\r
59 PyErr_SetNone(PyExc_StopIteration);\r
60 return NULL;\r
61 }\r
62\r
63 if (f->f_lasti == -1) {\r
64 if (arg && arg != Py_None) {\r
65 PyErr_SetString(PyExc_TypeError,\r
66 "can't send non-None value to a "\r
67 "just-started generator");\r
68 return NULL;\r
69 }\r
70 } else {\r
71 /* Push arg onto the frame's value stack */\r
72 result = arg ? arg : Py_None;\r
73 Py_INCREF(result);\r
74 *(f->f_stacktop++) = result;\r
75 }\r
76\r
77 /* Generators always return to their most recent caller, not\r
78 * necessarily their creator. */\r
79 f->f_tstate = tstate;\r
80 Py_XINCREF(tstate->frame);\r
81 assert(f->f_back == NULL);\r
82 f->f_back = tstate->frame;\r
83\r
84 gen->gi_running = 1;\r
85 result = PyEval_EvalFrameEx(f, exc);\r
86 gen->gi_running = 0;\r
87\r
88 /* Don't keep the reference to f_back any longer than necessary. It\r
89 * may keep a chain of frames alive or it could create a reference\r
90 * cycle. */\r
91 assert(f->f_back == tstate->frame);\r
92 Py_CLEAR(f->f_back);\r
93 /* Clear the borrowed reference to the thread state */\r
94 f->f_tstate = NULL;\r
95\r
96 /* If the generator just returned (as opposed to yielding), signal\r
97 * that the generator is exhausted. */\r
98 if (result == Py_None && f->f_stacktop == NULL) {\r
99 Py_DECREF(result);\r
100 result = NULL;\r
101 /* Set exception if not called by gen_iternext() */\r
102 if (arg)\r
103 PyErr_SetNone(PyExc_StopIteration);\r
104 }\r
105\r
106 if (!result || f->f_stacktop == NULL) {\r
107 /* generator can't be rerun, so release the frame */\r
108 Py_DECREF(f);\r
109 gen->gi_frame = NULL;\r
110 }\r
111\r
112 return result;\r
113}\r
114\r
115PyDoc_STRVAR(send_doc,\r
116"send(arg) -> send 'arg' into generator,\n\\r
117return next yielded value or raise StopIteration.");\r
118\r
119static PyObject *\r
120gen_send(PyGenObject *gen, PyObject *arg)\r
121{\r
122 return gen_send_ex(gen, arg, 0);\r
123}\r
124\r
125PyDoc_STRVAR(close_doc,\r
126"close() -> raise GeneratorExit inside generator.");\r
127\r
128static PyObject *\r
129gen_close(PyGenObject *gen, PyObject *args)\r
130{\r
131 PyObject *retval;\r
132 PyErr_SetNone(PyExc_GeneratorExit);\r
133 retval = gen_send_ex(gen, Py_None, 1);\r
134 if (retval) {\r
135 Py_DECREF(retval);\r
136 PyErr_SetString(PyExc_RuntimeError,\r
137 "generator ignored GeneratorExit");\r
138 return NULL;\r
139 }\r
140 if (PyErr_ExceptionMatches(PyExc_StopIteration)\r
141 || PyErr_ExceptionMatches(PyExc_GeneratorExit))\r
142 {\r
143 PyErr_Clear(); /* ignore these errors */\r
144 Py_INCREF(Py_None);\r
145 return Py_None;\r
146 }\r
147 return NULL;\r
148}\r
149\r
150static void\r
151gen_del(PyObject *self)\r
152{\r
153 PyObject *res;\r
154 PyObject *error_type, *error_value, *error_traceback;\r
155 PyGenObject *gen = (PyGenObject *)self;\r
156\r
157 if (gen->gi_frame == NULL || gen->gi_frame->f_stacktop == NULL)\r
158 /* Generator isn't paused, so no need to close */\r
159 return;\r
160\r
161 /* Temporarily resurrect the object. */\r
162 assert(self->ob_refcnt == 0);\r
163 self->ob_refcnt = 1;\r
164\r
165 /* Save the current exception, if any. */\r
166 PyErr_Fetch(&error_type, &error_value, &error_traceback);\r
167\r
168 res = gen_close(gen, NULL);\r
169\r
170 if (res == NULL)\r
171 PyErr_WriteUnraisable(self);\r
172 else\r
173 Py_DECREF(res);\r
174\r
175 /* Restore the saved exception. */\r
176 PyErr_Restore(error_type, error_value, error_traceback);\r
177\r
178 /* Undo the temporary resurrection; can't use DECREF here, it would\r
179 * cause a recursive call.\r
180 */\r
181 assert(self->ob_refcnt > 0);\r
182 if (--self->ob_refcnt == 0)\r
183 return; /* this is the normal path out */\r
184\r
185 /* close() resurrected it! Make it look like the original Py_DECREF\r
186 * never happened.\r
187 */\r
188 {\r
189 Py_ssize_t refcnt = self->ob_refcnt;\r
190 _Py_NewReference(self);\r
191 self->ob_refcnt = refcnt;\r
192 }\r
193 assert(PyType_IS_GC(self->ob_type) &&\r
194 _Py_AS_GC(self)->gc.gc_refs != _PyGC_REFS_UNTRACKED);\r
195\r
196 /* If Py_REF_DEBUG, _Py_NewReference bumped _Py_RefTotal, so\r
197 * we need to undo that. */\r
198 _Py_DEC_REFTOTAL;\r
199 /* If Py_TRACE_REFS, _Py_NewReference re-added self to the object\r
200 * chain, so no more to do there.\r
201 * If COUNT_ALLOCS, the original decref bumped tp_frees, and\r
202 * _Py_NewReference bumped tp_allocs: both of those need to be\r
203 * undone.\r
204 */\r
205#ifdef COUNT_ALLOCS\r
206 --self->ob_type->tp_frees;\r
207 --self->ob_type->tp_allocs;\r
208#endif\r
209}\r
210\r
211\r
212\r
213PyDoc_STRVAR(throw_doc,\r
214"throw(typ[,val[,tb]]) -> raise exception in generator,\n\\r
215return next yielded value or raise StopIteration.");\r
216\r
217static PyObject *\r
218gen_throw(PyGenObject *gen, PyObject *args)\r
219{\r
220 PyObject *typ;\r
221 PyObject *tb = NULL;\r
222 PyObject *val = NULL;\r
223\r
224 if (!PyArg_UnpackTuple(args, "throw", 1, 3, &typ, &val, &tb))\r
225 return NULL;\r
226\r
227 /* First, check the traceback argument, replacing None with\r
228 NULL. */\r
229 if (tb == Py_None)\r
230 tb = NULL;\r
231 else if (tb != NULL && !PyTraceBack_Check(tb)) {\r
232 PyErr_SetString(PyExc_TypeError,\r
233 "throw() third argument must be a traceback object");\r
234 return NULL;\r
235 }\r
236\r
237 Py_INCREF(typ);\r
238 Py_XINCREF(val);\r
239 Py_XINCREF(tb);\r
240\r
241 if (PyExceptionClass_Check(typ)) {\r
242 PyErr_NormalizeException(&typ, &val, &tb);\r
243 }\r
244\r
245 else if (PyExceptionInstance_Check(typ)) {\r
246 /* Raising an instance. The value should be a dummy. */\r
247 if (val && val != Py_None) {\r
248 PyErr_SetString(PyExc_TypeError,\r
249 "instance exception may not have a separate value");\r
250 goto failed_throw;\r
251 }\r
252 else {\r
253 /* Normalize to raise <class>, <instance> */\r
254 Py_XDECREF(val);\r
255 val = typ;\r
256 typ = PyExceptionInstance_Class(typ);\r
257 Py_INCREF(typ);\r
258 }\r
259 }\r
260 else {\r
261 /* Not something you can raise. throw() fails. */\r
262 PyErr_Format(PyExc_TypeError,\r
263 "exceptions must be classes, or instances, not %s",\r
264 typ->ob_type->tp_name);\r
265 goto failed_throw;\r
266 }\r
267\r
268 PyErr_Restore(typ, val, tb);\r
269 return gen_send_ex(gen, Py_None, 1);\r
270\r
271failed_throw:\r
272 /* Didn't use our arguments, so restore their original refcounts */\r
273 Py_DECREF(typ);\r
274 Py_XDECREF(val);\r
275 Py_XDECREF(tb);\r
276 return NULL;\r
277}\r
278\r
279\r
280static PyObject *\r
281gen_iternext(PyGenObject *gen)\r
282{\r
283 return gen_send_ex(gen, NULL, 0);\r
284}\r
285\r
286\r
287static PyObject *\r
288gen_repr(PyGenObject *gen)\r
289{\r
290 char *code_name;\r
291 code_name = PyString_AsString(((PyCodeObject *)gen->gi_code)->co_name);\r
292 if (code_name == NULL)\r
293 return NULL;\r
294 return PyString_FromFormat("<generator object %.200s at %p>",\r
295 code_name, gen);\r
296}\r
297\r
298\r
299static PyObject *\r
300gen_get_name(PyGenObject *gen)\r
301{\r
302 PyObject *name = ((PyCodeObject *)gen->gi_code)->co_name;\r
303 Py_INCREF(name);\r
304 return name;\r
305}\r
306\r
307\r
308PyDoc_STRVAR(gen__name__doc__,\r
309"Return the name of the generator's associated code object.");\r
310\r
311static PyGetSetDef gen_getsetlist[] = {\r
312 {"__name__", (getter)gen_get_name, NULL, gen__name__doc__},\r
313 {NULL}\r
314};\r
315\r
316\r
317static PyMemberDef gen_memberlist[] = {\r
318 {"gi_frame", T_OBJECT, offsetof(PyGenObject, gi_frame), RO},\r
319 {"gi_running", T_INT, offsetof(PyGenObject, gi_running), RO},\r
320 {"gi_code", T_OBJECT, offsetof(PyGenObject, gi_code), RO},\r
321 {NULL} /* Sentinel */\r
322};\r
323\r
324static PyMethodDef gen_methods[] = {\r
325 {"send",(PyCFunction)gen_send, METH_O, send_doc},\r
326 {"throw",(PyCFunction)gen_throw, METH_VARARGS, throw_doc},\r
327 {"close",(PyCFunction)gen_close, METH_NOARGS, close_doc},\r
328 {NULL, NULL} /* Sentinel */\r
329};\r
330\r
331PyTypeObject PyGen_Type = {\r
332 PyVarObject_HEAD_INIT(&PyType_Type, 0)\r
333 "generator", /* tp_name */\r
334 sizeof(PyGenObject), /* tp_basicsize */\r
335 0, /* tp_itemsize */\r
336 /* methods */\r
337 (destructor)gen_dealloc, /* tp_dealloc */\r
338 0, /* tp_print */\r
339 0, /* tp_getattr */\r
340 0, /* tp_setattr */\r
341 0, /* tp_compare */\r
342 (reprfunc)gen_repr, /* tp_repr */\r
343 0, /* tp_as_number */\r
344 0, /* tp_as_sequence */\r
345 0, /* tp_as_mapping */\r
346 0, /* tp_hash */\r
347 0, /* tp_call */\r
348 0, /* tp_str */\r
349 PyObject_GenericGetAttr, /* tp_getattro */\r
350 0, /* tp_setattro */\r
351 0, /* tp_as_buffer */\r
352 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */\r
353 0, /* tp_doc */\r
354 (traverseproc)gen_traverse, /* tp_traverse */\r
355 0, /* tp_clear */\r
356 0, /* tp_richcompare */\r
357 offsetof(PyGenObject, gi_weakreflist), /* tp_weaklistoffset */\r
358 PyObject_SelfIter, /* tp_iter */\r
359 (iternextfunc)gen_iternext, /* tp_iternext */\r
360 gen_methods, /* tp_methods */\r
361 gen_memberlist, /* tp_members */\r
362 gen_getsetlist, /* tp_getset */\r
363 0, /* tp_base */\r
364 0, /* tp_dict */\r
365\r
366 0, /* tp_descr_get */\r
367 0, /* tp_descr_set */\r
368 0, /* tp_dictoffset */\r
369 0, /* tp_init */\r
370 0, /* tp_alloc */\r
371 0, /* tp_new */\r
372 0, /* tp_free */\r
373 0, /* tp_is_gc */\r
374 0, /* tp_bases */\r
375 0, /* tp_mro */\r
376 0, /* tp_cache */\r
377 0, /* tp_subclasses */\r
378 0, /* tp_weaklist */\r
379 gen_del, /* tp_del */\r
380};\r
381\r
382PyObject *\r
383PyGen_New(PyFrameObject *f)\r
384{\r
385 PyGenObject *gen = PyObject_GC_New(PyGenObject, &PyGen_Type);\r
386 if (gen == NULL) {\r
387 Py_DECREF(f);\r
388 return NULL;\r
389 }\r
390 gen->gi_frame = f;\r
391 Py_INCREF(f->f_code);\r
392 gen->gi_code = (PyObject *)(f->f_code);\r
393 gen->gi_running = 0;\r
394 gen->gi_weakreflist = NULL;\r
395 _PyObject_GC_TRACK(gen);\r
396 return (PyObject *)gen;\r
397}\r
398\r
399int\r
400PyGen_NeedsFinalizing(PyGenObject *gen)\r
401{\r
402 int i;\r
403 PyFrameObject *f = gen->gi_frame;\r
404\r
405 if (f == NULL || f->f_stacktop == NULL || f->f_iblock <= 0)\r
406 return 0; /* no frame or empty blockstack == no finalization */\r
407\r
408 /* Any block type besides a loop requires cleanup. */\r
409 i = f->f_iblock;\r
410 while (--i >= 0) {\r
411 if (f->f_blockstack[i].b_type != SETUP_LOOP)\r
412 return 1;\r
413 }\r
414\r
415 /* No blocks except loops, it's safe to skip finalization. */\r
416 return 0;\r
417}\r