summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xscripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py9
-rw-r--r--scripts/automation/trex_control_plane/stl/services/scapy_server/unit_tests/test_scapy_service.py52
2 files changed, 55 insertions, 6 deletions
diff --git a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py
index c0a753c..35bcf1f 100755
--- a/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py
+++ b/scripts/automation/trex_control_plane/stl/services/scapy_server/scapy_service.py
@@ -644,8 +644,11 @@ class Scapy_service(Scapy_service_api):
for field_desc in pkt.fields_desc:
field_id = field_desc.name
ignored = field_id not in layer_full.fields
- offset = field_desc._offset
- protocol_offset = pkt._offset
+ # scapy offset/length calculation doesn't support dynamic size structures
+ # since PktClass.fields_desc is a singletone,
+ # _offset/size can be missing, uninitialized or contain values from the previous runs
+ offset = getattr(field_desc, '_offset', None)
+ protocol_offset = getattr(pkt, '_offset', None)
field_sz = field_desc.get_size_bytes()
# some values are unavailable in pkt(original model)
# at the same time,
@@ -708,7 +711,7 @@ class Scapy_service(Scapy_service_api):
fields.append(field_data)
layer_data = {
"id": layer_id,
- "offset": pkt._offset,
+ "offset": protocol_offset,
"fields": fields,
"real_id": real_layer_id,
"valid_structure": valid_struct,
diff --git a/scripts/automation/trex_control_plane/stl/services/scapy_server/unit_tests/test_scapy_service.py b/scripts/automation/trex_control_plane/stl/services/scapy_server/unit_tests/test_scapy_service.py
index 430e569..656290e 100644
--- a/scripts/automation/trex_control_plane/stl/services/scapy_server/unit_tests/test_scapy_service.py
+++ b/scripts/automation/trex_control_plane/stl/services/scapy_server/unit_tests/test_scapy_service.py
@@ -19,13 +19,38 @@ TEST_PKT_DEF = [
layer_def("TCP", sport="443")
]
+TEST_DNS_PKT = Ether(src='00:16:ce:6e:8b:24', dst='00:05:5d:21:99:4c', type=2048)/IP(frag=0L, src='192.168.0.114', proto=17, tos=0, dst='205.152.37.23', chksum=26561, len=59, options=[], version=4L, flags=0L, ihl=5L, ttl=128, id=7975)/UDP(dport=53, sport=1060, len=39, chksum=877)/DNS(aa=0L, qr=0L, an=None, ad=0L, nscount=0, qdcount=1, ns=None, tc=0L, rd=1L, arcount=0, ar=None, opcode=0L, ra=0L, cd=0L, z=0L, rcode=0L, id=6159, ancount=0, qd=DNSQR(qclass=1, qtype=1, qname='wireshark.org.'))
+
+TEST_DNS_PKT_B64 = (
+ "1MOyoQIABAAAAAAAAAAAAP//AAABAAAAdzmERaAVAwBJAAAASQAAAAAFXSGZTAAWzm6LJAgARQAA"
+ "Ox8nAACAEWfBwKgAcs2YJRcEJAA1ACcDbRgPAQAAAQAAAAAAAAl3aXJlc2hhcmsDb3JnAAABAAF3"
+ "OYRFvHkEAFkAAABZAAAAABbOboskAAVdIZlMCABFAABLdn1AADIRHlvNmCUXwKgAcgA1BCQANzwg"
+ "GA+BgAABAAEAAAAACXdpcmVzaGFyawNvcmcAAAEAAcAMAAEAAQAAOEAABIB5Mno="
+)
+
+def test_0_dns_pcap_read_base64():
+ # should be executed first
+ # this test could fail depending on the test execution order
+ # due to uninitialized/preserved _offset field, which is stored
+ # in Packet.fields_desc singletone
+ array_pkt = service.read_pcap(v_handler, TEST_DNS_PKT_B64)
+ pkt = build_pkt_to_scapy(array_pkt[0])
+ assert(pkt[DNS].id == 6159)
+ service._pkt_to_field_tree(pkt)
+
+def test_dns_pcap_read_and_write_multi():
+ pkt_b64 = bytes_to_b64(bytes(TEST_DNS_PKT))
+ pkts_to_write = [pkt_b64, pkt_b64]
+ pcap_b64 = service.write_pcap(v_handler, pkts_to_write)
+ array_pkt = service.read_pcap(v_handler, pcap_b64)
+ pkt = build_pkt_to_scapy(array_pkt[0])
+ assert(pkt[DNS].id == 6159)
+
def test_build_pkt_details():
pkt_data = build_pkt(TEST_PKT_DEF)
pkt = build_pkt_to_scapy(pkt_data)
assert(pkt[TCP].sport == 443)
- ether = pkt_data['data'][0]
- ip = pkt_data['data'][1]
- tcp = pkt_data['data'][2]
+ [ether, ip, tcp] = pkt_data['data']
assert(len(pkt_data["binary"]) == 72) #b64 encoded data
# absolute frame offset
@@ -49,6 +74,27 @@ def test_build_pkt_details():
assert(tcp_chksum["offset"] == 16)
assert(tcp_chksum["length"] == 2)
+def test_reconstruct_dns_packet():
+ modif = [
+ {"id": "Ether"},
+ {"id": "IP"},
+ {"id": "UDP"},
+ {"id": "DNS", "fields": [{"id": "id", "value": 777}]},
+ ]
+ pkt_data = reconstruct_pkt(base64.b64encode(bytes(TEST_DNS_PKT)), modif)
+ pkt = build_pkt_to_scapy(pkt_data)
+ assert(pkt[DNS].id == 777)
+
+ [ether, ip, udp, dns] = pkt_data['data'][:4]
+
+ assert(ether['offset'] == 0)
+ assert(ip['offset'] == 14)
+ assert(dns['offset'] == 42)
+
+ dns_id = dns['fields'][0]
+ assert(dns_id["value"] == 777)
+ assert("offset" in dns_id)
+
def test_build_invalid_structure_pkt():
ether_fields = {"dst": TEST_MAC_1, "type": "LOOP"}
pkt = build_pkt_get_scapy([