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