fix assertione error on block parsing

3 files changed, 8 insertions, 9 deletions

diff --git a/crates/libsyntax2/src/grammar/expressions/atom.rs b/crates/libsyntax2/src/grammar/expressions/atom.rs
index b0e270426..d9c3f998a 100644
--- a/crates/libsyntax2/src/grammar/expressions/atom.rs
+++ b/crates/libsyntax2/src/grammar/expressions/atom.rs
@@ -148,11 +148,7 @@ fn lambda_expr(p: &mut Parser) -> CompletedMarker {
     p.eat(MOVE_KW);
     params::param_list_opt_types(p);
     if opt_fn_ret_type(p) {
-        if p.at(L_CURLY) {
-            block(p);
-        } else {
-            p.error("expected a block");
-        }
+        block(p);
     } else {
         expr(p);
     }
diff --git a/crates/libsyntax2/src/grammar/expressions/mod.rs b/crates/libsyntax2/src/grammar/expressions/mod.rs
index bd6c84886..922d9f871 100644
--- a/crates/libsyntax2/src/grammar/expressions/mod.rs
+++ b/crates/libsyntax2/src/grammar/expressions/mod.rs
@@ -26,7 +26,10 @@ fn expr_no_struct(p: &mut Parser) {
 // fn c() { 1; 2; }
 // fn d() { 1; 2 }
 pub(crate) fn block(p: &mut Parser) {
-    assert!(p.at(L_CURLY));
+    if !p.at(L_CURLY) {
+        p.error("expected a block");
+        return;
+    }
     let m = p.start();
     p.bump();
     while !p.at(EOF) && !p.at(R_CURLY) {
diff --git a/crates/libsyntax2/src/grammar/items/mod.rs b/crates/libsyntax2/src/grammar/items/mod.rs
index 44ab92c63..32d0778c4 100644
--- a/crates/libsyntax2/src/grammar/items/mod.rs
+++ b/crates/libsyntax2/src/grammar/items/mod.rs
@@ -252,10 +252,10 @@ fn function(p: &mut Parser, flavor: ItemFlavor) {
 
     // test fn_decl
     // trait T { fn foo(); }
-    if p.at(L_CURLY) {
-        expressions::block(p);
+    if p.at(SEMI) {
+        p.bump();
     } else {
-        p.expect(SEMI);
+        expressions::block(p)
     }
 }