iced/Iced/Intel/FormatterUtils.cs

312 lines
8.2 KiB
C#
Raw Normal View History

2018-09-05 23:38:53 +00:00
/*
2019-01-01 11:10:41 +00:00
Copyright (C) 2018-2019 de4dot@gmail.com
2018-09-05 23:38:53 +00:00
This file is part of Iced.
Iced is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Iced is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with Iced. If not, see <https://www.gnu.org/licenses/>.
*/
#if (!NO_GAS_FORMATTER || !NO_INTEL_FORMATTER || !NO_MASM_FORMATTER || !NO_NASM_FORMATTER) && !NO_FORMATTER
using System;
2018-09-29 11:58:14 +00:00
using System.Diagnostics;
2018-09-05 23:38:53 +00:00
namespace Iced.Intel {
enum FormatterFlowControl {
AlwaysShortBranch,
ShortBranch,
NearBranch,
NearCall,
FarBranch,
FarCall,
Xbegin,
}
static class FormatterUtils {
static readonly string[] spaceStrings = new string[] {
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
" ",
};
static readonly string[] tabStrings = new string[] {
"\t",
"\t\t",
"\t\t\t",
"\t\t\t\t",
"\t\t\t\t\t",
"\t\t\t\t\t\t",
};
public static void AddTabs(FormatterOutput output, int column, int firstOperandCharIndex, int tabSize) {
#if DEBUG
for (int i = 0; i < spaceStrings.Length; i++)
System.Diagnostics.Debug.Assert(spaceStrings[i].Length == i + 1);
for (int i = 0; i < tabStrings.Length; i++)
System.Diagnostics.Debug.Assert(tabStrings[i].Length == i + 1);
#endif
const int max_firstOperandCharIndex = 256;
if (firstOperandCharIndex < 0)
firstOperandCharIndex = 0;
else if (firstOperandCharIndex > max_firstOperandCharIndex)
firstOperandCharIndex = max_firstOperandCharIndex;
if (tabSize <= 0) {
int charsLeft = firstOperandCharIndex - column;
if (charsLeft <= 0)
charsLeft = 1;
AddStrings(output, spaceStrings, charsLeft);
}
else {
int endCol = firstOperandCharIndex;
if (endCol <= column)
endCol = column + 1;
int endColRoundDown = endCol / tabSize * tabSize;
bool addedTabs = endColRoundDown > column;
if (addedTabs) {
int tabs = (endColRoundDown - (column / tabSize * tabSize)) / tabSize;
AddStrings(output, tabStrings, tabs);
column = endColRoundDown;
}
int spaces = firstOperandCharIndex - column;
if (spaces > 0)
AddStrings(output, spaceStrings, spaces);
else if (!addedTabs)
AddStrings(output, spaceStrings, 1);
}
}
static void AddStrings(FormatterOutput output, string[] strings, int count) {
while (count > 0) {
int n = count;
if (n >= strings.Length)
n = strings.Length;
output.Write(strings[n - 1], FormatterOutputTextKind.Text);
count -= n;
}
}
public static bool IsCall(FormatterFlowControl kind) => kind == FormatterFlowControl.NearCall || kind == FormatterFlowControl.FarCall;
public static FormatterFlowControl GetFlowControl(ref Instruction instruction) {
switch (instruction.Code) {
2018-09-17 17:49:26 +00:00
case Code.Jo_rel8_16:
case Code.Jo_rel8_32:
case Code.Jo_rel8_64:
case Code.Jno_rel8_16:
case Code.Jno_rel8_32:
case Code.Jno_rel8_64:
case Code.Jb_rel8_16:
case Code.Jb_rel8_32:
case Code.Jb_rel8_64:
case Code.Jae_rel8_16:
case Code.Jae_rel8_32:
case Code.Jae_rel8_64:
case Code.Je_rel8_16:
case Code.Je_rel8_32:
case Code.Je_rel8_64:
case Code.Jne_rel8_16:
case Code.Jne_rel8_32:
case Code.Jne_rel8_64:
case Code.Jbe_rel8_16:
case Code.Jbe_rel8_32:
case Code.Jbe_rel8_64:
case Code.Ja_rel8_16:
case Code.Ja_rel8_32:
case Code.Ja_rel8_64:
2018-09-05 23:38:53 +00:00
2018-09-17 17:49:26 +00:00
case Code.Js_rel8_16:
case Code.Js_rel8_32:
case Code.Js_rel8_64:
case Code.Jns_rel8_16:
case Code.Jns_rel8_32:
case Code.Jns_rel8_64:
case Code.Jp_rel8_16:
case Code.Jp_rel8_32:
case Code.Jp_rel8_64:
case Code.Jnp_rel8_16:
case Code.Jnp_rel8_32:
case Code.Jnp_rel8_64:
case Code.Jl_rel8_16:
case Code.Jl_rel8_32:
case Code.Jl_rel8_64:
case Code.Jge_rel8_16:
case Code.Jge_rel8_32:
case Code.Jge_rel8_64:
case Code.Jle_rel8_16:
case Code.Jle_rel8_32:
case Code.Jle_rel8_64:
case Code.Jg_rel8_16:
case Code.Jg_rel8_32:
case Code.Jg_rel8_64:
2018-09-05 23:38:53 +00:00
2018-09-17 17:49:26 +00:00
case Code.Jmp_rel8_16:
case Code.Jmp_rel8_32:
case Code.Jmp_rel8_64:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.ShortBranch;
2018-09-17 17:49:26 +00:00
case Code.Loopne_rel8_16_CX:
case Code.Loopne_rel8_32_CX:
case Code.Loopne_rel8_16_ECX:
case Code.Loopne_rel8_32_ECX:
case Code.Loopne_rel8_64_ECX:
case Code.Loopne_rel8_16_RCX:
2018-09-17 17:49:26 +00:00
case Code.Loopne_rel8_64_RCX:
case Code.Loope_rel8_16_CX:
case Code.Loope_rel8_32_CX:
case Code.Loope_rel8_16_ECX:
case Code.Loope_rel8_32_ECX:
case Code.Loope_rel8_64_ECX:
case Code.Loope_rel8_16_RCX:
2018-09-17 17:49:26 +00:00
case Code.Loope_rel8_64_RCX:
case Code.Loop_rel8_16_CX:
case Code.Loop_rel8_32_CX:
case Code.Loop_rel8_16_ECX:
case Code.Loop_rel8_32_ECX:
case Code.Loop_rel8_64_ECX:
case Code.Loop_rel8_16_RCX:
2018-09-17 17:49:26 +00:00
case Code.Loop_rel8_64_RCX:
case Code.Jcxz_rel8_16:
case Code.Jcxz_rel8_32:
case Code.Jecxz_rel8_16:
case Code.Jecxz_rel8_32:
case Code.Jecxz_rel8_64:
case Code.Jrcxz_rel8_16:
2018-09-17 17:49:26 +00:00
case Code.Jrcxz_rel8_64:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.AlwaysShortBranch;
2018-09-17 17:49:26 +00:00
case Code.Call_rel16:
case Code.Call_rel32_32:
case Code.Call_rel32_64:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.NearCall;
2018-09-17 17:49:26 +00:00
case Code.Jo_rel16:
case Code.Jo_rel32_32:
case Code.Jo_rel32_64:
case Code.Jno_rel16:
case Code.Jno_rel32_32:
case Code.Jno_rel32_64:
case Code.Jb_rel16:
case Code.Jb_rel32_32:
case Code.Jb_rel32_64:
case Code.Jae_rel16:
case Code.Jae_rel32_32:
case Code.Jae_rel32_64:
case Code.Je_rel16:
case Code.Je_rel32_32:
case Code.Je_rel32_64:
case Code.Jne_rel16:
case Code.Jne_rel32_32:
case Code.Jne_rel32_64:
case Code.Jbe_rel16:
case Code.Jbe_rel32_32:
case Code.Jbe_rel32_64:
case Code.Ja_rel16:
case Code.Ja_rel32_32:
case Code.Ja_rel32_64:
2018-09-05 23:38:53 +00:00
2018-09-17 17:49:26 +00:00
case Code.Js_rel16:
case Code.Js_rel32_32:
case Code.Js_rel32_64:
case Code.Jns_rel16:
case Code.Jns_rel32_32:
case Code.Jns_rel32_64:
case Code.Jp_rel16:
case Code.Jp_rel32_32:
case Code.Jp_rel32_64:
case Code.Jnp_rel16:
case Code.Jnp_rel32_32:
case Code.Jnp_rel32_64:
case Code.Jl_rel16:
case Code.Jl_rel32_32:
case Code.Jl_rel32_64:
case Code.Jge_rel16:
case Code.Jge_rel32_32:
case Code.Jge_rel32_64:
case Code.Jle_rel16:
case Code.Jle_rel32_32:
case Code.Jle_rel32_64:
case Code.Jg_rel16:
case Code.Jg_rel32_32:
case Code.Jg_rel32_64:
2018-09-05 23:38:53 +00:00
2018-09-17 17:49:26 +00:00
case Code.Jmp_rel16:
case Code.Jmp_rel32_32:
case Code.Jmp_rel32_64:
2018-09-28 20:44:53 +00:00
case Code.Jmpe_disp16:
case Code.Jmpe_disp32:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.NearBranch;
2018-09-17 17:49:26 +00:00
case Code.Call_ptr3216:
case Code.Call_ptr1616:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.FarCall;
2018-09-17 17:49:26 +00:00
case Code.Jmp_ptr3216:
case Code.Jmp_ptr1616:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.FarBranch;
2018-09-17 17:49:26 +00:00
case Code.Xbegin_rel16:
2018-09-17 18:06:31 +00:00
case Code.Xbegin_rel32:
case Code.Xbegin_rel32_REXW:
2018-09-05 23:38:53 +00:00
return FormatterFlowControl.Xbegin;
default:
throw new InvalidOperationException();
}
}
public static bool IsRepeOrRepneInstruction(Code code) {
switch (code) {
2018-09-17 17:49:26 +00:00
case Code.Cmpsb_m8_m8:
case Code.Cmpsw_m16_m16:
case Code.Cmpsd_m32_m32:
case Code.Cmpsq_m64_m64:
case Code.Scasb_AL_m8:
case Code.Scasw_AX_m16:
case Code.Scasd_EAX_m32:
case Code.Scasq_RAX_m64:
2018-09-05 23:38:53 +00:00
return true;
default:
return false;
}
}
2018-09-29 11:58:14 +00:00
public static bool IsNoTrackPrefixBranch(Code code) {
Debug.Assert(Code.Jmp_rm16 + 1 == Code.Jmp_rm32);
Debug.Assert(Code.Jmp_rm16 + 2 == Code.Jmp_rm64);
Debug.Assert(Code.Call_rm16 + 1 == Code.Call_rm32);
Debug.Assert(Code.Call_rm16 + 2 == Code.Call_rm64);
return (uint)code - (uint)Code.Jmp_rm16 <= 2 || (uint)code - (uint)Code.Call_rm16 <= 2;
}
2018-09-05 23:38:53 +00:00
}
}
#endif