- for (i = 0; i < set->vars->count; ++i) {
- decl_var *svar = set->vars->vars[i];
- if (!svar->arg && !locate_decl_var_arg(svar, ref_list, NULL)) {
- data->error(svar->token, PSI_WARNING, "Unknown variable '%s' in `set` statement", svar->name);
- return 0;
- }
- }
-
- if (!ref) {
- ref = set_var->arg;
- }
- ref_type = real_decl_type(ref->type);
-
- if (set->count) {
- int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
- int is_pointer_to_struct = (ref_type->type == PSI_T_STRUCT && ref->var->pointer_level);
-
- if (!is_to_array && !is_pointer_to_struct) {
- data->error(set->func->token, E_WARNING, "Inner `set` statement casts only work with "
- "to_array() casts on structs or pointers: %s(%s...", set->func->name, set->vars->vars[0]->name);
- return 0;
- }
- }
- if (set->num) {
- if (!validate_num_exp(data, ref_list, ref, set->num)) {
- return 0;
- }
- }
-
- if (ref_type->type == PSI_T_STRUCT) {
- /* to_array(struct, to_...) */
- for (i = 0; i < set->count; ++i) {
- decl_var *sub_var = set->inner[i]->vars->vars[0];
- decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
-
- if (sub_ref) {
- if (!validate_set_value_ex(data, set->inner[i], sub_ref, ref_type->strct->args)) {
- return 0;
- }
- }
- }
- } else if (set->count == 1) {
- /* to_array(ptr, to_string(*ptr)) */
- decl_var *sub_var = set->inner[0]->vars->vars[0];
- decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list, ref);
-
- if (sub_ref) {
- if (strcmp(sub_var->name, set_var->name)) {
- data->error(sub_var->token, E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
- return 0;
- }
- if (!validate_set_value_ex(data, set->inner[0], sub_ref, ref_list)) {
- return 0;
- }
- }
- } else if (set->count > 1) {
- data->error(set->func->token, E_WARNING, "Inner `set` statement casts on pointers may only occur once");
- return 0;
- }
-
- return 1;
-}
-static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
- va_list argp;
- decl_args args = {0};
- int check;
-
- va_start(argp, set);
- decl_var_arg_v(&args, argp);
- va_end(argp);
-
- check = validate_set_value_ex(data, set, NULL, &args);
- if (args.args) {
- free(args.args);
- }
- return check;
-}
-static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
- if (decls) {
- size_t i;
-
- for (i = 0; i < decls->count; ++i) {
- if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
- ret->decl = decls->list[i]->func;
- return decls->list[i];
- }
- }
- }
-
- return NULL;
-}
-static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
- return_stmt *ret;
-
- /* we must have exactly one ret stmt delcaring the native func to call */
- /* and which type cast to apply */
- if (impl->stmts->ret.count != 1) {
- if (impl->stmts->ret.count > 1) {
- data->error(impl->stmts->ret.list[1]->token, PSI_WARNING,
- "Too many `return` statements for implmentation %s;"
- " found %zu, exactly one is needed",
- impl->func->name, impl->stmts->ret.count);
- } else {
- data->error(impl->func->token, PSI_WARNING,
- "Missing `return` statement for implementation %s",
- impl->func->name);
- }
- return 0;
- }
-
- ret = impl->stmts->ret.list[0];
-
- if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
- data->error(ret->token, PSI_WARNING,
- "Missing declaration '%s' for `return` statment for implementation %s",
- ret->set->vars->vars[0]->name, impl->func->name);
- return 0;
- }
-
- if (!validate_set_value(data, ret->set, 1, &ret->decl, impl->decl->args ? (int) impl->decl->args->count : 0, impl->decl->args ? impl->decl->args->args : NULL, 0)) {
- return 0;
- }
-
- impl->decl->impl = impl;
-
- return 1;
-}
-
-static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
- size_t i, j;
- /* we can have multiple let stmts */
-
- /* check that we have a decl arg for every let stmt */
- for (i = 0; i < impl->stmts->let.count; ++i) {
- let_stmt *let = impl->stmts->let.list[i];
- decl_var *let_var;
- int check = 0;
-
- if (let->val && let->val->kind == PSI_LET_TMP) {
- let_var = let->val->data.var;
- } else {
- let_var = let->var;
- }
-
- if (!locate_decl_var_arg(let_var, impl->decl->args, impl->decl->func)) {
- data->error(let_var->token, PSI_WARNING, "Unknown variable '%s' in `let` statement"
- " of implementation '%s'", let_var->name, impl->func->name);
- return 0;
- }
-
- switch (let->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; */
- let->var->pointer_level = let->val->data.var->pointer_level;
- let->var->arg = init_decl_arg(
- init_decl_type(
- real_decl_type(let->val->data.var->arg->type)->type,
- real_decl_type(let->val->data.var->arg->type)->name),
- init_decl_var(
- let->var->name,
- let->var->pointer_level,
- let->var->array_size));
- break;
- case PSI_LET_NUMEXP:
- if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.num)) {
- return 0;
- }
- break;
- case PSI_LET_CALLOC:
- if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->nmemb)) {
- return 0;
- }
- if (!validate_num_exp(data, impl->decl->args, impl->decl->func, let->val->data.alloc->size)) {
- return 0;
- }
- break;
- case PSI_LET_FUNC:
- if (impl->func->args) {
- for (j = 0; j < impl->func->args->count; ++j) {
- impl_arg *iarg = impl->func->args->args[j];
-
- if (!strcmp(let->val->data.func->var->name, iarg->var->name)) {
- let->val->data.func->arg = iarg;
- check = 1;
- break;
- }
- }
- }
- if (!check) {
- data->error(let->var->token, PSI_WARNING, "Unknown value '$%s' of `let` statement"
- " for variable '%s' of implementation '%s'",
- let->val->data.func->var->name, let->var->name, impl->func->name);
- return 0;
- }
- break;
- }
- }
-
- /* check that we have a let stmt for every decl arg */
- if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
- decl_arg *darg = impl->decl->args->args[i];
- int check = 0;
-
- for (j = 0; j < impl->stmts->let.count; ++j) {
- let_stmt *let = impl->stmts->let.list[j];
-
- if (!strcmp(let->var->name, darg->var->name)) {
- darg->let = let;
- check = 1;
- break;
- }
- }
- if (!check) {
- data->error(impl->func->token, PSI_WARNING,
- "Missing `let` statement for arg '%s %.*s%s'"
- " of declaration '%s' for implementation '%s'",
- darg->type->name, (int) darg->var->pointer_level, "*****",
- darg->var->name, impl->decl->func->var->name, impl->func->name);
- return 0;
- }
- }
-
- return 1;
-}
-static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
- size_t i, j, k;
- /* we can have any count of set stmts; processing out vars */
- /* check that set stmts reference known variables */
- for (i = 0; i < impl->stmts->set.count; ++i) {
- set_stmt *set = impl->stmts->set.list[i];
- int check = 0;
-
- if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
- impl_arg *iarg = impl->func->args->args[j];
-
- if (!strcmp(set->var->name, iarg->var->name)) {
- set->arg = iarg;
- check = 1;
- break;
- }
- }
- if (!check) {
- data->error(set->var->token, PSI_WARNING, "Unknown variable '$%s' of `set` statement"
- " of implementation '%s'",
- set->var->name, impl->func->name);
- return 0;
- }
-
- for (j = 0; j < set->val->vars->count; ++j) {
- decl_var *set_var = set->val->vars->vars[j];
-
- check = 0;
- if (impl->decl->args) {
- for (k = 0; k < impl->decl->args->count; ++k) {
- decl_arg *set_arg = impl->decl->args->args[k];
-
- if (!strcmp(set_var->name, set_arg->var->name)) {
- check = 1;
- set_var->arg = set_arg;
- if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
- return 0;
- }
- break;
- }
- }
- }
- if (!check) {
- for (k = 0; k < impl->stmts->let.count; ++k) {
- let_stmt *let = impl->stmts->let.list[k];
-
- /* check temp vars */
- if (let->val && let->val->kind == PSI_LET_TMP) {
- if (!strcmp(set_var->name, let->var->name)) {
- check = 1;
- set_var->arg = let->var->arg;
- if (!validate_set_value(data, set->val, 1, &set_var->arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
- return 0;
- }
- break;
- }
- }
- }
- }
-
- if (!check) {
- data->error(set_var->token, PSI_WARNING, "Unknown value '%s' of `set` statement"
- " for variable '$%s' of implementation '%s'",
- set_var->name, set->arg->var->name, impl->func->name);
- return 0;
- }
- }
- }
- return 1;
-}
-static inline decl *locate_free_decl(decls *decls, free_call *f) {
- if (decls) {
- size_t i;
-
- for (i = 0; i < decls->count; ++i) {
- if (!strcmp(decls->list[i]->func->var->name, f->func)) {
- f->decl = decls->list[i];
- return decls->list[i];
- }
- }
- }
-
- return NULL;
-}
-static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
- size_t i, j, k, l;
- /* we can have any count of free stmts; freeing any out vars */
- for (i = 0; i < impl->stmts->fre.count; ++i) {
- free_stmt *fre = impl->stmts->fre.list[i];
-
- for (j = 0; j < fre->calls->count; ++j) {
- free_call *free_call = fre->calls->list[j];
-
- /* first find the decl of the free func */
- if (!locate_free_decl(data->decls, free_call)) {
- data->error(free_call->token, PSI_WARNING,
- "Missing declaration '%s' in `free` statement"
- " of implementation '%s'",
- free_call->func, impl->func->name);
- return 0;
- }
- if (!impl->decl->args) {
- data->error(free_call->token, PSI_WARNING,
- "Declaration '%s' of implementation '%s'"
- " does not have any arguments to free",
- impl->decl->func->var->name, impl->func->name);
- }
-
- /* now check for known vars */
- for (l = 0; l < free_call->vars->count; ++l) {
- int check = 0;
- decl_var *free_var = free_call->vars->vars[l];
-
- for (k = 0; k < impl->decl->args->count; ++k) {
- decl_arg *free_arg = impl->decl->args->args[k];
-
- if (!strcmp(free_var->name, free_arg->var->name)) {
- check = 1;
- free_var->arg = free_arg;
- break;
- }
- }
-
- if (!check) {
- data->error(free_var->token, PSI_WARNING,
- "Unknown variable '%s' of `free` statement"
- " of implementation '%s'",
- free_var->name, impl->func->name);
- return 0;
- }
- }
- }
- }
- return 1;
-}
-static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
- if (!impl->stmts) {
- data->error(impl->func->token, PSI_WARNING,
- "Missing body for implementation %s!",
- impl->func->name);
- return 0;
- }
-
- if (!validate_impl_ret_stmt(data, impl)) {
- return 0;
- }
-
- if (!validate_impl_let_stmts(data, impl)) {
- return 0;
- }
- if (!validate_impl_set_stmts(data, impl)) {
- return 0;
- }
- if (!validate_impl_free_stmts(data, impl)) {
- return 0;
- }
-
- return 1;
-}