520 lines
14 KiB
C#
520 lines
14 KiB
C#
//
|
|
// MethodBodyRocks.cs
|
|
//
|
|
// Author:
|
|
// Jb Evain (jbevain@gmail.com)
|
|
//
|
|
// Copyright (c) 2008 - 2011 Jb Evain
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining
|
|
// a copy of this software and associated documentation files (the
|
|
// "Software"), to deal in the Software without restriction, including
|
|
// without limitation the rights to use, copy, modify, merge, publish,
|
|
// distribute, sublicense, and/or sell copies of the Software, and to
|
|
// permit persons to whom the Software is furnished to do so, subject to
|
|
// the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be
|
|
// included in all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
//
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
using Mono.Cecil;
|
|
using Mono.Cecil.Cil;
|
|
|
|
namespace Mono.Tuner {
|
|
|
|
public static class MethodBodyRocks {
|
|
|
|
public static IEnumerable<TypeDefinition> GetAllTypes (this ModuleDefinition self)
|
|
{
|
|
return self.Types.SelectMany (t => t.GetAllTypes ());
|
|
}
|
|
|
|
static IEnumerable<TypeDefinition> GetAllTypes (this TypeDefinition self)
|
|
{
|
|
yield return self;
|
|
|
|
if (!self.HasNestedTypes)
|
|
yield break;
|
|
|
|
foreach (var type in self.NestedTypes.SelectMany (t => t.GetAllTypes ()))
|
|
yield return type;
|
|
}
|
|
|
|
public static IEnumerable<MethodDefinition> GetMethods (this TypeDefinition self)
|
|
{
|
|
return self.Methods.Where (m => !m.IsConstructor);
|
|
}
|
|
|
|
public static IEnumerable<MethodDefinition> GetConstructors (this TypeDefinition self)
|
|
{
|
|
return self.Methods.Where (m => m.IsConstructor);
|
|
}
|
|
|
|
public static MethodDefinition GetTypeConstructor (this TypeDefinition self)
|
|
{
|
|
return self.GetConstructors ().FirstOrDefault (c => c.IsStatic);
|
|
}
|
|
|
|
public static void SimplifyMacros (this MethodBody self)
|
|
{
|
|
if (self == null)
|
|
throw new ArgumentNullException ("self");
|
|
|
|
foreach (var instruction in self.Instructions) {
|
|
if (instruction.OpCode.OpCodeType != OpCodeType.Macro)
|
|
continue;
|
|
|
|
switch (instruction.OpCode.Code) {
|
|
case Code.Ldarg_0:
|
|
ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (0));
|
|
break;
|
|
case Code.Ldarg_1:
|
|
ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (1));
|
|
break;
|
|
case Code.Ldarg_2:
|
|
ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (2));
|
|
break;
|
|
case Code.Ldarg_3:
|
|
ExpandMacro (instruction, OpCodes.Ldarg, self.GetParameter (3));
|
|
break;
|
|
case Code.Ldloc_0:
|
|
ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [0]);
|
|
break;
|
|
case Code.Ldloc_1:
|
|
ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [1]);
|
|
break;
|
|
case Code.Ldloc_2:
|
|
ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [2]);
|
|
break;
|
|
case Code.Ldloc_3:
|
|
ExpandMacro (instruction, OpCodes.Ldloc, self.Variables [3]);
|
|
break;
|
|
case Code.Stloc_0:
|
|
ExpandMacro (instruction, OpCodes.Stloc, self.Variables [0]);
|
|
break;
|
|
case Code.Stloc_1:
|
|
ExpandMacro (instruction, OpCodes.Stloc, self.Variables [1]);
|
|
break;
|
|
case Code.Stloc_2:
|
|
ExpandMacro (instruction, OpCodes.Stloc, self.Variables [2]);
|
|
break;
|
|
case Code.Stloc_3:
|
|
ExpandMacro (instruction, OpCodes.Stloc, self.Variables [3]);
|
|
break;
|
|
case Code.Ldarg_S:
|
|
instruction.OpCode = OpCodes.Ldarg;
|
|
break;
|
|
case Code.Ldarga_S:
|
|
instruction.OpCode = OpCodes.Ldarga;
|
|
break;
|
|
case Code.Starg_S:
|
|
instruction.OpCode = OpCodes.Starg;
|
|
break;
|
|
case Code.Ldloc_S:
|
|
instruction.OpCode = OpCodes.Ldloc;
|
|
break;
|
|
case Code.Ldloca_S:
|
|
instruction.OpCode = OpCodes.Ldloca;
|
|
break;
|
|
case Code.Stloc_S:
|
|
instruction.OpCode = OpCodes.Stloc;
|
|
break;
|
|
case Code.Ldc_I4_M1:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, -1);
|
|
break;
|
|
case Code.Ldc_I4_0:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 0);
|
|
break;
|
|
case Code.Ldc_I4_1:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 1);
|
|
break;
|
|
case Code.Ldc_I4_2:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 2);
|
|
break;
|
|
case Code.Ldc_I4_3:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 3);
|
|
break;
|
|
case Code.Ldc_I4_4:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 4);
|
|
break;
|
|
case Code.Ldc_I4_5:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 5);
|
|
break;
|
|
case Code.Ldc_I4_6:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 6);
|
|
break;
|
|
case Code.Ldc_I4_7:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 7);
|
|
break;
|
|
case Code.Ldc_I4_8:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, 8);
|
|
break;
|
|
case Code.Ldc_I4_S:
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4, (int) (sbyte) instruction.Operand);
|
|
break;
|
|
case Code.Br_S:
|
|
instruction.OpCode = OpCodes.Br;
|
|
break;
|
|
case Code.Brfalse_S:
|
|
instruction.OpCode = OpCodes.Brfalse;
|
|
break;
|
|
case Code.Brtrue_S:
|
|
instruction.OpCode = OpCodes.Brtrue;
|
|
break;
|
|
case Code.Beq_S:
|
|
instruction.OpCode = OpCodes.Beq;
|
|
break;
|
|
case Code.Bge_S:
|
|
instruction.OpCode = OpCodes.Bge;
|
|
break;
|
|
case Code.Bgt_S:
|
|
instruction.OpCode = OpCodes.Bgt;
|
|
break;
|
|
case Code.Ble_S:
|
|
instruction.OpCode = OpCodes.Ble;
|
|
break;
|
|
case Code.Blt_S:
|
|
instruction.OpCode = OpCodes.Blt;
|
|
break;
|
|
case Code.Bne_Un_S:
|
|
instruction.OpCode = OpCodes.Bne_Un;
|
|
break;
|
|
case Code.Bge_Un_S:
|
|
instruction.OpCode = OpCodes.Bge_Un;
|
|
break;
|
|
case Code.Bgt_Un_S:
|
|
instruction.OpCode = OpCodes.Bgt_Un;
|
|
break;
|
|
case Code.Ble_Un_S:
|
|
instruction.OpCode = OpCodes.Ble_Un;
|
|
break;
|
|
case Code.Blt_Un_S:
|
|
instruction.OpCode = OpCodes.Blt_Un;
|
|
break;
|
|
case Code.Leave_S:
|
|
instruction.OpCode = OpCodes.Leave;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void ExpandMacro (Instruction instruction, OpCode opcode, object operand)
|
|
{
|
|
instruction.OpCode = opcode;
|
|
instruction.Operand = operand;
|
|
}
|
|
|
|
static void MakeMacro (Instruction instruction, OpCode opcode)
|
|
{
|
|
instruction.OpCode = opcode;
|
|
instruction.Operand = null;
|
|
}
|
|
|
|
public static void OptimizeMacros (this MethodBody self)
|
|
{
|
|
if (self == null)
|
|
throw new ArgumentNullException ("self");
|
|
|
|
var method = self.Method;
|
|
|
|
foreach (var instruction in self.Instructions) {
|
|
int index;
|
|
switch (instruction.OpCode.Code) {
|
|
case Code.Ldarg:
|
|
index = ((ParameterDefinition) instruction.Operand).Index;
|
|
if (index == -1 && instruction.Operand == self.ThisParameter)
|
|
index = 0;
|
|
else if (method.HasThis)
|
|
index++;
|
|
|
|
switch (index) {
|
|
case 0:
|
|
MakeMacro (instruction, OpCodes.Ldarg_0);
|
|
break;
|
|
case 1:
|
|
MakeMacro (instruction, OpCodes.Ldarg_1);
|
|
break;
|
|
case 2:
|
|
MakeMacro (instruction, OpCodes.Ldarg_2);
|
|
break;
|
|
case 3:
|
|
MakeMacro (instruction, OpCodes.Ldarg_3);
|
|
break;
|
|
default:
|
|
if (index < 256)
|
|
ExpandMacro (instruction, OpCodes.Ldarg_S, instruction.Operand);
|
|
break;
|
|
}
|
|
break;
|
|
case Code.Ldloc:
|
|
index = ((VariableDefinition) instruction.Operand).Index;
|
|
switch (index) {
|
|
case 0:
|
|
MakeMacro (instruction, OpCodes.Ldloc_0);
|
|
break;
|
|
case 1:
|
|
MakeMacro (instruction, OpCodes.Ldloc_1);
|
|
break;
|
|
case 2:
|
|
MakeMacro (instruction, OpCodes.Ldloc_2);
|
|
break;
|
|
case 3:
|
|
MakeMacro (instruction, OpCodes.Ldloc_3);
|
|
break;
|
|
default:
|
|
if (index < 256)
|
|
ExpandMacro (instruction, OpCodes.Ldloc_S, instruction.Operand);
|
|
break;
|
|
}
|
|
break;
|
|
case Code.Stloc:
|
|
index = ((VariableDefinition) instruction.Operand).Index;
|
|
switch (index) {
|
|
case 0:
|
|
MakeMacro (instruction, OpCodes.Stloc_0);
|
|
break;
|
|
case 1:
|
|
MakeMacro (instruction, OpCodes.Stloc_1);
|
|
break;
|
|
case 2:
|
|
MakeMacro (instruction, OpCodes.Stloc_2);
|
|
break;
|
|
case 3:
|
|
MakeMacro (instruction, OpCodes.Stloc_3);
|
|
break;
|
|
default:
|
|
if (index < 256)
|
|
ExpandMacro (instruction, OpCodes.Stloc_S, instruction.Operand);
|
|
break;
|
|
}
|
|
break;
|
|
case Code.Ldarga:
|
|
index = ((ParameterDefinition) instruction.Operand).Index;
|
|
if (index == -1 && instruction.Operand == self.ThisParameter)
|
|
index = 0;
|
|
else if (method.HasThis)
|
|
index++;
|
|
if (index < 256)
|
|
ExpandMacro (instruction, OpCodes.Ldarga_S, instruction.Operand);
|
|
break;
|
|
case Code.Ldloca:
|
|
if (((VariableDefinition) instruction.Operand).Index < 256)
|
|
ExpandMacro (instruction, OpCodes.Ldloca_S, instruction.Operand);
|
|
break;
|
|
case Code.Ldc_I4:
|
|
int i = (int) instruction.Operand;
|
|
switch (i) {
|
|
case -1:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_M1);
|
|
break;
|
|
case 0:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_0);
|
|
break;
|
|
case 1:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_1);
|
|
break;
|
|
case 2:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_2);
|
|
break;
|
|
case 3:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_3);
|
|
break;
|
|
case 4:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_4);
|
|
break;
|
|
case 5:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_5);
|
|
break;
|
|
case 6:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_6);
|
|
break;
|
|
case 7:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_7);
|
|
break;
|
|
case 8:
|
|
MakeMacro (instruction, OpCodes.Ldc_I4_8);
|
|
break;
|
|
default:
|
|
if (i >= -128 && i < 128)
|
|
ExpandMacro (instruction, OpCodes.Ldc_I4_S, (sbyte) i);
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
OptimizeBranches (self);
|
|
}
|
|
|
|
static void OptimizeBranches (MethodBody body)
|
|
{
|
|
ComputeOffsets (body);
|
|
|
|
foreach (var instruction in body.Instructions) {
|
|
if (instruction.OpCode.OperandType != OperandType.InlineBrTarget)
|
|
continue;
|
|
|
|
if (OptimizeBranch (instruction))
|
|
ComputeOffsets (body);
|
|
}
|
|
}
|
|
|
|
static bool OptimizeBranch (Instruction instruction)
|
|
{
|
|
var offset = ((Instruction) instruction.Operand).Offset - (instruction.Offset + instruction.OpCode.Size + 4);
|
|
if (!(offset >= -128 && offset <= 127))
|
|
return false;
|
|
|
|
switch (instruction.OpCode.Code) {
|
|
case Code.Br:
|
|
instruction.OpCode = OpCodes.Br_S;
|
|
break;
|
|
case Code.Brfalse:
|
|
instruction.OpCode = OpCodes.Brfalse_S;
|
|
break;
|
|
case Code.Brtrue:
|
|
instruction.OpCode = OpCodes.Brtrue_S;
|
|
break;
|
|
case Code.Beq:
|
|
instruction.OpCode = OpCodes.Beq_S;
|
|
break;
|
|
case Code.Bge:
|
|
instruction.OpCode = OpCodes.Bge_S;
|
|
break;
|
|
case Code.Bgt:
|
|
instruction.OpCode = OpCodes.Bgt_S;
|
|
break;
|
|
case Code.Ble:
|
|
instruction.OpCode = OpCodes.Ble_S;
|
|
break;
|
|
case Code.Blt:
|
|
instruction.OpCode = OpCodes.Blt_S;
|
|
break;
|
|
case Code.Bne_Un:
|
|
instruction.OpCode = OpCodes.Bne_Un_S;
|
|
break;
|
|
case Code.Bge_Un:
|
|
instruction.OpCode = OpCodes.Bge_Un_S;
|
|
break;
|
|
case Code.Bgt_Un:
|
|
instruction.OpCode = OpCodes.Bgt_Un_S;
|
|
break;
|
|
case Code.Ble_Un:
|
|
instruction.OpCode = OpCodes.Ble_Un_S;
|
|
break;
|
|
case Code.Blt_Un:
|
|
instruction.OpCode = OpCodes.Blt_Un_S;
|
|
break;
|
|
case Code.Leave:
|
|
instruction.OpCode = OpCodes.Leave_S;
|
|
break;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
static void ComputeOffsets (MethodBody body)
|
|
{
|
|
var offset = 0;
|
|
foreach (var instruction in body.Instructions) {
|
|
instruction.Offset = offset;
|
|
offset += instruction.GetSize ();
|
|
}
|
|
}
|
|
|
|
public static ParameterDefinition GetParameter (this MethodBody self, int index)
|
|
{
|
|
var method = self.Method;
|
|
|
|
if (method.HasThis) {
|
|
if (index == 0)
|
|
return self.ThisParameter;
|
|
|
|
index--;
|
|
}
|
|
|
|
var parameters = method.Parameters;
|
|
|
|
if (index < 0 || index >= parameters.Count)
|
|
return null;
|
|
|
|
return parameters [index];
|
|
}
|
|
|
|
public static bool Implements (this TypeReference self, string interfaceName)
|
|
{
|
|
if (interfaceName == null)
|
|
throw new ArgumentNullException ("interfaceName");
|
|
if (self == null)
|
|
return false;
|
|
|
|
TypeDefinition type = self.Resolve ();
|
|
if (type == null)
|
|
return false; // not enough information available
|
|
|
|
// special case, check if we implement ourselves
|
|
if (type.IsInterface && (type.FullName == interfaceName))
|
|
return true;
|
|
|
|
return Implements (type, interfaceName, (interfaceName.IndexOf ('`') >= 0));
|
|
}
|
|
|
|
public static bool Implements (TypeDefinition type, string interfaceName, bool generic)
|
|
{
|
|
while (type != null) {
|
|
// does the type implements it itself
|
|
if (type.HasInterfaces) {
|
|
foreach (var iface in type.Interfaces) {
|
|
string fullname = (generic) ? iface.InterfaceType.GetElementType ().FullName : iface.InterfaceType.FullName;
|
|
if (fullname == interfaceName)
|
|
return true;
|
|
//if not, then maybe one of its parent interfaces does
|
|
if (Implements (iface.InterfaceType.Resolve (), interfaceName, generic))
|
|
return true;
|
|
}
|
|
}
|
|
|
|
type = type.BaseType != null ? type.BaseType.Resolve () : null;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public static bool Inherits (this TypeReference self, string @namespace, string name)
|
|
{
|
|
if (@namespace == null)
|
|
throw new ArgumentNullException ("namespace");
|
|
if (name == null)
|
|
throw new ArgumentNullException ("name");
|
|
if (self == null)
|
|
return false;
|
|
|
|
TypeReference current = self.Resolve ();
|
|
while (current != null) {
|
|
if (current.Is (@namespace, name))
|
|
return true;
|
|
if (current.Is ("System", "Object"))
|
|
return false;
|
|
|
|
TypeDefinition td = current.Resolve ();
|
|
if (td == null)
|
|
return false; // could not resolve type
|
|
current = td.BaseType;
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|