diff --git a/llvm/lib/IR/Verifier.cpp b/llvm/lib/IR/Verifier.cpp --- a/llvm/lib/IR/Verifier.cpp +++ b/llvm/lib/IR/Verifier.cpp @@ -2501,6 +2501,15 @@ Assert(CBI.getOperand(i) != CBI.getOperand(j), "Duplicate callbr destination!", &CBI); } + { + SmallPtrSet ArgBBs; + for (Value *V : CBI.args()) + if (auto *BA = dyn_cast(V)) + ArgBBs.insert(BA->getBasicBlock()); + for (BasicBlock *BB : CBI.getIndirectDests()) + Assert(ArgBBs.find(BB) != ArgBBs.end(), + "Indirect label missing from arglist.", &CBI); + } visitTerminator(CBI); } diff --git a/llvm/test/CodeGen/AArch64/callbr-asm-obj-file.ll b/llvm/test/CodeGen/AArch64/callbr-asm-obj-file.ll --- a/llvm/test/CodeGen/AArch64/callbr-asm-obj-file.ll +++ b/llvm/test/CodeGen/AArch64/callbr-asm-obj-file.ll @@ -1,3 +1,4 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc < %s -mtriple=aarch64-unknown-linux-gnu -filetype=obj -o - \ ; RUN: | llvm-objdump -triple aarch64-unknown-linux-gnu -d - \ ; RUN: | FileCheck %s @@ -6,11 +7,11 @@ @l = common hidden local_unnamed_addr global i32 0, align 4 -; CHECK-LABEL: 0000000000000000 test1: -; CHECK-LABEL: 0000000000000018 $d.1: -; CHECK-LABEL: 0000000000000020 $x.2: +; CHECK-LABEL: test1: +; CHECK-LABEL: $d.1: +; CHECK-LABEL: $x.2: ; CHECK-NEXT: b #16 <$x.4+0x4> -; CHECK-LABEL: 000000000000002c $x.4: +; CHECK-LABEL: $x.4: ; CHECK-NEXT: b #4 <$x.4+0x4> ; CHECK-NEXT: mov w0, wzr ; CHECK-NEXT: ldr x30, [sp], #16 @@ -40,11 +41,11 @@ declare dso_local i32 @i(...) local_unnamed_addr -; CHECK-LABEL: 000000000000003c test2: -; CHECK: bl #0 -; CHECK-LABEL: 0000000000000064 $d.5: -; CHECK-LABEL: 000000000000006c $x.6: -; CHECK-NEXT: b #-24 +; CHECK-LABEL: test2: +; CHECK: bl #0 +; CHECK-LABEL: $d.5: +; CHECK-LABEL: $x.6: +; CHECK-NEXT: b #16 <$x.8+0x4> define hidden i32 @test2() local_unnamed_addr { %1 = load i32, i32* @l, align 4 %2 = icmp eq i32 %1, 0 @@ -57,7 +58,7 @@ 6: ; preds = %3 callbr void asm sideeffect "1: nop\0A\09.quad b\0A\09b ${1:l}\0A\09.quad ${0:c}", "i,X"(i32* null, i8* blockaddress(@test2, %7)) - to label %10 [label %9] + to label %10 [label %7] 7: ; preds = %3 %8 = tail call i32 bitcast (i32 (...)* @i to i32 ()*)() @@ -70,11 +71,11 @@ ret i32 undef } -; CHECK-LABEL: 0000000000000084 test3: -; CHECK-LABEL: 00000000000000a8 $d.9: -; CHECK-LABEL: 00000000000000b0 $x.10: +; CHECK-LABEL: test3: +; CHECK-LABEL: $d.9: +; CHECK-LABEL: $x.10: ; CHECK-NEXT: b #20 <$x.12+0x8> -; CHECK-LABEL: 00000000000000bc $x.12: +; CHECK-LABEL: $x.12: ; CHECK-NEXT: b #4 <$x.12+0x4> ; CHECK-NEXT: mov w0, wzr ; CHECK-NEXT: ldr x30, [sp], #16 diff --git a/llvm/test/Verifier/callbr.ll b/llvm/test/Verifier/callbr.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Verifier/callbr.ll @@ -0,0 +1,31 @@ +; RUN: not opt -S %s -verify 2>&1 | FileCheck %s + +; CHECK: Indirect label missing from arglist. +define void @foo() { + ; the %3 in the blockaddress doesn't match the %4 in the indirect label list (bad). + callbr void asm sideeffect "${0:l}", "X,X"(i8* blockaddress(@foo, %3), i8* blockaddress(@foo, %2)) + to label %1 [label %4, label %2] +1: + ret void +2: + ret void +3: + ret void +4: + ret void +} + +; CHECK-NOT: Indirect label missing from arglist. +define void @bar() { + ; the %4 in the blockaddress does match the %4 in the indirect label list (good). + callbr void asm sideeffect "${0:l}", "X,X"(i8* blockaddress(@bar, %4), i8* blockaddress(@bar, %2)) + to label %1 [label %4, label %2] +1: + ret void +2: + ret void +3: + ret void +4: + ret void +}