summaryrefslogtreecommitdiffstats
path: root/test/scripts/run_in_venv_with_cleanup.sh
blob: 454c9c9c4ad12431672877c49ab3ff6cd6262e1d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#!/usr/bin/env bash

rv=0

# Minimalist version of cleanup, used for signal handling.
# Sends a SIGKILL to the entire process group, including ourselves.
# Needs just two external commands, making it more
# robust in case of resource issues.
panic() {
	echo "$0(pid $$): Caught a signal, emergency clean-up"
	# use "pgid:1=" output format to get unpadded process group ID
	group_id=`ps -p $$ -o pgid:1=`
	echo "$0(pid $$): sending kill to process group ID:${group_id}"
	kill -9 -- -${group_id}
	# not reached
}

# Happy camper leisurely clean up - send the signal only to other
# processes in the process group, and also check
# that the processes exists before sending the signal.
atexit() {
	group_id=`ps -p $$ -o pgid=`
	my_id=$$
	ids=`pgrep -g $group_id -d ' ' | gsed "s/\b$my_id\b//g"`
	echo "Killing possible remaining process IDs: $ids"
	for id in $ids
	do
		if ps -p $id > /dev/null
		then
			kill -9 $id
		fi
	done
	exit ${rv}
}

trap "panic;" SIGINT SIGTERM

FORCE_FOREGROUND=$1
shift

source $1
shift

if [[ "${FORCE_FOREGROUND}" == "1" ]]
then
	$*
else
	$* &
	pid=$!
	wait ${pid}
fi

rv=$?
atexit
exit ${rv}