matthew/ostinato
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Feature (contd.): Device Emulatiom - converted emulation tests to use the py.test framework

Browse Source
This commit is contained in:
Srivats P 2015-11-29 12:52:08 +05:30
parent 3a5396c865
commit cad62c1fd7
3 changed files with 533 additions and 488 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
common/emulproto.proto
View File

@ -75,6 +75,7 @@ message Ip6Emulation {
}
extend OstProto.DeviceGroup {
// FIXME: add directly to DeviceGroup in protocol.proto
optional EncapEmulation encap = 2000;
optional MacEmulation mac = 2001;

3
common/protocol.proto
View File

@ -300,7 +300,8 @@ message DeviceGroupIdList {
message DeviceGroup {
required DeviceGroupId device_group_id = 1;
optional DeviceGroupCore core = 2;
optional uint32 device_count = 3 [default = 1];
// FIXME: add Encap here
optional uint32 device_count = 3 [default = 1]; // per-encap
extensions 2000 to 5999; // For use by Protocol Emulations
}

559
test/emultest.py
View File

@ -8,8 +8,10 @@ import subprocess
import sys
import time
import pytest
from fabric.api import run, env, sudo
from harness import Test, TestSuite, TestPreRequisiteError
#from harness import Test, TestSuite, TestPreRequisiteError
sys.path.insert(1, '../binding')
from core import ost_pb, emul, DroneProxy
@ -18,25 +20,21 @@ from protocols.mac_pb2 import mac, Mac
from protocols.ip4_pb2 import ip4, Ip4
from protocols.vlan_pb2 import vlan
use_defaults = False
# initialize defaults - drone
host_name = '127.0.0.1'
tx_port_number = -1
rx_port_number = -1
use_defaults = True
if sys.platform == 'win32':
tshark = r'C:\Program Files\Wireshark\tshark.exe'
else:
tshark = 'tshark'
# initialize defaults - drone
host_name = '127.0.0.1'
# initialize defaults - DUT
env.use_shell = False
env.user = 'tc'
env.password = 'tc'
env.host_string = 'localhost:50022'
dut_rx_port = 'eth1'
dut_tx_port = 'eth2'
dut_dst_mac = 0x0800278df2b4 #FIXME: hardcoding
@ -55,8 +53,7 @@ if len(sys.argv) > 1:
print(' -h --help show this help')
sys.exit(0)
print('')
print('This test uses the following topology -')
print('This module uses the following topology -')
print('')
print(' +-------+ +-------+')
print(' | |Tx--->---Rx|-+ |')
@ -69,16 +66,18 @@ print('are expected to be forwarded by the DUT and received back on the')
print('Rx port')
print('')
suite = TestSuite()
if not use_defaults:
s = raw_input('Drone\'s Hostname/IP [%s]: ' % (host_name))
host_name = s or host_name
s = raw_input('DUT\'s Hostname/IP [%s]: ' % (env.host_string))
env.host_string = s or env.host_string
# FIXME: get inputs for dut rx/tx ports
drone = DroneProxy(host_name)
try:
# ================================================================= #
# ----------------------------------------------------------------- #
# FIXTURES
# ----------------------------------------------------------------- #
# ================================================================= #
# ----------------------------------------------------------------- #
# Baseline Configuration for subsequent testcases
# NOTES
@ -87,119 +86,181 @@ try:
# the devices to check
# ----------------------------------------------------------------- #
@pytest.fixture(scope='module')
def drone(request):
drn = DroneProxy(host_name)
# FIXME: get inputs for dut rx/tx ports
log.info('connecting to drone(%s:%d)' % (drn.hostName(), drn.portNumber()))
drn.connect()
# Enable IP forwarding on the DUT (aka make it a router)
sudo('sysctl -w net.ipv4.ip_forward=1')
def fin():
drn.disconnect()
request.addfinalizer(fin)
# connect to drone
log.info('connecting to drone(%s:%d)'
% (drone.hostName(), drone.portNumber()))
drone.connect()
return drn
# retreive port id list
log.info('retreiving port list')
@pytest.fixture(scope='module')
def ports(request, drone):
port_id_list = drone.getPortIdList()
# retreive port config list
log.info('retreiving port config for all ports')
port_config_list = drone.getPortConfig(port_id_list)
if len(port_config_list.port) == 0:
log.warning('drone has no ports!')
sys.exit(1)
assert len(port_config_list.port) != 0
# print port list and find default tx/rx ports
tx_number = -1
rx_number = -1
print port_config_list
print('Port List')
print('---------')
for port in port_config_list.port:
print('%d.%s (%s)' % (port.port_id.id, port.name, port.description))
# use a vhost port as default tx/rx port
if ('vhost' in port.name or 'sun' in port.description.lower()):
if tx_port_number < 0:
tx_port_number = port.port_id.id
elif rx_port_number < 0:
rx_port_number = port.port_id.id
if tx_number < 0:
tx_number = port.port_id.id
elif rx_number < 0:
rx_number = port.port_id.id
if ('eth1' in port.name):
tx_port_number = port.port_id.id
tx_number = port.port_id.id
if ('eth2' in port.name):
rx_port_number = port.port_id.id
rx_number = port.port_id.id
if not use_defaults:
p = raw_input('Tx Port Id [%d]: ' % (tx_port_number))
if p:
tx_port_number = int(p)
assert tx_number >= 0
assert rx_number >= 0
p = raw_input('Rx Port Id [%d]: ' % (rx_port_number))
if p:
rx_port_number = int(p)
print('Using port %d as tx port(s)' % tx_number)
print('Using port %d as rx port(s)' % rx_number)
if tx_port_number < 0 or rx_port_number < 0:
log.warning('invalid tx/rx port')
sys.exit(1)
ports.tx = ost_pb.PortIdList()
ports.tx.port_id.add().id = tx_number;
print('Using port %d as tx port(s)' % tx_port_number)
print('Using port %d as rx port(s)' % rx_port_number)
ports.rx = ost_pb.PortIdList()
ports.rx.port_id.add().id = rx_number;
return ports
tx_port = ost_pb.PortIdList()
tx_port.port_id.add().id = tx_port_number;
@pytest.fixture(scope='module')
def dut(request):
# Enable IP forwarding on the DUT (aka make it a router)
sudo('sysctl -w net.ipv4.ip_forward=1')
rx_port = ost_pb.PortIdList()
rx_port.port_id.add().id = rx_port_number;
@pytest.fixture(scope='module')
def dut_ports(request):
dut_ports.rx = 'eth1'
dut_ports.tx = 'eth2'
# delete all configuration on the DUT interfaces
sudo('ip address flush dev ' + dut_ports.rx)
sudo('ip address flush dev ' + dut_ports.tx)
return dut_ports
@pytest.fixture(scope='module')
def emul_ports(request, drone, ports):
emul_ports = ost_pb.PortIdList()
emul_ports.port_id.add().id = ports.tx.port_id[0].id;
emul_ports.port_id.add().id = ports.rx.port_id[0].id;
return emul_ports
@pytest.fixture(scope='module')
def dgid_list(request, drone, ports):
# ----------------------------------------------------------------- #
# create emulated device(s) on tx/rx ports - each test case will
# modify and reuse these devices as per its needs
# ----------------------------------------------------------------- #
emul_ports = ost_pb.PortIdList()
emul_ports.port_id.add().id = tx_port_number;
emul_ports.port_id.add().id = rx_port_number;
# delete existing devices, if any, on tx port
tx_dgid_list = drone.getDeviceGroupIdList(tx_port.port_id[0])
drone.deleteDeviceGroup(tx_dgid_list)
dgid_list.tx = drone.getDeviceGroupIdList(ports.tx.port_id[0])
drone.deleteDeviceGroup(dgid_list.tx)
# add a emulated device on tx port
tx_dgid_list = ost_pb.DeviceGroupIdList()
tx_dgid_list.port_id.CopyFrom(tx_port.port_id[0])
tx_dgid_list.device_group_id.add().id = 1
log.info('adding tx device_group %d' % tx_dgid_list.device_group_id[0].id)
drone.addDeviceGroup(tx_dgid_list)
# add a emulated device group on tx port
dgid_list.tx = ost_pb.DeviceGroupIdList()
dgid_list.tx.port_id.CopyFrom(ports.tx.port_id[0])
dgid_list.tx.device_group_id.add().id = 1
log.info('adding tx device_group %d' % dgid_list.tx.device_group_id[0].id)
drone.addDeviceGroup(dgid_list.tx)
# delete existing devices, if any, on rx port
rx_dgid_list = drone.getDeviceGroupIdList(rx_port.port_id[0])
drone.deleteDeviceGroup(rx_dgid_list)
dgid_list.rx = drone.getDeviceGroupIdList(ports.rx.port_id[0])
drone.deleteDeviceGroup(dgid_list.rx)
# add a emulated device on rx port
rx_dgid_list = ost_pb.DeviceGroupIdList()
rx_dgid_list.port_id.CopyFrom(rx_port.port_id[0])
rx_dgid_list.device_group_id.add().id = 1
log.info('adding rx device_group %d' % rx_dgid_list.device_group_id[0].id)
drone.addDeviceGroup(rx_dgid_list)
# add a emulated device group on rx port
dgid_list.rx = ost_pb.DeviceGroupIdList()
dgid_list.rx.port_id.CopyFrom(ports.rx.port_id[0])
dgid_list.rx.device_group_id.add().id = 1
log.info('adding rx device_group %d' % dgid_list.rx.device_group_id[0].id)
drone.addDeviceGroup(dgid_list.rx)
def fin():
dgid_list = drone.getDeviceGroupIdList(ports.tx.port_id[0])
drone.deleteDeviceGroup(dgid_list)
dgid_list = drone.getDeviceGroupIdList(ports.rx.port_id[0])
drone.deleteDeviceGroup(dgid_list)
request.addfinalizer(fin)
return dgid_list
@pytest.fixture(scope='module')
def stream_id(request, drone, ports):
# ----------------------------------------------------------------- #
# create stream on tx port - each test case will modify and reuse
# this stream as per its needs
# ----------------------------------------------------------------- #
# delete existing streams, if any, on tx port
sid_list = drone.getStreamIdList(tx_port.port_id[0])
sid_list = drone.getStreamIdList(ports.tx.port_id[0])
drone.deleteStream(sid_list)
# add a stream
stream_id = ost_pb.StreamIdList()
stream_id.port_id.CopyFrom(tx_port.port_id[0])
stream_id.port_id.CopyFrom(ports.tx.port_id[0])
stream_id.stream_id.add().id = 1
log.info('adding tx_stream %d' % stream_id.stream_id[0].id)
drone.addStream(stream_id)
# ----------------------------------------------------------------- #
# delete all configuration on the DUT interfaces
# ----------------------------------------------------------------- #
sudo('ip address flush dev ' + dut_rx_port)
sudo('ip address flush dev ' + dut_tx_port)
def fin():
drone.deleteStream(stream_id)
request.addfinalizer(fin)
return stream_id
@pytest.fixture
def dut_vlans(request, dut_ports):
num_vlans = getattr(request.function, 'num_vlans')
vlan_base = getattr(request.function, 'vlan_base')
for i in range(num_vlans):
vlan_id = vlan_base+i
vrf = 'v' + str(vlan_id)
vlan_rx_dev = dut_ports.rx + '.' + str(vlan_id)
vlan_tx_dev = dut_ports.tx + '.' + str(vlan_id)
sudo('ip netns add ' + vrf)
sudo('ip link add link ' + dut_ports.rx
+ ' name ' + vlan_rx_dev
+ ' type vlan id ' + str(vlan_id))
sudo('ip link set ' + vlan_rx_dev
+ ' netns ' + vrf)
sudo('ip netns exec ' + vrf
+ ' ip addr add 10.1.1.1/24'
+ ' dev ' + vlan_rx_dev)
sudo('ip netns exec ' + vrf
+ ' ip link set ' + vlan_rx_dev + ' up')
sudo('ip link add link ' + dut_ports.tx
+ ' name ' + vlan_tx_dev
+ ' type vlan id ' + str(vlan_id))
sudo('ip link set ' + vlan_tx_dev
+ ' netns ' + vrf)
sudo('ip netns exec ' + vrf
+ ' ip addr add 10.1.2.1/24'
+ ' dev ' + vlan_tx_dev)
sudo('ip netns exec ' + vrf
+ ' ip link set ' + vlan_tx_dev + ' up')
def fin():
for i in range(num_vlans):
vlan_id = vlan_base + i
vrf = 'v' + str(vlan_id)
sudo('ip netns delete ' + vrf)
request.addfinalizer(fin)
# ================================================================= #
# ----------------------------------------------------------------- #
@ -207,6 +268,8 @@ try:
# ----------------------------------------------------------------- #
# ================================================================= #
def test_multiEmulDevNoVlan(drone, ports, dut, dut_ports, stream_id,
emul_ports, dgid_list):
# ----------------------------------------------------------------- #
# TESTCASE: Emulate multiple IPv4 devices (no vlans)
# DUT
@ -218,20 +281,17 @@ try:
# Host1(s) Host2(s)
# ----------------------------------------------------------------- #
passed = False
suite.test_begin('multiEmulDevNoVlan')
num_devs = 5
try:
# configure the DUT
sudo('ip address add 10.10.1.1/24 dev ' + dut_rx_port)
sudo('ip address add 10.10.2.1/24 dev ' + dut_tx_port)
sudo('ip address add 10.10.1.1/24 dev ' + dut_ports.rx)
sudo('ip address add 10.10.2.1/24 dev ' + dut_ports.tx)
# configure the tx device(s)
devgrp_cfg = ost_pb.DeviceGroupConfigList()
devgrp_cfg.port_id.CopyFrom(tx_port.port_id[0])
devgrp_cfg.port_id.CopyFrom(ports.tx.port_id[0])
dg = devgrp_cfg.device_group.add()
dg.device_group_id.id = tx_dgid_list.device_group_id[0].id
dg.device_group_id.id = dgid_list.tx.device_group_id[0].id
dg.core.name = "Host1"
dg.device_count = num_devs
dg.Extensions[emul.mac].address = 0x000102030a01
@ -244,9 +304,9 @@ try:
# configure the rx device(s)
devgrp_cfg = ost_pb.DeviceGroupConfigList()
devgrp_cfg.port_id.CopyFrom(rx_port.port_id[0])
devgrp_cfg.port_id.CopyFrom(ports.rx.port_id[0])
dg = devgrp_cfg.device_group.add()
dg.device_group_id.id = rx_dgid_list.device_group_id[0].id
dg.device_group_id.id = dgid_list.rx.device_group_id[0].id
dg.core.name = "Host1"
dg.device_count = num_devs
dg.Extensions[emul.mac].address = 0x000102030b01
@ -259,11 +319,11 @@ try:
# configure the tx stream
stream_cfg = ost_pb.StreamConfigList()
stream_cfg.port_id.CopyFrom(tx_port.port_id[0])
stream_cfg.port_id.CopyFrom(ports.tx.port_id[0])
s = stream_cfg.stream.add()
s.stream_id.id = stream_id.stream_id[0].id
s.core.is_enabled = True
s.control.packets_per_sec = 20
s.control.packets_per_sec = 100
s.control.num_packets = 10
# setup stream protocols as mac:eth2:ip4:udp:payload
@ -291,27 +351,24 @@ try:
log.info('configuring tx_stream %d' % stream_id.stream_id[0].id)
drone.modifyStream(stream_cfg)
# clear tx/rx stats
# FIXME(needed?): clear tx/rx stats
log.info('clearing tx/rx stats')
drone.clearStats(tx_port)
drone.clearStats(rx_port)
drone.clearStats(ports.tx)
drone.clearStats(ports.rx)
# clear arp on DUT
sudo('ip neigh flush all')
arp_cache = run('ip neigh show')
if re.search('10.10.2.10[1-5].*lladdr', arp_cache):
raise TestPreRequisiteError('ARP cache not cleared')
assert re.search('10.10.2.10[1-5].*lladdr', arp_cache) == None
# resolve ARP on tx/rx ports
log.info('resolving Neighbors on tx/rx ports ...')
drone.startCapture(emul_ports)
drone.clearDeviceNeighbors(emul_ports)
drone.resolveDeviceNeighbors(emul_ports)
time.sleep(10)
time.sleep(3)
drone.stopCapture(emul_ports)
fail = 0
# verify ARP Requests sent out from tx port
buff = drone.getCaptureBuffer(emul_ports.port_id[0])
drone.saveCaptureBuffer(buff, 'capture.pcap')
@ -325,8 +382,7 @@ try:
' && (arp.src.proto_ipv4 == 10.10.1.'+str(101+i)+')'
' && (arp.dst.proto_ipv4 == 10.10.1.1)'])
print(cap_pkts)
if cap_pkts.count('\n') != 1:
fail = fail + 1
assert cap_pkts.count('\n') == 1
# verify *no* ARP Requests sent out from rx port
buff = drone.getCaptureBuffer(emul_ports.port_id[1])
@ -341,8 +397,7 @@ try:
' && (arp.src.proto_ipv4 == 10.10.2.'+str(101+i)+')'
' && (arp.dst.proto_ipv4 == 10.10.2.1)'])
print(cap_pkts)
if cap_pkts.count('\n') != 0:
fail = fail + 1
assert cap_pkts.count('\n') == 0
# retrieve and verify ARP Table on tx/rx ports
log.info('retrieving ARP entries on tx port')
@ -359,8 +414,7 @@ try:
(dev_cfg.ip4, arp.ip4, arp.mac))
if (arp.ip4 == dev_cfg.ip4_default_gateway) and (arp.mac):
resolved = True
if not resolved:
fail = fail + 1
assert resolved
log.info('retrieving ARP entries on rx port')
device_list = drone.getDeviceList(emul_ports.port_id[0])
@ -370,21 +424,21 @@ try:
log.info('ARP Table on rx port')
for dev_cfg, device in zip(device_config, devices):
# verify *no* ARPs learnt on rx port
if len(device.arp):
fail = fail + 1
assert len(device.arp) == 0
for arp in device.arp:
# TODO: pretty print ip and mac
print('%08x: %08x %012x' %
(dev_cfg.ip4, arp.ip4, arp.mac))
drone.startCapture(rx_port)
drone.startTransmit(tx_port)
# We are all set now - so transmit the stream now
drone.startCapture(ports.rx)
drone.startTransmit(ports.tx)
log.info('waiting for transmit to finish ...')
time.sleep(7)
drone.stopTransmit(tx_port)
drone.stopCapture(rx_port)
time.sleep(3)
drone.stopTransmit(ports.tx)
drone.stopCapture(ports.rx)
buff = drone.getCaptureBuffer(rx_port.port_id[0])
buff = drone.getCaptureBuffer(ports.rx.port_id[0])
drone.saveCaptureBuffer(buff, 'capture.pcap')
log.info('dumping Rx capture buffer (all)')
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap'])
@ -397,27 +451,18 @@ try:
' && (eth.dst == 00:01:02:03:0b:'
+ format(1+i, '02x')+')'])
print(cap_pkts)
if cap_pkts.count('\n') != s.control.num_packets/ip.src_ip_count:
fail = fail + 1
if fail == 0:
passed = True
else:
log.info('failed checks: %d' % fail)
assert cap_pkts.count('\n') == s.control.num_packets/ip.src_ip_count
os.remove('capture.pcap')
except RpcError as e:
raise
finally:
drone.stopTransmit(tx_port)
drone.stopTransmit(ports.tx)
run('ip neigh show')
# unconfigure DUT
sudo('ip address delete 10.10.1.1/24 dev ' + dut_rx_port)
sudo('ip address delete 10.10.2.1/24 dev ' + dut_tx_port)
suite.test_end(passed)
sys.exit(1)
# FIXME: update the below test cases to resolve Neighbors and streams
# to derive src/dst mac from device
sudo('ip address delete 10.10.1.1/24 dev ' + dut_ports.rx)
sudo('ip address delete 10.10.2.1/24 dev ' + dut_ports.tx)
def test_multiEmulDevPerVlan(request, drone, ports, dut, dut_ports, stream_id,
emul_ports, dgid_list):
# ----------------------------------------------------------------- #
# TESTCASE: Emulate multiple IPv4 device per multiple single-tag VLANs
#
@ -431,80 +476,48 @@ try:
# Host1(s) Host2(s)
# ----------------------------------------------------------------- #
passed = False
suite.test_begin('multiEmulDevPerVlan')
num_vlans = 5
vlan_base = 201
num_devs = 3
dev_ip_base = 101
try:
# configure the DUT
for i in range(num_vlans):
vlan_id = vlan_base+i
vrf = 'v' + str(vlan_id)
vlan_rx_dev = dut_rx_port + '.' + str(vlan_id)
vlan_tx_dev = dut_tx_port + '.' + str(vlan_id)
sudo('ip netns add ' + vrf)
sudo('ip link add link ' + dut_rx_port
+ ' name ' + vlan_rx_dev
+ ' type vlan id ' + str(vlan_id))
sudo('ip link set ' + vlan_rx_dev
+ ' netns ' + vrf)
sudo('ip netns exec ' + vrf
+ ' ip addr add 10.1.1.1/24'
+ ' dev ' + vlan_rx_dev)
sudo('ip netns exec ' + vrf
+ ' ip link set ' + vlan_rx_dev + ' up')
sudo('ip link add link ' + dut_tx_port
+ ' name ' + vlan_tx_dev
+ ' type vlan id ' + str(vlan_id))
sudo('ip link set ' + vlan_tx_dev
+ ' netns ' + vrf)
sudo('ip netns exec ' + vrf
+ ' ip addr add 10.1.2.1/24'
+ ' dev ' + vlan_tx_dev)
sudo('ip netns exec ' + vrf
+ ' ip link set ' + vlan_tx_dev + ' up')
test_multiEmulDevPerVlan.num_vlans = num_vlans = 5
test_multiEmulDevPerVlan.vlan_base = vlan_base = 201
num_devs_per_vlan = 3
tx_ip_base = 0x0a010165
rx_ip_base = 0x0a010265
# configure vlans on the DUT
dut_vlans(request, dut_ports)
# configure the tx device(s)
devgrp_cfg = ost_pb.DeviceGroupConfigList()
devgrp_cfg.port_id.CopyFrom(tx_port.port_id[0])
devgrp_cfg.port_id.CopyFrom(ports.tx.port_id[0])
dg = devgrp_cfg.device_group.add()
dg.device_group_id.id = tx_dgid_list.device_group_id[0].id
dg.device_group_id.id = dgid_list.tx.device_group_id[0].id
dg.core.name = "Host1"
v = dg.Extensions[emul.vlan].stack.add()
v = dg.Extensions[emul.encap].vlan.stack.add()
v.vlan_tag = vlan_base
v.count = num_vlans
d = dg.Extensions[emul.device]
d.count = num_devs
d.mac.address = 0x000102030a01
d.ip4.address = 0x0a010165
d.ip4.prefix_length = 24
d.ip4.default_gateway = 0x0a0a0101
dg.device_count = num_devs_per_vlan
dg.Extensions[emul.mac].address = 0x000102030a01
ip = dg.Extensions[emul.ip4]
ip.address = tx_ip_base
ip.prefix_length = 24
ip.default_gateway = (tx_ip_base & 0xffffff00) | 0x01
drone.modifyDeviceGroup(devgrp_cfg)
# configure the rx device(s)
devgrp_cfg = ost_pb.DeviceGroupConfigList()
devgrp_cfg.port_id.CopyFrom(rx_port.port_id[0])
devgrp_cfg.port_id.CopyFrom(ports.rx.port_id[0])
dg = devgrp_cfg.device_group.add()
dg.device_group_id.id = rx_dgid_list.device_group_id[0].id
dg.device_group_id.id = dgid_list.rx.device_group_id[0].id
dg.core.name = "Host1"
v = dg.Extensions[emul.vlan].stack.add()
v = dg.Extensions[emul.encap].vlan.stack.add()
v.vlan_tag = vlan_base
v.count = num_vlans
d = dg.Extensions[emul.device]
d.count = num_devs
#d.mode = emul.Device.kNoRepeat
d.mac.address = 0x000102030b01
d.ip4.address = 0x0a010265
d.ip4.prefix_length = 24
d.ip4.default_gateway = 0x0a0a0201
dg.device_count = num_devs_per_vlan
dg.Extensions[emul.mac].address = 0x000102030b01
ip = dg.Extensions[emul.ip4]
ip.address = rx_ip_base
ip.prefix_length = 24
ip.default_gateway = (rx_ip_base & 0xffffff00) | 0x01
drone.modifyDeviceGroup(devgrp_cfg)
@ -516,7 +529,7 @@ try:
drone.deleteStream(stream_id)
stream_id = ost_pb.StreamIdList()
stream_id.port_id.CopyFrom(tx_port.port_id[0])
stream_id.port_id.CopyFrom(ports.tx.port_id[0])
for i in range(num_vlans):
stream_id.stream_id.add().id = i
log.info('adding tx_stream %d' % stream_id.stream_id[i].id)
@ -524,20 +537,20 @@ try:
drone.addStream(stream_id)
stream_cfg = ost_pb.StreamConfigList()
stream_cfg.port_id.CopyFrom(tx_port.port_id[0])
stream_cfg.port_id.CopyFrom(ports.tx.port_id[0])
for i in range(num_vlans):
s = stream_cfg.stream.add()
s.stream_id.id = stream_id.stream_id[i].id
s.core.is_enabled = True
s.core.ordinal = i
s.control.packets_per_sec = 10
s.control.num_packets = num_devs
s.control.packets_per_sec = 100
s.control.num_packets = num_devs_per_vlan
# setup stream protocols as mac:vlan:eth2:ip4:udp:payload
p = s.protocol.add()
p.protocol_id.id = ost_pb.Protocol.kMacFieldNumber
p.Extensions[mac].dst_mac = dut_dst_mac
p.Extensions[mac].src_mac = 0x00aabbccddee
p.Extensions[mac].dst_mac_mode = Mac.e_mm_resolve
p.Extensions[mac].src_mac_mode = Mac.e_mm_resolve
p = s.protocol.add()
p.protocol_id.id = ost_pb.Protocol.kVlanFieldNumber
@ -549,10 +562,12 @@ try:
p = s.protocol.add()
p.protocol_id.id = ost_pb.Protocol.kIp4FieldNumber
ip = p.Extensions[ip4]
ip.src_ip = 0x0a010165
ip.src_ip = tx_ip_base
ip.src_ip_mode = Ip4.e_im_inc_host
ip.dst_ip = 0x0a010265
ip.src_ip_count = num_devs_per_vlan
ip.dst_ip = rx_ip_base
ip.dst_ip_mode = Ip4.e_im_inc_host
ip.dst_ip_count = num_devs_per_vlan
p = s.protocol.add()
p.protocol_id.id = ost_pb.Protocol.kUdpFieldNumber
@ -563,69 +578,119 @@ try:
drone.modifyStream(stream_cfg)
# clear tx/rx stats
log.info('clearing tx/rx stats')
drone.clearStats(tx_port)
drone.clearStats(rx_port)
# clear arp on DUT
for i in range(num_vlans):
vlan_id = vlan_base + i
vrf = 'v' + str(vlan_id)
vlan_rx_dev = dut_rx_port + '.' + str(vlan_id)
vlan_tx_dev = dut_tx_port + '.' + str(vlan_id)
vlan_rx_dev = dut_ports.rx + '.' + str(vlan_id)
vlan_tx_dev = dut_ports.tx + '.' + str(vlan_id)
sudo('ip netns exec ' + vrf
+ ' ip neigh flush dev ' + vlan_rx_dev)
sudo('ip netns exec ' + vrf
+ ' ip neigh flush dev ' + vlan_tx_dev)
sudo('ip netns exec ' + vrf
+ ' ip neigh flush all')
arp_cache = sudo('ip netns exec ' + vrf
+ ' ip neigh show')
assert re.search('10.1.[1-2].20[1-5].*lladdr', arp_cache) == None
drone.startCapture(rx_port)
drone.startTransmit(tx_port)
log.info('waiting for transmit to finish ...')
time.sleep(5)
drone.stopTransmit(tx_port)
drone.stopCapture(rx_port)
# resolve ARP on tx/rx ports
log.info('resolving Neighbors on tx/rx ports ...')
drone.startCapture(emul_ports)
drone.clearDeviceNeighbors(emul_ports)
drone.resolveDeviceNeighbors(emul_ports)
time.sleep(3)
drone.stopCapture(emul_ports)
buff = drone.getCaptureBuffer(rx_port.port_id[0])
# verify ARP Requests sent out from tx port
buff = drone.getCaptureBuffer(emul_ports.port_id[0])
drone.saveCaptureBuffer(buff, 'capture.pcap')
log.info('dumping Tx capture buffer (all)')
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap'])
print(cap_pkts)
log.info('dumping Tx capture buffer (filtered)')
for i in range(num_vlans):
for j in range(num_devs_per_vlan):
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap',
'-R', '(vlan.id == '+str(vlan_base+i)+')'
' && (arp.opcode == 1)'
' && (arp.src.proto_ipv4 == 10.1.1.'
+str(101+j)+')'
' && (arp.dst.proto_ipv4 == 10.1.1.1)'])
print(cap_pkts)
assert cap_pkts.count('\n') == 1
# verify *no* ARP Requests sent out from rx port
buff = drone.getCaptureBuffer(emul_ports.port_id[1])
drone.saveCaptureBuffer(buff, 'capture.pcap')
log.info('dumping Rx capture buffer (all)')
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap'])
print(cap_pkts)
log.info('dumping Rx capture buffer (filtered)')
fail = 0
for i in range(num_vlans):
for j in range(num_devs):
for j in range(num_devs_per_vlan):
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap',
'-R', '(vlan.id == '+str(vlan_base+i)+')'
' && (arp.opcode == 1)'
' && (arp.src.proto_ipv4 == 10.10.2.'
+str(101+j)+')'
' && (arp.dst.proto_ipv4 == 10.10.2.1)'])
print(cap_pkts)
assert cap_pkts.count('\n') == 0
# retrieve and verify ARP Table on tx/rx ports
log.info('retrieving ARP entries on tx port')
device_list = drone.getDeviceList(emul_ports.port_id[0])
device_config = device_list.Extensions[emul.port_device]
neigh_list = drone.getDeviceNeighbors(emul_ports.port_id[0])
devices = neigh_list.Extensions[emul.devices]
log.info('ARP Table on tx port')
for dev_cfg, device in zip(device_config, devices):
resolved = False
for arp in device.arp:
# TODO: pretty print ip and mac
print('v%d|%08x: %08x %012x' %
(dev_cfg.vlan[0] & 0xffff, dev_cfg.ip4, arp.ip4, arp.mac))
if (arp.ip4 == dev_cfg.ip4_default_gateway) and (arp.mac):
resolved = True
assert resolved
log.info('retrieving ARP entries on rx port')
device_list = drone.getDeviceList(emul_ports.port_id[0])
device_config = device_list.Extensions[emul.port_device]
neigh_list = drone.getDeviceNeighbors(emul_ports.port_id[1])
devices = neigh_list.Extensions[emul.devices]
log.info('ARP Table on rx port')
for dev_cfg, device in zip(device_config, devices):
# verify *no* ARPs learnt on rx port
assert len(device.arp) == 0
for arp in device.arp:
# TODO: pretty print ip and mac
print('v%d|%08x: %08x %012x' %
(dev_cfg.vlan[0] & 0xffff, dev_cfg.ip4, arp.ip4, arp.mac))
drone.startCapture(ports.rx)
drone.startTransmit(ports.tx)
log.info('waiting for transmit to finish ...')
time.sleep(5)
drone.stopTransmit(ports.tx)
drone.stopCapture(ports.rx)
buff = drone.getCaptureBuffer(ports.rx.port_id[0])
drone.saveCaptureBuffer(buff, 'capture.pcap')
log.info('dumping Rx capture buffer (all)')
cap_pkts = subprocess.check_output([tshark, '-nr', 'capture.pcap'])
print(cap_pkts)
log.info('dumping Rx capture buffer (filtered)')
for i in range(num_vlans):
for j in range(num_devs_per_vlan):
cap_pkts = subprocess.check_output(
[tshark, '-nr', 'capture.pcap',
'-R', '(vlan.id == ' + str(201+i) + ')'
'-R', '(vlan.id == ' + str(vlan_base+i) + ')'
' && (ip.src == 10.1.1.' + str(101+j) + ') '
' && (ip.dst == 10.1.2.' + str(101+j) + ')'
' && (eth.dst == 00:01:02:03:0b:'
+ format(1+j, '02x')+')'])
print(cap_pkts)
if cap_pkts.count('\n') != 1:
fail = fail + 1
if fail == 0:
passed = True
assert cap_pkts.count('\n') == 1
os.remove('capture.pcap')
except RpcError as e:
raise
finally:
drone.stopTransmit(tx_port)
# show arp on DUT
for i in range(num_vlans):
vrf = 'v' + str(vlan_base + i)
sudo('ip netns exec ' + vrf
+ ' ip neigh show')
# un-configure the DUT
for i in range(num_vlans):
vlan_id = vlan_base + i
vrf = 'v' + str(vlan_id)
sudo('ip netns delete ' + vrf)
suite.test_end(passed)
# TODO:
# ----------------------------------------------------------------- #
@ -639,26 +704,4 @@ try:
# vlanMode: no-repeat; ip4Mode: no-repeat
# ----------------------------------------------------------------- #
suite.complete()
# delete stream(s)
log.info('deleting tx_stream %d' % stream_id.stream_id[0].id)
drone.deleteStream(stream_id)
# delete device(s)
dgid_list = drone.getDeviceGroupIdList(tx_port.port_id[0])
drone.deleteDeviceGroup(dgid_list)
dgid_list = drone.getDeviceGroupIdList(rx_port.port_id[0])
drone.deleteDeviceGroup(dgid_list)
# bye for now
drone.disconnect()
#disconnect_all()
except Exception as ex:
log.exception(ex)
finally:
suite.report()
if not suite.passed:
sys.exit(2);