// SPDX-License-Identifier: GPL-2.0 /* * Copyright (C) 2021, Qing Zhang <zhangqing@loongson.cn> * Loongson-2K1000 reset support */ #include <linux/of_address.h> #include <linux/pm.h> #include <asm/reboot.h> #define PM1_STS 0x0c /* Power Management 1 Status Register */ #define PM1_CNT 0x14 /* Power Management 1 Control Register */ #define RST_CNT 0x30 /* Reset Control Register */ static void __iomem *base; static void ls2k_restart(char *command) { writel(0x1, base + RST_CNT); } static void ls2k_poweroff(void) { /* Clear */ writel((readl(base + PM1_STS) & 0xffffffff), base + PM1_STS); /* Sleep Enable | Soft Off*/ writel(GENMASK(12, 10) | BIT(13), base + PM1_CNT); } static int ls2k_reset_init(void) { struct device_node *np; np = of_find_compatible_node(NULL, NULL, "loongson,ls2k-pm"); if (!np) { pr_info("Failed to get PM node\n"); return -ENODEV; } base = of_iomap(np, 0); of_node_put(np); if (!base) { pr_info("Failed to map PM register base address\n"); return -ENOMEM; } _machine_restart = ls2k_restart; pm_power_off = ls2k_poweroff; return 0; } arch_initcall(ls2k_reset_init);