blob: 794db98ec9b61ed2ea8fd849527b7ef58a6c1421 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System;
using Mono.Cecil.Cil;
using Mono.Cecil.Rocks;
using Mono.Cecil;
using CommandLine;
class Command {
protected string infile;
protected string outfile;
protected ModuleDefinition module;
public Command(GenericOptions options) {
if (options.outputFile == null)
this.outfile = options.inputFile;
else
this.outfile = options.outputFile;
this.infile = options.inputFile;
var rp = new ReaderParameters {
ReadWrite = this.infile == this.outfile
};
this.module = ModuleDefinition.ReadModule(this.infile, rp);
}
public void save() {
if (this.outfile == this.infile)
this.module.Write();
else
this.module.Write(this.outfile);
}
}
class FixFileStreams : Command {
public FixFileStreams(FixFileStreamsCmd options) : base(options) {
var filtered = this.module.Types
.Where(p => options.typesToPatch.Contains(p.Name));
foreach (var toPatch in filtered)
patch_type(toPatch);
this.save();
}
private void patch_method(MethodDefinition md) {
var il = md.Body.GetILProcessor();
var fileStreams = md.Body.Instructions
.Where(i => i.OpCode == OpCodes.Newobj)
.Where(i => (i.Operand as MethodReference).DeclaringType
.FullName == "System.IO.FileStream");
foreach (Instruction i in fileStreams.ToList()) {
var fileAccessRead = il.Create(OpCodes.Ldc_I4_1);
il.InsertBefore(i, fileAccessRead);
var ctorType = this.module.AssemblyReferences.Select(
x => new {
type = this.module.AssemblyResolver.Resolve(x)
.MainModule.GetType("System.IO.FileStream")
}
).Where(x => x.type != null).Select(x => x.type).First();
string wantedCtor = "System.Void System.IO.FileStream"
+ "::.ctor(System.String,"
+ "System.IO.FileMode,"
+ "System.IO.FileAccess)";
var newCtor = ctorType.GetConstructors()
.Single(x => x.ToString() == wantedCtor);
var refCtor = this.module.ImportReference(newCtor);
il.Replace(i, il.Create(OpCodes.Newobj, refCtor));
}
}
private void patch_type(TypeDefinition td) {
foreach (var nested in td.NestedTypes) patch_type(nested);
foreach (MethodDefinition md in td.Methods) patch_method(md);
}
}
class ReplaceCall : Command {
private string search;
private string replace;
public ReplaceCall(ReplaceCallCmd options) : base(options) {
this.search = options.replaceCall.ToList()[0];
this.replace = options.replaceCall.ToList()[1];
var filtered = this.module.Types
.Where(p => options.typesToPatch.Contains(p.Name));
foreach (var toPatch in filtered)
patch_type(toPatch);
this.save();
}
private void patch_method(MethodDefinition md) {
var il = md.Body.GetILProcessor();
var found = md.Body.Instructions
.Where(i => i.OpCode == OpCodes.Call)
.Where(i => i.Operand.ToString() == this.search);
var ctorType = this.module.AssemblyReferences.Select(
x => new {
// XXX: Don't hardcode UnityEngine.Application!
type = this.module.AssemblyResolver.Resolve(x)
.MainModule.GetType("UnityEngine.Application")
}
).Where(x => x.type != null).Select(x => x.type).First();
var newMethod = ctorType.GetMethods()
.Single(x => x.ToString() == this.replace);
var refMethod = this.module.ImportReference(newMethod);
foreach (Instruction i in found.ToList())
il.Replace(i, il.Create(OpCodes.Call, refMethod));
}
private void patch_type(TypeDefinition td) {
foreach (var nested in td.NestedTypes) patch_type(nested);
foreach (MethodDefinition md in td.Methods) patch_method(md);
}
}
public class patcher {
public static int Main(string[] args) {
Parser.Default.ParseArguments<FixFileStreamsCmd, ReplaceCallCmd>(args)
.WithParsed<FixFileStreamsCmd>(opts => new FixFileStreams(opts))
.WithParsed<ReplaceCallCmd>(opts => new ReplaceCall(opts));
return 0;
}
}
|
