mirror of
https://github.com/fortra/impacket
synced 2026-06-08 14:15:13 +00:00
gabrielg5
121 lines
2.8 KiB
Python
121 lines
2.8 KiB
Python
# Impacket - Collection of Python classes for working with network protocols.
|
|
#
|
|
# Copyright Fortra, LLC and its affiliated companies
|
|
#
|
|
# All rights reserved.
|
|
#
|
|
# This software is provided under a slightly modified version
|
|
# of the Apache Software License. See the accompanying LICENSE file
|
|
# for more information.
|
|
#
|
|
|
|
from impacket import structure
|
|
|
|
O_ETH = 0
|
|
O_IP = 1
|
|
O_ARP = 1
|
|
O_UDP = 2
|
|
O_TCP = 2
|
|
O_ICMP = 2
|
|
O_UDP_DATA = 3
|
|
O_ICMP_DATA = 3
|
|
|
|
MAGIC = '"\xD4\xC3\xB2\xA1'
|
|
|
|
class PCapFileHeader(structure.Structure):
|
|
structure = (
|
|
('magic', MAGIC),
|
|
('versionMajor', '<H=2'),
|
|
('versionMinor', '<H=4'),
|
|
('GMT2localCorrection', '<l=0'),
|
|
('timeAccuracy', '<L=0'),
|
|
('maxLength', '<L=0xffff'),
|
|
('linkType', '<L=1'),
|
|
('packets','*:=[]'),
|
|
)
|
|
|
|
class PCapFilePacket(structure.Structure):
|
|
structure = (
|
|
('tsec', '<L=0'),
|
|
('tmsec', '<L=0'),
|
|
('savedLength', '<L-data'),
|
|
('realLength', '<L-data'),
|
|
('data',':'),
|
|
)
|
|
|
|
def __init__(self, *args, **kargs):
|
|
structure.Structure.__init__(self, *args, **kargs)
|
|
self['data'] = b''
|
|
|
|
class PcapFile:
|
|
def __init__(self, fileName = None, mode = 'rb'):
|
|
if fileName is not None:
|
|
self.file = open(fileName, mode)
|
|
self.hdr = None
|
|
self.wroteHeader = False
|
|
|
|
def reset(self):
|
|
self.hdr = None
|
|
self.file.seek(0)
|
|
|
|
def close(self):
|
|
self.file.close()
|
|
|
|
def fileno(self):
|
|
return self.file.fileno()
|
|
|
|
def setFile(self, file):
|
|
self.file = file
|
|
|
|
def setSnapLen(self, snapLen):
|
|
self.createHeaderOnce()
|
|
self.hdr['maxLength'] = snapLen
|
|
|
|
def getSnapLen(self):
|
|
self.readHeaderOnce()
|
|
return self.hdr['maxLength']
|
|
|
|
def setLinkType(self, linkType):
|
|
self.createHeaderOnce()
|
|
self.hdr['linkType'] = linkType
|
|
|
|
def getLinkType(self):
|
|
self.readHeaderOnce()
|
|
return self.hdr['linkType']
|
|
|
|
def readHeaderOnce(self):
|
|
if self.hdr is None:
|
|
self.hdr = PCapFileHeader.fromFile(self.file)
|
|
|
|
def createHeaderOnce(self):
|
|
if self.hdr is None:
|
|
self.hdr = PCapFileHeader()
|
|
|
|
def writeHeaderOnce(self):
|
|
if not self.wroteHeader:
|
|
self.wroteHeader = True
|
|
self.file.seek(0)
|
|
self.createHeaderOnce()
|
|
self.file.write(self.hdr.getData())
|
|
|
|
def read(self):
|
|
self.readHeaderOnce()
|
|
try:
|
|
pkt = PCapFilePacket.fromFile(self.file)
|
|
pkt['data'] = self.file.read(pkt['savedLength'])
|
|
return pkt
|
|
except:
|
|
return None
|
|
|
|
def write(self, pkt):
|
|
self.writeHeaderOnce()
|
|
self.file.write(str(pkt))
|
|
|
|
def packets(self):
|
|
self.reset()
|
|
while 1:
|
|
answer = self.read()
|
|
if answer is None:
|
|
break
|
|
yield answer
|