使用Winpcap进行IP数据包统计

之前课设做过一回，然后过了半年不到，又华华丽丽的忘了。。。最近又要用，所以又捡了起来。

首先是环境的准备

首先是安装包 http://download.csdn.net/detail/u013529927/7174089

然后是开发包 http://download.csdn.net/detail/u013529927/7174095

上面的都是最新版的。当然，你也可以去官网上下 http://www.winpcap.org/第一个标签 里面有“Get Winpcap”和"Development"分别对应环境和开发包。

然后安装环境包，一直next就行。。。 

然后就可以打开VS了~（因为我用的VS，VC 6.0其实也差不多。。。）

把开发包中的 Include 和 Lib文件夹放到任意一个位置（因为后面都用的绝对路径，所以放哪都无所谓，当然还是推荐工程文件夹下面）

然后在 项目-> （工程名）属性（PS：VS2013里面是最后一个）->配置属性->VC++目录 中在包含目录里面添加 上面 Include文件夹的位置 在库目录里面添加 Lib文件夹的位置。

然后在工程中添加两个文件

第一个是头文件 IPNodeList.h

#include <fstream>
#include <iostream>
using namespace std;
class IPNode
{
private:
long m_lIPAddress;                //IP地址
long m_lCount;                   //发送数据包数
public:
IPNode*pNext;                   //指向下一个IP节点
//构造函数
IPNode(long sourceIP)
{
m_lIPAddress = sourceIP;
m_lCount = 1;                 //初始化数据包个数为1
}
//数据包个数加1
void addCount()
{
m_lCount++;
}
//返回数据包个数
long getCount()
{
return m_lCount;
}
//返回IP地址
long getIPAddress()
{
return m_lIPAddress;
}
};
//节点链表
class NodeList
{
IPNode*pHead;                //链表头
IPNode*pTail;                 //链表尾
public:
NodeList()
{
pHead = pTail = NULL;
}
~NodeList()
{

if (pHead != NULL)
{
IPNode*pTemp = pHead;
pHead = pHead->pNext;
delete  pTemp;
}
}
//IP节点加入链表
void addNode(long sourceIP)
{
IPNode* pTemp;
if (pHead == NULL)             //当链表为空时
{
pTail = new IPNode(sourceIP);
pHead = pTail;
pTail->pNext = NULL;
}
else                      //不为空时
{
for (pTemp = pHead; pTemp; pTemp = pTemp->pNext)
{
//如果链表中存在此IP，发送数据包个数加1
if (pTemp->getIPAddress() == sourceIP)
{
pTemp->addCount();
break;
}
}
//如果链表中没有此IP，则加入链表
if (pTemp == NULL)
{
pTail->pNext = new IPNode(sourceIP);
pTail = pTail->pNext;
pTail->pNext = NULL;
}
}
}
//输出IP结点，即IP地址和其它发送的IP包个数
ostream& print(ostream& os)
{
for (IPNode*pTemp = pHead; pTemp; pTemp = pTemp->pNext)
{
long lTemp = pTemp->getIPAddress();
os << inet_ntoa(*(in_addr*)&(lTemp)) << '\t';
os << pTemp->getCount() << endl;
}
return os;
}
};

第二个 因为是cpp文件，所以文件名随便起一个就行~

#include <iostream>
#include <iomanip>
#include <fstream>
#include <stdlib.h>
#include <stdio.h>
#include <conio.h>

#include "pcap.h"
#include "IPNodeList.h"
//等同于点击“Project→Setting→link"打开object/library modules编辑框后加入lib文件
#pragma comment(lib,"Wpcap.lib")
#pragma comment(lib,"Ws2_32.lib")

using namespace std;
//IP包的头部结构
struct ip_header{
unsigned char ver_ih1;          //版本号（4位）+头部长度（4位）
unsigned char tos;              //服务类型
unsigned short tlen;             //总长度
unsigned short identification;     //标识
unsigned short flags_fo;         //标志+片偏移
unsigned char tt1;              //生存时间
unsigned char proto;            //协议
unsigned short crc;             //校检和
DWORD saddr;               //源地址
DWORD daddr;               //目的地址
unsigned int    op_pad;        //选项+填充
};

