usr.bin/procstat/tests/procstat_test.sh

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