From 4f250df8f936fb4a76848b3e81256f1c4f999724 Mon Sep 17 00:00:00 2001
From: Christopher Roy Bratusek <nano@jpberlin.de>
Date: Tue, 8 Oct 2019 20:46:47 +0200
Subject: [PATCH] Patcher: update mount code

---
 patcher/CommonPatcher | 56 +++++++++++++++++++++++++++++++------------
 1 file changed, 41 insertions(+), 15 deletions(-)

diff --git a/patcher/CommonPatcher b/patcher/CommonPatcher
index c20a289..81b63b7 100644
--- a/patcher/CommonPatcher
+++ b/patcher/CommonPatcher
@@ -53,23 +53,30 @@ is_mounted () {
 
 # taken from Magisk, with minor modifications for NanoDroid
 mount_partitions () {
-	SLOT=$(grep_cmdline androidboot.slot_suffix)
-	if [ -z ${SLOT} ]; then
-		SLOT=_$(grep_cmdline androidboot.slot)
-		[ "${SLOT}" = "_" ] && SLOT=
+	DEVICE_AB=FALSE
+	VENDOR_COMPAT=FALSE
+
+	SYSTEM_AS_ROOT=$(getprop ro.build.system_root_image)
+	SYSTEM_BLOCK=/dev/block/bootdevice/by-name/system
+
+	RUNTIME_APEX=""
+	APEX_LOOP=""
+
+	if [ "${SYSTEM_AS_ROOT}" == "true" ]; then
+		SLOT=$(grep_cmdline androidboot.slot_suffix)
+		if [ -n "${SLOT}" ]; then
+			DEVICE_AB=TRUE
+			SYSTEM_BLOCK=/dev/block/bootdevice/by-name/system${SLOT}
+		fi
+		SYSTEM_MOUNT=/system_root
+	else
+		SYSTEM_MOUNT=/system
 	fi
 
 	is_mounted /data || mount /data || error "failed to mount /data!"
 
-	${BOOTMODE} || mount -o bind /dev/urandom /dev/random
-
-	! is_mounted /system && mount -o rw /system || mount -o rw,remount /system
-
-	if ! is_mounted /system; then
-		sys_tmp=$(find /dev/block -type l -iname system${SLOT} | head -n 1)
-		SYSTEMBLOCK=$(readlink -f ${sys_tmp})
-		mount -o rw ${SYSTEMBLOCK} /system
-	fi
+	mount -o bind /dev/urandom /dev/random
+	! is_mounted ${SYSTEM_MOUNT} && mount -o rw ${SYSTEM_BLOCK} ${SYSTEM_MOUNT} || mount -o rw,remount ${SYSTEM_MOUNT}
 
 	if [ -f /system/init.rc ]; then
 		[ -L /system_root ] && rm -f /system_root
@@ -80,8 +87,26 @@ mount_partitions () {
 
 	[ ! -f /system/build.prop ] && error "failed to mount /system (unsupported A/B device?)"
 
+	if [ -f /system/apex/com.android.runtime.apex ]; then
+		RUNTIME_APEX="/system/apex/com.android.runtime.apex"
+	elif [ -f /system/apex/com.android.runtime.release.apex ]; then
+		RUNTIME_APEX="/system/apex/com.android.runtime.release.apex"
+	fi
+
+	if [ -n "${RUNTIME_APEX}" ]; then
+		mkdir -p /dev/tmp/apex
+		unzip "${RUNTIME_APEX}" -d /dev/tmp/apex
+
+		mkdir -p /apex/com.android.runtime
+
+		mount -oloop,ro /dev/tmp/apex/apex_payload.img /apex/com.android.runtime
+
+		APEX_LOOP=$(mount | awk '/com.android.runtime/{print $1}')
+	fi
+
+	! is_mounted /vendor && mount /vendor
+
 	if [ -L /system/vendor ]; then
-		! is_mounted /vendor && mount /vendor
 		if ! is_mounted /vendor; then
 			vendor_tmp=$(find /dev/block -type l -iname vendor${SLOT} | head -n 1)
 			VENDORBLOCK=$(readlink -f ${vendor_tmp})
@@ -89,11 +114,12 @@ mount_partitions () {
 		fi
 	elif [ -d /system/vendor ]; then
 		### XXX work-around required for some ROMs
+		VENDOR_COMPAT=TRUE
 		echo " xxx compat /vendor link created!"
 		ln -sf /system/vendor /vendor >/dev/null
 	fi
 
-	mount | awk '{print $1 " on " $3}'
+	mount | awk '{print $1 " on " $3 " params: " $6}'
 }
 
 error () {