diff options
Diffstat (limited to 'contrib/Connection Patcher/Program.cs')
| -rw-r--r-- | contrib/Connection Patcher/Program.cs | 196 |
1 files changed, 196 insertions, 0 deletions
diff --git a/contrib/Connection Patcher/Program.cs b/contrib/Connection Patcher/Program.cs new file mode 100644 index 00000000000..956741b625b --- /dev/null +++ b/contrib/Connection Patcher/Program.cs @@ -0,0 +1,196 @@ +/* + * 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 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 = args[0].Replace(".exe", "_Patched.exe"); + var battleNetFileName = args[0].Replace("Wow.exe", "Battle.net.dll"); + var modulePatch = Patches.Windows.x86.Password; + var modulePattern = Patterns.Windows.x86.Password; + var realmListPatch = Patches.Windows.x86.RealmList; + var realmListoffset = Offsets.Windows.x86.RealmList; + var realmListBnPatch = Patches.Windows.x86.RealmListBn; + var realmListBnPattern = Patterns.Windows.x86.RealmListBn; + + 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; + // 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.Patch(realmListBnPatch, realmListBnPattern); + bnpatcher.Finish(); + } + + patcher.Patch(patchSend, null, offsetSend); + patcher.Patch(patchRecv, null, offsetRecv); + patcher.Patch(realmListPatch, null, realmListoffset); + + 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."); + } + } + 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; + } + } +} |