1 // escape.h -- Go escape analysis (based on Go compiler algorithm).
3 // Copyright 2016 The Go Authors. All rights reserved.
4 // Use of this source code is governed by a BSD-style
5 // license that can be found in the LICENSE file.
17 // There can be loops in the escape graph that lead to arbitrary recursions.
18 // See comment in gc/esc.go.
19 static const int MIN_LEVEL
= -2;
21 // Level models the escapement of a Node using two integers that are computed
22 // by backwards-analyzing the flow of a function from its sink and increasing or
23 // decreasing based on dereferences and addressing, respectively.
24 // One integer, known as the level's VALUE (think absolute value), is just the
25 // sum of indirections (via referencing or dereferencing) applied to the Node.
26 // The second, known as the level's SUFFIX_VALUE, is the amount of indirections
27 // applied after some data has been copied from the node. When accessing a
28 // field F of an object O and then applying indirections, for example, the field
29 // access O.F is assumed to copy that data from O before applying indirections.
30 // With this, even if O.F escapes, it might mean that the content of O escape,
31 // but not the object O itself.
37 : value_(0), suffix_value_(0)
40 Level(int value
, int suffix
)
41 : value_(value
), suffix_value_(suffix
)
44 // Return this level's value.
47 { return this->value_
; }
49 // Return this level's suffix value.
52 { return this->suffix_value_
; }
54 // Increase the level because a node is referenced.
58 if (this->value_
<= MIN_LEVEL
)
59 return Level(MIN_LEVEL
, 0);
61 return Level(this->value_
+ 1, this->suffix_value_
+ 1);
64 // Decrease the level because a node is dereferenced.
68 if (this->value_
<= MIN_LEVEL
)
69 return Level(MIN_LEVEL
, 0);
71 return Level(this->value_
- 1, this->suffix_value_
- 1);
74 // Model a node being copied.
78 return Level(this->value_
, std::max(this->suffix_value_
, 0));
81 // Return a level with the minimum values of this level and l.
83 min(const Level
& l
) const
85 return Level(std::min(this->value_
, l
.value()),
86 std::min(this->suffix_value_
, l
.suffix_value()));
89 // Compare two levels for equality.
91 operator==(const Level
& l
) const
93 return (this->value_
== l
.value()
94 && this->suffix_value_
== l
.suffix_value());
97 // Create a level from an integer value.
102 return Level(MIN_LEVEL
, 0);
107 // The sum of all indirects (-1) and references (+1) applied to a Node.
109 // The sum of all indirects (-1) abd references (+1) applied to a copied Node.
113 // A node in the escape graph. This node is an alias to a particular node
114 // in the Go parse tree. Specifically, it can represent an expression node,
115 // a statement node, or a named object node (a variable or function).
120 // This classification represents type of nodes in the Go parse tree that are
121 // interesting during the analysis.
122 enum Node_classification
129 // The state necessary to keep track of how a node escapes.
132 // The current function.
134 // A list of source nodes that flow into this node.
135 std::set
<Node
*> flows
;
136 // If the node is a function call, the list of nodes returned.
137 std::vector
<Node
*> retvals
;
138 // The node's loop depth.
140 // There is an extra loop depth in the flood phase used to account for
141 // variables referenced across closures. This is the maximum value of the
142 // extra loop depth seen during the flood that touches this node.
143 int max_extra_loop_depth
;
146 // An ID given to a node when it is encountered as a flow from the current
147 // dst node. This is used to avoid infinite recursion of cyclic nodes.
151 : fn(NULL
), loop_depth(0), max_extra_loop_depth(0), flood_id(0)
155 // Note: values in this enum appear in export data, and therefore MUST NOT
157 enum Escapement_encoding
160 // Does not escape to heap, result, or parameters.
162 // Is returned or reachable from a return statement.
164 // Allocated in an inner loop, assigned to an outer loop,
165 // which allows construction of non-escaping but arbitrarily large linked
166 // data structures (i.e., not eligible for allocation in a fixed-size stack
169 // Reachable from the heap.
171 // By construction will not escape.
175 // Multiple constructors for each classification.
176 Node(Named_object
* no
)
177 : classification_(NODE_OBJECT
), state_(NULL
), encoding_(ESCAPE_UNKNOWN
)
178 { this->u_
.object_val
= no
; }
181 : classification_(NODE_EXPRESSION
), state_(NULL
), encoding_(ESCAPE_UNKNOWN
)
182 { this->u_
.expression_val
= e
; }
185 : classification_(NODE_STATEMENT
), state_(NULL
), encoding_(ESCAPE_UNKNOWN
)
186 { this->u_
.statement_val
= s
; }
188 // Return this node's type.
192 // Return this node's location.
196 // Return this node's escape state.
198 state(Escape_context
* context
, Named_object
* fn
);
200 // Return this node's escape encoding.
203 { return this->encoding_
; }
205 // Set the node's escape encoding.
207 set_encoding(int enc
);
209 // Is this node a sink?
213 // Methods to return the underlying value in the Node union.
217 return (this->classification_
== NODE_OBJECT
218 ? this->u_
.object_val
225 return (this->classification_
== NODE_EXPRESSION
226 ? this->u_
.expression_val
233 return (this->classification_
== NODE_STATEMENT
234 ? this->u_
.statement_val
238 // Static creation methods for each value supported in the union.
240 make_node(Named_object
*);
243 make_node(Expression
*);
246 make_node(Statement
*);
248 // Return the maximum of an existing escape encoding E and a new
251 max_encoding(int e
, int etype
);
254 // The classification of this Node.
255 Node_classification classification_
;
260 Named_object
* object_val
;
261 // If NODE_EXPRESSION.
262 Expression
* expression_val
;
263 // If NODE_STATEMENT.
264 Statement
* statement_val
;
266 // The node's escape state.
267 Escape_state
* state_
;
268 // The node's escape encoding.
270 // | Return Encoding: (width - ESCAPE_RETURN_BITS) |
271 // | Content Escapes bit: 1 |
272 // | Escapement_encoding: ESCAPE_BITS |
275 // Cache all the Nodes created via Node::make_node to make the API simpler.
276 static std::map
<Named_object
*, Node
*> objects
;
277 static std::map
<Expression
*, Node
*> expressions
;
278 static std::map
<Statement
*, Node
*> statements
;
281 // The amount of bits used for the escapement encoding.
282 static const int ESCAPE_BITS
= 3;
284 // Mask used to extract encoding.
285 static const int ESCAPE_MASK
= (1 << ESCAPE_BITS
) - 1;
287 // Value obtained by indirect of parameter escapes to heap.
288 static const int ESCAPE_CONTENT_ESCAPES
= 1 << ESCAPE_BITS
;
290 // The amount of bits used in encoding of return values.
291 static const int ESCAPE_RETURN_BITS
= ESCAPE_BITS
+ 1;
293 // For each output, the number of bits for a tag.
294 static const int ESCAPE_BITS_PER_OUTPUT_IN_TAG
= 3;
296 // The bit max to extract a single tag.
297 static const int ESCAPE_BITS_MASK_FOR_TAG
= (1 << ESCAPE_BITS_PER_OUTPUT_IN_TAG
) - 1;
299 // The largest level that can be stored in a tag.
300 static const int ESCAPE_MAX_ENCODED_LEVEL
= ESCAPE_BITS_MASK_FOR_TAG
- 1;
302 // A helper for converting escape notes from encoded integers to a
303 // textual format and vice-versa.
308 // Return the string representation of an escapement encoding.
310 make_tag(int encoding
);
312 // Return the escapement encoding for a string tag.
314 parse_tag(std::string
* tag
);
317 // The escape context for a set of functions being analyzed.
322 Escape_context(Gogo
* gogo
, bool recursive
);
327 { return this->gogo_
; }
329 // Return the current function being analyzed.
331 current_function() const
332 { return this->current_function_
; }
334 // Change the function being analyzed.
336 set_current_function(Named_object
* fn
)
337 { this->current_function_
= fn
; }
339 // Return true if this is the context for a mutually recursive set of functions.
342 { return this->recursive_
; }
344 // Return the special sink node for this context.
347 { return this->sink_
; }
349 // Return the current loop depth.
352 { return this->loop_depth_
; }
354 // Increase the loop depth.
356 increase_loop_depth()
357 { this->loop_depth_
++; }
359 // Decrease the loop depth.
361 decrease_loop_depth()
362 { this->loop_depth_
--; }
365 set_loop_depth(int depth
)
366 { this->loop_depth_
= depth
; }
368 // Return the destination nodes encountered in this context.
369 const std::set
<Node
*>&
371 { return this->dsts_
; }
373 // Add a destination node.
376 { this->dsts_
.insert(dst
); }
378 // Return the nodes initially marked as non-escaping before flooding.
379 const std::vector
<Node
*>&
380 non_escaping_nodes() const
381 { return this->noesc_
; }
383 // Initialize the dummy return values for this Node N using the results
386 init_retvals(Node
* n
, Function_type
* fntype
);
388 // Return the indirection of Node N.
390 add_dereference(Node
* n
);
392 // Keep track of possibly non-escaping node N.
398 { return this->flood_id_
; }
402 { this->flood_id_
++; }
406 { return this->pdepth_
; }
419 // The current function being analyzed.
420 Named_object
* current_function_
;
421 // Return whether this is the context for a recursive function or a group of mutually
422 // recursive functions.
424 // The sink for this escape context. Nodes whose reference objects created
425 // outside the current function are assigned to the sink as well as nodes that
426 // the analysis loses track of.
428 // Used to detect nested loop scopes.
430 // All the destination nodes considered in this set of analyzed functions.
431 std::set
<Node
*> dsts_
;
432 // All the nodes that were noted as possibly not escaping in this context.
433 std::vector
<Node
*> noesc_
;
434 // An ID given to each dst and the flows discovered through DFS of that dst.
435 // This is used to avoid infinite recursion from nodes that point to each
436 // other within the flooding phase.
438 // The current level of recursion within a flooded section; used to debug.
442 #endif // !defined(GO_ESCAPE_H)