【笔记】wincap收集并统计网络流量

/////////////////////////////////////////////////////////////////////////////
// Name:        TrafficCount.cpp
// Purpose:     wincap收集并统计网络流量。
// Compiler:    VS2005
// Author:      陈相礼
// Modified by:
// Created:     09/18/09
// Copyright:
// Licence:
/////////////////////////////////////////////////////////////////////////////
#include <pcap.h>
#include <remote-ext.h>
#pragma comment(lib,"Packet.lib")
#pragma comment(lib,"wpcap.lib")
#pragma comment(lib,"ws2_32.lib")
void usage();
void dispatcher_handler( u_char *, const struct pcap_pkthdr *, const u_char * );
void main( int argc, char **argv)
{
pcap_t *fp;
char error[PCAP_ERRBUF_SIZE];
struct timeval st_ts;
u_int netmask;
struct bpf_program fcode;
/* 检查命令行参数的合法性 */
if ( argc != 2 )
{
usage();
return;
}
/* 打开输出适配器 */
if ( ( fp = pcap_open_live( argv[1], 100, 1, 1000, error ) ) == NULL )
{
fprintf_s( stderr, "/n适配器打开失败: %s/n", error );
return;
}
/* 不用关心掩码，在这个过滤器中，它不会被使用 */
netmask=0xffffff;
// 编译过滤器
if ( pcap_compile( fp, &fcode, "tcp", 1, netmask ) < 0 )
{
fprintf_s( stderr,"/n不能编译包过滤器，检查语法./n" );
/* 释放设备列表 */
return;
}
// 设置过滤器
if ( pcap_setfilter( fp, &fcode ) < 0 )
{
fprintf_s( stderr, "/n过滤器设置失败./n" );
/* 释放设备列表 */
return;
}
/* 将网卡设置为统计模式 */
pcap_setmode( fp, MODE_STAT );
printf_s( "TCP 流量统计:/n" );
/* 开始主循环 */
pcap_loop( fp, 0, dispatcher_handler, (PUCHAR)&st_ts );
pcap_close(fp);
return;
}
void dispatcher_handler( u_char *state, const struct pcap_pkthdr *header, const u_char *pkt_data )
{
struct timeval *old_ts = (struct timeval *)state;
u_int delay;
LARGE_INTEGER Bps,Pps;
struct tm ltime;
char timestr[16];
/* 以微妙计算上一次采样的延迟时间 */
/* 这个值通过采样到的时间戳获得 */
delay =(header->ts.tv_sec - old_ts->tv_sec) * 1000000 - old_ts->tv_usec + header->ts.tv_usec;
/* 获得每秒的比特数 */
Bps.QuadPart = (((*(LONGLONG*)(pkt_data + 8)) * 8 * 1000000) / (delay));
/*                                            ^      ^
|      |
|      |
|      |
将字节转换成比特 --   |
|
延时是以微妙表示的 --
*/
/* 获得每秒的数据包数 */
Pps.QuadPart = (((*(LONGLONG*)(pkt_data)) * 1000000) / (delay));
/* 将时间戳转变为易读的标准格式*/
time_t t = (time_t )&header->ts.tv_sec;
localtime_s( <ime, &t );
strftime( timestr, sizeof timestr, "%H:%M:%S", <ime);
printf_s("%s,/t%.6d毫秒/t长度:%d/n", timestr, header->ts.tv_usec, header->len );
/* 打印时间戳*/
printf_s("%s ", timestr);
/* 打印采样结果 */
printf_s( "BPS=%I64u  ", Bps.QuadPart );
printf_s( "PPS=%I64u/n", Pps.QuadPart );
//存储当前的时间戳
old_ts->tv_sec=header->ts.tv_sec;
old_ts->tv_usec=header->ts.tv_usec;
}
void usage()
{
printf_s("/nUsage:/n");
printf_s("/t tcptop adapter/n");
printf_s("/t You can use /"WinDump -D/" if you don‘t know the name of your adapters./n");
exit(0);
}