1143880Spjd#!/bin/sh
2143880Spjd# $FreeBSD$
3143880Spjd
4272549Srodrigcjail_name_to_jid()
5272549Srodrigc{
6272549Srodrigc	local check_name="$1"
7283312Srodrigc	jls -j "$check_name" -s | tr ' ' '\n' | grep jid= | sed -e 's/.*=//g'
8272549Srodrigc}
9143880Spjd
10272549Srodrigcbase=pgrep_j_test
11272549Srodrigc
12283312Srodrigcif [ `id -u` -ne 0 ]; then
13283312Srodrigc	echo "1..0 # skip Test needs uid 0."
14283312Srodrigc	exit 0
15283312Srodrigcfi
16283312Srodrigc
17287269Sjamieecho "1..4"
18143880Spjd
19283312Srodrigcsleep=$(pwd)/sleep.txt
20283312Srodrigcln -sf /bin/sleep $sleep
21283312Srodrigc
22143880Spjdname="pgrep -j <jid>"
23283312Srodrigcsleep_amount=5
24283312Srodrigcjail -c path=/ name=${base}_1_1 ip4.addr=127.0.0.1 \
25283312Srodrigc    command=daemon -p ${PWD}/${base}_1_1.pid $sleep $sleep_amount &
26272549Srodrigc
27283312Srodrigcjail -c path=/ name=${base}_1_2 ip4.addr=127.0.0.1 \
28283312Srodrigc    command=daemon -p ${PWD}/${base}_1_2.pid $sleep $sleep_amount &
29272549Srodrigc
30283312Srodrigcfor i in `seq 1 10`; do
31272549Srodrigc	jid1=$(jail_name_to_jid ${base}_1_1)
32272549Srodrigc	jid2=$(jail_name_to_jid ${base}_1_2)
33272549Srodrigc	jid="${jid1},${jid2}"
34283312Srodrigc	case "$jid" in
35283312Srodrigc	[0-9]+,[0-9]+)
36283312Srodrigc		break
37283312Srodrigc		;;
38283312Srodrigc	esac
39283312Srodrigc	sleep 0.1
40283312Srodrigcdone
41283312Srodrigcsleep 0.5
42283312Srodrigc
43283312Srodrigcpid1="$(pgrep -f -x -j "$jid" "$sleep $sleep_amount" | sort)"
44283312Srodrigcpid2=$(printf "%s\n%s" "$(cat ${PWD}/${base}_1_1.pid)" \
45283312Srodrigc    $(cat ${PWD}/${base}_1_2.pid) | sort)
46283312Srodrigcif [ "$pid1" = "$pid2" ]; then
47283312Srodrigc	echo "ok 1 - $name"
48143880Spjdelse
49283312Srodrigc	echo "not ok 1 - $name # pgrep output: '$(echo $pid1)', pidfile output: '$(echo $pid2)'"
50143880Spjdfi
51283312Srodrigc[ -f ${PWD}/${base}_1_1.pid ] && kill $(cat ${PWD}/${base}_1_1.pid)
52283312Srodrigc[ -f ${PWD}/${base}_1_2.pid ] && kill $(cat ${PWD}/${base}_1_2.pid)
53283312Srodrigcwait
54143880Spjd
55164558Syarname="pgrep -j any"
56283312Srodrigcsleep_amount=6
57283312Srodrigcjail -c path=/ name=${base}_2_1 ip4.addr=127.0.0.1 \
58283312Srodrigc    command=daemon -p ${PWD}/${base}_2_1.pid $sleep $sleep_amount &
59272549Srodrigc
60283312Srodrigcjail -c path=/ name=${base}_2_2 ip4.addr=127.0.0.1 \
61283312Srodrigc    command=daemon -p ${PWD}/${base}_2_2.pid $sleep $sleep_amount &
62272549Srodrigc
63283312Srodrigcsleep 2
64283312Srodrigcpid1="$(pgrep -f -x -j any "$sleep $sleep_amount" | sort)"
65283312Srodrigcpid2=$(printf "%s\n%s" "$(cat ${PWD}/${base}_2_1.pid)" \
66283312Srodrigc    $(cat ${PWD}/${base}_2_2.pid) | sort)
67283312Srodrigcif [ "$pid1" = "$pid2" ]; then
68283312Srodrigc	echo "ok 2 - $name"
69143880Spjdelse
70283312Srodrigc	echo "not ok 2 - $name # pgrep output: '$(echo $pid1)', pidfile output: '$(echo $pid2)'"
71143880Spjdfi
72283312Srodrigc[ -f ${PWD}/${base}_2_1.pid ] && kill $(cat ${PWD}/${base}_2_1.pid)
73283312Srodrigc[ -f ${PWD}/${base}_2_2.pid ] && kill $(cat ${PWD}/${base}_2_2.pid)
74283312Srodrigcwait
75164558Syar
76164558Syarname="pgrep -j none"
77283312Srodrigcsleep_amount=7
78283312Srodrigcdaemon -p ${PWD}/${base}_3_1.pid $sleep $sleep_amount &
79283312Srodrigcjail -c path=/ name=${base}_3_2 ip4.addr=127.0.0.1 \
80283312Srodrigc    command=daemon -p ${PWD}/${base}_3_2.pid $sleep $sleep_amount &
81283312Srodrigcsleep 2
82283312Srodrigcpid="$(pgrep -f -x -j none "$sleep $sleep_amount")"
83283312Srodrigcif [ "$pid" = "$(cat ${PWD}/${base}_3_1.pid)" ]; then
84283312Srodrigc	echo "ok 3 - $name"
85164558Syarelse
86283312Srodrigc	echo "not ok 3 - $name # pgrep output: '$(echo $pid1)', pidfile output: '$(echo $pid2)'"
87164558Syarfi
88283312Srodrigc[ -f ${PWD}/${base}_3_1.pid ] && kill $(cat $PWD/${base}_3_1.pid) 
89283312Srodrigc[ -f ${PWD}/${base}_3_2.pid ] && kill $(cat $PWD/${base}_3_2.pid) 
90287269Sjamiewait
91283312Srodrigc
92287269Sjamie# test 4 is like test 1 except with jname instead of jid.
93287269Sjamiename="pgrep -j <jname>"
94287269Sjamiesleep_amount=8
95287269Sjamiejail -c path=/ name=${base}_4_1 ip4.addr=127.0.0.1 \
96287269Sjamie    command=daemon -p ${PWD}/${base}_4_1.pid $sleep $sleep_amount &
97287269Sjamie
98287269Sjamiejail -c path=/ name=${base}_4_2 ip4.addr=127.0.0.1 \
99287269Sjamie    command=daemon -p ${PWD}/${base}_4_2.pid $sleep $sleep_amount &
100287269Sjamie
101287269Sjamiesleep 0.5
102287269Sjamie
103287269Sjamiejname="${base}_4_1,${base}_4_2"
104287269Sjamiepid1="$(pgrep -f -x -j "$jname" "$sleep $sleep_amount" | sort)"
105287269Sjamiepid2=$(printf "%s\n%s" "$(cat ${PWD}/${base}_4_1.pid)" \
106287269Sjamie    $(cat ${PWD}/${base}_4_2.pid) | sort)
107287269Sjamieif [ "$pid1" = "$pid2" ]; then
108287269Sjamie	echo "ok 4 - $name"
109287269Sjamieelse
110287269Sjamie	echo "not ok 4 - $name # pgrep output: '$(echo $pid1)', pidfile output: '$(echo $pid2)'"
111287269Sjamiefi
112287269Sjamie[ -f ${PWD}/${base}_4_1.pid ] && kill $(cat ${PWD}/${base}_4_1.pid)
113287269Sjamie[ -f ${PWD}/${base}_4_2.pid ] && kill $(cat ${PWD}/${base}_4_2.pid)
114287269Sjamiewait
115287269Sjamie
116283312Srodrigcrm -f $sleep
117