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