pxe: Move do_getfile() into the context
[platform/kernel/u-boot.git] / cmd / sysboot.c
1 // SPDX-License-Identifier: GPL-2.0+
2
3 #include <common.h>
4 #include <command.h>
5 #include <env.h>
6 #include <fs.h>
7 #include "pxe_utils.h"
8
9 static char *fs_argv[5];
10
11 static int do_get_ext2(struct pxe_context *ctx, const char *file_path,
12                        char *file_addr)
13 {
14 #ifdef CONFIG_CMD_EXT2
15         fs_argv[0] = "ext2load";
16         fs_argv[3] = file_addr;
17         fs_argv[4] = (void *)file_path;
18
19         if (!do_ext2load(ctx->cmdtp, 0, 5, fs_argv))
20                 return 1;
21 #endif
22         return -ENOENT;
23 }
24
25 static int do_get_fat(struct pxe_context *ctx, const char *file_path,
26                       char *file_addr)
27 {
28 #ifdef CONFIG_CMD_FAT
29         fs_argv[0] = "fatload";
30         fs_argv[3] = file_addr;
31         fs_argv[4] = (void *)file_path;
32
33         if (!do_fat_fsload(ctx->cmdtp, 0, 5, fs_argv))
34                 return 1;
35 #endif
36         return -ENOENT;
37 }
38
39 static int do_get_any(struct pxe_context *ctx, const char *file_path,
40                       char *file_addr)
41 {
42 #ifdef CONFIG_CMD_FS_GENERIC
43         fs_argv[0] = "load";
44         fs_argv[3] = file_addr;
45         fs_argv[4] = (void *)file_path;
46
47         if (!do_load(ctx->cmdtp, 0, 5, fs_argv, FS_TYPE_ANY))
48                 return 1;
49 #endif
50         return -ENOENT;
51 }
52
53 /*
54  * Boots a system using a local disk syslinux/extlinux file
55  *
56  * Returns 0 on success, 1 on error.
57  */
58 static int do_sysboot(struct cmd_tbl *cmdtp, int flag, int argc,
59                       char *const argv[])
60 {
61         unsigned long pxefile_addr_r;
62         struct pxe_context ctx;
63         struct pxe_menu *cfg;
64         char *pxefile_addr_str;
65         char *filename;
66         int prompt = 0;
67
68         is_pxe = false;
69
70         if (argc > 1 && strstr(argv[1], "-p")) {
71                 prompt = 1;
72                 argc--;
73                 argv++;
74         }
75
76         if (argc < 4)
77                 return cmd_usage(cmdtp);
78
79         if (argc < 5) {
80                 pxefile_addr_str = from_env("pxefile_addr_r");
81                 if (!pxefile_addr_str)
82                         return 1;
83         } else {
84                 pxefile_addr_str = argv[4];
85         }
86
87         if (argc < 6) {
88                 filename = env_get("bootfile");
89         } else {
90                 filename = argv[5];
91                 env_set("bootfile", filename);
92         }
93
94         pxe_setup_ctx(&ctx, cmdtp, NULL);
95         if (strstr(argv[3], "ext2")) {
96                 ctx.getfile = do_get_ext2;
97         } else if (strstr(argv[3], "fat")) {
98                 ctx.getfile = do_get_fat;
99         } else if (strstr(argv[3], "any")) {
100                 ctx.getfile = do_get_any;
101         } else {
102                 printf("Invalid filesystem: %s\n", argv[3]);
103                 return 1;
104         }
105         fs_argv[1] = argv[1];
106         fs_argv[2] = argv[2];
107
108         if (strict_strtoul(pxefile_addr_str, 16, &pxefile_addr_r) < 0) {
109                 printf("Invalid pxefile address: %s\n", pxefile_addr_str);
110                 return 1;
111         }
112
113         if (get_pxe_file(&ctx, filename, pxefile_addr_r) < 0) {
114                 printf("Error reading config file\n");
115                 return 1;
116         }
117
118         cfg = parse_pxefile(&ctx, pxefile_addr_r);
119
120         if (!cfg) {
121                 printf("Error parsing config file\n");
122                 return 1;
123         }
124
125         if (prompt)
126                 cfg->prompt = 1;
127
128         handle_pxe_menu(&ctx, cfg);
129
130         destroy_pxe_menu(cfg);
131
132         return 0;
133 }
134
135 U_BOOT_CMD(sysboot, 7, 1, do_sysboot,
136            "command to get and boot from syslinux files",
137            "[-p] <interface> <dev[:part]> <ext2|fat|any> [addr] [filename]\n"
138            "    - load and parse syslinux menu file 'filename' from ext2, fat\n"
139            "      or any filesystem on 'dev' on 'interface' to address 'addr'"
140 );