discover/grub2: Add initial command execution

[petitboot] / discover / grub2 / builtins.c

diff --git a/discover/grub2/builtins.c b/discover/grub2/builtins.c
new file mode 100644 (file)
index 0000000..379e870
--- /dev/null
+++ b/discover/grub2/builtins.c
@@ -0,0 +1,47 @@
+
+#include <stdio.h>
+#include <string.h>
+
+#include <talloc/talloc.h>
+#include <array-size/array-size.h>
+
+#include "grub2.h"
+
+
+static int builtin_set(struct grub2_script *script, int argc, char *argv[])
+{
+       char *name, *value, *p;
+       int i;
+
+       if (argc < 2)
+               return -1;
+
+       p = strchr(argv[1], '=');
+       if (!p)
+               return -1;
+
+       name = talloc_strndup(script, argv[1], p - argv[1]);
+       value = talloc_strdup(script, p+1);
+
+       for (i = 2; i < argc; i++)
+               value = talloc_asprintf_append(value, " %s", argv[i]);
+
+       script_env_set(script, name, value);
+
+       return 0;
+}
+
+static struct grub2_command commands[] = {
+       {
+               .name = "set",
+               .exec = builtin_set
+       },
+};
+
+void register_builtins(struct grub2_script *script)
+{
+       unsigned int i;
+
+       for (i = 0; i < ARRAY_SIZE(commands); i++)
+               script_register_command(script, &commands[i]);
+}