fix leaks
[m6w6/ext-psi] / src / context_validate.c
1 #ifdef HAVE_CONFIG_H
2 # include "config.h"
3 #endif
4
5 #include "php.h"
6
7 #include <sys/param.h>
8 #include <dlfcn.h>
9
10 #include "php_psi.h"
11 #include "php_psi_stdinc.h"
12 #include "php_psi_stdtypes.h"
13 #include "php_psi_macros.h"
14 #include "php_psi_redirs.h"
15
16 #include "parser.h"
17 #include "calc.h"
18 #include "marshal.h"
19 #include "engine.h"
20
21 static int validate_lib(PSI_Data *data, void **dlopened) {
22 char lib[MAXPATHLEN];
23 const char *ptr = data->psi.file.ln;
24 size_t len;
25
26 if (!ptr) {
27 /* FIXME: assume stdlib */
28 return 1;
29 } else if (!strchr(ptr, '/')) {
30 len = snprintf(lib, MAXPATHLEN, "lib%s.%s", ptr, PHP_PSI_SHLIB_SUFFIX);
31 if (MAXPATHLEN == len) {
32 data->error(data, NULL, PSI_WARNING, "Library name too long: '%s'", ptr);
33 }
34 lib[len] = 0;
35 ptr = lib;
36 }
37 if (!(*dlopened = dlopen(ptr, RTLD_LAZY|RTLD_LOCAL))) {
38 data->error(data, NULL, PSI_WARNING, "Could not open library '%s': %s.",
39 data->psi.file.ln, dlerror());
40 return 0;
41 }
42 return 1;
43 }
44
45 static inline int locate_decl_type_alias(decl_typedefs *defs, decl_type *type) {
46 size_t i;
47 struct psi_std_type *stdtyp;
48
49 if (type->real) {
50 return 1;
51 }
52 if (defs) for (i = 0; i < defs->count; ++i) {
53 decl_arg *def = defs->list[i];
54
55 if (def->type->type != type->type && !strcmp(def->var->name, type->name)) {
56 type->real = def->type;
57 return 1;
58 }
59 }
60 for (stdtyp = &psi_std_types[0]; stdtyp->type_tag; ++stdtyp) {
61 if (!strcmp(type->name, stdtyp->alias ?: stdtyp->type_name)) {
62 type->type = stdtyp->type_tag;
63 return 1;
64 }
65 }
66
67 return 0;
68 }
69
70 static inline int locate_decl_type_struct(decl_structs *structs, decl_type *type) {
71 size_t i;
72
73 if (type->strct) {
74 return 1;
75 }
76 if (structs) for (i = 0; i < structs->count; ++i) {
77 if (!strcmp(structs->list[i]->name, type->name)) {
78 type->strct = structs->list[i];
79 return 1;
80 }
81 }
82 return 0;
83 }
84
85 static inline int locate_decl_type_union(decl_unions *unions, decl_type *type) {
86 size_t i;
87
88 if (type->unn) {
89 return 1;
90 }
91 if (unions) for (i = 0; i < unions->count; ++i) {
92 if (!strcmp(unions->list[i]->name, type->name)) {
93 type->unn = unions->list[i];
94 return 1;
95 }
96 }
97 return 0;
98 }
99
100 static inline int locate_decl_type_enum(decl_enums *enums, decl_type *type) {
101 size_t i;
102
103 if (type->enm) {
104 return 1;
105 }
106 if (enums) for (i = 0; i < enums->count; ++i) {
107 if (!strcmp(enums->list[i]->name, type->name)) {
108 type->enm = enums->list[i];
109 return 1;
110 }
111 }
112 return 0;
113 }
114
115 static inline int locate_decl_type_decl(decls *decls, decl_type *type) {
116 size_t i;
117
118 if (type->func) {
119 return 1;
120 }
121 if (decls) for (i = 0; i < decls->count; ++i) {
122 if (!strcmp(decls->list[i]->func->var->name, type->name)) {
123 type->func = decls->list[i];
124 return 1;
125 }
126 }
127
128 return 0;
129 }
130
131 static inline int validate_decl_struct(PSI_Data *data, decl_struct *s);
132 static inline int validate_decl_union(PSI_Data *data, decl_union *u);
133 static inline int validate_decl_enum(PSI_Data *data, decl_enum *e);
134
135 static inline int validate_decl_type(PSI_Data *data, decl_type *type) {
136 switch (type->type) {
137 case PSI_T_CHAR:
138 case PSI_T_SHORT:
139 case PSI_T_INT:
140 case PSI_T_LONG:
141 case PSI_T_NAME:
142 if (!locate_decl_type_alias(data->defs, type)) {
143 return 0;
144 }
145 if (type->real) {
146 return validate_decl_type(data, type->real);
147 }
148 return 1;
149 case PSI_T_STRUCT:
150 if (!locate_decl_type_struct(data->structs, type)) {
151 return 0;
152 }
153 break;
154 case PSI_T_UNION:
155 if (!locate_decl_type_union(data->unions, type)) {
156 return 0;
157 }
158 break;
159 case PSI_T_ENUM:
160 if (!locate_decl_type_enum(data->enums, type)) {
161 return 0;
162 }
163 break;
164 case PSI_T_FUNCTION:
165 if (!locate_decl_type_decl(data->decls, type)) {
166 return 0;
167 }
168 break;
169 }
170 return 1;
171 }
172 static inline int validate_decl_typedef(PSI_Data *data, decl_arg *def) {
173 if (!validate_decl_type(data, def->type)) {
174 data->error(data, def->token, PSI_WARNING,
175 "Type '%s' cannot be aliased to %s'%s'",
176 def->type->name, def->type->type == PSI_T_STRUCT?"struct ":"",
177 def->var->name);
178 return 0;
179 }
180 if (def->type->type == PSI_T_VOID && def->var->pointer_level) {
181 def->type->type = PSI_T_POINTER;
182 }
183 return 1;
184 }
185
186 static inline int validate_constant(PSI_Data *data, constant *c) {
187 /* FIXME */
188 return 1;
189 }
190
191 static inline int validate_decl_arg(PSI_Data *data, decl_arg *arg) {
192 if (!validate_decl_type(data, arg->type)) {
193 data->error(data, arg->type->token, PSI_WARNING,
194 "Cannot use '%s' as type for '%s'",
195 arg->type->name, arg->var->name);
196 return 0;
197 }
198 return 1;
199 }
200
201 static int psi_sort_struct_arg_cmp(const void *_a, const void *_b) {
202 decl_arg *a = *(decl_arg **)_a, *b = *(decl_arg **)_b;
203
204 if (a->layout->pos == b->layout->pos) {
205 if (a->layout->len == b->layout->len) {
206 return 0;
207 } else if (a->layout->len > b->layout->len) {
208 return -1;
209 } else {
210 return 1;
211 }
212 } else if (a->layout->pos > b->layout->pos) {
213 return 1;
214 } else {
215 return -1;
216 }
217 }
218 static void psi_sort_struct_arg_swp(void *a, void *b) {
219 decl_arg **_a = a, **_b = b, *_c;
220
221 _c = *_b;
222 *_b = *_a;
223 *_a = _c;
224 }
225 static inline void psi_sort_struct_args(void **args, size_t count) {
226 zend_insert_sort(args, count, sizeof(*args),
227 psi_sort_struct_arg_cmp, psi_sort_struct_arg_swp);
228 }
229
230 static inline int validate_decl_struct_darg(PSI_Data *data, decl_arg *darg, void *current) {
231 decl_type *real = real_decl_type(darg->type);
232
233 /* pre-validate any structs/unions/enums */
234 switch (real->type) {
235 case PSI_T_STRUCT:
236 if (current && current == real->strct) {
237 return 1;
238 }
239 if (!validate_decl_struct(data, real->strct)) {
240 return 0;
241 }
242 break;
243 case PSI_T_UNION:
244 if (current && current == real->unn) {
245 return 1;
246 }
247 if (!validate_decl_union(data, real->unn)) {
248 return 0;
249 }
250 break;
251 case PSI_T_ENUM:
252 if (current && current == real->enm) {
253 return 1;
254 }
255 if (!validate_decl_enum(data, real->enm)) {
256 return 0;
257 }
258 break;
259 }
260
261 return 1;
262 }
263
264 static inline size_t sizeof_decl_arg(decl_arg *darg) {
265 size_t size;
266 decl_type *real = real_decl_type(darg->type);
267
268 if (darg->var->array_size) {
269 if (darg->var->pointer_level > 2) {
270 size = psi_t_size(PSI_T_POINTER) * darg->var->array_size;
271 } else {
272 size = psi_t_size(real->type) * darg->var->array_size;
273 }
274 } else if (darg->var->pointer_level) {
275 size = psi_t_size(PSI_T_POINTER);
276 } else {
277 switch (real->type) {
278 case PSI_T_UNION:
279 size = real->unn->size;
280 break;
281 case PSI_T_STRUCT:
282 size = real->strct->size;
283 break;
284 case PSI_T_ENUM:
285 default:
286 size = psi_t_size(real->type);
287 break;
288 }
289 }
290
291 ZEND_ASSERT(size);
292
293 return size;
294 }
295
296 static inline size_t alignof_decl_type(decl_type *t);
297 static inline size_t alignof_decl_arg(decl_arg *darg);
298 static inline size_t alignof_decl_union(decl_union *u);
299 static inline size_t alignof_decl_struct(decl_struct *s);
300
301 static inline size_t alignof_decl_args(decl_args *args) {
302 size_t i, maxalign = 0;
303
304 for (i = 0; i < args->count; ++i) {
305 decl_arg *darg = args->args[i];
306 size_t align = alignof_decl_arg(darg);
307
308 if (align > maxalign) {
309 maxalign = align;
310 }
311 }
312
313 return maxalign;
314 }
315
316 static inline size_t alignof_decl_struct(decl_struct *s) {
317 if (!s->align) {
318 s->align = alignof_decl_args(s->args);
319 }
320 return s->align;
321 }
322
323 static inline size_t alignof_decl_union(decl_union *u) {
324 if (!u->align) {
325 u->align = alignof_decl_args(u->args);
326 }
327 return u->align;
328 }
329
330 static inline size_t alignof_decl_type(decl_type *t) {
331 decl_type *real = real_decl_type(t);
332 size_t align;
333
334 switch (real->type) {
335 case PSI_T_STRUCT:
336 align = alignof_decl_struct(real->strct);
337 break;
338 case PSI_T_UNION:
339 align = alignof_decl_union(real->unn);
340 break;
341 case PSI_T_ENUM:
342 default:
343 align = psi_t_alignment(real->type);
344 }
345
346 return align;
347 }
348
349 static inline size_t alignof_decl_arg(decl_arg *darg) {
350 size_t align;
351
352 if (darg->var->pointer_level && (!darg->var->array_size || darg->var->pointer_level > 2)) {
353 align = psi_t_alignment(PSI_T_POINTER);
354 } else {
355 align = alignof_decl_type(darg->type);
356 }
357
358 return align;
359 }
360
361 static inline size_t align_decl_arg(decl_arg *darg, size_t *pos, size_t *len) {
362 size_t align = alignof_decl_arg(darg);
363
364 ZEND_ASSERT(align);
365
366 *len = sizeof_decl_arg(darg);
367 *pos = psi_align(align, *pos);
368
369 return align;
370 }
371
372 static inline int validate_decl_struct(PSI_Data *data, decl_struct *s) {
373 size_t i, pos, len, size, align;
374
375 if (!s->size && !s->args->count) {
376 data->error(data, s->token, PSI_WARNING,
377 "Cannot compute size of empty struct %s",
378 s->name);
379 return 0;
380 }
381
382 for (i = 0; i < s->args->count; ++i) {
383 decl_arg *darg = s->args->args[i];
384
385 if (!validate_decl_arg(data, darg)) {
386 return 0;
387 }
388
389 ZEND_ASSERT(!darg->var->arg || darg->var->arg == darg);
390 darg->var->arg = darg;
391
392 if (!validate_decl_struct_darg(data, darg, s)) {
393 return 0;
394 } else if (darg->layout) {
395 pos = darg->layout->pos;
396
397 align = align_decl_arg(darg, &pos, &len);
398
399 if (darg->layout->len != len) {
400 data->error(data, darg->token, PSI_WARNING,
401 "Computed size %zu of %s.%s does not match"
402 " pre-defined size %zu of type '%s'",
403 len, s->name, darg->var->name, darg->layout->len,
404 darg->type->name);
405 }
406 if (darg->layout->pos != pos) {
407 data->error(data, darg->token, PSI_WARNING,
408 "Computed offset %zu of %s.%s does not match"
409 " pre-defined offset %zu",
410 pos, s->name, darg->var->name, darg->layout->pos);
411 }
412 } else {
413 if (i) {
414 pos = s->args->args[i-1]->layout->pos +
415 s->args->args[i-1]->layout->len;
416 } else {
417 pos = 0;
418 }
419
420 align = align_decl_arg(darg, &pos, &len);
421 darg->layout = init_decl_struct_layout(pos, len);
422 }
423
424 if (align > s->align) {
425 s->align = align;
426 }
427 }
428
429 psi_sort_struct_args((void **) s->args->args, s->args->count);
430
431 if (s->args->count) {
432 decl_arg *darg = s->args->args[s->args->count-1];
433
434 size = darg->layout->pos + darg->layout->len;
435 if (s->size < size) {
436 s->size = psi_align(size, s->align);
437 }
438 }
439
440 return 1;
441 }
442
443 static inline int validate_decl_union(PSI_Data *data, decl_union *u) {
444 size_t i, pos, len, size = 0, align;
445
446 if (!u->size && !u->args->count) {
447 data->error(data, u->token, PSI_WARNING,
448 "Cannot compute size of empty union %s",
449 u->name);
450 return 0;
451 }
452
453 for (i = 0; i < u->args->count; ++i) {
454 decl_arg *darg = u->args->args[i];
455
456 if (!validate_decl_arg(data, darg)) {
457 return 0;
458 }
459
460 ZEND_ASSERT(!darg->var->arg || darg->var->arg == darg);
461 darg->var->arg = darg;
462
463 if (!validate_decl_struct_darg(data, darg, u)) {
464 return 0;
465 } else if (darg->layout) {
466 pos = darg->layout->pos;
467
468 align = align_decl_arg(darg, &pos, &len);
469
470 if (darg->layout->pos != 0) {
471 data->error(data, darg->token, PSI_WARNING,
472 "Offset of %s.%s should be 0",
473 u->name, darg->var->name);
474 darg->layout->pos = 0;
475 }
476 if (darg->layout->len != len) {
477 data->error(data, darg->token, PSI_WARNING,
478 "Computed size %zu of %s.%s does not match"
479 " pre-defined size %zu of type '%s'",
480 len, u->name, darg->var->name, darg->layout->len,
481 darg->type->name);
482 }
483 } else {
484 pos = 0;
485
486 align = align_decl_arg(darg, &pos, &len);
487 darg->layout = init_decl_struct_layout(pos, len);
488
489 }
490 if (len > size) {
491 size = len;
492 }
493 if (align > u->align) {
494 u->align = align;
495 }
496 }
497
498 psi_sort_struct_args((void **) u->args->args, u->args->count);
499
500 if (u->size < size) {
501 u->size = psi_align(size, u->align);
502 }
503
504 return 1;
505 }
506
507 static const char * const abi_ccs[] = {
508 "default", /* \ */
509 "extern", /* > - all the same */
510 "cdecl", /* / */
511 "mscdecl",
512 "stdcall",
513 "fastcall",
514 };
515 static inline int validate_decl_abi(PSI_Data *data, decl_abi *abi) {
516 size_t i;
517
518 for (i = 0; i < sizeof(abi_ccs)/sizeof(char*); ++i) {
519 if (strcasecmp(abi->convention, abi_ccs[i])) {
520 return 1;
521 }
522 }
523 return 0;
524 }
525 static inline int validate_decl_func(PSI_Data *data, void *dl, decl *decl, decl_arg *func)
526 {
527 struct psi_func_redir *redir;
528
529 if (!strcmp(func->var->name, "dlsym")) {
530 data->error(data, func->token, PSI_WARNING, "Cannot dlsym dlsym (sic!)");
531 return 0;
532 }
533
534 for (redir = &psi_func_redirs[0]; redir->name; ++redir) {
535 if (!strcmp(func->var->name, redir->name)) {
536 decl->call.sym = redir->func;
537 }
538 }
539 if (!decl->call.sym) {
540 #ifndef RTLD_NEXT
541 # define RTLD_NEXT ((void *) -1l)
542 #endif
543 decl->call.sym = dlsym(dl ?: RTLD_NEXT, func->var->name);
544 if (!decl->call.sym) {
545 data->error(data, func->token, PSI_WARNING,
546 "Failed to locate symbol '%s': %s",
547 func->var->name, dlerror() ?: "not found");
548 }
549 }
550 return 1;
551 }
552 static inline int validate_decl_nodl(PSI_Data *data, decl *decl) {
553 if (!validate_decl_abi(data, decl->abi)) {
554 data->error(data, decl->abi->token, PSI_WARNING,
555 "Invalid calling convention: '%s'", decl->abi->token->text);
556 return 0;
557 }
558 if (!validate_decl_arg(data, decl->func)) {
559 return 0;
560 }
561 if (decl->args) {
562 size_t i;
563
564 for (i = 0; i < decl->args->count; ++i) {
565 if (!validate_decl_arg(data, decl->args->args[i])) {
566 return 0;
567 }
568 }
569 }
570 return 1;
571 }
572 static inline int validate_decl(PSI_Data *data, void *dl, decl *decl) {
573 if (!validate_decl_nodl(data, decl)) {
574 return 0;
575 }
576 if (!validate_decl_func(data, dl, decl, decl->func)) {
577 return 0;
578 }
579 return 1;
580 }
581 static inline decl_arg *locate_decl_var_arg(decl_var *var, decl_args *args, decl_arg *func) {
582 size_t i;
583
584 if (args) for (i = 0; i < args->count; ++i) {
585 decl_arg *arg = args->args[i];
586
587 if (!strcmp(var->name, arg->var->name)) {
588 ZEND_ASSERT(!var->arg || var->arg == arg);
589 return var->arg = arg;
590 }
591 }
592 if (func && !strcmp(var->name, func->var->name)) {
593 return var->arg = func;
594 }
595
596 return NULL;
597 }
598 static inline decl_arg *locate_struct_member(decl_struct *s, decl_var *var) {
599 if (s->args) {
600 return locate_decl_var_arg(var, s->args, NULL);
601 }
602
603 return NULL;
604 }
605 static inline constant *locate_num_exp_constant(num_exp *exp, constants *consts) {
606 size_t i;
607
608 for (i = 0; i < consts->count; ++i) {
609 constant *cnst = consts->list[i];
610
611 if (!strcmp(cnst->name, exp->u.numb)) {
612 free(exp->u.numb);
613 return exp->u.cnst = cnst;
614 }
615 }
616
617 return NULL;
618 }
619 static inline decl_enum_item *locate_num_exp_enum_item_ex(num_exp *exp, decl_enum *e) {
620 size_t k;
621
622 if (e) for (k = 0; k < e->items->count; ++k) {
623 decl_enum_item *i = e->items->list[k];
624
625 if (!strcmp(i->name, exp->u.dvar->name)) {
626 free_decl_var(exp->u.dvar);
627 exp->t = PSI_T_ENUM;
628 exp->u.enm = i;
629 return i;
630 }
631 }
632 return NULL;
633 }
634 static inline decl_enum_item *locate_num_exp_enum_item(num_exp *exp, decl_enums *enums) {
635 size_t j;
636
637 if (enums) for (j = 0; j < enums->count; ++j) {
638 decl_enum *e = enums->list[j];
639 decl_enum_item *i = locate_num_exp_enum_item_ex(exp, e);
640
641 if (i) {
642 return i;
643 }
644 }
645 return NULL;
646 }
647 static inline int validate_num_exp(PSI_Data *data, num_exp *exp, decl_args *dargs, decl_arg *func, decl_enum *enm) {
648 if (exp->operand) {
649 switch (exp->operator) {
650 case PSI_T_PLUS:
651 exp->calculator = psi_calc_add;
652 break;
653 case PSI_T_MINUS:
654 exp->calculator = psi_calc_sub;
655 break;
656 case PSI_T_ASTERISK:
657 exp->calculator = psi_calc_mul;
658 break;
659 case PSI_T_SLASH:
660 exp->calculator = psi_calc_div;
661 break;
662 EMPTY_SWITCH_DEFAULT_CASE();
663 }
664 if (!validate_num_exp(data, exp->operand, dargs, func, enm)) {
665 return 0;
666 }
667 }
668 switch (exp->t) {
669 case PSI_T_NAME:
670 if (!locate_decl_var_arg(exp->u.dvar, dargs, func)) {
671 if (!locate_num_exp_enum_item(exp, data->enums) && !locate_num_exp_enum_item_ex(exp, enm)) {
672 data->error(data, exp->token, PSI_WARNING, "Unknown variable '%s' in numeric expression",
673 exp->u.dvar->name);
674 return 0;
675 }
676 }
677 return 1;
678 case PSI_T_NSNAME:
679 if (!locate_num_exp_constant(exp, data->consts)) {
680 data->error(data, exp->token, PSI_WARNING, "Unknown constant '%s' in numeric expression",
681 exp->u.numb);
682 return 0;
683 }
684 return 1;
685 case PSI_T_NUMBER:
686 case PSI_T_ENUM:
687 return 1;
688 default:
689 return 0;
690 }
691 }
692
693 static inline int validate_decl_enum(PSI_Data *data, decl_enum *e) {
694 size_t j;
695
696 if (!e->items || !e->items->count) {
697 data->error(data, e->token, PSI_WARNING, "Empty enum '%s'", e->name);
698 return 0;
699 }
700
701 for (j = 0; j < e->items->count; ++j) {
702 decl_enum_item *i = e->items->list[j];
703
704 if (!i->num) {
705 if (j) {
706 i->inc.t = PSI_T_NUMBER;
707 i->inc.u.numb = "1";
708 i->inc.operator = PSI_T_PLUS;
709 i->inc.operand = i->prev->num ?: &i->prev->inc;
710 i->num = &i->inc;
711 } else {
712 i->inc.t = PSI_T_NUMBER;
713 i->inc.u.numb = "0";
714 i->num = &i->inc;
715 }
716 }
717 if (!validate_num_exp(data, i->num, NULL, NULL, e)) {
718 return 0;
719 }
720 }
721
722 return 1;
723 }
724
725 static inline int validate_set_value_handler(set_value *set) {
726 switch (set->func->type) {
727 case PSI_T_TO_BOOL:
728 set->func->handler = psi_to_bool;
729 break;
730 case PSI_T_TO_INT:
731 set->func->handler = psi_to_int;
732 break;
733 case PSI_T_TO_FLOAT:
734 set->func->handler = psi_to_double;
735 break;
736 case PSI_T_TO_STRING:
737 set->func->handler = psi_to_string;
738 break;
739 case PSI_T_TO_ARRAY:
740 set->func->handler = psi_to_array;
741 break;
742 case PSI_T_TO_OBJECT:
743 set->func->handler = psi_to_object;
744 break;
745 case PSI_T_VOID:
746 set->func->handler = psi_to_void;
747 break;
748 case PSI_T_ELLIPSIS:
749 if (set->outer.set && set->outer.set->func->type == PSI_T_TO_ARRAY) {
750 set->func->handler = psi_to_recursive;
751 set->inner = set->outer.set->inner;
752 break;
753 }
754 /* no break */
755 default:
756 return 0;
757 }
758 return 1;
759 }
760 static inline void decl_var_arg_v(decl_args *args, va_list argp) {
761 int argc;
762 decl_arg **argv;
763
764 memset(args, 0, sizeof(*args));
765
766 while ((argc = va_arg(argp, int))) {
767 argv = va_arg(argp, decl_arg **);
768 while (argc--) {
769 add_decl_arg(args, *argv++);
770 }
771 }
772 }
773 static inline int validate_set_value_ex(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
774 size_t i;
775 decl_type *ref_type;
776 decl_var *set_var = set->vars->vars[0];
777
778 if (!validate_set_value_handler(set)) {
779 data->error(data, set->func->token, PSI_WARNING, "Invalid cast '%s' in `set` statement", set->func->name);
780 return 0;
781 }
782
783 for (i = 0; i < set->vars->count; ++i) {
784 decl_var *svar = set->vars->vars[i];
785 if (!svar->arg && !locate_decl_var_arg(svar, ref_list, NULL)) {
786 data->error(data, svar->token, PSI_WARNING, "Unknown variable '%s' in `set` statement", svar->name);
787 return 0;
788 }
789 }
790
791 if (!ref) {
792 ref = set_var->arg;
793 }
794 ref_type = real_decl_type(ref->type);
795
796 if (set->inner && set->inner->count) {
797 int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
798 int is_pointer_to_struct = (ref_type->type == PSI_T_STRUCT && ref->var->pointer_level);
799
800 if (!is_to_array && !is_pointer_to_struct) {
801 data->error(data, set->func->token, E_WARNING, "Inner `set` statement casts only work with "
802 "to_array() casts on structs or pointers: %s(%s...", set->func->name, set->vars->vars[0]->name);
803 return 0;
804 }
805 }
806 if (set->num) {
807 if (!validate_num_exp(data, set->num, ref_list, ref, NULL)) {
808 return 0;
809 }
810 }
811
812 if (set->inner && ref_type->type == PSI_T_STRUCT) {
813 /* to_array(struct, to_...) */
814 if (!set->outer.set || set->outer.set->inner->vals != set->inner->vals) {
815 for (i = 0; i < set->inner->count; ++i) {
816 decl_var *sub_var = set->inner->vals[i]->vars->vars[0];
817 decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
818
819 if (sub_ref) {
820 if (!validate_set_value_ex(data, set->inner->vals[i], sub_ref, ref_type->strct->args)) {
821 return 0;
822 }
823 }
824 }
825 }
826 } else if (set->inner && set->inner->count == 1) {
827 /* to_array(ptr, to_string(*ptr)) */
828 decl_var *sub_var = set->inner->vals[0]->vars->vars[0];
829 decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list, ref);
830
831 if (sub_ref) {
832 if (strcmp(sub_var->name, set_var->name)) {
833 data->error(data, sub_var->token, E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
834 return 0;
835 }
836 if (!validate_set_value_ex(data, set->inner->vals[0], sub_ref, ref_list)) {
837 return 0;
838 }
839 }
840 } else if (set->inner && set->inner->count > 1) {
841 data->error(data, set->func->token, E_WARNING, "Inner `set` statement casts on pointers may only occur once");
842 return 0;
843 }
844
845 return 1;
846 }
847 static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
848 va_list argp;
849 decl_args args = {0};
850 int check;
851
852 va_start(argp, set);
853 decl_var_arg_v(&args, argp);
854 va_end(argp);
855
856 check = validate_set_value_ex(data, set, NULL, &args);
857 if (args.args) {
858 free(args.args);
859 }
860 return check;
861 }
862 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
863 if (decls) {
864 size_t i;
865
866 for (i = 0; i < decls->count; ++i) {
867 if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
868 ret->decl = decls->list[i]->func;
869 return decls->list[i];
870 }
871 }
872 }
873
874 return NULL;
875 }
876 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
877 return_stmt *ret;
878
879 /* we must have exactly one ret stmt delcaring the native func to call */
880 /* and which type cast to apply */
881 if (impl->stmts->ret.count != 1) {
882 if (impl->stmts->ret.count > 1) {
883 data->error(data, impl->stmts->ret.list[1]->token, PSI_WARNING,
884 "Too many `return` statements for implmentation %s;"
885 " found %zu, exactly one is needed",
886 impl->func->name, impl->stmts->ret.count);
887 } else {
888 data->error(data, impl->func->token, PSI_WARNING,
889 "Missing `return` statement for implementation %s",
890 impl->func->name);
891 }
892 return 0;
893 }
894
895 ret = impl->stmts->ret.list[0];
896
897 if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
898 data->error(data, ret->token, PSI_WARNING,
899 "Missing declaration '%s' for `return` statment for implementation %s",
900 ret->set->vars->vars[0]->name, impl->func->name);
901 return 0;
902 }
903
904 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)) {
905 return 0;
906 }
907
908 //impl->decl->impl = impl;
909
910 return 1;
911 }
912
913 static inline impl_arg *locate_impl_var_arg(impl_var *var, impl_args *args) {
914 size_t i;
915
916 for (i = 0; i < args->count; ++i) {
917 impl_arg *iarg = args->args[i];
918
919 if (!strcmp(var->name, iarg->var->name)) {
920 return var->arg = iarg;
921 }
922 }
923
924 return NULL;
925 }
926
927 static inline int validate_let_func(PSI_Data *data, let_func *func, impl *impl) {
928 if (impl->func->args) {
929 if (!locate_impl_var_arg(func->var, impl->func->args)) {
930 data->error(data, func->var->token, PSI_WARNING,
931 "Unknown variable '$%s' of `let` statement"
932 " for cast '%s' of implementation '%s'",
933 func->var->name, func->name, impl->func->name);
934 return 0;
935 }
936 }
937 switch (func->type) {
938 case PSI_T_BOOLVAL: func->handler = psi_let_boolval; break;
939 case PSI_T_INTVAL: func->handler = psi_let_intval; break;
940 case PSI_T_FLOATVAL: func->handler = psi_let_floatval; break;
941 case PSI_T_STRVAL: func->handler = psi_let_strval; break;
942 case PSI_T_STRLEN: func->handler = psi_let_strlen; break;
943 case PSI_T_PATHVAL: func->handler = psi_let_pathval; break;
944 case PSI_T_ARRVAL: func->handler = psi_let_arrval; break;
945 case PSI_T_OBJVAL: func->handler = psi_let_objval; break;
946 EMPTY_SWITCH_DEFAULT_CASE();
947 }
948 return 1;
949 }
950
951 static inline int validate_let_callback(PSI_Data *data, decl_var *cb_var, let_callback *cb, impl *impl) {
952 size_t i;
953 decl *cb_func;
954 decl_type *cb_type = real_decl_type(cb_var->arg->type);
955
956 if (!validate_let_func(data, cb->func, impl)) {
957 return 0;
958 }
959
960 if (cb_type->type != PSI_T_FUNCTION) {
961 data->error(data, cb_var->token, PSI_WARNING, "Not a function: %s", cb_var->name);
962 return 0;
963 }
964 cb_func = cb_type->func;
965 for (i = 0; i < cb->args->count; ++i) {
966 if (!validate_set_value(data, cb->args->vals[i], cb_func->args->count, cb_func->args->args, 0)) {
967 return 0;
968 }
969 }
970
971 if (!validate_decl_nodl(data, cb_func)) {
972 return 0;
973 }
974
975 cb->decl = cb_func;
976
977 return 1;
978 }
979
980 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
981 size_t i, j;
982 /* we can have multiple let stmts */
983
984 /* check that we have a decl arg for every let stmt */
985 for (i = 0; i < impl->stmts->let.count; ++i) {
986 let_stmt *let = impl->stmts->let.list[i];
987 decl_var *let_var;
988
989 if (let->val && let->val->kind == PSI_LET_TMP) {
990 let_var = let->val->data.var;
991 } else {
992 let_var = let->var;
993 }
994
995 if (!locate_decl_var_arg(let_var, impl->decl->args, impl->decl->func)) {
996 data->error(data, let_var->token, PSI_WARNING, "Unknown variable '%s' in `let` statement"
997 " of implementation '%s'", let_var->name, impl->func->name);
998 return 0;
999 }
1000
1001 switch (let->val->kind) {
1002 case PSI_LET_NULL:
1003 break;
1004 case PSI_LET_TMP:
1005 /* e.g. let bar = &strval($bar); // decl_arg(char **bar) */
1006 /* e.g. let foo = *bar; */
1007 let->var->pointer_level = let->val->data.var->pointer_level;
1008 let->var->arg = init_decl_arg(
1009 init_decl_type(
1010 real_decl_type(let->val->data.var->arg->type)->type,
1011 real_decl_type(let->val->data.var->arg->type)->name),
1012 init_decl_var(
1013 let->var->name,
1014 let->var->pointer_level,
1015 let->var->array_size));
1016 break;
1017 case PSI_LET_NUMEXP:
1018 if (!validate_num_exp(data, let->val->data.num, impl->decl->args, impl->decl->func, NULL)) {
1019 return 0;
1020 }
1021 break;
1022 case PSI_LET_CALLOC:
1023 if (!validate_num_exp(data, let->val->data.alloc->nmemb, impl->decl->args, impl->decl->func, NULL)) {
1024 return 0;
1025 }
1026 if (!validate_num_exp(data, let->val->data.alloc->size, impl->decl->args, impl->decl->func, NULL)) {
1027 return 0;
1028 }
1029 break;
1030 case PSI_LET_CALLBACK:
1031 if (!validate_let_callback(data, let->var, let->val->data.callback, impl)) {
1032 return 0;
1033 }
1034 break;
1035 case PSI_LET_FUNC:
1036 if (!validate_let_func(data, let->val->data.func, impl)) {
1037 return 0;
1038 }
1039 break;
1040 }
1041 }
1042
1043 /* check that we have a let stmt for every decl arg */
1044 if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
1045 decl_arg *darg = impl->decl->args->args[i];
1046 int check = 0;
1047
1048 for (j = 0; j < impl->stmts->let.count; ++j) {
1049 let_stmt *let = impl->stmts->let.list[j];
1050
1051 if (!strcmp(let->var->name, darg->var->name)) {
1052 darg->let = let;
1053 check = 1;
1054 break;
1055 }
1056 }
1057 if (!check) {
1058 data->error(data, impl->func->token, PSI_WARNING,
1059 "Missing `let` statement for arg '%s %.*s%s'"
1060 " of declaration '%s' for implementation '%s'",
1061 darg->type->name, (int) darg->var->pointer_level, "*****",
1062 darg->var->name, impl->decl->func->var->name, impl->func->name);
1063 return 0;
1064 }
1065 }
1066
1067 return 1;
1068 }
1069 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
1070 size_t i, j, k;
1071 /* we can have any count of set stmts; processing out vars */
1072 /* check that set stmts reference known variables */
1073 for (i = 0; i < impl->stmts->set.count; ++i) {
1074 set_stmt *set = impl->stmts->set.list[i];
1075 int check = 0;
1076
1077 if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
1078 impl_arg *iarg = impl->func->args->args[j];
1079
1080 if (!strcmp(set->var->name, iarg->var->name)) {
1081 set->arg = iarg;
1082 check = 1;
1083 break;
1084 }
1085 }
1086 if (!check) {
1087 data->error(data, set->var->token, PSI_WARNING, "Unknown variable '$%s' of `set` statement"
1088 " of implementation '%s'",
1089 set->var->name, impl->func->name);
1090 return 0;
1091 }
1092
1093 for (j = 0; j < set->val->vars->count; ++j) {
1094 decl_var *set_var = set->val->vars->vars[j];
1095
1096 check = 0;
1097 if (impl->decl->args) {
1098 for (k = 0; k < impl->decl->args->count; ++k) {
1099 decl_arg *set_arg = impl->decl->args->args[k];
1100
1101 if (!strcmp(set_var->name, set_arg->var->name)) {
1102 check = 1;
1103 set_var->arg = set_arg;
1104 if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
1105 return 0;
1106 }
1107 break;
1108 }
1109 }
1110 }
1111 if (!check) {
1112 for (k = 0; k < impl->stmts->let.count; ++k) {
1113 let_stmt *let = impl->stmts->let.list[k];
1114
1115 /* check temp vars */
1116 if (let->val && let->val->kind == PSI_LET_TMP) {
1117 if (!strcmp(set_var->name, let->var->name)) {
1118 check = 1;
1119 set_var->arg = let->var->arg;
1120 if (!validate_set_value(data, set->val, 1, &set_var->arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
1121 return 0;
1122 }
1123 break;
1124 }
1125 }
1126 }
1127 }
1128
1129 if (!check) {
1130 data->error(data, set_var->token, PSI_WARNING, "Unknown value '%s' of `set` statement"
1131 " for variable '$%s' of implementation '%s'",
1132 set_var->name, set->arg->var->name, impl->func->name);
1133 return 0;
1134 }
1135 }
1136 }
1137 return 1;
1138 }
1139 static inline decl *locate_free_decl(decls *decls, free_call *f) {
1140 if (decls) {
1141 size_t i;
1142
1143 for (i = 0; i < decls->count; ++i) {
1144 if (!strcmp(decls->list[i]->func->var->name, f->func)) {
1145 f->decl = decls->list[i];
1146 return decls->list[i];
1147 }
1148 }
1149 }
1150
1151 return NULL;
1152 }
1153 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
1154 size_t i, j, k, l;
1155 /* we can have any count of free stmts; freeing any out vars */
1156 for (i = 0; i < impl->stmts->fre.count; ++i) {
1157 free_stmt *fre = impl->stmts->fre.list[i];
1158
1159 for (j = 0; j < fre->calls->count; ++j) {
1160 free_call *free_call = fre->calls->list[j];
1161
1162 /* first find the decl of the free func */
1163 if (!locate_free_decl(data->decls, free_call)) {
1164 data->error(data, free_call->token, PSI_WARNING,
1165 "Missing declaration '%s' in `free` statement"
1166 " of implementation '%s'",
1167 free_call->func, impl->func->name);
1168 return 0;
1169 }
1170
1171
1172
1173 /* now check for known vars */
1174 for (l = 0; l < free_call->vars->count; ++l) {
1175 int check = 0;
1176 decl_var *free_var = free_call->vars->vars[l];
1177
1178 if (!strcmp(free_var->name, impl->decl->func->var->name)) {
1179 check = 1;
1180 free_var->arg = impl->decl->func;
1181 } else if (impl->decl->args) {
1182 for (k = 0; k < impl->decl->args->count; ++k) {
1183 decl_arg *free_arg = impl->decl->args->args[k];
1184
1185 if (!strcmp(free_var->name, free_arg->var->name)) {
1186 check = 1;
1187 free_var->arg = free_arg;
1188 break;
1189 }
1190 }
1191 }
1192
1193 if (!check) {
1194 data->error(data, free_var->token, PSI_WARNING,
1195 "Unknown variable '%s' of `free` statement"
1196 " of implementation '%s'",
1197 free_var->name, impl->func->name);
1198 return 0;
1199 }
1200 }
1201 }
1202 }
1203 return 1;
1204 }
1205 static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
1206 if (!impl->stmts) {
1207 data->error(data, impl->func->token, PSI_WARNING,
1208 "Missing body for implementation %s!",
1209 impl->func->name);
1210 return 0;
1211 }
1212
1213 if (!validate_impl_ret_stmt(data, impl)) {
1214 return 0;
1215 }
1216
1217 if (!validate_impl_let_stmts(data, impl)) {
1218 return 0;
1219 }
1220 if (!validate_impl_set_stmts(data, impl)) {
1221 return 0;
1222 }
1223 if (!validate_impl_free_stmts(data, impl)) {
1224 return 0;
1225 }
1226
1227 return 1;
1228 }
1229
1230 static inline int validate_impl_args(PSI_Data *data, impl *impl) {
1231 int def = 0;
1232 size_t i;
1233
1234 for (i = 0; i < impl->func->args->count; ++i) {
1235 impl_arg *iarg = impl->func->args->args[i];
1236
1237 if (iarg->def) {
1238 def = 1;
1239 } else if (def) {
1240 data->error(data, impl->func->token, PSI_WARNING,
1241 "Non-optional argument %zu '$%s' of implementation '%s'"
1242 " follows optional argument",
1243 i+1, iarg->var->name, impl->func->name);
1244 return 0;
1245 }
1246 }
1247
1248 return 1;
1249 }
1250
1251 static inline int validate_impl(PSI_Data *data, impl *impl) {
1252 if (!validate_impl_args(data, impl)) {
1253 return 0;
1254 }
1255 return validate_impl_stmts(data, impl);
1256 }
1257
1258
1259 int PSI_ContextValidate(PSI_Context *C, PSI_Parser *P)
1260 {
1261 PSI_Data *D;
1262 void *dlopened = NULL;
1263 size_t i, count = C->count++, check_round, check_count;
1264 decl_typedefs *check_defs = P->defs;
1265 decl_structs *check_structs = P->structs;
1266 decl_unions *check_unions = P->unions;
1267 decl_enums *check_enums = P->enums;
1268 unsigned silent = C->flags & PSI_PARSER_SILENT;
1269
1270 C->data = realloc(C->data, C->count * sizeof(*C->data));
1271 D = PSI_DataExchange(&C->data[count], PSI_DATA(P));
1272
1273 #define REVALIDATE(what) do { \
1274 if (check_round && check_ ##what) { \
1275 free(check_ ##what->list); \
1276 free(check_ ##what); \
1277 } \
1278 check_ ##what = recheck_ ##what; \
1279 } while (0)
1280 #define CHECK_TOTAL (CHECK_COUNT(defs) + CHECK_COUNT(structs) + CHECK_COUNT(enums))
1281 #define CHECK_COUNT(of) (check_ ##of ? check_ ##of->count : 0)
1282
1283 if (!silent) {
1284 /* no warnings on first round */
1285 C->flags |= PSI_PARSER_SILENT;
1286 }
1287 for (check_round = 0, check_count = 0; CHECK_TOTAL && check_count != CHECK_TOTAL; ++check_round) {
1288 decl_typedefs *recheck_defs = NULL;
1289 decl_structs *recheck_structs = NULL;
1290 decl_unions *recheck_unions = NULL;
1291 decl_enums *recheck_enums = NULL;
1292
1293 check_count = CHECK_TOTAL;
1294
1295 for (i = 0; i < CHECK_COUNT(defs); ++i) {
1296 if (validate_decl_typedef(PSI_DATA(C), check_defs->list[i])) {
1297 C->defs = add_decl_typedef(C->defs, check_defs->list[i]);
1298 } else {
1299 recheck_defs = add_decl_typedef(recheck_defs, check_defs->list[i]);
1300 }
1301 }
1302 for (i = 0; i < CHECK_COUNT(structs); ++i) {
1303 if (validate_decl_struct(PSI_DATA(C), check_structs->list[i])) {
1304 C->structs = add_decl_struct(C->structs, check_structs->list[i]);
1305 } else {
1306 recheck_structs = add_decl_struct(recheck_structs, check_structs->list[i]);
1307 }
1308 }
1309 for (i = 0; i < CHECK_COUNT(unions); ++i) {
1310 if (validate_decl_union(PSI_DATA(C), check_unions->list[i])) {
1311 C->unions = add_decl_union(C->unions, check_unions->list[i]);
1312 } else {
1313 recheck_unions = add_decl_union(recheck_unions, check_unions->list[i]);
1314 }
1315 }
1316 for (i = 0; i < CHECK_COUNT(enums); ++i) {
1317 if (validate_decl_enum(PSI_DATA(C), check_enums->list[i])) {
1318 C->enums = add_decl_enum(C->enums, check_enums->list[i]);
1319 } else {
1320 recheck_enums = add_decl_enum(recheck_enums, check_enums->list[i]);
1321 }
1322 }
1323
1324 REVALIDATE(defs);
1325 REVALIDATE(structs);
1326 REVALIDATE(unions);
1327 REVALIDATE(enums);
1328
1329 if (check_round == 0 && !silent) {
1330 C->flags &= ~PSI_PARSER_SILENT;
1331 }
1332 }
1333
1334
1335 if (D->consts) {
1336 for (i = 0; i < D->consts->count; ++i) {
1337 if (validate_constant(PSI_DATA(C), D->consts->list[i])) {
1338 C->consts = add_constant(C->consts, D->consts->list[i]);
1339 }
1340 }
1341 }
1342
1343 if (!validate_lib(D, &dlopened)) {
1344 return 0;
1345 }
1346
1347 add_decl_lib(&C->psi.libs, dlopened);
1348
1349 if (D->decls) {
1350 for (i = 0; i < D->decls->count; ++i) {
1351 if (validate_decl(PSI_DATA(C), dlopened, D->decls->list[i])) {
1352 C->decls = add_decl(C->decls, D->decls->list[i]);
1353 }
1354 }
1355 }
1356 if (D->impls) {
1357 for (i = 0; i < D->impls->count; ++i) {
1358 if (validate_impl(PSI_DATA(C), D->impls->list[i])) {
1359 C->impls = add_impl(C->impls, D->impls->list[i]);
1360 }
1361 }
1362 }
1363
1364 return 1;
1365 }
1366
1367 int PSI_ContextValidateData(PSI_Data *dest, PSI_Data *source)
1368 {
1369 size_t i;
1370 int errors = 0;
1371
1372 if (source->defs) for (i = 0; i < source->defs->count; ++i) {
1373 decl_arg *def = source->defs->list[i];
1374
1375 if (validate_decl_typedef(source, def)) {
1376 if (dest) {
1377 dest->defs = add_decl_typedef(dest->defs, def);
1378 }
1379 } else {
1380 ++errors;
1381 }
1382 }
1383
1384 if (source->consts) for (i = 0; i < source->consts->count; ++i) {
1385 constant *constant = source->consts->list[i];
1386
1387 if (validate_constant(source, constant)) {
1388 if (dest) {
1389 dest->consts = add_constant(dest->consts, constant);
1390 }
1391 } else {
1392 ++errors;
1393 }
1394 }
1395
1396 if (source->structs) for (i = 0; i < source->structs->count; ++i) {
1397 decl_struct *dstruct = source->structs->list[i];
1398
1399 if (validate_decl_struct(source, dstruct)) {
1400 if (dest) {
1401 dest->structs = add_decl_struct(dest->structs, dstruct);
1402 }
1403 } else {
1404 ++errors;
1405 }
1406 }
1407
1408 if (source->unions) for (i = 0; i < source->unions->count; ++i) {
1409 decl_union *dunion = source->unions->list[i];
1410
1411 if (validate_decl_union(source, dunion)) {
1412 if (dest) {
1413 dest->unions = add_decl_union(dest->unions, dunion);
1414 }
1415 } else {
1416 ++errors;
1417 }
1418 }
1419
1420 if (source->enums) for (i = 0; i < source->enums->count; ++i) {
1421 decl_enum *denum = source->enums->list[i];
1422
1423 if (validate_decl_enum(source, denum)) {
1424 if (dest) {
1425 dest->enums = add_decl_enum(dest->enums, denum);
1426 }
1427 } else {
1428 ++errors;
1429 }
1430 }
1431
1432 if (source->decls) for (i = 0; i < source->decls->count; ++i) {
1433 decl *decl = source->decls->list[i];
1434
1435 if (validate_decl(source, NULL, decl)) {
1436 if (dest) {
1437 dest->decls = add_decl(dest->decls, decl);
1438 }
1439 } else {
1440 ++errors;
1441 }
1442 }
1443
1444 if (source->impls) for (i = 0; i < source->impls->count; ++i) {
1445 impl *impl = source->impls->list[i];
1446
1447 if (validate_impl(source, impl)) {
1448 if (dest) {
1449 dest->impls = add_impl(dest->impls, impl);
1450 }
1451 } else {
1452 ++errors;
1453 }
1454 }
1455
1456 return errors;
1457 }