X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/641204dfbdb9fc870cdd2e7f9e3169a44693e7bf..9b3f8c7252b92a7b32996aa3c2b5e5c16361c82d:/projects/stargazer/plugins/capture/ipq_linux/ipq_cap.cpp?ds=inline diff --git a/projects/stargazer/plugins/capture/ipq_linux/ipq_cap.cpp b/projects/stargazer/plugins/capture/ipq_linux/ipq_cap.cpp index 9fc946ca..18696ec2 100644 --- a/projects/stargazer/plugins/capture/ipq_linux/ipq_cap.cpp +++ b/projects/stargazer/plugins/capture/ipq_linux/ipq_cap.cpp @@ -1,69 +1,75 @@ -#include -#include +/* + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +/* +* Author : Boris Mikhailenko +*/ + #include #include +#include +#include +#include + +#include "stg/raw_ip_packet.h" +#include "stg/traffcounter.h" +#include "stg/plugin_creator.h" +#include "stg/common.h" + #include "ipq_cap.h" -#include "raw_ip_packet.h" extern "C" - { - #include "libipq.h" - } - -class IPQ_CAP_CREATOR { -private: - IPQ_CAP * ic; - -public: - IPQ_CAP_CREATOR() - : ic(new IPQ_CAP()) - { - }; - ~IPQ_CAP_CREATOR() - { - delete ic; - }; +#include "libipq.h" +} - IPQ_CAP * GetCapturer() - { - return ic; - }; -}; -//----------------------------------------------------------------------------- //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -IPQ_CAP_CREATOR icc; //----------------------------------------------------------------------------- -//----------------------------------------------------------------------------- -//----------------------------------------------------------------------------- -BASE_PLUGIN * GetPlugin() +namespace { -return icc.GetCapturer(); +PLUGIN_CREATOR icc; } + +extern "C" PLUGIN * GetPlugin(); //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -const string IPQ_CAP::GetVersion() const +PLUGIN * GetPlugin() { -return "ipq_cap v.1.2"; +return icc.GetPlugin(); } //----------------------------------------------------------------------------- -IPQ_CAP::IPQ_CAP() -{ -isRunning = false; -nonstop = false; -} //----------------------------------------------------------------------------- -void IPQ_CAP::SetTraffcounter(TRAFFCOUNTER * tc) +//----------------------------------------------------------------------------- +std::string IPQ_CAP::GetVersion() const { -traffCnt = tc; +return "cap_ipq v.1.2"; } //----------------------------------------------------------------------------- -const string & IPQ_CAP::GetStrError() const +IPQ_CAP::IPQ_CAP() + : ipq_h(NULL), + nonstop(false), + isRunning(false), + capSock(-1), + traffCnt(NULL), + logger(GetPluginLogger(GetStgLogger(), "cap_ipq")) { -return errorStr; +memset(buf, 0, BUFSIZE); } //----------------------------------------------------------------------------- int IPQ_CAP::Start() @@ -96,7 +102,8 @@ for (int i = 0; i < 25; i++) { if (!isRunning) break; - usleep(200000); + struct timespec ts = {0, 200000000}; + nanosleep(&ts, NULL); } //after 5 seconds waiting thread still running. now killing it if (isRunning) @@ -108,7 +115,8 @@ if (isRunning) } for (int i = 0; i < 25 && isRunning; ++i) { - usleep(200000); + struct timespec ts = {0, 200000000}; + nanosleep(&ts, NULL); } if (isRunning) { @@ -123,23 +131,21 @@ IPQCapClose(); return 0; } //----------------------------------------------------------------------------- -bool IPQ_CAP::IsRunning() -{ -return isRunning; -} -//----------------------------------------------------------------------------- void * IPQ_CAP::Run(void * d) { +sigset_t signalSet; +sigfillset(&signalSet); +pthread_sigmask(SIG_BLOCK, &signalSet, NULL); + RAW_PACKET raw_packet; -int status; -IPQ_CAP * dc = (IPQ_CAP *)d; +IPQ_CAP * dc = static_cast(d); dc->isRunning = true; memset(&raw_packet, 0, sizeof(raw_packet)); 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 || @@ -151,31 +157,21 @@ dc->isRunning = false; 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) { ipq_destroy_handle(ipq_h); + logger("Cannot create IPQ handle. Error: '%s', '%s'", ipq_errstr(), strerror(errno)); 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); + logger("Cannot set IPQ_COPY_PACKET mode."); errorStr = "Cannot set IPQ_COPY_PACKET mode!"; return -1; } @@ -190,11 +186,8 @@ return 0; //----------------------------------------------------------------------------- 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) @@ -203,7 +196,7 @@ if (status < 0) 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;