Auto merge of rust-lang#77317 - varkor:break-diverging-value, r=nikomatsakis

Fix control flow check for breaking with diverging values

Fixes rust-lang#77156.

Author: bors

compiler/rustc_typeck/src/check/expr.rs

@@ -627,7 +627,10 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                 assert!(expr_opt.is_none() || self.tcx.sess.has_errors());
             }
 
-            ctxt.may_break = true;
+            // If we encountered a `break`, then (no surprise) it may be possible to break from the
+            // loop... unless the value being returned from the loop diverges itself, e.g.
+            // `break return 5` or `break loop {}`.
+            ctxt.may_break |= !self.diverges.get().is_always();
 
             // the type of a `break` is always `!`, since it diverges
             tcx.types.never

src/test/ui/break-diverging-value.rs

@@ -0,0 +1,37 @@
+#![feature(never_type)]
+
+fn loop_break_return() -> i32 {
+    let loop_value = loop { break return 0 }; // ok
+}
+
+fn loop_break_loop() -> i32 {
+    let loop_value = loop { break loop {} }; // ok
+}
+
+fn loop_break_break() -> i32 { //~ ERROR mismatched types
+    let loop_value = loop { break break };
+}
+
+fn loop_break_return_2() -> i32 {
+    let loop_value = loop { break { return 0; () } }; // ok
+}
+
+enum Void {}
+
+fn get_void() -> Void {
+    panic!()
+}
+
+fn loop_break_void() -> i32 { //~ ERROR mismatched types
+    let loop_value = loop { break get_void() };
+}
+
+fn get_never() -> ! {
+    panic!()
+}
+
+fn loop_break_never() -> i32 {
+    let loop_value = loop { break get_never() }; // ok
+}
+
+fn main() {}

src/test/ui/break-diverging-value.stderr

@@ -0,0 +1,19 @@
+error[E0308]: mismatched types
+  --> $DIR/break-diverging-value.rs:11:26
+   |
+LL | fn loop_break_break() -> i32 {
+   |    ----------------      ^^^ expected `i32`, found `()`
+   |    |
+   |    implicitly returns `()` as its body has no tail or `return` expression
+
+error[E0308]: mismatched types
+  --> $DIR/break-diverging-value.rs:25:25
+   |
+LL | fn loop_break_void() -> i32 {
+   |    ---------------      ^^^ expected `i32`, found `()`
+   |    |
+   |    implicitly returns `()` as its body has no tail or `return` expression
+
+error: aborting due to 2 previous errors
+
+For more information about this error, try `rustc --explain E0308`.