2 * Copyright (c) 2015 The FreeBSD Foundation
3 * Copyright (c) 2016 Ruslan Bukin <br@bsdpad.com>
6 * Portions of this software were developed by Andrew Turner under
7 * sponsorship from the FreeBSD Foundation.
9 * Portions of this software were developed by SRI International and the
10 * University of Cambridge Computer Laboratory under DARPA/AFRL contract
11 * FA8750-10-C-0237 ("CTSRD"), as part of the DARPA CRASH research programme.
13 * Portions of this software were developed by the University of Cambridge
14 * Computer Laboratory as part of the CTSRD Project, with support from the
15 * UK Higher Education Innovation Fund (HEIF).
17 * Redistribution and use in source and binary forms, with or without
18 * modification, are permitted provided that the following conditions
20 * 1. Redistributions of source code must retain the above copyright
21 * notice, this list of conditions and the following disclaimer.
22 * 2. Redistributions in binary form must reproduce the above copyright
23 * notice, this list of conditions and the following disclaimer in the
24 * documentation and/or other materials provided with the distribution.
26 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
27 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
29 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
30 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
31 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
32 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
33 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
35 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
39 #include "opt_kstack_pages.h"
40 #include "opt_platform.h"
42 #include <sys/cdefs.h>
43 __FBSDID("$FreeBSD$");
45 #include <sys/param.h>
46 #include <sys/systm.h>
49 #include <sys/kernel.h>
50 #include <sys/malloc.h>
51 #include <sys/module.h>
52 #include <sys/mutex.h>
54 #include <sys/sched.h>
59 #include <vm/vm_extern.h>
60 #include <vm/vm_kern.h>
62 #include <machine/intr.h>
63 #include <machine/smp.h>
65 #include <machine/vfp.h>
69 #include <dev/ofw/openfirm.h>
70 #include <dev/ofw/ofw_cpu.h>
73 boolean_t ofw_cpu_reg(phandle_t node, u_int, cell_t *);
75 extern struct pcpu __pcpu[];
77 uint32_t __riscv_boot_ap[MAXCPU];
86 static device_identify_t riscv64_cpu_identify;
87 static device_probe_t riscv64_cpu_probe;
88 static device_attach_t riscv64_cpu_attach;
90 static int ipi_handler(void *);
92 struct mtx ap_boot_mtx;
93 struct pcb stoppcbs[MAXCPU];
96 static uint32_t cpu_reg[MAXCPU][2];
98 static device_t cpu_list[MAXCPU];
100 void mpentry(unsigned long cpuid);
101 void init_secondary(uint64_t);
103 uint8_t secondary_stacks[MAXCPU - 1][PAGE_SIZE * KSTACK_PAGES] __aligned(16);
105 /* Set to 1 once we're ready to let the APs out of the pen. */
106 volatile int aps_ready = 0;
108 /* Temporary variables for init_secondary() */
109 void *dpcpu[MAXCPU - 1];
111 static device_method_t riscv64_cpu_methods[] = {
112 /* Device interface */
113 DEVMETHOD(device_identify, riscv64_cpu_identify),
114 DEVMETHOD(device_probe, riscv64_cpu_probe),
115 DEVMETHOD(device_attach, riscv64_cpu_attach),
120 static devclass_t riscv64_cpu_devclass;
121 static driver_t riscv64_cpu_driver = {
127 DRIVER_MODULE(riscv64_cpu, cpu, riscv64_cpu_driver, riscv64_cpu_devclass, 0, 0);
130 riscv64_cpu_identify(driver_t *driver, device_t parent)
133 if (device_find_child(parent, "riscv64_cpu", -1) != NULL)
135 if (BUS_ADD_CHILD(parent, 0, "riscv64_cpu", -1) == NULL)
136 device_printf(parent, "add child failed\n");
140 riscv64_cpu_probe(device_t dev)
144 cpuid = device_get_unit(dev);
145 if (cpuid >= MAXCPU || cpuid > mp_maxid)
153 riscv64_cpu_attach(device_t dev)
160 cpuid = device_get_unit(dev);
162 if (cpuid >= MAXCPU || cpuid > mp_maxid)
164 KASSERT(cpu_list[cpuid] == NULL, ("Already have cpu %u", cpuid));
166 reg = cpu_get_cpuid(dev, ®_size);
171 device_printf(dev, "register <");
172 for (i = 0; i < reg_size; i++)
173 printf("%s%x", (i == 0) ? "" : " ", reg[i]);
177 /* Set the device to start it later */
178 cpu_list[cpuid] = dev;
184 release_aps(void *dummy __unused)
191 /* Setup the IPI handler */
192 riscv_setup_ipihandler(ipi_handler);
194 atomic_store_rel_int(&aps_ready, 1);
196 printf("Release APs\n");
198 for (i = 0; i < 2000; i++) {
200 for (cpu = 0; cpu <= mp_maxid; cpu++) {
209 printf("APs not started\n");
211 SYSINIT(start_aps, SI_SUB_SMP, SI_ORDER_FIRST, release_aps, NULL);
214 init_secondary(uint64_t cpu)
218 /* Setup the pcpu pointer */
219 pcpup = &__pcpu[cpu];
220 __asm __volatile("mv gp, %0" :: "r"(pcpup));
222 /* Spin until the BSP releases the APs */
224 __asm __volatile("wfi");
226 /* Initialize curthread */
227 KASSERT(PCPU_GET(idlethread) != NULL, ("no idle thread"));
228 pcpup->pc_curthread = pcpup->pc_idlethread;
229 pcpup->pc_curpcb = pcpup->pc_idlethread->td_pcb;
232 * Identify current CPU. This is necessary to setup
233 * affinity registers and to provide support for
234 * runtime chip identification.
238 /* Enable software interrupts */
241 /* Start per-CPU event timers. */
248 /* Enable interrupts */
251 mtx_lock_spin(&ap_boot_mtx);
253 atomic_add_rel_32(&smp_cpus, 1);
255 if (smp_cpus == mp_ncpus) {
256 /* enable IPI's, tlb shootdown, freezes etc */
257 atomic_store_rel_int(&smp_started, 1);
260 mtx_unlock_spin(&ap_boot_mtx);
262 /* Enter the scheduler */
265 panic("scheduler returned us to init_secondary");
270 ipi_handler(void *arg)
277 * We have shared interrupt line for both IPI and HTIF,
278 * so we don't really need to clear pending bit here
279 * as it will be cleared later in htif_intr.
280 * But lets assume HTIF is optional part, so do clear
281 * pending bit if there is no new entires in htif_ring.
283 machine_command(ECALL_CLEAR_IPI, 0);
285 cpu = PCPU_GET(cpuid);
289 ipi_bitmap = atomic_readandclear_int(PCPU_PTR(pending_ipis));
291 return (FILTER_HANDLED);
293 while ((bit = ffs(ipi_bitmap))) {
302 CTR0(KTR_SMP, "IPI_AST");
305 CTR1(KTR_SMP, "%s: IPI_PREEMPT", __func__);
306 sched_preempt(curthread);
309 CTR0(KTR_SMP, "IPI_RENDEZVOUS");
310 smp_rendezvous_action();
314 CTR0(KTR_SMP, (ipi == IPI_STOP) ? "IPI_STOP" : "IPI_STOP_HARD");
315 savectx(&stoppcbs[cpu]);
317 /* Indicate we are stopped */
318 CPU_SET_ATOMIC(cpu, &stopped_cpus);
320 /* Wait for restart */
321 while (!CPU_ISSET(cpu, &started_cpus))
324 CPU_CLR_ATOMIC(cpu, &started_cpus);
325 CPU_CLR_ATOMIC(cpu, &stopped_cpus);
326 CTR0(KTR_SMP, "IPI_STOP (restart)");
329 CTR1(KTR_SMP, "%s: IPI_HARDCLOCK", __func__);
333 panic("Unknown IPI %#0x on cpu %d", ipi, curcpu);
337 return (FILTER_HANDLED);
344 return (smp_topo_none());
347 /* Determine if we running MP machine */
352 return (mp_ncpus > 1);
357 cpu_init_fdt(u_int id, phandle_t node, u_int addr_size, pcell_t *reg)
362 /* Check we are able to start this cpu */
366 KASSERT(id < MAXCPU, ("Too many CPUs"));
368 KASSERT(addr_size == 1 || addr_size == 2, ("Invalid register size"));
370 cpu_reg[id][0] = reg[0];
372 cpu_reg[id][1] = reg[1];
376 if (addr_size == 2) {
378 target_cpu |= reg[1];
383 /* We are already running on cpu 0 */
385 pcpup->pc_reg = target_cpu;
389 pcpu_init(pcpup, id, sizeof(struct pcpu));
390 pcpup->pc_reg = target_cpu;
392 dpcpu[id - 1] = (void *)kmem_malloc(kernel_arena, DPCPU_SIZE,
394 dpcpu_init(dpcpu[id - 1], id);
396 printf("Starting CPU %u (%lx)\n", id, target_cpu);
397 __riscv_boot_ap[id] = 1;
399 CPU_SET(id, &all_cpus);
405 /* Initialize and fire up non-boot processors */
410 mtx_init(&ap_boot_mtx, "ap boot", NULL, MTX_SPIN);
412 CPU_SET(0, &all_cpus);
414 switch(cpu_enum_method) {
417 ofw_cpu_early_foreach(cpu_init_fdt, true);
425 /* Introduce rest of cores to the world */
427 cpu_mp_announce(void)
432 cpu_mp_setmaxid(void)
437 cores = ofw_cpu_early_foreach(NULL, false);
439 cores = MIN(cores, MAXCPU);
441 printf("Found %d CPUs in the device tree\n", cores);
443 mp_maxid = cores - 1;
444 cpu_enum_method = CPUS_FDT;
450 printf("No CPU data, limiting to 1 core\n");