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.
747 lines
28 KiB
747 lines
28 KiB
// Copyright (c) AlphaSierraPapa for the SharpDevelop Team (for details please see \doc\copyright.txt)
|
|
// This code is distributed under MIT X11 license (for details please see \doc\license.txt)
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using ICSharpCode.Decompiler.ILAst;
|
|
using ICSharpCode.NRefactory.CSharp;
|
|
using ICSharpCode.NRefactory.CSharp.PatternMatching;
|
|
using Mono.Cecil;
|
|
|
|
namespace ICSharpCode.Decompiler.Ast.Transforms
|
|
{
|
|
/// <summary>
|
|
/// Finds the expanded form of using statements using pattern matching and replaces it with a UsingStatement.
|
|
/// </summary>
|
|
public sealed class PatternStatementTransform : ContextTrackingVisitor<AstNode>, IAstTransform
|
|
{
|
|
public PatternStatementTransform(DecompilerContext context) : base(context)
|
|
{
|
|
}
|
|
|
|
#region Visitor Overrides
|
|
protected override AstNode VisitChildren(AstNode node, object data)
|
|
{
|
|
// Go through the children, and keep visiting a node as long as it changes.
|
|
// Because some transforms delete/replace nodes before and after the node being transformed, we rely
|
|
// on the transform's return value to know where we need to keep iterating.
|
|
for (AstNode child = node.FirstChild; child != null; child = child.NextSibling) {
|
|
AstNode oldChild;
|
|
do {
|
|
oldChild = child;
|
|
child = child.AcceptVisitor(this, data);
|
|
Debug.Assert(child != null && child.Parent == node);
|
|
} while (child != oldChild);
|
|
}
|
|
return node;
|
|
}
|
|
|
|
public override AstNode VisitExpressionStatement(ExpressionStatement expressionStatement, object data)
|
|
{
|
|
AstNode result;
|
|
if (context.Settings.UsingStatement) {
|
|
result = TransformUsings(expressionStatement);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
result = TransformFor(expressionStatement);
|
|
if (result != null)
|
|
return result;
|
|
if (context.Settings.LockStatement) {
|
|
result = TransformLock(expressionStatement);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
return base.VisitExpressionStatement(expressionStatement, data);
|
|
}
|
|
|
|
public override AstNode VisitUsingStatement(UsingStatement usingStatement, object data)
|
|
{
|
|
if (context.Settings.ForEachStatement) {
|
|
AstNode result = TransformForeach(usingStatement);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
return base.VisitUsingStatement(usingStatement, data);
|
|
}
|
|
|
|
public override AstNode VisitWhileStatement(WhileStatement whileStatement, object data)
|
|
{
|
|
return TransformDoWhile(whileStatement) ?? base.VisitWhileStatement(whileStatement, data);
|
|
}
|
|
|
|
public override AstNode VisitIfElseStatement(IfElseStatement ifElseStatement, object data)
|
|
{
|
|
if (context.Settings.SwitchStatementOnString) {
|
|
AstNode result = TransformSwitchOnString(ifElseStatement);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
return base.VisitIfElseStatement(ifElseStatement, data);
|
|
}
|
|
|
|
public override AstNode VisitPropertyDeclaration(PropertyDeclaration propertyDeclaration, object data)
|
|
{
|
|
if (context.Settings.AutomaticProperties) {
|
|
AstNode result = TransformAutomaticProperties(propertyDeclaration);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
return base.VisitPropertyDeclaration(propertyDeclaration, data);
|
|
}
|
|
|
|
public override AstNode VisitCustomEventDeclaration(CustomEventDeclaration eventDeclaration, object data)
|
|
{
|
|
// first apply transforms to the accessor bodies
|
|
base.VisitCustomEventDeclaration(eventDeclaration, data);
|
|
if (context.Settings.AutomaticEvents) {
|
|
AstNode result = TransformAutomaticEvents(eventDeclaration);
|
|
if (result != null)
|
|
return result;
|
|
}
|
|
return eventDeclaration;
|
|
}
|
|
|
|
public override AstNode VisitMethodDeclaration(MethodDeclaration methodDeclaration, object data)
|
|
{
|
|
return TransformDestructor(methodDeclaration) ?? base.VisitMethodDeclaration(methodDeclaration, data);
|
|
}
|
|
|
|
public override AstNode VisitTryCatchStatement(TryCatchStatement tryCatchStatement, object data)
|
|
{
|
|
return TransformTryCatchFinally(tryCatchStatement) ?? base.VisitTryCatchStatement(tryCatchStatement, data);
|
|
}
|
|
#endregion
|
|
|
|
/// <summary>
|
|
/// $type $variable = $initializer;
|
|
/// </summary>
|
|
static readonly AstNode variableAssignPattern = new ExpressionStatement(
|
|
new AssignmentExpression(
|
|
new NamedNode("variable", new IdentifierExpression()),
|
|
new AnyNode("initializer")
|
|
));
|
|
|
|
#region using
|
|
static readonly AstNode usingTryCatchPattern = new TryCatchStatement {
|
|
TryBlock = new AnyNode("body"),
|
|
FinallyBlock = new BlockStatement {
|
|
new Choice {
|
|
{ "valueType",
|
|
new ExpressionStatement(new NamedNode("ident", new IdentifierExpression()).ToExpression().Invoke("Dispose"))
|
|
},
|
|
{ "referenceType",
|
|
new IfElseStatement {
|
|
Condition = new BinaryOperatorExpression(
|
|
new NamedNode("ident", new IdentifierExpression()),
|
|
BinaryOperatorType.InEquality,
|
|
new NullReferenceExpression()
|
|
),
|
|
TrueStatement = new BlockStatement {
|
|
new ExpressionStatement(new Backreference("ident").ToExpression().Invoke("Dispose"))
|
|
}
|
|
}
|
|
}
|
|
}.ToStatement()
|
|
}
|
|
};
|
|
|
|
public UsingStatement TransformUsings(ExpressionStatement node)
|
|
{
|
|
Match m1 = variableAssignPattern.Match(node);
|
|
if (m1 == null) return null;
|
|
AstNode tryCatch = node.NextSibling;
|
|
Match m2 = usingTryCatchPattern.Match(tryCatch);
|
|
if (m2 == null) return null;
|
|
if (m1.Get<IdentifierExpression>("variable").Single().Identifier == m2.Get<IdentifierExpression>("ident").Single().Identifier) {
|
|
if (m2.Has("valueType")) {
|
|
// if there's no if(x!=null), then it must be a value type
|
|
ILVariable v = m1.Get("variable").Single().Annotation<ILVariable>();
|
|
if (v == null || v.Type == null || !v.Type.IsValueType)
|
|
return null;
|
|
}
|
|
BlockStatement body = m2.Get<BlockStatement>("body").Single();
|
|
UsingStatement usingStatement = new UsingStatement();
|
|
usingStatement.ResourceAcquisition = node.Detach();
|
|
usingStatement.EmbeddedStatement = body.Detach();
|
|
tryCatch.ReplaceWith(usingStatement);
|
|
// TODO: Move the variable declaration into the resource acquisition, if possible
|
|
// This is necessary for the foreach-pattern to work on the result of the using-pattern
|
|
return usingStatement;
|
|
}
|
|
return null;
|
|
}
|
|
#endregion
|
|
|
|
#region foreach
|
|
static readonly UsingStatement foreachPattern = new UsingStatement {
|
|
ResourceAcquisition = new VariableDeclarationStatement {
|
|
Type = new AnyNode("enumeratorType"),
|
|
Variables = {
|
|
new NamedNode(
|
|
"enumeratorVariable",
|
|
new VariableInitializer {
|
|
Initializer = new AnyNode("collection").ToExpression().Invoke("GetEnumerator")
|
|
}
|
|
)
|
|
}
|
|
},
|
|
EmbeddedStatement = new BlockStatement {
|
|
new Repeat(
|
|
new NamedNode("variableDeclaration", new VariableDeclarationStatement { Type = new AnyNode(), Variables = { new AnyNode() } })
|
|
).ToStatement(),
|
|
new WhileStatement {
|
|
Condition = new IdentifierExpressionBackreference("enumeratorVariable").ToExpression().Invoke("MoveNext"),
|
|
EmbeddedStatement = new BlockStatement {
|
|
new AssignmentExpression {
|
|
Left = new NamedNode("itemVariable", new IdentifierExpression()),
|
|
Operator = AssignmentOperatorType.Assign,
|
|
Right = new IdentifierExpressionBackreference("enumeratorVariable").ToExpression().Member("Current")
|
|
},
|
|
new Repeat(new AnyNode("statement")).ToStatement()
|
|
}
|
|
}
|
|
}};
|
|
|
|
public ForeachStatement TransformForeach(UsingStatement node)
|
|
{
|
|
Match m = foreachPattern.Match(node);
|
|
if (m == null)
|
|
return null;
|
|
VariableInitializer enumeratorVar = m.Get<VariableInitializer>("enumeratorVariable").Single();
|
|
ILVariable itemVar = m.Get("itemVariable").Single().Annotation<ILVariable>();
|
|
if (itemVar == null)
|
|
return null;
|
|
return null; // TODO
|
|
// if (m.Has("itemVariableInsideLoop") && itemVar.Annotation<DelegateConstruction.CapturedVariableAnnotation>() != null) {
|
|
// // cannot move captured variables out of loops
|
|
// return null;
|
|
// }
|
|
BlockStatement newBody = new BlockStatement();
|
|
foreach (Statement stmt in m.Get<Statement>("statement"))
|
|
newBody.Add(stmt.Detach());
|
|
|
|
ForeachStatement foreachStatement = new ForeachStatement {
|
|
VariableType = m.Get<AstType>("itemType").Single().Detach(),
|
|
VariableName = itemVar.Name,
|
|
InExpression = m.Get<Expression>("collection").Single().Detach(),
|
|
EmbeddedStatement = newBody
|
|
};
|
|
node.ReplaceWith(foreachStatement);
|
|
return foreachStatement;
|
|
}
|
|
#endregion
|
|
|
|
#region for
|
|
static readonly WhileStatement forPattern = new WhileStatement {
|
|
Condition = new BinaryOperatorExpression {
|
|
Left = new NamedNode("ident", new IdentifierExpression()),
|
|
Operator = BinaryOperatorType.Any,
|
|
Right = new AnyNode("endExpr")
|
|
},
|
|
EmbeddedStatement = new BlockStatement {
|
|
Statements = {
|
|
new Repeat(new AnyNode("statement")),
|
|
new NamedNode(
|
|
"increment",
|
|
new ExpressionStatement(
|
|
new AssignmentExpression {
|
|
Left = new Backreference("ident"),
|
|
Operator = AssignmentOperatorType.Any,
|
|
Right = new AnyNode()
|
|
}))
|
|
}
|
|
}};
|
|
|
|
public ForStatement TransformFor(ExpressionStatement node)
|
|
{
|
|
Match m1 = variableAssignPattern.Match(node);
|
|
if (m1 == null) return null;
|
|
AstNode next = node.NextSibling;
|
|
Match m2 = forPattern.Match(next);
|
|
if (m2 == null) return null;
|
|
// ensure the variable in the for pattern is the same as in the declaration
|
|
if (m1.Get<IdentifierExpression>("variable").Single().Identifier != m2.Get<IdentifierExpression>("ident").Single().Identifier)
|
|
return null;
|
|
WhileStatement loop = (WhileStatement)next;
|
|
node.Remove();
|
|
BlockStatement newBody = new BlockStatement();
|
|
foreach (Statement stmt in m2.Get<Statement>("statement"))
|
|
newBody.Add(stmt.Detach());
|
|
ForStatement forStatement = new ForStatement();
|
|
forStatement.Initializers.Add(node);
|
|
forStatement.Condition = loop.Condition.Detach();
|
|
forStatement.Iterators.Add(m2.Get<Statement>("increment").Single().Detach());
|
|
forStatement.EmbeddedStatement = newBody;
|
|
loop.ReplaceWith(forStatement);
|
|
return forStatement;
|
|
}
|
|
#endregion
|
|
|
|
#region doWhile
|
|
static readonly WhileStatement doWhilePattern = new WhileStatement {
|
|
Condition = new PrimitiveExpression(true),
|
|
EmbeddedStatement = new BlockStatement {
|
|
Statements = {
|
|
new Repeat(new AnyNode("statement")),
|
|
new IfElseStatement {
|
|
Condition = new AnyNode("condition"),
|
|
TrueStatement = new BlockStatement { new BreakStatement() }
|
|
}
|
|
}
|
|
}};
|
|
|
|
public DoWhileStatement TransformDoWhile(WhileStatement whileLoop)
|
|
{
|
|
Match m = doWhilePattern.Match(whileLoop);
|
|
if (m != null) {
|
|
DoWhileStatement doLoop = new DoWhileStatement();
|
|
doLoop.Condition = new UnaryOperatorExpression(UnaryOperatorType.Not, m.Get<Expression>("condition").Single().Detach());
|
|
doLoop.Condition.AcceptVisitor(new PushNegation(), null);
|
|
BlockStatement block = (BlockStatement)whileLoop.EmbeddedStatement;
|
|
block.Statements.Last().Remove(); // remove if statement
|
|
doLoop.EmbeddedStatement = block.Detach();
|
|
whileLoop.ReplaceWith(doLoop);
|
|
|
|
// we may have to extract variable definitions out of the loop if they were used in the condition:
|
|
foreach (var varDecl in block.Statements.OfType<VariableDeclarationStatement>()) {
|
|
VariableInitializer v = varDecl.Variables.Single();
|
|
if (doLoop.Condition.DescendantsAndSelf.OfType<IdentifierExpression>().Any(i => i.Identifier == v.Name)) {
|
|
AssignmentExpression assign = new AssignmentExpression(new IdentifierExpression(v.Name), v.Initializer.Detach());
|
|
// move annotations from v to assign:
|
|
assign.CopyAnnotationsFrom(v);
|
|
v.RemoveAnnotations<object>();
|
|
// remove varDecl with assignment; and move annotations from varDecl to the ExpressionStatement:
|
|
varDecl.ReplaceWith(new ExpressionStatement(assign).CopyAnnotationsFrom(varDecl));
|
|
varDecl.RemoveAnnotations<object>();
|
|
|
|
// insert the varDecl above the do-while loop:
|
|
doLoop.Parent.InsertChildBefore(doLoop, varDecl, BlockStatement.StatementRole);
|
|
}
|
|
}
|
|
return doLoop;
|
|
}
|
|
return null;
|
|
}
|
|
#endregion
|
|
|
|
#region lock
|
|
static readonly AstNode lockFlagInitPattern = new ExpressionStatement(
|
|
new AssignmentExpression(
|
|
new NamedNode("variable", new IdentifierExpression()),
|
|
new PrimitiveExpression(false)
|
|
));
|
|
|
|
static readonly AstNode lockTryCatchPattern = new TryCatchStatement {
|
|
TryBlock = new BlockStatement {
|
|
new TypePattern(typeof(System.Threading.Monitor)).ToType().Invoke(
|
|
"Enter", new AnyNode("enter"),
|
|
new DirectionExpression {
|
|
FieldDirection = FieldDirection.Ref,
|
|
Expression = new NamedNode("flag", new IdentifierExpression())
|
|
}),
|
|
new Repeat(new AnyNode()).ToStatement()
|
|
},
|
|
FinallyBlock = new BlockStatement {
|
|
new IfElseStatement {
|
|
Condition = new Backreference("flag"),
|
|
TrueStatement = new BlockStatement {
|
|
new TypePattern(typeof(System.Threading.Monitor)).ToType().Invoke("Exit", new NamedNode("exit", new IdentifierExpression()))
|
|
}
|
|
}
|
|
}};
|
|
|
|
public LockStatement TransformLock(ExpressionStatement node)
|
|
{
|
|
Match m1 = lockFlagInitPattern.Match(node);
|
|
if (m1 == null) return null;
|
|
AstNode tryCatch = node.NextSibling;
|
|
Match m2 = lockTryCatchPattern.Match(tryCatch);
|
|
if (m2 == null) return null;
|
|
if (m1.Get<IdentifierExpression>("variable").Single().Identifier == m2.Get<IdentifierExpression>("flag").Single().Identifier) {
|
|
Expression enter = m2.Get<Expression>("enter").Single();
|
|
IdentifierExpression exit = m2.Get<IdentifierExpression>("exit").Single();
|
|
if (exit.Match(enter) == null) {
|
|
// If exit and enter are not the same, then enter must be "exit = ..."
|
|
AssignmentExpression assign = enter as AssignmentExpression;
|
|
if (assign == null)
|
|
return null;
|
|
if (exit.Match(assign.Left) == null)
|
|
return null;
|
|
enter = assign.Right;
|
|
// Remove 'exit' variable:
|
|
bool ok = false;
|
|
for (AstNode tmp = node.NextSibling; tmp != tryCatch; tmp = tmp.NextSibling) {
|
|
VariableDeclarationStatement v = (VariableDeclarationStatement)tmp;
|
|
if (v.Variables.Single().Name == exit.Identifier) {
|
|
ok = true;
|
|
v.Remove();
|
|
break;
|
|
}
|
|
}
|
|
if (!ok)
|
|
return null;
|
|
}
|
|
// transform the code into a lock statement:
|
|
LockStatement l = new LockStatement();
|
|
l.Expression = enter.Detach();
|
|
l.EmbeddedStatement = ((TryCatchStatement)tryCatch).TryBlock.Detach();
|
|
((BlockStatement)l.EmbeddedStatement).Statements.First().Remove(); // Remove 'Enter()' call
|
|
tryCatch.ReplaceWith(l);
|
|
node.Remove(); // remove flag variable
|
|
return l;
|
|
}
|
|
return null;
|
|
}
|
|
#endregion
|
|
|
|
#region switch on strings
|
|
static readonly IfElseStatement switchOnStringPattern = new IfElseStatement {
|
|
Condition = new BinaryOperatorExpression {
|
|
Left = new AnyNode("switchExpr"),
|
|
Operator = BinaryOperatorType.InEquality,
|
|
Right = new NullReferenceExpression()
|
|
},
|
|
TrueStatement = new BlockStatement {
|
|
new IfElseStatement {
|
|
Condition = new BinaryOperatorExpression {
|
|
Left = new AnyNode("cachedDict"),
|
|
Operator = BinaryOperatorType.Equality,
|
|
Right = new NullReferenceExpression()
|
|
},
|
|
TrueStatement = new AnyNode("dictCreation")
|
|
},
|
|
new VariableDeclarationStatement {
|
|
Type = new PrimitiveType("int"),
|
|
Variables = { new NamedNode("intVar", new VariableInitializer()) }
|
|
},
|
|
new IfElseStatement {
|
|
Condition = new Backreference("cachedDict").ToExpression().Invoke(
|
|
"TryGetValue",
|
|
new NamedNode("switchVar", new IdentifierExpression()),
|
|
new DirectionExpression {
|
|
FieldDirection = FieldDirection.Out,
|
|
Expression = new IdentifierExpressionBackreference("intVar")
|
|
}),
|
|
TrueStatement = new BlockStatement {
|
|
Statements = {
|
|
new NamedNode(
|
|
"switch", new SwitchStatement {
|
|
Expression = new IdentifierExpressionBackreference("intVar"),
|
|
SwitchSections = { new Repeat(new AnyNode()) }
|
|
})
|
|
}
|
|
}
|
|
},
|
|
new Repeat(new AnyNode("nonNullDefaultStmt")).ToStatement()
|
|
},
|
|
FalseStatement = new OptionalNode("nullStmt", new BlockStatement { Statements = { new Repeat(new AnyNode()) } })
|
|
};
|
|
|
|
public SwitchStatement TransformSwitchOnString(IfElseStatement node)
|
|
{
|
|
Match m = switchOnStringPattern.Match(node);
|
|
if (m == null)
|
|
return null;
|
|
if (m.Has("nonNullDefaultStmt") && !m.Has("nullStmt"))
|
|
return null;
|
|
// switchVar must be the same as switchExpr; or switchExpr must be an assignment and switchVar the left side of that assignment
|
|
if (m.Get("switchVar").Single().Match(m.Get("switchExpr").Single()) == null) {
|
|
AssignmentExpression assign = m.Get("switchExpr").Single() as AssignmentExpression;
|
|
if (m.Get("switchVar").Single().Match(assign.Left) == null)
|
|
return null;
|
|
}
|
|
FieldReference cachedDictField = m.Get("cachedDict").Single().Annotation<FieldReference>();
|
|
if (cachedDictField == null || !cachedDictField.DeclaringType.Name.StartsWith("<PrivateImplementationDetails>", StringComparison.Ordinal))
|
|
return null;
|
|
List<Statement> dictCreation = m.Get<BlockStatement>("dictCreation").Single().Statements.ToList();
|
|
List<KeyValuePair<string, int>> dict = BuildDictionary(dictCreation);
|
|
SwitchStatement sw = m.Get<SwitchStatement>("switch").Single();
|
|
sw.Expression = m.Get<Expression>("switchExpr").Single().Detach();
|
|
foreach (SwitchSection section in sw.SwitchSections) {
|
|
List<CaseLabel> labels = section.CaseLabels.ToList();
|
|
section.CaseLabels.Clear();
|
|
foreach (CaseLabel label in labels) {
|
|
PrimitiveExpression expr = label.Expression as PrimitiveExpression;
|
|
if (expr == null || !(expr.Value is int))
|
|
continue;
|
|
int val = (int)expr.Value;
|
|
foreach (var pair in dict) {
|
|
if (pair.Value == val)
|
|
section.CaseLabels.Add(new CaseLabel { Expression = new PrimitiveExpression(pair.Key) });
|
|
}
|
|
}
|
|
}
|
|
if (m.Has("nullStmt")) {
|
|
SwitchSection section = new SwitchSection();
|
|
section.CaseLabels.Add(new CaseLabel { Expression = new NullReferenceExpression() });
|
|
BlockStatement block = m.Get<BlockStatement>("nullStmt").Single();
|
|
block.Statements.Add(new BreakStatement());
|
|
section.Statements.Add(block.Detach());
|
|
sw.SwitchSections.Add(section);
|
|
if (m.Has("nonNullDefaultStmt")) {
|
|
section = new SwitchSection();
|
|
section.CaseLabels.Add(new CaseLabel());
|
|
block = new BlockStatement();
|
|
block.Statements.AddRange(m.Get<Statement>("nonNullDefaultStmt").Select(s => s.Detach()));
|
|
block.Add(new BreakStatement());
|
|
section.Statements.Add(block);
|
|
sw.SwitchSections.Add(section);
|
|
}
|
|
}
|
|
node.ReplaceWith(sw);
|
|
return sw;
|
|
}
|
|
|
|
List<KeyValuePair<string, int>> BuildDictionary(List<Statement> dictCreation)
|
|
{
|
|
List<KeyValuePair<string, int>> dict = new List<KeyValuePair<string, int>>();
|
|
for (int i = 0; i < dictCreation.Count; i++) {
|
|
ExpressionStatement es = dictCreation[i] as ExpressionStatement;
|
|
if (es == null)
|
|
continue;
|
|
InvocationExpression ie = es.Expression as InvocationExpression;
|
|
if (ie == null)
|
|
continue;
|
|
PrimitiveExpression arg1 = ie.Arguments.ElementAtOrDefault(0) as PrimitiveExpression;
|
|
PrimitiveExpression arg2 = ie.Arguments.ElementAtOrDefault(1) as PrimitiveExpression;
|
|
if (arg1 != null && arg2 != null && arg1.Value is string && arg2.Value is int)
|
|
dict.Add(new KeyValuePair<string, int>((string)arg1.Value, (int)arg2.Value));
|
|
}
|
|
return dict;
|
|
}
|
|
#endregion
|
|
|
|
#region Automatic Properties
|
|
static readonly PropertyDeclaration automaticPropertyPattern = new PropertyDeclaration {
|
|
Attributes = { new Repeat(new AnyNode()) },
|
|
Modifiers = Modifiers.Any,
|
|
ReturnType = new AnyNode(),
|
|
Getter = new Accessor {
|
|
Attributes = { new Repeat(new AnyNode()) },
|
|
Modifiers = Modifiers.Any,
|
|
Body = new BlockStatement {
|
|
new ReturnStatement {
|
|
Expression = new NamedNode("fieldReference", new MemberReferenceExpression { Target = new ThisReferenceExpression() })
|
|
}
|
|
}
|
|
},
|
|
Setter = new Accessor {
|
|
Attributes = { new Repeat(new AnyNode()) },
|
|
Modifiers = Modifiers.Any,
|
|
Body = new BlockStatement {
|
|
new AssignmentExpression {
|
|
Left = new Backreference("fieldReference"),
|
|
Right = new IdentifierExpression("value")
|
|
}
|
|
}}};
|
|
|
|
PropertyDeclaration TransformAutomaticProperties(PropertyDeclaration property)
|
|
{
|
|
PropertyDefinition cecilProperty = property.Annotation<PropertyDefinition>();
|
|
if (cecilProperty == null || cecilProperty.GetMethod == null || cecilProperty.SetMethod == null)
|
|
return null;
|
|
if (!(cecilProperty.GetMethod.IsCompilerGenerated() && cecilProperty.SetMethod.IsCompilerGenerated()))
|
|
return null;
|
|
Match m = automaticPropertyPattern.Match(property);
|
|
if (m != null) {
|
|
FieldDefinition field = m.Get("fieldReference").Single().Annotation<FieldReference>().ResolveWithinSameModule();
|
|
if (field.IsCompilerGenerated()) {
|
|
RemoveCompilerGeneratedAttribute(property.Getter.Attributes);
|
|
RemoveCompilerGeneratedAttribute(property.Setter.Attributes);
|
|
property.Getter.Body = null;
|
|
property.Setter.Body = null;
|
|
}
|
|
}
|
|
// Since the event instance is not changed, we can continue in the visitor as usual, so return null
|
|
return null;
|
|
}
|
|
|
|
void RemoveCompilerGeneratedAttribute(AstNodeCollection<AttributeSection> attributeSections)
|
|
{
|
|
foreach (AttributeSection section in attributeSections) {
|
|
foreach (var attr in section.Attributes) {
|
|
TypeReference tr = attr.Type.Annotation<TypeReference>();
|
|
if (tr != null && tr.Namespace == "System.Runtime.CompilerServices" && tr.Name == "CompilerGeneratedAttribute") {
|
|
attr.Remove();
|
|
}
|
|
}
|
|
if (section.Attributes.Count == 0)
|
|
section.Remove();
|
|
}
|
|
}
|
|
#endregion
|
|
|
|
#region Automatic Events
|
|
static readonly Accessor automaticEventPatternV4 = new Accessor {
|
|
Body = new BlockStatement {
|
|
new VariableDeclarationStatement { Type = new AnyNode("type"), Variables = { new AnyNode() } },
|
|
new VariableDeclarationStatement { Type = new Backreference("type"), Variables = { new AnyNode() } },
|
|
new VariableDeclarationStatement { Type = new Backreference("type"), Variables = { new AnyNode() } },
|
|
new AssignmentExpression {
|
|
Left = new NamedNode("var1", new IdentifierExpression()),
|
|
Operator = AssignmentOperatorType.Assign,
|
|
Right = new NamedNode("field", new MemberReferenceExpression { Target = new ThisReferenceExpression() })
|
|
},
|
|
new DoWhileStatement {
|
|
EmbeddedStatement = new BlockStatement {
|
|
new AssignmentExpression(new NamedNode("var2", new IdentifierExpression()), new IdentifierExpressionBackreference("var1")),
|
|
new AssignmentExpression {
|
|
Left = new NamedNode("var3", new IdentifierExpression()),
|
|
Operator = AssignmentOperatorType.Assign,
|
|
Right = new AnyNode("delegateCombine").ToExpression().Invoke(
|
|
new IdentifierExpressionBackreference("var2"),
|
|
new IdentifierExpression("value")
|
|
).CastTo(new Backreference("type"))
|
|
},
|
|
new AssignmentExpression {
|
|
Left = new IdentifierExpressionBackreference("var1"),
|
|
Right = new TypePattern(typeof(System.Threading.Interlocked)).ToType().Invoke(
|
|
"CompareExchange",
|
|
new AstType[] { new Backreference("type") }, // type argument
|
|
new Expression[] { // arguments
|
|
new DirectionExpression { FieldDirection = FieldDirection.Ref, Expression = new Backreference("field") },
|
|
new IdentifierExpressionBackreference("var3"),
|
|
new IdentifierExpressionBackreference("var2")
|
|
}
|
|
)}
|
|
},
|
|
Condition = new BinaryOperatorExpression {
|
|
Left = new IdentifierExpressionBackreference("var1"),
|
|
Operator = BinaryOperatorType.InEquality,
|
|
Right = new IdentifierExpressionBackreference("var2")
|
|
}}
|
|
}};
|
|
|
|
bool CheckAutomaticEventV4Match(Match m, CustomEventDeclaration ev, bool isAddAccessor)
|
|
{
|
|
if (m == null)
|
|
return false;
|
|
if (m.Get<MemberReferenceExpression>("field").Single().MemberName != ev.Name)
|
|
return false; // field name must match event name
|
|
if (ev.ReturnType.Match(m.Get("type").Single()) == null)
|
|
return false; // variable types must match event type
|
|
var combineMethod = m.Get("delegateCombine").Single().Parent.Annotation<MethodReference>();
|
|
if (combineMethod == null || combineMethod.Name != (isAddAccessor ? "Combine" : "Remove"))
|
|
return false;
|
|
return combineMethod.DeclaringType.FullName == "System.Delegate";
|
|
}
|
|
|
|
EventDeclaration TransformAutomaticEvents(CustomEventDeclaration ev)
|
|
{
|
|
Match m1 = automaticEventPatternV4.Match(ev.AddAccessor);
|
|
if (!CheckAutomaticEventV4Match(m1, ev, true))
|
|
return null;
|
|
Match m2 = automaticEventPatternV4.Match(ev.RemoveAccessor);
|
|
if (!CheckAutomaticEventV4Match(m2, ev, false))
|
|
return null;
|
|
EventDeclaration ed = new EventDeclaration();
|
|
ev.Attributes.MoveTo(ed.Attributes);
|
|
ed.ReturnType = ev.ReturnType.Detach();
|
|
ed.Modifiers = ev.Modifiers;
|
|
ed.Variables.Add(new VariableInitializer(ev.Name));
|
|
ed.CopyAnnotationsFrom(ev);
|
|
|
|
EventDefinition eventDef = ev.Annotation<EventDefinition>();
|
|
if (eventDef != null) {
|
|
FieldDefinition field = eventDef.DeclaringType.Fields.FirstOrDefault(f => f.Name == ev.Name);
|
|
if (field != null) {
|
|
ed.AddAnnotation(field);
|
|
AstBuilder.ConvertAttributes(ed, field, AttributeTarget.Field);
|
|
}
|
|
}
|
|
|
|
ev.ReplaceWith(ed);
|
|
return ed;
|
|
}
|
|
#endregion
|
|
|
|
#region Destructor
|
|
static readonly MethodDeclaration destructorPattern = new MethodDeclaration {
|
|
Attributes = { new Repeat(new AnyNode()) },
|
|
Modifiers = Modifiers.Any,
|
|
ReturnType = new PrimitiveType("void"),
|
|
Name = "Finalize",
|
|
Body = new BlockStatement {
|
|
new TryCatchStatement {
|
|
TryBlock = new AnyNode("body"),
|
|
FinallyBlock = new BlockStatement {
|
|
new BaseReferenceExpression().Invoke("Finalize")
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
DestructorDeclaration TransformDestructor(MethodDeclaration methodDef)
|
|
{
|
|
Match m = destructorPattern.Match(methodDef);
|
|
if (m != null) {
|
|
DestructorDeclaration dd = new DestructorDeclaration();
|
|
methodDef.Attributes.MoveTo(dd.Attributes);
|
|
dd.Modifiers = methodDef.Modifiers & ~(Modifiers.Protected | Modifiers.Override);
|
|
dd.Body = m.Get<BlockStatement>("body").Single().Detach();
|
|
dd.Name = AstBuilder.CleanName(context.CurrentType.Name);
|
|
methodDef.ReplaceWith(dd);
|
|
return dd;
|
|
}
|
|
return null;
|
|
}
|
|
#endregion
|
|
|
|
#region Try-Catch-Finally
|
|
static readonly TryCatchStatement tryCatchFinallyPattern = new TryCatchStatement {
|
|
TryBlock = new BlockStatement {
|
|
new TryCatchStatement {
|
|
TryBlock = new AnyNode(),
|
|
CatchClauses = { new Repeat(new AnyNode()) }
|
|
}
|
|
},
|
|
FinallyBlock = new AnyNode()
|
|
};
|
|
|
|
/// <summary>
|
|
/// Simplify nested 'try { try {} catch {} } finally {}'.
|
|
/// This transformation must run after the using/lock tranformations.
|
|
/// </summary>
|
|
TryCatchStatement TransformTryCatchFinally(TryCatchStatement tryFinally)
|
|
{
|
|
if (tryCatchFinallyPattern.Match(tryFinally) != null) {
|
|
TryCatchStatement tryCatch = (TryCatchStatement)tryFinally.TryBlock.Statements.Single();
|
|
tryFinally.TryBlock = tryCatch.TryBlock.Detach();
|
|
tryCatch.CatchClauses.MoveTo(tryFinally.CatchClauses);
|
|
}
|
|
// Since the tryFinally instance is not changed, we can continue in the visitor as usual, so return null
|
|
return null;
|
|
}
|
|
#endregion
|
|
|
|
#region Pattern Matching Helpers
|
|
sealed class TypePattern : Pattern
|
|
{
|
|
readonly string ns;
|
|
readonly string name;
|
|
|
|
public TypePattern(Type type)
|
|
{
|
|
this.ns = type.Namespace;
|
|
this.name = type.Name;
|
|
}
|
|
|
|
protected override bool DoMatch(AstNode other, Match match)
|
|
{
|
|
if (other == null)
|
|
return false;
|
|
TypeReference tr = other.Annotation<TypeReference>();
|
|
return tr != null && tr.Namespace == ns && tr.Name == name;
|
|
}
|
|
|
|
public override S AcceptVisitor<T, S>(IAstVisitor<T, S> visitor, T data)
|
|
{
|
|
throw new NotImplementedException();
|
|
}
|
|
}
|
|
#endregion
|
|
}
|
|
}
|