#include "ipq_cap.h"
#include "raw_ip_packet.h"
+#include "traffcounter.h"
extern "C"
- {
- #include "libipq.h"
- }
-
-class IPQ_CAP_CREATOR
{
+#include "libipq.h"
+}
+
+class IPQ_CAP_CREATOR {
private:
IPQ_CAP * ic;
IPQ_CAP_CREATOR()
: ic(new IPQ_CAP())
{
- };
+ }
~IPQ_CAP_CREATOR()
{
delete ic;
- };
+ }
IPQ_CAP * GetCapturer()
{
return ic;
- };
+ }
};
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
-BASE_PLUGIN * GetPlugin()
+PLUGIN * GetPlugin()
{
return icc.GetCapturer();
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
-const string IPQ_CAP::GetVersion() const
+const std::string IPQ_CAP::GetVersion() const
{
return "ipq_cap v.1.2";
}
memset(buf, 0, BUFSIZE);
}
//-----------------------------------------------------------------------------
-void IPQ_CAP::SetTraffcounter(TRAFFCOUNTER * tc)
-{
-traffCnt = tc;
-}
-//-----------------------------------------------------------------------------
-const string & IPQ_CAP::GetStrError() const
-{
-return errorStr;
-}
-//-----------------------------------------------------------------------------
int IPQ_CAP::Start()
{
if (isRunning)
return 0;
}
//-----------------------------------------------------------------------------
-bool IPQ_CAP::IsRunning()
-{
-return isRunning;
-}
-//-----------------------------------------------------------------------------
void * IPQ_CAP::Run(void * d)
{
RAW_PACKET raw_packet;
-int status;
IPQ_CAP * dc = (IPQ_CAP *)d;
dc->isRunning = true;
raw_packet.dataLen = -1;
while (dc->nonstop)
{
- status = dc->IPQCapRead(&raw_packet, 68);
+ int status = dc->IPQCapRead(&raw_packet, 68);
if (status == -1 ||
status == -2 ||
status == -3 ||
return NULL;
}
//-----------------------------------------------------------------------------
-uint16_t IPQ_CAP::GetStartPosition() const
-{
-return 0;
-}
-//-----------------------------------------------------------------------------
-uint16_t IPQ_CAP::GetStopPosition() const
-{
-return 0;
-}
-//-----------------------------------------------------------------------------
int IPQ_CAP::IPQCapOpen()
{
-int status;
-
ipq_h = ipq_create_handle(0, PF_INET);
if (ipq_h == NULL)
{
errorStr = "Cannot create ipq handle!";
return -1;
}
-status = ipq_set_mode(ipq_h, IPQ_COPY_PACKET, PAYLOAD_LEN);
+int status = ipq_set_mode(ipq_h, IPQ_COPY_PACKET, PAYLOAD_LEN);
if (status < 0)
{
ipq_destroy_handle(ipq_h);
//-----------------------------------------------------------------------------
int IPQ_CAP::IPQCapRead(void * buffer, int blen)
{
-int status;
-static ipq_packet_msg_t *m;
-
memset(buf, 0, BUFSIZE);
-status = ipq_read(ipq_h, buf, BUFSIZE, 1);
+int status = ipq_read(ipq_h, buf, BUFSIZE, 1);
if (status == 0)
return -4;
if (errno == EINTR)
return -1;
if (ipq_message_type(buf) != IPQM_PACKET)
return -2;
-m = ipq_get_packet(buf);
+static ipq_packet_msg_t * m = ipq_get_packet(buf);
memcpy(buffer, m->payload, blen);
ipq_set_verdict(ipq_h, m->packet_id, NF_ACCEPT, 0, NULL);
return 0;