From 9e1b5e0be248150d3207e5b79716f9a6c27fcd79 Mon Sep 17 00:00:00 2001
From: aspargas2 <36926490+aspargas2@users.noreply.github.com>
Date: Thu, 28 Nov 2019 16:47:28 -0500
Subject: [PATCH] add a9lh stage2 payload removal to hax uninstall

---
 resources/gm9/scripts/GM9Megascript.gm9 | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/resources/gm9/scripts/GM9Megascript.gm9 b/resources/gm9/scripts/GM9Megascript.gm9
index 24aa886..7850fe3 100644
--- a/resources/gm9/scripts/GM9Megascript.gm9
+++ b/resources/gm9/scripts/GM9Megascript.gm9
@@ -438,7 +438,12 @@ if	allow -a S:
 	rm -o -s 1:/rw/luma
 	cp -n $[NATIVE_FIRM] S:/firm0.bin
 	cp -n $[NATIVE_FIRM] S:/firm1.bin
-	echo "CFW un-installed successfully"
+	shaget S:/nand.bin@57FFE00:200 PRE_STAGE2_HASH
+	if not sha S:/nand.bin@B800000:200 $[PRE_STAGE2_HASH]
+		fget S:/nand.bin@57FFE00:1 PRE_STAGE2
+		fill S:/nand.bin@B7FFE00:89C00 $[PRE_STAGE2]
+	end
+	echo "CFW uninstalled successfully"
 else
 	echo "Permissions denied. Aborting."
 end
@@ -457,12 +462,17 @@ set NATIVE_FIRM $[GM9OUT]/NATIVE_FIRM.firm
 cp -w G:/exefs/.firm $[NATIVE_FIRM]
 imgumount
 
-if	allow S:/nand.bin
-	allow 1:
+if	allow -a S:
+	allow -a 1:
 	rm -o -s 1:/boot.firm
 	rm -o -s 1:/rw/luma
 	cp -n $[NATIVE_FIRM] S:/firm0.bin
 	cp -n $[NATIVE_FIRM] S:/firm1.bin
+	shaget S:/nand.bin@57FFE00:200 PRE_STAGE2_HASH
+	if not sha S:/nand.bin@B800000:200 $[PRE_STAGE2_HASH]
+		fget S:/nand.bin@57FFE00:1 PRE_STAGE2
+		fill S:/nand.bin@B7FFE00:89C00 $[PRE_STAGE2]
+	end
 	echo "CFW uninstalled successfully"
 else
 	echo "Permissions denied. Aborting."