#!/usr/bin/env python3
|
"""Idempotent patches for 78/esp-wifi-connect captive portal.
|
|
The managed component is gitignored; this script is applied at CMake configure
|
so Android/WeChat locale codes (zh_CN) and socket exhaustion from captive DNS
|
hijack (WeChat /mmtls) do not break the config page.
|
"""
|
|
from __future__ import annotations
|
|
import re
|
import sys
|
from pathlib import Path
|
|
HTML_MARKER = "xiaozhi-portal-lang-patch"
|
HTTPD_MARKER = "xiaozhi-portal-httpd-patch"
|
|
|
def _normalize_ws(text: str) -> str:
|
return text.replace("\r\n", "\n")
|
|
|
def patch_html(path: Path) -> bool:
|
raw = _normalize_ws(path.read_text(encoding="utf-8"))
|
original = raw
|
|
if HTML_MARKER not in raw:
|
raw, n = re.subn(
|
r"function getSupportedLanguage\(lang\) \{\s*"
|
r"// First try exact match",
|
"function getSupportedLanguage(lang) {\n"
|
f" // {HTML_MARKER}\n"
|
" if (!lang) { return 'en-US'; }\n"
|
" lang = String(lang).replace(/_/g, '-');\n"
|
" // First try exact match",
|
raw,
|
count=1,
|
)
|
if n == 0:
|
print(f"warn: could not patch getSupportedLanguage in {path}")
|
|
if "getSupportedLanguage(langParam ||" not in raw:
|
raw, n = re.subn(
|
r"const savedLang = langParam \|\|\s*"
|
r"localStorage\.getItem\('preferred_language'\) \|\|\s*"
|
r"getSupportedLanguage\(browserLang\);",
|
"const savedLang = getSupportedLanguage(langParam ||\n"
|
" localStorage.getItem('preferred_language') ||\n"
|
" browserLang);",
|
raw,
|
count=1,
|
)
|
if n == 0:
|
print(f"warn: could not patch savedLang in {path}")
|
|
if raw == original:
|
print(f"already patched {path}")
|
return False
|
path.write_text(raw, encoding="utf-8", newline="\n")
|
print(f"patched {path}")
|
return True
|
|
|
def patch_httpd(path: Path) -> bool:
|
raw = _normalize_ws(path.read_text(encoding="utf-8"))
|
if HTTPD_MARKER in raw:
|
print(f"already patched {path}")
|
return False
|
raw2, n = re.subn(
|
r"httpd_config_t config = HTTPD_DEFAULT_CONFIG\(\);\s*"
|
r"config\.max_uri_handlers = 24;\s*"
|
r"config\.uri_match_fn = httpd_uri_match_wildcard;",
|
"httpd_config_t config = HTTPD_DEFAULT_CONFIG();\n"
|
" config.max_uri_handlers = 24;\n"
|
f" config.lru_purge_enable = true; // {HTTPD_MARKER}\n"
|
" config.uri_match_fn = httpd_uri_match_wildcard;",
|
raw,
|
count=1,
|
)
|
if n == 0:
|
print(f"warn: could not patch httpd config in {path}")
|
return False
|
path.write_text(raw2, encoding="utf-8", newline="\n")
|
print(f"patched {path}")
|
return True
|
|
|
def main() -> int:
|
if len(sys.argv) != 2:
|
print("usage: patch_wifi_connect.py <esp-wifi-connect component dir>")
|
return 2
|
root = Path(sys.argv[1])
|
if not root.is_dir():
|
print(f"error: not a directory: {root}")
|
return 1
|
html = root / "assets" / "wifi_configuration.html"
|
cc = root / "wifi_configuration_ap.cc"
|
if html.is_file():
|
patch_html(html)
|
else:
|
print(f"warn: missing {html}")
|
if cc.is_file():
|
patch_httpd(cc)
|
else:
|
print(f"warn: missing {cc}")
|
return 0
|
|
|
if __name__ == "__main__":
|
raise SystemExit(main())
|
