]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/other/smux/snmp.h
540bd3d7c328eba9d7519fa6ab29d0548b143405
[stg.git] / projects / stargazer / plugins / other / smux / snmp.h
1 #ifndef __SNMP_AGENT_H__
2 #define __SNMP_AGENT_H__
3
4 #include <pthread.h>
5
6 #include <string>
7 #include <map>
8
9 #include "asn1/SMUX-PDUs.h"
10 #include "asn1/ObjectSyntax.h"
11
12 #include "stg/os_int.h"
13 #include "stg/plugin.h"
14 #include "stg/module_settings.h"
15 #include "stg/users.h"
16 #include "stg/tariffs.h"
17
18 #include "sensors.h"
19
20 extern "C" PLUGIN * GetPlugin();
21
22 class USER;
23 class SETTINGS;
24 class SNMP_AGENT;
25
26 typedef bool (SNMP_AGENT::*SMUXPacketHandler)(const SMUX_PDUs_t * pdus);
27 typedef bool (SNMP_AGENT::*PDUsHandler)(const PDUs_t * pdus);
28 typedef std::map<SMUX_PDUs_PR, SMUXPacketHandler> SMUXHandlers;
29 typedef std::map<PDUs_PR, PDUsHandler> PDUsHandlers;
30 //-----------------------------------------------------------------------------
31 class SNMP_AGENT_SETTINGS {
32 public:
33     SNMP_AGENT_SETTINGS();
34     virtual ~SNMP_AGENT_SETTINGS() {}
35     const std::string & GetStrError() const { return errorStr; }
36     int ParseSettings(const MODULE_SETTINGS & s);
37
38     uint32_t GetIP() const { return ip; }
39     uint16_t GetPort() const { return port; }
40     const std::string GetPassword() const { return password; }
41
42 private:
43     mutable std::string errorStr;
44
45     uint32_t ip;
46     uint16_t port;
47     std::string password;
48 };
49 //-----------------------------------------------------------------------------
50 class SNMP_AGENT : public PLUGIN {
51 public:
52     SNMP_AGENT();
53     virtual ~SNMP_AGENT();
54
55     void SetUsers(USERS * u) { users = u; }
56     void SetTariffs(TARIFFS * t) { tariffs = t; }
57     void SetAdmins(ADMINS *) {}
58     void SetTraffcounter(TRAFFCOUNTER *) {}
59     void SetStore(STORE *) {}
60     void SetStgSettings(const SETTINGS *) {}
61     void SetSettings(const MODULE_SETTINGS & s) { settings = s; }
62     int ParseSettings();
63
64     int Start();
65     int Stop();
66     int Reload() { return 0; }
67     bool IsRunning() { return running && !stopped; }
68
69     const std::string & GetStrError() const { return errorStr; }
70     const std::string GetVersion() const { return "Stg SNMP Agent 1.0"; }
71     uint16_t GetStartPosition() const { return 100; }
72     uint16_t GetStopPosition() const { return 100; }
73
74 private:
75     static void * Runner(void * d);
76     void Run();
77     bool PrepareNet();
78
79     bool DispatchPDUs(const SMUX_PDUs_t * pdus);
80
81     bool CloseHandler(const SMUX_PDUs_t * pdus);
82     bool RegisterResponseHandler(const SMUX_PDUs_t * pdus);
83     bool PDUsHandler(const SMUX_PDUs_t * pdus);
84     bool CommitOrRollbackHandler(const SMUX_PDUs_t * pdus);
85
86     bool GetRequestHandler(const PDUs_t * pdus);
87     bool GetNextRequestHandler(const PDUs_t * pdus);
88     bool SetRequestHandler(const PDUs_t * pdus);
89
90     USERS * users;
91     TARIFFS * tariffs;
92
93     mutable std::string errorStr;
94     SNMP_AGENT_SETTINGS snmpAgentSettings;
95     MODULE_SETTINGS settings;
96
97     pthread_t thread;
98     pthread_mutex_t mutex;
99     bool running;
100     bool stopped;
101
102     int sock;
103
104     SMUXHandlers smuxHandlers;
105     PDUsHandlers pdusHandlers;
106     Sensors sensors;
107 };
108 //-----------------------------------------------------------------------------
109
110 extern "C" PLUGIN * GetPlugin();
111
112 #endif