欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-12 18:59:08
...

目前在完成一个网络抓包工具,用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查看。稍后提供比较完整的一个代码文件。

转载于:https://www.cnblogs.com/tt-0411/archive/2011/12/01/2270676.html