1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (C) 2018, Bin Meng <bmeng.cn@gmail.com>
4 */
5
6 #include <common.h>
7 #include <cpu.h>
8 #include <dm.h>
9 #include <init.h>
10 #include <log.h>
11 #include <asm/encoding.h>
12 #include <dm/uclass-internal.h>
13 #include <linux/bitops.h>
14
15 /*
16 * The variables here must be stored in the data section since they are used
17 * before the bss section is available.
18 */
19 #ifdef CONFIG_OF_PRIOR_STAGE
20 phys_addr_t prior_stage_fdt_address __attribute__((section(".data")));
21 #endif
22 #ifndef CONFIG_XIP
23 u32 hart_lottery __attribute__((section(".data"))) = 0;
24
25 /*
26 * The main hart running U-Boot has acquired available_harts_lock until it has
27 * finished initialization of global data.
28 */
29 u32 available_harts_lock = 1;
30 #endif
31
supports_extension(char ext)32 static inline bool supports_extension(char ext)
33 {
34 #ifdef CONFIG_CPU
35 struct udevice *dev;
36 char desc[32];
37
38 uclass_find_first_device(UCLASS_CPU, &dev);
39 if (!dev) {
40 debug("unable to find the RISC-V cpu device\n");
41 return false;
42 }
43 if (!cpu_get_desc(dev, desc, sizeof(desc))) {
44 /* skip the first 4 characters (rv32|rv64) */
45 if (strchr(desc + 4, ext))
46 return true;
47 }
48
49 return false;
50 #else /* !CONFIG_CPU */
51 #if CONFIG_IS_ENABLED(RISCV_MMODE)
52 return csr_read(CSR_MISA) & (1 << (ext - 'a'));
53 #else /* !CONFIG_IS_ENABLED(RISCV_MMODE) */
54 #warning "There is no way to determine the available extensions in S-mode."
55 #warning "Please convert your board to use the RISC-V CPU driver."
56 return false;
57 #endif /* CONFIG_IS_ENABLED(RISCV_MMODE) */
58 #endif /* CONFIG_CPU */
59 }
60
riscv_cpu_probe(void)61 static int riscv_cpu_probe(void)
62 {
63 #ifdef CONFIG_CPU
64 int ret;
65
66 /* probe cpus so that RISC-V timer can be bound */
67 ret = cpu_probe_all();
68 if (ret)
69 return log_msg_ret("RISC-V cpus probe failed\n", ret);
70 #endif
71
72 return 0;
73 }
74
arch_cpu_init_dm(void)75 int arch_cpu_init_dm(void)
76 {
77 int ret;
78
79 ret = riscv_cpu_probe();
80 if (ret)
81 return ret;
82
83 /* Enable FPU */
84 if (supports_extension('d') || supports_extension('f')) {
85 csr_set(MODE_PREFIX(status), MSTATUS_FS);
86 csr_write(CSR_FCSR, 0);
87 }
88
89 if (CONFIG_IS_ENABLED(RISCV_MMODE)) {
90 /*
91 * Enable perf counters for cycle, time,
92 * and instret counters only
93 */
94 csr_write(CSR_MCOUNTEREN, GENMASK(2, 0));
95
96 /* Disable paging */
97 if (supports_extension('s'))
98 csr_write(CSR_SATP, 0);
99 }
100
101 return 0;
102 }
103
arch_early_init_r(void)104 int arch_early_init_r(void)
105 {
106 return riscv_cpu_probe();
107 }
108