raising the head after a three-weeks refactoring
[m6w6/ext-psi] / src / types / let_exp.c
diff --git a/src/types/let_exp.c b/src/types/let_exp.c
new file mode 100644 (file)
index 0000000..1d8e2ec
--- /dev/null
@@ -0,0 +1,334 @@
+/*******************************************************************************
+ Copyright (c) 2016, Michael Wallner <mike@php.net>.
+ All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without
+ modification, are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright notice,
+ this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
+ FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *******************************************************************************/
+
+#include "php_psi_stdinc.h"
+#include "data.h"
+#include "call.h"
+
+#include <assert.h>
+
+struct psi_let_exp *psi_let_exp_init_ex(struct psi_decl_var *var,
+               enum psi_let_exp_kind kind, void *data)
+{
+       struct psi_let_exp *val = psi_let_exp_init(kind, data);
+       val->var = var;
+       return val;
+}
+
+struct psi_let_exp *psi_let_exp_init(enum psi_let_exp_kind kind, void *data)
+{
+       struct psi_let_exp *let = calloc(1, sizeof(*let));
+       switch (let->kind = kind) {
+       case PSI_LET_NULL:
+               assert(!data);
+               break;
+       case PSI_LET_NUMEXP:
+               let->data.num = data;
+               break;
+       case PSI_LET_CALLOC:
+               let->data.alloc = data;
+               break;
+       case PSI_LET_CALLBACK:
+               let->data.callback = data;
+               break;
+       case PSI_LET_FUNC:
+               let->data.func = data;
+               break;
+       case PSI_LET_TMP:
+               let->data.var = data;
+               break;
+       default:
+               assert(0);
+       }
+       return let;
+}
+
+void psi_let_exp_free(struct psi_let_exp **let_ptr)
+{
+       if (*let_ptr) {
+               struct psi_let_exp *let = *let_ptr;
+
+               *let_ptr = NULL;
+               switch (let->kind) {
+               case PSI_LET_NULL:
+                       break;
+               case PSI_LET_NUMEXP:
+                       psi_num_exp_free(&let->data.num);
+                       break;
+               case PSI_LET_CALLOC:
+                       psi_let_calloc_free(&let->data.alloc);
+                       break;
+               case PSI_LET_CALLBACK:
+                       psi_let_callback_free(&let->data.callback);
+                       break;
+               case PSI_LET_FUNC:
+                       psi_let_func_free(&let->data.func);
+                       break;
+               case PSI_LET_TMP:
+                       psi_decl_var_free(&let->data.var);
+                       break;
+               default:
+                       assert(0);
+               }
+               if (let->var) {
+                       psi_decl_var_free(&let->var);
+               }
+               free(let);
+       }
+}
+
+void psi_let_exp_dump(int fd, struct psi_let_exp *val, unsigned level, int last)
+{
+       if (level > 1) {
+               /* only if not directly after `set ...` */
+               dprintf(fd, "%s", psi_t_indent(level));
+       }
+
+       if (val->var) {
+               if (val->var->token) {
+                       psi_decl_var_dump(fd, val->var);
+                       dprintf(fd, " = ");
+               }
+       }
+       if (val->is_reference) {
+               dprintf(fd, "&");
+       }
+
+       switch (val->kind) {
+       case PSI_LET_NULL:
+               dprintf(fd, "NULL");
+               break;
+       case PSI_LET_TMP:
+               psi_decl_var_dump(fd, val->data.var);
+               dprintf(fd, "\t/* fqn=%s */", val->data.var->fqn);
+               break;
+       case PSI_LET_CALLOC:
+               psi_let_calloc_dump(fd, val->data.alloc);
+               break;
+       case PSI_LET_CALLBACK:
+               psi_let_callback_dump(fd, val->data.callback, level);
+               break;
+       case PSI_LET_FUNC:
+               psi_let_func_dump(fd, val->data.func, level);
+               break;
+       case PSI_LET_NUMEXP:
+               psi_num_exp_dump(fd, val->data.num);
+               break;
+
+       default:
+               assert(0);
+       }
+
+       if (val->var) {
+               dprintf(fd, "\t/* fqn=%s */", val->var->fqn);
+       }
+
+       if (level > 1) {
+               if (!last) {
+                       dprintf(fd, ",");
+               }
+       } else {
+               dprintf(fd, ";");
+       }
+}
+
+bool psi_let_exp_validate(struct psi_data *data, struct psi_let_exp *val,
+               struct psi_impl *impl)
+{
+       struct psi_decl_var *dvar = psi_let_exp_get_decl_var(val);
+
+       switch (val->kind) {
+       case PSI_LET_TMP:
+               if (!psi_decl_var_validate(data, val->data.var, impl->decl, val, NULL)) {
+                       data->error(data, dvar->token ? : **(struct psi_token ***) &val->data,
+                                       PSI_WARNING, "Unknown variable '%s'", dvar->name);
+                       return false;
+               }
+               break;
+
+       default:
+               if (!psi_decl_var_validate(data, dvar, impl->decl, val, NULL)) {
+                       data->error(data, dvar->token ? : **(struct psi_token ***) &val->data,
+                                       PSI_WARNING, "Unknown variable '%s'", dvar->name);
+                       return false;
+               }
+               break;
+       }
+
+       switch (val->kind) {
+       case PSI_LET_NULL:
+               break;
+       case PSI_LET_TMP:
+               /* e.g. let bar = &strval($bar); // decl_arg(char **bar) */
+               /* e.g. let foo = *bar;  */
+               val->var->pointer_level = val->data.var->pointer_level;
+               val->var->arg = val->data.var->arg;
+               break;
+       case PSI_LET_NUMEXP:
+               if (!psi_num_exp_validate(data, val->data.num, impl, NULL, val, NULL, NULL)) {
+                       return false;
+               }
+               break;
+       case PSI_LET_CALLOC:
+               if (!psi_num_exp_validate(data, val->data.alloc->nmemb, impl, NULL, val, NULL, NULL)) {
+                       return false;
+               }
+               if (!psi_num_exp_validate(data, val->data.alloc->size, impl, NULL, val, NULL, NULL)) {
+                       return false;
+               }
+               break;
+       case PSI_LET_CALLBACK:
+               if (!psi_let_func_validate(data, val, val->data.callback->func, impl)) {
+                       return false;
+               }
+               if (!psi_let_callback_validate(data, val, val->data.callback, impl)) {
+                       return false;
+               }
+               break;
+       case PSI_LET_FUNC:
+               if (!psi_let_func_validate(data, val, val->data.func, impl)) {
+                       return false;
+               }
+               break;
+       }
+
+       return true;
+}
+
+void *psi_let_exp_exec(struct psi_let_exp *val, struct psi_decl_arg *darg,
+               void *actual_location, size_t actual_size, struct psi_call_frame *frame)
+{
+       struct psi_call_frame_symbol *frame_sym;
+
+       frame_sym = psi_call_frame_fetch_symbol(frame, val->var);
+
+       switch (val->kind) {
+       case PSI_LET_NULL:
+               /* FIXME: sizeof_decl_arg */
+               if (darg && darg->var->array_size) {
+                       frame_sym->temp_val.ptr = ecalloc(1, psi_decl_arg_get_size(darg));
+               } else {
+                       memset(&frame_sym->temp_val, 0, sizeof(frame_sym->temp_val));
+               }
+               break;
+
+       case PSI_LET_TMP:
+               {
+                       struct psi_let_stmt *let_temp = psi_impl_get_let(frame->impl,
+                                       val->data.var);
+                       struct psi_call_frame_symbol *temp_arg;
+
+                       temp_arg = psi_call_frame_fetch_symbol(frame, let_temp->exp->var);
+                       frame_sym->temp_val = *deref_impl_val(temp_arg->ptr, val->data.var);
+               }
+               break;
+
+       case PSI_LET_CALLOC:
+               {
+                       zend_long n = psi_long_num_exp(val->data.alloc->nmemb, frame);
+                       zend_long s = psi_long_num_exp(val->data.alloc->size, frame);
+                       void *tmp = *psi_call_frame_push_auto(frame,
+                                       safe_emalloc(n, s, sizeof(void *)));
+
+                       memset(tmp, 0, n * s + sizeof(void *));
+                       frame_sym->temp_val.ptr = tmp;
+               }
+               break;
+
+       case PSI_LET_NUMEXP:
+               frame_sym->temp_val.zend.lval = psi_long_num_exp(val->data.num, frame);
+               break;
+
+       case PSI_LET_CALLBACK:
+               frame_sym->temp_val.ptr = val->data.callback->decl->sym;
+               break;
+
+       case PSI_LET_FUNC:
+               if (!psi_let_func_exec(val, val->data.func, darg, frame)) {
+                       return NULL;
+               }
+               break;
+       }
+
+       if (val->is_reference) {
+               frame_sym->ptr = &frame_sym->ival_ptr;
+       } else {
+               frame_sym->ptr = frame_sym->ival_ptr;
+       }
+
+       if (actual_location) {
+               assert(!val->is_reference || actual_size == SIZEOF_VOID_P);
+               memcpy(actual_location, frame_sym->ptr, actual_size);
+               frame_sym->ptr = actual_location;
+       }
+
+       return frame_sym->ptr;
+}
+
+struct psi_let_func* psi_let_exp_get_func(struct psi_let_exp* exp)
+{
+       if (exp) {
+               switch (exp->kind) {
+               case PSI_LET_CALLBACK:
+                       return exp->data.callback->func;
+               case PSI_LET_FUNC:
+                       return exp->data.func;
+               default:
+                       break;
+               }
+       }
+       return NULL;
+}
+
+struct psi_decl_var *psi_let_exp_get_decl_var(struct psi_let_exp *val)
+{
+       if (!val->var) {
+               const char *name = psi_let_exp_get_decl_var_name(val);
+
+               if (name) {
+                       val->var = psi_decl_var_init(name, 0, 0);
+               }
+       }
+       return val->var;
+}
+
+struct psi_impl_var *psi_let_exp_get_impl_var(struct psi_let_exp *val)
+{
+       struct psi_let_func* fn = psi_let_exp_get_func(val);
+       return fn ? fn->var : NULL;
+}
+
+const char* psi_let_exp_get_decl_var_name(struct psi_let_exp* val)
+{
+       struct psi_impl_var* var;
+       if (val->var) {
+               return val->var->name;
+       }
+       var = psi_let_exp_get_impl_var(val);
+       if (var) {
+               return &var->name[1];
+       }
+       return NULL;
+}