使用python解码tcp数据包

Question

我正在尝试解码通过 TCP 连接收到的数据。数据包很小，不超过100字节。然而，当它们很多时，我会收到一些连接在一起的数据包。有没有办法防止这种情况。我正在使用 python，

我尝试分离数据包，我的源代码如下。数据包以STX字节开始，以ETX字节结束，STX后面的字节是数据包长度，（数据包长度小于5无效）校验和是ETX之前的最后一个字节

def decode(data):
  while True:
    start = data.find(STX)
    if start == -1: #no stx in message
        pkt = ''
        data = ''
        break
    #stx found , next byte is the length
    pktlen = ord(data[1])
    #check message ends in ETX (pktken -1) or checksum invalid
    if pktlen < 5 or data[pktlen-1] != ETX or checksum_valid(data[start:pktlen]) == False:
        print "Invalid Pkt"
        data = data[start+1:]
        continue
    else:
        pkt = data[start:pktlen]
        data = data[pktlen:]
        break

return data , pkt

我这样使用它

#process reports
try:
    data = sock.recv(256) 
except: continue 
else:
    while data:
        data, pkt = decode(data) 
        if pkt:
           process(pkt)

此外，如果有多个数据流中的数据包，最好将数据包作为列表的集合返回还是只返回第一个数据包

我对python不太熟悉，只会C，这种方法可以吗？任何建议将不胜感激。提前致谢

谢谢

Answer 1

我将创建一个类，负责解码流中的数据包，如下所示：

class PacketDecoder(object):

    STX = ...
    ETX = ...

    def __init__(self):
        self._stream = ''

    def feed(self, buffer):
        self._stream += buffer

    def decode(self):
        '''
        Yields packets from the current stream.
        '''
        while len(self._stream) > 2:
            end = self._stream.find(self.ETX)
            if end == -1:
                break

            packet_len = ord(self._stream[1])
            packet = self._stream[:end]
            if packet_len >= 5 and check_sum_valid(packet):
                yield packet
            self._stream = self._stream[end+1:]

然后像这样使用：

decoder = PacketDecoder()
while True:
    data = sock.recv(256) 
    if not data:
        # handle lost connection... 
    decoder.feed(data)
    for packet in decoder.decode():
        process(packet)

Answer 2

TCP 在接口级别提供数据流，而不是单个数据包。如果您想要离散数据包，您可以使用 UDP（并自行处理丢失或无序的数据包），或内联一些数据分隔符。听起来您已经在这样做了，并使用 STX/ETX 作为分隔符。然而，正如您所注意到的，您从 TCP 堆栈的一个数据块中获取多条消息。

请注意，除非您正在进行其他处理，否则您显示的代码中的 data 不一定包含整数数量的消息。也就是说，最后一个STX很可能不会有匹配的ETX。 ETX 将位于下一个没有 STX 的data 块中。

您可能应该从 TCP 数据流中读取各个消息并在发生时返回它们。

Answer 3

尝试 scapy，一个强大的交互式数据包操作程序。

Answer 4

数据从哪里来？与其尝试手动解码，为什么不使用优秀的 Impacket 包：

http://oss .coresecurity.com/projects/impacket.html

Answer 5

漂亮又简单...:)
诀窍在于 file 对象。

f=sock.makefile()
while True:
  STX = f.read(1)
  pktlen = f.read(1)
  wholePacket = STX + pktlen + f.read(ord(pktlen)-2)
  doSomethingWithPacket(wholePacket)

就是这样！（使用 TCP 时也不需要检查校验和。）

这是一个更“健壮”（？）的版本（它使用 STX 和校验和）：

f=sock.makefile()
while True:
  while f.read(1)!=STX:
    continue
  pktlen = f.read(1)
  wholePacket = STX + pktlen + f.read(ord(pktlen)-2)
  if checksum_valid(wholePacket):
    doSomethingWithPacket(wholePacket)