X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/e9ae1f101b5418c0ba2e6c9d86b23c12f0140982..4f1ec5ae5fcb55e63412fd99103c72306341eff7:/stargazer/plugins/capture/divert_freebsd/divert_cap.cpp diff --git a/stargazer/plugins/capture/divert_freebsd/divert_cap.cpp b/stargazer/plugins/capture/divert_freebsd/divert_cap.cpp index 1885835b..0688150d 100644 --- a/stargazer/plugins/capture/divert_freebsd/divert_cap.cpp +++ b/stargazer/plugins/capture/divert_freebsd/divert_cap.cpp @@ -25,8 +25,9 @@ $Date: 2010/09/10 06:43:03 $ #include "divert_cap.h" -#include "stg/common.h" #include "stg/traffcounter.h" +#include "stg/raw_ip_packet.h" +#include "stg/common.h" #include #include @@ -64,7 +65,7 @@ DIVERT_DATA cddiv; //capture data //----------------------------------------------------------------------------- //----------------------------------------------------------------------------- -extern "C" Plugin* GetPlugin() +extern "C" STG::Plugin* GetPlugin() { static DIVERT_CAP plugin; return &plugin; @@ -83,7 +84,7 @@ DIVERT_CAP::DIVERT_CAP() nonstop(false), isRunning(false), traffCnt(NULL), - logger(PluginLogger::get("cap_divert")) + logger(STG::PluginLogger::get("cap_divert")) { } //----------------------------------------------------------------------------- @@ -159,7 +160,7 @@ dc->isRunning = true; char buffer[pcktSize + 14]; while (dc->nonstop) { - RAW_PACKET rp; + STG::RawPacket rp; dc->DivertCapRead(buffer, sizeof(buffer), NULL); if (buffer[12] != 0x8) @@ -167,7 +168,7 @@ while (dc->nonstop) memcpy(rp.rawPacket.pckt, &buffer[14], pcktSize); - dc->traffCnt->Process(rp); + dc->traffCnt->process(rp); } dc->isRunning = false; @@ -276,8 +277,8 @@ return 0; int DIVERT_CAP::ParseSettings() { int p; -PARAM_VALUE pv; -std::vector::const_iterator pvi; +STG::ParamValue pv; +std::vector::const_iterator pvi; pv.param = "Port"; pvi = std::find(settings.moduleParams.begin(), settings.moduleParams.end(), pv);