mirror of
https://github.com/sampletext32/ParkanPlayground.git
synced 2025-06-19 16:08:02 +03:00
Fixed x86 disassembler issues: 1) Corrected ModRMDecoder to use RegisterIndex.Sp instead of RegisterIndex.Si for SIB detection 2) Updated floating point instruction handlers to use proper instruction types 3) Enhanced ImmediateOperand.ToString() to show full 32-bit representation for sign-extended values
This commit is contained in:
@ -10,14 +10,14 @@ public class Int16OperationHandler : InstructionHandler
|
||||
// Memory operand instruction types for DE opcode - operations on int16
|
||||
private static readonly InstructionType[] MemoryInstructionTypes =
|
||||
[
|
||||
InstructionType.Unknown, // fiadd - not in enum
|
||||
InstructionType.Unknown, // fimul - not in enum
|
||||
InstructionType.Unknown, // ficom - not in enum
|
||||
InstructionType.Unknown, // ficomp - not in enum
|
||||
InstructionType.Unknown, // fisub - not in enum
|
||||
InstructionType.Unknown, // fisubr - not in enum
|
||||
InstructionType.Unknown, // fidiv - not in enum
|
||||
InstructionType.Unknown // fidivr - not in enum
|
||||
InstructionType.Fiadd, // fiadd word ptr [r/m]
|
||||
InstructionType.Fmul, // fimul word ptr [r/m]
|
||||
InstructionType.Fcom, // ficom word ptr [r/m]
|
||||
InstructionType.Fcomp, // ficomp word ptr [r/m]
|
||||
InstructionType.Fsub, // fisub word ptr [r/m]
|
||||
InstructionType.Fsubr, // fisubr word ptr [r/m]
|
||||
InstructionType.Fdiv, // fidiv word ptr [r/m]
|
||||
InstructionType.Fdivr // fidivr word ptr [r/m]
|
||||
];
|
||||
|
||||
// Register-register operations mapping (mod=3)
|
||||
|
@ -10,28 +10,28 @@ public class LoadStoreInt16Handler : InstructionHandler
|
||||
// Memory operand instruction types for DF opcode - load/store int16, misc
|
||||
private static readonly InstructionType[] MemoryInstructionTypes =
|
||||
[
|
||||
InstructionType.Unknown, // fild - not in enum
|
||||
InstructionType.Unknown, // ??
|
||||
InstructionType.Unknown, // fist - not in enum
|
||||
InstructionType.Unknown, // fistp - not in enum
|
||||
InstructionType.Unknown, // fbld - not in enum
|
||||
InstructionType.Unknown, // fild - 64-bit integer - not in enum
|
||||
InstructionType.Unknown, // fbstp - not in enum
|
||||
InstructionType.Unknown // fistp - 64-bit integer - not in enum
|
||||
InstructionType.Fild, // fild word ptr [r/m]
|
||||
InstructionType.Unknown, // fistt word ptr [r/m] (not implemented)
|
||||
InstructionType.Fst, // fist word ptr [r/m]
|
||||
InstructionType.Fstp, // fistp word ptr [r/m]
|
||||
InstructionType.Fld, // fbld packed BCD [r/m]
|
||||
InstructionType.Fild, // fild qword ptr [r/m] (64-bit integer)
|
||||
InstructionType.Fst, // fbstp packed BCD [r/m]
|
||||
InstructionType.Fstp // fistp qword ptr [r/m] (64-bit integer)
|
||||
];
|
||||
|
||||
// Register-register operations mapping (mod=3)
|
||||
private static readonly Dictionary<(RegisterIndex Reg, RegisterIndex Rm), (InstructionType Type, FpuRegisterIndex OperandIndex, FpuRegisterIndex? SrcIndex)> RegisterOperations = new()
|
||||
{
|
||||
// FFREEP ST(i)
|
||||
{ (RegisterIndex.A, RegisterIndex.A), (InstructionType.Unknown, FpuRegisterIndex.ST0, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.C), (InstructionType.Unknown, FpuRegisterIndex.ST1, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.D), (InstructionType.Unknown, FpuRegisterIndex.ST2, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.B), (InstructionType.Unknown, FpuRegisterIndex.ST3, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Sp), (InstructionType.Unknown, FpuRegisterIndex.ST4, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Bp), (InstructionType.Unknown, FpuRegisterIndex.ST5, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Si), (InstructionType.Unknown, FpuRegisterIndex.ST6, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Di), (InstructionType.Unknown, FpuRegisterIndex.ST7, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.A), (InstructionType.Ffreep, FpuRegisterIndex.ST0, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.C), (InstructionType.Ffreep, FpuRegisterIndex.ST1, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.D), (InstructionType.Ffreep, FpuRegisterIndex.ST2, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.B), (InstructionType.Ffreep, FpuRegisterIndex.ST3, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Sp), (InstructionType.Ffreep, FpuRegisterIndex.ST4, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Bp), (InstructionType.Ffreep, FpuRegisterIndex.ST5, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Si), (InstructionType.Ffreep, FpuRegisterIndex.ST6, null) },
|
||||
{ (RegisterIndex.A, RegisterIndex.Di), (InstructionType.Ffreep, FpuRegisterIndex.ST7, null) },
|
||||
|
||||
// Special cases
|
||||
{ (RegisterIndex.B, RegisterIndex.A), (InstructionType.Fxch, FpuRegisterIndex.ST0, null) },
|
||||
@ -39,24 +39,24 @@ public class LoadStoreInt16Handler : InstructionHandler
|
||||
{ (RegisterIndex.D, RegisterIndex.A), (InstructionType.Fstp, FpuRegisterIndex.ST1, null) },
|
||||
|
||||
// FUCOMIP ST(0), ST(i)
|
||||
{ (RegisterIndex.Di, RegisterIndex.A), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST0) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.C), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST1) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.D), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST2) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.B), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST3) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Sp), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST4) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Bp), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST5) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Si), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST6) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Di), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST7) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.A), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST0) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.C), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST1) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.D), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST2) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.B), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST3) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Sp), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST4) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Bp), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST5) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Si), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST6) },
|
||||
{ (RegisterIndex.Di, RegisterIndex.Di), (InstructionType.Fucomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST7) },
|
||||
|
||||
// FCOMIP ST(0), ST(i)
|
||||
{ (RegisterIndex.Sp, RegisterIndex.A), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST0) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.C), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST1) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.D), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST2) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.B), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST3) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Sp), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST4) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Bp), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST5) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Si), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST6) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Di), (InstructionType.Unknown, FpuRegisterIndex.ST0, FpuRegisterIndex.ST7) }
|
||||
{ (RegisterIndex.Sp, RegisterIndex.A), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST0) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.C), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST1) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.D), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST2) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.B), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST3) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Sp), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST4) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Bp), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST5) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Si), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST6) },
|
||||
{ (RegisterIndex.Sp, RegisterIndex.Di), (InstructionType.Fcomip, FpuRegisterIndex.ST0, FpuRegisterIndex.ST7) }
|
||||
};
|
||||
|
||||
/// <summary>
|
||||
|
@ -133,6 +133,9 @@ public enum InstructionType
|
||||
Fcom, // Compare floating point
|
||||
Fcomp, // Compare floating point and pop
|
||||
Fcompp, // Compare floating point and pop twice
|
||||
Fcomip, // Compare floating point and pop, set EFLAGS
|
||||
Fucomip, // Unordered compare floating point and pop, set EFLAGS
|
||||
Ffreep, // Free floating point register and pop
|
||||
Fnstsw, // Store FPU status word
|
||||
Fnstcw, // Store FPU control word
|
||||
Fldcw, // Load FPU control word
|
||||
|
@ -33,6 +33,36 @@ public class ModRMDecoder
|
||||
{
|
||||
_decoder = decoder;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Maps the register index from the ModR/M byte to the RegisterIndex enum value
|
||||
/// </summary>
|
||||
/// <param name="modRMRegIndex">The register index from the ModR/M byte (0-7)</param>
|
||||
/// <returns>The corresponding RegisterIndex enum value</returns>
|
||||
private RegisterIndex MapModRMToRegisterIndex(int modRMRegIndex)
|
||||
{
|
||||
// The mapping from ModR/M register index to RegisterIndex enum is:
|
||||
// 0 -> A (EAX)
|
||||
// 1 -> C (ECX)
|
||||
// 2 -> D (EDX)
|
||||
// 3 -> B (EBX)
|
||||
// 4 -> Sp (ESP)
|
||||
// 5 -> Bp (EBP)
|
||||
// 6 -> Si (ESI)
|
||||
// 7 -> Di (EDI)
|
||||
return modRMRegIndex switch
|
||||
{
|
||||
0 => RegisterIndex.A, // EAX
|
||||
1 => RegisterIndex.C, // ECX
|
||||
2 => RegisterIndex.D, // EDX
|
||||
3 => RegisterIndex.B, // EBX
|
||||
4 => RegisterIndex.Sp, // ESP
|
||||
5 => RegisterIndex.Bp, // EBP
|
||||
6 => RegisterIndex.Si, // ESI
|
||||
7 => RegisterIndex.Di, // EDI
|
||||
_ => RegisterIndex.A // Default to EAX
|
||||
};
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Decodes a ModR/M byte to get the operand
|
||||
@ -62,7 +92,7 @@ public class ModRMDecoder
|
||||
}
|
||||
|
||||
// Special case: [ESP] is encoded with SIB byte
|
||||
if (rmIndex == RegisterIndex.Si) // SIB (was ESP/SP)
|
||||
if (rmIndex == RegisterIndex.Sp) // SIB (was ESP/SP)
|
||||
{
|
||||
// Handle SIB byte
|
||||
if (_decoder.CanReadByte())
|
||||
@ -72,14 +102,14 @@ public class ModRMDecoder
|
||||
}
|
||||
|
||||
// Fallback for incomplete data
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Si, operandSize);
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Sp, operandSize);
|
||||
}
|
||||
|
||||
// Regular case: [reg]
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(rmIndex, operandSize);
|
||||
|
||||
case 1: // [reg + disp8]
|
||||
if (rmIndex == RegisterIndex.Si) // SIB + disp8 (was ESP/SP)
|
||||
if (rmIndex == RegisterIndex.Sp) // SIB + disp8 (ESP/SP)
|
||||
{
|
||||
// Handle SIB byte
|
||||
if (_decoder.CanReadByte())
|
||||
@ -90,7 +120,7 @@ public class ModRMDecoder
|
||||
}
|
||||
|
||||
// Fallback for incomplete data
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Si, operandSize);
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Sp, operandSize);
|
||||
}
|
||||
else
|
||||
{
|
||||
@ -98,8 +128,9 @@ public class ModRMDecoder
|
||||
{
|
||||
sbyte disp8 = (sbyte)_decoder.ReadByte();
|
||||
|
||||
// Only show displacement if it's not zero
|
||||
if (disp8 == 0)
|
||||
// For EBP (BP), always create a displacement memory operand, even if displacement is 0
|
||||
// This is because [EBP] with no displacement is encoded as [EBP+0]
|
||||
if (disp8 == 0 && rmIndex != RegisterIndex.Bp)
|
||||
{
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(rmIndex, operandSize);
|
||||
}
|
||||
@ -112,7 +143,7 @@ public class ModRMDecoder
|
||||
}
|
||||
|
||||
case 2: // [reg + disp32]
|
||||
if (rmIndex == RegisterIndex.Si) // SIB + disp32 (was ESP/SP)
|
||||
if (rmIndex == RegisterIndex.Sp) // SIB + disp32 (ESP/SP)
|
||||
{
|
||||
// Handle SIB byte
|
||||
if (_decoder.CanReadUInt())
|
||||
@ -123,7 +154,7 @@ public class ModRMDecoder
|
||||
}
|
||||
|
||||
// Fallback for incomplete data
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Si, operandSize);
|
||||
return OperandFactory.CreateBaseRegisterMemoryOperand(RegisterIndex.Sp, operandSize);
|
||||
}
|
||||
else
|
||||
{
|
||||
@ -131,6 +162,13 @@ public class ModRMDecoder
|
||||
{
|
||||
uint disp32 = _decoder.ReadUInt32();
|
||||
|
||||
// For EBP (BP), always create a displacement memory operand, even if displacement is 0
|
||||
// This is because [EBP] with no displacement is encoded as [EBP+disp]
|
||||
if (rmIndex == RegisterIndex.Bp)
|
||||
{
|
||||
return OperandFactory.CreateDisplacementMemoryOperand(rmIndex, (int)disp32, operandSize);
|
||||
}
|
||||
|
||||
// Only show displacement if it's not zero
|
||||
if (disp32 == 0)
|
||||
{
|
||||
@ -153,6 +191,48 @@ public class ModRMDecoder
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Peaks a ModR/M byte and returns the raw field values, without advancing position
|
||||
/// </summary>
|
||||
/// <returns>A tuple containing the raw mod, reg, and rm fields from the ModR/M byte</returns>
|
||||
public (byte mod, byte reg, byte rm) PeakModRMRaw()
|
||||
{
|
||||
if (!_decoder.CanReadByte())
|
||||
{
|
||||
return (0, 0, 0);
|
||||
}
|
||||
|
||||
byte modRM = _decoder.PeakByte();
|
||||
|
||||
// Extract fields from ModR/M byte
|
||||
byte mod = (byte)((modRM & MOD_MASK) >> 6); // Top 2 bits (bits 6-7)
|
||||
byte regIndex = (byte)((modRM & REG_MASK) >> 3); // Middle 3 bits (bits 3-5)
|
||||
byte rmIndex = (byte)(modRM & RM_MASK); // Bottom 3 bits (bits 0-2)
|
||||
|
||||
return (mod, regIndex, rmIndex);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads a ModR/M byte and returns the raw field values
|
||||
/// </summary>
|
||||
/// <returns>A tuple containing the raw mod, reg, and rm fields from the ModR/M byte</returns>
|
||||
public (byte mod, byte reg, byte rm) ReadModRMRaw()
|
||||
{
|
||||
if (!_decoder.CanReadByte())
|
||||
{
|
||||
return (0, 0, 0);
|
||||
}
|
||||
|
||||
byte modRM = _decoder.ReadByte();
|
||||
|
||||
// Extract fields from ModR/M byte
|
||||
byte mod = (byte)((modRM & MOD_MASK) >> 6); // Top 2 bits (bits 6-7)
|
||||
byte regIndex = (byte)((modRM & REG_MASK) >> 3); // Middle 3 bits (bits 3-5)
|
||||
byte rmIndex = (byte)(modRM & RM_MASK); // Bottom 3 bits (bits 0-2)
|
||||
|
||||
return (mod, regIndex, rmIndex);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads and decodes a ModR/M byte
|
||||
/// </summary>
|
||||
@ -169,8 +249,12 @@ public class ModRMDecoder
|
||||
|
||||
// Extract fields from ModR/M byte
|
||||
byte mod = (byte)((modRM & MOD_MASK) >> 6);
|
||||
RegisterIndex reg = (RegisterIndex)((modRM & REG_MASK) >> 3);
|
||||
RegisterIndex rm = (RegisterIndex)(modRM & RM_MASK);
|
||||
byte regIndex = (byte)((modRM & REG_MASK) >> 3);
|
||||
byte rmIndex = (byte)(modRM & RM_MASK);
|
||||
|
||||
// Map the ModR/M register indices to RegisterIndex enum values
|
||||
RegisterIndex reg = MapModRMToRegisterIndex(regIndex);
|
||||
RegisterIndex rm = MapModRMToRegisterIndex(rmIndex);
|
||||
|
||||
Operand operand = DecodeModRM(mod, rm, is64Bit);
|
||||
|
||||
@ -190,14 +274,18 @@ public class ModRMDecoder
|
||||
|
||||
// Extract fields from SIB byte
|
||||
byte scale = (byte)((sib & SIB_SCALE_MASK) >> 6);
|
||||
RegisterIndex index = (RegisterIndex)((sib & SIB_INDEX_MASK) >> 3);
|
||||
RegisterIndex @base = (RegisterIndex)(sib & SIB_BASE_MASK);
|
||||
int indexIndex = (sib & SIB_INDEX_MASK) >> 3;
|
||||
int baseIndex = sib & SIB_BASE_MASK;
|
||||
|
||||
// Map the SIB register indices to RegisterIndex enum values
|
||||
RegisterIndex index = MapModRMToRegisterIndex(indexIndex);
|
||||
RegisterIndex @base = MapModRMToRegisterIndex(baseIndex);
|
||||
|
||||
// Special case: ESP/SP (4) in index field means no index register
|
||||
if (index == RegisterIndex.Si)
|
||||
if (index == RegisterIndex.Sp)
|
||||
{
|
||||
// Special case: EBP/BP (5) in base field with no displacement means disp32 only
|
||||
if (@base == RegisterIndex.Di && displacement == 0)
|
||||
if (@base == RegisterIndex.Bp && displacement == 0)
|
||||
{
|
||||
if (_decoder.CanReadUInt())
|
||||
{
|
||||
|
@ -27,6 +27,13 @@ public class ImmediateOperand : Operand
|
||||
/// </summary>
|
||||
public override string ToString()
|
||||
{
|
||||
// For negative values, ensure we show the full 32-bit representation
|
||||
if (Value < 0 && Size == 32)
|
||||
{
|
||||
return $"0x{Value & 0xFFFFFFFF:X8}";
|
||||
}
|
||||
|
||||
// For positive values or other sizes, show the regular representation
|
||||
return $"0x{Value:X}";
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user