Merge tag 'u-boot-stm32-20211012' of https://source.denx.de/u-boot/custodians/u-boot-stm
[platform/kernel/u-boot.git] / cmd / qfw.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2015 Miao Yan <yanmiaobest@gmail.com>
4  */
5
6 #include <common.h>
7 #include <command.h>
8 #include <env.h>
9 #include <errno.h>
10 #include <qfw.h>
11 #include <dm.h>
12
13 static struct udevice *qfw_dev;
14
15 /*
16  * This function prepares kernel for zboot. It loads kernel data
17  * to 'load_addr', initrd to 'initrd_addr' and kernel command
18  * line using qemu fw_cfg interface.
19  */
20 static int qemu_fwcfg_cmd_setup_kernel(void *load_addr, void *initrd_addr)
21 {
22         char *data_addr;
23         uint32_t setup_size, kernel_size, cmdline_size, initrd_size;
24
25         qfw_read_entry(qfw_dev, FW_CFG_SETUP_SIZE, 4, &setup_size);
26         qfw_read_entry(qfw_dev, FW_CFG_KERNEL_SIZE, 4, &kernel_size);
27
28         if (setup_size == 0 || kernel_size == 0) {
29                 printf("warning: no kernel available\n");
30                 return -1;
31         }
32
33         data_addr = load_addr;
34         qfw_read_entry(qfw_dev, FW_CFG_SETUP_DATA,
35                        le32_to_cpu(setup_size), data_addr);
36         data_addr += le32_to_cpu(setup_size);
37
38         qfw_read_entry(qfw_dev, FW_CFG_KERNEL_DATA,
39                        le32_to_cpu(kernel_size), data_addr);
40         data_addr += le32_to_cpu(kernel_size);
41
42         data_addr = initrd_addr;
43         qfw_read_entry(qfw_dev, FW_CFG_INITRD_SIZE, 4, &initrd_size);
44         if (initrd_size == 0) {
45                 printf("warning: no initrd available\n");
46         } else {
47                 qfw_read_entry(qfw_dev, FW_CFG_INITRD_DATA,
48                                le32_to_cpu(initrd_size), data_addr);
49                 data_addr += le32_to_cpu(initrd_size);
50         }
51
52         qfw_read_entry(qfw_dev, FW_CFG_CMDLINE_SIZE, 4, &cmdline_size);
53         if (cmdline_size) {
54                 qfw_read_entry(qfw_dev, FW_CFG_CMDLINE_DATA,
55                                le32_to_cpu(cmdline_size), data_addr);
56                 /*
57                  * if kernel cmdline only contains '\0', (e.g. no -append
58                  * when invoking qemu), do not update bootargs
59                  */
60                 if (*data_addr != '\0') {
61                         if (env_set("bootargs", data_addr) < 0)
62                                 printf("warning: unable to change bootargs\n");
63                 }
64         }
65
66         printf("loading kernel to address %p size %x", load_addr,
67                le32_to_cpu(kernel_size));
68         if (initrd_size)
69                 printf(" initrd %p size %x\n",
70                        initrd_addr,
71                        le32_to_cpu(initrd_size));
72         else
73                 printf("\n");
74
75         return 0;
76 }
77
78 static int qemu_fwcfg_cmd_list_firmware(void)
79 {
80         int ret;
81         struct fw_cfg_file_iter iter;
82         struct fw_file *file;
83
84         /* make sure fw_list is loaded */
85         ret = qfw_read_firmware_list(qfw_dev);
86         if (ret)
87                 return ret;
88
89         for (file = qfw_file_iter_init(qfw_dev, &iter);
90              !qfw_file_iter_end(&iter);
91              file = qfw_file_iter_next(&iter)) {
92                 printf("%-56s\n", file->cfg.name);
93         }
94
95         return 0;
96 }
97
98 static int qemu_fwcfg_do_list(struct cmd_tbl *cmdtp, int flag,
99                               int argc, char *const argv[])
100 {
101         if (qemu_fwcfg_cmd_list_firmware() < 0)
102                 return CMD_RET_FAILURE;
103
104         return 0;
105 }
106
107 static int qemu_fwcfg_do_cpus(struct cmd_tbl *cmdtp, int flag,
108                               int argc, char *const argv[])
109 {
110         printf("%d cpu(s) online\n", qfw_online_cpus(qfw_dev));
111         return 0;
112 }
113
114 static int qemu_fwcfg_do_load(struct cmd_tbl *cmdtp, int flag,
115                               int argc, char *const argv[])
116 {
117         char *env;
118         void *load_addr;
119         void *initrd_addr;
120
121         env = env_get("loadaddr");
122         load_addr = env ?
123                 (void *)hextoul(env, NULL) :
124                 (void *)CONFIG_SYS_LOAD_ADDR;
125
126         env = env_get("ramdiskaddr");
127         initrd_addr = env ?
128                 (void *)hextoul(env, NULL) :
129 #ifdef CONFIG_RAMDISK_ADDR
130                 (void *)CONFIG_RAMDISK_ADDR;
131 #else
132                 NULL;
133 #endif
134
135         if (argc == 2) {
136                 load_addr = (void *)hextoul(argv[0], NULL);
137                 initrd_addr = (void *)hextoul(argv[1], NULL);
138         } else if (argc == 1) {
139                 load_addr = (void *)hextoul(argv[0], NULL);
140         }
141
142         if (!load_addr || !initrd_addr) {
143                 printf("missing load or initrd address\n");
144                 return CMD_RET_FAILURE;
145         }
146
147         return qemu_fwcfg_cmd_setup_kernel(load_addr, initrd_addr);
148 }
149
150 static struct cmd_tbl fwcfg_commands[] = {
151         U_BOOT_CMD_MKENT(list, 0, 1, qemu_fwcfg_do_list, "", ""),
152         U_BOOT_CMD_MKENT(cpus, 0, 1, qemu_fwcfg_do_cpus, "", ""),
153         U_BOOT_CMD_MKENT(load, 2, 1, qemu_fwcfg_do_load, "", ""),
154 };
155
156 static int do_qemu_fw(struct cmd_tbl *cmdtp, int flag, int argc,
157                       char *const argv[])
158 {
159         int ret;
160         struct cmd_tbl *fwcfg_cmd;
161
162         ret = qfw_get_dev(&qfw_dev);
163         if (ret) {
164                 printf("QEMU fw_cfg interface not found\n");
165                 return CMD_RET_USAGE;
166         }
167
168         fwcfg_cmd = find_cmd_tbl(argv[1], fwcfg_commands,
169                                  ARRAY_SIZE(fwcfg_commands));
170         argc -= 2;
171         argv += 2;
172         if (!fwcfg_cmd || argc > fwcfg_cmd->maxargs)
173                 return CMD_RET_USAGE;
174
175         ret = fwcfg_cmd->cmd(fwcfg_cmd, flag, argc, argv);
176
177         return cmd_process_error(fwcfg_cmd, ret);
178 }
179
180 U_BOOT_CMD(
181         qfw,    4,      1,      do_qemu_fw,
182         "QEMU firmware interface",
183         "<command>\n"
184         "    - list                             : print firmware(s) currently loaded\n"
185         "    - cpus                             : print online cpu number\n"
186         "    - load <kernel addr> <initrd addr> : load kernel and initrd (if any), and setup for zboot\n"
187 )