public VariableDescription(AALocalDecl localDecl, VariableTypes type)
{
Name = localDecl.GetName().Text;
Type = Util.TypeToString(localDecl.GetType());
switch (type)
{
case VariableTypes.LocalVariable:
PlacementPrefix = "Local";
break;
case VariableTypes.Parameter:
PlacementPrefix = "Parameter";
break;
case VariableTypes.StructVariable:
PlacementPrefix = "Struct field";
break;
default:
PlacementPrefix = "";
break;
}
VariableType = type;
Const = localDecl.GetConst() != null;
IsStatic = localDecl.GetStatic() != null;
Visibility = localDecl.GetVisibilityModifier();
realType = (PType) localDecl.GetType().Clone();
init = localDecl.GetInit();
Line = localDecl.GetName().Line;
Position = TextPoint.FromCompilerCoords(localDecl.GetName());
}
public override void OutAALocalDecl(AALocalDecl node)
{
if (!Util.HasAncestor<AABlock>(node) && !Util.HasAncestor<AMethodDecl>(node))
{
//OutStructFieldDecl(node);
return;
}
if (node.GetInit() != null)
return;
AABlock pBlock;
int insertIndex;
PLvalue lvalue;
if (Util.HasAncestor<AABlock>(node))
{
//A local variable
pBlock = Util.GetAncestor<AABlock>(node);
insertIndex = pBlock.GetStatements().IndexOf(Util.GetAncestor<PStm>(node)) + 1;
lvalue = new ALocalLvalue(new TIdentifier(node.GetName().Text));
data.LocalLinks[(ALocalLvalue) lvalue] = node;
data.LvalueTypes[lvalue] = node.GetType();
}
else
{
//Parameter
//Parameters will be set from the caller
return;
pBlock = (AABlock) Util.GetAncestor<AMethodDecl>(node).GetBlock();
insertIndex = 0;
lvalue = new ALocalLvalue(new TIdentifier(node.GetName().Text));
data.LocalLinks[(ALocalLvalue)lvalue] = node;
data.LvalueTypes[lvalue] = node.GetType();
}
AABlock block = new AABlock(new ArrayList(), new TRBrace("}"));
MakeAssignments(block, node.GetType(), lvalue, true);
if (block.GetStatements().Count != 0)
pBlock.GetStatements().Insert(insertIndex, new ABlockStm(new TLBrace("{"), block));
}
public override void CaseAALocalDecl(AALocalDecl node)
{
if (node.GetConst() != null) Write("const ");
node.GetType().Apply(this);
Write(" " + node.GetName().Text);
if (node.GetInit() != null)
{
Write(" = ");
node.GetInit().Apply(this);
}
}
//Convert struct variables to a collection of local variables
public override void CaseAALocalDecl(AALocalDecl node)
{
if (node.GetType() is ANamedType && data.StructTypeLinks.ContainsKey((ANamedType) node.GetType()) && Util.HasAncestor<PStm>(node))
{
//Can not have init - it would be bulk copy
AStructDecl str = data.StructTypeLinks[(ANamedType) node.GetType()];
Dictionary<AALocalDecl, AALocalDecl> variableMap = new Dictionary<AALocalDecl, AALocalDecl>();
PStm pStm = (PStm) node.Parent();
AABlock pBlock = (AABlock) pStm.Parent();
foreach (AALocalDecl structLocal in str.GetLocals())
{
AALocalDecl replacementLocal = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
Util.MakeClone(structLocal.GetType(), data),
new TIdentifier(node.GetName().Text + "_" +
structLocal.GetName().Text), null);
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(pStm), new ALocalDeclStm(new TSemicolon(";"), replacementLocal));
AALocalDecl baseLocal = structLocal;
if (data.EnheritanceLocalMap.ContainsKey(baseLocal))
baseLocal = data.EnheritanceLocalMap[baseLocal];
List<AALocalDecl> localsToAdd = new List<AALocalDecl>();
localsToAdd.AddRange(data.EnheritanceLocalMap.Where(pair => pair.Value == baseLocal).Select(pair => pair.Key));
localsToAdd.Add(baseLocal);
foreach (AALocalDecl localDecl in localsToAdd)
{
variableMap[localDecl] = replacementLocal;
}
}
List<ALocalLvalue> uses = new List<ALocalLvalue>();
uses.AddRange(data.LocalLinks.Where(k => k.Value == node && Util.GetAncestor<AAProgram>(k.Key) != null).Select(k => k.Key));
foreach (ALocalLvalue lvalue in uses)
{
AStructLvalue structLocalRef = (AStructLvalue) lvalue.Parent().Parent();
AALocalDecl replacementLocal = variableMap[data.StructFieldLinks[structLocalRef]];
ALocalLvalue replacementLvalue = new ALocalLvalue(new TIdentifier(replacementLocal.GetName().Text));
data.LocalLinks[replacementLvalue] = replacementLocal;
data.LvalueTypes[replacementLvalue] = replacementLocal.GetType();
structLocalRef.ReplaceBy(replacementLvalue);
}
foreach (AALocalDecl replacementLocal in variableMap.Select(k => k.Value))
{
replacementLocal.Apply(this);
}
}
base.CaseAALocalDecl(node);
}
private AMethodDecl CreateStringDeobfuscator()
{
AASourceFile file = (AASourceFile) finalTrans.mainEntry.Parent();
//Create fields for the string constants
AStringConstExp emptyStringConst = new AStringConstExp(new TStringLiteral("\"\""));
AFieldDecl emptyStringField = new AFieldDecl(new APublicVisibilityModifier(), null, new TConst("const"),
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("fOobar"), emptyStringConst);
file.GetDecl().Add(emptyStringField);
data.Fields.Add(new SharedData.DeclItem<AFieldDecl>(file, emptyStringField));
AFieldLvalue emptyStringRef1 = new AFieldLvalue(new TIdentifier(emptyStringField.GetName().Text));
AFieldLvalue emptyStringRef2 = new AFieldLvalue(new TIdentifier(emptyStringField.GetName().Text));
AFieldLvalue emptyStringRef3 = new AFieldLvalue(new TIdentifier(emptyStringField.GetName().Text));
ALvalueExp emptyStringRef1Exp = new ALvalueExp(emptyStringRef1);
ALvalueExp emptyStringRef2Exp = new ALvalueExp(emptyStringRef2);
ALvalueExp emptyStringRef3Exp = new ALvalueExp(emptyStringRef3);
AStringConstExp colonStringConst = new AStringConstExp(new TStringLiteral("\":\""));
AFieldDecl colonStringField = new AFieldDecl(new APublicVisibilityModifier(), null, new TConst("const"),
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("foObar"), colonStringConst);
file.GetDecl().Add(colonStringField);
data.Fields.Add(new SharedData.DeclItem<AFieldDecl>(file, colonStringField));
AFieldLvalue colonStringRef = new AFieldLvalue(new TIdentifier(colonStringField.GetName().Text));
ALvalueExp colonStringRefExp = new ALvalueExp(colonStringRef);
/*
string output = "";
string ch;
int length = StringLength(s);
int phase1 = (length - 1)%3;
int phase2 = (length - 1)%2;
*/
AALocalDecl stringParam = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("fo0bar"), emptyStringRef1Exp);
ALocalLvalue stringParamRef1 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef2 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef3 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef4 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef5 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef6 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef7 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALocalLvalue stringParamRef8 = new ALocalLvalue(new TIdentifier(stringParam.GetName().Text));
ALvalueExp stringParamRef1Exp = new ALvalueExp(stringParamRef1);
ALvalueExp stringParamRef2Exp = new ALvalueExp(stringParamRef2);
ALvalueExp stringParamRef4Exp = new ALvalueExp(stringParamRef4);
ALvalueExp stringParamRef5Exp = new ALvalueExp(stringParamRef5);
ALvalueExp stringParamRef7Exp = new ALvalueExp(stringParamRef7);
ALvalueExp stringParamRef8Exp = new ALvalueExp(stringParamRef8);
AABlock methodBlock = new AABlock(new ArrayList(), new TRBrace("}"));
AALocalDecl outputDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("foobar"), emptyStringRef1Exp);
methodBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), outputDecl));
ALocalLvalue outputRef1 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef2 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef3 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef4 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef5 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef6 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef7 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef8 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef9 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef10 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef11 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef12 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALocalLvalue outputRef13 = new ALocalLvalue(new TIdentifier(outputDecl.GetName().Text));
ALvalueExp outputRef2Exp = new ALvalueExp(outputRef2);
ALvalueExp outputRef4Exp = new ALvalueExp(outputRef4);
ALvalueExp outputRef5Exp = new ALvalueExp(outputRef5);
ALvalueExp outputRef6Exp = new ALvalueExp(outputRef6);
ALvalueExp outputRef7Exp = new ALvalueExp(outputRef7);
ALvalueExp outputRef8Exp = new ALvalueExp(outputRef8);
ALvalueExp outputRef10Exp = new ALvalueExp(outputRef10);
ALvalueExp outputRef11Exp = new ALvalueExp(outputRef11);
ALvalueExp outputRef12Exp = new ALvalueExp(outputRef12);
ALvalueExp outputRef13Exp = new ALvalueExp(outputRef13);
AALocalDecl chDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("f0obar"), null);
methodBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), chDecl));
ALocalLvalue chRef1 = new ALocalLvalue(new TIdentifier(chDecl.GetName().Text));
ALocalLvalue chRef2 = new ALocalLvalue(new TIdentifier(chDecl.GetName().Text));
ALocalLvalue chRef3 = new ALocalLvalue(new TIdentifier(chDecl.GetName().Text));
ALocalLvalue chRef4 = new ALocalLvalue(new TIdentifier(chDecl.GetName().Text));
ALocalLvalue chRef5 = new ALocalLvalue(new TIdentifier(chDecl.GetName().Text));
ALvalueExp chRef3Exp = new ALvalueExp(chRef3);
ALvalueExp chRef4Exp = new ALvalueExp(chRef4);
ALvalueExp chRef5Exp = new ALvalueExp(chRef5);
ASimpleInvokeExp stringLengthInvoke1 = new ASimpleInvokeExp(new TIdentifier("StringLength"),
new ArrayList() { stringParamRef1Exp });
AALocalDecl lengthDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("int"), null),
new TIdentifier("f0Obar"), stringLengthInvoke1);
methodBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), lengthDecl));
ALocalLvalue lengthRef1 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef2 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef3 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef4 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef5 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef6 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef7 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALocalLvalue lengthRef8 = new ALocalLvalue(new TIdentifier(lengthDecl.GetName().Text));
ALvalueExp lengthRef1Exp = new ALvalueExp(lengthRef1);
ALvalueExp lengthRef2Exp = new ALvalueExp(lengthRef2);
ALvalueExp lengthRef3Exp = new ALvalueExp(lengthRef3);
ALvalueExp lengthRef4Exp = new ALvalueExp(lengthRef4);
ALvalueExp lengthRef5Exp = new ALvalueExp(lengthRef5);
ALvalueExp lengthRef6Exp = new ALvalueExp(lengthRef6);
ALvalueExp lengthRef7Exp = new ALvalueExp(lengthRef7);
AIntConstExp intConstp1Init1 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConstp1Init2 = new AIntConstExp(new TIntegerLiteral("3"));
ABinopExp binopExpP1InitMinus = new ABinopExp(lengthRef1Exp, new AMinusBinop(new TMinus("-")), intConstp1Init1);
ABinopExp binopExpP1InitMod = new ABinopExp(binopExpP1InitMinus, new AModuloBinop(new TMod("%")), intConstp1Init2);
AALocalDecl phase1Decl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("int"), null),
new TIdentifier("fO0bar"), binopExpP1InitMod);
methodBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), phase1Decl));
ALocalLvalue phase1Ref1 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref2 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref3 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref4 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref5 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref6 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref7 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref8 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref9 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref10 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALocalLvalue phase1Ref11 = new ALocalLvalue(new TIdentifier(phase1Decl.GetName().Text));
ALvalueExp phase1Ref1Exp = new ALvalueExp(phase1Ref1);
ALvalueExp phase1Ref2Exp = new ALvalueExp(phase1Ref2);
ALvalueExp phase1Ref4Exp = new ALvalueExp(phase1Ref4);
ALvalueExp phase1Ref5Exp = new ALvalueExp(phase1Ref5);
ALvalueExp phase1Ref7Exp = new ALvalueExp(phase1Ref7);
ALvalueExp phase1Ref9Exp = new ALvalueExp(phase1Ref9);
ALvalueExp phase1Ref10Exp = new ALvalueExp(phase1Ref10);
ALvalueExp phase1Ref11Exp = new ALvalueExp(phase1Ref11);
AIntConstExp intConstp2Init1 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConstp2Init2 = new AIntConstExp(new TIntegerLiteral("2"));
ABinopExp binopExpP2InitMinus = new ABinopExp(lengthRef2Exp, new AMinusBinop(new TMinus("-")), intConstp2Init1);
ABinopExp binopExpP2InitMod = new ABinopExp(binopExpP2InitMinus, new AModuloBinop(new TMod("%")), intConstp2Init2);
AALocalDecl phase2Decl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("int"), null),
new TIdentifier("carl"), binopExpP2InitMod);
methodBlock.GetStatements().Add(new ALocalDeclStm(new TSemicolon(";"), phase2Decl));
ALocalLvalue phase2Ref1 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref2 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref3 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref4 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref5 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref6 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref7 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref8 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALocalLvalue phase2Ref9 = new ALocalLvalue(new TIdentifier(phase2Decl.GetName().Text));
ALvalueExp phase2Ref1Exp = new ALvalueExp(phase2Ref1);
ALvalueExp phase2Ref2Exp = new ALvalueExp(phase2Ref2);
ALvalueExp phase2Ref4Exp = new ALvalueExp(phase2Ref4);
ALvalueExp phase2Ref5Exp = new ALvalueExp(phase2Ref5);
ALvalueExp phase2Ref7Exp = new ALvalueExp(phase2Ref7);
ALvalueExp phase2Ref9Exp = new ALvalueExp(phase2Ref9);
/*
while(length > 0)
{
if(phase2 == 0)
{
ch = StringSub(s, 1, 1);
s = StringReplace(s, "", 1, 1);
}
else
{
if(phase2 == 1)
{
ch = StringSub(s, length, length);
s = StringReplace(s, "", length, length);
}
}
if(phase1 == 0)
{
output = ch + output;
}
else
{
if(phase1 == 1)
{
output = StringSub(output, 1, (StringLength(output) + 1)/2) + ch + StringSub(output, (StringLength(output) + 1)/2 + 1, StringLength(output));
}
else
{
output = output + ch;
}
}
phase1 = phase1 - 1;
if(phase1 < 0)
{
phase1 = phase1 + 3;
}
phase2 = phase2 - 1;
if(phase2 < 0)
{
phase2 = phase2 + 2;
}
length = StringLength(s);
}
*/
AABlock whileBlock = new AABlock(new ArrayList(), new TRBrace("}"));
AIntConstExp intConstWhileCond = new AIntConstExp(new TIntegerLiteral("0"));
ABinopExp binopWhileCond = new ABinopExp(lengthRef3Exp, new AGtBinop(new TGt(">")), intConstWhileCond);
methodBlock.GetStatements().Add(new AWhileStm(new TLParen("("), binopWhileCond,
new ABlockStm(new TLBrace("{"), whileBlock)));
/*
if(phase2 == 0)
{
ch = StringSub(s, 1, 1);
s = StringReplace(s, "", 1, 1);
}
else
{
if(phase2 == 1)
{
ch = StringSub(s, length, length);
s = StringReplace(s, "", length, length);
}
}
*/
AIntConstExp intConstIf1Cond = new AIntConstExp(new TIntegerLiteral("0"));
ABinopExp binopIf1Cond = new ABinopExp(phase2Ref1Exp, new AEqBinop(new TEq("==")), intConstIf1Cond);
AABlock thenBlock = new AABlock();
AABlock elseBlock = new AABlock();
whileBlock.GetStatements().Add(new AIfThenElseStm(new TLParen("("), binopIf1Cond,
new ABlockStm(new TLBrace("{"), thenBlock),
new ABlockStm(new TLBrace("{"), elseBlock)));
//ch = StringSub(s, 1, 1);
AIntConstExp intConst1 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst2 = new AIntConstExp(new TIntegerLiteral("1"));
ASimpleInvokeExp invokeStringSub1 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() {stringParamRef2Exp, intConst1, intConst2});
AAssignmentExp assignment1 = new AAssignmentExp(new TAssign("="), chRef1, invokeStringSub1);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment1));
//s = StringReplace(s, "", 1, 1);
AIntConstExp intConst3 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst4 = new AIntConstExp(new TIntegerLiteral("1"));
ASimpleInvokeExp invokeStringReplace1 = new ASimpleInvokeExp(new TIdentifier("StringReplace"),
new ArrayList() { stringParamRef4Exp, emptyStringRef2Exp, intConst3, intConst4 });
AAssignmentExp assignment2 = new AAssignmentExp(new TAssign("="), stringParamRef3, invokeStringReplace1);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment2));
//if(phase2 == 1)
AIntConstExp intConst5 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop1 = new ABinopExp(phase2Ref2Exp, new AEqBinop(new TEq("==")), intConst5);
thenBlock = new AABlock();
elseBlock.GetStatements().Add(new AIfThenStm(new TLParen("("), binop1,
new ABlockStm(new TLBrace("{"), thenBlock)));
//ch = StringSub(s, length, length);
ASimpleInvokeExp invokeStringSub2 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() { stringParamRef5Exp, lengthRef3Exp, lengthRef4Exp });
AAssignmentExp assignment3 = new AAssignmentExp(new TAssign("="), chRef2, invokeStringSub2);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment3));
//s = StringReplace(s, "", length, length);
ASimpleInvokeExp invokeStringReplace2 = new ASimpleInvokeExp(new TIdentifier("StringReplace"),
new ArrayList() { stringParamRef7Exp, emptyStringRef3Exp, lengthRef5Exp, lengthRef6Exp });
AAssignmentExp assignment4 = new AAssignmentExp(new TAssign("="), stringParamRef6, invokeStringReplace2);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment4));
//if(phase1 == 0)
AIntConstExp intConst6 = new AIntConstExp(new TIntegerLiteral("0"));
ABinopExp binop2 = new ABinopExp(phase1Ref1Exp, new AEqBinop(new TEq("==")), intConst6);
thenBlock = new AABlock();
elseBlock = new AABlock();
whileBlock.GetStatements().Add(new AIfThenElseStm(new TLParen("("), binop2,
new ABlockStm(new TLBrace("{"), thenBlock),
new ABlockStm(new TLBrace("{"), elseBlock)));
//output = ch + output;
ABinopExp binop3 = new ABinopExp(chRef3Exp, new APlusBinop(new TPlus("+")), outputRef2Exp);
AAssignmentExp assignment5 = new AAssignmentExp(new TAssign("="), outputRef1, binop3);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment5));
//if(phase1 == 1)
AABlock cBlock = elseBlock;
AIntConstExp intConst7 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop4 = new ABinopExp(phase1Ref2Exp, new AEqBinop(new TEq("==")), intConst7);
thenBlock = new AABlock();
elseBlock = new AABlock();
cBlock.GetStatements().Add(new AIfThenElseStm(new TLParen("("), binop4,
new ABlockStm(new TLBrace("{"), thenBlock),
new ABlockStm(new TLBrace("{"), elseBlock)));
//output = StringSub(output, 1, (StringLength(output) + 1)/2) + ch + StringSub(output, (StringLength(output) + 1)/2 + 1, StringLength(output));
AIntConstExp intConst8 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst9 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst10 = new AIntConstExp(new TIntegerLiteral("2"));
AIntConstExp intConst11 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst12 = new AIntConstExp(new TIntegerLiteral("2"));
AIntConstExp intConst13 = new AIntConstExp(new TIntegerLiteral("1"));
ASimpleInvokeExp invokeStringLength1 = new ASimpleInvokeExp(new TIdentifier("StringLength"),
new ArrayList() {outputRef5Exp});
ABinopExp binop5 = new ABinopExp(invokeStringLength1, new APlusBinop(new TPlus("+")), intConst9);
ABinopExp binop6 = new ABinopExp(binop5, new ADivideBinop(new TDiv("/")), intConst10);
ASimpleInvokeExp invokeStringSub3 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() { outputRef4Exp, intConst8, binop6});
ABinopExp binop7 = new ABinopExp(invokeStringSub3, new APlusBinop(new TPlus("+")), chRef4Exp);
ASimpleInvokeExp invokeStringLength2 = new ASimpleInvokeExp(new TIdentifier("StringLength"),
new ArrayList() { outputRef7Exp });
ABinopExp binop8 = new ABinopExp(invokeStringLength2, new APlusBinop(new TPlus("+")), intConst11);
ABinopExp binop9 = new ABinopExp(binop8, new ADivideBinop(new TDiv("/")), intConst12);
ABinopExp binop10 = new ABinopExp(binop9, new APlusBinop(new TPlus("+")), intConst13);
ASimpleInvokeExp invokeStringLength3 = new ASimpleInvokeExp(new TIdentifier("StringLength"),
new ArrayList() { outputRef8Exp });
ASimpleInvokeExp invokeStringSub4 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() { outputRef6Exp, binop10, invokeStringLength3 });
ABinopExp binop11 = new ABinopExp(binop7, new APlusBinop(new TPlus("+")), invokeStringSub4);
AAssignmentExp assignment6 = new AAssignmentExp(new TAssign("="), outputRef3, binop11);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment6));
//output = output + ch;
ABinopExp binop12 = new ABinopExp(outputRef10Exp, new APlusBinop(new TPlus("+")), chRef5Exp);
AAssignmentExp assignment7 = new AAssignmentExp(new TAssign("="), outputRef9, binop12);
elseBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment7));
//phase1 = phase1 - 1;
AIntConstExp intConst14 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop13 = new ABinopExp(phase1Ref4Exp, new AMinusBinop(new TMinus("-")), intConst14);
AAssignmentExp assignment8 = new AAssignmentExp(new TAssign("="), phase1Ref3, binop13);
whileBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment8));
//if(phase1 < 0)
AIntConstExp intConst15 = new AIntConstExp(new TIntegerLiteral("0"));
ABinopExp binop14 = new ABinopExp(phase1Ref5Exp, new ALtBinop(new TLt("<")), intConst15);
thenBlock = new AABlock();
whileBlock.GetStatements().Add(new AIfThenStm(new TLParen("("), binop14,
new ABlockStm(new TLBrace("{"), thenBlock)));
//phase1 = phase1 + 3;
AIntConstExp intConst16 = new AIntConstExp(new TIntegerLiteral("3"));
ABinopExp binop15 = new ABinopExp(phase1Ref7Exp, new APlusBinop(new TPlus("+")), intConst16);
AAssignmentExp assignment9 = new AAssignmentExp(new TAssign("="), phase1Ref6, binop15);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment9));
//phase2 = phase2 - 1;
AIntConstExp intConst17 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop16 = new ABinopExp(phase2Ref4Exp, new AMinusBinop(new TMinus("-")), intConst17);
AAssignmentExp assignment10 = new AAssignmentExp(new TAssign("="), phase2Ref3, binop16);
whileBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment10));
//if(phase2 < 0)
AIntConstExp intConst18 = new AIntConstExp(new TIntegerLiteral("0"));
ABinopExp binop17 = new ABinopExp(phase2Ref5Exp, new ALtBinop(new TLt("<")), intConst18);
thenBlock = new AABlock();
whileBlock.GetStatements().Add(new AIfThenStm(new TLParen("("), binop17,
new ABlockStm(new TLBrace("{"), thenBlock)));
//phase2 = phase2 + 2;
AIntConstExp intConst19 = new AIntConstExp(new TIntegerLiteral("2"));
ABinopExp binop18 = new ABinopExp(phase2Ref7Exp, new APlusBinop(new TPlus("+")), intConst19);
AAssignmentExp assignment11 = new AAssignmentExp(new TAssign("="), phase2Ref6, binop18);
thenBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment11));
//length = StringLength(s);
ASimpleInvokeExp invokeStringLength4 = new ASimpleInvokeExp(new TIdentifier("StringLength"),
new ArrayList() { stringParamRef8Exp });
AAssignmentExp assignment12 = new AAssignmentExp(new TAssign("="), lengthRef8, invokeStringLength4);
whileBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment12));
/*
phase1 = StringFind(output, ":", false);
phase2 = StringToInt(StringSub(output, 1, phase1 - 1));
return StringSub(output, phase1 + 1, phase2 + phase1);
*/
ABooleanConstExp boolConst1 = new ABooleanConstExp(new AFalseBool());
ASimpleInvokeExp invokeStringFind = new ASimpleInvokeExp(new TIdentifier("StringFind"),
new ArrayList()
{outputRef11Exp, colonStringRefExp, boolConst1});
AAssignmentExp assignment13 = new AAssignmentExp(new TAssign("="), phase1Ref8, invokeStringFind);
methodBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment13));
//phase2 = StringToInt(StringSub(output, 1, phase1 - 1));
AIntConstExp intConst20 = new AIntConstExp(new TIntegerLiteral("1"));
AIntConstExp intConst21 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop19 = new ABinopExp(phase1Ref9Exp, new AMinusBinop(new TMinus("-")), intConst21);
ASimpleInvokeExp invokeStringSub5 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() { outputRef12Exp, intConst20, binop19});
ASimpleInvokeExp invokeStringToInt = new ASimpleInvokeExp(new TIdentifier("StringToInt"),
new ArrayList() { invokeStringSub5 });
AAssignmentExp assignment14 = new AAssignmentExp(new TAssign("="), phase2Ref8, invokeStringToInt);
methodBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment14));
//return StringSub(output, phase1 + 1, phase2 + phase1);
AIntConstExp intConst22 = new AIntConstExp(new TIntegerLiteral("1"));
ABinopExp binop20 = new ABinopExp(phase1Ref10Exp, new APlusBinop(new TPlus("+")), intConst22);
ABinopExp binop21 = new ABinopExp(phase2Ref9Exp, new APlusBinop(new TPlus("+")), phase1Ref11Exp);
ASimpleInvokeExp invokeStringSub6 = new ASimpleInvokeExp(new TIdentifier("StringSub"),
new ArrayList() { outputRef12Exp, binop20, binop21 });
methodBlock.GetStatements().Add(new AValueReturnStm(new TReturn("return"), invokeStringSub6));
AMethodDecl method = new AMethodDecl(new APublicVisibilityModifier(), null, null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("Galaxypp_Deobfuscate"), new ArrayList() {stringParam},
methodBlock);
//Fix data refferences.. I got tired here.
return method;
}
public override void CaseAALocalDecl(AALocalDecl node)
{
InAALocalDecl(node);
if (node.GetInit() != null)
{
node.GetInit().Apply(this);
}
if (node.GetName() != null)
{
node.GetName().Apply(this);
}
if (node.GetType() != null)
{
node.GetType().Apply(this);
}
if (node.GetConst() != null)
{
node.GetConst().Apply(this);
}
if (node.GetOut() != null)
{
node.GetOut().Apply(this);
}
if (node.GetRef() != null)
{
node.GetRef().Apply(this);
}
if (node.GetStatic() != null)
{
node.GetStatic().Apply(this);
}
if (node.GetVisibilityModifier() != null)
{
node.GetVisibilityModifier().Apply(this);
}
OutAALocalDecl(node);
}
private void TransformBreaks(PStm stm, SwitchCaseData switchCase, AALocalDecl continueDecl, AALocalDecl fallthroughDecl, ref int currI)
{
if (stm is ABreakStm)
{
AABlock pBlock = (AABlock) stm.Parent();
if (!switchCase.IsLast && (switchCase.RequiresContinue || switchCase.ContainsFallthrough))
{
//Add continue = false;
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(continueDecl.GetName().Text));
ABooleanConstExp rightSide = new ABooleanConstExp(new AFalseBool());
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), lvalue, rightSide);
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(stm), new AExpStm(new TSemicolon(";"), assignment));
data.LocalLinks[lvalue] = continueDecl;
data.LvalueTypes[lvalue] = data.ExpTypes[rightSide] = data.ExpTypes[assignment] = continueDecl.GetType();
currI++;
}
if (!switchCase.IsLast && switchCase.TargetForFallThrough)
{
//Add fallthrough = false;
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(fallthroughDecl.GetName().Text));
ABooleanConstExp rightSide = new ABooleanConstExp(new AFalseBool());
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), lvalue, rightSide);
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(stm), new AExpStm(new TSemicolon(";"), assignment));
data.LocalLinks[lvalue] = fallthroughDecl;
data.LvalueTypes[lvalue] = data.ExpTypes[rightSide] = data.ExpTypes[assignment] = fallthroughDecl.GetType();
currI++;
}
if (!switchCase.RequiresWhile)
{//Remove break
stm.Parent().RemoveChild(stm);
currI--;
}
}
if (stm is ABlockStm)
TransformBreaks((AABlock)((ABlockStm)stm).GetBlock(), switchCase, continueDecl, fallthroughDecl);
if (stm is AIfThenStm)
TransformBreaks(((AIfThenStm)stm).GetBody(), switchCase, continueDecl, fallthroughDecl, ref currI);
if (stm is AIfThenElseStm)
{
TransformBreaks(((AIfThenElseStm)stm).GetThenBody(), switchCase, continueDecl, fallthroughDecl, ref currI);
TransformBreaks(((AIfThenElseStm)stm).GetElseBody(), switchCase, continueDecl, fallthroughDecl, ref currI);
}
}
public override void OutAIncDecExp(AIncDecExp node)
{
PIncDecOp op = node.GetIncDecOp();
if (!Util.HasAncestor<AABlock>(node))
{
Token token = null;
if (op is APostDecIncDecOp)
token = ((APostDecIncDecOp) op).GetToken();
else if (op is APreDecIncDecOp)
token = ((APreDecIncDecOp)op).GetToken();
else if (op is APostIncIncDecOp)
token = ((APostIncIncDecOp)op).GetToken();
else if (op is APreIncIncDecOp)
token = ((APreIncIncDecOp)op).GetToken();
errors.Add(new ErrorCollection.Error(token, LocRM.GetString("ErrorText114")));
throw new ParserException(token, "TypeChecking.OutAIncDecExp");
}
bool plus = op is APreIncIncDecOp || op is APostIncIncDecOp;
if (op is APreIncIncDecOp || op is APreDecIncDecOp || node.Parent() is AExpStm)
{//++i, --i, i++; or i--;
//Replace with assignment
//<exp>++ => <exp> + 1
//(... foo = <exp> ...)++ => (... foo = <exp> ...) = (... foo ...) + 1
//(... foo++ ...)++ => (... foo++ ...) = (... foo ...) + 1
PLvalue clone = Util.MakeClone(node.GetLvalue(), data);
clone.Apply(new AssignFixup(data));
PBinop binop;
if (plus)
{
binop = new APlusBinop(new TPlus("+"));
}
else
{
binop = new AMinusBinop(new TMinus("-"));
}
ABinopExp addExp = new ABinopExp(new ALvalueExp(clone), binop, new AIntConstExp(new TIntegerLiteral("1")));
AAssignmentExp exp = new AAssignmentExp(new TAssign("="), node.GetLvalue(), addExp);
node.ReplaceBy(exp);
exp.Apply(this);
return;
}
{//i++ or i--
//Make a new local so
//int newLocal = i;
//++i;
//...newLocal...;
PLvalue lvalueClone = Util.MakeClone(node.GetLvalue(), data);
PExp exp = new ALvalueExp(Util.MakeClone(node.GetLvalue(), data));
data.ExpTypes[exp] = data.LvalueTypes[node.GetLvalue()];
AALocalDecl localDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, Util.MakeClone(data.LvalueTypes[node.GetLvalue()], data), new TIdentifier("incDecVar"), exp);
ALocalDeclStm localDeclStm = new ALocalDeclStm(new TSemicolon(";"), localDecl);
node.SetIncDecOp(plus
? (PIncDecOp) new APreIncIncDecOp(new TPlusPlus("++"))
: new APreDecIncDecOp(new TMinusMinus("--")));
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(localDecl.GetName().Text));
exp = new ALvalueExp(lvalue);
data.ExpTypes[exp] = data.LvalueTypes[lvalue] = data.LvalueTypes[node.GetLvalue()];
data.LocalLinks[lvalue] = localDecl;
PStm pStm = Util.GetAncestor<PStm>(node);
node.ReplaceBy(exp);
PStm nodeStm = new AExpStm(new TSemicolon(";"), node);
AABlock block = (AABlock) pStm.Parent();
block.GetStatements().Insert(block.GetStatements().IndexOf(pStm), localDeclStm);
block.GetStatements().Insert(block.GetStatements().IndexOf(pStm), nodeStm);
localDeclStm.Apply(this);
nodeStm.Apply(this);
exp.Apply(this);
if (pStm is AWhileStm && Util.IsAncestor(exp, ((AWhileStm)pStm).GetCondition()))
{
AWhileStm aStm = (AWhileStm)pStm;
//Insert
// newLocal = i
// ++i
//Before each continue in the while, and at the end.
//Add continue statement, if not present
block = (AABlock)((ABlockStm)aStm.GetBody()).GetBlock();
if (block.GetStatements().Count == 0 || !(block.GetStatements()[block.GetStatements().Count - 1] is AContinueStm))
block.GetStatements().Add(new AContinueStm(new TContinue("continue")));
//Get all continue statements in the while
ContinueFinder finder = new ContinueFinder();
block.Apply(finder);
foreach (AContinueStm continueStm in finder.Continues)
{
PLvalue nodeLvalue1 = Util.MakeClone(lvalueClone, data);
PExp nodeLvalue1Exp = new ALvalueExp(nodeLvalue1);
PLvalue nodeLvalue2 = Util.MakeClone(lvalueClone, data);
ALocalLvalue newLocalLvalue = new ALocalLvalue(new TIdentifier("newLocal"));
data.LocalLinks[newLocalLvalue] = localDecl;
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), newLocalLvalue, nodeLvalue1Exp);
PStm assignmentStm = new AExpStm(new TSemicolon(";"), assignment);
AIncDecExp newIncDecExp = new AIncDecExp(nodeLvalue2, plus
? (PIncDecOp)
new APreIncIncDecOp(
new TPlusPlus("++"))
: new APreDecIncDecOp(
new TMinusMinus("--")));
PStm newIncDecExpStm = new AExpStm(new TSemicolon(";"), newIncDecExp);
block = (AABlock)continueStm.Parent();
block.GetStatements().Insert(block.GetStatements().IndexOf(continueStm), assignmentStm);
block.GetStatements().Insert(block.GetStatements().IndexOf(continueStm), newIncDecExpStm);
assignment.Apply(this);
newIncDecExp.Apply(this);
}
}
return;
}
}
public override void OutAALocalDecl(AALocalDecl node)
{
if (node.GetInit() != null)
{
PType from = data.ExpTypes[node.GetInit()];
PType to = node.GetType();
if (!Assignable(from, to))
{
if (ImplicitAssignable(from, to))
{
ANamedType namedTo = (ANamedType)to;
ACastExp cast = new ACastExp(new TLParen("("), new ANamedType(new TIdentifier(((AAName)namedTo.GetName()).AsString()), null), node.GetInit());
node.SetInit(cast);
OutACastExp(cast);
to = from;
}
else
errors.Add(new ErrorCollection.Error(node.GetName(), currentSourceFile,
LocRM.GetString("ErrorText151") + Util.TypeToString(from) +
LocRM.GetString("ErrorText152") + Util.TypeToString(to)));
}
}
//If the return type or the type of any formals is a private struct, and the method is a public context, give an error
if (node.Parent() is AStructDecl)
{
AStructDecl pStruct = Util.GetAncestor<AStructDecl>(node);
//Is public context
if ( pStruct.GetVisibilityModifier() is APublicVisibilityModifier && !(node.GetVisibilityModifier() is APrivateVisibilityModifier))
{
PType type = node.GetType();
int i = 0;
FindPrivateTypes finder = new FindPrivateTypes(data);
type.Apply(finder);
if (finder.PrivateTypes.Count > 0)
{
List<ErrorCollection.Error> subErrors = new List<ErrorCollection.Error>();
List<PDecl> usedDecls = new List<PDecl>();
foreach (ANamedType namedType in finder.PrivateTypes)
{
if (data.StructTypeLinks.ContainsKey(namedType))
{
AStructDecl decl = data.StructTypeLinks[namedType];
if (usedDecls.Contains(decl))
continue;
usedDecls.Add(decl);
subErrors.Add(new ErrorCollection.Error(decl.GetName(), LocRM.GetString("ErrorText64")));
}
else if (data.DelegateTypeLinks.ContainsKey(namedType))
{
AMethodDecl decl = data.DelegateTypeLinks[namedType];
if (usedDecls.Contains(decl))
continue;
usedDecls.Add(decl);
subErrors.Add(new ErrorCollection.Error(decl.GetName(), LocRM.GetString("ErrorText154")));
}
}
errors.Add(new ErrorCollection.Error(node.GetName(), LocRM.GetString("ErrorText155"), false, subErrors.ToArray()));
}
}
}
base.OutAALocalDecl(node);
}
public override void CaseASwitchStm(ASwitchStm node)
{
node.GetTest().Apply(this);
AALocalDecl fallThroughDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("bool"), null),
new TIdentifier(MakeUniqueLocalName(node, "switchFallThrough")),
new ABooleanConstExp(new AFalseBool()));
AALocalDecl continueDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("bool"), null),
new TIdentifier(MakeUniqueLocalName(node, "switchContinue")),
new ABooleanConstExp(new ATrueBool()));
AALocalDecl testVar = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
Util.MakeClone(data.ExpTypes[node.GetTest()], data),
new TIdentifier(MakeUniqueLocalName(node, "switchTestVar")),
node.GetTest());
AABlock bigBlock = new AABlock();
//AABlock previousBlock = bigBlock;
if (node.GetCases().Count > 0)
{
List<SwitchCaseData> switchCaseDatas = new List<SwitchCaseData>();
//Join cases without a body
for (int i = node.GetCases().Count - 1; i >= 0; i--)
{
ASwitchCaseStm caseStm = (ASwitchCaseStm) node.GetCases()[i];
SwitchCaseData caseData = new SwitchCaseData();
caseData.Block = (AABlock) caseStm.GetBlock();
if (caseStm.GetType() is ACaseSwitchCaseType)
caseData.Tests.Add(((ACaseSwitchCaseType)caseStm.GetType()).GetExp());
else
caseData.ContainsDefault = true;
caseData.IsLast = switchCaseDatas.Count == 0;
if (switchCaseDatas.Count == 0 || caseData.Block.GetStatements().Count > 0)
{
switchCaseDatas.Insert(0, caseData);
continue;
}
switchCaseDatas[0].Tests.AddRange(caseData.Tests);
}
for (int i = switchCaseDatas.Count - 1; i >= 0; i--)
{
switchCaseDatas[i].ContainsFallthrough = CanFallthrough(switchCaseDatas[i].Block, out switchCaseDatas[i].HasBreaks, out switchCaseDatas[i].RequiresWhile);
if (i == switchCaseDatas.Count - 1)
continue;
switchCaseDatas[i + 1].TargetForFallThrough = switchCaseDatas[i].ContainsFallthrough;
switchCaseDatas[i].RequiresContinue = !switchCaseDatas[i].ContainsFallthrough &&
(switchCaseDatas[i + 1].RequiresContinue ||
switchCaseDatas[i + 1].ContainsFallthrough);
}
AABlock previousBlock = bigBlock;
//Make code for specific case
foreach (SwitchCaseData switchCase in switchCaseDatas)
{
List<PExp> tests = new List<PExp>();
AABlock nextBlock;
if (switchCase.TargetForFallThrough)
{//Add if (continueSwitch) {}
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(continueDecl.GetName().Text));
ALvalueExp test = new ALvalueExp(lvalue);
nextBlock = new AABlock();
AIfThenStm ifStm = new AIfThenStm(new TLParen("("), test, new ABlockStm(new TLBrace("{"), nextBlock));
previousBlock.GetStatements().Add(ifStm);
previousBlock = nextBlock;
data.LocalLinks[lvalue] = continueDecl;
data.LvalueTypes[lvalue] = data.ExpTypes[test] = continueDecl.GetType();
//First test in next if: if (fallThrough || ...
lvalue = new ALocalLvalue(new TIdentifier(fallThroughDecl.GetName().Text));
test = new ALvalueExp(lvalue);
tests.Add(test);
data.LocalLinks[lvalue] = fallThroughDecl;
data.LvalueTypes[lvalue] = data.ExpTypes[test] = fallThroughDecl.GetType();
}
//Make code for the test in the if
foreach (PExp exp in switchCase.Tests)
{
ALocalLvalue leftSide = new ALocalLvalue(new TIdentifier(testVar.GetName().Text));
ALvalueExp lvalueExp = new ALvalueExp(leftSide);
ABinopExp test = new ABinopExp(lvalueExp, new AEqBinop(new TEq("==")), exp);
tests.Add(test);
data.LocalLinks[leftSide] = testVar;
data.LvalueTypes[leftSide] = data.ExpTypes[lvalueExp] = testVar.GetType();
data.ExpTypes[test] = new ANamedType(new TIdentifier("bool"), null);
}
if (switchCase.ContainsDefault)
{
ABooleanConstExp test = new ABooleanConstExp(new ATrueBool());
tests.Add(test);
data.ExpTypes[test] = new ANamedType(new TIdentifier("bool"), null);
}
PExp finalTest = tests[0];
tests.RemoveAt(0);
foreach (PExp exp in tests)
{
finalTest = new ABinopExp(finalTest, new ALazyOrBinop(new TOrOr("||")), exp);
data.ExpTypes[finalTest] = new ANamedType(new TIdentifier("bool"), null);
}
//Transform breaks into assignments
//If we can fallthrough, and there are breaks, encase in a while stm
AABlock testBlock = switchCase.Block;
if (switchCase.RequiresWhile)
{
AABlock newBlock = new AABlock();
PExp whileTest = new ABooleanConstExp(new ATrueBool());
AWhileStm whileStm = new AWhileStm(new TLParen("("), whileTest, new ABlockStm(new TLBrace("{"), switchCase.Block));
newBlock.GetStatements().Add(whileStm);
switchCase.Block = newBlock;
}
TransformBreaks(testBlock, switchCase, continueDecl, fallThroughDecl);
if (switchCase.ContainsFallthrough && !switchCase.TargetForFallThrough)
{//Add fallthrough = true;
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(fallThroughDecl.GetName().Text));
ABooleanConstExp rightSide = new ABooleanConstExp(new ATrueBool());
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), lvalue, rightSide);
testBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment));
data.LocalLinks[lvalue] = fallThroughDecl;
data.LvalueTypes[lvalue] = data.ExpTypes[rightSide] = data.ExpTypes[assignment] = fallThroughDecl.GetType();
}
if (switchCase.RequiresWhile)
{//Add break at the end of the while
testBlock.GetStatements().Add(new ABreakStm(new TBreak("break")));
}
//Make if
PStm finalIfStm;
if (finalTest is ABooleanConstExp)
{//Final if is if(true). dont add it.
finalIfStm = new ABlockStm(new TLBrace("{"), switchCase.Block);
nextBlock = new AABlock();
}
else if (switchCase.IsLast || switchCase.ContainsFallthrough)
{//One armed if
finalIfStm = new AIfThenStm(new TLParen("("), finalTest,
new ABlockStm(new TLBrace("{"), switchCase.Block));
nextBlock = bigBlock;
}
else
{//Two armed if
nextBlock = new AABlock();
finalIfStm = new AIfThenElseStm(new TLParen("("), finalTest,
new ABlockStm(new TLBrace("{"), switchCase.Block),
new ABlockStm(new TLBrace("{"), nextBlock));
}
previousBlock.GetStatements().Add(finalIfStm);
previousBlock = nextBlock;
}
//If needed, add fallThroughDecl and continueDecl
data.Locals.Add(bigBlock, new List<AALocalDecl>());
if (data.LocalLinks.Values.Contains(fallThroughDecl))
{
bigBlock.GetStatements().Insert(0, new ALocalDeclStm(new TSemicolon(";"), fallThroughDecl));
data.Locals[bigBlock].Add(fallThroughDecl);
}
if (data.LocalLinks.Values.Contains(continueDecl))
{
bigBlock.GetStatements().Insert(0, new ALocalDeclStm(new TSemicolon(";"), continueDecl));
data.Locals[bigBlock].Add(continueDecl);
}
bigBlock.GetStatements().Insert(0, new ALocalDeclStm(new TSemicolon(";"), testVar));
data.Locals[bigBlock].Add(testVar);
node.ReplaceBy(new ABlockStm(new TLBrace("{"), bigBlock));
bigBlock.Apply(this);
}
}
public override void CaseAALocalDecl(AALocalDecl node)
{
//Convert a static struct field into a global variable. All refferences to it are structFieldLvalues.
if (node.GetStatic() == null)
return;
AStructDecl str = (AStructDecl) node.Parent();
if (data.StructFields[str].Contains(node))
data.StructFields[str].Remove(node);
AFieldDecl replacementField;
//Don't enhrit static fields.
if (data.EnheritanceLocalMap.ContainsKey(node))
{
str.RemoveChild(node);
AALocalDecl realVar = data.EnheritanceLocalMap[node];
if (convertionMap.ContainsKey(realVar))
{
//Already converted to a field
replacementField = convertionMap[realVar];
foreach (AStructFieldLvalue lvalue in data.StructMethodFieldLinks.Where(link => link.Value == node).Select(link => link.Key))
{
AFieldLvalue newLvalue = new AFieldLvalue(new TIdentifier(replacementField.GetName().Text));
data.FieldLinks[newLvalue] = replacementField;
data.LvalueTypes[newLvalue] = replacementField.GetType();
lvalue.ReplaceBy(newLvalue);
}
}
else
{
List<AStructFieldLvalue> refferences = new List<AStructFieldLvalue>();
refferences.AddRange(data.StructMethodFieldLinks.Where(link => link.Value == node).Select(link => link.Key));
foreach (AStructFieldLvalue lvalue in refferences)
{
data.StructMethodFieldLinks[lvalue] = realVar;
}
}
return;
}
replacementField = new AFieldDecl(new APublicVisibilityModifier(), null, node.GetConst(), node.GetType(), node.GetName(), node.GetInit());
replacementField.GetName().Text = str.GetName().Text + "_" + replacementField.GetName().Text;
AASourceFile file = Util.GetAncestor<AASourceFile>(node);
file.GetDecl().Insert(file.GetDecl().IndexOf(node.Parent()), replacementField);
data.Fields.Add(new SharedData.DeclItem<AFieldDecl>(file, replacementField));
if (ContainsNewExp(replacementField.GetInit()))
data.FieldsToInitInMapInit.Add(replacementField);
foreach (AStructFieldLvalue lvalue in data.StructMethodFieldLinks.Where(link => link.Value == node).Select(link => link.Key))
{
AFieldLvalue newLvalue = new AFieldLvalue(new TIdentifier(replacementField.GetName().Text));
data.FieldLinks[newLvalue] = replacementField;
data.LvalueTypes[newLvalue] = replacementField.GetType();
lvalue.ReplaceBy(newLvalue);
}
convertionMap.Add(node, replacementField);
node.Parent().RemoveChild(node);
}
public override void OutAALocalDecl(AALocalDecl node)
{
if (node.GetInit() != null)
{
PType type = node.GetType();
if (Util.IsBulkCopy(type))
{
node.Apply(mover);
ALocalLvalue lvalue = new ALocalLvalue(new TIdentifier(node.GetName().Text));
data.LocalLinks[lvalue] = node;
data.LvalueTypes[lvalue] = type;
List<PStm> replacementStatements = MakeAssignments(lvalue, node.GetInit(),
new List<AALocalDecl>() { null });
PStm pStm = Util.GetAncestor<PStm>(node);
AABlock pBlock = (AABlock)pStm.Parent();
foreach (PStm stm in replacementStatements)
{
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(pStm) + 1, stm);
}
node.SetInit(null);
}
}
base.OutAALocalDecl(node);
}
public static List<AABlock> Inline(ASimpleInvokeExp node, FinalTransformations finalTrans)
{
/*if (Util.GetAncestor<AMethodDecl>(node) != null && Util.GetAncestor<AMethodDecl>(node).GetName().Text == "UIChatFrame_LeaveChannel")
node = node;*/
SharedData data = finalTrans.data;
//If this node is inside the condition of a while, replace it with a new local var,
//make a clone before the while, one before each continue in the while, and one at the end of the while
//(unless the end is a return or break)
AABlock pBlock;
if (Util.HasAncestor<AWhileStm>(node))
{
AWhileStm whileStm = Util.GetAncestor<AWhileStm>(node);
if (Util.IsAncestor(node, whileStm.GetCondition()))
{
List<ASimpleInvokeExp> toInline = new List<ASimpleInvokeExp>();
//Above while
AALocalDecl replaceVarDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
Util.MakeClone(data.ExpTypes[node], data),
new TIdentifier("whileVar"), null);
ALocalLvalue replaceVarRef = new ALocalLvalue(new TIdentifier("whileVar"));
ALvalueExp replaceVarRefExp = new ALvalueExp(replaceVarRef);
data.LocalLinks[replaceVarRef] = replaceVarDecl;
data.ExpTypes[replaceVarRefExp] = data.LvalueTypes[replaceVarRef] = replaceVarDecl.GetType();
node.ReplaceBy(replaceVarRefExp);
replaceVarDecl.SetInit(node);
pBlock = (AABlock) whileStm.Parent();
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(whileStm), new ALocalDeclStm(new TSemicolon(";"), replaceVarDecl));
toInline.Add(node);
//In the end of the while
PStm lastStm = whileStm.GetBody();
while (lastStm is ABlockStm)
{
AABlock block = (AABlock) ((ABlockStm) lastStm).GetBlock();
if (block.GetStatements().Count == 0)
{
lastStm = null;
break;
}
lastStm = (PStm) block.GetStatements()[block.GetStatements().Count - 1];
}
if (lastStm == null || !(lastStm is AValueReturnStm || lastStm is AVoidReturnStm || lastStm is ABreakStm))
{
lastStm = whileStm.GetBody();
AABlock block;
if (lastStm is ABlockStm)
{
block = (AABlock)((ABlockStm)lastStm).GetBlock();
}
else
{
block = new AABlock(new ArrayList(), new TRBrace("}"));
block.GetStatements().Add(lastStm);
whileStm.SetBody(new ABlockStm(new TLBrace("{"), block));
}
replaceVarRef = new ALocalLvalue(new TIdentifier("whileVar"));
ASimpleInvokeExp clone = (ASimpleInvokeExp)Util.MakeClone(node, data);
toInline.Add(clone);
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), replaceVarRef, clone);
data.LocalLinks[replaceVarRef] = replaceVarDecl;
data.ExpTypes[assignment] = data.LvalueTypes[replaceVarRef] = replaceVarDecl.GetType();
block.GetStatements().Add(new AExpStm(new TSemicolon(";"), assignment));
}
//After each continue
CloneBeforeContinue cloner = new CloneBeforeContinue(node, replaceVarDecl, data);
whileStm.GetBody().Apply(cloner);
toInline.AddRange(cloner.replacementExpressions);
List<AABlock> visitBlocks = new List<AABlock>();
foreach (ASimpleInvokeExp invoke in toInline)
{
visitBlocks.AddRange(Inline(invoke, finalTrans));
}
return visitBlocks;
}
}
AMethodDecl decl = finalTrans.data.SimpleMethodLinks[node];
FindAssignedToFormals assignedToFormalsFinder = new FindAssignedToFormals(finalTrans.data);
decl.Apply(assignedToFormalsFinder);
List<AALocalDecl> assignedToFormals = assignedToFormalsFinder.AssignedFormals;
/*
* inline int foo(int a)
* {
* int b = 2;
* int c;
* ...
* while(...)
* {
* ...
* break;
* ...
* return c;
* }
* ...
* return 2;
* }
*
* bar(foo(<arg for a>));
* ->
*
* {
* bool inlineMethodReturned = false;
* int inlineReturner;
* int a = <arg for a>;
* while (!inlineMethodReturned)
* {
* int b = 2;
* int c;
* ...
* while(...)
* {
* ...
* break
* ...
* inlineReturner = c;
* inlineMethodReturned = true;
* break;
* }
* if (inlineMethodReturned)
* {
* break;
* }
* ...
* inlineReturner = 2;
* inlineMethodReturned = true;
* break;
* break;
* }
* bar(inlineReturner);
* }
*
*
*/
AABlock outerBlock = new AABlock();
PExp exp = new ABooleanConstExp(new AFalseBool());
finalTrans.data.ExpTypes[exp] = new ANamedType(new TIdentifier("bool"), null);
AALocalDecl hasMethodReturnedVar = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, new ANamedType(new TIdentifier("bool"), null),
new TIdentifier("hasInlineReturned"), exp);
finalTrans.data.GeneratedVariables.Add(hasMethodReturnedVar);
PStm stm = new ALocalDeclStm(new TSemicolon(";"), hasMethodReturnedVar);
outerBlock.GetStatements().Add(stm);
AALocalDecl methodReturnerVar = null;
if (!(decl.GetReturnType() is AVoidType))
{
methodReturnerVar = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, Util.MakeClone(decl.GetReturnType(), finalTrans.data),
new TIdentifier("inlineReturner"), null);
stm = new ALocalDeclStm(new TSemicolon(";"), methodReturnerVar);
outerBlock.GetStatements().Add(stm);
}
AABlock afterBlock = new AABlock();
//A dictionary from the formals of the inline method to a cloneable replacement lvalue
Dictionary<AALocalDecl, PLvalue> Parameters = new Dictionary<AALocalDecl, PLvalue>();
Dictionary<AALocalDecl, PExp> ParameterExps = new Dictionary<AALocalDecl, PExp>();
for (int i = 0; i < decl.GetFormals().Count; i++)
{
AALocalDecl formal = (AALocalDecl)decl.GetFormals()[i];
PExp arg = (PExp)node.GetArgs()[0];
PLvalue lvalue;
//if ref, dont make a new var
if (formal.GetRef() != null && arg is ALvalueExp)
{
arg.Apply(new MoveMethodDeclsOut("inlineVar", finalTrans.data));
arg.Parent().RemoveChild(arg);
lvalue = ((ALvalueExp) arg).GetLvalue();
}
else if (!assignedToFormals.Contains(formal) && Util.IsLocal(arg, finalTrans.data))
{
lvalue = new ALocalLvalue(new TIdentifier("I hope I dont make it"));
finalTrans.data.LvalueTypes[lvalue] = formal.GetType();
finalTrans.data.LocalLinks[(ALocalLvalue) lvalue] = formal;
ParameterExps[formal] = arg;
arg.Parent().RemoveChild(arg);
}
else
{
AAssignmentExp assExp = null;
if (formal.GetOut() != null)
{
//Dont initialize with arg, but assign arg after
arg.Apply(new MoveMethodDeclsOut("inlineVar", finalTrans.data));
lvalue = ((ALvalueExp)arg).GetLvalue();
assExp = new AAssignmentExp(new TAssign("="), lvalue, null);
finalTrans.data.ExpTypes[assExp] = finalTrans.data.LvalueTypes[lvalue];
arg.Parent().RemoveChild(arg);
arg = null;
}
AALocalDecl parameter = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null, Util.MakeClone(formal.GetType(), finalTrans.data),
new TIdentifier(formal.GetName().Text),
arg);
stm = new ALocalDeclStm(new TSemicolon(";"), parameter);
outerBlock.GetStatements().Add(stm);
lvalue = new ALocalLvalue(new TIdentifier(parameter.GetName().Text));
finalTrans.data.LvalueTypes[lvalue] = parameter.GetType();
finalTrans.data.LocalLinks[(ALocalLvalue)lvalue] = parameter;
if (formal.GetOut() != null)
{
//Dont initialize with arg, but assign arg after
ALvalueExp lvalueExp = new ALvalueExp(Util.MakeClone(lvalue, finalTrans.data));
finalTrans.data.ExpTypes[lvalueExp] = finalTrans.data.LvalueTypes[lvalue];
assExp.SetExp(lvalueExp);
afterBlock.GetStatements().Add(new AExpStm(new TSemicolon(";"), assExp));
}
}
Parameters.Add(formal, lvalue);
}
AABlock innerBlock = (AABlock)decl.GetBlock().Clone();
exp = new ABooleanConstExp(new ATrueBool());
finalTrans.data.ExpTypes[exp] = new ANamedType(new TIdentifier("bool"), null);
ABlockStm innerBlockStm = new ABlockStm(new TLBrace("{"), innerBlock);
bool needWhile = CheckIfWhilesIsNeeded.IsWhileNeeded(decl.GetBlock());
if (needWhile)
stm = new AWhileStm(new TLParen("("), exp, innerBlockStm);
else
stm = innerBlockStm;
outerBlock.GetStatements().Add(stm);
outerBlock.GetStatements().Add(new ABlockStm(new TLBrace("{"), afterBlock));
//Clone method contents to inner block.
CloneMethod cloneFixer = new CloneMethod(finalTrans, Parameters, ParameterExps, innerBlockStm);
decl.GetBlock().Apply(cloneFixer);
foreach (KeyValuePair<PLvalue, PExp> pair in cloneFixer.ReplaceUsAfter)
{
PLvalue lvalue = pair.Key;
PExp replacement = Util.MakeClone(pair.Value, finalTrans.data);
ALvalueExp lvalueParent = (ALvalueExp) lvalue.Parent();
lvalueParent.ReplaceBy(replacement);
}
innerBlockStm.Apply(new FixTypes(finalTrans.data));
innerBlock.Apply(new FixReturnsAndWhiles(hasMethodReturnedVar, methodReturnerVar, finalTrans.data, needWhile));
GetNonBlockStm stmFinder = new GetNonBlockStm(false);
innerBlock.Apply(stmFinder);
if (needWhile && (stmFinder.Stm == null || !(stmFinder.Stm is ABreakStm)))
innerBlock.GetStatements().Add(new ABreakStm(new TBreak("break")));
//Insert before current statement
ABlockStm outerBlockStm = new ABlockStm(new TLBrace("{"), outerBlock);
PStm pStm = Util.GetAncestor<PStm>(node);
pBlock = (AABlock)pStm.Parent();
pBlock.GetStatements().Insert(pBlock.GetStatements().IndexOf(pStm), outerBlockStm);
if (node.Parent() == pStm && pStm is AExpStm)
{
pBlock.RemoveChild(pStm);
}
else
{
PLvalue lvalue = new ALocalLvalue(new TIdentifier(methodReturnerVar.GetName().Text));
finalTrans.data.LvalueTypes[lvalue] = methodReturnerVar.GetType();
finalTrans.data.LocalLinks[(ALocalLvalue)lvalue] = methodReturnerVar;
exp = new ALvalueExp(lvalue);
finalTrans.data.ExpTypes[exp] = methodReturnerVar.GetType();
node.ReplaceBy(exp);
}
return new List<AABlock>() { outerBlock };
}
