python基础教程之获取本机ip数据包示例

这几天用到了raw socket,用python写了些demo程序,这里记录下。

首先我们看一个简单的sniffer程序:

代码如下:

#! /usr/bin/python# code for linuximport 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接收数据,直接print操作。这个就几行代码,也没什么好解释的了,不懂的google下。

得到ip数据包后,接下来的工作就是对ip头进行解析,在这之前,我们先看看rfc中是怎么定义的(rfc791 : http://www.ietf.org/rfc/rfc791.txt ):

从rfc和上图中可以看到ip数据包头各个字段所占的位数,我们可以根据这些定义去解析ip数据包头,然后根据相应的策略处理数据。这里给出一段用python实现的解析ip头的代码(呵呵,是demo中的代码,只解析了前20个字节):

代码如下:

def decodeipheader(packet): mapret = {} mapret[“version”] = (int(ord(packet[0])) & 0xf0)>>4 mapret[“headerlen”] = (int(ord(packet[0])) & 0x0f)5 mapret[“fragoff”] = int(ord(packet[6]) & 0x1f)