From 4bf4d14afbc46d6bbf11839b1f8fdc9bf66dd16d Mon Sep 17 00:00:00 2001 From: Jeremy Kerr Date: Mon, 16 Sep 2013 16:53:22 +0800 Subject: [PATCH] discover/grub2: Implement 'search' command Signed-off-by: Jeremy Kerr --- discover/grub2/builtins.c | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/discover/grub2/builtins.c b/discover/grub2/builtins.c index cd56dcc..8171b8d 100644 --- a/discover/grub2/builtins.c +++ b/discover/grub2/builtins.c @@ -97,6 +97,32 @@ static int builtin_initrd(struct grub2_script *script, return 0; } +static int builtin_search(struct grub2_script *script, + void *data __attribute__((unused)), + int argc, char *argv[]) +{ + const char *env_var, *spec; + int i; + + env_var = NULL; + + for (i = 1; i < argc - 1; i++) { + if (!strncmp(argv[i], "--set=", strlen("--set="))) { + env_var = argv[i] + strlen("--set="); + break; + } + } + + if (!env_var) + return 0; + + spec = argv[argc - 1]; + + script_env_set(script, env_var, spec); + + return 0; +} + static struct { const char *name; grub2_function fn; @@ -112,6 +138,10 @@ static struct { { .name = "initrd", .fn = builtin_initrd, + }, + { + .name = "search", + .fn = builtin_search, } }; -- 2.39.2