dlls/jscript/function.c

   1 /*
   2  * Copyright 2008 Jacek Caban for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <assert.h>
  20
  21 #include "jscript.h"
  22 #include "engine.h"
  23
  24 #include "wine/debug.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  27
  28 typedef struct _function_vtbl_t function_vtbl_t;
  29
  30 typedef struct {
  31     jsdisp_t dispex;
  32     const function_vtbl_t *vtbl;
  33     DWORD flags;
  34     DWORD length;
  35 } FunctionInstance;
  36
  37 struct _function_vtbl_t {
  38     HRESULT (*call)(script_ctx_t*,FunctionInstance*,jsval_t,unsigned,unsigned,jsval_t*,jsval_t*);
  39     HRESULT (*toString)(FunctionInstance*,jsstr_t**);
  40     function_code_t* (*get_code)(FunctionInstance*);
  41     void (*destructor)(FunctionInstance*);
  42     HRESULT (*gc_traverse)(struct gc_ctx*,enum gc_traverse_op,FunctionInstance*);
  43 };
  44
  45 typedef struct {
  46     FunctionInstance function;
  47     scope_chain_t *scope_chain;
  48     bytecode_t *code;
  49     function_code_t *func_code;
  50 } InterpretedFunction;
  51
  52 typedef struct {
  53     FunctionInstance function;
  54     builtin_invoke_t proc;
  55     const WCHAR *name;
  56 } NativeFunction;
  57
  58 typedef struct {
  59     FunctionInstance function;
  60     FunctionInstance *target;
  61     jsval_t this;
  62     unsigned argc;
  63     jsval_t args[1];
  64 } BindFunction;
  65
  66 typedef struct {
  67     jsdisp_t jsdisp;
  68     InterpretedFunction *function;
  69     jsval_t *buf;
  70     call_frame_t *frame;
  71     unsigned argc;
  72 } ArgumentsInstance;
  73
  74 static HRESULT create_bind_function(script_ctx_t*,FunctionInstance*,jsval_t,unsigned,jsval_t*,jsdisp_t**r);
  75
  76 static HRESULT no_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, FunctionInstance *function)
  77 {
  78     return S_OK;
  79 }
  80
  81 static inline FunctionInstance *function_from_jsdisp(jsdisp_t *jsdisp)
  82 {
  83     return CONTAINING_RECORD(jsdisp, FunctionInstance, dispex);
  84 }
  85
  86 static inline FunctionInstance *function_this(jsval_t vthis)
  87 {
  88     jsdisp_t *jsdisp = is_object_instance(vthis) ? to_jsdisp(get_object(vthis)) : NULL;
  89     return (jsdisp && is_class(jsdisp, JSCLASS_FUNCTION)) ? function_from_jsdisp(jsdisp) : NULL;
  90 }
  91
  92 static inline ArgumentsInstance *arguments_from_jsdisp(jsdisp_t *jsdisp)
  93 {
  94     return CONTAINING_RECORD(jsdisp, ArgumentsInstance, jsdisp);
  95 }
  96
  97 static HRESULT Arguments_value(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
  98         jsval_t *r)
  99 {
 100     FIXME("\n");
 101     return E_NOTIMPL;
 102 }
 103
 104 static void Arguments_destructor(jsdisp_t *jsdisp)
 105 {
 106     ArgumentsInstance *arguments = arguments_from_jsdisp(jsdisp);
 107
 108     TRACE("(%p)\n", arguments);
 109
 110     if(arguments->buf) {
 111         unsigned i;
 112         for(i = 0; i < arguments->argc; i++)
 113             jsval_release(arguments->buf[i]);
 114         free(arguments->buf);
 115     }
 116
 117     if(arguments->function)
 118         jsdisp_release(&arguments->function->function.dispex);
 119     free(arguments);
 120 }
 121
 122 static unsigned Arguments_idx_length(jsdisp_t *jsdisp)
 123 {
 124     ArgumentsInstance *arguments = arguments_from_jsdisp(jsdisp);
 125     return arguments->argc;
 126 }
 127
 128 static jsval_t *get_argument_ref(ArgumentsInstance *arguments, unsigned idx)
 129 {
 130     if(arguments->buf)
 131         return arguments->buf + idx;
 132     if(arguments->frame->base_scope->frame || idx >= arguments->frame->function->param_cnt)
 133         return arguments->jsdisp.ctx->stack + arguments->frame->arguments_off + idx;
 134     return NULL;
 135 }
 136
 137 static HRESULT Arguments_idx_get(jsdisp_t *jsdisp, unsigned idx, jsval_t *r)
 138 {
 139     ArgumentsInstance *arguments = arguments_from_jsdisp(jsdisp);
 140     jsval_t *ref;
 141
 142     TRACE("%p[%u]\n", arguments, idx);
 143
 144     if((ref = get_argument_ref(arguments, idx)))
 145         return jsval_copy(*ref, r);
 146
 147     /* FIXME: Accessing by name won't work for duplicated argument names */
 148     return jsdisp_propget_name(arguments->frame->base_scope->jsobj,
 149                                arguments->function->func_code->params[idx], r);
 150 }
 151
 152 static HRESULT Arguments_idx_put(jsdisp_t *jsdisp, unsigned idx, jsval_t val)
 153 {
 154     ArgumentsInstance *arguments = arguments_from_jsdisp(jsdisp);
 155     jsval_t *ref;
 156     HRESULT hres;
 157
 158     TRACE("%p[%u] = %s\n", arguments, idx, debugstr_jsval(val));
 159
 160     if((ref = get_argument_ref(arguments, idx))) {
 161         jsval_t copy;
 162         hres = jsval_copy(val, &copy);
 163         if(FAILED(hres))
 164             return hres;
 165
 166         jsval_release(*ref);
 167         *ref = copy;
 168         return S_OK;
 169     }
 170
 171     /* FIXME: Accessing by name won't work for duplicated argument names */
 172     return jsdisp_propput_name(arguments->frame->base_scope->jsobj,
 173                                arguments->function->func_code->params[idx], val);
 174 }
 175
 176 static HRESULT Arguments_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, jsdisp_t *jsdisp)
 177 {
 178     ArgumentsInstance *arguments = arguments_from_jsdisp(jsdisp);
 179     HRESULT hres;
 180     unsigned i;
 181
 182     if(arguments->buf) {
 183         for(i = 0; i < arguments->argc; i++) {
 184             hres = gc_process_linked_val(gc_ctx, op, jsdisp, &arguments->buf[i]);
 185             if(FAILED(hres))
 186                 return hres;
 187         }
 188     }
 189
 190     return gc_process_linked_obj(gc_ctx, op, jsdisp, &arguments->function->function.dispex, (void**)&arguments->function);
 191 }
 192
 193 static const builtin_info_t Arguments_info = {
 194     JSCLASS_ARGUMENTS,
 195     Arguments_value,
 196     0, NULL,
 197     Arguments_destructor,
 198     NULL,
 199     Arguments_idx_length,
 200     Arguments_idx_get,
 201     Arguments_idx_put,
 202     Arguments_gc_traverse
 203 };
 204
 205 HRESULT setup_arguments_object(script_ctx_t *ctx, call_frame_t *frame)
 206 {
 207     ArgumentsInstance *args;
 208     HRESULT hres;
 209
 210     args = calloc(1, sizeof(*args));
 211     if(!args)
 212         return E_OUTOFMEMORY;
 213
 214     hres = init_dispex_from_constr(&args->jsdisp, ctx, &Arguments_info, ctx->object_constr);
 215     if(FAILED(hres)) {
 216         free(args);
 217         return hres;
 218     }
 219
 220     args->function = (InterpretedFunction*)function_from_jsdisp(jsdisp_addref(frame->function_instance));
 221     args->argc = frame->argc;
 222     args->frame = frame;
 223
 224     hres = jsdisp_define_data_property(&args->jsdisp, L"length", PROPF_WRITABLE | PROPF_CONFIGURABLE,
 225                                        jsval_number(args->argc));
 226     if(SUCCEEDED(hres))
 227         hres = jsdisp_define_data_property(&args->jsdisp, L"callee", PROPF_WRITABLE | PROPF_CONFIGURABLE,
 228                                            jsval_obj(&args->function->function.dispex));
 229     if(SUCCEEDED(hres))
 230         hres = jsdisp_propput(frame->base_scope->jsobj, L"arguments", PROPF_WRITABLE, TRUE, jsval_obj(&args->jsdisp));
 231     if(FAILED(hres)) {
 232         jsdisp_release(&args->jsdisp);
 233         return hres;
 234     }
 235
 236     frame->arguments_obj = &args->jsdisp;
 237     return S_OK;
 238 }
 239
 240 void detach_arguments_object(jsdisp_t *args_disp)
 241 {
 242     ArgumentsInstance *arguments = arguments_from_jsdisp(args_disp);
 243     call_frame_t *frame = arguments->frame;
 244     const BOOL on_stack = frame->base_scope->frame == frame;
 245     HRESULT hres;
 246
 247     /* Reset arguments value to cut the reference cycle. Note that since all activation contexts have
 248      * their own arguments property, it's impossible to use prototype's one during name lookup */
 249     jsdisp_propput_name(frame->base_scope->jsobj, L"arguments", jsval_undefined());
 250     arguments->frame = NULL;
 251
 252     /* Don't bother coppying arguments if call frame holds the last reference. */
 253     if(arguments->jsdisp.ref > 1) {
 254         arguments->buf = malloc(arguments->argc * sizeof(*arguments->buf));
 255         if(arguments->buf) {
 256             int i;
 257
 258             for(i = 0; i < arguments->argc ; i++) {
 259                 if(on_stack || i >= frame->function->param_cnt)
 260                     hres = jsval_copy(arguments->jsdisp.ctx->stack[frame->arguments_off + i], arguments->buf+i);
 261                 else
 262                     hres = jsdisp_propget_name(frame->base_scope->jsobj, frame->function->params[i], arguments->buf+i);
 263                 if(FAILED(hres))
 264                     arguments->buf[i] = jsval_undefined();
 265             }
 266         }else {
 267             ERR("out of memory\n");
 268             arguments->argc = 0;
 269         }
 270     }
 271
 272     jsdisp_release(frame->arguments_obj);
 273 }
 274
 275 HRESULT Function_invoke(jsdisp_t *func_this, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
 276 {
 277     FunctionInstance *function;
 278
 279     TRACE("func %p this %s\n", func_this, debugstr_jsval(vthis));
 280
 281     assert(is_class(func_this, JSCLASS_FUNCTION));
 282     function = function_from_jsdisp(func_this);
 283
 284     if(function->dispex.ctx->state == SCRIPTSTATE_UNINITIALIZED || function->dispex.ctx->state == SCRIPTSTATE_CLOSED) {
 285         WARN("Script engine state does not allow running code.\n");
 286         return E_UNEXPECTED;
 287     }
 288
 289     return function->vtbl->call(function->dispex.ctx, function, vthis, flags, argc, argv, r);
 290 }
 291
 292 static HRESULT Function_get_length(script_ctx_t *ctx, jsdisp_t *jsthis, jsval_t *r)
 293 {
 294     TRACE("%p\n", jsthis);
 295
 296     *r = jsval_number(function_from_jsdisp(jsthis)->length);
 297     return S_OK;
 298 }
 299
 300 static HRESULT Function_toString(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
 301         jsval_t *r)
 302 {
 303     FunctionInstance *function;
 304     jsstr_t *str;
 305     HRESULT hres;
 306
 307     TRACE("\n");
 308
 309     if(!(function = function_this(vthis)))
 310         return JS_E_FUNCTION_EXPECTED;
 311
 312     hres = function->vtbl->toString(function, &str);
 313     if(FAILED(hres))
 314         return hres;
 315
 316     if(r)
 317         *r = jsval_string(str);
 318     else
 319         jsstr_release(str);
 320     return S_OK;
 321 }
 322
 323 static HRESULT array_to_args(script_ctx_t *ctx, jsdisp_t *arg_array, unsigned *argc, jsval_t **ret)
 324 {
 325     jsval_t *argv, val;
 326     UINT32 length, i;
 327     HRESULT hres;
 328
 329     hres = jsdisp_propget_name(arg_array, L"length", &val);
 330     if(FAILED(hres))
 331         return hres;
 332
 333     hres = to_uint32(ctx, val, &length);
 334     jsval_release(val);
 335     if(FAILED(hres))
 336         return hres;
 337
 338     argv = malloc(length * sizeof(*argv));
 339     if(!argv)
 340         return E_OUTOFMEMORY;
 341
 342     for(i=0; i<length; i++) {
 343         hres = jsdisp_get_idx(arg_array, i, argv+i);
 344         if(hres == DISP_E_UNKNOWNNAME) {
 345             argv[i] = jsval_undefined();
 346         }else if(FAILED(hres)) {
 347             while(i--)
 348                 jsval_release(argv[i]);
 349             free(argv);
 350             return hres;
 351         }
 352     }
 353
 354     *argc = length;
 355     *ret = argv;
 356     return S_OK;
 357 }
 358
 359 static HRESULT Function_apply(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv, jsval_t *r)
 360 {
 361     jsval_t this_val = jsval_undefined();
 362     FunctionInstance *function;
 363     jsval_t *args = NULL;
 364     unsigned i, cnt = 0;
 365     HRESULT hres = S_OK;
 366
 367     TRACE("\n");
 368
 369     if(is_null_disp(vthis))
 370         return JS_E_OBJECT_REQUIRED;
 371     if(!is_object_instance(vthis) || (!(function = function_this(vthis)) && to_jsdisp(get_object(vthis))))
 372         return JS_E_FUNCTION_EXPECTED;
 373
 374     if(argc) {
 375         if(ctx->version < SCRIPTLANGUAGEVERSION_ES5 && !is_undefined(argv[0]) && !is_null(argv[0])) {
 376             IDispatch *this_obj;
 377             hres = to_object(ctx, argv[0], &this_obj);
 378             if(FAILED(hres))
 379                 return hres;
 380             this_val = jsval_disp(this_obj);
 381         }else {
 382             hres = jsval_copy(argv[0], &this_val);
 383             if(FAILED(hres))
 384                 return hres;
 385         }
 386     }
 387
 388     if(argc >= 2) {
 389         jsdisp_t *arg_array = NULL;
 390
 391         if(is_object_instance(argv[1])) {
 392             arg_array = iface_to_jsdisp(get_object(argv[1]));
 393             if(arg_array &&
 394                (!is_class(arg_array, JSCLASS_ARRAY) && !is_class(arg_array, JSCLASS_ARGUMENTS) )) {
 395                 jsdisp_release(arg_array);
 396                 arg_array = NULL;
 397             }
 398         }
 399
 400         if(arg_array) {
 401             hres = array_to_args(ctx, arg_array, &cnt, &args);
 402             jsdisp_release(arg_array);
 403         }else {
 404             FIXME("throw TypeError\n");
 405             hres = E_FAIL;
 406         }
 407     }
 408
 409     if(SUCCEEDED(hres)) {
 410         if(function) {
 411             hres = function->vtbl->call(ctx, function, this_val, flags, cnt, args, r);
 412         }else {
 413             jsval_t res;
 414             hres = disp_call_value(ctx, get_object(vthis), this_val, DISPATCH_METHOD, cnt, args, &res);
 415             if(SUCCEEDED(hres)) {
 416                 if(r)
 417                     *r = res;
 418                 else
 419                     jsval_release(res);
 420             }
 421         }
 422     }
 423
 424     jsval_release(this_val);
 425     for(i=0; i < cnt; i++)
 426         jsval_release(args[i]);
 427     free(args);
 428     return hres;
 429 }
 430
 431 static HRESULT Function_call(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
 432         jsval_t *r)
 433 {
 434     jsval_t this_val = jsval_undefined();
 435     FunctionInstance *function;
 436     unsigned cnt = 0;
 437     HRESULT hres;
 438
 439     TRACE("\n");
 440
 441     if(is_null_disp(vthis))
 442         return JS_E_OBJECT_REQUIRED;
 443     if(!(function = function_this(vthis)))
 444         return JS_E_FUNCTION_EXPECTED;
 445
 446     if(argc) {
 447         if(ctx->version < SCRIPTLANGUAGEVERSION_ES5 && !is_undefined(argv[0]) && !is_null(argv[0])) {
 448             IDispatch *this_obj;
 449             hres = to_object(ctx, argv[0], &this_obj);
 450             if(FAILED(hres))
 451                 return hres;
 452             this_val = jsval_disp(this_obj);
 453         }else {
 454             hres = jsval_copy(argv[0], &this_val);
 455             if(FAILED(hres))
 456                 return hres;
 457         }
 458         cnt = argc-1;
 459     }
 460
 461     hres = function->vtbl->call(ctx, function, this_val, flags, cnt, argv + 1, r);
 462
 463     jsval_release(this_val);
 464     return hres;
 465 }
 466
 467 static HRESULT Function_bind(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
 468         jsval_t *r)
 469 {
 470     jsval_t bound_this = jsval_undefined();
 471     FunctionInstance *function;
 472     jsdisp_t *new_function;
 473     HRESULT hres;
 474
 475     TRACE("\n");
 476
 477     if(!(function = function_this(vthis)))
 478         return JS_E_FUNCTION_EXPECTED;
 479
 480     if(argc < 1) {
 481         argc = 1;
 482     }else if(is_null(argv[0])) {
 483         bound_this = argv[0];
 484     }else if(!is_undefined(argv[0])) {
 485         IDispatch *obj;
 486         hres = to_object(ctx, argv[0], &obj);
 487         if(FAILED(hres))
 488             return hres;
 489         bound_this = jsval_disp(obj);
 490     }
 491
 492     hres = create_bind_function(ctx, function, bound_this, argc - 1, argv + 1, &new_function);
 493     jsval_release(bound_this);
 494     if(FAILED(hres))
 495         return hres;
 496
 497     if(r)
 498         *r = jsval_obj(new_function);
 499     else
 500         jsdisp_release(new_function);
 501     return S_OK;
 502 }
 503
 504 HRESULT Function_value(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
 505         jsval_t *r)
 506 {
 507     FunctionInstance *function;
 508
 509     TRACE("\n");
 510
 511     if(!(function = function_this(vthis))) {
 512         ERR("dispex is not a function\n");
 513         return E_FAIL;
 514     }
 515
 516     return function->vtbl->call(ctx, function, vthis, flags, argc, argv, r);
 517 }
 518
 519 HRESULT Function_get_value(script_ctx_t *ctx, jsdisp_t *jsthis, jsval_t *r)
 520 {
 521     FunctionInstance *function = function_from_jsdisp(jsthis);
 522     jsstr_t *str;
 523     HRESULT hres;
 524
 525     TRACE("\n");
 526
 527     hres = function->vtbl->toString(function, &str);
 528     if(FAILED(hres))
 529         return hres;
 530
 531     *r = jsval_string(str);
 532     return S_OK;
 533 }
 534
 535 static HRESULT Function_get_arguments(script_ctx_t *ctx, jsdisp_t *jsthis, jsval_t *r)
 536 {
 537     FunctionInstance *function = function_from_jsdisp(jsthis);
 538     call_frame_t *frame;
 539     HRESULT hres;
 540
 541     TRACE("\n");
 542
 543     for(frame = ctx->call_ctx; frame; frame = frame->prev_frame) {
 544         if(frame->function_instance == &function->dispex) {
 545             if(!frame->arguments_obj) {
 546                 hres = setup_arguments_object(ctx, frame);
 547                 if(FAILED(hres))
 548                     return hres;
 549             }
 550             *r = jsval_obj(jsdisp_addref(frame->arguments_obj));
 551             return S_OK;
 552         }
 553     }
 554
 555     *r = jsval_null();
 556     return S_OK;
 557 }
 558
 559 function_code_t *Function_get_code(jsdisp_t *jsthis)
 560 {
 561     FunctionInstance *function;
 562
 563     assert(is_class(jsthis, JSCLASS_FUNCTION));
 564     function = function_from_jsdisp(jsthis);
 565
 566     return function->vtbl->get_code(function);
 567 }
 568
 569 static void Function_destructor(jsdisp_t *dispex)
 570 {
 571     FunctionInstance *function = function_from_jsdisp(dispex);
 572     function->vtbl->destructor(function);
 573     free(function);
 574 }
 575
 576 static HRESULT Function_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, jsdisp_t *dispex)
 577 {
 578     FunctionInstance *function = function_from_jsdisp(dispex);
 579     return function->vtbl->gc_traverse(gc_ctx, op, function);
 580 }
 581
 582 static const builtin_prop_t Function_props[] = {
 583     {L"apply",               Function_apply,                 PROPF_METHOD|2},
 584     {L"arguments",           NULL, 0,                        Function_get_arguments},
 585     {L"bind",                Function_bind,                  PROPF_METHOD|PROPF_ES5|1},
 586     {L"call",                Function_call,                  PROPF_METHOD|1},
 587     {L"length",              NULL, 0,                        Function_get_length},
 588     {L"toString",            Function_toString,              PROPF_METHOD}
 589 };
 590
 591 static const builtin_info_t Function_info = {
 592     JSCLASS_FUNCTION,
 593     Function_value,
 594     ARRAY_SIZE(Function_props),
 595     Function_props,
 596     Function_destructor,
 597     NULL,
 598     NULL,
 599     NULL,
 600     NULL,
 601     Function_gc_traverse
 602 };
 603
 604 static const builtin_prop_t FunctionInst_props[] = {
 605     {L"arguments",           NULL, 0,                        Function_get_arguments},
 606     {L"length",              NULL, 0,                        Function_get_length}
 607 };
 608
 609 static const builtin_info_t FunctionInst_info = {
 610     JSCLASS_FUNCTION,
 611     Function_value,
 612     ARRAY_SIZE(FunctionInst_props),
 613     FunctionInst_props,
 614     Function_destructor,
 615     NULL,
 616     NULL,
 617     NULL,
 618     NULL,
 619     Function_gc_traverse
 620 };
 621
 622 static HRESULT create_function(script_ctx_t *ctx, const builtin_info_t *builtin_info, const function_vtbl_t *vtbl, size_t size,
 623         DWORD flags, BOOL funcprot, jsdisp_t *prototype, void **ret)
 624 {
 625     FunctionInstance *function;
 626     HRESULT hres;
 627
 628     function = calloc(1, size);
 629     if(!function)
 630         return E_OUTOFMEMORY;
 631
 632     if(funcprot)
 633         hres = init_dispex(&function->dispex, ctx, builtin_info, prototype);
 634     else if(builtin_info)
 635         hres = init_dispex_from_constr(&function->dispex, ctx, builtin_info, ctx->function_constr);
 636     else
 637         hres = init_dispex_from_constr(&function->dispex, ctx, &FunctionInst_info, ctx->function_constr);
 638     if(FAILED(hres)) {
 639         free(function);
 640         return hres;
 641     }
 642
 643     function->vtbl = vtbl;
 644     function->flags = flags;
 645     function->length = flags & PROPF_ARGMASK;
 646
 647     *ret = function;
 648     return S_OK;
 649 }
 650
 651 static HRESULT NativeFunction_call(script_ctx_t *ctx, FunctionInstance *func, jsval_t vthis, unsigned flags,
 652         unsigned argc, jsval_t *argv, jsval_t *r)
 653 {
 654     NativeFunction *function = (NativeFunction*)func;
 655
 656     if((flags & DISPATCH_CONSTRUCT) && !(function->function.flags & PROPF_CONSTR))
 657         return JS_E_INVALID_ACTION;
 658     return function->proc(ctx, vthis, flags & ~DISPATCH_JSCRIPT_INTERNAL_MASK, argc, argv, r);
 659 }
 660
 661 static HRESULT NativeFunction_toString(FunctionInstance *func, jsstr_t **ret)
 662 {
 663     NativeFunction *function = (NativeFunction*)func;
 664     DWORD name_len;
 665     jsstr_t *str;
 666     WCHAR *ptr;
 667
 668     static const WCHAR native_prefixW[] = L"\nfunction ";
 669     static const WCHAR native_suffixW[] = L"() {\n    [native code]\n}\n";
 670
 671     name_len = function->name ? lstrlenW(function->name) : 0;
 672     str = jsstr_alloc_buf(ARRAY_SIZE(native_prefixW) + ARRAY_SIZE(native_suffixW) + name_len - 2, &ptr);
 673     if(!str)
 674         return E_OUTOFMEMORY;
 675
 676     memcpy(ptr, native_prefixW, sizeof(native_prefixW));
 677     ptr += ARRAY_SIZE(native_prefixW) - 1;
 678     memcpy(ptr, function->name, name_len*sizeof(WCHAR));
 679     ptr += name_len;
 680     memcpy(ptr, native_suffixW, sizeof(native_suffixW));
 681
 682     *ret = str;
 683     return S_OK;
 684 }
 685
 686 static function_code_t *NativeFunction_get_code(FunctionInstance *function)
 687 {
 688     return NULL;
 689 }
 690
 691 static void NativeFunction_destructor(FunctionInstance *function)
 692 {
 693 }
 694
 695 static const function_vtbl_t NativeFunctionVtbl = {
 696     NativeFunction_call,
 697     NativeFunction_toString,
 698     NativeFunction_get_code,
 699     NativeFunction_destructor,
 700     no_gc_traverse
 701 };
 702
 703 HRESULT create_builtin_function(script_ctx_t *ctx, builtin_invoke_t value_proc, const WCHAR *name,
 704         const builtin_info_t *builtin_info, DWORD flags, jsdisp_t *prototype, jsdisp_t **ret)
 705 {
 706     NativeFunction *function;
 707     HRESULT hres;
 708
 709     if(!ctx->function_constr)
 710         return E_UNEXPECTED;
 711
 712     hres = create_function(ctx, builtin_info, &NativeFunctionVtbl, sizeof(NativeFunction), flags, FALSE, NULL, (void**)&function);
 713     if(FAILED(hres))
 714         return hres;
 715
 716     if(builtin_info)
 717         hres = jsdisp_define_data_property(&function->function.dispex, L"length", 0,
 718                                            jsval_number(function->function.length));
 719     if(SUCCEEDED(hres))
 720         hres = jsdisp_define_data_property(&function->function.dispex, L"prototype", 0, prototype ? jsval_obj(prototype) : jsval_null());
 721     if(FAILED(hres)) {
 722         jsdisp_release(&function->function.dispex);
 723         return hres;
 724     }
 725
 726     function->proc = value_proc;
 727     function->name = name;
 728
 729     *ret = &function->function.dispex;
 730     return S_OK;
 731 }
 732
 733 static HRESULT set_constructor_prop(script_ctx_t *ctx, jsdisp_t *constr, jsdisp_t *prot)
 734 {
 735     return jsdisp_define_data_property(prot, L"constructor", PROPF_WRITABLE | PROPF_CONFIGURABLE,
 736                                        jsval_obj(constr));
 737 }
 738
 739 HRESULT create_builtin_constructor(script_ctx_t *ctx, builtin_invoke_t value_proc, const WCHAR *name,
 740         const builtin_info_t *builtin_info, DWORD flags, jsdisp_t *prototype, jsdisp_t **ret)
 741 {
 742     jsdisp_t *constr;
 743     HRESULT hres;
 744
 745     hres = create_builtin_function(ctx, value_proc, name, builtin_info, flags, prototype, &constr);
 746     if(FAILED(hres))
 747         return hres;
 748
 749     hres = set_constructor_prop(ctx, constr, prototype);
 750     if(FAILED(hres)) {
 751         jsdisp_release(constr);
 752         return hres;
 753     }
 754
 755     *ret = constr;
 756     return S_OK;
 757 }
 758
 759 /*
 760  * Create the actual prototype on demand, since it is a circular ref, which prevents the vast
 761  * majority of functions from being released quickly, leading to unnecessary scope detach.
 762  */
 763 static HRESULT InterpretedFunction_get_prototype(script_ctx_t *ctx, jsdisp_t *jsthis, jsval_t *r)
 764 {
 765     jsdisp_t *prototype;
 766     HRESULT hres;
 767
 768     hres = create_object(ctx, NULL, &prototype);
 769     if(FAILED(hres))
 770         return hres;
 771
 772     hres = jsdisp_define_data_property(jsthis, L"prototype", PROPF_WRITABLE, jsval_obj(prototype));
 773     if(SUCCEEDED(hres))
 774         hres = set_constructor_prop(ctx, jsthis, prototype);
 775     if(FAILED(hres)) {
 776         jsdisp_release(prototype);
 777         return hres;
 778     }
 779
 780     *r = jsval_obj(prototype);
 781     return S_OK;
 782 }
 783
 784 static HRESULT InterpretedFunction_set_prototype(script_ctx_t *ctx, jsdisp_t *jsthis, jsval_t value)
 785 {
 786     return jsdisp_define_data_property(jsthis, L"prototype", PROPF_WRITABLE, value);
 787 }
 788
 789 static const builtin_prop_t InterpretedFunction_props[] = {
 790     {L"arguments",           NULL, 0,                        Function_get_arguments},
 791     {L"length",              NULL, 0,                        Function_get_length},
 792     {L"prototype",           NULL, 0,                        InterpretedFunction_get_prototype, InterpretedFunction_set_prototype}
 793 };
 794
 795 static const builtin_info_t InterpretedFunction_info = {
 796     JSCLASS_FUNCTION,
 797     Function_value,
 798     ARRAY_SIZE(InterpretedFunction_props),
 799     InterpretedFunction_props,
 800     Function_destructor,
 801     NULL,
 802     NULL,
 803     NULL,
 804     NULL,
 805     Function_gc_traverse
 806 };
 807
 808 static HRESULT InterpretedFunction_call(script_ctx_t *ctx, FunctionInstance *func, jsval_t vthis, unsigned flags,
 809          unsigned argc, jsval_t *argv, jsval_t *r)
 810 {
 811     InterpretedFunction *function = (InterpretedFunction*)func;
 812     IDispatch *this_obj = NULL;
 813     DWORD exec_flags = 0;
 814     jsdisp_t *new_obj;
 815     HRESULT hres;
 816
 817     TRACE("%p\n", function);
 818
 819     if(flags & DISPATCH_CONSTRUCT) {
 820         hres = create_object(ctx, &function->function.dispex, &new_obj);
 821         if(FAILED(hres))
 822             return hres;
 823         this_obj = to_disp(new_obj);
 824     }else if(is_object_instance(vthis)) {
 825         this_obj = get_object(vthis);
 826         IDispatch_AddRef(this_obj);
 827     }else if(ctx->version >= SCRIPTLANGUAGEVERSION_ES5 && !is_undefined(vthis) && !is_null(vthis)) {
 828         hres = to_object(ctx, vthis, &this_obj);
 829         if(FAILED(hres))
 830             return hres;
 831     }
 832
 833     if(flags & DISPATCH_JSCRIPT_CALLEREXECSSOURCE)
 834         exec_flags |= EXEC_RETURN_TO_INTERP;
 835     if(flags & DISPATCH_CONSTRUCT)
 836         exec_flags |= EXEC_CONSTRUCTOR;
 837     hres = exec_source(ctx, exec_flags, function->code, function->func_code, function->scope_chain, this_obj,
 838                        &function->function.dispex, argc, argv, r);
 839     if(this_obj)
 840         IDispatch_Release(this_obj);
 841     return hres;
 842 }
 843
 844 static HRESULT InterpretedFunction_toString(FunctionInstance *func, jsstr_t **ret)
 845 {
 846     InterpretedFunction *function = (InterpretedFunction*)func;
 847
 848     *ret = jsstr_alloc_len(function->func_code->source, function->func_code->source_len);
 849     return *ret ? S_OK : E_OUTOFMEMORY;
 850 }
 851
 852 static function_code_t *InterpretedFunction_get_code(FunctionInstance *func)
 853 {
 854     InterpretedFunction *function = (InterpretedFunction*)func;
 855
 856     return function->func_code;
 857 }
 858
 859 static void InterpretedFunction_destructor(FunctionInstance *func)
 860 {
 861     InterpretedFunction *function = (InterpretedFunction*)func;
 862
 863     release_bytecode(function->code);
 864     if(function->scope_chain)
 865         scope_release(function->scope_chain);
 866 }
 867
 868 static HRESULT InterpretedFunction_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, FunctionInstance *func)
 869 {
 870     InterpretedFunction *function = (InterpretedFunction*)func;
 871
 872     if(!function->scope_chain)
 873         return S_OK;
 874     return gc_process_linked_obj(gc_ctx, op, &function->function.dispex, &function->scope_chain->dispex,
 875                                  (void**)&function->scope_chain);
 876 }
 877
 878 static const function_vtbl_t InterpretedFunctionVtbl = {
 879     InterpretedFunction_call,
 880     InterpretedFunction_toString,
 881     InterpretedFunction_get_code,
 882     InterpretedFunction_destructor,
 883     InterpretedFunction_gc_traverse
 884 };
 885
 886 HRESULT create_source_function(script_ctx_t *ctx, bytecode_t *code, function_code_t *func_code,
 887         scope_chain_t *scope_chain, jsdisp_t **ret)
 888 {
 889     InterpretedFunction *function;
 890     HRESULT hres;
 891
 892     hres = create_function(ctx, &InterpretedFunction_info, &InterpretedFunctionVtbl, sizeof(InterpretedFunction),
 893                            PROPF_CONSTR, FALSE, NULL, (void**)&function);
 894     if(FAILED(hres))
 895         return hres;
 896
 897     if(scope_chain) {
 898         scope_addref(scope_chain);
 899         function->scope_chain = scope_chain;
 900     }
 901
 902     bytecode_addref(code);
 903     function->code = code;
 904     function->func_code = func_code;
 905     function->function.length = function->func_code->param_cnt;
 906
 907     *ret = &function->function.dispex;
 908     return S_OK;
 909 }
 910
 911 static HRESULT BindFunction_call(script_ctx_t *ctx, FunctionInstance *func, jsval_t vthis, unsigned flags,
 912          unsigned argc, jsval_t *argv, jsval_t *r)
 913 {
 914     BindFunction *function = (BindFunction*)func;
 915     jsval_t *call_args = NULL;
 916     unsigned call_argc;
 917     HRESULT hres;
 918
 919     TRACE("%p\n", function);
 920
 921     call_argc = function->argc + argc;
 922     if(call_argc) {
 923         call_args = malloc(call_argc * sizeof(*call_args));
 924         if(!call_args)
 925             return E_OUTOFMEMORY;
 926
 927         if(function->argc)
 928             memcpy(call_args, function->args, function->argc * sizeof(*call_args));
 929         if(argc)
 930             memcpy(call_args + function->argc, argv, argc * sizeof(*call_args));
 931     }
 932
 933     hres = function->target->vtbl->call(ctx, function->target, function->this, flags, call_argc, call_args, r);
 934
 935     free(call_args);
 936     return hres;
 937 }
 938
 939 static HRESULT BindFunction_toString(FunctionInstance *function, jsstr_t **ret)
 940 {
 941     *ret = jsstr_alloc(L"\nfunction() {\n    [native code]\n}\n");
 942     return *ret ? S_OK : E_OUTOFMEMORY;
 943 }
 944
 945 static function_code_t *BindFunction_get_code(FunctionInstance *function)
 946 {
 947     return NULL;
 948 }
 949
 950 static void BindFunction_destructor(FunctionInstance *func)
 951 {
 952     BindFunction *function = (BindFunction*)func;
 953     unsigned i;
 954
 955     TRACE("%p\n", function);
 956
 957     for(i = 0; i < function->argc; i++)
 958         jsval_release(function->args[i]);
 959     if(function->target)
 960         jsdisp_release(&function->target->dispex);
 961     jsval_release(function->this);
 962 }
 963
 964 static HRESULT BindFunction_gc_traverse(struct gc_ctx *gc_ctx, enum gc_traverse_op op, FunctionInstance *func)
 965 {
 966     BindFunction *function = (BindFunction*)func;
 967     HRESULT hres;
 968     unsigned i;
 969
 970     for(i = 0; i < function->argc; i++) {
 971         hres = gc_process_linked_val(gc_ctx, op, &function->function.dispex, &function->args[i]);
 972         if(FAILED(hres))
 973             return hres;
 974     }
 975
 976     hres = gc_process_linked_obj(gc_ctx, op, &function->function.dispex, &function->target->dispex, (void**)&function->target);
 977     if(FAILED(hres))
 978         return hres;
 979
 980     return gc_process_linked_val(gc_ctx, op, &function->function.dispex, &function->this);
 981 }
 982
 983 static const function_vtbl_t BindFunctionVtbl = {
 984     BindFunction_call,
 985     BindFunction_toString,
 986     BindFunction_get_code,
 987     BindFunction_destructor,
 988     BindFunction_gc_traverse
 989 };
 990
 991 static HRESULT create_bind_function(script_ctx_t *ctx, FunctionInstance *target, jsval_t bound_this, unsigned argc,
 992                                     jsval_t *argv, jsdisp_t **ret)
 993 {
 994     BindFunction *function;
 995     HRESULT hres;
 996
 997     hres = create_function(ctx, NULL, &BindFunctionVtbl, FIELD_OFFSET(BindFunction, args[argc]), PROPF_METHOD,
 998                            FALSE, NULL, (void**)&function);
 999     if(FAILED(hres))
1000         return hres;
1001
1002     jsdisp_addref(&target->dispex);
1003     function->target = target;
1004
1005     hres = jsval_copy(bound_this, &function->this);
1006     if(FAILED(hres)) {
1007         jsdisp_release(&function->function.dispex);
1008         return hres;
1009     }
1010
1011     for(function->argc = 0; function->argc < argc; function->argc++) {
1012         hres = jsval_copy(argv[function->argc], function->args + function->argc);
1013         if(FAILED(hres)) {
1014             jsdisp_release(&function->function.dispex);
1015             return hres;
1016         }
1017     }
1018
1019     function->function.length = target->length > argc ? target->length - argc : 0;
1020
1021     *ret = &function->function.dispex;
1022     return S_OK;
1023 }
1024
1025 static HRESULT construct_function(script_ctx_t *ctx, unsigned argc, jsval_t *argv, IDispatch **ret)
1026 {
1027     WCHAR *str = NULL, *ptr;
1028     unsigned len = 0, i = 0;
1029     bytecode_t *code;
1030     jsdisp_t *function;
1031     jsstr_t **params = NULL;
1032     int j = 0;
1033     HRESULT hres = S_OK;
1034
1035     static const WCHAR function_anonymousW[] = L"function anonymous(";
1036     static const WCHAR function_beginW[] = L") {\n";
1037     static const WCHAR function_endW[] = L"\n}";
1038
1039     if(argc) {
1040         params = malloc(argc*sizeof(*params));
1041         if(!params)
1042             return E_OUTOFMEMORY;
1043
1044         if(argc > 2)
1045             len = (argc-2)*2; /* separating commas */
1046         for(i=0; i < argc; i++) {
1047             hres = to_string(ctx, argv[i], params+i);
1048             if(FAILED(hres))
1049                 break;
1050             len += jsstr_length(params[i]);
1051         }
1052     }
1053
1054     if(SUCCEEDED(hres)) {
1055         len += ARRAY_SIZE(function_anonymousW) + ARRAY_SIZE(function_beginW) + ARRAY_SIZE(function_endW) - 2;
1056         str = malloc(len*sizeof(WCHAR));
1057         if(str) {
1058             memcpy(str, function_anonymousW, sizeof(function_anonymousW));
1059             ptr = str + ARRAY_SIZE(function_anonymousW) - 1;
1060             if(argc > 1) {
1061                 while(1) {
1062                     ptr += jsstr_flush(params[j], ptr);
1063                     if(++j == argc-1)
1064                         break;
1065                     *ptr++ = ',';
1066                     *ptr++ = ' ';
1067                 }
1068             }
1069             memcpy(ptr, function_beginW, sizeof(function_beginW));
1070             ptr += ARRAY_SIZE(function_beginW) - 1;
1071             if(argc)
1072                 ptr += jsstr_flush(params[argc-1], ptr);
1073             memcpy(ptr, function_endW, sizeof(function_endW));
1074
1075             TRACE("%s\n", debugstr_w(str));
1076         }else {
1077             hres = E_OUTOFMEMORY;
1078         }
1079     }
1080
1081     while(i)
1082         jsstr_release(params[--i]);
1083     free(params);
1084     if(FAILED(hres))
1085         return hres;
1086
1087     hres = compile_script(ctx, str, 0, 0, NULL, NULL, FALSE, FALSE,
1088                           ctx->call_ctx ? ctx->call_ctx->bytecode->named_item : NULL, &code);
1089     free(str);
1090     if(FAILED(hres))
1091         return hres;
1092
1093     if(code->global_code.func_cnt != 1 || code->global_code.var_cnt != 1) {
1094         ERR("Invalid parser result!\n");
1095         release_bytecode(code);
1096         return E_UNEXPECTED;
1097     }
1098
1099     hres = create_source_function(ctx, code, code->global_code.funcs, NULL, &function);
1100     release_bytecode(code);
1101     if(FAILED(hres))
1102         return hres;
1103
1104     *ret = to_disp(function);
1105     return S_OK;
1106 }
1107
1108 static HRESULT FunctionConstr_value(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
1109         jsval_t *r)
1110 {
1111     HRESULT hres;
1112
1113     TRACE("\n");
1114
1115     switch(flags) {
1116     case DISPATCH_METHOD:
1117     case DISPATCH_CONSTRUCT: {
1118         IDispatch *ret;
1119
1120         hres = construct_function(ctx, argc, argv, &ret);
1121         if(FAILED(hres))
1122             return hres;
1123
1124         if(r) *r = jsval_disp(ret);
1125         else  IDispatch_Release(ret);
1126         break;
1127     }
1128     default:
1129         FIXME("unimplemented flags %x\n", flags);
1130         return E_NOTIMPL;
1131     }
1132
1133     return S_OK;
1134 }
1135
1136 static HRESULT FunctionProt_value(script_ctx_t *ctx, jsval_t vthis, WORD flags, unsigned argc, jsval_t *argv,
1137         jsval_t *r)
1138 {
1139     FIXME("\n");
1140     return E_NOTIMPL;
1141 }
1142
1143 HRESULT init_function_constr(script_ctx_t *ctx, jsdisp_t *object_prototype)
1144 {
1145     NativeFunction *prot, *constr;
1146     HRESULT hres;
1147
1148     hres = create_function(ctx, &Function_info, &NativeFunctionVtbl, sizeof(NativeFunction), PROPF_CONSTR,
1149                            TRUE, object_prototype, (void**)&prot);
1150     if(FAILED(hres))
1151         return hres;
1152
1153     prot->proc = FunctionProt_value;
1154     prot->name = L"prototype";
1155
1156     hres = create_function(ctx, &FunctionInst_info, &NativeFunctionVtbl, sizeof(NativeFunction), PROPF_CONSTR|1,
1157                            TRUE, &prot->function.dispex, (void**)&constr);
1158     if(SUCCEEDED(hres)) {
1159         constr->proc = FunctionConstr_value;
1160         constr->name = L"Function";
1161         hres = jsdisp_define_data_property(&constr->function.dispex, L"prototype", 0, jsval_obj(&prot->function.dispex));
1162         if(SUCCEEDED(hres))
1163             hres = set_constructor_prop(ctx, &constr->function.dispex, &prot->function.dispex);
1164         if(FAILED(hres))
1165             jsdisp_release(&constr->function.dispex);
1166     }
1167     jsdisp_release(&prot->function.dispex);
1168     if(FAILED(hres))
1169         return hres;
1170
1171     ctx->function_constr = &constr->function.dispex;
1172     return S_OK;
1173 }