[llvm] 742250b - [ARM][CMSE] Issue an error if passing arguments through memory across
Momchil Velikov via llvm-commits
llvm-commits at lists.llvm.org
Mon Sep 21 09:26:19 PDT 2020
Author: Momchil Velikov
Date: 2020-09-21T17:26:10+01:00
New Revision: 742250bf62a92647d3462d1e0b75940bc0914fcb
URL: https://github.com/llvm/llvm-project/commit/742250bf62a92647d3462d1e0b75940bc0914fcb
DIFF: https://github.com/llvm/llvm-project/commit/742250bf62a92647d3462d1e0b75940bc0914fcb.diff
LOG: [ARM][CMSE] Issue an error if passing arguments through memory across
security boundary
It was never supported and that part was accidentally omitted when
upstreaming D76518.
Differential Revision: https://reviews.llvm.org/D86478
Change-Id: If6ba9506eb0431c87a1d42a38aa60e47ce263039
Added:
llvm/test/CodeGen/ARM/cmse-errors.ll
Modified:
llvm/lib/Target/ARM/ARMISelLowering.cpp
Removed:
################################################################################
diff --git a/llvm/lib/Target/ARM/ARMISelLowering.cpp b/llvm/lib/Target/ARM/ARMISelLowering.cpp
index 70e8a797f869..3d674868976e 100644
--- a/llvm/lib/Target/ARM/ARMISelLowering.cpp
+++ b/llvm/lib/Target/ARM/ARMISelLowering.cpp
@@ -4430,13 +4430,26 @@ SDValue ARMTargetLowering::LowerFormalArguments(
}
// varargs
- if (isVarArg && MFI.hasVAStart())
- VarArgStyleRegisters(CCInfo, DAG, dl, Chain,
- CCInfo.getNextStackOffset(),
+ if (isVarArg && MFI.hasVAStart()) {
+ VarArgStyleRegisters(CCInfo, DAG, dl, Chain, CCInfo.getNextStackOffset(),
TotalArgRegsSaveSize);
+ if (AFI->isCmseNSEntryFunction()) {
+ DiagnosticInfoUnsupported Diag(
+ DAG.getMachineFunction().getFunction(),
+ "secure entry function must not be variadic", dl.getDebugLoc());
+ DAG.getContext()->diagnose(Diag);
+ }
+ }
AFI->setArgumentStackSize(CCInfo.getNextStackOffset());
+ if (CCInfo.getNextStackOffset() > 0 && AFI->isCmseNSEntryFunction()) {
+ DiagnosticInfoUnsupported Diag(
+ DAG.getMachineFunction().getFunction(),
+ "secure entry function requires arguments on stack", dl.getDebugLoc());
+ DAG.getContext()->diagnose(Diag);
+ }
+
return Chain;
}
diff --git a/llvm/test/CodeGen/ARM/cmse-errors.ll b/llvm/test/CodeGen/ARM/cmse-errors.ll
new file mode 100644
index 000000000000..ea7976b0fe49
--- /dev/null
+++ b/llvm/test/CodeGen/ARM/cmse-errors.ll
@@ -0,0 +1,74 @@
+; RUN: not llc -mtriple=thumbv8m.main-eabi %s -o - 2>&1 | FileCheck %s
+
+%struct.two_ints = type { i32, i32 }
+%struct.__va_list = type { i8* }
+
+define void @test1(%struct.two_ints* noalias nocapture sret align 4 %agg.result) "cmse_nonsecure_entry" {
+entry:
+ %0 = bitcast %struct.two_ints* %agg.result to i64*
+ store i64 8589934593, i64* %0, align 4
+ ret void
+}
+; CHECK: error: {{.*}}test1{{.*}}: secure entry function would return value through pointer
+
+define void @test2(i32 %a, i32 %b, i32 %c, i32 %d, i32 %e) "cmse_nonsecure_entry" {
+entry:
+ ret void
+}
+; CHECK: error: {{.*}}test2{{.*}}: secure entry function requires arguments on stack
+
+define void @test3(void (i32, i32, i32, i32, i32)* nocapture %p) {
+entry:
+ tail call void %p(i32 1, i32 2, i32 3, i32 4, i32 5) "cmse_nonsecure_call"
+ ret void
+}
+; CHECK: error: {{.*}}test3{{.*}}: call to non-secure function would require passing arguments on stack
+
+
+define void @test4(void (%struct.two_ints*)* nocapture %p) {
+entry:
+ %r = alloca %struct.two_ints, align 4
+ %0 = bitcast %struct.two_ints* %r to i8*
+ call void %p(%struct.two_ints* nonnull sret align 4 %r) "cmse_nonsecure_call"
+ ret void
+}
+; CHECK: error: {{.*}}test4{{.*}}: call to non-secure function would return value through pointer
+
+declare void @llvm.va_start(i8*) "nounwind"
+
+declare void @llvm.va_end(i8*) "nounwind"
+
+define i32 @test5(i32 %a, ...) "cmse_nonsecure_entry" {
+entry:
+ %vl = alloca %struct.__va_list, align 4
+ %0 = bitcast %struct.__va_list* %vl to i8*
+ call void @llvm.va_start(i8* nonnull %0)
+ %1 = getelementptr inbounds %struct.__va_list, %struct.__va_list* %vl, i32 0, i32 0
+ %argp.cur = load i8*, i8** %1, align 4
+ %argp.next = getelementptr inbounds i8, i8* %argp.cur, i32 4
+ store i8* %argp.next, i8** %1, align 4
+ %2 = bitcast i8* %argp.cur to i32*
+ %3 = load i32, i32* %2, align 4
+ call void @llvm.va_end(i8* nonnull %0)
+ ret i32 %3
+}
+; CHECK: error: {{.*}}test5{{.*}}: secure entry function must not be variadic
+
+define void @test6(void (i32, ...)* nocapture %p) {
+entry:
+ tail call void (i32, ...) %p(i32 1, i32 2, i32 3, i32 4, i32 5) "cmse_nonsecure_call"
+ ret void
+}
+; CHECK: error: {{.*}}test6{{.*}}: call to non-secure function would require passing arguments on stack
+
+define void @neg_test1(void (i32, ...)* nocapture %p) {
+entry:
+ tail call void (i32, ...) %p(i32 1, i32 2, i32 3, i32 4) "cmse_nonsecure_call"
+ ret void
+}
+
+define void @neg_test2(i32 %a, ...) "cmse_nonsecure_entry" {
+entry:
+ ret void
+}
+; CHECK-NOT: error:
More information about the llvm-commits
mailing list