1#!/bin/sh
2# SPDX-License-Identifier: GPL-2.0
3
4# Test various socket options that can be set by attaching programs to cgroups.
5
6CGRP_MNT="/tmp/cgroupv2-test_cgrp2_sock"
7
8################################################################################
9#
10print_result()
11{
12	local rc=$1
13	local status=" OK "
14
15	[ $rc -ne 0 ] && status="FAIL"
16
17	printf "%-50s    [%4s]\n" "$2" "$status"
18}
19
20check_sock()
21{
22	out=$(test_cgrp2_sock)
23	echo $out | grep -q "$1"
24	if [ $? -ne 0 ]; then
25		print_result 1 "IPv4: $2"
26		echo "    expected: $1"
27		echo "        have: $out"
28		rc=1
29	else
30		print_result 0 "IPv4: $2"
31	fi
32}
33
34check_sock6()
35{
36	out=$(test_cgrp2_sock -6)
37	echo $out | grep -q "$1"
38	if [ $? -ne 0 ]; then
39		print_result 1 "IPv6: $2"
40		echo "    expected: $1"
41		echo "        have: $out"
42		rc=1
43	else
44		print_result 0 "IPv6: $2"
45	fi
46}
47
48################################################################################
49#
50
51cleanup()
52{
53	echo $$ >> ${CGRP_MNT}/cgroup.procs
54	rmdir ${CGRP_MNT}/sockopts
55}
56
57cleanup_and_exit()
58{
59	local rc=$1
60	local msg="$2"
61
62	[ -n "$msg" ] && echo "ERROR: $msg"
63
64	test_cgrp2_sock -d ${CGRP_MNT}/sockopts
65	ip li del cgrp2_sock
66	umount ${CGRP_MNT}
67
68	exit $rc
69}
70
71
72################################################################################
73# main
74
75rc=0
76
77ip li add cgrp2_sock type dummy 2>/dev/null
78
79set -e
80mkdir -p ${CGRP_MNT}
81mount -t cgroup2 none ${CGRP_MNT}
82set +e
83
84
85# make sure we have a known start point
86cleanup 2>/dev/null
87
88mkdir -p ${CGRP_MNT}/sockopts
89[ $? -ne 0 ] && cleanup_and_exit 1 "Failed to create cgroup hierarchy"
90
91
92# set pid into cgroup
93echo $$ > ${CGRP_MNT}/sockopts/cgroup.procs
94
95# no bpf program attached, so socket should show no settings
96check_sock "dev , mark 0, priority 0" "No programs attached"
97check_sock6 "dev , mark 0, priority 0" "No programs attached"
98
99# verify device is set
100#
101test_cgrp2_sock -b cgrp2_sock ${CGRP_MNT}/sockopts
102if [ $? -ne 0 ]; then
103	cleanup_and_exit 1 "Failed to install program to set device"
104fi
105check_sock "dev cgrp2_sock, mark 0, priority 0" "Device set"
106check_sock6 "dev cgrp2_sock, mark 0, priority 0" "Device set"
107
108# verify mark is set
109#
110test_cgrp2_sock -m 666 ${CGRP_MNT}/sockopts
111if [ $? -ne 0 ]; then
112	cleanup_and_exit 1 "Failed to install program to set mark"
113fi
114check_sock "dev , mark 666, priority 0" "Mark set"
115check_sock6 "dev , mark 666, priority 0" "Mark set"
116
117# verify priority is set
118#
119test_cgrp2_sock -p 123 ${CGRP_MNT}/sockopts
120if [ $? -ne 0 ]; then
121	cleanup_and_exit 1 "Failed to install program to set priority"
122fi
123check_sock "dev , mark 0, priority 123" "Priority set"
124check_sock6 "dev , mark 0, priority 123" "Priority set"
125
126# all 3 at once
127#
128test_cgrp2_sock -b cgrp2_sock -m 666 -p 123 ${CGRP_MNT}/sockopts
129if [ $? -ne 0 ]; then
130	cleanup_and_exit 1 "Failed to install program to set device, mark and priority"
131fi
132check_sock "dev cgrp2_sock, mark 666, priority 123" "Priority set"
133check_sock6 "dev cgrp2_sock, mark 666, priority 123" "Priority set"
134
135cleanup_and_exit $rc
136