带有自定义图层的 Scapy sr()

Scapy sr() with custom layers

使用 sr() 或 srp() 函数时 - Scapy 如何知道收到的数据包是对我发送的数据包的答复?

我写了一个模仿 BACNet 的自定义协议。我可以向 BACNet 设备发送一个 WHO_IS 数据包,设备用 I_AM 数据包进行应答,由于层绑定,该数据包被正确反汇编,但 sr 函数无法将其识别为应答。我如何让 Scapy 接受这些数据包作为答案?

更新: 这就是我的图层 classes 的样子。我认为 answers() 方法看起来不错,但它仍然不起作用。在实施 answers() 方法时,我可能有什么误解吗? 在我的理解中,self 指的是层本身的 class,other 是有问题的接收数据包。因此,为了将有效负载传递到下一个更高层,我传递 other.payload 并使用下一个更高层 class 调用 answers() 方法。 图层像 Ether/IP/UDP/BVLC/NPDU/APDU.

一样堆叠
class BVLC(Packet):

    name = "BVLC"
    fields_desc = [
       # many fields
                   ]

    def post_build(self, pkt, pay):
        if self.length == None:                
            pkt = pkt[0:2] + struct.pack('!H', len(pay)+len(pkt)) + pkt[4:]  
        return pkt+pay

    def answers(self, other):
        if not isinstance(other,BVLC):
            return 0
        else:
            return NPDU.answers(other.payload)


class NPDU(Packet):

    name = "NPDU"
    fields_desc = [ 
        # again, many fields
                   ]

    def answers(self, other):
        if not isinstance(other,NPDU):
            return 0
        else:
            return APDU.answers(other.payload)

class APDU(Packet):

    name = "APDU"
    fields_desc = [
          # many fields and the only relevant field in this case
            ConditionalField(
                ByteEnumField(
                    "serviceChoice", None, APDU_Service_Unconfirmed.revDict()),
                    lambda pkt: pkt.pduType == APDU_Type.CONFIRMED_SERVICE_REQUEST or
                                pkt.pduType == APDU_Type.UNCONFIRMED_SERVICE_REQUEST), 
                   ]

    def answers(self, other):
        if not isinstance(other,APDU):
            return 0
        else:
            if self.serviceChoice == APDU_Service_Unconfirmed.WHO_IS and\
               other.serviceChoice == APDU_Service_Unconfirmed.I_AM:
                return 1
        return 0

Scapy sr 只是为每个发送的数据包调用 recv_packet.answers(sent_packet)。 因此,您必须为您的图层实施 def answers(),请参阅:

def answers(self, other):
    """DEV: true if self is an answer from other"""
    if other.__class__ == self.__class__:
        return self.payload.answers(other.payload)
    return 0

这是原始 TCP 层的摘录:

def answers(self, other):
    if not isinstance(other, TCP):
        return 0
    if conf.checkIPsrc:
        if not ((self.sport == other.dport) and
                (self.dport == other.sport)):
            return 0
    if (abs(other.seq-self.ack) > 2+len(other.payload)):
        return 0
    return 1