vbscript: Added exit sub statement 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
38 static void *new_expression(parser_ctx_t*,expression_type_t,size_t);
39 static expression_t *new_bool_expression(parser_ctx_t*,VARIANT_BOOL);
40 static expression_t *new_string_expression(parser_ctx_t*,const WCHAR*);
41 static expression_t *new_long_expression(parser_ctx_t*,expression_type_t,LONG);
42 static expression_t *new_double_expression(parser_ctx_t*,double);
43 static expression_t *new_unary_expression(parser_ctx_t*,expression_type_t,expression_t*);
44 static expression_t *new_binary_expression(parser_ctx_t*,expression_type_t,expression_t*,expression_t*);
45
46 static member_expression_t *new_member_expression(parser_ctx_t*,expression_t*,const WCHAR*);
47
48 static void *new_statement(parser_ctx_t*,statement_type_t,size_t);
49 static statement_t *new_call_statement(parser_ctx_t*,member_expression_t*);
50 static statement_t *new_assign_statement(parser_ctx_t*,member_expression_t*,expression_t*);
51 static statement_t *new_dim_statement(parser_ctx_t*,dim_decl_t*);
52 static statement_t *new_if_statement(parser_ctx_t*,expression_t*,statement_t*,elseif_decl_t*,statement_t*);
53 static statement_t *new_function_statement(parser_ctx_t*,function_decl_t*);
54
55 static dim_decl_t *new_dim_decl(parser_ctx_t*,const WCHAR*,dim_decl_t*);
56 static elseif_decl_t *new_elseif_decl(parser_ctx_t*,expression_t*,statement_t*);
57 static function_decl_t *new_function_decl(parser_ctx_t*,const WCHAR*,function_type_t,arg_decl_t*,statement_t*);
58 static arg_decl_t *new_argument_decl(parser_ctx_t*,const WCHAR*,BOOL);
59
60 #define CHECK_ERROR if(((parser_ctx_t*)ctx)->hres != S_OK) YYABORT
61
62 %}
63
64 %pure_parser
65 %start Program
66
67 %union {
68     const WCHAR *string;
69     statement_t *statement;
70     expression_t *expression;
71     member_expression_t *member;
72     elseif_decl_t *elseif;
73     dim_decl_t *dim_decl;
74     function_decl_t *func_decl;
75     arg_decl_t *arg_decl;
76     LONG lng;
77     BOOL bool;
78     double dbl;
79 }
80
81 %token tEOF tNL tREM tEMPTYBRACKETS
82 %token tTRUE tFALSE
83 %token tNOT tAND tOR tXOR tEQV tIMP tNEQ
84 %token tIS tLTEQ tGTEQ tMOD
85 %token tCALL tDIM tSUB tFUNCTION tPROPERTY tGET tLET
86 %token tIF tELSE tELSEIF tEND tTHEN tEXIT
87 %token tWHILE tWEND tDO tLOOP tUNTIL
88 %token tBYREF tBYVAL
89 %token tOPTION tEXPLICIT
90 %token tSTOP
91 %token tNOTHING tEMPTY tNULL
92 %token tCLASS tSET tNEW tPUBLIC tPRIVATE tDEFAULT tME
93 %token tERROR tNEXT tON tRESUME tGOTO
94 %token <string> tIdentifier tString
95 %token <lng> tLong tShort
96 %token <dbl> tDouble
97
98 %type <statement> Statement StatementNl StatementsNl StatementsNl_opt IfStatement Else_opt
99 %type <expression> Expression LiteralExpression PrimaryExpression EqualityExpression CallExpression
100 %type <expression> ConcatExpression AdditiveExpression ModExpression IntdivExpression MultiplicativeExpression ExpExpression
101 %type <expression> NotExpression UnaryExpression
102 %type <member> MemberExpression
103 %type <expression> Arguments_opt ArgumentList_opt ArgumentList
104 %type <bool> OptionExplicit_opt
105 %type <arg_decl> ArgumentsDecl_opt ArgumentDeclList ArgumentDecl
106 %type <func_decl> FunctionDecl
107 %type <elseif> ElseIfs_opt ElseIfs ElseIf
108 %type <dim_decl> DimDeclList
109
110 %%
111
112 Program
113     : OptionExplicit_opt SourceElements tEOF    { parse_complete(ctx, $1); }
114
115 OptionExplicit_opt
116     : /* empty */                { $$ = FALSE; }
117     | tOPTION tEXPLICIT tNL      { $$ = TRUE; }
118
119 SourceElements
120     : /* empty */
121     | SourceElements StatementNl    { source_add_statement(ctx, $2); }
122
123 StatementsNl_opt
124     : /* empty */                           { $$ = NULL; }
125     | StatementsNl                          { $$ = $1; }
126
127 StatementsNl
128     : StatementNl                           { $$ = $1; }
129     | StatementNl StatementsNl              { $1->next = $2; $$ = $1; }
130
131 StatementNl
132     : Statement tNL                 { $$ = $1; }
133
134 Statement
135     : MemberExpression ArgumentList_opt     { $1->args = $2; $$ = new_call_statement(ctx, $1); CHECK_ERROR; }
136     | tCALL MemberExpression Arguments_opt  { $2->args = $3; $$ = new_call_statement(ctx, $2); CHECK_ERROR; }
137     | MemberExpression Arguments_opt '=' Expression
138                                             { $1->args = $2; $$ = new_assign_statement(ctx, $1, $4); CHECK_ERROR; }
139     | tDIM DimDeclList                      { $$ = new_dim_statement(ctx, $2); CHECK_ERROR; }
140     | IfStatement                           { $$ = $1; }
141     | FunctionDecl                          { $$ = new_function_statement(ctx, $1); CHECK_ERROR; }
142     | tEXIT tSUB                            { $$ = new_statement(ctx, STAT_EXITSUB, 0); CHECK_ERROR; }
143
144 MemberExpression
145     : tIdentifier                           { $$ = new_member_expression(ctx, NULL, $1); CHECK_ERROR; }
146     | CallExpression '.' tIdentifier        { $$ = new_member_expression(ctx, $1, $3); CHECK_ERROR; }
147
148 DimDeclList /* FIXME: Support arrays */
149     : tIdentifier                           { $$ = new_dim_decl(ctx, $1, NULL); CHECK_ERROR; }
150     | tIdentifier ',' DimDeclList           { $$ = new_dim_decl(ctx, $1, $3); CHECK_ERROR; }
151
152 IfStatement
153     : tIF Expression tTHEN tNL StatementsNl ElseIfs_opt Else_opt tEND tIF
154                                             { $$ = new_if_statement(ctx, $2, $5, $6, $7); CHECK_ERROR; }
155     /* FIXME: short if statement */
156
157 ElseIfs_opt
158     : /* empty */                           { $$ = NULL; }
159     | ElseIfs                               { $$ = $1; }
160
161 ElseIfs
162     : ElseIf                                { $$ = $1; }
163     | ElseIf ElseIfs                        { $1->next = $2; $$ = $1; }
164
165 ElseIf
166     : tELSEIF Expression tTHEN tNL StatementsNl
167                                             { $$ = new_elseif_decl(ctx, $2, $5); }
168
169 Else_opt
170     : /* empty */                           { $$ = NULL; }
171     | tELSE tNL StatementsNl                { $$ = $3; }
172
173 Arguments_opt
174     : EmptyBrackets_opt             { $$ = NULL; }
175     | '(' ArgumentList ')'          { $$ = $2; }
176
177 ArgumentList_opt
178     : EmptyBrackets_opt             { $$ = NULL; }
179     | ArgumentList                  { $$ = $1; }
180
181 ArgumentList
182     : Expression                    { $$ = $1; }
183     | Expression ',' ArgumentList   { $1->next = $3; $$ = $1; }
184
185 EmptyBrackets_opt
186     : /* empty */
187     | tEMPTYBRACKETS
188
189 Expression
190     : NotExpression                 { $$ = $1; }
191
192 NotExpression
193     : EqualityExpression            { $$ = $1; }
194     | tNOT NotExpression            { $$ = new_unary_expression(ctx, EXPR_NOT, $2); CHECK_ERROR; }
195
196 EqualityExpression
197     : ConcatExpression                          { $$ = $1; }
198     | EqualityExpression '=' ConcatExpression   { $$ = new_binary_expression(ctx, EXPR_EQUAL, $1, $3); CHECK_ERROR; }
199     | EqualityExpression tNEQ ConcatExpression  { $$ = new_binary_expression(ctx, EXPR_NEQUAL, $1, $3); CHECK_ERROR; }
200
201 ConcatExpression
202     : AdditiveExpression                        { $$ = $1; }
203     | ConcatExpression '&' AdditiveExpression   { $$ = new_binary_expression(ctx, EXPR_CONCAT, $1, $3); CHECK_ERROR; }
204
205 AdditiveExpression
206     : ModExpression                             { $$ = $1; }
207     | AdditiveExpression '+' ModExpression      { $$ = new_binary_expression(ctx, EXPR_ADD, $1, $3); CHECK_ERROR; }
208     | AdditiveExpression '-' ModExpression      { $$ = new_binary_expression(ctx, EXPR_SUB, $1, $3); CHECK_ERROR; }
209
210 ModExpression
211     : IntdivExpression                          { $$ = $1; }
212     | ModExpression tMOD IntdivExpression       { $$ = new_binary_expression(ctx, EXPR_MOD, $1, $3); CHECK_ERROR; }
213
214 IntdivExpression
215     : MultiplicativeExpression                  { $$ = $1; }
216     | IntdivExpression '\\' MultiplicativeExpression
217                                                 { $$ = new_binary_expression(ctx, EXPR_IDIV, $1, $3); CHECK_ERROR; }
218
219 MultiplicativeExpression
220     : ExpExpression                             { $$ = $1; }
221     | MultiplicativeExpression '*' ExpExpression
222                                                 { $$ = new_binary_expression(ctx, EXPR_MUL, $1, $3); CHECK_ERROR; }
223     | MultiplicativeExpression '/' ExpExpression
224                                                 { $$ = new_binary_expression(ctx, EXPR_DIV, $1, $3); CHECK_ERROR; }
225
226 ExpExpression
227     : UnaryExpression                           { $$ = $1; }
228     | ExpExpression '^' UnaryExpression         { $$ = new_binary_expression(ctx, EXPR_EXP, $1, $3); CHECK_ERROR; }
229
230 UnaryExpression
231     : LiteralExpression             { $$ = $1; }
232     | CallExpression                { $$ = $1; }
233     | '-' UnaryExpression           { $$ = new_unary_expression(ctx, EXPR_NEG, $2); CHECK_ERROR; }
234
235 CallExpression
236     : PrimaryExpression                 { $$ = $1; }
237     | MemberExpression Arguments_opt    { $1->args = $2; $$ = &$1->expr; }
238
239 LiteralExpression
240     : tTRUE                         { $$ = new_bool_expression(ctx, VARIANT_TRUE); CHECK_ERROR; }
241     | tFALSE                        { $$ = new_bool_expression(ctx, VARIANT_FALSE); CHECK_ERROR; }
242     | tString                       { $$ = new_string_expression(ctx, $1); CHECK_ERROR; }
243     | tShort                        { $$ = new_long_expression(ctx, EXPR_USHORT, $1); CHECK_ERROR; }
244     | '0'                           { $$ = new_long_expression(ctx, EXPR_USHORT, 0); CHECK_ERROR; }
245     | tLong                         { $$ = new_long_expression(ctx, EXPR_ULONG, $1); CHECK_ERROR; }
246     | tDouble                       { $$ = new_double_expression(ctx, $1); CHECK_ERROR; }
247     | tEMPTY                        { $$ = new_expression(ctx, EXPR_EMPTY, 0); CHECK_ERROR; }
248     | tNULL                         { $$ = new_expression(ctx, EXPR_NULL, 0); CHECK_ERROR; }
249
250 PrimaryExpression
251     : '(' Expression ')'            { $$ = $2; }
252
253 FunctionDecl
254     : /* Storage_opt */ tSUB tIdentifier ArgumentsDecl_opt tNL StatementsNl_opt tEND tSUB
255                                     { $$ = new_function_decl(ctx, $2, FUNC_SUB, $3, $5); CHECK_ERROR; }
256
257 ArgumentsDecl_opt
258     : EmptyBrackets_opt                         { $$ = NULL; }
259     | '(' ArgumentDeclList ')'                  { $$ = $2; }
260
261 ArgumentDeclList
262     : ArgumentDecl                              { $$ = $1; }
263     | ArgumentDecl ',' ArgumentDeclList         { $1->next = $3; $$ = $1; }
264
265 ArgumentDecl
266     : tIdentifier                               { $$ = new_argument_decl(ctx, $1, TRUE); }
267     | tBYREF tIdentifier                        { $$ = new_argument_decl(ctx, $2, TRUE); }
268     | tBYVAL tIdentifier                        { $$ = new_argument_decl(ctx, $2, FALSE); }
269
270 %%
271
272 static int parser_error(const char *str)
273 {
274     return 0;
275 }
276
277 static void source_add_statement(parser_ctx_t *ctx, statement_t *stat)
278 {
279     if(ctx->stats) {
280         ctx->stats_tail->next = stat;
281         ctx->stats_tail = stat;
282     }else {
283         ctx->stats = ctx->stats_tail = stat;
284     }
285 }
286
287 static void parse_complete(parser_ctx_t *ctx, BOOL option_explicit)
288 {
289     ctx->parse_complete = TRUE;
290     ctx->option_explicit = option_explicit;
291 }
292
293 static void *new_expression(parser_ctx_t *ctx, expression_type_t type, size_t size)
294 {
295     expression_t *expr;
296
297     expr = parser_alloc(ctx, size ? size : sizeof(*expr));
298     if(expr) {
299         expr->type = type;
300         expr->next = NULL;
301     }
302
303     return expr;
304 }
305
306 static expression_t *new_bool_expression(parser_ctx_t *ctx, VARIANT_BOOL value)
307 {
308     bool_expression_t *expr;
309
310     expr = new_expression(ctx, EXPR_BOOL, sizeof(*expr));
311     if(!expr)
312         return NULL;
313
314     expr->value = value;
315     return &expr->expr;
316 }
317
318 static expression_t *new_string_expression(parser_ctx_t *ctx, const WCHAR *value)
319 {
320     string_expression_t *expr;
321
322     expr = new_expression(ctx, EXPR_STRING, sizeof(*expr));
323     if(!expr)
324         return NULL;
325
326     expr->value = value;
327     return &expr->expr;
328 }
329
330 static expression_t *new_long_expression(parser_ctx_t *ctx, expression_type_t type, LONG value)
331 {
332     int_expression_t *expr;
333
334     expr = new_expression(ctx, type, sizeof(*expr));
335     if(!expr)
336         return NULL;
337
338     expr->value = value;
339     return &expr->expr;
340 }
341
342 static expression_t *new_double_expression(parser_ctx_t *ctx, double value)
343 {
344     double_expression_t *expr;
345
346     expr = new_expression(ctx, EXPR_DOUBLE, sizeof(*expr));
347     if(!expr)
348         return NULL;
349
350     expr->value = value;
351     return &expr->expr;
352 }
353
354 static expression_t *new_unary_expression(parser_ctx_t *ctx, expression_type_t type, expression_t *subexpr)
355 {
356     unary_expression_t *expr;
357
358     expr = new_expression(ctx, type, sizeof(*expr));
359     if(!expr)
360         return NULL;
361
362     expr->subexpr = subexpr;
363     return &expr->expr;
364 }
365
366 static expression_t *new_binary_expression(parser_ctx_t *ctx, expression_type_t type, expression_t *left, expression_t *right)
367 {
368     binary_expression_t *expr;
369
370     expr = new_expression(ctx, type, sizeof(*expr));
371     if(!expr)
372         return NULL;
373
374     expr->left = left;
375     expr->right = right;
376     return &expr->expr;
377 }
378
379 static member_expression_t *new_member_expression(parser_ctx_t *ctx, expression_t *obj_expr, const WCHAR *identifier)
380 {
381     member_expression_t *expr;
382
383     expr = new_expression(ctx, EXPR_MEMBER, sizeof(*expr));
384     if(!expr)
385         return NULL;
386
387     expr->obj_expr = obj_expr;
388     expr->identifier = identifier;
389     expr->args = NULL;
390     return expr;
391 }
392
393 static void *new_statement(parser_ctx_t *ctx, statement_type_t type, size_t size)
394 {
395     statement_t *stat;
396
397     stat = parser_alloc(ctx, size ? size : sizeof(*stat));
398     if(stat) {
399         stat->type = type;
400         stat->next = NULL;
401     }
402
403     return stat;
404 }
405
406 static statement_t *new_call_statement(parser_ctx_t *ctx, member_expression_t *expr)
407 {
408     call_statement_t *stat;
409
410     stat = new_statement(ctx, STAT_CALL, sizeof(*stat));
411     if(!stat)
412         return NULL;
413
414     stat->expr = expr;
415     return &stat->stat;
416 }
417
418 static statement_t *new_assign_statement(parser_ctx_t *ctx, member_expression_t *left, expression_t *right)
419 {
420     assign_statement_t *stat;
421
422     stat = new_statement(ctx, STAT_ASSIGN, sizeof(*stat));
423     if(!stat)
424         return NULL;
425
426     stat->member_expr = left;
427     stat->value_expr = right;
428     return &stat->stat;
429 }
430
431 static dim_decl_t *new_dim_decl(parser_ctx_t *ctx, const WCHAR *name, dim_decl_t *next)
432 {
433     dim_decl_t *decl;
434
435     decl = parser_alloc(ctx, sizeof(*decl));
436     if(!decl)
437         return NULL;
438
439     decl->name = name;
440     decl->next = next;
441     return decl;
442 }
443
444 static statement_t *new_dim_statement(parser_ctx_t *ctx, dim_decl_t *decls)
445 {
446     dim_statement_t *stat;
447
448     stat = new_statement(ctx, STAT_DIM, sizeof(*stat));
449     if(!stat)
450         return NULL;
451
452     stat->dim_decls = decls;
453     return &stat->stat;
454 }
455
456 static elseif_decl_t *new_elseif_decl(parser_ctx_t *ctx, expression_t *expr, statement_t *stat)
457 {
458     elseif_decl_t *decl;
459
460     decl = parser_alloc(ctx, sizeof(*decl));
461     if(!decl)
462         return NULL;
463
464     decl->expr = expr;
465     decl->stat = stat;
466     decl->next = NULL;
467     return decl;
468 }
469
470 static statement_t *new_if_statement(parser_ctx_t *ctx, expression_t *expr, statement_t *if_stat, elseif_decl_t *elseif_decl,
471         statement_t *else_stat)
472 {
473     if_statement_t *stat;
474
475     stat = new_statement(ctx, STAT_IF, sizeof(*stat));
476     if(!stat)
477         return NULL;
478
479     stat->expr = expr;
480     stat->if_stat = if_stat;
481     stat->elseifs = elseif_decl;
482     stat->else_stat = else_stat;
483     return &stat->stat;
484 }
485
486 static arg_decl_t *new_argument_decl(parser_ctx_t *ctx, const WCHAR *name, BOOL by_ref)
487 {
488     arg_decl_t *arg_decl;
489
490     arg_decl = parser_alloc(ctx, sizeof(*arg_decl));
491     if(!arg_decl)
492         return NULL;
493
494     arg_decl->name = name;
495     arg_decl->by_ref = by_ref;
496     arg_decl->next = NULL;
497     return arg_decl;
498 }
499
500 static function_decl_t *new_function_decl(parser_ctx_t *ctx, const WCHAR *name, function_type_t type,
501         arg_decl_t *arg_decl, statement_t *body)
502 {
503     function_decl_t *decl;
504
505     decl = parser_alloc(ctx, sizeof(*decl));
506     if(!decl)
507         return NULL;
508
509     decl->name = name;
510     decl->type = type;
511     decl->args = arg_decl;
512     decl->body = body;
513     return decl;
514 }
515
516 static statement_t *new_function_statement(parser_ctx_t *ctx, function_decl_t *decl)
517 {
518     function_statement_t *stat;
519
520     stat = new_statement(ctx, STAT_FUNC, sizeof(*stat));
521     if(!stat)
522         return NULL;
523
524     stat->func_decl = decl;
525     return &stat->stat;
526 }
527
528 void *parser_alloc(parser_ctx_t *ctx, size_t size)
529 {
530     void *ret;
531
532     ret = vbsheap_alloc(&ctx->heap, size);
533     if(!ret)
534         ctx->hres = E_OUTOFMEMORY;
535     return ret;
536 }
537
538 HRESULT parse_script(parser_ctx_t *ctx, const WCHAR *code)
539 {
540     ctx->code = ctx->ptr = code;
541     ctx->end = ctx->code + strlenW(ctx->code);
542
543     vbsheap_init(&ctx->heap);
544
545     ctx->parse_complete = FALSE;
546     ctx->hres = S_OK;
547
548     ctx->last_token = tNL;
549     ctx->last_nl = 0;
550     ctx->stats = ctx->stats_tail = NULL;
551     ctx->option_explicit = FALSE;
552
553     parser_parse(ctx);
554
555     if(FAILED(ctx->hres))
556         return ctx->hres;
557     if(!ctx->parse_complete) {
558         FIXME("parser failed on parsing %s\n", debugstr_w(ctx->ptr));
559         return E_FAIL;
560     }
561
562     return S_OK;
563 }
564
565 void parser_release(parser_ctx_t *ctx)
566 {
567     vbsheap_free(&ctx->heap);
568 }