test/scripts/run_in_venv_with_cleanup.sh

   1 #!/bin/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         ids=`pgrep -g $group_id -d ' ' | sed "s/\b$my_id\b//g"`
  25         echo "Killing possible remaining process IDs: $ids"
  26         for id in $ids
  27         do
  28                 if ps -p $id > /dev/null
  29                 then
  30                         kill -9 $id
  31                 fi
  32         done
  33         exit ${rv}
  34 }
  35
  36 trap "panic;" SIGINT SIGTERM
  37
  38 FORCE_FOREGROUND=$1
  39 shift
  40
  41 source $1
  42 shift
  43
  44 if [[ "${FORCE_FOREGROUND}" == "1" ]]
  45 then
  46         $*
  47 else
  48         $* &
  49         pid=$!
  50         wait ${pid}
  51 fi
  52
  53 rv=$?
  54 atexit
  55 exit ${rv}