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

Timestamp:

Jul 19, 2014, 12:47:32 PM (10 years ago)

Author:

alloc

Message:

fixes

File:

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

binary-improvements/assembly-patcher/Main.cs

@@ -14 +14 @@
                         connectLogPatch (module);
                         executionLogPatch( module);
+                        publicCommandPermissionsPatch (module);
                         module.Write ("Assembly-CSharp.dll");
                         Console.WriteLine ("Done");
+                }
+
+                private static void publicCommandPermissionsPatch (ModuleDefinition module)
+                {
+                        TypeDefinition type = module.GetType ("AdminTools");
+
+                        if (isPatched (type)) {
+                                return;
+                        }
+
+                        markTypePatched (module, type);
+                        replaceMethod (type, "GetAllowedCommandsList", true, 1, typeof(AdminToolsStuff).GetMethod ("GetAllowedCommandsList"));
                 }
 
@@ -51 +64 @@
 
                         markTypePatched (module, type);
-                        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"));
+                        replaceMethod (type, ".ctor", false, 1, typeof(AllocsNetTelnetServer).GetMethod ("init"));
+                        replaceMethod (type, "Disconnect", false, 0, typeof(AllocsNetTelnetServer).GetMethod ("Disconnect"));
+                        replaceMethod (type, "SetConsole", false, 1, typeof(AllocsNetTelnetServer).GetMethod ("SetConsole"));
+                        replaceMethod (type, "WriteToClient", false, 1, typeof(AllocsNetTelnetServer).GetMethod ("WriteToClient"));
                 }
 
@@ -92 +105 @@
                 }
 
-                private static void replaceMethod (TypeDefinition type, string methodName, int opCount, MethodBase targetMethod)
+                private static void replaceMethod (TypeDefinition type, string methodName, bool addThisRef, int opCount, MethodBase targetMethod)
                 {
                         foreach (MethodDefinition method in type.Methods) {
@@ -100 +113 @@
                                         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));