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