VNFs/DPPD-PROX/tools/flow_extract/stream2.hpp

   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 #ifndef _STREAM2_H_
  18 #define _STREAM2_H_
  19
  20 #include <inttypes.h>
  21 #include <fstream>
  22
  23 #include "netsocket.hpp"
  24
  25 using namespace std;
  26
  27 class Stream2 {
  28 public:
  29         Stream2() : clientServerHdrContent(NULL) {}
  30         ~Stream2() {delete [] clientServerHdrContent;}
  31         int fromFile(ifstream *f);
  32         void calcOffsets(ofstream *out);
  33         void toFile(ofstream *out) const;
  34         void toLua(ofstream *f, const string& binFileName, const string& streamTableName) const;
  35         NetSocket getServerNetSocket() const;
  36         NetSocket getClientNetSocket() const;
  37         void setServerNetSocket(const NetSocket& netSocket);
  38         void setClientNetSocket(const NetSocket& netSocket);
  39         Stream::Header      streamHdr;
  40 private:
  41         int actionsFromFile(ifstream *f, size_t actionCount);
  42         int setReferences(ifstream *f);
  43
  44         uint8_t *clientServerHdrContent;
  45
  46         uint32_t clientHdrBeg;
  47         uint32_t serverHdrBeg;
  48         uint32_t clientContentBeg;
  49         uint32_t serverContentBeg;
  50
  51         vector<Stream::ActionEntry> m_actions;
  52 };
  53
  54 #endif /* _STREAM2_H_ */