快捷導(dǎo)航

python基礎(chǔ)教程之獲取本機(jī)ip數(shù)據(jù)包示例

更新時(shí)間：2014年02月10日 10:29:10 作者：

本文主要介紹了python獲取本機(jī)ip數(shù)據(jù)包的示例，大家參考使用吧

這幾天用到了raw socket，用python寫了些demo程序，這里記錄下。

首先我們看一個(gè)簡(jiǎn)單的sniffer程序：

#! /usr/bin/python
# code for linux
import socket
#s = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_UDP)
s = socket.socket(socket.AF_INET, socket.SOCK_RAW, socket.IPPROTO_TCP)
while True:
    print s.recvfrom(65535)

這里直接用raw socket接收數(shù)據(jù)，直接print操作。這個(gè)就幾行代碼，也沒什么好解釋的了，不懂的google下。

得到IP數(shù)據(jù)包后，接下來的工作就是對(duì)IP頭進(jìn)行解析，在這之前，我們先看看RFC中是怎么定義的（RFC791 : http://www.ietf.org/rfc/rfc791.txt ）：

即對(duì)應(yīng)的圖：

從RFC和上圖中可以看到IP數(shù)據(jù)包頭各個(gè)字段所占的位數(shù)，我們可以根據(jù)這些定義去解析IP數(shù)據(jù)包頭，然后根據(jù)相應(yīng)的策略處理數(shù)據(jù)。
這里給出一段用python實(shí)現(xiàn)的解析IP頭的代碼（呵呵，是demo中的代碼，只解析了前20個(gè)字節(jié)）：

復(fù)制代碼代碼如下:

def decodeIpHeader(packet):
        mapRet = {}
        mapRet["version"] = (int(ord(packet[0])) & 0xF0)>>4
        mapRet["headerLen"] = (int(ord(packet[0])) & 0x0F)<<2
        mapRet["serviceType"] = hex(int(ord(packet[1])))
        mapRet["totalLen"] = (int(ord(packet[2])<<8))+(int(ord(packet[3])))
        mapRet["identification"] = (int( ord(packet[4])>>8 )) + (int( ord(packet[5])))
        mapRet["id"] = int(ord(packet[6]) & 0xE0)>>5
        mapRet["fragOff"] = int(ord(packet[6]) & 0x1F)<<8 + int(ord(packet[7]))
        mapRet["ttl"] = int(ord(packet[8]))
        mapRet["protocol"] = int(ord(packet[9]))
        mapRet["checkSum"] = int(ord(packet[10])<<8)+int(ord(packet[11]))
        mapRet["srcaddr"] = "%d.%d.%d.%d" % (int(ord(packet[12])),int(ord(packet[13])),int(ord(packet[14])), int(ord(packet[15])))
        mapRet["dstaddr"] = "%d.%d.%d.%d" % (int(ord(packet[16])),int(ord(packet[17])),int(ord(packet[18])), int(ord(packet[19])))
        return mapRet

調(diào)用代碼：

復(fù)制代碼代碼如下:

proto = socket.getprotobyname('tcp') # only tcp
sock = socket.socket(socket.AF_INET, socket.SOCK_RAW, proto)

while True:
        packet = sock.recvfrom(65535)[0]
        if len(packet) == 0:
                sck.close()
        else:
                #print str(packet)
                mapIpTmp = decodeIpHeader(packet)
                for k,v in mapIpTmp.items():
                        print k,"\t:\t",v

print ""

您可能感興趣的文章: