无法使用dpkt(python)提取大多数直径的AVP

时间:2013-06-28 09:40:37

标签: python pcap

我需要从直径数据包中提取所有AVPs详细信息。目前我正在使用dpkt lib Python,但我发现它不支持提取所有AVP。

在标题级别,仅支持以下

class Diameter(dpkt.Packet):
    __hdr__ = (
        ('v', 'B', 1),
        ('len', '3s', 0),
        ('flags', 'B', 0),
        ('cmd', '3s', 0),
        ('app_id', 'I', 0),
        ('hop_id', 'I', 0),
        ('end_id', 'I', 0)
        )

在AVP级别,支持,其余的AVP存储为二进制数据(如果我没有错)

class AVP(dpkt.Packet):
    __hdr__ = (
        ('code', 'I', 0),
        ('flags', 'B', 0),
        ('len', '3s', 0),
        )

import dpkt

f = open('snoop.pcap','r')
pcap = dpkt.pcap.Reader(f)


for ts, buf in pcap:
    eth = dpkt.ethernet.Ethernet(buf)
    ip = eth.data
    tcp = ip.data
    dia = dpkt.diameter.Diameter(tcp.data)
    avp = dpkt.diameter.AVP(tcp.data)
    if tcp.dport == 3868 and len(tcp.data) > 0:
        print (str(dia.cmd) + ' ' + str(dia.hop_id) + ' ' + str(dia.end_id) + ' ' + str(avp.code))

f.close() 

有没有比dpkt更好的其他lib或者使用这个lib提取AVP的其他任何方法?

版本:dpkt 1.8

1 个答案:

答案 0 :(得分:0)

dpkt的Diameter解析器有一个错误(从版本89开始)。 根据RFC 3588,第4节: “将多个零值字节添加到AVP数据字段的末尾,直到达到字边界。填充的长度不会反映在AVP长度字段中。”

此填充未实现。像下面这样修改diameter.py: 在AVP类的unpack()方法(第119行)中,添加:

self.padding = ( 4 - self.len % 4 ) % 4

在AVP类的第__len __()方法中(第140行),添加:

length += self.padding

这些修改为我解决了这个问题。