From 86f778c0aa21f1cc7a358331c102180488b3351b Mon Sep 17 00:00:00 2001
From: topjohnwu <topjohnwu@gmail.com>
Date: Sun, 24 Feb 2019 02:30:04 -0500
Subject: [PATCH] Group unsupported formats into the same code

---
 native/jni/magiskboot/bootimg.cpp | 14 +++++---------
 scripts/boot_patch.sh             |  9 +--------
 2 files changed, 6 insertions(+), 17 deletions(-)

diff --git a/native/jni/magiskboot/bootimg.cpp b/native/jni/magiskboot/bootimg.cpp
index 56d1a6687..0acf336c0 100644
--- a/native/jni/magiskboot/bootimg.cpp
+++ b/native/jni/magiskboot/bootimg.cpp
@@ -44,10 +44,8 @@ boot_img::~boot_img() {
 	delete b_hdr;
 }
 
-#define CHROMEOS_RET       2
-#define ELF32_RET          3
-#define ELF64_RET          4
-
+#define UNSUPP_RET  1
+#define CHROME_RET  2
 int boot_img::parse_file(const char *image) {
 	mmap_ro(image, (void **) &map_addr, &map_size);
 	fprintf(stderr, "Parsing boot image: [%s]\n", image);
@@ -74,15 +72,13 @@ int boot_img::parse_file(const char *image) {
 
 		/* Unsupported */
 		case ELF32:
-			exit(ELF32_RET);
 		case ELF64:
-			exit(ELF64_RET);
+			exit(UNSUPP_RET);
 		default:
 			break;
 		}
 	}
-	LOGE("No boot image magic found!\n");
-	exit(1);
+	exit(UNSUPP_RET);
 }
 
 #define pos_align() pos = do_align(pos, hdr.page_size())
@@ -177,7 +173,7 @@ int boot_img::parse_image(uint8_t *head) {
 	fprintf(stderr, "KERNEL_FMT      [%s]\n", fmt2name[k_fmt]);
 	fprintf(stderr, "RAMDISK_FMT     [%s]\n", fmt2name[r_fmt]);
 
-	return (flags & CHROMEOS_FLAG) ? CHROMEOS_RET : 0;
+	return (flags & CHROMEOS_FLAG) ? CHROME_RET : 0;
 }
 
 void boot_img::find_dtb() {
diff --git a/scripts/boot_patch.sh b/scripts/boot_patch.sh
index 4fc3add03..4c4836335 100644
--- a/scripts/boot_patch.sh
+++ b/scripts/boot_patch.sh
@@ -75,19 +75,12 @@ ui_print "- Unpacking boot image"
 
 case $? in
   1 )
-    abort "! Unable to unpack boot image"
+    abort "! Unsupported/Unknown image format"
     ;;
   2 )
     ui_print "- ChromeOS boot image detected"
     CHROMEOS=true
     ;;
-  3 )
-    ui_print "! Sony ELF32 format detected"
-    abort "! Please use BootBridge from @AdrianDC to flash Magisk"
-    ;;
-  4 )
-    ui_print "! Sony ELF64 format detected"
-    abort "! Stock kernel cannot be patched, please use a custom kernel"
 esac
 
 ##########################################################################################