1 files changed, 94 insertions, 0 deletions
diff --git a/selftest/in_screen b/selftest/in_screen
new file mode 100755
index 0000000..d7d1b53
--- /dev/null
+++ b/selftest/in_screen
@@ -0,0 +1,94 @@
+#!/usr/bin/env bash
+
+export TMPDIR="$SELFTEST_TMPDIR"
+
+SERVERNAME="$ENVNAME"
+[ -z "$SERVERNAME" ] && SERVERNAME="base"
+basedir=$TMPDIR
+
+[ -r $basedir/$SERVERNAME.pid ] && {
+	for i in {2..100}; do
+		if [ ! -r "$basedir/${SERVERNAME}-$i.pid" ]; then
+			SERVERNAME="${SERVERNAME}-$i"
+			break
+		fi
+	done
+}
+
+rm -f $basedir/$SERVERNAME.{launch,log,parent.pid,pid,status}
+
+# set most of the environment vars we have in the screen session too
+_ENV=""
+printenv |
+	egrep -v '^TERMCAP|^WINDOW|^SHELL|^STY|^SHLVL|^SAMBA_VALGRIND|\$' |
+	egrep '^[A-Z]' |
+	sed "s/\(^[^=]*=\)\(.*\)/export \1'\2'/g" >$basedir/$SERVERNAME.vars
+
+cat <<EOF >$basedir/$SERVERNAME.launch
+cd $PWD
+ echo \$\$ > $basedir/$SERVERNAME.pid
+ . $basedir/$SERVERNAME.vars
+ echo "\$(date) starting $SERVERNAME" >> $basedir/$SERVERNAME.log
+ $@
+ echo \$? > $basedir/$SERVERNAME.status
+ read parent < $basedir/$SERVERNAME.parent.pid
+ kill \$parent
+EOF
+pid=$$
+
+cleanup()
+{
+	trap "exit 1" SIGINT SIGTERM SIGPIPE
+	[ -r $basedir/$SERVERNAME.status ] && {
+		read status <$basedir/$SERVERNAME.status
+		echo "$(date) samba exited with status $status" >>$basedir/$SERVERNAME.log
+		exit $status
+	}
+
+	case $ENVNAME in
+	*.nmbd | *.smbd | *.winbindd | *.samba | *.samba_dcerpcd)
+		kill $(cat $basedir/../"${ENVNAME%\.*}"/pid/"${ENVNAME##*\.}".pid)
+		;;
+	esac
+
+	read pid <$basedir/$SERVERNAME.pid
+	echo "$(date) Killing samba pid $pid from $$" >>$basedir/$SERVERNAME.log
+	if [ "$pid" = "$$" ]; then
+		exit 1
+	fi
+	kill -9 $pid 2>&1
+	exit 1
+}
+
+echo $$ >$basedir/$SERVERNAME.parent.pid
+trap cleanup SIGINT SIGTERM SIGPIPE
+
+if [[ "$TMUX" ]]; then
+	TMUX_CMD=tmux
+	if [[ $TMUX = *tmate* ]]; then
+		TMUX_CMD=tmate
+	fi
+
+	$TMUX_CMD new-window -n test:$SERVERNAME "bash $basedir/$SERVERNAME.launch"
+
+	# tmux seems to lag a bit for new sessions. Don't create them too
+	# quickly one after another
+	sleep .1
+else
+	screen -r -X screen -t test:$SERVERNAME bash $basedir/$SERVERNAME.launch
+fi
+echo "$(date) waiting in $$" >>$basedir/$SERVERNAME.log
+read stdin_var
+echo "$(date) EOF on stdin" >>$basedir/$SERVERNAME.log
+
+case $ENVNAME in
+*.nmbd | *.smbd | *.winbindd | *.samba | *.samba_dcerpcd)
+	kill $(cat $basedir/../"${ENVNAME%\.*}"/pid/"${ENVNAME##*\.}".pid)
+	;;
+esac
+
+read pid <$basedir/$SERVERNAME.pid
+echo "$(date) killing $pid" >>$basedir/$SERVERNAME.log
+kill $pid 2>/dev/null
+echo "$(date) exiting" >>$basedir/$SERVERNAME.log
+exit 0