MA-7633-2 [Android-Reboot]reboot to fastboot\recovery mode

Bootloader decide to enter recovery or bootloader mode through SNVS register.
Kernel set the SNVS register through reboot notification.

Signed-off-by: zhang sanshan <b51434@freescale.com>
(cherry picked from commit 77a9d23)

Change-Id: I5583073dc49dc2fde51feda3df036d8b5cac0039
Signed-off-by: Eric Nelson <eric@nelint.com>
Signed-off-by: Otavio Salvador <otavio@ossystems.com.br>

arch/arm/mach-imx/Kconfig

@@ -270,4 +270,7 @@ endchoice
 
 endif
 
+config MXC_REBOOT_ANDROID_CMD
+	bool "Android reboot setting"
+
 endif

arch/arm/mach-imx/Makefile

@@ -1,6 +1,10 @@
 # SPDX-License-Identifier: GPL-2.0
 obj-y := cpu.o system.o irq-common.o
 
+obj-$(CONFIG_MXC_REBOOT_ANDROID_CMD) += reboot.o
+
+obj-$(CONFIG_SOC_IMX21) += mm-imx21.o
+
 obj-$(CONFIG_SOC_IMX25) += cpu-imx25.o mach-imx25.o pm-imx25.o
 
 obj-$(CONFIG_SOC_IMX27) += cpu-imx27.o pm-imx27.o mach-imx27.o

arch/arm/mach-imx/reboot.c

@@ -0,0 +1,133 @@
+/*
+ * Copyright (C) 2016 Freescale Semiconductor, Inc. All Rights Reserved.
+ * The code contained herein is licensed under the GNU General Public
+ * License. You may obtain a copy of the GNU General Public License
+ *
+*/
+#include <linux/clk.h>
+#include <linux/io.h>
+#include <linux/notifier.h>
+#include <linux/module.h>
+#include <linux/reboot.h>
+
+#include "hardware.h"
+
+#define MX6_AIPS1_ARB_BASE_ADDR			0x02000000
+#define MX6_ATZ1_BASE_ADDR		MX6_AIPS1_ARB_BASE_ADDR
+#define MX6_AIPS1_OFF_BASE_ADDR		(MX6_ATZ1_BASE_ADDR + 0x80000)
+#define MX6_SNVS_BASE_ADDR		(MX6_AIPS1_OFF_BASE_ADDR + 0x4C000)
+#define MX6_SNVS_LPGPR				0x68
+#define MX6_SNVS_SIZE				(1024*16)
+#define MX7_AIPS1_ARB_BASE_ADDR			0x30000000
+#define MX7_ATZ1_BASE_ADDR		MX7_AIPS1_ARB_BASE_ADDR
+#define MX7_AIPS1_OFF_BASE_ADDR		(MX7_ATZ1_BASE_ADDR + 0x200000)
+#define MX7_SNVS_BASE_ADDR		(MX7_AIPS1_OFF_BASE_ADDR + 0x170000)
+#define MX7_SNVS_LPGPR				0x68
+#define MX7_SNVS_SIZE				(1024*16)
+
+#define ANDROID_RECOVERY_BOOT  (1 << 7)
+#define ANDROID_FASTBOOT_BOOT  (1 << 8)
+
+void do_switch_recovery(void)
+{
+	u32 reg;
+	void *addr;
+	struct clk *snvs_root;
+	if (cpu_is_imx6()) {
+		addr = ioremap(MX6_SNVS_BASE_ADDR, MX6_SNVS_SIZE);
+		if (!addr) {
+			pr_warn("SNVS ioremap failed!\n");
+			return;
+		}
+		reg = __raw_readl(addr + MX6_SNVS_LPGPR);
+		reg |= ANDROID_RECOVERY_BOOT;
+		__raw_writel(reg, (addr + MX6_SNVS_LPGPR));
+	} else {
+		snvs_root = clk_get_sys("imx-snvs.0", "snvs");
+		addr = ioremap(MX7_SNVS_BASE_ADDR, MX7_SNVS_SIZE);
+		if (!addr) {
+			pr_warn("SNVS ioremap failed!\n");
+			return;
+		}
+		clk_enable(snvs_root);
+		reg = __raw_readl(addr + MX7_SNVS_LPGPR);
+		reg |= ANDROID_RECOVERY_BOOT;
+		__raw_writel(reg, (addr + MX7_SNVS_LPGPR));
+		clk_disable(snvs_root);
+	}
+	iounmap(addr);
+}
+
+void do_switch_fastboot(void)
+{
+	u32 reg;
+	void *addr;
+	struct clk *snvs_root;
+	if (cpu_is_imx6()) {
+		addr = ioremap(MX6_SNVS_BASE_ADDR, MX6_SNVS_SIZE);
+		if (!addr) {
+			pr_warn("SNVS ioremap failed!\n");
+			return;
+		}
+		reg = __raw_readl(addr + MX6_SNVS_LPGPR);
+		reg |= ANDROID_FASTBOOT_BOOT;
+		__raw_writel(reg, addr + MX6_SNVS_LPGPR);
+	} else {
+		snvs_root = clk_get_sys("imx-snvs.0", "snvs");
+		addr = ioremap(MX7_SNVS_BASE_ADDR, MX7_SNVS_SIZE);
+		if (!addr) {
+			pr_warn("SNVS ioremap failed!\n");
+			return;
+		}
+		clk_enable(snvs_root);
+		reg = __raw_readl(addr + MX7_SNVS_LPGPR);
+		reg |= ANDROID_FASTBOOT_BOOT;
+		__raw_writel(reg, addr + MX7_SNVS_LPGPR);
+		clk_disable(snvs_root);
+	}
+	iounmap(addr);
+}
+
+static void restart_special_mode(const char *cmd)
+{
+	if (cmd && strcmp(cmd, "recovery") == 0)
+		do_switch_recovery();
+	else if (cmd && strcmp(cmd, "bootloader") == 0)
+		do_switch_fastboot();
+}
+
+static int imx_reboot_notifier_call(
+		struct notifier_block *notifier,
+		unsigned long what, void *data)
+{
+	int ret = NOTIFY_DONE;
+	char *cmd = (char *)data;
+	restart_special_mode(cmd);
+
+	return ret;
+}
+
+
+static struct notifier_block imx_reboot_notifier = {
+	.notifier_call = imx_reboot_notifier_call,
+};
+
+static int __init reboot_init(void)
+{
+	if (register_reboot_notifier(&imx_reboot_notifier)) {
+		pr_err("unable to register reboot notifier\n");
+		return -1;
+	}
+	return 0;
+}
+module_init(reboot_init);
+
+static void __exit reboot_exit(void)
+{
+	unregister_reboot_notifier(&imx_reboot_notifier);
+}
+module_exit(reboot_exit);
+
+MODULE_AUTHOR("Freescale Semiconductor, Inc.");
+MODULE_DESCRIPTION("bootloader communication module");
+MODULE_LICENSE("GPL v2");