cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

msg_zerocopy.sh (3012B)

      1#!/bin/bash
      2#
      3# Send data between two processes across namespaces
      4# Run twice: once without and once with zerocopy
      5
      6set -e
      7
      8readonly DEV="veth0"
      9readonly DEV_MTU=65535
     10readonly BIN="./msg_zerocopy"
     11
     12readonly RAND="$(mktemp -u XXXXXX)"
     13readonly NSPREFIX="ns-${RAND}"
     14readonly NS1="${NSPREFIX}1"
     15readonly NS2="${NSPREFIX}2"
     16
     17readonly SADDR4='192.168.1.1'
     18readonly DADDR4='192.168.1.2'
     19readonly SADDR6='fd::1'
     20readonly DADDR6='fd::2'
     21
     22readonly path_sysctl_mem="net.core.optmem_max"
     23
     24# No arguments: automated test
     25if [[ "$#" -eq "0" ]]; then
     26	$0 4 tcp -t 1
     27	$0 6 tcp -t 1
     28	$0 4 udp -t 1
     29	$0 6 udp -t 1
     30	echo "OK. All tests passed"
     31	exit 0
     32fi
     33
     34# Argument parsing
     35if [[ "$#" -lt "2" ]]; then
     36	echo "Usage: $0 [4|6] [tcp|udp|raw|raw_hdrincl|packet|packet_dgram] <args>"
     37	exit 1
     38fi
     39
     40readonly IP="$1"
     41shift
     42readonly TXMODE="$1"
     43shift
     44readonly EXTRA_ARGS="$@"
     45
     46# Argument parsing: configure addresses
     47if [[ "${IP}" == "4" ]]; then
     48	readonly SADDR="${SADDR4}"
     49	readonly DADDR="${DADDR4}"
     50elif [[ "${IP}" == "6" ]]; then
     51	readonly SADDR="${SADDR6}"
     52	readonly DADDR="${DADDR6}"
     53else
     54	echo "Invalid IP version ${IP}"
     55	exit 1
     56fi
     57
     58# Argument parsing: select receive mode
     59#
     60# This differs from send mode for
     61# - packet:	use raw recv, because packet receives skb clones
     62# - raw_hdrinc: use raw recv, because hdrincl is a tx-only option
     63case "${TXMODE}" in
     64'packet' | 'packet_dgram' | 'raw_hdrincl')
     65	RXMODE='raw'
     66	;;
     67*)
     68	RXMODE="${TXMODE}"
     69	;;
     70esac
     71
     72# Start of state changes: install cleanup handler
     73save_sysctl_mem="$(sysctl -n ${path_sysctl_mem})"
     74
     75cleanup() {
     76	ip netns del "${NS2}"
     77	ip netns del "${NS1}"
     78	sysctl -w -q "${path_sysctl_mem}=${save_sysctl_mem}"
     79}
     80
     81trap cleanup EXIT
     82
     83# Configure system settings
     84sysctl -w -q "${path_sysctl_mem}=1000000"
     85
     86# Create virtual ethernet pair between network namespaces
     87ip netns add "${NS1}"
     88ip netns add "${NS2}"
     89
     90ip link add "${DEV}" mtu "${DEV_MTU}" netns "${NS1}" type veth \
     91  peer name "${DEV}" mtu "${DEV_MTU}" netns "${NS2}"
     92
     93# Bring the devices up
     94ip -netns "${NS1}" link set "${DEV}" up
     95ip -netns "${NS2}" link set "${DEV}" up
     96
     97# Set fixed MAC addresses on the devices
     98ip -netns "${NS1}" link set dev "${DEV}" address 02:02:02:02:02:02
     99ip -netns "${NS2}" link set dev "${DEV}" address 06:06:06:06:06:06
    100
    101# Add fixed IP addresses to the devices
    102ip -netns "${NS1}" addr add 192.168.1.1/24 dev "${DEV}"
    103ip -netns "${NS2}" addr add 192.168.1.2/24 dev "${DEV}"
    104ip -netns "${NS1}" addr add       fd::1/64 dev "${DEV}" nodad
    105ip -netns "${NS2}" addr add       fd::2/64 dev "${DEV}" nodad
    106
    107# Optionally disable sg or csum offload to test edge cases
    108# ip netns exec "${NS1}" ethtool -K "${DEV}" sg off
    109
    110do_test() {
    111	local readonly ARGS="$1"
    112
    113	echo "ipv${IP} ${TXMODE} ${ARGS}"
    114	ip netns exec "${NS2}" "${BIN}" "-${IP}" -i "${DEV}" -t 2 -C 2 -S "${SADDR}" -D "${DADDR}" ${ARGS} -r "${RXMODE}" &
    115	sleep 0.2
    116	ip netns exec "${NS1}" "${BIN}" "-${IP}" -i "${DEV}" -t 1 -C 3 -S "${SADDR}" -D "${DADDR}" ${ARGS} "${TXMODE}"
    117	wait
    118}
    119
    120do_test "${EXTRA_ARGS}"
    121do_test "-z ${EXTRA_ARGS}"
    122echo ok