From eac9385f098e6fc730ad9a3ba9933957f1650127 Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Sun, 7 Oct 2007 21:53:12 +0000
Subject: [PATCH] Fix many regressions on x86 by avoiding dereferencing the end
 iterator.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@42738 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Target/X86/X86RegisterInfo.cpp | 73 ++++++++++++++++--------------
 1 file changed, 38 insertions(+), 35 deletions(-)

diff --git a/lib/Target/X86/X86RegisterInfo.cpp b/lib/Target/X86/X86RegisterInfo.cpp
index e923606021e..d6b0ec84616 100644
--- a/lib/Target/X86/X86RegisterInfo.cpp
+++ b/lib/Target/X86/X86RegisterInfo.cpp
@@ -1481,47 +1481,50 @@ void emitSPUpdate(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI,
 static
 void mergeSPUpdatesUp(MachineBasicBlock &MBB, MachineBasicBlock::iterator &MBBI,
                       unsigned StackPtr, uint64_t *NumBytes = NULL) {
-  if (MBBI != MBB.begin()) {
-    MachineBasicBlock::iterator PI = prior(MBBI);
-    unsigned Opc = PI->getOpcode();
-    if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 ||
-         Opc == X86::ADD32ri || Opc == X86::ADD32ri8) &&
-        PI->getOperand(0).getReg() == StackPtr) {
-      if (NumBytes)
-        *NumBytes += PI->getOperand(2).getImm();
-      MBB.erase(PI);
-    } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 ||
-                Opc == X86::SUB32ri || Opc == X86::SUB32ri8) &&
-               PI->getOperand(0).getReg() == StackPtr) {
-      if (NumBytes)
-        *NumBytes -= PI->getOperand(2).getImm();
-      MBB.erase(PI);
-    }
+  if (MBBI == MBB.begin()) return;
+  
+  MachineBasicBlock::iterator PI = prior(MBBI);
+  unsigned Opc = PI->getOpcode();
+  if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 ||
+       Opc == X86::ADD32ri || Opc == X86::ADD32ri8) &&
+      PI->getOperand(0).getReg() == StackPtr) {
+    if (NumBytes)
+      *NumBytes += PI->getOperand(2).getImm();
+    MBB.erase(PI);
+  } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 ||
+              Opc == X86::SUB32ri || Opc == X86::SUB32ri8) &&
+             PI->getOperand(0).getReg() == StackPtr) {
+    if (NumBytes)
+      *NumBytes -= PI->getOperand(2).getImm();
+    MBB.erase(PI);
   }
 }
 
 // mergeSPUpdatesUp - Merge two stack-manipulating instructions lower iterator.
 static
-void mergeSPUpdatesDown(MachineBasicBlock &MBB,MachineBasicBlock::iterator &MBBI,
+void mergeSPUpdatesDown(MachineBasicBlock &MBB,
+                        MachineBasicBlock::iterator &MBBI,
                         unsigned StackPtr, uint64_t *NumBytes = NULL) {
-  if (MBBI != MBB.end()) {
-    MachineBasicBlock::iterator NI = next(MBBI);
-    unsigned Opc = NI->getOpcode();
-    if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 ||
-         Opc == X86::ADD32ri || Opc == X86::ADD32ri8) &&
-        NI->getOperand(0).getReg() == StackPtr) {
-      if (NumBytes)
-        *NumBytes -= NI->getOperand(2).getImm();
-      MBB.erase(NI);
-      MBBI = NI;
-    } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 ||
-                Opc == X86::SUB32ri || Opc == X86::SUB32ri8) &&
-               NI->getOperand(0).getReg() == StackPtr) {
-      if (NumBytes)
-        *NumBytes += NI->getOperand(2).getImm();
-      MBB.erase(NI);
-      MBBI = NI;
-    }
+  if (MBBI == MBB.end()) return;
+  
+  MachineBasicBlock::iterator NI = next(MBBI);
+  if (NI == MBB.end()) return;
+  
+  unsigned Opc = NI->getOpcode();
+  if ((Opc == X86::ADD64ri32 || Opc == X86::ADD64ri8 ||
+       Opc == X86::ADD32ri || Opc == X86::ADD32ri8) &&
+      NI->getOperand(0).getReg() == StackPtr) {
+    if (NumBytes)
+      *NumBytes -= NI->getOperand(2).getImm();
+    MBB.erase(NI);
+    MBBI = NI;
+  } else if ((Opc == X86::SUB64ri32 || Opc == X86::SUB64ri8 ||
+              Opc == X86::SUB32ri || Opc == X86::SUB32ri8) &&
+             NI->getOperand(0).getReg() == StackPtr) {
+    if (NumBytes)
+      *NumBytes += NI->getOperand(2).getImm();
+    MBB.erase(NI);
+    MBBI = NI;
   }
 }