1 /*
2 +----------------------------------------------------------------------+
3 | Zend Engine |
4 +----------------------------------------------------------------------+
5 | Copyright (c) Zend Technologies Ltd. (http://www.zend.com) |
6 +----------------------------------------------------------------------+
7 | This source file is subject to version 2.00 of the Zend license, |
8 | that is bundled with this package in the file LICENSE, and is |
9 | available through the world-wide-web at the following url: |
10 | http://www.zend.com/license/2_00.txt. |
11 | If you did not receive a copy of the Zend license and are unable to |
12 | obtain it through the world-wide-web, please send a note to |
13 | license@zend.com so we can mail you a copy immediately. |
14 +----------------------------------------------------------------------+
15 | Authors: Nikita Popov <nikic@php.net> |
16 | Bob Weinand <bobwei9@hotmail.com> |
17 +----------------------------------------------------------------------+
18 */
19
20 #ifndef ZEND_GENERATORS_H
21 #define ZEND_GENERATORS_H
22
23 BEGIN_EXTERN_C()
24
25 extern ZEND_API zend_class_entry *zend_ce_generator;
26 extern ZEND_API zend_class_entry *zend_ce_ClosedGeneratorException;
27
28 typedef struct _zend_generator_node zend_generator_node;
29 typedef struct _zend_generator zend_generator;
30
31 /* The concept of `yield from` exposes problems when accessed at different levels of the chain of delegated generators. We need to be able to reference the currently executed Generator in all cases and still being able to access the return values of finished Generators.
32 * The solution to this problem is a doubly-linked tree, which all Generators referenced in maintain a reference to. It should be impossible to avoid walking the tree in all cases. This way, we only need tree walks from leaf to root in case where some part of the `yield from` chain is passed to another `yield from`. (Update of leaf node pointer and list of multi-children nodes needed when leaf gets a child in direct path from leaf to root node.) But only in that case, which should be a fairly rare case (which is then possible, but not totally cheap).
33 * The root of the tree is then the currently executed Generator. The subnodes of the tree (all except the root node) are all Generators which do `yield from`. Each node of the tree knows a pointer to one leaf descendant node. Each node with multiple children needs a list of all leaf descendant nodes paired with pointers to their respective child node. (The stack is determined by leaf node pointers) Nodes with only one child just don't need a list, there it is enough to just have a pointer to the child node. Further, leaf nodes store a pointer to the root node.
34 * That way, when we advance any generator, we just need to look up a leaf node (which all have a reference to a root node). Then we can see at the root node whether current Generator is finished. If it isn't, all is fine and we can just continue. If the Generator finished, there will be two cases. Either it is a simple node with just one child, then go down to child node. Or it has multiple children and we now will remove the current leaf node from the list of nodes (unnecessary, is microoptimization) and go down to the child node whose reference was paired with current leaf node. Child node is then removed its parent reference and becomes new top node. Or the current node references the Generator we're currently executing, then we can continue from the YIELD_FROM opcode. When a node referenced as root node in a leaf node has a parent, then we go the way up until we find a root node without parent.
35 * In case we go into a new `yield from` level, a node is created on top of current root and becomes the new root. Leaf node needs to be updated with new root node then.
36 * When a Generator referenced by a node of the tree is added to `yield from`, that node now gets a list of children (we need to walk the descendants of that node and nodes of the tree of the other Generator down to the first multi-children node and copy all the leaf node pointers from there). In case there was no multi-children node (linear tree), we just add a pair (pointer to leaf node, pointer to child node), with the child node being in a direct path from leaf to this node.
37 */
38
39 struct _zend_generator_node {
40 zend_generator *parent; /* NULL for root */
41 uint32_t children;
42 union {
43 HashTable *ht; /* if multiple children */
44 struct { /* if one child */
45 zend_generator *leaf;
46 zend_generator *child;
47 } single;
48 } child;
49 union {
50 zend_generator *leaf; /* if > 0 children */
51 zend_generator *root; /* if 0 children */
52 } ptr;
53 };
54
55 struct _zend_generator {
56 zend_object std;
57
58 zend_object_iterator *iterator;
59
60 /* The suspended execution context. */
61 zend_execute_data *execute_data;
62
63 /* Frozen call stack for "yield" used in context of other calls */
64 zend_execute_data *frozen_call_stack;
65
66 /* Current value */
67 zval value;
68 /* Current key */
69 zval key;
70 /* Return value */
71 zval retval;
72 /* Variable to put sent value into */
73 zval *send_target;
74 /* Largest used integer key for auto-incrementing keys */
75 zend_long largest_used_integer_key;
76
77 /* Values specified by "yield from" to yield from this generator.
78 * This is only used for arrays or non-generator Traversables.
79 * This zval also uses the u2 structure in the same way as
80 * by-value foreach. */
81 zval values;
82
83 /* Node of waiting generators when multiple "yield from" expressions
84 * are nested. */
85 zend_generator_node node;
86
87 /* Fake execute_data for stacktraces */
88 zend_execute_data execute_fake;
89
90 /* ZEND_GENERATOR_* flags */
91 zend_uchar flags;
92
93 zval *gc_buffer;
94 uint32_t gc_buffer_size;
95 };
96
97 static const zend_uchar ZEND_GENERATOR_CURRENTLY_RUNNING = 0x1;
98 static const zend_uchar ZEND_GENERATOR_FORCED_CLOSE = 0x2;
99 static const zend_uchar ZEND_GENERATOR_AT_FIRST_YIELD = 0x4;
100 static const zend_uchar ZEND_GENERATOR_DO_INIT = 0x8;
101
102 void zend_register_generator_ce(void);
103 ZEND_API void zend_generator_close(zend_generator *generator, zend_bool finished_execution);
104 ZEND_API void zend_generator_resume(zend_generator *generator);
105
106 ZEND_API void zend_generator_restore_call_stack(zend_generator *generator);
107 ZEND_API zend_execute_data* zend_generator_freeze_call_stack(zend_execute_data *execute_data);
108
109 void zend_generator_yield_from(zend_generator *generator, zend_generator *from);
110 ZEND_API zend_execute_data *zend_generator_check_placeholder_frame(zend_execute_data *ptr);
111
112 ZEND_API zend_generator *zend_generator_update_current(zend_generator *generator, zend_generator *leaf);
zend_generator_get_current(zend_generator * generator)113 static zend_always_inline zend_generator *zend_generator_get_current(zend_generator *generator)
114 {
115 zend_generator *leaf;
116 zend_generator *root;
117
118 if (EXPECTED(generator->node.parent == NULL)) {
119 /* we're not in yield from mode */
120 return generator;
121 }
122
123 leaf = generator->node.children ? generator->node.ptr.leaf : generator;
124 root = leaf->node.ptr.root;
125
126 if (EXPECTED(root->execute_data && root->node.parent == NULL)) {
127 /* generator still running */
128 return root;
129 }
130
131 return zend_generator_update_current(generator, leaf);
132 }
133
134 END_EXTERN_C()
135
136 #endif
137
