blob: d5dbdb05dd53ad10c4a82dd5e217c5401bb45d0f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
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 _STREAMEXTRACT_H_
#define _STREAMEXTRACT_H_
#include <string>
#include <list>
#include <map>
#include <set>
#include "programconfig.hpp"
#include "bundle.hpp"
#include "pcapreader.hpp"
#include "flowtable.hpp"
#include "pcappkt.hpp"
#include "stream3.hpp"
#include "streamsorter.hpp"
#include "path.hpp"
using namespace std;
class StreamExtract {
public:
/* The size of the flow table determines the number of flows
that can be active at a given time. When a flow expires, it
is written out to a file and the memory is freed. */
StreamExtract(const ProgramConfig &cfg);
int run();
private:
int writeToPcaps(const string &sourceFilePath, const set<uint32_t> &streamIDs);
int writeToLua(const string& binFilePath, const Path &smallFinalBin, const string& luaFilePath, const string& orderedTemp);
int writeFinalBin(const string& sourceFilePath, const string& destFilePath);
string createStreamPcapFileName(int id);
vector<Bundle> createBundles(const string& streamPath);
set<uint32_t> getBundleStreamIDs(const vector<Bundle*>& bundleSamples);
FlowTable<pkt_tuple, Stream3> ft2;
StreamSorter streamSorter;
ProgramConfig cfg;
};
#endif /* _STREAMEXTRACT_H_ */
|
