Fix kill switch not cleared on reinstall; guard midclt delete; tighten sitecustomize imports

install.sh: running install.sh after a recover.sh left /data/truecloud-patch/disabled
in place, so apply.sh silently skipped all patching and middlewared restarted
without the patch. Clear the kill switch file before running apply.sh.

uninstall.sh: midclt initshutdownscript.delete was unguarded under set -euo pipefail,
so a delete failure (already-removed entry, transient API error) aborted the script
before sitecustomize.py was cleaned up or /data/truecloud-patch/ was removed. Now
guarded with an if/else that warns and continues.

sitecustomize.py: move importlib.machinery/.util imports inside the if block in
find_spec so they only execute when intercepting our two target modules, not on
every module import across the whole process. In _install(), import os before
importlib.util so the kill switch check (cheap) runs before the importlib import
(slightly heavier on first use). Remove the unused orig variable in _patch_restic.

patch_ui.py: add explicit encoding="utf-8" to both open() calls. errors="replace"
on read so malformed bytes in a bundle don't silently skip a candidate file.
This commit is contained in:
2026-06-15 02:39:34 +00:00
parent 48cf3a5601
commit 78f943a71b
4 changed files with 20 additions and 10 deletions
+8
View File
@@ -73,6 +73,14 @@ else
fi
echo ""
# ── Clear kill switch if set ──────────────────────────────────────────────────
if [ -f "$PATCH_DIR/disabled" ]; then
rm "$PATCH_DIR/disabled"
echo "Removed kill switch ($PATCH_DIR/disabled) left from a previous recovery."
echo ""
fi
# ── Apply now ─────────────────────────────────────────────────────────────────
echo "Applying patches ..."
+2 -2
View File
@@ -60,7 +60,7 @@ def find_bundle(webui):
continue
path = os.path.join(root, name)
try:
with open(path) as fh:
with open(path, encoding="utf-8", errors="replace") as fh:
content = fh.read()
if FIND.search(content):
matches.append((path, content))
@@ -112,7 +112,7 @@ def main():
patched, count = FIND.subn(REPLACE, content)
try:
with open(path, "w") as fh:
with open(path, "w", encoding="utf-8") as fh:
fh.write(patched)
except OSError as exc:
print(f"[truecloud-patch] ERROR: Could not write {path}: {exc}")
+5 -6
View File
@@ -41,14 +41,14 @@ class _Finder:
self._done = set() # modules already patched
def find_spec(self, fullname, path, target=None): # noqa: ARG002
import importlib.machinery
import importlib.util
if (
fullname in self._targets
and fullname not in self._done
and fullname not in self._loading
):
import importlib.machinery
import importlib.util
# Find the real file spec HERE, before Python adds the module to
# sys.modules. If we deferred this to exec_module, find_spec would
# short-circuit via sys.modules[fullname].__spec__ (our own spec) and
@@ -135,8 +135,7 @@ def _patch_b2(module):
def _patch_restic(module):
orig = module.get_restic_config
if getattr(orig, "_truecloud_patched", False):
if getattr(module.get_restic_config, "_truecloud_patched", False):
return
# ResticConfig is safe to capture now (it's a dataclass defined in the module).
@@ -182,10 +181,10 @@ _PATCHES = {
# ── Entry point ───────────────────────────────────────────────────────────────
def _install():
import importlib.util
import os
if os.path.exists("/data/truecloud-patch/disabled"):
return # kill switch: touch /data/truecloud-patch/disabled to bypass this hook
import importlib.util
if importlib.util.find_spec("middlewared") is None:
return # not a middlewared Python process; nothing to do
sys.meta_path.append(_Finder())
+5 -2
View File
@@ -32,8 +32,11 @@ for s in json.load(sys.stdin):
if [ -n "$IDS" ]; then
for id in $IDS; do
midclt call initshutdownscript.delete "$id" > /dev/null
echo " Removed initshutdownscript id=$id"
if midclt call initshutdownscript.delete "$id" > /dev/null; then
echo " Removed initshutdownscript id=$id"
else
echo " WARNING: could not delete id=$id (already gone?)"
fi
done
else
echo " No entry found (already removed or never installed)."