def testPacketDumper(self):
"""
#6 test that the dumper writes correct payload
"""
try:
r = pcapy.open_offline(TestPcapy._96PINGS)
dumper = r.dump_open('tmp.pcap')
hdr, body = r.next()
i = 0
while hdr is not None:
dumper.dump(hdr, body)
i += 1
hdr, body = r.next()
# make sure file closes
del dumper
# check that the dumper wrote a legal pcap
# file with same packer data
r = pcapy.open_offline(TestPcapy._96PINGS)
r2 = pcapy.open_offline('tmp.pcap')
h1, b1 = r.next()
h2, b2 = r2.next()
while h1 is not None and h2 is not None:
self.assertEqual(b1, b2)
h1, b1 = r.next()
h2, b2 = r2.next()
self.assertTrue(h1 is None)
self.assertTrue(h2 is None)
del r2
finally:
os.unlink('tmp.pcap')
def testContextManager(self):
"""
#8 Test the context manager support
"""
with pcapy.open_offline(TestPcapy._96PINGS) as r:
hdr, body = r.next()
assert hdr is not None
with self.assertRaises(ValueError):
r.next()
def testClose(self):
"""
#7 Test the close method
"""
r = pcapy.open_offline(TestPcapy._96PINGS)
hdr, body = r.next()
assert hdr is not None
r.close()
with self.assertRaises(ValueError):
r.next()
def testBPFFilter(self):
"""
#3 disabled -- bpf not supported yettest offline BPFFilter
"""
r = pcapy.open_offline(TestPcapy._96PINGS)
bpf = pcapy.BPFProgram("ip dst host 192.168.1.1")
hdr, pkt = r.next()
while hdr is not None:
f = bpf.filter(pkt)
self.assertNotEqual(f, 0)
hdr, pkt = r.next()
def testPacketHeaderRefCount(self):
"""
#1: when next() creates a pkthdr it makes one extra reference
"""
class _Simple:
pass
r = pcapy.open_offline(TestPcapy._96PINGS)
# get one & check its refcount
self.assertEqual(sys.getrefcount(r.next()[0]),
sys.getrefcount(_Simple()))
def _testSendPacket(self):
"""
#5 (disabled -- requires interface info) test sendpacket
"""
r = pcapy.open_offline(TestPcapy._96PINGS)
w = pcapy.open_live(TestPcapy._IFACE, 60000, 1, 1500)
# get one & check its refcount
i = 0
hdr, pkt = r.next()
while hdr is not None:
w.sendpacket(pkt)
hdr, pkt = r.next()
i += 1
def testEOFValue(self):
"""
#2 empty string is returned as packet body at end of file
"""
r = pcapy.open_offline(TestPcapy._96PINGS)
# get one & check its refcount
i = 0
refNone = sys.getrefcount(None)
hdr, pkt = r.next()
while hdr is not None:
hdr, pkt = r.next()
i += 1
self.assertEqual(96, i)
self.assertTrue(hdr is None)
self.assertEqual(pkt, b'')
del hdr
self.assertEqual(refNone, sys.getrefcount(None))