LXR linux/samples/bpf/test_cgrp2

   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