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