test.sh

#!/bin/bash
# Copyright (c) 2017-2018 The Swipp developers
# Distributed under the MIT/X11 software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.

SWIPP_BINARY=src/swippd
SWIPP_ARGS="-debug -debugbacktrace -testnet -staking -datadir="
MINER_BINARY=cpuminer-opt/cpuminer
MINER_ARGS=" -a x11 --userpass=%s:%s --url=http://127.0.0.1:%d"
TMP_TEMPLATE=/tmp/swipp.XXXXXXX
STATUS_COMMAND="ps -eo pid,args | grep swippd | grep testnet"

RED=$(tput setaf 1)
GREEN=$(tput setaf 2)
MAGENTA=$(tput setaf 5)
RESET=$(tput sgr0)

# Return a randomly generated UUID.

uuid() {
	cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 32 | head -n 1
}

# Stall and wait for the process in $1 to exit. Returns instantly if the
# process is not running.
#
# $1 [pid] The process id to wait for.

waitpid() {
	tail --pid=$1 -f /dev/null
}

# Return true/false exit value depending on if the process in $1 exists.
#
# $1 [pid] The process id to check for.

pidexists() {
	kill -0 $1 &> /dev/null
}

# Start the swipp daemon executable specified in SWIPP_BINARY.
#
# $1 [ip]      IP address to bind daemon to.
# $2 [port]    Port to bind daemon to.
# $3 [rpcport] Port to run the rpc interface on.

start_swipp_exe() {
	dir=$(mktemp -d $TMP_TEMPLATE)

	user=$(uuid)
	echo $user > $dir/.user

	password=$(uuid)
	echo $password > $dir/.password

	extraargs="-rpcuser=$user -rpcpassword=$password "
	extraargs+="-externalip=$1 -bind=$1:$2 -rpcport=$3 "

	for i in "${peers[@]}"; do
		extraargs+="-addnode=$i "
	done

	echo $GREEN"Starting swipp instance"$RESET ">" $SWIPP_BINARY $extraargs
	$SWIPP_BINARY $SWIPP_ARGS$dir $extraargs &
	pid=$!

	echo $pid > $dir/.pid
	sleep 2

	if ! pidexists $pid; then
		echo Failed to start swipp instance "$1$dir $extrargs" \
		     Aborting startup...
		stop
		exit 1
	fi
}

# Create a list of peers in the given interval and put them into $peers.
# Exclude the IP specified in $1. Peers are generated frm 127.0.10.<startindex>
# to 127.0.10.<endindex>.
#
# $1 [ip]         IP (peer) to exclude from the final list.
# $2 [startindex] The index to start from when generating peer list.
# $3 [endindex]   The index to end at when generating peer list.

get_peers() {
	tmp_peers=()
	for j in $(seq $2 $3); do
		tmp_peers[$j]="127.0.10."$j
	done
	peers=(${tmp_peers[@]/$1*/})
}

start() {
	echo Preparing Swipp instances...

	for i in $(seq 1 $1); do
		ip="127.0.10."$i
		get_peers $ip 1 $1
		start_swipp_exe $ip 18065 $((15074 + $i))
	done
}

stop() {
	echo Stopping all running Swipp instances...


	if ls /tmp/swipp.* 1> /dev/null 2>&1; then
		for i in /tmp/swipp.*/.pid; do
			kill $(<$i) &> /dev/null # We don't care if the kill fails
			waitpid $(<$i)
		done
	fi

	echo Deleting data directories...
	rm -rf /tmp/swipp.*
}

command() {
	get_node_parameters $1
	$SWIPP_BINARY -rpcuser=$node_user -rpcpassword=$node_password\
	              -rpcport=$node_rpcport "${@:2}"
}

status() {
	running_nodes=$(eval $STATUS_COMMAND)
	SAVEIFS=$IFS; IFS=$'\n'; running_nodes=($running_nodes); IFS=$SAVEIFS

	i=0;

	if [ "$running_nodes" != "" ]; then
		for (( ; i < ${#running_nodes[@]}; i++ )); do
			privkey=$(command $i dumpprivkey $(command $i getaddressesbyaccount "" | grep -Po "[a-zA-Z0-9]+"))
			echo $GREEN[$i, $privkey]$RESET: ${running_nodes[$i]}
		done
	fi

	echo $MAGENTA"There are $i nodes up and running."$RESET
}

get_node_parameters() {
	running_nodes=$(eval $STATUS_COMMAND)
	SAVEIFS=$IFS; IFS=$'\n'; running_nodes=($running_nodes); IFS=$SAVEIFS

	re="-datadir=([^ ]+) -rpcuser=([^ ]+) -rpcpassword=([^ ]+) [^ ]+ [^ ]+ -rpcport=([^ ]+)"
	if [[ ${running_nodes[$1]} =~ $re ]]; then
		node_datadir=${BASH_REMATCH[1]}
		node_user=${BASH_REMATCH[2]}
		node_password=${BASH_REMATCH[3]}
		node_rpcport=${BASH_REMATCH[4]}
	else
		echo $RED"The" specified Swipp node is not running and could \
		     not be found. Exiting...$RESET
		exit 1
	fi
}

mine() {
	echo Initiating mining on Swipp node instance $1. Please hit CTRL-C to stop...
	get_node_parameters $1
	$MINER_BINARY $(printf "$MINER_ARGS" $node_user $node_password $node_rpcport)
}

case $1 in
	start)
	if [ "$#" -lt 2 ]; then
		echo $RED"Syntax" for starting swipp test instances is \
		     "\"start <n>\"", where n denotes the number of nodes. \
		     Exiting...$RESET
		exit 1
	fi

	if [ "$(eval $STATUS_COMMAND)" != "" ]; then
		echo $RED"Already started."$RESET
		exit 1
	fi

	start $2
	;;

	stop)
	stop
	;;

	command)
	if [ "$#" -lt 3 ]; then
		echo $RED"Syntax" for sending a command to a Swipp node \
		     is incorrect. The correct syntax is \
		     "\"command <n> <command> <args>\"", where n denotes \
		     the node number as shown by "\"status\"".$RESET
		     exit 1
	fi

	command $2 "${@:3}"
	;;

	status)
	status
	;;

	mine)
	if [ "$#" -lt 2 ]; then
		echo $RED"Syntax" for initiating a mining session is \
		     "\"mine <n>\"", where n denotes the node number as shown \
		     by "\"status\"".$RESET
		exit 1
	fi

	if [ ! -f $MINER_BINARY ]; then
		echo $RED"Miner" binary not found. Please compile the miner \
		     before executing this operation.
		exit 1
	fi

	mine $2
	;;

	# unknown option
	*)
	echo Please specify one of the following: start/stop/run. \
	     Exiting...
	exit 1
	;;
esac

exit 0