SEGAy/beerpsi-x
1
0
Fork 0
mirror of https://gitea.tendokyu.moe/beerpsi/x.git synced 2024-12-18 02:25:57 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

idk

Browse Source
This commit is contained in:
beerpsi 2024-07-24 08:07:18 +07:00
parent d5fd61fd5e
commit 421c2dc09b
2 changed files with 30 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
chuni/calc_title_server_enc_key.py
View File

@ -4,11 +4,8 @@ import struct
import sys
import re
from Crypto.Hash import SHA1
import pefile
from Crypto.Protocol.KDF import PBKDF2
def rva2offset(pe: pefile.PE, rva: int):
for section in pe.sections:
if section.contains_rva(rva):
@ -41,16 +38,10 @@ if (pmatch := KEY_PASSWORD_RE.search(exe)) and (smatch := KEY_SALT_RE.search(exe
soffset = rva2offset(pe, struct.unpack("<I", smatch.group("offset"))[0] - base_address)
poffset_end = poffset + exe[poffset:].index(0)
password = exe[poffset:poffset_end].decode("utf-8")
password = exe[poffset:poffset_end]
salt = exe[soffset:soffset + 16]
key = PBKDF2(
password,
salt,
dkLen=32,
count=smatch.group("iterations")[0],
hmac_hash_module=SHA1,
)
key = pbkdf2_hmac("sha1", password, salt, iterations=smatch.group("iterations")[0], dklen=32)
key = bytes((x % 0x5E) + 0x21 for x in key)
print(f"Key: {key.hex()}")
@ -71,16 +62,10 @@ if (pmatch := SALT_PASSWORD_RE.search(exe)) and (smatch := SALT_SALT_RE.search(e
soffset = rva2offset(pe, struct.unpack("<I", smatch.group("offset"))[0] - base_address)
poffset_end = poffset + exe[poffset:].index(0)
password = exe[poffset:poffset_end].decode("utf-8")
password = exe[poffset:poffset_end]
salt = exe[soffset:soffset + 16]
key = PBKDF2(
password,
salt,
dkLen=8,
count=smatch.group("iterations")[0],
hmac_hash_module=SHA1,
)
key = pbkdf2_hmac("sha1", password, salt, iterations=smatch.group("iterations")[0], dklen=8)
print(f"Endpoint salt: {key.hex()}")
if (match := ITER_COUNT_RE.search(exe)):

52
geki/calc_title_server_enc_key.py
View File

@ -1,6 +1,10 @@
import argparse
from hashlib import blake2b
import os
import sys
from pathlib import Path
from typing import cast
from UnityPy import Environment
from UnityPy.classes import TextAsset
# MU3.Sys.System.keyIvDigestFixed
KEY_IV_DIGEST_FIXED = bytes([
@ -9,34 +13,30 @@ KEY_IV_DIGEST_FIXED = bytes([
57, 72, 120, 103, 17, 124, 18, 64, 15, 225,
169, 39
])
FILE_MAP = {
"noise0": "Key",
"noise1": "IV",
"noise2": "Endpoint salt",
}
if len(sys.argv) < 4:
print(f"Usage: python {os.path.basename(__file__)} <noise0> <noise1> <noise2>")
exit(1)
parser = argparse.ArgumentParser()
_ = parser.add_argument("mu3_data_path", type=Path)
_ = parser.add_argument("key_iv_digest_fixed", type=bytes.fromhex, nargs="?", default=KEY_IV_DIGEST_FIXED)
with open(sys.argv[1], "rb") as f:
key = blake2b(
f.read(),
digest_size=32,
key=KEY_IV_DIGEST_FIXED,
).hexdigest()
args = parser.parse_args()
print(f"Key: {key}")
env = Environment()
with open(sys.argv[2], "rb") as f:
iv = blake2b(
f.read(),
digest_size=32,
key=KEY_IV_DIGEST_FIXED,
).hexdigest()
with (cast(Path, args.mu3_data_path) / "resources.assets").open("rb") as f:
env.load_file(f)
print(f"IV: {iv}")
for object in env.objects:
if object.type.name != "TextAsset":
continue
with open(sys.argv[3], "rb") as f:
salt = blake2b(
f.read(),
digest_size=32,
key=KEY_IV_DIGEST_FIXED,
).hexdigest()
data = cast(TextAsset, object.read())
print(f"Endpoint salt: {salt}")
if data.name in FILE_MAP:
key = blake2b(data.m_Script, digest_size=32, key=args.key_iv_digest_fixed).hexdigest()
print(f"{FILE_MAP[data.name]}: {key}")