[llvm] r375287 - [X86] Fix register parsing in .seh_* in Intel syntax

[Reid Kleckner via llvm-commits]

Author: rnk
Date: Fri Oct 18 14:01:41 2019
New Revision: 375287

URL: http://llvm.org/viewvc/llvm-project?rev=375287&view=rev
Log:
[X86] Fix register parsing in .seh_* in Intel syntax

Previously, the parser checked for a '%' prefix to indicate a register.
In Intel syntax mode, LLVM does not print a '%' prefix on registers, so
LLVM could not parse its own assembly output. Instead, require that
register numbers be integer literals, or at least start with an integer
literal, which is consistent with .cfi_* directive register parsing.

Modified:
    llvm/trunk/lib/Target/X86/AsmParser/X86AsmParser.cpp
    llvm/trunk/test/MC/AsmParser/directive_seh.s

Modified: llvm/trunk/lib/Target/X86/AsmParser/X86AsmParser.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/X86/AsmParser/X86AsmParser.cpp?rev=375287&r1=375286&r2=375287&view=diff
==============================================================================
--- llvm/trunk/lib/Target/X86/AsmParser/X86AsmParser.cpp (original)
+++ llvm/trunk/lib/Target/X86/AsmParser/X86AsmParser.cpp Fri Oct 18 14:01:41 2019
@@ -3749,11 +3749,10 @@ bool X86AsmParser::parseSEHRegisterNumbe
   SMLoc startLoc = getLexer().getLoc();
   const MCRegisterInfo *MRI = getContext().getRegisterInfo();
 
-  // A percent indicates a symbolic register name. Parse it as usual and check
-  // the register class.
-  if (getLexer().is(AsmToken::Percent)) {
+  // Try parsing the argument as a register first.
+  if (getLexer().getTok().isNot(AsmToken::Integer)) {
     SMLoc endLoc;
-    if (getParser().getTargetParser().ParseRegister(RegNo, startLoc, endLoc))
+    if (ParseRegister(RegNo, startLoc, endLoc))
       return true;
 
     if (!X86MCRegisterClasses[RegClassID].contains(RegNo)) {

Modified: llvm/trunk/test/MC/AsmParser/directive_seh.s
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/MC/AsmParser/directive_seh.s?rev=375287&r1=375286&r2=375287&view=diff
==============================================================================
--- llvm/trunk/test/MC/AsmParser/directive_seh.s (original)
+++ llvm/trunk/test/MC/AsmParser/directive_seh.s Fri Oct 18 14:01:41 2019
@@ -1,5 +1,9 @@
 # RUN: llvm-mc -triple x86_64-pc-win32 %s | FileCheck %s
 
+#   Round trip via intel syntax printing and back.
+# RUN: llvm-mc -triple x86_64-pc-win32 %s -output-asm-variant=1 | \
+# RUN:     llvm-mc -triple x86_64-pc-win32 -x86-asm-syntax=intel | FileCheck %s
+
     .text
     .globl func
     .def func; .scl 2; .type 32; .endef
@@ -51,3 +55,45 @@ func:
     ret
     .seh_endproc
 # CHECK: .seh_endproc
+
+# Re-run more or less the same test, but with intel syntax. Previously LLVM
+# required percent prefixing in the .seh_* directives that take registers.
+
+    .intel_syntax noprefix
+    .text
+    .globl func_intel
+    .def func_intel; .scl 2; .type 32; .endef
+    .seh_proc func_intel
+# CHECK: .seh_proc func_intel
+func_intel:
+    sub RSP, 24
+    .seh_stackalloc 24
+# CHECK: .seh_stackalloc 24
+    mov [16+RSP], RSI
+    .seh_savereg rsi, 16
+# CHECK: .seh_savereg %rsi, 16
+    .seh_savereg 6, 16
+# CHECK: .seh_savereg %rsi, 16
+    movups [RSP], XMM8
+    .seh_savexmm XMM8, 0
+# CHECK: .seh_savexmm %xmm8, 0
+    .seh_savexmm 8, 0
+# CHECK: .seh_savexmm %xmm8, 0
+    push rbx
+    .seh_pushreg rbx
+# CHECK: .seh_pushreg %rbx
+    .seh_pushreg 3
+# CHECK: .seh_pushreg %rbx
+    mov rbx, rsp
+    .seh_setframe rbx, 0
+# CHECK: .seh_setframe %rbx, 0
+    .seh_endprologue
+# CHECK: .seh_endprologue
+    .seh_handler __C_specific_handler, @except
+# CHECK: .seh_handler __C_specific_handler, @except
+    .seh_handlerdata
+# CHECK-NOT: .section{{.*}}.xdata
+# CHECK: .seh_handlerdata
+    .long 0
+    .text
+    .seh_endproc