f204753bfbf9190c13a0eb96f5bea4769a46f5b1
[m6w6/ext-psi] / src / marshal.c
1 #ifdef HAVE_CONFIG_H
2 # include "config.h"
3 #endif
4
5 #include "php.h"
6 #include "php_psi.h"
7 #include "parser.h"
8 #include "marshal.h"
9 #include "calc.h"
10
11 void psi_to_void(zval *return_value, set_value *set, impl_val *ret_val)
12 {
13 RETVAL_NULL();
14 }
15
16 void psi_to_zval(zval *return_value, set_value *set, impl_val *ret_val) {
17 RETVAL_ZVAL(ret_val->ptr, 1, 0);
18 }
19
20 impl_val *psi_let_zval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
21 {
22 *to_free = tmp->ptr = emalloc(sizeof(zval));
23 ZVAL_COPY_VALUE(tmp->ptr, iarg->_zv);
24 return tmp;
25 }
26
27 void psi_to_bool(zval *return_value, set_value *set, impl_val *ret_val)
28 {
29 psi_to_int(return_value, set, ret_val);
30 convert_to_boolean(return_value);
31 }
32
33 static inline impl_val *psi_val_boolval(impl_val *tmp, token_t real_type, zend_bool boolval) {
34 switch (real_type) {
35 case PSI_T_INT8: tmp->i8 = boolval; break;
36 case PSI_T_UINT8: tmp->u8 = boolval; break;
37 case PSI_T_INT16: tmp->i16 = boolval; break;
38 case PSI_T_UINT16: tmp->u16 = boolval; break;
39 case PSI_T_INT32: tmp->i32 = boolval; break;
40 case PSI_T_UINT32: tmp->u32 = boolval; break;
41 case PSI_T_INT64: tmp->i64 = boolval; break;
42 case PSI_T_UINT64: tmp->u64 = boolval; break;
43 case PSI_T_FLOAT: tmp->fval = boolval; break;
44 case PSI_T_DOUBLE: tmp->dval = boolval; break;
45 #ifdef HAVE_LONG_DOUBLE
46 case PSI_T_LONG_DOUBLE: tmp->ldval = boolval; break;
47 #endif
48 EMPTY_SWITCH_DEFAULT_CASE();
49 }
50 return tmp;
51 }
52
53 impl_val *psi_let_boolval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
54 {
55 zend_bool boolval;
56 token_t real_type = type ? real_decl_type(type)->type : PSI_T_UINT8;
57
58 if (iarg->type->type == PSI_T_BOOL) {
59 boolval = iarg->val.zend.bval;
60 } else {
61 boolval = zend_is_true(iarg->_zv);
62 }
63
64 return psi_val_boolval(tmp, real_type, boolval);
65 }
66
67 # define RETVAL_LONG_U64(V) \
68 if (V > ZEND_LONG_MAX) { \
69 char d[24] = {0}; \
70 RETVAL_STRING(zend_print_ulong_to_buf(&d[22], V)); \
71 } else { \
72 RETVAL_LONG(V); \
73 }
74
75 void psi_to_int(zval *return_value, set_value *set, impl_val *ret_val)
76 {
77 decl_var *var = set->vars->vars[0];
78 token_t t = real_decl_type(var->arg->type)->type;
79 impl_val *v = deref_impl_val(ret_val, var);
80
81 switch (t) {
82 case PSI_T_INT8: RETVAL_LONG(v->i8); break;
83 case PSI_T_UINT8: RETVAL_LONG(v->u8); break;
84 case PSI_T_INT16: RETVAL_LONG(v->i16); break;
85 case PSI_T_UINT16: RETVAL_LONG(v->u16); break;
86 case PSI_T_INT32: RETVAL_LONG(v->i32); break;
87 case PSI_T_UINT32: RETVAL_LONG(v->u32); break;
88 case PSI_T_INT64: RETVAL_LONG(v->i64); break;
89 case PSI_T_UINT64: RETVAL_LONG_U64(v->u64); break;
90 case PSI_T_FLOAT: RETVAL_DOUBLE((double) v->fval); break;
91 case PSI_T_DOUBLE: RETVAL_DOUBLE(v->dval); break;
92 #ifdef HAVE_LONG_DOUBLE
93 case PSI_T_LONG_DOUBLE: RETVAL_DOUBLE((double) v->ldval); break;
94 #endif
95 EMPTY_SWITCH_DEFAULT_CASE();
96 }
97
98 convert_to_long(return_value);
99 }
100
101 static inline impl_val *psi_val_intval(impl_val *tmp, token_t real_type, zend_long intval) {
102 switch (real_type) {
103 case PSI_T_INT8: tmp->i8 = intval; break;
104 case PSI_T_UINT8: tmp->u8 = intval; break;
105 case PSI_T_INT16: tmp->i16 = intval; break;
106 case PSI_T_UINT16: tmp->u16 = intval; break;
107 case PSI_T_INT32: tmp->i32 = intval; break;
108 case PSI_T_UINT32: tmp->u32 = intval; break;
109 case PSI_T_INT: tmp->ival = intval; break;
110 case PSI_T_INT64: tmp->i64 = intval; break;
111 case PSI_T_UINT64: tmp->u64 = intval; break;
112 case PSI_T_FLOAT: tmp->fval = intval; break;
113 case PSI_T_DOUBLE: tmp->dval = intval; break;
114 #ifdef HAVE_LONG_DOUBLE
115 case PSI_T_LONG_DOUBLE: tmp->ldval = intval; break;
116 #endif
117 EMPTY_SWITCH_DEFAULT_CASE();
118 }
119
120 return tmp;
121 }
122
123 impl_val *psi_let_intval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
124 {
125 zend_long intval;
126 token_t real_type = type ? real_decl_type(type)->type : PSI_T_INT;
127
128 if (iarg->type->type == PSI_T_INT) {
129 intval = iarg->val.zend.lval;
130 } else {
131 intval = zval_get_long(iarg->_zv);
132 }
133
134 return psi_val_intval(tmp, real_type, intval);
135 }
136
137 void psi_to_double(zval *return_value, set_value *set, impl_val *ret_val)
138 {
139 decl_var *var = set->vars->vars[0];
140 token_t t = real_decl_type(var->arg->type)->type;
141 impl_val *v = deref_impl_val(ret_val, var);
142
143 switch (t) {
144 case PSI_T_FLOAT: RETVAL_DOUBLE((double) v->fval); break;
145 case PSI_T_DOUBLE: RETVAL_DOUBLE(v->dval); break;
146 #ifdef HAVE_LONG_DOUBLE
147 case PSI_T_LONG_DOUBLE: RETVAL_DOUBLE((double) v->ldval); break;
148 #endif
149 case PSI_T_INT8: RETVAL_DOUBLE((double) v->i8); break;
150 case PSI_T_UINT8: RETVAL_DOUBLE((double) v->u8); break;
151 case PSI_T_INT16: RETVAL_DOUBLE((double) v->i16); break;
152 case PSI_T_UINT16: RETVAL_DOUBLE((double) v->u16); break;
153 case PSI_T_INT32: RETVAL_DOUBLE((double) v->i32); break;
154 case PSI_T_UINT32: RETVAL_DOUBLE((double) v->u32); break;
155 case PSI_T_INT64: RETVAL_DOUBLE((double) v->i64); break;
156 case PSI_T_UINT64: RETVAL_DOUBLE((double) v->u64); break;
157 EMPTY_SWITCH_DEFAULT_CASE();
158 }
159 }
160
161 static inline impl_val *psi_val_floatval(impl_val *tmp, token_t real_type, double floatval) {
162 switch (real_type) {
163 case PSI_T_INT8: tmp->i8 = floatval; break;
164 case PSI_T_UINT8: tmp->u8 = floatval; break;
165 case PSI_T_INT16: tmp->i16 = floatval; break;
166 case PSI_T_UINT16: tmp->u16 = floatval; break;
167 case PSI_T_INT32: tmp->i32 = floatval; break;
168 case PSI_T_UINT32: tmp->u32 = floatval; break;
169 case PSI_T_INT64: tmp->i64 = floatval; break;
170 case PSI_T_UINT64: tmp->u64 = floatval; break;
171 case PSI_T_FLOAT: tmp->fval = floatval; break;
172 case PSI_T_DOUBLE: tmp->dval = floatval; break;
173 #ifdef HAVE_LONG_DOUBLE
174 case PSI_T_LONG_DOUBLE: tmp->ldval = floatval; break;
175 #endif
176 EMPTY_SWITCH_DEFAULT_CASE();
177 }
178
179 return tmp;
180 }
181
182 impl_val *psi_let_floatval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
183 {
184 double floatval;
185 token_t real_type = type ? real_decl_type(type)->type : PSI_T_DOUBLE;
186
187 if (iarg->type->type == PSI_T_FLOAT || iarg->type->type == PSI_T_DOUBLE) {
188 floatval = iarg->val.dval;
189 } else {
190 floatval = zval_get_double(iarg->_zv);
191 }
192
193 return psi_val_floatval(tmp, real_type, floatval);
194 }
195
196 void psi_to_string(zval *return_value, set_value *set, impl_val *ret_val)
197 {
198 char *str;
199 decl_var *var = set->vars->vars[0];
200 token_t t = real_decl_type(var->arg->type)->type;
201
202 switch (t) {
203 case PSI_T_FLOAT: RETVAL_DOUBLE((double) deref_impl_val(ret_val, var)->fval); break;
204 case PSI_T_DOUBLE: RETVAL_DOUBLE(deref_impl_val(ret_val, var)->dval); break;
205 #ifdef HAVE_LONG_DOUBLE
206 case PSI_T_LONG_DOUBLE: RETVAL_DOUBLE((double) deref_impl_val(ret_val, var)->ldval); break;
207 #endif
208 default:
209 if (!var->arg->var->pointer_level) {
210 RETVAL_STRINGL(&ret_val->cval, 1);
211 } else {
212 ret_val = deref_impl_val(ret_val, var);
213 if (var->arg->var->array_size) {
214 str = (char *) ret_val;
215 } else {
216 str = ret_val->ptr;
217 }
218 if (str) {
219 if (set->num) {
220 zend_long n = psi_long_num_exp(set->num, set->outer.val);
221 RETVAL_STRINGL(str, n);
222 } else {
223 RETVAL_STRING(str);
224 }
225 } else {
226 RETVAL_EMPTY_STRING();
227 }
228 }
229 return;
230 }
231
232 convert_to_string(return_value);
233 }
234
235 impl_val *psi_let_strval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
236 {
237 if (iarg->type->type == PSI_T_STRING) {
238 if (iarg->val.zend.str) {
239 tmp->ptr = estrndup(iarg->val.zend.str->val, iarg->val.zend.str->len);
240 *to_free = tmp->ptr;
241 } else {
242 tmp->ptr = "";
243 }
244 } else {
245 zend_string *zs = zval_get_string(iarg->_zv);
246 tmp->ptr = estrdup(zs->val);
247 *to_free = tmp->ptr;
248 zend_string_release(zs);
249 }
250
251 return tmp;
252 }
253
254 impl_val *psi_let_pathval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
255 {
256 tmp = psi_let_strval(tmp, type, iarg, to_free);
257 if (SUCCESS != php_check_open_basedir(tmp->ptr)) {
258 efree(tmp->ptr);
259 return *to_free = NULL;
260 }
261 return tmp;
262 }
263
264 impl_val *psi_let_strlen(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
265 {
266 if (iarg->type->type == PSI_T_STRING) {
267 if (iarg->val.zend.str) {
268 tmp->lval = iarg->val.zend.str->len;
269 } else {
270 tmp->lval = 0;
271 }
272 } else {
273 zend_string *zs = zval_get_string(iarg->_zv);
274 tmp->lval = zs->len;
275 zend_string_release(zs);
276 }
277
278 return tmp;
279 }
280
281 static impl_val *iterate(impl_val *val, size_t size, unsigned i, impl_val *tmp)
282 {
283 memset(tmp, 0, sizeof(*tmp));
284 memcpy(tmp, ((void*) val) + size * i, size);
285 return tmp;
286 }
287
288 void psi_from_zval_ex(impl_val **ptr, decl_arg *spec, token_t cast, zval *zv, void **tmp)
289 {
290 decl_type *real = real_decl_type(spec->type);
291 impl_val *val = *ptr;
292
293 switch (real->type) {
294 default:
295 ZEND_ASSERT(0);
296 /* no break */
297 case PSI_T_INT8:
298 val->i8 = zval_get_long(zv);
299 break;
300 case PSI_T_UINT8:
301 val->u8 = zval_get_long(zv);
302 break;
303 case PSI_T_INT16:
304 val->i16 = zval_get_long(zv);
305 break;
306 case PSI_T_UINT16:
307 val->u16 = zval_get_long(zv);
308 break;
309 case PSI_T_INT32:
310 val->i32 = zval_get_long(zv);
311 break;
312 case PSI_T_UINT32:
313 val->u32 = zval_get_long(zv);
314 break;
315 case PSI_T_INT64:
316 val->i64 = zval_get_long(zv);
317 break;
318 case PSI_T_UINT64:
319 val->u64 = zval_get_long(zv);
320 break;
321 case PSI_T_FLOAT:
322 val->fval = zval_get_double(zv);
323 break;
324 case PSI_T_DOUBLE:
325 val->dval = zval_get_double(zv);
326 break;
327 #ifdef HAVE_LONG_DOUBLE
328 case PSI_T_LONG_DOUBLE:
329 val->ldval = zval_get_double(zv);
330 break;
331 #endif
332 case PSI_T_ENUM:
333 val->ival = zval_get_long(zv);
334 break;
335 case PSI_T_STRUCT:
336 *tmp = *ptr = psi_array_to_struct(real->strct, HASH_OF(zv));
337 break;
338 }
339 }
340
341 void psi_from_zval(impl_val *mem, decl_arg *spec, zval *zv, void **tmp)
342 {
343 decl_type *type = real_decl_type(spec->type);
344
345 switch (type->type) {
346 case PSI_T_FLOAT:
347 mem->fval = (float) zval_get_double(zv);
348 break;
349 case PSI_T_DOUBLE:
350 mem->dval = zval_get_double(zv);
351 break;
352 case PSI_T_VOID:
353 case PSI_T_INT8:
354 case PSI_T_UINT8:
355 if (spec->var->pointer_level) {
356 zend_string *zs = zval_get_string(zv);
357 *tmp = mem->ptr = estrndup(zs->val, zs->len);
358 zend_string_release(zs);
359 break;
360 }
361 /* no break */
362 default:
363 mem->zend.lval = zval_get_long(zv);
364 break;
365 }
366 }
367
368 void *psi_array_to_struct(decl_struct *s, HashTable *arr)
369 {
370 size_t i, j = 0;
371 char *mem = ecalloc(1, s->size + s->args->count * sizeof(void *));
372
373 if (arr) for (i = 0; i < s->args->count; ++i) {
374 decl_arg *darg = s->args->args[i];
375 zval *entry = zend_hash_str_find_ind(arr, darg->var->name, strlen(darg->var->name));
376
377 if (entry) {
378 impl_val val;
379 void *tmp = NULL;
380
381 memset(&tmp, 0, sizeof(tmp));
382 psi_from_zval(&val, darg, entry, &tmp);
383 memcpy(mem + darg->layout->pos, &val, darg->layout->len);
384 if (tmp) {
385 ((void **)(mem + s->size))[j++] = tmp;
386 }
387 }
388 }
389 return mem;
390 }
391
392 void psi_to_recursive(zval *return_value, set_value *set, impl_val *r_val)
393 {
394 set->outer.set->func->handler(return_value, set, r_val);
395 }
396
397 void psi_to_array(zval *return_value, set_value *set, impl_val *r_val)
398 {
399 size_t i;
400 decl_var *var = set->vars->vars[0];
401 token_t t = real_decl_type(var->arg->type)->type;
402 impl_val tmp, *ret_val = deref_impl_val(r_val, var);
403
404 if ((intptr_t) ret_val <= (intptr_t) 0) {
405 RETURN_NULL();
406 }
407
408 array_init(return_value);
409
410 if (t == PSI_T_STRUCT) {
411 // decl_struct *s = real_decl_type(var->arg->type)->strct;
412
413 if (set->inner && set->inner->count) {
414 /* explicit member casts */
415 for (i = 0; i < set->inner->count; ++i) {
416 set_value *sub_set = set->inner->vals[i];
417 decl_var *sub_var = sub_set->vars->vars[0];
418
419 sub_set->outer.val = ret_val;
420
421 if (sub_var->arg) {
422 impl_val *tmp = NULL, *val;
423 zval ztmp;
424
425 val = struct_member_ref(sub_var->arg, ret_val, &tmp);
426 sub_set->func->handler(&ztmp, sub_set, val);
427 add_assoc_zval(return_value, sub_var->name, &ztmp);
428
429 if (tmp) {
430 free(tmp);
431 }
432 }
433 }
434 }
435 return;
436 }
437
438 if (var->arg->var->array_size) {
439 /* to_array(foo[NUMBER]) */
440 for (i = 0; i < var->arg->var->array_size; ++i) {
441 size_t size = psi_t_size(var->arg->var->pointer_level > 1 ? PSI_T_POINTER : t);
442 impl_val *ptr = iterate(ret_val, size, i, &tmp);
443 zval ele;
444
445 switch (t) {
446 case PSI_T_FLOAT:
447 ZVAL_DOUBLE(&ele, (double) ptr->fval);
448 break;
449 case PSI_T_DOUBLE:
450 ZVAL_DOUBLE(&ele, ptr->dval);
451 break;
452 default:
453 ZVAL_LONG(&ele, ptr->lval);
454 break;
455 }
456
457 add_next_index_zval(return_value, &ele);
458 }
459 return;
460 } else if (set->num) {
461 /* to_array(arr_var, num_expr, to_int(*arr_var)) */
462 zval ele;
463 char *ptr;
464 zend_long i, n = psi_long_num_exp(set->num, set->outer.val);
465 size_t size = psi_t_size(var->arg->var->pointer_level ? PSI_T_POINTER : t);
466 set_value *sub_set = set->inner->vals[0];
467
468 sub_set->outer.val = set->outer.val;
469 for (i = 0; i < n; ++i) {
470 ptr = (char *) ret_val->ptr + i * size;
471 sub_set->func->handler(&ele, sub_set, (void *) ptr);
472 add_next_index_zval(return_value, &ele);
473 }
474 } else {
475 /* to_array(arr_var, to_int(*arr_var)) */
476 zval ele;
477 char *ptr = ret_val->ptr;
478 size_t size = psi_t_size(var->arg->var->pointer_level ? PSI_T_POINTER : t);
479 set_value *sub_set = set->inner->vals[0];
480
481 sub_set->outer.val = set->outer.val;
482 while (*(void **) ptr) {
483 sub_set->func->handler(&ele, sub_set, (void *) ptr);
484 add_next_index_zval(return_value, &ele);
485 ptr += size;
486 }
487 }
488 }
489
490 impl_val *psi_let_arrval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
491 {
492 decl_type *real = real_decl_type(type);
493 HashTable *arr;
494
495 if (iarg->type->type != PSI_T_ARRAY) {
496 SEPARATE_ARG_IF_REF(iarg->_zv);
497 convert_to_array(iarg->_zv);
498 }
499 arr = HASH_OF(iarg->_zv);
500
501 switch (real->type) {
502 case PSI_T_STRUCT:
503 *to_free = tmp = psi_array_to_struct(real->strct, arr);
504 break;
505 EMPTY_SWITCH_DEFAULT_CASE();
506 }
507
508 return tmp;
509 }
510
511 void psi_to_object(zval *return_value, set_value *set, impl_val *r_val)
512 {
513 decl_var *var = set->vars->vars[0];
514 impl_val *ret_val = deref_impl_val(r_val, var);
515 psi_object *obj;
516
517 if ((intptr_t) ret_val->ptr > (intptr_t) 0) {
518 object_init_ex(return_value, psi_object_get_class_entry());
519 obj = PSI_OBJ(return_value, NULL);
520 obj->data = ret_val->ptr;
521 } else {
522 RETVAL_NULL();
523 }
524 }
525
526 impl_val *psi_let_objval(impl_val *tmp, decl_type *type, impl_arg *iarg, void **to_free)
527 {
528 psi_object *obj;
529
530 if (Z_TYPE_P(iarg->_zv) != IS_OBJECT
531 || !instanceof_function(Z_OBJCE_P(iarg->_zv), psi_object_get_class_entry())) {
532 return NULL;
533 }
534
535 obj = PSI_OBJ(iarg->_zv, NULL);
536 tmp->ptr = obj->data;
537
538 return tmp;
539 }
540