1 files changed, 46 insertions, 0 deletions
diff --git a/VNFs/DPPD-PROX/tools/flow_extract/pcapwriter.cpp b/VNFs/DPPD-PROX/tools/flow_extract/pcapwriter.cpp
new file mode 100644
index 00000000..4c7c4cea
--- /dev/null
+++ b/VNFs/DPPD-PROX/tools/flow_extract/pcapwriter.cpp
@@ -0,0 +1,46 @@
+/*
+// Copyright (c) 2010-2017 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+*/
+
+#include "pcapwriter.hpp"
+
+int PcapWriter::open(const string& file_path)
+{
+	m_handle = pcap_open_dead_with_tstamp_precision(DLT_EN10MB, 65536, PCAP_TSTAMP_PRECISION_NANO);
+	if (m_handle == NULL)
+		return -1;
+
+	m_pcap_dumper = pcap_dump_open(m_handle, file_path.c_str());
+	if (m_pcap_dumper == NULL) {
+		pcap_close(m_handle);
+		return -1;
+	}
+
+	return 0;
+}
+
+int PcapWriter::write(const PcapPkt& pkt)
+{
+	pcap_dump((unsigned char *)m_pcap_dumper, &pkt.hdr(), pkt.payload());
+	return 0;
+}
+
+void PcapWriter::close()
+{
+	if (m_pcap_dumper)
+		pcap_dump_close(m_pcap_dumper);
+	if (m_handle)
+		pcap_close(m_handle);
+}