Skip to content
Snippets Groups Projects
Commit 8d15ff5a authored by Pierre LALET's avatar Pierre LALET
Browse files

wrpcap() / (Raw)PcapWriter: allow the use of a file-like object instead of a filename

parent 0eae79be
No related branches found
No related tags found
No related merge requests found
...@@ -839,17 +839,23 @@ class RawPcapWriter: ...@@ -839,17 +839,23 @@ class RawPcapWriter:
self.linktype = linktype self.linktype = linktype
self.header_present = 0 self.header_present = 0
self.append=append self.append = append
self.gz = gz self.gz = gz
self.endian = endianness self.endian = endianness
self.filename=filename self.sync = sync
self.sync=sync
bufsz=4096 bufsz=4096
if sync: if sync:
bufsz=0 bufsz = 0
if isinstance(filename, basestring):
self.filename = filename
self.f = [open,gzip.open][gz](filename,append and "ab" or "wb", gz and 9 or bufsz)
else:
self.f = filename
self.filename = (filename.name
if hasattr(filename, "name") else
"No name")
self.f = [open,gzip.open][gz](filename,append and "ab" or "wb", gz and 9 or bufsz)
def fileno(self): def fileno(self):
return self.f.fileno() return self.f.fileno()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment