test/scripts/run_in_venv_with_cleanup.sh

   1 #!/usr/bin/env bash
   2
   3 rv=0
   4
   5 # Minimalist version of cleanup, used for signal handling.
   6 # Sends a SIGKILL to the entire process group, including ourselves.
   7 # Needs just two external commands, making it more
   8 # robust in case of resource issues.
   9 panic() {
  10         echo "$0(pid $$): Caught a signal, emergency clean-up"
  11         # use "pgid:1=" output format to get unpadded process group ID
  12         group_id=`ps -p $$ -o pgid:1=`
  13         echo "$0(pid $$): sending kill to process group ID:${group_id}"
  14         kill -9 -- -${group_id}
  15         # not reached
  16 }
  17
  18 # Happy camper leisurely clean up - send the signal only to other
  19 # processes in the process group, and also check
  20 # that the processes exists before sending the signal.
  21 atexit() {
  22         group_id=`ps -p $$ -o pgid=`
  23         my_id=$$
  24         SED=`which gsed`
  25         SED=$(basename "${SED:-sed}")
  26         ids=`pgrep -g $group_id -d ' ' | ${SED} "s/\b$my_id\b//g"`
  27         echo "Killing possible remaining process IDs: $ids"
  28         for id in $ids
  29         do
  30                 if ps -p $id > /dev/null
  31                 then
  32                         kill -9 $id
  33                 fi
  34         done
  35         exit ${rv}
  36 }
  37
  38 trap "panic;" SIGINT SIGTERM
  39
  40 FORCE_FOREGROUND=$1
  41 shift
  42
  43 source $1
  44 shift
  45
  46 if [[ "${FORCE_FOREGROUND}" == "1" ]]
  47 then
  48         $*
  49 else
  50         $* &
  51         pid=$!
  52         wait ${pid}
  53 fi
  54
  55 rv=$?
  56 atexit
  57 exit ${rv}