Scapy:从具有自己的init的Packet派生的层

时间:2015-07-10 09:04:52

标签: python constructor initialization scapy

是否可以使用自己的__init__()函数构建自己的图层? 我试图这样做的方式是:

>>> class SomePacket(Packet):
...     def __init__(self, number):
...             fields_desc = [IntField("Number",number)]
... 
>>> packet = SomePacket(1)
>>> ls(packet)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python2.7/dist-packages/scapy/packet.py", line 1213, in ls
    if not isinstance(obj.payload, NoPayload):
  File "/usr/lib/python2.7/dist-packages/scapy/packet.py", line 180, in __getattr__
    raise AttributeError(attr)
AttributeError: payload

这样做是错误的,我应该如何编写类以便将它与一种构造函数一起使用?

1 个答案:

答案 0 :(得分:1)

天空是极限,你可以按自己喜欢的方式扩展或滥用scapy。因此,是的,您可以拥有自己的__init__但是您必须重新实现一些scapy代码或者至少调用超类来适应接口。此外,您必须将fields_desc定义为类/实例var而不是__init__本地var(即将代码更改为self.fields_desc以使其成为实例var或使其成为一个明确的类var。

但是,如果您只是尝试为图层字段指定值,那么最好不要重新发明轮子并使用scapy的设计方式,因为它已经提供__init__之类的接口

class SomePacket(Packet):
    fields_desc = [IntField("Number",1),  # capital letters are not convenient
                   IntField("yournumber",1),]


packet = SomePacket(Number=1, yournumber=99)
packet.show()

导致:

###[ SomePacket ]###
  Number    = 1
  yournumber= 99

我唯一必须覆盖__init__的层是SSL / TLS层,它为跨越多个层的ssl上下文(加密/密钥)提供了一个隐藏字段(用于解密,...)< / p>