コード例 #1
0
    def stream(self):
        """
            Yields Flow objects from the dump.
        """

        # There is a weird mingw bug that breaks .tell() when reading from stdin.
        try:
            self.fo.tell()
        except IOError:  # pragma: no cover
            can_tell = False
        else:
            can_tell = True

        off = 0
        try:
            while True:
                data = tnetstring.load(self.fo)
                try:
                    data = io_compat.migrate_flow(data)
                except ValueError as e:
                    raise exceptions.FlowReadException(str(e))
                if can_tell:
                    off = self.fo.tell()
                if data["type"] not in models.FLOW_TYPES:
                    raise exceptions.FlowReadException(
                        "Unknown flow type: {}".format(data["type"]))
                yield models.FLOW_TYPES[data["type"]].from_state(data)
        except ValueError:
            # Error is due to EOF
            if can_tell and self.fo.tell() == off and self.fo.read() == b'':
                return
            raise exceptions.FlowReadException("Invalid data format.")
コード例 #2
0
ファイル: io.py プロジェクト: AlTune/mitmproxy
    def stream(self):
        """
            Yields Flow objects from the dump.
        """

        # There is a weird mingw bug that breaks .tell() when reading from stdin.
        try:
            self.fo.tell()
        except IOError:  # pragma: no cover
            can_tell = False
        else:
            can_tell = True

        off = 0
        try:
            while True:
                data = tnetstring.load(self.fo)
                try:
                    data = io_compat.migrate_flow(data)
                except ValueError as e:
                    raise exceptions.FlowReadException(str(e))
                if can_tell:
                    off = self.fo.tell()
                if data["type"] not in models.FLOW_TYPES:
                    raise exceptions.FlowReadException("Unknown flow type: {}".format(data["type"]))
                yield models.FLOW_TYPES[data["type"]].from_state(data)
        except ValueError:
            # Error is due to EOF
            if can_tell and self.fo.tell() == off and self.fo.read() == '':
                return
            raise exceptions.FlowReadException("Invalid data format.")
コード例 #3
0
ファイル: io.py プロジェクト: mkagenius/mitmproxy
 def stream(self):
     """
         Yields Flow objects from the dump.
     """
     try:
         while True:
             data = tnetstring.load(self.fo)
             try:
                 data = io_compat.migrate_flow(data)
             except ValueError as e:
                 raise exceptions.FlowReadException(str(e))
             if data["type"] not in models.FLOW_TYPES:
                 raise exceptions.FlowReadException("Unknown flow type: {}".format(data["type"]))
             yield models.FLOW_TYPES[data["type"]].from_state(data)
     except ValueError as e:
         if str(e) == "not a tnetstring: empty file":
             return  # Error is due to EOF
         raise exceptions.FlowReadException("Invalid data format.")
コード例 #4
0
ファイル: io.py プロジェクト: skywind0218/mitmproxy
 def stream(self):
     """
         Yields Flow objects from the dump.
     """
     try:
         while True:
             data = tnetstring.load(self.fo)
             try:
                 data = io_compat.migrate_flow(data)
             except ValueError as e:
                 raise exceptions.FlowReadException(str(e))
             if data["type"] not in models.FLOW_TYPES:
                 raise exceptions.FlowReadException("Unknown flow type: {}".format(data["type"]))
             yield models.FLOW_TYPES[data["type"]].from_state(data)
     except ValueError as e:
         if str(e) == "not a tnetstring: empty file":
             return  # Error is due to EOF
         raise exceptions.FlowReadException("Invalid data format.")