dlls/jscript/lex.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 "config.h"
  20 #include "wine/port.h"
  21
  22 #include <limits.h>
  23
  24 #include "jscript.h"
  25 #include "activscp.h"
  26 #include "objsafe.h"
  27 #include "engine.h"
  28 #include "parser.h"
  29
  30 #include "parser.tab.h"
  31
  32 #include "wine/debug.h"
  33 #include "wine/unicode.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(jscript);
  36
  37 static const WCHAR breakW[] = {'b','r','e','a','k',0};
  38 static const WCHAR caseW[] = {'c','a','s','e',0};
  39 static const WCHAR catchW[] = {'c','a','t','c','h',0};
  40 static const WCHAR continueW[] = {'c','o','n','t','i','n','u','e',0};
  41 static const WCHAR defaultW[] = {'d','e','f','a','u','l','t',0};
  42 static const WCHAR deleteW[] = {'d','e','l','e','t','e',0};
  43 static const WCHAR doW[] = {'d','o',0};
  44 static const WCHAR elseW[] = {'e','l','s','e',0};
  45 static const WCHAR falseW[] = {'f','a','l','s','e',0};
  46 static const WCHAR finallyW[] = {'f','i','n','a','l','l','y',0};
  47 static const WCHAR forW[] = {'f','o','r',0};
  48 static const WCHAR functionW[] = {'f','u','n','c','t','i','o','n',0};
  49 static const WCHAR ifW[] = {'i','f',0};
  50 static const WCHAR inW[] = {'i','n',0};
  51 static const WCHAR instanceofW[] = {'i','n','s','t','a','n','c','e','o','f',0};
  52 static const WCHAR newW[] = {'n','e','w',0};
  53 static const WCHAR nullW[] = {'n','u','l','l',0};
  54 static const WCHAR returnW[] = {'r','e','t','u','r','n',0};
  55 static const WCHAR switchW[] = {'s','w','i','t','c','h',0};
  56 static const WCHAR thisW[] = {'t','h','i','s',0};
  57 static const WCHAR throwW[] = {'t','h','r','o','w',0};
  58 static const WCHAR trueW[] = {'t','r','u','e',0};
  59 static const WCHAR tryW[] = {'t','r','y',0};
  60 static const WCHAR typeofW[] = {'t','y','p','e','o','f',0};
  61 static const WCHAR varW[] = {'v','a','r',0};
  62 static const WCHAR voidW[] = {'v','o','i','d',0};
  63 static const WCHAR whileW[] = {'w','h','i','l','e',0};
  64 static const WCHAR withW[] = {'w','i','t','h',0};
  65
  66 static const WCHAR elifW[] = {'e','l','i','f',0};
  67 static const WCHAR endW[] = {'e','n','d',0};
  68
  69 static const struct {
  70     const WCHAR *word;
  71     int token;
  72     BOOL no_nl;
  73 } keywords[] = {
  74     {breakW,       kBREAK, TRUE},
  75     {caseW,        kCASE},
  76     {catchW,       kCATCH},
  77     {continueW,    kCONTINUE, TRUE},
  78     {defaultW,     kDEFAULT},
  79     {deleteW,      kDELETE},
  80     {doW,          kDO},
  81     {elseW,        kELSE},
  82     {falseW,       kFALSE},
  83     {finallyW,     kFINALLY},
  84     {forW,         kFOR},
  85     {functionW,    kFUNCTION},
  86     {ifW,          kIF},
  87     {inW,          kIN},
  88     {instanceofW,  kINSTANCEOF},
  89     {newW,         kNEW},
  90     {nullW,        kNULL},
  91     {returnW,      kRETURN, TRUE},
  92     {switchW,      kSWITCH},
  93     {thisW,        kTHIS},
  94     {throwW,       kTHROW},
  95     {trueW,        kTRUE},
  96     {tryW,         kTRY},
  97     {typeofW,      kTYPEOF},
  98     {varW,         kVAR},
  99     {voidW,        kVOID},
 100     {whileW,       kWHILE},
 101     {withW,        kWITH}
 102 };
 103
 104 static int lex_error(parser_ctx_t *ctx, HRESULT hres)
 105 {
 106     ctx->hres = hres;
 107     ctx->lexer_error = TRUE;
 108     return -1;
 109 }
 110
 111 /* ECMA-262 3rd Edition    7.6 */
 112 BOOL is_identifier_char(WCHAR c)
 113 {
 114     return isalnumW(c) || c == '$' || c == '_' || c == '\\';
 115 }
 116
 117 static BOOL is_identifier_first_char(WCHAR c)
 118 {
 119     return isalphaW(c) || c == '$' || c == '_' || c == '\\';
 120 }
 121
 122 static int check_keyword(parser_ctx_t *ctx, const WCHAR *word, const WCHAR **lval)
 123 {
 124     const WCHAR *p1 = ctx->ptr;
 125     const WCHAR *p2 = word;
 126
 127     while(p1 < ctx->end && *p2) {
 128         if(*p1 != *p2)
 129             return *p1 - *p2;
 130         p1++;
 131         p2++;
 132     }
 133
 134     if(*p2 || (p1 < ctx->end && is_identifier_char(*p1)))
 135         return 1;
 136
 137     if(lval)
 138         *lval = ctx->ptr;
 139     ctx->ptr = p1;
 140     return 0;
 141 }
 142
 143 /* ECMA-262 3rd Edition    7.3 */
 144 static BOOL is_endline(WCHAR c)
 145 {
 146     return c == '\n' || c == '\r' || c == 0x2028 || c == 0x2029;
 147 }
 148
 149 static int hex_to_int(WCHAR c)
 150 {
 151     if('0' <= c && c <= '9')
 152         return c-'0';
 153
 154     if('a' <= c && c <= 'f')
 155         return c-'a'+10;
 156
 157     if('A' <= c && c <= 'F')
 158         return c-'A'+10;
 159
 160     return -1;
 161 }
 162
 163 static int check_keywords(parser_ctx_t *ctx, const WCHAR **lval)
 164 {
 165     int min = 0, max = sizeof(keywords)/sizeof(keywords[0])-1, r, i;
 166
 167     while(min <= max) {
 168         i = (min+max)/2;
 169
 170         r = check_keyword(ctx, keywords[i].word, lval);
 171         if(!r) {
 172             ctx->implicit_nl_semicolon = keywords[i].no_nl;
 173             return keywords[i].token;
 174         }
 175
 176         if(r > 0)
 177             min = i+1;
 178         else
 179             max = i-1;
 180     }
 181
 182     return 0;
 183 }
 184
 185 static BOOL skip_html_comment(parser_ctx_t *ctx)
 186 {
 187     const WCHAR html_commentW[] = {'<','!','-','-',0};
 188
 189     if(!ctx->is_html || ctx->ptr+3 >= ctx->end ||
 190         memcmp(ctx->ptr, html_commentW, sizeof(WCHAR)*4))
 191         return FALSE;
 192
 193     ctx->nl = TRUE;
 194     while(ctx->ptr < ctx->end && !is_endline(*ctx->ptr++));
 195
 196     return TRUE;
 197 }
 198
 199 static BOOL skip_comment(parser_ctx_t *ctx)
 200 {
 201     if(ctx->ptr+1 >= ctx->end)
 202         return FALSE;
 203
 204     if(*ctx->ptr != '/') {
 205         if(*ctx->ptr == '@' && ctx->ptr+2 < ctx->end && ctx->ptr[1] == '*' && ctx->ptr[2] == '/') {
 206             ctx->ptr += 3;
 207             return TRUE;
 208         }
 209
 210         return FALSE;
 211     }
 212
 213     switch(ctx->ptr[1]) {
 214     case '*':
 215         ctx->ptr += 2;
 216         if(ctx->ptr+2 < ctx->end && *ctx->ptr == '@' && is_identifier_char(ctx->ptr[1]))
 217             return FALSE;
 218         while(ctx->ptr+1 < ctx->end && (ctx->ptr[0] != '*' || ctx->ptr[1] != '/'))
 219             ctx->ptr++;
 220
 221         if(ctx->ptr[0] == '*' && ctx->ptr[1] == '/') {
 222             ctx->ptr += 2;
 223         }else {
 224             WARN("unexpected end of file (missing end of comment)\n");
 225             ctx->ptr = ctx->end;
 226         }
 227         break;
 228     case '/':
 229         ctx->ptr += 2;
 230         if(ctx->ptr+2 < ctx->end && *ctx->ptr == '@' && is_identifier_char(ctx->ptr[1]))
 231             return FALSE;
 232         while(ctx->ptr < ctx->end && !is_endline(*ctx->ptr))
 233             ctx->ptr++;
 234         break;
 235     default:
 236         return FALSE;
 237     }
 238
 239     return TRUE;
 240 }
 241
 242 static BOOL skip_spaces(parser_ctx_t *ctx)
 243 {
 244     while(ctx->ptr < ctx->end && (isspaceW(*ctx->ptr) || *ctx->ptr == 0xFEFF /* UTF16 BOM */)) {
 245         if(is_endline(*ctx->ptr++))
 246             ctx->nl = TRUE;
 247     }
 248
 249     return ctx->ptr != ctx->end;
 250 }
 251
 252 BOOL unescape(WCHAR *str)
 253 {
 254     WCHAR *pd, *p, c;
 255     int i;
 256
 257     pd = p = str;
 258     while(*p) {
 259         if(*p != '\\') {
 260             *pd++ = *p++;
 261             continue;
 262         }
 263
 264         p++;
 265
 266         switch(*p) {
 267         case '\'':
 268         case '\"':
 269         case '\\':
 270             c = *p;
 271             break;
 272         case 'b':
 273             c = '\b';
 274             break;
 275         case 't':
 276             c = '\t';
 277             break;
 278         case 'n':
 279             c = '\n';
 280             break;
 281         case 'f':
 282             c = '\f';
 283             break;
 284         case 'r':
 285             c = '\r';
 286             break;
 287         case 'x':
 288             i = hex_to_int(*++p);
 289             if(i == -1)
 290                 return FALSE;
 291             c = i << 4;
 292
 293             i = hex_to_int(*++p);
 294             if(i == -1)
 295                 return FALSE;
 296             c += i;
 297             break;
 298         case 'u':
 299             i = hex_to_int(*++p);
 300             if(i == -1)
 301                 return FALSE;
 302             c = i << 12;
 303
 304             i = hex_to_int(*++p);
 305             if(i == -1)
 306                 return FALSE;
 307             c += i << 8;
 308
 309             i = hex_to_int(*++p);
 310             if(i == -1)
 311                 return FALSE;
 312             c += i << 4;
 313
 314             i = hex_to_int(*++p);
 315             if(i == -1)
 316                 return FALSE;
 317             c += i;
 318             break;
 319         default:
 320             if(isdigitW(*p)) {
 321                 c = *p++ - '0';
 322                 if(isdigitW(*p)) {
 323                     c = c*8 + (*p++ - '0');
 324                     if(isdigitW(*p))
 325                         c = c*8 + (*p++ - '0');
 326                 }
 327                 p--;
 328             }
 329             else
 330                 c = *p;
 331         }
 332
 333         *pd++ = c;
 334         p++;
 335     }
 336
 337     *pd = 0;
 338     return TRUE;
 339 }
 340
 341 static int parse_identifier(parser_ctx_t *ctx, const WCHAR **ret)
 342 {
 343     const WCHAR *ptr = ctx->ptr++;
 344     WCHAR *wstr;
 345     int len;
 346
 347     while(ctx->ptr < ctx->end && is_identifier_char(*ctx->ptr))
 348         ctx->ptr++;
 349
 350     len = ctx->ptr-ptr;
 351
 352     *ret = wstr = parser_alloc(ctx, (len+1)*sizeof(WCHAR));
 353     memcpy(wstr, ptr, len*sizeof(WCHAR));
 354     wstr[len] = 0;
 355
 356     /* FIXME: unescape */
 357     return tIdentifier;
 358 }
 359
 360 static int parse_string_literal(parser_ctx_t *ctx, const WCHAR **ret, WCHAR endch)
 361 {
 362     const WCHAR *ptr = ++ctx->ptr;
 363     WCHAR *wstr;
 364     int len;
 365
 366     while(ctx->ptr < ctx->end && *ctx->ptr != endch) {
 367         if(*ctx->ptr++ == '\\')
 368             ctx->ptr++;
 369     }
 370
 371     if(ctx->ptr == ctx->end)
 372         return lex_error(ctx, JS_E_UNTERMINATED_STRING);
 373
 374     len = ctx->ptr-ptr;
 375
 376     *ret = wstr = parser_alloc(ctx, (len+1)*sizeof(WCHAR));
 377     memcpy(wstr, ptr, len*sizeof(WCHAR));
 378     wstr[len] = 0;
 379
 380     ctx->ptr++;
 381
 382     if(!unescape(wstr)) {
 383         WARN("unescape failed\n");
 384         return lex_error(ctx, E_FAIL);
 385     }
 386
 387     return tStringLiteral;
 388 }
 389
 390 static literal_t *new_double_literal(parser_ctx_t *ctx, DOUBLE d)
 391 {
 392     literal_t *ret = parser_alloc(ctx, sizeof(literal_t));
 393
 394     ret->type = LT_DOUBLE;
 395     ret->u.dval = d;
 396     return ret;
 397 }
 398
 399 literal_t *new_boolean_literal(parser_ctx_t *ctx, BOOL bval)
 400 {
 401     literal_t *ret = parser_alloc(ctx, sizeof(literal_t));
 402
 403     ret->type = LT_BOOL;
 404     ret->u.bval = bval;
 405
 406     return ret;
 407 }
 408
 409 HRESULT parse_decimal(const WCHAR **iter, const WCHAR *end, double *ret)
 410 {
 411     const WCHAR *ptr = *iter;
 412     LONGLONG d = 0, hlp;
 413     int exp = 0;
 414
 415     while(ptr < end && isdigitW(*ptr)) {
 416         hlp = d*10 + *(ptr++) - '0';
 417         if(d>MAXLONGLONG/10 || hlp<0) {
 418             exp++;
 419             break;
 420         }
 421         else
 422             d = hlp;
 423     }
 424     while(ptr < end && isdigitW(*ptr)) {
 425         exp++;
 426         ptr++;
 427     }
 428
 429     if(*ptr == '.') {
 430         ptr++;
 431
 432         while(ptr < end && isdigitW(*ptr)) {
 433             hlp = d*10 + *(ptr++) - '0';
 434             if(d>MAXLONGLONG/10 || hlp<0)
 435                 break;
 436
 437             d = hlp;
 438             exp--;
 439         }
 440         while(ptr < end && isdigitW(*ptr))
 441             ptr++;
 442     }
 443
 444     if(ptr < end && (*ptr == 'e' || *ptr == 'E')) {
 445         int sign = 1, e = 0;
 446
 447         if(++ptr < end) {
 448             if(*ptr == '+') {
 449                 ptr++;
 450             }else if(*ptr == '-') {
 451                 sign = -1;
 452                 ptr++;
 453             }else if(!isdigitW(*ptr)) {
 454                 WARN("Expected exponent part\n");
 455                 return E_FAIL;
 456             }
 457         }
 458
 459         if(ptr == end) {
 460             WARN("unexpected end of file\n");
 461             return E_FAIL;
 462         }
 463
 464         while(ptr < end && isdigitW(*ptr)) {
 465             if(e > INT_MAX/10 || (e = e*10 + *ptr++ - '0')<0)
 466                 e = INT_MAX;
 467         }
 468         e *= sign;
 469
 470         if(exp<0 && e<0 && e+exp>0) exp = INT_MIN;
 471         else if(exp>0 && e>0 && e+exp<0) exp = INT_MAX;
 472         else exp += e;
 473     }
 474
 475     if(is_identifier_char(*ptr)) {
 476         WARN("wrong char after zero\n");
 477         return JS_E_MISSING_SEMICOLON;
 478     }
 479
 480     *ret = exp>=0 ? d*pow(10, exp) : d/pow(10, -exp);
 481     *iter = ptr;
 482     return S_OK;
 483 }
 484
 485 static BOOL parse_numeric_literal(parser_ctx_t *ctx, double *ret)
 486 {
 487     HRESULT hres;
 488
 489     if(*ctx->ptr == '0') {
 490         LONG d, l = 0;
 491
 492         ctx->ptr++;
 493
 494         if(*ctx->ptr == 'x' || *ctx->ptr == 'X') {
 495             if(++ctx->ptr == ctx->end) {
 496                 ERR("unexpected end of file\n");
 497                 return FALSE;
 498             }
 499
 500             while(ctx->ptr < ctx->end && (d = hex_to_int(*ctx->ptr)) != -1) {
 501                 l = l*16 + d;
 502                 ctx->ptr++;
 503             }
 504
 505             if(ctx->ptr < ctx->end && is_identifier_char(*ctx->ptr)) {
 506                 WARN("unexpected identifier char\n");
 507                 lex_error(ctx, JS_E_MISSING_SEMICOLON);
 508                 return FALSE;
 509             }
 510
 511             *ret = l;
 512             return TRUE;
 513         }
 514
 515         if(isdigitW(*ctx->ptr)) {
 516             unsigned base = 8;
 517             const WCHAR *ptr;
 518             double val = 0;
 519
 520             for(ptr = ctx->ptr; ptr < ctx->end && isdigitW(*ptr); ptr++) {
 521                 if(*ptr > '7') {
 522                     base = 10;
 523                     break;
 524                 }
 525             }
 526
 527             do {
 528                 val = val*base + *ctx->ptr-'0';
 529             }while(++ctx->ptr < ctx->end && isdigitW(*ctx->ptr));
 530
 531             /* FIXME: Do we need it here? */
 532             if(ctx->ptr < ctx->end && (is_identifier_char(*ctx->ptr) || *ctx->ptr == '.')) {
 533                 WARN("wrong char after octal literal: '%c'\n", *ctx->ptr);
 534                 lex_error(ctx, JS_E_MISSING_SEMICOLON);
 535                 return FALSE;
 536             }
 537
 538             *ret = val;
 539             return TRUE;
 540         }
 541
 542         if(is_identifier_char(*ctx->ptr)) {
 543             WARN("wrong char after zero\n");
 544             lex_error(ctx, JS_E_MISSING_SEMICOLON);
 545             return FALSE;
 546         }
 547     }
 548
 549     hres = parse_decimal(&ctx->ptr, ctx->end, ret);
 550     if(FAILED(hres)) {
 551         lex_error(ctx, hres);
 552         return FALSE;
 553     }
 554
 555     return TRUE;
 556 }
 557
 558 static int next_token(parser_ctx_t *ctx, void *lval)
 559 {
 560     do {
 561         if(!skip_spaces(ctx))
 562             return tEOF;
 563     }while(skip_comment(ctx) || skip_html_comment(ctx));
 564
 565     if(ctx->implicit_nl_semicolon) {
 566         if(ctx->nl)
 567             return ';';
 568         ctx->implicit_nl_semicolon = FALSE;
 569     }
 570
 571     if(isalphaW(*ctx->ptr)) {
 572         int ret = check_keywords(ctx, lval);
 573         if(ret)
 574             return ret;
 575
 576         return parse_identifier(ctx, lval);
 577     }
 578
 579     if(isdigitW(*ctx->ptr)) {
 580         double n;
 581
 582         if(!parse_numeric_literal(ctx, &n))
 583             return -1;
 584
 585         *(literal_t**)lval = new_double_literal(ctx, n);
 586         return tNumericLiteral;
 587     }
 588
 589     switch(*ctx->ptr) {
 590     case '{':
 591     case '(':
 592     case ')':
 593     case '[':
 594     case ']':
 595     case ';':
 596     case ',':
 597     case '~':
 598     case '?':
 599         return *ctx->ptr++;
 600
 601     case '}':
 602         *(const WCHAR**)lval = ctx->ptr++;
 603         return '}';
 604
 605     case '.':
 606         if(++ctx->ptr < ctx->end && isdigitW(*ctx->ptr)) {
 607             double n;
 608             HRESULT hres;
 609             hres = parse_decimal(&ctx->ptr, ctx->end, &n);
 610             if(FAILED(hres)) {
 611                 lex_error(ctx, hres);
 612                 return -1;
 613             }
 614             *(literal_t**)lval = new_double_literal(ctx, n);
 615             return tNumericLiteral;
 616         }
 617         return '.';
 618
 619     case '<':
 620         if(++ctx->ptr == ctx->end) {
 621             *(int*)lval = EXPR_LESS;
 622             return tRelOper;
 623         }
 624
 625         switch(*ctx->ptr) {
 626         case '=':  /* <= */
 627             ctx->ptr++;
 628             *(int*)lval = EXPR_LESSEQ;
 629             return tRelOper;
 630         case '<':  /* << */
 631             if(++ctx->ptr < ctx->end && *ctx->ptr == '=') { /* <<= */
 632                 ctx->ptr++;
 633                 *(int*)lval = EXPR_ASSIGNLSHIFT;
 634                 return tAssignOper;
 635             }
 636             *(int*)lval = EXPR_LSHIFT;
 637             return tShiftOper;
 638         default: /* < */
 639             *(int*)lval = EXPR_LESS;
 640             return tRelOper;
 641         }
 642
 643     case '>':
 644         if(++ctx->ptr == ctx->end) { /* > */
 645             *(int*)lval = EXPR_GREATER;
 646             return tRelOper;
 647         }
 648
 649         switch(*ctx->ptr) {
 650         case '=':  /* >= */
 651             ctx->ptr++;
 652             *(int*)lval = EXPR_GREATEREQ;
 653             return tRelOper;
 654         case '>':  /* >> */
 655             if(++ctx->ptr < ctx->end) {
 656                 if(*ctx->ptr == '=') {  /* >>= */
 657                     ctx->ptr++;
 658                     *(int*)lval = EXPR_ASSIGNRSHIFT;
 659                     return tAssignOper;
 660                 }
 661                 if(*ctx->ptr == '>') {  /* >>> */
 662                     if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* >>>= */
 663                         ctx->ptr++;
 664                         *(int*)lval = EXPR_ASSIGNRRSHIFT;
 665                         return tAssignOper;
 666                     }
 667                     *(int*)lval = EXPR_RRSHIFT;
 668                     return tRelOper;
 669                 }
 670             }
 671             *(int*)lval = EXPR_RSHIFT;
 672             return tShiftOper;
 673         default:
 674             *(int*)lval = EXPR_GREATER;
 675             return tRelOper;
 676         }
 677
 678     case '+':
 679         ctx->ptr++;
 680         if(ctx->ptr < ctx->end) {
 681             switch(*ctx->ptr) {
 682             case '+':  /* ++ */
 683                 ctx->ptr++;
 684                 return tINC;
 685             case '=':  /* += */
 686                 ctx->ptr++;
 687                 *(int*)lval = EXPR_ASSIGNADD;
 688                 return tAssignOper;
 689             }
 690         }
 691         return '+';
 692
 693     case '-':
 694         ctx->ptr++;
 695         if(ctx->ptr < ctx->end) {
 696             switch(*ctx->ptr) {
 697             case '-':  /* -- or --> */
 698                 ctx->ptr++;
 699                 if(ctx->is_html && ctx->nl && ctx->ptr < ctx->end && *ctx->ptr == '>') {
 700                     ctx->ptr++;
 701                     return tHTMLCOMMENT;
 702                 }
 703                 return tDEC;
 704             case '=':  /* -= */
 705                 ctx->ptr++;
 706                 *(int*)lval = EXPR_ASSIGNSUB;
 707                 return tAssignOper;
 708             }
 709         }
 710         return '-';
 711
 712     case '*':
 713         if(++ctx->ptr < ctx->end && *ctx->ptr == '=') { /* *= */
 714             ctx->ptr++;
 715             *(int*)lval = EXPR_ASSIGNMUL;
 716             return tAssignOper;
 717         }
 718         return '*';
 719
 720     case '%':
 721         if(++ctx->ptr < ctx->end && *ctx->ptr == '=') { /* %= */
 722             ctx->ptr++;
 723             *(int*)lval = EXPR_ASSIGNMOD;
 724             return tAssignOper;
 725         }
 726         return '%';
 727
 728     case '&':
 729         if(++ctx->ptr < ctx->end) {
 730             switch(*ctx->ptr) {
 731             case '=':  /* &= */
 732                 ctx->ptr++;
 733                 *(int*)lval = EXPR_ASSIGNAND;
 734                 return tAssignOper;
 735             case '&':  /* && */
 736                 ctx->ptr++;
 737                 return tANDAND;
 738             }
 739         }
 740         return '&';
 741
 742     case '|':
 743         if(++ctx->ptr < ctx->end) {
 744             switch(*ctx->ptr) {
 745             case '=':  /* |= */
 746                 ctx->ptr++;
 747                 *(int*)lval = EXPR_ASSIGNOR;
 748                 return tAssignOper;
 749             case '|':  /* || */
 750                 ctx->ptr++;
 751                 return tOROR;
 752             }
 753         }
 754         return '|';
 755
 756     case '^':
 757         if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* ^= */
 758             ctx->ptr++;
 759             *(int*)lval = EXPR_ASSIGNXOR;
 760             return tAssignOper;
 761         }
 762         return '^';
 763
 764     case '!':
 765         if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* != */
 766             if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* !== */
 767                 ctx->ptr++;
 768                 *(int*)lval = EXPR_NOTEQEQ;
 769                 return tEqOper;
 770             }
 771             *(int*)lval = EXPR_NOTEQ;
 772             return tEqOper;
 773         }
 774         return '!';
 775
 776     case '=':
 777         if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* == */
 778             if(++ctx->ptr < ctx->end && *ctx->ptr == '=') {  /* === */
 779                 ctx->ptr++;
 780                 *(int*)lval = EXPR_EQEQ;
 781                 return tEqOper;
 782             }
 783             *(int*)lval = EXPR_EQ;
 784             return tEqOper;
 785         }
 786         return '=';
 787
 788     case '/':
 789         if(++ctx->ptr < ctx->end) {
 790             if(*ctx->ptr == '=') {  /* /= */
 791                 ctx->ptr++;
 792                 *(int*)lval = EXPR_ASSIGNDIV;
 793                 return kDIVEQ;
 794             }
 795         }
 796         return '/';
 797
 798     case ':':
 799         if(++ctx->ptr < ctx->end && *ctx->ptr == ':') {
 800             ctx->ptr++;
 801             return kDCOL;
 802         }
 803         return ':';
 804
 805     case '\"':
 806     case '\'':
 807         return parse_string_literal(ctx, lval, *ctx->ptr);
 808
 809     case '_':
 810     case '$':
 811         return parse_identifier(ctx, lval);
 812
 813     case '@':
 814         return '@';
 815     }
 816
 817     WARN("unexpected char '%c' %d\n", *ctx->ptr, *ctx->ptr);
 818     return 0;
 819 }
 820
 821 struct _cc_var_t {
 822     ccval_t val;
 823     struct _cc_var_t *next;
 824     unsigned name_len;
 825     WCHAR name[0];
 826 };
 827
 828 void release_cc(cc_ctx_t *cc)
 829 {
 830     cc_var_t *iter, *next;
 831
 832     for(iter = cc->vars; iter; iter = next) {
 833         next = iter->next;
 834         heap_free(iter);
 835     }
 836
 837     heap_free(cc);
 838 }
 839
 840 static BOOL new_cc_var(cc_ctx_t *cc, const WCHAR *name, int len, ccval_t v)
 841 {
 842     cc_var_t *new_v;
 843
 844     if(len == -1)
 845         len = strlenW(name);
 846
 847     new_v = heap_alloc(sizeof(cc_var_t) + (len+1)*sizeof(WCHAR));
 848     if(!new_v)
 849         return FALSE;
 850
 851     new_v->val = v;
 852     memcpy(new_v->name, name, (len+1)*sizeof(WCHAR));
 853     new_v->name_len = len;
 854     new_v->next = cc->vars;
 855     cc->vars = new_v;
 856     return TRUE;
 857 }
 858
 859 static cc_var_t *find_cc_var(cc_ctx_t *cc, const WCHAR *name, unsigned name_len)
 860 {
 861     cc_var_t *iter;
 862
 863     for(iter = cc->vars; iter; iter = iter->next) {
 864         if(iter->name_len == name_len && !memcmp(iter->name, name, name_len*sizeof(WCHAR)))
 865             return iter;
 866     }
 867
 868     return NULL;
 869 }
 870
 871 static BOOL init_cc(parser_ctx_t *ctx)
 872 {
 873     cc_ctx_t *cc;
 874
 875     static const WCHAR _win32W[] = {'_','w','i','n','3','2',0};
 876     static const WCHAR _win64W[] = {'_','w','i','n','6','4',0};
 877     static const WCHAR _x86W[] = {'_','x','8','6',0};
 878     static const WCHAR _amd64W[] = {'_','a','m','d','6','4',0};
 879     static const WCHAR _jscriptW[] = {'_','j','s','c','r','i','p','t',0};
 880     static const WCHAR _jscript_buildW[] = {'_','j','s','c','r','i','p','t','_','b','u','i','l','d',0};
 881     static const WCHAR _jscript_versionW[] = {'_','j','s','c','r','i','p','t','_','v','e','r','s','i','o','n',0};
 882
 883     if(ctx->script->cc)
 884         return TRUE;
 885
 886     cc = heap_alloc(sizeof(cc_ctx_t));
 887     if(!cc) {
 888         lex_error(ctx, E_OUTOFMEMORY);
 889         return FALSE;
 890     }
 891
 892     cc->vars = NULL;
 893
 894     if(!new_cc_var(cc, _jscriptW, -1, ccval_bool(TRUE))
 895        || !new_cc_var(cc, sizeof(void*) == 8 ? _win64W : _win32W, -1, ccval_bool(TRUE))
 896        || !new_cc_var(cc, sizeof(void*) == 8 ? _amd64W : _x86W, -1, ccval_bool(TRUE))
 897        || !new_cc_var(cc, _jscript_versionW, -1, ccval_num(JSCRIPT_MAJOR_VERSION + (DOUBLE)JSCRIPT_MINOR_VERSION/10.0))
 898        || !new_cc_var(cc, _jscript_buildW, -1, ccval_num(JSCRIPT_BUILD_VERSION))) {
 899         release_cc(cc);
 900         lex_error(ctx, E_OUTOFMEMORY);
 901         return FALSE;
 902     }
 903
 904     ctx->script->cc = cc;
 905     return TRUE;
 906 }
 907
 908 static BOOL parse_cc_identifier(parser_ctx_t *ctx, const WCHAR **ret, unsigned *ret_len)
 909 {
 910     if(*ctx->ptr != '@') {
 911         lex_error(ctx, JS_E_EXPECTED_AT);
 912         return FALSE;
 913     }
 914
 915     if(!is_identifier_first_char(*++ctx->ptr)) {
 916         lex_error(ctx, JS_E_EXPECTED_IDENTIFIER);
 917         return FALSE;
 918     }
 919
 920     *ret = ctx->ptr;
 921     while(++ctx->ptr < ctx->end && is_identifier_char(*ctx->ptr));
 922     *ret_len = ctx->ptr - *ret;
 923     return TRUE;
 924 }
 925
 926 int try_parse_ccval(parser_ctx_t *ctx, ccval_t *r)
 927 {
 928     if(!skip_spaces(ctx))
 929         return -1;
 930
 931     if(isdigitW(*ctx->ptr)) {
 932         double n;
 933
 934         if(!parse_numeric_literal(ctx, &n))
 935             return -1;
 936
 937         *r = ccval_num(n);
 938         return 1;
 939     }
 940
 941     if(*ctx->ptr == '@') {
 942         const WCHAR *ident;
 943         unsigned ident_len;
 944         cc_var_t *cc_var;
 945
 946         if(!parse_cc_identifier(ctx, &ident, &ident_len))
 947             return -1;
 948
 949         cc_var = find_cc_var(ctx->script->cc, ident, ident_len);
 950         *r = cc_var ? cc_var->val : ccval_num(NAN);
 951         return 1;
 952     }
 953
 954     if(!check_keyword(ctx, trueW, NULL)) {
 955         *r = ccval_bool(TRUE);
 956         return 1;
 957     }
 958
 959     if(!check_keyword(ctx, falseW, NULL)) {
 960         *r = ccval_bool(FALSE);
 961         return 1;
 962     }
 963
 964     return 0;
 965 }
 966
 967 static int skip_code(parser_ctx_t *ctx, BOOL exec_else)
 968 {
 969     int if_depth = 1;
 970     const WCHAR *ptr;
 971
 972     while(1) {
 973         ptr = strchrW(ctx->ptr, '@');
 974         if(!ptr) {
 975             WARN("No @end\n");
 976             return lex_error(ctx, JS_E_EXPECTED_CCEND);
 977         }
 978         ctx->ptr = ptr+1;
 979
 980         if(!check_keyword(ctx, endW, NULL)) {
 981             if(--if_depth)
 982                 continue;
 983             return 0;
 984         }
 985
 986         if(exec_else && !check_keyword(ctx, elifW, NULL)) {
 987             if(if_depth > 1)
 988                 continue;
 989
 990             if(!skip_spaces(ctx) || *ctx->ptr != '(')
 991                 return lex_error(ctx, JS_E_MISSING_LBRACKET);
 992
 993             if(!parse_cc_expr(ctx))
 994                 return -1;
 995
 996             if(!get_ccbool(ctx->ccval))
 997                 continue; /* skip block of code */
 998
 999             /* continue parsing */
1000             ctx->cc_if_depth++;
1001             return 0;
1002         }
1003
1004         if(exec_else && !check_keyword(ctx, elseW, NULL)) {
1005             if(if_depth > 1)
1006                 continue;
1007
1008             /* parse else block */
1009             ctx->cc_if_depth++;
1010             return 0;
1011         }
1012
1013         if(!check_keyword(ctx, ifW, NULL)) {
1014             if_depth++;
1015             continue;
1016         }
1017
1018         ctx->ptr++;
1019     }
1020 }
1021
1022 static int cc_token(parser_ctx_t *ctx, void *lval)
1023 {
1024     unsigned id_len = 0;
1025     cc_var_t *var;
1026
1027     static const WCHAR cc_onW[] = {'c','c','_','o','n',0};
1028     static const WCHAR setW[] = {'s','e','t',0};
1029
1030     ctx->ptr++;
1031
1032     if(!check_keyword(ctx, cc_onW, NULL))
1033         return init_cc(ctx) ? 0 : -1;
1034
1035     if(!check_keyword(ctx, setW, NULL)) {
1036         const WCHAR *ident;
1037         unsigned ident_len;
1038         cc_var_t *var;
1039
1040         if(!init_cc(ctx))
1041             return -1;
1042
1043         if(!skip_spaces(ctx))
1044             return lex_error(ctx, JS_E_EXPECTED_AT);
1045
1046         if(!parse_cc_identifier(ctx, &ident, &ident_len))
1047             return -1;
1048
1049         if(!skip_spaces(ctx) || *ctx->ptr != '=')
1050             return lex_error(ctx, JS_E_EXPECTED_ASSIGN);
1051         ctx->ptr++;
1052
1053         if(!parse_cc_expr(ctx)) {
1054             WARN("parsing CC expression failed\n");
1055             return -1;
1056         }
1057
1058         var = find_cc_var(ctx->script->cc, ident, ident_len);
1059         if(var) {
1060             var->val = ctx->ccval;
1061         }else {
1062             if(!new_cc_var(ctx->script->cc, ident, ident_len, ctx->ccval))
1063                 return lex_error(ctx, E_OUTOFMEMORY);
1064         }
1065
1066         return 0;
1067     }
1068
1069     if(!check_keyword(ctx, ifW, NULL)) {
1070         if(!init_cc(ctx))
1071             return -1;
1072
1073         if(!skip_spaces(ctx) || *ctx->ptr != '(')
1074             return lex_error(ctx, JS_E_MISSING_LBRACKET);
1075
1076         if(!parse_cc_expr(ctx))
1077             return -1;
1078
1079         if(get_ccbool(ctx->ccval)) {
1080             /* continue parsing block inside if */
1081             ctx->cc_if_depth++;
1082             return 0;
1083         }
1084
1085         return skip_code(ctx, TRUE);
1086     }
1087
1088     if(!check_keyword(ctx, elifW, NULL) || !check_keyword(ctx, elseW, NULL)) {
1089         if(!ctx->cc_if_depth)
1090             return lex_error(ctx, JS_E_SYNTAX);
1091
1092         return skip_code(ctx, FALSE);
1093     }
1094
1095     if(!check_keyword(ctx, endW, NULL)) {
1096         if(!ctx->cc_if_depth)
1097             return lex_error(ctx, JS_E_SYNTAX);
1098
1099         ctx->cc_if_depth--;
1100         return 0;
1101     }
1102
1103     if(!ctx->script->cc)
1104         return lex_error(ctx, JS_E_DISABLED_CC);
1105
1106     while(ctx->ptr+id_len < ctx->end && is_identifier_char(ctx->ptr[id_len]))
1107         id_len++;
1108     if(!id_len)
1109         return '@';
1110
1111     TRACE("var %s\n", debugstr_wn(ctx->ptr, id_len));
1112
1113     var = find_cc_var(ctx->script->cc, ctx->ptr, id_len);
1114     ctx->ptr += id_len;
1115     if(!var || var->val.is_num) {
1116         *(literal_t**)lval = new_double_literal(ctx, var ? var->val.u.n : NAN);
1117         return tNumericLiteral;
1118     }
1119
1120     *(literal_t**)lval = new_boolean_literal(ctx, var->val.u.b);
1121     return tBooleanLiteral;
1122 }
1123
1124 int parser_lex(void *lval, parser_ctx_t *ctx)
1125 {
1126     int ret;
1127
1128     ctx->nl = ctx->ptr == ctx->begin;
1129
1130     do {
1131         ret = next_token(ctx, lval);
1132     } while(ret == '@' && !(ret = cc_token(ctx, lval)));
1133
1134     return ret;
1135 }
1136
1137 literal_t *parse_regexp(parser_ctx_t *ctx)
1138 {
1139     const WCHAR *re, *flags_ptr;
1140     BOOL in_class = FALSE;
1141     DWORD re_len, flags;
1142     literal_t *ret;
1143     HRESULT hres;
1144
1145     TRACE("\n");
1146
1147     while(*--ctx->ptr != '/');
1148
1149     /* Simple regexp pre-parser; '/' if used in char class does not terminate regexp literal */
1150     re = ++ctx->ptr;
1151     while(ctx->ptr < ctx->end) {
1152         if(*ctx->ptr == '\\') {
1153             if(++ctx->ptr == ctx->end)
1154                 break;
1155         }else if(in_class) {
1156             if(*ctx->ptr == '\n')
1157                 break;
1158             if(*ctx->ptr == ']')
1159                 in_class = FALSE;
1160         }else {
1161             if(*ctx->ptr == '/')
1162                 break;
1163
1164             if(*ctx->ptr == '[')
1165                 in_class = TRUE;
1166         }
1167         ctx->ptr++;
1168     }
1169
1170     if(ctx->ptr == ctx->end || *ctx->ptr != '/') {
1171         WARN("pre-parsing failed\n");
1172         return NULL;
1173     }
1174
1175     re_len = ctx->ptr-re;
1176
1177     flags_ptr = ++ctx->ptr;
1178     while(ctx->ptr < ctx->end && isalnumW(*ctx->ptr))
1179         ctx->ptr++;
1180
1181     hres = parse_regexp_flags(flags_ptr, ctx->ptr-flags_ptr, &flags);
1182     if(FAILED(hres))
1183         return NULL;
1184
1185     ret = parser_alloc(ctx, sizeof(literal_t));
1186     ret->type = LT_REGEXP;
1187     ret->u.regexp.str = re;
1188     ret->u.regexp.str_len = re_len;
1189     ret->u.regexp.flags = flags;
1190     return ret;
1191 }