From 044fd47e3d574589d42e86c57a1762b2fe3153a8 Mon Sep 17 00:00:00 2001
From: opa334 <opa334@users.noreply.github.com>
Date: Sat, 3 Sep 2022 12:18:57 +0200
Subject: [PATCH] Add exploit value picker

---
 .../TrollInstaller/Base.lproj/Main.storyboard | 33 +++++++++++--
 .../TrollInstaller/ViewController.m           | 46 +++++++++++++++++++
 .../TrollInstaller/exploit/IOGPU.c            |  4 +-
 3 files changed, 79 insertions(+), 4 deletions(-)

diff --git a/Installer/TrollInstaller/TrollInstaller/Base.lproj/Main.storyboard b/Installer/TrollInstaller/TrollInstaller/Base.lproj/Main.storyboard
index 207e557..7445d4c 100644
--- a/Installer/TrollInstaller/TrollInstaller/Base.lproj/Main.storyboard
+++ b/Installer/TrollInstaller/TrollInstaller/Base.lproj/Main.storyboard
@@ -53,7 +53,25 @@ Installer</string>
                                 <nil key="highlightedColor"/>
                             </label>
                             <label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Supports iOS 15.0 - 15.1.1" textAlignment="natural" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="s80-eR-fY3">
-                                <rect key="frame" x="110.5" y="249" width="193" height="21"/>
+                                <rect key="frame" x="110.5" y="238" width="193" height="21"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="17"/>
+                                <nil key="textColor"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Exploit value: -1" textAlignment="natural" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="iBq-Z7-lRu">
+                                <rect key="frame" x="87" y="267" width="120" height="21"/>
+                                <fontDescription key="fontDescription" type="system" pointSize="17"/>
+                                <nil key="textColor"/>
+                                <nil key="highlightedColor"/>
+                            </label>
+                            <stepper opaque="NO" contentMode="scaleToFill" horizontalHuggingPriority="750" verticalHuggingPriority="750" contentHorizontalAlignment="center" contentVerticalAlignment="center" maximumValue="100" translatesAutoresizingMaskIntoConstraints="NO" id="DFM-s0-Qyv">
+                                <rect key="frame" x="215" y="261" width="94" height="32"/>
+                                <connections>
+                                    <action selector="stepperValueChanged:" destination="BYZ-38-t0r" eventType="valueChanged" id="R0o-Gd-byG"/>
+                                </connections>
+                            </stepper>
+                            <label opaque="NO" userInteractionEnabled="NO" contentMode="left" horizontalHuggingPriority="251" verticalHuggingPriority="251" text="Model: " textAlignment="natural" lineBreakMode="tailTruncation" baselineAdjustment="alignBaselines" adjustsFontSizeToFit="NO" translatesAutoresizingMaskIntoConstraints="NO" id="2NU-06-Bab">
+                                <rect key="frame" x="178.5" y="771" width="57" height="21"/>
                                 <fontDescription key="fontDescription" type="system" pointSize="17"/>
                                 <nil key="textColor"/>
                                 <nil key="highlightedColor"/>
@@ -62,20 +80,29 @@ Installer</string>
                         <viewLayoutGuide key="safeArea" id="6Tk-OE-BBY"/>
                         <color key="backgroundColor" systemColor="systemBackgroundColor"/>
                         <constraints>
+                            <constraint firstItem="DFM-s0-Qyv" firstAttribute="top" secondItem="s80-eR-fY3" secondAttribute="bottom" constant="2" id="33d-ZZ-Sav"/>
                             <constraint firstItem="Wzs-Lb-XiN" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="3hT-dN-t3y"/>
                             <constraint firstItem="s80-eR-fY3" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="CRX-Jh-CcX"/>
+                            <constraint firstItem="DFM-s0-Qyv" firstAttribute="leading" secondItem="iBq-Z7-lRu" secondAttribute="trailing" constant="8" symbolic="YES" id="Dcz-cB-4cr"/>
+                            <constraint firstItem="iBq-Z7-lRu" firstAttribute="top" secondItem="s80-eR-fY3" secondAttribute="bottom" constant="8" symbolic="YES" id="Du5-wG-JGr"/>
                             <constraint firstItem="JVS-Tm-8qS" firstAttribute="top" secondItem="6Tk-OE-BBY" secondAttribute="top" id="QXJ-WA-nNb"/>
+                            <constraint firstItem="s80-eR-fY3" firstAttribute="top" secondItem="1a2-Wm-Mcd" secondAttribute="bottom" constant="13.5" id="ZBp-CB-kGa"/>
                             <constraint firstItem="1a2-Wm-Mcd" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="c2Z-Vi-FvF"/>
                             <constraint firstItem="Wzs-Lb-XiN" firstAttribute="centerY" secondItem="6Tk-OE-BBY" secondAttribute="centerY" id="gpk-JM-ZKB"/>
                             <constraint firstItem="1a2-Wm-Mcd" firstAttribute="top" secondItem="JVS-Tm-8qS" secondAttribute="bottom" constant="19" id="kjo-EL-B0G"/>
