X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/641204dfbdb9fc870cdd2e7f9e3169a44693e7bf..a1211c21a5e253a1068042fa9bfdf19056c534ac:/projects/stargazer/plugins/capture/ether_linux/ether_cap.cpp?ds=sidebyside diff --git a/projects/stargazer/plugins/capture/ether_linux/ether_cap.cpp b/projects/stargazer/plugins/capture/ether_linux/ether_cap.cpp index 5c76be91..bd075fee 100644 --- a/projects/stargazer/plugins/capture/ether_linux/ether_cap.cpp +++ b/projects/stargazer/plugins/capture/ether_linux/ether_cap.cpp @@ -44,67 +44,39 @@ $Date: 2009/12/13 13:45:13 $ #include #include +#include "stg/common.h" +#include "stg/raw_ip_packet.h" +#include "stg/traffcounter.h" +#include "stg/plugin_creator.h" #include "ether_cap.h" -#include "common.h" -#include "raw_ip_packet.h" //#define CAP_DEBUG 1 - -//----------------------------------------------------------------------------- -class ETHER_CAP_CREATOR -{ -private: - ETHER_CAP * ec; - -public: - ETHER_CAP_CREATOR() - : ec(new ETHER_CAP()) - { - }; - ~ETHER_CAP_CREATOR() - { - delete ec; - }; - - ETHER_CAP * GetCapturer() - { - return ec; - }; -}; //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -ETHER_CAP_CREATOR ecc; +PLUGIN_CREATOR ecc; //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -BASE_PLUGIN * GetPlugin() +PLUGIN * GetPlugin() { -return ecc.GetCapturer(); +return ecc.GetPlugin(); } //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -const string ETHER_CAP::GetVersion() const +const std::string ETHER_CAP::GetVersion() const { return "Ether_cap v.1.2"; } //----------------------------------------------------------------------------- ETHER_CAP::ETHER_CAP() + : nonstop(false), + isRunning(false), + capSock(-1), + traffCnt(NULL) { -isRunning = false; -nonstop = false; -} -//----------------------------------------------------------------------------- -void ETHER_CAP::SetTraffcounter(TRAFFCOUNTER * tc) -{ -traffCnt = tc; -} -//----------------------------------------------------------------------------- -const string & ETHER_CAP::GetStrError() const -{ -return errorStr; } //----------------------------------------------------------------------------- int ETHER_CAP::Start() @@ -169,11 +141,6 @@ EthCapClose(); return 0; } //----------------------------------------------------------------------------- -bool ETHER_CAP::IsRunning() -{ -return isRunning; -} -//----------------------------------------------------------------------------- void * ETHER_CAP::Run(void * d) { ETHER_CAP * dc = (ETHER_CAP *)d; @@ -215,16 +182,6 @@ dc->isRunning = false; return NULL; } //----------------------------------------------------------------------------- -uint16_t ETHER_CAP::GetStartPosition() const -{ -return 10; -} -//----------------------------------------------------------------------------- -uint16_t ETHER_CAP::GetStopPosition() const -{ -return 10; -} -//----------------------------------------------------------------------------- int ETHER_CAP::EthCapOpen() { capSock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));