Add packet injector test frame

src/packet_io/dumpfile_io.cpp

@@ -26,6 +26,7 @@ struct dumpfile_io
     struct io_stat stat[MAX_THREAD_NUM];
     uint64_t io_thread_need_exit;
     uint64_t io_thread_is_runing;
+    uint64_t io_thread_wait_exit;
 };
 
 struct pcap_pkt
@@ -34,10 +35,90 @@ struct pcap_pkt
     int len;
 };
 
+struct pcap_file_hdr
+{
+    unsigned int magic;
+    unsigned short version_major;
+    unsigned short version_minor;
+    unsigned int thiszone; // gmt to local correction
+    unsigned int sigfigs;  // accuracy of timestamps
+    unsigned int snaplen;  // max length saved portion of each pkt
+    unsigned int linktype; // data link type (LINKTYPE_*)
+};
+
+struct pcap_pkt_hdr
+{
+    unsigned int tv_sec;  // time stamp
+    unsigned int tv_usec; // time stamp
+    unsigned int caplen;  // length of portion present
+    unsigned int len;     // length this packet (off wire)
+};
+
+struct pcap_file_hdr DEFAULT_PCAP_FILE_HDR =
+    {
+        .magic = 0xA1B2C3D4,
+        .version_major = 0x0002,
+        .version_minor = 0x0004,
+        .thiszone = 0,
+        .sigfigs = 0,
+        .snaplen = 0xFFFF,
+        .linktype = 1};
+
 /******************************************************************************
  * Private API
  ******************************************************************************/
 
+static void save_packet(struct packet *pkt, uint64_t idx)
+{
+    int len = 0;
+    FILE *fp = NULL;
+    struct tuple6 tuple;
+    struct timeval ts = {0};
+    struct pcap_pkt_hdr pcap_hdr = {0};
+
+    char file[256] = {0};
+    char src_addr[INET6_ADDRSTRLEN] = {0};
+    char dst_addr[INET6_ADDRSTRLEN] = {0};
+
+    len = packet_get_len(pkt);
+    memset(&tuple, 0, sizeof(struct tuple6));
+    packet_get_innermost_tuple6(pkt, &tuple);
+
+    if (tuple.ip_type == IP_TYPE_V4)
+    {
+        inet_ntop(AF_INET, &tuple.src_addr.v4, src_addr, INET6_ADDRSTRLEN);
+        inet_ntop(AF_INET, &tuple.dst_addr.v4, dst_addr, INET6_ADDRSTRLEN);
+    }
+    else
+    {
+        inet_ntop(AF_INET6, &tuple.src_addr.v6, src_addr, INET6_ADDRSTRLEN);
+        inet_ntop(AF_INET6, &tuple.dst_addr.v6, dst_addr, INET6_ADDRSTRLEN);
+    }
+    snprintf(file, sizeof(file), "/tmp/inject-%s:%u-%s:%u-%lu.pcap", src_addr, ntohs(tuple.src_port), dst_addr, ntohs(tuple.dst_port), idx);
+
+    fp = fopen(file, "w+");
+    if (fp)
+    {
+        gettimeofday(&ts, NULL);
+        pcap_hdr.tv_sec = ts.tv_sec;
+        pcap_hdr.tv_usec = ts.tv_usec;
+
+        pcap_hdr.caplen = len;
+        pcap_hdr.len = len;
+
+        fwrite(&DEFAULT_PCAP_FILE_HDR, sizeof(DEFAULT_PCAP_FILE_HDR), 1, fp);
+        fwrite(&pcap_hdr, sizeof(struct pcap_pkt_hdr), 1, fp);
+        fwrite(packet_get_data(pkt), 1, len, fp);
+        fflush(fp);
+        fclose(fp);
+        PACKET_IO_LOG_DEBUG("save packet to %s", file);
+    }
+    else
+    {
+        PACKET_IO_LOG_ERROR("unable to write pcap file: %s, %s", file, strerror(errno));
+    }
+}
+
 typedef int file_handle(const char *file, void *arg);
 
 static int scan_directory(const char *dir, file_handle *handler, void *arg)
@@ -165,6 +246,18 @@ static int dumpfile_handler(const char *file, void *arg)
     return 0;
 }
 
+static int all_packet_processed(struct dumpfile_io *handle)
+{
+    for (uint16_t i = 0; i < handle->nr_threads; i++)
+    {
+        if (!lock_free_queue_empty(handle->queue[i]))
+        {
+            return 0;
+        }
+    }
+    return 1;
+}
+
 static void *dumpfile_thread(void *arg)
 {
     struct dumpfile_io *handle = (struct dumpfile_io *)arg;
@@ -176,6 +269,11 @@ static void *dumpfile_thread(void *arg)
 
     while (ATOMIC_READ(&handle->io_thread_need_exit) == 0)
     {
+        if (all_packet_processed(handle))
+        {
+            ATOMIC_SET(&handle->io_thread_wait_exit, 1);
+        }
+
         PACKET_IO_LOG_STATE("dumpfile io thread waiting");
         sleep(1);
     }
@@ -259,6 +357,11 @@ void dumpfile_io_free(struct dumpfile_io *handle)
     }
 }
 
+int dumpfile_io_wait_exit(struct dumpfile_io *handle)
+{
+    return ATOMIC_READ(&handle->io_thread_wait_exit);
+}
+
 int dumpfile_io_init(struct dumpfile_io *handle, uint16_t thr_idx)
 {
     return 0;
@@ -365,6 +468,8 @@ int dumpfile_io_inject(struct dumpfile_io *handle, uint16_t thr_idx, struct pack
         stat->dev_tx_pkts++;
         stat->dev_tx_bytes += len;
 
+        save_packet(pkt, stat->inject_pkts);
+
         packet_free(pkt);
     }