使用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

數據從哪裏來？不要試圖用手將其解碼，爲什麼不使用優秀Impacket包：

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

Answer 4

嘗試scapy，一個強大的交互式數據包處理程序。

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)