#!/bin/bash

# vim: dict=/usr/share/beakerlib/dictionary.vim cpt=.,w,b,u,t,i,k

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#

#   runtest.sh of /CoreOS/iproute/Sanity/bridge-utility

#   Description: Test basic bridge funcionality

#   Author: David Spurek <dspurek@redhat.com>

#

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#

#   Copyright (c) 2013 Red Hat, Inc. All rights reserved.

#

#   This copyrighted material is made available to anyone wishing

#   to use, modify, copy, or redistribute it subject to the terms

#   and conditions of the GNU General Public License version 2.

#

#   This program is distributed in the hope that it will be

#   useful, but WITHOUT ANY WARRANTY; without even the implied

#   warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR

#   PURPOSE. See the GNU General Public License for more details.

#

#   You should have received a copy of the GNU General Public

#   License along with this program; if not, write to the Free

#   Software Foundation, Inc., 51 Franklin Street, Fifth Floor,

#   Boston, MA 02110-1301, USA.

#

# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

# Include Beaker environment

. /usr/bin/rhts-environment.sh || exit 1

. /usr/share/beakerlib/beakerlib.sh || exit 1

PACKAGE="iproute"

PACKAGES="$PACKAGE"

rlIsRHEL 6 && PACKAGES=( ${PACKAGES[@]} "bridge-utils" )

vxlan_name="testvxlan"

bridge_name="testbridge"

lsmod | grep dummy

dummy_loaded=$?

rlJournalStart

    rlPhaseStartSetup

        # Check reqiured packages.

        for P in ${PACKAGES[@]}; do rlCheckRpm $P || rlDie "Package $P is missing"; done

        rlRun "TmpDir=\$(mktemp -d)" 0 "Creating tmp directory"

        rlRun "pushd $TmpDir"

        no_dummy=0

        if [ $dummy_loaded -eq 1 ] ; then

                # dummy module doesn't loaded before the test

                modprobe dummy numdummies=2

        else

                # dummy module loaded before the test, backup number of loaded dummy devices (nmdumies parameter), it is doesn't show under /sys/module/dummy/parameters

                dummies_count=`ip a | grep dummy | tail -n 1 | sed -r 's/.*dummy([0-9]+).*/\1/'`

                if [ -z $dummies_count] ; then

                        # dummy module is loaded but no dummy device exists

                        no_dummy=1

                else

                        # get correct count, dummy0 is the first

                        let "dummies_count=$dummies_count+1"

                fi

                rmmod dummy

                modprobe dummy numdummies=2

        fi

        rlRun "ip addr flush dev dummy0"

        rlRun "ip link set dummy0 up"

        rlRun "ip addr flush dev dummy1"

        rlRun "ip link set dummy1 up"

        rlRun "ip addr add 127.0.0.13/24 dev dummy0" 0 "Setting IPv4 address to 

dummy0 interface"

        rlRun "ip addr add 127.0.0.14/24 dev dummy1" 0 "Setting IPv4 address to 

