Scapy

Question

我正在尝试向数据包添加一个字段，该字段的类型依赖于另一个字段值。其中一个 "types" 是可变长度字符串，如果适用，还有一个额外的 "len" 属性。

考虑以下示例：

from scapy.all import *


class TestPacket(Packet):
        fields_desc=[ 
                  ShortField("index", 0),
                  ByteEnumField("type", 0 , { 1: "String", 2: "Integer" }),
                  ConditionalField(FieldLenField("len", None, length_of="data"), lambda x: x.type==1),
                  ConditionalField(StrLenField("data", "", length_from=lambda pkt:pkt.len), lambda x: x.type==7 or x.type==11),
                  ConditionalField(IntField("data", 1), lambda pkt: pkt.type==2)]


p = TestPacket(index=12, type=1, data="My Sample String")
p.show2()

p = TestPacket(index=13, type=2, data=5)
p.show2()

问题是 "len" 总是设置为 4（IntField 的长度，因为它是最后一个 "data" 条件字段）。

实现此功能的最佳方法是什么？具有不同名称的条件字段？将原始字节存储为可变长度字符串，无论类型如何？

Answer 1

您可以使用 MultipleTypeField。看 https://scapy.readthedocs.io/en/latest/api/scapy.fields.html#scapy.fields.MultipleTypeField

例如：

class DebugPacket(Packet):
    fields_desc = [
        ByteEnumField("atyp", 0x1, {0x1: "IPv4", 0x3: "DNS", 0x4: "IPv6"}),
        MultipleTypeField(
            [
                # IPv4
                (IPField("addr", "0.0.0.0"), lambda pkt: pkt.atyp == 0x1),
                # DNS
                (DNSStrField("addr", ""), lambda pkt: pkt.atyp == 0x3),
                # IPv6
                (IP6Field("addr", "::"), lambda pkt: pkt.atyp == 0x4),
            ],
            StrField("addr", "")  # By default
        ),
    ]

它允许您根据条件使同一字段具有多种类型

Scapy - 添加类型依赖于其他字段的字段

Scapy - Add field with type dependant on other field

python

python-3.x