From 6ef3d4a66016327b2b6e677e92127b9ac28bf196 Mon Sep 17 00:00:00 2001 From: Quinn Date: Sun, 6 Apr 2025 17:27:55 +0200 Subject: [PATCH] add bounds to numbered input --- src/main.c | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/main.c b/src/main.c index 08fff93..01c32aa 100644 --- a/src/main.c +++ b/src/main.c @@ -1,4 +1,5 @@ #include +#include #include #include #include @@ -39,6 +40,7 @@ int main(int argc, char** argv) { errno = 0; long const c = strtol(argv[i], NULL, 10); if (errno == EINVAL) error("syntax error for string: %s\n", argv[i]); + if (c < 1 || c > (LONG_MAX - 1)) error("count may only be in between 1 and %l. Received %l", LONG_MAX, c); // perform for the input count for (long j = 0; j < c; ++j) {