X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fgrub2%2Fscript.c;h=9c087373ac0454524308fa6da5370a0b86b69cc5;hp=ec37fbb0c06bc1b8da5984e339c90a27502689a3;hb=e8a50ad2461a8efaa4d71ea19692a1b63a0f9bc2;hpb=d808f5d2ec8e232d736a5f811d647df91e2a5b43

diff --git a/discover/grub2/script.c b/discover/grub2/script.c
index ec37fbb..9c08737 100644
--- a/discover/grub2/script.c
+++ b/discover/grub2/script.c
@@ -13,6 +13,8 @@
 	container_of(stmt, struct grub2_statement_if, st)
 #define to_stmt_menuentry(stmt) \
 	container_of(stmt, struct grub2_statement_menuentry, st)
+#define to_stmt_function(stmt) \
+	container_of(stmt, struct grub2_statement_function, st)
 
 struct env_entry {
 	const char		*name;
@@ -297,6 +299,37 @@ int statement_menuentry_execute(struct grub2_script *script,
 	return 0;
 }
 
+static int function_invoke(struct grub2_script *script,
+		void *data, int argc, char **argv)
+{
+	struct grub2_statement_function *fn = data;
+	char *name;
+	int i;
+
+	/* set positional parameters */
+	for (i = 0; i < argc; i++) {
+		name = talloc_asprintf(script, "$%d", i);
+		script_env_set(script, name, argv[i]);
+	}
+
+	return statements_execute(script, fn->body);
+}
+
+int statement_function_execute(struct grub2_script *script,
+		struct grub2_statement *statement)
+{
+	struct grub2_statement_function *st = to_stmt_function(statement);
+	const char *name;
+
+	if (st->name->type == GRUB2_WORD_VAR)
+		expand_var(script, st->name);
+
+	name = st->name->text;
+	script_register_function(script, name, function_invoke, st);
+
+	return 0;
+}
+
 static void init_env(struct grub2_script *script)
 {
 	struct env_entry *env;