[MC] Fix nested macro body parsing

Add missing .rep case in nestlevel checking for macro body parsing.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@337398 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/MC/MCParser/AsmParser.cpp b/lib/MC/MCParser/AsmParser.cpp
index 68e6e94..39a7608 100644 (file)
--- a/lib/MC/MCParser/AsmParser.cpp
+++ b/lib/MC/MCParser/AsmParser.cpp
@@ -5317,7 +5317,8 @@ MCAsmMacro *AsmParser::parseMacroLikeBody(SMLoc DirectiveLoc) {
     }
 
     if (Lexer.is(AsmToken::Identifier) &&
-        (getTok().getIdentifier() == ".rept" ||
+        (getTok().getIdentifier() == ".rep" ||
+         getTok().getIdentifier() == ".rept" ||
          getTok().getIdentifier() == ".irp" ||
          getTok().getIdentifier() == ".irpc")) {
       ++NestLevel;

diff --git a/test/MC/AsmParser/directive_rept.s b/test/MC/AsmParser/directive_rept.s
index 4f8ed7c..7bc20f7 100644 (file)
--- a/test/MC/AsmParser/directive_rept.s
+++ b/test/MC/AsmParser/directive_rept.s
@@ -28,3 +28,16 @@ half_a_dozen_daffodils:
 # CHECK: .long 3674148113
 # CHECK: .long 3674148113
 
+
+nested_reps:   
+       .rep 2
+       .rep 2
+       .long 0xdeadbeef
+       .endr
+       .endr
+
+# CHECK-LABEL: nested_reps
+# CHECK: .long   3735928559
+# CHECK: .long   3735928559
+# CHECK: .long   3735928559
+# CHECK: .long   3735928559