Merge 50beeef63a into 5468724771

install.sh

@@ -1,5 +1,67 @@
 #!/usr/bin/env bash
 
+
+print_help() {
+  echo "Usage: $0 [-s y|n] [-t h|s|l|i|m|t] -u <webuser name> -p <password for web user>"
+  echo "  -s: yes or no (optional)"
+  echo "  -t: h (host),s (sensor), l (llm), i(mini),m(mobile),t(tarpit) (optional)"
+  echo "  -u: web username (optional)"
+  echo "  -p: password for web user (optional)"
+  exit 1
+}
+
+validate_s() {
+  if [[ -n "$myQST" ]]; then
+    if [[ "$myQST" =~ ^[yYnN]$ ]]; then
+      return 1 # Valid
+    else
+      print_help
+    fi
+  else
+    print_help
+  fi
+
+}
+
+validate_t() {
+  if [[ -n "$myTPOT_TYPE" ]]; then
+    if [[ "$myTPOT_TYPE" =~ ^[hslimtHSLIMT]$ ]]; then
+      return 1 # Valid
+    else
+      print_help
+    fi
+  else
+    print_help
+  fi
+
+}
+
+while getopts ":s:t:u:p:" opt; do
+  case "$opt" in
+  s)
+    myQST="${OPTARG}"
+    validate_s
+    ;;
+  t)
+    myTPOT_TYPE="${OPTARG}"
+    validate_t
+    ;;
+  u)
+    export myWEB_USER="${OPTARG}"
+    ;;
+  p)
+    export myWEB_PW="${OPTARG}"
+    ;;
+  :)
+    echo "Option -${OPTARG} requires an argument."
+    print_help
+    exit 1
+    ;;
+  \?)
+    print_help
+    ;;
+  esac
+done
 myINSTALL_NOTIFICATION="### Now installing required packages ..."
 myUSER=$(whoami)
 myTPOT_CONF_FILE="/home/${myUSER}/tpotce/.env"
@@ -43,12 +105,13 @@ echo "$myINSTALLER"
 echo
 echo
 echo "### This script will now install T-Pot and all of its dependencies."
-while [ "${myQST}" != "y" ] && [ "${myQST}" != "n" ];
+if [[ -z "$myQST" ]]; then
-  do
+  while [ "${myQST}" != "y" ] && [ "${myQST}" != "n" ]; do
     echo
     read -p "### Install? (y/n) " myQST
     echo
   done
+fi
 if [ "${myQST}" = "n" ];
   then
     echo
@@ -183,7 +246,10 @@ echo "###            Feed data endlessly to attackers, bots and scanners."
 echo "###            Also runs a Denial of Service Honeypot (ddospot)."
 echo
 while true; do
+  if [[ -z "$myTPOT_TYPE" ]]; then
     read -p "### Install Type? (h/s/l/i/m/t) " myTPOT_TYPE
+  fi  
+  
   case "${myTPOT_TYPE}" in
     h|H)
       echo
@@ -239,55 +305,49 @@ if [ "${myTPOT_TYPE}" == "HIVE" ];
   echo "### T-Pot User Configuration ..."
   echo
   # Asking for web user name
+  if [[ -z "$myWEB_USER" ]]; then
     myWEB_USER=""
-	while [ 1 != 2 ];
+    while [ 1 != 2 ]; do
-	  do
       myOK=""
       read -rp "### Enter your web user name: " myWEB_USER
       myWEB_USER=$(echo $myWEB_USER | tr -cd "[:alnum:]_.-")
       echo "### Your username is: ${myWEB_USER}"
-	    while [[ ! "${myOK}" =~ [YyNn] ]];
+      while [[ ! "${myOK}" =~ [YyNn] ]]; do    
-	      do
         read -rp "### Is this correct? (y/n) " myOK
       done
-	    if [[ "${myOK}" =~ [Yy] ]] && [ "$myWEB_USER" != "" ];
+      if [[ "${myOK}" =~ [Yy] ]] && [ "$myWEB_USER" != "" ]; then
-	      then
         break
       else
         echo
       fi
     done
+  fi
 
   # Asking for web user password
+  if [[ -z "$myWEB_PW" ]]; then
     myWEB_PW="pass1"
     myWEB_PW2="pass2"
     mySECURE=0
     myOK=""
-	while [ "${myWEB_PW}" != "${myWEB_PW2}"  ] && [ "${mySECURE}" == "0" ]
+    while [ "${myWEB_PW}" != "${myWEB_PW2}" ] && [ "${mySECURE}" == "0" ]; do
-	  do
       echo
-	    while [ "${myWEB_PW}" == "pass1"  ] || [ "${myWEB_PW}" == "" ]
+      while [ "${myWEB_PW}" == "pass1" ] || [ "${myWEB_PW}" == "" ]; do
-	      do
         read -rsp "### Enter password for your web user: " myWEB_PW
         echo
       done
       read -rsp "### Repeat password you your web user: " myWEB_PW2
       echo
-	    if [ "${myWEB_PW}" != "${myWEB_PW2}" ];
+      if [ "${myWEB_PW}" != "${myWEB_PW2}" ]; then
-	      then
         echo "### Passwords do not match."
         myWEB_PW="pass1"
         myWEB_PW2="pass2"
       fi
       mySECURE=$(printf "%s" "$myWEB_PW" | /usr/sbin/cracklib-check | grep -c "OK")
-	    if [ "$mySECURE" == "0" ] && [ "$myWEB_PW" == "$myWEB_PW2" ];
+      if [ "$mySECURE" == "0" ] && [ "$myWEB_PW" == "$myWEB_PW2" ]; then
-	      then
+        while [[ ! "${myOK}" =~ [YyNn] ]]; do
-	        while [[ ! "${myOK}" =~ [YyNn] ]];
-	          do
           read -rp "### Keep insecure password? (y/n) " myOK
         done
-	        if [[ "${myOK}" =~ [Nn] ]] || [ "$myWEB_PW" == "" ];
+        if [[ "${myOK}" =~ [Nn] ]] || [ "$myWEB_PW" == "" ]; then
-	          then
           myWEB_PW="pass1"
           myWEB_PW2="pass2"
           mySECURE=0
@@ -295,6 +355,8 @@ if [ "${myTPOT_TYPE}" == "HIVE" ];
         fi
       fi
     done
+  fi
+
 
   # Write username and password to T-Pot config file
   echo "### Creating base64 encoded htpasswd username and password for T-Pot config file: ${myTPOT_CONF_FILE}"