59 lines
2.7 KiB
Diff
59 lines
2.7 KiB
Diff
Index: lib/CodeGen/LocalStackSlotAllocation.cpp
|
|
===================================================================
|
|
--- lib/CodeGen/LocalStackSlotAllocation.cpp
|
|
+++ lib/CodeGen/LocalStackSlotAllocation.cpp
|
|
@@ -201,6 +201,14 @@
|
|
SmallSet<int, 16> ProtectedObjs;
|
|
if (MFI.hasStackProtectorIndex()) {
|
|
int StackProtectorFI = MFI.getStackProtectorIndex();
|
|
+
|
|
+ // We need to make sure we didn't pre-allocate the stack protector when
|
|
+ // doing this.
|
|
+ // If we already have a stack protector, this will re-assign it to a slot
|
|
+ // that is **not** covering the protected objects.
|
|
+ assert(!MFI.isObjectPreAllocated(StackProtectorFI) &&
|
|
+ "Stack protector pre-allocated in LocalStackSlotAllocation");
|
|
+
|
|
StackObjSet LargeArrayObjs;
|
|
StackObjSet SmallArrayObjs;
|
|
StackObjSet AddrOfObjs;
|
|
Index: lib/CodeGen/PrologEpilogInserter.cpp
|
|
===================================================================
|
|
--- lib/CodeGen/PrologEpilogInserter.cpp
|
|
+++ lib/CodeGen/PrologEpilogInserter.cpp
|
|
@@ -933,8 +933,16 @@
|
|
StackObjSet SmallArrayObjs;
|
|
StackObjSet AddrOfObjs;
|
|
|
|
- AdjustStackOffset(MFI, StackProtectorFI, StackGrowsDown, Offset, MaxAlign,
|
|
- Skew);
|
|
+ // If we need a stack protector, we need to make sure that
|
|
+ // LocalStackSlotPass didn't already allocate a slot for it.
|
|
+ // If we are told to use the LocalStackAllocationBlock, the stack protector
|
|
+ // is expected to be already pre-allocated.
|
|
+ if (!MFI.getUseLocalStackAllocationBlock())
|
|
+ AdjustStackOffset(MFI, StackProtectorFI, StackGrowsDown, Offset, MaxAlign,
|
|
+ Skew);
|
|
+ else if (!MFI.isObjectPreAllocated(MFI.getStackProtectorIndex()))
|
|
+ llvm_unreachable(
|
|
+ "Stack protector not pre-allocated by LocalStackSlotPass.");
|
|
|
|
// Assign large stack objects first.
|
|
for (unsigned i = 0, e = MFI.getObjectIndexEnd(); i != e; ++i) {
|
|
@@ -968,6 +976,15 @@
|
|
llvm_unreachable("Unexpected SSPLayoutKind.");
|
|
}
|
|
|
|
+ // We expect **all** the protected stack objects to be pre-allocated by
|
|
+ // LocalStackSlotPass. If it turns out that PEI still has to allocate some
|
|
+ // of them, we may end up messing up the expected order of the objects.
|
|
+ if (MFI.getUseLocalStackAllocationBlock() &&
|
|
+ !(LargeArrayObjs.empty() && SmallArrayObjs.empty() &&
|
|
+ AddrOfObjs.empty()))
|
|
+ llvm_unreachable("Found protected stack objects not pre-allocated by "
|
|
+ "LocalStackSlotPass.");
|
|
+
|
|
AssignProtectedObjSet(LargeArrayObjs, ProtectedObjs, MFI, StackGrowsDown,
|
|
Offset, MaxAlign, Skew);
|
|
AssignProtectedObjSet(SmallArrayObjs, ProtectedObjs, MFI, StackGrowsDown,
|