summaryrefslogtreecommitdiffstats
path: root/src/common/pcap.h
diff options
context:
space:
mode:
authorHanoh Haim <hhaim@cisco.com>2015-06-24 14:03:29 +0300
committerHanoh Haim <hhaim@cisco.com>2015-06-24 14:03:29 +0300
commit8b52a31ed2c299b759f330c4f976b9c70f5765f4 (patch)
tree9d6da5438b5b56b1d2d57e6c13494b4e65d000e7 /src/common/pcap.h
first version
Diffstat (limited to 'src/common/pcap.h')
-rwxr-xr-xsrc/common/pcap.h150
1 files changed, 150 insertions, 0 deletions
diff --git a/src/common/pcap.h b/src/common/pcap.h
new file mode 100755
index 00000000..fb266e31
--- /dev/null
+++ b/src/common/pcap.h
@@ -0,0 +1,150 @@
+#ifndef __LIBPCAP_H__
+#define __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();
+
+
+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();
+
+private:
+
+ bool init();
+ FILE * m_file_handler;
+ uint64_t m_timestamp;
+ bool m_is_open;
+ uint32_t m_pkt_count;
+};
+#endif