vbscript: Support one-line sub/function.

dlls/vbscript/parser.y

@@ -145,7 +145,7 @@ static statement_t *link_statements(statement_t*,statement_t*);
 %type <member> MemberExpression
 %type <expression> Arguments ArgumentList ArgumentList_opt Step_opt ExpressionList
 %type <boolean> DoType Preserve_opt
-%type <arg_decl> ArgumentsDecl_opt ArgumentDeclList ArgumentDecl
+%type <arg_decl> ArgumentsDecl ArgumentsDecl_opt ArgumentDeclList ArgumentDecl
 %type <func_decl> FunctionDecl PropertyDecl
 %type <elseif> ElseIfs_opt ElseIfs ElseIf
 %type <class_decl> ClassDeclaration ClassBody
@@ -475,9 +475,13 @@ PropertyDecl
                                     { $$ = new_function_decl(ctx, $4, FUNC_PROPSET, $1, $6, $9); CHECK_ERROR; }
 
 FunctionDecl
-    : Storage_opt tSUB Identifier ArgumentsDecl_opt StSep BodyStatements tEND tSUB
+    : Storage_opt tSUB Identifier StSep BodyStatements tEND tSUB
+                                    { $$ = new_function_decl(ctx, $3, FUNC_SUB, $1, NULL, $5); CHECK_ERROR; }
+    | Storage_opt tSUB Identifier ArgumentsDecl Nl_opt BodyStatements tEND tSUB
                                     { $$ = new_function_decl(ctx, $3, FUNC_SUB, $1, $4, $6); CHECK_ERROR; }
-    | Storage_opt tFUNCTION Identifier ArgumentsDecl_opt StSep BodyStatements tEND tFUNCTION
+    | Storage_opt tFUNCTION Identifier StSep BodyStatements tEND tFUNCTION
+                                    { $$ = new_function_decl(ctx, $3, FUNC_FUNCTION, $1, NULL, $5); CHECK_ERROR; }
+    | Storage_opt tFUNCTION Identifier ArgumentsDecl Nl_opt BodyStatements tEND tFUNCTION
                                     { $$ = new_function_decl(ctx, $3, FUNC_FUNCTION, $1, $4, $6); CHECK_ERROR; }
 
 Storage_opt
@@ -490,7 +494,11 @@ Storage
     | tPRIVATE                      { $$ = STORAGE_IS_PRIVATE; }
 
 ArgumentsDecl_opt
-    : EmptyBrackets_opt                         { $$ = NULL; }
+    : /* empty*/                                { $$ = 0; }
+    | ArgumentsDecl                             { $$ = $1; }
+
+ArgumentsDecl
+    : tEMPTYBRACKETS                            { $$ = NULL; }
     | '(' ArgumentDeclList ')'                  { $$ = $2; }
 
 ArgumentDeclList
@@ -515,6 +523,10 @@ StSep_opt
     : /* empty */
     | StSep
 
+Nl_opt
+    : /* empty */
+    | tNL Nl_opt
+
 /* Most statements accept both new line and ':' as separators */
 StSep
     : tNL
@@ -1163,6 +1175,8 @@ static statement_t *link_statements(statement_t *head, statement_t *tail)
 {
     statement_t *iter;
 
+    if (!head) return tail;
+
     for(iter = head; iter->next; iter = iter->next);
     iter->next = tail;
 

dlls/vbscript/tests/lang.vbs

@@ -827,6 +827,14 @@ x = false
 Call testsub
 Call ok(x, "x is false, testsub not called?")
 
+if false then
+Sub testsub_one_line() x = true End Sub
+end if
+
+x = false
+Call testsub_one_line
+Call ok(x, "x is false, testsub_one_line not called?")
+
 Sub SubSetTrue(v)
     Call ok(not v, "v is not true")
     v = true
@@ -920,6 +928,14 @@ x = false
 Call TestFunc
 Call ok(x, "x is false, testfunc not called?")
 
+if false then
+Function testfunc_one_line() x = true End Function
+end if
+
+x = false
+Call testfunc_one_line
+Call ok(x, "x is false, testfunc_one_line not called?")
+
 Function FuncSetTrue(v)
     Call ok(not v, "v is not true")
     v = true