git.m6w6.name Git - m6w6/ext-psi/blob - src/engine.c

   1 #ifdef HAVE_CONFIG_H
   2 # include "config.h"
   3 #endif
   4
   5 #include "php.h"
   6 #include "php_psi.h"
   7
   8 #include "zend_exceptions.h"
   9
  10 #include "engine.h"
  11 #include "calc.h"
  12 #include "marshal.h"
  13
  14 size_t psi_t_alignment(token_t t)
  15 {
  16 #define PSI_ALIGNOF(T) case PSI_T_## T: return ALIGNOF_## T ##_T;
  17         switch (t) {
  18         PSI_ALIGNOF(INT8);
  19         PSI_ALIGNOF(UINT8);
  20         PSI_ALIGNOF(INT16);
  21         PSI_ALIGNOF(UINT16);
  22         PSI_ALIGNOF(INT32);
  23         PSI_ALIGNOF(UINT32);
  24         PSI_ALIGNOF(INT64);
  25         PSI_ALIGNOF(UINT64);
  26         case PSI_T_FLOAT:
  27                 return ALIGNOF_FLOAT;
  28         case PSI_T_DOUBLE:
  29                 return ALIGNOF_DOUBLE;
  30         case PSI_T_POINTER:
  31         case PSI_T_FUNCTION:
  32         case PSI_T_STRUCT:
  33                 return ALIGNOF_VOID_P;
  34         case PSI_T_ENUM:
  35                 return ALIGNOF_INT;
  36         EMPTY_SWITCH_DEFAULT_CASE();
  37         }
  38         return 0;
  39 }
  40
  41 size_t psi_t_size(token_t t)
  42 {
  43 #define PSI_SIZEOF(T) case PSI_T_## T : return SIZEOF_## T ##_T;
  44         switch (t) {
  45         PSI_SIZEOF(INT8);
  46         PSI_SIZEOF(UINT8);
  47         PSI_SIZEOF(INT16);
  48         PSI_SIZEOF(UINT16);
  49         PSI_SIZEOF(INT32);
  50         PSI_SIZEOF(UINT32);
  51         PSI_SIZEOF(INT64);
  52         PSI_SIZEOF(UINT64);
  53         case PSI_T_FLOAT:
  54                 return SIZEOF_FLOAT;
  55         case PSI_T_DOUBLE:
  56                 return SIZEOF_DOUBLE;
  57         case PSI_T_POINTER:
  58         case PSI_T_FUNCTION:
  59                 return SIZEOF_VOID_P;
  60         case PSI_T_ENUM:
  61                 return SIZEOF_INT;
  62         EMPTY_SWITCH_DEFAULT_CASE();
  63         }
  64         return 0;
  65 }
  66
  67 size_t psi_t_align(token_t t, size_t s)
  68 {
  69         size_t a = psi_t_alignment(t);
  70         return ((s - 1) | (a - 1)) + 1;
  71 }
  72
  73 size_t psi_offset_padding(size_t diff, size_t alignment)
  74 {
  75         if (diff && diff <= ((diff - 1) | (alignment -1)) + 1) {
  76                 diff = 0;
  77         }
  78
  79         return diff;
  80 }
  81
  82 int psi_internal_type(impl_type *type)
  83 {
  84         switch (type->type) {
  85         case PSI_T_BOOL:
  86                 return _IS_BOOL;
  87         case PSI_T_INT:
  88                 return IS_LONG;
  89         case PSI_T_FLOAT:
  90         case PSI_T_DOUBLE:
  91                 return IS_DOUBLE;
  92         case PSI_T_STRING:
  93                 return IS_STRING;
  94         case PSI_T_ARRAY:
  95                 return IS_ARRAY;
  96         default:
  97                 return 0;
  98         }
  99 }
 100
 101 zend_internal_arg_info *psi_internal_arginfo(impl *impl)
 102 {
 103         size_t i;
 104         zend_internal_arg_info *aip;
 105         zend_internal_function_info *fi;
 106
 107         aip = calloc(impl->func->args->count + 1 + !!impl->func->args->vararg.name, sizeof(*aip));
 108
 109         fi = (zend_internal_function_info *) &aip[0];
 110         fi->allow_null = 1;
 111         fi->required_num_args = psi_num_min_args(impl);
 112         fi->return_reference = impl->func->return_reference;
 113         fi->type_hint = psi_internal_type(impl->func->return_type);
 114
 115         if (impl->func->args->vararg.name) {
 116                 impl_arg *vararg = impl->func->args->vararg.name;
 117                 zend_internal_arg_info *ai = &aip[impl->func->args->count];
 118
 119                 ai->name = vararg->var->name;
 120                 ai->allow_null = 1;
 121                 ai->type_hint = psi_internal_type(vararg->type);
 122                 if (vararg->var->reference) {
 123                         ai->pass_by_reference = 1;
 124                 }
 125                 ai->is_variadic = 1;
 126         }
 127
 128         for (i = 0; i < impl->func->args->count; ++i) {
 129                 impl_arg *iarg = impl->func->args->args[i];
 130                 zend_internal_arg_info *ai = &aip[i+1];
 131
 132                 ai->name = iarg->var->name;
 133                 ai->type_hint = psi_internal_type(iarg->type);
 134                 if (iarg->var->reference) {
 135                         ai->pass_by_reference = 1;
 136                 }
 137                 //if (iarg->var->reference || (iarg->def && iarg->def->type == PSI_T_NULL)) {
 138                         ai->allow_null = 1;
 139                 //}
 140         }
 141
 142         return aip;
 143 }
 144
 145 size_t psi_num_min_args(impl *impl)
 146 {
 147         size_t i, n = impl->func->args->count;
 148
 149         for (i = 0; i < impl->func->args->count; ++i) {
 150                 if (impl->func->args->args[i]->def) {
 151                         --n;
 152                 }
 153         }
 154         return n;
 155 }
 156
 157
 158 static inline ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
 159 {
 160         size_t i;
 161         impl_arg *iarg;
 162         zend_error_handling zeh;
 163
 164         zend_replace_error_handling(EH_THROW, zend_exception_get_default(), &zeh);
 165
 166         if (!impl->func->args->count) {
 167                 ZEND_RESULT_CODE rv;
 168
 169                 rv = zend_parse_parameters_none();
 170                 zend_restore_error_handling(&zeh);
 171                 return rv;
 172         }
 173
 174         ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->vararg.name ? -1 : impl->func->args->count)
 175         nextarg:
 176                 if (impl->func->args->vararg.name && _i >= impl->func->args->count) {
 177                         impl_arg *varg = impl->func->args->vararg.name;
 178                         iarg = init_impl_arg(
 179                                         init_impl_type(varg->type->type, varg->type->name),
 180                                         init_impl_var(varg->var->name, varg->var->reference),
 181                                         NULL);
 182
 183                         Z_PARAM_OPTIONAL;
 184                         if (_i == impl->func->args->count) {
 185                                 impl->func->args->vararg.args = init_impl_args(iarg);
 186                         } else {
 187                                 add_impl_arg(impl->func->args->vararg.args, iarg);
 188                         }
 189                 } else {
 190                         iarg = impl->func->args->args[_i];
 191                         if (iarg->def) {
 192                                 Z_PARAM_OPTIONAL;
 193                         }
 194                 }
 195                 if (PSI_T_BOOL == iarg->type->type) {
 196                         Z_PARAM_BOOL(iarg->val.zend.bval);
 197                 } else if (PSI_T_INT == iarg->type->type) {
 198                         Z_PARAM_LONG(iarg->val.zend.lval);
 199                 } else if (PSI_T_FLOAT == iarg->type->type || PSI_T_DOUBLE == iarg->type->type) {
 200                         Z_PARAM_DOUBLE(iarg->val.dval);
 201                 } else if (PSI_T_STRING == iarg->type->type) {
 202                         Z_PARAM_STR_EX(iarg->val.zend.str, 1, iarg->var->reference);
 203                         if (iarg->val.zend.str) {
 204                                 zend_string_addref(iarg->val.zend.str);
 205                         }
 206                 } else if (PSI_T_ARRAY == iarg->type->type) {
 207                         Z_PARAM_PROLOGUE(0);
 208                 } else if (PSI_T_OBJECT == iarg->type->type) {
 209                         Z_PARAM_PROLOGUE(0);
 210                 } else if (PSI_T_MIXED == iarg->type->type) {
 211                         Z_PARAM_PROLOGUE(0);
 212                 } else {
 213                         error_code = ZPP_ERROR_FAILURE;
 214                         break;
 215                 }
 216                 iarg->_zv = _arg;
 217                 ZVAL_DEREF(iarg->_zv);
 218                 if (_i < _num_args) {
 219                         goto nextarg;
 220                 }
 221         ZEND_PARSE_PARAMETERS_END_EX(
 222                 zend_restore_error_handling(&zeh);
 223                 return FAILURE
 224         );
 225
 226         /* set up defaults */
 227         for (i = 0; i < impl->func->args->count; ++i) {
 228                 if (i >= EX_NUM_ARGS() && iarg->def) {
 229                         iarg = impl->func->args->args[i];
 230
 231                         switch (iarg->type->type) {
 232                         case PSI_T_BOOL:
 233                                 iarg->val.zend.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
 234                                 break;
 235                         case PSI_T_INT:
 236                                 iarg->val.zend.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
 237                                 break;
 238                         case PSI_T_FLOAT:
 239                         case PSI_T_DOUBLE:
 240                                 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
 241                                 break;
 242                         case PSI_T_STRING:
 243                                 /* FIXME */
 244                                 iarg->val.zend.str = zend_string_init(&iarg->def->text[1], strlen(iarg->def->text) - 2, 0);
 245                                 break;
 246                         }
 247                 }
 248         }
 249
 250         zend_restore_error_handling(&zeh);
 251         return SUCCESS;
 252 }
 253
 254 static inline void *psi_do_calloc(let_calloc *alloc)
 255 {
 256         zend_long n = psi_long_num_exp(alloc->nmemb, NULL), s = psi_long_num_exp(alloc->size, NULL);
 257         void *mem = safe_emalloc(n, s, sizeof(void *));
 258         memset(mem, 0, n * s + sizeof(void *));
 259 #if 0
 260         fprintf(stderr, "calloc: %p\n", mem);
 261 #endif
 262         return mem;
 263 }
 264
 265 static inline impl_val *psi_let_val(token_t let_func, impl_arg *iarg, impl_val *arg_val, decl_struct *strct, void **to_free)
 266 {
 267         switch (let_func) {
 268         case PSI_T_BOOLVAL:
 269                 if (iarg->type->type == PSI_T_BOOL) {
 270                         arg_val->cval = iarg->val.zend.bval;
 271                 } else {
 272                         arg_val->cval = zend_is_true(iarg->_zv);
 273                 }
 274                 break;
 275         case PSI_T_INTVAL:
 276                 if (iarg->type->type == PSI_T_INT) {
 277                         arg_val->lval = iarg->val.zend.lval;
 278                 } else {
 279                         arg_val->lval = zval_get_long(iarg->_zv);
 280                 }
 281                 break;
 282         case PSI_T_FLOATVAL:
 283                 if (iarg->type->type == PSI_T_FLOAT || iarg->type->type == PSI_T_DOUBLE) {
 284                         arg_val->dval = iarg->val.dval;
 285                 } else {
 286                         arg_val->dval = zval_get_double(iarg->_zv);
 287                 }
 288                 break;
 289         case PSI_T_PATHVAL:
 290         case PSI_T_STRVAL:
 291                 if (iarg->type->type == PSI_T_STRING) {
 292                         if (iarg->val.zend.str) {
 293                                 arg_val->ptr = estrndup(iarg->val.zend.str->val, iarg->val.zend.str->len);
 294                                 *to_free = arg_val->ptr;
 295                         } else {
 296                                 arg_val->ptr = "";
 297                         }
 298                 } else {
 299                         zend_string *zs = zval_get_string(iarg->_zv);
 300                         arg_val->ptr = estrdup(zs->val);
 301                         *to_free = arg_val->ptr;
 302                         zend_string_release(zs);
 303                 }
 304                 if (PSI_T_PATHVAL == let_func) {
 305                         if (SUCCESS != php_check_open_basedir(arg_val->ptr)) {
 306                                 efree(arg_val->ptr);
 307                                 return NULL;
 308                         }
 309                 }
 310                 break;
 311         case PSI_T_STRLEN:
 312                 if (iarg->type->type == PSI_T_STRING) {
 313                         if (iarg->val.zend.str) {
 314                                 arg_val->lval = iarg->val.zend.str->len;
 315                         } else {
 316                                 arg_val->lval = 0;
 317                         }
 318                 } else {
 319                         zend_string *zs = zval_get_string(iarg->_zv);
 320                         arg_val->lval = zs->len;
 321                         zend_string_release(zs);
 322                 }
 323                 break;
 324         case PSI_T_ARRVAL:
 325                 if (iarg->type->type == PSI_T_ARRAY) {
 326                         arg_val = psi_array_to_struct(strct, HASH_OF(iarg->_zv));
 327                         *to_free = arg_val;
 328                 }
 329                 break;
 330         case PSI_T_OBJVAL:
 331                 if (iarg->type->type == PSI_T_OBJECT) {
 332                         psi_object *obj;
 333
 334                         if (!instanceof_function(Z_OBJCE_P(iarg->_zv), psi_object_get_class_entry())) {
 335                                 return NULL;
 336                         }
 337
 338                         obj = PSI_OBJ(iarg->_zv, NULL);
 339                         arg_val->ptr = obj->data;
 340                 }
 341                 break;
 342         EMPTY_SWITCH_DEFAULT_CASE();
 343         }
 344         return arg_val;
 345 }
 346
 347 static inline void *psi_do_let(let_stmt *let)
 348 {
 349         decl_arg *darg = let->var->arg;
 350         impl_val *arg_val = darg->ptr;
 351         impl_arg *iarg;
 352
 353         switch (let->val ? let->val->kind : PSI_LET_NULL) {
 354         case PSI_LET_TMP:
 355                 memcpy(arg_val, deref_impl_val(let->val->data.var->arg->let->ptr, let->val->data.var), sizeof(*arg_val));
 356 #if 0
 357                 fprintf(stderr, "LET TMP: %p -> %p\n",
 358                                 let->val->data.var->arg->let->ptr,
 359                                 arg_val->ptr);
 360 #endif
 361                 break;
 362         case PSI_LET_NULL:
 363                 if (darg->var->array_size) {
 364                         arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val));
 365                         darg->mem = arg_val->ptr;
 366                 } else {
 367                         memset(arg_val, 0, sizeof(*arg_val));
 368                 }
 369                 break;
 370         case PSI_LET_CALLOC:
 371                 arg_val->ptr = psi_do_calloc(let->val->data.alloc);
 372                 darg->mem = arg_val->ptr;
 373                 break;
 374         case PSI_LET_NUMEXP:
 375                 arg_val->zend.lval = psi_long_num_exp(let->val->data.num, NULL);
 376                 break;
 377         case PSI_LET_FUNC:
 378                 iarg = let->val->data.func->arg;
 379
 380                 if (!(darg->ptr = psi_let_val(let->val->data.func->type, iarg, darg->ptr, real_decl_type(darg->type)->strct, &darg->mem))) {
 381                         return NULL;
 382                 }
 383         }
 384
 385         if (let->val && let->val->flags.one.is_reference) {
 386                 return let->ptr = &darg->ptr;
 387         } else {
 388                 return let->ptr = darg->ptr;
 389         }
 390 }
 391
 392 static inline void psi_do_set(zval *return_value, set_value *set)
 393 {
 394         decl_arg *set_arg = set->vars->vars[0]->arg;
 395
 396         zval_dtor(return_value);
 397         set->func->handler(return_value, set, set_arg->let ? set_arg->let->ptr : set_arg->ptr);
 398 }
 399
 400 static inline void psi_do_return(zval *return_value, return_stmt *ret)
 401 {
 402         ret->set->func->handler(return_value, ret->set, ret->set->vars->vars[0]->arg->ptr);
 403 }
 404
 405 static inline void psi_do_free(free_stmt *fre)
 406 {
 407         size_t i, j;
 408
 409         for (i = 0; i < fre->calls->count; ++i) {
 410                 free_call *f = fre->calls->list[i];
 411
 412                 for (j = 0; j < f->vars->count; ++j) {
 413                         decl_var *dvar = f->vars->vars[j];
 414                         decl_arg *darg = dvar->arg;
 415                         impl_val *fval = darg->let ? darg->let->ptr : darg->ptr;
 416
 417                         f->decl->call.args[j] = deref_impl_val(fval, dvar);
 418                 }
 419
 420                 /* FIXME: check in validate_* that free functions return scalar */
 421                 PSI_ContextCall(&PSI_G(context), &f->decl->call, NULL);
 422         }
 423 }
 424
 425 static inline void psi_clean_array_struct(decl_arg *darg) {
 426         if (darg->let
 427         &&      darg->let->val->kind == PSI_LET_FUNC
 428         &&      darg->let->val->data.func->type == PSI_T_ARRVAL) {
 429                 decl_type *type = real_decl_type(darg->type);
 430
 431                 if (type->type == PSI_T_STRUCT) {
 432                         void **ptr = (void **) ((char *) darg->mem + type->strct->size);
 433
 434                         while (*ptr) {
 435                                 efree(*ptr++);
 436                         }
 437                 }
 438         }
 439 }
 440
 441 static inline void psi_do_clean(impl *impl)
 442 {
 443         size_t i;
 444
 445         if (impl->decl->func->ptr != &impl->decl->func->val) {
 446                 efree(impl->decl->func->ptr);
 447                 impl->decl->func->ptr = &impl->decl->func->val;
 448         }
 449         for (i = 0; i < impl->func->args->count; ++i ) {
 450                 impl_arg *iarg = impl->func->args->args[i];
 451
 452                 switch (iarg->type->type) {
 453                 case PSI_T_STRING:
 454                         if (iarg->val.zend.str) {
 455                                 zend_string_release(iarg->val.zend.str);
 456                         }
 457                         break;
 458                 }
 459         }
 460
 461         if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 462                 decl_arg *darg = impl->decl->args->args[i];
 463
 464                 if (darg->mem) {
 465                         psi_clean_array_struct(darg);
 466                         efree(darg->mem);
 467                         darg->mem = NULL;
 468                 }
 469                 darg->ptr = &darg->val;
 470         }
 471
 472         if (impl->func->args->vararg.args) {
 473                 free_impl_args(impl->func->args->vararg.args);
 474                 impl->func->args->vararg.args = NULL;
 475         }
 476         if (impl->func->args->vararg.types) {
 477                 efree(impl->func->args->vararg.types);
 478                 impl->func->args->vararg.types = NULL;
 479         }
 480         if (impl->func->args->vararg.values) {
 481                 efree(impl->func->args->vararg.values);
 482                 impl->func->args->vararg.values = NULL;
 483         }
 484         if (impl->func->args->vararg.free_list) {
 485                 void **list = impl->func->args->vararg.free_list;
 486
 487                 while (*list) {
 488                         efree(*list++);
 489                 }
 490
 491                 efree(impl->func->args->vararg.free_list);
 492                 impl->func->args->vararg.free_list = NULL;
 493         }
 494 }
 495
 496
 497 static inline void psi_do_args(impl *impl) {
 498         size_t i;
 499
 500         for (i = 0; i < impl->decl->args->count; ++i) {
 501                 impl->decl->call.args[i] = impl->decl->args->args[i]->let->ptr;
 502         }
 503
 504         if (!impl->decl->func->var->pointer_level) {
 505                 decl_type *real = real_decl_type(impl->decl->func->type);
 506
 507                 switch (real->type) {
 508                 case PSI_T_STRUCT:
 509                         impl->decl->func->ptr = psi_array_to_struct(real->strct, NULL);
 510                         break;
 511                 }
 512         }
 513 }
 514
 515 static inline impl_vararg *psi_do_varargs(impl *impl) {
 516         size_t i, j;
 517         impl_vararg *va = &impl->func->args->vararg;
 518         size_t vacount = va->args->count;
 519
 520
 521         if (!vacount) {
 522                 return NULL;
 523         }
 524
 525         va->types = ecalloc(vacount, sizeof(*va->types));
 526         va->values = ecalloc(vacount, sizeof(*va->values));
 527
 528         for (i = 0, j = 0; i < vacount; ++i) {
 529                 impl_arg *vaarg = va->args->args[i];
 530                 void *to_free = NULL;
 531                 token_t let_fn, vatype = va->name->type->type;
 532
 533                 if (vatype == PSI_T_MIXED) {
 534                         switch (Z_TYPE_P(vaarg->_zv)) {
 535                         case IS_TRUE:
 536                         case IS_FALSE:  vatype = PSI_T_BOOL;    break;
 537                         case IS_LONG:   vatype = PSI_T_INT;             break;
 538                         case IS_DOUBLE: vatype = PSI_T_FLOAT;   break;
 539                         default:                vatype = PSI_T_STRING;  break;
 540                         }
 541                 }
 542
 543
 544                 switch (vatype) {
 545                 case PSI_T_BOOL:        let_fn = PSI_T_BOOLVAL; break;
 546                 case PSI_T_INT:         let_fn = PSI_T_INTVAL;  break;
 547                 case PSI_T_FLOAT:
 548                 case PSI_T_DOUBLE:      let_fn = PSI_T_FLOATVAL;break;
 549                 case PSI_T_STRING:      let_fn = PSI_T_STRVAL;  break;
 550                 EMPTY_SWITCH_DEFAULT_CASE();
 551                 }
 552
 553                 va->types[i] = vatype;
 554                 /* FIXME: varargs with struct-by-value :) */
 555                 if (!psi_let_val(let_fn, vaarg, &va->values[i], NULL, &to_free)) {
 556                         return NULL;
 557                 }
 558
 559                 if (to_free) {
 560                         if (!va->free_list) {
 561                                 va->free_list = ecalloc(vacount - i + 1, sizeof(*va->free_list));
 562                         }
 563                         va->free_list[j++] = to_free;
 564                 }
 565         }
 566
 567         return va;
 568 }
 569
 570 void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl)
 571 {
 572         size_t i;
 573         impl_vararg *va = NULL;
 574
 575         memset(impl->decl->func->ptr, 0, sizeof(impl_val));
 576
 577         if (SUCCESS != psi_parse_args(execute_data, impl)) {
 578                 return;
 579         }
 580
 581         for (i = 0; i < impl->stmts->let.count; ++i) {
 582                 let_stmt *let = impl->stmts->let.list[i];
 583
 584                 if (!psi_do_let(let)) {
 585                         psi_do_return(return_value, impl->stmts->ret.list[0]);
 586                         psi_do_clean(impl);
 587                         return;
 588                 }
 589         }
 590
 591         if (impl->decl->args) {
 592                 psi_do_args(impl);
 593
 594                 if (impl->func->args->vararg.args) {
 595                         va = psi_do_varargs(impl);
 596                 }
 597         }
 598
 599         PSI_ContextCall(&PSI_G(context), &impl->decl->call, va);
 600         psi_do_return(return_value, impl->stmts->ret.list[0]);
 601
 602         for (i = 0; i < impl->stmts->set.count; ++i) {
 603                 set_stmt *set = impl->stmts->set.list[i];
 604
 605                 if (set->arg->_zv) {
 606                         psi_do_set(set->arg->_zv, set->val);
 607                 }
 608         }
 609
 610         for (i = 0; i < impl->stmts->fre.count; ++i) {
 611                 free_stmt *fre = impl->stmts->fre.list[i];
 612
 613                 psi_do_free(fre);
 614         }
 615         psi_do_clean(impl);
 616 }