1 /* SPDX-License-Identifier: GPL-2.0-only */
2
3 #include <program_loading.h>
4 #include <vm.h>
5 #include <arch/boot.h>
6 #include <arch/encoding.h>
7 #include <arch/smp/smp.h>
8 #include <mcall.h>
9 #include <cbfs.h>
10 #include <console/console.h>
11
12 struct arch_prog_run_args {
13 struct prog *prog;
14 struct prog *opensbi;
15 };
16
17 /*
18 * A pointer to the Flattened Device Tree passed to coreboot by the boot ROM.
19 * Presumably this FDT is also in ROM.
20 *
21 * This pointer is only used in ramstage!
22 */
23
do_arch_prog_run(struct arch_prog_run_args * args)24 static void do_arch_prog_run(struct arch_prog_run_args *args)
25 {
26 int hart_id = HLS()->hart_id;
27 struct prog *prog = args->prog;
28 void *fdt = HLS()->fdt;
29
30 if (prog_cbfs_type(prog) == CBFS_TYPE_FIT_PAYLOAD)
31 fdt = prog_entry_arg(prog);
32
33 if (ENV_RAMSTAGE && prog_type(prog) == PROG_PAYLOAD) {
34 if (CONFIG(RISCV_OPENSBI)) {
35 // tell OpenSBI to switch to Supervisor mode before jumping to payload
36 run_payload_opensbi(prog, fdt, args->opensbi, RISCV_PAYLOAD_MODE_S);
37 } else {
38 run_payload(prog, fdt, RISCV_PAYLOAD_MODE_S);
39 }
40 } else {
41 void (*doit)(int hart_id, void *fdt, void *arg) = prog_entry(prog);
42 doit(hart_id, fdt, prog_entry_arg(prog));
43 }
44
45 die("Failed to run stage");
46 }
47
arch_prog_run(struct prog * prog)48 void arch_prog_run(struct prog *prog)
49 {
50 struct arch_prog_run_args args = {};
51
52 args.prog = prog;
53
54 /* In case of OpenSBI we have to load it before resuming all HARTs */
55 if (ENV_RAMSTAGE && CONFIG(RISCV_OPENSBI)) {
56 struct prog sbi = PROG_INIT(PROG_OPENSBI, CONFIG_CBFS_PREFIX"/opensbi");
57
58 if (!selfload_check(&sbi, BM_MEM_OPENSBI))
59 die("OpenSBI load failed");
60
61 args.opensbi = &sbi;
62 }
63
64 smp_resume((void (*)(void *))do_arch_prog_run, &args);
65 }
66