X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/a05654cf045d0e8ecc104b019145f627ad6ce4eb..5dfab5bea026e33031d041b233c73e196469b4ac:/projects/stargazer/plugins/capture/pcap/pcap_cap.h?ds=inline diff --git a/projects/stargazer/plugins/capture/pcap/pcap_cap.h b/projects/stargazer/plugins/capture/pcap/pcap_cap.h index b39f736a..e72c3e74 100644 --- a/projects/stargazer/plugins/capture/pcap/pcap_cap.h +++ b/projects/stargazer/plugins/capture/pcap/pcap_cap.h @@ -18,8 +18,7 @@ * Author : Maxim Mamontov */ -#ifndef PCAP_CAP_H -#define PCAP_CAP_H +#pragma once #include "stg/plugin.h" #include "stg/module_settings.h" @@ -27,20 +26,32 @@ #include #include +#pragma GCC diagnostic push +#pragma GCC diagnostic ignored "-Wshadow" +#include +#pragma GCC diagnostic pop #include -#include +#include -class USERS; -class TARIFFS; -class ADMINS; -class TRAFFCOUNTER; -class SETTINGS; +namespace STG +{ + +struct Users; +struct Tariffs; +struct Admins; +struct TraffCounter; +struct Settings; -class TRAFFCOUNTER; +} struct DEV { + DEV() : device("any"), filterExpression("ip"), handle(NULL), fd(-1) {} + explicit DEV(const std::string & d) : device(d), filterExpression("ip"), handle(NULL), fd(-1) {} + DEV(const std::string & d, const std::string & f) + : device(d), filterExpression(f), handle(NULL), fd(-1) {} + std::string device; std::string filterExpression; pcap_t * handle; @@ -50,41 +61,42 @@ struct DEV typedef std::vector DEV_MAP; -class PCAP_CAP : public PLUGIN { +class PCAP_CAP : public STG::Plugin { public: PCAP_CAP(); - virtual ~PCAP_CAP() {} - void SetTraffcounter(TRAFFCOUNTER * tc) { traffCnt = tc; } + void SetTraffcounter(STG::TraffCounter * tc) override { traffCnt = tc; } - int Start(); - int Stop(); - int Reload() { return 0; } - bool IsRunning() { return isRunning; } + int Start() override; + int Stop() override; + int Reload(const STG::ModuleSettings & /*ms*/) override { return 0; } + bool IsRunning() override { return isRunning; } - int ParseSettings() { return 0; } - const std::string & GetStrError() const { return errorStr; } - std::string GetVersion() const; - uint16_t GetStartPosition() const { return 40; } - uint16_t GetStopPosition() const { return 40; } + 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; } private: PCAP_CAP(const PCAP_CAP & rvalue); PCAP_CAP & operator=(const PCAP_CAP & rvalue); - static void * Run(void *); + void TryRead(const fd_set & set); + void TryReadDev(const DEV & dev); + + void Run(std::stop_token token); mutable std::string errorStr; - pthread_t thread; - bool nonstop; + std::jthread m_thread; bool isRunning; + STG::ModuleSettings settings; DEV_MAP devices; - TRAFFCOUNTER * traffCnt; + STG::TraffCounter * traffCnt; - PLUGIN_LOGGER logger; + STG::PluginLogger logger; }; -//----------------------------------------------------------------------------- - -#endif