1 // SPDX-License-Identifier: GPL-2.0+
3 * Copyright 2010-2011 Calxeda, Inc.
4 * Copyright (c) 2014, NVIDIA CORPORATION. All rights reserved.
11 #include "pxe_utils.h"
14 const char *pxe_default_paths[] = {
16 #ifdef CONFIG_SYS_BOARD
17 "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC "-" CONFIG_SYS_BOARD,
19 "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC,
21 "default-" CONFIG_SYS_ARCH,
26 static int do_get_tftp(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
28 char *tftp_argv[] = {"tftp", NULL, NULL, NULL};
30 tftp_argv[1] = file_addr;
31 tftp_argv[2] = (void *)file_path;
33 if (do_tftpb(cmdtp, 0, 3, tftp_argv))
40 static char *fs_argv[5];
42 static int do_get_ext2(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
44 #ifdef CONFIG_CMD_EXT2
45 fs_argv[0] = "ext2load";
46 fs_argv[3] = file_addr;
47 fs_argv[4] = (void *)file_path;
49 if (!do_ext2load(cmdtp, 0, 5, fs_argv))
55 static int do_get_fat(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
58 fs_argv[0] = "fatload";
59 fs_argv[3] = file_addr;
60 fs_argv[4] = (void *)file_path;
62 if (!do_fat_fsload(cmdtp, 0, 5, fs_argv))
68 static int do_get_any(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
70 #ifdef CONFIG_CMD_FS_GENERIC
72 fs_argv[3] = file_addr;
73 fs_argv[4] = (void *)file_path;
75 if (!do_load(cmdtp, 0, 5, fs_argv, FS_TYPE_ANY))
84 * Looks for a pxe file with a name based on the pxeuuid environment variable.
86 * Returns 1 on success or < 0 on error.
88 static int pxe_uuid_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
92 uuid_str = from_env("pxeuuid");
97 return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r);
101 * Looks for a pxe file with a name based on the 'ethaddr' environment
104 * Returns 1 on success or < 0 on error.
106 static int pxe_mac_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
111 err = format_mac_pxe(mac_str, sizeof(mac_str));
116 return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r);
120 * Looks for pxe files with names based on our IP address. See pxelinux
121 * documentation for details on what these file names look like. We match
124 * Returns 1 on success or < 0 on error.
126 static int pxe_ipaddr_paths(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
131 sprintf(ip_addr, "%08X", ntohl(net_ip.s_addr));
133 for (mask_pos = 7; mask_pos >= 0; mask_pos--) {
134 err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r);
139 ip_addr[mask_pos] = '\0';
145 * Entry point for the 'pxe get' command.
146 * This Follows pxelinux's rules to download a config file from a tftp server.
147 * The file is stored at the location given by the pxefile_addr_r environment
148 * variable, which must be set.
150 * UUID comes from pxeuuid env variable, if defined
151 * MAC addr comes from ethaddr env variable, if defined
154 * see http://syslinux.zytor.com/wiki/index.php/PXELINUX
156 * Returns 0 on success or 1 on error.
159 do_pxe_get(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
161 char *pxefile_addr_str;
162 unsigned long pxefile_addr_r;
165 do_getfile = do_get_tftp;
168 return CMD_RET_USAGE;
170 pxefile_addr_str = from_env("pxefile_addr_r");
172 if (!pxefile_addr_str)
175 err = strict_strtoul(pxefile_addr_str, 16,
176 (unsigned long *)&pxefile_addr_r);
181 * Keep trying paths until we successfully get a file we're looking
184 if (pxe_uuid_path(cmdtp, pxefile_addr_r) > 0 ||
185 pxe_mac_path(cmdtp, pxefile_addr_r) > 0 ||
186 pxe_ipaddr_paths(cmdtp, pxefile_addr_r) > 0) {
187 printf("Config file found\n");
192 while (pxe_default_paths[i]) {
193 if (get_pxelinux_path(cmdtp, pxe_default_paths[i],
194 pxefile_addr_r) > 0) {
195 printf("Config file found\n");
201 printf("Config file not found\n");
207 * Boots a system using a pxe file
209 * Returns 0 on success, 1 on error.
212 do_pxe_boot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
214 unsigned long pxefile_addr_r;
215 struct pxe_menu *cfg;
216 char *pxefile_addr_str;
218 do_getfile = do_get_tftp;
221 pxefile_addr_str = from_env("pxefile_addr_r");
222 if (!pxefile_addr_str)
225 } else if (argc == 2) {
226 pxefile_addr_str = argv[1];
228 return CMD_RET_USAGE;
231 if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
232 printf("Invalid pxefile address: %s\n", pxefile_addr_str);
236 cfg = parse_pxefile(cmdtp, pxefile_addr_r);
239 printf("Error parsing config file\n");
243 handle_pxe_menu(cmdtp, cfg);
245 destroy_pxe_menu(cfg);
247 copy_filename(net_boot_file_name, "", sizeof(net_boot_file_name));
252 static cmd_tbl_t cmd_pxe_sub[] = {
253 U_BOOT_CMD_MKENT(get, 1, 1, do_pxe_get, "", ""),
254 U_BOOT_CMD_MKENT(boot, 2, 1, do_pxe_boot, "", "")
257 static int do_pxe(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
262 return CMD_RET_USAGE;
266 /* drop initial "pxe" arg */
270 cp = find_cmd_tbl(argv[0], cmd_pxe_sub, ARRAY_SIZE(cmd_pxe_sub));
273 return cp->cmd(cmdtp, flag, argc, argv);
275 return CMD_RET_USAGE;
280 "commands to get and boot from pxe files",
281 "get - try to retrieve a pxe file using tftp\npxe "
282 "boot [pxefile_addr_r] - boot from the pxe file at pxefile_addr_r\n"
287 * Boots a system using a local disk syslinux/extlinux file
289 * Returns 0 on success, 1 on error.
291 static int do_sysboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
293 unsigned long pxefile_addr_r;
294 struct pxe_menu *cfg;
295 char *pxefile_addr_str;
301 if (argc > 1 && strstr(argv[1], "-p")) {
308 return cmd_usage(cmdtp);
311 pxefile_addr_str = from_env("pxefile_addr_r");
312 if (!pxefile_addr_str)
315 pxefile_addr_str = argv[4];
319 filename = env_get("bootfile");
322 env_set("bootfile", filename);
325 if (strstr(argv[3], "ext2"))
326 do_getfile = do_get_ext2;
327 else if (strstr(argv[3], "fat"))
328 do_getfile = do_get_fat;
329 else if (strstr(argv[3], "any"))
330 do_getfile = do_get_any;
332 printf("Invalid filesystem: %s\n", argv[3]);
335 fs_argv[1] = argv[1];
336 fs_argv[2] = argv[2];
338 if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
339 printf("Invalid pxefile address: %s\n", pxefile_addr_str);
343 if (get_pxe_file(cmdtp, filename, pxefile_addr_r) < 0) {
344 printf("Error reading config file\n");
348 cfg = parse_pxefile(cmdtp, pxefile_addr_r);
351 printf("Error parsing config file\n");
358 handle_pxe_menu(cmdtp, cfg);
360 destroy_pxe_menu(cfg);
366 sysboot, 7, 1, do_sysboot,
367 "command to get and boot from syslinux files",
368 "[-p] <interface> <dev[:part]> <ext2|fat|any> [addr] [filename]\n"
369 " - load and parse syslinux menu file 'filename' from ext2, fat\n"
370 " or any filesystem on 'dev' on 'interface' to address 'addr'"