python / dpkt: выяснить, является ли пакет пакетом tcp или пакетом udp, - PullRequest
Новая
       61

python / dpkt: выяснить, является ли пакет пакетом tcp или пакетом udp,

9 голосов
/ 13 января 2012

У меня есть сценарии Python, которые захватывают пакеты в Ethernet с помощью dpkt, но как мне различить, какие пакеты являются tcp, а какие - для udp.

В конце концов я хотел бы иметь список пакетов для каждого соединения TCP, которое было установлено в течение интервала времени.

мой код: 

import dpkt
import pcapy
cap=pcap.open_live('eth0',100000,1,0)
(header,payload)=cap.next()
while header:
    eth=dpkt.ethernet.Ethernet(str(payload))
    ip=eth.data
    tcp=ip.data 
    # i need to know whether it is a tcp or  a udp packet here!!!
    (header,payload)=cap.next()

Ответы [ 2 ]

7 голосов
/ 13 января 2012

IP-заголовок содержит поле протокола.dpkt должен позволить вам получить это значение и, используя его, вы сможете угадать, что находится на вершине IP.Вот список допустимых номеров протоколов http://www.iana.org/assignments/protocol-numbers/protocol-numbers.xml. UDP равен 17, а TCP - 6.

Редактировать: я проверил эту проблему и, как я уже упоминал, dpkg предоставляет p свойства для доступа к протоколуполе IP.Таким образом, вы можете проверить это снова.Но он также автоматически анализирует пакет и присваивает свойству data экземпляр класса, который представляет протокол более высокого уровня, такой как UDP или TCP.Таким образом, вы можете проверить тип свойства data и распознать этот протокол. 

from dpkt.ip import IP, IP_PROTO_UDP
from dpkt.udp import UDP
ip = IP('E\x00\x00"\x00\x00\x00\x00@\x11r\xc0\x01\x02\x03\x04\x01\x02\x03\x04\x00o\x00\xde\x00\x0e\xbf5foobar')
#if ip.p == IP_PROTO_UDP:  # checking for protocol field in ip header
if type(ip.data) == UDP :  # checking of type of data that was recognized by dpkg
    udp = ip.data
    print udp.sport
else:
    print "Not UDP"
6 голосов
/ 05 апреля 2012

Сценарий python, который захватывает пакеты на сетевом адаптере eth0 с использованием dpkt и различает TCP и UDP пакетов IP . 

import dpkt
import pcapy

cap=pcapy.open_live('eth0',100000,1,0)
(header,payload)=cap.next()

while header:
    eth=dpkt.ethernet.Ethernet(str(payload))

    # Check whether IP packets: to consider only IP packets 
    if eth.type!=dpkt.ethernet.ETH_TYPE_IP:
            continue
            # Skip if it is not an IP packet
    ip=eth.data
    if ip.p==dpkt.ip.IP_PROTO_TCP: # Check for TCP packets
           TCP=ip.data 
           # ADD TCP packets Analysis code here
    elif ip.p==dpkt.ip.IP_PROTO_UDP: # Check for UDP packets
           UDP=ip.data 
           # UDP packets Analysis code here

    (header,payload)=cap.next()
...