1*7eb7819aSQing Zhang // SPDX-License-Identifier: GPL-2.0 2*7eb7819aSQing Zhang /* 3*7eb7819aSQing Zhang * Copyright (C) 2021, Qing Zhang <zhangqing@loongson.cn> 4*7eb7819aSQing Zhang * Loongson-2K1000 reset support 5*7eb7819aSQing Zhang */ 6*7eb7819aSQing Zhang 7*7eb7819aSQing Zhang #include <linux/of_address.h> 8*7eb7819aSQing Zhang #include <linux/pm.h> 9*7eb7819aSQing Zhang #include <asm/reboot.h> 10*7eb7819aSQing Zhang 11*7eb7819aSQing Zhang #define PM1_STS 0x0c /* Power Management 1 Status Register */ 12*7eb7819aSQing Zhang #define PM1_CNT 0x14 /* Power Management 1 Control Register */ 13*7eb7819aSQing Zhang #define RST_CNT 0x30 /* Reset Control Register */ 14*7eb7819aSQing Zhang 15*7eb7819aSQing Zhang static void __iomem *base; 16*7eb7819aSQing Zhang 17*7eb7819aSQing Zhang static void ls2k_restart(char *command) 18*7eb7819aSQing Zhang { 19*7eb7819aSQing Zhang writel(0x1, base + RST_CNT); 20*7eb7819aSQing Zhang } 21*7eb7819aSQing Zhang 22*7eb7819aSQing Zhang static void ls2k_poweroff(void) 23*7eb7819aSQing Zhang { 24*7eb7819aSQing Zhang /* Clear */ 25*7eb7819aSQing Zhang writel((readl(base + PM1_STS) & 0xffffffff), base + PM1_STS); 26*7eb7819aSQing Zhang /* Sleep Enable | Soft Off*/ 27*7eb7819aSQing Zhang writel(GENMASK(12, 10) | BIT(13), base + PM1_CNT); 28*7eb7819aSQing Zhang } 29*7eb7819aSQing Zhang 30*7eb7819aSQing Zhang static int ls2k_reset_init(void) 31*7eb7819aSQing Zhang { 32*7eb7819aSQing Zhang struct device_node *np; 33*7eb7819aSQing Zhang 34*7eb7819aSQing Zhang np = of_find_compatible_node(NULL, NULL, "loongson,ls2k-pm"); 35*7eb7819aSQing Zhang if (!np) { 36*7eb7819aSQing Zhang pr_info("Failed to get PM node\n"); 37*7eb7819aSQing Zhang return -ENODEV; 38*7eb7819aSQing Zhang } 39*7eb7819aSQing Zhang 40*7eb7819aSQing Zhang base = of_iomap(np, 0); 41*7eb7819aSQing Zhang if (!base) { 42*7eb7819aSQing Zhang pr_info("Failed to map PM register base address\n"); 43*7eb7819aSQing Zhang return -ENOMEM; 44*7eb7819aSQing Zhang } 45*7eb7819aSQing Zhang 46*7eb7819aSQing Zhang _machine_restart = ls2k_restart; 47*7eb7819aSQing Zhang pm_power_off = ls2k_poweroff; 48*7eb7819aSQing Zhang 49*7eb7819aSQing Zhang of_node_put(np); 50*7eb7819aSQing Zhang return 0; 51*7eb7819aSQing Zhang } 52*7eb7819aSQing Zhang 53*7eb7819aSQing Zhang arch_initcall(ls2k_reset_init); 54