ui/ncurses: Interpret left/right keys as up/down
[petitboot] / discover / pxe-parser.c
1
2 #if defined(HAVE_CONFIG_H)
3 #include "config.h"
4 #endif
5 #include <stdlib.h>
6 #include <string.h>
7
8 #include <talloc/talloc.h>
9 #include <url/url.h>
10 #include <log/log.h>
11
12 #include "parser.h"
13 #include "parser-conf.h"
14 #include "parser-utils.h"
15 #include "resource.h"
16 #include "paths.h"
17 #include "event.h"
18 #include "user-event.h"
19 #include "network.h"
20
21 static const char *pxelinux_prefix = "pxelinux.cfg/";
22
23 struct pxe_parser_info {
24         struct discover_boot_option *opt;
25         const char *default_name;
26 };
27
28 static void pxe_finish(struct conf_context *conf)
29 {
30         struct pxe_parser_info *info = conf->parser_info;
31         if (info->opt)
32                 discover_context_add_boot_option(conf->dc, info->opt);
33 }
34
35 /* We need a slightly modified version of pb_url_join, to allow for the
36  * pxelinux "::filename" syntax for absolute URLs
37  */
38 static struct pb_url *pxe_url_join(void *ctx, const struct pb_url *url,
39                 const char *s)
40 {
41         struct pb_url *new_url;
42         int len;
43
44         len = strlen(s);
45
46         if (len > 2 && s[0] == ':' && s[1] == ':') {
47                 char *tmp;
48
49                 if (s[2] == '/') {
50                         /* ::/path -> /path */
51                         tmp = talloc_strdup(ctx, s+2);
52                 } else {
53                         /* ::path -> /path */
54                         tmp = talloc_strdup(ctx, s+1);
55                         tmp[0] = '/';
56                 }
57
58                 new_url = pb_url_join(ctx, url, tmp);
59
60                 talloc_free(tmp);
61
62         } else {
63                 const char *tmp;
64                 /* strip leading slashes */
65                 for (tmp = s; *tmp == '/'; tmp++)
66                         ;
67                 new_url = pb_url_join(ctx, url, tmp);
68         }
69
70         return new_url;
71 }
72
73 static void pxe_append_string(struct discover_boot_option *opt,
74                 const char *str)
75 {
76         if (opt->option->boot_args)
77                 opt->option->boot_args = talloc_asprintf_append(
78                                 opt->option->boot_args, " %s", str);
79         else
80                 opt->option->boot_args = talloc_strdup(opt->option, str);
81 }
82
83 static char *pxe_sysappend_mac(void *ctx, uint8_t *mac)
84 {
85         if (!mac)
86                 return NULL;
87
88         return talloc_asprintf(ctx,
89                 "BOOTIF=01-%02x-%02x-%02x-%02x-%02x-%02x",
90                 mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
91 }
92
93 static void pxe_process_sysappend(struct discover_context *ctx,
94                 struct discover_boot_option *opt,
95                 unsigned long val)
96 {
97         struct event *event = ctx->event;
98         char *str = NULL;
99
100         if (!event)
101                 return;
102
103         if (val & 0x2) {
104                 uint8_t *mac = find_mac_by_name(ctx, ctx->network,
105                                         event->device);
106                 str = pxe_sysappend_mac(ctx, mac);
107                 if (str) {
108                         pxe_append_string(opt, str);
109                         talloc_free(str);
110                 }
111                 val &= ~0x2;
112         }
113
114         if (val)
115                 pb_log("pxe: unsupported features requested in "
116                                 "ipappend/sysappend: 0x%04lx", val);
117
118 }
119
120 static void pxe_process_pair(struct conf_context *ctx,
121                 const char *name, char *value)
122 {
123         struct pxe_parser_info *parser_info = ctx->parser_info;
124         struct discover_boot_option *opt = parser_info->opt;
125         struct pb_url *url;
126
127         /* quirk in the syslinux config format: initrd can be separated
128          * by an '=' */
129         if (!name && !strncasecmp(value, "initrd=", strlen("initrd="))) {
130                 name = "initrd";
131                 value += strlen("initrd=");
132         }
133
134         if (!name)
135                 return;
136
137         if (streq(name, "DEFAULT")) {
138                 parser_info->default_name = talloc_strdup(parser_info, value);
139                 return;
140         }
141
142         if (streq(name, "LABEL")) {
143                 if (opt)
144                         pxe_finish(ctx);
145
146                 opt = discover_boot_option_create(ctx->dc, ctx->dc->device);
147
148                 opt->option->name = talloc_strdup(opt, value);
149                 opt->option->id = talloc_asprintf(opt, "%s@%p",
150                                 ctx->dc->device->device->id, opt);
151
152                 opt->option->is_default = parser_info->default_name &&
153                                         streq(parser_info->default_name, value);
154
155                 parser_info->opt = opt;
156                 return;
157         }
158
159         /* all other parameters need an option */
160         if (!opt)
161                 return;
162
163         if (streq(name, "KERNEL")) {
164                 url = pxe_url_join(ctx->dc, ctx->dc->conf_url, value);
165                 opt->boot_image = create_url_resource(opt, url);
166
167         } else if (streq(name, "INITRD")) {
168                 url = pxe_url_join(ctx->dc, ctx->dc->conf_url, value);
169                 opt->initrd = create_url_resource(opt, url);
170
171         } else if (streq(name, "APPEND")) {
172                 char *str, *end;
173
174                 pxe_append_string(opt, value);
175
176                 str = strcasestr(value, "INITRD=");
177                 if (str) {
178                         str += strlen("INITRD=");
179                         end = strchrnul(str, ' ');
180                         *end = '\0';
181
182                         url = pxe_url_join(ctx->dc, ctx->dc->conf_url, str);
183                         opt->initrd = create_url_resource(opt, url);
184                 }
185         } else if (streq(name, "SYSAPPEND") || streq(name, "IPAPPEND")) {
186                 unsigned long type;
187                 char *end;
188
189                 type = strtoul(value, &end, 10);
190                 if (end != value && !(*end))
191                         pxe_process_sysappend(ctx->dc, opt, type);
192
193         } else if (streq(name, "DTB") || streq(name, "FDT")) {
194                 url = pxe_url_join(ctx->dc, ctx->dc->conf_url, value);
195                 opt->dtb = create_url_resource(opt, url);
196         }
197
198 }
199
200 static int pxe_parse(struct discover_context *dc)
201 {
202         struct pb_url *pxe_base_url, *url;
203         struct pxe_parser_info *parser_info;
204         char **pxe_conf_files, **filename;
205         struct conf_context *conf;
206         bool complete_url;
207         int len, rc;
208         char *buf;
209
210         /* Expects dhcp event parameters to support network boot */
211         if (!dc->event)
212                 return -1;
213
214         conf = talloc_zero(dc, struct conf_context);
215
216         if (!conf)
217                 goto out;
218
219         conf->dc = dc;
220         conf->get_pair = conf_get_pair_space;
221         conf->process_pair = pxe_process_pair;
222         conf->finish = pxe_finish;
223
224         parser_info = talloc_zero(conf, struct pxe_parser_info);
225         conf->parser_info = parser_info;
226
227         dc->conf_url = user_event_parse_conf_url(dc, dc->event, &complete_url);
228         if (!dc->conf_url)
229                 goto out_conf;
230
231         if (complete_url) {
232                 /* we have a complete URL; use this and we're done. */
233                 rc = parser_request_url(dc, dc->conf_url, &buf, &len);
234                 if (rc)
235                         goto out_conf;
236         } else {
237                 pxe_conf_files = user_event_parse_conf_filenames(dc, dc->event);
238                 if (!pxe_conf_files)
239                         goto out_conf;
240
241                 rc = -1;
242
243                 pxe_base_url = pb_url_join(dc, dc->conf_url, pxelinux_prefix);
244                 if (!pxe_base_url)
245                         goto out_pxe_conf;
246
247                 for (filename = pxe_conf_files; *filename; filename++) {
248                         url = pb_url_join(dc, pxe_base_url, *filename);
249                         if (!url)
250                                 continue;
251
252                         rc = parser_request_url(dc, url, &buf, &len);
253                         if (!rc) /* found one, just break */
254                                 break;
255
256                         talloc_free(url);
257                 }
258
259                 talloc_free(pxe_base_url);
260
261                 /* No configuration file found on the boot server */
262                 if (rc)
263                         goto out_pxe_conf;
264
265                 talloc_free(pxe_conf_files);
266         }
267
268         /* Call the config file parser with the data read from the file */
269         conf_parse_buf(conf, buf, len);
270
271         talloc_free(buf);
272         talloc_free(conf);
273
274         return 0;
275
276 out_pxe_conf:
277         talloc_free(pxe_conf_files);
278 out_conf:
279         talloc_free(conf);
280 out:
281         return -1;
282 }
283
284 static struct parser pxe_parser = {
285         .name                   = "pxe",
286         .parse                  = pxe_parse,
287 };
288
289 register_parser(pxe_parser);