capture: re-order classes (#6768)

This better reflects the inheritance / smartness with regard to raw or
encoded.

- FDCaptureBinary
- FDCapture
- SysCaptureBinary
- SysCapture
- TeeSysCapture
This commit is contained in:
Daniel Hahler 2020-02-20 11:00:19 +01:00 committed by GitHub
parent fb16d3e27a
commit 82f5986424
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 14 deletions

View File

@ -621,9 +621,9 @@ class FDCapture(FDCaptureBinary):
return res
class SysCapture:
class SysCaptureBinary:
EMPTY_BUFFER = str()
EMPTY_BUFFER = b""
_state = None
def __init__(self, fd, tmpfile=None):
@ -651,7 +651,7 @@ class SysCapture:
self._state = "started"
def snap(self):
res = self.tmpfile.getvalue()
res = self.tmpfile.buffer.getvalue()
self.tmpfile.seek(0)
self.tmpfile.truncate()
return res
@ -675,6 +675,16 @@ class SysCapture:
self._old.flush()
class SysCapture(SysCaptureBinary):
EMPTY_BUFFER = str() # type: ignore[assignment] # noqa: F821
def snap(self):
res = self.tmpfile.getvalue()
self.tmpfile.seek(0)
self.tmpfile.truncate()
return res
class TeeSysCapture(SysCapture):
def __init__(self, fd, tmpfile=None):
name = patchsysdict[fd]
@ -688,17 +698,6 @@ class TeeSysCapture(SysCapture):
self.tmpfile = tmpfile
class SysCaptureBinary(SysCapture):
# Ignore type because it doesn't match the type in the superclass (str).
EMPTY_BUFFER = b"" # type: ignore
def snap(self):
res = self.tmpfile.buffer.getvalue()
self.tmpfile.seek(0)
self.tmpfile.truncate()
return res
map_fixname_class = {
"capfd": FDCapture,
"capfdbinary": FDCaptureBinary,