mirror of
https://github.com/leminlimez/Nugget.git
synced 2025-04-08 04:23:05 +08:00
113 lines
4.5 KiB
Python
113 lines
4.5 KiB
Python
from . import backup, perform_restore
|
|
from pymobiledevice3.lockdown import LockdownClient
|
|
|
|
class FileToRestore:
|
|
def __init__(self, contents: str, restore_path: str, restore_name: str, owner: int = 501, group: int = 501):
|
|
self.contents = contents
|
|
self.restore_path = restore_path
|
|
self.restore_name = restore_name
|
|
self.owner = owner
|
|
self.group = group
|
|
|
|
# files is a list of FileToRestore objects
|
|
def restore_files(files: list, reboot: bool = False, lockdown_client: LockdownClient = None):
|
|
# create the files to be backed up
|
|
files_list = [
|
|
backup.Directory("", "RootDomain"),
|
|
backup.Directory("Library", "RootDomain"),
|
|
backup.Directory("Library/Preferences", "RootDomain"),
|
|
]
|
|
# create the links
|
|
for file_num in range(len(files)):
|
|
files_list.append(backup.ConcreteFile(
|
|
f"Library/Preferences/temp{file_num}",
|
|
"RootDomain",
|
|
owner=files[file_num].owner,
|
|
group=files[file_num].group,
|
|
contents=files[file_num].contents,
|
|
inode=file_num
|
|
))
|
|
# add the file paths
|
|
for file_num in range(len(files)):
|
|
file = files[file_num]
|
|
base_path = "/var/backup"
|
|
# set it to work in the separate volumes (prevents a bootloop)
|
|
if file.restore_path.startswith("/var/mobile/"):
|
|
# required on iOS 17.0+ since /var/mobile is on a separate partition
|
|
base_path = "/var/mobile/backup"
|
|
elif file.restore_path.startswith("/private/var/mobile/"):
|
|
base_path = "/private/var/mobile/backup"
|
|
elif file.restore_path.startswith("/private/var/"):
|
|
base_path = "/private/var/backup"
|
|
files_list.append(backup.Directory(
|
|
"",
|
|
f"SysContainerDomain-../../../../../../../..{base_path}{file.restore_path}",
|
|
owner=file.owner,
|
|
group=file.group
|
|
))
|
|
files_list.append(backup.ConcreteFile(
|
|
"",
|
|
f"SysContainerDomain-../../../../../../../..{base_path}{file.restore_path}{file.restore_name}",
|
|
owner=file.owner,
|
|
group=file.group,
|
|
contents=b"",
|
|
inode=file_num
|
|
))
|
|
# break the hard links
|
|
for file_num in range(len(files)):
|
|
files_list.append(backup.ConcreteFile(
|
|
"",
|
|
f"SysContainerDomain-../../../../../../../../var/.backup.i/var/root/Library/Preferences/temp{file_num}",
|
|
owner=501,
|
|
group=501,
|
|
contents=b"",
|
|
)) # Break the hard link
|
|
files_list.append(backup.ConcreteFile("", "SysContainerDomain-../../../../../../../.." + "/crash_on_purpose", contents=b""))
|
|
|
|
# create the backup
|
|
back = backup.Backup(files=files_list)
|
|
|
|
perform_restore(backup=back, reboot=reboot, lockdown_client=lockdown_client)
|
|
|
|
|
|
def restore_file(fp: str, restore_path: str, restore_name: str, reboot: bool = False, lockdown_client: LockdownClient = None):
|
|
# open the file and read the contents
|
|
contents = open(fp, "rb").read()
|
|
|
|
base_path = "/var/backup"
|
|
if restore_path.startswith("/var/mobile/"):
|
|
# required on iOS 17.0+ since /var/mobile is on a separate partition
|
|
base_path = "/var/mobile/backup"
|
|
|
|
# create the backup
|
|
back = backup.Backup(files=[
|
|
# backup.Directory("", "HomeDomain"),
|
|
# backup.Directory("Library", "HomeDomain"),
|
|
# backup.Directory("Library/Preferences", "HomeDomain"),
|
|
# backup.ConcreteFile("Library/Preferences/temp", "HomeDomain", owner=501, group=501, contents=contents, inode=0),
|
|
backup.Directory(
|
|
"",
|
|
f"SysContainerDomain-../../../../../../../..{base_path}{restore_path}",
|
|
owner=501,
|
|
group=501
|
|
),
|
|
backup.ConcreteFile(
|
|
"",
|
|
f"SysContainerDomain-../../../../../../../..{base_path}{restore_path}{restore_name}",
|
|
owner=501,
|
|
group=501,
|
|
contents=contents#b"",
|
|
# inode=0
|
|
),
|
|
# backup.ConcreteFile(
|
|
# "",
|
|
# "SysContainerDomain-../../../../../../../../var/.backup.i/var/root/Library/Preferences/temp",
|
|
# owner=501,
|
|
# group=501,
|
|
# contents=b"",
|
|
# ), # Break the hard link
|
|
backup.ConcreteFile("", "SysContainerDomain-../../../../../../../.." + "/crash_on_purpose", contents=b""),
|
|
])
|
|
|
|
|
|
perform_restore(backup=back, reboot=reboot, lockdown_client=lockdown_client) |