diff options
author | bloerwald <bloerwald@googlemail.com> | 2014-10-28 18:33:19 +0000 |
---|---|---|
committer | DDuarte <dnpd.dd@gmail.com> | 2014-10-28 18:33:19 +0000 |
commit | ff0f84e3aa23a32106e88e84c7abe10229200e9e (patch) | |
tree | 7aafb946356dc4f82031badbcbbfbe53eac3e5d0 /src/tools/connection_patcher/Helper.cpp | |
parent | 6b6b6e8f09732ef0306b2cf3c6d5f8953a4fddca (diff) |
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
Diffstat (limited to 'src/tools/connection_patcher/Helper.cpp')
-rw-r--r-- | src/tools/connection_patcher/Helper.cpp | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/src/tools/connection_patcher/Helper.cpp b/src/tools/connection_patcher/Helper.cpp new file mode 100644 index 00000000000..75de2452d26 --- /dev/null +++ b/src/tools/connection_patcher/Helper.cpp @@ -0,0 +1,58 @@ +/* + * 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 <SHA256.h> +#include <Util.h> + +#include <stdexcept> + +namespace Connection_Patcher +{ + namespace Helper + { + Constants::BinaryTypes GetBinaryType(std::vector<unsigned char> const& data) + { + // Check MS-DOS magic + if (*reinterpret_cast<uint16_t const*>(data.data()) == 0x5A4D) + { + uint32_t const peOffset(*reinterpret_cast<uint32_t const*>(data.data() + 0x3C)); + + // Check PE magic + if (*reinterpret_cast<uint32_t const*>(data.data() + peOffset) != 0x4550) + throw std::invalid_argument("Not a PE file!"); + + return Constants::BinaryTypes(*reinterpret_cast<uint16_t const*>(data.data() + peOffset + 4)); + } + else + { + return Constants::BinaryTypes(*reinterpret_cast<uint32_t const*>(data.data())); + } + } + + std::string GetFileChecksum(std::vector<unsigned char> const& data) + { + SHA256Hash h; + h.UpdateData(data.data(), data.size()); + h.Finalize(); + + return ByteArrayToHexStr(h.GetDigest(), h.GetLength()); + } + } +} |