X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Futils%2Futils.sh;h=5b81febb6bfb5c52ba9e3eff2c451175603326ad;hp=3964e3b12820901957d68f1bf1bbe91497a99b88;hb=1c362dc78cf2e28c8935efcb5d4a85ef5d5967ba;hpb=b89c3a68195f2b41a803a66e8acc57ea8eac5289 diff --git a/tests/utils/utils.sh b/tests/utils/utils.sh index 3964e3b12..5b81febb6 100644 --- a/tests/utils/utils.sh +++ b/tests/utils/utils.sh @@ -16,8 +16,11 @@ # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA SESSIOND_BIN="lttng-sessiond" +SESSIOND_PIDS="" +RUNAS_BIN="lttng-runas" CONSUMERD_BIN="lttng-consumerd" RELAYD_BIN="lttng-relayd" +RELAYD_PIDS="" LTTNG_BIN="lttng" BABELTRACE_BIN="babeltrace" OUTPUT_DEST=/dev/null @@ -41,6 +44,22 @@ export LTTNG_SESSIOND_PATH="/bin/true" source $TESTDIR/utils/tap/tap.sh +function full_cleanup () +{ + if [ -n "${SESSIOND_PIDS}" ] || [ -n "${RELAYD_PIDS}" ]; then + kill -9 ${SESSIOND_PIDS} ${RELAYD_PIDS} > /dev/null 2>&1 + fi + + # Disable trap for SIGTERM since the following kill to the + # pidgroup will be SIGTERM. Otherwise it loops. + # The '-' before the pid number ($$) indicates 'kill' to signal the + # whole process group. + trap - SIGTERM && kill -- -$$ +} + + +trap full_cleanup SIGINT SIGTERM + function print_ok () { # Check if we are a terminal @@ -301,6 +320,8 @@ function start_lttng_relayd_opt() else pass "Start lttng-relayd (opt: $opt)" fi + + RELAYD_PIDS=$(pgrep --full lt-$RELAYD_BIN) } function start_lttng_relayd() @@ -317,12 +338,10 @@ function stop_lttng_relayd_opt() { local withtap=$1 - PID_RELAYD=`pgrep --full lt-$RELAYD_BIN` - if [ $withtap -eq "1" ]; then - diag "Killing lttng-relayd (pid: $PID_RELAYD)" + diag "Killing lttng-relayd (pid: $RELAYD_PIDS)" fi - kill $PID_RELAYD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST + kill $RELAYD_PIDS 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST retval=$? if [ $? -eq 1 ]; then @@ -340,6 +359,7 @@ function stop_lttng_relayd_opt() pass "Kill relay daemon" fi fi + RELAYD_PIDS="" return $retval } @@ -388,6 +408,7 @@ function start_lttng_sessiond_opt() ok $status "Start session daemon" fi fi + SESSIOND_PIDS=$(pgrep --full lt-$SESSIOND_BIN) } function start_lttng_sessiond() @@ -411,13 +432,15 @@ function stop_lttng_sessiond_opt() return fi - PID_SESSIOND=`pgrep --full lt-$SESSIOND_BIN` + local pids="${SESSIOND_PIDS} $(pgrep --full $RUNAS_BIN)" if [ -n "$2" ]; then kill_opt="$kill_opt -s $signal" fi - - kill $kill_opt $PID_SESSIOND 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST + if [ $withtap -eq "1" ]; then + diag "Killing lt-$SESSIOND_BIN pids: $(echo $pids | tr '\n' ' ')" + fi + kill $kill_opt $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST if [ $? -eq 1 ]; then if [ $withtap -eq "1" ]; then @@ -434,6 +457,8 @@ function stop_lttng_sessiond_opt() out=$(pgrep --full $CONSUMERD_BIN) sleep 0.5 done + + SESSIOND_PIDS="" if [ $withtap -eq "1" ]; then pass "Kill session daemon" fi @@ -450,6 +475,62 @@ function stop_lttng_sessiond_notap() stop_lttng_sessiond_opt 0 "$@" } +function sigstop_lttng_sessiond_opt() +{ + local withtap=$1 + local signal=SIGSTOP + local kill_opt="" + + if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then + # Env variable requested no session daemon + return + fi + + PID_SESSIOND="$(pgrep --full lt-$SESSIOND_BIN) $(pgrep --full $RUNAS_BIN)" + + kill_opt="$kill_opt -s $signal" + + if [ $withtap -eq "1" ]; then + diag "Sending SIGSTOP to lt-$SESSIOND_BIN pids: $(echo $PID_SESSIOND | tr '\n' ' ')" + fi + kill $kill_opt $PID_SESSIOND 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST + + if [ $? -eq 1 ]; then + if [ $withtap -eq "1" ]; then + fail "Sending SIGSTOP to session daemon" + fi + else + out=1 + while [ $out -ne 0 ]; do + pid=$(pgrep --full lt-$SESSIOND_BIN) + + # Wait until state becomes stopped for session + # daemon(s). + out=0 + for sessiond_pid in $pid; do + state=$(ps -p $sessiond_pid -o state= ) + if [[ -n "$state" && "$state" != "T" ]]; then + out=1 + fi + done + sleep 0.5 + done + if [ $withtap -eq "1" ]; then + pass "Sending SIGSTOP to session daemon" + fi + fi +} + +function sigstop_lttng_sessiond() +{ + sigstop_lttng_sessiond_opt 1 "$@" +} + +function sigstop_lttng_sessiond_notap() +{ + sigstop_lttng_sessiond_opt 0 "$@" +} + function stop_lttng_consumerd_opt() { local withtap=$1 @@ -463,7 +544,7 @@ function stop_lttng_consumerd_opt() fi if [ $withtap -eq "1" ]; then - diag "Killing lttng-consumerd (pid: $PID_CONSUMERD)" + diag "Killing $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')" fi kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST retval=$? @@ -507,6 +588,61 @@ function stop_lttng_consumerd_notap() stop_lttng_consumerd_opt 0 "$@" } +function sigstop_lttng_consumerd_opt() +{ + local withtap=$1 + local signal=SIGSTOP + local kill_opt="" + + PID_CONSUMERD=`pgrep --full $CONSUMERD_BIN` + + kill_opt="$kill_opt -s $signal" + + if [ $withtap -eq "1" ]; then + diag "Sending SIGSTOP to $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')" + fi + kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST + retval=$? + set +x + + if [ $? -eq 1 ]; then + if [ $withtap -eq "1" ]; then + fail "Sending SIGSTOP to consumer daemon" + fi + return 1 + else + out=1 + while [ $out -ne 0 ]; do + pid=$(pgrep --full $CONSUMERD_BIN) + + # Wait until state becomes stopped for all + # consumers. + out=0 + for consumer_pid in $pid; do + state=$(ps -p $consumer_pid -o state= ) + if [[ -n "$state" && "$state" != "T" ]]; then + out=1 + fi + done + sleep 0.5 + done + if [ $withtap -eq "1" ]; then + pass "Sending SIGSTOP to consumer daemon" + fi + fi + return $retval +} + +function sigstop_lttng_consumerd() +{ + sigstop_lttng_consumerd_opt 1 "$@" +} + +function sigstop_lttng_consumerd_notap() +{ + sigstop_lttng_consumerd_opt 0 "$@" +} + function list_lttng_with_opts () { local opts=$1 @@ -1195,3 +1331,37 @@ function validate_trace_empty() ret=$? return $ret } + +function metadata_regenerate () +{ + local expected_to_fail=$1 + local sess_name=$2 + + $TESTDIR/../src/bin/lttng/$LTTNG_BIN metadata regenerate -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST + ret=$? + if [[ $expected_to_fail -eq "1" ]]; then + test "$ret" -ne "0" + ok $? "Expected fail on regenerate $sess_name" + else + ok $ret "Metadata regenerate $sess_name" + fi +} + +function metadata_regenerate_ok () +{ + metadata_regenerate 0 "$@" +} + +function metadata_regenerate_fail () +{ + metadata_regenerate 1 "$@" +} + +function destructive_tests_enabled () +{ + if [ ${LTTNG_ENABLE_DESTRUCTIVE_TESTS} = "will-break-my-system" ]; then + return 0 + else + return 1 + fi +}