diff options
author | Jeremy Kerr <jk@ozlabs.org> | 2019-11-14 15:06:26 +0800 |
---|---|---|
committer | Jeremy Kerr <jk@ozlabs.org> | 2019-11-29 13:54:10 +0800 |
commit | 967cfa7e5c1bfb4d2cf78bb3de3dc6d36b78c440 (patch) | |
tree | 880bc62b0e2aeacb70105f923a19821fb1bd4291 /test/parser | |
parent | 9711179694bb0e52c5951dc7222f1f79fcba814d (diff) | |
download | talos-petitboot-967cfa7e5c1bfb4d2cf78bb3de3dc6d36b78c440.tar.gz talos-petitboot-967cfa7e5c1bfb4d2cf78bb3de3dc6d36b78c440.zip |
discover/grub2: implement 'source' command
This change add support for the grub2 'source' command, executing a
referenced script in the current parse context.
We impose a limit of 10 (concurrent) source commands, to prevent
infinite recursion.
Signed-off-by: Jeremy Kerr <jk@ozlabs.org>
Diffstat (limited to 'test/parser')
-rw-r--r-- | test/parser/Makefile.am | 4 | ||||
-rw-r--r-- | test/parser/test-grub2-source-functions.c | 46 | ||||
-rw-r--r-- | test/parser/test-grub2-source-recursion-infinite.c | 43 | ||||
-rw-r--r-- | test/parser/test-grub2-source-recursion.c | 58 | ||||
-rw-r--r-- | test/parser/test-grub2-source.c | 54 |
5 files changed, 205 insertions, 0 deletions
diff --git a/test/parser/Makefile.am b/test/parser/Makefile.am index c8e059b..5f1a93b 100644 --- a/test/parser/Makefile.am +++ b/test/parser/Makefile.am @@ -46,6 +46,10 @@ parser_TESTS = \ test/parser/test-grub2-lexer-error \ test/parser/test-grub2-parser-error \ test/parser/test-grub2-test-file-ops \ + test/parser/test-grub2-source \ + test/parser/test-grub2-source-functions \ + test/parser/test-grub2-source-recursion \ + test/parser/test-grub2-source-recursion-infinite \ test/parser/test-grub2-single-yocto \ test/parser/test-grub2-blscfg-default-filename \ test/parser/test-grub2-blscfg-default-index \ diff --git a/test/parser/test-grub2-source-functions.c b/test/parser/test-grub2-source-functions.c new file mode 100644 index 0000000..a9da934 --- /dev/null +++ b/test/parser/test-grub2-source-functions.c @@ -0,0 +1,46 @@ + +/* check that we can source other scripts, and functions can be defined + * and called across sourced scripts */ + +#include "parser-test.h" + +#if 0 /* PARSER_EMBEDDED_CONFIG */ + +function f1 { + menuentry "f1$1" { linux $2 } +} + +source /grub/2.cfg + +f2 a /vmlinux + +#endif + +void run_test(struct parser_test *test) +{ + struct discover_boot_option *opt; + struct discover_context *ctx; + struct discover_device *dev; + + ctx = test->ctx; + dev = ctx->device; + + test_read_conf_embedded(test, "/grub/grub.cfg"); + + test_add_file_string(test, dev, + "/grub/2.cfg", + "function f2 { menuentry \"f2$1\" { linux $2 } }\n" + "f1 a /vmlinux\n"); + + test_run_parser(test, "grub2"); + + check_boot_option_count(ctx, 2); + + opt = get_boot_option(ctx, 0); + check_name(opt, "f1a"); + check_resolved_local_resource(opt->boot_image, dev, "/vmlinux"); + + opt = get_boot_option(ctx, 1); + check_name(opt, "f2a"); + check_resolved_local_resource(opt->boot_image, dev, "/vmlinux"); +} diff --git a/test/parser/test-grub2-source-recursion-infinite.c b/test/parser/test-grub2-source-recursion-infinite.c new file mode 100644 index 0000000..fbcc5a3 --- /dev/null +++ b/test/parser/test-grub2-source-recursion-infinite.c @@ -0,0 +1,43 @@ + +/* check that have a maximum source recursion limit */ + +#include "parser-test.h" + +#if 0 /* PARSER_EMBEDDED_CONFIG */ + +name=a$name + +menuentry $name { + linux /a +} + +source /grub/grub.cfg + +#endif + +void run_test(struct parser_test *test) +{ + struct discover_boot_option *opt; + struct discover_context *ctx; + struct discover_device *dev; + + ctx = test->ctx; + dev = ctx->device; + + test_read_conf_embedded(test, "/grub/grub.cfg"); + + test_run_parser(test, "grub2"); + + /* we error out after 10 levels, but we should still have + * parse results up to that point + */ + check_boot_option_count(ctx, 11); + + opt = get_boot_option(ctx, 0); + check_name(opt, "a"); + check_resolved_local_resource(opt->boot_image, dev, "/a"); + + opt = get_boot_option(ctx,10); + check_name(opt, "aaaaaaaaaaa"); + check_resolved_local_resource(opt->boot_image, dev, "/a"); +} diff --git a/test/parser/test-grub2-source-recursion.c b/test/parser/test-grub2-source-recursion.c new file mode 100644 index 0000000..21b6bd2 --- /dev/null +++ b/test/parser/test-grub2-source-recursion.c @@ -0,0 +1,58 @@ +/* check that we can source other files recursively */ + +#include "parser-test.h" + +#if 0 /* PARSER_EMBEDDED_CONFIG */ + +menuentry a { + linux /a +} + +source /grub/2.cfg + +menuentry c { + linux /c +} + +#endif + +void run_test(struct parser_test *test) +{ + struct discover_boot_option *opt; + struct discover_context *ctx; + struct discover_device *dev; + + ctx = test->ctx; + dev = ctx->device; + + test_read_conf_embedded(test, "/grub/grub.cfg"); + + /* four levels of config files, the last defining a boot option */ + test_add_file_string(test, dev, + "/grub/2.cfg", + "source /grub/3.cfg\n"); + + test_add_file_string(test, dev, + "/grub/3.cfg", + "source /grub/4.cfg\n"); + + test_add_file_string(test, dev, + "/grub/4.cfg", + "menuentry b { linux /b }\n"); + + test_run_parser(test, "grub2"); + + check_boot_option_count(ctx, 3); + + opt = get_boot_option(ctx, 0); + check_name(opt, "a"); + check_resolved_local_resource(opt->boot_image, dev, "/a"); + + opt = get_boot_option(ctx, 1); + check_name(opt, "b"); + check_resolved_local_resource(opt->boot_image, dev, "/b"); + + opt = get_boot_option(ctx, 2); + check_name(opt, "c"); + check_resolved_local_resource(opt->boot_image, dev, "/c"); +} diff --git a/test/parser/test-grub2-source.c b/test/parser/test-grub2-source.c new file mode 100644 index 0000000..a14bef7 --- /dev/null +++ b/test/parser/test-grub2-source.c @@ -0,0 +1,54 @@ + +/* check that we can source other scripts, and variables get passed + * in to and out of sourced scripts */ + +#include "parser-test.h" + +#if 0 /* PARSER_EMBEDDED_CONFIG */ + +menuentry a { + linux /a +} + +# var: outer -> inner -> outer +v=b + +source /grub/2.cfg + +menuentry $v { + linux /c +} + +#endif + +void run_test(struct parser_test *test) +{ + struct discover_boot_option *opt; + struct discover_context *ctx; + struct discover_device *dev; + + ctx = test->ctx; + dev = ctx->device; + + test_read_conf_embedded(test, "/grub/grub.cfg"); + + test_add_file_string(test, dev, + "/grub/2.cfg", + "menuentry $v { linux /b }\nv=c\n"); + + test_run_parser(test, "grub2"); + + check_boot_option_count(ctx, 3); + + opt = get_boot_option(ctx, 0); + check_name(opt, "a"); + check_resolved_local_resource(opt->boot_image, dev, "/a"); + + opt = get_boot_option(ctx, 1); + check_name(opt, "b"); + check_resolved_local_resource(opt->boot_image, dev, "/b"); + + opt = get_boot_option(ctx, 2); + check_name(opt, "c"); + check_resolved_local_resource(opt->boot_image, dev, "/c"); +} |