arch/mips/loongson64/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
  25 static 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
  34 static 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
  51 static int kexec_argc;
  52 static int kdump_argc;
  53 static void *kexec_argv;
  54 static void *kdump_argv;
  55 static void *kexec_envp;
  56
  57 static 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
 107 static 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
 126 static 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
 141 static 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
 179 arch_initcall(mips_reboot_setup);