mirror of
https://github.com/dortania/OpenCore-Legacy-Patcher.git
synced 2026-04-14 12:48:18 +10:00
61 lines
2.9 KiB
Python
61 lines
2.9 KiB
Python
# Reoute binaries to tmp directory, and mount a disk image of the payloads
|
|
# Implements a shadowfile to avoid direct writes to the dmg
|
|
# Copyright (C) 2022, Mykola Grymalyuk
|
|
|
|
import sys
|
|
import plistlib
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tempfile
|
|
import atexit
|
|
|
|
class reroute_payloads:
|
|
def __init__(self, constants):
|
|
self.constants = constants
|
|
|
|
def setup_tmp_disk_image(self):
|
|
# Create a temp directory to mount the payloads.dmg
|
|
# Then reroute r/w to this new temp directory
|
|
# Currently only applicable for GUI variant
|
|
if self.constants.launcher_binary and self.constants.wxpython_variant is True and not self.constants.launcher_script:
|
|
print("- Running in Binary GUI mode, switching to tmp directory")
|
|
self.temp_dir = tempfile.TemporaryDirectory()
|
|
print(f"- New payloads location: {self.temp_dir.name}")
|
|
print("- Creating payloads directory")
|
|
Path(self.temp_dir.name / Path("payloads")).mkdir(parents=True, exist_ok=True)
|
|
self.unmount_active_dmgs()
|
|
output = subprocess.run(
|
|
[
|
|
"hdiutil", "attach", "-noverify", f"{self.constants.payload_path}.dmg",
|
|
"-mountpoint", Path(self.temp_dir.name / Path("payloads")),
|
|
"-nobrowse", "-shadow", Path(self.temp_dir.name / Path("payloads_overlay")),
|
|
"-passphrase", "password"
|
|
],
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT
|
|
)
|
|
if output.returncode == 0:
|
|
print("- Mounted payloads.dmg")
|
|
self.constants.current_path = Path(self.temp_dir.name)
|
|
self.constants.payload_path = Path(self.temp_dir.name) / Path("payloads")
|
|
atexit.register(self.unmount_active_dmgs)
|
|
else:
|
|
print("- Failed to mount payloads.dmg")
|
|
print(f"Output: {output.stdout.decode()}")
|
|
print(f"Return Code: {output.returncode}")
|
|
print("- Exiting...")
|
|
sys.exit(1)
|
|
|
|
def unmount_active_dmgs(self):
|
|
# Find all DMGs that are mounted, and forcefully unmount them
|
|
# If our disk image was previously mounted, we need to unmount it to use again
|
|
# This can happen if we crash during a previous scession, however 'atexit' class should hopefully avoid this
|
|
dmg_info = subprocess.run(["hdiutil", "info", "-plist"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
dmg_info = plistlib.loads(dmg_info.stdout)
|
|
|
|
for image in dmg_info["images"]:
|
|
if image["image-path"].endswith("payloads.dmg"):
|
|
print(f"- Unmounting payloads.dmg")
|
|
subprocess.run(
|
|
["hdiutil", "detach", image["system-entities"][0]["dev-entry"], "-force"],
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT
|
|
) |