dummy1 interface"

    rlPhaseEnd

    rlPhaseStartTest "Test bridge fdb basic funcionality with vxlan device"

        rlRun "ip link add $vxlan_name type vxlan id 10 group 239.0.0.10 ttl 4 dev dummy0" 0 "add vxlan interface"

        rlRun "ip addr add 192.168.1.1/24 broadcast 192.168.1.255 dev $vxlan_name" 0 "setting address to vxlan interface"

        rlRun "ip -d link show $vxlan_name" 0 "show details about vxlan device" 

        vxlan_ether_address=`ip -d link show $vxlan_name | grep link/ether | awk '{print $2}'`

        echo "ethernet address of vxlan device is: $vxlan_ether_address"

        # add new entry to bridge fdb database (device must by type vxlan)

        rlRun "bridge fdb add $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name"

        # check if entry was successfuly added

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2" bridge_show.out "-i"

        # try replace entry in bridge fdb database

        rlRun "bridge fdb replace $vxlan_ether_address dst 192.19.0.3 dev $vxlan_name"

        # check if entry was successfuly changed

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.3" bridge_show.out "-i"

        rlRun "bridge fdb del $vxlan_ether_address dev $vxlan_name"

        # check if entry was successfuly deleted

        # 'default' entry added by ip link command should be still listed

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertNotGrep "$vxlan_ether_address dst 192.19.0.2" bridge_show.out "-i"

        rlAssertGrep "dst 239.0.0.10 via dummy0" bridge_show.out "-i"

        # add new entry to bridge fdb database with port,vni and via options

        rlRun "bridge fdb add $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name port 10000 vni 100 via dummy0"

        # check if entry was successfuly added

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2 port 10000 vni 100 via dummy0" bridge_show.out "-i"

        rlRun "bridge fdb del $vxlan_ether_address dev $vxlan_name"

        # add new entry to bridge fdb database with self option

        rlRun "bridge fdb add $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name self"

        # check if entry was successfuly added

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2 self" bridge_show.out "-i"

        # replace entry in bridge fdb database with temp option

        rlRun "bridge fdb replace $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name temp"

        # check if entry was successfuly changed

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2 self static" bridge_show.out "-i"

        # replace entry in bridge fdb database with local option

        rlRun "bridge fdb replace $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name local"

        # check if entry was successfuly changed

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2 self permanent" bridge_show.out "-i"

        # replace entry in bridge fdb database with router option

        rlRun "bridge fdb replace $vxlan_ether_address dst 192.19.0.2 dev $vxlan_name router"

        # check if entry was successfuly changed

        bridge fdb show dev $vxlan_name &> bridge_show.out

        cat bridge_show.out

        rlAssertGrep "$vxlan_ether_address dst 192.19.0.2 self router permanent" bridge_show.out "-i"

        rlRun "bridge fdb del $vxlan_ether_address dev $vxlan_name"

        rlRun "ip link del $vxlan_name" 0

    rlPhaseEnd

    rlPhaseStartTest "Test bridge fdb basic funcionality with bridge device, test bridge link set command"

        # on rhels < 7 must be bridge device added with brctl (add type bridge is not supported)

        rlIsRHEL '>=7' && rlRun "ip link add $bridge_name type bridge" 0 || rlRun "brctl addbr $bridge_name" 0

        if rlIsRHEL '>=7'; then

                rlRun "ip link set dummy0 master $bridge_name" 0 "Add dummy interface to bridge"

                rlRun "bridge link show dev dummy0"

                # test bridge link set, command is not supported on rhel < 7 (->ndo_bridge_setlink() is not in our kernel.)

                rlRun "bridge link set dev dummy0 cost 10"

                rlRun "bridge link show dev dummy0 &> bridge_show.out" 0

                cat bridge_show.out

                rlAssertGrep "dummy0.*cost 10 $" bridge_show.out "-i"

                # add new entry to bridge fdb database with self option

                rlRun "bridge fdb add 00:1b:21:55:23:61 dev dummy0 self"

                bridge fdb show dev dummy0 &> bridge_show.out

                cat bridge_show.out

                rlAssertGrep "00:1b:21:55:23:61 self" bridge_show.out "-i"

                # add new entry to bridge fdb database with master option

                rlRun "bridge fdb add 00:1b:21:55:23:62 dev dummy0 master"

                bridge fdb show dev dummy0 &> bridge_show.out

                cat bridge_show.out

                rlAssertGrep "00:1b:21:55:23:62 vlan 1" bridge_show.out "-i"

                # add new entry to bridge fdb database with master and self options (entries for both should be added)

                rlRun "bridge fdb add 00:1b:21:55:23:63 dev dummy0 self master"

                bridge fdb show dev dummy0 &> bridge_show.out

                cat bridge_show.out

                rlAssertGrep "00:1b:21:55:23:63 self" bridge_show.out "-i"

                rlAssertGrep "00:1b:21:55:23:63 vlan 1" bridge_show.out "-i"

        else

                rlRun "brctl addif $bridge_name dummy0" 0 "Add dummy interface to bridge"

                rlRun "brctl show $bridge_name"

        fi

        rlIsRHEL '>=7' && rlRun "ip link set dummy0 nomaster" 0 "Remove dummy vlan interface from bridge" || rlRun "brctl delif $bridge_name dummy0" 0 "Remove dummy interface from bridge"

        rlIsRHEL '>=7' && rlRun "ip link del $bridge_name" 0 || rlRun "brctl delbr $bridge_name" 0

    rlPhaseEnd

    rlPhaseStartTest "Test bridge vlan basic funcionality"

        # on rhels < 7 must be bridge device added with brctl (add type bridge is not supported)

        rlIsRHEL '>=7' && rlRun "ip link add $bridge_name type bridge" 0 || rlRun "brctl addbr $bridge_name" 0

        rlRun "ip link add link dummy0 name dummy0.10 type vlan id 10"

        if rlIsRHEL '>=7' ; then

                rlRun "ip link set dummy0.10 master $bridge_name" 0 "Add dummy vlan interface to bridge"

                rlRun "bridge link show dev dummy0.10"

        else

                rlRun "brctl addif $bridge_name dummy0.10" 0 "Add dummy vlan interface to bridge"

                rlRun "brctl show $bridge_name"

        fi

        # bridge vlan is not supported on rhel < 6.8

        if rlIsRHEL '>=7' || rlIsRHEL '>=6.8' || rlIsFedora; then

            rlRun "bridge vlan add dev dummy0.10 vid 5" 0

        else

            rlRun "bridge vlan add dev dummy0.10 vid 5" 2

        fi

        # test correct funcionality only on rhel 7

        if rlIsRHEL '>=7' ; then

                bridge vlan &> bridge_vlan.out

                cat bridge_vlan.out

                #rlAssertGrep "dummy0.10.*5" bridge_vlan.out "-i"

                #rlAssertGrep "dummy0.10.*10" bridge_vlan.out "-i"

                rlRun "grep -A 2 'dummy0.10' bridge_vlan.out | grep '5'"

                rlRun "bridge vlan del dev dummy0.10 vid 5"

                bridge vlan &> bridge_vlan.out

                cat bridge_vlan.out

                #rlAssertNotGrep "dummy0.10" bridge_vlan.out "-i"

                rlRun "grep -A 2 'dummy0.10' bridge_vlan.out | grep '5'" 1

        fi

        rlIsRHEL '>=7' && rlRun "ip link set dummy0.10 nomaster" 0 "Remove dummy vlan interface from bridge" || rlRun "brctl delif $bridge_name dummy0.10" 0 "Remove dummy vlan interface from bridge"

        rlRun "ip link del dev dummy0.10"

        rlIsRHEL '>=7' && rlRun "ip link del $bridge_name" 0 || rlRun "brctl delbr $bridge_name" 0

    rlPhaseEnd

    rlPhaseStartTest "Test bridge mdb basic funcionality"

        rlRun "bridge mdb show" 0

    rlPhaseEnd

    if rlIsRHEL '>=7'; then

        rlPhaseStartTest

            rlRun "ip link add test_bridge type bridge"

            rlRun "bridge fdb show | grep 'dev test_bridge' | grep 'master test_bridge'"

            rlRun "ip link del test_bridge"

        rlPhaseEnd

    fi

    rlPhaseStartCleanup

        rlRun "ip route flush dev dummy0"

        rlRun "ip link set dummy0 down"

        rlRun "ip addr flush dev dummy0"

        rlRun "ip route flush dev dummy1"

        rlRun "ip link set dummy1 down"

        rlRun "ip addr flush dev dummy1"

        if [ $dummy_loaded -eq 1 ] ; then

                rmmod dummy

        else

                rmmod dummy

                if [ $no_dummy -eq 1 ] ; then

                        # load dummy module and delete dummy0 with ip link

                        modprobe dummy

                         rlIsRHEL '>=7' && rlRun "ip link del dummy0"

                else

                        modprobe dummy numdummies=$dummies_count

                fi

        fi

        rlRun "service network restart" 0,1 "Restarting network, just for sure"

        rlRun "popd"

        rlRun "rm -r $TmpDir" 0 "Removing tmp directory"

    rlPhaseEnd

rlJournalPrintText

rlJournalEnd