libpcap报文解析: ipv4、ipv6 @ 2014.7.2

#include <string.h>
#include <stdlib.h>
#include <pcap.h>
#include <stdio.h>
#include <sys/time.h>
#include <unistd.h>
#include <netinet/in.h>
#include <pthread.h>
#include "packet_header.h"
#include <iostream>
#include <string>
using namespace std;

#define MAXBYTE2CAPTURE 2048

pthread_t g_thread[2];
pthread_mutex_t g_mutex;

int isprint(char c)
{
return 0;
}

void print_buf(u_char* pBuf, u_int32 len)
{
if (!pBuf)
{
return;
}

for(int i=0; i<len; i++)
{
printf("%02x ",  (u_char*)pBuf[i]);

if ((i%16 == 0 && i!=0) || i == len-1)
{
printf("\r\n");
}
}
}

void parse_ethII(u_char* pData, u_int32 len)
{
if (!pData || len <14)
{
return;
}

printf("eth II frame: \r\n");
print_buf(pData, 14);

/* parse src mac and dst mac */
EthHeader_t* pEth = (EthHeader_t*)pData;
printf("destination: %02x:%02x:%02x:%02x:%02x:%02x ",
pEth->dest_hwaddr[0],
pEth->dest_hwaddr[1],
pEth->dest_hwaddr[2],
pEth->dest_hwaddr[3],
pEth->dest_hwaddr[4],
pEth->dest_hwaddr[5]);

printf("source : %02x:%02x:%02x:%02x:%02x:%02x",
pEth->source_hwaddr[0],
pEth->source_hwaddr[1],
pEth->source_hwaddr[2],
pEth->source_hwaddr[3],
pEth->source_hwaddr[4],
pEth->source_hwaddr[5]);

/* parse frame type */
printf("\r\nframe type: 0x%x\r\n", ntohs(pEth->frame_type));
}

void parse_ipheader(u_char* pData, u_int32 len)
{
if (!pData || len <14)
{
return;
}

printf("ip header: \r\n");
print_buf(pData, 20);

/* parse ip header */
IPHeader_t* pIpHeader = (IPHeader_t*)pData;
printf("\tversion     : %02x\r\n"
"\ttos         : %02x\r\n"
"\ttotal length: %d(0x%02x)\r\n"
"\tid          : %d(0x%02x)\r\n"
"\tsegment flag: %d(0x%02x)\r\n"
"\tttl         : %02x\r\n"
"\tprotocol    : %02x\r\n"
"\tchecksum    : %d(0x%02x)\r\n"
"\tsrc ip      : %d.%d.%d.%d\r\n"
"\tdst ip      : %d.%d.%d.%d\r\n",
pIpHeader->Ver_HLen,
pIpHeader->TOS,
ntohs(pIpHeader->TotalLen), ntohs(pIpHeader->TotalLen),
ntohs(pIpHeader->ID), ntohs(pIpHeader->ID),
ntohs(pIpHeader->Flag_Segment), ntohs(pIpHeader->Flag_Segment),
pIpHeader->TTL,
pIpHeader->Protocol,
ntohs(pIpHeader->Checksum), ntohs(pIpHeader->Checksum),
pIpHeader->SrcIP[0],pIpHeader->SrcIP[1],pIpHeader->SrcIP[2],pIpHeader->SrcIP[3],
pIpHeader->DstIP[0],pIpHeader->DstIP[1],pIpHeader->DstIP[2],pIpHeader->DstIP[3]);
}

