blob: c5a770a26995fabc5a1b3b0b7450997d29e110db [file] [log] [blame]
Jason Hobbs06283a62011-08-31 10:37:30 -05001/*
2 * Copyright 2010-2011 Calxeda, Inc.
Bryan Wu1fb7d0e2014-07-31 17:39:59 -07003 * Copyright (c) 2014, NVIDIA CORPORATION. All rights reserved.
Jason Hobbs06283a62011-08-31 10:37:30 -05004 *
Wolfgang Denk1a459662013-07-08 09:37:19 +02005 * SPDX-License-Identifier: GPL-2.0+
Jason Hobbs06283a62011-08-31 10:37:30 -05006 */
Wolfgang Denk1a459662013-07-08 09:37:19 +02007
Jason Hobbs06283a62011-08-31 10:37:30 -05008#include <common.h>
9#include <command.h>
10#include <malloc.h>
Joe Hershberger0eb25b62015-03-22 17:08:59 -050011#include <mapmem.h>
Jason Hobbs06283a62011-08-31 10:37:30 -050012#include <linux/string.h>
13#include <linux/ctype.h>
14#include <errno.h>
15#include <linux/list.h>
Dennis Gilmore6d1a3e52014-02-04 05:25:46 -060016#include <fs.h>
Sjoerd Simons4a0bd102015-04-13 22:54:25 +020017#include <asm/io.h>
Jason Hobbs06283a62011-08-31 10:37:30 -050018
19#include "menu.h"
Hans de Goedeb1ba62d2014-08-06 09:37:39 +020020#include "cli.h"
Jason Hobbs06283a62011-08-31 10:37:30 -050021
22#define MAX_TFTP_PATH_LEN 127
23
Rob Herring39f98552012-12-02 21:00:28 -060024const char *pxe_default_paths[] = {
Joe Hershberger58d9ff92013-06-24 17:21:04 -050025#ifdef CONFIG_SYS_SOC
Rob Herring39f98552012-12-02 21:00:28 -060026 "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC,
Joe Hershberger58d9ff92013-06-24 17:21:04 -050027#endif
Rob Herring39f98552012-12-02 21:00:28 -060028 "default-" CONFIG_SYS_ARCH,
29 "default",
30 NULL
31};
32
Rob Herringe5a9a402013-10-18 13:04:42 -050033static bool is_pxe;
34
Jason Hobbs06283a62011-08-31 10:37:30 -050035/*
Simon Glass00caae62017-08-03 12:22:12 -060036 * Like env_get, but prints an error if envvar isn't defined in the
37 * environment. It always returns what env_get does, so it can be used in
38 * place of env_get without changing error handling otherwise.
Jason Hobbs06283a62011-08-31 10:37:30 -050039 */
Rob Herring23b71942012-12-02 21:00:21 -060040static char *from_env(const char *envvar)
Jason Hobbs06283a62011-08-31 10:37:30 -050041{
42 char *ret;
43
Simon Glass00caae62017-08-03 12:22:12 -060044 ret = env_get(envvar);
Jason Hobbs06283a62011-08-31 10:37:30 -050045
46 if (!ret)
47 printf("missing environment variable: %s\n", envvar);
48
49 return ret;
50}
51
Stephen Warrenb81fdb02014-02-05 20:49:20 -070052#ifdef CONFIG_CMD_NET
Jason Hobbs06283a62011-08-31 10:37:30 -050053/*
54 * Convert an ethaddr from the environment to the format used by pxelinux
55 * filenames based on mac addresses. Convert's ':' to '-', and adds "01-" to
56 * the beginning of the ethernet address to indicate a hardware type of
57 * Ethernet. Also converts uppercase hex characters into lowercase, to match
58 * pxelinux's behavior.
59 *
60 * Returns 1 for success, -ENOENT if 'ethaddr' is undefined in the
61 * environment, or some other value < 0 on error.
62 */
63static int format_mac_pxe(char *outbuf, size_t outbuf_len)
64{
Rob Herringef034c92012-12-02 21:00:20 -060065 uchar ethaddr[6];
Jason Hobbs06283a62011-08-31 10:37:30 -050066
Rob Herringef034c92012-12-02 21:00:20 -060067 if (outbuf_len < 21) {
David Feng5cea95c2013-12-14 11:47:30 +080068 printf("outbuf is too small (%zd < 21)\n", outbuf_len);
Jason Hobbs06283a62011-08-31 10:37:30 -050069
70 return -EINVAL;
71 }
72
Simon Glass35affd72017-08-03 12:22:14 -060073 if (!eth_env_get_enetaddr_by_index("eth", eth_get_dev_index(), ethaddr))
Rob Herringef034c92012-12-02 21:00:20 -060074 return -ENOENT;
Jason Hobbs06283a62011-08-31 10:37:30 -050075
Rob Herringef034c92012-12-02 21:00:20 -060076 sprintf(outbuf, "01-%02x-%02x-%02x-%02x-%02x-%02x",
77 ethaddr[0], ethaddr[1], ethaddr[2],
78 ethaddr[3], ethaddr[4], ethaddr[5]);
Jason Hobbs06283a62011-08-31 10:37:30 -050079
80 return 1;
81}
Stephen Warrenb81fdb02014-02-05 20:49:20 -070082#endif
Jason Hobbs06283a62011-08-31 10:37:30 -050083
84/*
85 * Returns the directory the file specified in the bootfile env variable is
86 * in. If bootfile isn't defined in the environment, return NULL, which should
87 * be interpreted as "don't prepend anything to paths".
88 */
Rob Herring90ba7d72012-03-28 05:51:36 +000089static int get_bootfile_path(const char *file_path, char *bootfile_path,
90 size_t bootfile_path_size)
Jason Hobbs06283a62011-08-31 10:37:30 -050091{
92 char *bootfile, *last_slash;
Rob Herring90ba7d72012-03-28 05:51:36 +000093 size_t path_len = 0;
94
Rob Herringe5a9a402013-10-18 13:04:42 -050095 /* Only syslinux allows absolute paths */
96 if (file_path[0] == '/' && !is_pxe)
Rob Herring90ba7d72012-03-28 05:51:36 +000097 goto ret;
Jason Hobbs06283a62011-08-31 10:37:30 -050098
99 bootfile = from_env("bootfile");
100
Rob Herring90ba7d72012-03-28 05:51:36 +0000101 if (!bootfile)
102 goto ret;
Jason Hobbs06283a62011-08-31 10:37:30 -0500103
104 last_slash = strrchr(bootfile, '/');
105
Rob Herring90ba7d72012-03-28 05:51:36 +0000106 if (last_slash == NULL)
107 goto ret;
Jason Hobbs06283a62011-08-31 10:37:30 -0500108
109 path_len = (last_slash - bootfile) + 1;
110
111 if (bootfile_path_size < path_len) {
David Feng5cea95c2013-12-14 11:47:30 +0800112 printf("bootfile_path too small. (%zd < %zd)\n",
Jason Hobbs06283a62011-08-31 10:37:30 -0500113 bootfile_path_size, path_len);
114
115 return -1;
116 }
117
118 strncpy(bootfile_path, bootfile, path_len);
119
Rob Herring90ba7d72012-03-28 05:51:36 +0000120 ret:
Jason Hobbs06283a62011-08-31 10:37:30 -0500121 bootfile_path[path_len] = '\0';
122
123 return 1;
124}
125
Steven Falco0e3f3f82013-10-07 09:51:48 -0400126static int (*do_getfile)(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr);
Rob Herring669df7e2012-05-25 10:47:39 +0000127
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700128#ifdef CONFIG_CMD_NET
Steven Falco0e3f3f82013-10-07 09:51:48 -0400129static int do_get_tftp(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
Rob Herring669df7e2012-05-25 10:47:39 +0000130{
131 char *tftp_argv[] = {"tftp", NULL, NULL, NULL};
132
133 tftp_argv[1] = file_addr;
Rob Herring23b71942012-12-02 21:00:21 -0600134 tftp_argv[2] = (void *)file_path;
Rob Herring669df7e2012-05-25 10:47:39 +0000135
Steven Falco0e3f3f82013-10-07 09:51:48 -0400136 if (do_tftpb(cmdtp, 0, 3, tftp_argv))
Rob Herring669df7e2012-05-25 10:47:39 +0000137 return -ENOENT;
138
139 return 1;
140}
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700141#endif
Rob Herring669df7e2012-05-25 10:47:39 +0000142
143static char *fs_argv[5];
144
Steven Falco0e3f3f82013-10-07 09:51:48 -0400145static int do_get_ext2(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
Rob Herring669df7e2012-05-25 10:47:39 +0000146{
147#ifdef CONFIG_CMD_EXT2
148 fs_argv[0] = "ext2load";
149 fs_argv[3] = file_addr;
Rob Herring23b71942012-12-02 21:00:21 -0600150 fs_argv[4] = (void *)file_path;
Rob Herring669df7e2012-05-25 10:47:39 +0000151
Steven Falco0e3f3f82013-10-07 09:51:48 -0400152 if (!do_ext2load(cmdtp, 0, 5, fs_argv))
Rob Herring669df7e2012-05-25 10:47:39 +0000153 return 1;
154#endif
155 return -ENOENT;
156}
157
Steven Falco0e3f3f82013-10-07 09:51:48 -0400158static int do_get_fat(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
Rob Herring669df7e2012-05-25 10:47:39 +0000159{
160#ifdef CONFIG_CMD_FAT
161 fs_argv[0] = "fatload";
162 fs_argv[3] = file_addr;
Rob Herring23b71942012-12-02 21:00:21 -0600163 fs_argv[4] = (void *)file_path;
Rob Herring669df7e2012-05-25 10:47:39 +0000164
Steven Falco0e3f3f82013-10-07 09:51:48 -0400165 if (!do_fat_fsload(cmdtp, 0, 5, fs_argv))
Rob Herring669df7e2012-05-25 10:47:39 +0000166 return 1;
167#endif
168 return -ENOENT;
169}
170
Dennis Gilmore6d1a3e52014-02-04 05:25:46 -0600171static int do_get_any(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
172{
173#ifdef CONFIG_CMD_FS_GENERIC
174 fs_argv[0] = "load";
175 fs_argv[3] = file_addr;
176 fs_argv[4] = (void *)file_path;
177
178 if (!do_load(cmdtp, 0, 5, fs_argv, FS_TYPE_ANY))
179 return 1;
180#endif
181 return -ENOENT;
182}
183
Jason Hobbs06283a62011-08-31 10:37:30 -0500184/*
185 * As in pxelinux, paths to files referenced from files we retrieve are
186 * relative to the location of bootfile. get_relfile takes such a path and
187 * joins it with the bootfile path to get the full path to the target file. If
188 * the bootfile path is NULL, we use file_path as is.
189 *
190 * Returns 1 for success, or < 0 on error.
191 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200192static int get_relfile(cmd_tbl_t *cmdtp, const char *file_path,
193 unsigned long file_addr)
Jason Hobbs06283a62011-08-31 10:37:30 -0500194{
195 size_t path_len;
196 char relfile[MAX_TFTP_PATH_LEN+1];
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200197 char addr_buf[18];
Jason Hobbs06283a62011-08-31 10:37:30 -0500198 int err;
199
Rob Herring90ba7d72012-03-28 05:51:36 +0000200 err = get_bootfile_path(file_path, relfile, sizeof(relfile));
Jason Hobbs06283a62011-08-31 10:37:30 -0500201
202 if (err < 0)
203 return err;
204
205 path_len = strlen(file_path);
206 path_len += strlen(relfile);
207
208 if (path_len > MAX_TFTP_PATH_LEN) {
209 printf("Base path too long (%s%s)\n",
210 relfile,
211 file_path);
212
213 return -ENAMETOOLONG;
214 }
215
216 strcat(relfile, file_path);
217
218 printf("Retrieving file: %s\n", relfile);
219
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200220 sprintf(addr_buf, "%lx", file_addr);
Jason Hobbs06283a62011-08-31 10:37:30 -0500221
Steven Falco0e3f3f82013-10-07 09:51:48 -0400222 return do_getfile(cmdtp, relfile, addr_buf);
Jason Hobbs06283a62011-08-31 10:37:30 -0500223}
224
225/*
226 * Retrieve the file at 'file_path' to the locate given by 'file_addr'. If
227 * 'bootfile' was specified in the environment, the path to bootfile will be
228 * prepended to 'file_path' and the resulting path will be used.
229 *
230 * Returns 1 on success, or < 0 for error.
231 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200232static int get_pxe_file(cmd_tbl_t *cmdtp, const char *file_path,
233 unsigned long file_addr)
Jason Hobbs06283a62011-08-31 10:37:30 -0500234{
235 unsigned long config_file_size;
236 char *tftp_filesize;
237 int err;
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200238 char *buf;
Jason Hobbs06283a62011-08-31 10:37:30 -0500239
Steven Falco0e3f3f82013-10-07 09:51:48 -0400240 err = get_relfile(cmdtp, file_path, file_addr);
Jason Hobbs06283a62011-08-31 10:37:30 -0500241
242 if (err < 0)
243 return err;
244
245 /*
246 * the file comes without a NUL byte at the end, so find out its size
247 * and add the NUL byte.
248 */
249 tftp_filesize = from_env("filesize");
250
251 if (!tftp_filesize)
252 return -ENOENT;
253
254 if (strict_strtoul(tftp_filesize, 16, &config_file_size) < 0)
255 return -EINVAL;
256
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200257 buf = map_sysmem(file_addr + config_file_size, 1);
258 *buf = '\0';
259 unmap_sysmem(buf);
Jason Hobbs06283a62011-08-31 10:37:30 -0500260
261 return 1;
262}
263
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700264#ifdef CONFIG_CMD_NET
265
Jason Hobbs06283a62011-08-31 10:37:30 -0500266#define PXELINUX_DIR "pxelinux.cfg/"
267
268/*
269 * Retrieves a file in the 'pxelinux.cfg' folder. Since this uses get_pxe_file
270 * to do the hard work, the location of the 'pxelinux.cfg' folder is generated
271 * from the bootfile path, as described above.
272 *
273 * Returns 1 on success or < 0 on error.
274 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200275static int get_pxelinux_path(cmd_tbl_t *cmdtp, const char *file,
276 unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -0500277{
278 size_t base_len = strlen(PXELINUX_DIR);
279 char path[MAX_TFTP_PATH_LEN+1];
280
281 if (base_len + strlen(file) > MAX_TFTP_PATH_LEN) {
282 printf("path (%s%s) too long, skipping\n",
283 PXELINUX_DIR, file);
284 return -ENAMETOOLONG;
285 }
286
287 sprintf(path, PXELINUX_DIR "%s", file);
288
Steven Falco0e3f3f82013-10-07 09:51:48 -0400289 return get_pxe_file(cmdtp, path, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500290}
291
292/*
293 * Looks for a pxe file with a name based on the pxeuuid environment variable.
294 *
295 * Returns 1 on success or < 0 on error.
296 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200297static int pxe_uuid_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -0500298{
299 char *uuid_str;
300
301 uuid_str = from_env("pxeuuid");
302
303 if (!uuid_str)
304 return -ENOENT;
305
Steven Falco0e3f3f82013-10-07 09:51:48 -0400306 return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500307}
308
309/*
310 * Looks for a pxe file with a name based on the 'ethaddr' environment
311 * variable.
312 *
313 * Returns 1 on success or < 0 on error.
314 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200315static int pxe_mac_path(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -0500316{
317 char mac_str[21];
318 int err;
319
320 err = format_mac_pxe(mac_str, sizeof(mac_str));
321
322 if (err < 0)
323 return err;
324
Steven Falco0e3f3f82013-10-07 09:51:48 -0400325 return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500326}
327
328/*
329 * Looks for pxe files with names based on our IP address. See pxelinux
330 * documentation for details on what these file names look like. We match
331 * that exactly.
332 *
333 * Returns 1 on success or < 0 on error.
334 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200335static int pxe_ipaddr_paths(cmd_tbl_t *cmdtp, unsigned long pxefile_addr_r)
Jason Hobbs06283a62011-08-31 10:37:30 -0500336{
337 char ip_addr[9];
338 int mask_pos, err;
339
Joe Hershberger049a95a2015-04-08 01:41:01 -0500340 sprintf(ip_addr, "%08X", ntohl(net_ip.s_addr));
Jason Hobbs06283a62011-08-31 10:37:30 -0500341
342 for (mask_pos = 7; mask_pos >= 0; mask_pos--) {
Steven Falco0e3f3f82013-10-07 09:51:48 -0400343 err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -0500344
345 if (err > 0)
346 return err;
347
348 ip_addr[mask_pos] = '\0';
349 }
350
351 return -ENOENT;
352}
353
354/*
355 * Entry point for the 'pxe get' command.
356 * This Follows pxelinux's rules to download a config file from a tftp server.
357 * The file is stored at the location given by the pxefile_addr_r environment
358 * variable, which must be set.
359 *
360 * UUID comes from pxeuuid env variable, if defined
361 * MAC addr comes from ethaddr env variable, if defined
362 * IP
363 *
364 * see http://syslinux.zytor.com/wiki/index.php/PXELINUX
365 *
366 * Returns 0 on success or 1 on error.
367 */
368static int
369do_pxe_get(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
370{
371 char *pxefile_addr_str;
Jason Hobbs834c9382012-03-05 08:12:28 +0000372 unsigned long pxefile_addr_r;
Rob Herring39f98552012-12-02 21:00:28 -0600373 int err, i = 0;
Jason Hobbs06283a62011-08-31 10:37:30 -0500374
Rob Herring669df7e2012-05-25 10:47:39 +0000375 do_getfile = do_get_tftp;
376
Jason Hobbs06283a62011-08-31 10:37:30 -0500377 if (argc != 1)
Simon Glass4c12eeb2011-12-10 08:44:01 +0000378 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -0500379
Jason Hobbs06283a62011-08-31 10:37:30 -0500380 pxefile_addr_str = from_env("pxefile_addr_r");
381
382 if (!pxefile_addr_str)
383 return 1;
384
385 err = strict_strtoul(pxefile_addr_str, 16,
386 (unsigned long *)&pxefile_addr_r);
387 if (err < 0)
388 return 1;
389
390 /*
391 * Keep trying paths until we successfully get a file we're looking
392 * for.
393 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200394 if (pxe_uuid_path(cmdtp, pxefile_addr_r) > 0 ||
395 pxe_mac_path(cmdtp, pxefile_addr_r) > 0 ||
396 pxe_ipaddr_paths(cmdtp, pxefile_addr_r) > 0) {
Jason Hobbs06283a62011-08-31 10:37:30 -0500397 printf("Config file found\n");
398
399 return 0;
400 }
401
Rob Herring39f98552012-12-02 21:00:28 -0600402 while (pxe_default_paths[i]) {
Steven Falco0e3f3f82013-10-07 09:51:48 -0400403 if (get_pxelinux_path(cmdtp, pxe_default_paths[i],
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200404 pxefile_addr_r) > 0) {
Rob Herring39f98552012-12-02 21:00:28 -0600405 printf("Config file found\n");
406 return 0;
407 }
408 i++;
409 }
410
Jason Hobbs06283a62011-08-31 10:37:30 -0500411 printf("Config file not found\n");
412
413 return 1;
414}
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700415#endif
Jason Hobbs06283a62011-08-31 10:37:30 -0500416
417/*
418 * Wrapper to make it easier to store the file at file_path in the location
419 * specified by envaddr_name. file_path will be joined to the bootfile path,
420 * if any is specified.
421 *
422 * Returns 1 on success or < 0 on error.
423 */
Steven Falco0e3f3f82013-10-07 09:51:48 -0400424static int get_relfile_envaddr(cmd_tbl_t *cmdtp, const char *file_path, const char *envaddr_name)
Jason Hobbs06283a62011-08-31 10:37:30 -0500425{
Jason Hobbs834c9382012-03-05 08:12:28 +0000426 unsigned long file_addr;
Jason Hobbs06283a62011-08-31 10:37:30 -0500427 char *envaddr;
428
429 envaddr = from_env(envaddr_name);
430
431 if (!envaddr)
432 return -ENOENT;
433
Jason Hobbs834c9382012-03-05 08:12:28 +0000434 if (strict_strtoul(envaddr, 16, &file_addr) < 0)
Jason Hobbs06283a62011-08-31 10:37:30 -0500435 return -EINVAL;
436
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200437 return get_relfile(cmdtp, file_path, file_addr);
Jason Hobbs06283a62011-08-31 10:37:30 -0500438}
439
440/*
441 * A note on the pxe file parser.
442 *
443 * We're parsing files that use syslinux grammar, which has a few quirks.
444 * String literals must be recognized based on context - there is no
445 * quoting or escaping support. There's also nothing to explicitly indicate
446 * when a label section completes. We deal with that by ending a label
447 * section whenever we see a line that doesn't include.
448 *
449 * As with the syslinux family, this same file format could be reused in the
450 * future for non pxe purposes. The only action it takes during parsing that
451 * would throw this off is handling of include files. It assumes we're using
452 * pxe, and does a tftp download of a file listed as an include file in the
453 * middle of the parsing operation. That could be handled by refactoring it to
454 * take a 'include file getter' function.
455 */
456
457/*
458 * Describes a single label given in a pxe file.
459 *
460 * Create these with the 'label_create' function given below.
461 *
462 * name - the name of the menu as given on the 'menu label' line.
463 * kernel - the path to the kernel file to use for this label.
464 * append - kernel command line to use when booting this label
465 * initrd - path to the initrd to use for this label.
466 * attempted - 0 if we haven't tried to boot this label, 1 if we have.
467 * localboot - 1 if this label specified 'localboot', 0 otherwise.
468 * list - lets these form a list, which a pxe_menu struct will hold.
469 */
470struct pxe_label {
Rob Herring32d2ffe2012-12-02 21:00:26 -0600471 char num[4];
Jason Hobbs06283a62011-08-31 10:37:30 -0500472 char *name;
Rob Herring7815c4e2012-03-28 05:51:34 +0000473 char *menu;
Jason Hobbs06283a62011-08-31 10:37:30 -0500474 char *kernel;
475 char *append;
476 char *initrd;
Chander Kashyapa6559382012-09-06 19:36:31 +0000477 char *fdt;
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700478 char *fdtdir;
Rob Herring98f64672012-12-02 21:00:29 -0600479 int ipappend;
Jason Hobbs06283a62011-08-31 10:37:30 -0500480 int attempted;
481 int localboot;
Rob Herring500f3042012-12-02 21:00:22 -0600482 int localboot_val;
Jason Hobbs06283a62011-08-31 10:37:30 -0500483 struct list_head list;
484};
485
486/*
487 * Describes a pxe menu as given via pxe files.
488 *
489 * title - the name of the menu as given by a 'menu title' line.
490 * default_label - the name of the default label, if any.
491 * timeout - time in tenths of a second to wait for a user key-press before
492 * booting the default label.
493 * prompt - if 0, don't prompt for a choice unless the timeout period is
494 * interrupted. If 1, always prompt for a choice regardless of
495 * timeout.
496 * labels - a list of labels defined for the menu.
497 */
498struct pxe_menu {
499 char *title;
500 char *default_label;
501 int timeout;
502 int prompt;
503 struct list_head labels;
504};
505
506/*
507 * Allocates memory for and initializes a pxe_label. This uses malloc, so the
508 * result must be free()'d to reclaim the memory.
509 *
510 * Returns NULL if malloc fails.
511 */
512static struct pxe_label *label_create(void)
513{
514 struct pxe_label *label;
515
516 label = malloc(sizeof(struct pxe_label));
517
518 if (!label)
519 return NULL;
520
521 memset(label, 0, sizeof(struct pxe_label));
522
523 return label;
524}
525
526/*
527 * Free the memory used by a pxe_label, including that used by its name,
528 * kernel, append and initrd members, if they're non NULL.
529 *
530 * So - be sure to only use dynamically allocated memory for the members of
531 * the pxe_label struct, unless you want to clean it up first. These are
532 * currently only created by the pxe file parsing code.
533 */
534static void label_destroy(struct pxe_label *label)
535{
536 if (label->name)
537 free(label->name);
538
539 if (label->kernel)
540 free(label->kernel);
541
542 if (label->append)
543 free(label->append);
544
545 if (label->initrd)
546 free(label->initrd);
547
Chander Kashyapa6559382012-09-06 19:36:31 +0000548 if (label->fdt)
549 free(label->fdt);
550
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700551 if (label->fdtdir)
552 free(label->fdtdir);
553
Jason Hobbs06283a62011-08-31 10:37:30 -0500554 free(label);
555}
556
557/*
558 * Print a label and its string members if they're defined.
559 *
560 * This is passed as a callback to the menu code for displaying each
561 * menu entry.
562 */
563static void label_print(void *data)
564{
565 struct pxe_label *label = data;
Rob Herring32d2ffe2012-12-02 21:00:26 -0600566 const char *c = label->menu ? label->menu : label->name;
Jason Hobbs06283a62011-08-31 10:37:30 -0500567
Rob Herring32d2ffe2012-12-02 21:00:26 -0600568 printf("%s:\t%s\n", label->num, c);
Jason Hobbs06283a62011-08-31 10:37:30 -0500569}
570
571/*
572 * Boot a label that specified 'localboot'. This requires that the 'localcmd'
Bin Menga1875592016-02-05 19:30:11 -0800573 * environment variable is defined. Its contents will be executed as U-Boot
Jason Hobbs06283a62011-08-31 10:37:30 -0500574 * command. If the label specified an 'append' line, its contents will be
575 * used to overwrite the contents of the 'bootargs' environment variable prior
576 * to running 'localcmd'.
577 *
578 * Returns 1 on success or < 0 on error.
579 */
580static int label_localboot(struct pxe_label *label)
581{
Simon Glassd51004a2012-03-30 21:30:55 +0000582 char *localcmd;
Jason Hobbs06283a62011-08-31 10:37:30 -0500583
584 localcmd = from_env("localcmd");
585
586 if (!localcmd)
587 return -ENOENT;
588
Hans de Goedeb1ba62d2014-08-06 09:37:39 +0200589 if (label->append) {
590 char bootargs[CONFIG_SYS_CBSIZE];
591
592 cli_simple_process_macros(label->append, bootargs);
Simon Glass382bee52017-08-03 12:22:09 -0600593 env_set("bootargs", bootargs);
Hans de Goedeb1ba62d2014-08-06 09:37:39 +0200594 }
Jason Hobbs06283a62011-08-31 10:37:30 -0500595
Simon Glassd51004a2012-03-30 21:30:55 +0000596 debug("running: %s\n", localcmd);
Jason Hobbs06283a62011-08-31 10:37:30 -0500597
Simon Glassd51004a2012-03-30 21:30:55 +0000598 return run_command_list(localcmd, strlen(localcmd), 0);
Jason Hobbs06283a62011-08-31 10:37:30 -0500599}
600
601/*
602 * Boot according to the contents of a pxe_label.
603 *
604 * If we can't boot for any reason, we return. A successful boot never
605 * returns.
606 *
607 * The kernel will be stored in the location given by the 'kernel_addr_r'
608 * environment variable.
609 *
610 * If the label specifies an initrd file, it will be stored in the location
611 * given by the 'ramdisk_addr_r' environment variable.
612 *
613 * If the label specifies an 'append' line, its contents will overwrite that
614 * of the 'bootargs' environment variable.
615 */
Tom Rinid7884e02013-09-24 09:05:08 -0400616static int label_boot(cmd_tbl_t *cmdtp, struct pxe_label *label)
Jason Hobbs06283a62011-08-31 10:37:30 -0500617{
618 char *bootm_argv[] = { "bootm", NULL, NULL, NULL, NULL };
Rob Herringe6b6ccf2012-12-03 13:17:21 -0600619 char initrd_str[22];
Rob Herring98f64672012-12-02 21:00:29 -0600620 char mac_str[29] = "";
621 char ip_str[68] = "";
York Sunf63963f2016-09-01 16:28:21 +0800622 int bootm_argc = 2;
Rob Herring98f64672012-12-02 21:00:29 -0600623 int len = 0;
Bryan Wu1fb7d0e2014-07-31 17:39:59 -0700624 ulong kernel_addr;
625 void *buf;
Jason Hobbs06283a62011-08-31 10:37:30 -0500626
627 label_print(label);
628
629 label->attempted = 1;
630
631 if (label->localboot) {
Rob Herring500f3042012-12-02 21:00:22 -0600632 if (label->localboot_val >= 0)
633 label_localboot(label);
634 return 0;
Jason Hobbs06283a62011-08-31 10:37:30 -0500635 }
636
637 if (label->kernel == NULL) {
638 printf("No kernel given, skipping %s\n",
639 label->name);
Rob Herring500f3042012-12-02 21:00:22 -0600640 return 1;
Jason Hobbs06283a62011-08-31 10:37:30 -0500641 }
642
643 if (label->initrd) {
Steven Falco0e3f3f82013-10-07 09:51:48 -0400644 if (get_relfile_envaddr(cmdtp, label->initrd, "ramdisk_addr_r") < 0) {
Jason Hobbs06283a62011-08-31 10:37:30 -0500645 printf("Skipping %s for failure retrieving initrd\n",
646 label->name);
Rob Herring500f3042012-12-02 21:00:22 -0600647 return 1;
Jason Hobbs06283a62011-08-31 10:37:30 -0500648 }
649
Rob Herringe6b6ccf2012-12-03 13:17:21 -0600650 bootm_argv[2] = initrd_str;
Simon Glass00caae62017-08-03 12:22:12 -0600651 strcpy(bootm_argv[2], env_get("ramdisk_addr_r"));
Rob Herringe6b6ccf2012-12-03 13:17:21 -0600652 strcat(bootm_argv[2], ":");
Simon Glass00caae62017-08-03 12:22:12 -0600653 strcat(bootm_argv[2], env_get("filesize"));
Jason Hobbs06283a62011-08-31 10:37:30 -0500654 }
655
Steven Falco0e3f3f82013-10-07 09:51:48 -0400656 if (get_relfile_envaddr(cmdtp, label->kernel, "kernel_addr_r") < 0) {
Jason Hobbs06283a62011-08-31 10:37:30 -0500657 printf("Skipping %s for failure retrieving kernel\n",
658 label->name);
Rob Herring500f3042012-12-02 21:00:22 -0600659 return 1;
Jason Hobbs06283a62011-08-31 10:37:30 -0500660 }
661
Rob Herring98f64672012-12-02 21:00:29 -0600662 if (label->ipappend & 0x1) {
663 sprintf(ip_str, " ip=%s:%s:%s:%s",
Simon Glass00caae62017-08-03 12:22:12 -0600664 env_get("ipaddr"), env_get("serverip"),
665 env_get("gatewayip"), env_get("netmask"));
Rob Herring98f64672012-12-02 21:00:29 -0600666 }
667
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700668#ifdef CONFIG_CMD_NET
Rob Herring98f64672012-12-02 21:00:29 -0600669 if (label->ipappend & 0x2) {
670 int err;
671 strcpy(mac_str, " BOOTIF=");
672 err = format_mac_pxe(mac_str + 8, sizeof(mac_str) - 8);
673 if (err < 0)
674 mac_str[0] = '\0';
Rob Herring98f64672012-12-02 21:00:29 -0600675 }
Stephen Warrenb81fdb02014-02-05 20:49:20 -0700676#endif
Rob Herring98f64672012-12-02 21:00:29 -0600677
Hans de Goedeb1ba62d2014-08-06 09:37:39 +0200678 if ((label->ipappend & 0x3) || label->append) {
679 char bootargs[CONFIG_SYS_CBSIZE] = "";
680 char finalbootargs[CONFIG_SYS_CBSIZE];
Rob Herring98f64672012-12-02 21:00:29 -0600681
Ian Campbell64a0c242014-10-03 14:29:01 +0100682 if (strlen(label->append ?: "") +
683 strlen(ip_str) + strlen(mac_str) + 1 > sizeof(bootargs)) {
684 printf("bootarg overflow %zd+%zd+%zd+1 > %zd\n",
685 strlen(label->append ?: ""),
686 strlen(ip_str), strlen(mac_str),
687 sizeof(bootargs));
688 return 1;
689 }
690
Rob Herring98f64672012-12-02 21:00:29 -0600691 if (label->append)
692 strcpy(bootargs, label->append);
693 strcat(bootargs, ip_str);
694 strcat(bootargs, mac_str);
695
Hans de Goedeb1ba62d2014-08-06 09:37:39 +0200696 cli_simple_process_macros(bootargs, finalbootargs);
Simon Glass382bee52017-08-03 12:22:09 -0600697 env_set("bootargs", finalbootargs);
Hans de Goedeb1ba62d2014-08-06 09:37:39 +0200698 printf("append: %s\n", finalbootargs);
Rob Herring32d2ffe2012-12-02 21:00:26 -0600699 }
Jason Hobbs06283a62011-08-31 10:37:30 -0500700
Simon Glass00caae62017-08-03 12:22:12 -0600701 bootm_argv[1] = env_get("kernel_addr_r");
Jason Hobbs06283a62011-08-31 10:37:30 -0500702
703 /*
Chander Kashyapa6559382012-09-06 19:36:31 +0000704 * fdt usage is optional:
705 * It handles the following scenarios. All scenarios are exclusive
706 *
707 * Scenario 1: If fdt_addr_r specified and "fdt" label is defined in
708 * pxe file, retrieve fdt blob from server. Pass fdt_addr_r to bootm,
709 * and adjust argc appropriately.
710 *
711 * Scenario 2: If there is an fdt_addr specified, pass it along to
712 * bootm, and adjust argc appropriately.
713 *
714 * Scenario 3: fdt blob is not available.
Jason Hobbs06283a62011-08-31 10:37:30 -0500715 */
Simon Glass00caae62017-08-03 12:22:12 -0600716 bootm_argv[3] = env_get("fdt_addr_r");
Chander Kashyapa6559382012-09-06 19:36:31 +0000717
718 /* if fdt label is defined then get fdt from server */
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700719 if (bootm_argv[3]) {
720 char *fdtfile = NULL;
721 char *fdtfilefree = NULL;
722
723 if (label->fdt) {
724 fdtfile = label->fdt;
725 } else if (label->fdtdir) {
Stephen Warrene22361a2014-02-12 14:30:04 -0700726 char *f1, *f2, *f3, *f4, *slash;
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700727
Simon Glass00caae62017-08-03 12:22:12 -0600728 f1 = env_get("fdtfile");
Stephen Warrene22361a2014-02-12 14:30:04 -0700729 if (f1) {
730 f2 = "";
731 f3 = "";
732 f4 = "";
733 } else {
734 /*
735 * For complex cases where this code doesn't
736 * generate the correct filename, the board
737 * code should set $fdtfile during early boot,
738 * or the boot scripts should set $fdtfile
739 * before invoking "pxe" or "sysboot".
740 */
Simon Glass00caae62017-08-03 12:22:12 -0600741 f1 = env_get("soc");
Stephen Warrene22361a2014-02-12 14:30:04 -0700742 f2 = "-";
Simon Glass00caae62017-08-03 12:22:12 -0600743 f3 = env_get("board");
Stephen Warrene22361a2014-02-12 14:30:04 -0700744 f4 = ".dtb";
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700745 }
Stephen Warrene22361a2014-02-12 14:30:04 -0700746
747 len = strlen(label->fdtdir);
748 if (!len)
749 slash = "./";
750 else if (label->fdtdir[len - 1] != '/')
751 slash = "/";
752 else
753 slash = "";
754
755 len = strlen(label->fdtdir) + strlen(slash) +
756 strlen(f1) + strlen(f2) + strlen(f3) +
757 strlen(f4) + 1;
758 fdtfilefree = malloc(len);
759 if (!fdtfilefree) {
760 printf("malloc fail (FDT filename)\n");
761 return 1;
762 }
763
764 snprintf(fdtfilefree, len, "%s%s%s%s%s%s",
765 label->fdtdir, slash, f1, f2, f3, f4);
766 fdtfile = fdtfilefree;
Chander Kashyapa6559382012-09-06 19:36:31 +0000767 }
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700768
769 if (fdtfile) {
770 int err = get_relfile_envaddr(cmdtp, fdtfile, "fdt_addr_r");
771 free(fdtfilefree);
772 if (err < 0) {
773 printf("Skipping %s for failure retrieving fdt\n",
774 label->name);
775 return 1;
776 }
777 } else {
778 bootm_argv[3] = NULL;
779 }
780 }
781
782 if (!bootm_argv[3])
Simon Glass00caae62017-08-03 12:22:12 -0600783 bootm_argv[3] = env_get("fdt_addr");
Jason Hobbs06283a62011-08-31 10:37:30 -0500784
York Sunf63963f2016-09-01 16:28:21 +0800785 if (bootm_argv[3]) {
786 if (!bootm_argv[2])
787 bootm_argv[2] = "-";
Jason Hobbs06283a62011-08-31 10:37:30 -0500788 bootm_argc = 4;
York Sunf63963f2016-09-01 16:28:21 +0800789 }
Jason Hobbs06283a62011-08-31 10:37:30 -0500790
Bryan Wu1fb7d0e2014-07-31 17:39:59 -0700791 kernel_addr = genimg_get_kernel_addr(bootm_argv[1]);
792 buf = map_sysmem(kernel_addr, 0);
793 /* Try bootm for legacy and FIT format image */
794 if (genimg_get_format(buf) != IMAGE_FORMAT_INVALID)
795 do_bootm(cmdtp, 0, bootm_argc, bootm_argv);
Stephen Warren8b5c7382015-07-21 17:49:41 -0600796#ifdef CONFIG_CMD_BOOTI
797 /* Try booting an AArch64 Linux kernel image */
798 else
799 do_booti(cmdtp, 0, bootm_argc, bootm_argv);
800#elif defined(CONFIG_CMD_BOOTZ)
801 /* Try booting a Image */
Bryan Wu1fb7d0e2014-07-31 17:39:59 -0700802 else
803 do_bootz(cmdtp, 0, bootm_argc, bootm_argv);
Rob Herringe6b6ccf2012-12-03 13:17:21 -0600804#endif
Sjoerd Simons4a0bd102015-04-13 22:54:25 +0200805 unmap_sysmem(buf);
Rob Herring500f3042012-12-02 21:00:22 -0600806 return 1;
Jason Hobbs06283a62011-08-31 10:37:30 -0500807}
808
809/*
810 * Tokens for the pxe file parser.
811 */
812enum token_type {
813 T_EOL,
814 T_STRING,
815 T_EOF,
816 T_MENU,
817 T_TITLE,
818 T_TIMEOUT,
819 T_LABEL,
820 T_KERNEL,
Rob Herringbeb9f6c2012-03-28 05:51:35 +0000821 T_LINUX,
Jason Hobbs06283a62011-08-31 10:37:30 -0500822 T_APPEND,
823 T_INITRD,
824 T_LOCALBOOT,
825 T_DEFAULT,
826 T_PROMPT,
827 T_INCLUDE,
Chander Kashyapa6559382012-09-06 19:36:31 +0000828 T_FDT,
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700829 T_FDTDIR,
Rob Herring8577fec2012-12-02 21:00:27 -0600830 T_ONTIMEOUT,
Rob Herring98f64672012-12-02 21:00:29 -0600831 T_IPAPPEND,
Jason Hobbs06283a62011-08-31 10:37:30 -0500832 T_INVALID
833};
834
835/*
836 * A token - given by a value and a type.
837 */
838struct token {
839 char *val;
840 enum token_type type;
841};
842
843/*
844 * Keywords recognized.
845 */
846static const struct token keywords[] = {
847 {"menu", T_MENU},
848 {"title", T_TITLE},
849 {"timeout", T_TIMEOUT},
850 {"default", T_DEFAULT},
851 {"prompt", T_PROMPT},
852 {"label", T_LABEL},
853 {"kernel", T_KERNEL},
Rob Herringbeb9f6c2012-03-28 05:51:35 +0000854 {"linux", T_LINUX},
Jason Hobbs06283a62011-08-31 10:37:30 -0500855 {"localboot", T_LOCALBOOT},
856 {"append", T_APPEND},
857 {"initrd", T_INITRD},
858 {"include", T_INCLUDE},
Stephen Warrenf43c4012014-01-28 14:50:09 -0700859 {"devicetree", T_FDT},
Chander Kashyapa6559382012-09-06 19:36:31 +0000860 {"fdt", T_FDT},
Stephen Warrenc61d94d2014-01-28 14:50:10 -0700861 {"devicetreedir", T_FDTDIR},
862 {"fdtdir", T_FDTDIR},
Rob Herring8577fec2012-12-02 21:00:27 -0600863 {"ontimeout", T_ONTIMEOUT,},
Rob Herring98f64672012-12-02 21:00:29 -0600864 {"ipappend", T_IPAPPEND,},
Jason Hobbs06283a62011-08-31 10:37:30 -0500865 {NULL, T_INVALID}
866};
867
868/*
869 * Since pxe(linux) files don't have a token to identify the start of a
870 * literal, we have to keep track of when we're in a state where a literal is
871 * expected vs when we're in a state a keyword is expected.
872 */
873enum lex_state {
874 L_NORMAL = 0,
875 L_KEYWORD,
876 L_SLITERAL
877};
878
879/*
880 * get_string retrieves a string from *p and stores it as a token in
881 * *t.
882 *
883 * get_string used for scanning both string literals and keywords.
884 *
885 * Characters from *p are copied into t-val until a character equal to
886 * delim is found, or a NUL byte is reached. If delim has the special value of
887 * ' ', any whitespace character will be used as a delimiter.
888 *
889 * If lower is unequal to 0, uppercase characters will be converted to
890 * lowercase in the result. This is useful to make keywords case
891 * insensitive.
892 *
893 * The location of *p is updated to point to the first character after the end
894 * of the token - the ending delimiter.
895 *
896 * On success, the new value of t->val is returned. Memory for t->val is
897 * allocated using malloc and must be free()'d to reclaim it. If insufficient
898 * memory is available, NULL is returned.
899 */
900static char *get_string(char **p, struct token *t, char delim, int lower)
901{
902 char *b, *e;
903 size_t len, i;
904
905 /*
906 * b and e both start at the beginning of the input stream.
907 *
908 * e is incremented until we find the ending delimiter, or a NUL byte
909 * is reached. Then, we take e - b to find the length of the token.
910 */
911 b = e = *p;
912
913 while (*e) {
914 if ((delim == ' ' && isspace(*e)) || delim == *e)
915 break;
916 e++;
917 }
918
919 len = e - b;
920
921 /*
922 * Allocate memory to hold the string, and copy it in, converting
923 * characters to lowercase if lower is != 0.
924 */
925 t->val = malloc(len + 1);
926 if (!t->val)
927 return NULL;
928
929 for (i = 0; i < len; i++, b++) {
930 if (lower)
931 t->val[i] = tolower(*b);
932 else
933 t->val[i] = *b;
934 }
935
936 t->val[len] = '\0';
937
938 /*
939 * Update *p so the caller knows where to continue scanning.
940 */
941 *p = e;
942
943 t->type = T_STRING;
944
945 return t->val;
946}
947
948/*
949 * Populate a keyword token with a type and value.
950 */
951static void get_keyword(struct token *t)
952{
953 int i;
954
955 for (i = 0; keywords[i].val; i++) {
956 if (!strcmp(t->val, keywords[i].val)) {
957 t->type = keywords[i].type;
958 break;
959 }
960 }
961}
962
963/*
964 * Get the next token. We have to keep track of which state we're in to know
965 * if we're looking to get a string literal or a keyword.
966 *
967 * *p is updated to point at the first character after the current token.
968 */
969static void get_token(char **p, struct token *t, enum lex_state state)
970{
971 char *c = *p;
972
973 t->type = T_INVALID;
974
975 /* eat non EOL whitespace */
976 while (isblank(*c))
977 c++;
978
979 /*
980 * eat comments. note that string literals can't begin with #, but
981 * can contain a # after their first character.
982 */
983 if (*c == '#') {
984 while (*c && *c != '\n')
985 c++;
986 }
987
988 if (*c == '\n') {
989 t->type = T_EOL;
990 c++;
991 } else if (*c == '\0') {
992 t->type = T_EOF;
993 c++;
994 } else if (state == L_SLITERAL) {
995 get_string(&c, t, '\n', 0);
996 } else if (state == L_KEYWORD) {
997 /*
998 * when we expect a keyword, we first get the next string
999 * token delimited by whitespace, and then check if it
1000 * matches a keyword in our keyword list. if it does, it's
1001 * converted to a keyword token of the appropriate type, and
1002 * if not, it remains a string token.
1003 */
1004 get_string(&c, t, ' ', 1);
1005 get_keyword(t);
1006 }
1007
1008 *p = c;
1009}
1010
1011/*
1012 * Increment *c until we get to the end of the current line, or EOF.
1013 */
1014static void eol_or_eof(char **c)
1015{
1016 while (**c && **c != '\n')
1017 (*c)++;
1018}
1019
1020/*
1021 * All of these parse_* functions share some common behavior.
1022 *
1023 * They finish with *c pointing after the token they parse, and return 1 on
1024 * success, or < 0 on error.
1025 */
1026
1027/*
1028 * Parse a string literal and store a pointer it at *dst. String literals
1029 * terminate at the end of the line.
1030 */
1031static int parse_sliteral(char **c, char **dst)
1032{
1033 struct token t;
1034 char *s = *c;
1035
1036 get_token(c, &t, L_SLITERAL);
1037
1038 if (t.type != T_STRING) {
1039 printf("Expected string literal: %.*s\n", (int)(*c - s), s);
1040 return -EINVAL;
1041 }
1042
1043 *dst = t.val;
1044
1045 return 1;
1046}
1047
1048/*
1049 * Parse a base 10 (unsigned) integer and store it at *dst.
1050 */
1051static int parse_integer(char **c, int *dst)
1052{
1053 struct token t;
1054 char *s = *c;
Jason Hobbs06283a62011-08-31 10:37:30 -05001055
1056 get_token(c, &t, L_SLITERAL);
1057
1058 if (t.type != T_STRING) {
1059 printf("Expected string: %.*s\n", (int)(*c - s), s);
1060 return -EINVAL;
1061 }
1062
Rob Herring500f3042012-12-02 21:00:22 -06001063 *dst = simple_strtol(t.val, NULL, 10);
Jason Hobbs06283a62011-08-31 10:37:30 -05001064
1065 free(t.val);
1066
1067 return 1;
1068}
1069
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001070static int parse_pxefile_top(cmd_tbl_t *cmdtp, char *p, unsigned long base,
1071 struct pxe_menu *cfg, int nest_level);
Jason Hobbs06283a62011-08-31 10:37:30 -05001072
1073/*
1074 * Parse an include statement, and retrieve and parse the file it mentions.
1075 *
1076 * base should point to a location where it's safe to store the file, and
1077 * nest_level should indicate how many nested includes have occurred. For this
1078 * include, nest_level has already been incremented and doesn't need to be
1079 * incremented here.
1080 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001081static int handle_include(cmd_tbl_t *cmdtp, char **c, unsigned long base,
Jason Hobbs06283a62011-08-31 10:37:30 -05001082 struct pxe_menu *cfg, int nest_level)
1083{
1084 char *include_path;
1085 char *s = *c;
1086 int err;
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001087 char *buf;
1088 int ret;
Jason Hobbs06283a62011-08-31 10:37:30 -05001089
1090 err = parse_sliteral(c, &include_path);
1091
1092 if (err < 0) {
1093 printf("Expected include path: %.*s\n",
1094 (int)(*c - s), s);
1095 return err;
1096 }
1097
Steven Falco0e3f3f82013-10-07 09:51:48 -04001098 err = get_pxe_file(cmdtp, include_path, base);
Jason Hobbs06283a62011-08-31 10:37:30 -05001099
1100 if (err < 0) {
1101 printf("Couldn't retrieve %s\n", include_path);
1102 return err;
1103 }
1104
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001105 buf = map_sysmem(base, 0);
1106 ret = parse_pxefile_top(cmdtp, buf, base, cfg, nest_level);
1107 unmap_sysmem(buf);
1108
1109 return ret;
Jason Hobbs06283a62011-08-31 10:37:30 -05001110}
1111
1112/*
1113 * Parse lines that begin with 'menu'.
1114 *
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001115 * base and nest are provided to handle the 'menu include' case.
Jason Hobbs06283a62011-08-31 10:37:30 -05001116 *
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001117 * base should point to a location where it's safe to store the included file.
Jason Hobbs06283a62011-08-31 10:37:30 -05001118 *
1119 * nest_level should be 1 when parsing the top level pxe file, 2 when parsing
1120 * a file it includes, 3 when parsing a file included by that file, and so on.
1121 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001122static int parse_menu(cmd_tbl_t *cmdtp, char **c, struct pxe_menu *cfg,
1123 unsigned long base, int nest_level)
Jason Hobbs06283a62011-08-31 10:37:30 -05001124{
1125 struct token t;
1126 char *s = *c;
Heiko Schocher43d4a5e2011-12-12 20:37:17 +00001127 int err = 0;
Jason Hobbs06283a62011-08-31 10:37:30 -05001128
1129 get_token(c, &t, L_KEYWORD);
1130
1131 switch (t.type) {
1132 case T_TITLE:
1133 err = parse_sliteral(c, &cfg->title);
1134
1135 break;
1136
1137 case T_INCLUDE:
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001138 err = handle_include(cmdtp, c, base, cfg,
Jason Hobbs06283a62011-08-31 10:37:30 -05001139 nest_level + 1);
1140 break;
1141
1142 default:
1143 printf("Ignoring malformed menu command: %.*s\n",
1144 (int)(*c - s), s);
1145 }
1146
1147 if (err < 0)
1148 return err;
1149
1150 eol_or_eof(c);
1151
1152 return 1;
1153}
1154
1155/*
1156 * Handles parsing a 'menu line' when we're parsing a label.
1157 */
1158static int parse_label_menu(char **c, struct pxe_menu *cfg,
1159 struct pxe_label *label)
1160{
1161 struct token t;
1162 char *s;
1163
1164 s = *c;
1165
1166 get_token(c, &t, L_KEYWORD);
1167
1168 switch (t.type) {
1169 case T_DEFAULT:
Rob Herring8577fec2012-12-02 21:00:27 -06001170 if (!cfg->default_label)
1171 cfg->default_label = strdup(label->name);
Jason Hobbs06283a62011-08-31 10:37:30 -05001172
1173 if (!cfg->default_label)
1174 return -ENOMEM;
1175
1176 break;
Rob Herring7815c4e2012-03-28 05:51:34 +00001177 case T_LABEL:
1178 parse_sliteral(c, &label->menu);
1179 break;
Jason Hobbs06283a62011-08-31 10:37:30 -05001180 default:
1181 printf("Ignoring malformed menu command: %.*s\n",
1182 (int)(*c - s), s);
1183 }
1184
1185 eol_or_eof(c);
1186
1187 return 0;
1188}
1189
1190/*
1191 * Parses a label and adds it to the list of labels for a menu.
1192 *
1193 * A label ends when we either get to the end of a file, or
1194 * get some input we otherwise don't have a handler defined
1195 * for.
1196 *
1197 */
1198static int parse_label(char **c, struct pxe_menu *cfg)
1199{
1200 struct token t;
Rob Herring34bd23e2012-03-28 05:51:37 +00001201 int len;
Jason Hobbs06283a62011-08-31 10:37:30 -05001202 char *s = *c;
1203 struct pxe_label *label;
1204 int err;
1205
1206 label = label_create();
1207 if (!label)
1208 return -ENOMEM;
1209
1210 err = parse_sliteral(c, &label->name);
1211 if (err < 0) {
1212 printf("Expected label name: %.*s\n", (int)(*c - s), s);
1213 label_destroy(label);
1214 return -EINVAL;
1215 }
1216
1217 list_add_tail(&label->list, &cfg->labels);
1218
1219 while (1) {
1220 s = *c;
1221 get_token(c, &t, L_KEYWORD);
1222
1223 err = 0;
1224 switch (t.type) {
1225 case T_MENU:
1226 err = parse_label_menu(c, cfg, label);
1227 break;
1228
1229 case T_KERNEL:
Rob Herringbeb9f6c2012-03-28 05:51:35 +00001230 case T_LINUX:
Jason Hobbs06283a62011-08-31 10:37:30 -05001231 err = parse_sliteral(c, &label->kernel);
1232 break;
1233
1234 case T_APPEND:
1235 err = parse_sliteral(c, &label->append);
Rob Herring34bd23e2012-03-28 05:51:37 +00001236 if (label->initrd)
1237 break;
1238 s = strstr(label->append, "initrd=");
1239 if (!s)
1240 break;
1241 s += 7;
1242 len = (int)(strchr(s, ' ') - s);
1243 label->initrd = malloc(len + 1);
1244 strncpy(label->initrd, s, len);
1245 label->initrd[len] = '\0';
1246
Jason Hobbs06283a62011-08-31 10:37:30 -05001247 break;
1248
1249 case T_INITRD:
Rob Herring34bd23e2012-03-28 05:51:37 +00001250 if (!label->initrd)
1251 err = parse_sliteral(c, &label->initrd);
Jason Hobbs06283a62011-08-31 10:37:30 -05001252 break;
1253
Chander Kashyapa6559382012-09-06 19:36:31 +00001254 case T_FDT:
1255 if (!label->fdt)
1256 err = parse_sliteral(c, &label->fdt);
1257 break;
1258
Stephen Warrenc61d94d2014-01-28 14:50:10 -07001259 case T_FDTDIR:
1260 if (!label->fdtdir)
1261 err = parse_sliteral(c, &label->fdtdir);
1262 break;
1263
Jason Hobbs06283a62011-08-31 10:37:30 -05001264 case T_LOCALBOOT:
Rob Herring500f3042012-12-02 21:00:22 -06001265 label->localboot = 1;
1266 err = parse_integer(c, &label->localboot_val);
Jason Hobbs06283a62011-08-31 10:37:30 -05001267 break;
1268
Rob Herring98f64672012-12-02 21:00:29 -06001269 case T_IPAPPEND:
1270 err = parse_integer(c, &label->ipappend);
1271 break;
1272
Jason Hobbs06283a62011-08-31 10:37:30 -05001273 case T_EOL:
1274 break;
1275
1276 default:
1277 /*
1278 * put the token back! we don't want it - it's the end
1279 * of a label and whatever token this is, it's
1280 * something for the menu level context to handle.
1281 */
1282 *c = s;
1283 return 1;
1284 }
1285
1286 if (err < 0)
1287 return err;
1288 }
1289}
1290
1291/*
1292 * This 16 comes from the limit pxelinux imposes on nested includes.
1293 *
1294 * There is no reason at all we couldn't do more, but some limit helps prevent
1295 * infinite (until crash occurs) recursion if a file tries to include itself.
1296 */
1297#define MAX_NEST_LEVEL 16
1298
1299/*
1300 * Entry point for parsing a menu file. nest_level indicates how many times
1301 * we've nested in includes. It will be 1 for the top level menu file.
1302 *
1303 * Returns 1 on success, < 0 on error.
1304 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001305static int parse_pxefile_top(cmd_tbl_t *cmdtp, char *p, unsigned long base,
1306 struct pxe_menu *cfg, int nest_level)
Jason Hobbs06283a62011-08-31 10:37:30 -05001307{
1308 struct token t;
1309 char *s, *b, *label_name;
1310 int err;
1311
1312 b = p;
1313
1314 if (nest_level > MAX_NEST_LEVEL) {
1315 printf("Maximum nesting (%d) exceeded\n", MAX_NEST_LEVEL);
1316 return -EMLINK;
1317 }
1318
1319 while (1) {
1320 s = p;
1321
1322 get_token(&p, &t, L_KEYWORD);
1323
1324 err = 0;
1325 switch (t.type) {
1326 case T_MENU:
Rob Herringe82eeb52012-12-02 21:00:25 -06001327 cfg->prompt = 1;
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001328 err = parse_menu(cmdtp, &p, cfg,
1329 base + ALIGN(strlen(b) + 1, 4),
1330 nest_level);
Jason Hobbs06283a62011-08-31 10:37:30 -05001331 break;
1332
1333 case T_TIMEOUT:
1334 err = parse_integer(&p, &cfg->timeout);
1335 break;
1336
1337 case T_LABEL:
1338 err = parse_label(&p, cfg);
1339 break;
1340
1341 case T_DEFAULT:
Rob Herring8577fec2012-12-02 21:00:27 -06001342 case T_ONTIMEOUT:
Jason Hobbs06283a62011-08-31 10:37:30 -05001343 err = parse_sliteral(&p, &label_name);
1344
1345 if (label_name) {
1346 if (cfg->default_label)
1347 free(cfg->default_label);
1348
1349 cfg->default_label = label_name;
1350 }
1351
1352 break;
1353
Rob Herring1e085222012-05-25 10:43:16 +00001354 case T_INCLUDE:
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001355 err = handle_include(cmdtp, &p,
1356 base + ALIGN(strlen(b), 4), cfg,
1357 nest_level + 1);
Rob Herring1e085222012-05-25 10:43:16 +00001358 break;
1359
Jason Hobbs06283a62011-08-31 10:37:30 -05001360 case T_PROMPT:
Rob Herringe82eeb52012-12-02 21:00:25 -06001361 eol_or_eof(&p);
Jason Hobbs06283a62011-08-31 10:37:30 -05001362 break;
1363
1364 case T_EOL:
1365 break;
1366
1367 case T_EOF:
1368 return 1;
1369
1370 default:
1371 printf("Ignoring unknown command: %.*s\n",
1372 (int)(p - s), s);
1373 eol_or_eof(&p);
1374 }
1375
1376 if (err < 0)
1377 return err;
1378 }
1379}
1380
1381/*
1382 * Free the memory used by a pxe_menu and its labels.
1383 */
1384static void destroy_pxe_menu(struct pxe_menu *cfg)
1385{
1386 struct list_head *pos, *n;
1387 struct pxe_label *label;
1388
1389 if (cfg->title)
1390 free(cfg->title);
1391
1392 if (cfg->default_label)
1393 free(cfg->default_label);
1394
1395 list_for_each_safe(pos, n, &cfg->labels) {
1396 label = list_entry(pos, struct pxe_label, list);
1397
1398 label_destroy(label);
1399 }
1400
1401 free(cfg);
1402}
1403
1404/*
1405 * Entry point for parsing a pxe file. This is only used for the top level
1406 * file.
1407 *
1408 * Returns NULL if there is an error, otherwise, returns a pointer to a
1409 * pxe_menu struct populated with the results of parsing the pxe file (and any
1410 * files it includes). The resulting pxe_menu struct can be free()'d by using
1411 * the destroy_pxe_menu() function.
1412 */
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001413static struct pxe_menu *parse_pxefile(cmd_tbl_t *cmdtp, unsigned long menucfg)
Jason Hobbs06283a62011-08-31 10:37:30 -05001414{
1415 struct pxe_menu *cfg;
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001416 char *buf;
1417 int r;
Jason Hobbs06283a62011-08-31 10:37:30 -05001418
1419 cfg = malloc(sizeof(struct pxe_menu));
1420
1421 if (!cfg)
1422 return NULL;
1423
1424 memset(cfg, 0, sizeof(struct pxe_menu));
1425
1426 INIT_LIST_HEAD(&cfg->labels);
1427
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001428 buf = map_sysmem(menucfg, 0);
1429 r = parse_pxefile_top(cmdtp, buf, menucfg, cfg, 1);
1430 unmap_sysmem(buf);
1431
1432 if (r < 0) {
Jason Hobbs06283a62011-08-31 10:37:30 -05001433 destroy_pxe_menu(cfg);
1434 return NULL;
1435 }
1436
1437 return cfg;
1438}
1439
1440/*
Bin Menga1875592016-02-05 19:30:11 -08001441 * Converts a pxe_menu struct into a menu struct for use with U-Boot's generic
Jason Hobbs06283a62011-08-31 10:37:30 -05001442 * menu code.
1443 */
1444static struct menu *pxe_menu_to_menu(struct pxe_menu *cfg)
1445{
1446 struct pxe_label *label;
1447 struct list_head *pos;
1448 struct menu *m;
1449 int err;
Rob Herring32d2ffe2012-12-02 21:00:26 -06001450 int i = 1;
1451 char *default_num = NULL;
Jason Hobbs06283a62011-08-31 10:37:30 -05001452
1453 /*
1454 * Create a menu and add items for all the labels.
1455 */
Pali Rohárfc9d64f2013-03-23 14:50:40 +00001456 m = menu_create(cfg->title, cfg->timeout, cfg->prompt, label_print,
1457 NULL, NULL);
Jason Hobbs06283a62011-08-31 10:37:30 -05001458
1459 if (!m)
1460 return NULL;
1461
1462 list_for_each(pos, &cfg->labels) {
1463 label = list_entry(pos, struct pxe_label, list);
1464
Rob Herring32d2ffe2012-12-02 21:00:26 -06001465 sprintf(label->num, "%d", i++);
1466 if (menu_item_add(m, label->num, label) != 1) {
Jason Hobbs06283a62011-08-31 10:37:30 -05001467 menu_destroy(m);
1468 return NULL;
1469 }
Rob Herring32d2ffe2012-12-02 21:00:26 -06001470 if (cfg->default_label &&
Rob Herring8577fec2012-12-02 21:00:27 -06001471 (strcmp(label->name, cfg->default_label) == 0))
Rob Herring32d2ffe2012-12-02 21:00:26 -06001472 default_num = label->num;
1473
Jason Hobbs06283a62011-08-31 10:37:30 -05001474 }
1475
1476 /*
1477 * After we've created items for each label in the menu, set the
1478 * menu's default label if one was specified.
1479 */
Rob Herring32d2ffe2012-12-02 21:00:26 -06001480 if (default_num) {
1481 err = menu_default_set(m, default_num);
Jason Hobbs06283a62011-08-31 10:37:30 -05001482 if (err != 1) {
1483 if (err != -ENOENT) {
1484 menu_destroy(m);
1485 return NULL;
1486 }
1487
1488 printf("Missing default: %s\n", cfg->default_label);
1489 }
1490 }
1491
1492 return m;
1493}
1494
1495/*
1496 * Try to boot any labels we have yet to attempt to boot.
1497 */
Tom Rinid7884e02013-09-24 09:05:08 -04001498static void boot_unattempted_labels(cmd_tbl_t *cmdtp, struct pxe_menu *cfg)
Jason Hobbs06283a62011-08-31 10:37:30 -05001499{
1500 struct list_head *pos;
1501 struct pxe_label *label;
1502
1503 list_for_each(pos, &cfg->labels) {
1504 label = list_entry(pos, struct pxe_label, list);
1505
1506 if (!label->attempted)
Tom Rinid7884e02013-09-24 09:05:08 -04001507 label_boot(cmdtp, label);
Jason Hobbs06283a62011-08-31 10:37:30 -05001508 }
1509}
1510
1511/*
1512 * Boot the system as prescribed by a pxe_menu.
1513 *
1514 * Use the menu system to either get the user's choice or the default, based
1515 * on config or user input. If there is no default or user's choice,
1516 * attempted to boot labels in the order they were given in pxe files.
1517 * If the default or user's choice fails to boot, attempt to boot other
1518 * labels in the order they were given in pxe files.
1519 *
1520 * If this function returns, there weren't any labels that successfully
1521 * booted, or the user interrupted the menu selection via ctrl+c.
1522 */
Tom Rinid7884e02013-09-24 09:05:08 -04001523static void handle_pxe_menu(cmd_tbl_t *cmdtp, struct pxe_menu *cfg)
Jason Hobbs06283a62011-08-31 10:37:30 -05001524{
1525 void *choice;
1526 struct menu *m;
1527 int err;
1528
1529 m = pxe_menu_to_menu(cfg);
1530 if (!m)
1531 return;
1532
1533 err = menu_get_choice(m, &choice);
1534
1535 menu_destroy(m);
1536
Jason Hobbs6f40f272011-11-07 03:07:15 +00001537 /*
1538 * err == 1 means we got a choice back from menu_get_choice.
1539 *
1540 * err == -ENOENT if the menu was setup to select the default but no
1541 * default was set. in that case, we should continue trying to boot
1542 * labels that haven't been attempted yet.
1543 *
1544 * otherwise, the user interrupted or there was some other error and
1545 * we give up.
1546 */
Jason Hobbs06283a62011-08-31 10:37:30 -05001547
Rob Herring500f3042012-12-02 21:00:22 -06001548 if (err == 1) {
Tom Rinid7884e02013-09-24 09:05:08 -04001549 err = label_boot(cmdtp, choice);
Rob Herring500f3042012-12-02 21:00:22 -06001550 if (!err)
1551 return;
1552 } else if (err != -ENOENT) {
Jason Hobbs6f40f272011-11-07 03:07:15 +00001553 return;
Rob Herring500f3042012-12-02 21:00:22 -06001554 }
Jason Hobbs06283a62011-08-31 10:37:30 -05001555
Tom Rinid7884e02013-09-24 09:05:08 -04001556 boot_unattempted_labels(cmdtp, cfg);
Jason Hobbs06283a62011-08-31 10:37:30 -05001557}
1558
Stephen Warrenb81fdb02014-02-05 20:49:20 -07001559#ifdef CONFIG_CMD_NET
Jason Hobbs06283a62011-08-31 10:37:30 -05001560/*
1561 * Boots a system using a pxe file
1562 *
1563 * Returns 0 on success, 1 on error.
1564 */
1565static int
1566do_pxe_boot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
1567{
1568 unsigned long pxefile_addr_r;
1569 struct pxe_menu *cfg;
1570 char *pxefile_addr_str;
1571
Rob Herring669df7e2012-05-25 10:47:39 +00001572 do_getfile = do_get_tftp;
1573
Jason Hobbs06283a62011-08-31 10:37:30 -05001574 if (argc == 1) {
1575 pxefile_addr_str = from_env("pxefile_addr_r");
1576 if (!pxefile_addr_str)
1577 return 1;
1578
1579 } else if (argc == 2) {
1580 pxefile_addr_str = argv[1];
1581 } else {
Simon Glass4c12eeb2011-12-10 08:44:01 +00001582 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -05001583 }
1584
1585 if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
1586 printf("Invalid pxefile address: %s\n", pxefile_addr_str);
1587 return 1;
1588 }
1589
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001590 cfg = parse_pxefile(cmdtp, pxefile_addr_r);
Jason Hobbs06283a62011-08-31 10:37:30 -05001591
1592 if (cfg == NULL) {
1593 printf("Error parsing config file\n");
1594 return 1;
1595 }
1596
Tom Rinid7884e02013-09-24 09:05:08 -04001597 handle_pxe_menu(cmdtp, cfg);
Jason Hobbs06283a62011-08-31 10:37:30 -05001598
1599 destroy_pxe_menu(cfg);
1600
Joe Hershberger14111572015-04-08 01:41:02 -05001601 copy_filename(net_boot_file_name, "", sizeof(net_boot_file_name));
Stephen Warrended2e202014-07-22 18:06:46 -06001602
Jason Hobbs06283a62011-08-31 10:37:30 -05001603 return 0;
1604}
1605
1606static cmd_tbl_t cmd_pxe_sub[] = {
1607 U_BOOT_CMD_MKENT(get, 1, 1, do_pxe_get, "", ""),
1608 U_BOOT_CMD_MKENT(boot, 2, 1, do_pxe_boot, "", "")
1609};
1610
Jeroen Hofstee0e350f82014-06-23 00:22:08 +02001611static int do_pxe(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
Jason Hobbs06283a62011-08-31 10:37:30 -05001612{
1613 cmd_tbl_t *cp;
1614
1615 if (argc < 2)
Simon Glass4c12eeb2011-12-10 08:44:01 +00001616 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -05001617
Rob Herringe5a9a402013-10-18 13:04:42 -05001618 is_pxe = true;
1619
Jason Hobbs06283a62011-08-31 10:37:30 -05001620 /* drop initial "pxe" arg */
1621 argc--;
1622 argv++;
1623
1624 cp = find_cmd_tbl(argv[0], cmd_pxe_sub, ARRAY_SIZE(cmd_pxe_sub));
1625
1626 if (cp)
1627 return cp->cmd(cmdtp, flag, argc, argv);
1628
Simon Glass4c12eeb2011-12-10 08:44:01 +00001629 return CMD_RET_USAGE;
Jason Hobbs06283a62011-08-31 10:37:30 -05001630}
1631
1632U_BOOT_CMD(
1633 pxe, 3, 1, do_pxe,
1634 "commands to get and boot from pxe files",
1635 "get - try to retrieve a pxe file using tftp\npxe "
1636 "boot [pxefile_addr_r] - boot from the pxe file at pxefile_addr_r\n"
1637);
Stephen Warrenb81fdb02014-02-05 20:49:20 -07001638#endif
Rob Herring669df7e2012-05-25 10:47:39 +00001639
1640/*
1641 * Boots a system using a local disk syslinux/extlinux file
1642 *
1643 * Returns 0 on success, 1 on error.
1644 */
Jeroen Hofstee0e350f82014-06-23 00:22:08 +02001645static int do_sysboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
Rob Herring669df7e2012-05-25 10:47:39 +00001646{
1647 unsigned long pxefile_addr_r;
1648 struct pxe_menu *cfg;
1649 char *pxefile_addr_str;
1650 char *filename;
1651 int prompt = 0;
1652
Rob Herringe5a9a402013-10-18 13:04:42 -05001653 is_pxe = false;
1654
Tuomas Tynkkynen0ece6b52015-05-07 21:29:18 +03001655 if (argc > 1 && strstr(argv[1], "-p")) {
Rob Herring669df7e2012-05-25 10:47:39 +00001656 prompt = 1;
1657 argc--;
1658 argv++;
1659 }
1660
1661 if (argc < 4)
1662 return cmd_usage(cmdtp);
1663
1664 if (argc < 5) {
1665 pxefile_addr_str = from_env("pxefile_addr_r");
1666 if (!pxefile_addr_str)
1667 return 1;
1668 } else {
1669 pxefile_addr_str = argv[4];
1670 }
1671
1672 if (argc < 6)
Simon Glass00caae62017-08-03 12:22:12 -06001673 filename = env_get("bootfile");
Rob Herring669df7e2012-05-25 10:47:39 +00001674 else {
1675 filename = argv[5];
Simon Glass382bee52017-08-03 12:22:09 -06001676 env_set("bootfile", filename);
Rob Herring669df7e2012-05-25 10:47:39 +00001677 }
1678
1679 if (strstr(argv[3], "ext2"))
1680 do_getfile = do_get_ext2;
1681 else if (strstr(argv[3], "fat"))
1682 do_getfile = do_get_fat;
Dennis Gilmore6d1a3e52014-02-04 05:25:46 -06001683 else if (strstr(argv[3], "any"))
1684 do_getfile = do_get_any;
Rob Herring669df7e2012-05-25 10:47:39 +00001685 else {
1686 printf("Invalid filesystem: %s\n", argv[3]);
1687 return 1;
1688 }
1689 fs_argv[1] = argv[1];
1690 fs_argv[2] = argv[2];
1691
1692 if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
1693 printf("Invalid pxefile address: %s\n", pxefile_addr_str);
1694 return 1;
1695 }
1696
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001697 if (get_pxe_file(cmdtp, filename, pxefile_addr_r) < 0) {
Rob Herring669df7e2012-05-25 10:47:39 +00001698 printf("Error reading config file\n");
1699 return 1;
1700 }
1701
Sjoerd Simons4a0bd102015-04-13 22:54:25 +02001702 cfg = parse_pxefile(cmdtp, pxefile_addr_r);
Rob Herring669df7e2012-05-25 10:47:39 +00001703
1704 if (cfg == NULL) {
1705 printf("Error parsing config file\n");
1706 return 1;
1707 }
1708
1709 if (prompt)
1710 cfg->prompt = 1;
1711
Tom Rinid7884e02013-09-24 09:05:08 -04001712 handle_pxe_menu(cmdtp, cfg);
Rob Herring669df7e2012-05-25 10:47:39 +00001713
1714 destroy_pxe_menu(cfg);
1715
1716 return 0;
1717}
1718
1719U_BOOT_CMD(
1720 sysboot, 7, 1, do_sysboot,
1721 "command to get and boot from syslinux files",
Dennis Gilmore6d1a3e52014-02-04 05:25:46 -06001722 "[-p] <interface> <dev[:part]> <ext2|fat|any> [addr] [filename]\n"
1723 " - load and parse syslinux menu file 'filename' from ext2, fat\n"
1724 " or any filesystem on 'dev' on 'interface' to address 'addr'"
Rob Herring669df7e2012-05-25 10:47:39 +00001725);