mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-16 07:30:42 +01:00
Tools/connection_patcher: remove c# code, add c++ code with minor modification, move to tools, integrate into cmake
note: no longer downloads nonexistent modules note: now throws on not finding patterns new dependency: boost.filesystem Closes #13471
This commit is contained in:
bloerwald
197
src/tools/connection_patcher/Program.cpp
Normal file
197
src/tools/connection_patcher/Program.cpp
Normal file
@@ -0,0 +1,197 @@
|
||||
/*
|
||||
* 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/>.
|
||||
*/
|
||||
|
||||
#include "Helper.hpp"
|
||||
#include "Patcher.hpp"
|
||||
#include "Patches/Mac.hpp"
|
||||
#include "Patches/Windows.hpp"
|
||||
#include "Patterns/Mac.hpp"
|
||||
#include "Patterns/Windows.hpp"
|
||||
|
||||
#include <CompilerDefs.h>
|
||||
|
||||
#include <boost/algorithm/string/replace.hpp>
|
||||
#include <boost/filesystem.hpp>
|
||||
|
||||
#include <iostream>
|
||||
|
||||
#if PLATFORM == PLATFORM_WINDOWS
|
||||
#include <Shlobj.h>
|
||||
#endif
|
||||
|
||||
namespace Connection_Patcher
|
||||
{
|
||||
namespace
|
||||
{
|
||||
template<typename PATCH, typename PATTERN>
|
||||
void PatchModule(boost::filesystem::path file, boost::filesystem::path path)
|
||||
{
|
||||
std::cout << "Patching module...\n";
|
||||
|
||||
Patcher patcher(file);
|
||||
std::cout << "patching Password\n";
|
||||
patcher.Patch(PATCH::Password(), PATTERN::Password());
|
||||
|
||||
std::string const moduleName(Helper::GetFileChecksum(patcher.binary) + ".auth");
|
||||
boost::filesystem::path const modulePath
|
||||
(path / std::string(&moduleName[0], 2) / std::string(&moduleName[2], 2));
|
||||
|
||||
if (!boost::filesystem::exists(modulePath))
|
||||
boost::filesystem::create_directories(modulePath);
|
||||
|
||||
patcher.Finish(modulePath / moduleName);
|
||||
|
||||
std::cout << "Patching module finished.\n";
|
||||
}
|
||||
|
||||
template<typename PATCH, typename PATTERN>
|
||||
void do_module(std::string moduleName, boost::filesystem::path path)
|
||||
{
|
||||
boost::filesystem::path const modulePath
|
||||
(path / std::string(&moduleName[0], 2) / std::string(&moduleName[2], 2));
|
||||
boost::filesystem::path const module(modulePath / moduleName);
|
||||
|
||||
if (!boost::filesystem::exists (module))
|
||||
throw std::runtime_error("base module does not exist. run client once.");
|
||||
|
||||
PatchModule<PATCH, PATTERN>(module, path);
|
||||
}
|
||||
|
||||
template<typename PATCH, typename PATTERN>
|
||||
void do_patches(Patcher* patcher, boost::filesystem::path output)
|
||||
{
|
||||
std::cout << "patching BNet\n";
|
||||
patcher->Patch(PATCH::BNet(), PATTERN::BNet());
|
||||
std::cout << "patching Portal\n";
|
||||
patcher->Patch(PATCH::Portal(), PATTERN::Portal());
|
||||
std::cout << "patching Connect\n";
|
||||
patcher->Patch(PATCH::Connect(), PATTERN::Connect());
|
||||
std::cout << "patching Signature\n";
|
||||
patcher->Patch(PATCH::Signature(), PATTERN::Signature());
|
||||
patcher->Finish(output);
|
||||
|
||||
std::cout << "Patching done.\n";
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// adapted from http://stackoverflow.com/questions/8593608/how-can-i-copy-a-directory-using-boost-filesystem
|
||||
void copyDir(boost::filesystem::path const & source, boost::filesystem::path const & destination)
|
||||
{
|
||||
namespace fs = boost::filesystem;
|
||||
if (!fs::exists(source) || !fs::is_directory(source))
|
||||
throw std::invalid_argument("Source directory " + source.string() + " does not exist or is not a directory.");
|
||||
|
||||
if (fs::exists(destination))
|
||||
throw std::invalid_argument("Destination directory " + destination.string() + " already exists.");
|
||||
|
||||
if (!fs::create_directory(destination))
|
||||
throw std::runtime_error("Unable to create destination directory" + destination.string());
|
||||
|
||||
for (fs::directory_iterator file(source); file != fs::directory_iterator(); ++file)
|
||||
{
|
||||
fs::path current(file->path());
|
||||
if (fs::is_directory(current))
|
||||
copyDir(current, destination / current.filename());
|
||||
else
|
||||
fs::copy_file(current, destination / current.filename());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
using namespace Connection_Patcher;
|
||||
|
||||
try
|
||||
{
|
||||
if (argc != 2)
|
||||
throw std::invalid_argument("Wrong number of arguments: Missing client file.");
|
||||
|
||||
std::string const binary_path(argv[1]);
|
||||
|
||||
std::string renamed_binary_path(binary_path);
|
||||
|
||||
wchar_t* commonAppData (nullptr);
|
||||
#if PLATFORM == PLATFORM_WINDOWS
|
||||
SHGetKnownFolderPath(FOLDERID_ProgramData, 0, NULL, &commonAppData);
|
||||
#endif
|
||||
|
||||
std::cout << "Creating patched binaries for ";
|
||||
|
||||
Patcher patcher(binary_path);
|
||||
|
||||
switch (patcher.Type)
|
||||
{
|
||||
case Constants::BinaryTypes::Pe32:
|
||||
std::cout << "Win32 client...\n";
|
||||
|
||||
boost::algorithm::replace_all(renamed_binary_path, ".exe", "_Patched.exe");
|
||||
do_patches<Patches::Windows::x86, Patterns::Windows::x86>
|
||||
(&patcher, renamed_binary_path);
|
||||
|
||||
do_module<Patches::Windows::x86, Patterns::Windows::x86>
|
||||
( "8f52906a2c85b416a595702251570f96d3522f39237603115f2f1ab24962043c.auth"
|
||||
, std::wstring(commonAppData) + std::wstring(L"/Blizzard Entertainment/Battle.net/Cache/")
|
||||
);
|
||||
|
||||
break;
|
||||
case Constants::BinaryTypes::Pe64:
|
||||
std::cout << "Win64 client...\n";
|
||||
|
||||
boost::algorithm::replace_all(renamed_binary_path, ".exe", "_Patched.exe");
|
||||
do_patches<Patches::Windows::x64, Patterns::Windows::x64>
|
||||
(&patcher, renamed_binary_path);
|
||||
|
||||
do_module<Patches::Windows::x64, Patterns::Windows::x64>
|
||||
( "0a3afee2cade3a0e8b458c4b4660104cac7fc50e2ca9bef0d708942e77f15c1d.auth"
|
||||
, std::wstring(commonAppData) + std::wstring(L"/Blizzard Entertainment/Battle.net/Cache/")
|
||||
);
|
||||
|
||||
break;
|
||||
case Constants::BinaryTypes::Mach64:
|
||||
std::cout << "Mac client...\n";
|
||||
|
||||
boost::algorithm::replace_all (renamed_binary_path, ".app", " Patched.app");
|
||||
copyDir ( boost::filesystem::path(binary_path).parent_path()/*MacOS*/.parent_path()/*Contents*/.parent_path()
|
||||
, boost::filesystem::path(renamed_binary_path).parent_path()/*MacOS*/.parent_path()/*Contents*/.parent_path()
|
||||
);
|
||||
|
||||
do_patches<Patches::Mac::x64, Patterns::Mac::x64>
|
||||
(&patcher, renamed_binary_path);
|
||||
|
||||
do_module<Patches::Windows::x64, Patterns::Windows::x64>
|
||||
( "97eeb2e28e9e56ed6a22d09f44e2ff43c93315e006bbad43bafc0defaa6f50ae.auth"
|
||||
, "/Users/Shared/Blizzard/Battle.net/Cache/"
|
||||
);
|
||||
|
||||
break;
|
||||
default:
|
||||
throw std::runtime_error("Type: " + std::to_string(static_cast<uint32_t>(patcher.Type)) + " not supported!");
|
||||
}
|
||||
|
||||
std::cout << "Successfully created your patched binaries.\n";
|
||||
|
||||
return 0;
|
||||
}
|
||||
catch (std::exception const& ex)
|
||||
{
|
||||
std::cerr << "EX: " << ex.what() << std::endl;
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user