catarrh
cb4c07f3c2
pcap.py: LINKTYPE_GSMTAP (155) was wrong — 155 is DLT_USER8, so Wireshark showed 'USER 8 / DLT=155'. Now write Ethernet-encapsulated PCAP (LINKTYPE 1): each GSMTAP message wrapped in Ethernet + IPv4 + UDP (port 4729), with correct checksums — matching a real tcpdump capture of GSMTAP-over-UDP and how Wireshark's gsmtap dissector is keyed. gsmtap.py: GSMTAP_TYPE_SIM was 0x07 (WiMAX burst) — corrected to 0x04; GSMTAP_SIM_APDU was 0x02 (PPS req) — corrected to 0x00; added PPS_REQ/ PPS_RSP. These match libosmocore/Wireshark packet-gsmtap.h. Verified with tshark: PCAP now decodes as GSM SIM (SELECT/STATUS/ATR) instead of 'USER 8'. tests: pcap tests assert Ethernet linktype + framing + GSMTAP payload.
80 lines
3.1 KiB
Python
80 lines
3.1 KiB
Python
"""Tests for PCAP generation and Content-Disposition filename."""
|
||
|
||
import struct
|
||
import unittest
|
||
|
||
from simtrace2_pysniff.pcap import build_pcap, PCAP_MAGIC, LINKTYPE_ETHERNET
|
||
from simtrace2_pysniff.server.server import _content_disposition
|
||
|
||
FRAMING = 14 + 20 + 8 # Ethernet + IPv4 + UDP
|
||
|
||
|
||
class TestBuildPcap(unittest.TestCase):
|
||
def _gsmtap_hdr(self, sub_type):
|
||
return struct.pack('!BBBBHBBIBBBB', 2, 4, 7, 0, 0, 0, 0, 0, sub_type, 0, 0, 0)
|
||
|
||
def test_empty(self):
|
||
data = build_pcap([])
|
||
self.assertEqual(len(data), 24) # just the global header
|
||
magic, vmaj, vmin, _tz, _sig, snaplen, network = struct.unpack('<IHHiIII', data[:24])
|
||
self.assertEqual(magic, PCAP_MAGIC)
|
||
self.assertEqual((vmaj, vmin), (2, 4))
|
||
self.assertEqual(snaplen, 65535)
|
||
self.assertEqual(network, LINKTYPE_ETHERNET)
|
||
|
||
def test_one_packet(self):
|
||
pkt = build_pcap([(self._gsmtap_hdr(2), b'\xa0\xa4\x00\x00\x02\x3f\x00', 1234567890.5)])
|
||
# global + pkt hdr + framing (Eth+IP+UDP) + gsmtap hdr + data
|
||
self.assertEqual(len(pkt), 24 + 16 + FRAMING + 16 + 7)
|
||
ts_sec, ts_usec, incl_len, orig_len = struct.unpack('<IIII', pkt[24:40])
|
||
self.assertEqual(ts_sec, 1234567890)
|
||
self.assertEqual(ts_usec, 500000)
|
||
self.assertEqual(incl_len, FRAMING + 16 + 7)
|
||
self.assertEqual(orig_len, FRAMING + 16 + 7)
|
||
|
||
def test_ethernet_and_udp_headers(self):
|
||
gsmtap_hdr = self._gsmtap_hdr(2)
|
||
data = b'\xa0\xa4\x00\x00\x02\x3f\x00'
|
||
pkt = build_pcap([(gsmtap_hdr, data, 1.0)])
|
||
payload = pkt[24 + 16:] # skip global + pkt header
|
||
|
||
# Ethernet: dst MAC, src MAC, ethertype 0x0800 (IPv4)
|
||
self.assertEqual(payload[12:14], b'\x08\x00')
|
||
# IPv4: protocol UDP (17), src/dst 127.0.0.1
|
||
self.assertEqual(payload[23], 17)
|
||
self.assertEqual(payload[26:30], b'\x7f\x00\x00\x01')
|
||
self.assertEqual(payload[30:34], b'\x7f\x00\x00\x01')
|
||
# UDP: dst port 4729
|
||
self.assertEqual(struct.unpack('!H', payload[36:38])[0], 4729)
|
||
# GSMTAP payload follows framing
|
||
self.assertEqual(payload[FRAMING:], gsmtap_hdr + data)
|
||
|
||
def test_multiple_packets(self):
|
||
packets = [
|
||
(self._gsmtap_hdr(1), b'\x3b\x00', 1.0),
|
||
(self._gsmtap_hdr(2), b'\x80\xf2\x00\x00\x00', 2.0),
|
||
]
|
||
data = build_pcap(packets)
|
||
# global header + 2 packets (each: 16 pkt hdr + framing + 16 gsmtap hdr + data)
|
||
self.assertEqual(len(data), 24 + (16 + FRAMING + 16 + 2) + (16 + FRAMING + 16 + 5))
|
||
|
||
|
||
class TestContentDisposition(unittest.TestCase):
|
||
def test_ascii(self):
|
||
h = _content_disposition('My Capture.pcap')
|
||
self.assertIn('filename="My Capture.pcap"', h)
|
||
|
||
def test_unicode(self):
|
||
h = _content_disposition('Моя сессия.pcap')
|
||
# ASCII fallback uses '?' replacements; UTF-8 filename* carries the real name
|
||
self.assertIn("filename*=UTF-8''", h)
|
||
self.assertIn('%D0%9C', h) # 'М' = %D0%9C
|
||
|
||
def test_special_chars(self):
|
||
h = _content_disposition('a"b.pcap')
|
||
self.assertIn('filename="a_b.pcap"', h)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|