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 atf_skip "https://bugs.freebsd.org/233587" 83 84 arguments="my arguments" 85 86 start_program $arguments 87 88 line_format="$SP*%s$SP+%s$SP+%s$SP*" 89 header_re=$(printf "$line_format" "PID" "COMM" "ARGS") 90 line_re=$(printf "$line_format" $PROG_PID "$PROG_COMM" "$PROG_PATH $arguments") 91 92 atf_check -o save:procstat.out procstat arguments $PROG_PID 93 atf_check -o match:"$header_re" head -n 1 procstat.out 94 atf_check -o match:"$line_re" tail -n 1 procstat.out 95 96 atf_check -o save:procstat.out procstat -c $PROG_PID 97 atf_check -o match:"$header_re" head -n 1 procstat.out 98 atf_check -o match:"$line_re" tail -n 1 procstat.out 99} 100 101atf_test_case environment 102environment_head() 103{ 104 atf_set "descr" "Checks -e support" 105} 106environment_body() 107{ 108 atf_skip "https://bugs.freebsd.org/233588" 109 110 var="MY_VARIABLE=foo" 111 eval "export $var" 112 113 start_program my arguments 114 115 line_format="$SP*%s$SP+%s$SP+%s$SP*" 116 header_re=$(printf "$line_format" "PID" "COMM" "ENVIRONMENT") 117 line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".*$var.*") 118 119 atf_check -o save:procstat.out procstat environment $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 atf_check -o save:procstat.out procstat -e $PROG_PID 124 atf_check -o match:"$header_re" head -n 1 procstat.out 125 atf_check -o match:"$line_re" tail -n 1 procstat.out 126} 127 128atf_test_case file_descriptor 129file_descriptor_head() 130{ 131 atf_set "descr" "Checks -f support" 132} 133file_descriptor_body() 134{ 135 start_program my arguments 136 137 line_format="$SP*%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP%s$SP*" 138 header_re=$(printf "$line_format" "PID" "COMM" "FD" "T" "V" "FLAGS" "REF" "OFFSET" "PRO" "NAME") 139 # XXX: write a more sensible feature test 140 line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".+" ".+" ".+" ".+" ".+" ".+" ".+" ".+") 141 142 atf_check -o save:procstat.out procstat files $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 atf_check -o save:procstat.out procstat -f $PROG_PID 147 atf_check -o match:"$header_re" head -n 1 procstat.out 148 atf_check -o match:"$line_re" awk 'NR > 1' procstat.out 149} 150 151atf_test_case kernel_stacks 152kernel_stacks_head() 153{ 154 atf_set "descr" "Captures kernel stacks for all visible threads" 155} 156kernel_stacks_body() 157{ 158 atf_check -o save:procstat.out procstat -a kstack 159 atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out 160 161 atf_check -o save:procstat.out procstat -kka 162 atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out 163} 164 165atf_init_test_cases() 166{ 167 atf_add_test_case binary_info 168 atf_add_test_case command_line_arguments 169 atf_add_test_case environment 170 atf_add_test_case file_descriptor 171 atf_add_test_case kernel_stacks 172} 173