public virtual void OutANullExp(ANullExp node)
{
DefaultOut(node);
}
public override void OutAAProgram(AAProgram node)
{
if (strings.Count == 0)
return;
//Obfuscate all strings
List<string> obfuscated = new List<string>();
foreach (AStringConstExp stringConstExp in strings)
{
TStringLiteral token = stringConstExp.GetStringLiteral();
string s = token.Text.Substring(1, token.Text.Length - 2);
obfuscated.Add(Obfuscate(s));
}
//Set invokes instead of string constants, and move varaiabes down
List<AFieldDecl> ignoredFields = new List<AFieldDecl>();
List<AFieldDecl> moveFieldsIn = new List<AFieldDecl>();
Dictionary<AFieldDecl, AMethodDecl> fieldMethods = new Dictionary<AFieldDecl, AMethodDecl>();
for (int i = 0; i < strings.Count; i++)
{
AStringConstExp stringExp = strings[i];
Token token = stringExp.GetStringLiteral();
bool inDeobfuscator = Util.GetAncestor<AMethodDecl>(stringExp) == finalTrans.data.DeobfuscateMethod;
if (inDeobfuscator)
{
AFieldDecl field = finalTrans.data.UnobfuscatedStrings[stringExp];
AStringConstExp newStringConst = new AStringConstExp(stringExp.GetStringLiteral());
field.SetInit(newStringConst);
AFieldLvalue fieldRef = new AFieldLvalue(new TIdentifier(field.GetName().Text, token.Line, token.Pos));
finalTrans.data.FieldLinks[fieldRef] = field;
stringExp.ReplaceBy(new ALvalueExp(fieldRef));
}
else
{
AFieldDecl field;
if (!finalTrans.data.ObfuscatedStrings.ContainsKey(stringExp))
{
int line = -finalTrans.data.ObfuscatedStrings.Count - 1;
field = new AFieldDecl(new APublicVisibilityModifier(), null, new TConst("const", line, 0),
new ANamedType(new TIdentifier("string", line, 1), null),
new TIdentifier("Galaxy_pp_stringO" +
finalTrans.data.ObfuscatedStrings.Count), null);
//If the strings are the same - point them to same field
bool newField = true;
foreach (AStringConstExp oldStringConstExp in finalTrans.data.ObfuscatedStrings.Keys)
{
if (stringExp.GetStringLiteral().Text == oldStringConstExp.GetStringLiteral().Text)
{
field = finalTrans.data.ObfuscatedStrings[oldStringConstExp];
newField = false;
break;
}
}
if (newField)
{
AASourceFile file = (AASourceFile)finalTrans.data.DeobfuscateMethod.Parent();
file.GetDecl().Insert(file.GetDecl().IndexOf(finalTrans.data.DeobfuscateMethod) + 1, field);
finalTrans.data.Fields.Add(new SharedData.DeclItem<AFieldDecl>(file, field));
}
finalTrans.data.ObfuscatedStrings.Add(stringExp, field);
}
field = finalTrans.data.ObfuscatedStrings[stringExp];
string obfuscatedString = obfuscated[i];
ASimpleInvokeExp invoke = new ASimpleInvokeExp();
invoke.SetName(new TIdentifier(finalTrans.data.DeobfuscateMethod.GetName().Text,
stringExp.GetStringLiteral().Line,
stringExp.GetStringLiteral().Pos));
AStringConstExp newStringConst =
new AStringConstExp(new TStringLiteral("\"" + obfuscatedString + "\""));
invoke.GetArgs().Add(newStringConst);
finalTrans.data.SimpleMethodLinks[invoke] = finalTrans.data.DeobfuscateMethod;
if (Util.GetAncestor<PStm>(stringExp) == null && false)
{
ignoredFields.Add(field);
/*if (Util.GetAncestor<ASimpleInvokeExp>(stringExp) == null)
stringExp.ReplaceBy(invoke);*/
//Add obfuscate call to this location);
continue;
/*ASimpleInvokeExp invoke = new ASimpleInvokeExp();
invoke.SetName(new TIdentifier(finalTrans.data.DeobfuscateMethod.GetName().Text,
stringExp.GetStringLiteral().Line,
stringExp.GetStringLiteral().Pos));
AStringConstExp newStringConst =
new AStringConstExp(new TStringLiteral("\"" + obfuscatedString + "\""));
invoke.GetArgs().Add(newStringConst);
stringExp.ReplaceBy(invoke);
finalTrans.data.SimpleMethodLinks[invoke] = finalTrans.data.DeobfuscateMethod;
continue;*/
}
if (field.GetInit() == null)
{
/*field.SetInit(invoke);
field.SetConst(null);*/
if (
stringExp.GetStringLiteral().Text.Remove(0, 1).Substring(0,
stringExp.GetStringLiteral().Text.
Length - 2) == "")
{
//Make method
/*
string <field>Method()
{
return "";
}
*
*/
ANullExp nullExp = new ANullExp();
field.SetInit(nullExp);
field.SetConst(null);
AStringConstExp stringConst = new AStringConstExp(new TStringLiteral("\"\""));
AMethodDecl method = new AMethodDecl(new APublicVisibilityModifier(), null, null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("Get" + field.GetName()),
new ArrayList(),
new AABlock(
new ArrayList()
{
new AValueReturnStm(new TReturn("return"),
stringConst)
},
new TRBrace("}")));
AASourceFile pFile = (AASourceFile)field.Parent();
pFile.GetDecl().Insert(pFile.GetDecl().IndexOf(field) + 1, method);
finalTrans.data.ExpTypes[stringConst] = new ANamedType(new TIdentifier("string"), null);
finalTrans.data.ExpTypes[nullExp] = new ANamedType(new TIdentifier("null"), null);
fieldMethods[field] = method;
}
else
{
//Make method
/*
string <field>Method()
{
if (field == null)
{
field = Invoke;
}
if (field == null)
{
return Invoke;
}
return field;
}
*/
ANullExp nullExp1 = new ANullExp();
field.SetInit(nullExp1);
field.SetConst(null);
ANullExp nullExp2 = new ANullExp();
AFieldLvalue fieldRef1 = new AFieldLvalue(new TIdentifier(field.GetName().Text));
AFieldLvalue fieldRef2 = new AFieldLvalue(new TIdentifier(field.GetName().Text));
AFieldLvalue fieldRef3 = new AFieldLvalue(new TIdentifier(field.GetName().Text));
ALvalueExp fieldRef1Exp = new ALvalueExp(fieldRef1);
ALvalueExp fieldRef3Exp = new ALvalueExp(fieldRef3);
ABinopExp binop1 = new ABinopExp(fieldRef1Exp, new AEqBinop(new TEq("==")), nullExp2);
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), fieldRef2, invoke);
AIfThenStm ifStm1 = new AIfThenStm(new TLParen("("), binop1,
new ABlockStm(new TLBrace("{"),
new AABlock(
new ArrayList()
{
new AExpStm(new TSemicolon(";"),
assignment)
},
new TRBrace("}"))));
/*ANullExp nullExp3 = new ANullExp();
AFieldLvalue fieldRef4 = new AFieldLvalue(new TIdentifier(field.GetName().Text));
ALvalueExp fieldRef4Exp = new ALvalueExp(fieldRef4);
AStringConstExp invokeArgClone =
new AStringConstExp(new TStringLiteral("\"" + obfuscatedString + "\""));
ASimpleInvokeExp invokeClone = new ASimpleInvokeExp(new TIdentifier(invoke.GetName().Text),
new ArrayList() { invokeArgClone });
finalTrans.data.SimpleMethodLinks[invokeClone] = finalTrans.data.DeobfuscateMethod;
ABinopExp binop2 = new ABinopExp(fieldRef4Exp, new AEqBinop(new TEq("==")), nullExp3);
AIfThenStm ifStm2 = new AIfThenStm(new TLParen("("), binop2,
new ABlockStm(new TLBrace("{"),
new AABlock(
new ArrayList()
{
new AValueReturnStm(new TReturn("return"), invokeClone)
},
new TRBrace("}"))));*/
AMethodDecl method = new AMethodDecl(new APublicVisibilityModifier(), null, null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("Get" + field.GetName()),
new ArrayList(),
new AABlock(
new ArrayList()
{
ifStm1,
//ifStm2,
new AValueReturnStm(new TReturn("return"),
fieldRef3Exp)
},
new TRBrace("}")));
AASourceFile pFile = (AASourceFile) field.Parent();
pFile.GetDecl().Insert(pFile.GetDecl().IndexOf(field) + 1, method);
finalTrans.data.FieldLinks[fieldRef1] =
finalTrans.data.FieldLinks[fieldRef2] =
finalTrans.data.FieldLinks[fieldRef3] =
/*finalTrans.data.FieldLinks[fieldRef4] = */field;
finalTrans.data.LvalueTypes[fieldRef1] =
finalTrans.data.LvalueTypes[fieldRef2] =
finalTrans.data.LvalueTypes[fieldRef3] =
//finalTrans.data.LvalueTypes[fieldRef4] =
finalTrans.data.ExpTypes[fieldRef1Exp] =
finalTrans.data.ExpTypes[fieldRef3Exp] =
//finalTrans.data.ExpTypes[fieldRef4Exp] =
finalTrans.data.ExpTypes[assignment] = field.GetType();
finalTrans.data.ExpTypes[nullExp1] =
finalTrans.data.ExpTypes[nullExp2] =
/*finalTrans.data.ExpTypes[nullExp3] =*/ new ANamedType(new TIdentifier("null"), null);
finalTrans.data.ExpTypes[binop1] =
/*finalTrans.data.ExpTypes[binop2] = */new ANamedType(new TIdentifier("bool"), null);
fieldMethods[field] = method;
}
/* AFieldLvalue fieldRef = new AFieldLvalue(new TIdentifier(field.GetName().Text, token.Line, token.Pos));
finalTrans.data.FieldLinks[fieldRef] = field;*/
//stringExp.ReplaceBy(new ALvalueExp(fieldRef));
}
ASimpleInvokeExp invoke2 =
new ASimpleInvokeExp(new TIdentifier(fieldMethods[field].GetName().Text), new ArrayList());
finalTrans.data.SimpleMethodLinks[invoke2] = fieldMethods[field];
stringExp.ReplaceBy(invoke2);
//If we are in a field, move it in
if (Util.GetAncestor<AFieldDecl>(invoke2) != null)
moveFieldsIn.Add(Util.GetAncestor<AFieldDecl>(invoke2));
}
}
foreach (AFieldDecl field in finalTrans.data.ObfuscationFields)
{
if (field.GetInit() == null && field.Parent() != null)
{
field.Parent().RemoveChild(field);
}
}
//A constant field, or a field used by a constant field cannot be moved in
List<AFieldDecl> constantFields = new List<AFieldDecl>();
foreach (SharedData.DeclItem<AFieldDecl> field in finalTrans.data.Fields)
{
if (field.Decl.GetConst() != null)
constantFields.Add(field.Decl);
}
for (int i = 0; i < constantFields.Count; i++)
{
GetFieldLvalues lvalues = new GetFieldLvalues();
constantFields[i].Apply(lvalues);
foreach (AFieldLvalue lvalue in lvalues.Lvalues)
{
AFieldDecl field = finalTrans.data.FieldLinks[lvalue];
if (!constantFields.Contains(field))
constantFields.Add(field);
}
}
moveFieldsIn.RemoveAll(constantFields.Contains);
Dictionary<AFieldDecl, List<AFieldDecl>> dependancies = new Dictionary<AFieldDecl, List<AFieldDecl>>();
//Order the fields so any dependancies are instansiated first
foreach (AFieldDecl field in moveFieldsIn)
{
dependancies.Add(field, new List<AFieldDecl>());
GetFieldLvalues lvalues = new GetFieldLvalues();
field.Apply(lvalues);
foreach (AFieldLvalue lvalue in lvalues.Lvalues)
{
AFieldDecl dependancy = finalTrans.data.FieldLinks[lvalue];
if (!dependancies[field].Contains(dependancy))
dependancies[field].Add(dependancy);
}
}
List<PStm> newStatements = new List<PStm>();
while (dependancies.Keys.Count > 0)
{
AFieldDecl field = dependancies.FirstOrDefault(f1 => f1.Value.Count == 0).Key ??
dependancies.Keys.First(f => true);
AFieldLvalue fieldRef = new AFieldLvalue(new TIdentifier(field.GetName().Text));
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), fieldRef, field.GetInit());
field.SetInit(null);
newStatements.Add(new AExpStm(new TSemicolon(";"), assignment));
finalTrans.data.FieldLinks[fieldRef] = field;
finalTrans.data.LvalueTypes[fieldRef] =
finalTrans.data.ExpTypes[assignment] = field.GetType();
foreach (KeyValuePair<AFieldDecl, List<AFieldDecl>> dependancy in dependancies)
{
if (dependancy.Value.Contains(field))
dependancy.Value.Remove(field);
}
dependancies.Remove(field);
}
AABlock initBody = (AABlock) finalTrans.mainEntry.GetBlock();
for (int i = newStatements.Count - 1; i >= 0; i--)
{
initBody.GetStatements().Insert(0, newStatements[i]);
}
}
public override void CaseANullExp(ANullExp node)
{
InANullExp(node);
OutANullExp(node);
}
public virtual void InANullExp(ANullExp node)
{
DefaultIn(node);
}
private List<PStm> AssignDefault(PLvalue lvalue)
{
List<PStm> returner = new List<PStm>();
PType type = data.LvalueTypes[lvalue];
PExp rightSide = null;
if (type is ANamedType)
{
ANamedType aType = (ANamedType)type;
if (aType.IsPrimitive("string"))//aType.GetName().Text == "string")
{
rightSide = new AStringConstExp(new TStringLiteral("\"\""));
data.ExpTypes[rightSide] = new ANamedType(new TIdentifier("string"), null);
}
else if (aType.IsPrimitive(GalaxyKeywords.NullablePrimitives.words)) //GalaxyKeywords.NullablePrimitives.words.Contains(aType.GetName().Text))
{
rightSide = new ANullExp();
data.ExpTypes[rightSide] = new ANamedType(new TIdentifier("null"), null);
}
else if (aType.IsPrimitive(new []{"int", "byte", "fixed"}))
/*aType.GetName().Text == "int" ||
aType.GetName().Text == "byte" ||
aType.GetName().Text == "fixed")*/
{
rightSide = new AIntConstExp(new TIntegerLiteral("0"));
data.ExpTypes[rightSide] = type;
}
else if (aType.IsPrimitive("bool"))//aType.GetName().Text == "bool")
{
rightSide = new ABooleanConstExp(new AFalseBool());
data.ExpTypes[rightSide] = type;
}
else if (aType.IsPrimitive("color"))//aType.GetName().Text == "color")
{
PExp arg1 = new AIntConstExp(new TIntegerLiteral("0"));
PExp arg2 = new AIntConstExp(new TIntegerLiteral("0"));
PExp arg3 = new AIntConstExp(new TIntegerLiteral("0"));
ASimpleInvokeExp invoke = new ASimpleInvokeExp(new TIdentifier("Color"), new ArrayList() { arg1, arg2, arg3 });
rightSide = invoke;
data.ExpTypes[rightSide] = type;
data.ExpTypes[arg1] =
data.ExpTypes[arg2] =
data.ExpTypes[arg3] = new ANamedType(new TIdentifier("int"), null);
data.SimpleMethodLinks[invoke] =
data.Libraries.Methods.First(func => func.GetName().Text == invoke.GetName().Text);
}
else if (aType.IsPrimitive("char"))//aType.GetName().Text == "char")
{
//Dunno?!
rightSide = new ACharConstExp(new TCharLiteral("'\0'"));
data.ExpTypes[rightSide] = type;
}
else //Struct
{
AStructDecl str = data.StructTypeLinks[aType];
foreach (AALocalDecl localDecl in str.GetLocals())
{
ALvalueExp reciever = new ALvalueExp(Util.MakeClone(lvalue, data));
AStructLvalue newLvalue = new AStructLvalue(reciever, new ADotDotType(new TDot(".")), new TIdentifier(localDecl.GetName().Text));
data.StructFieldLinks[newLvalue] = localDecl;
data.ExpTypes[reciever] = type;
data.LvalueTypes[newLvalue] = localDecl.GetType();
returner.AddRange(AssignDefault(newLvalue));
}
return returner;
}
AAssignmentExp assignment = new AAssignmentExp(new TAssign("="), Util.MakeClone(lvalue, data), rightSide);
data.ExpTypes[assignment] = type;
return new List<PStm>() { new AExpStm(new TSemicolon(";"), assignment) };
}
if (type is AArrayTempType)
{
AArrayTempType aType = (AArrayTempType)type;
for (int i = 0; i < int.Parse(aType.GetIntDim().Text); i++)
{
ALvalueExp reciever = new ALvalueExp(Util.MakeClone(lvalue, data));
AArrayLvalue newLvalue = new AArrayLvalue(new TLBracket("["), reciever, new AIntConstExp(new TIntegerLiteral(i.ToString())));
data.ExpTypes[reciever] = type;
data.LvalueTypes[newLvalue] = aType.GetType();
data.ExpTypes[newLvalue.GetIndex()] = new ANamedType(new TIdentifier("int"), null);
returner.AddRange(AssignDefault(newLvalue));
}
return returner;
}
throw new Exception("Unexpected type. (LivenessAnalasys.AssignDefault), got " + type);
}
public virtual void CaseANullExp(ANullExp node)
{
DefaultCase(node);
}
public override void OutANullExp(ANullExp node)
{
data.ExpTypes[node] = new ANamedType(new TIdentifier("null"), null);
base.OutANullExp(node);
}
public override void CaseANullExp(ANullExp node)
{
Write("null");
}
ArrayList New326()
{
ArrayList nodeList = new ArrayList();
ArrayList nodeArrayList1 = (ArrayList) Pop();
ANullExp pexpNode1 = new ANullExp (
);
nodeList.Add(pexpNode1);
return nodeList;
}
private static AMethodDecl CreateGeneralNullCheckMethod(Node node, SharedData data)
{
if (generalCompareNullMethod != null)
return generalCompareNullMethod;
/*
* bool IsNull(string pointer)
* {
* if (pointer == null)
* {
* return true;
* }
* return !DataTableGetBool(true, pointer + "\\Exists");
* }
*/
AALocalDecl pointerDecl = new AALocalDecl(new APublicVisibilityModifier(), null, null, null, null,
new ANamedType(new TIdentifier("string"), null),
new TIdentifier("pointer"), null);
ALocalLvalue pointerRef1 = new ALocalLvalue(new TIdentifier("pointer"));
ALocalLvalue pointerRef2 = new ALocalLvalue(new TIdentifier("pointer"));
ALvalueExp pointerRef1Exp = new ALvalueExp(pointerRef1);
ALvalueExp pointerRef2Exp = new ALvalueExp(pointerRef2);
ANullExp nullExp = new ANullExp();
ABooleanConstExp boolConst1 = new ABooleanConstExp(new ATrueBool());
ABooleanConstExp boolConst2 = new ABooleanConstExp(new ATrueBool());
AStringConstExp stringConst = new AStringConstExp(new TStringLiteral("\"\\\\Exists\""));
ABinopExp binop1 = new ABinopExp(pointerRef1Exp, new AEqBinop(new TEq("==")), nullExp);
ABinopExp binop2 = new ABinopExp(pointerRef2Exp, new APlusBinop(new TPlus("+")), stringConst);
ASimpleInvokeExp dataTableGetBoolCall = new ASimpleInvokeExp(new TIdentifier("DataTableGetBool"), new ArrayList(){boolConst2, binop2});
AUnopExp unopExp = new AUnopExp(new AComplementUnop(new TComplement("!")), dataTableGetBoolCall);
generalCompareNullMethod = new AMethodDecl(new APublicVisibilityModifier(), null, null, null, null, null,
new ANamedType(new TIdentifier("bool"), null),
new TIdentifier("IsNull"), new ArrayList() {pointerDecl},
new AABlock(
new ArrayList()
{
new AIfThenStm(new TLParen("("), binop1,
new ABlockStm(new TLBrace("{"),
new AABlock(
new ArrayList()
{
new AValueReturnStm
(new TReturn
("return"),
boolConst1)
},
new TRBrace("}")))),
new AValueReturnStm(new TReturn("return"),
unopExp)
}, new TRBrace("}")));
AASourceFile sourceFile = Util.GetAncestor<AASourceFile>(node);
sourceFile.GetDecl().Add(generalCompareNullMethod);
data.LocalLinks[pointerRef1] =
data.LocalLinks[pointerRef2] = pointerDecl;
data.LvalueTypes[pointerRef1] =
data.LvalueTypes[pointerRef2] =
data.ExpTypes[pointerRef1Exp] =
data.ExpTypes[pointerRef2Exp] =
data.ExpTypes[stringConst] =
data.ExpTypes[binop2] = new ANamedType(new TIdentifier("string"), null);
data.ExpTypes[binop1] =
data.ExpTypes[boolConst1] =
data.ExpTypes[boolConst2] =
data.ExpTypes[dataTableGetBoolCall] =
data.ExpTypes[unopExp] = new ANamedType(new TIdentifier("string"), null);
foreach (AMethodDecl methodDecl in data.Libraries.Methods)
{
if (methodDecl.GetName().Text == dataTableGetBoolCall.GetName().Text)
{
data.SimpleMethodLinks[dataTableGetBoolCall] = methodDecl;
break;
}
}
return generalCompareNullMethod;
}
public override void CaseANullExp(ANullExp node)
{
Value += "null";
}