void main(int argc, char*argv[])
{
if (argc != 3)          //判断参数是否正确
{
cout << "Usage:IPStatistic  time  logfile" << endl;
cout << "Press any key to continue…" << endl;
_getch();
return;
}
double sec = atof(argv[1]);
pcap_if_t *alldevs;    //网络设备结构
pcap_if_t *d, *head = NULL;
pcap_t *fp;          //网卡描叙符
char errbuf[PCAP_ERRBUF_SIZE];	 //错误信息
unsigned int netmask;               //子网掩码
char packet_filter[] = "ip";            //过滤，选择IP协议
struct bpf_program fcode;
struct pcap_pkthdr *header;
const unsigned char *pkt_data;         //获取网络设备列表
if (pcap_findalldevs(&alldevs, errbuf) == -1)
{
cout << "Error in pcap_findalldevs : " << errbuf;
return;
}
int i = 1;                            //网卡数
if (i == 0)                           //无设备
{
cout << "\nNo interfaces found!Make sure winPacp is installed.\n";
return;
}

if (i >= 1)
{
int j = 0;
for (d = alldevs; d; d = d->next)      //列出网卡列表，让用户进行选择
{
cout << ++j << ":" << d->name;
if (d->description)
cout << " " << d->description << endl;
}
cout << "\nEneter the interface number(1 - " << j << ") :";
int k;
cin >> k;

if (k<1 || k>j)
{
cout << "out of range" << endl;
return;
}
for (d = alldevs, i = 1; i<k; d = d->next, i++);   //找到选择的网卡
head = d;
}

//以混杂模式打开网卡
if ((fp = pcap_open_live(head->name, 1000, 1, 1000, errbuf)) == NULL)
{
cout << "\nUnable to open the adapter." << endl;
pcap_freealldevs(alldevs);
return;
}

//获得子网掩码
if (head->addresses != NULL)
netmask = ((sockaddr_in*)(head->addresses->netmask))->sin_addr.S_un.S_addr;
else
//没有地址假设为C类地址
netmask = 0xffffff;

//编译过滤器
if (pcap_compile(fp, &fcode, packet_filter, i, netmask)<0)
{
cout << "\nUnable to compile the packet filter.Check the syntax.\n";
pcap_freealldevs(alldevs);
return;
}

//设置过滤器
if (pcap_setfilter(fp, &fcode)<0)
{
cout << "\nEeeor setting the filter.\n";
pcap_freealldevs(alldevs);
return;
}

//显示提示信息及每项含义
cout << "\t\tlistening on " << head->description << " " << endl << endl;
ofstream fout(argv[2], ios::app);        //日志记录文件
fout << "\tIP Statistic : (" << sec << "minutes)" << endl;
time_t tmp = time(NULL);
fout << ctime(&tmp);
cout << "IP Statistic : (" << sec << "Seconds)" << endl;
fout << "   Sour IP     " << "\tpacket numbers" << endl;
//释放设备列表
pcap_freealldevs(alldevs);
NodeList link;           //存储数据用链表
int res;
time_t beg;
time_t end;
time(&beg);            //获得当前时间
while ((res = pcap_next_ex(fp, &header, &pkt_data)) >= 0)
{
time(&end);         //获得系统时间
if (end - beg >= sec)  //计算系统时间
break;
if (res == 0)
continue;        //超时
ip_header*ih;
//找到IP头位置
ih = (ip_header*)(pkt_data + 14);   //14为以太头的长度
link.addNode(ih->saddr);       //将源IP地址假如链表
}
cout << "Sour IP   " << '\t' << "packet numbers" << endl;
link.print(cout);          //输出到屏幕
link.print(fout);          //输出到日志文件
fout << endl;
}

要说明的一点是：上面的cpp文件中有一行 fout << ctime(&tmp); 这个是在文件中输出记录时间的 如果要用这个要在 项目属性 -> 配置属性->C/C++ ->命令行中的其它选项中添加/D _CRT_SECURE_NO_WARNINGS 如果有强迫症的同学就直接把这个注释了吧-_-||

最后一步，在项目属性 -> 配置属性->调试中的命令参数中填参数 第一个是时间 单位是秒，第二个是记录文件的名称 例如 "10 1.txt"（PS：去掉引号）

然后运行就行了，第一步是选择网卡，然后静静地等待出结果就行了~