void parse_ip6header(u_char* pData, u_int32 len)
{
if (!pData || len <14)
{
return;
}

printf("ipv6 header: \r\n");
print_buf(pData, 40);

/* parse ipv6 header */
IPv6Header_t* pIpv6Header = (IPv6Header_t*)pData;
printf("\tversion           : %x\r\n"
"\ttraffic class     : %x\r\n"
"\tflow label        : %x\r\n"
"\tpayload length    : %x\r\n"
"\tnext header       : %x\r\n"
"\thop limit         : %x\r\n"
"\tsource            : %x\r\n"
"\tdestination       : %x\r\n",
pIpv6Header->ip6_ctlun.ip6_un2_vfc,
pIpv6Header->ip6_ctlun.ip6_unl.ip6_unl_flow,
pIpv6Header->ip6_ctlun.ip6_unl.ip6_unl_flow,
pIpv6Header->ip6_ctlun.ip6_unl.ip6_unl_plen,
pIpv6Header->ip6_ctlun.ip6_unl.ip6_unl_nxt,
pIpv6Header->ip6_ctlun.ip6_unl.ip6_unl_hlim,
pIpv6Header->ip6_src,
pIpv6Header->ip6_dst);
}

void parse_packet(const u_char* packet, u_int32 len)
{
u_short ftype = 0;

if (!packet)
{
return ;
}

u_char* pMbuf = (u_char*)packet;
parse_ethII(pMbuf, len);

ftype = ntohs(((EthHeader_t*)pMbuf)->frame_type);
switch(ftype)
{
case 0x0800:  /* ipv4 */
pMbuf = (u_char*)packet + 14;
parse_ipheader(pMbuf, len-14);
break;
case 0x86dd: /* ipv6 */
pMbuf = (u_char*)packet + 14;
parse_ip6header(pMbuf, len-14);
break;
default:
printf("frame type : 0x%x\r\n", ftype);
break;
}

printf("\r\n");
}

void processPacket(u_char *arg, const struct pcap_pkthdr *pkthdr, const u_char *packet)
{
int i = 0, *counter = (int *)arg;

printf("--------------------------------------------\r\n");
printf("Packet Count: %d\n", ++(*counter));
printf("Received Packet Size: %d\n", pkthdr->len);
printf("Payload:\n");

#if 1
for (i = 0; i < pkthdr->len; i++)
{
if (isprint(packet[i]))
{
printf("%02d ", packet[i]);
}
else
{
printf("%02x ", packet[i]);
}

if ((i % 16 == 0 && i != 0) || i == pkthdr->len-1)
{
printf("\n");
}

}
#endif

parse_packet(packet, pkthdr->len);

return;
}

void* thread_recv_pkt(void *)
{
while(1)
{
cout << "recv pkt: " << endl;
sleep(1);
}
}

void* thread_send_pkt(void *)
{
while (1)
{
cout << "send pkt: " << endl;
sleep(1);
}
}

int create_pkt_process_task()
{
int ret = 0;

memset(&g_thread, 0, sizeof(g_thread));

ret = pthread_create(&g_thread[0], NULL, thread_send_pkt, NULL);
if (0 == ret)
{
cout << "packet send thread create successfully." << endl;
}
else
{
cout << "packet send thread create failed." << endl;
}

ret = pthread_create(&g_thread[1], NULL, thread_recv_pkt, NULL);
if (0 == ret)
{
cout << "packet send thread create successfully." << endl;
}
else
{
cout << "packet send thread create failed." << endl;
}

return 0;
}

void pkt_process_task_wait()
{
if(g_thread[0] !=0)
{                   //comment4
pthread_join(g_thread[0],NULL);
printf("线程1 已经结束\n");
}

if(g_thread[1] !=0)
{                //comment5
pthread_join(g_thread[1],NULL);
printf("线程2 已经结束\n");
}
}

int main()
{

int i = 0, count = 0;
pcap_t *descr = NULL;
char errbuf[PCAP_ERRBUF_SIZE], *device = NULL;

memset(errbuf, 0, PCAP_ERRBUF_SIZE);

create_pkt_process_task();
pkt_process_task_wait();

/* Get the name of the first device suitable for capture */
device = pcap_lookupdev(errbuf);
if (!device)
{
printf("Open device failed.");
return -1;
}

printf("Opening device %s\n", device);

/* Open device in promiscuous mode */
descr = pcap_open_live(device, MAXBYTE2CAPTURE, 1, 512, errbuf);

/* Loop forever & call processPacket() for every received packet */
pcap_loop(descr, -1, processPacket, (u_char *)&count);

return 0;
}