zend_ast.h 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374
  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: Bob Weinand <bwoebi@php.net> |
  16. | Dmitry Stogov <dmitry@php.net> |
  17. | Nikita Popov <nikic@php.net> |
  18. +----------------------------------------------------------------------+
  19. */
  20. #ifndef ZEND_AST_H
  21. #define ZEND_AST_H
  22. #include "zend.h"
  23. #ifndef ZEND_AST_SPEC
  24. # define ZEND_AST_SPEC 1
  25. #endif
  26. #define ZEND_AST_SPECIAL_SHIFT 6
  27. #define ZEND_AST_IS_LIST_SHIFT 7
  28. #define ZEND_AST_NUM_CHILDREN_SHIFT 8
  29. enum _zend_ast_kind {
  30. /* special nodes */
  31. ZEND_AST_ZVAL = 1 << ZEND_AST_SPECIAL_SHIFT,
  32. ZEND_AST_CONSTANT,
  33. ZEND_AST_ZNODE,
  34. /* declaration nodes */
  35. ZEND_AST_FUNC_DECL,
  36. ZEND_AST_CLOSURE,
  37. ZEND_AST_METHOD,
  38. ZEND_AST_CLASS,
  39. ZEND_AST_ARROW_FUNC,
  40. /* list nodes */
  41. ZEND_AST_ARG_LIST = 1 << ZEND_AST_IS_LIST_SHIFT,
  42. ZEND_AST_ARRAY,
  43. ZEND_AST_ENCAPS_LIST,
  44. ZEND_AST_EXPR_LIST,
  45. ZEND_AST_STMT_LIST,
  46. ZEND_AST_IF,
  47. ZEND_AST_SWITCH_LIST,
  48. ZEND_AST_CATCH_LIST,
  49. ZEND_AST_PARAM_LIST,
  50. ZEND_AST_CLOSURE_USES,
  51. ZEND_AST_PROP_DECL,
  52. ZEND_AST_CONST_DECL,
  53. ZEND_AST_CLASS_CONST_DECL,
  54. ZEND_AST_NAME_LIST,
  55. ZEND_AST_TRAIT_ADAPTATIONS,
  56. ZEND_AST_USE,
  57. ZEND_AST_TYPE_UNION,
  58. ZEND_AST_TYPE_INTERSECTION,
  59. ZEND_AST_ATTRIBUTE_LIST,
  60. ZEND_AST_ATTRIBUTE_GROUP,
  61. ZEND_AST_MATCH_ARM_LIST,
  62. /* 0 child nodes */
  63. ZEND_AST_MAGIC_CONST = 0 << ZEND_AST_NUM_CHILDREN_SHIFT,
  64. ZEND_AST_TYPE,
  65. ZEND_AST_CONSTANT_CLASS,
  66. ZEND_AST_CALLABLE_CONVERT,
  67. /* 1 child node */
  68. ZEND_AST_VAR = 1 << ZEND_AST_NUM_CHILDREN_SHIFT,
  69. ZEND_AST_CONST,
  70. ZEND_AST_UNPACK,
  71. ZEND_AST_UNARY_PLUS,
  72. ZEND_AST_UNARY_MINUS,
  73. ZEND_AST_CAST,
  74. ZEND_AST_EMPTY,
  75. ZEND_AST_ISSET,
  76. ZEND_AST_SILENCE,
  77. ZEND_AST_SHELL_EXEC,
  78. ZEND_AST_CLONE,
  79. ZEND_AST_EXIT,
  80. ZEND_AST_PRINT,
  81. ZEND_AST_INCLUDE_OR_EVAL,
  82. ZEND_AST_UNARY_OP,
  83. ZEND_AST_PRE_INC,
  84. ZEND_AST_PRE_DEC,
  85. ZEND_AST_POST_INC,
  86. ZEND_AST_POST_DEC,
  87. ZEND_AST_YIELD_FROM,
  88. ZEND_AST_CLASS_NAME,
  89. ZEND_AST_GLOBAL,
  90. ZEND_AST_UNSET,
  91. ZEND_AST_RETURN,
  92. ZEND_AST_LABEL,
  93. ZEND_AST_REF,
  94. ZEND_AST_HALT_COMPILER,
  95. ZEND_AST_ECHO,
  96. ZEND_AST_THROW,
  97. ZEND_AST_GOTO,
  98. ZEND_AST_BREAK,
  99. ZEND_AST_CONTINUE,
  100. /* 2 child nodes */
  101. ZEND_AST_DIM = 2 << ZEND_AST_NUM_CHILDREN_SHIFT,
  102. ZEND_AST_PROP,
  103. ZEND_AST_NULLSAFE_PROP,
  104. ZEND_AST_STATIC_PROP,
  105. ZEND_AST_CALL,
  106. ZEND_AST_CLASS_CONST,
  107. ZEND_AST_ASSIGN,
  108. ZEND_AST_ASSIGN_REF,
  109. ZEND_AST_ASSIGN_OP,
  110. ZEND_AST_BINARY_OP,
  111. ZEND_AST_GREATER,
  112. ZEND_AST_GREATER_EQUAL,
  113. ZEND_AST_AND,
  114. ZEND_AST_OR,
  115. ZEND_AST_ARRAY_ELEM,
  116. ZEND_AST_NEW,
  117. ZEND_AST_INSTANCEOF,
  118. ZEND_AST_YIELD,
  119. ZEND_AST_COALESCE,
  120. ZEND_AST_ASSIGN_COALESCE,
  121. ZEND_AST_STATIC,
  122. ZEND_AST_WHILE,
  123. ZEND_AST_DO_WHILE,
  124. ZEND_AST_IF_ELEM,
  125. ZEND_AST_SWITCH,
  126. ZEND_AST_SWITCH_CASE,
  127. ZEND_AST_DECLARE,
  128. ZEND_AST_USE_TRAIT,
  129. ZEND_AST_TRAIT_PRECEDENCE,
  130. ZEND_AST_METHOD_REFERENCE,
  131. ZEND_AST_NAMESPACE,
  132. ZEND_AST_USE_ELEM,
  133. ZEND_AST_TRAIT_ALIAS,
  134. ZEND_AST_GROUP_USE,
  135. ZEND_AST_CLASS_CONST_GROUP,
  136. ZEND_AST_ATTRIBUTE,
  137. ZEND_AST_MATCH,
  138. ZEND_AST_MATCH_ARM,
  139. ZEND_AST_NAMED_ARG,
  140. /* 3 child nodes */
  141. ZEND_AST_METHOD_CALL = 3 << ZEND_AST_NUM_CHILDREN_SHIFT,
  142. ZEND_AST_NULLSAFE_METHOD_CALL,
  143. ZEND_AST_STATIC_CALL,
  144. ZEND_AST_CONDITIONAL,
  145. ZEND_AST_TRY,
  146. ZEND_AST_CATCH,
  147. ZEND_AST_PROP_GROUP,
  148. ZEND_AST_PROP_ELEM,
  149. ZEND_AST_CONST_ELEM,
  150. // Pseudo node for initializing enums
  151. ZEND_AST_CONST_ENUM_INIT,
  152. /* 4 child nodes */
  153. ZEND_AST_FOR = 4 << ZEND_AST_NUM_CHILDREN_SHIFT,
  154. ZEND_AST_FOREACH,
  155. ZEND_AST_ENUM_CASE,
  156. /* 5 child nodes */
  157. ZEND_AST_PARAM = 5 << ZEND_AST_NUM_CHILDREN_SHIFT,
  158. };
  159. typedef uint16_t zend_ast_kind;
  160. typedef uint16_t zend_ast_attr;
  161. struct _zend_ast {
  162. zend_ast_kind kind; /* Type of the node (ZEND_AST_* enum constant) */
  163. zend_ast_attr attr; /* Additional attribute, use depending on node type */
  164. uint32_t lineno; /* Line number */
  165. zend_ast *child[1]; /* Array of children (using struct hack) */
  166. };
  167. /* Same as zend_ast, but with children count, which is updated dynamically */
  168. typedef struct _zend_ast_list {
  169. zend_ast_kind kind;
  170. zend_ast_attr attr;
  171. uint32_t lineno;
  172. uint32_t children;
  173. zend_ast *child[1];
  174. } zend_ast_list;
  175. /* Lineno is stored in val.u2.lineno */
  176. typedef struct _zend_ast_zval {
  177. zend_ast_kind kind;
  178. zend_ast_attr attr;
  179. zval val;
  180. } zend_ast_zval;
  181. /* Separate structure for function and class declaration, as they need extra information. */
  182. typedef struct _zend_ast_decl {
  183. zend_ast_kind kind;
  184. zend_ast_attr attr; /* Unused - for structure compatibility */
  185. uint32_t start_lineno;
  186. uint32_t end_lineno;
  187. uint32_t flags;
  188. unsigned char *lex_pos;
  189. zend_string *doc_comment;
  190. zend_string *name;
  191. zend_ast *child[5];
  192. } zend_ast_decl;
  193. typedef void (*zend_ast_process_t)(zend_ast *ast);
  194. extern ZEND_API zend_ast_process_t zend_ast_process;
  195. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_zval_with_lineno(zval *zv, uint32_t lineno);
  196. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_zval_ex(zval *zv, zend_ast_attr attr);
  197. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_zval(zval *zv);
  198. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_zval_from_str(zend_string *str);
  199. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_zval_from_long(zend_long lval);
  200. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_constant(zend_string *name, zend_ast_attr attr);
  201. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_class_const_or_name(zend_ast *class_name, zend_ast *name);
  202. #if ZEND_AST_SPEC
  203. # define ZEND_AST_SPEC_CALL(name, ...) \
  204. ZEND_EXPAND_VA(ZEND_AST_SPEC_CALL_(name, __VA_ARGS__, _5, _4, _3, _2, _1, _0)(__VA_ARGS__))
  205. # define ZEND_AST_SPEC_CALL_(name, _, _5, _4, _3, _2, _1, suffix, ...) \
  206. name ## suffix
  207. # define ZEND_AST_SPEC_CALL_EX(name, ...) \
  208. ZEND_EXPAND_VA(ZEND_AST_SPEC_CALL_EX_(name, __VA_ARGS__, _5, _4, _3, _2, _1, _0)(__VA_ARGS__))
  209. # define ZEND_AST_SPEC_CALL_EX_(name, _, _6, _5, _4, _3, _2, _1, suffix, ...) \
  210. name ## suffix
  211. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_0(zend_ast_kind kind);
  212. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_1(zend_ast_kind kind, zend_ast *child);
  213. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_2(zend_ast_kind kind, zend_ast *child1, zend_ast *child2);
  214. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_3(zend_ast_kind kind, zend_ast *child1, zend_ast *child2, zend_ast *child3);
  215. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_4(zend_ast_kind kind, zend_ast *child1, zend_ast *child2, zend_ast *child3, zend_ast *child4);
  216. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_5(zend_ast_kind kind, zend_ast *child1, zend_ast *child2, zend_ast *child3, zend_ast *child4, zend_ast *child5);
  217. static zend_always_inline zend_ast * zend_ast_create_ex_0(zend_ast_kind kind, zend_ast_attr attr) {
  218. zend_ast *ast = zend_ast_create_0(kind);
  219. ast->attr = attr;
  220. return ast;
  221. }
  222. static zend_always_inline zend_ast * zend_ast_create_ex_1(zend_ast_kind kind, zend_ast_attr attr, zend_ast *child) {
  223. zend_ast *ast = zend_ast_create_1(kind, child);
  224. ast->attr = attr;
  225. return ast;
  226. }
  227. static zend_always_inline zend_ast * zend_ast_create_ex_2(zend_ast_kind kind, zend_ast_attr attr, zend_ast *child1, zend_ast *child2) {
  228. zend_ast *ast = zend_ast_create_2(kind, child1, child2);
  229. ast->attr = attr;
  230. return ast;
  231. }
  232. static zend_always_inline zend_ast * zend_ast_create_ex_3(zend_ast_kind kind, zend_ast_attr attr, zend_ast *child1, zend_ast *child2, zend_ast *child3) {
  233. zend_ast *ast = zend_ast_create_3(kind, child1, child2, child3);
  234. ast->attr = attr;
  235. return ast;
  236. }
  237. static zend_always_inline zend_ast * zend_ast_create_ex_4(zend_ast_kind kind, zend_ast_attr attr, zend_ast *child1, zend_ast *child2, zend_ast *child3, zend_ast *child4) {
  238. zend_ast *ast = zend_ast_create_4(kind, child1, child2, child3, child4);
  239. ast->attr = attr;
  240. return ast;
  241. }
  242. static zend_always_inline zend_ast * zend_ast_create_ex_5(zend_ast_kind kind, zend_ast_attr attr, zend_ast *child1, zend_ast *child2, zend_ast *child3, zend_ast *child4, zend_ast *child5) {
  243. zend_ast *ast = zend_ast_create_5(kind, child1, child2, child3, child4, child5);
  244. ast->attr = attr;
  245. return ast;
  246. }
  247. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_list_0(zend_ast_kind kind);
  248. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_list_1(zend_ast_kind kind, zend_ast *child);
  249. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_create_list_2(zend_ast_kind kind, zend_ast *child1, zend_ast *child2);
  250. # define zend_ast_create(...) \
  251. ZEND_AST_SPEC_CALL(zend_ast_create, __VA_ARGS__)
  252. # define zend_ast_create_ex(...) \
  253. ZEND_AST_SPEC_CALL_EX(zend_ast_create_ex, __VA_ARGS__)
  254. # define zend_ast_create_list(init_children, ...) \
  255. ZEND_AST_SPEC_CALL(zend_ast_create_list, __VA_ARGS__)
  256. #else
  257. ZEND_API zend_ast *zend_ast_create(zend_ast_kind kind, ...);
  258. ZEND_API zend_ast *zend_ast_create_ex(zend_ast_kind kind, zend_ast_attr attr, ...);
  259. ZEND_API zend_ast *zend_ast_create_list(uint32_t init_children, zend_ast_kind kind, ...);
  260. #endif
  261. ZEND_API zend_ast * ZEND_FASTCALL zend_ast_list_add(zend_ast *list, zend_ast *op);
  262. ZEND_API zend_ast *zend_ast_create_decl(
  263. zend_ast_kind kind, uint32_t flags, uint32_t start_lineno, zend_string *doc_comment,
  264. zend_string *name, zend_ast *child0, zend_ast *child1, zend_ast *child2, zend_ast *child3, zend_ast *child4
  265. );
  266. ZEND_API zend_result ZEND_FASTCALL zend_ast_evaluate(zval *result, zend_ast *ast, zend_class_entry *scope);
  267. ZEND_API zend_string *zend_ast_export(const char *prefix, zend_ast *ast, const char *suffix);
  268. ZEND_API zend_ast_ref * ZEND_FASTCALL zend_ast_copy(zend_ast *ast);
  269. ZEND_API void ZEND_FASTCALL zend_ast_destroy(zend_ast *ast);
  270. ZEND_API void ZEND_FASTCALL zend_ast_ref_destroy(zend_ast_ref *ast);
  271. typedef void (*zend_ast_apply_func)(zend_ast **ast_ptr, void *context);
  272. ZEND_API void zend_ast_apply(zend_ast *ast, zend_ast_apply_func fn, void *context);
  273. static zend_always_inline size_t zend_ast_size(uint32_t children) {
  274. return sizeof(zend_ast) - sizeof(zend_ast *) + sizeof(zend_ast *) * children;
  275. }
  276. static zend_always_inline bool zend_ast_is_special(zend_ast *ast) {
  277. return (ast->kind >> ZEND_AST_SPECIAL_SHIFT) & 1;
  278. }
  279. static zend_always_inline bool zend_ast_is_list(zend_ast *ast) {
  280. return (ast->kind >> ZEND_AST_IS_LIST_SHIFT) & 1;
  281. }
  282. static zend_always_inline zend_ast_list *zend_ast_get_list(zend_ast *ast) {
  283. ZEND_ASSERT(zend_ast_is_list(ast));
  284. return (zend_ast_list *) ast;
  285. }
  286. static zend_always_inline zval *zend_ast_get_zval(zend_ast *ast) {
  287. ZEND_ASSERT(ast->kind == ZEND_AST_ZVAL);
  288. return &((zend_ast_zval *) ast)->val;
  289. }
  290. static zend_always_inline zend_string *zend_ast_get_str(zend_ast *ast) {
  291. zval *zv = zend_ast_get_zval(ast);
  292. ZEND_ASSERT(Z_TYPE_P(zv) == IS_STRING);
  293. return Z_STR_P(zv);
  294. }
  295. static zend_always_inline zend_string *zend_ast_get_constant_name(zend_ast *ast) {
  296. ZEND_ASSERT(ast->kind == ZEND_AST_CONSTANT);
  297. ZEND_ASSERT(Z_TYPE(((zend_ast_zval *) ast)->val) == IS_STRING);
  298. return Z_STR(((zend_ast_zval *) ast)->val);
  299. }
  300. static zend_always_inline uint32_t zend_ast_get_num_children(zend_ast *ast) {
  301. ZEND_ASSERT(!zend_ast_is_list(ast));
  302. return ast->kind >> ZEND_AST_NUM_CHILDREN_SHIFT;
  303. }
  304. static zend_always_inline uint32_t zend_ast_get_lineno(zend_ast *ast) {
  305. if (ast->kind == ZEND_AST_ZVAL) {
  306. zval *zv = zend_ast_get_zval(ast);
  307. return Z_LINENO_P(zv);
  308. } else {
  309. return ast->lineno;
  310. }
  311. }
  312. static zend_always_inline zend_ast *zend_ast_create_binary_op(uint32_t opcode, zend_ast *op0, zend_ast *op1) {
  313. return zend_ast_create_ex(ZEND_AST_BINARY_OP, opcode, op0, op1);
  314. }
  315. static zend_always_inline zend_ast *zend_ast_create_assign_op(uint32_t opcode, zend_ast *op0, zend_ast *op1) {
  316. return zend_ast_create_ex(ZEND_AST_ASSIGN_OP, opcode, op0, op1);
  317. }
  318. static zend_always_inline zend_ast *zend_ast_create_cast(uint32_t type, zend_ast *op0) {
  319. return zend_ast_create_ex(ZEND_AST_CAST, type, op0);
  320. }
  321. static zend_always_inline zend_ast *zend_ast_list_rtrim(zend_ast *ast) {
  322. zend_ast_list *list = zend_ast_get_list(ast);
  323. if (list->children && list->child[list->children - 1] == NULL) {
  324. list->children--;
  325. }
  326. return ast;
  327. }
  328. zend_ast * ZEND_FASTCALL zend_ast_with_attributes(zend_ast *ast, zend_ast *attr);
  329. #endif