dab231a11e
Perform accounting of expected packets based on created packet infos. Use this accounting info to automatically expect (and verify) the correct number of packets to be captured. Automatically retry the read of the capture file if scapy raises an exception while doing so to handle rare cases when capture file is read while only partially written during busy wait. Don't fail assert_nothing_captured if only junk packets arrived. Change-Id: I16ec2e9410ef510d313ec16b7e13c57d0b2a63f5 Signed-off-by: Klement Sekera <ksekera@cisco.com>
85 lines
1.9 KiB
Python
85 lines
1.9 KiB
Python
import socket
|
|
import sys
|
|
from abc import abstractmethod, ABCMeta
|
|
from cStringIO import StringIO
|
|
|
|
|
|
def ppp(headline, packet):
|
|
""" Return string containing the output of scapy packet.show() call. """
|
|
o = StringIO()
|
|
old_stdout = sys.stdout
|
|
sys.stdout = o
|
|
print(headline)
|
|
packet.show()
|
|
sys.stdout = old_stdout
|
|
return o.getvalue()
|
|
|
|
|
|
def ppc(headline, capture, limit=10):
|
|
""" Return string containing ppp() printout for a capture.
|
|
|
|
:param headline: printed as first line of output
|
|
:param capture: packets to print
|
|
:param limit: limit the print to # of packets
|
|
"""
|
|
if not capture:
|
|
return headline
|
|
tail = ""
|
|
if limit < len(capture):
|
|
tail = "\nPrint limit reached, %s out of %s packets printed" % (
|
|
len(capture), limit)
|
|
limit = len(capture)
|
|
body = "".join([ppp("Packet #%s:" % count, p)
|
|
for count, p in zip(range(0, limit), capture)])
|
|
return "%s\n%s%s" % (headline, body, tail)
|
|
|
|
|
|
class NumericConstant(object):
|
|
__metaclass__ = ABCMeta
|
|
|
|
desc_dict = {}
|
|
|
|
@abstractmethod
|
|
def __init__(self, value):
|
|
self._value = value
|
|
|
|
def __int__(self):
|
|
return self._value
|
|
|
|
def __long__(self):
|
|
return self._value
|
|
|
|
def __str__(self):
|
|
if self._value in self.desc_dict:
|
|
return self.desc_dict[self._value]
|
|
return ""
|
|
|
|
|
|
class Host(object):
|
|
""" Generic test host "connected" to VPPs interface. """
|
|
|
|
@property
|
|
def mac(self):
|
|
""" MAC address """
|
|
return self._mac
|
|
|
|
@property
|
|
def ip4(self):
|
|
""" IPv4 address """
|
|
return self._ip4
|
|
|
|
@property
|
|
def ip4n(self):
|
|
""" IPv4 address """
|
|
return socket.inet_pton(socket.AF_INET, self._ip4)
|
|
|
|
@property
|
|
def ip6(self):
|
|
""" IPv6 address """
|
|
return self._ip6
|
|
|
|
def __init__(self, mac=None, ip4=None, ip6=None):
|
|
self._mac = mac
|
|
self._ip4 = ip4
|
|
self._ip6 = ip6
|