Skip to content

Commit 80dadb2

Browse files
mkaringmkaring
authored
Merge a7d0821 into e2f8d09 
2 parents e2f8d09 + a7d0821 commit 80dadb2

File tree

6 files changed

+113
-9
lines changed

6 files changed

+113
-9
lines changed

‎Confuser.Core/Helpers/MutationHelper.cs‎

Lines changed: 21 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -86,22 +86,34 @@ public static void ReplacePlaceholder(MethodDef method, Func<Instruction[], Inst
8686
var operand = (IMethod)instr.Operand;
8787
if (operand.DeclaringType.FullName == mutationType &&
8888
operand.Name == "Placeholder") {
89-
int[] argIndexes = trace.TraceArguments(instr);
90-
if (argIndexes == null)
91-
throw new ArgumentException("Failed to trace placeholder argument.");
89+
var initialLoadInstructions = new List<Instruction>();
90+
var pendingInstructions = new Queue<Instruction>();
91+
pendingInstructions.Enqueue(instr);
92+
while (pendingInstructions.Count > 0) {
93+
var currentInstr = pendingInstructions.Dequeue();
94+
int[] argIndexes = trace.TraceArguments(currentInstr);
95+
if (argIndexes == null)
96+
throw new ArgumentException("Failed to trace placeholder argument.");
9297

93-
int argIndex = argIndexes[0];
94-
Instruction[] arg = method.Body.Instructions.Skip(argIndex).Take(i - argIndex).ToArray();
98+
if (argIndexes.Length == 0)
99+
initialLoadInstructions.Add(currentInstr);
100+
101+
foreach (int argIndex in argIndexes)
102+
pendingInstructions.Enqueue(method.Body.Instructions[argIndex]);
103+
}
104+
105+
var firstArgIndex = initialLoadInstructions.Select(method.Body.Instructions.IndexOf).Min();
106+
Instruction[] arg = method.Body.Instructions.Skip(firstArgIndex).Take(i - firstArgIndex).ToArray();
95107
for (int j = 0; j < arg.Length; j++)
96-
method.Body.Instructions.RemoveAt(argIndex);
97-
method.Body.Instructions.RemoveAt(argIndex);
108+
method.Body.Instructions.RemoveAt(firstArgIndex);
109+
method.Body.Instructions.RemoveAt(firstArgIndex);
98110
arg = repl(arg);
99111
for (int j = arg.Length - 1; j >= 0; j--)
100-
method.Body.Instructions.Insert(argIndex, arg[j]);
112+
method.Body.Instructions.Insert(firstArgIndex, arg[j]);
101113
return;
102114
}
103115
}
104116
}
105117
}
106118
}
107-
}
119+
}

