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

   1
   2 #ifdef HAVE_CONFIG_H
   3 #include "config.h"
   4 #endif
   5
   6 #include "php.h"
   7 #include "php_ini.h"
   8 #include "ext/standard/info.h"
   9
  10 #include "php_psi.h"
  11 #include "parser.h"
  12
  13 #include "libjit.h"
  14 #include "libffi.h"
  15
  16 ZEND_DECLARE_MODULE_GLOBALS(psi);
  17
  18 PHP_INI_BEGIN()
  19         STD_PHP_INI_ENTRY("psi.engine", "ffi", PHP_INI_SYSTEM, OnUpdateString, engine, zend_psi_globals, psi_globals)
  20         STD_PHP_INI_ENTRY("psi.directory", "psis", PHP_INI_SYSTEM, OnUpdateString, directory, zend_psi_globals, psi_globals)
  21 PHP_INI_END();
  22
  23 void psi_error(int type, const char *msg, ...)
  24 {
  25         char buf[0x1000];
  26         va_list argv;
  27
  28         va_start(argv, msg);
  29         vslprintf(buf, 0x1000, msg, argv);
  30         va_end(argv);
  31
  32         php_error(type, buf);
  33 }
  34
  35 int psi_internal_type(impl_type *type)
  36 {
  37         switch (type->type) {
  38         case PSI_T_BOOL:
  39                 return _IS_BOOL;
  40         case PSI_T_INT:
  41         case PSI_T_LONG:
  42                 return IS_LONG;
  43         case PSI_T_FLOAT:
  44         case PSI_T_DOUBLE:
  45                 return IS_DOUBLE;
  46         case PSI_T_STRING:
  47                 return IS_STRING;
  48         case PSI_T_ARRAY:
  49                 return IS_ARRAY;
  50         default:
  51                 return 0;
  52         }
  53 }
  54
  55 zend_internal_arg_info *psi_internal_arginfo(impl *impl)
  56 {
  57         size_t i;
  58         zend_internal_arg_info *aip;
  59         zend_internal_function_info *fi;
  60
  61         aip = calloc(impl->func->args->count + 1, sizeof(*aip));
  62
  63         fi = (zend_internal_function_info *) &aip[0];
  64         fi->required_num_args = psi_num_min_args(impl);
  65         fi->return_reference = impl->func->return_reference;
  66         fi->type_hint = psi_internal_type(impl->func->return_type);
  67
  68         for (i = 0; i < impl->func->args->count; ++i) {
  69                 impl_arg *iarg = impl->func->args->args[i];
  70                 zend_internal_arg_info *ai = &aip[i+1];
  71
  72                 ai->name = iarg->var->name;
  73                 ai->type_hint = psi_internal_type(iarg->type);
  74                 if (iarg->var->reference) {
  75                         ai->pass_by_reference = 1;
  76                 }
  77                 if (iarg->var->reference || (iarg->def && iarg->def->type == PSI_T_NULL)) {
  78                         ai->allow_null = 1;
  79                 }
  80         }
  81
  82         return aip;
  83 }
  84
  85 size_t psi_num_min_args(impl *impl)
  86 {
  87         size_t i, n = impl->func->args->count;
  88
  89         for (i = 0; i < impl->func->args->count; ++i) {
  90                 if (impl->func->args->args[i]->def) {
  91                         --n;
  92                 }
  93         }
  94         return n;
  95 }
  96
  97 void psi_to_int(impl_val *ret_val, decl_arg *func, zval *return_value)
  98 {
  99         switch (real_decl_type(func->type)->type) {
 100         case PSI_T_FLOAT:
 101         case PSI_T_DOUBLE:
 102                 RETVAL_DOUBLE(deref_impl_val(0, ret_val, func)->dval);
 103                 convert_to_long(return_value);
 104                 break;
 105         default:
 106                 RETVAL_LONG(deref_impl_val(0, ret_val, func)->lval);
 107         }
 108 }
 109
 110 void psi_to_double(impl_val *ret_val, decl_arg *func, zval *return_value)
 111 {
 112         RETVAL_DOUBLE(deref_impl_val(0, ret_val, func)->dval);
 113 }
 114
 115 void psi_to_string(impl_val *ret_val, decl_arg *func, zval *return_value)
 116 {
 117         switch (real_decl_type(func->type)->type) {
 118         case PSI_T_CHAR:
 119         case PSI_T_SINT8:
 120         case PSI_T_UINT8:
 121                 if (!func->var->pointer_level) {
 122                         char chr = ret_val->lval;
 123                         RETVAL_STRINGL(&chr, 1);
 124                 } else {
 125                         ret_val = deref_impl_val(1, ret_val, func);
 126                         if (ret_val->ptr) {
 127                                 RETVAL_STRING(ret_val->ptr);
 128                         } else {
 129                                 RETVAL_EMPTY_STRING();
 130                         }
 131                 }
 132                 break;
 133         case PSI_T_FLOAT:
 134         case PSI_T_DOUBLE:
 135                 RETVAL_DOUBLE(deref_impl_val(0, ret_val, func)->dval);
 136                 convert_to_string(return_value);
 137                 break;
 138         default:
 139                 RETVAL_LONG(deref_impl_val(0, ret_val, func)->lval);
 140                 convert_to_string(return_value);
 141                 break;
 142         }
 143 }
 144
 145 size_t psi_t_alignment(token_t t)
 146 {
 147         size_t align;
 148 #define PSI_TAS_D(T) struct PSI_TAS_ ##T { \
 149         char c; \
 150         T x; \
 151 }
 152 #define PSI_TAS_P(T) struct PSI_TAS_ ## T ## _pointer { \
 153         char c; \
 154         T *x; \
 155 }
 156 #define PSI_TAS_C(T) align = offsetof(struct PSI_TAS_ ##T, x)
 157 #define PSI_TAS_CASE(T) { \
 158         PSI_TAS_D(T); \
 159         PSI_TAS_C(T); \
 160 }
 161         switch (t) {
 162         case PSI_T_CHAR:
 163                 PSI_TAS_CASE(char);
 164                 break;
 165         case PSI_T_SINT8:
 166                 PSI_TAS_CASE(int8_t);
 167                 break;
 168         case PSI_T_UINT8:
 169                 PSI_TAS_CASE(uint8_t);
 170                 break;
 171         case PSI_T_SHORT:
 172                 PSI_TAS_CASE(short);
 173                 break;
 174         case PSI_T_SINT16:
 175                 PSI_TAS_CASE(int16_t);
 176                 break;
 177         case PSI_T_UINT16:
 178                 PSI_TAS_CASE(uint16_t);
 179                 break;
 180         case PSI_T_INT:
 181                 PSI_TAS_CASE(int);
 182                 break;
 183         case PSI_T_SINT32:
 184                 PSI_TAS_CASE(int32_t);
 185                 break;
 186         case PSI_T_UINT32:
 187                 PSI_TAS_CASE(uint32_t);
 188                 break;
 189         case PSI_T_LONG:
 190                 PSI_TAS_CASE(long);
 191                 break;
 192         case PSI_T_SINT64:
 193                 PSI_TAS_CASE(int64_t);
 194                 break;
 195         case PSI_T_UINT64:
 196                 PSI_TAS_CASE(uint64_t);
 197                 break;
 198         case PSI_T_FLOAT:
 199                 PSI_TAS_CASE(float);
 200                 break;
 201         case PSI_T_DOUBLE:
 202                 PSI_TAS_CASE(double);
 203                 break;
 204         case PSI_T_SIZE_T:
 205                 PSI_TAS_CASE(size_t);
 206                 break;
 207         case PSI_T_POINTER:
 208                 {
 209                         PSI_TAS_P(char);
 210                         PSI_TAS_C(char_pointer);
 211                 }
 212                 break;
 213         EMPTY_SWITCH_DEFAULT_CASE();
 214         }
 215
 216         return align;
 217 }
 218
 219 size_t psi_t_size(token_t t)
 220 {
 221         size_t size;
 222
 223         switch (t) {
 224         case PSI_T_CHAR:
 225                 size = sizeof(char);
 226                 break;
 227         case PSI_T_SINT8:
 228         case PSI_T_UINT8:
 229                 size = 1;
 230                 break;
 231         case PSI_T_SHORT:
 232                 size = sizeof(short);
 233                 break;
 234         case PSI_T_SINT16:
 235         case PSI_T_UINT16:
 236                 size = 2;
 237                 break;
 238         case PSI_T_INT:
 239                 size = sizeof(int);
 240                 break;
 241         case PSI_T_SINT32:
 242         case PSI_T_UINT32:
 243                 size = 4;
 244                 break;
 245         case PSI_T_LONG:
 246                 size = sizeof(long);
 247                 break;
 248         case PSI_T_SINT64:
 249         case PSI_T_UINT64:
 250                 size = 8;
 251                 break;
 252         case PSI_T_FLOAT:
 253                 size = sizeof(float);
 254                 break;
 255         case PSI_T_DOUBLE:
 256                 size = sizeof(double);
 257                 break;
 258         case PSI_T_SIZE_T:
 259                 size = sizeof(size_t);
 260                 break;
 261         case PSI_T_POINTER:
 262                 size = sizeof(char *);
 263                 break;
 264         EMPTY_SWITCH_DEFAULT_CASE();
 265         }
 266         return size;
 267 }
 268
 269 size_t psi_t_align(token_t t, size_t s)
 270 {
 271         size_t a = psi_t_alignment(t);
 272         return ((s - 1) | (a - 1)) + 1;
 273 }
 274
 275 static impl_val *iterate(impl_val *val, token_t t, unsigned i, impl_val *tmp)
 276 {
 277         size_t size = psi_t_size(t);
 278
 279         memset(tmp, 0, sizeof(*tmp));
 280         memcpy(tmp, val->ptr + size * i, size);
 281         return tmp;
 282 }
 283
 284 void psi_to_array(impl_val *ret_val, decl_arg *func, zval *return_value)
 285 {
 286         zval ele;
 287         unsigned i;
 288         impl_val tmp;
 289         decl_type *type = real_decl_type(func->type);
 290         token_t t = type->type;
 291         printf("%d:%s real=%p,strct=%p\n", t, type->name, type->real, type->strct);
 292
 293         array_init(return_value);
 294
 295         if (t == PSI_T_STRUCT) {
 296                 decl_struct *s = type->strct;
 297                 ret_val = deref_impl_val(func->var->pointer_level, ret_val, func);
 298
 299                 for (i = 0; i < s->args->count; ++i) {
 300                         decl_arg *darg = s->args->args[i];
 301                         decl_struct_layout layout = s->layout[i];
 302                         impl_val tmp;
 303                         zval ztmp;
 304                         char *ptr = (char *) ret_val->ptr + layout.pos;
 305
 306                         memset(&tmp, 0, sizeof(tmp));
 307                         memcpy(&tmp, ptr, layout.len);
 308                         switch (real_decl_type(darg->type)->type) {
 309                         case PSI_T_INT:
 310                         case PSI_T_LONG:
 311                                 psi_to_int(&tmp, darg, &ztmp);
 312                                 break;
 313                         case PSI_T_CHAR:
 314                                 psi_to_string(&tmp, darg, &ztmp);
 315                                 break;
 316                         EMPTY_SWITCH_DEFAULT_CASE();
 317                         }
 318                         add_assoc_zval(return_value, darg->var->name, &ztmp);
 319                 }
 320                 return;
 321         }
 322         ret_val = deref_impl_val(0, ret_val, func);
 323         for (i = 0; i < func->var->array_size; ++i) {
 324                 impl_val *ptr = iterate(ret_val, t, i, &tmp);
 325
 326                 switch (t) {
 327                 case PSI_T_FLOAT:
 328                 case PSI_T_DOUBLE:
 329                         ZVAL_DOUBLE(&ele, ptr->dval);
 330                         break;
 331                 default:
 332                         ZVAL_LONG(&ele, ptr->lval);
 333                         break;
 334                 }
 335
 336                 add_next_index_zval(return_value, &ele);
 337         }
 338 }
 339
 340 ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
 341 {
 342         impl_arg *iarg;
 343
 344         if (!impl->func->args->count) {
 345                 return zend_parse_parameters_none();
 346         }
 347
 348         ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->count)
 349         nextarg:
 350                 iarg = impl->func->args->args[_i];
 351                 if (iarg->def) {
 352                         Z_PARAM_OPTIONAL;
 353                 }
 354                 if (PSI_T_BOOL == iarg->type->type) {
 355                         if (iarg->def) {
 356                                 iarg->val.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
 357                         }
 358                         Z_PARAM_BOOL(iarg->val.bval);
 359                 } else if (PSI_T_INT == iarg->type->type) {
 360                         if (iarg->def) {
 361                                 iarg->val.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
 362                         }
 363                         Z_PARAM_LONG(iarg->val.lval);
 364                 } else if (PSI_T_FLOAT == iarg->type->type) {
 365                         if (iarg->def) {
 366                                 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
 367                         }
 368                         Z_PARAM_DOUBLE(iarg->val.dval);
 369                 } else if (PSI_T_STRING == iarg->type->type) {
 370                         struct {char *val; size_t len;} str;
 371                         if (iarg->def) {
 372                                 /* FIXME */
 373                                 str.len = strlen(iarg->def->text) - 2;
 374                                 str.val = &iarg->def->text[1];
 375                         }
 376                         Z_PARAM_STR_EX(iarg->val.str, 1, 0);
 377                         if (iarg->val.str) {
 378                                 zend_string_addref(iarg->val.str);
 379                         } else if (iarg->def) {
 380                                 iarg->val.str = zend_string_init(str.val, str.len, 0);
 381                         }
 382                 } else if (PSI_T_ARRAY == iarg->type->type) {
 383                         /* handled as _zv in let or set */
 384                         Z_PARAM_PROLOGUE(0);
 385                 } else {
 386                         error_code = ZPP_ERROR_FAILURE;
 387                         break;
 388                 }
 389                 iarg->_zv = _arg;
 390                 if (_i < _max_num_args) {
 391                         goto nextarg;
 392                 }
 393         ZEND_PARSE_PARAMETERS_END_EX(return FAILURE);
 394
 395         return SUCCESS;
 396 }
 397
 398 impl_val *psi_do_let(decl_arg *darg)
 399 {
 400         impl_val *arg_val = &darg->let->out;
 401         impl_arg *iarg = darg->let->arg;
 402
 403         if (!iarg) {
 404                 /*
 405                  * let foo = NULL;
 406                  * let foo;
 407                  */
 408                 if (darg->var->array_size) {
 409                         arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val));
 410                         darg->let->mem = arg_val->ptr;
 411                 } else {
 412                         memset(arg_val, 0, sizeof(*arg_val));
 413                 }
 414                 return arg_val;
 415         }
 416         switch (darg->let->val->func->type) {
 417         case PSI_T_BOOLVAL:
 418                 if (iarg->type->type == PSI_T_BOOL) {
 419                         arg_val->cval = iarg->val.cval;
 420                 } else {
 421                         arg_val->cval = zend_is_true(iarg->_zv);
 422                 }
 423                 break;
 424         case PSI_T_INTVAL:
 425                 if (iarg->type->type == PSI_T_INT) {
 426                         arg_val->lval = iarg->val.lval;
 427                 } else {
 428                         arg_val->lval = zval_get_long(iarg->_zv);
 429                 }
 430                 break;
 431         case PSI_T_STRVAL:
 432                 if (iarg->type->type == PSI_T_STRING) {
 433                         arg_val->ptr = estrdup(iarg->val.str->val);
 434                         darg->let->mem = arg_val->ptr;
 435                         zend_string_release(iarg->val.str);
 436                 } else {
 437                         zend_string *zs = zval_get_string(iarg->_zv);
 438                         arg_val->ptr = estrdup(zs->val);
 439                         darg->let->mem = arg_val->ptr;
 440                         zend_string_release(zs);
 441                 }
 442                 break;
 443         case PSI_T_STRLEN:
 444                 if (iarg->type->type == PSI_T_STRING) {
 445                         arg_val->lval = iarg->val.str->len;
 446                         zend_string_release(iarg->val.str);
 447                 } else {
 448                         zend_string *zs = zval_get_string(iarg->_zv);
 449                         arg_val->lval = zs->len;
 450                         zend_string_release(zs);
 451                 }
 452                 break;
 453         case PSI_T_CALLOC:
 454                 arg_val->ptr = calloc(1, darg->let->val->func->size);
 455                 darg->let->mem = arg_val->ptr;
 456                 break;
 457         EMPTY_SWITCH_DEFAULT_CASE();
 458         }
 459
 460         return arg_val;
 461 }
 462
 463 void psi_do_set(zval *return_value, set_func *func, decl_vars *vars)
 464 {
 465         impl_val *val = (impl_val *) &vars->vars[0]->arg->let->ptr;
 466
 467         ZVAL_DEREF(return_value);
 468         zval_dtor(return_value);
 469
 470         switch (func->type) {
 471         case PSI_T_TO_STRING:
 472                 psi_to_string(val, vars->vars[0]->arg, return_value);
 473                 break;
 474         case PSI_T_TO_ARRAY:
 475                 psi_to_array(val, vars->vars[0]->arg, return_value);
 476                 break;
 477         EMPTY_SWITCH_DEFAULT_CASE();
 478         }
 479 }
 480
 481 void psi_do_return(impl *impl, impl_val *ret_val, zval *return_value)
 482 {
 483         switch (impl->stmts->ret.list[0]->func->type) {
 484         case PSI_T_TO_STRING:
 485                 psi_to_string(ret_val, impl->decl->func, return_value);
 486                 break;
 487         case PSI_T_TO_INT:
 488                 psi_to_int(ret_val, impl->decl->func, return_value);
 489                 break;
 490         case PSI_T_TO_ARRAY:
 491                 psi_to_array(ret_val, impl->decl->func, return_value);
 492                 break;
 493         EMPTY_SWITCH_DEFAULT_CASE();
 494         }
 495 }
 496
 497 void psi_do_free(free_stmt *fre)
 498 {
 499         size_t i;
 500
 501         for (i = 0; i < fre->vars->count; ++i) {
 502                 decl_var *dvar = fre->vars->vars[i];
 503
 504                 if (dvar->arg && dvar->arg->let->out.ptr) {
 505                         free(dvar->arg->let->out.ptr);
 506                         dvar->arg->let->out.ptr = NULL;
 507                 }
 508         }
 509 }
 510
 511 void psi_do_clean(impl *impl)
 512 {
 513         size_t i;
 514
 515         for (i = 0; i < impl->func->args->count; ++i ) {
 516                 impl_arg *iarg = impl->func->args->args[i];
 517
 518                 switch (iarg->type->type) {
 519                 case PSI_T_STRING:
 520                         if (iarg->val.str) {
 521                                 zend_string_release(iarg->val.str);
 522                         }
 523                         break;
 524                 }
 525         }
 526
 527         for (i = 0; i < impl->decl->args->count; ++i) {
 528                 decl_arg *darg = impl->decl->args->args[i];
 529
 530                 if (darg->let && darg->let->mem) {
 531                         efree(darg->let->mem);
 532                         darg->let->mem = NULL;
 533                 }
 534         }
 535 }
 536
 537 PHP_MINIT_FUNCTION(psi)
 538 {
 539         PSI_ContextOps *ops;
 540
 541         REGISTER_INI_ENTRIES();
 542
 543         if (!strcasecmp(PSI_G(engine), "jit")) {
 544                 ops = PSI_Libjit();
 545         } else {
 546                 ops = PSI_Libffi();
 547         }
 548
 549         PSI_ContextInit(&PSI_G(context), ops, psi_error);
 550         PSI_ContextBuild(&PSI_G(context), PSI_G(directory));
 551
 552         return SUCCESS;
 553 }
 554 PHP_MSHUTDOWN_FUNCTION(psi)
 555 {
 556         PSI_ContextDtor(&PSI_G(context));
 557
 558         UNREGISTER_INI_ENTRIES();
 559
 560         return SUCCESS;
 561 }
 562
 563 /* Remove if there's nothing to do at request start */
 564 /* {{{ PHP_RINIT_FUNCTION
 565  */
 566 PHP_RINIT_FUNCTION(psi)
 567 {
 568 #if defined(COMPILE_DL_PSI) && defined(ZTS)
 569         ZEND_TSRMLS_CACHE_UPDATE();
 570 #endif
 571         return SUCCESS;
 572 }
 573 /* }}} */
 574
 575 /* Remove if there's nothing to do at request end */
 576 /* {{{ PHP_RSHUTDOWN_FUNCTION
 577  */
 578 PHP_RSHUTDOWN_FUNCTION(psi)
 579 {
 580         return SUCCESS;
 581 }
 582 /* }}} */
 583
 584 PHP_MINFO_FUNCTION(psi)
 585 {
 586         php_info_print_table_start();
 587         php_info_print_table_header(2, "psi support", "enabled");
 588         php_info_print_table_end();
 589
 590         DISPLAY_INI_ENTRIES();
 591 }
 592 const zend_function_entry psi_functions[] = {
 593         PHP_FE_END
 594 };
 595
 596 zend_module_entry psi_module_entry = {
 597         STANDARD_MODULE_HEADER,
 598         "psi",
 599         psi_functions,
 600         PHP_MINIT(psi),
 601         PHP_MSHUTDOWN(psi),
 602         PHP_RINIT(psi),         /* Replace with NULL if there's nothing to do at request start */
 603         PHP_RSHUTDOWN(psi),     /* Replace with NULL if there's nothing to do at request end */
 604         PHP_MINFO(psi),
 605         PHP_PSI_VERSION,
 606         STANDARD_MODULE_PROPERTIES
 607 };
 608
 609 #ifdef COMPILE_DL_PSI
 610 #ifdef ZTS
 611 ZEND_TSRMLS_CACHE_DEFINE();
 612 #endif
 613 ZEND_GET_MODULE(psi)
 614 #endif
 615
 616 /*
 617  * Local variables:
 618  * tab-width: 4
 619  * c-basic-offset: 4
 620  * End:
 621  * vim600: noet sw=4 ts=4 fdm=marker
 622  * vim<600: noet sw=4 ts=4
 623  */