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         token_t t = real_decl_type(func->type)->type;
 290
 291         array_init(return_value);
 292         ret_val = deref_impl_val(0, ret_val, func);
 293         for (i = 0; i < func->var->array_size; ++i) {
 294                 impl_val *ptr = iterate(ret_val, t, i, &tmp);
 295
 296                 switch (t) {
 297                 case PSI_T_FLOAT:
 298                 case PSI_T_DOUBLE:
 299                         ZVAL_DOUBLE(&ele, ptr->dval);
 300                         break;
 301                 default:
 302                         ZVAL_LONG(&ele, ptr->lval);
 303                         break;
 304                 }
 305
 306                 add_next_index_zval(return_value, &ele);
 307         }
 308 }
 309
 310 ZEND_RESULT_CODE psi_parse_args(zend_execute_data *execute_data, impl *impl)
 311 {
 312         impl_arg *iarg;
 313
 314         if (!impl->func->args->count) {
 315                 return zend_parse_parameters_none();
 316         }
 317
 318         ZEND_PARSE_PARAMETERS_START(psi_num_min_args(impl), impl->func->args->count)
 319         nextarg:
 320                 iarg = impl->func->args->args[_i];
 321                 if (iarg->def) {
 322                         Z_PARAM_OPTIONAL;
 323                 }
 324                 if (PSI_T_BOOL == iarg->type->type) {
 325                         if (iarg->def) {
 326                                 iarg->val.bval = iarg->def->type == PSI_T_TRUE ? 1 : 0;
 327                         }
 328                         Z_PARAM_BOOL(iarg->val.bval);
 329                 } else if (PSI_T_INT == iarg->type->type) {
 330                         if (iarg->def) {
 331                                 iarg->val.lval = zend_atol(iarg->def->text, strlen(iarg->def->text));
 332                         }
 333                         Z_PARAM_LONG(iarg->val.lval);
 334                 } else if (PSI_T_FLOAT == iarg->type->type) {
 335                         if (iarg->def) {
 336                                 iarg->val.dval = zend_strtod(iarg->def->text, NULL);
 337                         }
 338                         Z_PARAM_DOUBLE(iarg->val.dval);
 339                 } else if (PSI_T_STRING == iarg->type->type) {
 340                         struct {char *val; size_t len;} str;
 341                         if (iarg->def) {
 342                                 /* FIXME */
 343                                 str.len = strlen(iarg->def->text) - 2;
 344                                 str.val = &iarg->def->text[1];
 345                         }
 346                         Z_PARAM_STR_EX(iarg->val.str, 1, 0);
 347                         if (iarg->val.str) {
 348                                 zend_string_addref(iarg->val.str);
 349                         } else if (iarg->def) {
 350                                 iarg->val.str = zend_string_init(str.val, str.len, 0);
 351                         }
 352                 } else if (PSI_T_ARRAY == iarg->type->type) {
 353                         /* handled as _zv in let or set */
 354                         Z_PARAM_PROLOGUE(0);
 355                 } else {
 356                         error_code = ZPP_ERROR_FAILURE;
 357                         break;
 358                 }
 359                 iarg->_zv = _arg;
 360                 if (_i < _max_num_args) {
 361                         goto nextarg;
 362                 }
 363         ZEND_PARSE_PARAMETERS_END_EX(return FAILURE);
 364
 365         return SUCCESS;
 366 }
 367
 368 impl_val *psi_do_let(decl_arg *darg)
 369 {
 370         impl_val *arg_val = &darg->let->out;
 371         impl_arg *iarg = darg->let->arg;
 372
 373         if (!iarg) {
 374                 /*
 375                  * let foo = NULL;
 376                  * let foo;
 377                  */
 378                 if (darg->var->array_size) {
 379                         arg_val->ptr = ecalloc(darg->var->array_size, sizeof(*arg_val));
 380                         darg->let->mem = arg_val->ptr;
 381                 } else {
 382                         memset(arg_val, 0, sizeof(*arg_val));
 383                 }
 384                 return arg_val;
 385         }
 386         switch (darg->let->val->func->type) {
 387         case PSI_T_BOOLVAL:
 388                 if (iarg->type->type == PSI_T_BOOL) {
 389                         arg_val->cval = iarg->val.cval;
 390                 } else {
 391                         arg_val->cval = zend_is_true(iarg->_zv);
 392                 }
 393                 break;
 394         case PSI_T_INTVAL:
 395                 if (iarg->type->type == PSI_T_INT) {
 396                         arg_val->lval = iarg->val.lval;
 397                 } else {
 398                         arg_val->lval = zval_get_long(iarg->_zv);
 399                 }
 400                 break;
 401         case PSI_T_STRVAL:
 402                 if (iarg->type->type == PSI_T_STRING) {
 403                         arg_val->ptr = estrdup(iarg->val.str->val);
 404                         darg->let->mem = arg_val->ptr;
 405                         zend_string_release(iarg->val.str);
 406                 } else {
 407                         zend_string *zs = zval_get_string(iarg->_zv);
 408                         arg_val->ptr = estrdup(zs->val);
 409                         darg->let->mem = arg_val->ptr;
 410                         zend_string_release(zs);
 411                 }
 412                 break;
 413         case PSI_T_STRLEN:
 414                 if (iarg->type->type == PSI_T_STRING) {
 415                         arg_val->lval = iarg->val.str->len;
 416                         zend_string_release(iarg->val.str);
 417                 } else {
 418                         zend_string *zs = zval_get_string(iarg->_zv);
 419                         arg_val->lval = zs->len;
 420                         zend_string_release(zs);
 421                 }
 422                 break;
 423         case PSI_T_CALLOC:
 424                 arg_val->ptr = calloc(1, darg->let->val->func->size);
 425                 darg->let->mem = arg_val->ptr;
 426                 break;
 427         EMPTY_SWITCH_DEFAULT_CASE();
 428         }
 429
 430         return arg_val;
 431 }
 432
 433 void psi_do_set(zval *return_value, set_func *func, decl_vars *vars)
 434 {
 435         impl_val *val = (impl_val *) &vars->vars[0]->arg->let->ptr;
 436
 437         ZVAL_DEREF(return_value);
 438         zval_dtor(return_value);
 439
 440         switch (func->type) {
 441         case PSI_T_TO_STRING:
 442                 psi_to_string(val, vars->vars[0]->arg, return_value);
 443                 break;
 444         case PSI_T_TO_ARRAY:
 445                 psi_to_array(val, vars->vars[0]->arg, return_value);
 446                 break;
 447         EMPTY_SWITCH_DEFAULT_CASE();
 448         }
 449 }
 450
 451 void psi_do_return(impl *impl, impl_val *ret_val, zval *return_value)
 452 {
 453         switch (impl->stmts->ret.list[0]->func->type) {
 454         case PSI_T_TO_STRING:
 455                 psi_to_string(ret_val, impl->decl->func, return_value);
 456                 break;
 457         case PSI_T_TO_INT:
 458                 psi_to_int(ret_val, impl->decl->func, return_value);
 459                 break;
 460         EMPTY_SWITCH_DEFAULT_CASE();
 461         }
 462 }
 463
 464 void psi_do_free(free_stmt *fre)
 465 {
 466         size_t i;
 467
 468         for (i = 0; i < fre->vars->count; ++i) {
 469                 decl_var *dvar = fre->vars->vars[i];
 470
 471                 if (dvar->arg && dvar->arg->let->out.ptr) {
 472                         free(dvar->arg->let->out.ptr);
 473                         dvar->arg->let->out.ptr = NULL;
 474                 }
 475         }
 476 }
 477
 478 void psi_do_clean(impl *impl)
 479 {
 480         size_t i;
 481
 482         for (i = 0; i < impl->func->args->count; ++i ) {
 483                 impl_arg *iarg = impl->func->args->args[i];
 484
 485                 switch (iarg->type->type) {
 486                 case PSI_T_STRING:
 487                         if (iarg->val.str) {
 488                                 zend_string_release(iarg->val.str);
 489                         }
 490                         break;
 491                 }
 492         }
 493
 494         for (i = 0; i < impl->decl->args->count; ++i) {
 495                 decl_arg *darg = impl->decl->args->args[i];
 496
 497                 if (darg->let && darg->let->mem) {
 498                         efree(darg->let->mem);
 499                         darg->let->mem = NULL;
 500                 }
 501         }
 502 }
 503
 504 PHP_MINIT_FUNCTION(psi)
 505 {
 506         PSI_ContextOps *ops;
 507
 508         REGISTER_INI_ENTRIES();
 509
 510         if (!strcasecmp(PSI_G(engine), "jit")) {
 511                 ops = PSI_Libjit();
 512         } else {
 513                 ops = PSI_Libffi();
 514         }
 515
 516         PSI_ContextInit(&PSI_G(context), ops, psi_error);
 517         PSI_ContextBuild(&PSI_G(context), PSI_G(directory));
 518
 519         return SUCCESS;
 520 }
 521 PHP_MSHUTDOWN_FUNCTION(psi)
 522 {
 523         PSI_ContextDtor(&PSI_G(context));
 524
 525         UNREGISTER_INI_ENTRIES();
 526
 527         return SUCCESS;
 528 }
 529
 530 /* Remove if there's nothing to do at request start */
 531 /* {{{ PHP_RINIT_FUNCTION
 532  */
 533 PHP_RINIT_FUNCTION(psi)
 534 {
 535 #if defined(COMPILE_DL_PSI) && defined(ZTS)
 536         ZEND_TSRMLS_CACHE_UPDATE();
 537 #endif
 538         return SUCCESS;
 539 }
 540 /* }}} */
 541
 542 /* Remove if there's nothing to do at request end */
 543 /* {{{ PHP_RSHUTDOWN_FUNCTION
 544  */
 545 PHP_RSHUTDOWN_FUNCTION(psi)
 546 {
 547         return SUCCESS;
 548 }
 549 /* }}} */
 550
 551 PHP_MINFO_FUNCTION(psi)
 552 {
 553         php_info_print_table_start();
 554         php_info_print_table_header(2, "psi support", "enabled");
 555         php_info_print_table_end();
 556
 557         DISPLAY_INI_ENTRIES();
 558 }
 559 const zend_function_entry psi_functions[] = {
 560         PHP_FE_END
 561 };
 562
 563 zend_module_entry psi_module_entry = {
 564         STANDARD_MODULE_HEADER,
 565         "psi",
 566         psi_functions,
 567         PHP_MINIT(psi),
 568         PHP_MSHUTDOWN(psi),
 569         PHP_RINIT(psi),         /* Replace with NULL if there's nothing to do at request start */
 570         PHP_RSHUTDOWN(psi),     /* Replace with NULL if there's nothing to do at request end */
 571         PHP_MINFO(psi),
 572         PHP_PSI_VERSION,
 573         STANDARD_MODULE_PROPERTIES
 574 };
 575
 576 #ifdef COMPILE_DL_PSI
 577 #ifdef ZTS
 578 ZEND_TSRMLS_CACHE_DEFINE();
 579 #endif
 580 ZEND_GET_MODULE(psi)
 581 #endif
 582
 583 /*
 584  * Local variables:
 585  * tab-width: 4
 586  * c-basic-offset: 4
 587  * End:
 588  * vim600: noet sw=4 ts=4 fdm=marker
 589  * vim<600: noet sw=4 ts=4
 590  */