-
Notifications
You must be signed in to change notification settings - Fork 1
/
PcapFileDevice.h
480 lines (395 loc) · 19.1 KB
/
PcapFileDevice.h
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
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
#ifndef PCAPPP_FILE_DEVICE
#define PCAPPP_FILE_DEVICE
#include <PcapDevice.h>
#include <RawPacket.h>
/// @file
/**
* \namespace pcpp
* \brief The main namespace for the PcapPlusPlus lib
*/
namespace pcpp
{
/**
* @class IFileDevice
* An abstract class (cannot be instantiated, has a private c'tor) which is the parent class for all file devices
*/
class IFileDevice : public IPcapDevice
{
protected:
char* m_FileName;
IFileDevice(const char* fileName);
virtual ~IFileDevice();
public:
/**
* @return The name of the file
*/
std::string getFileName();
//override methods
/**
* Close the file
*/
virtual void close();
};
/**
* @class IFileReaderDevice
* An abstract class (cannot be instantiated, has a private c'tor) which is the parent class for file reader devices
*/
class IFileReaderDevice : public IFileDevice
{
protected:
uint32_t m_NumOfPacketsRead;
uint32_t m_NumOfPacketsNotParsed;
/**
* A constructor for this class that gets the pcap full path file name to open. Notice that after calling this constructor the file
* isn't opened yet, so reading packets will fail. For opening the file call open()
* @param[in] fileName The full path of the file to read
*/
IFileReaderDevice(const char* fileName);
public:
/**
* A destructor for this class
*/
virtual ~IFileReaderDevice() {}
/**
* @return The file size in bytes
*/
uint64_t getFileSize();
virtual bool getNextPacket(RawPacket& rawPacket) = 0;
/**
* A static method that creates an instance of the reader best fit to read the file. It decides by the file extension: for .pcapng
* files it returns an instance of PcapNgFileReaderDevice and for all other extensions it returns an instance of PcapFileReaderDevice
* @param[in] fileName The file name to open
* @return An instance of the reader to read the file. Notice you should free this instance when done using it
*/
static IFileReaderDevice* getReader(const char* fileName);
};
/**
* @class PcapFileReaderDevice
* A class for opening a pcap file in read-only mode. This class enable to open the file and read all packets, packet-by-packet
*/
class PcapFileReaderDevice : public IFileReaderDevice
{
private:
LinkLayerType m_PcapLinkLayerType;
// private copy c'tor
PcapFileReaderDevice(const PcapFileReaderDevice& other);
PcapFileReaderDevice& operator=(const PcapFileReaderDevice& other);
public:
/**
* A constructor for this class that gets the pcap full path file name to open. Notice that after calling this constructor the file
* isn't opened yet, so reading packets will fail. For opening the file call open()
* @param[in] fileName The full path of the file to read
*/
PcapFileReaderDevice(const char* fileName);
/**
* A destructor for this class
*/
virtual ~PcapFileReaderDevice() {}
/**
* @return The link layer type of this file
*/
LinkLayerType getLinkLayerType();
//overridden methods
/**
* Read the next packet from the file. Before using this method please verify the file is opened using open()
* @param[out] rawPacket A reference for an empty RawPacket where the packet will be written
* @return True if a packet was read successfully. False will be returned if the file isn't opened (also, an error log will be printed)
* or if reached end-of-file
*/
bool getNextPacket(RawPacket& rawPacket);
/**
* Open the file name which path was specified in the constructor in a read-only mode
* @return True if file was opened successfully or if file is already opened. False if opening the file failed for some reason (for example:
* file path does not exist)
*/
bool open();
/**
* Get statistics of packets read so far. In the pcap_stat struct, only ps_recv member is relevant. The rest of the members will contain 0
* @param[out] stats The stats struct where stats are returned
*/
void getStatistics(pcap_stat& stats);
};
/**
* @class PcapNgFileReaderDevice
* A class for opening a pcap-ng file in read-only mode. This class enable to open the file and read all packets, packet-by-packet
*/
class PcapNgFileReaderDevice : public IFileReaderDevice
{
private:
void* m_LightPcapNg;
struct bpf_program m_Bpf;
bool m_BpfInitialized;
int m_BpfLinkType;
std::string m_CurFilter;
// private copy c'tor
PcapNgFileReaderDevice(const PcapNgFileReaderDevice& other);
PcapNgFileReaderDevice& operator=(const PcapNgFileReaderDevice& other);
bool matchPacketWithFilter(const uint8_t* packetData, size_t packetLen, timeval packetTimestamp, uint16_t linkType);
public:
/**
* A constructor for this class that gets the pcap-ng full path file name to open. Notice that after calling this constructor the file
* isn't opened yet, so reading packets will fail. For opening the file call open()
* @param[in] fileName The full path of the file to read
*/
PcapNgFileReaderDevice(const char* fileName);
/**
* A destructor for this class
*/
virtual ~PcapNgFileReaderDevice() { close(); }
/**
* The pcap-ng format allows storing metadata at the header of the file. Part of this metadata is a string specifying the
* operating system that was used for capturing the packets. This method reads this string from the metadata (if exists) and
* returns it
* @return The operating system string if exists, or an empty string otherwise
*/
std::string getOS();
/**
* The pcap-ng format allows storing metadata at the header of the file. Part of this metadata is a string specifying the
* hardware that was used for capturing the packets. This method reads this string from the metadata (if exists) and
* returns it
* @return The hardware string if exists, or an empty string otherwise
*/
std::string getHardware();
/**
* The pcap-ng format allows storing metadata at the header of the file. Part of this metadata is a string specifying the
* capture application that was used for capturing the packets. This method reads this string from the metadata (if exists) and
* returns it
* @return The capture application string if exists, or an empty string otherwise
*/
std::string getCaptureApplication();
/**
* The pcap-ng format allows storing metadata at the header of the file. Part of this metadata is a string containing a user-defined
* comment (can be any string). This method reads this string from the metadata (if exists) and
* returns it
* @return The comment written inside the file if exists, or an empty string otherwise
*/
std::string getCaptureFileComment();
/**
* The pcap-ng format allows storing a user-defined comment for every packet (besides the comment per-file). This method reads
* the next packet and the comment attached to it (if such comment exists), and returns them both
* @param[out] rawPacket A reference for an empty RawPacket where the packet will be written
* @param[out] packetComment The comment attached to the packet or an empty string if no comment exists
* @return True if a packet was read successfully. False will be returned if the file isn't opened (also, an error log will be printed)
* or if reached end-of-file
*/
bool getNextPacket(RawPacket& rawPacket, std::string& packetComment);
//overridden methods
/**
* Read the next packet from the file. Before using this method please verify the file is opened using open()
* @param[out] rawPacket A reference for an empty RawPacket where the packet will be written
* @return True if a packet was read successfully. False will be returned if the file isn't opened (also, an error log will be printed)
* or if reached end-of-file
*/
bool getNextPacket(RawPacket& rawPacket);
/**
* Open the file name which path was specified in the constructor in a read-only mode
* @return True if file was opened successfully or if file is already opened. False if opening the file failed for some reason (for example:
* file path does not exist)
*/
bool open();
/**
* Get statistics of packets read so far. In the pcap_stat struct, only ps_recv member is relevant. The rest of the members will contain 0
* @param[out] stats The stats struct where stats are returned
*/
void getStatistics(pcap_stat& stats);
/**
* Set a filter for PcapNG reader device. Only packets that match the filter will be received
* @param[in] filterAsString The filter to be set in Berkeley Packet Filter (BPF) syntax (http://biot.com/capstats/bpf.html)
* @return True if filter set successfully, false otherwise
*/
bool setFilter(std::string filterAsString);
/**
* Close the pacp-ng file
*/
void close();
};
/**
* @class IFileWriterDevice
* An abstract class (cannot be instantiated, has a private c'tor) which is the parent class for file writer devices
*/
class IFileWriterDevice : public IFileDevice
{
protected:
uint32_t m_NumOfPacketsWritten;
uint32_t m_NumOfPacketsNotWritten;
IFileWriterDevice(const char* fileName);
public:
/**
* A destructor for this class
*/
virtual ~IFileWriterDevice() {}
virtual bool writePacket(RawPacket const& packet) = 0;
virtual bool writePackets(const RawPacketVector& packets) = 0;
using IFileDevice::open;
virtual bool open(bool appendMode) = 0;
};
/**
* @class PcapFileWriterDevice
* A class for opening a pcap file for writing or create a new pcap file and write packets to it. This class adds
* a unique capability that isn't supported in WinPcap and in older libpcap versions which is to open a pcap file
* in append mode where packets are written at the end of the pcap file instead of running it over
*/
class PcapFileWriterDevice : public IFileWriterDevice
{
private:
pcap_dumper_t* m_PcapDumpHandler;
LinkLayerType m_PcapLinkLayerType;
bool m_AppendMode;
FILE* m_File;
// private copy c'tor
PcapFileWriterDevice(const PcapFileWriterDevice& other);
PcapFileWriterDevice& operator=(const PcapFileWriterDevice& other);
void closeFile();
public:
/**
* A constructor for this class that gets the pcap full path file name to open for writing or create. Notice that after calling this
* constructor the file isn't opened yet, so writing packets will fail. For opening the file call open()
* @param[in] fileName The full path of the file
* @param[in] linkLayerType The link layer type all packet in this file will be based on. The default is Ethernet
*/
PcapFileWriterDevice(const char* fileName, LinkLayerType linkLayerType = LINKTYPE_ETHERNET);
/**
* A destructor for this class
*/
~PcapFileWriterDevice();
/**
* Write a RawPacket to the file. Before using this method please verify the file is opened using open(). This method won't change the
* written packet
* @param[in] packet A reference for an existing RawPcket to write to the file
* @return True if a packet was written successfully. False will be returned if the file isn't opened
* or if the packet link layer type is different than the one defined for the file
* (in all cases, an error will be printed to log)
*/
bool writePacket(RawPacket const& packet);
/**
* Write multiple RawPacket to the file. Before using this method please verify the file is opened using open(). This method won't change
* the written packets or the RawPacketVector instance
* @param[in] packets A reference for an existing RawPcketVector, all of its packets will be written to the file
* @return True if all packets were written successfully to the file. False will be returned if the file isn't opened (also, an error
* log will be printed) or if at least one of the packets wasn't written successfully to the file
*/
bool writePackets(const RawPacketVector& packets);
//override methods
/**
* Open the file in a write mode. If file doesn't exist, it will be created. If it does exist it will be
* overwritten, meaning all its current content will be deleted
* @return True if file was opened/created successfully or if file is already opened. False if opening the file failed for some reason
* (an error will be printed to log)
*/
virtual bool open();
/**
* Same as open(), but enables to open the file in append mode in which packets will be appended to the file
* instead of overwrite its current content. In append mode file must exist, otherwise opening will fail
* @param[in] appendMode A boolean indicating whether to open the file in append mode or not. If set to false
* this method will act exactly like open(). If set to true, file will be opened in append mode
* @return True of managed to open the file successfully. In case appendMode is set to true, false will be returned
* if file wasn't found or couldn't be read, if file type is not pcap, or if link type specified in c'tor is
* different from current file link type. In case appendMode is set to false, please refer to open() for return
* values
*/
bool open(bool appendMode);
/**
* Flush and close the pacp file
*/
virtual void close();
/**
* Get statistics of packets written so far. In the pcap_stat struct, only ps_recv member is relevant. The rest of the members will contain 0
* @param[out] stats The stats struct where stats are returned
*/
virtual void getStatistics(pcap_stat& stats);
};
/**
* @class PcapNgFileWriterDevice
* A class for opening a pcap-ng file for writing or creating a new pcap-ng file and write packets to it. This class adds
* unique capabilities such as writing metadata attributes into the file header, adding comments per packet and opening
* the file in append mode where packets are added to a file instead of overriding it. This capabilities are part of the
* pcap-ng standard but aren't supported in most tools and libraries
*/
class PcapNgFileWriterDevice : public IFileWriterDevice
{
private:
void* m_LightPcapNg;
// private copy c'tor
PcapNgFileWriterDevice(const PcapFileWriterDevice& other);
PcapNgFileWriterDevice& operator=(const PcapNgFileWriterDevice& other);
public:
/**
* A constructor for this class that gets the pcap-ng full path file name to open for writing or create. Notice that after calling this
* constructor the file isn't opened yet, so writing packets will fail. For opening the file call open()
* @param[in] fileName The full path of the file
*/
PcapNgFileWriterDevice(const char* fileName);
/**
* A destructor for this class
*/
virtual ~PcapNgFileWriterDevice() { close(); }
/**
* Open the file in a write mode. If file doesn't exist, it will be created. If it does exist it will be
* overwritten, meaning all its current content will be deleted. As opposed to open(), this method also allows writing several
* metadata attributes that will be stored in the header of the file
* @param[in] os A string describing the operating system that was used to capture the packets. If this string is empty or null it
* will be ignored
* @param[in] hardware A string describing the hardware that was used to capture the packets. If this string is empty or null it
* will be ignored
* @param[in] captureApp A string describing the application that was used to capture the packets. If this string is empty or null it
* will be ignored
* @param[in] fileComment A string containing a user-defined comment that will be part of the metadata of the file.
* If this string is empty or null it will be ignored
* @return True if file was opened/created successfully or if file is already opened. False if opening the file failed for some reason
* (an error will be printed to log)
*/
bool open(const char* os, const char* hardware, const char* captureApp, const char* fileComment);
/**
* The pcap-ng format allows adding a user-defined comment for each stored packet. This method writes a RawPacket to the file and
* adds a comment to it. Before using this method please verify the file is opened using open(). This method won't change the
* written packet or the input comment
* @param[in] packet A reference for an existing RawPcket to write to the file
* @param[in] comment The comment to be written for the packet. If this string is empty or null it will be ignored
* @return True if a packet was written successfully. False will be returned if the file isn't opened (an error will be printed to log)
*/
bool writePacket(RawPacket const& packet, const char* comment);
//overridden methods
/**
* Write a RawPacket to the file. Before using this method please verify the file is opened using open(). This method won't change the
* written packet
* @param[in] packet A reference for an existing RawPcket to write to the file
* @return True if a packet was written successfully. False will be returned if the file isn't opened (an error will be printed to log)
*/
bool writePacket(RawPacket const& packet);
/**
* Write multiple RawPacket to the file. Before using this method please verify the file is opened using open(). This method won't change
* the written packets or the RawPacketVector instance
* @param[in] packets A reference for an existing RawPcketVector, all of its packets will be written to the file
* @return True if all packets were written successfully to the file. False will be returned if the file isn't opened (also, an error
* log will be printed) or if at least one of the packets wasn't written successfully to the file
*/
bool writePackets(const RawPacketVector& packets);
/**
* Open the file in a write mode. If file doesn't exist, it will be created. If it does exist it will be
* overwritten, meaning all its current content will be deleted
* @return True if file was opened/created successfully or if file is already opened. False if opening the file failed for some reason
* (an error will be printed to log)
*/
bool open();
/**
* Same as open(), but enables to open the file in append mode in which packets will be appended to the file
* instead of overwrite its current content. In append mode file must exist, otherwise opening will fail
* @param[in] appendMode A boolean indicating whether to open the file in append mode or not. If set to false
* this method will act exactly like open(). If set to true, file will be opened in append mode
* @return True of managed to open the file successfully. In case appendMode is set to true, false will be returned
* if file wasn't found or couldn't be read, if file type is not pcap-ng. In case appendMode is set to false, please refer to open()
* for return values
*/
bool open(bool appendMode);
/**
* Flush and close the pacp-ng file
*/
void close();
/**
* Get statistics of packets written so far. In the pcap_stat struct, only ps_recv member is relevant. The rest of the members will contain 0
* @param[out] stats The stats struct where stats are returned
*/
void getStatistics(pcap_stat& stats);
};
}// namespace pcpp
#endif