diff --git a/scapy/utils.py b/scapy/utils.py index 07e3a8ae89af2d99124fc94ad0f0e0c7323ef422..661385671ada06d6226b8920699f9b9a6b857690 100644 --- a/scapy/utils.py +++ b/scapy/utils.py @@ -560,7 +560,12 @@ class RawPcapReader: def close(self): return self.f.close() - + def __enter__(self): + return self + + def __exit__(self, exc_type, exc_value, tracback): + pass + class PcapReader(RawPcapReader): def __init__(self, filename): @@ -673,9 +678,16 @@ class RawPcapWriter: def flush(self): return self.f.flush() + def close(self): return self.f.close() - + + def __enter__(self): + return self + def __exit__(self, exc_type, exc_value, tracback): + self.flush() + + class PcapWriter(RawPcapWriter): def _write_header(self, pkt): if self.linktype == None: