Contrib: Added client patcher, original author @Fabi

1 files changed, 244 insertions, 0 deletions

diff --git a/contrib/Connection Patcher/Program.cs b/contrib/Connection Patcher/Program.cs
new file mode 100644
index 00000000000..5ba7b3e0a87
--- /dev/null
+++ b/contrib/Connection Patcher/Program.cs
@@ -0,0 +1,244 @@
+/*
+ * Copyright (C) 2012-2014 Arctium Emulation <http://arctium.org>
+ * Copyright (C) 2008-2014 TrinityCore <http://www.trinitycore.org/>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+using Connection_Patcher.Constants;
+using System;
+using System.IO;
+using System.Net;
+using System.Text;
+using System.Threading;
+
+namespace Connection_Patcher
+{
+    class Program
+    {
+        static void Main(string[] args)
+        {
+            if (args.Length >= 1)
+            {
+                Console.ForegroundColor = ConsoleColor.Cyan;
+
+                Console.WriteLine("Client Connection Patcher");
+                Console.WriteLine("Press Enter to patch...");
+
+                Console.ReadKey(true);
+
+                var commonAppData = Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData);
+                var system32      = Environment.GetFolderPath(Environment.SpecialFolder.System);
+                var hostsPath     = Path.Combine(system32, "drivers/etc/hosts");
+                var modulePath    = "";
+                var moduleFile    = "";
+
+                // Let's use Win32 as default module
+                var patchSend         = Patches.Windows.x86.Send;
+                var offsetSend        = Offsets.Windows.x86.Send;
+                var patchRecv         = Patches.Windows.x86.Recv;
+                var offsetRecv        = Offsets.Windows.x86.Recv;
+                var patchBNet         = Patches.Windows.x86.BNet;
+                var offsetBNet        = Offsets.Windows.x86.BNet;
+                var patchSignature    = Patches.Windows.x86.Signature;
+                var offsetSignature   = Offsets.Windows.x86.Signature;
+                var fileName          = "";
+                var battleNetFileName = args[0].Replace("Wow.exe", "Battle.net.dll");
+                var modulePatch       = Patches.Windows.x86.Password;
+                var modulePattern     = Patterns.Windows.x86.Password;
+
+                Console.ForegroundColor = ConsoleColor.White;
+                Console.Write("Creating patched binaries for ");
+
+                using (var patcher = new Patcher(args[0]))
+                {
+                    switch (patcher.Type)
+                    {
+                        case BinaryTypes.Pe32:
+                            Console.WriteLine("Win32 client...");
+                            modulePath        = commonAppData + "/Blizzard Entertainment/Battle.net/Cache/";
+                            moduleFile        = "8f52906a2c85b416a595702251570f96d3522f39237603115f2f1ab24962043c.auth";
+                            break;
+                        //case BinaryTypes.Pe64:
+                        //    Console.WriteLine("Win64 client...");
+                        //    fileName   = patcher.Binary.Replace(".exe", "") + "_Patched.exe";
+
+                        //    modulePath = commonAppData + "/Blizzard Entertainment/Battle.net/Cache/";
+                        //    moduleFile = "0a3afee2cade3a0e8b458c4b4660104cac7fc50e2ca9bef0d708942e77f15c1d.auth";
+                        //    break;
+                        //case BinaryTypes.Mach32:
+                        //    break;
+                        //case BinaryTypes.Mach64:
+                        //    Console.WriteLine("Mc64 client...");
+                        //    patchBNet        = Patches.Mac.x64.BNet;
+                        //    patternBNet      = Patterns.Mac.x64.BNet;
+                        //    patchSend        = Patches.Mac.x64.Send;
+                        //    patternSend      = Patterns.Mac.x64.Send;
+                        //    patchSignature   = Patches.Mac.x64.Signature;
+                        //    patternSignature = Patterns.Mac.x64.Signature;
+                        //    fileName         = patcher.Binary + " Patched";
+
+                        //    modulePath       = "/Users/Shared/Blizzard/Battle.net/Cache/";
+                        //    moduleFile       = "97eeb2e28e9e56ed6a22d09f44e2ff43c93315e006bbad43bafc0defaa6f50ae.auth";
+                        //    modulePatch      = Patches.Mac.x64.Password;
+                        //    modulePattern    = Patterns.Mac.x64.Password;
+                        //    hostsPath        = "/private/etc/hosts";
+                        //    break;
+                        default:
+                            throw new NotSupportedException("Type: " + patcher.Type + " not supported!");
+                    }
+
+                    if (!File.Exists(battleNetFileName + "_backup"))
+                    {
+                        File.Copy(battleNetFileName, battleNetFileName + "_backup");
+                        File.SetAttributes(battleNetFileName + "_backup", File.GetAttributes(battleNetFileName + "_backup") | FileAttributes.ReadOnly);
+                    }
+
+                    using (var bnpatcher = new Patcher(battleNetFileName))
+                    {
+                        bnpatcher.Patch(patchBNet, null, offsetBNet);
+                        bnpatcher.Patch(patchSignature, null, offsetSignature);
+                        bnpatcher.Finish();
+                    }
+
+                    patcher.Patch(patchSend, null, offsetSend);
+                    patcher.Patch(patchRecv, null, offsetRecv);
+
+                    patcher.Binary = fileName;
+
+                    patcher.Finish();
+
+                    Console.ForegroundColor = ConsoleColor.Green;
+                    Console.WriteLine("Patching done.");
+
+                    CreateModule(moduleFile, modulePath, modulePatch, modulePattern);
+
+                    Console.ForegroundColor = ConsoleColor.Green;
+                    Console.WriteLine("Successfully created your patched binaries.");
+                }
+
+                Console.ForegroundColor = ConsoleColor.White;
+                Console.WriteLine("Adding host rewrite...");
+
+                var host = args.Length == 2 ? args[1] : "127.0.0.1";
+                var hostName = args.Length == 3 ? (" " + args[2] + ".logon.battle.net") : " tc.logon.battle.net";
+                var exists = false;
+
+                using (var sr = new StreamReader(hostsPath))
+                {
+                    while (!sr.EndOfStream)
+                    {
+                        var line = sr.ReadLine();
+
+                        if (line.Contains(hostName))
+                        {
+                            Console.ForegroundColor = ConsoleColor.Yellow;
+                            Console.WriteLine("Host rewrite not needed.");
+
+                            exists = true;
+                            break;
+                        }
+                    }
+                }
+
+                if (!exists)
+                {
+                    try
+                    {
+                        using (var stream = new StreamWriter(hostsPath, true, Encoding.UTF8))
+                        {
+                            stream.WriteLine("");
+                            stream.WriteLine("{0}{1}", host, hostName);
+                        }
+
+                        Console.ForegroundColor = ConsoleColor.Green;
+                        Console.WriteLine("Host rewrite successfully added.");
+                    }
+                    catch (Exception e)
+                    {
+                        Console.ForegroundColor = ConsoleColor.Red;
+                        Console.WriteLine(string.Format("Can't write host file! Exception type: {0}", e.GetType()));
+                        Console.WriteLine("You must add the following line:");
+
+                        Console.ForegroundColor = ConsoleColor.White;
+                        Console.WriteLine("{0}{1}", host, hostName);
+
+                        Console.ForegroundColor = ConsoleColor.Red;
+                        Console.WriteLine("to your host file before using Arctium WoD Sandbox!");
+                    }
+                }
+            }
+            else
+            {
+                Console.ForegroundColor = ConsoleColor.Red;
+                Console.WriteLine("Wrong number of arguments: Missing client file.");
+            }
+
+            Console.ForegroundColor = ConsoleColor.Gray;
+            Thread.Sleep(5000);
+            Environment.Exit(0);
+        }
+
+        static void CreateModule(string moduleName, string path, byte[] patches, byte[] patterns)
+        {
+            var modulePath = path + moduleName[0] + moduleName[1] + "/" + moduleName[2] + moduleName[3];
+            var module = modulePath + "/" + moduleName;
+
+            if (!File.Exists(module))
+            {
+                Console.ForegroundColor = ConsoleColor.White;
+                Console.WriteLine("Base module doesn't exist, downloading it...");
+
+                if (!Directory.Exists(modulePath))
+                    Directory.CreateDirectory(modulePath);
+
+                var webClient = new WebClient();
+
+                webClient.DownloadFileCompleted += (o, e) => PatchModule(module, path, patches, patterns);
+                webClient.DownloadFileAsync(new Uri("http://xx.depot.battle.net:1119/" + moduleName), module);
+
+                Console.ForegroundColor = ConsoleColor.Green;
+                Console.WriteLine("Done.");
+            }
+            else
+                PatchModule(module, path, patches, patterns);
+        }
+
+        static void PatchModule(string file, string path, byte[] patches, byte[] pattern)
+        {
+            Console.ForegroundColor = ConsoleColor.White;
+            Console.WriteLine("Patching module...");
+
+            using (var patcher2 = new Patcher(file))
+            {
+                patcher2.Patch(patches, pattern);
+
+                var moduleName = Helper.GetFileChecksum(patcher2.binary) + ".auth";
+                var modulePath = path + moduleName[0] + moduleName[1] + "/" + moduleName[2] + moduleName[3];
+
+                if (!Directory.Exists(modulePath))
+                    Directory.CreateDirectory(modulePath);
+
+                patcher2.Binary = modulePath + "/" + moduleName;
+
+                patcher2.Finish();
+            }
+
+            Console.ForegroundColor = ConsoleColor.Green;
+            Console.WriteLine("Patching module finished.");
+
+            Console.ForegroundColor = ConsoleColor.Gray;
+        }
+    }
+}