diff options
author | Jeremy Kerr <jk@ozlabs.org> | 2013-09-17 13:02:05 +0800 |
---|---|---|
committer | Jeremy Kerr <jk@ozlabs.org> | 2013-09-24 13:14:59 +0800 |
commit | 9638c29114c8d575d1b89c26dcf1f274ea6ab6c4 (patch) | |
tree | d39b3d16444474f7018a1948d59c7e479f3fd5ee /discover/grub2/script.c | |
parent | 63575c2045b1505da7a2abca8a51a4a7aa3df6f7 (diff) | |
download | talos-petitboot-9638c29114c8d575d1b89c26dcf1f274ea6ab6c4.tar.gz talos-petitboot-9638c29114c8d575d1b89c26dcf1f274ea6ab6c4.zip |
discover/grub2: Create 'conditional' statements
Rather than holding the condition and conditional-statements in struct
grub2_statment_if, create a new conditional type that contains these. We
can then use this to implement elif statements.
Signed-off-by: Jeremy Kerr <jk@ozlabs.org>
Diffstat (limited to 'discover/grub2/script.c')
-rw-r--r-- | discover/grub2/script.c | 34 |
1 files changed, 24 insertions, 10 deletions
diff --git a/discover/grub2/script.c b/discover/grub2/script.c index 9c08737..c0d6e1a 100644 --- a/discover/grub2/script.c +++ b/discover/grub2/script.c @@ -15,6 +15,8 @@ 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; @@ -252,24 +254,36 @@ int statement_simple_execute(struct grub2_script *script, return rc; } +/* 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); + + return rc; +} + int statement_if_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 *conditional; + bool executed; int rc; - rc = st->condition->exec(script, st->condition); + conditional = st->conditional; - if (rc == 0) - case_stmts = st->true_case; - else - case_stmts = st->false_case; + rc = statement_conditional_execute(script, + conditional, &executed); - if (case_stmts) - statements_execute(script, case_stmts); - else - rc = 0; + if (!executed && st->else_case) + rc = statements_execute(script, st->else_case); return rc; } |