pe*alloc
[m6w6/ext-psi] / src / types / let_callback.c
1 /*******************************************************************************
2 Copyright (c) 2016, Michael Wallner <mike@php.net>.
3 All rights reserved.
4
5 Redistribution and use in source and binary forms, with or without
6 modification, are permitted provided that the following conditions are met:
7
8 * Redistributions of source code must retain the above copyright notice,
9 this list of conditions and the following disclaimer.
10 * Redistributions in binary form must reproduce the above copyright
11 notice, this list of conditions and the following disclaimer in the
12 documentation and/or other materials provided with the distribution.
13
14 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
15 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
17 DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
18 FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
20 SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
21 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
22 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 *******************************************************************************/
25
26 #include "php_psi_stdinc.h"
27 #include "data.h"
28
29 struct psi_let_callback *psi_let_callback_init(struct psi_let_func *func,
30 struct psi_plist *args, struct psi_plist *cb_args)
31 {
32 struct psi_let_callback *cb = pecalloc(1, sizeof(*cb), 1);
33 cb->func = func;
34 cb->args = args;
35 cb->cb_args = cb_args;
36 return cb;
37 }
38
39 void psi_let_callback_free(struct psi_let_callback **cb_ptr)
40 {
41 if (*cb_ptr) {
42 struct psi_let_callback *cb = *cb_ptr;
43
44 *cb_ptr = NULL;
45 psi_let_func_free(&cb->func);
46 psi_plist_free(cb->args);
47 if (cb->cb_args) {
48 psi_plist_free(cb->cb_args);
49 }
50 psi_token_free(&cb->token);
51 free(cb);
52 }
53 }
54
55
56 static inline bool psi_let_callback_validate_decl_args(struct psi_data *data,
57 struct psi_let_callback *cb, struct psi_decl *cb_decl,
58 struct psi_impl *impl)
59 {
60 size_t i;
61 struct psi_decl_var *call_arg;
62
63 if (cb->cb_args) {
64 if (psi_plist_count(cb->cb_args) != psi_plist_count(cb_decl->args)) {
65 data->error(data, cb->token, PSI_WARNING,
66 "Argument count of callback statement of implementation '%s'"
67 "does not match argument count of callback declaration '%s'",
68 impl->func->name->val, cb_decl->func->var->name->val);
69 return false;
70 }
71
72 for (i = 0; psi_plist_get(cb->cb_args, i, &call_arg); ++i) {
73 psi_plist_get(cb_decl->args, i, &call_arg->arg);
74 }
75 }
76 return true;
77 }
78
79 bool psi_let_callback_validate(struct psi_data *data, struct psi_let_callback *cb,
80 struct psi_validate_scope *scope)
81 {
82 size_t i = 0;
83 struct psi_decl_type *cb_type;
84 struct psi_let_exp *exp = scope->current_let;
85 struct psi_decl_var *cb_var = exp->var;
86 struct psi_set_exp *set_exp, *parent_set = scope->current_set;
87
88 cb_type = psi_decl_type_get_real(cb_var->arg->type);
89 if (cb_type->type != PSI_T_FUNCTION) {
90 data->error(data, cb_var->token, PSI_WARNING,
91 "Expected a function: %s",
92 cb_var->name->val);
93 return false;
94 }
95 cb->decl = cb_type->real.func;
96
97 if (!psi_decl_validate_nodl(data, cb->decl, scope)) {
98 return false;
99 }
100 if (!psi_let_callback_validate_decl_args(data, cb, cb->decl, scope->impl)) {
101 return false;
102 }
103 while (psi_plist_get(cb->args, i++, &set_exp)) {
104 if (cb->cb_args) {
105 struct psi_decl_var *cb_var;
106
107 if (psi_plist_get(cb->cb_args, i - 1, &cb_var)) {
108 struct psi_decl_var *dvar = psi_set_exp_get_decl_var(set_exp);
109 dvar->arg = cb_var->arg;
110 }
111 }
112
113 scope->current_set = set_exp;
114 scope->cb_decl = cb->decl;
115
116 if (!psi_set_exp_validate(data, set_exp, scope)) {
117 scope->cb_decl = NULL;
118 scope->current_set = parent_set;
119 return false;
120 }
121 scope->cb_decl = NULL;
122 scope->current_set = parent_set;
123 }
124
125 return true;
126 }
127
128 void psi_let_callback_dump(int fd, struct psi_let_callback *callback,
129 unsigned level)
130 {
131 dprintf(fd, "callback(");
132 if (callback->cb_args) {
133 size_t i = 0;
134 struct psi_decl_var *cb_arg;
135
136 while (psi_plist_get(callback->cb_args, i++, &cb_arg)) {
137 if (i > 1) {
138 dprintf(fd, ", ");
139 }
140 psi_decl_var_dump(fd, cb_arg);
141 }
142 }
143 dprintf(fd, ") as %s(%s(",
144 callback->func->name->val,
145 callback->func->var->name->val);
146
147 if (callback->args) {
148 size_t i = 0, last = psi_plist_count(callback->args);
149 struct psi_set_exp *set;
150
151 dprintf(fd, "\n");
152 ++level;
153 while (psi_plist_get(callback->args, i++, &set)) {
154 psi_set_exp_dump(fd, set, level, i == last);
155 dprintf(fd, "\n");
156 }
157 --level;
158 dprintf(fd, "%s", psi_t_indent(level));
159 }
160 dprintf(fd, "))");
161 }