2 // Copyright (c) 2010-2017 Intel Corporation
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
8 // http://www.apache.org/licenses/LICENSE-2.0
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.
17 #ifndef _STREAMSORTER_H_
18 #define _STREAMSORTER_H_
20 #include "stream3.hpp"
21 #include "flowtable.hpp"
22 #include "allocator.hpp"
26 StreamSorter(size_t flowTableSize, const string& workingDirectory, size_t memoryLimit);
27 void sort(const string &inputPcapFile, const string &outputBinFile);
29 void sortChunks(const string &inputPcapFilePath);
30 void mergeChunks(const string &outputBinFilePath);
31 void setTempFileName();
32 void processPkt(const PcapPkt &pkt);
34 FlowTable<pkt_tuple, uint32_t>::entry* getFlowEntry(const PcapPkt &pkt);
35 void flushStreams(ofstream *outputTempFile);
36 Stream3 *addNewStream(PcapPkt::L4Proto proto);
38 FlowTable<pkt_tuple, uint32_t> *ft;
39 vector<size_t> flushOffsets;
40 vector<Stream3> streams;
42 const string workingDirectory;
47 #endif /* _STREAMSORTER_H_ */