vbscript: Added ME expression parser implementation.
[wine] / dlls / vbscript / parser.y
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 %{
20
21 #include "vbscript.h"
22 #include "parse.h"
23
24 #include "wine/debug.h"
25
26 WINE_DEFAULT_DEBUG_CHANNEL(vbscript);
27
28
29 #define YYLEX_PARAM ctx
30 #define YYPARSE_PARAM ctx
31
32 static int parser_error(const char*);
33
34 static void parse_complete(parser_ctx_t*,BOOL);
35
36 static void source_add_statement(parser_ctx_t*,statement_t*);
37 static void source_add_class(parser_ctx_t*,class_decl_t*);
38
39 static void *new_expression(parser_ctx_t*,expression_type_t,size_t);
40 static expression_t *new_bool_expression(parser_ctx_t*,VARIANT_BOOL);
41 static expression_t *new_string_expression(parser_ctx_t*,const WCHAR*);
42 static expression_t *new_long_expression(parser_ctx_t*,expression_type_t,LONG);
43 static expression_t *new_double_expression(parser_ctx_t*,double);
44 static expression_t *new_unary_expression(parser_ctx_t*,expression_type_t,expression_t*);
45 static expression_t *new_binary_expression(parser_ctx_t*,expression_type_t,expression_t*,expression_t*);
46 static expression_t *new_new_expression(parser_ctx_t*,const WCHAR*);
47
48 static member_expression_t *new_member_expression(parser_ctx_t*,expression_t*,const WCHAR*);
49
50 static void *new_statement(parser_ctx_t*,statement_type_t,size_t);
51 static statement_t *new_call_statement(parser_ctx_t*,member_expression_t*);
52 static statement_t *new_assign_statement(parser_ctx_t*,member_expression_t*,expression_t*);
53 static statement_t *new_set_statement(parser_ctx_t*,member_expression_t*,expression_t*);
54 static statement_t *new_dim_statement(parser_ctx_t*,dim_decl_t*);
55 static statement_t *new_while_statement(parser_ctx_t*,statement_type_t,expression_t*,statement_t*);
56 static statement_t *new_if_statement(parser_ctx_t*,expression_t*,statement_t*,elseif_decl_t*,statement_t*);
57 static statement_t *new_function_statement(parser_ctx_t*,function_decl_t*);
58
59 static dim_decl_t *new_dim_decl(parser_ctx_t*,const WCHAR*,dim_decl_t*);
60 static elseif_decl_t *new_elseif_decl(parser_ctx_t*,expression_t*,statement_t*);
61 static function_decl_t *new_function_decl(parser_ctx_t*,const WCHAR*,function_type_t,unsigned,arg_decl_t*,statement_t*);
62 static arg_decl_t *new_argument_decl(parser_ctx_t*,const WCHAR*,BOOL);
63
64 static class_decl_t *new_class_decl(parser_ctx_t*);
65 static class_decl_t *add_class_function(parser_ctx_t*,class_decl_t*,function_decl_t*);
66 static class_decl_t *add_variant_prop(parser_ctx_t*,class_decl_t*,const WCHAR*,unsigned);
67
68 #define STORAGE_IS_PRIVATE    1
69 #define STORAGE_IS_DEFAULT    2
70
71 #define CHECK_ERROR if(((parser_ctx_t*)ctx)->hres != S_OK) YYABORT
72
73 %}
74
75 %pure_parser
76 %start Program
77
78 %union {
79     const WCHAR *string;
80     statement_t *statement;
81     expression_t *expression;
82     member_expression_t *member;
83     elseif_decl_t *elseif;
84     dim_decl_t *dim_decl;
85     function_decl_t *func_decl;
86     arg_decl_t *arg_decl;
87     class_decl_t *class_decl;
88     unsigned uint;
89     LONG lng;
90     BOOL bool;
91     double dbl;
92 }
93
94 %token tEOF tNL tREM tEMPTYBRACKETS
95 %token tTRUE tFALSE
96 %token tNOT tAND tOR tXOR tEQV tIMP tNEQ
97 %token tIS tLTEQ tGTEQ tMOD
98 %token tCALL tDIM tSUB tFUNCTION tPROPERTY tGET tLET
99 %token tIF tELSE tELSEIF tEND tTHEN tEXIT
100 %token tWHILE tWEND tDO tLOOP tUNTIL
101 %token tBYREF tBYVAL
102 %token tOPTION tEXPLICIT
103 %token tSTOP
104 %token tNOTHING tEMPTY tNULL
105 %token tCLASS tSET tNEW tPUBLIC tPRIVATE tDEFAULT tME
106 %token tERROR tNEXT tON tRESUME tGOTO
107 %token <string> tIdentifier tString
108 %token <lng> tLong tShort
109 %token <dbl> tDouble
110
111 %type <statement> Statement StatementNl StatementsNl StatementsNl_opt IfStatement Else_opt
112 %type <expression> Expression LiteralExpression PrimaryExpression EqualityExpression CallExpression
113 %type <expression> ConcatExpression AdditiveExpression ModExpression IntdivExpression MultiplicativeExpression ExpExpression
114 %type <expression> NotExpression UnaryExpression AndExpression OrExpression XorExpression EqvExpression
115 %type <member> MemberExpression
116 %type <expression> Arguments_opt ArgumentList_opt ArgumentList
117 %type <bool> OptionExplicit_opt DoType
118 %type <arg_decl> ArgumentsDecl_opt ArgumentDeclList ArgumentDecl
119 %type <func_decl> FunctionDecl PropertyDecl
120 %type <elseif> ElseIfs_opt ElseIfs ElseIf
121 %type <class_decl> ClassDeclaration ClassBody
122 %type <uint> Storage Storage_opt
123 %type <dim_decl> DimDeclList
124
125 %%
126
127 Program
128     : OptionExplicit_opt SourceElements tEOF    { parse_complete(ctx, $1); }
129
130 OptionExplicit_opt
131     : /* empty */                { $$ = FALSE; }
132     | tOPTION tEXPLICIT tNL      { $$ = TRUE; }
133
134 SourceElements
135     : /* empty */
136     | SourceElements StatementNl            { source_add_statement(ctx, $2); }
137     | SourceElements ClassDeclaration       { source_add_class(ctx, $2); }
138
139 StatementsNl_opt
140     : /* empty */                           { $$ = NULL; }
141     | StatementsNl                          { $$ = $1; }
142
143 StatementsNl
144     : StatementNl                           { $$ = $1; }
145     | StatementNl StatementsNl              { $1->next = $2; $$ = $1; }
146
147 StatementNl
148     : Statement tNL                 { $$ = $1; }
149
150 Statement
151     : MemberExpression ArgumentList_opt     { $1->args = $2; $$ = new_call_statement(ctx, $1); CHECK_ERROR; }
152     | tCALL MemberExpression Arguments_opt  { $2->args = $3; $$ = new_call_statement(ctx, $2); CHECK_ERROR; }
153     | MemberExpression Arguments_opt '=' Expression
154                                             { $1->args = $2; $$ = new_assign_statement(ctx, $1, $4); CHECK_ERROR; }
155     | tDIM DimDeclList                      { $$ = new_dim_statement(ctx, $2); CHECK_ERROR; }
156     | IfStatement                           { $$ = $1; }
157     | tWHILE Expression tNL StatementsNl_opt tWEND
158                                             { $$ = new_while_statement(ctx, STAT_WHILE, $2, $4); CHECK_ERROR; }
159     | tDO DoType Expression tNL StatementsNl_opt tLOOP
160                                             { $$ = new_while_statement(ctx, $2 ? STAT_WHILELOOP : STAT_UNTIL, $3, $5);
161                                               CHECK_ERROR; }
162     | tDO tNL StatementsNl_opt tLOOP DoType Expression
163                                             { $$ = new_while_statement(ctx, $5 ? STAT_DOWHILE : STAT_DOUNTIL, $6, $3);
164                                               CHECK_ERROR; }
165     | FunctionDecl                          { $$ = new_function_statement(ctx, $1); CHECK_ERROR; }
166     | tEXIT tDO                             { $$ = new_statement(ctx, STAT_EXITDO, 0); CHECK_ERROR; }
167     | tEXIT tFUNCTION                       { $$ = new_statement(ctx, STAT_EXITFUNC, 0); CHECK_ERROR; }
168     | tEXIT tPROPERTY                       { $$ = new_statement(ctx, STAT_EXITPROP, 0); CHECK_ERROR; }
169     | tEXIT tSUB                            { $$ = new_statement(ctx, STAT_EXITSUB, 0); CHECK_ERROR; }
170     | tSET MemberExpression Arguments_opt '=' Expression
171                                             { $2->args = $3; $$ = new_set_statement(ctx, $2, $5); CHECK_ERROR; }
172     | tSTOP                                 { $$ = new_statement(ctx, STAT_STOP, 0); CHECK_ERROR; }
173
174 MemberExpression
175     : tIdentifier                           { $$ = new_member_expression(ctx, NULL, $1); CHECK_ERROR; }
176     | CallExpression '.' tIdentifier        { $$ = new_member_expression(ctx, $1, $3); CHECK_ERROR; }
177
178 DimDeclList /* FIXME: Support arrays */
179     : tIdentifier                           { $$ = new_dim_decl(ctx, $1, NULL); CHECK_ERROR; }
180     | tIdentifier ',' DimDeclList           { $$ = new_dim_decl(ctx, $1, $3); CHECK_ERROR; }
181
182 DoType
183     : tWHILE        { $$ = TRUE; }
184     | tUNTIL        { $$ = FALSE; }
185
186 IfStatement
187     : tIF Expression tTHEN tNL StatementsNl ElseIfs_opt Else_opt tEND tIF
188                                             { $$ = new_if_statement(ctx, $2, $5, $6, $7); CHECK_ERROR; }
189     /* FIXME: short if statement */
190
191 ElseIfs_opt
192     : /* empty */                           { $$ = NULL; }
193     | ElseIfs                               { $$ = $1; }
194
195 ElseIfs
196     : ElseIf                                { $$ = $1; }
197     | ElseIf ElseIfs                        { $1->next = $2; $$ = $1; }
198
199 ElseIf
200     : tELSEIF Expression tTHEN tNL StatementsNl
201                                             { $$ = new_elseif_decl(ctx, $2, $5); }
202
203 Else_opt
204     : /* empty */                           { $$ = NULL; }
205     | tELSE tNL StatementsNl                { $$ = $3; }
206
207 Arguments_opt
208     : EmptyBrackets_opt             { $$ = NULL; }
209     | '(' ArgumentList ')'          { $$ = $2; }
210
211 ArgumentList_opt
212     : EmptyBrackets_opt             { $$ = NULL; }
213     | ArgumentList                  { $$ = $1; }
214
215 ArgumentList
216     : Expression                    { $$ = $1; }
217     | Expression ',' ArgumentList   { $1->next = $3; $$ = $1; }
218
219 EmptyBrackets_opt
220     : /* empty */
221     | tEMPTYBRACKETS
222
223 Expression
224     : EqvExpression                             { $$ = $1; }
225     | Expression tIMP EqvExpression             { $$ = new_binary_expression(ctx, EXPR_IMP, $1, $3); CHECK_ERROR; }
226
227 EqvExpression
228     : XorExpression                             { $$ = $1; }
229     | EqvExpression tEQV XorExpression          { $$ = new_binary_expression(ctx, EXPR_EQV, $1, $3); CHECK_ERROR; }
230
231 XorExpression
232     : OrExpression                              { $$ = $1; }
233     | XorExpression tXOR OrExpression           { $$ = new_binary_expression(ctx, EXPR_XOR, $1, $3); CHECK_ERROR; }
234
235 OrExpression
236     : AndExpression                             { $$ = $1; }
237     | OrExpression tOR AndExpression            { $$ = new_binary_expression(ctx, EXPR_OR, $1, $3); CHECK_ERROR; }
238
239 AndExpression
240     : NotExpression                             { $$ = $1; }
241     | AndExpression tAND NotExpression          { $$ = new_binary_expression(ctx, EXPR_AND, $1, $3); CHECK_ERROR; }
242
243 NotExpression
244     : EqualityExpression            { $$ = $1; }
245     | tNOT NotExpression            { $$ = new_unary_expression(ctx, EXPR_NOT, $2); CHECK_ERROR; }
246
247 EqualityExpression
248     : ConcatExpression                          { $$ = $1; }
249     | EqualityExpression '=' ConcatExpression   { $$ = new_binary_expression(ctx, EXPR_EQUAL, $1, $3); CHECK_ERROR; }
250     | EqualityExpression tNEQ ConcatExpression  { $$ = new_binary_expression(ctx, EXPR_NEQUAL, $1, $3); CHECK_ERROR; }
251     | EqualityExpression '>' ConcatExpression   { $$ = new_binary_expression(ctx, EXPR_GT, $1, $3); CHECK_ERROR; }
252     | EqualityExpression '<' ConcatExpression   { $$ = new_binary_expression(ctx, EXPR_LT, $1, $3); CHECK_ERROR; }
253     | EqualityExpression tGTEQ ConcatExpression { $$ = new_binary_expression(ctx, EXPR_GTEQ, $1, $3); CHECK_ERROR; }
254     | EqualityExpression tLTEQ ConcatExpression { $$ = new_binary_expression(ctx, EXPR_LTEQ, $1, $3); CHECK_ERROR; }
255     | EqualityExpression tIS ConcatExpression   { $$ = new_binary_expression(ctx, EXPR_IS, $1, $3); CHECK_ERROR; }
256
257 ConcatExpression
258     : AdditiveExpression                        { $$ = $1; }
259     | ConcatExpression '&' AdditiveExpression   { $$ = new_binary_expression(ctx, EXPR_CONCAT, $1, $3); CHECK_ERROR; }
260
261 AdditiveExpression
262     : ModExpression                             { $$ = $1; }
263     | AdditiveExpression '+' ModExpression      { $$ = new_binary_expression(ctx, EXPR_ADD, $1, $3); CHECK_ERROR; }
264     | AdditiveExpression '-' ModExpression      { $$ = new_binary_expression(ctx, EXPR_SUB, $1, $3); CHECK_ERROR; }
265
266 ModExpression
267     : IntdivExpression                          { $$ = $1; }
268     | ModExpression tMOD IntdivExpression       { $$ = new_binary_expression(ctx, EXPR_MOD, $1, $3); CHECK_ERROR; }
269
270 IntdivExpression
271     : MultiplicativeExpression                  { $$ = $1; }
272     | IntdivExpression '\\' MultiplicativeExpression
273                                                 { $$ = new_binary_expression(ctx, EXPR_IDIV, $1, $3); CHECK_ERROR; }
274
275 MultiplicativeExpression
276     : ExpExpression                             { $$ = $1; }
277     | MultiplicativeExpression '*' ExpExpression
278                                                 { $$ = new_binary_expression(ctx, EXPR_MUL, $1, $3); CHECK_ERROR; }
279     | MultiplicativeExpression '/' ExpExpression
280                                                 { $$ = new_binary_expression(ctx, EXPR_DIV, $1, $3); CHECK_ERROR; }
281
282 ExpExpression
283     : UnaryExpression                           { $$ = $1; }
284     | ExpExpression '^' UnaryExpression         { $$ = new_binary_expression(ctx, EXPR_EXP, $1, $3); CHECK_ERROR; }
285
286 UnaryExpression
287     : LiteralExpression             { $$ = $1; }
288     | CallExpression                { $$ = $1; }
289     | tNEW tIdentifier              { $$ = new_new_expression(ctx, $2); CHECK_ERROR; }
290     | '-' UnaryExpression           { $$ = new_unary_expression(ctx, EXPR_NEG, $2); CHECK_ERROR; }
291
292 CallExpression
293     : PrimaryExpression                 { $$ = $1; }
294     | MemberExpression Arguments_opt    { $1->args = $2; $$ = &$1->expr; }
295
296 LiteralExpression
297     : tTRUE                         { $$ = new_bool_expression(ctx, VARIANT_TRUE); CHECK_ERROR; }
298     | tFALSE                        { $$ = new_bool_expression(ctx, VARIANT_FALSE); CHECK_ERROR; }
299     | tString                       { $$ = new_string_expression(ctx, $1); CHECK_ERROR; }
300     | tShort                        { $$ = new_long_expression(ctx, EXPR_USHORT, $1); CHECK_ERROR; }
301     | '0'                           { $$ = new_long_expression(ctx, EXPR_USHORT, 0); CHECK_ERROR; }
302     | tLong                         { $$ = new_long_expression(ctx, EXPR_ULONG, $1); CHECK_ERROR; }
303     | tDouble                       { $$ = new_double_expression(ctx, $1); CHECK_ERROR; }
304     | tEMPTY                        { $$ = new_expression(ctx, EXPR_EMPTY, 0); CHECK_ERROR; }
305     | tNULL                         { $$ = new_expression(ctx, EXPR_NULL, 0); CHECK_ERROR; }
306     | tNOTHING                      { $$ = new_expression(ctx, EXPR_NOTHING, 0); CHECK_ERROR; }
307
308 PrimaryExpression
309     : '(' Expression ')'            { $$ = $2; }
310     | tME                           { $$ = new_expression(ctx, EXPR_ME, 0); CHECK_ERROR; }
311
312 ClassDeclaration
313     : tCLASS tIdentifier tNL ClassBody tEND tCLASS tNL      { $4->name = $2; $$ = $4; }
314
315 ClassBody
316     : /* empty */                               { $$ = new_class_decl(ctx); }
317     | FunctionDecl tNL ClassBody                { $$ = add_class_function(ctx, $3, $1); CHECK_ERROR; }
318     | Storage tIdentifier tNL ClassBody         { $$ = add_variant_prop(ctx, $4, $2, $1); CHECK_ERROR; }
319     | PropertyDecl tNL ClassBody                { $$ = add_class_function(ctx, $3, $1); CHECK_ERROR; }
320
321 PropertyDecl
322     : Storage_opt tPROPERTY tGET tIdentifier EmptyBrackets_opt tNL StatementsNl_opt tEND tPROPERTY
323                                     { $$ = new_function_decl(ctx, $4, FUNC_PROPGET, $1, NULL, $7); CHECK_ERROR; }
324     | Storage_opt tPROPERTY tLET tIdentifier '(' ArgumentDecl ')' tNL StatementsNl_opt tEND tPROPERTY
325                                     { $$ = new_function_decl(ctx, $4, FUNC_PROPLET, $1, $6, $9); CHECK_ERROR; }
326     | Storage_opt tPROPERTY tSET tIdentifier '(' ArgumentDecl ')' tNL StatementsNl_opt tEND tPROPERTY
327                                     { $$ = new_function_decl(ctx, $4, FUNC_PROPSET, $1, $6, $9); CHECK_ERROR; }
328
329 FunctionDecl
330     : Storage_opt tSUB tIdentifier ArgumentsDecl_opt tNL StatementsNl_opt tEND tSUB
331                                     { $$ = new_function_decl(ctx, $3, FUNC_SUB, $1, $4, $6); CHECK_ERROR; }
332     | Storage_opt tFUNCTION tIdentifier ArgumentsDecl_opt tNL StatementsNl_opt tEND tFUNCTION
333                                     { $$ = new_function_decl(ctx, $3, FUNC_FUNCTION, $1, $4, $6); CHECK_ERROR; }
334
335 Storage_opt
336     : /* empty*/                    { $$ = 0; }
337     | Storage                       { $$ = $1; }
338
339 Storage
340     : tPUBLIC tDEFAULT              { $$ = STORAGE_IS_DEFAULT; }
341     | tPUBLIC                       { $$ = 0; }
342     | tPRIVATE                      { $$ = STORAGE_IS_PRIVATE; }
343
344 ArgumentsDecl_opt
345     : EmptyBrackets_opt                         { $$ = NULL; }
346     | '(' ArgumentDeclList ')'                  { $$ = $2; }
347
348 ArgumentDeclList
349     : ArgumentDecl                              { $$ = $1; }
350     | ArgumentDecl ',' ArgumentDeclList         { $1->next = $3; $$ = $1; }
351
352 ArgumentDecl
353     : tIdentifier                               { $$ = new_argument_decl(ctx, $1, TRUE); }
354     | tBYREF tIdentifier                        { $$ = new_argument_decl(ctx, $2, TRUE); }
355     | tBYVAL tIdentifier                        { $$ = new_argument_decl(ctx, $2, FALSE); }
356
357 %%
358
359 static int parser_error(const char *str)
360 {
361     return 0;
362 }
363
364 static void source_add_statement(parser_ctx_t *ctx, statement_t *stat)
365 {
366     if(ctx->stats) {
367         ctx->stats_tail->next = stat;
368         ctx->stats_tail = stat;
369     }else {
370         ctx->stats = ctx->stats_tail = stat;
371     }
372 }
373
374 static void source_add_class(parser_ctx_t *ctx, class_decl_t *class_decl)
375 {
376     class_decl->next = ctx->class_decls;
377     ctx->class_decls = class_decl;
378 }
379
380 static void parse_complete(parser_ctx_t *ctx, BOOL option_explicit)
381 {
382     ctx->parse_complete = TRUE;
383     ctx->option_explicit = option_explicit;
384 }
385
386 static void *new_expression(parser_ctx_t *ctx, expression_type_t type, size_t size)
387 {
388     expression_t *expr;
389
390     expr = parser_alloc(ctx, size ? size : sizeof(*expr));
391     if(expr) {
392         expr->type = type;
393         expr->next = NULL;
394     }
395
396     return expr;
397 }
398
399 static expression_t *new_bool_expression(parser_ctx_t *ctx, VARIANT_BOOL value)
400 {
401     bool_expression_t *expr;
402
403     expr = new_expression(ctx, EXPR_BOOL, sizeof(*expr));
404     if(!expr)
405         return NULL;
406
407     expr->value = value;
408     return &expr->expr;
409 }
410
411 static expression_t *new_string_expression(parser_ctx_t *ctx, const WCHAR *value)
412 {
413     string_expression_t *expr;
414
415     expr = new_expression(ctx, EXPR_STRING, sizeof(*expr));
416     if(!expr)
417         return NULL;
418
419     expr->value = value;
420     return &expr->expr;
421 }
422
423 static expression_t *new_long_expression(parser_ctx_t *ctx, expression_type_t type, LONG value)
424 {
425     int_expression_t *expr;
426
427     expr = new_expression(ctx, type, sizeof(*expr));
428     if(!expr)
429         return NULL;
430
431     expr->value = value;
432     return &expr->expr;
433 }
434
435 static expression_t *new_double_expression(parser_ctx_t *ctx, double value)
436 {
437     double_expression_t *expr;
438
439     expr = new_expression(ctx, EXPR_DOUBLE, sizeof(*expr));
440     if(!expr)
441         return NULL;
442
443     expr->value = value;
444     return &expr->expr;
445 }
446
447 static expression_t *new_unary_expression(parser_ctx_t *ctx, expression_type_t type, expression_t *subexpr)
448 {
449     unary_expression_t *expr;
450
451     expr = new_expression(ctx, type, sizeof(*expr));
452     if(!expr)
453         return NULL;
454
455     expr->subexpr = subexpr;
456     return &expr->expr;
457 }
458
459 static expression_t *new_binary_expression(parser_ctx_t *ctx, expression_type_t type, expression_t *left, expression_t *right)
460 {
461     binary_expression_t *expr;
462
463     expr = new_expression(ctx, type, sizeof(*expr));
464     if(!expr)
465         return NULL;
466
467     expr->left = left;
468     expr->right = right;
469     return &expr->expr;
470 }
471
472 static member_expression_t *new_member_expression(parser_ctx_t *ctx, expression_t *obj_expr, const WCHAR *identifier)
473 {
474     member_expression_t *expr;
475
476     expr = new_expression(ctx, EXPR_MEMBER, sizeof(*expr));
477     if(!expr)
478         return NULL;
479
480     expr->obj_expr = obj_expr;
481     expr->identifier = identifier;
482     expr->args = NULL;
483     return expr;
484 }
485
486 static expression_t *new_new_expression(parser_ctx_t *ctx, const WCHAR *identifier)
487 {
488     string_expression_t *expr;
489
490     expr = new_expression(ctx, EXPR_NEW, sizeof(*expr));
491     if(!expr)
492         return NULL;
493
494     expr->value = identifier;
495     return &expr->expr;
496 }
497
498 static void *new_statement(parser_ctx_t *ctx, statement_type_t type, size_t size)
499 {
500     statement_t *stat;
501
502     stat = parser_alloc(ctx, size ? size : sizeof(*stat));
503     if(stat) {
504         stat->type = type;
505         stat->next = NULL;
506     }
507
508     return stat;
509 }
510
511 static statement_t *new_call_statement(parser_ctx_t *ctx, member_expression_t *expr)
512 {
513     call_statement_t *stat;
514
515     stat = new_statement(ctx, STAT_CALL, sizeof(*stat));
516     if(!stat)
517         return NULL;
518
519     stat->expr = expr;
520     return &stat->stat;
521 }
522
523 static statement_t *new_assign_statement(parser_ctx_t *ctx, member_expression_t *left, expression_t *right)
524 {
525     assign_statement_t *stat;
526
527     stat = new_statement(ctx, STAT_ASSIGN, sizeof(*stat));
528     if(!stat)
529         return NULL;
530
531     stat->member_expr = left;
532     stat->value_expr = right;
533     return &stat->stat;
534 }
535
536 static statement_t *new_set_statement(parser_ctx_t *ctx, member_expression_t *left, expression_t *right)
537 {
538     assign_statement_t *stat;
539
540     stat = new_statement(ctx, STAT_SET, sizeof(*stat));
541     if(!stat)
542         return NULL;
543
544     stat->member_expr = left;
545     stat->value_expr = right;
546     return &stat->stat;
547 }
548
549 static dim_decl_t *new_dim_decl(parser_ctx_t *ctx, const WCHAR *name, dim_decl_t *next)
550 {
551     dim_decl_t *decl;
552
553     decl = parser_alloc(ctx, sizeof(*decl));
554     if(!decl)
555         return NULL;
556
557     decl->name = name;
558     decl->next = next;
559     return decl;
560 }
561
562 static statement_t *new_dim_statement(parser_ctx_t *ctx, dim_decl_t *decls)
563 {
564     dim_statement_t *stat;
565
566     stat = new_statement(ctx, STAT_DIM, sizeof(*stat));
567     if(!stat)
568         return NULL;
569
570     stat->dim_decls = decls;
571     return &stat->stat;
572 }
573
574 static elseif_decl_t *new_elseif_decl(parser_ctx_t *ctx, expression_t *expr, statement_t *stat)
575 {
576     elseif_decl_t *decl;
577
578     decl = parser_alloc(ctx, sizeof(*decl));
579     if(!decl)
580         return NULL;
581
582     decl->expr = expr;
583     decl->stat = stat;
584     decl->next = NULL;
585     return decl;
586 }
587
588 static statement_t *new_while_statement(parser_ctx_t *ctx, statement_type_t type, expression_t *expr, statement_t *body)
589 {
590     while_statement_t *stat;
591
592     stat = new_statement(ctx, type, sizeof(*stat));
593     if(!stat)
594         return NULL;
595
596     stat->expr = expr;
597     stat->body = body;
598     return &stat->stat;
599 }
600
601 static statement_t *new_if_statement(parser_ctx_t *ctx, expression_t *expr, statement_t *if_stat, elseif_decl_t *elseif_decl,
602         statement_t *else_stat)
603 {
604     if_statement_t *stat;
605
606     stat = new_statement(ctx, STAT_IF, sizeof(*stat));
607     if(!stat)
608         return NULL;
609
610     stat->expr = expr;
611     stat->if_stat = if_stat;
612     stat->elseifs = elseif_decl;
613     stat->else_stat = else_stat;
614     return &stat->stat;
615 }
616
617 static arg_decl_t *new_argument_decl(parser_ctx_t *ctx, const WCHAR *name, BOOL by_ref)
618 {
619     arg_decl_t *arg_decl;
620
621     arg_decl = parser_alloc(ctx, sizeof(*arg_decl));
622     if(!arg_decl)
623         return NULL;
624
625     arg_decl->name = name;
626     arg_decl->by_ref = by_ref;
627     arg_decl->next = NULL;
628     return arg_decl;
629 }
630
631 static function_decl_t *new_function_decl(parser_ctx_t *ctx, const WCHAR *name, function_type_t type,
632         unsigned storage_flags, arg_decl_t *arg_decl, statement_t *body)
633 {
634     function_decl_t *decl;
635
636     if(storage_flags & STORAGE_IS_DEFAULT) {
637         if(type == FUNC_PROPGET) {
638             type = FUNC_DEFGET;
639         }else {
640             FIXME("Invalid default property\n");
641             ctx->hres = E_FAIL;
642             return NULL;
643         }
644     }
645
646     decl = parser_alloc(ctx, sizeof(*decl));
647     if(!decl)
648         return NULL;
649
650     decl->name = name;
651     decl->type = type;
652     decl->is_public = !(storage_flags & STORAGE_IS_PRIVATE);
653     decl->args = arg_decl;
654     decl->body = body;
655     decl->next = NULL;
656     decl->next_prop_func = NULL;
657     return decl;
658 }
659
660 static statement_t *new_function_statement(parser_ctx_t *ctx, function_decl_t *decl)
661 {
662     function_statement_t *stat;
663
664     stat = new_statement(ctx, STAT_FUNC, sizeof(*stat));
665     if(!stat)
666         return NULL;
667
668     stat->func_decl = decl;
669     return &stat->stat;
670 }
671
672 static class_decl_t *new_class_decl(parser_ctx_t *ctx)
673 {
674     class_decl_t *class_decl;
675
676     class_decl = parser_alloc(ctx, sizeof(*class_decl));
677     if(!class_decl)
678         return NULL;
679
680     class_decl->funcs = NULL;
681     class_decl->props = NULL;
682     class_decl->next = NULL;
683     return class_decl;
684 }
685
686 static class_decl_t *add_class_function(parser_ctx_t *ctx, class_decl_t *class_decl, function_decl_t *decl)
687 {
688     function_decl_t *iter;
689
690     for(iter = class_decl->funcs; iter; iter = iter->next) {
691         if(!strcmpiW(iter->name, decl->name)) {
692             if(decl->type == FUNC_SUB || decl->type == FUNC_FUNCTION) {
693                 FIXME("Redefinition of %s::%s\n", debugstr_w(class_decl->name), debugstr_w(decl->name));
694                 ctx->hres = E_FAIL;
695                 return NULL;
696             }
697
698             while(1) {
699                 if(iter->type == decl->type) {
700                     FIXME("Redefinition of %s::%s\n", debugstr_w(class_decl->name), debugstr_w(decl->name));
701                     ctx->hres = E_FAIL;
702                     return NULL;
703                 }
704                 if(!iter->next_prop_func)
705                     break;
706                 iter = iter->next_prop_func;
707             }
708
709             iter->next_prop_func = decl;
710             return class_decl;
711         }
712     }
713
714     decl->next = class_decl->funcs;
715     class_decl->funcs = decl;
716     return class_decl;
717 }
718
719 static class_decl_t *add_variant_prop(parser_ctx_t *ctx, class_decl_t *class_decl, const WCHAR *identifier, unsigned storage_flags)
720 {
721     class_prop_decl_t *prop;
722
723     if(storage_flags & STORAGE_IS_DEFAULT) {
724         FIXME("variant prop van't be default value\n");
725         ctx->hres = E_FAIL;
726         return NULL;
727     }
728
729     prop = parser_alloc(ctx, sizeof(*prop));
730     if(!prop)
731         return NULL;
732
733     prop->name = identifier;
734     prop->is_public = !(storage_flags & STORAGE_IS_PRIVATE);
735     prop->next = class_decl->props;
736     class_decl->props = prop;
737     return class_decl;
738 }
739
740 void *parser_alloc(parser_ctx_t *ctx, size_t size)
741 {
742     void *ret;
743
744     ret = vbsheap_alloc(&ctx->heap, size);
745     if(!ret)
746         ctx->hres = E_OUTOFMEMORY;
747     return ret;
748 }
749
750 HRESULT parse_script(parser_ctx_t *ctx, const WCHAR *code)
751 {
752     ctx->code = ctx->ptr = code;
753     ctx->end = ctx->code + strlenW(ctx->code);
754
755     vbsheap_init(&ctx->heap);
756
757     ctx->parse_complete = FALSE;
758     ctx->hres = S_OK;
759
760     ctx->last_token = tNL;
761     ctx->last_nl = 0;
762     ctx->stats = ctx->stats_tail = NULL;
763     ctx->class_decls = NULL;
764     ctx->option_explicit = FALSE;
765
766     parser_parse(ctx);
767
768     if(FAILED(ctx->hres))
769         return ctx->hres;
770     if(!ctx->parse_complete) {
771         FIXME("parser failed on parsing %s\n", debugstr_w(ctx->ptr));
772         return E_FAIL;
773     }
774
775     return S_OK;
776 }
777
778 void parser_release(parser_ctx_t *ctx)
779 {
780     vbsheap_free(&ctx->heap);
781 }