Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 1 | #!/bin/bash -x |
| 2 | |
Ajay Lotan Thakur | 8ff4a2c | 2022-08-05 12:49:33 -0600 | [diff] [blame^] | 3 | # Copyright 2022-present Open Networking Foundation |
| 4 | # |
| 5 | # SPDX-License-Identifier: Apache-2.0 |
| 6 | |
| 7 | |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 8 | set -o errexit |
| 9 | set -o pipefail |
| 10 | set -o nounset |
| 11 | |
| 12 | exec 3>&1 |
| 13 | exec &>>/var/log/$(basename $0).log |
| 14 | |
| 15 | PATH="$CNI_PATH:$(dirname "${BASH_SOURCE[0]}"):$PATH" |
| 16 | CNI_CONF=$(cat /dev/stdin) |
| 17 | |
| 18 | get_peer_name() { |
| 19 | echo "$1-vdev" |
| 20 | } |
| 21 | |
| 22 | get_mac_with_vfpci() { |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 23 | local vfdir=$(find /sys/devices/pci* -name $1 | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 24 | local pf=$(readlink $vfdir/physfn | awk '{print substr($1,4)}') |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 25 | local pfdir=$(find /sys/devices/pci* -name $pf | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 26 | local pfName=$(ls $pfdir/net/ | head -1) |
| 27 | local idx=$(ls -l $pfdir | awk -v vf=$1 'substr($11,4)==vf {print substr($9,7)}') |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 28 | local mac=$(ip link show dev $pfName | awk -v idx="$idx" '$1=="vf" && $2==idx {print substr($4,1,17)}') |
| 29 | echo $mac |
| 30 | } |
| 31 | |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 32 | add_vlan() { |
| 33 | vfpci=$(echo $CNI_CONF | jq -r '.deviceID') |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 34 | local vfdir=$(find /sys/devices/pci* -name $vfpci | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 35 | local pf=$(readlink $vfdir/physfn | awk '{print substr($1,4)}') |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 36 | local pfdir=$(find /sys/devices/pci* -name $pf | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 37 | local pfName=$(ls $pfdir/net/ | head -1) |
| 38 | local idx=$(ls -l $pfdir | awk -v vf=$vfpci 'substr($11,4)==vf {print substr($9,7)}') |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 39 | |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 40 | vlan=$(echo $CNI_CONF | jq -r '.vlan // empty') |
| 41 | if [ ! -z "$vlan" ]; then |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 42 | ip link set $pfName vf $idx vlan $vlan |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 43 | fi |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 44 | } |
| 45 | |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 46 | ipam() { |
| 47 | local plugin=$(echo $CNI_CONF | jq -r '.ipam.type') |
| 48 | local res=$(echo $"$CNI_CONF" | "$plugin" | jq -c '.') |
| 49 | echo $res |
| 50 | } |
| 51 | |
| 52 | add_pair_ns() { |
| 53 | vfpci=$(echo $CNI_CONF | jq -r '.deviceID') |
Hyunsun Moon | 1711b2c | 2021-05-07 15:31:01 -0700 | [diff] [blame] | 54 | mac=$(echo $CNI_CONF | jq -r '.runtimeConfig.mac // empty') |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 55 | if [ -z "$mac" ]; then |
| 56 | mac=$(get_mac_with_vfpci $vfpci) |
| 57 | fi |
| 58 | peer=$(get_peer_name $CNI_IFNAME) |
| 59 | ip=$1 |
| 60 | |
| 61 | mkdir -p /var/run/netns/ |
| 62 | ln -sfT $CNI_NETNS /var/run/netns/$CNI_CONTAINERID |
| 63 | |
| 64 | ip netns exec $CNI_CONTAINERID ip link add $CNI_IFNAME type veth peer name $peer |
| 65 | ip netns exec $CNI_CONTAINERID ip link set $CNI_IFNAME addr $mac up alias $vfpci |
| 66 | ip netns exec $CNI_CONTAINERID ip link set $peer up |
| 67 | ip netns exec $CNI_CONTAINERID ip addr add $ip dev $CNI_IFNAME |
| 68 | } |
| 69 | |
| 70 | delete_pair_ns() { |
| 71 | ip netns exec $CNI_CONTAINERID ip link del $CNI_IFNAME |
| 72 | } |
| 73 | |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 74 | delete_vlan() { |
| 75 | vfpci=$(echo $CNI_CONF | jq -r '.deviceID') |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 76 | local vfdir=$(find /sys/devices/pci* -name $vfpci | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 77 | local pf=$(readlink $vfdir/physfn | awk '{print substr($1,4)}') |
Hyunsun Moon | fa4c8e4 | 2022-04-11 23:29:26 -0700 | [diff] [blame] | 78 | local pfdir=$(find /sys/devices/pci* -name $pf | grep -v iommu) |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 79 | local pfName=$(ls $pfdir/net/ | head -1) |
| 80 | local idx=$(ls -l $pfdir | awk -v vf=$vfpci 'substr($11,4)==vf {print substr($9,7)}') |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 81 | |
Hyunsun Moon | 56d7092 | 2021-11-19 10:38:14 -0800 | [diff] [blame] | 82 | vlan=$(echo $CNI_CONF | jq -r '.vlan // empty') |
| 83 | if [ ! -z "$vlan" ]; then |
| 84 | ip link set $pfName vf $idx vlan 0 |
| 85 | fi |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 86 | } |
| 87 | |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 88 | case $CNI_COMMAND in |
| 89 | ADD) |
| 90 | res=$(ipam) |
| 91 | ip=$(echo $res | jq -r '.ips[0].address') |
| 92 | add_pair_ns $ip |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 93 | add_vlan |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 94 | echo '{"cniVersion":"0.3.1"}' | jq -c --arg ip $ip '.ips[0].address = $ip' >&3 |
| 95 | ;; |
| 96 | DEL) |
| 97 | set +o errexit |
| 98 | ipam |
| 99 | delete_pair_ns |
Hyunsun Moon | 6d8b21b | 2021-05-20 18:12:09 -0700 | [diff] [blame] | 100 | delete_vlan |
Hyunsun Moon | f9cac59 | 2020-11-10 00:46:41 -0800 | [diff] [blame] | 101 | set -o errexit |
| 102 | ;; |
| 103 | *) |
| 104 | echo "CNI_COMMAND=[ADD|DEL] only supported" |
| 105 | exit 1 |
| 106 | ;; |
| 107 | esac |