/// <summary>
/// Optimize
/// </summary>
/// <param name="script">Script</param>
/// <param name="parsers">Optmize parser, currently, there are four parsers:
/// <para> DELETE_DEAD_CODDE -- delete dead code parser, default parser</para>
/// <para> USE_SHORT_ADDRESS -- use short address parser. eg: JMP_L -> JMP, JMPIF_L -> JMPIF, default parser</para>
/// <para> DELETE_NOP -- delete nop parser</para>
/// <para> DELETE_USELESS_JMP -- delete useless jmp parser, eg: JPM 2</para></param>
/// <returns>Optimized script</returns>
public static byte[] Optimize(byte[] script, OptimizeParserType[] parserTypes)
{
var optimizer = new NefOptimizer();
foreach (var parserType in parserTypes)
{
object[] objAttrs = parserType.GetType().GetField(parserType.ToString()).GetCustomAttributes(typeof(OptimizeParserAttribute), false);
if (objAttrs is null || objAttrs.Length == 0)
{
continue;
}
var attribute = (OptimizeParserAttribute)objAttrs[0];
var obj = Activator.CreateInstance(attribute.Type);
if (obj is null)
{
continue;
}
IOptimizeParser parser = (IOptimizeParser)obj;
optimizer.AddOptimizeParser(parser);
}
//step01 Load
using (var ms = new MemoryStream(script))
{
optimizer.LoadNef(ms);
}
//step02 doOptimize
optimizer.Optimize();
//step03 link
using (var ms = new MemoryStream())
{
optimizer.LinkNef(ms);
var bytes = ms.ToArray();
return(bytes);
}
}
/// <summary>
/// Optimize
/// </summary>
/// <param name="script">Script</param>
/// <param name="parserTypes">Optmize parser, currently, there are four parsers:
/// <para> DELETE_DEAD_CODDE -- delete dead code parser, default parser</para>
/// <para> USE_SHORT_ADDRESS -- use short address parser. eg: JMP_L -> JMP, JMPIF_L -> JMPIF, default parser</para>
/// <para> DELETE_NOP -- delete nop parser</para>
/// <para> DELETE_USELESS_JMP -- delete useless jmp parser, eg: JPM 2</para>
/// <para> DELETE_USELESS_EQUAL -- delete useless equal parser, eg: EQUAL 01 01 </para></param>
/// <returns>Optimized script</returns>
public static byte[] Optimize(byte[] script, OptimizeParserType[] parserTypes, out Dictionary <int, int> addrConvertTable)
{
var optimizer = new NefOptimizer();
foreach (var parserType in parserTypes)
{
object[] objAttrs = parserType.GetType().GetField(parserType.ToString()).GetCustomAttributes(typeof(OptimizeParserAttribute), false);
if (objAttrs is null || objAttrs.Length == 0)
{
continue;
}
var attribute = (OptimizeParserAttribute)objAttrs[0];
var obj = Activator.CreateInstance(attribute.Type);
if (obj is null)
{
continue;
}
IOptimizeParser parser = (IOptimizeParser)obj;
optimizer.AddOptimizeParser(parser);
}
bool optimized;
addrConvertTable = null;
do
{
//step01 Load
using (var ms = new MemoryStream(script))
{
optimizer.LoadNef(ms);
}
//step02 doOptimize
optimizer.Optimize();
//step03 link
using (var ms = new MemoryStream())
{
optimizer.LinkNef(ms);
if (addrConvertTable is null)
{
addrConvertTable = optimizer.GetAddrConvertTable();
}
else
{
Dictionary <int, int> addrConvertTableTemp = optimizer.GetAddrConvertTable();
addrConvertTable = optimizer.RebuildAddrConvertTable(addrConvertTable, addrConvertTableTemp);
}
var bytes = ms.ToArray();
optimized = bytes.Length < script.Length;
if (optimized)
{
script = bytes;
}
}
// Execute it while decrease the size
}while (optimized);
return(script);
}
/// <summary>
/// Optimize
/// </summary>
/// <param name="script">Script</param>
/// <param name="parserTypes">Optmize parser, currently, there are four parsers:
/// <para> DELETE_DEAD_CODDE -- delete dead code parser, default parser</para>
/// <para> USE_SHORT_ADDRESS -- use short address parser. eg: JMP_L -> JMP, JMPIF_L -> JMPIF, default parser</para>
/// <para> DELETE_NOP -- delete nop parser</para>
/// <para> DELETE_USELESS_JMP -- delete useless jmp parser, eg: JPM 2</para>
/// <para> DELETE_USELESS_EQUAL -- delete useless equal parser, eg: EQUAL 01 01 </para></param>
/// <returns>Optimized script</returns>
public static byte[] Optimize(byte[] script, OptimizeParserType[] parserTypes, out Dictionary <int, int> addrConvertTable)
{
var optimizer = new NefOptimizer();
foreach (var parserType in parserTypes)
{
object[] objAttrs = parserType.GetType().GetField(parserType.ToString()).GetCustomAttributes(typeof(OptimizeParserAttribute), false);
if (objAttrs is null || objAttrs.Length == 0)
{
continue;
}
var attribute = (OptimizeParserAttribute)objAttrs[0];
var obj = Activator.CreateInstance(attribute.Type);
if (obj is null)
{
continue;
}
IOptimizeParser parser = (IOptimizeParser)obj;
optimizer.AddOptimizeParser(parser);
}
addrConvertTable = null;
// 10 iterations max
for (int x = 0; x < 10; x++)
{
//step01 Load
using (var ms = new MemoryStream(script))
{
optimizer.LoadNef(ms);
}
//step02 doOptimize
optimizer.Optimize();
//step03 link
using (var ms = new MemoryStream())
{
optimizer.LinkNef(ms);
if (addrConvertTable is null)
{
addrConvertTable = optimizer.GetAddrConvertTable();
}
else
{
Dictionary <int, int> addrConvertTableTemp = optimizer.GetAddrConvertTable();
addrConvertTable = optimizer.RebuildAddrConvertTable(addrConvertTable, addrConvertTableTemp);
}
var bytes = ms.ToArray();
if (bytes.SequenceEqual(script))
{
// Sometimes the script could be bigger but more efficient
// int.Max+INC (6 bytes) => PUSHInt64 (9 bytes)
break;
}
script = bytes;
}
// Execute it while decrease the size
}
return(script);
}
