scripts/lttng-tools/hang_processes.sh

   1 #!/bin/bash
   2 #
   3 # Copyright (C) 2018 - Jonathan Rajotte-Julien <jonathan.rajotte-julien@efficios.com>
   4 #
   5 # This program is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 set -exu
  19
  20 PGREP=pgrep
  21 pids=""
  22 file_list=$(mktemp)
  23 ret=0
  24
  25 # WORKSPACE is normally set by Jenkins.  Use the current directory otherwise,
  26 # like when testing this script manually.
  27 WORKSPACE=${WORKSPACE:-$PWD}
  28
  29 lttng_processes="$("$PGREP" -l 'lttng|gen-ust-.+')" || true
  30 if [ -n "$lttng_processes" ]; then
  31
  32     pids="$(cut -d ' ' -f 1 <<< "$lttng_processes" | tr '\n' ' ')"
  33     echo "The following LTTng processes were detected running on the system and will be aborted:"
  34     echo "$lttng_processes"
  35
  36     # Stop the processes to make sure everything is frozen
  37     kill -SIGSTOP $pids
  38     kill -SIGABRT $pids
  39     kill -SIGCONT $pids
  40     ret=1
  41 fi
  42
  43 # Add the file passed as $1 to the list of files to collect.
  44 #
  45 # If that file is a symlink, follow it and collect the target, recursively.
  46
  47 function collect_recursive
  48 {
  49     file_to_collect=$1
  50
  51     if [ -f "$file_to_collect" ]; then
  52         echo "$file_to_collect" >> "$file_list"
  53
  54         if [ -L "$file_to_collect" ]; then
  55             collect_recursive "$(readlink "$file_to_collect")"
  56         fi
  57     fi
  58 }
  59
  60 # For each core file...
  61 while read -r core_file; do
  62     # Make sure the coredump is finished using fuser.
  63     while fuser "$core_file"; do
  64         sleep 1
  65     done
  66
  67     # Collect everything in the core file that looks like a reference to a
  68     # shared lib.
  69     strings "$core_file" | grep '^/.*\.so.*' | while read -r str; do
  70         collect_recursive "$str"
  71     done
  72
  73     echo "$core_file" >> $file_list
  74     ret=1
  75 done < <(find "/tmp" -maxdepth 1 -name "core\.[0-9]*" -type f 2>/dev/null)
  76
  77 # If we recorded some files to collect, pack them up.
  78 if [ -s "$file_list" ]; then
  79     mkdir -p "${WORKSPACE}/build"
  80     tar cfzh "${WORKSPACE}/build/core.tar.gz" -T <(sort "$file_list" | uniq)
  81 fi
  82
  83 # Remove core file
  84 while read -r core_file; do
  85         rm -rf "$core_file"
  86 done < <(find "/tmp" -maxdepth 1 -name "core\.[0-9]*" -type f 2>/dev/null)
  87
  88 rm -f "$file_list"
  89 exit $ret