discover/grub2: Add a reference to the context from grub2_script

We'll need the context to add boot options.

Signed-off-by: Jeremy Kerr <jk@ozlabs.org>

1 files changed, 5 insertions, 2 deletions

diff --git a/discover/grub2/script.c b/discover/grub2/script.c
index 9cc0cfb..5b38156 100644
--- a/discover/grub2/script.c
+++ b/discover/grub2/script.c
@@ -300,13 +300,16 @@ void script_execute(struct grub2_script *script)
 	statements_execute(script, script->statements);
 }
 
-struct grub2_script *create_script(void *ctx)
+struct grub2_script *create_script(struct grub2_parser *parser,
+		struct discover_context *ctx)
 {
 	struct grub2_script *script;
 
-	script = talloc(ctx, struct grub2_script);
+	script = talloc(parser, struct grub2_script);
 
 	init_env(script);
+	script->ctx = ctx;
+
 	list_init(&script->commands);
 	register_builtins(script);