From 201f9e290b59838ed249b7d1be03e5b8230bef3e Mon Sep 17 00:00:00 2001
From: funemy
Date: Sun, 23 Oct 2022 19:21:33 -0700
Subject: add proper bound check

---
 entries/funemy/z3/z3fib.sh | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

(limited to 'entries/funemy/z3/z3fib.sh')

diff --git a/entries/funemy/z3/z3fib.sh b/entries/funemy/z3/z3fib.sh
index 9faa228..03a7209 100755
--- a/entries/funemy/z3/z3fib.sh
+++ b/entries/funemy/z3/z3fib.sh
@@ -14,6 +14,11 @@ else
     touch fib.smt2
 fi
 
+if [ "$1" -lt "0" ]; then
+  echo "Argument must be larger than 0."
+  exit 1
+fi
+
 for i in $(seq 0 $1);
 do
   echo "(declare-const x$i Int)" >> fib.smt2
@@ -32,10 +37,13 @@ done
 echo "    (= x0 0)" >> fib.smt2
 echo "    (= x1 1)" >> fib.smt2
 
-for i in $(seq 2 $1);
-do
-  echo "    (= x$i (+ x$(($i - 2)) x$(($i - 1))))" >> fib.smt2
-done
+if [ "$1" -ge "2" ]; then
+  for i in $(seq 2 $1);
+  do
+    echo "    (= x$i (+ x$(($i - 2)) x$(($i - 1))))" >> fib.smt2
+  done
+fi
+
 
 echo "  )" >> fib.smt2
 echo ")" >> fib.smt2
-- 
cgit v1.2.3-70-g09d2