blob: 265ea17b17fad911e65148ed01df4a0420c59f7f (
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#ifndef __TREX_LIBPCAP_H__
#define __TREX_LIBPCAP_H__
/*
Copyright (c) 2015-2015 Cisco Systems, Inc.
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 "captureFile.h"
#include <stdio.h>
typedef struct pcaptime {
uint32_t sec;
uint32_t msec;
} pcaptime_t;
typedef struct packet_file_header
{
uint32_t magic;
uint16_t version_major;
uint16_t version_minor;
uint32_t thiszone;
uint32_t sigfigs;
uint32_t snaplen;
uint32_t linktype;
}packet_file_header_t ;
typedef struct sf_pkthdr {
pcaptime_t ts;
uint32_t caplen;
uint32_t len;
} sf_pkthdr_t;
/**
* Implements the CCAPReaderBase interface.
*
*/
class LibPCapReader : public CCapReaderBase
{
public:
LibPCapReader();
virtual ~LibPCapReader();
/**
* open file for reading.
* (can be called once).
* @param name
*
* @return bool
*/
bool Create(char * name, int loops = 0);
/**
* When called after open will return true only if
* capture file is libpcap format.
*
* @return bool
*/
bool isValid() { return m_is_valid; }
/**
* Fill the structure with the new packet.
* @param lpPacket
*
* @return bool - return true if packet were read and false
* otherwise (reached eof)
*/
virtual bool ReadPacket(CCapPktRaw *lpPacket);
virtual void Rewind();
virtual capture_type_e get_type() {
return LIBPCAP;
}
private:
LibPCapReader(LibPCapReader &);
bool init();
void flip(sf_pkthdr_t * tofilp);
bool m_is_open;
uint64_t m_last_time;
bool m_is_valid;
FILE * m_file_handler;
bool m_is_flip;
};
/**
* Libpcap file format writer.
* Implements CFileWrirerBase interface
*/
class LibPCapWriter: public CFileWriterBase
{
public:
LibPCapWriter();
virtual ~LibPCapWriter();
/**
* Open file for writing. Rewrite from scratch (no append).
* @param name - the file name
*
* @return bool - return true if File was open successfully.
*/
bool Create(char * name);
/**
* Write packet to file (must be called only after successfull
* Create).
*
* @param p - buffer pointer
* @param size - buffer length
*
* @return true on success.
*/
virtual bool write_packet(CCapPktRaw * lpPacket);
/**
*
* returns the count of packets so far written
*
* @return uint32_t
*/
uint32_t get_pkt_count();
/**
* Close file and flush all.
*/
void Close();
/**
* flush all packets to disk
*
* @author imarom (11/24/2016)
*/
void flush_to_disk();
private:
bool init();
FILE * m_file_handler;
uint64_t m_timestamp;
bool m_is_open;
uint32_t m_pkt_count;
};
#endif
|
