VNFs/DPPD-PROX/tools/flow_extract/pcapwriter.cpp

   1 /*
   2 // Copyright (c) 2010-2017 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //     http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 */
  16
  17 #include "pcapwriter.hpp"
  18
  19 int PcapWriter::open(const string& file_path)
  20 {
  21         m_handle = pcap_open_dead_with_tstamp_precision(DLT_EN10MB, 65536, PCAP_TSTAMP_PRECISION_NANO);
  22         if (m_handle == NULL)
  23                 return -1;
  24
  25         m_pcap_dumper = pcap_dump_open(m_handle, file_path.c_str());
  26         if (m_pcap_dumper == NULL) {
  27                 pcap_close(m_handle);
  28                 return -1;
  29         }
  30
  31         return 0;
  32 }
  33
  34 int PcapWriter::write(const PcapPkt& pkt)
  35 {
  36         pcap_dump((unsigned char *)m_pcap_dumper, &pkt.hdr(), pkt.payload());
  37         return 0;
  38 }
  39
  40 void PcapWriter::close()
  41 {
  42         if (m_pcap_dumper)
  43                 pcap_dump_close(m_pcap_dumper);
  44         if (m_handle)
  45                 pcap_close(m_handle);
  46 }