| 1 | #!/bin/bash -exu |
| 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 | PGREP=pgrep |
| 19 | pids="" |
| 20 | file_list=$(mktemp) |
| 21 | ret=0 |
| 22 | |
| 23 | # WORKSPACE is normally set by Jenkins. Use the current directory otherwise, |
| 24 | # like when testing this script manually. |
| 25 | WORKSPACE=${WORKSPACE:-$PWD} |
| 26 | |
| 27 | lttng_processes="$("$PGREP" -l 'lttng|gen-ust-.+')" || true |
| 28 | if [ -n "$lttng_processes" ]; then |
| 29 | |
| 30 | pids="$(cut -d ' ' -f 1 <<< "$lttng_processes" | tr '\n' ' ')" |
| 31 | echo "The following LTTng processes were detected running on the system and will be aborted:" |
| 32 | echo "$lttng_processes" |
| 33 | |
| 34 | # Stop the processes to make sure everything is frozen |
| 35 | kill -SIGSTOP $pids |
| 36 | kill -SIGABRT $pids |
| 37 | kill -SIGCONT $pids |
| 38 | ret=1 |
| 39 | fi |
| 40 | |
| 41 | # Add the file passed as $1 to the list of files to collect. |
| 42 | # |
| 43 | # If that file is a symlink, follow it and collect the target, recursively. |
| 44 | |
| 45 | function collect_recursive |
| 46 | { |
| 47 | file_to_collect=$1 |
| 48 | |
| 49 | if [ -f "$file_to_collect" ]; then |
| 50 | echo "$file_to_collect" >> "$file_list" |
| 51 | |
| 52 | if [ -L "$file_to_collect" ]; then |
| 53 | collect_recursive "$(readlink "$file_to_collect")" |
| 54 | fi |
| 55 | fi |
| 56 | } |
| 57 | |
| 58 | # For each core file... |
| 59 | while read -r core_file; do |
| 60 | # Make sure the coredump is finished using fuser. |
| 61 | while fuser "$core_file"; do |
| 62 | sleep 1 |
| 63 | done |
| 64 | |
| 65 | # Collect everything in the core file that looks like a reference to a |
| 66 | # shared lib. |
| 67 | strings "$core_file" | grep '^/.*\.so.*' | while read -r str; do |
| 68 | collect_recursive "$str" |
| 69 | done |
| 70 | |
| 71 | echo "$core_file" >> $file_list |
| 72 | ret=1 |
| 73 | done < <(find "/tmp" -maxdepth 1 -name "core\.[0-9]*" -type f 2>/dev/null) |
| 74 | |
| 75 | # If we recorded some files to collect, pack them up. |
| 76 | if [ -s "$file_list" ]; then |
| 77 | mkdir -p "${WORKSPACE}/build" |
| 78 | tar cfzh "${WORKSPACE}/build/core.tar.gz" -T <(sort "$file_list" | uniq) |
| 79 | fi |
| 80 | |
| 81 | # Remove core file |
| 82 | while read -r core_file; do |
| 83 | rm -rf "$core_file" |
| 84 | done < <(find "/tmp" -maxdepth 1 -name "core\.[0-9]*" -type f 2>/dev/null) |
| 85 | |
| 86 | rm -f "$file_list" |
| 87 | exit $ret |