X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fgrub2%2Fbuiltins.c;fp=discover%2Fgrub2%2Fbuiltins.c;h=8171b8d4907bb2f2067f73ee790d6cb20fd74697;hp=cd56dcc033f8eb67d414e68d241fd5e53b315edb;hb=4bf4d14afbc46d6bbf11839b1f8fdc9bf66dd16d;hpb=1513dd5d05f6408a64e19ea18cfaeb0cb8c5fb73

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,
 	}
 };