#include <cerrno>
#include <cstring>
-#include "common.h"
-#include "cap_nf.h"
-#include "raw_ip_packet.h"
+#include <vector>
-#include "../../../traffcounter.h"
+#include "stg/common.h"
+#include "stg/raw_ip_packet.h"
+#include "stg/traffcounter.h"
+#include "cap_nf.h"
-class CAP_NF_CREATOR
-{
+class CAP_NF_CREATOR {
public:
CAP_NF_CREATOR()
: nf(new NF_CAP())
{
- };
+ }
~CAP_NF_CREATOR()
{
delete nf;
- };
+ }
- NF_CAP * GetCapturer() { return nf; };
+ NF_CAP * GetCapturer() { return nf; }
private:
NF_CAP * nf;
} cnc;
-BASE_PLUGIN * GetPlugin()
+PLUGIN * GetPlugin()
{
return cnc.GetCapturer();
}
int NF_CAP::ParseSettings()
{
-vector<PARAM_VALUE>::iterator it;
+std::vector<PARAM_VALUE>::iterator it;
for (it = settings.moduleParams.begin(); it != settings.moduleParams.end(); ++it)
{
if (it->param == "TCPPort")
continue;
}
-
- // Wrong logic!
- // Need to check actual data length and wait all data to receive
if (res < 24)
{
if (errno != EINTR)
{
NF_DATA * data = reinterpret_cast<NF_DATA *>(buf + 24 + i * 48);
- /*ip.pckt[0] = 4 << 4;
- ip.pckt[0] |= 5;
- ip.pckt[9] = data->proto;
- ip.dataLen = ntohl(data->octets);
- *(uint32_t *)(ip.pckt + 12) = data->srcAddr;
- *(uint32_t *)(ip.pckt + 16) = data->dstAddr;
- *(uint16_t *)(ip.pckt + 20) = data->srcPort;
- *(uint16_t *)(ip.pckt + 22) = data->dstPort;*/
ip.header.ipHeader.ip_v = 4;
ip.header.ipHeader.ip_hl = 5;
ip.header.ipHeader.ip_p = data->proto;