Ensure asm noreturn works with labels

Author: nbdd0121

compiler/rustc_hir_typeck/src/expr.rs

@@ -3222,6 +3222,12 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
     }
 
     fn check_expr_asm(&self, asm: &'tcx hir::InlineAsm<'tcx>) -> Ty<'tcx> {
+        let ret_ty = if asm.options.contains(ast::InlineAsmOptions::NORETURN) {
+            self.tcx.types.never
+        } else {
+            Ty::new_unit(self.tcx)
+        };
+
         for (op, _op_sp) in asm.operands {
             match op {
                 hir::InlineAsmOperand::In { expr, .. } => {
@@ -3244,15 +3250,22 @@ impl<'a, 'tcx> FnCtxt<'a, 'tcx> {
                 hir::InlineAsmOperand::Const { .. } | hir::InlineAsmOperand::SymFn { .. } => {}
                 hir::InlineAsmOperand::SymStatic { .. } => {}
                 hir::InlineAsmOperand::Label { block } => {
-                    self.check_block_no_value(block);
+                    let diverges = self.diverges.get();
+
+                    // The label blocks should have unit return value, expect when the assembly
+                    // itself is noreturn, in which case the label blocks also become noreturn.
+                    let ty = self.check_block_with_expected(block, ExpectHasType(ret_ty));
+                    if !ty.is_never() {
+                        self.demand_suptype(block.span, ret_ty, ty);
+                    }
+
+                    // We need this to avoid false unreachable warning when a label diverges.
+                    self.diverges.set(diverges);
                 }
             }
         }
-        if asm.options.contains(ast::InlineAsmOptions::NORETURN) {
-            self.tcx.types.never
-        } else {
-            Ty::new_unit(self.tcx)
-        }
+
+        ret_ty
     }
 
     fn check_offset_of(