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