用Python做个网络抓包工具之为Winpcapy写个过滤器filter

用Python做个网络抓包工具之为Winpcapy写个过滤器filter

http://www.cnblogs.com/tt-0411/archive/2011/12/01/2270676.html

目前在完成一个网络抓包工具，用Python实现。在Windows上抓包首选的库或接口自然是Winpcap。自然Winpcap这种接近系统底层和硬件的程序一般都是c语言写的，他提供的开发接口原生也是c的。不过还好，人们已经为其他语言的使用提供了封装，比如java和.net，好像连vb都有了，可以在winpcap官网上找找。

Winpcapy是Python语言对Winpcap的封装，可以在google code上找到这个项目，并下载源代码。其实和winpcap的c语言代码头文件对比可以发现，Winpcapy主要是用ctype把winpcap提供的c函数和一些数据结构转化成了Python的函数和数据结构，就像c语言是一个pcap.h，winpcapy就是一个winpcapy.py，只要在代码中导入这个模块，就能和c语言一样使用winpcap了。另外，源代码提供了winpcapy.py之外，也像winpcap开发包一样提供了几个sample。比较一下这两套sample会发现，winpcapy的sample基本上就是把c语言写的sample翻译成了Python语言的写法，对应关系是很显而易见的。不过，sample里面没有filter方面的代码，于是我就仿照winpcap sample中的pcap_filter.c，自己写一个Python版的pcap_filter。

pcap_filter.c中关于过滤器filter的主要代码是这些:

//pcap_t *fp;
//struct bpf_program fcode;
//bpf_u_int32 NetMask;
//char *filter;

NetMask=0xffffff;

//compile the filter
if (pcap_compile(fp, &fcode, filter, 1, NetMask) < 0)
{
    printf (stderr, "\nError compiling filter: wrong syntax.\n" );

    pcap_close(fp);
    return  -3;
}

//set the filter
if (pcap_setfilter(fp, &fcode)<0)
{
    fprintf (stderr, "\nError setting the filter\n" );

    pcap_close(fp);
    return  -4;
}

　　开始几行给出了几个参数的声明，具体请参见完整代码。下面给出我写的filter部分的代码：

fcode =  bpf_program()
NetMask =  0xffffff
filter  =  "tcp"

## compile the filter
if  pcap_compile(adhandle,byref(fcode), filter , 1 ,NetMask) < 0 :
    print ( '\nError compiling filter: wrong syntax.\n' )
    pcap_close(adhandle)
    sys.exit( - 3 )

## set the filter

if  pcap_setfilter(adhandle,byref(fcode)) < 0 :
    print ( '\nError setting the filter\n' )
    pcap_close(adhandle)
    sys.exit( - 4 )

　　当然看上去基本上是对着翻译，没什么技术含量。这段代码可以放到winpcapy示例basic_dump_ex.py开始抓包的代码行之前（

res = pcap_next_ex( adhandle, byref(header), byref(pkt_data))

）就能设置过滤器是“tcp”了。

如果要验证的话，可以在用pcap_dump把抓包保存下来，然后用wireshark查看。稍后提供比较完整的一个代码文件。