Thumb assemmbly parsing diagnostic improvements for LDM.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@138287 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/Target/ARM/AsmParser/ARMAsmParser.cpp b/lib/Target/ARM/AsmParser/ARMAsmParser.cpp
index 31300ae..0632e85 100644
--- a/lib/Target/ARM/AsmParser/ARMAsmParser.cpp
+++ b/lib/Target/ARM/AsmParser/ARMAsmParser.cpp
@@ -3084,6 +3084,9 @@
     // Thumb LDM instructions are writeback iff the base register is not
     // in the register list.
     unsigned Rn = Inst.getOperand(0).getReg();
+    bool hasWritebackToken =
+      (static_cast<ARMOperand*>(Operands[3])->isToken() &&
+       static_cast<ARMOperand*>(Operands[3])->getToken() == "!");
     bool doesWriteback = true;
     for (unsigned i = 3; i < Inst.getNumOperands(); ++i) {
       unsigned Reg = Inst.getOperand(i).getReg();
@@ -3091,15 +3094,18 @@
         doesWriteback = false;
       // Anything other than a low register isn't legal here.
       if (!isARMLowRegister(Reg))
-        return Error(Operands[4]->getStartLoc(),
+        return Error(Operands[3 + hasWritebackToken]->getStartLoc(),
                      "registers must be in range r0-r7");
     }
     // If we should have writeback, then there should be a '!' token.
-    if (doesWriteback &&
-        (!static_cast<ARMOperand*>(Operands[3])->isToken() ||
-         static_cast<ARMOperand*>(Operands[3])->getToken() != "!"))
+    if (doesWriteback && !hasWritebackToken)
       return Error(Operands[2]->getStartLoc(),
                    "writeback operator '!' expected");
+    // Likewise, if we should not have writeback, there must not be a '!'
+    if (!doesWriteback && hasWritebackToken)
+      return Error(Operands[3]->getStartLoc(),
+                   "writeback operator '!' not allowed when base register "
+                   "in register list");
 
     break;
   }