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: stable/11/usr.bin/procstat/tests/procstat_test.sh 352492 2019-09-18 19:28:17Z jilles $
26#
27
28PROG_PID=
29PROG_PATH=$(atf_get_srcdir)/while1
30
31SP='[[:space:]]'
32
33start_program()
34{
35	echo "Starting program in background"
36	PROG_COMM=while1
37	PROG_PATH=$(atf_get_srcdir)/$PROG_COMM
38
39	mkfifo wait_for_start || atf_fail "mkfifo"
40	$PROG_PATH $* >wait_for_start &
41	PROG_PID=$!
42	if ! read dummy <wait_for_start; then
43		atf_fail "Program did not start properly"
44	fi
45	rm wait_for_start
46}
47
48atf_test_case binary_info
49binary_info_head()
50{
51	atf_set "descr" "Checks -b support"
52}
53binary_info_body()
54{
55	start_program bogus-arg
56
57	line_format="$SP*%s$SP+%s$SP+%s$SP+%s$SP*"
58	header_re=$(printf "$line_format" "PID" "COMM" "OSREL" "PATH")
59	line_re=$(printf "$line_format" $PROG_PID $PROG_COMM "[[:digit:]]+" "$PROG_PATH")
60
61	atf_check -o save:procstat.out procstat -b $PROG_PID
62
63	atf_check -o match:"$header_re" head -n 1 procstat.out
64	atf_check -o match:"$line_re" tail -n 1 procstat.out
65}
66
67atf_test_case command_line_arguments
68command_line_arguments_head()
69{
70	atf_set "descr" "Checks -c support"
71}
72command_line_arguments_body()
73{
74	arguments="my arguments"
75
76	start_program $arguments
77
78	line_format="$SP*%s$SP+%s$SP+%s$SP*"
79	header_re=$(printf "$line_format" "PID" "COMM" "ARGS")
80	line_re=$(printf "$line_format" $PROG_PID "$PROG_COMM" "$PROG_PATH $arguments")
81
82	atf_check -o save:procstat.out procstat -c $PROG_PID
83	atf_check -o match:"$header_re" head -n 1 procstat.out
84	atf_check -o match:"$line_re" tail -n 1 procstat.out
85}
86
87atf_test_case environment
88environment_head()
89{
90	atf_set "descr" "Checks -e support"
91}
92environment_body()
93{
94	var="MY_VARIABLE=foo"
95	eval "export $var"
96
97	start_program my arguments
98
99	line_format="$SP*%s$SP+%s$SP+%s$SP*"
100	header_re=$(printf "$line_format" "PID" "COMM" "ENVIRONMENT")
101	line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".*$var.*")
102
103	atf_check -o save:procstat.out procstat -e $PROG_PID
104
105	atf_check -o match:"$header_re" head -n 1 procstat.out
106	atf_check -o match:"$line_re" tail -n 1 procstat.out
107}
108
109atf_test_case file_descriptor
110file_descriptor_head()
111{
112	atf_set "descr" "Checks -f support"
113}
114file_descriptor_body()
115{
116	start_program my arguments
117
118	line_format="$SP*%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP+%s$SP%s$SP*"
119	header_re=$(printf "$line_format" "PID" "COMM" "FD" "T" "V" "FLAGS" "REF" "OFFSET" "PRO" "NAME")
120	# XXX: write a more sensible feature test
121	line_re=$(printf "$line_format" $PROG_PID $PROG_COMM ".+" ".+" ".+" ".+" ".+" ".+" ".+" ".+")
122
123	atf_check -o save:procstat.out procstat -f $PROG_PID
124
125	atf_check -o match:"$header_re" head -n 1 procstat.out
126	atf_check -o match:"$line_re" awk 'NR > 1' procstat.out
127}
128
129atf_test_case kernel_stacks
130kernel_stacks_head()
131{
132	atf_set "descr" "Captures kernel stacks for all visible threads"
133}
134kernel_stacks_body()
135{
136	# Bug 237445: checks will fail because of missing MFCs on branch
137	#atf_check -o save:procstat.out procstat -a kstack
138	#atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out
139
140	atf_check -o save:procstat.out procstat -kka
141	atf_check -o not-empty awk '{if ($3 == "procstat") print}' procstat.out
142}
143
144atf_init_test_cases()
145{
146	atf_add_test_case binary_info
147	atf_add_test_case command_line_arguments
148	atf_add_test_case environment
149	atf_add_test_case file_descriptor
150	atf_add_test_case kernel_stacks
151}
152