aben126/kgg-dec
1
0
Fork 0
forked from um/kgg-dec
Code Pull Requests Activity

chore: reformat files

Browse Source
This commit is contained in:
鲁树人 2024-11-19 16:59:28 +09:00
parent 84585531d4
commit edbe9ae708
5 changed files with 108 additions and 111 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/base64.h
View File

@ -4,4 +4,4 @@
#include <cstdint>
#include <vector>
std::vector<uint8_t> b64_decode(const uint8_t *input, size_t len);
std::vector<uint8_t> b64_decode(const uint8_t* input, size_t len);

28
src/qmc2/ekey.cpp
View File

@ -3,38 +3,36 @@
#include "tc_tea.h"
#include <algorithm>
#include <string>
#include <array>
#include <string>
#include <vector>
const static std::string kEKeyV2Prefix = "UVFNdXNpYyBFbmNWMixLZXk6";
const static std::array<uint8_t, 16> kEKeyV2Key1{
0x33, 0x38, 0x36, 0x5A, 0x4A, 0x59, 0x21, 0x40,
0x23, 0x2A, 0x24, 0x25, 0x5E, 0x26, 0x29, 0x28,
0x33, 0x38, 0x36, 0x5A, 0x4A, 0x59, 0x21, 0x40, 0x23, 0x2A, 0x24, 0x25, 0x5E, 0x26, 0x29, 0x28,
};
const static std::array<uint8_t, 16> kEKeyV2Key2{
0x2A, 0x2A, 0x23, 0x21, 0x28, 0x23, 0x24, 0x25,
0x26, 0x5E, 0x61, 0x31, 0x63, 0x5A, 0x2C, 0x54,
0x2A, 0x2A, 0x23, 0x21, 0x28, 0x23, 0x24, 0x25, 0x26, 0x5E, 0x61, 0x31, 0x63, 0x5A, 0x2C, 0x54,
};
template <typename T> std::span<T> ss2span(std::string_view sv) {
auto *data = reinterpret_cast<const T *>(sv.data());
return std::span<T>(const_cast<T *>(data), sv.size());
template <typename T>
std::span<T> ss2span(std::string_view sv) {
auto* data = reinterpret_cast<const T*>(sv.data());
return std::span<T>(const_cast<T*>(data), sv.size());
}
template <typename T> std::string_view span2ss(std::span<T> span) {
return std::string_view(reinterpret_cast<char *>(span.data()), span.size());
template <typename T>
std::string_view span2ss(std::span<T> span) {
return std::string_view(reinterpret_cast<char*>(span.data()), span.size());
}
void remove_trailing_zeros(std::vector<uint8_t> &vec) {
auto it = std::find_if(vec.rbegin(), vec.rend(),
[](uint8_t value) { return value != 0; });
void remove_trailing_zeros(std::vector<uint8_t>& vec) {
auto it = std::find_if(vec.rbegin(), vec.rend(), [](uint8_t value) { return value != 0; });
vec.erase(it.base(), vec.end());
}
std::vector<uint8_t> decrypt_ekey_v1(std::string_view ekey) {
std::vector<uint8_t> result =
b64_decode(reinterpret_cast<const uint8_t *>(ekey.data()), ekey.size());
std::vector<uint8_t> result = b64_decode(reinterpret_cast<const uint8_t*>(ekey.data()), ekey.size());
remove_trailing_zeros(result);
uint32_t tea_key[4] = {

15
src/tc_tea/tc_tea.cpp
View File

@ -10,8 +10,11 @@ constexpr size_t kTeaBlockSize = 8;
constexpr size_t kFixedSaltLen = 2;
constexpr size_t kZeroPadLen = 7;
inline void decrypt_round(uint8_t *p_plain, const uint8_t *p_cipher,
uint64_t *iv1, uint64_t *iv2, const uint32_t *key) {
inline void decrypt_round(uint8_t* p_plain,
const uint8_t* p_cipher,
uint64_t* iv1,
uint64_t* iv2,
const uint32_t* key) {
uint64_t iv1_next = Endian::be_u64_read(p_cipher);
uint64_t iv2_next = tc_tea_ecb_decrypt(iv1_next ^ *iv2, key);
uint64_t plain = iv2_next ^ *iv1;
@ -20,8 +23,7 @@ inline void decrypt_round(uint8_t *p_plain, const uint8_t *p_cipher,
Endian::be_u64_write(p_plain, plain);
}
std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher,
const uint32_t *key) {
std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher, const uint32_t* key) {
// It needs to have at least 2 blocks long, due to the nature of the padding
// scheme used.
if (cipher.size() % kTeaBlockSize != 0 || cipher.size() < kTeaBlockSize * 2) {
@ -31,7 +33,7 @@ std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher,
uint64_t iv1 = 0;
uint64_t iv2 = 0;
uint8_t header[kTeaBlockSize * 2];
const uint8_t *in_cipher = cipher.data();
const uint8_t* in_cipher = cipher.data();
decrypt_round(header, in_cipher, &iv1, &iv2, key);
in_cipher += kTeaBlockSize;
decrypt_round(header + kTeaBlockSize, in_cipher, &iv1, &iv2, key);
@ -50,8 +52,7 @@ std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher,
if (real_plain_len != copy_len) {
// Decrypt the rest of the blocks
for (size_t i = cipher.size() - kTeaBlockSize * 3; i != 0;
i -= kTeaBlockSize) {
for (size_t i = cipher.size() - kTeaBlockSize * 3; i != 0; i -= kTeaBlockSize) {
decrypt_round(p_output, in_cipher, &iv1, &iv2, key);
in_cipher += kTeaBlockSize;
p_output += kTeaBlockSize;

6
src/tc_tea/tc_tea.h
View File

@ -6,11 +6,9 @@
#include <span>
#include <vector>
std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher,
const uint32_t *key);
std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher, const uint32_t* key);
inline std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher,
const uint8_t *key) {
inline std::vector<uint8_t> tc_tea_cbc_decrypt(std::span<uint8_t> cipher, const uint8_t* key) {
uint32_t key_u32[4];
for (int i = 0; i < 4; i++) {
key_u32[i] = Endian::be_u32_read(key + i * 4);