run

#!/bin/bash


# <command> <stub(file)> <conf> <ca> <curpath> <stub(printable name)> <timeout>
#init-------------------------------------------------------------
curpath=$0
if [[ ! $curpath == \/* ]]; then
  curpath=${curpath#./}         ##remove ./
  curpath=`pwd`/$curpath  ##add .../..
fi

if [[ $curpath =~ "/"  ]]; then
  curpath=${curpath%/*}   ##remove /run..
else
  curpath=""
fi
command=$1
#execcommand='exec'
if [[ $command == "_" ]]; then
  command=""
  extracommand='./'
fi

stubspath=$8
if [[ ! $stubspath ]]; then
  stubspath=$curpath"/../../../../stubs"
fi

filename=$2
if [[ ! $filename =~ '/' ]]; then
  if [[ $filename =~ ":" ]]; then
    path=$stubspath/${filename%:*}
    cd $path
    filename=${filename##*:}
  else
    cd $stubspath/$filename
    filename=`find -name "[Rr]un*"`' Run run'
  fi
  filename=${filename##*/}
fi

status=""
for f in $filename
do
  #echo `pwd` - $command $extracommand''$f google.com 443
  valid=`$command $extracommand''$f google.com 443`
  if [[ $(echo $valid | grep "REJECT") || $(echo $valid | grep "ACCEPT") || $(echo $valid | grep "UNSUPPORTED") ]]; then
    filename=$f #this is propably the correct file, if that is not the case, well then you will see some errors
    status="ok"
    break
  fi
done
if [[ ! $status ]]; then  #no correct file found or might be wrong command?
  exit 1
fi



confpath=$3
if [[ ! $confpath =~ '/' ]]; then
  confpath=$curpath"/conf/"$confpath
fi


certspath=$4
FIS=IFS; IFS=+ read certspath cabundle <<< $certspath; IFS=FIS
if [[ "$certspath" ]]; then
  if [[ "$certspath" != '_' ]]; then
    if [[ $certspath == "--" ]]; then
      certspath=$curpath/certs
    else
      certspath=${certspath//[[:blank:]]/}
    fi
  else
    certspath=""
  fi
  if [[ "$5" ]]; then
    if [[ $5 != "_" ]]; then
      curpath=$5
    fi
  fi
fi


stub=$6
if [[ ! $stub ]]; then
  stub="${filename##*\/}"
fi
filepath=${curpath%/*}

timeout=$7
if [[ ! $timeout ]]; then
  timeout=15
fi

#read info
infopath=$curpath/info
if [[ -f "$infopath/info" ]]; then
  info=`cat $infopath/info`
  parallel=`echo $info | grep 'parallel'`
  printerror=`echo $info | grep 'print-error'`
  printcontext=`echo $info | grep 'print-context'`
fi

#functions-------------------------------------------------------------

#echo timeout: $timeout

checkreturnvalue()
{
  if [[ $1 ]]; then
    if [[ $1 == "BREAK" ]]; then
      return 1
    fi
  fi
  return 0
}


handleresponse()
{
  #echo "hello world"
  start=0
  process=$1
  while [[ $process ]]; do

    if ps -p $process >> /dev/null 2>&1
    then
      (( start++ ))
      sleep 1
      if [[  $start -gt $timeout ]]; then
        kill $process >> /dev/null 2>&1
        echo "TIMED OUT"
        return
      fi
    else
      return
    fi
  done

}

connect()
{
  port=$1
  msg=$2
  expected=$3
  crt=$4
  hostname=$5
  extra=$6

  returnvalue=0

  if [[ "$crt" == "certs/test.crt" ]]; then
    response=`$command $extracommand''$filename $hostname $port $curpath/$crt 2>&1 & handleresponse $!`
  elif [[ "$crt" == "_" ]]; then
    if [[ $cabundle ]]; then
      response=`$command $extracommand''$filename $hostname $port $cabundle  2>&1 & handleresponse $!`
    elif [[ "$certspath" =~ "." ]]; then
      response=`$command $extracommand''$filename $hostname $port $certspath  2>&1 & handleresponse $!`
    else
      response=`$command $extracommand''$filename $hostname $port  2>&1 & handleresponse $!`
    fi
  elif [[ "$certspath" ]]; then
    if [[ "$certspath" != "-k" ]]; then
      #echo $command $filename $hostname $port "$certspath"/"$crt"
      response=`$command $extracommand''$filename $hostname $port $certspath/$crt 2>&1 & handleresponse $!`
    else
      response=`$command $extracommand''$filename $hostname $port $certspath  2>&1 & handleresponse $!`
    fi
  else
    response=`$command $extracommand''$filename $hostname $port  2>&1 & handleresponse $!`
  fi


  if [[ $printcontext ]]; then
    context="`echo -e $response | sed -e '$d'`"
  fi
  if [[ $printerror ]]; then
    error=$response
  fi
  response=`echo -e $response | sed -e '$!d'`
  #choose color

  if [[ "$response" == "UNSUPPORTED" ]]; then
    color='\e[0;37m'    #allright
    response='[ OK?  ][ UNSUPPORTED  ]'
    #msg=$filename
  elif [[ "$response" == "TIMED OUT" ]]; then
    color='\e[0;40m'      #most likely the connection was succesfull, but can be sure
    response="[      ][  TIMED OUT   ]"
    `checkreturnvalue $extra`; returnvalue=$?
  elif [[ "$expected" == "REJECT" ]]; then

    if [[ "$response" == "REJECT" ]]; then
      color='\e[0;32m'    #green, success
      response='[ PASS ]'[$response]
    else
      color='\e[0;41m'    #red, failure(either error or verify did succeed)
      if [[ "$response" == "ACCEPT" ]]; then
        response='[ FAIL ]'[$response]
        `checkreturnvalue $extra`; returnvalue=$?
      else
        response='[ ERROR][------]'
        if [[ $printerror ]]; then
          context=$error
        fi
        `checkreturnvalue $extra`; returnvalue=$?
      fi
    fi

  elif [[ "$expected" == "ACCEPT" ]]; then
    if [[ "$response" == "ACCEPT" ]]; then
      color='\e[0;36m'    #green, success
      response="[ PASS ][$response]"
    else
      color='\e[0;41m'    #red, failure(either error or verify did not succeed)
      if [[ "$response" == "REJECT" ]]; then
        response="[ FAIL ][$response]"
        `checkreturnvalue $extra`; returnvalue=$?
      else
        response='[ ERROR][------]'
        if [[ $printerror ]]; then
          context=$error
        fi
        `checkreturnvalue $extra`; returnvalue=$?
      fi
    fi
  else
    color='\e[0;37m'    #allright
    if [[ "$response" == "ACCEPT" || "$response" == "REJECT" ]]; then
      response="[ OK?  ][$response]"
    else
      response='[ ERROR][------]'
      if [[ $printerror ]]; then
        context=$error
      fi
      `checkreturnvalue $extra`; returnvalue=$?
    fi
  fi

  if [[ $returnvalue != 0 ]]; then
    response=$response"[-> SKIP $hostname (till CONTINUE)    ]"
    msg=""
  fi

  if [[ $printcontext && $context ]]; then
    echo -e "$color[$stub]$response[$msg]\e[0m[$hostname]\n"$context
  else
    echo -e "$color[$stub]$response[$msg]\e[0m[$hostname]"
  fi
  return $returnvalue

}

#main-------------------------------------------------------------

FIS=$IFS
IFS='&'

trytoconnect=true

while read -r port msg expected crt hostname extra; do

  port=${port//[[:blank:]]/}
  #msg=${msg//[[:blank:]]/}
  expected=${expected//[[:blank:]]/}
  crt=${crt//[[:blank:]]/}
  hostname=${hostname//[[:blank:]]/}
  extra=${extra//[[:blank:]]/}


  #echo "port: " .$port.
  #echo "msg: " .$msg.
  #echo "crt: " .$crt.
  #echo "expected: " .$expected.
  #echo "hostname: " .$hostname.
  #echo "extra:"$extra"-"
  #echo $filename
  #echo $confpath
  #echo $certspath

  #try to connect
  if [ $trytoconnect = true ]; then
    if [[ $parallel && $extra != "BREAK" ]]; then
      connect $port $msg $expected $crt $hostname &
    else
      connect $port $msg $expected $crt $hostname $extra || trytoconnect=false      #should i skip the rest of this backend?
    fi
  elif [[ $extra == "CONTINUE" ]]; then  #resume control
    trytoconnect=true
  fi

done < "$confpath"


IFS=$FIS

exit 0