e34330dee6ad521647169778ae4670df8cccf406
[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: set->func->handler = psi_to_bool; break;
728 case PSI_T_TO_INT: set->func->handler = psi_to_int; break;
729 case PSI_T_TO_FLOAT: set->func->handler = psi_to_double; break;
730 case PSI_T_TO_STRING: set->func->handler = psi_to_string; break;
731 case PSI_T_TO_ARRAY: set->func->handler = psi_to_array; break;
732 case PSI_T_TO_OBJECT: set->func->handler = psi_to_object; break;
733 case PSI_T_VOID: set->func->handler = psi_to_void; break;
734 case PSI_T_ZVAL: set->func->handler = psi_to_zval; break;
735 case PSI_T_ELLIPSIS:
736 if (set->outer.set && set->outer.set->func->type == PSI_T_TO_ARRAY) {
737 set->func->handler = psi_to_recursive;
738 set->inner = set->outer.set->inner;
739 break;
740 }
741 /* no break */
742 default:
743 return 0;
744 }
745 return 1;
746 }
747 static inline void decl_var_arg_v(decl_args *args, va_list argp) {
748 int argc;
749 decl_arg **argv;
750
751 memset(args, 0, sizeof(*args));
752
753 while ((argc = va_arg(argp, int))) {
754 argv = va_arg(argp, decl_arg **);
755 while (argc--) {
756 add_decl_arg(args, *argv++);
757 }
758 }
759 }
760 static inline int validate_set_value_ex(PSI_Data *data, set_value *set, decl_arg *ref, decl_args *ref_list) {
761 size_t i;
762 decl_type *ref_type;
763 decl_var *set_var = set->vars->vars[0];
764
765 if (!validate_set_value_handler(set)) {
766 data->error(data, set->func->token, PSI_WARNING, "Invalid cast '%s' in `set` statement", set->func->name);
767 return 0;
768 }
769
770 for (i = 0; i < set->vars->count; ++i) {
771 decl_var *svar = set->vars->vars[i];
772 if (!svar->arg && !locate_decl_var_arg(svar, ref_list, NULL)) {
773 data->error(data, svar->token, PSI_WARNING, "Unknown variable '%s' in `set` statement", svar->name);
774 return 0;
775 }
776 }
777
778 if (!ref) {
779 ref = set_var->arg;
780 }
781 ref_type = real_decl_type(ref->type);
782
783 if (set->inner && set->inner->count) {
784 int is_to_array = (set->func->type == PSI_T_TO_ARRAY);
785 int is_pointer_to_struct = (ref_type->type == PSI_T_STRUCT && ref->var->pointer_level);
786
787 if (!is_to_array && !is_pointer_to_struct) {
788 data->error(data, set->func->token, E_WARNING, "Inner `set` statement casts only work with "
789 "to_array() casts on structs or pointers: %s(%s...", set->func->name, set->vars->vars[0]->name);
790 return 0;
791 }
792 }
793 if (set->num) {
794 if (!validate_num_exp(data, set->num, ref_list, ref, NULL)) {
795 return 0;
796 }
797 }
798
799 if (set->inner && ref_type->type == PSI_T_STRUCT) {
800 /* to_array(struct, to_...) */
801 if (!set->outer.set || set->outer.set->inner->vals != set->inner->vals) {
802 for (i = 0; i < set->inner->count; ++i) {
803 decl_var *sub_var = set->inner->vals[i]->vars->vars[0];
804 decl_arg *sub_ref = locate_struct_member(ref_type->strct, sub_var);
805
806 if (sub_ref) {
807 if (!validate_set_value_ex(data, set->inner->vals[i], sub_ref, ref_type->strct->args)) {
808 return 0;
809 }
810 }
811 }
812 }
813 } else if (set->inner && set->inner->count == 1) {
814 /* to_array(ptr, to_string(*ptr)) */
815 decl_var *sub_var = set->inner->vals[0]->vars->vars[0];
816 decl_arg *sub_ref = locate_decl_var_arg(sub_var, ref_list, ref);
817
818 if (sub_ref) {
819 if (strcmp(sub_var->name, set_var->name)) {
820 data->error(data, sub_var->token, E_WARNING, "Inner `set` statement casts on pointers must reference the same variable");
821 return 0;
822 }
823 if (!validate_set_value_ex(data, set->inner->vals[0], sub_ref, ref_list)) {
824 return 0;
825 }
826 }
827 } else if (set->inner && set->inner->count > 1) {
828 data->error(data, set->func->token, E_WARNING, "Inner `set` statement casts on pointers may only occur once");
829 return 0;
830 }
831
832 return 1;
833 }
834 static inline int validate_set_value(PSI_Data *data, set_value *set, ...) {
835 va_list argp;
836 decl_args args = {0};
837 int check;
838
839 va_start(argp, set);
840 decl_var_arg_v(&args, argp);
841 va_end(argp);
842
843 check = validate_set_value_ex(data, set, NULL, &args);
844 if (args.args) {
845 free(args.args);
846 }
847 return check;
848 }
849 static inline decl *locate_impl_decl(decls *decls, return_stmt *ret) {
850 if (decls) {
851 size_t i;
852
853 for (i = 0; i < decls->count; ++i) {
854 if (!strcmp(decls->list[i]->func->var->name, ret->set->vars->vars[0]->name)) {
855 ret->decl = decls->list[i]->func;
856 return decls->list[i];
857 }
858 }
859 }
860
861 return NULL;
862 }
863 static inline int validate_impl_ret_stmt(PSI_Data *data, impl *impl) {
864 return_stmt *ret;
865
866 /* we must have exactly one ret stmt delcaring the native func to call */
867 /* and which type cast to apply */
868 if (impl->stmts->ret.count != 1) {
869 if (impl->stmts->ret.count > 1) {
870 data->error(data, impl->stmts->ret.list[1]->token, PSI_WARNING,
871 "Too many `return` statements for implmentation %s;"
872 " found %zu, exactly one is needed",
873 impl->func->name, impl->stmts->ret.count);
874 } else {
875 data->error(data, impl->func->token, PSI_WARNING,
876 "Missing `return` statement for implementation %s",
877 impl->func->name);
878 }
879 return 0;
880 }
881
882 ret = impl->stmts->ret.list[0];
883
884 if (!(impl->decl = locate_impl_decl(data->decls, ret))) {
885 data->error(data, ret->token, PSI_WARNING,
886 "Missing declaration '%s' for `return` statment for implementation %s",
887 ret->set->vars->vars[0]->name, impl->func->name);
888 return 0;
889 }
890
891 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)) {
892 return 0;
893 }
894
895 //impl->decl->impl = impl;
896
897 return 1;
898 }
899
900 static inline impl_arg *locate_impl_var_arg(impl_var *var, impl_args *args) {
901 size_t i;
902
903 for (i = 0; i < args->count; ++i) {
904 impl_arg *iarg = args->args[i];
905
906 if (!strcmp(var->name, iarg->var->name)) {
907 return var->arg = iarg;
908 }
909 }
910
911 return NULL;
912 }
913
914 static inline int validate_let_func(PSI_Data *data, let_func *func, impl *impl) {
915 if (impl->func->args) {
916 if (!locate_impl_var_arg(func->var, impl->func->args)) {
917 data->error(data, func->var->token, PSI_WARNING,
918 "Unknown variable '$%s' of `let` statement"
919 " for cast '%s' of implementation '%s'",
920 func->var->name, func->name, impl->func->name);
921 return 0;
922 }
923 }
924 switch (func->type) {
925 case PSI_T_BOOLVAL: func->handler = psi_let_boolval; break;
926 case PSI_T_INTVAL: func->handler = psi_let_intval; break;
927 case PSI_T_FLOATVAL: func->handler = psi_let_floatval; break;
928 case PSI_T_STRVAL: func->handler = psi_let_strval; break;
929 case PSI_T_STRLEN: func->handler = psi_let_strlen; break;
930 case PSI_T_PATHVAL: func->handler = psi_let_pathval; break;
931 case PSI_T_ARRVAL: func->handler = psi_let_arrval; break;
932 case PSI_T_OBJVAL: func->handler = psi_let_objval; break;
933 case PSI_T_ZVAL: func->handler = psi_let_zval; break;
934 EMPTY_SWITCH_DEFAULT_CASE();
935 }
936 return 1;
937 }
938
939 static inline int validate_let_callback(PSI_Data *data, decl_var *cb_var, let_callback *cb, impl *impl) {
940 size_t i;
941 decl *cb_func;
942 decl_type *cb_type = real_decl_type(cb_var->arg->type);
943
944 if (!validate_let_func(data, cb->func, impl)) {
945 return 0;
946 }
947
948 if (cb_type->type != PSI_T_FUNCTION) {
949 data->error(data, cb_var->token, PSI_WARNING, "Not a function: %s", cb_var->name);
950 return 0;
951 }
952 cb_func = cb_type->func;
953 for (i = 0; i < cb->args->count; ++i) {
954 if (!validate_set_value(data, cb->args->vals[i], cb_func->args->count, cb_func->args->args, 0)) {
955 return 0;
956 }
957 }
958
959 if (!validate_decl_nodl(data, cb_func)) {
960 return 0;
961 }
962
963 cb->decl = cb_func;
964
965 return 1;
966 }
967
968 static inline int validate_impl_let_stmts(PSI_Data *data, impl *impl) {
969 size_t i, j;
970 /* we can have multiple let stmts */
971
972 /* check that we have a decl arg for every let stmt */
973 for (i = 0; i < impl->stmts->let.count; ++i) {
974 let_stmt *let = impl->stmts->let.list[i];
975 decl_var *let_var;
976
977 if (let->val && let->val->kind == PSI_LET_TMP) {
978 let_var = let->val->data.var;
979 } else {
980 let_var = let->var;
981 }
982
983 if (!locate_decl_var_arg(let_var, impl->decl->args, impl->decl->func)) {
984 data->error(data, let_var->token, PSI_WARNING, "Unknown variable '%s' in `let` statement"
985 " of implementation '%s'", let_var->name, impl->func->name);
986 return 0;
987 }
988
989 switch (let->val->kind) {
990 case PSI_LET_NULL:
991 break;
992 case PSI_LET_TMP:
993 /* e.g. let bar = &strval($bar); // decl_arg(char **bar) */
994 /* e.g. let foo = *bar; */
995 let->var->pointer_level = let->val->data.var->pointer_level;
996 let->var->arg = init_decl_arg(
997 init_decl_type(
998 real_decl_type(let->val->data.var->arg->type)->type,
999 real_decl_type(let->val->data.var->arg->type)->name),
1000 init_decl_var(
1001 let->var->name,
1002 let->var->pointer_level,
1003 let->var->array_size));
1004 break;
1005 case PSI_LET_NUMEXP:
1006 if (!validate_num_exp(data, let->val->data.num, impl->decl->args, impl->decl->func, NULL)) {
1007 return 0;
1008 }
1009 break;
1010 case PSI_LET_CALLOC:
1011 if (!validate_num_exp(data, let->val->data.alloc->nmemb, impl->decl->args, impl->decl->func, NULL)) {
1012 return 0;
1013 }
1014 if (!validate_num_exp(data, let->val->data.alloc->size, impl->decl->args, impl->decl->func, NULL)) {
1015 return 0;
1016 }
1017 break;
1018 case PSI_LET_CALLBACK:
1019 if (!validate_let_callback(data, let->var, let->val->data.callback, impl)) {
1020 return 0;
1021 }
1022 break;
1023 case PSI_LET_FUNC:
1024 if (!validate_let_func(data, let->val->data.func, impl)) {
1025 return 0;
1026 }
1027 break;
1028 }
1029 }
1030
1031 /* check that we have a let stmt for every decl arg */
1032 if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
1033 decl_arg *darg = impl->decl->args->args[i];
1034 int check = 0;
1035
1036 for (j = 0; j < impl->stmts->let.count; ++j) {
1037 let_stmt *let = impl->stmts->let.list[j];
1038
1039 if (!strcmp(let->var->name, darg->var->name)) {
1040 check = 1;
1041 break;
1042 }
1043 }
1044 if (!check) {
1045 data->error(data, impl->func->token, PSI_WARNING,
1046 "Missing `let` statement for arg '%s %.*s%s'"
1047 " of declaration '%s' for implementation '%s'",
1048 darg->type->name, (int) darg->var->pointer_level, "*****",
1049 darg->var->name, impl->decl->func->var->name, impl->func->name);
1050 return 0;
1051 }
1052 }
1053
1054 return 1;
1055 }
1056 static inline int validate_impl_set_stmts(PSI_Data *data, impl *impl) {
1057 size_t i, j, k;
1058 /* we can have any count of set stmts; processing out vars */
1059 /* check that set stmts reference known variables */
1060 for (i = 0; i < impl->stmts->set.count; ++i) {
1061 set_stmt *set = impl->stmts->set.list[i];
1062 int check = 0;
1063
1064 if (impl->func->args) for (j = 0; j < impl->func->args->count; ++j) {
1065 impl_arg *iarg = impl->func->args->args[j];
1066
1067 if (!strcmp(set->var->name, iarg->var->name)) {
1068 set->arg = iarg;
1069 check = 1;
1070 break;
1071 }
1072 }
1073 if (!check) {
1074 data->error(data, set->var->token, PSI_WARNING, "Unknown variable '$%s' of `set` statement"
1075 " of implementation '%s'",
1076 set->var->name, impl->func->name);
1077 return 0;
1078 }
1079
1080 for (j = 0; j < set->val->vars->count; ++j) {
1081 decl_var *set_var = set->val->vars->vars[j];
1082
1083 check = 0;
1084 if (impl->decl->args) {
1085 for (k = 0; k < impl->decl->args->count; ++k) {
1086 decl_arg *set_arg = impl->decl->args->args[k];
1087
1088 if (!strcmp(set_var->name, set_arg->var->name)) {
1089 check = 1;
1090 set_var->arg = set_arg;
1091 if (!validate_set_value(data, set->val, 1, &set_arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
1092 return 0;
1093 }
1094 break;
1095 }
1096 }
1097 }
1098 if (!check) {
1099 for (k = 0; k < impl->stmts->let.count; ++k) {
1100 let_stmt *let = impl->stmts->let.list[k];
1101
1102 /* check temp vars */
1103 if (let->val && let->val->kind == PSI_LET_TMP) {
1104 if (!strcmp(set_var->name, let->var->name)) {
1105 check = 1;
1106 set_var->arg = let->var->arg;
1107 if (!validate_set_value(data, set->val, 1, &set_var->arg, 1, &impl->decl->func, impl->decl->args->count, impl->decl->args->args, 0)) {
1108 return 0;
1109 }
1110 break;
1111 }
1112 }
1113 }
1114 }
1115
1116 if (!check) {
1117 data->error(data, set_var->token, PSI_WARNING, "Unknown value '%s' of `set` statement"
1118 " for variable '$%s' of implementation '%s'",
1119 set_var->name, set->arg->var->name, impl->func->name);
1120 return 0;
1121 }
1122 }
1123 }
1124 return 1;
1125 }
1126 static inline decl *locate_free_decl(decls *decls, free_call *f) {
1127 if (decls) {
1128 size_t i;
1129
1130 for (i = 0; i < decls->count; ++i) {
1131 if (!strcmp(decls->list[i]->func->var->name, f->func)) {
1132 f->decl = decls->list[i];
1133 return decls->list[i];
1134 }
1135 }
1136 }
1137
1138 return NULL;
1139 }
1140 static inline int validate_impl_free_stmts(PSI_Data *data, impl *impl) {
1141 size_t i, j, k, l;
1142 /* we can have any count of free stmts; freeing any out vars */
1143 for (i = 0; i < impl->stmts->fre.count; ++i) {
1144 free_stmt *fre = impl->stmts->fre.list[i];
1145
1146 for (j = 0; j < fre->calls->count; ++j) {
1147 free_call *free_call = fre->calls->list[j];
1148
1149 /* first find the decl of the free func */
1150 if (!locate_free_decl(data->decls, free_call)) {
1151 data->error(data, free_call->token, PSI_WARNING,
1152 "Missing declaration '%s' in `free` statement"
1153 " of implementation '%s'",
1154 free_call->func, impl->func->name);
1155 return 0;
1156 }
1157
1158
1159
1160 /* now check for known vars */
1161 for (l = 0; l < free_call->vars->count; ++l) {
1162 int check = 0;
1163 decl_var *free_var = free_call->vars->vars[l];
1164
1165 if (!strcmp(free_var->name, impl->decl->func->var->name)) {
1166 check = 1;
1167 free_var->arg = impl->decl->func;
1168 } else if (impl->decl->args) {
1169 for (k = 0; k < impl->decl->args->count; ++k) {
1170 decl_arg *free_arg = impl->decl->args->args[k];
1171
1172 if (!strcmp(free_var->name, free_arg->var->name)) {
1173 check = 1;
1174 free_var->arg = free_arg;
1175 break;
1176 }
1177 }
1178 }
1179
1180 if (!check) {
1181 data->error(data, free_var->token, PSI_WARNING,
1182 "Unknown variable '%s' of `free` statement"
1183 " of implementation '%s'",
1184 free_var->name, impl->func->name);
1185 return 0;
1186 }
1187 }
1188 }
1189 }
1190 return 1;
1191 }
1192 static inline int validate_impl_stmts(PSI_Data *data, impl *impl) {
1193 if (!impl->stmts) {
1194 data->error(data, impl->func->token, PSI_WARNING,
1195 "Missing body for implementation %s!",
1196 impl->func->name);
1197 return 0;
1198 }
1199
1200 if (!validate_impl_ret_stmt(data, impl)) {
1201 return 0;
1202 }
1203
1204 if (!validate_impl_let_stmts(data, impl)) {
1205 return 0;
1206 }
1207 if (!validate_impl_set_stmts(data, impl)) {
1208 return 0;
1209 }
1210 if (!validate_impl_free_stmts(data, impl)) {
1211 return 0;
1212 }
1213
1214 return 1;
1215 }
1216
1217 static inline int validate_impl_args(PSI_Data *data, impl *impl) {
1218 int def = 0;
1219 size_t i;
1220
1221 for (i = 0; i < impl->func->args->count; ++i) {
1222 impl_arg *iarg = impl->func->args->args[i];
1223
1224 if (iarg->def) {
1225 def = 1;
1226 } else if (def) {
1227 data->error(data, impl->func->token, PSI_WARNING,
1228 "Non-optional argument %zu '$%s' of implementation '%s'"
1229 " follows optional argument",
1230 i+1, iarg->var->name, impl->func->name);
1231 return 0;
1232 }
1233 }
1234
1235 return 1;
1236 }
1237
1238 static inline int validate_impl(PSI_Data *data, impl *impl) {
1239 if (!validate_impl_args(data, impl)) {
1240 return 0;
1241 }
1242 return validate_impl_stmts(data, impl);
1243 }
1244
1245
1246 int PSI_ContextValidate(PSI_Context *C, PSI_Parser *P)
1247 {
1248 PSI_Data *D;
1249 void *dlopened = NULL;
1250 size_t i, count = C->count++, check_round, check_count;
1251 decl_typedefs *check_defs = P->defs;
1252 decl_structs *check_structs = P->structs;
1253 decl_unions *check_unions = P->unions;
1254 decl_enums *check_enums = P->enums;
1255 unsigned silent = C->flags & PSI_PARSER_SILENT;
1256
1257 C->data = realloc(C->data, C->count * sizeof(*C->data));
1258 D = PSI_DataExchange(&C->data[count], PSI_DATA(P));
1259
1260 #define REVALIDATE(what) do { \
1261 if (check_round && check_ ##what) { \
1262 free(check_ ##what->list); \
1263 free(check_ ##what); \
1264 } \
1265 check_ ##what = recheck_ ##what; \
1266 } while (0)
1267 #define CHECK_TOTAL (CHECK_COUNT(defs) + CHECK_COUNT(structs) + CHECK_COUNT(enums))
1268 #define CHECK_COUNT(of) (check_ ##of ? check_ ##of->count : 0)
1269
1270 if (!silent) {
1271 /* no warnings on first round */
1272 C->flags |= PSI_PARSER_SILENT;
1273 }
1274 for (check_round = 0, check_count = 0; CHECK_TOTAL && check_count != CHECK_TOTAL; ++check_round) {
1275 decl_typedefs *recheck_defs = NULL;
1276 decl_structs *recheck_structs = NULL;
1277 decl_unions *recheck_unions = NULL;
1278 decl_enums *recheck_enums = NULL;
1279
1280 check_count = CHECK_TOTAL;
1281
1282 for (i = 0; i < CHECK_COUNT(defs); ++i) {
1283 if (validate_decl_typedef(PSI_DATA(C), check_defs->list[i])) {
1284 C->defs = add_decl_typedef(C->defs, check_defs->list[i]);
1285 } else {
1286 recheck_defs = add_decl_typedef(recheck_defs, check_defs->list[i]);
1287 }
1288 }
1289 for (i = 0; i < CHECK_COUNT(structs); ++i) {
1290 if (validate_decl_struct(PSI_DATA(C), check_structs->list[i])) {
1291 C->structs = add_decl_struct(C->structs, check_structs->list[i]);
1292 } else {
1293 recheck_structs = add_decl_struct(recheck_structs, check_structs->list[i]);
1294 }
1295 }
1296 for (i = 0; i < CHECK_COUNT(unions); ++i) {
1297 if (validate_decl_union(PSI_DATA(C), check_unions->list[i])) {
1298 C->unions = add_decl_union(C->unions, check_unions->list[i]);
1299 } else {
1300 recheck_unions = add_decl_union(recheck_unions, check_unions->list[i]);
1301 }
1302 }
1303 for (i = 0; i < CHECK_COUNT(enums); ++i) {
1304 if (validate_decl_enum(PSI_DATA(C), check_enums->list[i])) {
1305 C->enums = add_decl_enum(C->enums, check_enums->list[i]);
1306 } else {
1307 recheck_enums = add_decl_enum(recheck_enums, check_enums->list[i]);
1308 }
1309 }
1310
1311 REVALIDATE(defs);
1312 REVALIDATE(structs);
1313 REVALIDATE(unions);
1314 REVALIDATE(enums);
1315
1316 if (check_round == 0 && !silent) {
1317 C->flags &= ~PSI_PARSER_SILENT;
1318 }
1319 }
1320
1321
1322 if (D->consts) {
1323 for (i = 0; i < D->consts->count; ++i) {
1324 if (validate_constant(PSI_DATA(C), D->consts->list[i])) {
1325 C->consts = add_constant(C->consts, D->consts->list[i]);
1326 }
1327 }
1328 }
1329
1330 if (!validate_lib(D, &dlopened)) {
1331 return 0;
1332 }
1333
1334 add_decl_lib(&C->psi.libs, dlopened);
1335
1336 if (D->decls) {
1337 for (i = 0; i < D->decls->count; ++i) {
1338 if (validate_decl(PSI_DATA(C), dlopened, D->decls->list[i])) {
1339 C->decls = add_decl(C->decls, D->decls->list[i]);
1340 }
1341 }
1342 }
1343 if (D->impls) {
1344 for (i = 0; i < D->impls->count; ++i) {
1345 if (validate_impl(PSI_DATA(C), D->impls->list[i])) {
1346 C->impls = add_impl(C->impls, D->impls->list[i]);
1347 }
1348 }
1349 }
1350
1351 return 1;
1352 }
1353
1354 int PSI_ContextValidateData(PSI_Data *dest, PSI_Data *source)
1355 {
1356 size_t i;
1357 int errors = 0;
1358
1359 if (source->defs) for (i = 0; i < source->defs->count; ++i) {
1360 decl_arg *def = source->defs->list[i];
1361
1362 if (validate_decl_typedef(source, def)) {
1363 if (dest) {
1364 dest->defs = add_decl_typedef(dest->defs, def);
1365 }
1366 } else {
1367 ++errors;
1368 }
1369 }
1370
1371 if (source->consts) for (i = 0; i < source->consts->count; ++i) {
1372 constant *constant = source->consts->list[i];
1373
1374 if (validate_constant(source, constant)) {
1375 if (dest) {
1376 dest->consts = add_constant(dest->consts, constant);
1377 }
1378 } else {
1379 ++errors;
1380 }
1381 }
1382
1383 if (source->structs) for (i = 0; i < source->structs->count; ++i) {
1384 decl_struct *dstruct = source->structs->list[i];
1385
1386 if (validate_decl_struct(source, dstruct)) {
1387 if (dest) {
1388 dest->structs = add_decl_struct(dest->structs, dstruct);
1389 }
1390 } else {
1391 ++errors;
1392 }
1393 }
1394
1395 if (source->unions) for (i = 0; i < source->unions->count; ++i) {
1396 decl_union *dunion = source->unions->list[i];
1397
1398 if (validate_decl_union(source, dunion)) {
1399 if (dest) {
1400 dest->unions = add_decl_union(dest->unions, dunion);
1401 }
1402 } else {
1403 ++errors;
1404 }
1405 }
1406
1407 if (source->enums) for (i = 0; i < source->enums->count; ++i) {
1408 decl_enum *denum = source->enums->list[i];
1409
1410 if (validate_decl_enum(source, denum)) {
1411 if (dest) {
1412 dest->enums = add_decl_enum(dest->enums, denum);
1413 }
1414 } else {
1415 ++errors;
1416 }
1417 }
1418
1419 if (source->decls) for (i = 0; i < source->decls->count; ++i) {
1420 decl *decl = source->decls->list[i];
1421
1422 if (validate_decl(source, NULL, decl)) {
1423 if (dest) {
1424 dest->decls = add_decl(dest->decls, decl);
1425 }
1426 } else {
1427 ++errors;
1428 }
1429 }
1430
1431 if (source->impls) for (i = 0; i < source->impls->count; ++i) {
1432 impl *impl = source->impls->list[i];
1433
1434 if (validate_impl(source, impl)) {
1435 if (dest) {
1436 dest->impls = add_impl(dest->impls, impl);
1437 }
1438 } else {
1439 ++errors;
1440 }
1441 }
1442
1443 return errors;
1444 }