X-Git-Url: http://git.ozlabs.org/?a=blobdiff_plain;f=discover%2Fgrub2%2Fscript.c;h=0cf21967450efb3fbdf22c76f901c1c312b4d9d1;hb=4844eaae93e9e32708f02f7e5a43edcc87435699;hp=ec37fbb0c06bc1b8da5984e339c90a27502689a3;hpb=64899475f9b895628fd7b654f7b549e50494229a;p=petitboot diff --git a/discover/grub2/script.c b/discover/grub2/script.c index ec37fbb..0cf2196 100644 --- a/discover/grub2/script.c +++ b/discover/grub2/script.c @@ -1,6 +1,7 @@ #include #include +#include #include #include @@ -9,10 +10,16 @@ #define to_stmt_simple(stmt) \ container_of(stmt, struct grub2_statement_simple, st) +#define to_stmt_block(stmt) \ + container_of(stmt, struct grub2_statement_block, st) #define to_stmt_if(stmt) \ 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) +#define to_stmt_conditional(stmt) \ + container_of(stmt, struct grub2_statement_conditional, st) struct env_entry { const char *name; @@ -90,6 +97,24 @@ static bool is_delim(char c) return c == ' ' || c == '\t'; } +static bool option_is_default(struct grub2_script *script, + struct discover_boot_option *opt) +{ + unsigned int default_idx; + const char *var; + char *end; + + var = script_env_get(script, "default"); + if (!var) + return false; + + default_idx = strtoul(var, &end, 10); + if (end != var && *end == '\0') + return default_idx == script->n_options; + + return !strcmp(opt->option->name, var); +} + /* For non-double-quoted variable expansions, we may need to split the * variable's value into multiple argv items. * @@ -229,6 +254,7 @@ int statement_simple_execute(struct grub2_script *script, { struct grub2_statement_simple *st = to_stmt_simple(statement); struct grub2_symtab_entry *entry; + char *pos; int rc; if (!st->argv) @@ -239,10 +265,21 @@ int statement_simple_execute(struct grub2_script *script, if (!st->argv->argc) return 0; + /* is this a var=value assignment? */ + pos = strchr(st->argv->argv[0], '='); + if (pos) { + char *name, *value; + name = st->argv->argv[0]; + name = talloc_strndup(st, name, pos - name); + value = pos + 1; + script_env_set(script, name, value); + return 0; + } + entry = script_lookup_function(script, st->argv->argv[0]); if (!entry) { fprintf(stderr, "undefined function '%s'\n", st->argv->argv[0]); - return 0; + return 1; } rc = entry->fn(script, entry->data, st->argv->argc, st->argv->argv); @@ -250,24 +287,45 @@ int statement_simple_execute(struct grub2_script *script, return rc; } -int statement_if_execute(struct grub2_script *script, +int statement_block_execute(struct grub2_script *script, struct grub2_statement *statement) { - struct grub2_statement_if *st = to_stmt_if(statement); - struct grub2_statements *case_stmts; + struct grub2_statement_block *st = to_stmt_block(statement); + return statements_execute(script, st->statements); +} + +/* returns 0 if the statement was executed, 1 otherwise */ +static int statement_conditional_execute(struct grub2_script *script, + struct grub2_statement *statement, bool *executed) +{ + struct grub2_statement_conditional *st = to_stmt_conditional(statement); int rc; rc = st->condition->exec(script, st->condition); + *executed = (!rc); + if (*executed) + rc = statements_execute(script, st->statements); - if (rc == 0) - case_stmts = st->true_case; - else - case_stmts = st->false_case; + return rc; +} - if (case_stmts) - statements_execute(script, case_stmts); - else - rc = 0; +int statement_if_execute(struct grub2_script *script, + struct grub2_statement *statement) +{ + struct grub2_statement_if *st = to_stmt_if(statement); + struct grub2_statement *conditional; + bool executed; + int rc; + + list_for_each_entry(&st->conditionals->list, conditional, list) { + rc = statement_conditional_execute(script, + conditional, &executed); + if (executed) + break; + } + + if (!executed && st->else_case) + rc = statements_execute(script, st->else_case); return rc; } @@ -286,17 +344,54 @@ int statement_menuentry_execute(struct grub2_script *script, } else { opt->option->name = talloc_strdup(opt, "(unknown)"); } + opt->option->id = talloc_asprintf(opt->option, "%s#%s", + script->ctx->device->device->id, + opt->option->name); script->opt = opt; statements_execute(script, st->statements); + opt->option->is_default = option_is_default(script, opt); + discover_context_add_boot_option(script->ctx, opt); + script->n_options++; script->opt = NULL; 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; @@ -334,11 +429,10 @@ struct grub2_script *create_script(struct grub2_parser *parser, { struct grub2_script *script; - script = talloc(parser, struct grub2_script); + script = talloc_zero(parser, struct grub2_script); init_env(script); script->ctx = ctx; - script->opt = NULL; list_init(&script->symtab); register_builtins(script);