2 * Copyright Geoff Levand <geoff@infradead.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; version 2 of the License.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License
14 * along with this program; if not, write to the Free Software
15 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 #if defined(HAVE_CONFIG_H)
29 #include "talloc/talloc.h"
30 #include "types/types.h"
31 #include "parser-conf.h"
32 #include "parser-utils.h"
41 struct discover_boot_option *opt;
46 const char *const *known_names;
47 struct grub2_root *root;
50 struct grub2_resource_info {
51 struct grub2_root *root;
55 /* we use slightly different resources for grub2 */
56 static struct resource *create_grub2_resource(void *ctx,
57 struct discover_device *orig_device,
58 struct grub2_root *root, const char *path)
60 struct grub2_resource_info *info;
63 res = talloc(ctx, struct resource);
66 info = talloc(res, struct grub2_resource_info);
68 talloc_reference(info, root);
69 info->path = talloc_strdup(info, path);
71 res->resolved = false;
75 resolve_resource_against_device(res, orig_device, path);
80 static bool resolve_grub2_resource(struct device_handler *handler,
83 struct grub2_resource_info *info = res->info;
84 struct discover_device *dev;
86 assert(!res->resolved);
88 dev = device_lookup_by_uuid(handler, info->root->uuid);
93 resolve_resource_against_device(res, dev, info->path);
99 static bool current_option_is_default(struct grub2_state *state)
101 if (state->default_idx < 0)
103 return state->cur_idx == state->default_idx;
106 static void grub2_finish(struct conf_context *conf)
108 struct device *dev = conf->dc->device->device;
109 struct grub2_state *state = conf->parser_info;
110 struct boot_option *opt;
112 if (!state->desc_image) {
113 pb_log("%s: %s: no image found\n", __func__, dev->id);
118 opt = state->opt->option;
122 assert(opt->boot_args);
124 opt->description = talloc_asprintf(opt, "%s %s %s",
126 (state->desc_initrd ? state->desc_initrd : ""),
129 talloc_free(state->desc_initrd);
130 state->desc_initrd = NULL;
132 conf_strip_str(opt->boot_args);
133 conf_strip_str(opt->description);
135 state->opt->option->is_default = current_option_is_default(state);
137 discover_context_add_boot_option(conf->dc, state->opt);
143 static void grub2_process_pair(struct conf_context *conf, const char *name,
146 struct device *dev = conf->dc->device->device;
147 struct grub2_state *state = conf->parser_info;
148 struct discover_boot_option *opt = state->opt;
150 if (!name || !conf_param_in_list(state->known_names, name))
153 if (streq(name, "menuentry")) {
154 /* complete any existing option... */
158 /* ... then start the new one */
159 opt = discover_boot_option_create(conf->dc, conf->dc->device);
160 opt->option->boot_args = talloc_strdup(opt->option, "");
162 value = strtok(value, "\'{\"");
164 opt->option->id = talloc_asprintf(opt->option,
165 "%s#%s", dev->id, value);
166 opt->option->name = talloc_strdup(opt->option, value);
167 opt->option->boot_args = talloc_strdup(opt, "");
174 if (streq(name, "linux") || streq(name, "linux16")) {
177 sep = strchr(value, ' ');
182 opt->boot_image = create_grub2_resource(opt, conf->dc->device,
184 state->desc_image = talloc_strdup(opt, value);
187 opt->option->boot_args = talloc_strdup(opt, sep + 1);
192 if (streq(name, "initrd")) {
193 opt->initrd = create_grub2_resource(opt, conf->dc->device,
195 state->desc_initrd = talloc_asprintf(state, "initrd=%s",
200 if (streq(name, "search")) {
201 struct grub2_root *root;
204 if (!strstr(value, "--set=root")) {
205 pb_log("%s: no root\n", __func__);
209 /* The UUID should be the last argument to the search command.
210 * FIXME: this is a little fragile; would be nice to have some
211 * parser helpers to deal with "command args" parsing
213 uuid = strrchr(value, ' ');
220 talloc_unlink(state, state->root);
222 root = talloc(state, struct grub2_root);
223 root->uuid = talloc_strdup(root, uuid);
228 if (streq(name, "set")) {
229 char *sep, *var_name, *var_value;
231 /* this is pretty nasty, but works until we implement a proper
234 sep = strchr(value, '=');
242 if (var_value[0] == '"' || var_value[0] == '\'')
245 if (!strlen(var_name) || !strlen(var_value))
248 if (streq(var_name, "default"))
249 state->default_idx = atoi(var_value);
254 pb_log("%s: unknown name: %s\n", __func__, name);
257 static const char *const grub2_conf_files[] = {
263 "/boot/grub/grub.cfg",
264 "/boot/grub/menu.lst",
270 "/BOOT/GRUB/GRUB.CFG",
271 "/BOOT/GRUB/MENU.LST",
275 static const char *grub2_known_names[] = {
285 static int grub2_parse(struct discover_context *dc, char *buf, int len)
287 struct conf_context *conf;
288 struct grub2_state *state;
290 conf = talloc_zero(dc, struct conf_context);
296 conf_init_global_options(conf);
297 conf->get_pair = conf_get_pair_space;
298 conf->process_pair = grub2_process_pair;
299 conf->finish = grub2_finish;
300 conf->parser_info = state = talloc_zero(conf, struct grub2_state);
302 state->known_names = grub2_known_names;
303 state->default_idx = -1;
305 conf_parse_buf(conf, buf, len);
311 static struct parser grub2_parser = {
313 .method = CONF_METHOD_LOCAL_FILE,
314 .parse = grub2_parse,
315 .filenames = grub2_conf_files,
316 .resolve_resource = resolve_grub2_resource,
319 register_parser(grub2_parser);