917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncfrom __future__ import print_function;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync__copyright__ = \

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync__version__ = "$Revision$";

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncimport io;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncimport os;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncfrom optparse import OptionParser

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncimport random;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncimport sys;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncRTEXITCODE_SUCCESS = 0;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncRTEXITCODE_SYNTAX = 2;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncUINT8_MAX = 0xff

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncUINT16_MAX = 0xffff

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncUINT32_MAX = 0xffffffff

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncUINT64_MAX = 0xffffffffffffffff

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef RT_BIT_32(iBit): # pylint: disable=C0103

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ 32-bit one bit mask. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 1 << iBit;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef RT_BIT_64(iBit): # pylint: disable=C0103

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ 64-bit one bit mask. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 1 << iBit;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_RM_MASK = 0x07;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_REG_MASK = 0x38;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_REG_SMASK = 0x07;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_REG_SHIFT = 3;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_MOD_MASK = 0xc0;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_MOD_SMASK = 0x03;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_MODRM_MOD_SHIFT = 6;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_BASE_MASK = 0x07;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_INDEX_MASK = 0x38;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_INDEX_SMASK = 0x07;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_INDEX_SHIFT = 3;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_SCALE_MASK = 0xc0;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_SCALE_SMASK = 0x03;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_SIB_SCALE_SHIFT = 6;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_CS = 0x2e;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_SS = 0x36;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_DS = 0x3e;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_ES = 0x26;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_FS = 0x64;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_GS = 0x65;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_SIZE_OP = 0x66;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_SIZE_ADDR = 0x67;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_LOCK = 0xf0;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_REPZ = 0xf2;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_PRF_REPNZ = 0xf3;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_REX_B = 0x41;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_REX_X = 0x42;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_REX_R = 0x44;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncX86_OP_REX_W = 0x48;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs64NoSp = ('rax', 'rcx', 'rdx', 'rbx', None, 'rbp', 'rsi', 'rdi', 'r8', 'r9', 'r10', 'r11', 'r12', 'r13', 'r14', 'r15');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs64 = ('rax', 'rcx', 'rdx', 'rbx', 'rsp', 'rbp', 'rsi', 'rdi', 'r8', 'r9', 'r10', 'r11', 'r12', 'r13', 'r14', 'r15');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs32NoSp = ('eax', 'ecx', 'edx', 'ebx', None, 'ebp', 'esi', 'edi',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'r8d', 'r9d', 'r10d', 'r11d', 'r12d', 'r13d', 'r14d', 'r15d');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs32 = ('eax', 'ecx', 'edx', 'ebx', 'esp', 'ebp', 'esi', 'edi',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'r8d', 'r9d', 'r10d', 'r11d', 'r12d', 'r13d', 'r14d', 'r15d');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs16NoSp = ('ax', 'cx', 'dx', 'bx', None, 'bp', 'si', 'di',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'r8w', 'r9w', 'r10w', 'r11w', 'r12w', 'r13w', 'r14w', 'r15w');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_asGRegs16 = ('ax', 'cx', 'dx', 'bx', 'sp', 'bp', 'si', 'di',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'r8w', 'r9w', 'r10w', 'r11w', 'r12w', 'r13w', 'r14w', 'r15w');

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsyncg_asGRegs8 = ('al', 'cl', 'dl', 'bl', 'ah', 'ah', 'dh', 'bh');

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsyncg_asGRegs8Rex = ('al', 'cl', 'dl', 'bl', 'spl', 'bpl', 'sil', 'dil',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'r8b', 'r9b', 'r10b', 'r11b', 'r12b', 'r13b', 'r14b', 'r15b');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_oMyRand = random.SystemRandom()

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef randU16():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Unsigned 16-bit random number. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return g_oMyRand.getrandbits(16);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef randU32():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Unsigned 32-bit random number. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return g_oMyRand.getrandbits(32);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef randU64():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Unsigned 64-bit random number. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return g_oMyRand.getrandbits(64);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef randUxx(cBits):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Unsigned 8-, 16-, 32-, or 64-bit random number. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return g_oMyRand.getrandbits(cBits);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef randUxxList(cBits, cElements):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ List of nsigned 8-, 16-, 32-, or 64-bit random numbers. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return [randUxx(cBits) for _ in range(cElements)];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef calcRexPrefixForTwoModRmRegs(iReg, iRm, bOtherRexPrefixes = 0):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync bRex = bOtherRexPrefixes;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iReg >= 8:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync bRex = bRex | X86_OP_REX_R;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iRm >= 8:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync bRex = bRex | X86_OP_REX_B;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if bRex == 0:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return [];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return [bRex,];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncdef calcModRmForTwoRegs(iReg, iRm):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync bRm = (0x3 << X86_MODRM_MOD_SHIFT) \

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync | ((iReg << X86_MODRM_REG_SHIFT) & X86_MODRM_REG_MASK) \

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync | (iRm & X86_MODRM_RM_MASK);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return [bRm,];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsyncdef convU32ToSigned(u32):

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync """ Converts a 32-bit unsigned value to 32-bit signed. """

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync if u32 < 0x80000000:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return u32;

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return u32 - UINT32_MAX - 1;

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsyncdef gregName(iReg, cBits, fRexByteRegs = True):

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync """ Gets the name of a general register by index and width. """

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync if cBits == 64:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return g_asGRegs64[iReg];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync if cBits == 32:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return g_asGRegs32[iReg];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync if cBits == 16:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return g_asGRegs16[iReg];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync assert cBits == 8;

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync if fRexByteRegs:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return g_asGRegs8Rex[iReg];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return g_asGRegs8[iReg];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncclass TargetEnv(object):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @name CPU Modes

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @{

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksCpuMode_Real = 'real';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksCpuMode_Protect = 'prot';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksCpuMode_Paged = 'paged';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksCpuMode_Long = 'long';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksCpuMode_V86 = 'v86';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @}

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @name Instruction set.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @{

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksInstrSet_16 = '16';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksInstrSet_32 = '32';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ksInstrSet_64 = '64';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ## @}

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def __init__(self, sName,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sInstrSet = ksInstrSet_32,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sCpuMode = ksCpuMode_Paged,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iRing = 3,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sName = sName;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sInstrSet = sInstrSet;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sCpuMode = sCpuMode;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.iRing = iRing;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.asGRegs = g_asGRegs64 if self.is64Bit() else g_asGRegs32;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.asGRegsNoSp = g_asGRegs64NoSp if self.is64Bit() else g_asGRegs32NoSp;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def isUsingIprt(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Whether it's an IPRT environment or not. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.sName.startswith('iprt');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def is64Bit(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Whether it's a 64-bit environment or not. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.sInstrSet == self.ksInstrSet_64;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def getDefOpBits(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Get the default operand size as a bit count. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.sInstrSet == self.ksInstrSet_16:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 16;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 32;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def getDefOpBytes(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Get the default operand size as a byte count. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.getDefOpBits() / 8;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def getMaxOpBits(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Get the max operand size as a bit count. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.sInstrSet == self.ksInstrSet_64:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 64;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 32;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def getMaxOpBytes(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Get the max operand size as a byte count. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.getMaxOpBits() / 8;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def getDefAddrBits(self):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync """ Get the default address size as a bit count. """

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if self.sInstrSet == self.ksInstrSet_16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return 16;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if self.sInstrSet == self.ksInstrSet_32:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return 32;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return 64;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def getDefAddrBytes(self):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync """ Get the default address size as a byte count. """

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return self.getDefAddrBits() / 8;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def getGRegCount(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Get the number of general registers. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.sInstrSet == self.ksInstrSet_64:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 16;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 8;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync def randGRegNoSp(self):

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync """ Returns a random general register number, excluding the SP register. """

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync iReg = randU16();

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync return iReg % (16 if self.is64Bit() else 8);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync def getAddrModes(self):

270236340676d2385b27ea992e07fcb643bb78b6vboxsync """ Gets a list of addressing mode (16, 32, or/and 64). """

270236340676d2385b27ea992e07fcb643bb78b6vboxsync if self.sInstrSet == self.ksInstrSet_16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return [16, 32];

270236340676d2385b27ea992e07fcb643bb78b6vboxsync if self.sInstrSet == self.ksInstrSet_32:

270236340676d2385b27ea992e07fcb643bb78b6vboxsync return [32, 16];

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync return [64, 32];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_dTargetEnvs = {

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'iprt-r3-32': TargetEnv('iprt-r3-32', TargetEnv.ksInstrSet_32, TargetEnv.ksCpuMode_Protect, 3),

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'iprt-r3-64': TargetEnv('iprt-r3-64', TargetEnv.ksInstrSet_64, TargetEnv.ksCpuMode_Long, 3),

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync};

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsyncclass InstrTestBase(object):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def __init__(self, sName, sInstr = None):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sName = sName;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sInstr = sInstr if sInstr else sName.split()[0];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync def isApplicable(self, oGen):

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync """

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync _ = oGen;

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def generateTest(self, oGen, sTestFnName):

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.write(';; @todo not implemented. This is for the linter: %s, %s\n' % (oGen, sTestFnName));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync def generateInputs(self, cbEffOp, cbMaxOp, oGen, fLong = False):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Generate a list of inputs. """

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if fLong:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync #

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync # Try do extremes as well as different ranges of random numbers.

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync #

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet = [0, 1, ];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cbMaxOp >= 1:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += [ UINT8_MAX / 2, UINT8_MAX / 2 + 1, UINT8_MAX ];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cbMaxOp >= 2:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += [ UINT16_MAX / 2, UINT16_MAX / 2 + 1, UINT16_MAX ];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cbMaxOp >= 4:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += [ UINT32_MAX / 2, UINT32_MAX / 2 + 1, UINT32_MAX ];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cbMaxOp >= 8:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += [ UINT64_MAX / 2, UINT64_MAX / 2 + 1, UINT64_MAX ];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if oGen.oOptions.sTestSize == InstructionTestGen.ksTestSize_Tiny:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync for cBits, cValues in ( (8, 4), (16, 4), (32, 8), (64, 8) ):

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cBits < cbMaxOp * 8:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cBits, cValues);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync cWanted = 16;

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync elif oGen.oOptions.sTestSize == InstructionTestGen.ksTestSize_Medium:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync for cBits, cValues in ( (8, 8), (16, 8), (24, 2), (32, 16), (40, 1), (48, 1), (56, 1), (64, 16) ):

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cBits < cbMaxOp * 8:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cBits, cValues);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync cWanted = 64;

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync else:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync for cBits, cValues in ( (8, 16), (16, 16), (24, 4), (32, 64), (40, 4), (48, 4), (56, 4), (64, 64) ):

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cBits < cbMaxOp * 8:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cBits, cValues);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync cWanted = 168;

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if len(auRet) < cWanted:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cbEffOp * 8, cWanted - len(auRet));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync #

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync # Short list, just do some random numbers.

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync #

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet = [];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if oGen.oOptions.sTestSize == InstructionTestGen.ksTestSize_Tiny:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cbMaxOp, 1);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync elif oGen.oOptions.sTestSize == InstructionTestGen.ksTestSize_Medium:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cbMaxOp, 2);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync else:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet = [];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync for cBits in (8, 16, 32, 64):

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if cBits < cbMaxOp * 8:

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auRet += randUxxList(cBits, 1);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return auRet;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsyncclass InstrTest_MemOrGreg_2_Greg(InstrTestBase):

270236340676d2385b27ea992e07fcb643bb78b6vboxsync """

270236340676d2385b27ea992e07fcb643bb78b6vboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync def __init__(self, sName, fnCalcResult, sInstr = None, acbOpVars = None):

270236340676d2385b27ea992e07fcb643bb78b6vboxsync InstrTestBase.__init__(self, sName, sInstr);

270236340676d2385b27ea992e07fcb643bb78b6vboxsync self.fnCalcResult = fnCalcResult;

270236340676d2385b27ea992e07fcb643bb78b6vboxsync self.acbOpVars = [ 1, 2, 4, 8 ] if not acbOpVars else list(acbOpVars);

270236340676d2385b27ea992e07fcb643bb78b6vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def writeInstrGregGreg(self, cbEffOp, iOp1, iOp2, oGen):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Writes the instruction with two general registers as operands. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if cbEffOp == 8:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' %s %s, %s\n' % (self.sInstr, g_asGRegs64[iOp1], g_asGRegs64[iOp2]));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync elif cbEffOp == 4:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' %s %s, %s\n' % (self.sInstr, g_asGRegs32[iOp1], g_asGRegs32[iOp2]));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync elif cbEffOp == 2:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' %s %s, %s\n' % (self.sInstr, g_asGRegs16[iOp1], g_asGRegs16[iOp2]));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cbEffOp == 1:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync oGen.write(' %s %s, %s\n' % (self.sInstr, g_asGRegs8Rex[iOp1], g_asGRegs8Rex[iOp2]));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync assert False;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync return True;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync def writeInstrGregPureRM(self, cbEffOp, iOp1, cAddrBits, iOp2, iMod, offDisp, oGen):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync """ Writes the instruction with two general registers as operands. """

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if cbEffOp == 8:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' %s %s, [' % (self.sInstr, g_asGRegs64[iOp1],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync elif cbEffOp == 4:

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.write(' %s %s, [' % (self.sInstr, g_asGRegs32[iOp1],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync elif cbEffOp == 2:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' %s %s, [' % (self.sInstr, g_asGRegs16[iOp1],));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cbEffOp == 1:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync oGen.write(' %s %s, [' % (self.sInstr, g_asGRegs8Rex[iOp1],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync assert False;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync if (iOp2 == 5 or iOp2 == 13) and iMod == 0:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write('VBINSTST_NAME(g_u%sData)' % (cbEffOp * 8,))

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if oGen.oTarget.is64Bit():

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' wrt rip');

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if iMod == 1:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write('byte %d + ' % (offDisp,));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync elif iMod == 2:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write('dword %d + ' % (offDisp,));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync assert iMod == 0;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if cAddrBits == 64:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(g_asGRegs64[iOp2]);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync elif cAddrBits == 32:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(g_asGRegs32[iOp2]);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync elif cAddrBits == 16:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync assert False; ## @todo implement 16-bit addressing.

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync assert False, str(cAddrBits);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(']\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync def generateOneStdTestGregGreg(self, oGen, cbEffOp, cbMaxOp, iOp1, iOp2, uInput, uResult):

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync """ Generate one standard test. """

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync oGen.write(' call VBINSTST_NAME(Common_LoadKnownValues)\n');

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync oGen.write(' mov %s, 0x%x\n' % (oGen.oTarget.asGRegs[iOp2], uInput,));

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync oGen.write(' push %s\n' % (oGen.oTarget.asGRegs[iOp2],));

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync self.writeInstrGregGreg(cbEffOp, iOp1, iOp2, oGen);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.pushConst(uResult);

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync oGen.write(' call VBINSTST_NAME(%s)\n' % (oGen.needGRegChecker(iOp1, iOp2),));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync _ = cbMaxOp;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync return True;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync def generateMemSetupPureRM(self, oGen, cAddrBits, cbEffOp, iOp2, iMod, uInput, offDisp = None):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync """ Sets up memory for a pure R/M addressed read, iOp2 being the R/M value. """

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.pushConst(uInput);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync assert offDisp is None or iMod != 0;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if (iOp2 != 5 and iOp2 != 13) or iMod != 0:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' call VBINSTST_NAME(%s)\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync % (oGen.needGRegMemSetup(cAddrBits, cbEffOp, iOp2, offDisp),));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync ## @todo generate REX.B=1 variant.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync oGen.write(' call VBINSTST_NAME(Common_SetupMemReadU%u)\n' % (cbEffOp*8,));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' push %s\n' % (oGen.oTarget.asGRegs[iOp2],));

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync return True;

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync def generateOneStdTestGregMemNoSib(self, oGen, cAddrBits, cbEffOp, cbMaxOp, iOp1, iOp2, uInput, uResult):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync """ Generate mode 0, 1 and 2 test for the R/M=iOp2. """

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if cAddrBits == 16:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync _ = cbMaxOp;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync iMod = 0; # No disp, except for i=5.

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' call VBINSTST_NAME(Common_LoadKnownValues)\n');

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.generateMemSetupPureRM(oGen, cAddrBits, cbEffOp, iOp2, iMod, uInput);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.writeInstrGregPureRM(cbEffOp, iOp1, cAddrBits, iOp2, iMod, None, oGen);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.pushConst(uResult);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.write(' call VBINSTST_NAME(%s)\n' % (oGen.needGRegChecker(iOp1, iOp2),));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync if iOp2 != 5 and iOp2 != 13:

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync iMod = 1;

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync for offDisp in (127, -128):

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.write(' call VBINSTST_NAME(Common_LoadKnownValues)\n');

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync self.generateMemSetupPureRM(oGen, cAddrBits, cbEffOp, iOp2, iMod, uInput, offDisp);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync self.writeInstrGregPureRM(cbEffOp, iOp1, cAddrBits, iOp2, iMod, offDisp, oGen);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.pushConst(uResult);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.write(' call VBINSTST_NAME(%s)\n' % (oGen.needGRegChecker(iOp1, iOp2),));

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync iMod = 2;

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync for offDisp in (2147483647, -2147483648):

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.write(' call VBINSTST_NAME(Common_LoadKnownValues)\n');

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync self.generateMemSetupPureRM(oGen, cAddrBits, cbEffOp, iOp2, iMod, uInput, offDisp);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync self.writeInstrGregPureRM(cbEffOp, iOp1, cAddrBits, iOp2, iMod, offDisp, oGen);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.pushConst(uResult);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync oGen.write(' call VBINSTST_NAME(%s)\n' % (oGen.needGRegChecker(iOp1, iOp2),));

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync return True;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def generateStandardTests(self, oGen):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Generate standard tests. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync # Parameters.

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync cbDefOp = oGen.oTarget.getDefOpBytes();

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync cbMaxOp = oGen.oTarget.getMaxOpBytes();

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auShortInputs = self.generateInputs(cbDefOp, cbMaxOp, oGen);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync auLongInputs = self.generateInputs(cbDefOp, cbMaxOp, oGen, fLong = True);

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync iLongOp1 = oGen.oTarget.randGRegNoSp();

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync iLongOp2 = oGen.oTarget.randGRegNoSp();

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync oOp2Range = range(oGen.oTarget.getGRegCount());

270236340676d2385b27ea992e07fcb643bb78b6vboxsync oOp1MemRange = range(oGen.oTarget.getGRegCount());

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync if oGen.oOptions.sTestSize == InstructionTestGen.ksTestSize_Tiny:

270236340676d2385b27ea992e07fcb643bb78b6vboxsync oOp2Range = [iLongOp2,];

270236340676d2385b27ea992e07fcb643bb78b6vboxsync oOp1MemRange = [iLongOp1,];

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

270236340676d2385b27ea992e07fcb643bb78b6vboxsync # Register tests

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if False:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for cbEffOp in self.acbOpVars:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if cbEffOp > cbMaxOp:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync continue;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for iOp1 in range(oGen.oTarget.getGRegCount()):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if oGen.oTarget.asGRegsNoSp[iOp1] is None:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync continue;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for iOp2 in oOp2Range:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if oGen.oTarget.asGRegsNoSp[iOp2] is None:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync continue;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for uInput in (auLongInputs if iOp1 == iLongOp1 and iOp2 == iLongOp2 else auShortInputs):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync uResult = self.fnCalcResult(cbEffOp, uInput,

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.auRegValues[iOp1] if iOp1 != iOp2 else uInput, oGen);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.newSubTest();

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.generateOneStdTestGregGreg(oGen, cbEffOp, cbMaxOp, iOp1, iOp2, uInput, uResult);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync # Memory test.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for cbEffOp in self.acbOpVars:

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync if cbEffOp > cbMaxOp:

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync continue;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for iOp1 in oOp1MemRange:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if oGen.oTarget.asGRegsNoSp[iOp1] is None:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync continue;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for cAddrBits in oGen.oTarget.getAddrModes():

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for iOp2 in range(len(oGen.oTarget.asGRegs)):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if iOp2 != 4:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync for uInput in (auLongInputs if iOp1 == iLongOp1 and False else auShortInputs):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync oGen.newSubTest();

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if iOp1 == iOp2 and iOp2 != 5 and iOp2 != 13 and cbEffOp != cbMaxOp:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync continue; # Don't know the high bit of the address ending up the result - skip it.

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync uResult = self.fnCalcResult(cbEffOp, uInput, oGen.auRegValues[iOp1], oGen);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.generateOneStdTestGregMemNoSib(oGen, cAddrBits, cbEffOp, cbMaxOp,

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync iOp1, iOp2, uInput, uResult);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync pass; # SIB

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def generateTest(self, oGen, sTestFnName):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.write('VBINSTST_BEGINPROC %s\n' % (sTestFnName,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #oGen.write(' int3\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.generateStandardTests(oGen);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #oGen.write(' int3\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.write(' ret\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.write('VBINSTST_ENDPROC %s\n' % (sTestFnName,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsyncclass InstrTest_Mov_Gv_Ev(InstrTest_MemOrGreg_2_Greg):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync """

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def __init__(self):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync InstrTest_MemOrGreg_2_Greg.__init__(self, 'mov Gv,Ev', self.calc_mov);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync @staticmethod

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def calc_mov(cbEffOp, uInput, uCur, oGen):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync """ Calculates the result of a mov instruction."""

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cbEffOp == 8:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return uInput & UINT64_MAX;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cbEffOp == 4:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return uInput & UINT32_MAX;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cbEffOp == 2:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return (uCur & 0xffffffffffff0000) | (uInput & UINT16_MAX);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync assert cbEffOp == 1; _ = oGen;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync return (uCur & 0xffffffffffffff00) | (uInput & UINT8_MAX);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncclass InstrTest_MovSxD_Gv_Ev(InstrTest_MemOrGreg_2_Greg):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def __init__(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync InstrTest_MemOrGreg_2_Greg.__init__(self, 'movsxd Gv,Ev', self.calc_movsxd, acbOpVars = [ 8, 4, 2, ]);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def writeInstrGregGreg(self, cbEffOp, iOp1, iOp2, oGen):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if cbEffOp == 8:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.write(' %s %s, %s\n' % (self.sInstr, g_asGRegs64[iOp1], g_asGRegs32[iOp2]));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync elif cbEffOp == 4 or cbEffOp == 2:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync abInstr = [];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if cbEffOp != oGen.oTarget.getDefOpBytes():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync abInstr.append(X86_OP_PRF_SIZE_OP);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync abInstr += calcRexPrefixForTwoModRmRegs(iOp1, iOp2);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync abInstr.append(0x63);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync abInstr += calcModRmForTwoRegs(iOp1, iOp2);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oGen.writeInstrBytes(abInstr);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync assert False;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync assert False;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync def isApplicable(self, oGen):

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync return oGen.oTarget.is64Bit();

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync @staticmethod

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def calc_movsxd(cbEffOp, uInput, uCur, oGen):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync _ = oGen;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if cbEffOp == 8 and (uInput & RT_BIT_32(31)):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return (UINT32_MAX << 32) | (uInput & UINT32_MAX);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if cbEffOp == 2:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return (uCur & 0xffffffffffff0000) | (uInput & 0xffff);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return uInput & UINT32_MAX;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncg_aoInstructionTests = [

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync InstrTest_Mov_Gv_Ev(),

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync #InstrTest_MovSxD_Gv_Ev(),

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncclass InstructionTestGen(object):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ## @name Test size

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ## @{

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ksTestSize_Large = 'large';

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ksTestSize_Medium = 'medium';

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ksTestSize_Tiny = 'tiny';

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync ## @}

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync kasTestSizes = ( ksTestSize_Large, ksTestSize_Medium, ksTestSize_Tiny );

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def __init__(self, oOptions):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oOptions = oOptions;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oTarget = g_dTargetEnvs[oOptions.sTargetEnv];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Calculate the number of output files.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.cFiles = 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if len(g_aoInstructionTests) > self.oOptions.cInstrPerFile:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.cFiles = len(g_aoInstructionTests) / self.oOptions.cInstrPerFile;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.cFiles * self.oOptions.cInstrPerFile < len(g_aoInstructionTests):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.cFiles += 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Fix the known register values.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.au64Regs = randUxxList(64, 16);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.au32Regs = [(self.au64Regs[i] & UINT32_MAX) for i in range(8)];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.au16Regs = [(self.au64Regs[i] & UINT16_MAX) for i in range(8)];

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync self.auRegValues = self.au64Regs if self.oTarget.is64Bit() else self.au32Regs;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Declare state variables used while generating.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oFile = sys.stderr;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.iFile = -1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sFile = '';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.dCheckFns = dict();

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.dMemSetupFns = dict();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.d64BitConsts = dict();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Methods used by instruction tests.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def write(self, sText):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Writes to the current output file. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.oFile.write(unicode(sText));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def writeln(self, sText):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """ Writes a line to the current output file. """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(sText);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.write('\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def writeInstrBytes(self, abInstr):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' db %#04x' % (abInstr[0],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(1, len(abInstr)):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(', %#04x' % (abInstr[i],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self.write('\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def newSubTest(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov dword [VBINSTST_NAME(g_uVBInsTstSubTestIndicator) xWrtRIP], __LINE__\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def needGRegChecker(self, iReg1, iReg2):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync assert iReg1 < 32; assert iReg2 < 32;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iRegs = iReg1 + iReg2 * 32;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iRegs in self.dCheckFns:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.dCheckFns[iRegs] += 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.dCheckFns[iRegs] = 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 'Common_Check_%s_%s' % (self.oTarget.asGRegs[iReg1], self.oTarget.asGRegs[iReg2]);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync def needGRegMemSetup(self, cAddrBits, cbEffOp, iReg1, offDisp = None):

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync """

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync sName = '%ubit_U%u_%s' % (cAddrBits, cbEffOp * 8, gregName(iReg1, cAddrBits),);

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if offDisp is not None:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sName += '_%#010x' % (offDisp & UINT32_MAX, );

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if sName in self.dCheckFns:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.dMemSetupFns[sName] += 1;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync else:

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync self.dMemSetupFns[sName] = 1;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync return 'Common_MemSetup_' + sName;

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def need64BitConstant(self, uVal):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync assert uVal >= 0 and uVal <= UINT64_MAX;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if uVal in self.d64BitConsts:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.d64BitConsts[uVal] += 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.d64BitConsts[uVal] = 1;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return 'g_u64Const_0x%016x' % (uVal, );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync def pushConst(self, uResult):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync if self.oTarget.is64Bit() and uResult >= 0x80000000:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' push qword [%s wrt rip]\n' % (self.need64BitConstant(uResult),));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' push dword 0x%x\n' % (uResult,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Internal machinery.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def _calcTestFunctionName(self, oInstrTest, iInstrTest):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sName = 'TestInstr%03u_%s' % (iInstrTest, oInstrTest.sName);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return sName.replace(',', '_').replace(' ', '_').replace('%', '_');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def _generateFileHeader(self, ):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('; $Id$\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';; @file %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; Autogenerate by %s %s. DO NOT EDIT\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; Headers\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '%%include "env-%s.mac"\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( os.path.basename(self.sFile),

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync os.path.basename(__file__), __version__[11:-1],

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oTarget.sName,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ) );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Target environment specific init stuff.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Global variables.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; Globals\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('VBINSTST_BEGINDATA\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync 'VBINSTST_GLOBALNAME_EX g_pvLow16Mem4K, data hidden\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync ' dq 0\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync 'VBINSTST_GLOBALNAME_EX g_pvLow32Mem4K, data hidden\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync ' dq 0\n'

f372af8e6ee2a011213b11cc69f4a29530ff7ce5vboxsync 'VBINSTST_GLOBALNAME_EX g_pvMem4K, data hidden\n'

270236340676d2385b27ea992e07fcb643bb78b6vboxsync ' dq 0\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_GLOBALNAME_EX g_uVBInsTstSubTestIndicator, data hidden\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' dd 0\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_BEGINCODE\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%ifdef RT_ARCH_AMD64\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(len(g_asGRegs64)):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('g_u64KnownValue_%s: dq 0x%x\n' % (g_asGRegs64[i], self.au64Regs[i]));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%endif\n\n')

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Common functions.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Loading common values.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_BEGINPROC Common_LoadKnownValues\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '%ifdef RT_ARCH_AMD64\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(len(g_asGRegs64NoSp)):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if g_asGRegs64NoSp[i]:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, 0x%x\n' % (g_asGRegs64NoSp[i], self.au64Regs[i],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%else\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(8):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if g_asGRegs32NoSp[i]:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, 0x%x\n' % (g_asGRegs32NoSp[i], self.au32Regs[i],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%endif\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' ret\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_ENDPROC Common_LoadKnownValues\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('VBINSTST_BEGINPROC Common_CheckKnownValues\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '%ifdef RT_ARCH_AMD64\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(len(g_asGRegs64NoSp)):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if g_asGRegs64NoSp[i]:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' cmp %s, [g_u64KnownValue_%s wrt rip]\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' je .ok_%u\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %u ; register number\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; actual\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push qword [g_u64KnownValue_%s wrt rip] ; expected\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' call VBINSTST_NAME(Common_BadValue)\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '.ok_%u:\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( g_asGRegs64NoSp[i], g_asGRegs64NoSp[i], i, i, g_asGRegs64NoSp[i], g_asGRegs64NoSp[i], i,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%else\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for i in range(8):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if g_asGRegs32NoSp[i]:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' cmp %s, 0x%x\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' je .ok_%u\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %u ; register number\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; actual\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push dword 0x%x ; expected\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' call VBINSTST_NAME(Common_BadValue)\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '.ok_%u:\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( g_asGRegs32NoSp[i], self.au32Regs[i], i, i, g_asGRegs32NoSp[i], self.au32Regs[i], i,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('%endif\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' ret\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_ENDPROC Common_CheckKnownValues\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync def _generateMemSetupFunctions(self):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync """

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync cDefAddrBits = self.oTarget.getDefAddrBits();

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync for sName in self.dMemSetupFns:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Unpack it.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync asParams = sName.split('_');

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync cAddrBits = int(asParams[0][:-3]);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync cEffOpBits = int(asParams[1][1:]);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sBaseReg = asParams[2];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cAddrBits == 64: asAddrGRegs = g_asGRegs64;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cAddrBits == 32: asAddrGRegs = g_asGRegs32;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else: asAddrGRegs = g_asGRegs16;

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync try:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync iBaseReg = asAddrGRegs.index(sBaseReg);

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync except ValueError:

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync assert False, 'sBaseReg=%s' % (sBaseReg,);

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync raise;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync i = 3;

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync u32Disp = None;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if i < len(asParams) and len(asParams[i]) == 10:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync u32Disp = long(asParams[i], 16);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync i += 1;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sIndexReg = None;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync iIndexReg = None;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if i < len(asParams):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sIndexReg = asParams[i];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync iBaseReg = asAddrGRegs.index(sBaseReg);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync i += 1;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync iScale = 1;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if i < len(asParams):

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync iScale = int(asParams[i]); assert iScale in [2, 4, 8];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync i += 1;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync assert i == len(asParams), 'i=%d len=%d len[i]=%d (%s)' % (i, len(asParams), len(asParams[i]), asParams[i],);

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Prologue.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync iTmpReg1 = 0 if iBaseReg != 0 else 1;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write('\n\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync 'VBINSTST_BEGINPROC Common_MemSetup_%s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync ' MY_PUSH_FLAGS\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync ' push %s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % (sName, self.oTarget.asGRegs[iTmpReg1], ));

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync self.write('; cAddrBits=%s cEffOpBits=%s iBaseReg=%s u32Disp=%s iIndexReg=%s iScale=%s\n'

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync % (cAddrBits, cEffOpBits, iBaseReg, u32Disp, iIndexReg, iScale,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Figure out what to use.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cEffOpBits == 64:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sTmpReg1 = g_asGRegs64[iTmpReg1];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sDataVar = 'VBINSTST_NAME(g_u64Data)';

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cEffOpBits == 32:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sTmpReg1 = g_asGRegs32[iTmpReg1];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sDataVar = 'VBINSTST_NAME(g_u32Data)';

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cEffOpBits == 16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sTmpReg1 = g_asGRegs16[iTmpReg1];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sDataVar = 'VBINSTST_NAME(g_u16Data)';

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync assert cEffOpBits == 8;

ac91d9d0626a52f79cf554c81b1fdf5f262092bbvboxsync sTmpReg1 = g_asGRegs8Rex[iTmpReg1];

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync sDataVar = 'VBINSTST_NAME(g_u8Data)';

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Load the value and mem address, sorting the value there.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, [xSP + sCB + MY_PUSH_FLAGS_SIZE + xCB]\n' % (sTmpReg1,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cAddrBits >= cDefAddrBits:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov [%s xWrtRIP], %s\n' % (sDataVar, sTmpReg1,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' lea %s, [%s xWrtRIP]\n' % (sBaseReg, sDataVar,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cAddrBits == 16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, [VBINSTST_NAME(g_pvLow16Mem4K) xWrtRIP]\n' % (sBaseReg,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, [VBINSTST_NAME(g_pvLow32Mem4K) xWrtRIP]\n' % (sBaseReg,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' add %s, %s\n' % (sBaseReg, (randU16() << cEffOpBits) & 0xfff, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov [%s], %s\n' % (sBaseReg, sTmpReg1, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Adjust for disposition and scaling.

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync if u32Disp is not None:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' sub %s, %d\n' % ( sBaseReg, convU32ToSigned(u32Disp), ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if iIndexReg is not None:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync uIdxRegVal = randUxx(cAddrBits);

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, %u\n' % ( sIndexReg, uIdxRegVal,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cAddrBits == 64:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' sub %s, %#06x\n' % ( sBaseReg, uIdxRegVal * iScale, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync elif cAddrBits == 16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' sub %s, %#06x\n' % ( sBaseReg, (uIdxRegVal * iScale) & UINT32_MAX, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync assert cAddrBits == 16;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' sub %s, %#06x\n' % ( sBaseReg, (uIdxRegVal * iScale) & UINT16_MAX, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Set upper bits that's supposed to be unused.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cDefAddrBits > cAddrBits or cAddrBits == 16:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if cDefAddrBits == 64:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync assert cAddrBits == 32;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, %#018x\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync ' or %s, %s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % ( g_asGRegs64[iTmpReg1], randU64() & 0xffffffff00000000,

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync g_asGRegs64[iBaseReg], g_asGRegs64[iTmpReg1],));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if iIndexReg is not None:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' mov %s, %#018x\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync ' or %s, %s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % ( g_asGRegs64[iTmpReg1], randU64() & 0xffffffff00000000,

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync g_asGRegs64[iIndexReg], g_asGRegs64[iTmpReg1],));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync else:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync assert cDefAddrBits == 32; assert cAddrBits == 16;

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' or %s, %#010x\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % ( g_asGRegs32[iBaseReg], randU32() & 0xffff0000, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync if iIndexReg is not None:

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' or %s, %#010x\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % ( g_asGRegs32[iIndexReg], randU32() & 0xffff0000, ));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # Epilogue.

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self.write(' pop %s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync ' MY_POP_FLAGS\n'

bec4d1c0274e4712fe01426313aab120b5ad1c17vboxsync ' ret sCB\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync 'VBINSTST_ENDPROC Common_MemSetup_%s\n'

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync % ( self.oTarget.asGRegs[iTmpReg1], sName,));

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def _generateFileFooter(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Register checking functions.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for iRegs in self.dCheckFns:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iReg1 = iRegs & 0x1f;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sReg1 = self.oTarget.asGRegs[iReg1];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iReg2 = (iRegs >> 5) & 0x1f;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sReg2 = self.oTarget.asGRegs[iReg2];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sPushSize = 'dword';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; Checks two register values, expected values pushed on the stack.\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; To save space, the callee cleans up the stack.'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; Ref count: %u\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_BEGINPROC Common_Check_%s_%s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' MY_PUSH_FLAGS\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( self.dCheckFns[iRegs], sReg1, sReg2, ) );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' cmp %s, [xSP + MY_PUSH_FLAGS_SIZE + xCB]\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' je .equal1\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s %u ; register number\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; actual\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' mov %s, [xSP + sCB*2 + MY_PUSH_FLAGS_SIZE + xCB]\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; expected\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' call VBINSTST_NAME(Common_BadValue)\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '.equal1:\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( sReg1, sPushSize, iReg1, sReg1, sReg1, sReg1, sReg1, sReg1, sReg1, ) );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iReg1 != iReg2: # If input and result regs are the same, only check the result.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' cmp %s, [xSP + sCB + MY_PUSH_FLAGS_SIZE + xCB]\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' je .equal2\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s %u ; register number\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; actual\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' mov %s, [xSP + sCB*3 + MY_PUSH_FLAGS_SIZE + xCB]\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' push %s ; expected\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' call VBINSTST_NAME(Common_BadValue)\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' pop %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '.equal2:\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % ( sReg2, sPushSize, iReg2, sReg2, sReg2, sReg2, sReg2, sReg2, sReg2, ) );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.oTarget.is64Bit():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, [g_u64KnownValue_%s wrt rip]\n' % (sReg1, sReg1,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iReg1 != iReg2:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, [g_u64KnownValue_%s wrt rip]\n' % (sReg2, sReg2,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, 0x%x\n' % (sReg1, self.au32Regs[iReg1],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iReg1 != iReg2:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' mov %s, 0x%x\n' % (sReg2, self.au32Regs[iReg2],));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write(' MY_POP_FLAGS\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' call VBINSTST_NAME(Common_CheckKnownValues)\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' ret sCB*2\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_ENDPROC Common_Check_%s_%s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % (sReg1, sReg2,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync # memory setup functions

c6383709c15c809f8cfb09b5cfe670760f06e2b9vboxsync self._generateMemSetupFunctions();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # 64-bit constants.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if len(self.d64BitConsts) > 0:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; 64-bit constants\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for uVal in self.d64BitConsts:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('g_u64Const_0x%016x: dq 0x%016x ; Ref count: %d\n' % (uVal, uVal, self.d64BitConsts[uVal], ) );

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return True;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def _generateTests(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for self.iFile in range(self.cFiles):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.cFiles == 1:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sFile = '%s.asm' % (self.oOptions.sOutputBase,)

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sFile = '%s-%u.asm' % (self.oOptions.sOutputBase, self.iFile)

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oFile = sys.stdout;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.oOptions.sOutputBase != '-':

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oFile = io.open(self.sFile, 'w', encoding = 'utf-8');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self._generateFileHeader();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Calc the range.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iInstrTestStart = self.iFile * self.oOptions.cInstrPerFile;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iInstrTestEnd = iInstrTestStart + self.oOptions.cInstrPerFile;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if iInstrTestEnd > len(g_aoInstructionTests):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync iInstrTestEnd = len(g_aoInstructionTests);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Generate the instruction tests.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for iInstrTest in range(iInstrTestStart, iInstrTestEnd):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oInstrTest = g_aoInstructionTests[iInstrTest];

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '; %s\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ';\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % (oInstrTest.sName,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oInstrTest.generateTest(self, self._calcTestFunctionName(oInstrTest, iInstrTest));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Generate the main function.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'VBINSTST_BEGINPROC TestInstrMain\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' MY_PUSH_ALL\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' sub xSP, 40h\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync '\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for iInstrTest in range(iInstrTestStart, iInstrTestEnd):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oInstrTest = g_aoInstructionTests[iInstrTest];

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync if oInstrTest.isApplicable(self):

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync self.write('%%ifdef ASM_CALL64_GCC\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' lea rdi, [.szInstr%03u wrt rip]\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync '%%elifdef ASM_CALL64_MSC\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' lea rcx, [.szInstr%03u wrt rip]\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync '%%else\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' mov xAX, .szInstr%03u\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' mov [xSP], xAX\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync '%%endif\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' VBINSTST_CALL_FN_SUB_TEST\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync ' call VBINSTST_NAME(%s)\n'

51ef69064b4ea4d571ed129ab883b0c08967c901vboxsync % ( iInstrTest, iInstrTest, iInstrTest, self._calcTestFunctionName(oInstrTest, iInstrTest)));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' add xSP, 40h\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' MY_POP_ALL\n'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync ' ret\n\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync for iInstrTest in range(iInstrTestStart, iInstrTestEnd):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('.szInstr%03u: db \'%s\', 0\n' % (iInstrTest, g_aoInstructionTests[iInstrTest].sName,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.write('VBINSTST_ENDPROC TestInstrMain\n\n');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self._generateFileFooter();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.oOptions.sOutputBase != '-':

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oFile.close();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.oFile = None;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync self.sFile = '';

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return RTEXITCODE_SUCCESS;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def _runMakefileMode(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.cFiles == 1:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync print('%s.asm' % (self.oOptions.sOutputBase,));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync else:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync print(' '.join('%s-%s.asm' % (self.oOptions.sOutputBase, i) for i in range(self.cFiles)));

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return RTEXITCODE_SUCCESS;

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def run(self):

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if self.oOptions.fMakefileMode:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self._runMakefileMode();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return self._generateTests();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync @staticmethod

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync def main():

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync """

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Parse the command line.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser = OptionParser(version = __version__[11:-1].strip());

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser.add_option('--makefile-mode', dest = 'fMakefileMode', action = 'store_true', default = False,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync help = 'Special mode for use to output a list of output files for the benefit of '

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync 'the make program (kmk).');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser.add_option('--split', dest = 'cInstrPerFile', metavar = '<instr-per-file>', type = 'int', default = 9999999,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync help = 'Number of instruction to test per output file.');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser.add_option('--output-base', dest = 'sOutputBase', metavar = '<file>', default = None,

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync help = 'The output file base name, no suffix please. Required.');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser.add_option('--target', dest = 'sTargetEnv', metavar = '<target>',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync default = 'iprt-r3-32',

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync choices = g_dTargetEnvs.keys(),

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync help = 'The target environment. Choices: %s'

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync % (', '.join(sorted(g_dTargetEnvs.keys())),));

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync oParser.add_option('--test-size', dest = 'sTestSize', default = InstructionTestGen.ksTestSize_Medium,

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync choices = InstructionTestGen.kasTestSizes,

0ab80a5f847b8c908ee2d9db5cc37da8e8dd5697vboxsync help = 'Selects the test size.');

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync (oOptions, asArgs) = oParser.parse_args();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if len(asArgs) > 0:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oParser.print_help();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return RTEXITCODE_SYNTAX

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync if oOptions.sOutputBase is None:

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync print('syntax error: Missing required option --output-base.', file = sys.stderr);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return RTEXITCODE_SYNTAX

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync # Instantiate the program class and run it.

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync #

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync oProgram = InstructionTestGen(oOptions);

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync return oProgram.run();

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsyncif __name__ == '__main__':

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync sys.exit(InstructionTestGen.main());

917f4ee9f101c9786cf09ea0fe7923a7f6dfe40cvboxsync