123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240 |
- // SPDX-License-Identifier: GPL-2.0+
- /*
- * Copyright 2010-2011 Calxeda, Inc.
- * Copyright (c) 2014, NVIDIA CORPORATION. All rights reserved.
- */
- #include <common.h>
- #include <command.h>
- #include <fs.h>
- #include <net.h>
- #include "pxe_utils.h"
- #ifdef CONFIG_CMD_NET
- const char *pxe_default_paths[] = {
- #ifdef CONFIG_SYS_SOC
- #ifdef CONFIG_SYS_BOARD
- "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC "-" CONFIG_SYS_BOARD,
- #endif
- "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC,
- #endif
- "default-" CONFIG_SYS_ARCH,
- "default",
- NULL
- };
- static int do_get_tftp(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
- {
- char *tftp_argv[] = {"tftp", NULL, NULL, NULL};
- tftp_argv[1] = file_addr;
- tftp_argv[2] = (void *)file_path;
- if (do_tftpb(cmdtp, 0, 3, tftp_argv))
- return -ENOENT;
- return 1;
- }
- /*
- * Looks for a pxe file with a name based on the pxeuuid environment variable.
- *
- * Returns 1 on success or < 0 on error.
- */
- static int pxe_uuid_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
- {
- char *uuid_str;
- uuid_str = from_env("pxeuuid");
- if (!uuid_str)
- return -ENOENT;
- return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r);
- }
- /*
- * Looks for a pxe file with a name based on the 'ethaddr' environment
- * variable.
- *
- * Returns 1 on success or < 0 on error.
- */
- static int pxe_mac_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
- {
- char mac_str[21];
- int err;
- err = format_mac_pxe(mac_str, sizeof(mac_str));
- if (err < 0)
- return err;
- return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r);
- }
- /*
- * Looks for pxe files with names based on our IP address. See pxelinux
- * documentation for details on what these file names look like. We match
- * that exactly.
- *
- * Returns 1 on success or < 0 on error.
- */
- static int pxe_ipaddr_paths(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
- {
- char ip_addr[9];
- int mask_pos, err;
- sprintf(ip_addr, "%08X", ntohl(net_ip.s_addr));
- for (mask_pos = 7; mask_pos >= 0; mask_pos--) {
- err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r);
- if (err > 0)
- return err;
- ip_addr[mask_pos] = '\0';
- }
- return -ENOENT;
- }
- /*
- * Entry point for the 'pxe get' command.
- * This Follows pxelinux's rules to download a config file from a tftp server.
- * The file is stored at the location given by the pxefile_addr_r environment
- * variable, which must be set.
- *
- * UUID comes from pxeuuid env variable, if defined
- * MAC addr comes from ethaddr env variable, if defined
- * IP
- *
- * see http://syslinux.zytor.com/wiki/index.php/PXELINUX
- *
- * Returns 0 on success or 1 on error.
- */
- static int
- do_pxe_get(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
- {
- char *pxefile_addr_str;
- unsigned long pxefile_addr_r;
- int err, i = 0;
- do_getfile = do_get_tftp;
- if (argc != 1)
- return CMD_RET_USAGE;
- pxefile_addr_str = from_env("pxefile_addr_r");
- if (!pxefile_addr_str)
- return 1;
- err = strict_strtoul(pxefile_addr_str, 16,
- (unsigned long *)&pxefile_addr_r);
- if (err < 0)
- return 1;
- /*
- * Keep trying paths until we successfully get a file we're looking
- * for.
- */
- if (pxe_uuid_path(cmdtp, pxefile_addr_r) > 0 ||
- pxe_mac_path(cmdtp, pxefile_addr_r) > 0 ||
- pxe_ipaddr_paths(cmdtp, pxefile_addr_r) > 0) {
- printf("Config file found\n");
- return 0;
- }
- while (pxe_default_paths[i]) {
- if (get_pxelinux_path(cmdtp, pxe_default_paths[i],
- pxefile_addr_r) > 0) {
- printf("Config file found\n");
- return 0;
- }
- i++;
- }
- printf("Config file not found\n");
- return 1;
- }
- /*
- * Boots a system using a pxe file
- *
- * Returns 0 on success, 1 on error.
- */
- static int
- do_pxe_boot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
- {
- unsigned long pxefile_addr_r;
- struct pxe_menu *cfg;
- char *pxefile_addr_str;
- do_getfile = do_get_tftp;
- if (argc == 1) {
- pxefile_addr_str = from_env("pxefile_addr_r");
- if (!pxefile_addr_str)
- return 1;
- } else if (argc == 2) {
- pxefile_addr_str = argv[1];
- } else {
- return CMD_RET_USAGE;
- }
- if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
- printf("Invalid pxefile address: %s\n", pxefile_addr_str);
- return 1;
- }
- cfg = parse_pxefile(cmdtp, pxefile_addr_r);
- if (!cfg) {
- printf("Error parsing config file\n");
- return 1;
- }
- handle_pxe_menu(cmdtp, cfg);
- destroy_pxe_menu(cfg);
- copy_filename(net_boot_file_name, "", sizeof(net_boot_file_name));
- return 0;
- }
- static cmd_tbl_t cmd_pxe_sub[] = {
- U_BOOT_CMD_MKENT(get, 1, 1, do_pxe_get, "", ""),
- U_BOOT_CMD_MKENT(boot, 2, 1, do_pxe_boot, "", "")
- };
- static int do_pxe(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
- {
- cmd_tbl_t *cp;
- if (argc < 2)
- return CMD_RET_USAGE;
- is_pxe = true;
- /* drop initial "pxe" arg */
- argc--;
- argv++;
- cp = find_cmd_tbl(argv[0], cmd_pxe_sub, ARRAY_SIZE(cmd_pxe_sub));
- if (cp)
- return cp->cmd(cmdtp, flag, argc, argv);
- return CMD_RET_USAGE;
- }
- U_BOOT_CMD(pxe, 3, 1, do_pxe,
- "commands to get and boot from pxe files",
- "get - try to retrieve a pxe file using tftp\n"
- "pxe boot [pxefile_addr_r] - boot from the pxe file at pxefile_addr_r\n"
- );
- #endif
|