diff options
Diffstat (limited to 'test/functional/fixtures/shell-test.c')
| -rw-r--r-- | test/functional/fixtures/shell-test.c | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/test/functional/fixtures/shell-test.c b/test/functional/fixtures/shell-test.c index 6a1f9226d2..1ff1598e2b 100644 --- a/test/functional/fixtures/shell-test.c +++ b/test/functional/fixtures/shell-test.c @@ -33,13 +33,12 @@ static void help(void) puts(" Prints \"ready $ prog args...\\n\" to stderr."); puts(" shell-test -t {prompt text} EXE \"prog args...\""); puts(" Prints \"{prompt text} $ progs args...\" to stderr."); - puts(" shell-test REP {byte} \"line line line\""); - puts(" Prints \"{lnr}: line line line\\n\" to stdout {byte} times."); - puts(" I.e. for `shell-test REP ab \"test\"'"); + puts(" shell-test REP {count} \"line line line\""); + puts(" Prints \"{lnr}: line line line\\n\" to stdout {count} times."); + puts(" I.e. for `shell-test REP 97 \"test\"'"); puts(" 0: test"); puts(" ..."); puts(" 96: test"); - puts(" will be printed because byte `a' is equal to 97."); puts(" shell-test INTERACT"); puts(" Prints \"interact $ \" to stderr, and waits for \"exit\" input."); } @@ -71,8 +70,12 @@ int main(int argc, char **argv) fprintf(stderr, "Not enough REP arguments\n"); return 4; } - uint8_t number = (uint8_t) *argv[2]; - for (uint8_t i = 0; i < number; i++) { + int count = 0; + if (sscanf(argv[2], "%d", &count) != 1) { + fprintf(stderr, "Invalid count: %s\n", argv[2]); + return 4; + } + for (uint8_t i = 0; i < count; i++) { printf("%d: %s\n", (int) i, argv[3]); } } else if (strcmp(argv[1], "UTF-8") == 0) { @@ -104,7 +107,7 @@ int main(int argc, char **argv) break; // EOF } - input_argc = sscanf(input, "%s %d", cmd, &arg); + input_argc = sscanf(input, "%99s %d", cmd, &arg); if(1 == input_argc) { arg = 0; } |