mirror of https://github.com/icsharpcode/ILSpy.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
332 lines
12 KiB
332 lines
12 KiB
// Copyright (c) 2018 Siegfried Pammer |
|
// |
|
// 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.Diagnostics; |
|
using System.Linq; |
|
|
|
using ICSharpCode.Decompiler.FlowAnalysis; |
|
using ICSharpCode.Decompiler.IL.Transforms; |
|
using ICSharpCode.Decompiler.TypeSystem; |
|
|
|
namespace ICSharpCode.Decompiler.IL.ControlFlow |
|
{ |
|
class AwaitInFinallyTransform |
|
{ |
|
public static void Run(ILFunction function, ILTransformContext context) |
|
{ |
|
if (!context.Settings.AwaitInCatchFinally) |
|
return; |
|
HashSet<BlockContainer> changedContainers = new HashSet<BlockContainer>(); |
|
|
|
// analyze all try-catch statements in the function |
|
foreach (var tryCatch in function.Descendants.OfType<TryCatch>().ToArray()) |
|
{ |
|
if (!(tryCatch.Parent?.Parent is BlockContainer container)) |
|
continue; |
|
// } catch exceptionVariable : 02000078 System.Object when (ldc.i4 1) BlockContainer { |
|
// Block IL_004a (incoming: 1) { |
|
// stloc objectVariable(ldloc exceptionVariable) |
|
// br finallyBlock |
|
// } |
|
// |
|
// } |
|
// } |
|
// |
|
// Block finallyBlock (incoming: 2) { |
|
// if (comp.o(ldloc b == ldnull)) br afterFinallyBlock |
|
// br finallyBlockContinuation |
|
// } |
|
// |
|
// Block finallyBlockContinuation (incoming: 1) { |
|
// await(addressof System.Threading.Tasks.ValueTask(callvirt DisposeAsync(ldloc b))) |
|
// br afterFinallyBlock |
|
// } |
|
// |
|
// Block afterFinallyBlock (incoming: 2) { |
|
// stloc V_1(ldloc objectVariable) |
|
// if (comp.o(ldloc V_1 == ldnull)) br IL_00ea |
|
// br IL_00cf |
|
// } |
|
|
|
// await in finally uses a single catch block with catch-type object |
|
if (tryCatch.Handlers.Count != 1) |
|
{ |
|
continue; |
|
} |
|
var handler = tryCatch.Handlers[0]; |
|
var exceptionVariable = handler.Variable; |
|
if (handler.Body is not BlockContainer catchBlockContainer) |
|
continue; |
|
if (!exceptionVariable.Type.IsKnownType(KnownTypeCode.Object)) |
|
continue; |
|
// Matches the await finally pattern: |
|
// [stloc V_3(ldloc E_100) - copy exception variable to a temporary] |
|
// stloc V_6(ldloc V_3) - store exception in 'global' object variable |
|
// br IL_0075 - jump out of catch block to the head of the finallyBlock |
|
var catchBlockEntry = catchBlockContainer.EntryPoint; |
|
ILVariable objectVariable; |
|
switch (catchBlockEntry.Instructions.Count) |
|
{ |
|
case 2: |
|
if (!catchBlockEntry.Instructions[0].MatchStLoc(out objectVariable, out var value)) |
|
continue; |
|
if (!value.MatchLdLoc(exceptionVariable)) |
|
continue; |
|
break; |
|
case 3: |
|
if (!catchBlockEntry.Instructions[0].MatchStLoc(out var temporaryVariable, out value)) |
|
continue; |
|
if (!value.MatchLdLoc(exceptionVariable)) |
|
continue; |
|
if (!catchBlockEntry.Instructions[1].MatchStLoc(out objectVariable, out value)) |
|
continue; |
|
if (!value.MatchLdLoc(temporaryVariable)) |
|
continue; |
|
break; |
|
default: |
|
continue; |
|
} |
|
if (!catchBlockEntry.Instructions[catchBlockEntry.Instructions.Count - 1].MatchBranch(out var entryPointOfFinally)) |
|
continue; |
|
// globalCopyVar should only be used once, at the end of the finally-block |
|
if (objectVariable.LoadCount != 1 || objectVariable.StoreCount > 2) |
|
continue; |
|
|
|
var beforeExceptionCaptureBlock = (Block)LocalFunctionDecompiler.GetStatement(objectVariable.LoadInstructions[0])?.Parent; |
|
if (beforeExceptionCaptureBlock == null) |
|
continue; |
|
|
|
var (noThrowBlock, exceptionCaptureBlock, objectVariableCopy) = FindBlockAfterFinally(context, beforeExceptionCaptureBlock, objectVariable); |
|
if (noThrowBlock == null || exceptionCaptureBlock == null) |
|
continue; |
|
|
|
var initOfStateVariable = tryCatch.Parent.Children.ElementAtOrDefault(tryCatch.ChildIndex - 1) as StLoc; |
|
if (initOfStateVariable == null || !initOfStateVariable.Value.MatchLdcI4(0)) |
|
continue; |
|
|
|
var stateVariable = initOfStateVariable.Variable; |
|
if (!ValidateStateVariable(stateVariable, initOfStateVariable, tryCatch, entryPointOfFinally)) |
|
continue; |
|
|
|
StateRangeAnalysis sra = new StateRangeAnalysis(StateRangeAnalysisMode.AwaitInFinally, null, stateVariable); |
|
sra.AssignStateRanges(noThrowBlock, Util.LongSet.Universe); |
|
var mapping = sra.GetBlockStateSetMapping((BlockContainer)noThrowBlock.Parent); |
|
var mappingForLeave = sra.GetBlockStateSetMappingForLeave(); |
|
|
|
context.StepStartGroup("Inline finally block with await", tryCatch.Handlers[0]); |
|
var cfg = new ControlFlowGraph(container, context.CancellationToken); |
|
changedContainers.Add(container); |
|
|
|
var finallyContainer = new BlockContainer().WithILRange(catchBlockContainer); |
|
tryCatch.ReplaceWith(new TryFinally(tryCatch.TryBlock, finallyContainer).WithILRange(tryCatch.TryBlock)); |
|
|
|
context.Step("Move blocks into finally", finallyContainer); |
|
MoveDominatedBlocksToContainer(entryPointOfFinally, beforeExceptionCaptureBlock, cfg, finallyContainer); |
|
|
|
SimplifyEndOfFinally(context, objectVariable, beforeExceptionCaptureBlock, objectVariableCopy, finallyContainer); |
|
|
|
if (noThrowBlock.Instructions[0].MatchLdLoc(stateVariable)) |
|
{ |
|
noThrowBlock.Instructions.RemoveAt(0); |
|
} |
|
|
|
foreach (var branch in tryCatch.TryBlock.Descendants.OfType<Branch>()) |
|
{ |
|
if (branch.TargetBlock == entryPointOfFinally) |
|
{ |
|
if (!(branch.Parent is Block block && branch.ChildIndex > 0 |
|
&& block.Instructions[branch.ChildIndex - 1].MatchStLoc(stateVariable, out var v) |
|
&& v.MatchLdcI4(out int value))) |
|
{ |
|
value = 0; |
|
} |
|
if (mapping.TryGetValue(value, out Block targetBlock)) |
|
{ |
|
context.Step($"branch to finally with state {value} => branch to state target " + targetBlock.Label, branch); |
|
branch.TargetBlock = targetBlock; |
|
} |
|
else if (mappingForLeave.TryGetValue(value, out BlockContainer targetContainer)) |
|
{ |
|
context.Step($"branch to finally with state {value} => leave to state target " + targetContainer, branch); |
|
branch.ReplaceWith(new Leave(targetContainer).WithILRange(branch)); |
|
} |
|
else |
|
{ |
|
context.Step("branch to finally => branch after finally", branch); |
|
branch.TargetBlock = noThrowBlock; |
|
} |
|
} |
|
else |
|
{ |
|
Debug.Assert(branch.TargetBlock.IsDescendantOf(tryCatch.TryBlock)); |
|
} |
|
} |
|
|
|
context.StepEndGroup(keepIfEmpty: true); |
|
} |
|
|
|
context.Step("Clean up", function); |
|
|
|
// clean up all modified containers |
|
foreach (var container in changedContainers) |
|
container.SortBlocks(deleteUnreachableBlocks: true); |
|
|
|
((BlockContainer)function.Body).SortBlocks(deleteUnreachableBlocks: true); |
|
|
|
void MoveDominatedBlocksToContainer(Block newEntryPoint, Block endBlock, ControlFlowGraph graph, |
|
BlockContainer targetContainer) |
|
{ |
|
var node = graph.GetNode(newEntryPoint); |
|
var endNode = endBlock == null ? null : graph.GetNode(endBlock); |
|
|
|
MoveBlock(newEntryPoint, targetContainer); |
|
|
|
foreach (var n in graph.cfg) |
|
{ |
|
Block block = (Block)n.UserData; |
|
|
|
if (node.Dominates(n)) |
|
{ |
|
if (endNode != null && endNode != n && endNode.Dominates(n)) |
|
continue; |
|
|
|
if (block.Parent == targetContainer) |
|
continue; |
|
|
|
MoveBlock(block, targetContainer); |
|
} |
|
} |
|
} |
|
|
|
void MoveBlock(Block block, BlockContainer target) |
|
{ |
|
context.Step($"Move {block.Label} to container at IL_{target.StartILOffset:x4}", target); |
|
block.Remove(); |
|
target.Blocks.Add(block); |
|
} |
|
|
|
static void SimplifyEndOfFinally(ILTransformContext context, ILVariable objectVariable, Block beforeExceptionCaptureBlock, ILVariable objectVariableCopy, BlockContainer finallyContainer) |
|
{ |
|
if (beforeExceptionCaptureBlock.Instructions.Count >= 3 |
|
&& beforeExceptionCaptureBlock.Instructions.SecondToLastOrDefault().MatchIfInstruction(out var cond, out var brInst) |
|
&& beforeExceptionCaptureBlock.Instructions.LastOrDefault() is Branch branch |
|
&& beforeExceptionCaptureBlock.Instructions[beforeExceptionCaptureBlock.Instructions.Count - 3].MatchStLoc(objectVariableCopy, out var value) |
|
&& value.MatchLdLoc(objectVariable)) |
|
{ |
|
if (cond.MatchCompEqualsNull(out var arg) && arg.MatchLdLoc(objectVariableCopy)) |
|
{ |
|
context.Step("Simplify end of finally", beforeExceptionCaptureBlock); |
|
beforeExceptionCaptureBlock.Instructions.RemoveRange(beforeExceptionCaptureBlock.Instructions.Count - 3, 2); |
|
branch.ReplaceWith(new Leave(finallyContainer).WithILRange(branch)); |
|
} |
|
else if (cond.MatchCompNotEqualsNull(out arg) && arg.MatchLdLoc(objectVariableCopy)) |
|
{ |
|
context.Step("Simplify end of finally", beforeExceptionCaptureBlock); |
|
beforeExceptionCaptureBlock.Instructions.RemoveRange(beforeExceptionCaptureBlock.Instructions.Count - 3, 2); |
|
branch.ReplaceWith(new Leave(finallyContainer).WithILRange(branch)); |
|
} |
|
else |
|
{ |
|
Debug.Fail("Broken beforeExceptionCaptureBlock"); |
|
} |
|
} |
|
else |
|
{ |
|
Debug.Fail("Broken beforeExceptionCaptureBlock"); |
|
} |
|
} |
|
} |
|
|
|
private static bool ValidateStateVariable(ILVariable stateVariable, StLoc initializer, TryCatch tryCatch, Block entryPointOfFinally) |
|
{ |
|
if (stateVariable.AddressCount > 0) |
|
return false; |
|
|
|
foreach (var store in stateVariable.StoreInstructions) |
|
{ |
|
if (store == initializer) |
|
continue; |
|
if (store is not StLoc stloc) |
|
return false; |
|
if (!stloc.Value.MatchLdcI4(out _)) |
|
return false; |
|
if (!stloc.IsDescendantOf(tryCatch)) |
|
return false; |
|
if (stloc.Parent is not Block block) |
|
return false; |
|
if (block.Instructions.ElementAtOrDefault(stloc.ChildIndex + 1)?.MatchBranch(entryPointOfFinally) != true) |
|
return false; |
|
} |
|
|
|
return true; |
|
} |
|
|
|
static (Block, Block, ILVariable) FindBlockAfterFinally(ILTransformContext context, Block block, ILVariable objectVariable) |
|
{ |
|
// Block IL_0327 (incoming: 2) { |
|
// stloc V_7(ldloc I_0) |
|
// if (comp.o(ldloc V_7 == ldnull)) br IL_034a |
|
// br IL_0333 |
|
// } |
|
int count = block.Instructions.Count; |
|
if (count < 3) |
|
return default; |
|
|
|
if (!block.Instructions[count - 3].MatchStLoc(out var objectVariableCopy, out var value)) |
|
return default; |
|
|
|
if (!value.MatchLdLoc(objectVariable)) |
|
return default; |
|
|
|
if (!block.Instructions[count - 2].MatchIfInstruction(out var cond, out var noThrowBlockBranch)) |
|
return default; |
|
|
|
if (!noThrowBlockBranch.MatchBranch(out var noThrowBlock)) |
|
return default; |
|
|
|
if (!block.Instructions[count - 1].MatchBranch(out var exceptionCaptureBlock)) |
|
return default; |
|
|
|
if (cond.MatchCompEqualsNull(out var arg)) |
|
{ |
|
if (!arg.MatchLdLoc(objectVariableCopy)) |
|
return default; |
|
} |
|
else if (cond.MatchCompNotEqualsNull(out arg)) |
|
{ |
|
if (!arg.MatchLdLoc(objectVariableCopy)) |
|
return default; |
|
(noThrowBlock, exceptionCaptureBlock) = (exceptionCaptureBlock, noThrowBlock); |
|
} |
|
else |
|
{ |
|
return default; |
|
} |
|
|
|
if (!AwaitInCatchTransform.MatchExceptionCaptureBlock(context, exceptionCaptureBlock, |
|
ref objectVariableCopy, out _, out _, out _)) |
|
{ |
|
return default; |
|
} |
|
|
|
return (noThrowBlock, exceptionCaptureBlock, objectVariableCopy); |
|
} |
|
} |
|
}
|
|
|