blob: 6372fc987187a35667ce978e50a6ceb94ad7231f [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Jason Hobbs06283a62011-08-31 10:37:30 -05002/*
3 * Copyright 2010-2011 Calxeda, Inc.
Bryan Wu1fb7d0e2014-07-31 17:39:59 -07004 * Copyright (c) 2014, NVIDIA CORPORATION. All rights reserved.
Jason Hobbs06283a62011-08-31 10:37:30 -05005 */
Wolfgang Denk1a459662013-07-08 09:37:19 +02006
Jason Hobbs06283a62011-08-31 10:37:30 -05007#include <common.h>
8#include <command.h>
Simon Glass14449982019-12-28 10:44:44 -07009#include <fs.h>
Simon Glass77f4e472019-12-28 10:44:43 -070010#include <net.h>
Jason Hobbs06283a62011-08-31 10:37:30 -050011
Patrice Chotard2373cba2019-11-25 09:07:37 +010012#include "pxe_utils.h"
Jason Hobbs06283a62011-08-31 10:37:30 -050013
Patrice Chotard2373cba2019-11-25 09:07:37 +010014#ifdef CONFIG_CMD_NET
Rob Herring39f98552012-12-02 21:00:28 -060015const char *pxe_default_paths[] = {
Joe Hershberger58d9ff92013-06-24 17:21:04 -050016#ifdef CONFIG_SYS_SOC
Marek BehĂșn2455efa2019-05-02 15:29:12 +020017#ifdef CONFIG_SYS_BOARD
18 "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC "-" CONFIG_SYS_BOARD,
19#endif
Rob Herring39f98552012-12-02 21:00:28 -060020 "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC,
Joe Hershberger58d9ff92013-06-24 17:21:04 -050021#endif
Rob Herring39f98552012-12-02 21:00:28 -060022 "default-" CONFIG_SYS_ARCH,
23 "default",
24 NULL
25};
26
Simon Glass09140112020-05-10 11:40:03 -060027static int do_get_tftp(struct cmd_tbl *cmdtp, const char *file_path,
28 char *file_addr)
Rob Herring669df7e2012-05-25 10:47:39 +000029{
30 char *tftp_argv[] = {"tftp", NULL, NULL, NULL};
31
32 tftp_argv[1] = file_addr;
Rob Herring23b71942012-12-02 21:00:21 -060033 tftp_argv[2] = (void *)file_path;
Rob Herring669df7e2012-05-25 10:47:39 +000034
Steven Falco0e3f3f82013-10-07 09:51:48 -040035 if (do_tftpb(cmdtp, 0, 3, tftp_argv))
Rob Herring669df7e2012-05-25 10:47:39 +000036 return -ENOENT;
37
38 return 1;
39}
Stephen Warrenb81fdb02014-02-05 20:49:20 -070040
Jason Hobbs06283a62011-08-31 10:37:30 -050041/*
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 Glass09140112020-05-10 11:40:03 -060046static int pxe_uuid_path(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -050047{
48 char *uuid_str;
49
50 uuid_str = from_env("pxeuuid");
51
52 if (!uuid_str)
53 return -ENOENT;
54
Steven Falco0e3f3f82013-10-07 09:51:48 -040055 return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -050056}
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 Glass09140112020-05-10 11:40:03 -060064static int pxe_mac_path(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -050065{
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 Falco0e3f3f82013-10-07 09:51:48 -040074 return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -050075}
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 Glass09140112020-05-10 11:40:03 -060084static int pxe_ipaddr_paths(struct cmd_tbl *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -050085{
86 char ip_addr[9];
87 int mask_pos, err;
88
Joe Hershberger049a95a2015-04-08 01:41:01 -050089 sprintf(ip_addr, "%08X", ntohl(net_ip.s_addr));
Jason Hobbs06283a62011-08-31 10:37:30 -050090
91 for (mask_pos = 7; mask_pos >= 0; mask_pos--) {
Steven Falco0e3f3f82013-10-07 09:51:48 -040092 err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -050093
94 if (err > 0)
95 return err;
96
97 ip_addr[mask_pos] = '\0';
98 }
99
100 return -ENOENT;
101}
Jason Hobbs06283a62011-08-31 10:37:30 -0500102/*
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 */
116static int
Simon Glass09140112020-05-10 11:40:03 -0600117do_pxe_get(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
Jason Hobbs06283a62011-08-31 10:37:30 -0500118{
119 char *pxefile_addr_str;
Jason Hobbs834c9382012-03-05 08:12:28 +0000120 unsigned long pxefile_addr_r;
Rob Herring39f98552012-12-02 21:00:28 -0600121 int err, i = 0;
Jason Hobbs06283a62011-08-31 10:37:30 -0500122
Rob Herring669df7e2012-05-25 10:47:39 +0000123 do_getfile = do_get_tftp;
124
Jason Hobbs06283a62011-08-31 10:37:30 -0500125 if (argc != 1)
Simon Glass4c12eeb2011-12-10 08:44:01 +0000126 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -0500127
Jason Hobbs06283a62011-08-31 10:37:30 -0500128 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 Chotard31839dc2019-11-25 09:07:41 +0100134 (unsigned long *)&pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500135 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 Simons4a0bd102015-04-13 22:54:25 +0200142 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 Hobbs06283a62011-08-31 10:37:30 -0500145 printf("Config file found\n");
146
147 return 0;
148 }
149
Rob Herring39f98552012-12-02 21:00:28 -0600150 while (pxe_default_paths[i]) {
Steven Falco0e3f3f82013-10-07 09:51:48 -0400151 if (get_pxelinux_path(cmdtp, pxe_default_paths[i],
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200152 pxefile_addr_r) > 0) {
Rob Herring39f98552012-12-02 21:00:28 -0600153 printf("Config file found\n");
154 return 0;
155 }
156 i++;
157 }
158
Jason Hobbs06283a62011-08-31 10:37:30 -0500159 printf("Config file not found\n");
160
161 return 1;
162}
Jason Hobbs06283a62011-08-31 10:37:30 -0500163
164/*
165 * Boots a system using a pxe file
166 *
167 * Returns 0 on success, 1 on error.
168 */
169static int
Simon Glass09140112020-05-10 11:40:03 -0600170do_pxe_boot(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
Jason Hobbs06283a62011-08-31 10:37:30 -0500171{
172 unsigned long pxefile_addr_r;
173 struct pxe_menu *cfg;
174 char *pxefile_addr_str;
175
Rob Herring669df7e2012-05-25 10:47:39 +0000176 do_getfile = do_get_tftp;
177
Jason Hobbs06283a62011-08-31 10:37:30 -0500178 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 Glass4c12eeb2011-12-10 08:44:01 +0000186 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -0500187 }
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 Simons4a0bd102015-04-13 22:54:25 +0200194 cfg = parse_pxefile(cmdtp, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500195
Patrice Chotard31839dc2019-11-25 09:07:41 +0100196 if (!cfg) {
Jason Hobbs06283a62011-08-31 10:37:30 -0500197 printf("Error parsing config file\n");
198 return 1;
199 }
200
Tom Rinid7884e02013-09-24 09:05:08 -0400201 handle_pxe_menu(cmdtp, cfg);
Jason Hobbs06283a62011-08-31 10:37:30 -0500202
203 destroy_pxe_menu(cfg);
204
Joe Hershberger14111572015-04-08 01:41:02 -0500205 copy_filename(net_boot_file_name, "", sizeof(net_boot_file_name));
Stephen Warrended2e202014-07-22 18:06:46 -0600206
Jason Hobbs06283a62011-08-31 10:37:30 -0500207 return 0;
208}
209
Simon Glass09140112020-05-10 11:40:03 -0600210static struct cmd_tbl cmd_pxe_sub[] = {
Jason Hobbs06283a62011-08-31 10:37:30 -0500211 U_BOOT_CMD_MKENT(get, 1, 1, do_pxe_get, "", ""),
212 U_BOOT_CMD_MKENT(boot, 2, 1, do_pxe_boot, "", "")
213};
214
Simon Glass09140112020-05-10 11:40:03 -0600215static int do_pxe(struct cmd_tbl *cmdtp, int flag, int argc, char *const argv[])
Jason Hobbs06283a62011-08-31 10:37:30 -0500216{
Simon Glass09140112020-05-10 11:40:03 -0600217 struct cmd_tbl *cp;
Jason Hobbs06283a62011-08-31 10:37:30 -0500218
219 if (argc < 2)
Simon Glass4c12eeb2011-12-10 08:44:01 +0000220 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -0500221
Rob Herringe5a9a402013-10-18 13:04:42 -0500222 is_pxe = true;
223
Jason Hobbs06283a62011-08-31 10:37:30 -0500224 /* 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 Glass4c12eeb2011-12-10 08:44:01 +0000233 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -0500234}
235
Patrice Chotard31839dc2019-11-25 09:07:41 +0100236U_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 Hobbs06283a62011-08-31 10:37:30 -0500240);
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700241#endif