dlls/jscript/compile.c

   1 /*
   2  * Copyright 2011 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 <math.h>
  20 #include <assert.h>
  21
  22 #include "jscript.h"
  23 #include "engine.h"
  24 #include "parser.h"
  25
  26 #include "wine/rbtree.h"
  27 #include "wine/debug.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  30 WINE_DECLARE_DEBUG_CHANNEL(jscript_disas);
  31
  32 typedef struct _statement_ctx_t {
  33     unsigned stack_use;
  34     BOOL using_scope;
  35     BOOL using_except;
  36
  37     unsigned break_label;
  38     unsigned continue_label;
  39
  40     const labelled_statement_t *labelled_stat;
  41
  42     struct _statement_ctx_t *next;
  43 } statement_ctx_t;
  44
  45 typedef struct {
  46     struct wine_rb_entry entry;
  47     BSTR name;
  48     int ref;
  49 } function_local_t;
  50
  51 typedef struct _compiler_ctx_t {
  52     parser_ctx_t *parser;
  53     bytecode_t *code;
  54
  55     BOOL from_eval;
  56
  57     unsigned code_off;
  58     unsigned code_size;
  59
  60     unsigned *labels;
  61     unsigned labels_size;
  62     unsigned labels_cnt;
  63
  64     struct wine_rb_tree locals;
  65     unsigned locals_cnt;
  66
  67     statement_ctx_t *stat_ctx;
  68     function_code_t *func;
  69
  70     unsigned loc;
  71
  72     function_expression_t *func_head;
  73     function_expression_t *func_tail;
  74
  75     heap_pool_t heap;
  76 } compiler_ctx_t;
  77
  78 static const struct {
  79     const char *op_str;
  80     instr_arg_type_t arg1_type;
  81     instr_arg_type_t arg2_type;
  82 } instr_info[] = {
  83 #define X(n,a,b,c) {#n,b,c},
  84 OP_LIST
  85 #undef X
  86 };
  87
  88 static void dump_instr_arg(instr_arg_type_t type, instr_arg_t *arg)
  89 {
  90     switch(type) {
  91     case ARG_STR:
  92         TRACE_(jscript_disas)("\t%s", debugstr_jsstr(arg->str));
  93         break;
  94     case ARG_BSTR:
  95         TRACE_(jscript_disas)("\t%s", debugstr_wn(arg->bstr, SysStringLen(arg->bstr)));
  96         break;
  97     case ARG_INT:
  98         TRACE_(jscript_disas)("\t%d", arg->uint);
  99         break;
 100     case ARG_UINT:
 101     case ARG_ADDR:
 102         TRACE_(jscript_disas)("\t%u", arg->uint);
 103         break;
 104     case ARG_FUNC:
 105     case ARG_NONE:
 106         break;
 107     DEFAULT_UNREACHABLE;
 108     }
 109 }
 110
 111 static void dump_code(compiler_ctx_t *ctx, unsigned off)
 112 {
 113     instr_t *instr;
 114
 115     for(instr = ctx->code->instrs+off; instr < ctx->code->instrs+ctx->code_off; instr++) {
 116         TRACE_(jscript_disas)("%d:\t%s", (int)(instr-ctx->code->instrs), instr_info[instr->op].op_str);
 117         if(instr_info[instr->op].arg1_type == ARG_DBL) {
 118             TRACE_(jscript_disas)("\t%lf", instr->u.dbl);
 119         }else {
 120             dump_instr_arg(instr_info[instr->op].arg1_type, instr->u.arg);
 121             dump_instr_arg(instr_info[instr->op].arg2_type, instr->u.arg+1);
 122         }
 123         TRACE_(jscript_disas)("\n");
 124     }
 125 }
 126
 127 static HRESULT compile_expression(compiler_ctx_t*,expression_t*,BOOL);
 128 static HRESULT compile_statement(compiler_ctx_t*,statement_ctx_t*,statement_t*);
 129
 130 static inline void *compiler_alloc(bytecode_t *code, size_t size)
 131 {
 132     return heap_pool_alloc(&code->heap, size);
 133 }
 134
 135 jsstr_t *compiler_alloc_string_len(compiler_ctx_t *ctx, const WCHAR *str, unsigned len)
 136 {
 137     jsstr_t *new_str;
 138
 139     if(!ctx->code->str_pool_size) {
 140         ctx->code->str_pool = heap_alloc(8 * sizeof(jsstr_t*));
 141         if(!ctx->code->str_pool)
 142             return NULL;
 143         ctx->code->str_pool_size = 8;
 144     }else if(ctx->code->str_pool_size == ctx->code->str_cnt) {
 145         jsstr_t **new_pool;
 146
 147         new_pool = heap_realloc(ctx->code->str_pool, ctx->code->str_pool_size*2*sizeof(jsstr_t*));
 148         if(!new_pool)
 149             return NULL;
 150
 151         ctx->code->str_pool = new_pool;
 152         ctx->code->str_pool_size *= 2;
 153     }
 154
 155     new_str = jsstr_alloc_len(str, len);
 156     if(!new_str)
 157         return NULL;
 158
 159     ctx->code->str_pool[ctx->code->str_cnt++] = new_str;
 160     return new_str;
 161 }
 162
 163 static jsstr_t *compiler_alloc_string(compiler_ctx_t *ctx, const WCHAR *str)
 164 {
 165     return compiler_alloc_string_len(ctx, str, lstrlenW(str));
 166 }
 167
 168 static BOOL ensure_bstr_slot(compiler_ctx_t *ctx)
 169 {
 170     if(!ctx->code->bstr_pool_size) {
 171         ctx->code->bstr_pool = heap_alloc(8 * sizeof(BSTR));
 172         if(!ctx->code->bstr_pool)
 173             return FALSE;
 174         ctx->code->bstr_pool_size = 8;
 175     }else if(ctx->code->bstr_pool_size == ctx->code->bstr_cnt) {
 176         BSTR *new_pool;
 177
 178         new_pool = heap_realloc(ctx->code->bstr_pool, ctx->code->bstr_pool_size*2*sizeof(BSTR));
 179         if(!new_pool)
 180             return FALSE;
 181
 182         ctx->code->bstr_pool = new_pool;
 183         ctx->code->bstr_pool_size *= 2;
 184     }
 185
 186     return TRUE;
 187 }
 188
 189 static BSTR compiler_alloc_bstr(compiler_ctx_t *ctx, const WCHAR *str)
 190 {
 191     if(!ensure_bstr_slot(ctx))
 192         return NULL;
 193
 194     ctx->code->bstr_pool[ctx->code->bstr_cnt] = SysAllocString(str);
 195     if(!ctx->code->bstr_pool[ctx->code->bstr_cnt])
 196         return NULL;
 197
 198     return ctx->code->bstr_pool[ctx->code->bstr_cnt++];
 199 }
 200
 201 static BSTR compiler_alloc_bstr_len(compiler_ctx_t *ctx, const WCHAR *str, size_t len)
 202 {
 203     if(!ensure_bstr_slot(ctx))
 204         return NULL;
 205
 206     ctx->code->bstr_pool[ctx->code->bstr_cnt] = SysAllocStringLen(str, len);
 207     if(!ctx->code->bstr_pool[ctx->code->bstr_cnt])
 208         return NULL;
 209
 210     return ctx->code->bstr_pool[ctx->code->bstr_cnt++];
 211 }
 212
 213 void set_compiler_loc(compiler_ctx_t *ctx, unsigned loc)
 214 {
 215     ctx->loc = loc;
 216 }
 217
 218 static unsigned push_instr(compiler_ctx_t *ctx, jsop_t op)
 219 {
 220     assert(ctx->code_size >= ctx->code_off);
 221
 222     if(ctx->code_size == ctx->code_off) {
 223         instr_t *new_instrs;
 224
 225         new_instrs = heap_realloc(ctx->code->instrs, ctx->code_size*2*sizeof(instr_t));
 226         if(!new_instrs)
 227             return 0;
 228
 229         ctx->code->instrs = new_instrs;
 230         ctx->code_size *= 2;
 231     }
 232
 233     ctx->code->instrs[ctx->code_off].op = op;
 234     ctx->code->instrs[ctx->code_off].loc = ctx->loc;
 235     return ctx->code_off++;
 236 }
 237
 238 static inline instr_t *instr_ptr(compiler_ctx_t *ctx, unsigned off)
 239 {
 240     assert(off < ctx->code_off);
 241     return ctx->code->instrs + off;
 242 }
 243
 244 static HRESULT push_instr_int(compiler_ctx_t *ctx, jsop_t op, LONG arg)
 245 {
 246     unsigned instr;
 247
 248     instr = push_instr(ctx, op);
 249     if(!instr)
 250         return E_OUTOFMEMORY;
 251
 252     instr_ptr(ctx, instr)->u.arg->lng = arg;
 253     return S_OK;
 254 }
 255
 256 static HRESULT push_instr_str(compiler_ctx_t *ctx, jsop_t op, jsstr_t *str)
 257 {
 258     unsigned instr;
 259
 260     instr = push_instr(ctx, op);
 261     if(!instr)
 262         return E_OUTOFMEMORY;
 263
 264     instr_ptr(ctx, instr)->u.arg->str = str;
 265     return S_OK;
 266 }
 267
 268 static HRESULT push_instr_str_uint(compiler_ctx_t *ctx, jsop_t op, jsstr_t *str, unsigned arg2)
 269 {
 270     unsigned instr;
 271
 272     instr = push_instr(ctx, op);
 273     if(!instr)
 274         return E_OUTOFMEMORY;
 275
 276     instr_ptr(ctx, instr)->u.arg[0].str = str;
 277     instr_ptr(ctx, instr)->u.arg[1].uint = arg2;
 278     return S_OK;
 279 }
 280
 281 static HRESULT push_instr_bstr(compiler_ctx_t *ctx, jsop_t op, const WCHAR *arg)
 282 {
 283     unsigned instr;
 284     WCHAR *str;
 285
 286     str = compiler_alloc_bstr(ctx, arg);
 287     if(!str)
 288         return E_OUTOFMEMORY;
 289
 290     instr = push_instr(ctx, op);
 291     if(!instr)
 292         return E_OUTOFMEMORY;
 293
 294     instr_ptr(ctx, instr)->u.arg->bstr = str;
 295     return S_OK;
 296 }
 297
 298 static HRESULT push_instr_bstr_uint(compiler_ctx_t *ctx, jsop_t op, const WCHAR *arg1, unsigned arg2)
 299 {
 300     unsigned instr;
 301     WCHAR *str;
 302
 303     str = compiler_alloc_bstr(ctx, arg1);
 304     if(!str)
 305         return E_OUTOFMEMORY;
 306
 307     instr = push_instr(ctx, op);
 308     if(!instr)
 309         return E_OUTOFMEMORY;
 310
 311     instr_ptr(ctx, instr)->u.arg[0].bstr = str;
 312     instr_ptr(ctx, instr)->u.arg[1].uint = arg2;
 313     return S_OK;
 314 }
 315
 316 static HRESULT push_instr_uint_str(compiler_ctx_t *ctx, jsop_t op, unsigned arg1, const WCHAR *arg2)
 317 {
 318     unsigned instr;
 319     jsstr_t *str;
 320
 321     str = compiler_alloc_string(ctx, arg2);
 322     if(!str)
 323         return E_OUTOFMEMORY;
 324
 325     instr = push_instr(ctx, op);
 326     if(!instr)
 327         return E_OUTOFMEMORY;
 328
 329     instr_ptr(ctx, instr)->u.arg[0].uint = arg1;
 330     instr_ptr(ctx, instr)->u.arg[1].str = str;
 331     return S_OK;
 332 }
 333
 334 static HRESULT push_instr_double(compiler_ctx_t *ctx, jsop_t op, double arg)
 335 {
 336     unsigned instr;
 337
 338     instr = push_instr(ctx, op);
 339     if(!instr)
 340         return E_OUTOFMEMORY;
 341
 342     instr_ptr(ctx, instr)->u.dbl = arg;
 343     return S_OK;
 344 }
 345
 346 static inline void set_arg_uint(compiler_ctx_t *ctx, unsigned instr, unsigned arg)
 347 {
 348     instr_ptr(ctx, instr)->u.arg->uint = arg;
 349 }
 350
 351 static HRESULT push_instr_uint(compiler_ctx_t *ctx, jsop_t op, unsigned arg)
 352 {
 353     unsigned instr;
 354
 355     instr = push_instr(ctx, op);
 356     if(!instr)
 357         return E_OUTOFMEMORY;
 358
 359     set_arg_uint(ctx, instr, arg);
 360     return S_OK;
 361 }
 362
 363 static HRESULT compile_binary_expression(compiler_ctx_t *ctx, binary_expression_t *expr, jsop_t op)
 364 {
 365     HRESULT hres;
 366
 367     hres = compile_expression(ctx, expr->expression1, TRUE);
 368     if(FAILED(hres))
 369         return hres;
 370
 371     hres = compile_expression(ctx, expr->expression2, TRUE);
 372     if(FAILED(hres))
 373         return hres;
 374
 375     return push_instr(ctx, op) ? S_OK : E_OUTOFMEMORY;
 376 }
 377
 378 static HRESULT compile_unary_expression(compiler_ctx_t *ctx, unary_expression_t *expr, jsop_t op)
 379 {
 380     HRESULT hres;
 381
 382     hres = compile_expression(ctx, expr->expression, TRUE);
 383     if(FAILED(hres))
 384         return hres;
 385
 386     return push_instr(ctx, op) ? S_OK : E_OUTOFMEMORY;
 387 }
 388
 389 /* ECMA-262 3rd Edition    11.2.1 */
 390 static HRESULT compile_member_expression(compiler_ctx_t *ctx, member_expression_t *expr)
 391 {
 392     HRESULT hres;
 393
 394     hres = compile_expression(ctx, expr->expression, TRUE);
 395     if(FAILED(hres))
 396         return hres;
 397
 398     return push_instr_bstr(ctx, OP_member, expr->identifier);
 399 }
 400
 401 #define LABEL_FLAG 0x80000000
 402
 403 static unsigned alloc_label(compiler_ctx_t *ctx)
 404 {
 405     if(!ctx->labels_size) {
 406         ctx->labels = heap_alloc(8 * sizeof(*ctx->labels));
 407         if(!ctx->labels)
 408             return 0;
 409         ctx->labels_size = 8;
 410     }else if(ctx->labels_size == ctx->labels_cnt) {
 411         unsigned *new_labels;
 412
 413         new_labels = heap_realloc(ctx->labels, 2*ctx->labels_size*sizeof(*ctx->labels));
 414         if(!new_labels)
 415             return 0;
 416
 417         ctx->labels = new_labels;
 418         ctx->labels_size *= 2;
 419     }
 420
 421     return ctx->labels_cnt++ | LABEL_FLAG;
 422 }
 423
 424 static void label_set_addr(compiler_ctx_t *ctx, unsigned label)
 425 {
 426     assert(label & LABEL_FLAG);
 427     ctx->labels[label & ~LABEL_FLAG] = ctx->code_off;
 428 }
 429
 430 static inline BOOL is_memberid_expr(expression_type_t type)
 431 {
 432     return type == EXPR_IDENT || type == EXPR_MEMBER || type == EXPR_ARRAY;
 433 }
 434
 435 static BOOL bind_local(compiler_ctx_t *ctx, const WCHAR *identifier, int *ret_ref)
 436 {
 437     statement_ctx_t *iter;
 438     local_ref_t *ref;
 439
 440     for(iter = ctx->stat_ctx; iter; iter = iter->next) {
 441         if(iter->using_scope)
 442             return FALSE;
 443     }
 444
 445     ref = lookup_local(ctx->func, identifier);
 446     if(!ref)
 447         return FALSE;
 448
 449     *ret_ref = ref->ref;
 450     return TRUE;
 451 }
 452
 453 static HRESULT emit_identifier_ref(compiler_ctx_t *ctx, const WCHAR *identifier, unsigned flags)
 454 {
 455     int local_ref;
 456     if(bind_local(ctx, identifier, &local_ref))
 457         return push_instr_int(ctx, OP_local_ref, local_ref);
 458     return push_instr_bstr_uint(ctx, OP_identid, identifier, flags);
 459 }
 460
 461 static HRESULT emit_identifier(compiler_ctx_t *ctx, const WCHAR *identifier)
 462 {
 463     int local_ref;
 464     if(bind_local(ctx, identifier, &local_ref))
 465         return push_instr_int(ctx, OP_local, local_ref);
 466     return push_instr_bstr(ctx, OP_ident, identifier);
 467 }
 468
 469 static HRESULT emit_member_expression(compiler_ctx_t *ctx, expression_t *expr)
 470 {
 471     HRESULT hres;
 472
 473     if(expr->type == EXPR_ARRAY) {
 474         binary_expression_t *array_expr = (binary_expression_t*)expr;
 475
 476         hres = compile_expression(ctx, array_expr->expression1, TRUE);
 477         if(FAILED(hres))
 478             return hres;
 479
 480         hres = compile_expression(ctx, array_expr->expression2, TRUE);
 481         if(FAILED(hres))
 482             return hres;
 483
 484         if(!push_instr(ctx, OP_to_string))
 485             return E_OUTOFMEMORY;
 486     }else {
 487         member_expression_t *member_expr = (member_expression_t*)expr;
 488         jsstr_t *jsstr;
 489
 490         assert(expr->type == EXPR_MEMBER);
 491
 492         hres = compile_expression(ctx, member_expr->expression, TRUE);
 493         if(FAILED(hres))
 494             return hres;
 495
 496         jsstr = compiler_alloc_string(ctx, member_expr->identifier);
 497         if(!jsstr)
 498             return E_OUTOFMEMORY;
 499
 500         hres = push_instr_str(ctx, OP_str, jsstr);
 501         if(FAILED(hres))
 502             return hres;
 503     }
 504
 505     return S_OK;
 506 }
 507
 508 static HRESULT compile_memberid_expression(compiler_ctx_t *ctx, expression_t *expr, unsigned flags)
 509 {
 510     HRESULT hres;
 511
 512     if(expr->type == EXPR_IDENT) {
 513         identifier_expression_t *ident_expr = (identifier_expression_t*)expr;
 514         return emit_identifier_ref(ctx, ident_expr->identifier, flags);
 515     }
 516
 517     hres = emit_member_expression(ctx, expr);
 518     if(FAILED(hres))
 519         return hres;
 520
 521     return push_instr_uint(ctx, OP_memberid, flags);
 522 }
 523
 524 static HRESULT compile_increment_expression(compiler_ctx_t *ctx, unary_expression_t *expr, jsop_t op, int n)
 525 {
 526     HRESULT hres;
 527
 528     if(!is_memberid_expr(expr->expression->type)) {
 529         hres = compile_expression(ctx, expr->expression, TRUE);
 530         if(FAILED(hres))
 531             return hres;
 532
 533         return push_instr_uint(ctx, OP_throw_ref, JS_E_ILLEGAL_ASSIGN);
 534     }
 535
 536     hres = compile_memberid_expression(ctx, expr->expression, fdexNameEnsure);
 537     if(FAILED(hres))
 538         return hres;
 539
 540     return push_instr_int(ctx, op, n);
 541 }
 542
 543 /* ECMA-262 3rd Edition    11.14 */
 544 static HRESULT compile_comma_expression(compiler_ctx_t *ctx, binary_expression_t *expr, BOOL emit_ret)
 545 {
 546     HRESULT hres;
 547
 548     hres = compile_expression(ctx, expr->expression1, FALSE);
 549     if(FAILED(hres))
 550         return hres;
 551
 552     return compile_expression(ctx, expr->expression2, emit_ret);
 553 }
 554
 555 /* ECMA-262 3rd Edition    11.11 */
 556 static HRESULT compile_logical_expression(compiler_ctx_t *ctx, binary_expression_t *expr, jsop_t op)
 557 {
 558     unsigned instr;
 559     HRESULT hres;
 560
 561     hres = compile_expression(ctx, expr->expression1, TRUE);
 562     if(FAILED(hres))
 563         return hres;
 564
 565     instr = push_instr(ctx, op);
 566     if(!instr)
 567         return E_OUTOFMEMORY;
 568
 569     hres = compile_expression(ctx, expr->expression2, TRUE);
 570     if(FAILED(hres))
 571         return hres;
 572
 573     set_arg_uint(ctx, instr, ctx->code_off);
 574     return S_OK;
 575 }
 576
 577 /* ECMA-262 3rd Edition    11.12 */
 578 static HRESULT compile_conditional_expression(compiler_ctx_t *ctx, conditional_expression_t *expr)
 579 {
 580     unsigned jmp_false, jmp_end;
 581     HRESULT hres;
 582
 583     hres = compile_expression(ctx, expr->expression, TRUE);
 584     if(FAILED(hres))
 585         return hres;
 586
 587     jmp_false = push_instr(ctx, OP_cnd_z);
 588     if(!jmp_false)
 589         return E_OUTOFMEMORY;
 590
 591     hres = compile_expression(ctx, expr->true_expression, TRUE);
 592     if(FAILED(hres))
 593         return hres;
 594
 595     jmp_end = push_instr(ctx, OP_jmp);
 596     if(!jmp_end)
 597         return E_OUTOFMEMORY;
 598
 599     set_arg_uint(ctx, jmp_false, ctx->code_off);
 600     hres = push_instr_uint(ctx, OP_pop, 1);
 601     if(FAILED(hres))
 602         return hres;
 603
 604     hres = compile_expression(ctx, expr->false_expression, TRUE);
 605     if(FAILED(hres))
 606         return hres;
 607
 608     set_arg_uint(ctx, jmp_end, ctx->code_off);
 609     return S_OK;
 610 }
 611
 612 static HRESULT compile_new_expression(compiler_ctx_t *ctx, call_expression_t *expr)
 613 {
 614     unsigned arg_cnt = 0;
 615     argument_t *arg;
 616     HRESULT hres;
 617
 618     hres = compile_expression(ctx, expr->expression, TRUE);
 619     if(FAILED(hres))
 620         return hres;
 621
 622     for(arg = expr->argument_list; arg; arg = arg->next) {
 623         hres = compile_expression(ctx, arg->expr, TRUE);
 624         if(FAILED(hres))
 625             return hres;
 626         arg_cnt++;
 627     }
 628
 629     hres = push_instr_uint(ctx, OP_new, arg_cnt);
 630     if(FAILED(hres))
 631         return hres;
 632
 633     hres = push_instr_uint(ctx, OP_pop, arg_cnt+1);
 634     if(FAILED(hres))
 635         return hres;
 636
 637     return push_instr(ctx, OP_push_acc) ? S_OK : E_OUTOFMEMORY;
 638 }
 639
 640 static HRESULT compile_call_expression(compiler_ctx_t *ctx, call_expression_t *expr, BOOL emit_ret)
 641 {
 642     unsigned arg_cnt = 0, extra_args;
 643     argument_t *arg;
 644     unsigned instr;
 645     jsop_t op;
 646     HRESULT hres;
 647
 648     if(is_memberid_expr(expr->expression->type)) {
 649         op = OP_call_member;
 650         extra_args = 2;
 651         hres = compile_memberid_expression(ctx, expr->expression, 0);
 652     }else {
 653         op = OP_call;
 654         extra_args = 1;
 655         hres = compile_expression(ctx, expr->expression, TRUE);
 656     }
 657
 658     if(FAILED(hres))
 659         return hres;
 660
 661     for(arg = expr->argument_list; arg; arg = arg->next) {
 662         hres = compile_expression(ctx, arg->expr, TRUE);
 663         if(FAILED(hres))
 664             return hres;
 665         arg_cnt++;
 666     }
 667
 668     instr = push_instr(ctx, op);
 669     if(!instr)
 670         return E_OUTOFMEMORY;
 671
 672     instr_ptr(ctx, instr)->u.arg[0].uint = arg_cnt;
 673     instr_ptr(ctx, instr)->u.arg[1].lng = emit_ret;
 674
 675     hres = push_instr_uint(ctx, OP_pop, arg_cnt + extra_args);
 676     if(FAILED(hres))
 677         return hres;
 678
 679     return !emit_ret || push_instr(ctx, OP_push_acc) ? S_OK : E_OUTOFMEMORY;
 680 }
 681
 682 static HRESULT compile_delete_expression(compiler_ctx_t *ctx, unary_expression_t *expr)
 683 {
 684     HRESULT hres;
 685
 686     switch(expr->expression->type) {
 687     case EXPR_ARRAY: {
 688         binary_expression_t *array_expr = (binary_expression_t*)expr->expression;
 689
 690         hres = compile_expression(ctx, array_expr->expression1, TRUE);
 691         if(FAILED(hres))
 692             return hres;
 693
 694         hres = compile_expression(ctx, array_expr->expression2, TRUE);
 695         if(FAILED(hres))
 696             return hres;
 697
 698         if(!push_instr(ctx, OP_delete))
 699             return E_OUTOFMEMORY;
 700         break;
 701     }
 702     case EXPR_MEMBER: {
 703         member_expression_t *member_expr = (member_expression_t*)expr->expression;
 704         jsstr_t *jsstr;
 705
 706         hres = compile_expression(ctx, member_expr->expression, TRUE);
 707         if(FAILED(hres))
 708             return hres;
 709
 710         /* FIXME: Potential optimization */
 711         jsstr = compiler_alloc_string(ctx, member_expr->identifier);
 712         if(!jsstr)
 713             return E_OUTOFMEMORY;
 714
 715         hres = push_instr_str(ctx, OP_str, jsstr);
 716         if(FAILED(hres))
 717             return hres;
 718
 719         if(!push_instr(ctx, OP_delete))
 720             return E_OUTOFMEMORY;
 721         break;
 722     }
 723     case EXPR_IDENT:
 724         return push_instr_bstr(ctx, OP_delete_ident, ((identifier_expression_t*)expr->expression)->identifier);
 725     default: {
 726         WARN("invalid delete, unimplemented exception message\n");
 727
 728         hres = compile_expression(ctx, expr->expression, TRUE);
 729         if(FAILED(hres))
 730             return hres;
 731
 732         return push_instr_uint_str(ctx, OP_throw_type, JS_E_INVALID_DELETE, L"FIXME");
 733     }
 734     }
 735
 736     return S_OK;
 737 }
 738
 739 static HRESULT compile_assign_expression(compiler_ctx_t *ctx, binary_expression_t *expr, jsop_t op)
 740 {
 741     jsop_t assign_op = OP_throw_ref;
 742     unsigned arg_cnt = 0;
 743     HRESULT hres;
 744
 745     if(expr->expression1->type == EXPR_CALL) {
 746         call_expression_t *call_expr = (call_expression_t*)expr->expression1;
 747         argument_t *arg;
 748
 749         if(is_memberid_expr(call_expr->expression->type) && call_expr->argument_list) {
 750             hres = compile_memberid_expression(ctx, call_expr->expression, fdexNameEnsure);
 751             if(FAILED(hres))
 752                 return hres;
 753
 754             for(arg = call_expr->argument_list; arg; arg = arg->next) {
 755                 hres = compile_expression(ctx, arg->expr, TRUE);
 756                 if(FAILED(hres))
 757                     return hres;
 758                 arg_cnt++;
 759             }
 760
 761             if(op != OP_LAST) {
 762                 unsigned instr;
 763
 764                 /* We need to call the functions twice: to get the value and to set it.
 765                  * JavaScript interpreted functions may to modify value on the stack,
 766                  * but assignment calls are allowed only on external functions, so we
 767                  * may reuse the stack here. */
 768                 instr = push_instr(ctx, OP_call_member);
 769                 if(!instr)
 770                     return E_OUTOFMEMORY;
 771                 instr_ptr(ctx, instr)->u.arg[0].uint = arg_cnt;
 772                 instr_ptr(ctx, instr)->u.arg[1].lng = 1;
 773
 774                 if(!push_instr(ctx, OP_push_acc))
 775                     return E_OUTOFMEMORY;
 776             }
 777             assign_op = OP_assign_call;
 778         }
 779     }else if(is_memberid_expr(expr->expression1->type)) {
 780         if(op != OP_LAST || expr->expression1->type == EXPR_IDENT) {
 781             hres = compile_memberid_expression(ctx, expr->expression1, fdexNameEnsure);
 782             if(FAILED(hres))
 783                 return hres;
 784             if(op != OP_LAST && !push_instr(ctx, OP_refval))
 785                 return E_OUTOFMEMORY;
 786             assign_op = OP_assign;
 787         }else {
 788             hres = emit_member_expression(ctx, expr->expression1);
 789             if(FAILED(hres))
 790                 return hres;
 791             assign_op = OP_set_member;
 792         }
 793     }
 794
 795     if(assign_op == OP_throw_ref) {
 796         /* Illegal assignment: evaluate and throw */
 797         hres = compile_expression(ctx, expr->expression1, TRUE);
 798         if(FAILED(hres))
 799             return hres;
 800         arg_cnt = JS_E_ILLEGAL_ASSIGN;
 801     }
 802
 803     hres = compile_expression(ctx, expr->expression2, TRUE);
 804     if(FAILED(hres))
 805         return hres;
 806
 807     if(op != OP_LAST && !push_instr(ctx, op))
 808         return E_OUTOFMEMORY;
 809
 810     return push_instr_uint(ctx, assign_op, arg_cnt);
 811 }
 812
 813 static HRESULT compile_typeof_expression(compiler_ctx_t *ctx, unary_expression_t *expr)
 814 {
 815     jsop_t op;
 816     HRESULT hres;
 817
 818     if(is_memberid_expr(expr->expression->type)) {
 819         if(expr->expression->type == EXPR_IDENT)
 820             return push_instr_bstr(ctx, OP_typeofident, ((identifier_expression_t*)expr->expression)->identifier);
 821
 822         op = OP_typeofid;
 823         hres = compile_memberid_expression(ctx, expr->expression, 0);
 824     }else {
 825         op = OP_typeof;
 826         hres = compile_expression(ctx, expr->expression, TRUE);
 827     }
 828     if(FAILED(hres))
 829         return hres;
 830
 831     return push_instr(ctx, op) ? S_OK : E_OUTOFMEMORY;
 832 }
 833
 834 static HRESULT compile_literal(compiler_ctx_t *ctx, literal_t *literal)
 835 {
 836     switch(literal->type) {
 837     case LT_BOOL:
 838         return push_instr_int(ctx, OP_bool, literal->u.bval);
 839     case LT_DOUBLE:
 840         return push_instr_double(ctx, OP_double, literal->u.dval);
 841     case LT_NULL:
 842         return push_instr(ctx, OP_null) ? S_OK : E_OUTOFMEMORY;
 843     case LT_STRING:
 844         return push_instr_str(ctx, OP_str, literal->u.str);
 845     case LT_REGEXP:
 846         return push_instr_str_uint(ctx, OP_regexp, literal->u.regexp.str, literal->u.regexp.flags);
 847     DEFAULT_UNREACHABLE;
 848     }
 849     return E_FAIL;
 850 }
 851
 852 static HRESULT literal_as_string(compiler_ctx_t *ctx, literal_t *literal, jsstr_t **str)
 853 {
 854     switch(literal->type) {
 855     case LT_STRING:
 856         *str = literal->u.str;
 857         break;
 858     case LT_DOUBLE:
 859         return double_to_string(literal->u.dval, str);
 860     DEFAULT_UNREACHABLE;
 861     }
 862
 863     return *str ? S_OK : E_OUTOFMEMORY;
 864 }
 865
 866 static HRESULT compile_array_literal(compiler_ctx_t *ctx, array_literal_expression_t *expr)
 867 {
 868     unsigned length = 0;
 869     array_element_t *iter;
 870     unsigned array_instr;
 871     HRESULT hres;
 872
 873     array_instr = push_instr(ctx, OP_carray);
 874
 875     for(iter = expr->element_list; iter; iter = iter->next) {
 876         length += iter->elision;
 877
 878         hres = compile_expression(ctx, iter->expr, TRUE);
 879         if(FAILED(hres))
 880             return hres;
 881
 882         hres = push_instr_uint(ctx, OP_carray_set, length);
 883         if(FAILED(hres))
 884             return hres;
 885
 886         length++;
 887     }
 888
 889     instr_ptr(ctx, array_instr)->u.arg[0].uint = length + expr->length;
 890     return S_OK;
 891 }
 892
 893 static HRESULT compile_object_literal(compiler_ctx_t *ctx, property_value_expression_t *expr)
 894 {
 895     property_definition_t *iter;
 896     jsstr_t *name;
 897     HRESULT hres;
 898
 899     if(!push_instr(ctx, OP_new_obj))
 900         return E_OUTOFMEMORY;
 901
 902     for(iter = expr->property_list; iter; iter = iter->next) {
 903         hres = literal_as_string(ctx, iter->name, &name);
 904         if(FAILED(hres))
 905             return hres;
 906
 907         hres = compile_expression(ctx, iter->value, TRUE);
 908         if(FAILED(hres))
 909             return hres;
 910
 911         hres = push_instr_str_uint(ctx, OP_obj_prop, name, iter->type);
 912         if(FAILED(hres))
 913             return hres;
 914     }
 915
 916     return S_OK;
 917 }
 918
 919 static HRESULT compile_function_expression(compiler_ctx_t *ctx, function_expression_t *expr, BOOL emit_ret)
 920 {
 921     return emit_ret ? push_instr_uint(ctx, OP_func, expr->func_id) : S_OK;
 922 }
 923
 924 static HRESULT compile_expression(compiler_ctx_t *ctx, expression_t *expr, BOOL emit_ret)
 925 {
 926     HRESULT hres;
 927
 928     switch(expr->type) {
 929     case EXPR_ADD:
 930         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_add);
 931         break;
 932     case EXPR_AND:
 933         hres = compile_logical_expression(ctx, (binary_expression_t*)expr, OP_cnd_z);
 934         break;
 935     case EXPR_ARRAY:
 936         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_array);
 937         break;
 938     case EXPR_ARRAYLIT:
 939         hres = compile_array_literal(ctx, (array_literal_expression_t*)expr);
 940         break;
 941     case EXPR_ASSIGN:
 942         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_LAST);
 943         break;
 944     case EXPR_ASSIGNADD:
 945         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_add);
 946         break;
 947     case EXPR_ASSIGNAND:
 948         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_and);
 949         break;
 950     case EXPR_ASSIGNSUB:
 951         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_sub);
 952         break;
 953     case EXPR_ASSIGNMUL:
 954         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_mul);
 955         break;
 956     case EXPR_ASSIGNDIV:
 957         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_div);
 958         break;
 959     case EXPR_ASSIGNMOD:
 960         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_mod);
 961         break;
 962     case EXPR_ASSIGNOR:
 963         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_or);
 964         break;
 965     case EXPR_ASSIGNLSHIFT:
 966         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_lshift);
 967         break;
 968     case EXPR_ASSIGNRSHIFT:
 969         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_rshift);
 970         break;
 971     case EXPR_ASSIGNRRSHIFT:
 972         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_rshift2);
 973         break;
 974     case EXPR_ASSIGNXOR:
 975         hres = compile_assign_expression(ctx, (binary_expression_t*)expr, OP_xor);
 976         break;
 977     case EXPR_BAND:
 978         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_and);
 979         break;
 980     case EXPR_BITNEG:
 981         hres = compile_unary_expression(ctx, (unary_expression_t*)expr, OP_bneg);
 982         break;
 983     case EXPR_BOR:
 984         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_or);
 985         break;
 986     case EXPR_CALL:
 987         return compile_call_expression(ctx, (call_expression_t*)expr, emit_ret);
 988     case EXPR_COMMA:
 989         return compile_comma_expression(ctx, (binary_expression_t*)expr, emit_ret);
 990     case EXPR_COND:
 991         hres = compile_conditional_expression(ctx, (conditional_expression_t*)expr);
 992         break;
 993     case EXPR_DELETE:
 994         hres = compile_delete_expression(ctx, (unary_expression_t*)expr);
 995         break;
 996     case EXPR_DIV:
 997         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_div);
 998         break;
 999     case EXPR_EQ:
