From 77c5692bd85d4dcee4864cf945d44b2cf720b0e3 Mon Sep 17 00:00:00 2001
From: Daniel Gibbs <me@danielgibbs.co.uk>
Date: Fri, 4 Mar 2016 00:34:34 +0000
Subject: [PATCH] Getting sdtd graceful working better

---
 functions/command_stop.sh | 313 ++++++++++++++++++++------------------
 1 file changed, 164 insertions(+), 149 deletions(-)

diff --git a/functions/command_stop.sh b/functions/command_stop.sh
index 3e6a7e398..69b799839 100644
--- a/functions/command_stop.sh
+++ b/functions/command_stop.sh
@@ -11,27 +11,29 @@ function_selfname="$(basename $(readlink -f "${BASH_SOURCE[0]}"))"
 
 # Attempts Graceful of source using rcon 'quit' command.
 fn_stop_graceful_source(){
-fn_print_dots "Graceful: rcon quit"
-fn_scriptlog "Graceful: rcon quit"
-# sends quit
-tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
-# waits up to 30 seconds giving the server time to shutdown gracefuly
-for seconds in {1..30}; do
-	pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
-	if [ "${pid}" == "0" ]; then
-		fn_print_ok_nl "Graceful: rcon quit: ${seconds}"
-		fn_scriptlog "Graceful: rcon quit: OK: ${seconds} seconds"
-		break
+	fn_print_dots "Graceful: rcon quit"
+	fn_scriptlog "Graceful: rcon quit"
+	# sends quit
+	tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
+	# waits up to 30 seconds giving the server time to shutdown gracefuly
+	for seconds in {1..30}; do
+		pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
+		if [ "${pid}" == "0" ]; then
+			fn_print_ok "Graceful: rcon quit: ${seconds}: "
+			fn_print_ok_eol_nl
+			fn_scriptlog "Graceful: rcon quit: OK: ${seconds} seconds"
+			break
+		fi
+		sleep 1
+		fn_print_dots "Graceful: rcon quit: ${seconds}"
+	done
+	if [ "${pid}" != "0" ]; then
+		fn_print_fail "Graceful: rcon quit"
+		fn_print_fail_eol_nl
+		fn_scriptlog "Graceful: rcon quit: FAIL"
+		fn_stop_tmux
 	fi
 	sleep 1
-	fn_print_dots "Graceful: rcon quit: ${seconds}"
-done
-if [ "${pid}" != "0" ]; then
-	fn_print_fail_nl "Graceful: rcon quit"
-	fn_scriptlog "Graceful: rcon quit: FAIL"
-	fn_stop_tmux
-fi
-sleep 1
 }
 
 # Attempts Graceful of goldsource using rcon 'quit' command.
