mirror of
https://github.com/NohamR/pyzule-rw.git
synced 2026-10-11 02:49:40 +00:00
92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
import os
|
|
import sys
|
|
import shutil
|
|
from argparse import ArgumentParser
|
|
from tempfile import TemporaryDirectory
|
|
|
|
from cyan import tbhutils, tbhtypes
|
|
|
|
|
|
def main(parser: ArgumentParser) -> None:
|
|
args = parser.parse_args()
|
|
args.i = os.path.normpath(args.input)
|
|
|
|
if args.output is not None:
|
|
args.o = os.path.normpath(args.output)
|
|
if not (args.o.endswith(".app") or args.o.endswith(".ipa")):
|
|
print("[?] output's file extension not specified; will create ipa")
|
|
args.o += ".ipa"
|
|
else:
|
|
args.o = args.i
|
|
|
|
# this also modifies some args, like -f,
|
|
# to ensure there are no duplicates, etc
|
|
arg_err = tbhutils.validate_inputs(args)
|
|
if arg_err is not None:
|
|
parser.error(arg_err)
|
|
|
|
INPUT_IS_IPA = True if args.i.endswith(".ipa") else False
|
|
OUTPUT_IS_IPA = True if args.o.endswith(".ipa") else False
|
|
|
|
with TemporaryDirectory() as tmpdir, tbhtypes.LeavingCM():
|
|
app_path = tbhutils.get_app(args.i, tmpdir, INPUT_IS_IPA)
|
|
app = tbhtypes.AppBundle(app_path)
|
|
|
|
if app.executable.is_encrypted():
|
|
if args.ignore_encrypted:
|
|
print("[?] main binary is encrypted, ignoring")
|
|
else:
|
|
sys.exit("[!] main binary is encrypted; exiting")
|
|
|
|
if args.cyan is not None:
|
|
changing = vars(args)
|
|
tbhutils.parse_cyan(changing, tmpdir)
|
|
|
|
# this goes before injection,
|
|
# since user might inject their own extensions
|
|
if args.remove_extensions:
|
|
app.remove_all_extensions()
|
|
elif args.remove_encrypted:
|
|
app.remove_encrypted_extensions()
|
|
|
|
if args.f is not None:
|
|
app.executable.inject(args.f, tmpdir)
|
|
if args.n is not None:
|
|
app.plist.change_name(args.n)
|
|
if args.v is not None:
|
|
app.plist.change_version(args.v)
|
|
if args.b is not None:
|
|
app.plist.change_bundle_id(args.b)
|
|
if args.m is not None:
|
|
app.plist.change_minimum_version(args.m)
|
|
if args.k is not None:
|
|
app.change_icon(args.k, tmpdir)
|
|
|
|
if args.remove_supported_devices:
|
|
app.plist.remove_uisd()
|
|
if args.no_watch:
|
|
app.remove_watch_apps()
|
|
if args.enable_documents:
|
|
app.plist.enable_documents()
|
|
if args.fakesign:
|
|
app.fakesign_all()
|
|
if args.thin:
|
|
app.thin_all()
|
|
|
|
# create subdirectories if necessary
|
|
if "/" in args.o:
|
|
os.makedirs(os.path.dirname(args.o), exist_ok=True)
|
|
|
|
# done !
|
|
if OUTPUT_IS_IPA:
|
|
print(f"[*] generating ipa with compression level {args.compress}..")
|
|
tbhutils.make_ipa(tmpdir, os.path.realpath(args.o), args.compress)
|
|
print(f"[*] generated ipa at {args.o}")
|
|
else:
|
|
if os.path.isdir(args.o):
|
|
shutil.rmtree(args.o)
|
|
|
|
shutil.move(app_path, args.o)
|
|
print(f"[*] generated app at {args.o}")
|
|
|