Quelle  mirror_lib.sh   Sprache: Shell

# SPDX-License-Identifier: GPL-2.0

mirror_install()
{
 local from_dev=$1; shift
 local direction=$1; shift
 local to_dev=$1; shift
 local filter=$1; shift

 tc filter add dev $from_dev $direction \
    pref 1000 $filter \
    action mirred egress mirror dev $to_dev
}

mirror_uninstall()
{
 local from_dev=$1; shift
 local direction=$1; shift

 tc filter del dev $swp1 $direction pref 1000
}

is_ipv6()
{
 local addr=$1; shift

 [[ -z ${addr//[0-9a-fA-F:]/} ]]
}

mirror_test()
{
 local vrf_name=$1; shift
 local sip=$1; shift
 local dip=$1; shift
 local dev=$1; shift
 local pref=$1; shift
 local expect=$1; shift

 if is_ipv6 $dip; then
  local proto=-6
  local type="icmp6 type=128" # Echo request.
 else
  local proto=
  local type="icmp echoreq"
 fi

 if [[ -z ${expect//[[:digit:]]/} ]]; then
  expect="== $expect"
 fi

 local t0=$(tc_rule_stats_get $dev $pref)
 $MZ $proto $vrf_name ${sip:+-A $sip} -B $dip -a own -b bc -q \
     -c 10 -d 100msec -t $type
 sleep 0.5
 local t1=$(tc_rule_stats_get $dev $pref)
 local delta=$((t1 - t0))
 ((delta $expect))
 check_err $? "Expected to capture $expect packets, got $delta."
}

do_test_span_dir_ips()
{
 local expect=$1; shift
 local dev=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift
 local forward_type=${1-8}; shift
 local backward_type=${1-0}; shift

 icmp_capture_install $dev "type $forward_type"
 mirror_test v$h1 $ip1 $ip2 $dev 100 $expect
 icmp_capture_uninstall $dev

 icmp_capture_install $dev "type $backward_type"
 mirror_test v$h2 $ip2 $ip1 $dev 100 $expect
 icmp_capture_uninstall $dev
}

quick_test_span_dir_ips()
{
 local dev=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift
 local forward_type=${1-8}; shift
 local backward_type=${1-0}; shift

 do_test_span_dir_ips 10 "$dev" "$ip1" "$ip2" \
        "$forward_type" "$backward_type"
}

test_span_dir_ips()
{
 local dev=$1; shift
 local forward_type=$1; shift
 local backward_type=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift

 quick_test_span_dir_ips "$dev" "$ip1" "$ip2" \
    "$forward_type" "$backward_type"

 icmp_capture_install $dev "type $forward_type"
 mirror_test v$h1 $ip1 $ip2 $dev 100 10
 icmp_capture_uninstall $dev

 icmp_capture_install $dev "type $backward_type"
 mirror_test v$h2 $ip2 $ip1 $dev 100 10
 icmp_capture_uninstall $dev
}

test_span_dir()
{
 local dev=$1; shift
 local forward_type=$1; shift
 local backward_type=$1; shift

 test_span_dir_ips "$dev" "$forward_type" "$backward_type" \
     192.0.2.1 192.0.2.2
}

do_test_span_vlan_dir_ips()
{
 local expect=$1; shift
 local dev=$1; shift
 local vid=$1; shift
 local ul_proto=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift

 # Install the capture as skip_hw to avoid double-counting of packets.
 # The traffic is meant for local box anyway, so will be trapped to
 # kernel.
 vlan_capture_install $dev "skip_hw vlan_id $vid vlan_ethtype $ul_proto"
 mirror_test v$h1 $ip1 $ip2 $dev 100 "$expect"
 mirror_test v$h2 $ip2 $ip1 $dev 100 "$expect"
 vlan_capture_uninstall $dev
}

quick_test_span_vlan_dir_ips()
{
 local dev=$1; shift
 local vid=$1; shift
 local ul_proto=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift

 do_test_span_vlan_dir_ips '>= 10' "$dev" "$vid" "$ul_proto" \
      "$ip1" "$ip2"
}

fail_test_span_vlan_dir_ips()
{
 local dev=$1; shift
 local vid=$1; shift
 local ul_proto=$1; shift
 local ip1=$1; shift
 local ip2=$1; shift

 do_test_span_vlan_dir_ips 0 "$dev" "$vid" "$ul_proto" "$ip1" "$ip2"
}

quick_test_span_vlan_dir()
{
 local dev=$1; shift
 local vid=$1; shift
 local ul_proto=$1; shift

 quick_test_span_vlan_dir_ips "$dev" "$vid" "$ul_proto" \
         192.0.2.1 192.0.2.2
}

fail_test_span_vlan_dir()
{
 local dev=$1; shift
 local vid=$1; shift
 local ul_proto=$1; shift

 fail_test_span_vlan_dir_ips "$dev" "$vid" "$ul_proto" \
        192.0.2.1 192.0.2.2
}