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