Validator_ragel: REX prefix support in gen_dfa.py.

src/trusted/validator_ragel/gen_dfa_test.py

 #!/usr/bin/python
 # Copyright (c) 2012 The Native Client Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import sys
 import unittest
 
 import gen_dfa
 
@@ skipping 168 matching lines @@
         printer.GetContent().split(),
         """
         0xeb
         @instruction_jmp
         @operands_count_is_1
         @operand0_8bit
         rel8
         @operand0_jmp_to
         """.split())
 
+  def test_no_modrm_with_rex(self):
+    printer = gen_dfa.InstructionPrinter(gen_dfa.DECODER, 64)
+    instr = gen_dfa.Instruction.Parse(
+        'movabs Od !ad, 0xa0, amd64 nacl-forbidden')
+
+    printer.PrintInstructionWithoutModRM(instr)
+
+    self.assertEquals(
+        printer.GetContent().split(),
+        """
+        REX_RXB?
+        0xa0
+        @instruction_movabs
+        @operands_count_is_2
+        @operand0_32bit
+        @operand1_32bit
+        @set_spurious_rex_b
+        @set_spurious_rex_x
+        @set_spurious_rex_r
+        @operand1_rax
+        @operand0_absolute_disp
+        disp64
+        """.split())
+
+  def test_nop_with_rex(self):
+    printer = gen_dfa.InstructionPrinter(gen_dfa.DECODER, 64)
+    instr = gen_dfa.Instruction.Parse(
+        'nop, 0x40 0x90, amd64 norex')
+
+    printer.PrintInstructionWithoutModRM(instr)
+
+    self.assertEquals(
+        printer.GetContent().split(),
+        """
+        0x40 0x90
+        @instruction_nop
+        @operands_count_is_0
+        @set_spurious_rex_b
+        @set_spurious_rex_x
+        @set_spurious_rex_r
+        """.split())
+
 
 class TestParser(unittest.TestCase):
 
   def test_instruction_definitions(self):
     def_filenames = sys.argv[1:]
     assert len(def_filenames) > 0
     for filename in def_filenames:
       gen_dfa.ParseDefFile(filename)
 
 
 if __name__ == '__main__':
   unittest.main(argv=[sys.argv[0], '--verbose'])