recovery: add truedualboot support

Change-Id: I80501250ee23146f57009d7968eb65b73e1fceaf

Author: M1cha

BoardConfig.mk

@@ -89,6 +89,7 @@ TARGET_USES_SF_BYPASS := true
 TARGET_USES_C2D_COMPOSITION := true
 
 BOARD_NATIVE_DUALBOOT := true
+BOARD_NATIVE_DUALBOOT_SINGLEDATA := true
 TARGET_RECOVERY_PIXEL_FORMAT := "RGBX_8888"
 TARGET_RECOVERY_UI_LIB := librecovery_ui_aries
 BOARD_USE_CUSTOM_RECOVERY_FONT := \"roboto_15x24.h\"

device.mk

@@ -42,6 +42,11 @@ PRODUCT_PACKAGES += \
         VisualizationWallpapers \
         librs_jni
 
+# Dualboot
+PRODUCT_COPY_FILES += \
+	device/xiaomi/aries/recovery/mount_ext4_default.sh:recovery/root/res/dualboot/mount_ext4_default.sh \
+	device/xiaomi/aries/recovery/mount_ext4_tdb.sh:recovery/root/res/dualboot/mount_ext4_tdb.sh
+
 PRODUCT_COPY_FILES += \
 	device/xiaomi/aries/WCNSS_cfg.dat:system/vendor/firmware/wlan/prima/WCNSS_cfg.dat \
 	device/xiaomi/aries/WCNSS_qcom_cfg.ini:system/etc/wifi/WCNSS_qcom_cfg.ini \

recovery/dualboot.c

@@ -6,10 +6,12 @@
 #include "common.h"
 #include "recovery_ui.h"
 #include "mounts.h"
+#include "roots.h"
 
 extern int ui_menu_level;
 extern int ui_root_menu;
 static enum dualboot_system selected_system = INVALID_SYSTEM;
