104 lines
3.3 KiB
C
104 lines
3.3 KiB
C
/*
|
|
* Copyright (C) 2017 THL A29 Limited, a Tencent company.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice, this
|
|
* list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
|
* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*
|
|
*/
|
|
|
|
#include <sys/time.h>
|
|
#include <unistd.h>
|
|
|
|
#include "ff_dpdk_pcap.h"
|
|
|
|
struct pcap_file_header {
|
|
uint32_t magic;
|
|
u_short version_major;
|
|
u_short version_minor;
|
|
int32_t thiszone; /* gmt to local correction */
|
|
uint32_t sigfigs; /* accuracy of timestamps */
|
|
uint32_t snaplen; /* max length saved portion of each pkt */
|
|
uint32_t linktype; /* data link type (LINKTYPE_*) */
|
|
};
|
|
|
|
struct pcap_pkthdr {
|
|
uint32_t sec; /* time stamp */
|
|
uint32_t usec; /* struct timeval time_t, in linux64: 8*2=16, in cap: 4 */
|
|
uint32_t caplen; /* length of portion present */
|
|
uint32_t len; /* length this packet (off wire) */
|
|
};
|
|
|
|
int
|
|
ff_enable_pcap(const char* dump_path)
|
|
{
|
|
FILE* fp = fopen(dump_path, "w");
|
|
if (fp == NULL) {
|
|
rte_exit(EXIT_FAILURE, "Cannot open pcap dump path: %s\n", dump_path);
|
|
return -1;
|
|
}
|
|
|
|
struct pcap_file_header pcap_file_hdr;
|
|
void* file_hdr = &pcap_file_hdr;
|
|
|
|
pcap_file_hdr.magic = 0xA1B2C3D4;
|
|
pcap_file_hdr.version_major = 0x0002;
|
|
pcap_file_hdr.version_minor = 0x0004;
|
|
pcap_file_hdr.thiszone = 0x00000000;
|
|
pcap_file_hdr.sigfigs = 0x00000000;
|
|
pcap_file_hdr.snaplen = 0x0000FFFF; //65535
|
|
pcap_file_hdr.linktype = 0x00000001; //DLT_EN10MB, Ethernet (10Mb)
|
|
|
|
fwrite(file_hdr, sizeof(struct pcap_file_header), 1, fp);
|
|
fclose(fp);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
ff_dump_packets(const char* dump_path, struct rte_mbuf* pkt)
|
|
{
|
|
FILE* fp = fopen(dump_path, "a");
|
|
if (fp == NULL) {
|
|
return -1;
|
|
}
|
|
|
|
struct pcap_pkthdr pcap_hdr;
|
|
void* hdr = &pcap_hdr;
|
|
|
|
struct timeval ts;
|
|
gettimeofday(&ts, NULL);
|
|
pcap_hdr.sec = ts.tv_sec;
|
|
pcap_hdr.usec = ts.tv_usec;
|
|
pcap_hdr.caplen = pkt->pkt_len;
|
|
pcap_hdr.len = pkt->pkt_len;
|
|
fwrite(hdr, sizeof(struct pcap_pkthdr), 1, fp);
|
|
|
|
while(pkt != NULL) {
|
|
fwrite(rte_pktmbuf_mtod(pkt, char*), pkt->data_len, 1, fp);
|
|
pkt = pkt->next;
|
|
}
|
|
|
|
fclose(fp);
|
|
|
|
return 0;
|
|
}
|
|
|