//#define CAP_DEBUG 1
-extern "C" Plugin* GetPlugin()
+extern "C" STG::Plugin* GetPlugin()
{
-static BPF_CAP plugin;
-return &plugin;
+ static BPF_CAP plugin;
+ return &plugin;
}
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
-int BPF_CAP_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
+int BPF_CAP_SETTINGS::ParseSettings(const STG::ModuleSettings & s)
{
iface.erase(iface.begin(), iface.end());
isRunning(false),
capSock(-1),
traffCnt(NULL),
- logger(PluginLogger::get("cap_bpf"))
+ logger(STG::PluginLogger::get("cap_bpf"))
{
}
//-----------------------------------------------------------------------------
uint8_t hdr[96]; //68 + 14 + 4(size) + 9(SYS_IFACE) + 1(align to 4) = 96
-RAW_PACKET * rpp = (RAW_PACKET *)&hdr[14];
+STG::RawPacket * rpp = (STG::RawPacket *)&hdr[14];
memset(hdr, 0, sizeof(hdr));
rpp->dataLen = -1;
if (!(hdr[12] == 0x8 && hdr[13] == 0x0))
continue;
- dc->traffCnt->Process(*rpp);
+ dc->traffCnt->process(*rpp);
}
dc->isRunning = false;
* Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
*/
-#ifndef ETHER_CAP_H
-#define ETHER_CAP_H
+#pragma once
-#include <pthread.h>
+#include "stg/plugin.h"
+#include "stg/module_settings.h"
+#include "stg/logger.h"
#include <string>
#include <vector>
#include <cstdint>
-#include "stg/plugin.h"
-#include "stg/module_settings.h"
-#include "stg/logger.h"
+#include <pthread.h>
#define BUFF_LEN (128)
-class TRAFFCOUNTER;
+namespace STG
+{
+struct TraffCounter;
+}
//-----------------------------------------------------------------------------
struct BPF_DATA {
//-----------------------------------------------------------------------------
class BPF_CAP_SETTINGS {
public:
- virtual ~BPF_CAP_SETTINGS() {}
const std::string & GetStrError() const { return errorStr; }
- int ParseSettings(const MODULE_SETTINGS & s);
+ int ParseSettings(const STG::ModuleSettings & s);
std::string GetIface(unsigned int num);
private:
mutable std::string errorStr;
};
//-----------------------------------------------------------------------------
-class BPF_CAP : public PLUGIN {
+class BPF_CAP : public STG::Plugin {
public:
BPF_CAP();
- virtual ~BPF_CAP() {}
- void SetTraffcounter(TRAFFCOUNTER * tc) { traffCnt = tc; }
+ void SetTraffcounter(STG::TraffCounter * tc) override { traffCnt = tc; }
- int Start();
- int Stop();
- int Reload(const MODULE_SETTINGS & /*ms*/) { 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; }
- void SetSettings(const MODULE_SETTINGS & s) { settings = s; }
- int ParseSettings();
+ void SetSettings(const STG::ModuleSettings & s) override { settings = s; }
+ int ParseSettings() override;
- const std::string & GetStrError() const { return errorStr; }
- std::string GetVersion() const;
- uint16_t GetStartPosition() const { return 40; }
- uint16_t GetStopPosition() const { return 40; }
+ 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:
BPF_CAP(const BPF_CAP & rvalue);
bool nonstop;
bool isRunning;
int capSock;
- MODULE_SETTINGS settings;
+ STG::ModuleSettings settings;
- TRAFFCOUNTER * traffCnt;
+ STG::TraffCounter * traffCnt;
- PLUGIN_LOGGER logger;
+ STG::PluginLogger logger;
};
-//-----------------------------------------------------------------------------
-
-#endif