1 files changed, 55 insertions, 0 deletions
diff --git a/VNFs/DPPD-PROX/tools/flow_extract/stream3.hpp b/VNFs/DPPD-PROX/tools/flow_extract/stream3.hpp
new file mode 100644
index 00000000..7e94814e
--- /dev/null
+++ b/VNFs/DPPD-PROX/tools/flow_extract/stream3.hpp
@@ -0,0 +1,55 @@
+/*
+// 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.
+*/
+
+#ifndef _STREAM3_H_
+#define _STREAM3_H_
+
+#include <inttypes.h>
+#include <vector>
+
+#include "pcappkt.hpp"
+#include "timestamp.hpp"
+
+using namespace std;
+class Allocator;
+
+class Stream3 {
+public:
+	PcapPkt::L4Proto getProto(void) const {return m_proto;}
+	Stream3(uint32_t id, PcapPkt::L4Proto proto);
+	Stream3() : m_id(UINT32_MAX), m_proto(PcapPkt::PROTO_UDP), m_pktCount(0), m_flushCount(0) {}
+	void addPkt(const PcapPkt& pkt);
+	void flush(ofstream *outputFile);
+	void addFromMemory(uint8_t *mem, size_t *len);
+	static uint32_t getIDFromMem(uint8_t *mem);
+	bool hasFlushablePackets() const {return !!m_flushCount;}
+	Timestamp getTimeout() const;
+	uint32_t getID() const {return m_id;}
+	void removeAllPackets();
+	void setID(const uint32_t id) {m_id = id;}
+private:
+	void writeHeader(ofstream *outputFile) const;
+	void writePackets(ofstream *outputFile) const;
+	void clearPackets();
+
+	uint32_t m_id;
+	PcapPkt::L4Proto m_proto;
+	vector<PcapPkt *> m_pkts;
+	uint32_t m_pktCount;
+	uint32_t m_flushCount;
+};
+
+#endif /* _STREAM3_H_ */