245 lines
10 KiB
C#
245 lines
10 KiB
C#
/*
|
|
* 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;
|
|
}
|
|
}
|
|
}
|