From: Maxim Mamontov Date: Sun, 8 Jun 2014 10:33:42 +0000 (+0300) Subject: Implemented NFQ capturing. X-Git-Url: https://git.stg.codes/stg.git/commitdiff_plain/9447011b1029f3fcd82f05632673d3a85a4ae5d4 Implemented NFQ capturing. --- diff --git a/projects/stargazer/plugins/capture/nfqueue/nfqueue.cpp b/projects/stargazer/plugins/capture/nfqueue/nfqueue.cpp index a007c890..245fc649 100644 --- a/projects/stargazer/plugins/capture/nfqueue/nfqueue.cpp +++ b/projects/stargazer/plugins/capture/nfqueue/nfqueue.cpp @@ -25,6 +25,14 @@ #include "stg/common.h" #include "stg/raw_ip_packet.h" +extern "C" { + +#include /* Defines verdicts (NF_ACCEPT, etc) */ +#include + +} + +#include // ntohl #include //----------------------------------------------------------------------------- @@ -32,7 +40,36 @@ //----------------------------------------------------------------------------- namespace { + PLUGIN_CREATOR ncc; + +int callback(struct nfq_q_handle * /*queueHandle*/, struct nfgenmsg * /*msg*/, + struct nfq_data * nfqData, void *data) +{ +int id = 0; + +struct nfqnl_msg_packet_hdr * packetHeader = nfq_get_msg_packet_hdr(nfqData); +if (packetHeader == NULL) + return 0; + +id = ntohl(packetHeader->packet_id); + +unsigned char * payload = NULL; + +if (nfq_get_payload(nfqData) < 0) + return id; + +RAW_PACKET packet; + +memcpy(&packet.rawPacket, payload, sizeof(ip.rawPacket)); + +NFQ_CAP * cap = static_cast(data); + +cap->Process(packet); + +return nfq_set_verdict(qh, id, NF_ACCEPT, 0, NULL); +} + } extern "C" PLUGIN * GetPlugin(); @@ -71,6 +108,43 @@ int NFQ_CAP::Start() if (isRunning) return 0; +nfqHandle = nfq_open(); +if (nfqHandle == NULL) + { + errorStr = "Failed to initialize netfilter queue."; + logger(errorStr); + return -1; + } + +if (nfq_unbind_pf(nfqHandle, AF_INET) < 0) + { + errorStr = "Failed to unbind netfilter queue from IP handling."; + logger(errorStr); + return -1; + } + +if (nfq_bind_pf(nfqHandle, AF_INET) < 0) + { + errorStr = "Failed to bind netfilter queue to IP handling."; + logger(errorStr); + return -1; + } + +queueHandle = nfq_create_queue(nfqHandle, queueNumber, &Callback, this); +if (queueHandle == NULL) + { + errorStr = "Failed to create queue " + x2str(queueNumber) + "."; + logger(errorStr); + return -1; + } + +if (nfq_set_mode(queueHandle, NFQNL_COPY_PACKET, 0xffFF) < 0) + { + errorStr = "Failed to set queue " + x2str(queueNumber) + " mode."; + logger(errorStr); + return -1; + } + nonstop = true; if (pthread_create(&thread, NULL, Run, this)) @@ -122,6 +196,9 @@ if (isRunning) pthread_join(thread, NULL); +nfq_destroy_queue(queueHandle); +nfq_close(nfqHandle); + return 0; } //----------------------------------------------------------------------------- @@ -134,8 +211,22 @@ pthread_sigmask(SIG_BLOCK, &signalSet, NULL); NFQ_CAP * dc = static_cast(d); dc->isRunning = true; +int fd = nfq_fd(nfqHandle); +char buf[4096]; + while (dc->nonstop) { + if (!WaitPackets(fd)) + continue; + + int rv = read(fd, buf, sizeof(buf), 0); + if (rv < 0) + { + errorStr = "Read error: " + strerror(errno); + logger(errorStr); + break; + } + nfq_handle_packet(nfqHandle, buf, rv); } dc->isRunning = false; diff --git a/projects/stargazer/plugins/capture/nfqueue/nfqueue.h b/projects/stargazer/plugins/capture/nfqueue/nfqueue.h index 02452bb8..45342be0 100644 --- a/projects/stargazer/plugins/capture/nfqueue/nfqueue.h +++ b/projects/stargazer/plugins/capture/nfqueue/nfqueue.h @@ -38,6 +38,9 @@ class SETTINGS; class TRAFFCOUNTER; +struct nfq_handle; +struct nfq_q_handle; + class NFQ_CAP : public PLUGIN { public: NFQ_CAP(); @@ -58,6 +61,8 @@ public: uint16_t GetStartPosition() const { return 40; } uint16_t GetStopPosition() const { return 40; } + void Process(const RAW_PACKET & packet) { traffCnt->Process(packet); } + private: NFQ_CAP(const NFQ_CAP & rvalue); NFQ_CAP & operator=(const NFQ_CAP & rvalue); @@ -71,6 +76,9 @@ private: bool isRunning; MODULE_SETTINGS settings; + struct nfq_handle * nfqHandle; + struct nfq_q_handle * queueHandle; + TRAFFCOUNTER * traffCnt; PLUGIN_LOGGER logger;