Skip to content
Snippets Groups Projects
Commit d2f8660d authored by Guillaume Valadon's avatar Guillaume Valadon
Browse files

Merged secdev/scapy into default

parents b77e3cb7 d628e3b8
No related branches found
No related tags found
No related merge requests found
...@@ -560,7 +560,12 @@ class RawPcapReader: ...@@ -560,7 +560,12 @@ class RawPcapReader:
def close(self): def close(self):
return self.f.close() return self.f.close()
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, tracback):
pass
class PcapReader(RawPcapReader): class PcapReader(RawPcapReader):
def __init__(self, filename): def __init__(self, filename):
...@@ -673,9 +678,16 @@ class RawPcapWriter: ...@@ -673,9 +678,16 @@ class RawPcapWriter:
def flush(self): def flush(self):
return self.f.flush() return self.f.flush()
def close(self): def close(self):
return self.f.close() return self.f.close()
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, tracback):
self.flush()
class PcapWriter(RawPcapWriter): class PcapWriter(RawPcapWriter):
def _write_header(self, pkt): def _write_header(self, pkt):
if self.linktype == None: if self.linktype == None:
......
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