Author : Boris Mikhailenko <stg34@stg.dp.ua>
*/
-/*
-$Revision: 1.6 $
-$Date: 2009/06/23 11:32:27 $
-*/
+#pragma once
-#ifndef DIVERT_CAP_H
-#define DIVERT_CAP_H
-
-#include <pthread.h>
+#include "stg/plugin.h"
+#include "stg/module_settings.h"
+#include "stg/logger.h"
#include <string>
-#include "plugin.h"
-#include "module_settings.h"
-#include "../../../traffcounter.h"
-
-class USERS;
-class TARIFFS;
-class ADMINS;
-class TRAFFCOUNTER;
-class SETTINGS;
+#include <pthread.h>
-extern "C" PLUGIN * GetPlugin();
+namespace STG
+{
+struct Users;
+struct Tariffs;
+struct Admins;
+struct TraffCounter;
+struct Settings;
+}
//-----------------------------------------------------------------------------
-class DIVERT_CAP : public PLUGIN {
+class DIVERT_CAP : public STG::Plugin {
public:
DIVERT_CAP();
- virtual ~DIVERT_CAP() {}
-
- void SetUsers(USERS *) {}
- void SetTariffs(TARIFFS *) {}
- void SetAdmins(ADMINS *) {}
- void SetTraffcounter(TRAFFCOUNTER * tc);
- void SetStore(STORE *) {}
- void SetStgSettings(const SETTINGS *) {}
-
- int Start();
- int Stop();
- int Reload() { return 0; }
- bool IsRunning();
-
- void SetSettings(const MODULE_SETTINGS & s);
- int ParseSettings();
- const std::string & GetStrError() const;
- const std::string GetVersion() const;
- uint16_t GetStartPosition() const;
- uint16_t GetStopPosition() const;
+
+ 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; }
private:
+ DIVERT_CAP(const DIVERT_CAP & rvalue);
+ DIVERT_CAP & operator=(const DIVERT_CAP & rvalue);
+
static void * Run(void *);
int DivertCapOpen();
int DivertCapRead(char * buffer, int blen, char ** iface, int n);
int DivertCapClose();
- int ParseIntInRange(const std::string & str, int min, int max, int * val);
-
- MODULE_SETTINGS settings;
+ STG::ModuleSettings settings;
int port;
+ bool disableForwarding;
mutable std::string errorStr;
bool nonstop;
bool isRunning;
- TRAFFCOUNTER * traffCnt;
-};
-//-----------------------------------------------------------------------------
+ STG::TraffCounter * traffCnt;
-#endif
+ STG::PluginLogger logger;
+};