ui/ncurses: Always provide a key definition for backtab
[petitboot] / discover / grub2 / builtins.c
index cd56dcc033f8eb67d414e68d241fd5e53b315edb..6ada2a64c130ae05c66c12c15753e84e0af684e2 100644 (file)
@@ -5,8 +5,9 @@
 #include <log/log.h>
 #include <types/types.h>
 #include <talloc/talloc.h>
-#include <array-size/array-size.h>
+#include <util/util.h>
 
+#include "discover/parser.h"
 #include "grub2.h"
 
 
@@ -97,6 +98,217 @@ 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 bool builtin_test_op_file(struct grub2_script *script, char op,
+               const char *file)
+{
+       bool result;
+       int len, rc;
+       char *buf;
+
+       rc = parser_request_file(script->ctx, script->ctx->device,
+                       file, &buf, &len);
+       if (rc)
+               return false;
+
+       switch (op) {
+       case 's':
+               /* -s: return true if file exists and has non-zero size */
+               result = len > 0;
+               break;
+       case 'f':
+               /* -f: return true if file exists */
+               result = true;
+               break;
+       default:
+               result = false;
+
+       }
+
+       talloc_free(buf);
+       return result;
+}
+
+static bool builtin_test_op_dir(struct grub2_script *script, char op,
+               const char *dir)
+{
+       if (op != 'd')
+               return false;
+
+       return parser_check_dir(script->ctx, script->ctx->device, dir) == 0;
+}
+
+static bool builtin_test_op(struct grub2_script *script,
+               int argc, char **argv, int *consumed)
+{
+       char *op;
+
+       if (argc >= 3) {
+               const char *a1, *a2;
+
+               a1 = argv[0];
+               op = argv[1];
+               a2 = argv[2];
+
+               if (!strcmp(op, "=") || !strcmp(op, "==")) {
+                       *consumed = 3;
+                       return !strcmp(a1, a2);
+               }
+
+               if (!strcmp(op, "!=")) {
+                       *consumed = 3;
+                       return strcmp(a1, a2);
+               }
+
+               if (!strcmp(op, "<")) {
+                       *consumed = 3;
+                       return strcmp(a1, a2) < 0;
+               }
+
+               if (!strcmp(op, ">")) {
+                       *consumed = 3;
+                       return strcmp(a1, a2) > 0;
+               }
+       }
+
+       if (argc >= 2) {
+               const char *a1;
+
+               op = argv[0];
+               a1 = argv[1];
+
+               if (!strcmp(op, "-z")) {
+                       *consumed = 2;
+                       return strlen(a1) == 0;
+               }
+
+               if (!strcmp(op, "-n")) {
+                       *consumed = 2;
+                       return strlen(a1) != 0;
+               }
+
+               if (!strcmp(op, "-s") || !strcmp(op, "-f")) {
+                       *consumed = 2;
+                       return builtin_test_op_file(script, op[1], a1);
+               }
+
+               if (!strcmp(op, "-d")) {
+                       *consumed = 2;
+                       return builtin_test_op_dir(script, op[1], a1);
+               }
+       }
+
+       op = argv[0];
+       *consumed = 1;
+       return strlen(op) > 0;
+}
+
+static int builtin_test(struct grub2_script *script,
+               void *data __attribute__((unused)),
+               int argc, char *argv[])
+{
+       int consumed;
+       bool not, rc;
+
+       if (!strcmp(argv[0], "[") && !strcmp(argv[argc - 1], "]"))
+               argc--;
+
+       /* skip command name */
+       argc--;
+       argv++;
+
+       not = false;
+       rc = false;
+
+       for (consumed = 0; argc > 0; argv += consumed, argc -= consumed) {
+
+               if (!strcmp(argv[0], "!")) {
+                       not = true;
+                       consumed = 1;
+                       continue;
+               }
+
+               if (!strcmp(argv[0], "-a")) {
+                       if (!rc)
+                               return 1;
+                       consumed = 1;
+                       continue;
+               }
+
+               if (!strcmp(argv[0], "-o")) {
+                       if (rc)
+                               return 0;
+                       consumed = 1;
+                       continue;
+               }
+
+               rc = builtin_test_op(script, argc, argv, &consumed);
+               if (not) {
+                       rc = !rc;
+                       not = false;
+               }
+       }
+
+       return rc ? 0 : 1;
+}
+
+static int builtin_true(struct grub2_script *script __attribute__((unused)),
+               void *data __attribute__((unused)),
+               int argc __attribute__((unused)),
+               char *argv[] __attribute__((unused)))
+{
+       return 0;
+}
+
+static int builtin_false(struct grub2_script *script __attribute__((unused)),
+               void *data __attribute__((unused)),
+               int argc __attribute__((unused)),
+               char *argv[] __attribute__((unused)))
+{
+       return 1;
+}
+
+static int builtin_nop(struct grub2_script *script __attribute__((unused)),
+               void *data __attribute__((unused)),
+               int argc __attribute__((unused)),
+               char *argv[] __attribute__((unused)))
+{
+       return 0;
+}
+
+extern int builtin_load_env(struct grub2_script *script,
+               void *data __attribute__((unused)),
+               int argc, char *argv[]);
+int builtin_save_env(struct grub2_script *script,
+               void *data __attribute__((unused)),
+               int argc, char *argv[]);
+
+
 static struct {
        const char *name;
        grub2_function fn;
@@ -109,10 +321,46 @@ static struct {
                .name = "linux",
                .fn = builtin_linux,
        },
+       {
+               .name = "linux16",
+               .fn = builtin_linux,
+       },
        {
                .name = "initrd",
                .fn = builtin_initrd,
-       }
+       },
+       {
+               .name = "search",
+               .fn = builtin_search,
+       },
+       {
+               .name = "[",
+               .fn = builtin_test,
+       },
+       {
+               .name = "test",
+               .fn = builtin_test,
+       },
+       {
+               .name = "true",
+               .fn = builtin_true,
+       },
+       {
+               .name = "false",
+               .fn = builtin_false,
+       },
+       {
+               .name = "load_env",
+               .fn = builtin_load_env,
+       },
+       {
+               .name = "save_env",
+               .fn = builtin_save_env,
+       },
+};
+
+static const char *nops[] = {
+       "echo", "export", "insmod", "loadfont", "terminfo",
 };
 
 void register_builtins(struct grub2_script *script)
@@ -122,4 +370,7 @@ void register_builtins(struct grub2_script *script)
        for (i = 0; i < ARRAY_SIZE(builtins); i++)
                script_register_function(script, builtins[i].name,
                                builtins[i].fn, NULL);
+
+       for (i = 0; i < ARRAY_SIZE(nops); i++)
+               script_register_function(script, nops[i], builtin_nop, NULL);
 }