Program.cpp File Reference

#include "Helper.hpp"
#include "Patcher.hpp"
#include "Patches/Common.hpp"
#include "Patches/Mac.hpp"
#include "Patches/Windows.hpp"
#include "Patterns/Common.hpp"
#include "Patterns/Mac.hpp"
#include "Patterns/Windows.hpp"
#include <CompilerDefs.h>
#include <boost/algorithm/string/replace.hpp>
#include <boost/program_options.hpp>
#include <Shlobj.h>

Include dependency graph for Program.cpp:

Namespaces
	Connection_Patcher

Functions
po::variables_map	Connection_Patcher::GetConsoleArguments (int argc, char **argv)
int	main (int argc, char **argv)

Function Documentation

int main	(	int	argc,
		char **	argv
	)

{
 using namespace Connection_Patcher;

 try
 {
 auto vm = GetConsoleArguments(argc, argv);

 // exit if help is enabled
 if (vm.count("help"))
 {
 std::cin.get();
 return 0;
 }

 std::string const binary_path(std::move(vm["path"].as<std::string>()));
 std::string renamed_binary_path(binary_path);

 std::cout << "Creating patched binary..." << std::endl;

 Patcher patcher(binary_path);

 // always set wowBuild to current build of the .exe files
 int wowBuild = Helper::GetBuildNumber(patcher.GetBinary());

 // define logical limits in case the exe was tinkered with and the build number was changed
 if (wowBuild == 0 || wowBuild < 10000 || wowBuild > 65535) // Build number has to be exactly 5 characters long
 throw std::runtime_error("Build number was out of range. Build: " + std::to_string(wowBuild));

 std::cout << "Determined build number: " << std::to_string(wowBuild) << std::endl;

 switch (patcher.GetType())
 {
 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, wowBuild);
 WriteCertificateBundle(boost::filesystem::path(binary_path).remove_filename() / "tc_bundle.txt");
 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, wowBuild);
 WriteCertificateBundle(boost::filesystem::path(binary_path).remove_filename() / "tc_bundle.txt");
 break;
 case Constants::BinaryTypes::Mach64:
 std::cout << "Mac client...\n";

 boost::algorithm::replace_all(renamed_binary_path, ".app", " Patched.app");
 Helper::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, wowBuild);

 {
 namespace fs = boost::filesystem;
 fs::permissions(renamed_binary_path, fs::add_perms | fs::others_exe | fs::group_exe | fs::owner_exe);
 }
 WriteCertificateBundle(boost::filesystem::path(binary_path).parent_path()/*MacOS*/.parent_path()/*Contents*/.parent_path()/*World of Warcraft.app*/.parent_path() / "tc_bundle.txt");
 break;
 default:
 throw std::runtime_error("Type: " + std::to_string(static_cast<uint32_t>(patcher.GetType())) + " not supported!");
 }

 std::cout << "Successfully created your patched binaries.\n";

 return 0;
 }
 catch (std::exception const& ex)
 {
 std::cerr << "EX: " << ex.what() << std::endl;
 std::cerr << "An error occurred. Press ENTER to continue...";
 std::cin.get();
 return 1;
 }
}

Here is the call graph for this function: