diff options
| -rw-r--r-- | lib/CodeGen/PrologEpilogInserter.cpp | 13 | ||||
| -rw-r--r-- | test/CodeGen/ARM/warn-stack.ll | 24 | ||||
| -rw-r--r-- | test/CodeGen/X86/warn-stack.ll | 24 |
3 files changed, 61 insertions, 0 deletions
diff --git a/lib/CodeGen/PrologEpilogInserter.cpp b/lib/CodeGen/PrologEpilogInserter.cpp index 959dd7d..3a77af0 100644 --- a/lib/CodeGen/PrologEpilogInserter.cpp +++ b/lib/CodeGen/PrologEpilogInserter.cpp @@ -35,6 +35,7 @@ #include "llvm/Support/CommandLine.h" #include "llvm/Support/Compiler.h" #include "llvm/Support/Debug.h" +#include "llvm/Support/raw_ostream.h" #include "llvm/Target/TargetFrameLowering.h" #include "llvm/Target/TargetInstrInfo.h" #include "llvm/Target/TargetMachine.h" @@ -46,6 +47,11 @@ using namespace llvm; char PEI::ID = 0; char &llvm::PrologEpilogCodeInserterID = PEI::ID; +static cl::opt<uint64_t> +WarnStackSize("warn-stack-size", cl::Hidden, + cl::desc("Warn for stack size bigger than the given" + " number")); + INITIALIZE_PASS_BEGIN(PEI, "prologepilog", "Prologue/Epilogue Insertion", false, false) INITIALIZE_PASS_DEPENDENCY(MachineLoopInfo) @@ -128,6 +134,13 @@ bool PEI::runOnMachineFunction(MachineFunction &Fn) { // Clear any vregs created by virtual scavenging. Fn.getRegInfo().clearVirtRegs(); + // Warn on stack size when we exceeds the given limit. + MachineFrameInfo *MFI = Fn.getFrameInfo(); + if (WarnStackSize.getNumOccurrences() > 0 && + WarnStackSize < MFI->getStackSize()) + errs() << "warning: Stack size limit exceeded (" << MFI->getStackSize() + << ") in " << Fn.getName() << ".\n"; + delete RS; clearAllSets(); return true; diff --git a/test/CodeGen/ARM/warn-stack.ll b/test/CodeGen/ARM/warn-stack.ll new file mode 100644 index 0000000..4ddc846 --- /dev/null +++ b/test/CodeGen/ARM/warn-stack.ll @@ -0,0 +1,24 @@ +; RUN: llc -march arm -warn-stack-size=80 < %s 2>&1 >/dev/null | FileCheck %s +; Check the internal option that warns when the stack size exceeds the +; given amount. +; <rdar://13987214> + +; CHECK-NOT: nowarn +define void @nowarn() nounwind ssp { +entry: + %buffer = alloca [12 x i8], align 1 + %arraydecay = getelementptr inbounds [12 x i8]* %buffer, i64 0, i64 0 + call void @doit(i8* %arraydecay) nounwind + ret void +} + +; CHECK: warning: Stack size limit exceeded (92) in warn. +define void @warn() nounwind ssp { +entry: + %buffer = alloca [80 x i8], align 1 + %arraydecay = getelementptr inbounds [80 x i8]* %buffer, i64 0, i64 0 + call void @doit(i8* %arraydecay) nounwind + ret void +} + +declare void @doit(i8*) diff --git a/test/CodeGen/X86/warn-stack.ll b/test/CodeGen/X86/warn-stack.ll new file mode 100644 index 0000000..0a6a506 --- /dev/null +++ b/test/CodeGen/X86/warn-stack.ll @@ -0,0 +1,24 @@ +; RUN: llc -march x86 -warn-stack-size=80 < %s 2>&1 >/dev/null | FileCheck %s +; Check the internal option that warns when the stack size exceeds the +; given amount. +; <rdar://13987214> + +; CHECK-NOT: nowarn +define void @nowarn() nounwind ssp { +entry: + %buffer = alloca [12 x i8], align 1 + %arraydecay = getelementptr inbounds [12 x i8]* %buffer, i64 0, i64 0 + call void @doit(i8* %arraydecay) nounwind + ret void +} + +; CHECK: warning: Stack size limit exceeded (92) in warn. +define void @warn() nounwind ssp { +entry: + %buffer = alloca [80 x i8], align 1 + %arraydecay = getelementptr inbounds [80 x i8]* %buffer, i64 0, i64 0 + call void @doit(i8* %arraydecay) nounwind + ret void +} + +declare void @doit(i8*) |