restituere.py

#!/usr/bin/env python3

import argparse
import os
import shutil
import sys
import subprocess
import platform
from resources import ipsw, patch, restore
import glob
import time

if platform.system() == 'Darwin':
    pass
else:
    sys.exit('This script can only be ran on macOS. Please run this on a macOS computer.')

parser = argparse.ArgumentParser(description='Restituere - Restore custom IPSWs onto your 64bit iOS device!', usage="./restituere.py -d 'device' -i 'iOS version' -f 'IPSW' [-v]")
parser.add_argument('-d', '--device', help='Your device identifier (e.g. iPhone10,2)', nargs=1)
parser.add_argument('-i', '--version', help='The version of your custom IPSW', nargs=1)
parser.add_argument('-f', '--ipsw', help='Path to custom IPSW', nargs=1)
parser.add_argument('-v', '--verbose', help='Print verbose output for debugging', action='store_true')
parser.add_argument('-u', '--update', help='Keep data while restoring (EXPERIMENTAL)', action='store_true')
args = parser.parse_args()

if args.ipsw:
    if args.device:
        pass
    else:
        sys.exit('Error: You must specify a device identifier!\nExiting...')
    if args.version:
        version_str = args.version[0]
        if version_str.startswith('10.'):
            sys.exit('Error: iOS 10.x IPSWs are not currently supported!\nExiting...')
    else:
        sys.exit('Error: You must specify an iOS version!\nExiting...')
    if args.update:
        keep_data = True
    else:
        keep_data = False
    device_check = input('Is your device is connected in Pwned DFU mode with signature checks removed? [Y/N]: ')
    if 'Y' or 'y' in device_check.lower():
        pass
    else:
        sys.exit('Exiting...')
    lsusb = subprocess.Popen('./resources/bin/lsusb', stdout=subprocess.PIPE, shell=True)
    time.sleep(10)
    lsusb_output = str(lsusb.stdout.read())
    if 'Apple Mobile Device (DFU Mode)' in lsusb_output:
        pass
    else:
        sys.exit('DFU device not found!\nExiting...')
    print('Fetching some required info...')
    if args.verbose:
        print('[VERBOSE] Fetching ECID...')
    fetch_ecid = subprocess.Popen('./resources/bin/igetnonce | grep ECID', stdout=subprocess.PIPE, shell=True)
    time.sleep(3)
    ecid = str(fetch_ecid.stdout.read())
    ecid = ecid[7:-3]
    dec_ecid = str(int(ecid, 16))
    if args.verbose:
        print(f'[VERBOSE] ECID: {ecid}')
    if args.verbose:
        print('[VERBOSE] Fetching apnonce...')
    fetch_apnonce = subprocess.Popen('./resources/bin/igetnonce | grep ApNonce', stdout=subprocess.PIPE, shell=True)
    time.sleep(3)
    apnonce = str(fetch_apnonce.stdout.read())
    apnonce = apnonce[10:-3]
    if args.verbose:
        print(f'[VERBOSE] apnonce: {apnonce}')
    if os.path.exists(f'work'): # In case work directory is still here from a previous run, remove it
        shutil.rmtree(f'work')
    os.makedirs('work/ipsw', exist_ok = True)
    print('Saving SHSH blobs for restore...')
    if len(glob.glob('work/ipsw/*.shsh2')) != 0:
        for shsh in glob.glob('work/ipsw/*.shsh2'):
            os.remove(shsh)
    subprocess.run(f'./resources/bin/tsschecker -d {args.device[0]} -l -e 0x{ecid} -s --apnonce {apnonce} --save-path work/ipsw/', stdout=open('resources/restituere.log','w'), shell=True)
    if len(glob.glob('work/ipsw/*.shsh2')) == 0:
        sys.exit("SHSH Blobs didn't save! Make sure you're connected to the internet, then try again.\nExiting...")
    for shsh in glob.glob('work/ipsw/*.shsh2'):
        os.rename(shsh, 'work/ipsw/blob.shsh2')
    print('SHSH blobs saved!')
    if args.verbose:
        print(f'Finding Firmware bundle for {args.device[0]}, {args.version[0]}...')
    if args.verbose:
        firmware_bundle = ipsw.find_bundle(args.device[0], args.version[0], 'yes')
    else:
        firmware_bundle = ipsw.find_bundle(args.device[0], args.version[0])
    is_a9 = ipsw.a9_check(firmware_bundle)
    if is_a9:
        if args.verbose:
            print('Device is A9, fetching correct board config...')
        board_configs = ipsw.fetch_a9_boardconfigs(firmware_bundle)
        if len(board_configs) != 2:
            sys.exit('Firmware Bundle for A9 is invalid.\nExiting...')
        firm_bundle_number = input(f'A9 device detected, please choose the correct board config for your device:\n[1] {board_configs[0]}\n[2] {board_configs[1]}\nChoice: ')
        try:
            int(firm_bundle_number)
        except ValueError:
            sys.exit('Input not a number!.\nExiting...')
        firm_bundle_number = int(firm_bundle_number)
        if 0 < firm_bundle_number < 3:
            firm_bundle_number = firm_bundle_number - 1
            pass
        else:
            sys.exit('Invalid input given.\nExiting...')
    else:
        firm_bundle_number = 1337
        if args.verbose:
            print('Device is not A9, continuing...')
    if args.verbose:
        print('Extracting iBSS and iBEC from custom IPSW...')
    if args.verbose:
        ibss_path, ibec_path = ipsw.extract_ibss_ibec(args.ipsw[0], firmware_bundle, firm_bundle_number, 'yes')
    else:
        ibss_path, ibec_path = ipsw.extract_ibss_ibec(args.ipsw[0], firmware_bundle, firm_bundle_number)
    print('Signing iBSS and iBEC with SHSH blob...')
    if args.verbose:
        patch.sign_ibss_ibec(ibss_path, ibec_path, 'yes')
    else:
        patch.sign_ibss_ibec(ibss_path, ibec_path)
    processor = ipsw.fetch_processor(firmware_bundle)
    print('Preparations done! Beginning restore...')
    if args.verbose:
        restore.send_ibss_ibec(processor, 'yes')
    else:
        restore.send_ibss_ibec(processor)
    if args.verbose:
        restore.restore(args.ipsw[0], restore.is_cellular(args.device[0]), keep_data, 'yes')
    else:
        restore.restore(args.ipsw[0], restore.is_cellular(args.device[0]), keep_data)
    print('Restore finished! Cleaning up...')
    shutil.rmtree('work')
    print('Done.')
else:
    exit(parser.print_help(sys.stderr))