OSDN Git Service

[X86] Add segment and address-size override prefixes
authorBill Wendling <isanbard@gmail.com>
Thu, 14 Jan 2021 09:51:04 +0000 (01:51 -0800)
committerBill Wendling <isanbard@gmail.com>
Wed, 20 Jan 2021 07:54:31 +0000 (23:54 -0800)
X86 allows for the "addr32" and "addr16" address size override prefixes.
Also, these and the segment override prefixes should be recognized as
valid prefixes.

Differential Revision: https://reviews.llvm.org/D94726

llvm/lib/Target/X86/AsmParser/X86AsmParser.cpp
llvm/lib/Target/X86/X86InstrSystem.td
llvm/test/MC/X86/addr16-32.s [new file with mode: 0644]
llvm/test/MC/X86/segment-prefix.s [new file with mode: 0644]

index 406327e..e4ffe3f 100644 (file)
@@ -3260,11 +3260,13 @@ bool X86AsmParser::ParseInstruction(ParseInstructionInfo &Info, StringRef Name,
   // repz repnz <insn>    ; GAS errors for the use of two similar prefixes
   // lock addq %rax, %rbx ; Destination operand must be of memory type
   // xacquire <insn>      ; xacquire must be accompanied by 'lock'
-  bool isPrefix = StringSwitch<bool>(Name)
-                      .Cases("rex64", "data32", "data16", true)
-                      .Cases("xacquire", "xrelease", true)
-                      .Cases("acquire", "release", isParsingIntelSyntax())
-                      .Default(false);
+  bool IsPrefix =
+      StringSwitch<bool>(Name)
+          .Cases("cs", "ds", "es", "fs", "gs", "ss", true)
+          .Cases("rex64", "data32", "data16", "addr32", "addr16", true)
+          .Cases("xacquire", "xrelease", true)
+          .Cases("acquire", "release", isParsingIntelSyntax())
+          .Default(false);
 
   auto isLockRepeatNtPrefix = [](StringRef N) {
     return StringSwitch<bool>(N)
@@ -3333,7 +3335,7 @@ bool X86AsmParser::ParseInstruction(ParseInstructionInfo &Info, StringRef Name,
       Name = Next;
       PatchedName = Name;
       ForcedDataPrefix = X86::Mode32Bit;
-      isPrefix = false;
+      IsPrefix = false;
     }
   }
 
@@ -3350,7 +3352,7 @@ bool X86AsmParser::ParseInstruction(ParseInstructionInfo &Info, StringRef Name,
   // prefix juxtaposed with an operation like "lock incl 4(%rax)", because we
   // just want to parse the "lock" as the first instruction and the "incl" as
   // the next one.
-  if (getLexer().isNot(AsmToken::EndOfStatement) && !isPrefix) {
+  if (getLexer().isNot(AsmToken::EndOfStatement) && !IsPrefix) {
     // Parse '*' modifier.
     if (getLexer().is(AsmToken::Star))
       Operands.push_back(X86Operand::CreateToken("*", consumeToken()));
@@ -3387,7 +3389,7 @@ bool X86AsmParser::ParseInstruction(ParseInstructionInfo &Info, StringRef Name,
 
   // Consume the EndOfStatement or the prefix separator Slash
   if (getLexer().is(AsmToken::EndOfStatement) ||
-      (isPrefix && getLexer().is(AsmToken::Slash)))
+      (IsPrefix && getLexer().is(AsmToken::Slash)))
     Parser.Lex();
   else if (CurlyAsEndOfStatement)
     // Add an actual EndOfStatement before the curly brace
index f57ca7f..eb87408 100644 (file)
@@ -172,6 +172,17 @@ def GS_PREFIX : I<0x65, PrefixByte, (outs), (ins), "gs", []>;
 } // SchedRW
 
 //===----------------------------------------------------------------------===//
+// Address-size override prefixes.
+//
+
+let SchedRW = [WriteNop] in {
+def ADDR16_PREFIX : I<0x67, PrefixByte, (outs), (ins), "addr16", []>,
+                      Requires<[In32BitMode]>;
+def ADDR32_PREFIX : I<0x67, PrefixByte, (outs), (ins), "addr32", []>,
+                      Requires<[In64BitMode]>;
+} // SchedRW
+
+//===----------------------------------------------------------------------===//
 // Moves to and from segment registers.
 //
 
diff --git a/llvm/test/MC/X86/addr16-32.s b/llvm/test/MC/X86/addr16-32.s
new file mode 100644 (file)
index 0000000..8812ee8
--- /dev/null
@@ -0,0 +1,23 @@
+# RUN: llvm-mc %s -triple x86_64-linux-gnu -filetype=obj -o - | llvm-objdump -d - | FileCheck %s
+
+.text
+.global foo
+foo:
+       insl
+       gs outsl
+       .code64
+       addr32 insl
+       addr32 gs outsl
+       .code32
+       addr16 insl
+       addr16 gs outsl
+       .code64
+       retq
+
+# CHECK: <foo>:
+# CHECK-NEXT: 6d                            insl   %dx, %es:(%rdi)
+# CHECK-NEXT: 65 6f                         outsl  %gs:(%rsi), %dx
+# CHECK-NEXT: 67 6d                         insl   %dx, %es:(%edi)
+# CHECK-NEXT: 67 65 6f                      outsl  %gs:(%esi), %dx
+# CHECK-NEXT: 67 6d                         insl   %dx, %es:(%edi)
+# CHECK-NEXT: 67 65 6f                      outsl  %gs:(%esi), %dx
diff --git a/llvm/test/MC/X86/segment-prefix.s b/llvm/test/MC/X86/segment-prefix.s
new file mode 100644 (file)
index 0000000..ba61d48
--- /dev/null
@@ -0,0 +1,20 @@
+# RUN: llvm-mc %s -triple x86_64-linux-gnu -filetype=obj -o - | llvm-objdump -d - | FileCheck %s
+
+.text
+.global foo
+foo:
+       cs outsl
+       ds outsl
+       es outsw
+       fs outsw
+       gs outsl
+       ss outsl
+       retq
+
+# CHECK: <foo>:
+# CHECK-NEXT: 2e 6f                         outsl  %cs:(%rsi), %dx
+# CHECK-NEXT: 3e 6f                         outsl  %ds:(%rsi), %dx
+# CHECK-NEXT: 26 66 6f                      outsw  %es:(%rsi), %dx
+# CHECK-NEXT: 64 66 6f                      outsw  %fs:(%rsi), %dx
+# CHECK-NEXT: 65 6f                         outsl  %gs:(%rsi), %dx
+# CHECK-NEXT: 36 6f                         outsl  %ss:(%rsi), %dx