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