@@ -39,162 +41,175 @@ sleep 1
 # this function will only wait 3 seconds then force a tmux shutdown.
 # preventing the server from coming back online.
 fn_stop_graceful_goldsource(){
-fn_print_dots "Graceful: rcon quit"
-fn_scriptlog "Graceful: rcon quit"
-# sends quit
-tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
-# waits 3 seconds as goldsource servers restart with the quit command
-for seconds in {1..3}; do
+	fn_print_dots "Graceful: rcon quit"
+	fn_scriptlog "Graceful: rcon quit"
+	# sends quit
+	tmux send -t "${servicename}" quit ENTER > /dev/null 2>&1
+	# waits 3 seconds as goldsource servers restart with the quit command
+	for seconds in {1..3}; do
+		sleep 1
+		fn_print_dots "Graceful: rcon quit: ${seconds}"
+	done
+	fn_print_ok "Graceful: rcon quit: ${seconds}: "
+	fn_print_ok_eol_nl
+	fn_scriptlog "Graceful: rcon quit: OK: ${seconds} seconds"
 	sleep 1
-	fn_print_dots "Graceful: rcon quit: ${seconds}"
-done
-fn_print_ok_nl "Graceful: rcon quit: ${seconds}"
-sleep 1
-fn_stop_tmux
+	fn_stop_tmux
 }
 
 # Attempts Graceful of 7 Days To Die using telnet.
 fn_stop_telnet_sdtd(){
-sdtdshutdown=$( expect -c '
-proc abort {} {
-	puts "Timeout or EOF\n"
-	exit 1
-}
-spawn telnet '"${telnetip}"' '"${telnetport}"'
-expect {
-	"password:"     { send "'"${telnetpass}"'\r" }
-	default         abort
-}
-expect {
-	"session."  { send "shutdown\r" }
-	default         abort
-}
-expect { eof }
-puts "Completed.\n"
-')
+	sdtd_telnet_shutdown=$( expect -c '
+	proc abort {} {
+		puts "Timeout or EOF\n"
+		exit 1
+	}
+	spawn telnet '"${telnetip}"' '"${telnetport}"'
+	expect {
+		"password:"     { send "'"${telnetpass}"'\r" }
+		default         abort
+	}
+	expect {
+		"session."  { send "shutdown\r" }
+		default         abort
+	}
+	expect { eof }
+	puts "Completed.\n"
+	')
+	
 }
 
 fn_stop_graceful_sdtd(){
-# Gets server IP.
-info_config.sh
-
-fn_print_dots "Graceful: telnet"
-fn_scriptlog "Graceful: telnet"
-sleep 1
+	# Gets server IP.
+	info_config.sh
 
-# uses localhost on first attempt.
-telnetip=127.0.0.1
-fn_print_dots "Graceful: telnet: ${telnetip}"
-fn_scriptlog "Graceful: telnet: ${telnetip}"
-fn_stop_telnet_sdtd
-sleep 1
-
-# falls back to the server ip if localhost fails.
-refused=$(echo -en "\n ${sdtdshutdown}"| grep "Timeout or EOF")
-if [ -n "${refused}" ]; then
-	fn_print_warn_nl "Graceful: telnet: localhost: "
-	fn_print_fail_eol
-	fn_scriptlog "Graceful: telnet: localhost: FAIL"
+	fn_print_dots "Graceful: telnet"
+	fn_scriptlog "Graceful: telnet"
 	sleep 1
-
-	telnetip=${ip}
-	fn_print_dots "Graceful: telnet: ${telnetip}"
-	fn_scriptlog "Graceful: telnet: ${telnetip}"
-	fn_stop_telnet_sdtd
-	refused=$(echo -en "\n ${sdtdshutdown}"| grep "Timeout or EOF")
-
-	fn_print_warnnl "Graceful: telnet: ${telnetip}: "
-	fn_print_fail_eol
-	fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
-	sleep 1 
-fi
-
-# Checks if attempts have worked.
-completed=$(echo -en "\n ${sdtdshutdown}"| grep "Completed.")
-if [ -n "${completed}" ]; then
-	fn_print_ok_nl "Graceful: telnet: "
-	fn_print_ok_eol
-	fn_scriptlog "Graceful: telnet: OK"
-elif [ -n "${refused}" ]; then
-	fn_print_fail_nl "Graceful: telnet: "
-	fn_print_fail_eol
-	fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
-	echo -en "\n\n" | tee -a "${scriptlog}"
-	echo -en "Telnet output:" | tee -a "${scriptlog}"
-	echo -en "\n ${sdtdshutdown}" | tee -a "${scriptlog}"
-	echo -en "\n\n" | tee -a "${scriptlog}"
-else
-	fn_print_fail_nl "Graceful: telnet: Unknown error"
-	fn_scriptlog "Graceful: telnet: Unknown error"
-	echo -en "\n\n" | tee -a "${scriptlog}"
-	echo -en "Telnet output:" | tee -a "${scriptlog}"
-	echo -en "\n ${sdtdshutdown}" | tee -a "${scriptlog}"
-	echo -en "\n\n" | tee -a "${scriptlog}" 
-fi
+	if [ "${telnetenabled}" == "false" ]; then
+		fn_print_info_nl "Graceful: telnet: DISABLED: Enable in ${servercfg}"
+	elif [ "$(command -v expect)" ]||[ "$(which expect >/dev/null 2>&1)" ]; then
+		# Tries to shutdown with both localhost and server IP.
+		for telnetip in 127.0.0.1 ${ip}; do
+			fn_print_dots "Graceful: telnet: ${telnetip}"
+			fn_scriptlog "Graceful: telnet: ${telnetip}"
+			sleep 1
+			fn_stop_telnet_sdtd
+			completed=$(echo -en "\n ${sdtd_telnet_shutdown}"|grep "Completed.")
+			refused=$(echo -en "\n ${sdtd_telnet_shutdown}"|grep "Timeout or EOF")
+			if [ -n "${refused}" ]; then
+				fn_print_warn "Graceful: telnet: ${telnetip}: "
+				fn_print_fail_eol_nl
+				fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
+				sleep 1
+			elif [ -n "${completed}" ]; then
+				break
+			fi
+		done
+
+		# If telnet was successful will use telnet again to check the connection has closed
+		# This confirms that the tmux session can now be killed.
+		if [ -n "${completed}" ]; then
+			for seconds in {1..30}; do
+				fn_stop_telnet_sdtd
+				refused=$(echo -en "\n ${sdtd_telnet_shutdown}"|grep "Timeout or EOF")
+				if [ -n "${refused}" ]; then
+					fn_print_ok "Graceful: telnet: ${telnetip}: "
+					fn_print_ok_eol_nl
+					fn_scriptlog "Graceful: telnet: ${telnetip}: ${seconds} seconds"
+					break
+				fi
+				sleep 1
+				fn_print_dots "Graceful: rcon quit: ${seconds}"
+			done
+		# If telnet failed will go straight to tmux shutdown. 
+		# If cannot shutdown correctly world save may be lost
+		else
+			if [ -n "${refused}" ]; then
+				fn_print_fail "Graceful: telnet: "
+				fn_print_fail_eol_nl
+				fn_scriptlog "Graceful: telnet: ${telnetip}: FAIL"
+			else
+				fn_print_fail_nl "Graceful: telnet: Unknown error"
+				fn_scriptlog "Graceful: telnet: Unknown error"
+			fi
+			echo -en "\n\n" | tee -a "${scriptlog}"
+			echo -en "Telnet output:" | tee -a "${scriptlog}"
+			echo -en "\n ${sdtd_telnet_shutdown}" | tee -a "${scriptlog}"
+			echo -en "\n\n" | tee -a "${scriptlog}"
+		fi
+	else
+		fn_print_dots "Graceful: telnet: "
+		fn_scriptlog "Graceful: telnet: "
+		fn_print_fail "Graceful: telnet: expect not installed: "
+		fn_print_fail_eol_nl
+		fn_scriptlog "Graceful: telnet: expect not installed: FAIL"
+	fi
+	sleep 1
+	fn_stop_tmux
 }
 
 fn_stop_graceful_select(){
-if [ "${gamename}" == "7 Days to Die" ]; then
-	fn_stop_graceful_sdtd
-elif [ "${engine}" == "source" ]; then
-	fn_stop_graceful_source
-elif [ "${engine}" == "goldsource" ]; then
-	fn_stop_graceful_goldsource
-else
-	fn_stop_tmux
-fi		
+	if [ "${gamename}" == "7 Days To Die" ]; then
+		fn_stop_graceful_sdtd
+	elif [ "${engine}" == "source" ]; then
+		fn_stop_graceful_source
+	elif [ "${engine}" == "goldsource" ]; then
+		fn_stop_graceful_goldsource
+	else
+		fn_stop_tmux
+	fi		
 }
 
 fn_stop_teamspeak3(){
-fn_print_dots "${servername}"
-fn_scriptlog "${servername}"
-sleep 1
-${filesdir}/ts3server_startscript.sh stop > /dev/null 2>&1
-# Remove lock file
-rm -f "${rootdir}/${lockselfname}"
-fn_print_ok_nl "${servername}"
-fn_scriptlog "Stopped ${servername}"
-}
-
-fn_stop_tmux(){
-fn_print_dots "${servername}"
-fn_scriptlog "tmux kill-session: ${servername}"
-sleep 1
-# Kill tmux session
-tmux kill-session -t "${servicename}" > /dev/null 2>&1
-pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
-if [ "${pid}" == "0" ]; then
-	fn_print_ok_nl "${servername}"
-	fn_scriptlog "Stopped ${servername}"
+	fn_print_dots "${servername}"
+	fn_scriptlog "${servername}"
 	sleep 1
+	${filesdir}/ts3server_startscript.sh stop > /dev/null 2>&1
 	# Remove lock file
 	rm -f "${rootdir}/${lockselfname}"
-else
-	fn_print_fail_nl "Unable to stop${servername}"
-	fn_scriptlog "Unable to stop${servername}"
-fi
+	fn_print_ok_nl "${servername}"
+	fn_scriptlog "Stopped ${servername}"
+	}
 
+	fn_stop_tmux(){
+	fn_print_dots "${servername}"
+	fn_scriptlog "tmux kill-session: ${servername}"
+	sleep 1
+	# Kill tmux session
+	tmux kill-session -t "${servicename}" > /dev/null 2>&1
+	sleep 0.5
+	pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
+	if [ "${pid}" == "0" ]; then
+		# Remove lock file
+		rm -f "${rootdir}/${lockselfname}"
+		fn_print_ok_nl "${servername}"
+		fn_scriptlog "Stopped ${servername}"
+	else
+		fn_print_fail_nl "Unable to stop${servername}"
+		fn_scriptlog "Unable to stop${servername}"
+	fi
 }
 
 # checks if the server is already stopped before trying to stop.
 fn_stop_pre_check(){
-if [ "${gamename}" == "Teamspeak 3" ]; then
-	info_ts3status.sh
-	if [ "${ts3status}" = "No server running (ts3server.pid is missing)" ]; then
-		fn_print_ok_nl "${servername} is already stopped"
-		fn_scriptlog "${servername} is already stopped"
-	else
-		fn_stop_teamspeak3
-	fi      
-else
-	pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
-	if [ "${pid}" == "0" ]; then
-		fn_print_ok_nl "${servername} is already stopped"
-		fn_scriptlog "${servername} is already stopped"
+	if [ "${gamename}" == "Teamspeak 3" ]; then
+		info_ts3status.sh
+		if [ "${ts3status}" = "No server running (ts3server.pid is missing)" ]; then
+			fn_print_ok_nl "${servername} is already stopped"
+			fn_scriptlog "${servername} is already stopped"
+		else
+			fn_stop_teamspeak3
+		fi      
 	else
-		fn_stop_graceful_select
+		pid=$(tmux list-sessions 2>&1|awk '{print $1}'|grep -Ec "^${servicename}:")
+		if [ "${pid}" == "0" ]; then
+			fn_print_ok_nl "${servername} is already stopped"
+			fn_scriptlog "${servername} is already stopped"
+		else
+			fn_stop_graceful_select
+		fi
 	fi
-fi
 }
 
 check.sh