mirror of
https://github.com/dortania/OpenCore-Legacy-Patcher.git
synced 2026-04-14 04:38:20 +10:00
382 lines
22 KiB
Python
382 lines
22 KiB
Python
# Framework for mounting and patching macOS root volume
|
|
# Copyright (C) 2020-2022, Dhinak G, Mykola Grymalyuk
|
|
|
|
# System based off of Apple's Kernel Development Kit (KDK)
|
|
# - https://developer.apple.com/download/all/
|
|
|
|
# The system relies on mounting the APFS volume as a live read/write volume
|
|
# We perform our required edits, then create a new snapshot for the system boot
|
|
|
|
# The manual process is as follows:
|
|
# 1. Mount the APFS volume as a read/write volume
|
|
# 'sudo mount -o nobrowse -t apfs /dev/disk5s5 /System/Volumes/Update/mnt1'
|
|
# 2. Perform edits to the system (ie. create new KernelCollection)
|
|
# 'sudo kmutil install --volume-root /System/Volumes/Update/mnt1/ --update-all'
|
|
# 3. Create a new snapshot for the system boot
|
|
# 'sudo bless --folder /System/Volumes/Update/mnt1/System/Library/CoreServices --bootefi --create-snapshot'
|
|
|
|
# Additionally Apple's APFS snapshot system supports system rollbacks:
|
|
# 'sudo bless --mount /System/Volumes/Update/mnt1 --bootefi --last-sealed-snapshot'
|
|
# Note: root volume rollbacks are unstable in Big Sur due to quickly discarding the original snapshot
|
|
# - Generally within 2~ boots, the original snapshot is discarded
|
|
# - Monterey always preserves the original snapshot allowing for reliable rollbacks
|
|
|
|
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from resources import constants, generate_smbios, utilities, sys_patch_download, sys_patch_detect, sys_patch_auto
|
|
from data import os_data
|
|
|
|
|
|
class PatchSysVolume:
|
|
def __init__(self, model, versions, hardware_details=None):
|
|
self.model = model
|
|
self.constants: constants.Constants() = versions
|
|
self.computer = self.constants.computer
|
|
self.root_mount_path = None
|
|
self.root_supports_snapshot = utilities.check_if_root_is_apfs_snapshot()
|
|
self.constants.root_patcher_succeded = False # Reset Variable each time we start
|
|
self.patch_set_dictionary = {}
|
|
|
|
# GUI will detect hardware patches before starting PatchSysVolume()
|
|
# However the TUI will not, so allow for data to be passed in manually avoiding multiple calls
|
|
if hardware_details is None:
|
|
hardware_details = sys_patch_detect.detect_root_patch(self.computer.real_model, self.constants).detect_patch_set()
|
|
self.hardware_details = hardware_details
|
|
self.init_pathing(custom_root_mount_path=None, custom_data_mount_path=None)
|
|
|
|
def init_pathing(self, custom_root_mount_path=None, custom_data_mount_path=None):
|
|
if custom_root_mount_path and custom_data_mount_path:
|
|
self.mount_location = custom_root_mount_path
|
|
self.data_mount_location = custom_data_mount_path
|
|
elif self.root_supports_snapshot is True:
|
|
# Big Sur and newer use APFS snapshots
|
|
self.mount_location = "/System/Volumes/Update/mnt1"
|
|
self.mount_location_data = ""
|
|
else:
|
|
self.mount_location = ""
|
|
self.mount_location_data = ""
|
|
self.mount_extensions = f"{self.mount_location}/System/Library/Extensions"
|
|
self.mount_application_support = f"{self.mount_location_data}/Library/Application Support"
|
|
|
|
def find_mount_root_vol(self, patch):
|
|
self.root_mount_path = utilities.get_disk_path()
|
|
if self.root_mount_path.startswith("disk"):
|
|
print(f"- Found Root Volume at: {self.root_mount_path}")
|
|
if Path(self.mount_extensions).exists():
|
|
print("- Root Volume is already mounted")
|
|
if patch is True:
|
|
self.patch_root_vol()
|
|
return True
|
|
else:
|
|
self.unpatch_root_vol()
|
|
return True
|
|
else:
|
|
if self.constants.detected_os > os_data.os_data.catalina and self.root_supports_snapshot is True:
|
|
print("- Mounting APFS Snapshot as writable")
|
|
result = utilities.elevated(["mount", "-o", "nobrowse", "-t", "apfs", f"/dev/{self.root_mount_path}", self.mount_location], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
if result.returncode == 0:
|
|
print(f"- Mounted APFS Snapshot as writable at: {self.mount_location}")
|
|
if Path(self.mount_extensions).exists():
|
|
print("- Successfully mounted the Root Volume")
|
|
if patch is True:
|
|
self.patch_root_vol()
|
|
return True
|
|
else:
|
|
self.unpatch_root_vol()
|
|
return True
|
|
else:
|
|
print("- Failed to mount the Root Volume")
|
|
print("- Recommend rebooting the machine and trying to patch again")
|
|
if self.constants.gui_mode is False:
|
|
input("- Press [ENTER] to exit: ")
|
|
else:
|
|
print("- Could not find root volume")
|
|
if self.constants.gui_mode is False:
|
|
input("- Press [ENTER] to exit: ")
|
|
|
|
def unpatch_root_vol(self):
|
|
if self.constants.detected_os > os_data.os_data.catalina and self.root_supports_snapshot is True:
|
|
print("- Reverting to last signed APFS snapshot")
|
|
result = utilities.elevated(["bless", "--mount", self.mount_location, "--bootefi", "--last-sealed-snapshot"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
if result.returncode != 0:
|
|
print("- Unable to revert root volume patches")
|
|
print("Reason for unpatch Failure:")
|
|
print(result.stdout.decode())
|
|
print("- Failed to revert snapshot via Apple's 'bless' command")
|
|
else:
|
|
self.clean_skylight_plugins()
|
|
self.constants.root_patcher_succeded = True
|
|
print("- Unpatching complete")
|
|
print("\nPlease reboot the machine for patches to take effect")
|
|
|
|
def rebuild_snapshot(self):
|
|
print("- Rebuilding Kernel Cache (This may take some time)")
|
|
result = utilities.elevated(["kmutil", "install", "--volume-root", self.mount_location, "--update-all"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|
|
|
# kextcache notes:
|
|
# - kextcache always returns 0, even if it fails
|
|
# - Check the output for 'KernelCache ID' to see if the cache was successfully rebuilt
|
|
# kmutil notes:
|
|
# - will return 71 on failure to build KCs
|
|
# - will return 31 on 'No binaries or codeless kexts were provided'
|
|
# - will return -10 if the volume is missing (ie. unmounted by another process)
|
|
if result.returncode != 0:
|
|
print("- Unable to build new kernel cache")
|
|
print(f"\nReason for Patch Failure ({result.returncode}):")
|
|
print(result.stdout.decode())
|
|
print("")
|
|
print("\nPlease reboot the machine to avoid potential issues rerunning the patcher")
|
|
if self.constants.gui_mode is False:
|
|
input("Press [ENTER] to continue")
|
|
else:
|
|
print("- Successfully built new kernel cache")
|
|
if self.root_supports_snapshot is True:
|
|
print("- Creating new APFS snapshot")
|
|
bless = utilities.elevated(
|
|
["bless", "--folder", f"{self.mount_location}/System/Library/CoreServices", "--bootefi", "--create-snapshot"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT
|
|
)
|
|
if bless.returncode != 0:
|
|
print("- Unable to create new snapshot")
|
|
print("Reason for snapshot failure:")
|
|
print(bless.stdout.decode())
|
|
if "Can't use last-sealed-snapshot or create-snapshot on non system volume" in bless.stdout.decode():
|
|
print("- This is an APFS bug with Monterey! Perform a clean installation to ensure your APFS volume is built correctly")
|
|
return
|
|
else:
|
|
self.unmount_drive()
|
|
print("- Patching complete")
|
|
print("\nPlease reboot the machine for patches to take effect")
|
|
self.constants.root_patcher_succeded = True
|
|
if self.constants.gui_mode is False:
|
|
input("\nPress [ENTER] to continue")
|
|
|
|
def unmount_drive(self):
|
|
print("- Unmounting Root Volume (Don't worry if this fails)")
|
|
utilities.elevated(["diskutil", "unmount", self.root_mount_path], stdout=subprocess.PIPE).stdout.decode().strip().encode()
|
|
|
|
|
|
def clean_skylight_plugins(self):
|
|
if (Path(self.mount_application_support) / Path("SkyLightPlugins/")).exists():
|
|
print("- Found SkylightPlugins folder, removing old plugins")
|
|
utilities.process_status(utilities.elevated(["rm", "-Rf", f"{self.mount_application_support}/SkyLightPlugins"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
utilities.process_status(utilities.elevated(["mkdir", f"{self.mount_application_support}/SkyLightPlugins"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
else:
|
|
print("- Creating SkylightPlugins folder")
|
|
utilities.process_status(utilities.elevated(["mkdir", "-p", f"{self.mount_application_support}/SkyLightPlugins/"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
|
|
|
|
def patch_root_vol(self):
|
|
print(f"- Running patches for {self.model}")
|
|
if self.patch_set_dictionary != {}:
|
|
self.execute_patchset(self.patch_set_dictionary)
|
|
else:
|
|
self.execute_patchset(sys_patch_detect.detect_root_patch(self.computer.real_model, self.constants).generate_patchset(self.hardware_details))
|
|
|
|
if self.constants.wxpython_variant is True and self.constants.detected_os >= os_data.os_data.big_sur:
|
|
sys_patch_auto.AutomaticSysPatch.install_auto_patcher_launch_agent(self.constants)
|
|
|
|
self.rebuild_snapshot()
|
|
|
|
def execute_patchset(self, required_patches):
|
|
source_files_path = str(self.constants.payload_local_binaries_root_path)
|
|
self.preflight_checks(required_patches, source_files_path)
|
|
for patch in required_patches:
|
|
print("- Installing Patchset: " + patch)
|
|
if "Remove" in required_patches[patch]:
|
|
for remove_patch_directory in required_patches[patch]["Remove"]:
|
|
print("- Remove Files at: " + remove_patch_directory)
|
|
for remove_patch_file in required_patches[patch]["Remove"][remove_patch_directory]:
|
|
destination_folder_path = str(self.mount_location) + remove_patch_directory
|
|
self.remove_file(destination_folder_path, remove_patch_file)
|
|
|
|
|
|
for method_install in ["Install", "Install Non-Root"]:
|
|
if method_install in required_patches[patch]:
|
|
for install_patch_directory in required_patches[patch][method_install]:
|
|
print(f"- Handling Installs in: {install_patch_directory}")
|
|
for install_file in required_patches[patch][method_install][install_patch_directory]:
|
|
source_folder_path = source_files_path + "/" + required_patches[patch][method_install][install_patch_directory][install_file] + install_patch_directory
|
|
if method_install == "Install":
|
|
destination_folder_path = str(self.mount_location) + install_patch_directory
|
|
else:
|
|
destination_folder_path = str(self.mount_location_data) + install_patch_directory
|
|
self.install_new_file(source_folder_path, destination_folder_path, install_file)
|
|
|
|
if "Processes" in required_patches[patch]:
|
|
for process in required_patches[patch]["Processes"]:
|
|
# Some processes need sudo, however we cannot directly call sudo in some scenarios
|
|
# Instead, call elevated funtion and strip sudo from argument
|
|
process_array = process.split(" ")
|
|
if required_patches[patch]["Processes"][process] is True:
|
|
utilities.process_status(utilities.elevated(process_array, stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
else:
|
|
utilities.process_status(subprocess.run(process_array, stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
|
|
def preflight_checks(self, required_patches, source_files_path):
|
|
print("- Running Preflight Checks before patching")
|
|
|
|
# Make sure old SkyLight plugins aren't being used
|
|
self.clean_skylight_plugins()
|
|
|
|
# Make sure SNB kexts are compatible with the host
|
|
if "Intel Sandy Bridge" in required_patches:
|
|
if self.computer.reported_board_id not in self.constants.sandy_board_id_stock:
|
|
print(f"- Found unspported Board ID {self.computer.reported_board_id}, performing AppleIntelSNBGraphicsFB bin patching")
|
|
board_to_patch = generate_smbios.determine_best_board_id_for_sandy(self.computer.reported_board_id, self.computer.gpus)
|
|
print(f"- Replacing {board_to_patch} with {self.computer.reported_board_id}")
|
|
|
|
board_to_patch_hex = bytes.fromhex(board_to_patch.encode('utf-8').hex())
|
|
reported_board_hex = bytes.fromhex(self.computer.reported_board_id.encode('utf-8').hex())
|
|
|
|
if len(board_to_patch_hex) != len(reported_board_hex):
|
|
print(f"- Error: Board ID {self.computer.reported_board_id} is not the same length as {board_to_patch}")
|
|
raise Exception("Host's Board ID is not the same length as the kext's Board ID, cannot patch!!!")
|
|
else:
|
|
path = source_files_path + "10.13.6/System/Library/Extensions/AppleIntelSNBGraphicsFB.kext/Contents/MacOS/AppleIntelSNBGraphicsFB"
|
|
if Path(path).exists:
|
|
with open(path, 'rb') as f:
|
|
data = f.read()
|
|
data = data.replace(board_to_patch_hex, reported_board_hex)
|
|
with open(path, 'wb') as f:
|
|
f.write(data)
|
|
else:
|
|
raise Exception("Failed to find AppleIntelSNBGraphicsFB.kext, cannot patch!!!")
|
|
|
|
# Check all the files are present
|
|
for patch in required_patches:
|
|
for method_type in ["Install", "Install Non-Root"]:
|
|
if method_type in required_patches[patch]:
|
|
for install_patch_directory in required_patches[patch][method_type]:
|
|
for install_file in required_patches[patch][method_type][install_patch_directory]:
|
|
source_file = source_files_path + "/" + required_patches[patch][method_type][install_patch_directory][install_file] + install_patch_directory + "/" + install_file
|
|
if not Path(source_file).exists:
|
|
raise Exception(f"Failed to find {source_file}")
|
|
|
|
print("- Finished Preflight, starting patching")
|
|
|
|
def install_new_file(self, source_folder, destination_folder, file_name):
|
|
# .frameworks are merged
|
|
# .kexts and .apps are deleted and replaced
|
|
file_name_str = str(file_name)
|
|
|
|
if file_name_str.endswith(".framework"):
|
|
# merge with rsync
|
|
print(f" - Installing: {file_name}")
|
|
utilities.elevated(["rsync", "-r", "-i", "-a", f"{source_folder}/{file_name}", f"{destination_folder}/"], stdout=subprocess.PIPE)
|
|
self.fix_permissions(destination_folder + "/" + file_name)
|
|
elif Path(source_folder + "/" + file_name_str).is_dir():
|
|
# Applicable for .kext, .app, .plugin, .bundle, all of which are directories
|
|
if Path(destination_folder + "/" + file_name).exists():
|
|
print(f" - Found existing {file_name}, overwritting...")
|
|
utilities.process_status(utilities.elevated(["rm", "-R", f"{destination_folder}/{file_name}"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
else:
|
|
print(f" - Installing: {file_name}")
|
|
utilities.process_status(utilities.elevated(["cp", "-R", f"{source_folder}/{file_name}", destination_folder], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
self.fix_permissions(destination_folder + "/" + file_name)
|
|
else:
|
|
# Assume it's an individual file, replace as normal
|
|
if Path(destination_folder + "/" + file_name).exists():
|
|
print(f" - Found existing {file_name}, overwritting...")
|
|
utilities.process_status(utilities.elevated(["rm", f"{destination_folder}/{file_name}"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
else:
|
|
print(f" - Installing: {file_name}")
|
|
utilities.process_status(utilities.elevated(["cp", f"{source_folder}/{file_name}", destination_folder], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
self.fix_permissions(destination_folder + "/" + file_name)
|
|
|
|
def remove_file(self, destination_folder, file_name):
|
|
if Path(destination_folder + "/" + file_name).exists():
|
|
print(f" - Removing: {file_name}")
|
|
if Path(destination_folder + "/" + file_name).is_dir():
|
|
utilities.process_status(utilities.elevated(["rm", "-R", f"{destination_folder}/{file_name}"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
else:
|
|
utilities.process_status(utilities.elevated(["rm", f"{destination_folder}/{file_name}"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
|
|
|
|
def fix_permissions(self, destination_file):
|
|
chmod_args = ["chmod", "-Rf", "755", destination_file]
|
|
chown_args = ["chown", "-Rf", "root:wheel", destination_file]
|
|
if not Path(destination_file).is_dir():
|
|
# Strip recursive arguments
|
|
chmod_args.pop(1)
|
|
chown_args.pop(1)
|
|
utilities.process_status(utilities.elevated(chmod_args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
utilities.process_status(utilities.elevated(chown_args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT))
|
|
|
|
|
|
def check_files(self):
|
|
if Path(self.constants.payload_local_binaries_root_path).exists():
|
|
print("- Found local Apple Binaries")
|
|
if self.constants.gui_mode is False:
|
|
patch_input = input("Would you like to redownload?(y/n): ")
|
|
if patch_input in {"y", "Y", "yes", "Yes"}:
|
|
shutil.rmtree(Path(self.constants.payload_local_binaries_root_path))
|
|
output = self.download_files()
|
|
else:
|
|
output = self.download_files()
|
|
else:
|
|
output = self.download_files()
|
|
return output
|
|
|
|
def download_files(self):
|
|
if self.constants.gui_mode is False or "Library/InstallerSandboxes/" in str(self.constants.payload_path):
|
|
download_result, link = sys_patch_download.grab_patcher_support_pkg(self.constants).download_files()
|
|
else:
|
|
download_result = True
|
|
link = sys_patch_download.grab_patcher_support_pkg(self.constants).generate_pkg_link()
|
|
|
|
if download_result and self.constants.payload_local_binaries_root_path_zip.exists():
|
|
print("- Unzipping binaries...")
|
|
utilities.process_status(subprocess.run(["ditto", "-V", "-x", "-k", "--sequesterRsrc", "--rsrc", self.constants.payload_local_binaries_root_path_zip, self.constants.payload_path]))
|
|
print("- Binaries downloaded to:")
|
|
print(self.constants.payload_path)
|
|
return self.constants.payload_local_binaries_root_path
|
|
else:
|
|
if self.constants.gui_mode is True:
|
|
print("- Download failed, please verify the below link work:")
|
|
print(link)
|
|
print("\nIf you continue to have issues, try using the Offline builds")
|
|
print("located on Github next to the other builds")
|
|
else:
|
|
input("\nPress enter to continue")
|
|
return None
|
|
|
|
# Entry Function
|
|
def start_patch(self):
|
|
print("- Starting Patch Process")
|
|
print(f"- Determining Required Patch set for Darwin {self.constants.detected_os}")
|
|
self.patch_set_dictionary = sys_patch_detect.detect_root_patch(self.computer.real_model, self.constants).generate_patchset(self.hardware_details)
|
|
|
|
if self.patch_set_dictionary == {}:
|
|
change_menu = None
|
|
print("- No Root Patches required for your machine!")
|
|
if self.constants.gui_mode is False:
|
|
input("\nPress [ENTER] to return to the main menu: ")
|
|
elif self.constants.gui_mode is False:
|
|
change_menu = input("Would you like to continue with Root Volume Patching?(y/n): ")
|
|
else:
|
|
change_menu = "y"
|
|
print("- Continuing root patching")
|
|
if change_menu in ["y", "Y"]:
|
|
print("- Verifying whether Root Patching possible")
|
|
if sys_patch_detect.detect_root_patch(self.computer.real_model, self.constants).verify_patch_allowed(print_errors=not self.constants.wxpython_variant) is True:
|
|
print("- Patcher is capable of patching")
|
|
if self.check_files():
|
|
self.find_mount_root_vol(True)
|
|
elif self.constants.gui_mode is False:
|
|
input("\nPress [ENTER] to return to the main menu: ")
|
|
|
|
else:
|
|
print("- Returning to main menu")
|
|
|
|
def start_unpatch(self):
|
|
print("- Starting Unpatch Process")
|
|
if sys_patch_detect.detect_root_patch(self.computer.real_model, self.constants).verify_patch_allowed(print_errors=True) is True:
|
|
self.find_mount_root_vol(False)
|
|
if self.constants.gui_mode is False:
|
|
input("\nPress [ENTER] to return to the main menu")
|
|
elif self.constants.gui_mode is False:
|
|
input("\nPress [ENTER] to return to the main menu")
|