1000         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_eq);
1001         break;
1002     case EXPR_EQEQ:
1003         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_eq2);
1004         break;
1005     case EXPR_FUNC:
1006         return compile_function_expression(ctx, (function_expression_t*)expr, emit_ret);
1007     case EXPR_GREATER:
1008         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_gt);
1009         break;
1010     case EXPR_GREATEREQ:
1011         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_gteq);
1012         break;
1013     case EXPR_IDENT:
1014         hres = emit_identifier(ctx, ((identifier_expression_t*)expr)->identifier);
1015         break;
1016     case EXPR_IN:
1017         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_in);
1018         break;
1019     case EXPR_INSTANCEOF:
1020         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_instanceof);
1021         break;
1022     case EXPR_LESS:
1023         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_lt);
1024         break;
1025     case EXPR_LESSEQ:
1026         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_lteq);
1027         break;
1028     case EXPR_LITERAL:
1029         hres = compile_literal(ctx, ((literal_expression_t*)expr)->literal);
1030         break;
1031     case EXPR_LOGNEG:
1032         hres = compile_unary_expression(ctx, (unary_expression_t*)expr, OP_neg);
1033         break;
1034     case EXPR_LSHIFT:
1035         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_lshift);
1036         break;
1037     case EXPR_MEMBER:
1038         hres = compile_member_expression(ctx, (member_expression_t*)expr);
1039         break;
1040     case EXPR_MINUS:
1041         hres = compile_unary_expression(ctx, (unary_expression_t*)expr, OP_minus);
1042         break;
1043     case EXPR_MOD:
1044         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_mod);
1045         break;
1046     case EXPR_MUL:
1047         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_mul);
1048         break;
1049     case EXPR_NEW:
1050         hres = compile_new_expression(ctx, (call_expression_t*)expr);
1051         break;
1052     case EXPR_NOTEQ:
1053         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_neq);
1054         break;
1055     case EXPR_NOTEQEQ:
1056         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_neq2);
1057         break;
1058     case EXPR_OR:
1059         hres = compile_logical_expression(ctx, (binary_expression_t*)expr, OP_cnd_nz);
1060         break;
1061     case EXPR_PLUS:
1062         hres = compile_unary_expression(ctx, (unary_expression_t*)expr, OP_tonum);
1063         break;
1064     case EXPR_POSTDEC:
1065         hres = compile_increment_expression(ctx, (unary_expression_t*)expr, OP_postinc, -1);
1066         break;
1067     case EXPR_POSTINC:
1068         hres = compile_increment_expression(ctx, (unary_expression_t*)expr, OP_postinc, 1);
1069         break;
1070     case EXPR_PREDEC:
1071         hres = compile_increment_expression(ctx, (unary_expression_t*)expr, OP_preinc, -1);
1072         break;
1073     case EXPR_PREINC:
1074         hres = compile_increment_expression(ctx, (unary_expression_t*)expr, OP_preinc, 1);
1075         break;
1076     case EXPR_PROPVAL:
1077         hres = compile_object_literal(ctx, (property_value_expression_t*)expr);
1078         break;
1079     case EXPR_RSHIFT:
1080         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_rshift);
1081         break;
1082     case EXPR_RRSHIFT:
1083         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_rshift2);
1084         break;
1085     case EXPR_SUB:
1086         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_sub);
1087         break;
1088     case EXPR_THIS:
1089         return !emit_ret || push_instr(ctx, OP_this) ? S_OK : E_OUTOFMEMORY;
1090     case EXPR_TYPEOF:
1091         hres = compile_typeof_expression(ctx, (unary_expression_t*)expr);
1092         break;
1093     case EXPR_VOID:
1094         hres = compile_unary_expression(ctx, (unary_expression_t*)expr, OP_void);
1095         break;
1096     case EXPR_BXOR:
1097         hres = compile_binary_expression(ctx, (binary_expression_t*)expr, OP_xor);
1098         break;
1099     DEFAULT_UNREACHABLE;
1100     }
1101
1102     if(FAILED(hres))
1103         return hres;
1104
1105     return emit_ret ? S_OK : push_instr_uint(ctx, OP_pop, 1);
1106 }
1107
1108 static inline BOOL is_loop_statement(statement_type_t type)
1109 {
1110     return type == STAT_FOR || type == STAT_FORIN || type == STAT_WHILE;
1111 }
1112
1113 /* ECMA-262 3rd Edition    12.1 */
1114 static HRESULT compile_block_statement(compiler_ctx_t *ctx, statement_t *iter)
1115 {
1116     HRESULT hres;
1117
1118     while(iter) {
1119         hres = compile_statement(ctx, NULL, iter);
1120         if(FAILED(hres))
1121             return hres;
1122
1123         iter = iter->next;
1124     }
1125
1126     return S_OK;
1127 }
1128
1129 /* ECMA-262 3rd Edition    12.2 */
1130 static HRESULT compile_variable_list(compiler_ctx_t *ctx, variable_declaration_t *list)
1131 {
1132     variable_declaration_t *iter;
1133     HRESULT hres;
1134
1135     assert(list != NULL);
1136
1137     for(iter = list; iter; iter = iter->next) {
1138         if(!iter->expr)
1139             continue;
1140
1141         hres = emit_identifier_ref(ctx, iter->identifier, 0);
1142         if(FAILED(hres))
1143             return hres;
1144
1145         hres = compile_expression(ctx, iter->expr, TRUE);
1146         if(FAILED(hres))
1147             return hres;
1148
1149         if(!push_instr(ctx, OP_assign))
1150             return E_OUTOFMEMORY;
1151
1152         hres = push_instr_uint(ctx, OP_pop, 1);
1153         if(FAILED(hres))
1154             return hres;
1155     }
1156
1157     return S_OK;
1158 }
1159
1160 /* ECMA-262 3rd Edition    12.2 */
1161 static HRESULT compile_var_statement(compiler_ctx_t *ctx, var_statement_t *stat)
1162 {
1163     return compile_variable_list(ctx, stat->variable_list);
1164 }
1165
1166 /* ECMA-262 3rd Edition    12.4 */
1167 static HRESULT compile_expression_statement(compiler_ctx_t *ctx, expression_statement_t *stat)
1168 {
1169     HRESULT hres;
1170
1171     hres = compile_expression(ctx, stat->expr, ctx->from_eval);
1172     if(FAILED(hres))
1173         return hres;
1174
1175     return !ctx->from_eval || push_instr(ctx, OP_setret) ? S_OK : E_OUTOFMEMORY;
1176 }
1177
1178 /* ECMA-262 3rd Edition    12.5 */
1179 static HRESULT compile_if_statement(compiler_ctx_t *ctx, if_statement_t *stat)
1180 {
1181     unsigned jmp_else;
1182     HRESULT hres;
1183
1184     hres = compile_expression(ctx, stat->expr, TRUE);
1185     if(FAILED(hres))
1186         return hres;
1187
1188     jmp_else = push_instr(ctx, OP_jmp_z);
1189     if(!jmp_else)
1190         return E_OUTOFMEMORY;
1191
1192     hres = compile_statement(ctx, NULL, stat->if_stat);
1193     if(FAILED(hres))
1194         return hres;
1195
1196     if(stat->else_stat) {
1197         unsigned jmp_end;
1198
1199         jmp_end = push_instr(ctx, OP_jmp);
1200         if(!jmp_end)
1201             return E_OUTOFMEMORY;
1202
1203         set_arg_uint(ctx, jmp_else, ctx->code_off);
1204
1205         hres = compile_statement(ctx, NULL, stat->else_stat);
1206         if(FAILED(hres))
1207             return hres;
1208
1209         set_arg_uint(ctx, jmp_end, ctx->code_off);
1210     }else {
1211         set_arg_uint(ctx, jmp_else, ctx->code_off);
1212     }
1213
1214     return S_OK;
1215 }
1216
1217 /* ECMA-262 3rd Edition    12.6.2 */
1218 static HRESULT compile_while_statement(compiler_ctx_t *ctx, while_statement_t *stat)
1219 {
1220     statement_ctx_t stat_ctx = {0, FALSE, FALSE};
1221     unsigned jmp_off;
1222     HRESULT hres;
1223
1224     stat_ctx.break_label = alloc_label(ctx);
1225     if(!stat_ctx.break_label)
1226         return E_OUTOFMEMORY;
1227
1228     stat_ctx.continue_label = alloc_label(ctx);
1229     if(!stat_ctx.continue_label)
1230         return E_OUTOFMEMORY;
1231
1232     jmp_off = ctx->code_off;
1233
1234     if(!stat->do_while) {
1235         label_set_addr(ctx, stat_ctx.continue_label);
1236         hres = compile_expression(ctx, stat->expr, TRUE);
1237         if(FAILED(hres))
1238             return hres;
1239
1240         hres = push_instr_uint(ctx, OP_jmp_z, stat_ctx.break_label);
1241         if(FAILED(hres))
1242             return hres;
1243     }
1244
1245     hres = compile_statement(ctx, &stat_ctx, stat->statement);
1246     if(FAILED(hres))
1247         return hres;
1248
1249     set_compiler_loc(ctx, stat->stat.loc);
1250     if(stat->do_while) {
1251         label_set_addr(ctx, stat_ctx.continue_label);
1252         hres = compile_expression(ctx, stat->expr, TRUE);
1253         if(FAILED(hres))
1254             return hres;
1255
1256         hres = push_instr_uint(ctx, OP_jmp_z, stat_ctx.break_label);
1257         if(FAILED(hres))
1258             return hres;
1259     }
1260
1261     hres = push_instr_uint(ctx, OP_jmp, jmp_off);
1262     if(FAILED(hres))
1263         return hres;
1264
1265     label_set_addr(ctx, stat_ctx.break_label);
1266     return S_OK;
1267 }
1268
1269 /* ECMA-262 3rd Edition    12.6.3 */
1270 static HRESULT compile_for_statement(compiler_ctx_t *ctx, for_statement_t *stat)
1271 {
1272     statement_ctx_t stat_ctx = {0, FALSE, FALSE};
1273     unsigned expr_off;
1274     HRESULT hres;
1275
1276     if(stat->variable_list) {
1277         hres = compile_variable_list(ctx, stat->variable_list);
1278         if(FAILED(hres))
1279             return hres;
1280     }else if(stat->begin_expr) {
1281         hres = compile_expression(ctx, stat->begin_expr, FALSE);
1282         if(FAILED(hres))
1283             return hres;
1284     }
1285
1286     stat_ctx.break_label = alloc_label(ctx);
1287     if(!stat_ctx.break_label)
1288         return E_OUTOFMEMORY;
1289
1290     stat_ctx.continue_label = alloc_label(ctx);
1291     if(!stat_ctx.continue_label)
1292         return E_OUTOFMEMORY;
1293
1294     expr_off = ctx->code_off;
1295
1296     if(stat->expr) {
1297         set_compiler_loc(ctx, stat->expr_loc);
1298         hres = compile_expression(ctx, stat->expr, TRUE);
1299         if(FAILED(hres))
1300             return hres;
1301
1302         hres = push_instr_uint(ctx, OP_jmp_z, stat_ctx.break_label);
1303         if(FAILED(hres))
1304             return hres;
1305     }
1306
1307     hres = compile_statement(ctx, &stat_ctx, stat->statement);
1308     if(FAILED(hres))
1309         return hres;
1310
1311     label_set_addr(ctx, stat_ctx.continue_label);
1312
1313     if(stat->end_expr) {
1314         set_compiler_loc(ctx, stat->end_loc);
1315         hres = compile_expression(ctx, stat->end_expr, FALSE);
1316         if(FAILED(hres))
1317             return hres;
1318     }
1319
1320     hres = push_instr_uint(ctx, OP_jmp, expr_off);
1321     if(FAILED(hres))
1322         return hres;
1323
1324     label_set_addr(ctx, stat_ctx.break_label);
1325     return S_OK;
1326 }
1327
1328 /* ECMA-262 3rd Edition    12.6.4 */
1329 static HRESULT compile_forin_statement(compiler_ctx_t *ctx, forin_statement_t *stat)
1330 {
1331     statement_ctx_t stat_ctx = {4, FALSE, FALSE};
1332     HRESULT hres;
1333
1334     if(stat->variable) {
1335         hres = compile_variable_list(ctx, stat->variable);
1336         if(FAILED(hres))
1337             return hres;
1338     }
1339
1340     stat_ctx.break_label = alloc_label(ctx);
1341     if(!stat_ctx.break_label)
1342         return E_OUTOFMEMORY;
1343
1344     stat_ctx.continue_label = alloc_label(ctx);
1345     if(!stat_ctx.continue_label)
1346         return E_OUTOFMEMORY;
1347
1348     hres = compile_expression(ctx, stat->in_expr, TRUE);
1349     if(FAILED(hres))
1350         return hres;
1351
1352     if(stat->variable) {
1353         hres = emit_identifier_ref(ctx, stat->variable->identifier, fdexNameEnsure);
1354         if(FAILED(hres))
1355             return hres;
1356     }else if(is_memberid_expr(stat->expr->type)) {
1357         hres = compile_memberid_expression(ctx, stat->expr, fdexNameEnsure);
1358         if(FAILED(hres))
1359             return hres;
1360     }else {
1361         hres = push_instr_uint(ctx, OP_throw_ref, JS_E_ILLEGAL_ASSIGN);
1362         if(FAILED(hres))
1363             return hres;
1364
1365         /* FIXME: compile statement anyways when we depend on compiler to check errors */
1366         return S_OK;
1367     }
1368
1369     hres = push_instr_int(ctx, OP_int, DISPID_STARTENUM);
1370     if(FAILED(hres))
1371         return hres;
1372
1373     label_set_addr(ctx, stat_ctx.continue_label);
1374     hres = push_instr_uint(ctx, OP_forin, stat_ctx.break_label);
1375     if(FAILED(hres))
1376         return E_OUTOFMEMORY;
1377
1378     hres = compile_statement(ctx, &stat_ctx, stat->statement);
1379     if(FAILED(hres))
1380         return hres;
1381
1382     hres = push_instr_uint(ctx, OP_jmp, stat_ctx.continue_label);
1383     if(FAILED(hres))
1384         return hres;
1385
1386     label_set_addr(ctx, stat_ctx.break_label);
1387     return S_OK;
1388 }
1389
1390 static HRESULT pop_to_stat(compiler_ctx_t *ctx, statement_ctx_t *stat_ctx)
1391 {
1392     unsigned stack_pop = 0;
1393     statement_ctx_t *iter;
1394     HRESULT hres;
1395
1396     for(iter = ctx->stat_ctx; iter != stat_ctx; iter = iter->next) {
1397         if(iter->using_scope && !push_instr(ctx, OP_pop_scope))
1398             return E_OUTOFMEMORY;
1399         if(iter->using_except) {
1400             if(stack_pop) {
1401                 hres = push_instr_uint(ctx, OP_pop, stack_pop);
1402                 if(FAILED(hres))
1403                     return hres;
1404                 stack_pop = 0;
1405             }
1406             hres = push_instr_uint(ctx, OP_pop_except, ctx->code_off+1);
1407             if(FAILED(hres))
1408                 return hres;
1409         }
1410         stack_pop += iter->stack_use;
1411     }
1412
1413     if(stack_pop) {
1414         hres = push_instr_uint(ctx, OP_pop, stack_pop);
1415         if(FAILED(hres))
1416             return hres;
1417     }
1418
1419     return S_OK;
1420 }
1421
1422 /* ECMA-262 3rd Edition    12.7 */
1423 static HRESULT compile_continue_statement(compiler_ctx_t *ctx, branch_statement_t *stat)
1424 {
1425     statement_ctx_t *pop_ctx;
1426     HRESULT hres;
1427
1428     if(stat->identifier) {
1429         statement_t *label_stat;
1430         statement_ctx_t *iter;
1431
1432         pop_ctx = NULL;
1433
1434         for(iter = ctx->stat_ctx; iter; iter = iter->next) {
1435             if(iter->continue_label)
1436                 pop_ctx = iter;
1437             if(iter->labelled_stat && !wcscmp(iter->labelled_stat->identifier, stat->identifier))
1438                 break;
1439         }
1440
1441         if(!iter) {
1442             WARN("Label not found\n");
1443             return JS_E_LABEL_NOT_FOUND;
1444         }
1445
1446         /* Labelled continue are allowed only on loops */
1447         for(label_stat = iter->labelled_stat->statement;
1448             label_stat->type == STAT_LABEL;
1449             label_stat = ((labelled_statement_t*)label_stat)->statement);
1450         if(!is_loop_statement(label_stat->type)) {
1451             WARN("Label is not a loop\n");
1452             return JS_E_INVALID_CONTINUE;
1453         }
1454
1455         assert(pop_ctx != NULL);
1456     }else {
1457         for(pop_ctx = ctx->stat_ctx; pop_ctx; pop_ctx = pop_ctx->next) {
1458             if(pop_ctx->continue_label)
1459                 break;
1460         }
1461
1462         if(!pop_ctx) {
1463             WARN("continue outside loop\n");
1464             return JS_E_INVALID_CONTINUE;
1465         }
1466     }
1467
1468     hres = pop_to_stat(ctx, pop_ctx);
1469     if(FAILED(hres))
1470         return hres;
1471
1472     return push_instr_uint(ctx, OP_jmp, pop_ctx->continue_label);
1473 }
1474
1475 /* ECMA-262 3rd Edition    12.8 */
1476 static HRESULT compile_break_statement(compiler_ctx_t *ctx, branch_statement_t *stat)
1477 {
1478     statement_ctx_t *pop_ctx;
1479     HRESULT hres;
1480
1481     if(stat->identifier) {
1482         for(pop_ctx = ctx->stat_ctx; pop_ctx; pop_ctx = pop_ctx->next) {
1483             if(pop_ctx->labelled_stat && !wcscmp(pop_ctx->labelled_stat->identifier, stat->identifier)) {
1484                 assert(pop_ctx->break_label);
1485                 break;
1486             }
1487         }
1488
1489         if(!pop_ctx) {
1490             WARN("Label not found\n");
1491             return JS_E_LABEL_NOT_FOUND;
1492         }
1493     }else {
1494         for(pop_ctx = ctx->stat_ctx; pop_ctx; pop_ctx = pop_ctx->next) {
1495             if(pop_ctx->break_label && !pop_ctx->labelled_stat)
1496                 break;
1497         }
1498
1499         if(!pop_ctx) {
1500             WARN("Break outside loop\n");
1501             return JS_E_INVALID_BREAK;
1502         }
1503     }
1504
1505     hres = pop_to_stat(ctx, pop_ctx->next);
1506     if(FAILED(hres))
1507         return hres;
1508
1509     return push_instr_uint(ctx, OP_jmp, pop_ctx->break_label);
1510 }
1511
1512 /* ECMA-262 3rd Edition    12.9 */
1513 static HRESULT compile_return_statement(compiler_ctx_t *ctx, expression_statement_t *stat)
1514 {
1515     HRESULT hres;
1516
1517     if(ctx->from_eval) {
1518         WARN("misplaced return statement\n");
1519         return JS_E_MISPLACED_RETURN;
1520     }
1521
1522     if(stat->expr) {
1523         hres = compile_expression(ctx, stat->expr, TRUE);
1524         if(FAILED(hres))
1525             return hres;
1526         if(!push_instr(ctx, OP_setret))
1527             return E_OUTOFMEMORY;
1528     }
1529
1530     hres = pop_to_stat(ctx, NULL);
1531     if(FAILED(hres))
1532         return hres;
1533
1534     return push_instr_uint(ctx, OP_ret, !stat->expr);
1535 }
1536
1537 /* ECMA-262 3rd Edition    12.10 */
1538 static HRESULT compile_with_statement(compiler_ctx_t *ctx, with_statement_t *stat)
1539 {
1540     statement_ctx_t stat_ctx = {0, TRUE, FALSE};
1541     HRESULT hres;
1542
1543     hres = compile_expression(ctx, stat->expr, TRUE);
1544     if(FAILED(hres))
1545         return hres;
1546
1547     if(!push_instr(ctx, OP_push_scope))
1548         return E_OUTOFMEMORY;
1549
1550     hres = compile_statement(ctx, &stat_ctx, stat->statement);
1551     if(FAILED(hres))
1552         return hres;
1553
1554     if(!push_instr(ctx, OP_pop_scope))
1555         return E_OUTOFMEMORY;
1556
1557     return S_OK;
1558 }
1559
1560 /* ECMA-262 3rd Edition    12.10 */
1561 static HRESULT compile_labelled_statement(compiler_ctx_t *ctx, labelled_statement_t *stat)
1562 {
1563     statement_ctx_t stat_ctx = {0, FALSE, FALSE, 0, 0, stat}, *iter;
1564     HRESULT hres;
1565
1566     for(iter = ctx->stat_ctx; iter; iter = iter->next) {
1567         if(iter->labelled_stat && !wcscmp(iter->labelled_stat->identifier, stat->identifier)) {
1568             WARN("Label %s redefined\n", debugstr_w(stat->identifier));
1569             return JS_E_LABEL_REDEFINED;
1570         }
1571     }
1572
1573     /* Labelled breaks are allowed for any labelled statements, not only loops (violating spec) */
1574     stat_ctx.break_label = alloc_label(ctx);
1575     if(!stat_ctx.break_label)
1576         return E_OUTOFMEMORY;
1577
1578     hres = compile_statement(ctx, &stat_ctx, stat->statement);
1579     if(FAILED(hres))
1580         return hres;
1581
1582     label_set_addr(ctx, stat_ctx.break_label);
1583     return S_OK;
1584 }
1585
1586 /* ECMA-262 3rd Edition    12.13 */
1587 static HRESULT compile_switch_statement(compiler_ctx_t *ctx, switch_statement_t *stat)
1588 {
1589     statement_ctx_t stat_ctx = {0, FALSE, FALSE};
1590     unsigned case_cnt = 0, *case_jmps, i, default_jmp;
1591     BOOL have_default = FALSE;
1592     statement_t *stat_iter;
1593     case_clausule_t *iter;
1594     HRESULT hres;
1595
1596     hres = compile_expression(ctx, stat->expr, TRUE);
1597     if(FAILED(hres))
1598         return hres;
1599
1600     stat_ctx.break_label = alloc_label(ctx);
1601     if(!stat_ctx.break_label)
1602         return E_OUTOFMEMORY;
1603
1604     for(iter = stat->case_list; iter; iter = iter->next) {
1605         if(iter->expr)
1606             case_cnt++;
1607     }
1608
1609     case_jmps = heap_alloc(case_cnt * sizeof(*case_jmps));
1610     if(!case_jmps)
1611         return E_OUTOFMEMORY;
1612
1613     i = 0;
1614     for(iter = stat->case_list; iter; iter = iter->next) {
1615         if(!iter->expr) {
1616             have_default = TRUE;
1617             continue;
1618         }
1619
1620         set_compiler_loc(ctx, iter->loc);
1621         hres = compile_expression(ctx, iter->expr, TRUE);
1622         if(FAILED(hres))
1623             break;
1624
1625         case_jmps[i] = push_instr(ctx, OP_case);
1626         if(!case_jmps[i]) {
1627             hres = E_OUTOFMEMORY;
1628             break;
1629         }
1630         i++;
1631     }
1632
1633     if(SUCCEEDED(hres)) {
1634         hres = push_instr_uint(ctx, OP_pop, 1);
1635         if(SUCCEEDED(hres)) {
1636             default_jmp = push_instr(ctx, OP_jmp);
1637             if(!default_jmp)
1638                 hres = E_OUTOFMEMORY;
1639         }
1640     }
1641
1642     if(FAILED(hres)) {
1643         heap_free(case_jmps);
1644         return hres;
1645     }
1646
1647     i = 0;
1648     for(iter = stat->case_list; iter; iter = iter->next) {
1649         while(iter->next && iter->next->stat == iter->stat) {
1650             set_arg_uint(ctx, iter->expr ? case_jmps[i++] : default_jmp, ctx->code_off);
1651             iter = iter->next;
1652         }
1653
1654         set_arg_uint(ctx, iter->expr ? case_jmps[i++] : default_jmp, ctx->code_off);
1655
1656         for(stat_iter = iter->stat; stat_iter && (!iter->next || iter->next->stat != stat_iter);
1657             stat_iter = stat_iter->next) {
1658             hres = compile_statement(ctx, &stat_ctx, stat_iter);
1659             if(FAILED(hres))
1660                 break;
1661         }
1662         if(FAILED(hres))
1663             break;
1664     }
1665
1666     heap_free(case_jmps);
1667     if(FAILED(hres))
1668         return hres;
1669     assert(i == case_cnt);
1670
1671     if(!have_default) {
1672         hres = push_instr_uint(ctx, OP_jmp, stat_ctx.break_label);
1673         if(FAILED(hres))
1674             return hres;
1675         set_arg_uint(ctx, default_jmp, ctx->code_off);
1676     }
1677
1678     label_set_addr(ctx, stat_ctx.break_label);
1679     return S_OK;
1680 }
1681
1682 /* ECMA-262 3rd Edition    12.13 */
1683 static HRESULT compile_throw_statement(compiler_ctx_t *ctx, expression_statement_t *stat)
1684 {
1685     HRESULT hres;
1686
1687     hres = compile_expression(ctx, stat->expr, TRUE);
1688     if(FAILED(hres))
1689         return hres;
1690
1691     return push_instr(ctx, OP_throw) ? S_OK : E_OUTOFMEMORY;
1692 }
1693
1694 /* ECMA-262 3rd Edition    12.14 */
1695 static HRESULT compile_try_statement(compiler_ctx_t *ctx, try_statement_t *stat)
1696 {
1697     statement_ctx_t try_ctx = {0, FALSE, TRUE}, finally_ctx = {2, FALSE, FALSE};
1698     unsigned push_except, finally_off = 0, catch_off = 0, pop_except, catch_pop_except = 0;
1699     BSTR ident;
1700     HRESULT hres;
1701
1702     push_except = push_instr(ctx, OP_push_except);
1703     if(!push_except)
1704         return E_OUTOFMEMORY;
1705
1706     if(stat->catch_block) {
1707         ident = compiler_alloc_bstr(ctx, stat->catch_block->identifier);
1708         if(!ident)
1709             return E_OUTOFMEMORY;
1710     }else {
1711         ident = NULL;
1712     }
1713
1714     hres = compile_statement(ctx, &try_ctx, stat->try_statement);
1715     if(FAILED(hres))
1716         return hres;
1717
1718     pop_except = push_instr(ctx, OP_pop_except);
1719     if(!pop_except)
1720         return E_OUTOFMEMORY;
1721
1722     if(stat->catch_block) {
1723         statement_ctx_t catch_ctx = {0, TRUE, stat->finally_statement != NULL};
1724
1725         if(stat->finally_statement)
1726             catch_ctx.using_except = TRUE;
1727
1728         catch_off = ctx->code_off;
1729
1730         hres = push_instr_bstr(ctx, OP_enter_catch, ident);
1731         if(FAILED(hres))
1732             return hres;
1733
1734         hres = compile_statement(ctx, &catch_ctx, stat->catch_block->statement);
1735         if(FAILED(hres))
1736             return hres;
1737
1738         if(!push_instr(ctx, OP_pop_scope))
1739             return E_OUTOFMEMORY;
1740
1741         if(stat->finally_statement) {
1742             catch_pop_except = push_instr(ctx, OP_pop_except);
1743             if(!catch_pop_except)
1744                 return E_OUTOFMEMORY;
1745         }
1746     }
1747
1748     if(stat->finally_statement) {
1749         /*
1750          * finally block expects two elements on the stack, which may be:
1751          * - (true, return_addr) set by OP_pop_except, OP_end_finally jumps back to passed address
1752          * - (false, exception_value) set when unwinding an exception, which OP_end_finally rethrows
1753          */
1754         finally_off = ctx->code_off;
1755         hres = compile_statement(ctx, &finally_ctx, stat->finally_statement);
1756         if(FAILED(hres))
1757             return hres;
1758
1759         set_compiler_loc(ctx, stat->finally_loc);
1760         if(!push_instr(ctx, OP_end_finally))
1761             return E_OUTOFMEMORY;
1762     }
1763
1764     instr_ptr(ctx, pop_except)->u.arg[0].uint = ctx->code_off;
1765     if(catch_pop_except)
1766         instr_ptr(ctx, catch_pop_except)->u.arg[0].uint = ctx->code_off;
1767     instr_ptr(ctx, push_except)->u.arg[0].uint = catch_off;
1768     instr_ptr(ctx, push_except)->u.arg[1].uint = finally_off;
1769     return S_OK;
1770 }
1771
1772 static HRESULT compile_statement(compiler_ctx_t *ctx, statement_ctx_t *stat_ctx, statement_t *stat)
1773 {
1774     HRESULT hres;
1775
1776     if(stat_ctx) {
1777         stat_ctx->next = ctx->stat_ctx;
1778         ctx->stat_ctx = stat_ctx;
1779     }
1780
1781     set_compiler_loc(ctx, stat->loc);
1782
1783     switch(stat->type) {
1784     case STAT_BLOCK:
1785         hres = compile_block_statement(ctx, ((block_statement_t*)stat)->stat_list);
1786         break;
1787     case STAT_BREAK:
1788         hres = compile_break_statement(ctx, (branch_statement_t*)stat);
1789         break;
1790     case STAT_CONTINUE:
1791         hres = compile_continue_statement(ctx, (branch_statement_t*)stat);
1792         break;
1793     case STAT_EMPTY:
1794         /* nothing to do */
1795         hres = S_OK;
1796         break;
1797     case STAT_EXPR:
1798         hres = compile_expression_statement(ctx, (expression_statement_t*)stat);
1799         break;
1800     case STAT_FOR:
1801         hres = compile_for_statement(ctx, (for_statement_t*)stat);
1802         break;
1803     case STAT_FORIN:
1804         hres = compile_forin_statement(ctx, (forin_statement_t*)stat);
1805         break;
1806     case STAT_IF:
1807         hres = compile_if_statement(ctx, (if_statement_t*)stat);
1808         break;
1809     case STAT_LABEL:
1810         hres = compile_labelled_statement(ctx, (labelled_statement_t*)stat);
1811         break;
1812     case STAT_RETURN:
1813         hres = compile_return_statement(ctx, (expression_statement_t*)stat);
1814         break;
1815     case STAT_SWITCH:
1816         hres = compile_switch_statement(ctx, (switch_statement_t*)stat);
1817         break;
1818     case STAT_THROW:
1819         hres = compile_throw_statement(ctx, (expression_statement_t*)stat);
1820         break;
1821     case STAT_TRY:
1822         hres = compile_try_statement(ctx, (try_statement_t*)stat);
1823         break;
1824     case STAT_VAR:
1825         hres = compile_var_statement(ctx, (var_statement_t*)stat);
1826         break;
1827     case STAT_WHILE:
1828         hres = compile_while_statement(ctx, (while_statement_t*)stat);
1829         break;
1830     case STAT_WITH:
1831         hres = compile_with_statement(ctx, (with_statement_t*)stat);
1832         break;
1833     DEFAULT_UNREACHABLE;
1834     }
1835
1836     if(stat_ctx) {
1837         assert(ctx->stat_ctx == stat_ctx);
1838         ctx->stat_ctx = stat_ctx->next;
1839     }
1840
1841     return hres;
1842 }
1843
1844 static int function_local_cmp(const void *key, const struct wine_rb_entry *entry)
1845 {
1846     function_local_t *local = WINE_RB_ENTRY_VALUE(entry, function_local_t, entry);
1847     return wcscmp(key, local->name);
1848 }
1849
1850 static inline function_local_t *find_local(compiler_ctx_t *ctx, const WCHAR *name)
1851 {
1852     struct wine_rb_entry *entry = wine_rb_get(&ctx->locals, name);
1853     return entry ? WINE_RB_ENTRY_VALUE(entry, function_local_t, entry) : NULL;
1854 }
1855
1856 static BOOL alloc_local(compiler_ctx_t *ctx, BSTR name, int ref)
1857 {
1858     function_local_t *local;
1859
1860     local = heap_pool_alloc(&ctx->heap, sizeof(*local));
1861     if(!local)
1862         return FALSE;
1863
1864     local->name = name;
1865     local->ref = ref;
1866     wine_rb_put(&ctx->locals, name, &local->entry);
1867     ctx->locals_cnt++;
1868     return TRUE;
1869 }
1870
1871 static BOOL alloc_variable(compiler_ctx_t *ctx, const WCHAR *name)
1872 {
1873     BSTR ident;
1874
1875     if(find_local(ctx, name))
1876         return TRUE;
1877
1878     ident = compiler_alloc_bstr(ctx, name);
1879     if(!ident)
1880         return FALSE;
1881
1882     return alloc_local(ctx, ident, ctx->func->var_cnt++);
1883 }
1884
1885 static HRESULT visit_function_expression(compiler_ctx_t *ctx, function_expression_t *expr)
1886 {
1887     expr->func_id = ctx->func->func_cnt++;
1888     ctx->func_tail = ctx->func_tail ? (ctx->func_tail->next = expr) : (ctx->func_head = expr);
1889
1890     if(!expr->identifier || expr->event_target)
1891         return S_OK;
1892     if(!expr->is_statement && ctx->parser->script->version >= SCRIPTLANGUAGEVERSION_ES5)
1893         return S_OK;
1894
1895     return alloc_variable(ctx, expr->identifier) ? S_OK : E_OUTOFMEMORY;
1896 }
1897
1898 static HRESULT visit_expression(compiler_ctx_t *ctx, expression_t *expr)
1899 {
1900     HRESULT hres = S_OK;
1901
1902     switch(expr->type) {
1903     case EXPR_ADD:
1904     case EXPR_AND:
1905     case EXPR_ARRAY:
1906     case EXPR_ASSIGN:
1907     case EXPR_ASSIGNADD:
1908     case EXPR_ASSIGNAND:
1909     case EXPR_ASSIGNSUB:
1910     case EXPR_ASSIGNMUL:
1911     case EXPR_ASSIGNDIV:
1912     case EXPR_ASSIGNMOD:
1913     case EXPR_ASSIGNOR:
1914     case EXPR_ASSIGNLSHIFT:
1915     case EXPR_ASSIGNRSHIFT:
1916     case EXPR_ASSIGNRRSHIFT:
1917     case EXPR_ASSIGNXOR:
1918     case EXPR_BAND:
1919     case EXPR_BOR:
1920     case EXPR_COMMA:
1921     case EXPR_DIV:
1922     case EXPR_EQ:
1923     case EXPR_EQEQ:
1924     case EXPR_GREATER:
1925     case EXPR_GREATEREQ:
1926     case EXPR_IN:
1927     case EXPR_INSTANCEOF:
1928     case EXPR_LESS:
1929     case EXPR_LESSEQ:
1930     case EXPR_LSHIFT:
1931     case EXPR_MOD:
1932     case EXPR_MUL:
1933     case EXPR_NOTEQ:
1934     case EXPR_NOTEQEQ:
1935     case EXPR_OR:
1936     case EXPR_RSHIFT:
1937     case EXPR_RRSHIFT:
1938     case EXPR_SUB:
1939     case EXPR_BXOR: {
1940         binary_expression_t *binary_expr = (binary_expression_t*)expr;
1941
1942         hres = visit_expression(ctx, binary_expr->expression1);
1943         if(FAILED(hres))
1944             return hres;
1945
1946         hres = visit_expression(ctx, binary_expr->expression2);
1947         break;
1948     }
1949     case EXPR_BITNEG:
1950     case EXPR_DELETE:
1951     case EXPR_LOGNEG:
1952     case EXPR_MINUS:
1953     case EXPR_PLUS:
1954     case EXPR_POSTDEC:
1955     case EXPR_POSTINC:
1956     case EXPR_PREDEC:
1957     case EXPR_PREINC:
1958     case EXPR_TYPEOF:
1959     case EXPR_VOID:
1960         hres = visit_expression(ctx, ((unary_expression_t*)expr)->expression);
1961         break;
1962     case EXPR_IDENT:
1963     case EXPR_LITERAL:
1964     case EXPR_THIS:
1965         break;
1966     case EXPR_ARRAYLIT: {
1967         array_literal_expression_t *array_expr = (array_literal_expression_t*)expr;
1968         array_element_t *iter;
1969
1970         for(iter = array_expr->element_list; iter; iter = iter->next) {
1971             hres = visit_expression(ctx, iter->expr);
1972             if(FAILED(hres))
1973                 return hres;
1974         }
1975         break;
1976     }
1977     case EXPR_CALL:
1978     case EXPR_NEW: {
1979         call_expression_t *call_expr = (call_expression_t*)expr;
1980         argument_t *arg;
1981
1982         hres = visit_expression(ctx, call_expr->expression);
1983         if(FAILED(hres))
1984             return hres;
1985
1986         for(arg = call_expr->argument_list; arg; arg = arg->next) {
1987             hres = visit_expression(ctx, arg->expr);
1988             if(FAILED(hres))
1989                 return hres;
1990         }
1991         break;
1992     }
1993     case EXPR_COND: {
1994         conditional_expression_t *cond_expr = (conditional_expression_t*)expr;
1995
1996         hres = visit_expression(ctx, cond_expr->expression);
1997         if(FAILED(hres))
1998             return hres;
1999
2000         hres = visit_expression(ctx, cond_expr->true_expression);
2001         if(FAILED(hres))
2002             return hres;
2003
2004         hres = visit_expression(ctx, cond_expr->false_expression);
2005         break;
2006     }
2007     case EXPR_FUNC:
2008         hres = visit_function_expression(ctx, (function_expression_t*)expr);
2009         break;
2010     case EXPR_MEMBER:
2011         hres = visit_expression(ctx, ((member_expression_t*)expr)->expression);
2012         break;
2013     case EXPR_PROPVAL: {
2014         property_definition_t *iter;
2015         for(iter = ((property_value_expression_t*)expr)->property_list; iter; iter = iter->next) {
2016             hres = visit_expression(ctx, iter->value);
2017             if(FAILED(hres))
2018                 return hres;
2019         }
2020         break;
2021     }
2022     DEFAULT_UNREACHABLE;
2023     }
2024
2025     return hres;
2026 }
2027
2028 static HRESULT visit_variable_list(compiler_ctx_t *ctx, variable_declaration_t *list)
2029 {
2030     variable_declaration_t *iter;
2031     HRESULT hres;
2032
2033     for(iter = list; iter; iter = iter->next) {
2034         if(!alloc_variable(ctx, iter->identifier))
2035             return E_OUTOFMEMORY;
2036
2037         if(iter->expr) {
2038             hres = visit_expression(ctx, iter->expr);
2039             if(FAILED(hres))
2040                 return hres;
2041         }
2042     }
2043
2044     return S_OK;
2045 }
2046
2047 static HRESULT visit_statement(compiler_ctx_t*,statement_t*);
2048
2049 static HRESULT visit_block_statement(compiler_ctx_t *ctx, statement_t *iter)
2050 {
2051     HRESULT hres;
2052
2053     while(iter) {
2054         hres = visit_statement(ctx, iter);
2055         if(FAILED(hres))
2056             return hres;
2057
2058         iter = iter->next;
2059     }
2060
2061     return S_OK;
2062 }
2063
2064 static HRESULT visit_statement(compiler_ctx_t *ctx, statement_t *stat)
2065 {
2066     HRESULT hres = S_OK;
2067
2068     switch(stat->type) {
2069     case STAT_BLOCK:
2070         hres = visit_block_statement(ctx, ((block_statement_t*)stat)->stat_list);
2071         break;
2072     case STAT_BREAK:
2073     case STAT_CONTINUE:
2074     case STAT_EMPTY:
2075         break;
2076     case STAT_EXPR: {
2077         expression_statement_t *expr_stat = (expression_statement_t*)stat;
2078         if(expr_stat->expr) {
2079             if(expr_stat->expr->type == EXPR_FUNC)
2080                 ((function_expression_t*)expr_stat->expr)->is_statement = TRUE;
2081             hres = visit_expression(ctx, expr_stat->expr);
2082         }
2083         break;
2084     }
2085     case STAT_RETURN:
2086     case STAT_THROW: {
2087         expression_statement_t *expr_stat = (expression_statement_t*)stat;
2088         if(expr_stat->expr)
2089             hres = visit_expression(ctx, expr_stat->expr);
2090         break;
2091     }
2092     case STAT_FOR: {
2093         for_statement_t *for_stat = (for_statement_t*)stat;
2094
2095         if(for_stat->variable_list)
2096             hres = visit_variable_list(ctx, for_stat->variable_list);
2097         else if(for_stat->begin_expr)
2098             hres = visit_expression(ctx, for_stat->begin_expr);
2099         if(FAILED(hres))
2100             break;
2101
2102         if(for_stat->expr) {
2103             hres = visit_expression(ctx, for_stat->expr);
2104             if(FAILED(hres))
2105                 break;
2106         }
2107
2108         hres = visit_statement(ctx, for_stat->statement);
2109         if(FAILED(hres))
2110             break;
2111
2112         if(for_stat->end_expr)
2113             hres = visit_expression(ctx, for_stat->end_expr);
2114         break;
2115     }
2116     case STAT_FORIN:  {
2117         forin_statement_t *forin_stat = (forin_statement_t*)stat;
2118
2119         if(forin_stat->variable) {
2120             hres = visit_variable_list(ctx, forin_stat->variable);
2121             if(FAILED(hres))
2122                 break;
2123         }
2124
2125         hres = visit_expression(ctx, forin_stat->in_expr);
2126         if(FAILED(hres))
2127             return hres;
2128
2129         if(forin_stat->expr) {
2130             hres = visit_expression(ctx, forin_stat->expr);
2131             if(FAILED(hres))
2132                 return hres;
2133         }
2134
2135         hres = visit_statement(ctx, forin_stat->statement);
2136         break;
2137     }
2138     case STAT_IF: {
2139         if_statement_t *if_stat = (if_statement_t*)stat;
2140
2141         hres = visit_expression(ctx, if_stat->expr);
2142         if(FAILED(hres))
2143             return hres;
2144
2145         hres = visit_statement(ctx, if_stat->if_stat);
2146         if(FAILED(hres))
2147             return hres;
2148
2149         if(if_stat->else_stat)
2150             hres = visit_statement(ctx, if_stat->else_stat);
2151         break;
2152     }
2153     case STAT_LABEL:
2154         hres = visit_statement(ctx, ((labelled_statement_t*)stat)->statement);
2155         break;
2156     case STAT_SWITCH: {
2157         switch_statement_t *switch_stat = (switch_statement_t*)stat;
2158         statement_t *stat_iter;
2159         case_clausule_t *iter;
2160
2161         hres = visit_expression(ctx, switch_stat->expr);
2162         if(FAILED(hres))
2163             return hres;
2164
2165         for(iter = switch_stat->case_list; iter; iter = iter->next) {
2166             if(!iter->expr)
2167                 continue;
2168             hres = visit_expression(ctx, iter->expr);
2169             if(FAILED(hres))
2170                 return hres;
2171         }
2172
2173         for(iter = switch_stat->case_list; iter; iter = iter->next) {
2174             while(iter->next && iter->next->stat == iter->stat)
2175                 iter = iter->next;
2176             for(stat_iter = iter->stat; stat_iter && (!iter->next || iter->next->stat != stat_iter);
2177                 stat_iter = stat_iter->next) {
2178                 hres = visit_statement(ctx, stat_iter);
2179                 if(FAILED(hres))
2180                     return hres;
2181             }
2182         }
2183         break;
2184     }
2185     case STAT_TRY: {
2186         try_statement_t *try_stat = (try_statement_t*)stat;
2187
2188         hres = visit_statement(ctx, try_stat->try_statement);
2189         if(FAILED(hres))
2190             return hres;
2191
2192         if(try_stat->catch_block) {
2193             hres = visit_statement(ctx, try_stat->catch_block->statement);
2194             if(FAILED(hres))
2195                 return hres;
2196         }
2197
2198         if(try_stat->finally_statement)
2199             hres = visit_statement(ctx, try_stat->finally_statement);
2200         break;
2201     }
2202     case STAT_VAR:
2203         hres = visit_variable_list(ctx, ((var_statement_t*)stat)->variable_list);
2204         break;
2205     case STAT_WHILE: {
2206         while_statement_t *while_stat = (while_statement_t*)stat;
2207
2208         hres = visit_expression(ctx, while_stat->expr);
2209         if(FAILED(hres))
2210             return hres;
2211
2212         hres = visit_statement(ctx, while_stat->statement);
2213         break;
2214     }
2215     case STAT_WITH: {
2216         with_statement_t *with_stat = (with_statement_t*)stat;
2217
2218         hres = visit_expression(ctx, with_stat->expr);
2219         if(FAILED(hres))
2220             return hres;
2221
2222         hres = visit_statement(ctx, with_stat->statement);
2223         break;
2224     }
2225     DEFAULT_UNREACHABLE;
2226     }
2227
2228     return hres;
2229 }
2230
2231 static void resolve_labels(compiler_ctx_t *ctx, unsigned off)
2232 {
2233     instr_t *instr;
2234
2235     for(instr = ctx->code->instrs+off; instr < ctx->code->instrs+ctx->code_off; instr++) {
2236         if(instr_info[instr->op].arg1_type == ARG_ADDR && (instr->u.arg->uint & LABEL_FLAG)) {
2237             assert((instr->u.arg->uint & ~LABEL_FLAG) < ctx->labels_cnt);
2238             instr->u.arg->uint = ctx->labels[instr->u.arg->uint & ~LABEL_FLAG];
2239         }
2240         assert(instr_info[instr->op].arg2_type != ARG_ADDR);
2241     }
2242
2243     ctx->labels_cnt = 0;
2244 }
2245
2246 unsigned get_location_line(bytecode_t *code, unsigned loc, unsigned *char_pos)
2247 {
2248     unsigned line = code->start_line;
2249     const WCHAR *nl, *p;
2250
2251     for(nl = p = code->source; p < code->source + loc; p++) {
2252         if(*p != '\n') continue;
2253         line++;
2254         nl = p + 1;
2255     }
2256     *char_pos = loc - (nl - code->source);
2257     return line;
2258 }
2259
2260 void release_bytecode(bytecode_t *code)
2261 {
2262     unsigned i;
2263
2264     if(--code->ref)
2265         return;
2266
2267     for(i=0; i < code->bstr_cnt; i++)
2268         SysFreeString(code->bstr_pool[i]);
2269     for(i=0; i < code->str_cnt; i++)
2270         jsstr_release(code->str_pool[i]);
2271
2272     if(code->named_item)
2273         release_named_item(code->named_item);
2274     heap_free(code->source);
2275     heap_pool_free(&code->heap);
2276     heap_free(code->bstr_pool);
2277     heap_free(code->str_pool);
2278     heap_free(code->instrs);
2279     heap_free(code);
2280 }
2281
2282 static HRESULT init_code(compiler_ctx_t *compiler, const WCHAR *source, UINT64 source_context, unsigned start_line)
2283 {
2284     size_t len = source ? lstrlenW(source) : 0;
2285
2286     if(len > INT32_MAX)
2287         return E_OUTOFMEMORY;
2288
2289     compiler->code = heap_alloc_zero(sizeof(bytecode_t));
2290     if(!compiler->code)
2291         return E_OUTOFMEMORY;
2292
2293     compiler->code->ref = 1;
2294     compiler->code->source_context = source_context;
2295     compiler->code->start_line = start_line;
2296     heap_pool_init(&compiler->code->heap);
2297
2298     compiler->code->source = heap_alloc((len + 1) * sizeof(WCHAR));
2299     if(!compiler->code->source) {
2300         release_bytecode(compiler->code);
2301         return E_OUTOFMEMORY;
2302     }
2303     if(len)
2304         memcpy(compiler->code->source, source, len * sizeof(WCHAR));
2305     compiler->code->source[len] = 0;
2306
2307     compiler->code->instrs = heap_alloc(64 * sizeof(instr_t));
2308     if(!compiler->code->instrs) {
2309         release_bytecode(compiler->code);
2310         return E_OUTOFMEMORY;
2311     }
2312
2313     compiler->code_size = 64;
2314     compiler->code_off = 1;
2315     return S_OK;
2316 }
2317
2318 static HRESULT compile_function(compiler_ctx_t *ctx, source_elements_t *source, function_expression_t *func_expr,
2319         BOOL from_eval, function_code_t *func)
2320 {
2321     function_expression_t *iter;
2322     function_local_t *local;
2323     unsigned off, i;
2324     HRESULT hres;
2325
2326     TRACE("\n");
2327
2328     func->bytecode = ctx->code;
2329     func->local_ref = INVALID_LOCAL_REF;
2330     ctx->func_head = ctx->func_tail = NULL;
2331     ctx->from_eval = from_eval;
2332     ctx->func = func;
2333     ctx->locals_cnt = 0;
2334     wine_rb_init(&ctx->locals, function_local_cmp);
2335
2336     if(func_expr) {
2337         parameter_t *param_iter;
2338
2339         if(func_expr->identifier) {
2340             func->name = compiler_alloc_bstr(ctx, func_expr->identifier);
2341             if(!func->name)
2342                 return E_OUTOFMEMORY;
2343         }
2344
2345         if(func_expr->event_target) {
2346             func->event_target = compiler_alloc_bstr(ctx, func_expr->event_target);
2347             if(!func->event_target)
2348                 return E_OUTOFMEMORY;
2349         }
2350
2351         func->source = func_expr->src_str;
2352         func->source_len = func_expr->src_len;
2353
2354         for(param_iter = func_expr->parameter_list; param_iter; param_iter = param_iter->next)
2355             func->param_cnt++;
2356
2357         func->params = compiler_alloc(ctx->code, func->param_cnt * sizeof(*func->params));
2358         if(!func->params)
2359             return E_OUTOFMEMORY;
2360
2361         for(param_iter = func_expr->parameter_list, i=0; param_iter; param_iter = param_iter->next, i++) {
2362             func->params[i] = compiler_alloc_bstr(ctx, param_iter->identifier);
2363             if(!func->params[i])
2364                 return E_OUTOFMEMORY;
2365         }
2366     }
2367
2368     for(i = 0; i < func->param_cnt; i++) {
2369         if(!find_local(ctx, func->params[i]) && !alloc_local(ctx, func->params[i], -i-1))
2370             return E_OUTOFMEMORY;
2371     }
2372
2373     hres = visit_block_statement(ctx, source->statement);
2374     if(FAILED(hres))
2375         return hres;
2376
2377     func->locals = compiler_alloc(ctx->code, ctx->locals_cnt * sizeof(*func->locals));
2378     if(!func->locals)
2379         return E_OUTOFMEMORY;
2380     func->locals_cnt = ctx->locals_cnt;
2381
2382     func->variables = compiler_alloc(ctx->code, func->var_cnt * sizeof(*func->variables));
2383     if(!func->variables)
2384         return E_OUTOFMEMORY;
2385
2386     i = 0;
2387     WINE_RB_FOR_EACH_ENTRY(local, &ctx->locals, function_local_t, entry) {
2388         func->locals[i].name = local->name;
2389         func->locals[i].ref = local->ref;
2390         if(local->ref >= 0) {
2391             func->variables[local->ref].name = local->name;
2392             func->variables[local->ref].func_id = -1;
2393         }
2394         i++;
2395     }
2396     assert(i == ctx->locals_cnt);
2397
2398     func->funcs = compiler_alloc(ctx->code, func->func_cnt * sizeof(*func->funcs));
2399     if(!func->funcs)
2400         return E_OUTOFMEMORY;
2401     memset(func->funcs, 0, func->func_cnt * sizeof(*func->funcs));
2402
2403     off = ctx->code_off;
2404     hres = compile_block_statement(ctx, source->statement);
2405     if(FAILED(hres))
2406         return hres;
2407
2408     resolve_labels(ctx, off);
2409
2410     hres = push_instr_uint(ctx, OP_ret, !from_eval);
2411     if(FAILED(hres))
2412         return hres;
2413
2414     if(TRACE_ON(jscript_disas))
2415         dump_code(ctx, off);
2416
2417     func->instr_off = off;
2418
2419     for(iter = ctx->func_head, i=0; iter; iter = iter->next, i++) {
2420         hres = compile_function(ctx, iter->source_elements, iter, FALSE, func->funcs+i);
2421         if(FAILED(hres))
2422             return hres;
2423
2424         TRACE("[%d] func %s\n", i, debugstr_w(func->funcs[i].name));
2425         if((ctx->parser->script->version < SCRIPTLANGUAGEVERSION_ES5 || iter->is_statement) &&
2426            func->funcs[i].name && !func->funcs[i].event_target) {
2427             local_ref_t *local_ref = lookup_local(func, func->funcs[i].name);
2428             func->funcs[i].local_ref = local_ref->ref;
2429             TRACE("found ref %s %d for %s\n", debugstr_w(local_ref->name), local_ref->ref, debugstr_w(func->funcs[i].name));
2430             if(local_ref->ref >= 0)
2431                 func->variables[local_ref->ref].func_id = i;
2432         }
2433     }
2434
2435     assert(i == func->func_cnt);
2436
2437     return S_OK;
2438 }
2439
2440 static HRESULT parse_arguments(compiler_ctx_t *ctx, const WCHAR *args, BSTR *arg_array, unsigned *args_size)
2441 {
2442     const WCHAR *ptr = args, *ptr2;
2443     unsigned arg_cnt = 0;
2444
2445     while(iswspace(*ptr))
2446         ptr++;
2447     if(!*ptr) {
2448         if(args_size)
2449             *args_size = 0;
2450         return S_OK;
2451     }
2452
2453     while(1) {
2454         if(!iswalpha(*ptr) && *ptr != '_') {
2455             FIXME("expected alpha or '_': %s\n", debugstr_w(ptr));
2456             return E_FAIL;
2457         }
2458
2459         ptr2 = ptr;
2460         while(iswalnum(*ptr) || *ptr == '_')
2461             ptr++;
2462
2463         if(*ptr && *ptr != ',' && !iswspace(*ptr)) {
2464             FIXME("unexpected har %s\n", debugstr_w(ptr));
2465             return E_FAIL;
2466         }
2467
2468         if(arg_array) {
2469             arg_array[arg_cnt] = compiler_alloc_bstr_len(ctx, ptr2, ptr-ptr2);
2470             if(!arg_array[arg_cnt])
2471                 return E_OUTOFMEMORY;
2472         }
2473         arg_cnt++;
2474
2475         while(iswspace(*ptr))
2476             ptr++;
2477         if(!*ptr)
2478             break;
2479         if(*ptr != ',') {
2480             FIXME("expected ',': %s\n", debugstr_w(ptr));
2481             return E_FAIL;
2482         }
2483
2484         ptr++;
2485         while(iswspace(*ptr))
2486             ptr++;
2487     }
2488
2489     if(args_size)
2490         *args_size = arg_cnt;
2491     return S_OK;
2492 }
2493
2494 static HRESULT compile_arguments(compiler_ctx_t *ctx, const WCHAR *args)
2495 {
2496     HRESULT hres;
2497
2498     hres = parse_arguments(ctx, args, NULL, &ctx->code->global_code.param_cnt);
2499     if(FAILED(hres))
2500         return hres;
2501
2502     ctx->code->global_code.params = compiler_alloc(ctx->code,
2503             ctx->code->global_code.param_cnt * sizeof(*ctx->code->global_code.params));
2504     if(!ctx->code->global_code.params)
2505         return E_OUTOFMEMORY;
2506
2507     return parse_arguments(ctx, args, ctx->code->global_code.params, NULL);
2508 }
2509
2510 HRESULT compile_script(script_ctx_t *ctx, const WCHAR *code, UINT64 source_context, unsigned start_line,
2511                        const WCHAR *args, const WCHAR *delimiter, BOOL from_eval, BOOL use_decode,
2512                        named_item_t *named_item, bytecode_t **ret)
2513 {
2514     compiler_ctx_t compiler = {0};
2515     HRESULT hres;
2516
2517     hres = init_code(&compiler, code, source_context, start_line);
2518     if(FAILED(hres))
2519         return hres;
2520
2521     if(args) {
2522         hres = compile_arguments(&compiler, args);
2523         if(FAILED(hres))
2524             return hres;
2525     }
2526
2527     if(use_decode) {
2528         hres = decode_source(compiler.code->source);
2529         if(FAILED(hres)) {
2530             WARN("Decoding failed\n");
2531             return hres;
2532         }
2533     }
2534
2535     hres = script_parse(ctx, &compiler, compiler.code, delimiter, from_eval, &compiler.parser);
2536     if(FAILED(hres)) {
2537         release_bytecode(compiler.code);
2538         return hres;
2539     }
2540
2541     heap_pool_init(&compiler.heap);
2542     hres = compile_function(&compiler, compiler.parser->source, NULL, from_eval, &compiler.code->global_code);
2543     heap_pool_free(&compiler.heap);
2544     parser_release(compiler.parser);
2545     if(FAILED(hres)) {
2546         if(hres != DISP_E_EXCEPTION)
2547             throw_error(ctx, hres, NULL);
2548         set_error_location(ctx->ei, compiler.code, compiler.loc, IDS_COMPILATION_ERROR, NULL);
2549         release_bytecode(compiler.code);
2550         return DISP_E_EXCEPTION;
2551     }
2552
2553     if(named_item) {
2554         compiler.code->named_item = named_item;
2555         named_item->ref++;
2556     }
2557
2558     *ret = compiler.code;
2559     return S_OK;
2560 }