Tom Rini | 83d290c | 2018-05-06 17:58:06 -0400 | [diff] [blame] | 1 | // SPDX-License-Identifier: GPL-2.0+ |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 2 | /* |
| 3 | * Copyright 2010-2011 Calxeda, Inc. |
Bryan Wu | 1fb7d0e | 2014-07-31 17:39:59 -0700 | [diff] [blame] | 4 | * Copyright (c) 2014, NVIDIA CORPORATION. All rights reserved. |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 5 | */ |
Wolfgang Denk | 1a45966 | 2013-07-08 09:37:19 +0200 | [diff] [blame] | 6 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 7 | #include <common.h> |
| 8 | #include <command.h> |
Simon Glass | 1444998 | 2019-12-28 10:44:44 -0700 | [diff] [blame] | 9 | #include <fs.h> |
Simon Glass | 77f4e47 | 2019-12-28 10:44:43 -0700 | [diff] [blame] | 10 | #include <net.h> |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 11 | |
Patrice Chotard | 2373cba | 2019-11-25 09:07:37 +0100 | [diff] [blame] | 12 | #include "pxe_utils.h" |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 13 | |
Patrice Chotard | 2373cba | 2019-11-25 09:07:37 +0100 | [diff] [blame] | 14 | #ifdef CONFIG_CMD_NET |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 15 | const char *pxe_default_paths[] = { |
Joe Hershberger | 58d9ff9 | 2013-06-24 17:21:04 -0500 | [diff] [blame] | 16 | #ifdef CONFIG_SYS_SOC |
Marek BehĂșn | 2455efa | 2019-05-02 15:29:12 +0200 | [diff] [blame] | 17 | #ifdef CONFIG_SYS_BOARD |
| 18 | "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC "-" CONFIG_SYS_BOARD, |
| 19 | #endif |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 20 | "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC, |
Joe Hershberger | 58d9ff9 | 2013-06-24 17:21:04 -0500 | [diff] [blame] | 21 | #endif |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 22 | "default-" CONFIG_SYS_ARCH, |
| 23 | "default", |
| 24 | NULL |
| 25 | }; |
| 26 | |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 27 | static int do_get_tftp(struct cmd_tbl *cmdtp, const char *file_path, |
| 28 | char *file_addr) |
Rob Herring | 669df7e | 2012-05-25 10:47:39 +0000 | [diff] [blame] | 29 | { |
| 30 | char *tftp_argv[] = {"tftp", NULL, NULL, NULL}; |
| 31 | |
| 32 | tftp_argv[1] = file_addr; |
Rob Herring | 23b7194 | 2012-12-02 21:00:21 -0600 | [diff] [blame] | 33 | tftp_argv[2] = (void *)file_path; |
Rob Herring | 669df7e | 2012-05-25 10:47:39 +0000 | [diff] [blame] | 34 | |
Steven Falco | 0e3f3f8 | 2013-10-07 09:51:48 -0400 | [diff] [blame] | 35 | if (do_tftpb(cmdtp, 0, 3, tftp_argv)) |
Rob Herring | 669df7e | 2012-05-25 10:47:39 +0000 | [diff] [blame] | 36 | return -ENOENT; |
| 37 | |
| 38 | return 1; |
| 39 | } |
Stephen Warren | b81fdb0 | 2014-02-05 20:49:20 -0700 | [diff] [blame] | 40 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 41 | /* |
| 42 | * Looks for a pxe file with a name based on the pxeuuid environment variable. |
| 43 | * |
| 44 | * Returns 1 on success or < 0 on error. |
| 45 | */ |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 46 | static int pxe_uuid_path(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 47 | { |
| 48 | char *uuid_str; |
| 49 | |
| 50 | uuid_str = from_env("pxeuuid"); |
| 51 | |
| 52 | if (!uuid_str) |
| 53 | return -ENOENT; |
| 54 | |
Steven Falco | 0e3f3f8 | 2013-10-07 09:51:48 -0400 | [diff] [blame] | 55 | return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 56 | } |
| 57 | |
| 58 | /* |
| 59 | * Looks for a pxe file with a name based on the 'ethaddr' environment |
| 60 | * variable. |
| 61 | * |
| 62 | * Returns 1 on success or < 0 on error. |
| 63 | */ |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 64 | static int pxe_mac_path(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 65 | { |
| 66 | char mac_str[21]; |
| 67 | int err; |
| 68 | |
| 69 | err = format_mac_pxe(mac_str, sizeof(mac_str)); |
| 70 | |
| 71 | if (err < 0) |
| 72 | return err; |
| 73 | |
Steven Falco | 0e3f3f8 | 2013-10-07 09:51:48 -0400 | [diff] [blame] | 74 | return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 75 | } |
| 76 | |
| 77 | /* |
| 78 | * Looks for pxe files with names based on our IP address. See pxelinux |
| 79 | * documentation for details on what these file names look like. We match |
| 80 | * that exactly. |
| 81 | * |
| 82 | * Returns 1 on success or < 0 on error. |
| 83 | */ |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 84 | static int pxe_ipaddr_paths(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 85 | { |
| 86 | char ip_addr[9]; |
| 87 | int mask_pos, err; |
| 88 | |
Joe Hershberger | 049a95a | 2015-04-08 01:41:01 -0500 | [diff] [blame] | 89 | sprintf(ip_addr, "%08X", ntohl(net_ip.s_addr)); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 90 | |
| 91 | for (mask_pos = 7; mask_pos >= 0; mask_pos--) { |
Steven Falco | 0e3f3f8 | 2013-10-07 09:51:48 -0400 | [diff] [blame] | 92 | err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 93 | |
| 94 | if (err > 0) |
| 95 | return err; |
| 96 | |
| 97 | ip_addr[mask_pos] = '\0'; |
| 98 | } |
| 99 | |
| 100 | return -ENOENT; |
| 101 | } |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 102 | /* |
| 103 | * Entry point for the 'pxe get' command. |
| 104 | * This Follows pxelinux's rules to download a config file from a tftp server. |
| 105 | * The file is stored at the location given by the pxefile_addr_r environment |
| 106 | * variable, which must be set. |
| 107 | * |
| 108 | * UUID comes from pxeuuid env variable, if defined |
| 109 | * MAC addr comes from ethaddr env variable, if defined |
| 110 | * IP |
| 111 | * |
| 112 | * see http://syslinux.zytor.com/wiki/index.php/PXELINUX |
| 113 | * |
| 114 | * Returns 0 on success or 1 on error. |
| 115 | */ |
| 116 | static int |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 117 | do_pxe_get(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 118 | { |
| 119 | char *pxefile_addr_str; |
Jason Hobbs | 834c938 | 2012-03-05 08:12:28 +0000 | [diff] [blame] | 120 | unsigned long pxefile_addr_r; |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 121 | int err, i = 0; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 122 | |
Rob Herring | 669df7e | 2012-05-25 10:47:39 +0000 | [diff] [blame] | 123 | do_getfile = do_get_tftp; |
| 124 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 125 | if (argc != 1) |
Simon Glass | 4c12eeb | 2011-12-10 08:44:01 +0000 | [diff] [blame] | 126 | return CMD_RET_USAGE; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 127 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 128 | pxefile_addr_str = from_env("pxefile_addr_r"); |
| 129 | |
| 130 | if (!pxefile_addr_str) |
| 131 | return 1; |
| 132 | |
| 133 | err = strict_strtoul(pxefile_addr_str, 16, |
Patrice Chotard | 31839dc | 2019-11-25 09:07:41 +0100 | [diff] [blame] | 134 | (unsigned long *)&pxefile_addr_r); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 135 | if (err < 0) |
| 136 | return 1; |
| 137 | |
| 138 | /* |
| 139 | * Keep trying paths until we successfully get a file we're looking |
| 140 | * for. |
| 141 | */ |
Sjoerd Simons | 4a0bd10 | 2015-04-13 22:54:25 +0200 | [diff] [blame] | 142 | if (pxe_uuid_path(cmdtp, pxefile_addr_r) > 0 || |
| 143 | pxe_mac_path(cmdtp, pxefile_addr_r) > 0 || |
| 144 | pxe_ipaddr_paths(cmdtp, pxefile_addr_r) > 0) { |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 145 | printf("Config file found\n"); |
| 146 | |
| 147 | return 0; |
| 148 | } |
| 149 | |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 150 | while (pxe_default_paths[i]) { |
Steven Falco | 0e3f3f8 | 2013-10-07 09:51:48 -0400 | [diff] [blame] | 151 | if (get_pxelinux_path(cmdtp, pxe_default_paths[i], |
Sjoerd Simons | 4a0bd10 | 2015-04-13 22:54:25 +0200 | [diff] [blame] | 152 | pxefile_addr_r) > 0) { |
Rob Herring | 39f9855 | 2012-12-02 21:00:28 -0600 | [diff] [blame] | 153 | printf("Config file found\n"); |
| 154 | return 0; |
| 155 | } |
| 156 | i++; |
| 157 | } |
| 158 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 159 | printf("Config file not found\n"); |
| 160 | |
| 161 | return 1; |
| 162 | } |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 163 | |
| 164 | /* |
| 165 | * Boots a system using a pxe file |
| 166 | * |
| 167 | * Returns 0 on success, 1 on error. |
| 168 | */ |
| 169 | static int |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 170 | do_pxe_boot(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 171 | { |
| 172 | unsigned long pxefile_addr_r; |
| 173 | struct pxe_menu *cfg; |
| 174 | char *pxefile_addr_str; |
| 175 | |
Rob Herring | 669df7e | 2012-05-25 10:47:39 +0000 | [diff] [blame] | 176 | do_getfile = do_get_tftp; |
| 177 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 178 | if (argc == 1) { |
| 179 | pxefile_addr_str = from_env("pxefile_addr_r"); |
| 180 | if (!pxefile_addr_str) |
| 181 | return 1; |
| 182 | |
| 183 | } else if (argc == 2) { |
| 184 | pxefile_addr_str = argv[1]; |
| 185 | } else { |
Simon Glass | 4c12eeb | 2011-12-10 08:44:01 +0000 | [diff] [blame] | 186 | return CMD_RET_USAGE; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 187 | } |
| 188 | |
| 189 | if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) { |
| 190 | printf("Invalid pxefile address: %s\n", pxefile_addr_str); |
| 191 | return 1; |
| 192 | } |
| 193 | |
Sjoerd Simons | 4a0bd10 | 2015-04-13 22:54:25 +0200 | [diff] [blame] | 194 | cfg = parse_pxefile(cmdtp, pxefile_addr_r); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 195 | |
Patrice Chotard | 31839dc | 2019-11-25 09:07:41 +0100 | [diff] [blame] | 196 | if (!cfg) { |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 197 | printf("Error parsing config file\n"); |
| 198 | return 1; |
| 199 | } |
| 200 | |
Tom Rini | d7884e0 | 2013-09-24 09:05:08 -0400 | [diff] [blame] | 201 | handle_pxe_menu(cmdtp, cfg); |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 202 | |
| 203 | destroy_pxe_menu(cfg); |
| 204 | |
Joe Hershberger | 1411157 | 2015-04-08 01:41:02 -0500 | [diff] [blame] | 205 | copy_filename(net_boot_file_name, "", sizeof(net_boot_file_name)); |
Stephen Warren | ded2e20 | 2014-07-22 18:06:46 -0600 | [diff] [blame] | 206 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 207 | return 0; |
| 208 | } |
| 209 | |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 210 | static struct cmd_tbl cmd_pxe_sub[] = { |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 211 | U_BOOT_CMD_MKENT(get, 1, 1, do_pxe_get, "", ""), |
| 212 | U_BOOT_CMD_MKENT(boot, 2, 1, do_pxe_boot, "", "") |
| 213 | }; |
| 214 | |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 215 | static int do_pxe(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[]) |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 216 | { |
Simon Glass | 0914011 | 2020-05-10 11:40:03 -0600 | [diff] [blame^] | 217 | struct cmd_tbl *cp; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 218 | |
| 219 | if (argc < 2) |
Simon Glass | 4c12eeb | 2011-12-10 08:44:01 +0000 | [diff] [blame] | 220 | return CMD_RET_USAGE; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 221 | |
Rob Herring | e5a9a40 | 2013-10-18 13:04:42 -0500 | [diff] [blame] | 222 | is_pxe = true; |
| 223 | |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 224 | /* drop initial "pxe" arg */ |
| 225 | argc--; |
| 226 | argv++; |
| 227 | |
| 228 | cp = find_cmd_tbl(argv[0], cmd_pxe_sub, ARRAY_SIZE(cmd_pxe_sub)); |
| 229 | |
| 230 | if (cp) |
| 231 | return cp->cmd(cmdtp, flag, argc, argv); |
| 232 | |
Simon Glass | 4c12eeb | 2011-12-10 08:44:01 +0000 | [diff] [blame] | 233 | return CMD_RET_USAGE; |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 234 | } |
| 235 | |
Patrice Chotard | 31839dc | 2019-11-25 09:07:41 +0100 | [diff] [blame] | 236 | U_BOOT_CMD(pxe, 3, 1, do_pxe, |
| 237 | "commands to get and boot from pxe files", |
| 238 | "get - try to retrieve a pxe file using tftp\n" |
| 239 | "pxe boot [pxefile_addr_r] - boot from the pxe file at pxefile_addr_r\n" |
Jason Hobbs | 06283a6 | 2011-08-31 10:37:30 -0500 | [diff] [blame] | 240 | ); |
Stephen Warren | b81fdb0 | 2014-02-05 20:49:20 -0700 | [diff] [blame] | 241 | #endif |