diff --git a/lib/CodeGen/PrologEpilogInserter.cpp b/lib/CodeGen/PrologEpilogInserter.cpp index 959dd7df58e..3a77af0cc49 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 +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 00000000000..4ddc846dc7e --- /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. +; + +; 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 00000000000..0a6a506c7a6 --- /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. +; + +; 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*)