feat(option): Restrict the timeout input to non-negative integers

mrako · Feb 2, 2021 · c4d125f · c4d125f
1 parent 89ec638
commit c4d125f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/wait-for b/wait-for
@@ -55,7 +55,6 @@ do
     ;;
     -t)
     TIMEOUT="$2"
-    if [ "$TIMEOUT" = "" ]; then break; fi
     shift 2
     ;;
     --timeout=*)
@@ -76,6 +75,11 @@ do
   esac
 done
 
+if ! [ "$TIMEOUT" -ge 0 ] 2>/dev/null; then
+  echoerr "Error: invalid timeout '$TIMEOUT'"
+  usage 3
+fi
+
 if [ "$HOST" = "" -o "$PORT" = "" ]; then
   echoerr "Error: you need to provide a host and port to test."
   usage 2