#include "types/types.h"
#include "parser-conf.h"
#include "parser-utils.h"
+#include "paths.h"
#include "resource.h"
+struct grub2_root {
+ char *uuid;
+};
+
struct grub2_state {
struct discover_boot_option *opt;
+ int default_idx;
+ int cur_idx;
char *desc_image;
char *desc_initrd;
const char *const *known_names;
+ struct grub2_root *root;
+};
+
+struct grub2_resource_info {
+ struct grub2_root *root;
+ char *path;
};
+/* we use slightly different resources for grub2 */
+static struct resource *create_grub2_resource(void *ctx,
+ struct discover_device *orig_device,
+ struct grub2_root *root, const char *path)
+{
+ struct grub2_resource_info *info;
+ struct resource *res;
+
+ res = talloc(ctx, struct resource);
+
+ if (root) {
+ info = talloc(res, struct grub2_resource_info);
+ info->root = root;
+ talloc_reference(info, root);
+ info->path = talloc_strdup(info, path);
+
+ res->resolved = false;
+ res->info = info;
+
+ } else
+ resolve_resource_against_device(res, orig_device, path);
+
+ return res;
+}
+
+static bool resolve_grub2_resource(struct device_handler *handler,
+ struct resource *res)
+{
+ struct grub2_resource_info *info = res->info;
+ struct discover_device *dev;
+
+ assert(!res->resolved);
+
+ dev = device_lookup_by_uuid(handler, info->root->uuid);
+
+ if (!dev)
+ return false;
+
+ resolve_resource_against_device(res, dev, info->path);
+ talloc_free(info);
+
+ return true;
+}
+
+static bool current_option_is_default(struct grub2_state *state)
+{
+ if (state->default_idx < 0)
+ return false;
+ return state->cur_idx == state->default_idx;
+}
+
static void grub2_finish(struct conf_context *conf)
{
struct device *dev = conf->dc->device->device;
conf_strip_str(opt->boot_args);
conf_strip_str(opt->description);
- /* opt is persistent, so must be associated with device */
+ state->opt->option->is_default = current_option_is_default(state);
discover_context_add_boot_option(conf->dc, state->opt);
- state->opt = discover_boot_option_create(conf->dc, conf->dc->device);
- opt = state->opt->option;
- opt->boot_args = talloc_strdup(opt, "");
-
- talloc_free(state->desc_image);
- state->desc_image = NULL;
+ state->opt = NULL;
+ state->cur_idx++;
}
static void grub2_process_pair(struct conf_context *conf, const char *name,
return;
if (streq(name, "menuentry")) {
- char *sep;
+ /* complete any existing option... */
+ if (state->opt)
+ grub2_finish(conf);
- grub2_finish(conf);
+ /* ... then start the new one */
+ opt = discover_boot_option_create(conf->dc, conf->dc->device);
+ opt->option->boot_args = talloc_strdup(opt->option, "");
- /* Then start the new image. */
-
- sep = strchr(value, '\'');
-
- if (sep)
- *sep = 0;
+ value = strtok(value, "\'{\"");
opt->option->id = talloc_asprintf(opt->option,
"%s#%s", dev->id, value);
opt->option->name = talloc_strdup(opt->option, value);
+ opt->option->boot_args = talloc_strdup(opt, "");
+
+ state->opt = opt;
return;
}
- if (streq(name, "linux")) {
+ if (streq(name, "linux") || streq(name, "linux16")) {
char *sep;
sep = strchr(value, ' ');
if (sep)
*sep = 0;
- opt->boot_image = create_devpath_resource(opt,
- conf->dc->device, value);
+ opt->boot_image = create_grub2_resource(opt, conf->dc->device,
+ state->root, value);
state->desc_image = talloc_strdup(opt, value);
if (sep)
}
if (streq(name, "initrd")) {
- opt->initrd = create_devpath_resource(opt,
- conf->dc->device, value);
+ opt->initrd = create_grub2_resource(opt, conf->dc->device,
+ state->root, value);
state->desc_initrd = talloc_asprintf(state, "initrd=%s",
value);
return;
}
+ if (streq(name, "search")) {
+ struct grub2_root *root;
+ char *uuid;
+
+ if (!strstr(value, "--set=root")) {
+ pb_log("%s: no root\n", __func__);
+ return;
+ }
+
+ /* The UUID should be the last argument to the search command.
+ * FIXME: this is a little fragile; would be nice to have some
+ * parser helpers to deal with "command args" parsing
+ */
+ uuid = strrchr(value, ' ');
+ if (!uuid)
+ return;
+
+ uuid++;
+
+ if (state->root)
+ talloc_unlink(state, state->root);
+
+ root = talloc(state, struct grub2_root);
+ root->uuid = talloc_strdup(root, uuid);
+ state->root = root;
+ return;
+ }
+
+ if (streq(name, "set")) {
+ char *sep, *var_name, *var_value;
+
+ /* this is pretty nasty, but works until we implement a proper
+ * parser... */
+
+ sep = strchr(value, '=');
+ if (!sep)
+ return;
+
+ *sep = '\0';
+
+ var_name = value;
+ var_value = sep + 1;
+ if (var_value[0] == '"' || var_value[0] == '\'')
+ var_value++;
+
+ if (!strlen(var_name) || !strlen(var_value))
+ return;
+
+ if (streq(var_name, "default"))
+ state->default_idx = atoi(var_value);
+
+ return;
+ }
+
pb_log("%s: unknown name: %s\n", __func__, name);
}
"/grub.cfg",
"/menu.lst",
"/grub/grub.cfg",
+ "/grub2/grub.cfg",
"/grub/menu.lst",
"/boot/grub/grub.cfg",
+ "/boot/grub2/grub.cfg",
"/boot/grub/menu.lst",
"/GRUB.CFG",
"/MENU.LST",
"/GRUB/GRUB.CFG",
+ "/GRUB2/GRUB.CFG",
"/GRUB/MENU.LST",
"/BOOT/GRUB/GRUB.CFG",
"/BOOT/GRUB/MENU.LST",
static const char *grub2_known_names[] = {
"menuentry",
"linux",
+ "linux16",
"initrd",
+ "search",
+ "set",
NULL
};
conf->parser_info = state = talloc_zero(conf, struct grub2_state);
state->known_names = grub2_known_names;
-
- /* opt is persistent, so must be associated with device */
-
- state->opt = discover_boot_option_create(dc, dc->device);
- state->opt->option->boot_args = talloc_strdup(state->opt->option, "");
+ state->default_idx = -1;
conf_parse_buf(conf, buf, len);
return 1;
}
-struct parser __grub2_parser = {
- .name = "grub2",
- .parse = grub2_parse,
- .filenames = grub2_conf_files,
+static struct parser grub2_parser = {
+ .name = "grub2",
+ .method = CONF_METHOD_LOCAL_FILE,
+ .parse = grub2_parse,
+ .filenames = grub2_conf_files,
+ .resolve_resource = resolve_grub2_resource,
};
+
+register_parser(grub2_parser);