Changeset 75 for binary-improvements/assembly-patcher/Main.cs

Timestamp:

Jul 16, 2014, 7:36:46 PM (10 years ago)

Author:

alloc

Message:

fixes

File:

• binary-improvements/assembly-patcher/Main.cs (modified) (3 diffs)

binary-improvements/assembly-patcher/Main.cs

@@ -12 +12 @@
                         ModuleDefinition module = ModuleDefinition.ReadModule ("Assembly-CSharp.dll");
                         telnetPatch (module);
+                        connectLogPatch(module);
                         module.Write ("Assembly-CSharp.dll");
                         Console.WriteLine ("Done");
+                }
+
+                private static void connectLogPatch (ModuleDefinition module)
+                {
+                        TypeDefinition type = module.GetType ("GameManager");
+
+                        if (isPatched (type)) {
+                                return;
+                        }
+
+                        markTypePatched (module, type);
+                        addHook (type, "RequestToSpawnPlayer", true, 5, typeof(AllocsRequestToSpawnPlayer).GetMethod ("RequestToSpawnPlayer"));
                 }
 
@@ -25 +38 @@
 
                         markTypePatched (module, type);
-                        patchMethod (type, ".ctor", 1, typeof(AllocsNetTelnetServer).GetMethod ("init"));
-                        patchMethod (type, "Disconnect", 0, typeof(AllocsNetTelnetServer).GetMethod ("Disconnect"));
-                        patchMethod (type, "SetConsole", 1, typeof(AllocsNetTelnetServer).GetMethod ("SetConsole"));
-                        patchMethod (type, "WriteToClient", 1, typeof(AllocsNetTelnetServer).GetMethod ("WriteToClient"));
+                        replaceMethod (type, ".ctor", 1, typeof(AllocsNetTelnetServer).GetMethod ("init"));
+                        replaceMethod (type, "Disconnect", 0, typeof(AllocsNetTelnetServer).GetMethod ("Disconnect"));
+                        replaceMethod (type, "SetConsole", 1, typeof(AllocsNetTelnetServer).GetMethod ("SetConsole"));
+                        replaceMethod (type, "WriteToClient", 1, typeof(AllocsNetTelnetServer).GetMethod ("WriteToClient"));
                 }
 
@@ -36 +49 @@
                 }
 
-                private static void patchMethod (TypeDefinition type, string methodName, int opCount, MethodBase targetMethod)
+                private static void addHook (TypeDefinition type, string methodName, bool addThisRef, int opCount, MethodBase targetMethod)
+                {
+                        foreach (MethodDefinition method in type.Methods) {
+                                if (method.Name.Equals (methodName)) {
+                                        Console.WriteLine ("Patching " + methodName);
+                                        var il = method.Body.GetILProcessor ();
+                                        var call = il.Create (OpCodes.Call, method.Module.Import (targetMethod));
+                                        var i = 0;
+                                        if (addThisRef)
+                                                il.InsertBefore (method.Body.Instructions [i++], il.Create (OpCodes.Ldarg, 0));
+                                        for (int op = 0; op < opCount; op++) {
+                                                il.InsertBefore (method.Body.Instructions [i++], il.Create (OpCodes.Ldarg, op + 1));
+                                        }
+                                        il.InsertBefore (method.Body.Instructions [i++], call);
+                                }
+                        }
+                }
+
+                private static void replaceMethod (TypeDefinition type, string methodName, int opCount, MethodBase targetMethod)
                 {
                         foreach (MethodDefinition method in type.Methods) {