Assembler.cs 70 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439
  1. using ARMeilleure.IntermediateRepresentation;
  2. using ARMeilleure.Translation.PTC;
  3. using System;
  4. using System.Diagnostics;
  5. using System.IO;
  6. namespace ARMeilleure.CodeGen.X86
  7. {
  8. class Assembler
  9. {
  10. private const int BadOp = 0;
  11. private const int OpModRMBits = 24;
  12. private const byte RexPrefix = 0x40;
  13. private const byte RexWPrefix = 0x48;
  14. private const byte LockPrefix = 0xf0;
  15. private const int MaxRegNumber = 15;
  16. [Flags]
  17. private enum InstructionFlags
  18. {
  19. None = 0,
  20. RegOnly = 1 << 0,
  21. Reg8Src = 1 << 1,
  22. Reg8Dest = 1 << 2,
  23. RexW = 1 << 3,
  24. Vex = 1 << 4,
  25. PrefixBit = 16,
  26. PrefixMask = 7 << PrefixBit,
  27. Prefix66 = 1 << PrefixBit,
  28. PrefixF3 = 2 << PrefixBit,
  29. PrefixF2 = 4 << PrefixBit
  30. }
  31. private struct InstructionInfo
  32. {
  33. public int OpRMR { get; }
  34. public int OpRMImm8 { get; }
  35. public int OpRMImm32 { get; }
  36. public int OpRImm64 { get; }
  37. public int OpRRM { get; }
  38. public InstructionFlags Flags { get; }
  39. public InstructionInfo(
  40. int opRMR,
  41. int opRMImm8,
  42. int opRMImm32,
  43. int opRImm64,
  44. int opRRM,
  45. InstructionFlags flags)
  46. {
  47. OpRMR = opRMR;
  48. OpRMImm8 = opRMImm8;
  49. OpRMImm32 = opRMImm32;
  50. OpRImm64 = opRImm64;
  51. OpRRM = opRRM;
  52. Flags = flags;
  53. }
  54. }
  55. private static InstructionInfo[] _instTable;
  56. private Stream _stream;
  57. private PtcInfo _ptcInfo;
  58. private bool _ptcDisabled;
  59. static Assembler()
  60. {
  61. _instTable = new InstructionInfo[(int)X86Instruction.Count];
  62. // Name RM/R RM/I8 RM/I32 R/I64 R/RM Flags
  63. Add(X86Instruction.Add, new InstructionInfo(0x00000001, 0x00000083, 0x00000081, BadOp, 0x00000003, InstructionFlags.None));
  64. Add(X86Instruction.Addpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f58, InstructionFlags.Vex | InstructionFlags.Prefix66));
  65. Add(X86Instruction.Addps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f58, InstructionFlags.Vex));
  66. Add(X86Instruction.Addsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f58, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  67. Add(X86Instruction.Addss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f58, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  68. Add(X86Instruction.Aesdec, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38de, InstructionFlags.Vex | InstructionFlags.Prefix66));
  69. Add(X86Instruction.Aesdeclast, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38df, InstructionFlags.Vex | InstructionFlags.Prefix66));
  70. Add(X86Instruction.Aesenc, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38dc, InstructionFlags.Vex | InstructionFlags.Prefix66));
  71. Add(X86Instruction.Aesenclast, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38dd, InstructionFlags.Vex | InstructionFlags.Prefix66));
  72. Add(X86Instruction.Aesimc, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38db, InstructionFlags.Vex | InstructionFlags.Prefix66));
  73. Add(X86Instruction.And, new InstructionInfo(0x00000021, 0x04000083, 0x04000081, BadOp, 0x00000023, InstructionFlags.None));
  74. Add(X86Instruction.Andnpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f55, InstructionFlags.Vex | InstructionFlags.Prefix66));
  75. Add(X86Instruction.Andnps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f55, InstructionFlags.Vex));
  76. Add(X86Instruction.Andpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f54, InstructionFlags.Vex | InstructionFlags.Prefix66));
  77. Add(X86Instruction.Andps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f54, InstructionFlags.Vex));
  78. Add(X86Instruction.Blendvpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3815, InstructionFlags.Prefix66));
  79. Add(X86Instruction.Blendvps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3814, InstructionFlags.Prefix66));
  80. Add(X86Instruction.Bsr, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fbd, InstructionFlags.None));
  81. Add(X86Instruction.Bswap, new InstructionInfo(0x00000fc8, BadOp, BadOp, BadOp, BadOp, InstructionFlags.RegOnly));
  82. Add(X86Instruction.Call, new InstructionInfo(0x020000ff, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  83. Add(X86Instruction.Cmovcc, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f40, InstructionFlags.None));
  84. Add(X86Instruction.Cmp, new InstructionInfo(0x00000039, 0x07000083, 0x07000081, BadOp, 0x0000003b, InstructionFlags.None));
  85. Add(X86Instruction.Cmppd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc2, InstructionFlags.Vex | InstructionFlags.Prefix66));
  86. Add(X86Instruction.Cmpps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc2, InstructionFlags.Vex));
  87. Add(X86Instruction.Cmpsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc2, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  88. Add(X86Instruction.Cmpss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc2, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  89. Add(X86Instruction.Cmpxchg, new InstructionInfo(0x00000fb1, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  90. Add(X86Instruction.Cmpxchg16b, new InstructionInfo(0x01000fc7, BadOp, BadOp, BadOp, BadOp, InstructionFlags.RexW));
  91. Add(X86Instruction.Comisd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2f, InstructionFlags.Vex | InstructionFlags.Prefix66));
  92. Add(X86Instruction.Comiss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2f, InstructionFlags.Vex));
  93. Add(X86Instruction.Cpuid, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fa2, InstructionFlags.RegOnly));
  94. Add(X86Instruction.Crc32, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38f1, InstructionFlags.PrefixF2));
  95. Add(X86Instruction.Crc32_16, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38f1, InstructionFlags.PrefixF2 | InstructionFlags.Prefix66));
  96. Add(X86Instruction.Crc32_8, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f38f0, InstructionFlags.PrefixF2 | InstructionFlags.Reg8Src));
  97. Add(X86Instruction.Cvtdq2pd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fe6, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  98. Add(X86Instruction.Cvtdq2ps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5b, InstructionFlags.Vex));
  99. Add(X86Instruction.Cvtpd2dq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fe6, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  100. Add(X86Instruction.Cvtpd2ps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5a, InstructionFlags.Vex | InstructionFlags.Prefix66));
  101. Add(X86Instruction.Cvtps2dq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5b, InstructionFlags.Vex | InstructionFlags.Prefix66));
  102. Add(X86Instruction.Cvtps2pd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5a, InstructionFlags.Vex));
  103. Add(X86Instruction.Cvtsd2si, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2d, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  104. Add(X86Instruction.Cvtsd2ss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5a, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  105. Add(X86Instruction.Cvtsi2sd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2a, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  106. Add(X86Instruction.Cvtsi2ss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2a, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  107. Add(X86Instruction.Cvtss2sd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5a, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  108. Add(X86Instruction.Cvtss2si, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f2d, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  109. Add(X86Instruction.Div, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x060000f7, InstructionFlags.None));
  110. Add(X86Instruction.Divpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5e, InstructionFlags.Vex | InstructionFlags.Prefix66));
  111. Add(X86Instruction.Divps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5e, InstructionFlags.Vex));
  112. Add(X86Instruction.Divsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5e, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  113. Add(X86Instruction.Divss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5e, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  114. Add(X86Instruction.Haddpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f7c, InstructionFlags.Vex | InstructionFlags.Prefix66));
  115. Add(X86Instruction.Haddps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f7c, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  116. Add(X86Instruction.Idiv, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x070000f7, InstructionFlags.None));
  117. Add(X86Instruction.Imul, new InstructionInfo(BadOp, 0x0000006b, 0x00000069, BadOp, 0x00000faf, InstructionFlags.None));
  118. Add(X86Instruction.Imul128, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x050000f7, InstructionFlags.None));
  119. Add(X86Instruction.Insertps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a21, InstructionFlags.Vex | InstructionFlags.Prefix66));
  120. Add(X86Instruction.Jmp, new InstructionInfo(0x040000ff, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  121. Add(X86Instruction.Lea, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x0000008d, InstructionFlags.None));
  122. Add(X86Instruction.Maxpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5f, InstructionFlags.Vex | InstructionFlags.Prefix66));
  123. Add(X86Instruction.Maxps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5f, InstructionFlags.Vex));
  124. Add(X86Instruction.Maxsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5f, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  125. Add(X86Instruction.Maxss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5f, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  126. Add(X86Instruction.Minpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5d, InstructionFlags.Vex | InstructionFlags.Prefix66));
  127. Add(X86Instruction.Minps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5d, InstructionFlags.Vex));
  128. Add(X86Instruction.Minsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5d, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  129. Add(X86Instruction.Minss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5d, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  130. Add(X86Instruction.Mov, new InstructionInfo(0x00000089, BadOp, 0x000000c7, 0x000000b8, 0x0000008b, InstructionFlags.None));
  131. Add(X86Instruction.Mov16, new InstructionInfo(0x00000089, BadOp, 0x000000c7, BadOp, 0x0000008b, InstructionFlags.Prefix66));
  132. Add(X86Instruction.Mov8, new InstructionInfo(0x00000088, 0x000000c6, BadOp, BadOp, 0x0000008a, InstructionFlags.Reg8Src | InstructionFlags.Reg8Dest));
  133. Add(X86Instruction.Movd, new InstructionInfo(0x00000f7e, BadOp, BadOp, BadOp, 0x00000f6e, InstructionFlags.Vex | InstructionFlags.Prefix66));
  134. Add(X86Instruction.Movdqu, new InstructionInfo(0x00000f7f, BadOp, BadOp, BadOp, 0x00000f6f, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  135. Add(X86Instruction.Movhlps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f12, InstructionFlags.Vex));
  136. Add(X86Instruction.Movlhps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f16, InstructionFlags.Vex));
  137. Add(X86Instruction.Movq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f7e, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  138. Add(X86Instruction.Movsd, new InstructionInfo(0x00000f11, BadOp, BadOp, BadOp, 0x00000f10, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  139. Add(X86Instruction.Movss, new InstructionInfo(0x00000f11, BadOp, BadOp, BadOp, 0x00000f10, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  140. Add(X86Instruction.Movsx16, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fbf, InstructionFlags.None));
  141. Add(X86Instruction.Movsx32, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000063, InstructionFlags.None));
  142. Add(X86Instruction.Movsx8, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fbe, InstructionFlags.Reg8Src));
  143. Add(X86Instruction.Movzx16, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fb7, InstructionFlags.None));
  144. Add(X86Instruction.Movzx8, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fb6, InstructionFlags.Reg8Src));
  145. Add(X86Instruction.Mul128, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x040000f7, InstructionFlags.None));
  146. Add(X86Instruction.Mulpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f59, InstructionFlags.Vex | InstructionFlags.Prefix66));
  147. Add(X86Instruction.Mulps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f59, InstructionFlags.Vex));
  148. Add(X86Instruction.Mulsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f59, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  149. Add(X86Instruction.Mulss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f59, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  150. Add(X86Instruction.Neg, new InstructionInfo(0x030000f7, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  151. Add(X86Instruction.Not, new InstructionInfo(0x020000f7, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  152. Add(X86Instruction.Or, new InstructionInfo(0x00000009, 0x01000083, 0x01000081, BadOp, 0x0000000b, InstructionFlags.None));
  153. Add(X86Instruction.Paddb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ffc, InstructionFlags.Vex | InstructionFlags.Prefix66));
  154. Add(X86Instruction.Paddd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ffe, InstructionFlags.Vex | InstructionFlags.Prefix66));
  155. Add(X86Instruction.Paddq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fd4, InstructionFlags.Vex | InstructionFlags.Prefix66));
  156. Add(X86Instruction.Paddw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ffd, InstructionFlags.Vex | InstructionFlags.Prefix66));
  157. Add(X86Instruction.Pand, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fdb, InstructionFlags.Vex | InstructionFlags.Prefix66));
  158. Add(X86Instruction.Pandn, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fdf, InstructionFlags.Vex | InstructionFlags.Prefix66));
  159. Add(X86Instruction.Pavgb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fe0, InstructionFlags.Vex | InstructionFlags.Prefix66));
  160. Add(X86Instruction.Pavgw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fe3, InstructionFlags.Vex | InstructionFlags.Prefix66));
  161. Add(X86Instruction.Pblendvb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3810, InstructionFlags.Prefix66));
  162. Add(X86Instruction.Pclmulqdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a44, InstructionFlags.Vex | InstructionFlags.Prefix66));
  163. Add(X86Instruction.Pcmpeqb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f74, InstructionFlags.Vex | InstructionFlags.Prefix66));
  164. Add(X86Instruction.Pcmpeqd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f76, InstructionFlags.Vex | InstructionFlags.Prefix66));
  165. Add(X86Instruction.Pcmpeqq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3829, InstructionFlags.Vex | InstructionFlags.Prefix66));
  166. Add(X86Instruction.Pcmpeqw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f75, InstructionFlags.Vex | InstructionFlags.Prefix66));
  167. Add(X86Instruction.Pcmpgtb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f64, InstructionFlags.Vex | InstructionFlags.Prefix66));
  168. Add(X86Instruction.Pcmpgtd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f66, InstructionFlags.Vex | InstructionFlags.Prefix66));
  169. Add(X86Instruction.Pcmpgtq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3837, InstructionFlags.Vex | InstructionFlags.Prefix66));
  170. Add(X86Instruction.Pcmpgtw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f65, InstructionFlags.Vex | InstructionFlags.Prefix66));
  171. Add(X86Instruction.Pextrb, new InstructionInfo(0x000f3a14, BadOp, BadOp, BadOp, BadOp, InstructionFlags.Vex | InstructionFlags.Prefix66));
  172. Add(X86Instruction.Pextrd, new InstructionInfo(0x000f3a16, BadOp, BadOp, BadOp, BadOp, InstructionFlags.Vex | InstructionFlags.Prefix66));
  173. Add(X86Instruction.Pextrq, new InstructionInfo(0x000f3a16, BadOp, BadOp, BadOp, BadOp, InstructionFlags.Vex | InstructionFlags.RexW | InstructionFlags.Prefix66));
  174. Add(X86Instruction.Pextrw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc5, InstructionFlags.Vex | InstructionFlags.Prefix66));
  175. Add(X86Instruction.Pinsrb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a20, InstructionFlags.Vex | InstructionFlags.Prefix66));
  176. Add(X86Instruction.Pinsrd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a22, InstructionFlags.Vex | InstructionFlags.Prefix66));
  177. Add(X86Instruction.Pinsrq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a22, InstructionFlags.Vex | InstructionFlags.RexW | InstructionFlags.Prefix66));
  178. Add(X86Instruction.Pinsrw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc4, InstructionFlags.Vex | InstructionFlags.Prefix66));
  179. Add(X86Instruction.Pmaxsb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383c, InstructionFlags.Vex | InstructionFlags.Prefix66));
  180. Add(X86Instruction.Pmaxsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383d, InstructionFlags.Vex | InstructionFlags.Prefix66));
  181. Add(X86Instruction.Pmaxsw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fee, InstructionFlags.Vex | InstructionFlags.Prefix66));
  182. Add(X86Instruction.Pmaxub, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fde, InstructionFlags.Vex | InstructionFlags.Prefix66));
  183. Add(X86Instruction.Pmaxud, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383f, InstructionFlags.Vex | InstructionFlags.Prefix66));
  184. Add(X86Instruction.Pmaxuw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383e, InstructionFlags.Vex | InstructionFlags.Prefix66));
  185. Add(X86Instruction.Pminsb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3838, InstructionFlags.Vex | InstructionFlags.Prefix66));
  186. Add(X86Instruction.Pminsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3839, InstructionFlags.Vex | InstructionFlags.Prefix66));
  187. Add(X86Instruction.Pminsw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fea, InstructionFlags.Vex | InstructionFlags.Prefix66));
  188. Add(X86Instruction.Pminub, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fda, InstructionFlags.Vex | InstructionFlags.Prefix66));
  189. Add(X86Instruction.Pminud, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383b, InstructionFlags.Vex | InstructionFlags.Prefix66));
  190. Add(X86Instruction.Pminuw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f383a, InstructionFlags.Vex | InstructionFlags.Prefix66));
  191. Add(X86Instruction.Pmovsxbw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3820, InstructionFlags.Vex | InstructionFlags.Prefix66));
  192. Add(X86Instruction.Pmovsxdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3825, InstructionFlags.Vex | InstructionFlags.Prefix66));
  193. Add(X86Instruction.Pmovsxwd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3823, InstructionFlags.Vex | InstructionFlags.Prefix66));
  194. Add(X86Instruction.Pmovzxbw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3830, InstructionFlags.Vex | InstructionFlags.Prefix66));
  195. Add(X86Instruction.Pmovzxdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3835, InstructionFlags.Vex | InstructionFlags.Prefix66));
  196. Add(X86Instruction.Pmovzxwd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3833, InstructionFlags.Vex | InstructionFlags.Prefix66));
  197. Add(X86Instruction.Pmulld, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3840, InstructionFlags.Vex | InstructionFlags.Prefix66));
  198. Add(X86Instruction.Pmullw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fd5, InstructionFlags.Vex | InstructionFlags.Prefix66));
  199. Add(X86Instruction.Pop, new InstructionInfo(0x0000008f, BadOp, BadOp, BadOp, BadOp, InstructionFlags.None));
  200. Add(X86Instruction.Popcnt, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fb8, InstructionFlags.PrefixF3));
  201. Add(X86Instruction.Por, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000feb, InstructionFlags.Vex | InstructionFlags.Prefix66));
  202. Add(X86Instruction.Pshufb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3800, InstructionFlags.Vex | InstructionFlags.Prefix66));
  203. Add(X86Instruction.Pshufd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f70, InstructionFlags.Vex | InstructionFlags.Prefix66));
  204. Add(X86Instruction.Pslld, new InstructionInfo(BadOp, 0x06000f72, BadOp, BadOp, 0x00000ff2, InstructionFlags.Vex | InstructionFlags.Prefix66));
  205. Add(X86Instruction.Pslldq, new InstructionInfo(BadOp, 0x07000f73, BadOp, BadOp, BadOp, InstructionFlags.Vex | InstructionFlags.Prefix66));
  206. Add(X86Instruction.Psllq, new InstructionInfo(BadOp, 0x06000f73, BadOp, BadOp, 0x00000ff3, InstructionFlags.Vex | InstructionFlags.Prefix66));
  207. Add(X86Instruction.Psllw, new InstructionInfo(BadOp, 0x06000f71, BadOp, BadOp, 0x00000ff1, InstructionFlags.Vex | InstructionFlags.Prefix66));
  208. Add(X86Instruction.Psrad, new InstructionInfo(BadOp, 0x04000f72, BadOp, BadOp, 0x00000fe2, InstructionFlags.Vex | InstructionFlags.Prefix66));
  209. Add(X86Instruction.Psraw, new InstructionInfo(BadOp, 0x04000f71, BadOp, BadOp, 0x00000fe1, InstructionFlags.Vex | InstructionFlags.Prefix66));
  210. Add(X86Instruction.Psrld, new InstructionInfo(BadOp, 0x02000f72, BadOp, BadOp, 0x00000fd2, InstructionFlags.Vex | InstructionFlags.Prefix66));
  211. Add(X86Instruction.Psrlq, new InstructionInfo(BadOp, 0x02000f73, BadOp, BadOp, 0x00000fd3, InstructionFlags.Vex | InstructionFlags.Prefix66));
  212. Add(X86Instruction.Psrldq, new InstructionInfo(BadOp, 0x03000f73, BadOp, BadOp, BadOp, InstructionFlags.Vex | InstructionFlags.Prefix66));
  213. Add(X86Instruction.Psrlw, new InstructionInfo(BadOp, 0x02000f71, BadOp, BadOp, 0x00000fd1, InstructionFlags.Vex | InstructionFlags.Prefix66));
  214. Add(X86Instruction.Psubb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ff8, InstructionFlags.Vex | InstructionFlags.Prefix66));
  215. Add(X86Instruction.Psubd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ffa, InstructionFlags.Vex | InstructionFlags.Prefix66));
  216. Add(X86Instruction.Psubq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ffb, InstructionFlags.Vex | InstructionFlags.Prefix66));
  217. Add(X86Instruction.Psubw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000ff9, InstructionFlags.Vex | InstructionFlags.Prefix66));
  218. Add(X86Instruction.Punpckhbw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f68, InstructionFlags.Vex | InstructionFlags.Prefix66));
  219. Add(X86Instruction.Punpckhdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f6a, InstructionFlags.Vex | InstructionFlags.Prefix66));
  220. Add(X86Instruction.Punpckhqdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f6d, InstructionFlags.Vex | InstructionFlags.Prefix66));
  221. Add(X86Instruction.Punpckhwd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f69, InstructionFlags.Vex | InstructionFlags.Prefix66));
  222. Add(X86Instruction.Punpcklbw, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f60, InstructionFlags.Vex | InstructionFlags.Prefix66));
  223. Add(X86Instruction.Punpckldq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f62, InstructionFlags.Vex | InstructionFlags.Prefix66));
  224. Add(X86Instruction.Punpcklqdq, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f6c, InstructionFlags.Vex | InstructionFlags.Prefix66));
  225. Add(X86Instruction.Punpcklwd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f61, InstructionFlags.Vex | InstructionFlags.Prefix66));
  226. Add(X86Instruction.Push, new InstructionInfo(BadOp, 0x0000006a, 0x00000068, BadOp, 0x060000ff, InstructionFlags.None));
  227. Add(X86Instruction.Pxor, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fef, InstructionFlags.Vex | InstructionFlags.Prefix66));
  228. Add(X86Instruction.Rcpps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f53, InstructionFlags.Vex));
  229. Add(X86Instruction.Rcpss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f53, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  230. Add(X86Instruction.Ror, new InstructionInfo(0x010000d3, 0x010000c1, BadOp, BadOp, BadOp, InstructionFlags.None));
  231. Add(X86Instruction.Roundpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a09, InstructionFlags.Vex | InstructionFlags.Prefix66));
  232. Add(X86Instruction.Roundps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a08, InstructionFlags.Vex | InstructionFlags.Prefix66));
  233. Add(X86Instruction.Roundsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a0b, InstructionFlags.Vex | InstructionFlags.Prefix66));
  234. Add(X86Instruction.Roundss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a0a, InstructionFlags.Vex | InstructionFlags.Prefix66));
  235. Add(X86Instruction.Rsqrtps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f52, InstructionFlags.Vex));
  236. Add(X86Instruction.Rsqrtss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f52, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  237. Add(X86Instruction.Sar, new InstructionInfo(0x070000d3, 0x070000c1, BadOp, BadOp, BadOp, InstructionFlags.None));
  238. Add(X86Instruction.Setcc, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f90, InstructionFlags.Reg8Dest));
  239. Add(X86Instruction.Shl, new InstructionInfo(0x040000d3, 0x040000c1, BadOp, BadOp, BadOp, InstructionFlags.None));
  240. Add(X86Instruction.Shr, new InstructionInfo(0x050000d3, 0x050000c1, BadOp, BadOp, BadOp, InstructionFlags.None));
  241. Add(X86Instruction.Shufpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc6, InstructionFlags.Vex | InstructionFlags.Prefix66));
  242. Add(X86Instruction.Shufps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000fc6, InstructionFlags.Vex));
  243. Add(X86Instruction.Sqrtpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f51, InstructionFlags.Vex | InstructionFlags.Prefix66));
  244. Add(X86Instruction.Sqrtps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f51, InstructionFlags.Vex));
  245. Add(X86Instruction.Sqrtsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f51, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  246. Add(X86Instruction.Sqrtss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f51, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  247. Add(X86Instruction.Sub, new InstructionInfo(0x00000029, 0x05000083, 0x05000081, BadOp, 0x0000002b, InstructionFlags.None));
  248. Add(X86Instruction.Subpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5c, InstructionFlags.Vex | InstructionFlags.Prefix66));
  249. Add(X86Instruction.Subps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5c, InstructionFlags.Vex));
  250. Add(X86Instruction.Subsd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5c, InstructionFlags.Vex | InstructionFlags.PrefixF2));
  251. Add(X86Instruction.Subss, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f5c, InstructionFlags.Vex | InstructionFlags.PrefixF3));
  252. Add(X86Instruction.Test, new InstructionInfo(0x00000085, BadOp, 0x000000f7, BadOp, BadOp, InstructionFlags.None));
  253. Add(X86Instruction.Unpckhpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f15, InstructionFlags.Vex | InstructionFlags.Prefix66));
  254. Add(X86Instruction.Unpckhps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f15, InstructionFlags.Vex));
  255. Add(X86Instruction.Unpcklpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f14, InstructionFlags.Vex | InstructionFlags.Prefix66));
  256. Add(X86Instruction.Unpcklps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f14, InstructionFlags.Vex));
  257. Add(X86Instruction.Vblendvpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a4b, InstructionFlags.Vex | InstructionFlags.Prefix66));
  258. Add(X86Instruction.Vblendvps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a4a, InstructionFlags.Vex | InstructionFlags.Prefix66));
  259. Add(X86Instruction.Vpblendvb, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x000f3a4c, InstructionFlags.Vex | InstructionFlags.Prefix66));
  260. Add(X86Instruction.Xor, new InstructionInfo(0x00000031, 0x06000083, 0x06000081, BadOp, 0x00000033, InstructionFlags.None));
  261. Add(X86Instruction.Xorpd, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f57, InstructionFlags.Vex | InstructionFlags.Prefix66));
  262. Add(X86Instruction.Xorps, new InstructionInfo(BadOp, BadOp, BadOp, BadOp, 0x00000f57, InstructionFlags.Vex));
  263. }
  264. private static void Add(X86Instruction inst, InstructionInfo info)
  265. {
  266. _instTable[(int)inst] = info;
  267. }
  268. public Assembler(Stream stream, PtcInfo ptcInfo = null)
  269. {
  270. _stream = stream;
  271. _ptcInfo = ptcInfo;
  272. _ptcDisabled = ptcInfo == null;
  273. }
  274. public void Add(Operand dest, Operand source, OperandType type)
  275. {
  276. WriteInstruction(dest, source, type, X86Instruction.Add);
  277. }
  278. public void Addsd(Operand dest, Operand src1, Operand src2)
  279. {
  280. WriteInstruction(dest, src1, src2, X86Instruction.Addsd);
  281. }
  282. public void Addss(Operand dest, Operand src1, Operand src2)
  283. {
  284. WriteInstruction(dest, src1, src2, X86Instruction.Addss);
  285. }
  286. public void And(Operand dest, Operand source, OperandType type)
  287. {
  288. WriteInstruction(dest, source, type, X86Instruction.And);
  289. }
  290. public void Bsr(Operand dest, Operand source, OperandType type)
  291. {
  292. WriteInstruction(dest, source, type, X86Instruction.Bsr);
  293. }
  294. public void Bswap(Operand dest)
  295. {
  296. WriteInstruction(dest, null, dest.Type, X86Instruction.Bswap);
  297. }
  298. public void Call(Operand dest)
  299. {
  300. WriteInstruction(dest, null, OperandType.None, X86Instruction.Call);
  301. }
  302. public void Cdq()
  303. {
  304. WriteByte(0x99);
  305. }
  306. public void Cmovcc(Operand dest, Operand source, OperandType type, X86Condition condition)
  307. {
  308. InstructionInfo info = _instTable[(int)X86Instruction.Cmovcc];
  309. WriteOpCode(dest, null, source, type, info.Flags, info.OpRRM | (int)condition, rrm: true);
  310. }
  311. public void Cmp(Operand src1, Operand src2, OperandType type)
  312. {
  313. WriteInstruction(src1, src2, type, X86Instruction.Cmp);
  314. }
  315. public void Cqo()
  316. {
  317. WriteByte(0x48);
  318. WriteByte(0x99);
  319. }
  320. public void Cmpxchg(MemoryOperand memOp, Operand src)
  321. {
  322. WriteByte(LockPrefix);
  323. WriteInstruction(memOp, src, src.Type, X86Instruction.Cmpxchg);
  324. }
  325. public void Cmpxchg16b(MemoryOperand memOp)
  326. {
  327. WriteByte(LockPrefix);
  328. WriteInstruction(memOp, null, OperandType.None, X86Instruction.Cmpxchg16b);
  329. }
  330. public void Comisd(Operand src1, Operand src2)
  331. {
  332. WriteInstruction(src1, null, src2, X86Instruction.Comisd);
  333. }
  334. public void Comiss(Operand src1, Operand src2)
  335. {
  336. WriteInstruction(src1, null, src2, X86Instruction.Comiss);
  337. }
  338. public void Cpuid()
  339. {
  340. WriteInstruction(null, null, OperandType.None, X86Instruction.Cpuid);
  341. }
  342. public void Cvtsd2ss(Operand dest, Operand src1, Operand src2)
  343. {
  344. WriteInstruction(dest, src1, src2, X86Instruction.Cvtsd2ss);
  345. }
  346. public void Cvtsi2sd(Operand dest, Operand src1, Operand src2, OperandType type)
  347. {
  348. WriteInstruction(dest, src1, src2, X86Instruction.Cvtsi2sd, type);
  349. }
  350. public void Cvtsi2ss(Operand dest, Operand src1, Operand src2, OperandType type)
  351. {
  352. WriteInstruction(dest, src1, src2, X86Instruction.Cvtsi2ss, type);
  353. }
  354. public void Cvtss2sd(Operand dest, Operand src1, Operand src2)
  355. {
  356. WriteInstruction(dest, src1, src2, X86Instruction.Cvtss2sd);
  357. }
  358. public void Div(Operand source)
  359. {
  360. WriteInstruction(null, source, source.Type, X86Instruction.Div);
  361. }
  362. public void Divsd(Operand dest, Operand src1, Operand src2)
  363. {
  364. WriteInstruction(dest, src1, src2, X86Instruction.Divsd);
  365. }
  366. public void Divss(Operand dest, Operand src1, Operand src2)
  367. {
  368. WriteInstruction(dest, src1, src2, X86Instruction.Divss);
  369. }
  370. public void Idiv(Operand source)
  371. {
  372. WriteInstruction(null, source, source.Type, X86Instruction.Idiv);
  373. }
  374. public void Imul(Operand source)
  375. {
  376. WriteInstruction(null, source, source.Type, X86Instruction.Imul128);
  377. }
  378. public void Imul(Operand dest, Operand source, OperandType type)
  379. {
  380. if (source.Kind != OperandKind.Register)
  381. {
  382. throw new ArgumentException($"Invalid source operand kind \"{source.Kind}\".");
  383. }
  384. WriteInstruction(dest, source, type, X86Instruction.Imul);
  385. }
  386. public void Imul(Operand dest, Operand src1, Operand src2, OperandType type)
  387. {
  388. InstructionInfo info = _instTable[(int)X86Instruction.Imul];
  389. if (src2.Kind != OperandKind.Constant)
  390. {
  391. throw new ArgumentException($"Invalid source 2 operand kind \"{src2.Kind}\".");
  392. }
  393. if (IsImm8(src2.Value, src2.Type) && info.OpRMImm8 != BadOp)
  394. {
  395. WriteOpCode(dest, null, src1, type, info.Flags, info.OpRMImm8, rrm: true);
  396. WriteByte(src2.AsByte());
  397. }
  398. else if (IsImm32(src2.Value, src2.Type) && info.OpRMImm32 != BadOp)
  399. {
  400. WriteOpCode(dest, null, src1, type, info.Flags, info.OpRMImm32, rrm: true);
  401. WriteInt32(src2.AsInt32());
  402. }
  403. else
  404. {
  405. throw new ArgumentException($"Failed to encode constant 0x{src2.Value:X}.");
  406. }
  407. }
  408. public void Insertps(Operand dest, Operand src1, Operand src2, byte imm)
  409. {
  410. WriteInstruction(dest, src1, src2, X86Instruction.Insertps);
  411. WriteByte(imm);
  412. }
  413. public void Jcc(X86Condition condition, long offset)
  414. {
  415. if (_ptcDisabled && ConstFitsOnS8(offset))
  416. {
  417. WriteByte((byte)(0x70 | (int)condition));
  418. WriteByte((byte)offset);
  419. }
  420. else if (ConstFitsOnS32(offset))
  421. {
  422. WriteByte(0x0f);
  423. WriteByte((byte)(0x80 | (int)condition));
  424. WriteInt32((int)offset);
  425. }
  426. else
  427. {
  428. throw new ArgumentOutOfRangeException(nameof(offset));
  429. }
  430. }
  431. public void Jmp(long offset)
  432. {
  433. if (_ptcDisabled && ConstFitsOnS8(offset))
  434. {
  435. WriteByte(0xeb);
  436. WriteByte((byte)offset);
  437. }
  438. else if (ConstFitsOnS32(offset))
  439. {
  440. WriteByte(0xe9);
  441. WriteInt32((int)offset);
  442. }
  443. else
  444. {
  445. throw new ArgumentOutOfRangeException(nameof(offset));
  446. }
  447. }
  448. public void Jmp(Operand dest)
  449. {
  450. WriteInstruction(dest, null, OperandType.None, X86Instruction.Jmp);
  451. }
  452. public void Lea(Operand dest, Operand source, OperandType type)
  453. {
  454. WriteInstruction(dest, source, type, X86Instruction.Lea);
  455. }
  456. public void Mov(Operand dest, Operand source, OperandType type)
  457. {
  458. WriteInstruction(dest, source, type, X86Instruction.Mov);
  459. }
  460. public void Mov16(Operand dest, Operand source)
  461. {
  462. WriteInstruction(dest, source, OperandType.None, X86Instruction.Mov16);
  463. }
  464. public void Mov8(Operand dest, Operand source)
  465. {
  466. WriteInstruction(dest, source, OperandType.None, X86Instruction.Mov8);
  467. }
  468. public void Movd(Operand dest, Operand source)
  469. {
  470. InstructionInfo info = _instTable[(int)X86Instruction.Movd];
  471. if (source.Type.IsInteger() || source.Kind == OperandKind.Memory)
  472. {
  473. WriteOpCode(dest, null, source, OperandType.None, info.Flags, info.OpRRM, rrm: true);
  474. }
  475. else
  476. {
  477. WriteOpCode(dest, null, source, OperandType.None, info.Flags, info.OpRMR);
  478. }
  479. }
  480. public void Movdqu(Operand dest, Operand source)
  481. {
  482. WriteInstruction(dest, null, source, X86Instruction.Movdqu);
  483. }
  484. public void Movhlps(Operand dest, Operand src1, Operand src2)
  485. {
  486. WriteInstruction(dest, src1, src2, X86Instruction.Movhlps);
  487. }
  488. public void Movlhps(Operand dest, Operand src1, Operand src2)
  489. {
  490. WriteInstruction(dest, src1, src2, X86Instruction.Movlhps);
  491. }
  492. public void Movq(Operand dest, Operand source)
  493. {
  494. InstructionInfo info = _instTable[(int)X86Instruction.Movd];
  495. InstructionFlags flags = info.Flags | InstructionFlags.RexW;
  496. if (source.Type.IsInteger() || source.Kind == OperandKind.Memory)
  497. {
  498. WriteOpCode(dest, null, source, OperandType.None, flags, info.OpRRM, rrm: true);
  499. }
  500. else if (dest.Type.IsInteger() || dest.Kind == OperandKind.Memory)
  501. {
  502. WriteOpCode(dest, null, source, OperandType.None, flags, info.OpRMR);
  503. }
  504. else
  505. {
  506. WriteInstruction(dest, source, OperandType.None, X86Instruction.Movq);
  507. }
  508. }
  509. public void Movsd(Operand dest, Operand src1, Operand src2)
  510. {
  511. WriteInstruction(dest, src1, src2, X86Instruction.Movsd);
  512. }
  513. public void Movss(Operand dest, Operand src1, Operand src2)
  514. {
  515. WriteInstruction(dest, src1, src2, X86Instruction.Movss);
  516. }
  517. public void Movsx16(Operand dest, Operand source, OperandType type)
  518. {
  519. WriteInstruction(dest, source, type, X86Instruction.Movsx16);
  520. }
  521. public void Movsx32(Operand dest, Operand source, OperandType type)
  522. {
  523. WriteInstruction(dest, source, type, X86Instruction.Movsx32);
  524. }
  525. public void Movsx8(Operand dest, Operand source, OperandType type)
  526. {
  527. WriteInstruction(dest, source, type, X86Instruction.Movsx8);
  528. }
  529. public void Movzx16(Operand dest, Operand source, OperandType type)
  530. {
  531. WriteInstruction(dest, source, type, X86Instruction.Movzx16);
  532. }
  533. public void Movzx8(Operand dest, Operand source, OperandType type)
  534. {
  535. WriteInstruction(dest, source, type, X86Instruction.Movzx8);
  536. }
  537. public void Mul(Operand source)
  538. {
  539. WriteInstruction(null, source, source.Type, X86Instruction.Mul128);
  540. }
  541. public void Mulsd(Operand dest, Operand src1, Operand src2)
  542. {
  543. WriteInstruction(dest, src1, src2, X86Instruction.Mulsd);
  544. }
  545. public void Mulss(Operand dest, Operand src1, Operand src2)
  546. {
  547. WriteInstruction(dest, src1, src2, X86Instruction.Mulss);
  548. }
  549. public void Neg(Operand dest)
  550. {
  551. WriteInstruction(dest, null, dest.Type, X86Instruction.Neg);
  552. }
  553. public void Not(Operand dest)
  554. {
  555. WriteInstruction(dest, null, dest.Type, X86Instruction.Not);
  556. }
  557. public void Or(Operand dest, Operand source, OperandType type)
  558. {
  559. WriteInstruction(dest, source, type, X86Instruction.Or);
  560. }
  561. public void Pclmulqdq(Operand dest, Operand source, byte imm)
  562. {
  563. WriteInstruction(dest, null, source, X86Instruction.Pclmulqdq);
  564. WriteByte(imm);
  565. }
  566. public void Pcmpeqw(Operand dest, Operand src1, Operand src2)
  567. {
  568. WriteInstruction(dest, src1, src2, X86Instruction.Pcmpeqw);
  569. }
  570. public void Pextrb(Operand dest, Operand source, byte imm)
  571. {
  572. WriteInstruction(dest, null, source, X86Instruction.Pextrb);
  573. WriteByte(imm);
  574. }
  575. public void Pextrd(Operand dest, Operand source, byte imm)
  576. {
  577. WriteInstruction(dest, null, source, X86Instruction.Pextrd);
  578. WriteByte(imm);
  579. }
  580. public void Pextrq(Operand dest, Operand source, byte imm)
  581. {
  582. WriteInstruction(dest, null, source, X86Instruction.Pextrq);
  583. WriteByte(imm);
  584. }
  585. public void Pextrw(Operand dest, Operand source, byte imm)
  586. {
  587. WriteInstruction(dest, null, source, X86Instruction.Pextrw);
  588. WriteByte(imm);
  589. }
  590. public void Pinsrb(Operand dest, Operand src1, Operand src2, byte imm)
  591. {
  592. WriteInstruction(dest, src1, src2, X86Instruction.Pinsrb);
  593. WriteByte(imm);
  594. }
  595. public void Pinsrd(Operand dest, Operand src1, Operand src2, byte imm)
  596. {
  597. WriteInstruction(dest, src1, src2, X86Instruction.Pinsrd);
  598. WriteByte(imm);
  599. }
  600. public void Pinsrq(Operand dest, Operand src1, Operand src2, byte imm)
  601. {
  602. WriteInstruction(dest, src1, src2, X86Instruction.Pinsrq);
  603. WriteByte(imm);
  604. }
  605. public void Pinsrw(Operand dest, Operand src1, Operand src2, byte imm)
  606. {
  607. WriteInstruction(dest, src1, src2, X86Instruction.Pinsrw);
  608. WriteByte(imm);
  609. }
  610. public void Pop(Operand dest)
  611. {
  612. if (dest.Kind == OperandKind.Register)
  613. {
  614. WriteCompactInst(dest, 0x58);
  615. }
  616. else
  617. {
  618. WriteInstruction(dest, null, dest.Type, X86Instruction.Pop);
  619. }
  620. }
  621. public void Popcnt(Operand dest, Operand source, OperandType type)
  622. {
  623. WriteInstruction(dest, source, type, X86Instruction.Popcnt);
  624. }
  625. public void Pshufd(Operand dest, Operand source, byte imm)
  626. {
  627. WriteInstruction(dest, null, source, X86Instruction.Pshufd);
  628. WriteByte(imm);
  629. }
  630. public void Push(Operand source)
  631. {
  632. if (source.Kind == OperandKind.Register)
  633. {
  634. WriteCompactInst(source, 0x50);
  635. }
  636. else
  637. {
  638. WriteInstruction(null, source, source.Type, X86Instruction.Push);
  639. }
  640. }
  641. public void Return()
  642. {
  643. WriteByte(0xc3);
  644. }
  645. public void Ror(Operand dest, Operand source, OperandType type)
  646. {
  647. WriteShiftInst(dest, source, type, X86Instruction.Ror);
  648. }
  649. public void Sar(Operand dest, Operand source, OperandType type)
  650. {
  651. WriteShiftInst(dest, source, type, X86Instruction.Sar);
  652. }
  653. public void Shl(Operand dest, Operand source, OperandType type)
  654. {
  655. WriteShiftInst(dest, source, type, X86Instruction.Shl);
  656. }
  657. public void Shr(Operand dest, Operand source, OperandType type)
  658. {
  659. WriteShiftInst(dest, source, type, X86Instruction.Shr);
  660. }
  661. public void Setcc(Operand dest, X86Condition condition)
  662. {
  663. InstructionInfo info = _instTable[(int)X86Instruction.Setcc];
  664. WriteOpCode(dest, null, null, OperandType.None, info.Flags, info.OpRRM | (int)condition);
  665. }
  666. public void Sub(Operand dest, Operand source, OperandType type)
  667. {
  668. WriteInstruction(dest, source, type, X86Instruction.Sub);
  669. }
  670. public void Subsd(Operand dest, Operand src1, Operand src2)
  671. {
  672. WriteInstruction(dest, src1, src2, X86Instruction.Subsd);
  673. }
  674. public void Subss(Operand dest, Operand src1, Operand src2)
  675. {
  676. WriteInstruction(dest, src1, src2, X86Instruction.Subss);
  677. }
  678. public void Test(Operand src1, Operand src2, OperandType type)
  679. {
  680. WriteInstruction(src1, src2, type, X86Instruction.Test);
  681. }
  682. public void Xor(Operand dest, Operand source, OperandType type)
  683. {
  684. WriteInstruction(dest, source, type, X86Instruction.Xor);
  685. }
  686. public void Xorps(Operand dest, Operand src1, Operand src2)
  687. {
  688. WriteInstruction(dest, src1, src2, X86Instruction.Xorps);
  689. }
  690. public void WriteInstruction(
  691. X86Instruction inst,
  692. Operand dest,
  693. Operand source,
  694. OperandType type = OperandType.None)
  695. {
  696. WriteInstruction(dest, null, source, inst, type);
  697. }
  698. public void WriteInstruction(X86Instruction inst, Operand dest, Operand src1, Operand src2)
  699. {
  700. if (src2.Kind == OperandKind.Constant)
  701. {
  702. WriteInstruction(src1, dest, src2, inst);
  703. }
  704. else
  705. {
  706. WriteInstruction(dest, src1, src2, inst);
  707. }
  708. }
  709. public void WriteInstruction(
  710. X86Instruction inst,
  711. Operand dest,
  712. Operand src1,
  713. Operand src2,
  714. OperandType type)
  715. {
  716. WriteInstruction(dest, src1, src2, inst, type);
  717. }
  718. public void WriteInstruction(X86Instruction inst, Operand dest, Operand source, byte imm)
  719. {
  720. WriteInstruction(dest, null, source, inst);
  721. WriteByte(imm);
  722. }
  723. public void WriteInstruction(
  724. X86Instruction inst,
  725. Operand dest,
  726. Operand src1,
  727. Operand src2,
  728. Operand src3)
  729. {
  730. // 3+ operands can only be encoded with the VEX encoding scheme.
  731. Debug.Assert(HardwareCapabilities.SupportsVexEncoding);
  732. WriteInstruction(dest, src1, src2, inst);
  733. WriteByte((byte)(src3.AsByte() << 4));
  734. }
  735. public void WriteInstruction(
  736. X86Instruction inst,
  737. Operand dest,
  738. Operand src1,
  739. Operand src2,
  740. byte imm)
  741. {
  742. WriteInstruction(dest, src1, src2, inst);
  743. WriteByte(imm);
  744. }
  745. private void WriteShiftInst(Operand dest, Operand source, OperandType type, X86Instruction inst)
  746. {
  747. if (source.Kind == OperandKind.Register)
  748. {
  749. X86Register shiftReg = (X86Register)source.GetRegister().Index;
  750. Debug.Assert(shiftReg == X86Register.Rcx, $"Invalid shift register \"{shiftReg}\".");
  751. source = null;
  752. }
  753. else if (source.Kind == OperandKind.Constant)
  754. {
  755. source = source.With((uint)source.Value & (dest.Type == OperandType.I32 ? 0x1f : 0x3f));
  756. }
  757. WriteInstruction(dest, source, type, inst);
  758. }
  759. private void WriteInstruction(Operand dest, Operand source, OperandType type, X86Instruction inst)
  760. {
  761. InstructionInfo info = _instTable[(int)inst];
  762. if (source != null)
  763. {
  764. if (source.Kind == OperandKind.Constant)
  765. {
  766. ulong imm = source.Value;
  767. if (inst == X86Instruction.Mov8)
  768. {
  769. WriteOpCode(dest, null, null, type, info.Flags, info.OpRMImm8);
  770. WriteByte((byte)imm);
  771. }
  772. else if (inst == X86Instruction.Mov16)
  773. {
  774. WriteOpCode(dest, null, null, type, info.Flags, info.OpRMImm32);
  775. WriteInt16((short)imm);
  776. }
  777. else if (IsImm8(imm, type) && info.OpRMImm8 != BadOp)
  778. {
  779. WriteOpCode(dest, null, null, type, info.Flags, info.OpRMImm8);
  780. WriteByte((byte)imm);
  781. }
  782. else if (!source.Relocatable && IsImm32(imm, type) && info.OpRMImm32 != BadOp)
  783. {
  784. WriteOpCode(dest, null, null, type, info.Flags, info.OpRMImm32);
  785. WriteInt32((int)imm);
  786. }
  787. else if (dest != null && dest.Kind == OperandKind.Register && info.OpRImm64 != BadOp)
  788. {
  789. int? index = source.PtcIndex;
  790. int rexPrefix = GetRexPrefix(dest, source, type, rrm: false);
  791. if (rexPrefix != 0)
  792. {
  793. WriteByte((byte)rexPrefix);
  794. }
  795. WriteByte((byte)(info.OpRImm64 + (dest.GetRegister().Index & 0b111)));
  796. if (_ptcInfo != null && index != null)
  797. {
  798. _ptcInfo.WriteRelocEntry(new RelocEntry((int)_stream.Position, (int)index));
  799. }
  800. WriteUInt64(imm);
  801. }
  802. else
  803. {
  804. throw new ArgumentException($"Failed to encode constant 0x{imm:X}.");
  805. }
  806. }
  807. else if (source.Kind == OperandKind.Register && info.OpRMR != BadOp)
  808. {
  809. WriteOpCode(dest, null, source, type, info.Flags, info.OpRMR);
  810. }
  811. else if (info.OpRRM != BadOp)
  812. {
  813. WriteOpCode(dest, null, source, type, info.Flags, info.OpRRM, rrm: true);
  814. }
  815. else
  816. {
  817. throw new ArgumentException($"Invalid source operand kind \"{source.Kind}\".");
  818. }
  819. }
  820. else if (info.OpRRM != BadOp)
  821. {
  822. WriteOpCode(dest, null, source, type, info.Flags, info.OpRRM, rrm: true);
  823. }
  824. else if (info.OpRMR != BadOp)
  825. {
  826. WriteOpCode(dest, null, source, type, info.Flags, info.OpRMR);
  827. }
  828. else
  829. {
  830. throw new ArgumentNullException(nameof(source));
  831. }
  832. }
  833. private void WriteInstruction(
  834. Operand dest,
  835. Operand src1,
  836. Operand src2,
  837. X86Instruction inst,
  838. OperandType type = OperandType.None)
  839. {
  840. InstructionInfo info = _instTable[(int)inst];
  841. if (src2 != null)
  842. {
  843. if (src2.Kind == OperandKind.Constant)
  844. {
  845. ulong imm = src2.Value;
  846. if ((byte)imm == imm && info.OpRMImm8 != BadOp)
  847. {
  848. WriteOpCode(dest, src1, null, type, info.Flags, info.OpRMImm8);
  849. WriteByte((byte)imm);
  850. }
  851. else
  852. {
  853. throw new ArgumentException($"Failed to encode constant 0x{imm:X}.");
  854. }
  855. }
  856. else if (src2.Kind == OperandKind.Register && info.OpRMR != BadOp)
  857. {
  858. WriteOpCode(dest, src1, src2, type, info.Flags, info.OpRMR);
  859. }
  860. else if (info.OpRRM != BadOp)
  861. {
  862. WriteOpCode(dest, src1, src2, type, info.Flags, info.OpRRM, rrm: true);
  863. }
  864. else
  865. {
  866. throw new ArgumentException($"Invalid source operand kind \"{src2.Kind}\".");
  867. }
  868. }
  869. else if (info.OpRRM != BadOp)
  870. {
  871. WriteOpCode(dest, src1, src2, type, info.Flags, info.OpRRM, rrm: true);
  872. }
  873. else if (info.OpRMR != BadOp)
  874. {
  875. WriteOpCode(dest, src1, src2, type, info.Flags, info.OpRMR);
  876. }
  877. else
  878. {
  879. throw new ArgumentNullException(nameof(src2));
  880. }
  881. }
  882. private void WriteOpCode(
  883. Operand dest,
  884. Operand src1,
  885. Operand src2,
  886. OperandType type,
  887. InstructionFlags flags,
  888. int opCode,
  889. bool rrm = false)
  890. {
  891. int rexPrefix = GetRexPrefix(dest, src2, type, rrm);
  892. if ((flags & InstructionFlags.RexW) != 0)
  893. {
  894. rexPrefix |= RexWPrefix;
  895. }
  896. int modRM = (opCode >> OpModRMBits) << 3;
  897. MemoryOperand memOp = null;
  898. if (dest != null)
  899. {
  900. if (dest.Kind == OperandKind.Register)
  901. {
  902. int regIndex = dest.GetRegister().Index;
  903. modRM |= (regIndex & 0b111) << (rrm ? 3 : 0);
  904. if ((flags & InstructionFlags.Reg8Dest) != 0 && regIndex >= 4)
  905. {
  906. rexPrefix |= RexPrefix;
  907. }
  908. }
  909. else if (dest.Kind == OperandKind.Memory)
  910. {
  911. memOp = dest as MemoryOperand;
  912. }
  913. else
  914. {
  915. throw new ArgumentException("Invalid destination operand kind \"" + dest.Kind + "\".");
  916. }
  917. }
  918. if (src2 != null)
  919. {
  920. if (src2.Kind == OperandKind.Register)
  921. {
  922. int regIndex = src2.GetRegister().Index;
  923. modRM |= (regIndex & 0b111) << (rrm ? 0 : 3);
  924. if ((flags & InstructionFlags.Reg8Src) != 0 && regIndex >= 4)
  925. {
  926. rexPrefix |= RexPrefix;
  927. }
  928. }
  929. else if (src2.Kind == OperandKind.Memory && memOp == null)
  930. {
  931. memOp = src2 as MemoryOperand;
  932. }
  933. else
  934. {
  935. throw new ArgumentException("Invalid source operand kind \"" + src2.Kind + "\".");
  936. }
  937. }
  938. bool needsSibByte = false;
  939. bool needsDisplacement = false;
  940. int sib = 0;
  941. if (memOp != null)
  942. {
  943. // Either source or destination is a memory operand.
  944. Register baseReg = memOp.BaseAddress.GetRegister();
  945. X86Register baseRegLow = (X86Register)(baseReg.Index & 0b111);
  946. needsSibByte = memOp.Index != null || baseRegLow == X86Register.Rsp;
  947. needsDisplacement = memOp.Displacement != 0 || baseRegLow == X86Register.Rbp;
  948. if (needsDisplacement)
  949. {
  950. if (ConstFitsOnS8(memOp.Displacement))
  951. {
  952. modRM |= 0x40;
  953. }
  954. else /* if (ConstFitsOnS32(memOp.Displacement)) */
  955. {
  956. modRM |= 0x80;
  957. }
  958. }
  959. if (baseReg.Index >= 8)
  960. {
  961. Debug.Assert((uint)baseReg.Index <= MaxRegNumber);
  962. rexPrefix |= RexPrefix | (baseReg.Index >> 3);
  963. }
  964. if (needsSibByte)
  965. {
  966. sib = (int)baseRegLow;
  967. if (memOp.Index != null)
  968. {
  969. int indexReg = memOp.Index.GetRegister().Index;
  970. Debug.Assert(indexReg != (int)X86Register.Rsp, "Using RSP as index register on the memory operand is not allowed.");
  971. if (indexReg >= 8)
  972. {
  973. Debug.Assert((uint)indexReg <= MaxRegNumber);
  974. rexPrefix |= RexPrefix | (indexReg >> 3) << 1;
  975. }
  976. sib |= (indexReg & 0b111) << 3;
  977. }
  978. else
  979. {
  980. sib |= 0b100 << 3;
  981. }
  982. sib |= (int)memOp.Scale << 6;
  983. modRM |= 0b100;
  984. }
  985. else
  986. {
  987. modRM |= (int)baseRegLow;
  988. }
  989. }
  990. else
  991. {
  992. // Source and destination are registers.
  993. modRM |= 0xc0;
  994. }
  995. Debug.Assert(opCode != BadOp, "Invalid opcode value.");
  996. if ((flags & InstructionFlags.Vex) != 0 && HardwareCapabilities.SupportsVexEncoding)
  997. {
  998. // In a vex encoding, only one prefix can be active at a time. The active prefix is encoded in the second byte using two bits.
  999. int vexByte2 = (flags & InstructionFlags.PrefixMask) switch
  1000. {
  1001. InstructionFlags.Prefix66 => 1,
  1002. InstructionFlags.PrefixF3 => 2,
  1003. InstructionFlags.PrefixF2 => 3,
  1004. _ => 0
  1005. };
  1006. if (src1 != null)
  1007. {
  1008. vexByte2 |= (src1.GetRegister().Index ^ 0xf) << 3;
  1009. }
  1010. else
  1011. {
  1012. vexByte2 |= 0b1111 << 3;
  1013. }
  1014. ushort opCodeHigh = (ushort)(opCode >> 8);
  1015. if ((rexPrefix & 0b1011) == 0 && opCodeHigh == 0xf)
  1016. {
  1017. // Two-byte form.
  1018. WriteByte(0xc5);
  1019. vexByte2 |= (~rexPrefix & 4) << 5;
  1020. WriteByte((byte)vexByte2);
  1021. }
  1022. else
  1023. {
  1024. // Three-byte form.
  1025. WriteByte(0xc4);
  1026. int vexByte1 = (~rexPrefix & 7) << 5;
  1027. switch (opCodeHigh)
  1028. {
  1029. case 0xf: vexByte1 |= 1; break;
  1030. case 0xf38: vexByte1 |= 2; break;
  1031. case 0xf3a: vexByte1 |= 3; break;
  1032. default: Debug.Assert(false, $"Failed to VEX encode opcode 0x{opCode:X}."); break;
  1033. }
  1034. vexByte2 |= (rexPrefix & 8) << 4;
  1035. WriteByte((byte)vexByte1);
  1036. WriteByte((byte)vexByte2);
  1037. }
  1038. opCode &= 0xff;
  1039. }
  1040. else
  1041. {
  1042. if (flags.HasFlag(InstructionFlags.Prefix66))
  1043. {
  1044. WriteByte(0x66);
  1045. }
  1046. if (flags.HasFlag(InstructionFlags.PrefixF2))
  1047. {
  1048. WriteByte(0xf2);
  1049. }
  1050. if (flags.HasFlag(InstructionFlags.PrefixF3))
  1051. {
  1052. WriteByte(0xf3);
  1053. }
  1054. if (rexPrefix != 0)
  1055. {
  1056. WriteByte((byte)rexPrefix);
  1057. }
  1058. }
  1059. if (dest != null && (flags & InstructionFlags.RegOnly) != 0)
  1060. {
  1061. opCode += dest.GetRegister().Index & 7;
  1062. }
  1063. if ((opCode & 0xff0000) != 0)
  1064. {
  1065. WriteByte((byte)(opCode >> 16));
  1066. }
  1067. if ((opCode & 0xff00) != 0)
  1068. {
  1069. WriteByte((byte)(opCode >> 8));
  1070. }
  1071. WriteByte((byte)opCode);
  1072. if ((flags & InstructionFlags.RegOnly) == 0)
  1073. {
  1074. WriteByte((byte)modRM);
  1075. if (needsSibByte)
  1076. {
  1077. WriteByte((byte)sib);
  1078. }
  1079. if (needsDisplacement)
  1080. {
  1081. if (ConstFitsOnS8(memOp.Displacement))
  1082. {
  1083. WriteByte((byte)memOp.Displacement);
  1084. }
  1085. else /* if (ConstFitsOnS32(memOp.Displacement)) */
  1086. {
  1087. WriteInt32(memOp.Displacement);
  1088. }
  1089. }
  1090. }
  1091. }
  1092. private void WriteCompactInst(Operand operand, int opCode)
  1093. {
  1094. int regIndex = operand.GetRegister().Index;
  1095. if (regIndex >= 8)
  1096. {
  1097. WriteByte(0x41);
  1098. }
  1099. WriteByte((byte)(opCode + (regIndex & 0b111)));
  1100. }
  1101. private static int GetRexPrefix(Operand dest, Operand source, OperandType type, bool rrm)
  1102. {
  1103. int rexPrefix = 0;
  1104. if (Is64Bits(type))
  1105. {
  1106. rexPrefix = RexWPrefix;
  1107. }
  1108. void SetRegisterHighBit(Register reg, int bit)
  1109. {
  1110. if (reg.Index >= 8)
  1111. {
  1112. rexPrefix |= RexPrefix | (reg.Index >> 3) << bit;
  1113. }
  1114. }
  1115. if (dest != null && dest.Kind == OperandKind.Register)
  1116. {
  1117. SetRegisterHighBit(dest.GetRegister(), rrm ? 2 : 0);
  1118. }
  1119. if (source != null && source.Kind == OperandKind.Register)
  1120. {
  1121. SetRegisterHighBit(source.GetRegister(), rrm ? 0 : 2);
  1122. }
  1123. return rexPrefix;
  1124. }
  1125. private static bool Is64Bits(OperandType type)
  1126. {
  1127. return type == OperandType.I64 || type == OperandType.FP64;
  1128. }
  1129. private static bool IsImm8(ulong immediate, OperandType type)
  1130. {
  1131. long value = type == OperandType.I32 ? (int)immediate : (long)immediate;
  1132. return ConstFitsOnS8(value);
  1133. }
  1134. private static bool IsImm32(ulong immediate, OperandType type)
  1135. {
  1136. long value = type == OperandType.I32 ? (int)immediate : (long)immediate;
  1137. return ConstFitsOnS32(value);
  1138. }
  1139. public static int GetJccLength(long offset, bool ptcDisabled = true)
  1140. {
  1141. if (ptcDisabled && ConstFitsOnS8(offset < 0 ? offset - 2 : offset))
  1142. {
  1143. return 2;
  1144. }
  1145. else if (ConstFitsOnS32(offset < 0 ? offset - 6 : offset))
  1146. {
  1147. return 6;
  1148. }
  1149. else
  1150. {
  1151. throw new ArgumentOutOfRangeException(nameof(offset));
  1152. }
  1153. }
  1154. public static int GetJmpLength(long offset, bool ptcDisabled = true)
  1155. {
  1156. if (ptcDisabled && ConstFitsOnS8(offset < 0 ? offset - 2 : offset))
  1157. {
  1158. return 2;
  1159. }
  1160. else if (ConstFitsOnS32(offset < 0 ? offset - 5 : offset))
  1161. {
  1162. return 5;
  1163. }
  1164. else
  1165. {
  1166. throw new ArgumentOutOfRangeException(nameof(offset));
  1167. }
  1168. }
  1169. private static bool ConstFitsOnS8(long value)
  1170. {
  1171. return value == (sbyte)value;
  1172. }
  1173. private static bool ConstFitsOnS32(long value)
  1174. {
  1175. return value == (int)value;
  1176. }
  1177. private void WriteInt16(short value)
  1178. {
  1179. WriteUInt16((ushort)value);
  1180. }
  1181. private void WriteInt32(int value)
  1182. {
  1183. WriteUInt32((uint)value);
  1184. }
  1185. private void WriteByte(byte value)
  1186. {
  1187. _stream.WriteByte(value);
  1188. }
  1189. private void WriteUInt16(ushort value)
  1190. {
  1191. _stream.WriteByte((byte)(value >> 0));
  1192. _stream.WriteByte((byte)(value >> 8));
  1193. }
  1194. private void WriteUInt32(uint value)
  1195. {
  1196. _stream.WriteByte((byte)(value >> 0));
  1197. _stream.WriteByte((byte)(value >> 8));
  1198. _stream.WriteByte((byte)(value >> 16));
  1199. _stream.WriteByte((byte)(value >> 24));
  1200. }
  1201. private void WriteUInt64(ulong value)
  1202. {
  1203. _stream.WriteByte((byte)(value >> 0));
  1204. _stream.WriteByte((byte)(value >> 8));
  1205. _stream.WriteByte((byte)(value >> 16));
  1206. _stream.WriteByte((byte)(value >> 24));
  1207. _stream.WriteByte((byte)(value >> 32));
  1208. _stream.WriteByte((byte)(value >> 40));
  1209. _stream.WriteByte((byte)(value >> 48));
  1210. _stream.WriteByte((byte)(value >> 56));
  1211. }
  1212. }
  1213. }