+static int force_raw_format = 0;
 
 struct partition {
 	char path[PATH_MAX];
@@ -28,7 +30,7 @@ enum dualboot_system get_selected_system(void) {
 }
 
 static int select_system(const char* title) {
-	char* headers[] = { title, "", NULL };
+	const char* headers[] = { title, "", NULL };
 	char* items[] = { "System1", "System2", NULL };
 	enum dualboot_system chosen_item = -1;
 
@@ -85,13 +87,13 @@ static int replace_device_node(int num, struct stat* statbuf) {
 	MountedVolume * mv = find_mounted_volume_by_real_node(part_table[num].path);
 
 	if(mv!=NULL && unmount_mounted_volume(mv)!=0) {
-		LOGE("replace_device_node: could not unmount device!\n");
+		LOGE("could not unmount device!\n");
 		return -1;
 	} 
 
 	unlink(part_table[num].path);
 	if(mknod(part_table[num].path, statbuf->st_mode, statbuf->st_rdev)!=0) {
-		LOGE("replace_device_node: could not create node!\n");
+		LOGE("could not create node!\n");
 		return -1;
 	}
 
@@ -152,3 +154,42 @@ void dualboot_set_system(enum dualboot_system sys) {
 	selected_system = sys;
 	dualboot_setup_env();
 }
+
+static int dualboot_mount_dataroot(void) {
+	int rc = ensure_path_mounted_at_mount_point("/data", MOUNTPOINT_DATAROOT);
+	if(rc)
+		LOGE("failed mounting dataroot!\n");
+	return rc;
+}
+
+int dualboot_is_tdb_enabled(void) {
+	struct stat st;
+
+	if(dualboot_mount_dataroot())
+		return 0;
+
+	return lstat(TDB_FILE, &st)==0;
+}
+
+void dualboot_set_tdb_enabled(int enable) {
+	if(dualboot_mount_dataroot())
+		return;
+
+		if(enable) {
+			FILE * pFile = fopen(TDB_FILE, "w");
+			if(pFile==NULL) {
+				LOGE("TrueDualBoot: failed creating file");
+			return;
+		}
+		fclose(pFile);
+	}
+	else remove(TDB_FILE);
+}
+
+int is_force_raw_format_enabled(void) {
+	return !!force_raw_format;
+}
+
+void set_force_raw_format_enabled(int enabled) {
+	force_raw_format = enabled;
+}

recovery/dualboot.h

@@ -1,3 +1,8 @@
+#define MOUNTPOINT_DATAROOT "/data_root"
+#define TDB_FILE MOUNTPOINT_DATAROOT "/.truedualboot"
+#define PATH_USERDATA_NODE "/dev/block/mmcblk0p26"
+#define PATH_USERDATA_NODE_BACKUP "/dev/part_backup_userdata"
+
 enum dualboot_system {
 	INVALID_SYSTEM = -1,
 	SYSTEM1,
@@ -9,3 +14,8 @@ void dualboot_show_selection_ui(void);
 enum dualboot_system get_selected_system(void);
 void dualboot_setup_env(void);
 void dualboot_set_system(enum dualboot_system sys);
+
+void dualboot_set_tdb_enabled(int enable);
+int dualboot_is_tdb_enabled(void);
+int is_force_raw_format_enabled(void);
+void set_force_raw_format_enabled(int enabled);

recovery/mount_ext4_default.sh

@@ -0,0 +1,24 @@
+#!/system/bin/sh
+export PATH=/system/bin:/system/xbin:$PATH
+BLOCK_DEVICE=$1
+MOUNT_POINT=$2
+LOG_FILE="/dev/null"
+LOG_LOCATION="/data/.fsck_log/"
+
+# storage log
+if [ "${MOUNT_POINT}" == "/storage_int" ]; then
+    mkdir ${LOG_LOCATION}
+    busybox find /data/.fsck_log/ -type f -mtime +7  -exec rm {} \;
+    TIMESTAMP=`date +%F_%H-%M-%S`
+    LOG_FILE=${LOG_LOCATION}/storage_${TIMESTAMP}.log
+fi
+
+# mount partition
+if [ -e ${BLOCK_DEVICE} ]; then
+    mount -t ext4 -o nosuid,nodev,barrier=1,noauto_da_alloc ${BLOCK_DEVICE} ${MOUNT_POINT}
+fi
+
+# hide recovery partition
+RECOVERY_NODE="$(busybox readlink -f /dev/block/platform/msm_sdcc.1/by-name/recovery)"
+busybox mv "${RECOVERY_NODE}" /dev/recovery_moved
+busybox mknod -m 0600 "${RECOVERY_NODE}" b 1 3

recovery/mount_ext4_tdb.sh

@@ -0,0 +1,57 @@
+#!/system/bin/sh
+export PATH=/system/bin:/system/xbin:$PATH
+BLOCK_DEVICE=$1
+MOUNT_POINT=$2
+LOG_FILE="/dev/null"
+LOG_LOCATION="/data/.fsck_log/"
+
+# get syspart-flag from cmdline
+set -- $(cat /proc/cmdline)
+for x in "$@"; do
+    case "$x" in
+        syspart=*)
+        SYSPART=$(echo "${x#syspart=}")
+        ;;
+    esac
+done
+
+# storage log
+if [ "${MOUNT_POINT}" == "/storage_int" ]; then
+    mkdir ${LOG_LOCATION}
+    busybox find /data/.fsck_log/ -type f -mtime +7  -exec rm {} \;
+    TIMESTAMP=`date +%F_%H-%M-%S`
+    LOG_FILE=${LOG_LOCATION}/storage_${TIMESTAMP}.log
+fi
+
+# mount partition
+if [ -e ${BLOCK_DEVICE} ]; then
+    # userdata
+    if [ "${BLOCK_DEVICE}" == "/dev/block/platform/msm_sdcc.1/by-name/userdata" ];then
+        if [ "${SYSPART}" == "system" ];then
+            BINDMOUNT_PATH="/data_root/system0"
+        elif [ "${SYSPART}" == "system1" ];then
+            BINDMOUNT_PATH="/data_root/system1"
+        else
+            reboot recovery
+        fi
+
+        # mount /data_root
+        mkdir -p /data_root
+        chmod 0755 /data_root
+        mount -t ext4 -o nosuid,nodev,barrier=1,noauto_da_alloc ${BLOCK_DEVICE} /data_root
+
+        # bind mount
+        mkdir -p ${BINDMOUNT_PATH}
+        chmod 0755 ${BINDMOUNT_PATH}
+        mount -o bind ${BINDMOUNT_PATH} ${MOUNT_POINT}
+
+    # normal mount
+    else
+        mount -t ext4 -o nosuid,nodev,barrier=1,noauto_da_alloc ${BLOCK_DEVICE} ${MOUNT_POINT}
+    fi
+fi
+
+# hide recovery partition
+RECOVERY_NODE="$(busybox readlink -f /dev/block/platform/msm_sdcc.1/by-name/recovery)"
+busybox mv "${RECOVERY_NODE}" /dev/recovery_moved
+busybox mknod -m 0600 "${RECOVERY_NODE}" b 1 3