mirror of https://github.com/icsharpcode/ILSpy.git
3 changed files with 129 additions and 1 deletions
@ -0,0 +1,126 @@ |
|||||||
|
using System.Diagnostics; |
||||||
|
using ICSharpCode.Decompiler.TypeSystem; |
||||||
|
using Mono.Cecil; |
||||||
|
|
||||||
|
namespace ICSharpCode.Decompiler.IL.Transforms |
||||||
|
{ |
||||||
|
class ProxyCallReplacer : ILVisitor, IILTransform |
||||||
|
{ |
||||||
|
ILTransformContext context; |
||||||
|
|
||||||
|
public void Run(ILFunction function, ILTransformContext context) |
||||||
|
{ |
||||||
|
this.context = context; |
||||||
|
function.AcceptVisitor(this); |
||||||
|
} |
||||||
|
|
||||||
|
protected override void Default(ILInstruction inst) |
||||||
|
{ |
||||||
|
foreach (var child in inst.Children) { |
||||||
|
child.AcceptVisitor(this); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
protected internal override void VisitCallVirt(CallVirt inst) |
||||||
|
{ |
||||||
|
VisitCally(inst); |
||||||
|
} |
||||||
|
|
||||||
|
protected internal override void VisitCall(Call inst) |
||||||
|
{ |
||||||
|
VisitCally(inst); |
||||||
|
} |
||||||
|
|
||||||
|
protected internal void VisitCally(CallInstruction inst) |
||||||
|
{ |
||||||
|
if (inst.Method.DeclaringTypeDefinition == null) // TODO: investigate why
|
||||||
|
return; |
||||||
|
foreach (IMethod method in inst.Method.DeclaringTypeDefinition.Methods) { |
||||||
|
if (method.FullName.Equals(inst.Method.FullName)) { |
||||||
|
MethodDefinition methodDef = context.TypeSystem.GetCecil(method) as MethodDefinition; |
||||||
|
if (methodDef != null && methodDef.Body != null) { |
||||||
|
if (method.IsCompilerGeneratedOrIsInCompilerGeneratedClass()) { |
||||||
|
// partially copied from CSharpDecompiler
|
||||||
|
var specializingTypeSystem = this.context.TypeSystem.GetSpecializingTypeSystem(this.context.TypeSystem.Compilation.TypeResolveContext); |
||||||
|
var ilReader = new ILReader(specializingTypeSystem); |
||||||
|
System.Threading.CancellationToken cancellationToken = new System.Threading.CancellationToken(); |
||||||
|
var function = ilReader.ReadIL(methodDef.Body, cancellationToken); |
||||||
|
var context = new ILTransformContext(function, specializingTypeSystem, this.context.Settings) { |
||||||
|
CancellationToken = cancellationToken |
||||||
|
}; |
||||||
|
foreach (var transform in CSharp.CSharpDecompiler.GetILTransforms()) { |
||||||
|
if (transform.GetType() != typeof(ProxyCallReplacer)) { // don't call itself on itself
|
||||||
|
cancellationToken.ThrowIfCancellationRequested(); |
||||||
|
transform.Run(function, context); |
||||||
|
} |
||||||
|
} |
||||||
|
Call currentCall = new ProxyMethodVisitor().GetCalledMethod(function, context); |
||||||
|
if (currentCall != null) { |
||||||
|
Call newInst = (Call)currentCall.Clone(); |
||||||
|
newInst.Arguments.Clear(); |
||||||
|
|
||||||
|
ILInstruction thisArg = inst.Arguments.ElementAtOrDefault(0).Clone(); |
||||||
|
|
||||||
|
// special handling for first argument (this) - the underlying issue may be somewhere else
|
||||||
|
// normally
|
||||||
|
// leave IL_0000(await(callvirt<> n__0(ldobj xxHandler(ldloca this), ldobj System.Net.Http.HttpRequestMessage(ldloca request), ldobj System.Threading.CancellationToken(ldloca cancellationToken))))
|
||||||
|
// would be decompiled to
|
||||||
|
// return await((DelegatingHandler)this).SendAsync(request, cancellationToken);
|
||||||
|
// this changes it to
|
||||||
|
// return await base.SendAsync(request, cancellationToken);
|
||||||
|
if (thisArg.OpCode == OpCode.LdObj && thisArg.Children.Count > 0 && thisArg.Children[0].OpCode == OpCode.LdLoca) { |
||||||
|
thisArg = new LdLoc(((LdLoca)thisArg.Children[0]).Variable); |
||||||
|
} |
||||||
|
|
||||||
|
newInst.Arguments.Add(thisArg); |
||||||
|
|
||||||
|
// add everything except first argument
|
||||||
|
for (int i = 1; i < inst.Arguments.Count; i++) { |
||||||
|
newInst.Arguments.Add(inst.Arguments.ElementAtOrDefault(i).Clone()); |
||||||
|
} |
||||||
|
inst.ReplaceWith(newInst); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Checks if the method is a proxy method by checking if it only contains a call instruction and if it has special compiler attributes.
|
||||||
|
private class ProxyMethodVisitor : ILVisitor |
||||||
|
{ |
||||||
|
ILTransformContext context; |
||||||
|
int invalidInstructions = 0; |
||||||
|
Call currentCall; |
||||||
|
|
||||||
|
public Call GetCalledMethod(ILFunction function, ILTransformContext context) |
||||||
|
{ |
||||||
|
this.context = context; |
||||||
|
function.AcceptVisitor(this); |
||||||
|
if (invalidInstructions == 0) { |
||||||
|
return currentCall; |
||||||
|
} |
||||||
|
return null; |
||||||
|
} |
||||||
|
|
||||||
|
protected override void Default(ILInstruction inst) |
||||||
|
{ |
||||||
|
if (inst.OpCode != OpCode.ILFunction && |
||||||
|
inst.OpCode != OpCode.BlockContainer && |
||||||
|
inst.OpCode != OpCode.Block && |
||||||
|
inst.OpCode != OpCode.Leave && |
||||||
|
inst.OpCode != OpCode.Nop) { |
||||||
|
invalidInstructions++; |
||||||
|
} |
||||||
|
foreach (var child in inst.Children) { |
||||||
|
child.AcceptVisitor(this); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
protected internal override void VisitCall(Call inst) |
||||||
|
{ |
||||||
|
currentCall = inst; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue