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                         zend_fcall_info fci;
 198                         zend_fcall_info_cache fcc;
 199
 200                         Z_PARAM_FUNC_EX(fci, fcc, 1, 0);
 201
 202                         if (fci.size) {
 203                                 iarg->val.zend.cb = calloc(1, sizeof(zend_fcall));
 204                                 iarg->val.zend.cb->fci = fci;
 205                                 iarg->val.zend.cb->fcc = fcc;
 206                         }
 207                 } else {
 208                         error_code = ZPP_ERROR_FAILURE;
 209                         break;
 210                 }
 211                 iarg->_zv = _arg;
 212                 ZVAL_DEREF(iarg->_zv);
 213                 if (_i < _num_args) {
 214                         goto nextarg;
 215                 }
 216         ZEND_PARSE_PARAMETERS_END_EX(
 217                 zend_restore_error_handling(&zeh);
 218                 return FAILURE
 219         );
 220
 221         /* set up defaults */
 222         for (i = 0; i < impl->func->args->count; ++i) {
 223                 if (i >= EX_NUM_ARGS() && iarg->def) {
 224                         iarg = impl->func->args->args[i];
 225
 226                         switch (iarg->type->type) {
 227                         case PSI_T_BOOL:
 228                                 iarg->val.zend.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
 229                                 break;
 230                         case PSI_T_INT:
 231                                 iarg->val.zend.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
 232                                 break;
 233                         case PSI_T_FLOAT:
 234                         case PSI_T_DOUBLE:
 235                                 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
 236                                 break;
 237                         case PSI_T_STRING:
 238                                 /* FIXME */
 239                                 iarg->val.zend.str = zend_string_init(&iarg->def->text[1], strlen(iarg->def->text) - 2, 0);
 240                                 break;
 241                         }
 242                 }
 243         }
 244
 245         zend_restore_error_handling(&zeh);
 246         return SUCCESS;
 247 }
 248
 249 static inline void *psi_do_calloc(let_calloc *alloc)
 250 {
 251         zend_long n = psi_long_num_exp(alloc->nmemb, NULL), s = psi_long_num_exp(alloc->size, NULL);
 252         void *mem = safe_emalloc(n, s, sizeof(void *));
 253         memset(mem, 0, n * s + sizeof(void *));
 254 #if 0
 255         fprintf(stderr, "calloc: %p\n", mem);
 256 #endif
 257         return mem;
 258 }
 259
 260 static inline impl_val *psi_let_val(token_t let_func, impl_arg *iarg, impl_val *arg_val, decl_struct *strct, void **to_free)
 261 {
 262         switch (let_func) {
 263         case PSI_T_BOOLVAL:
 264                 if (iarg->type->type == PSI_T_BOOL) {
 265                         arg_val->cval = iarg->val.zend.bval;
 266                 } else {
 267                         arg_val->cval = zend_is_true(iarg->_zv);
 268                 }
 269                 break;
 270         case PSI_T_INTVAL:
 271                 if (iarg->type->type == PSI_T_INT) {
 272                         arg_val->lval = iarg->val.zend.lval;
 273                 } else {
 274                         arg_val->lval = zval_get_long(iarg->_zv);
 275                 }
 276                 break;
 277         case PSI_T_FLOATVAL:
 278                 if (iarg->type->type == PSI_T_FLOAT || iarg->type->type == PSI_T_DOUBLE) {
 279                         arg_val->dval = iarg->val.dval;
 280                 } else {
 281                         arg_val->dval = zval_get_double(iarg->_zv);
 282                 }
 283                 break;
 284         case PSI_T_PATHVAL:
 285         case PSI_T_STRVAL:
 286                 if (iarg->type->type == PSI_T_STRING) {
 287                         if (iarg->val.zend.str) {
 288                                 arg_val->ptr = estrndup(iarg->val.zend.str->val, iarg->val.zend.str->len);
 289                                 *to_free = arg_val->ptr;
 290                         } else {
 291                                 arg_val->ptr = "";
 292                         }
 293                 } else {
 294                         zend_string *zs = zval_get_string(iarg->_zv);
 295                         arg_val->ptr = estrdup(zs->val);
 296                         *to_free = arg_val->ptr;
 297                         zend_string_release(zs);
 298                 }
 299                 if (PSI_T_PATHVAL == let_func) {
 300                         if (SUCCESS != php_check_open_basedir(arg_val->ptr)) {
 301                                 efree(arg_val->ptr);
 302                                 return NULL;
 303                         }
 304                 }
 305                 break;
 306         case PSI_T_STRLEN:
 307                 if (iarg->type->type == PSI_T_STRING) {
 308                         if (iarg->val.zend.str) {
 309                                 arg_val->lval = iarg->val.zend.str->len;
 310                         } else {
 311                                 arg_val->lval = 0;
 312                         }
 313                 } else {
 314                         zend_string *zs = zval_get_string(iarg->_zv);
 315                         arg_val->lval = zs->len;
 316                         zend_string_release(zs);
 317                 }
 318                 break;
 319         case PSI_T_ARRVAL:
 320                 if (iarg->type->type == PSI_T_ARRAY) {
 321                         arg_val = psi_array_to_struct(strct, HASH_OF(iarg->_zv));
 322                         *to_free = arg_val;
 323                 }
 324                 break;
 325         case PSI_T_OBJVAL:
 326                 if (iarg->type->type == PSI_T_OBJECT) {
 327                         psi_object *obj;
 328
 329                         if (!instanceof_function(Z_OBJCE_P(iarg->_zv), psi_object_get_class_entry())) {
 330                                 return NULL;
 331                         }
 332
 333                         obj = PSI_OBJ(iarg->_zv, NULL);
 334                         arg_val->ptr = obj->data;
 335                 }
 336                 break;
 337         case PSI_T_CALLBACK:
 338                 if (iarg->type->type == PSI_T_CALLABLE) {
 339
 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_CALLBACK:
 375                 arg_val->ptr = 0 /* callback closure */;
 376                 break;
 377         case PSI_LET_NUMEXP:
 378                 arg_val->zend.lval = psi_long_num_exp(let->val->data.num, NULL);
 379                 break;
 380         case PSI_LET_FUNC:
 381                 iarg = let->val->data.func->arg;
 382
 383                 if (!(darg->ptr = psi_let_val(let->val->data.func->type, iarg, darg->ptr, real_decl_type(darg->type)->strct, &darg->mem))) {
 384                         return NULL;
 385                 }
 386         }
 387
 388         if (let->val && let->val->flags.one.is_reference) {
 389                 return let->ptr = &darg->ptr;
 390         } else {
 391                 return let->ptr = darg->ptr;
 392         }
 393 }
 394
 395 static inline void psi_do_return(zval *return_value, return_stmt *ret)
 396 {
 397         ret->set->func->handler(return_value, ret->set, ret->set->vars->vars[0]->arg->ptr);
 398 }
 399
 400 static inline void psi_do_free(free_stmt *fre)
 401 {
 402         size_t i, j;
 403
 404         for (i = 0; i < fre->calls->count; ++i) {
 405                 free_call *f = fre->calls->list[i];
 406
 407                 for (j = 0; j < f->vars->count; ++j) {
 408                         decl_var *dvar = f->vars->vars[j];
 409                         decl_arg *darg = dvar->arg;
 410                         impl_val *fval = darg->let ? darg->let->ptr : darg->ptr;
 411
 412                         f->decl->call.args[j] = deref_impl_val(fval, dvar);
 413                 }
 414
 415                 /* FIXME: check in validate_* that free functions return scalar */
 416                 PSI_ContextCall(&PSI_G(context), &f->decl->call, NULL);
 417         }
 418 }
 419
 420 static inline void psi_clean_array_struct(decl_arg *darg) {
 421         if (darg->let
 422         &&      darg->let->val->kind == PSI_LET_FUNC
 423         &&      darg->let->val->data.func->type == PSI_T_ARRVAL) {
 424                 decl_type *type = real_decl_type(darg->type);
 425
 426                 if (type->type == PSI_T_STRUCT) {
 427                         void **ptr = (void **) ((char *) darg->mem + type->strct->size);
 428
 429                         while (*ptr) {
 430                                 efree(*ptr++);
 431                         }
 432                 }
 433         }
 434 }
 435
 436 static inline void psi_do_clean(impl *impl)
 437 {
 438         size_t i;
 439
 440         if (impl->decl->func->ptr != &impl->decl->func->val) {
 441                 efree(impl->decl->func->ptr);
 442                 impl->decl->func->ptr = &impl->decl->func->val;
 443         }
 444         for (i = 0; i < impl->func->args->count; ++i ) {
 445                 impl_arg *iarg = impl->func->args->args[i];
 446
 447                 switch (iarg->type->type) {
 448                 case PSI_T_STRING:
 449                         if (iarg->val.zend.str) {
 450                                 zend_string_release(iarg->val.zend.str);
 451                         }
 452                         break;
 453                 }
 454         }
 455
 456         if (impl->decl->args) for (i = 0; i < impl->decl->args->count; ++i) {
 457                 decl_arg *darg = impl->decl->args->args[i];
 458
 459                 if (darg->mem) {
 460                         psi_clean_array_struct(darg);
 461                         efree(darg->mem);
 462                         darg->mem = NULL;
 463                 }
 464                 darg->ptr = &darg->val;
 465         }
 466
 467         if (impl->func->args->vararg.args) {
 468                 free_impl_args(impl->func->args->vararg.args);
 469                 impl->func->args->vararg.args = NULL;
 470         }
 471         if (impl->func->args->vararg.types) {
 472                 efree(impl->func->args->vararg.types);
 473                 impl->func->args->vararg.types = NULL;
 474         }
 475         if (impl->func->args->vararg.values) {
 476                 efree(impl->func->args->vararg.values);
 477                 impl->func->args->vararg.values = NULL;
 478         }
 479         if (impl->func->args->vararg.free_list) {
 480                 void **list = impl->func->args->vararg.free_list;
 481
 482                 while (*list) {
 483                         efree(*list++);
 484                 }
 485
 486                 efree(impl->func->args->vararg.free_list);
 487                 impl->func->args->vararg.free_list = NULL;
 488         }
 489 }
 490
 491
 492 static inline void psi_do_args(impl *impl) {
 493         size_t i;
 494
 495         for (i = 0; i < impl->decl->args->count; ++i) {
 496                 impl->decl->call.args[i] = impl->decl->args->args[i]->let->ptr;
 497         }
 498
 499         if (!impl->decl->func->var->pointer_level) {
 500                 decl_type *real = real_decl_type(impl->decl->func->type);
 501
 502                 switch (real->type) {
 503                 case PSI_T_STRUCT:
 504                         impl->decl->func->ptr = psi_array_to_struct(real->strct, NULL);
 505                         break;
 506                 }
 507         }
 508 }
 509
 510 static inline impl_vararg *psi_do_varargs(impl *impl) {
 511         size_t i, j;
 512         impl_vararg *va = &impl->func->args->vararg;
 513         size_t vacount = va->args->count;
 514
 515
 516         if (!vacount) {
 517                 return NULL;
 518         }
 519
 520         va->types = ecalloc(vacount, sizeof(*va->types));
 521         va->values = ecalloc(vacount, sizeof(*va->values));
 522
 523         for (i = 0, j = 0; i < vacount; ++i) {
 524                 impl_arg *vaarg = va->args->args[i];
 525                 void *to_free = NULL;
 526                 token_t let_fn, vatype = va->name->type->type;
 527
 528                 if (vatype == PSI_T_MIXED) {
 529                         switch (Z_TYPE_P(vaarg->_zv)) {
 530                         case IS_TRUE:
 531                         case IS_FALSE:  vatype = PSI_T_BOOL;    break;
 532                         case IS_LONG:   vatype = PSI_T_INT;             break;
 533                         case IS_DOUBLE: vatype = PSI_T_FLOAT;   break;
 534                         default:                vatype = PSI_T_STRING;  break;
 535                         }
 536                 }
 537
 538
 539                 switch (vatype) {
 540                 case PSI_T_BOOL:        let_fn = PSI_T_BOOLVAL; break;
 541                 case PSI_T_INT:         let_fn = PSI_T_INTVAL;  break;
 542                 case PSI_T_FLOAT:
 543                 case PSI_T_DOUBLE:      let_fn = PSI_T_FLOATVAL;break;
 544                 case PSI_T_STRING:      let_fn = PSI_T_STRVAL;  break;
 545                 EMPTY_SWITCH_DEFAULT_CASE();
 546                 }
 547
 548                 va->types[i] = vatype;
 549                 /* FIXME: varargs with struct-by-value :) */
 550                 if (!psi_let_val(let_fn, vaarg, &va->values[i], NULL, &to_free)) {
 551                         return NULL;
 552                 }
 553
 554                 if (to_free) {
 555                         if (!va->free_list) {
 556                                 va->free_list = ecalloc(vacount - i + 1, sizeof(*va->free_list));
 557                         }
 558                         va->free_list[j++] = to_free;
 559                 }
 560         }
 561
 562         return va;
 563 }
 564
 565 void psi_call(zend_execute_data *execute_data, zval *return_value, impl *impl)
 566 {
 567         size_t i;
 568         impl_vararg *va = NULL;
 569
 570         memset(impl->decl->func->ptr, 0, sizeof(impl_val));
 571
 572         if (SUCCESS != psi_parse_args(execute_data, impl)) {
 573                 return;
 574         }
 575
 576         for (i = 0; i < impl->stmts->let.count; ++i) {
 577                 let_stmt *let = impl->stmts->let.list[i];
 578
 579                 if (!psi_do_let(let)) {
 580                         psi_do_return(return_value, impl->stmts->ret.list[0]);
 581                         psi_do_clean(impl);
 582                         return;
 583                 }
 584         }
 585
 586         if (impl->decl->args) {
 587                 psi_do_args(impl);
 588
 589                 if (impl->func->args->vararg.args) {
 590                         va = psi_do_varargs(impl);
 591                 }
 592         }
 593
 594         PSI_ContextCall(&PSI_G(context), &impl->decl->call, va);
 595         psi_do_return(return_value, impl->stmts->ret.list[0]);
 596
 597         for (i = 0; i < impl->stmts->set.count; ++i) {
 598                 set_stmt *set = impl->stmts->set.list[i];
 599
 600                 if (set->arg->_zv) {
 601                         psi_do_set(set->arg->_zv, set->val);
 602                 }
 603         }
 604
 605         for (i = 0; i < impl->stmts->fre.count; ++i) {
 606                 free_stmt *fre = impl->stmts->fre.list[i];
 607
 608                 psi_do_free(fre);
 609         }
 610         psi_do_clean(impl);
 611 }