From 89d83fd0e81825eeec4070d94851b8efc8e8c117 Mon Sep 17 00:00:00 2001
From: Philip Herron <herron.philip@googlemail.com>
Date: Wed, 29 Mar 2023 16:14:04 +0100
Subject: gccrs: fix ICE when closure body is not a block

Fixes: #2052

gcc/rust/ChangeLog:

	* backend/rust-compile-expr.cc (CompileExpr::generate_closure_function):
	when its not a block we dont have any ribs to generate locals from

gcc/testsuite/ChangeLog:

	* rust/execute/torture/issue-2052.rs: New test.

Signed-off-by: Philip Herron <herron.philip@googlemail.com>
---
 gcc/rust/backend/rust-compile-expr.cc            | 23 ++++++++++++++---------
 gcc/testsuite/rust/execute/torture/issue-2052.rs | 15 +++++++++++++++
 2 files changed, 29 insertions(+), 9 deletions(-)
 create mode 100644 gcc/testsuite/rust/execute/torture/issue-2052.rs

(limited to 'gcc')

diff --git a/gcc/rust/backend/rust-compile-expr.cc b/gcc/rust/backend/rust-compile-expr.cc
index 6a7b2eb..904c47b 100644
--- a/gcc/rust/backend/rust-compile-expr.cc
+++ b/gcc/rust/backend/rust-compile-expr.cc
@@ -2880,20 +2880,25 @@ CompileExpr::generate_closure_function (HIR::ClosureExpr &expr,
 
   // lookup locals
   HIR::Expr *function_body = expr.get_expr ().get ();
-  auto body_mappings = function_body->get_mappings ();
-  Resolver::Rib *rib = nullptr;
-  bool ok
-    = ctx->get_resolver ()->find_name_rib (body_mappings.get_nodeid (), &rib);
-  rust_assert (ok);
+  bool is_block_expr
+    = function_body->get_expression_type () == HIR::Expr::ExprType::Block;
+
+  std::vector<Bvariable *> locals = {};
+  if (is_block_expr)
+    {
+      auto body_mappings = function_body->get_mappings ();
+      Resolver::Rib *rib = nullptr;
+      bool ok
+	= ctx->get_resolver ()->find_name_rib (body_mappings.get_nodeid (),
+					       &rib);
+      rust_assert (ok);
 
-  std::vector<Bvariable *> locals
-    = compile_locals_for_block (ctx, *rib, fndecl);
+      locals = compile_locals_for_block (ctx, *rib, fndecl);
+    }
 
   tree enclosing_scope = NULL_TREE;
   Location start_location = function_body->get_locus ();
   Location end_location = function_body->get_locus ();
-  bool is_block_expr
-    = function_body->get_expression_type () == HIR::Expr::ExprType::Block;
   if (is_block_expr)
     {
       HIR::BlockExpr *body = static_cast<HIR::BlockExpr *> (function_body);
diff --git a/gcc/testsuite/rust/execute/torture/issue-2052.rs b/gcc/testsuite/rust/execute/torture/issue-2052.rs
new file mode 100644
index 0000000..6c15eb3
--- /dev/null
+++ b/gcc/testsuite/rust/execute/torture/issue-2052.rs
@@ -0,0 +1,15 @@
+#[lang = "fn_once"]
+pub trait FnOnce<Args> {
+    #[lang = "fn_once_output"]
+    type Output;
+
+    extern "rust-call" fn call_once(self, args: Args) -> Self::Output;
+}
+
+pub fn f() -> i32 {
+    (|| 42)()
+}
+
+pub fn main() -> i32 {
+    f() - 42
+}
-- 
cgit v1.1