l4/pkg/udis86/lib/contrib/tests/testgen.py

   1 # udis86 - test/testgen.py
   2 #
   3 # Copyright (c) 2009 Vivek Thampi
   4 # All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without modification,
   7 # are permitted provided that the following conditions are met:
   8 #
   9 #     * Redistributions of source code must retain the above copyright notice,
  10 #       this list of conditions and the following disclaimer.
  11 #     * Redistributions in binary form must reproduce the above copyright notice,
  12 #       this list of conditions and the following disclaimer in the documentation
  13 #       and/or other materials provided with the distribution.
  14 #
  15 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  17 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18 # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
  19 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  20 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  21 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
  22 # ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  24 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25
  26 import os
  27 import sys
  28 import random
  29
  30 if ( len( os.getenv( 'UD_SCRIPT_DIR' ) ) ):
  31     scriptsPath = os.getenv( 'UD_SCRIPT_DIR' ) + "/scripts"
  32 else:
  33     scriptsPath = '../scripts'
  34 sys.path.append( scriptsPath );
  35
  36 import ud_optable
  37 import ud_opcode
  38 import testgen_opr
  39
  40 class UdTestGenerator( ud_opcode.UdOpcodeTables ):
  41
  42     OprTable = []
  43
  44     ExcludeList = ( 'fcomp3', 'fcom2', 'fcomp5', 'fstp1', 'fstp8', 'fstp9',
  45                     'fxch4', 'fxch7' )
  46
  47     def __init__( self ):
  48         pass
  49
  50     def generate_yasm( self, mode, seed ):
  51         random.seed( seed )
  52         print "[bits %s]" % mode
  53         for insn in self.InsnTable:
  54             if insn[ 'mnemonic' ] in self.ExcludeList:
  55                 continue
  56             if 'inv64' in insn[ 'prefixes' ] and mode == '64':
  57                 continue
  58             if 'def64' in insn[ 'prefixes' ] and mode != '64':
  59                 continue
  60             opr = '_'.join( insn[ 'operands' ] )
  61             if len( opr ):
  62                 if not opr in testgen_opr.OperandSet.keys():
  63                     print "Warning: no test-case for operand type '%s' (insn=%s)" % ( opr, insn[ 'mnemonic' ] )
  64                 testcase = testgen_opr.OperandSet[ opr ][ mode ]
  65                 if len( testcase ):
  66                     print "\t%s %s" % ( insn[ 'mnemonic' ], random.choice( testcase ) )
  67
  68 def main():
  69     generator = UdTestGenerator()
  70     optableXmlParser = ud_optable.UdOptableXmlParser()
  71     optableXmlParser.parse( sys.argv[ 1 ], generator.addInsnDef )
  72
  73     generator.generate_yasm( sys.argv[ 3 ], int( sys.argv[ 2 ] ) )
  74
  75 if __name__ == '__main__':
  76     main()