apply fix and add test case

wrongnull · Nov 23, 2023 · 8afbb8e · 8afbb8e
1 parent 6f8c597
commit 8afbb8e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/src/Sema.zig b/src/Sema.zig
@@ -6189,7 +6189,10 @@ fn zirSetCold(sema: *Sema, block: *Block, extended: Zir.Inst.Extended.InstData)
     const mod = sema.mod;
     const ip = &mod.intern_pool;
     const extra = sema.code.extraData(Zir.Inst.UnNode, extended.operand).data;
+    const builtin_src = LazySrcLoc.nodeOffset(extra.node);
     const operand_src: LazySrcLoc = .{ .node_offset_builtin_call_arg0 = extra.node };
+    if (sema.func_index == .none)
+        return sema.fail(block, builtin_src, "@setCold outside function body", .{});
     const is_cold = try sema.resolveConstBool(block, operand_src, extra.operand, .{
         .needed_comptime_reason = "operand to @setCold must be comptime-known",
     });
@@ -16756,7 +16759,9 @@ fn zirRetAddr(
     block: *Block,
     extended: Zir.Inst.Extended.InstData,
 ) CompileError!Air.Inst.Ref {
-    _ = extended;
+    const src = LazySrcLoc.nodeOffset(@bitCast(extended.operand));
+    if (sema.func_index == .none)
+        return sema.fail(block, src, "@returnAddress outside function body", .{});
     if (block.is_comptime) {
         // TODO: we could give a meaningful lazy value here. #14938
         return sema.mod.intRef(Type.usize, 0);

diff --git a/test/cases/compile_errors/setAlignStack_outside_function.zig b/test/cases/compile_errors/setAlignStack_outside_function.zig