Class IPFIXDecoder

test/ipfix.py:517–557

IPFIX data set decoder

class IPFIXDecoder:
    """IPFIX data set decoder"""

    def __init__(self):
        self._templates = []

    def add_template(self, template):
        """
        Add IPFIX template

        :param template: IPFIX template
        """
        templateID = template.templateID
        fields = []
        rec_len = 0
        for field in template.templateFields:
            fields.append({"name": field.informationElement, "len": field.fieldLength})
            rec_len += field.fieldLength
        self._templates.append({"id": templateID, "fields": fields, "rec_len": rec_len})

    def decode_data_set(self, data_set):
        """
        Decode IPFIX data

        :param data_set: IPFIX data set
        :returns: List of decoded data records.
        """
        data = []
        for template in self._templates:
            if template["id"] == data_set.setID:
                offset = 0
                d = data_set[Data].data
                for i in range(len(d) // template["rec_len"]):
                    record = {}
                    for field in template["fields"]:
                        f = d[offset : offset + field["len"]]
                        offset += field["len"]
                        record.update({field["name"]: f})
                    data.append(record)
                break
        return data