1# 2# Copyright (c) 2017 Ngie Cooper <ngie@FreeBSD.org> 3# All rights reserved. 4# 5# Redistribution and use in source and binary forms, with or without 6# modification, are permitted provided that the following conditions 7# are met: 8# 1. Redistributions of source code must retain the above copyright 9# notice, this list of conditions and the following disclaimer. 10# 2. Redistributions in binary form must reproduce the above copyright 11# notice, this list of conditions and the following disclaimer in the 12# documentation and/or other materials provided with the distribution. 13# 14# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 15# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 16# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 17# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 18# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 19# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 20# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 21# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 22# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 23# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 24# POSSIBILITY OF SUCH DAMAGE. 25# 26# $FreeBSD$ 27# 28 29MAX_TRIES=20 30PROG_PID= 31PROG_PATH=$(atf_get_srcdir)/while1 32 33SP='[[:space:]]' 34 35start_program() 36{ 37 echo "Starting program in background" 38 PROG_COMM=while1 39 PROG_PATH=$(atf_get_srcdir)/$PROG_COMM 40 41 $PROG_PATH $* & 42 PROG_PID=$! 43 try=0 44 while [ $try -lt $MAX_TRIES ] && ! kill -0 $PROG_PID; do 45 sleep 0.5 46 : $(( try += 1 )) 47 done 48 if [ $try -ge $MAX_TRIES ]; then 49 atf_fail "Polled for program start $MAX_TRIES tries and failed" 50 fi 51} 52 53atf_test_case binary_info 54binary_info_head() 55{ 56 atf_set "descr" "Checks -b support" 57} 58binary_info_body() 59{ 60 start_program bogus-arg 61 62 line_format="$SP*%s$SP+%s$SP+%s$SP+%s$SP*" 63 header_re=$(printf "$line_format" "PID" "COMM" "OSREL" "PATH") 64 line_re=$(printf "$line_format" $PROG_PID $PROG_COMM "[[:digit:]]+" "$PROG_PATH") 65 66 atf_check -o save:procstat.out procstat binary $PROG_PID 67 atf_check -o match:"$header_re" head -n 1 procstat.out 68 atf_check -o match:"$line_re" tail -n 1 procstat.out 69 70 atf_check -o save:procstat.out procstat -b $PROG_PID 71 atf_check -o match:"$header_re" head -n 1 procstat.out 72 atf_check -o match:"$line_re" tail -n 1 procstat.out 73} 74 75atf_test_case command_line_arguments 76command_line_arguments_head() 77{ 78 atf_set "descr" "Checks -c support" 79} 80command_line_arguments_body() 81{ 82 arguments="my arguments" 83 84 start_program $arguments 85 86 line_format="$SP*%s$SP+%s$SP+%s$SP*" 87 header_re=$(printf "$line_format" "PID" "COMM" "ARGS") 88 line_re=$(printf "$line_format" $PROG_PID "$PROG_COMM" "$PROG_PATH $arguments") 89 90 atf_check -o save:procstat.out procstat arguments $PROG_PID 91 atf_check -o match:"$header_re" head -n 1 procstat.out 92 atf_check -o match:"$line_re" tail -n 1 procstat.out 93 94 atf_check -o save:procstat.out procstat -c $PROG_PID 95 atf_check -o match:"$header_re" head -n 1 procstat.out 96 atf_check -o match:"$line_re" tail -n 1 procstat.out 97} 98 99atf_test_case environment 100environment_head() 101{ 102 atf_set "descr" "Checks -e support" 103} 104environment_body() 105{ 106 var="MY_VARIABLE=foo" 107 eval "export $var" 108 109 start_program my arguments 110 111 line_format="$SP*%s$SP+%s$SP+%s$SP*" 112 header_re=$(printf "$line_format" "PID" "COMM" "ENVIRONMENT") 113 line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".*$var.*") 114 115 atf_check -o save:procstat.out procstat environment $PROG_PID 116 atf_check -o match:"$header_re" head -n 1 procstat.out 117 atf_check -o match:"$line_re" tail -n 1 procstat.out 118 119 atf_check -o save:procstat.out procstat -e $PROG_PID 120 atf_check -o match:"$header_re" head -n 1 procstat.out 121 atf_check -o match:"$line_re" tail -n 1 procstat.out 122} 123 124atf_test_case file_descriptor 125file_descriptor_head() 126{ 127 atf_set "descr" "Checks -f support" 128} 129file_descriptor_body() 130{ 131 start_program my arguments 132 133 line_format="$SP*%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP%s$SP*" 134 header_re=$(printf "$line_format" "PID" "COMM" "FD" "T" "V" "FLAGS" "REF" "OFFSET" "PRO" "NAME") 135 # XXX: write a more sensible feature test 136 line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".+" ".+" ".+" ".+" ".+" ".+" ".+" ".+") 137 138 atf_check -o save:procstat.out procstat files $PROG_PID 139 atf_check -o match:"$header_re" head -n 1 procstat.out 140 atf_check -o match:"$line_re" awk 'NR > 1' procstat.out 141 142 atf_check -o save:procstat.out procstat -f $PROG_PID 143 atf_check -o match:"$header_re" head -n 1 procstat.out 144 atf_check -o match:"$line_re" awk 'NR > 1' procstat.out 145} 146 147atf_test_case kernel_stacks 148kernel_stacks_head() 149{ 150 atf_set "descr" "Captures kernel stacks for all visible threads" 151} 152kernel_stacks_body() 153{ 154 atf_check -o save:procstat.out procstat -a kstack 155 atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out 156 157 atf_check -o save:procstat.out procstat -kka 158 atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out 159} 160 161atf_init_test_cases() 162{ 163 atf_add_test_case binary_info 164 atf_add_test_case command_line_arguments 165 atf_add_test_case environment 166 atf_add_test_case file_descriptor 167 atf_add_test_case kernel_stacks 168} 169