#!/bin/bash
FAMILY=ixgbe
echo 1024 > /sys/devices/system/node/node0/hugepages/hugepages-2048kB/nr_hugepages
echo 1024 > /sys/devices/system/node/node1/hugepages/hugepages-2048kB/nr_hugepages

# Remove old modules (if loaded)
rmmod ixgbe
rmmod ixgbe_zc
rmmod pf_ring

# We assume that you have compiled PF_RING
insmod /usr/lib/modules/3.10.0-327.18.2.el7.x86_64/extra/pf_ring.ko transparent_mode=2

# Required by ixgbe
modprobe ptp
modprobe vxlan
modprobe dca

# As many queues as the number of processors
insmod /usr/lib/modules/3.10.0-327.18.2.el7.x86_64/extra/ixgbe_zc.ko RSS=12

sleep 1

killall irqbalance

INTERFACES=$(cat /proc/net/dev|grep 'enp5s0'|grep -v 'lo'|grep -v 'sit'|awk -F":" '{print $1}'|tr -d ' ')
for IF in $INTERFACES ; do
        TOCONFIG=$(ethtool -i $IF|grep $FAMILY|wc -l)
        if [ "$TOCONFIG" -eq 1 ]; then
                printf "Configuring %s\n" "$IF"
                ifconfig $IF up
                sleep 1
                bash /root/set_irq_affinity $IF

                # Max number of RX slots
                ethtool -G $IF rx 32768

                # Max number of TX slots
                ethtool -G $IF tx 32768

                # Disabling VLAN stripping
                ethtool -K $IF rxvlan off

                # Disabling Flow Control (actually it should be automatically disabled by the driver)
                #ethtool -A $IF autoneg off
                ethtool -A $IF rx off
                ethtool -A $IF tx off
                #ethtool -s $IF speed 10000

                # Enable n-tuple hw filters
                #ethtool -K $IF ntuple on

        fi
done

HUGEPAGES=1024
if [ `cat /proc/mounts | grep hugetlbfs | wc -l` -eq 0 ]; then
        sync && echo 3 > /proc/sys/vm/drop_caches
        echo $HUGEPAGES > /sys/kernel/mm/hugepages/hugepages-2048kB/nr_hugepages
        mkdir /mnt/huge
        mount -t hugetlbfs nodev /mnt/huge
fi
AVAILHUGEPAGES=$(grep HugePages_Total /sys/devices/system/node/node0/meminfo | cut -d ':' -f 2|sed 's/ //g')
if [ $AVAILHUGEPAGES -ne $HUGEPAGES ]; then
        printf "Warning: %s hugepages available, %s requested\n" "$AVAILHUGEPAGES" "$HUGEPAGES"
fi