reset.c

  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 *
  4 * Copyright (C) 2007 Lemote, Inc. & Institute of Computing Technology
  5 * Author: Fuxin Zhang, zhangfx@lemote.com
  6 * Copyright (C) 2009 Lemote, Inc.
  7 * Author: Zhangjin Wu, wuzhangjin@gmail.com
  8 */
  9#include <linux/cpu.h>
 10#include <linux/delay.h>
 11#include <linux/init.h>
 12#include <linux/kexec.h>
 13#include <linux/pm.h>
 14#include <linux/reboot.h>
 15#include <linux/slab.h>
 16
 17#include <asm/bootinfo.h>
 18#include <asm/idle.h>
 19#include <asm/reboot.h>
 20#include <asm/bug.h>
 21
 22#include <loongson.h>
 23#include <boot_param.h>
 24
 25static int firmware_restart(struct sys_off_data *unusedd)
 26{
 27
 28	void (*fw_restart)(void) = (void *)loongson_sysconf.restart_addr;
 29
 30	fw_restart();
 31	return NOTIFY_DONE;
 32}
 33
 34static int firmware_poweroff(struct sys_off_data *unused)
 35{
 36	void (*fw_poweroff)(void) = (void *)loongson_sysconf.poweroff_addr;
 37
 38	fw_poweroff();
 39	return NOTIFY_DONE;
 40}
 41
 42#ifdef CONFIG_KEXEC_CORE
 43
 44/* 0X80000000~0X80200000 is safe */
 45#define MAX_ARGS	64
 46#define KEXEC_CTRL_CODE	0xFFFFFFFF80100000UL
 47#define KEXEC_ARGV_ADDR	0xFFFFFFFF80108000UL
 48#define KEXEC_ARGV_SIZE	COMMAND_LINE_SIZE
 49#define KEXEC_ENVP_SIZE	4800
 50
 51static int kexec_argc;
 52static int kdump_argc;
 53static void *kexec_argv;
 54static void *kdump_argv;
 55static void *kexec_envp;
 56
 57static int loongson_kexec_prepare(struct kimage *image)
 58{
 59	int i, argc = 0;
 60	unsigned int *argv;
 61	char *str, *ptr, *bootloader = "kexec";
 62
 63	/* argv at offset 0, argv[] at offset KEXEC_ARGV_SIZE/2 */
 64	if (image->type == KEXEC_TYPE_DEFAULT)
 65		argv = (unsigned int *)kexec_argv;
 66	else
 67		argv = (unsigned int *)kdump_argv;
 68
 69	argv[argc++] = (unsigned int)(KEXEC_ARGV_ADDR + KEXEC_ARGV_SIZE/2);
 70
 71	for (i = 0; i < image->nr_segments; i++) {
 72		if (!strncmp(bootloader, (char *)image->segment[i].buf,
 73				strlen(bootloader))) {
 74			/*
 75			 * convert command line string to array
 76			 * of parameters (as bootloader does).
 77			 */
 78			int offt;
 79			str = (char *)argv + KEXEC_ARGV_SIZE/2;
 80			memcpy(str, image->segment[i].buf, KEXEC_ARGV_SIZE/2);
 81			ptr = strchr(str, ' ');
 82
 83			while (ptr && (argc < MAX_ARGS)) {
 84				*ptr = '\0';
 85				if (ptr[1] != ' ') {
 86					offt = (int)(ptr - str + 1);
 87					argv[argc] = KEXEC_ARGV_ADDR + KEXEC_ARGV_SIZE/2 + offt;
 88					argc++;
 89				}
 90				ptr = strchr(ptr + 1, ' ');
 91			}
 92			break;
 93		}
 94	}
 95
 96	if (image->type == KEXEC_TYPE_DEFAULT)
 97		kexec_argc = argc;
 98	else
 99		kdump_argc = argc;
100
101	/* kexec/kdump need a safe page to save reboot_code_buffer */
102	image->control_code_page = virt_to_page((void *)KEXEC_CTRL_CODE);
103
104	return 0;
105}
106
107static void loongson_kexec_shutdown(void)
108{
109#ifdef CONFIG_SMP
110	int cpu;
111
112	/* All CPUs go to reboot_code_buffer */
113	for_each_possible_cpu(cpu)
114		if (!cpu_online(cpu))
115			cpu_device_up(get_cpu_device(cpu));
116
117	secondary_kexec_args[0] = TO_UNCAC(0x3ff01000);
118#endif
119	kexec_args[0] = kexec_argc;
120	kexec_args[1] = fw_arg1;
121	kexec_args[2] = fw_arg2;
122	memcpy((void *)fw_arg1, kexec_argv, KEXEC_ARGV_SIZE);
123	memcpy((void *)fw_arg2, kexec_envp, KEXEC_ENVP_SIZE);
124}
125
126static void loongson_crash_shutdown(struct pt_regs *regs)
127{
128	default_machine_crash_shutdown(regs);
129	kexec_args[0] = kdump_argc;
130	kexec_args[1] = fw_arg1;
131	kexec_args[2] = fw_arg2;
132#ifdef CONFIG_SMP
133	secondary_kexec_args[0] = TO_UNCAC(0x3ff01000);
134#endif
135	memcpy((void *)fw_arg1, kdump_argv, KEXEC_ARGV_SIZE);
136	memcpy((void *)fw_arg2, kexec_envp, KEXEC_ENVP_SIZE);
137}
138
139#endif
140
141static int __init mips_reboot_setup(void)
142{
143	if (loongson_sysconf.restart_addr) {
144		register_sys_off_handler(SYS_OFF_MODE_RESTART,
145				 SYS_OFF_PRIO_FIRMWARE,
146				 firmware_restart, NULL);
147	}
148
149	if (loongson_sysconf.poweroff_addr) {
150		register_sys_off_handler(SYS_OFF_MODE_POWER_OFF,
151				 SYS_OFF_PRIO_FIRMWARE,
152				 firmware_poweroff, NULL);
153	}
154
155#ifdef CONFIG_KEXEC_CORE
156	kexec_argv = kmalloc(KEXEC_ARGV_SIZE, GFP_KERNEL);
157	if (WARN_ON(!kexec_argv))
158		return -ENOMEM;
159
160	kdump_argv = kmalloc(KEXEC_ARGV_SIZE, GFP_KERNEL);
161	if (WARN_ON(!kdump_argv))
162		return -ENOMEM;
163
164	kexec_envp = kmalloc(KEXEC_ENVP_SIZE, GFP_KERNEL);
165	if (WARN_ON(!kexec_envp))
166		return -ENOMEM;
167
168	fw_arg1 = KEXEC_ARGV_ADDR;
169	memcpy(kexec_envp, (void *)fw_arg2, KEXEC_ENVP_SIZE);
170
171	_machine_kexec_prepare = loongson_kexec_prepare;
172	_machine_kexec_shutdown = loongson_kexec_shutdown;
173	_machine_crash_shutdown = loongson_crash_shutdown;
174#endif
175
176	return 0;
177}
178
179arch_initcall(mips_reboot_setup);