projects/stargazer/plugins/other/smux/smux.cpp

   1 #include <sys/types.h>
   2 #include <sys/socket.h>
   3 #include <arpa/inet.h>
   4
   5 #include <cstring>
   6 #include <cerrno>
   7 #include <ctime>
   8 #include <csignal>
   9
  10 #include <vector>
  11 #include <algorithm>
  12
  13 #include "stg/common.h"
  14
  15 #include "smux.h"
  16 #include "utils.h"
  17
  18 class SMUX_CREATOR
  19 {
  20 private:
  21     SMUX * smux;
  22
  23 public:
  24     SMUX_CREATOR() : smux(new SMUX()) {}
  25     ~SMUX_CREATOR() { delete smux; }
  26
  27     SMUX * GetPlugin() { return smux; }
  28 };
  29
  30 SMUX_CREATOR sac;
  31
  32 PLUGIN * GetPlugin()
  33 {
  34 return sac.GetPlugin();
  35 }
  36
  37 SMUX_SETTINGS::SMUX_SETTINGS()
  38     : ip(0),
  39       port(0)
  40 {}
  41
  42 int SMUX_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
  43 {
  44 PARAM_VALUE pv;
  45 std::vector<PARAM_VALUE>::const_iterator pvi;
  46 int p;
  47 ///////////////////////////
  48 pv.param = "Port";
  49 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
  50 if (pvi == s.moduleParams.end())
  51     {
  52     errorStr = "Parameter \'Port\' not found.";
  53     printfd(__FILE__, "Parameter 'Port' not found\n");
  54     return -1;
  55     }
  56 if (ParseIntInRange(pvi->value[0], 2, 65535, &p))
  57     {
  58     errorStr = "Cannot parse parameter \'Port\': " + errorStr;
  59     printfd(__FILE__, "Cannot parse parameter 'Port'\n");
  60     return -1;
  61     }
  62 port = p;
  63
  64 pv.param = "Password";
  65 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
  66 if (pvi == s.moduleParams.end())
  67     {
  68     errorStr = "Parameter \'Password\' not found.";
  69     printfd(__FILE__, "Parameter 'Password' not found\n");
  70     password = "";
  71     }
  72 else
  73     {
  74     password = pvi->value[0];
  75     }
  76
  77 pv.param = "Server";
  78 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
  79 if (pvi == s.moduleParams.end())
  80     {
  81     errorStr = "Parameter \'Server\' not found.";
  82     printfd(__FILE__, "Parameter 'Server' not found\n");
  83     return -1;
  84     }
  85 ip = inet_strington(pvi->value[0]);
  86
  87 return 0;
  88 }
  89
  90 SMUX::SMUX()
  91     : PLUGIN(),
  92       users(NULL),
  93       tariffs(NULL),
  94       running(false),
  95       stopped(true),
  96       sock(-1)
  97 {
  98 pthread_mutex_init(&mutex, NULL);
  99
 100 smuxHandlers[SMUX_PDUs_PR_close] = &SMUX::CloseHandler;
 101 smuxHandlers[SMUX_PDUs_PR_registerResponse] = &SMUX::RegisterResponseHandler;
 102 smuxHandlers[SMUX_PDUs_PR_pdus] = &SMUX::PDUsRequestHandler;
 103 smuxHandlers[SMUX_PDUs_PR_commitOrRollback] = &SMUX::CommitOrRollbackHandler;
 104
 105 pdusHandlers[PDUs_PR_get_request] = &SMUX::GetRequestHandler;
 106 pdusHandlers[PDUs_PR_get_next_request] = &SMUX::GetNextRequestHandler;
 107 pdusHandlers[PDUs_PR_set_request] = &SMUX::SetRequestHandler;
 108 }
 109
 110 SMUX::~SMUX()
 111 {
 112 Sensors::iterator it;
 113 for (it = sensors.begin(); it != sensors.end(); ++it)
 114     delete it->second;
 115 printfd(__FILE__, "SMUX::~SMUX()\n");
 116 pthread_mutex_destroy(&mutex);
 117 }
 118
 119 int SMUX::ParseSettings()
 120 {
 121 return smuxSettings.ParseSettings(settings);
 122 }
 123
 124 int SMUX::Start()
 125 {
 126 if (PrepareNet())
 127     return -1;
 128
 129 // Users
 130 sensors[".1.3.6.1.4.1.38313.1.1.1"] = new TotalUsersSensor(*users);
 131 sensors[".1.3.6.1.4.1.38313.1.1.2"] = new ConnectedUsersSensor(*users);
 132 sensors[".1.3.6.1.4.1.38313.1.1.3"] = new AuthorizedUsersSensor(*users);
 133 sensors[".1.3.6.1.4.1.38313.1.1.4"] = new AlwaysOnlineUsersSensor(*users);
 134 sensors[".1.3.6.1.4.1.38313.1.1.5"] = new NoCashUsersSensor(*users);
 135 sensors[".1.3.6.1.4.1.38313.1.1.7"] = new DisabledDetailStatsUsersSensor(*users);
 136 sensors[".1.3.6.1.4.1.38313.1.1.8"] = new DisabledUsersSensor(*users);
 137 sensors[".1.3.6.1.4.1.38313.1.1.9"] = new PassiveUsersSensor(*users);
 138 sensors[".1.3.6.1.4.1.38313.1.1.10"] = new CreditUsersSensor(*users);
 139 sensors[".1.3.6.1.4.1.38313.1.1.11"] = new FreeMbUsersSensor(*users);
 140 sensors[".1.3.6.1.4.1.38313.1.1.12"] = new TariffChangeUsersSensor(*users);
 141 // Tariffs
 142 sensors[".1.3.6.1.4.1.38313.1.2.1"] = new TotalTariffsSensor(*tariffs);
 143
 144 if (!running)
 145     {
 146     if (pthread_create(&thread, NULL, Runner, this))
 147         {
 148         errorStr = "Cannot create thread.";
 149         printfd(__FILE__, "Cannot create thread\n");
 150         return -1;
 151         }
 152     }
 153
 154 return 0;
 155 }
 156
 157 int SMUX::Stop()
 158 {
 159 printfd(__FILE__, "SMUX::Stop() - Before\n");
 160 running = false;
 161
 162 if (!stopped)
 163     {
 164     //5 seconds to thread stops itself
 165     for (int i = 0; i < 25 && !stopped; i++)
 166         {
 167         struct timespec ts = {0, 200000000};
 168         nanosleep(&ts, NULL);
 169         }
 170
 171     //after 5 seconds waiting thread still running. now killing it
 172     if (!stopped)
 173         {
 174         printfd(__FILE__, "SMUX::Stop() - failed to stop thread, killing it\n");
 175         if (pthread_kill(thread, SIGINT))
 176             {
 177             errorStr = "Cannot kill thread.";
 178             printfd(__FILE__, "SMUX::Stop() - Cannot kill thread\n");
 179             return -1;
 180             }
 181         printfd(__FILE__, "SMUX::Stop() -  killed Run\n");
 182         }
 183     }
 184
 185 pthread_join(thread, NULL);
 186
 187 close(sock);
 188
 189 printfd(__FILE__, "SMUX::Stop() - After\n");
 190 return 0;
 191 }
 192
 193 void * SMUX::Runner(void * d)
 194 {
 195 SMUX * smux = static_cast<SMUX *>(d);
 196
 197 smux->Run();
 198
 199 return NULL;
 200 }
 201
 202 void SMUX::Run()
 203 {
 204 SendOpenPDU(sock);
 205 SendRReqPDU(sock);
 206 running = true;
 207 stopped = false;
 208 while(running)
 209     {
 210     if (WaitPackets(sock))
 211         {
 212         SMUX_PDUs_t * pdus = RecvSMUXPDUs(sock);
 213         if (pdus)
 214             DispatchPDUs(pdus);
 215         }
 216     if (!running)
 217         break;
 218     }
 219 SendClosePDU(sock);
 220 stopped = true;
 221 }
 222
 223 bool SMUX::PrepareNet()
 224 {
 225 sock = socket(AF_INET, SOCK_STREAM, 0);
 226
 227 if (sock < 0)
 228     {
 229     errorStr = "Cannot create socket.";
 230     printfd(__FILE__, "Cannot create socket\n");
 231     return true;
 232     }
 233
 234 struct sockaddr_in addr;
 235
 236 addr.sin_family = AF_INET;
 237 addr.sin_port = htons(smuxSettings.GetPort());
 238 addr.sin_addr.s_addr = smuxSettings.GetIP();
 239
 240 if (connect(sock, reinterpret_cast<struct sockaddr *>(&addr), sizeof(addr)))
 241     {
 242     errorStr = "Cannot connect.";
 243     printfd(__FILE__, "Cannot connect. Message: '%s'\n", strerror(errno));
 244     return true;
 245     }
 246
 247 return false;
 248 }
 249
 250 bool SMUX::DispatchPDUs(const SMUX_PDUs_t * pdus)
 251 {
 252 SMUXHandlers::iterator it;
 253 it = smuxHandlers.find(pdus->present);
 254 if (it != smuxHandlers.end())
 255     {
 256     return (this->*(it->second))(pdus);
 257     }
 258 else
 259     {
 260     switch (pdus->present)
 261         {
 262         case SMUX_PDUs_PR_NOTHING:
 263             printfd(__FILE__, "PDUs: nothing\n");
 264             break;
 265         case SMUX_PDUs_PR_open:
 266             printfd(__FILE__, "PDUs: open\n");
 267             break;
 268         case SMUX_PDUs_PR_registerRequest:
 269             printfd(__FILE__, "PDUs: registerRequest\n");
 270             break;
 271         default:
 272             printfd(__FILE__, "PDUs: undefined\n");
 273         }
 274     asn_fprint(stderr, &asn_DEF_SMUX_PDUs, pdus);
 275     }
 276 return false;
 277 }