first version

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