From 559ea8e1b5479bd738898c8c7d63d071cb2e3279 Mon Sep 17 00:00:00 2001 From: Linus Groh Date: Wed, 5 Aug 2020 20:03:21 +0200 Subject: [PATCH] Userland: Use Core::ArgsParser for 'tt' --- Userland/tt.cpp | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/Userland/tt.cpp b/Userland/tt.cpp index 924c59cb16..194acc93cc 100644 --- a/Userland/tt.cpp +++ b/Userland/tt.cpp @@ -24,6 +24,7 @@ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ +#include #include #include #include @@ -39,15 +40,21 @@ static int set_stack_test(); int main(int argc, char** argv) { - if (argc == 2 && *argv[1] == 'm') + const char* test_name = nullptr; + + Core::ArgsParser args_parser; + args_parser.add_positional_argument(test_name, "Test to run (m = mutex, d = detached, p = priority, s = stack size, x = set stack)", "test-name", Core::ArgsParser::Required::No); + args_parser.parse(argc, argv); + + if (*test_name == 'm') return mutex_test(); - if (argc == 2 && *argv[1] == 'd') + if (*test_name == 'd') return detached_test(); - if (argc == 2 && *argv[1] == 'p') + if (*test_name == 'p') return priority_test(); - if (argc == 2 && *argv[1] == 's') + if (*test_name == 's') return stack_size_test(); - if (argc == 2 && *argv[1] == 'x') + if (*test_name == 'x') return set_stack_test(); printf("Hello from the first thread!\n");