common-bf.sh

function on_bf() {
    __on_remote $BF_IP "$@"
}

function on_remote_bf() {
    __on_remote $REMOTE_BF_IP "$@"
}

function on_bf_exec() {
    __on_remote_exec $BF_IP "$@"
}

function on_remote_bf_exec() {
    __on_remote_exec $REMOTE_BF_IP "$@"
}

function require_bf() {
    if [ -z "$BF_IP" ]; then
        fail "BF IP is not configured"
    fi

    log "BF $BF_IP"
    on_bf true || fail "BF command failed"
    print_remote_test_separator $BF_IP
    on_bf "echo MLNX_OFED \`modinfo --field version mlx5_core\` | tee -a /dev/kmsg"
}

function require_remote_bf() {
    if [ -z "$REMOTE_BF_IP" ]; then
        fail "Remote BF IP is not configured"
    fi

    log "Remote BF $REMOTE_BF_IP"
    on_remote_bf true || fail "Remote BF command failed"
}

function remote_bf_wrap() {
    if is_bf_host; then
        on_remote_bf "$@"
    else
        on_remote "$@"
    fi
}

function remote_bf_wrap_exec() {
    if is_bf_host; then
        on_remote_bf_exec "$@"
    else
        on_remote_exec "$@"
    fi
}

function bf_wrap() {
    if is_bf_host; then
        on_bf "$@"
    else
        eval "$@"
    fi
}

function bf_wrap_exec() {
    if is_bf_host; then
        on_bf_exec "$@"
    else
        eval "$@"
    fi
}

function __config_vf() {
    local ns=$1
    local vf=$2
    local ip=$3  # optional
    local mac=$4 # optional
    local prefix=24

    if [[ "$ip" == *":"* ]]; then
        # ipv6
        prefix=64
    fi

    ip netns add $ns
    ${mac:+ip link set $vf address $mac}
    ip link set $vf netns $ns
    ${ip:+ip -netns $ns address replace dev $vf $ip/$prefix}
    ip -netns $ns link set $vf up
}

function config_remote_bf_vf() {
    local ns=$1
    local vf=$2
    local rep=$3
    local ip=$4  # optional
    local mac=$5 # optional

    echo "[$ns] $vf (${mac:+$mac/}$ip) -> BF $rep"
    on_remote_exec "__config_vf $ns $vf $ip $mac"
    on_remote_bf_exec "__config_rep $rep"
}

function ovs-vsctl() {
    if is_bf_host; then
        on_bf command ovs-vsctl "$@"
    else
        command ovs-vsctl "$@"
    fi
}

function ovs-ofctl() {
    if is_bf_host; then
        local cmd="$@"
        # Note: Escape special character and remove any extra space after comma.
        cmd=$(echo $cmd | sed -e 's/)/\\)/g' -e 's/(/\\(/g' -e 's/->/\\-\\>/g' -e 's/\]/\\]/g' -e 's/\[/\\[/g' -e 's/}/\\}/g' -e 's/{/\\{/g' -e 's/, /,/g')
        on_bf command ovs-ofctl "$cmd"
    else
        command ovs-ofctl "$@"
    fi
}

function ovs-appctl() {
    if is_bf_host; then
        on_bf command ovs-appctl "$@"
    else
        command ovs-appctl "$@"
    fi
}

function __setup_common_bf() {
    if ! is_bf_host && ! is_bf ; then
        return
    fi

    BF_PCI=$(bf_wrap "basename \$(readlink /sys/class/net/${BF_NIC}/device)")
    BF_PCI2=$(bf_wrap "basename \$(readlink /sys/class/net/${BF_NIC2}/device)")

    bf_wrap """
        source /etc/os-release
        echo -e \"ARM ${ANSI_COLOR0}\${PRETTY_NAME}$NOCOLOR\"
        uname -a
    """

    echo "ARM NIC $BF_NIC PCI $BF_PCI"
}

__setup_common_bf