‎Confuser2.sln‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,10 @@ Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "270_EnumArrayConstantProtec
103103
EndProject
104104
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "270_EnumArrayConstantProtection.Test", "Tests\270_EnumArrayConstantProtection.Test\270_EnumArrayConstantProtection.Test.csproj", "{DB234158-233E-4EC4-A2CE-EF02699563A2}"
105105
EndProject
106+
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "ReferenceProxy", "Tests\ReferenceProxy\ReferenceProxy.csproj", "{FE068381-F170-4C37-82C4-11A81FE60F1A}"
107+
EndProject
108+
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "ReferenceProxy.Test", "Tests\ReferenceProxy.Test\ReferenceProxy.Test.csproj", "{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1}"
109+
EndProject
106110
Global
107111
GlobalSection(SolutionConfigurationPlatforms) = preSolution
108112
Debug|Any CPU = Debug|Any CPU
@@ -281,6 +285,14 @@ Global
281285
{DB234158-233E-4EC4-A2CE-EF02699563A2}.Debug|Any CPU.Build.0 = Debug|Any CPU
282286
{DB234158-233E-4EC4-A2CE-EF02699563A2}.Release|Any CPU.ActiveCfg = Release|Any CPU
283287
{DB234158-233E-4EC4-A2CE-EF02699563A2}.Release|Any CPU.Build.0 = Release|Any CPU
288+
{FE068381-F170-4C37-82C4-11A81FE60F1A}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
289+
{FE068381-F170-4C37-82C4-11A81FE60F1A}.Debug|Any CPU.Build.0 = Debug|Any CPU
290+
{FE068381-F170-4C37-82C4-11A81FE60F1A}.Release|Any CPU.ActiveCfg = Release|Any CPU
291+
{FE068381-F170-4C37-82C4-11A81FE60F1A}.Release|Any CPU.Build.0 = Release|Any CPU
292+
{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
293+
{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1}.Debug|Any CPU.Build.0 = Debug|Any CPU
294+
{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1}.Release|Any CPU.ActiveCfg = Release|Any CPU
295+
{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1}.Release|Any CPU.Build.0 = Release|Any CPU
284296
EndGlobalSection
285297
GlobalSection(SolutionProperties) = preSolution
286298
HideSolutionNode = FALSE
@@ -321,6 +333,8 @@ Global
321333
{C10599E3-5A79-484F-940B-E4B61F256466} = {356BDB31-853E-43BB-8F9A-D8AC08F69EBB}
322334
{7C6D1CCD-D4DF-426A-B5D6-A6B5F13D0091} = {356BDB31-853E-43BB-8F9A-D8AC08F69EBB}
323335
{DB234158-233E-4EC4-A2CE-EF02699563A2} = {356BDB31-853E-43BB-8F9A-D8AC08F69EBB}
336+
{FE068381-F170-4C37-82C4-11A81FE60F1A} = {356BDB31-853E-43BB-8F9A-D8AC08F69EBB}
337+
{2C059FE7-C868-4C6D-AFA0-D62BA3C1B2E1} = {356BDB31-853E-43BB-8F9A-D8AC08F69EBB}
324338
EndGlobalSection
325339
GlobalSection(ExtensibilityGlobals) = postSolution
326340
SolutionGuid = {0D937D9E-E04B-4A68-B639-D4260473A388}

‎Tests/ReferenceProxy.Test/ReferenceProxy.Test.csproj‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
<Project Sdk="Microsoft.NET.Sdk">
2+
3+
<PropertyGroup>
4+
<TargetFramework>net461</TargetFramework>
5+
<IsPackable>False</IsPackable>
6+
</PropertyGroup>
7+
8+
<ItemGroup>
9+
<ProjectReference Include="..\Confuser.UnitTest\Confuser.UnitTest.csproj" />
10+
<ProjectReference Include="..\ReferenceProxy\ReferenceProxy.csproj" />
11+
</ItemGroup>
12+
13+
14+
</Project>

‎Tests/ReferenceProxy.Test/ReferenceProxyTest.cs‎

Lines changed: 42 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,42 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Threading.Tasks;
4+
using Confuser.Core;
5+
using Confuser.Core.Project;
6+
using Confuser.UnitTest;
7+
using Xunit;
8+
using Xunit.Abstractions;
9+
10+
namespace ReferenceProxy.Test {
11+
public class ReferenceProxyTest : TestBase {
12+
public ReferenceProxyTest(ITestOutputHelper outputHelper) : base(outputHelper) { }
13+
14+
[Theory]
15+
[MemberData(nameof(ReferenceProxyData))]
16+
[Trait("Category", "Protection")]
17+
[Trait("Protection", "ref proxy")]
18+
[Trait("Issue", "https://github.com/mkaring/ConfuserEx/issues/229")]
19+
public async Task ReferenceProxy(string mode, string encoding, bool internalRefs, bool typeErasure) =>
20+
await Run(
21+
"ReferenceProxy.exe",
22+
Array.Empty<string>(),
23+
new SettingItem<Protection>("ref proxy") {
24+
["mode"] = mode,
25+
["encoding"] = encoding,
26+
["internal"] = internalRefs.ToString(),
27+
["typeErasure"] = typeErasure.ToString()
28+
},
29+
outputDirSuffix: $"_{mode}_{encoding}_{internalRefs}_{typeErasure}"
30+
);
31+
32+
public static IEnumerable<object[]> ReferenceProxyData() {
33+
foreach (var mode in new[] { "mild", "strong" })
34+
foreach (var encoding in new[] { "normal", "expression", "x86" })
35+
foreach (var internalRefs in new[] { true, false })
36+
foreach (var typeErasure in new[] { true, false }) {
37+
if (mode.Equals("mild") && !encoding.Equals("normal")) continue;
38+
yield return new object[] { mode, encoding, internalRefs, typeErasure };
39+
}
40+
}
41+
}
42+
}

‎Tests/ReferenceProxy/Program.cs‎

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
using System;
2+
3+
namespace ReferenceProxy {
4+
class Program {
5+
static int Main(string[] args) {
6+
Console.WriteLine("START");
7+
foreach (var arg in args)
8+
Console.WriteLine(arg);
9+
Console.WriteLine("END");
10+
return 42;
11+
}
12+
}
13+
}

‎Tests/ReferenceProxy/ReferenceProxy.csproj‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
<Project Sdk="Microsoft.NET.Sdk">
2+
3+
<PropertyGroup>
4+
<OutputType>Exe</OutputType>
5+
<TargetFramework>net461</TargetFramework>
6+
<StartupObject>ReferenceProxy.Program</StartupObject>
7+
</PropertyGroup>
8+
9+
</Project>

0 commit comments

Comments
 (0)