[llvm-commits] [llvm] r39978 - /llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp

[Evan Cheng]

Author: evancheng
Date: Tue Jul 17 13:40:47 2007
New Revision: 39978

URL: http://llvm.org/viewvc/llvm-project?rev=39978&view=rev
Log:
Make sure not to break eh_return.

Modified:
    llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp

Modified: llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp?rev=39978&r1=39977&r2=39978&view=diff

==============================================================================
--- llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp (original)
+++ llvm/trunk/lib/Target/X86/X86RegisterInfo.cpp Tue Jul 17 13:40:47 2007
@@ -1332,14 +1332,6 @@
     NumBytes -= SlotSize;
   }
 
-  // We're returning from function via eh_return.
-  if (RetOpcode == X86::EH_RETURN) {
-    MachineOperand &DestAddr  = MBBI->getOperand(0);
-    assert(DestAddr.isReg() && "Offset should be in register!");
-    BuildMI(MBB, MBBI, TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr),StackPtr).
-      addReg(DestAddr.getReg());
-  }
-
   // Skip the callee-saved pop instructions.
   while (MBBI != MBB.begin()) {
     MachineBasicBlock::iterator PI = prior(MBBI);      
@@ -1355,7 +1347,7 @@
     MachineInstr *MI = addRegOffset(BuildMI(TII.get(Opc), StackPtr),
                                     FramePtr, -CSSize);
     MBB.insert(MBBI, MI);
-    return;
+    NumBytes = 0;
   }
 
   if (NumBytes) {    // adjust stack pointer back: ESP += numbytes
@@ -1380,6 +1372,15 @@
     if (NumBytes)
       emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII);
   }
+
+  // We're returning from function via eh_return.
+  if (RetOpcode == X86::EH_RETURN) {
+    MBBI = prior(MBB.end());
+    MachineOperand &DestAddr  = MBBI->getOperand(0);
+    assert(DestAddr.isReg() && "Offset should be in register!");
+    BuildMI(MBB, MBBI, TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr),StackPtr).
+      addReg(DestAddr.getReg());
+  }
 }
 
 unsigned X86RegisterInfo::getRARegister() const {