trafficgen-base

#!/bin/bash

if ! source ${TOOLBOX_HOME}/bash/library/bench-base; then echo "ERROR: Could not source bench-base from \$TOOLBOX_HOME [${TOOLBOX_HOME}]"; exit 1; fi

function validate_other_prereqs() {
    if [ ! -e /dev/hugepages ]; then
        echo "ERROR: could not find /dev/hugepages"
        exit 1
    fi
}

function expand_number_list() {
    # expand a list of numbers to no longer include a range
    # ie. "1-3,5" becomes "1,2,3,5"
    local range=$1
    local list=""
    local items=""
    local item=""
    for items in $(echo "$range" | sed -e "s/,/ /g"); do
	if echo $items | grep -q -- "-"; then
	    items=$(echo "$items" | sed -e "s/-/ /")
	    items=$(seq $items)
	fi
	for item in $items; do
	    list="$list,$item"
	done
    done
    list=$(echo "$list" | sed -e "s/^,//")
    echo "$list"
}

function separate_comma_list() {
    echo "$1" | sed -e "s/,/ /g"
}

function resolve_device() {
    typeset -n resolved_dev=$1; shift # caller-provided variable name (call-by-reference)
    resolved_dev=""
    local dev=$1; shift
    local count=$1; shift
    pciloc_re='^[0-9,a-f]{4}:[0-9,a-f]{2}:[0-9,a-f]{2}\.[0-9,a-f]{1}$' # 0001:0a.b
    #envvar_re='^VAR:(\w+)[:(\d+)]{0,1}' # VAR:SOME_ENV_VAR or VAR:SOME_ENV_VAR:<number>
    envvar_re='^VAR:(\w+)(:(.+)){0,1}' # VAR:SOME_ENV_VAR or VAR:SOME_ENV_VAR:<number>
    devtype_re='^([virtio|vf]-\d+)$' # virtio-2
    echo "Resolving device [$dev]"
    if [[ "$dev" =~ $envvar_re ]]; then
        envvar="${BASH_REMATCH[1]}"
        position="${BASH_REMATCH[3]}"
        if [ -z "$position" ]; then
            position=1
        fi
        echo "Device [$dev] matched for environment variable pattern"
        echo "Looking for environment varible [$envvar]"
        # On some endpoints (k8s/openshift) the device is described as a
        # environment variable.  Users can specify which variable this is
        # in the benchmark params with
        #    "client-devices": "VAR:<varname>,VAR:<varname>"
        # Addtionally, when specifying the k8s enpoint for a trafficgen
        # client, the user must include a annotation for the pod which
        # ensures the pass-through device(s) are provisioned to the pod.
        # This of course requires that MULTUS networking is configured
        # for the cluster.
        if env | grep -q "^${envvar}="; then
            # There may be multiple devices listed.  Take the first and leave the rest.
            # If a user specifies this env var for multple devices, then they will
            # be used in order of appearance in the env var's value.
            dev=`env | grep "^${envvar}=" | awk -F= '{print $2}' | cut -d, -f$position`
            echo "device [$dev] found in environment variable [$envvar]"
            #local remaining=`env | grep "^${envvar}=" | awk -F= '{print $2}' | sed -e s/^$dev// -e s/^,//`
            #echo "$envvar is now: $remaining"
            #export $envvar="$remaining"
        else
            echo "The variable string [$envvar] does not match any environment variable"
            echo "Here are all of the envronment variables:"
            env
            echo -e "\n\nExiting"
            exit 1
        fi
    elif [[ "$dev" =~ $devtype_re ]]; then
        echo "Searching for the Nth virtio or VF device is not yet supported, exiting"
        exit 1
    else
        echo "Device [$dev] did not match regex for environemnt variable [$envvar_re] or devtype [$devtype_re]"
    fi
    # At this point $dev should only be a PCI location ID
    if [[ "$dev" =~ $pciloc_re ]]; then
	echo "The device [$dev] matched the regex for a PCI location ID"
        resolved_dev="$dev"
    else
        echo "The device [$dev] did not match the regex for a PCI location ID [$pciloc_re], exiting"
        exit 1
    fi
}