+                            <constraint firstItem="2NU-06-Bab" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="kxx-DH-gPg"/>
+                            <constraint firstItem="Xbu-Hl-yex" firstAttribute="top" secondItem="2NU-06-Bab" secondAttribute="bottom" constant="8" symbolic="YES" id="lnR-go-0yB"/>
                             <constraint firstItem="Xbu-Hl-yex" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="pWV-JU-YTc"/>
                             <constraint firstItem="6Tk-OE-BBY" firstAttribute="bottom" secondItem="Xbu-Hl-yex" secondAttribute="bottom" constant="18" id="rj9-jo-r9k"/>
-                            <constraint firstItem="s80-eR-fY3" firstAttribute="top" secondItem="1a2-Wm-Mcd" secondAttribute="bottom" constant="24.5" id="xcO-OB-aq4"/>
+                            <constraint firstItem="iBq-Z7-lRu" firstAttribute="leading" secondItem="6Tk-OE-BBY" secondAttribute="leading" constant="87" id="tiJ-nD-cFR"/>
                             <constraint firstItem="JVS-Tm-8qS" firstAttribute="centerX" secondItem="6Tk-OE-BBY" secondAttribute="centerX" id="yNr-e8-C4N"/>
                         </constraints>
                     </view>
                     <connections>
-                        <outlet property="statusLabel" destination="Wzs-Lb-XiN" id="Vrz-B5-otQ"/>
+                        <outlet property="exploitValueLabel" destination="iBq-Z7-lRu" id="oOb-6o-snU"/>
+                        <outlet property="exploitValueStepper" destination="DFM-s0-Qyv" id="bXF-zH-1ha"/>
+                        <outlet property="modelLabel" destination="2NU-06-Bab" id="YDx-Kj-Wuo"/>
+                        <outlet property="statusLabel" destination="2NU-06-Bab" id="YAZ-G0-Avm"/>
                     </connections>
                 </viewController>
                 <placeholder placeholderIdentifier="IBFirstResponder" id="dkx-z0-nzr" sceneMemberID="firstResponder"/>
diff --git a/Installer/TrollInstaller/TrollInstaller/ViewController.m b/Installer/TrollInstaller/TrollInstaller/ViewController.m
index c656e6f..6508614 100644
--- a/Installer/TrollInstaller/TrollInstaller/ViewController.m
+++ b/Installer/TrollInstaller/TrollInstaller/ViewController.m
@@ -13,8 +13,10 @@
 #import "unarchive.h"
 #import <spawn.h>
 #import <sys/stat.h>
+#include <sys/utsname.h>
 
 extern uint64_t g_self_proc;
+extern int g_exp_fallback;
 
 void badLog(const char* a, ...)
 {
@@ -154,6 +156,9 @@ int dropRoot(void)
 }
 
 @interface ViewController ()
+@property (weak, nonatomic) IBOutlet UILabel *modelLabel;
+@property (weak, nonatomic) IBOutlet UILabel *exploitValueLabel;
+@property (weak, nonatomic) IBOutlet UIStepper *exploitValueStepper;
 @property (weak, nonatomic) IBOutlet UILabel *statusLabel;
 @end
 
@@ -162,6 +167,47 @@ int dropRoot(void)
 - (void)viewDidLoad {
     [super viewDidLoad];
     // Do any additional setup after loading the view.
+    
+    struct utsname u;
+    uname(&u);
+    NSString* nsMachine = [NSString stringWithUTF8String:u.machine];
+    _modelLabel.text = [NSString stringWithFormat:@"Model: %@", nsMachine];
+}
+
+- (void)reloadExploitValue {
+    _exploitValueLabel.text = [NSString stringWithFormat:@"Exploit value: %d", g_exp_fallback];
+}
+
+- (IBAction)stepperValueChanged:(id)sender {
+    int toSet = _exploitValueStepper.value;
+    if(toSet == 0)
+    {
+        if(g_exp_fallback == -1)
+        {
+            toSet = 1;
+        }
+        else
+        {
+            toSet = -1;
+        }
+    }
+    
+    if(toSet < -1)
+    {
+        toSet = -1;
+    }
+    else if(toSet > 5)
+    {
+        toSet = 5;
+    }
+    
+    if(_exploitValueStepper.value != toSet)
+    {
+        _exploitValueStepper.value = toSet;
+    }
+    
+    g_exp_fallback = toSet;
+    [self reloadExploitValue];
 }
 
 - (void)updateStatus:(NSString*)status
diff --git a/Installer/TrollInstaller/TrollInstaller/exploit/IOGPU.c b/Installer/TrollInstaller/TrollInstaller/exploit/IOGPU.c
index db475a7..e248e1e 100644
--- a/Installer/TrollInstaller/TrollInstaller/exploit/IOGPU.c
+++ b/Installer/TrollInstaller/TrollInstaller/exploit/IOGPU.c
@@ -2,6 +2,8 @@
 
 #include <sys/utsname.h>
 
+int g_exp_fallback = -1;
+
 io_connect_t IOGPU_init(void)
 {
     mach_port_t mp = MACH_PORT_NULL;
@@ -79,5 +81,5 @@ int IOGPU_get_command_queue_extra_refills_needed(void)
     
     printf("IOGPU_get_command_queue_extra_refills_needed(): Unknown device %s! May panic in generic part until correct number 1-5 is provided for this device!\n", u.machine);
     
-    return -1;
+    return g_exp_fallback;
 }