cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

pm-imx6.c (19198B)


      1// SPDX-License-Identifier: GPL-2.0-or-later
      2/*
      3 * Copyright 2011-2014 Freescale Semiconductor, Inc.
      4 * Copyright 2011 Linaro Ltd.
      5 */
      6
      7#include <linux/clk/imx.h>
      8#include <linux/delay.h>
      9#include <linux/init.h>
     10#include <linux/io.h>
     11#include <linux/irq.h>
     12#include <linux/genalloc.h>
     13#include <linux/irqchip/arm-gic.h>
     14#include <linux/mfd/syscon.h>
     15#include <linux/mfd/syscon/imx6q-iomuxc-gpr.h>
     16#include <linux/of.h>
     17#include <linux/of_address.h>
     18#include <linux/of_platform.h>
     19#include <linux/regmap.h>
     20#include <linux/suspend.h>
     21#include <asm/cacheflush.h>
     22#include <asm/fncpy.h>
     23#include <asm/proc-fns.h>
     24#include <asm/suspend.h>
     25#include <asm/tlb.h>
     26
     27#include "common.h"
     28#include "hardware.h"
     29
     30#define CCR				0x0
     31#define BM_CCR_WB_COUNT			(0x7 << 16)
     32#define BM_CCR_RBC_BYPASS_COUNT		(0x3f << 21)
     33#define BM_CCR_RBC_EN			(0x1 << 27)
     34
     35#define CLPCR				0x54
     36#define BP_CLPCR_LPM			0
     37#define BM_CLPCR_LPM			(0x3 << 0)
     38#define BM_CLPCR_BYPASS_PMIC_READY	(0x1 << 2)
     39#define BM_CLPCR_ARM_CLK_DIS_ON_LPM	(0x1 << 5)
     40#define BM_CLPCR_SBYOS			(0x1 << 6)
     41#define BM_CLPCR_DIS_REF_OSC		(0x1 << 7)
     42#define BM_CLPCR_VSTBY			(0x1 << 8)
     43#define BP_CLPCR_STBY_COUNT		9
     44#define BM_CLPCR_STBY_COUNT		(0x3 << 9)
     45#define BM_CLPCR_COSC_PWRDOWN		(0x1 << 11)
     46#define BM_CLPCR_WB_PER_AT_LPM		(0x1 << 16)
     47#define BM_CLPCR_WB_CORE_AT_LPM		(0x1 << 17)
     48#define BM_CLPCR_BYP_MMDC_CH0_LPM_HS	(0x1 << 19)
     49#define BM_CLPCR_BYP_MMDC_CH1_LPM_HS	(0x1 << 21)
     50#define BM_CLPCR_MASK_CORE0_WFI		(0x1 << 22)
     51#define BM_CLPCR_MASK_CORE1_WFI		(0x1 << 23)
     52#define BM_CLPCR_MASK_CORE2_WFI		(0x1 << 24)
     53#define BM_CLPCR_MASK_CORE3_WFI		(0x1 << 25)
     54#define BM_CLPCR_MASK_SCU_IDLE		(0x1 << 26)
     55#define BM_CLPCR_MASK_L2CC_IDLE		(0x1 << 27)
     56
     57#define CGPR				0x64
     58#define BM_CGPR_INT_MEM_CLK_LPM		(0x1 << 17)
     59
     60#define MX6Q_SUSPEND_OCRAM_SIZE		0x1000
     61#define MX6_MAX_MMDC_IO_NUM		33
     62
     63static void __iomem *ccm_base;
     64static void __iomem *suspend_ocram_base;
     65static void (*imx6_suspend_in_ocram_fn)(void __iomem *ocram_vbase);
     66
     67/*
     68 * suspend ocram space layout:
     69 * ======================== high address ======================
     70 *                              .
     71 *                              .
     72 *                              .
     73 *                              ^
     74 *                              ^
     75 *                              ^
     76 *                      imx6_suspend code
     77 *              PM_INFO structure(imx6_cpu_pm_info)
     78 * ======================== low address =======================
     79 */
     80
     81struct imx6_pm_base {
     82	phys_addr_t pbase;
     83	void __iomem *vbase;
     84};
     85
     86struct imx6_pm_socdata {
     87	u32 ddr_type;
     88	const char *mmdc_compat;
     89	const char *src_compat;
     90	const char *iomuxc_compat;
     91	const char *gpc_compat;
     92	const char *pl310_compat;
     93	const u32 mmdc_io_num;
     94	const u32 *mmdc_io_offset;
     95};
     96
     97static const u32 imx6q_mmdc_io_offset[] __initconst = {
     98	0x5ac, 0x5b4, 0x528, 0x520, /* DQM0 ~ DQM3 */
     99	0x514, 0x510, 0x5bc, 0x5c4, /* DQM4 ~ DQM7 */
    100	0x56c, 0x578, 0x588, 0x594, /* CAS, RAS, SDCLK_0, SDCLK_1 */
    101	0x5a8, 0x5b0, 0x524, 0x51c, /* SDQS0 ~ SDQS3 */
    102	0x518, 0x50c, 0x5b8, 0x5c0, /* SDQS4 ~ SDQS7 */
    103	0x784, 0x788, 0x794, 0x79c, /* GPR_B0DS ~ GPR_B3DS */
    104	0x7a0, 0x7a4, 0x7a8, 0x748, /* GPR_B4DS ~ GPR_B7DS */
    105	0x59c, 0x5a0, 0x750, 0x774, /* SODT0, SODT1, MODE_CTL, MODE */
    106	0x74c,			    /* GPR_ADDS */
    107};
    108
    109static const u32 imx6dl_mmdc_io_offset[] __initconst = {
    110	0x470, 0x474, 0x478, 0x47c, /* DQM0 ~ DQM3 */
    111	0x480, 0x484, 0x488, 0x48c, /* DQM4 ~ DQM7 */
    112	0x464, 0x490, 0x4ac, 0x4b0, /* CAS, RAS, SDCLK_0, SDCLK_1 */
    113	0x4bc, 0x4c0, 0x4c4, 0x4c8, /* DRAM_SDQS0 ~ DRAM_SDQS3 */
    114	0x4cc, 0x4d0, 0x4d4, 0x4d8, /* DRAM_SDQS4 ~ DRAM_SDQS7 */
    115	0x764, 0x770, 0x778, 0x77c, /* GPR_B0DS ~ GPR_B3DS */
    116	0x780, 0x784, 0x78c, 0x748, /* GPR_B4DS ~ GPR_B7DS */
    117	0x4b4, 0x4b8, 0x750, 0x760, /* SODT0, SODT1, MODE_CTL, MODE */
    118	0x74c,			    /* GPR_ADDS */
    119};
    120
    121static const u32 imx6sl_mmdc_io_offset[] __initconst = {
    122	0x30c, 0x310, 0x314, 0x318, /* DQM0 ~ DQM3 */
    123	0x5c4, 0x5cc, 0x5d4, 0x5d8, /* GPR_B0DS ~ GPR_B3DS */
    124	0x300, 0x31c, 0x338, 0x5ac, /* CAS, RAS, SDCLK_0, GPR_ADDS */
    125	0x33c, 0x340, 0x5b0, 0x5c0, /* SODT0, SODT1, MODE_CTL, MODE */
    126	0x330, 0x334, 0x320,        /* SDCKE0, SDCKE1, RESET */
    127};
    128
    129static const u32 imx6sll_mmdc_io_offset[] __initconst = {
    130	0x294, 0x298, 0x29c, 0x2a0, /* DQM0 ~ DQM3 */
    131	0x544, 0x54c, 0x554, 0x558, /* GPR_B0DS ~ GPR_B3DS */
    132	0x530, 0x540, 0x2ac, 0x52c, /* MODE_CTL, MODE, SDCLK_0, GPR_ADDDS */
    133	0x2a4, 0x2a8,		    /* SDCKE0, SDCKE1*/
    134};
    135
    136static const u32 imx6sx_mmdc_io_offset[] __initconst = {
    137	0x2ec, 0x2f0, 0x2f4, 0x2f8, /* DQM0 ~ DQM3 */
    138	0x60c, 0x610, 0x61c, 0x620, /* GPR_B0DS ~ GPR_B3DS */
    139	0x300, 0x2fc, 0x32c, 0x5f4, /* CAS, RAS, SDCLK_0, GPR_ADDS */
    140	0x310, 0x314, 0x5f8, 0x608, /* SODT0, SODT1, MODE_CTL, MODE */
    141	0x330, 0x334, 0x338, 0x33c, /* SDQS0 ~ SDQS3 */
    142};
    143
    144static const u32 imx6ul_mmdc_io_offset[] __initconst = {
    145	0x244, 0x248, 0x24c, 0x250, /* DQM0, DQM1, RAS, CAS */
    146	0x27c, 0x498, 0x4a4, 0x490, /* SDCLK0, GPR_B0DS-B1DS, GPR_ADDS */
    147	0x280, 0x284, 0x260, 0x264, /* SDQS0~1, SODT0, SODT1 */
    148	0x494, 0x4b0,	            /* MODE_CTL, MODE, */
    149};
    150
    151static const struct imx6_pm_socdata imx6q_pm_data __initconst = {
    152	.mmdc_compat = "fsl,imx6q-mmdc",
    153	.src_compat = "fsl,imx6q-src",
    154	.iomuxc_compat = "fsl,imx6q-iomuxc",
    155	.gpc_compat = "fsl,imx6q-gpc",
    156	.pl310_compat = "arm,pl310-cache",
    157	.mmdc_io_num = ARRAY_SIZE(imx6q_mmdc_io_offset),
    158	.mmdc_io_offset = imx6q_mmdc_io_offset,
    159};
    160
    161static const struct imx6_pm_socdata imx6dl_pm_data __initconst = {
    162	.mmdc_compat = "fsl,imx6q-mmdc",
    163	.src_compat = "fsl,imx6q-src",
    164	.iomuxc_compat = "fsl,imx6dl-iomuxc",
    165	.gpc_compat = "fsl,imx6q-gpc",
    166	.pl310_compat = "arm,pl310-cache",
    167	.mmdc_io_num = ARRAY_SIZE(imx6dl_mmdc_io_offset),
    168	.mmdc_io_offset = imx6dl_mmdc_io_offset,
    169};
    170
    171static const struct imx6_pm_socdata imx6sl_pm_data __initconst = {
    172	.mmdc_compat = "fsl,imx6sl-mmdc",
    173	.src_compat = "fsl,imx6sl-src",
    174	.iomuxc_compat = "fsl,imx6sl-iomuxc",
    175	.gpc_compat = "fsl,imx6sl-gpc",
    176	.pl310_compat = "arm,pl310-cache",
    177	.mmdc_io_num = ARRAY_SIZE(imx6sl_mmdc_io_offset),
    178	.mmdc_io_offset = imx6sl_mmdc_io_offset,
    179};
    180
    181static const struct imx6_pm_socdata imx6sll_pm_data __initconst = {
    182	.mmdc_compat = "fsl,imx6sll-mmdc",
    183	.src_compat = "fsl,imx6sll-src",
    184	.iomuxc_compat = "fsl,imx6sll-iomuxc",
    185	.gpc_compat = "fsl,imx6sll-gpc",
    186	.pl310_compat = "arm,pl310-cache",
    187	.mmdc_io_num = ARRAY_SIZE(imx6sll_mmdc_io_offset),
    188	.mmdc_io_offset = imx6sll_mmdc_io_offset,
    189};
    190
    191static const struct imx6_pm_socdata imx6sx_pm_data __initconst = {
    192	.mmdc_compat = "fsl,imx6sx-mmdc",
    193	.src_compat = "fsl,imx6sx-src",
    194	.iomuxc_compat = "fsl,imx6sx-iomuxc",
    195	.gpc_compat = "fsl,imx6sx-gpc",
    196	.pl310_compat = "arm,pl310-cache",
    197	.mmdc_io_num = ARRAY_SIZE(imx6sx_mmdc_io_offset),
    198	.mmdc_io_offset = imx6sx_mmdc_io_offset,
    199};
    200
    201static const struct imx6_pm_socdata imx6ul_pm_data __initconst = {
    202	.mmdc_compat = "fsl,imx6ul-mmdc",
    203	.src_compat = "fsl,imx6ul-src",
    204	.iomuxc_compat = "fsl,imx6ul-iomuxc",
    205	.gpc_compat = "fsl,imx6ul-gpc",
    206	.pl310_compat = NULL,
    207	.mmdc_io_num = ARRAY_SIZE(imx6ul_mmdc_io_offset),
    208	.mmdc_io_offset = imx6ul_mmdc_io_offset,
    209};
    210
    211/*
    212 * This structure is for passing necessary data for low level ocram
    213 * suspend code(arch/arm/mach-imx/suspend-imx6.S), if this struct
    214 * definition is changed, the offset definition in
    215 * arch/arm/mach-imx/suspend-imx6.S must be also changed accordingly,
    216 * otherwise, the suspend to ocram function will be broken!
    217 */
    218struct imx6_cpu_pm_info {
    219	phys_addr_t pbase; /* The physical address of pm_info. */
    220	phys_addr_t resume_addr; /* The physical resume address for asm code */
    221	u32 ddr_type;
    222	u32 pm_info_size; /* Size of pm_info. */
    223	struct imx6_pm_base mmdc_base;
    224	struct imx6_pm_base src_base;
    225	struct imx6_pm_base iomuxc_base;
    226	struct imx6_pm_base ccm_base;
    227	struct imx6_pm_base gpc_base;
    228	struct imx6_pm_base l2_base;
    229	u32 mmdc_io_num; /* Number of MMDC IOs which need saved/restored. */
    230	u32 mmdc_io_val[MX6_MAX_MMDC_IO_NUM][2]; /* To save offset and value */
    231} __aligned(8);
    232
    233void imx6_set_int_mem_clk_lpm(bool enable)
    234{
    235	u32 val = readl_relaxed(ccm_base + CGPR);
    236
    237	val &= ~BM_CGPR_INT_MEM_CLK_LPM;
    238	if (enable)
    239		val |= BM_CGPR_INT_MEM_CLK_LPM;
    240	writel_relaxed(val, ccm_base + CGPR);
    241}
    242
    243void imx6_enable_rbc(bool enable)
    244{
    245	u32 val;
    246
    247	/*
    248	 * need to mask all interrupts in GPC before
    249	 * operating RBC configurations
    250	 */
    251	imx_gpc_mask_all();
    252
    253	/* configure RBC enable bit */
    254	val = readl_relaxed(ccm_base + CCR);
    255	val &= ~BM_CCR_RBC_EN;
    256	val |= enable ? BM_CCR_RBC_EN : 0;
    257	writel_relaxed(val, ccm_base + CCR);
    258
    259	/* configure RBC count */
    260	val = readl_relaxed(ccm_base + CCR);
    261	val &= ~BM_CCR_RBC_BYPASS_COUNT;
    262	val |= enable ? BM_CCR_RBC_BYPASS_COUNT : 0;
    263	writel(val, ccm_base + CCR);
    264
    265	/*
    266	 * need to delay at least 2 cycles of CKIL(32K)
    267	 * due to hardware design requirement, which is
    268	 * ~61us, here we use 65us for safe
    269	 */
    270	udelay(65);
    271
    272	/* restore GPC interrupt mask settings */
    273	imx_gpc_restore_all();
    274}
    275
    276static void imx6q_enable_wb(bool enable)
    277{
    278	u32 val;
    279
    280	/* configure well bias enable bit */
    281	val = readl_relaxed(ccm_base + CLPCR);
    282	val &= ~BM_CLPCR_WB_PER_AT_LPM;
    283	val |= enable ? BM_CLPCR_WB_PER_AT_LPM : 0;
    284	writel_relaxed(val, ccm_base + CLPCR);
    285
    286	/* configure well bias count */
    287	val = readl_relaxed(ccm_base + CCR);
    288	val &= ~BM_CCR_WB_COUNT;
    289	val |= enable ? BM_CCR_WB_COUNT : 0;
    290	writel_relaxed(val, ccm_base + CCR);
    291}
    292
    293int imx6_set_lpm(enum mxc_cpu_pwr_mode mode)
    294{
    295	u32 val = readl_relaxed(ccm_base + CLPCR);
    296
    297	val &= ~BM_CLPCR_LPM;
    298	switch (mode) {
    299	case WAIT_CLOCKED:
    300		break;
    301	case WAIT_UNCLOCKED:
    302		val |= 0x1 << BP_CLPCR_LPM;
    303		val |= BM_CLPCR_ARM_CLK_DIS_ON_LPM;
    304		break;
    305	case STOP_POWER_ON:
    306		val |= 0x2 << BP_CLPCR_LPM;
    307		val &= ~BM_CLPCR_VSTBY;
    308		val &= ~BM_CLPCR_SBYOS;
    309		if (cpu_is_imx6sl())
    310			val |= BM_CLPCR_BYPASS_PMIC_READY;
    311		if (cpu_is_imx6sl() || cpu_is_imx6sx() || cpu_is_imx6ul() ||
    312		    cpu_is_imx6ull() || cpu_is_imx6sll() || cpu_is_imx6ulz())
    313			val |= BM_CLPCR_BYP_MMDC_CH0_LPM_HS;
    314		else
    315			val |= BM_CLPCR_BYP_MMDC_CH1_LPM_HS;
    316		break;
    317	case WAIT_UNCLOCKED_POWER_OFF:
    318		val |= 0x1 << BP_CLPCR_LPM;
    319		val &= ~BM_CLPCR_VSTBY;
    320		val &= ~BM_CLPCR_SBYOS;
    321		break;
    322	case STOP_POWER_OFF:
    323		val |= 0x2 << BP_CLPCR_LPM;
    324		val |= 0x3 << BP_CLPCR_STBY_COUNT;
    325		val |= BM_CLPCR_VSTBY;
    326		val |= BM_CLPCR_SBYOS;
    327		if (cpu_is_imx6sl() || cpu_is_imx6sx())
    328			val |= BM_CLPCR_BYPASS_PMIC_READY;
    329		if (cpu_is_imx6sl() || cpu_is_imx6sx() || cpu_is_imx6ul() ||
    330		    cpu_is_imx6ull() || cpu_is_imx6sll() || cpu_is_imx6ulz())
    331			val |= BM_CLPCR_BYP_MMDC_CH0_LPM_HS;
    332		else
    333			val |= BM_CLPCR_BYP_MMDC_CH1_LPM_HS;
    334		break;
    335	default:
    336		return -EINVAL;
    337	}
    338
    339	/*
    340	 * ERR007265: CCM: When improper low-power sequence is used,
    341	 * the SoC enters low power mode before the ARM core executes WFI.
    342	 *
    343	 * Software workaround:
    344	 * 1) Software should trigger IRQ #32 (IOMUX) to be always pending
    345	 *    by setting IOMUX_GPR1_GINT.
    346	 * 2) Software should then unmask IRQ #32 in GPC before setting CCM
    347	 *    Low-Power mode.
    348	 * 3) Software should mask IRQ #32 right after CCM Low-Power mode
    349	 *    is set (set bits 0-1 of CCM_CLPCR).
    350	 *
    351	 * Note that IRQ #32 is GIC SPI #0.
    352	 */
    353	if (mode != WAIT_CLOCKED)
    354		imx_gpc_hwirq_unmask(0);
    355	writel_relaxed(val, ccm_base + CLPCR);
    356	if (mode != WAIT_CLOCKED)
    357		imx_gpc_hwirq_mask(0);
    358
    359	return 0;
    360}
    361
    362static int imx6q_suspend_finish(unsigned long val)
    363{
    364	if (!imx6_suspend_in_ocram_fn) {
    365		cpu_do_idle();
    366	} else {
    367		/*
    368		 * call low level suspend function in ocram,
    369		 * as we need to float DDR IO.
    370		 */
    371		local_flush_tlb_all();
    372		/* check if need to flush internal L2 cache */
    373		if (!((struct imx6_cpu_pm_info *)
    374			suspend_ocram_base)->l2_base.vbase)
    375			flush_cache_all();
    376		imx6_suspend_in_ocram_fn(suspend_ocram_base);
    377	}
    378
    379	return 0;
    380}
    381
    382static int imx6q_pm_enter(suspend_state_t state)
    383{
    384	switch (state) {
    385	case PM_SUSPEND_STANDBY:
    386		imx6_set_lpm(STOP_POWER_ON);
    387		imx6_set_int_mem_clk_lpm(true);
    388		imx_gpc_pre_suspend(false);
    389		if (cpu_is_imx6sl())
    390			imx6sl_set_wait_clk(true);
    391		/* Zzz ... */
    392		cpu_do_idle();
    393		if (cpu_is_imx6sl())
    394			imx6sl_set_wait_clk(false);
    395		imx_gpc_post_resume();
    396		imx6_set_lpm(WAIT_CLOCKED);
    397		break;
    398	case PM_SUSPEND_MEM:
    399		imx6_set_lpm(STOP_POWER_OFF);
    400		imx6_set_int_mem_clk_lpm(false);
    401		imx6q_enable_wb(true);
    402		/*
    403		 * For suspend into ocram, asm code already take care of
    404		 * RBC setting, so we do NOT need to do that here.
    405		 */
    406		if (!imx6_suspend_in_ocram_fn)
    407			imx6_enable_rbc(true);
    408		imx_gpc_pre_suspend(true);
    409		imx_anatop_pre_suspend();
    410		/* Zzz ... */
    411		cpu_suspend(0, imx6q_suspend_finish);
    412		if (cpu_is_imx6q() || cpu_is_imx6dl())
    413			imx_smp_prepare();
    414		imx_anatop_post_resume();
    415		imx_gpc_post_resume();
    416		imx6_enable_rbc(false);
    417		imx6q_enable_wb(false);
    418		imx6_set_int_mem_clk_lpm(true);
    419		imx6_set_lpm(WAIT_CLOCKED);
    420		break;
    421	default:
    422		return -EINVAL;
    423	}
    424
    425	return 0;
    426}
    427
    428static int imx6q_pm_valid(suspend_state_t state)
    429{
    430	return (state == PM_SUSPEND_STANDBY || state == PM_SUSPEND_MEM);
    431}
    432
    433static const struct platform_suspend_ops imx6q_pm_ops = {
    434	.enter = imx6q_pm_enter,
    435	.valid = imx6q_pm_valid,
    436};
    437
    438static int __init imx6_pm_get_base(struct imx6_pm_base *base,
    439				const char *compat)
    440{
    441	struct device_node *node;
    442	struct resource res;
    443	int ret = 0;
    444
    445	node = of_find_compatible_node(NULL, NULL, compat);
    446	if (!node)
    447		return -ENODEV;
    448
    449	ret = of_address_to_resource(node, 0, &res);
    450	if (ret)
    451		goto put_node;
    452
    453	base->pbase = res.start;
    454	base->vbase = ioremap(res.start, resource_size(&res));
    455	if (!base->vbase)
    456		ret = -ENOMEM;
    457
    458put_node:
    459	of_node_put(node);
    460	return ret;
    461}
    462
    463static int __init imx6q_suspend_init(const struct imx6_pm_socdata *socdata)
    464{
    465	phys_addr_t ocram_pbase;
    466	struct device_node *node;
    467	struct platform_device *pdev;
    468	struct imx6_cpu_pm_info *pm_info;
    469	struct gen_pool *ocram_pool;
    470	unsigned long ocram_base;
    471	int i, ret = 0;
    472	const u32 *mmdc_offset_array;
    473
    474	suspend_set_ops(&imx6q_pm_ops);
    475
    476	if (!socdata) {
    477		pr_warn("%s: invalid argument!\n", __func__);
    478		return -EINVAL;
    479	}
    480
    481	node = of_find_compatible_node(NULL, NULL, "mmio-sram");
    482	if (!node) {
    483		pr_warn("%s: failed to find ocram node!\n", __func__);
    484		return -ENODEV;
    485	}
    486
    487	pdev = of_find_device_by_node(node);
    488	if (!pdev) {
    489		pr_warn("%s: failed to find ocram device!\n", __func__);
    490		ret = -ENODEV;
    491		goto put_node;
    492	}
    493
    494	ocram_pool = gen_pool_get(&pdev->dev, NULL);
    495	if (!ocram_pool) {
    496		pr_warn("%s: ocram pool unavailable!\n", __func__);
    497		ret = -ENODEV;
    498		goto put_device;
    499	}
    500
    501	ocram_base = gen_pool_alloc(ocram_pool, MX6Q_SUSPEND_OCRAM_SIZE);
    502	if (!ocram_base) {
    503		pr_warn("%s: unable to alloc ocram!\n", __func__);
    504		ret = -ENOMEM;
    505		goto put_device;
    506	}
    507
    508	ocram_pbase = gen_pool_virt_to_phys(ocram_pool, ocram_base);
    509
    510	suspend_ocram_base = __arm_ioremap_exec(ocram_pbase,
    511		MX6Q_SUSPEND_OCRAM_SIZE, false);
    512
    513	memset(suspend_ocram_base, 0, sizeof(*pm_info));
    514	pm_info = suspend_ocram_base;
    515	pm_info->pbase = ocram_pbase;
    516	pm_info->resume_addr = __pa_symbol(v7_cpu_resume);
    517	pm_info->pm_info_size = sizeof(*pm_info);
    518
    519	/*
    520	 * ccm physical address is not used by asm code currently,
    521	 * so get ccm virtual address directly.
    522	 */
    523	pm_info->ccm_base.vbase = ccm_base;
    524
    525	ret = imx6_pm_get_base(&pm_info->mmdc_base, socdata->mmdc_compat);
    526	if (ret) {
    527		pr_warn("%s: failed to get mmdc base %d!\n", __func__, ret);
    528		goto put_device;
    529	}
    530
    531	ret = imx6_pm_get_base(&pm_info->src_base, socdata->src_compat);
    532	if (ret) {
    533		pr_warn("%s: failed to get src base %d!\n", __func__, ret);
    534		goto src_map_failed;
    535	}
    536
    537	ret = imx6_pm_get_base(&pm_info->iomuxc_base, socdata->iomuxc_compat);
    538	if (ret) {
    539		pr_warn("%s: failed to get iomuxc base %d!\n", __func__, ret);
    540		goto iomuxc_map_failed;
    541	}
    542
    543	ret = imx6_pm_get_base(&pm_info->gpc_base, socdata->gpc_compat);
    544	if (ret) {
    545		pr_warn("%s: failed to get gpc base %d!\n", __func__, ret);
    546		goto gpc_map_failed;
    547	}
    548
    549	if (socdata->pl310_compat) {
    550		ret = imx6_pm_get_base(&pm_info->l2_base, socdata->pl310_compat);
    551		if (ret) {
    552			pr_warn("%s: failed to get pl310-cache base %d!\n",
    553				__func__, ret);
    554			goto pl310_cache_map_failed;
    555		}
    556	}
    557
    558	pm_info->ddr_type = imx_mmdc_get_ddr_type();
    559	pm_info->mmdc_io_num = socdata->mmdc_io_num;
    560	mmdc_offset_array = socdata->mmdc_io_offset;
    561
    562	for (i = 0; i < pm_info->mmdc_io_num; i++) {
    563		pm_info->mmdc_io_val[i][0] =
    564			mmdc_offset_array[i];
    565		pm_info->mmdc_io_val[i][1] =
    566			readl_relaxed(pm_info->iomuxc_base.vbase +
    567			mmdc_offset_array[i]);
    568	}
    569
    570	imx6_suspend_in_ocram_fn = fncpy(
    571		suspend_ocram_base + sizeof(*pm_info),
    572		&imx6_suspend,
    573		MX6Q_SUSPEND_OCRAM_SIZE - sizeof(*pm_info));
    574
    575	__arm_iomem_set_ro(suspend_ocram_base, MX6Q_SUSPEND_OCRAM_SIZE);
    576
    577	goto put_device;
    578
    579pl310_cache_map_failed:
    580	iounmap(pm_info->gpc_base.vbase);
    581gpc_map_failed:
    582	iounmap(pm_info->iomuxc_base.vbase);
    583iomuxc_map_failed:
    584	iounmap(pm_info->src_base.vbase);
    585src_map_failed:
    586	iounmap(pm_info->mmdc_base.vbase);
    587put_device:
    588	put_device(&pdev->dev);
    589put_node:
    590	of_node_put(node);
    591
    592	return ret;
    593}
    594
    595static void __init imx6_pm_common_init(const struct imx6_pm_socdata
    596					*socdata)
    597{
    598	struct regmap *gpr;
    599	int ret;
    600
    601	WARN_ON(!ccm_base);
    602
    603	if (IS_ENABLED(CONFIG_SUSPEND)) {
    604		ret = imx6q_suspend_init(socdata);
    605		if (ret)
    606			pr_warn("%s: No DDR LPM support with suspend %d!\n",
    607				__func__, ret);
    608	}
    609
    610	/*
    611	 * This is for SW workaround step #1 of ERR007265, see comments
    612	 * in imx6_set_lpm for details of this errata.
    613	 * Force IOMUXC irq pending, so that the interrupt to GPC can be
    614	 * used to deassert dsm_request signal when the signal gets
    615	 * asserted unexpectedly.
    616	 */
    617	gpr = syscon_regmap_lookup_by_compatible("fsl,imx6q-iomuxc-gpr");
    618	if (!IS_ERR(gpr))
    619		regmap_update_bits(gpr, IOMUXC_GPR1, IMX6Q_GPR1_GINT,
    620				   IMX6Q_GPR1_GINT);
    621}
    622
    623static void imx6_pm_stby_poweroff(void)
    624{
    625	gic_cpu_if_down(0);
    626	imx6_set_lpm(STOP_POWER_OFF);
    627	imx6q_suspend_finish(0);
    628
    629	mdelay(1000);
    630
    631	pr_emerg("Unable to poweroff system\n");
    632}
    633
    634static int imx6_pm_stby_poweroff_probe(void)
    635{
    636	if (pm_power_off) {
    637		pr_warn("%s: pm_power_off already claimed  %p %ps!\n",
    638			__func__, pm_power_off, pm_power_off);
    639		return -EBUSY;
    640	}
    641
    642	pm_power_off = imx6_pm_stby_poweroff;
    643	return 0;
    644}
    645
    646void __init imx6_pm_ccm_init(const char *ccm_compat)
    647{
    648	struct device_node *np;
    649	u32 val;
    650
    651	np = of_find_compatible_node(NULL, NULL, ccm_compat);
    652	ccm_base = of_iomap(np, 0);
    653	BUG_ON(!ccm_base);
    654
    655	/*
    656	 * Initialize CCM_CLPCR_LPM into RUN mode to avoid ARM core
    657	 * clock being shut down unexpectedly by WAIT mode.
    658	 */
    659	val = readl_relaxed(ccm_base + CLPCR);
    660	val &= ~BM_CLPCR_LPM;
    661	writel_relaxed(val, ccm_base + CLPCR);
    662
    663	if (of_property_read_bool(np, "fsl,pmic-stby-poweroff"))
    664		imx6_pm_stby_poweroff_probe();
    665
    666	of_node_put(np);
    667}
    668
    669void __init imx6q_pm_init(void)
    670{
    671	imx6_pm_common_init(&imx6q_pm_data);
    672}
    673
    674void __init imx6dl_pm_init(void)
    675{
    676	imx6_pm_common_init(&imx6dl_pm_data);
    677}
    678
    679void __init imx6sl_pm_init(void)
    680{
    681	struct regmap *gpr;
    682
    683	if (cpu_is_imx6sl()) {
    684		imx6_pm_common_init(&imx6sl_pm_data);
    685	} else {
    686		imx6_pm_common_init(&imx6sll_pm_data);
    687		gpr = syscon_regmap_lookup_by_compatible("fsl,imx6q-iomuxc-gpr");
    688		if (!IS_ERR(gpr))
    689			regmap_update_bits(gpr, IOMUXC_GPR5,
    690				IMX6SLL_GPR5_AFCG_X_BYPASS_MASK, 0);
    691	}
    692}
    693
    694void __init imx6sx_pm_init(void)
    695{
    696	imx6_pm_common_init(&imx6sx_pm_data);
    697}
    698
    699void __init imx6ul_pm_init(void)
    700{
    701	imx6_pm_common_init(&imx6ul_pm_data);
    702}