18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
28c2ecf20Sopenharmony_ci/*
38c2ecf20Sopenharmony_ci * Copyright (c) 2011 Zhang, Keguang <keguang.zhang@gmail.com>
48c2ecf20Sopenharmony_ci */
58c2ecf20Sopenharmony_ci
68c2ecf20Sopenharmony_ci#include <linux/io.h>
78c2ecf20Sopenharmony_ci#include <linux/pm.h>
88c2ecf20Sopenharmony_ci#include <linux/sizes.h>
98c2ecf20Sopenharmony_ci#include <asm/idle.h>
108c2ecf20Sopenharmony_ci#include <asm/reboot.h>
118c2ecf20Sopenharmony_ci
128c2ecf20Sopenharmony_ci#include <loongson1.h>
138c2ecf20Sopenharmony_ci
148c2ecf20Sopenharmony_cistatic void __iomem *wdt_reg_base;
158c2ecf20Sopenharmony_ci
168c2ecf20Sopenharmony_cistatic void ls1x_halt(void)
178c2ecf20Sopenharmony_ci{
188c2ecf20Sopenharmony_ci	while (1) {
198c2ecf20Sopenharmony_ci		if (cpu_wait)
208c2ecf20Sopenharmony_ci			cpu_wait();
218c2ecf20Sopenharmony_ci	}
228c2ecf20Sopenharmony_ci}
238c2ecf20Sopenharmony_ci
248c2ecf20Sopenharmony_cistatic void ls1x_restart(char *command)
258c2ecf20Sopenharmony_ci{
268c2ecf20Sopenharmony_ci	__raw_writel(0x1, wdt_reg_base + WDT_EN);
278c2ecf20Sopenharmony_ci	__raw_writel(0x1, wdt_reg_base + WDT_TIMER);
288c2ecf20Sopenharmony_ci	__raw_writel(0x1, wdt_reg_base + WDT_SET);
298c2ecf20Sopenharmony_ci
308c2ecf20Sopenharmony_ci	ls1x_halt();
318c2ecf20Sopenharmony_ci}
328c2ecf20Sopenharmony_ci
338c2ecf20Sopenharmony_cistatic void ls1x_power_off(void)
348c2ecf20Sopenharmony_ci{
358c2ecf20Sopenharmony_ci	ls1x_halt();
368c2ecf20Sopenharmony_ci}
378c2ecf20Sopenharmony_ci
388c2ecf20Sopenharmony_cistatic int __init ls1x_reboot_setup(void)
398c2ecf20Sopenharmony_ci{
408c2ecf20Sopenharmony_ci	wdt_reg_base = ioremap(LS1X_WDT_BASE, (SZ_4 + SZ_8));
418c2ecf20Sopenharmony_ci	if (!wdt_reg_base)
428c2ecf20Sopenharmony_ci		panic("Failed to remap watchdog registers");
438c2ecf20Sopenharmony_ci
448c2ecf20Sopenharmony_ci	_machine_restart = ls1x_restart;
458c2ecf20Sopenharmony_ci	_machine_halt = ls1x_halt;
468c2ecf20Sopenharmony_ci	pm_power_off = ls1x_power_off;
478c2ecf20Sopenharmony_ci
488c2ecf20Sopenharmony_ci	return 0;
498c2ecf20Sopenharmony_ci}
508c2ecf20Sopenharmony_ci
518c2ecf20Sopenharmony_ciarch_initcall(ls1x_reboot_setup);
52