X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fgrub2%2Fbuiltins.c;h=6ada2a64c130ae05c66c12c15753e84e0af684e2;hp=55597c18467d4cd3f697816a804543fcf45366df;hb=37bff93c8b0a71432613f41f2319dc073ca64619;hpb=64899475f9b895628fd7b654f7b549e50494229a diff --git a/discover/grub2/builtins.c b/discover/grub2/builtins.c index 55597c1..6ada2a6 100644 --- a/discover/grub2/builtins.c +++ b/discover/grub2/builtins.c @@ -2,9 +2,12 @@ #include #include +#include +#include #include -#include +#include +#include "discover/parser.h" #include "grub2.h" @@ -33,16 +36,333 @@ static int builtin_set(struct grub2_script *script, return 0; } +static int builtin_linux(struct grub2_script *script, + void *data __attribute__((unused)), + int argc, char *argv[]) +{ + struct discover_boot_option *opt = script->opt; + const char *root; + int i; + + if (!opt) { + pb_log("grub2 syntax error: 'linux' statement outside " + "a menuentry.\n"); + return -1; + } + + if (argc < 2) { + pb_log("grub2 syntax error: no filename provided to " + "linux statement\n"); + return -1; + } + + root = script_env_get(script, "root"); + + opt->boot_image = create_grub2_resource(opt, script->ctx->device, + root, argv[1]); + opt->option->boot_args = NULL; + + if (argc > 2) + opt->option->boot_args = talloc_strdup(opt, argv[2]); + + for (i = 3; i < argc; i++) + opt->option->boot_args = talloc_asprintf_append( + opt->option->boot_args, + " %s", argv[i]); + return 0; +} + +static int builtin_initrd(struct grub2_script *script, + void *data __attribute__((unused)), + int argc, char *argv[]) +{ + struct discover_boot_option *opt = script->opt; + const char *root; + + if (!opt) { + pb_log("grub2 syntax error: 'initrd' statement outside " + "a menuentry.\n"); + return -1; + } + + if (argc < 2) { + pb_log("grub2 syntax error: no filename provided to " + "initrd statement\n"); + return -1; + } + + root = script_env_get(script, "root"); + opt->initrd = create_grub2_resource(opt, script->ctx->device, + root, argv[1]); + + 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; } builtins[] = { { .name = "set", - .fn = builtin_set + .fn = builtin_set, + }, + { + .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) { unsigned int i; @@ -50,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); }