/* * 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 : Maxim Mamontov */ #pragma once #include "stg/plugin.h" #include "stg/module_settings.h" #include "stg/logger.h" #include #include #pragma GCC diagnostic push #pragma GCC diagnostic ignored "-Wshadow" #include #pragma GCC diagnostic pop namespace STG { class Users; class Tariffs; struct Admins; struct TraffCounter; struct Settings; struct RawPacket; } struct nfq_handle; struct nfq_q_handle; class NFQ_CAP : public STG::Plugin { public: NFQ_CAP(); void SetTraffcounter(STG::TraffCounter * tc) override { traffCnt = tc; } int Start() override; int Stop() override; int Reload(const STG::ModuleSettings & /*ms*/) override { return 0; } bool IsRunning() override { return isRunning; } void SetSettings(const STG::ModuleSettings & s) override { settings = s; } int ParseSettings() override; const std::string & GetStrError() const override { return errorStr; } std::string GetVersion() const override; uint16_t GetStartPosition() const override { return 40; } uint16_t GetStopPosition() const override { return 40; } void Process(const STG::RawPacket & packet); private: NFQ_CAP(const NFQ_CAP & rvalue); NFQ_CAP & operator=(const NFQ_CAP & rvalue); void Run(std::stop_token token); mutable std::string errorStr; std::jthread m_thread; bool isRunning; STG::ModuleSettings settings; size_t queueNumber; struct nfq_handle * nfqHandle; struct nfq_q_handle * queueHandle; STG::TraffCounter * traffCnt; STG::PluginLogger logger; };