stargazer/plugins/other/ping/ping.cpp

   1 #include "ping.h"
   2
   3 #include "stg/user.h"
   4 #include "stg/locker.h"
   5 #include "stg/user_property.h"
   6 #include "stg/plugin_creator.h"
   7
   8 #include <cstdio>
   9 #include <cassert>
  10 #include <csignal>
  11 #include <ctime>
  12 #include <algorithm>
  13
  14 namespace
  15 {
  16 PLUGIN_CREATOR<PING> pc;
  17
  18 //-----------------------------------------------------------------------------
  19 //-----------------------------------------------------------------------------
  20 //-----------------------------------------------------------------------------
  21 template <typename varType>
  22 class HAS_USER: public std::binary_function<varType, USER_PTR, bool>
  23 {
  24 public:
  25     explicit HAS_USER(const USER_PTR & u) : user(u) {}
  26     bool operator()(varType notifier) const
  27         {
  28         return notifier.GetUser() == user;
  29         }
  30 private:
  31     const USER_PTR & user;
  32 };
  33 }
  34
  35 extern "C" PLUGIN * GetPlugin();
  36 //-----------------------------------------------------------------------------
  37 //-----------------------------------------------------------------------------
  38 //-----------------------------------------------------------------------------
  39 PLUGIN * GetPlugin()
  40 {
  41 return pc.GetPlugin();
  42 }
  43 //-----------------------------------------------------------------------------
  44 //-----------------------------------------------------------------------------
  45 //-----------------------------------------------------------------------------
  46 int PING_SETTINGS::ParseSettings(const MODULE_SETTINGS & s)
  47 {
  48 PARAM_VALUE pv;
  49 std::vector<PARAM_VALUE>::const_iterator pvi;
  50
  51 pv.param = "PingDelay";
  52 pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv);
  53 if (pvi == s.moduleParams.end() || pvi->value.empty())
  54     {
  55     errorStr = "Parameter \'PingDelay\' not found.";
  56     printfd(__FILE__, "Parameter 'PingDelay' not found\n");
  57     return -1;
  58     }
  59 if (ParseIntInRange(pvi->value[0], 5, 3600, &pingDelay))
  60     {
  61     errorStr = "Cannot parse parameter \'PingDelay\': " + errorStr;
  62     printfd(__FILE__, "Canot parse parameter 'PingDelay'\n");
  63     return -1;
  64     }
  65
  66 return 0;
  67 }
  68 //-----------------------------------------------------------------------------
  69 PING::PING()
  70     : users(NULL),
  71       nonstop(false),
  72       isRunning(false),
  73       onAddUserNotifier(*this),
  74       onDelUserNotifier(*this),
  75       logger(GetPluginLogger(GetStgLogger(), "ping"))
  76 {
  77 pthread_mutex_init(&mutex, NULL);
  78 }
  79 //-----------------------------------------------------------------------------
  80 PING::~PING()
  81 {
  82 pthread_mutex_destroy(&mutex);
  83 }
  84 //-----------------------------------------------------------------------------
  85 int PING::ParseSettings()
  86 {
  87 int ret = pingSettings.ParseSettings(settings);
  88 if (ret)
  89     errorStr = pingSettings.GetStrError();
  90 return ret;
  91 }
  92 //-----------------------------------------------------------------------------
  93 int PING::Start()
  94 {
  95 GetUsers();
  96
  97 users->AddNotifierUserAdd(&onAddUserNotifier);
  98 users->AddNotifierUserDel(&onDelUserNotifier);
  99
 100 nonstop = true;
 101
 102 pinger.SetDelayTime(pingSettings.GetPingDelay());
 103 pinger.Start();
 104
 105 if (pthread_create(&thread, NULL, Run, this))
 106     {
 107     errorStr = "Cannot start thread.";
 108     logger("Cannot create thread.");
 109     printfd(__FILE__, "Cannot start thread\n");
 110     return -1;
 111     }
 112
 113 return 0;
 114 }
 115 //-----------------------------------------------------------------------------
 116 int PING::Stop()
 117 {
 118 STG_LOCKER lock(&mutex);
 119
 120 if (!isRunning)
 121     return 0;
 122
 123 pinger.Stop();
 124 nonstop = false;
 125 //5 seconds to thread stops itself
 126 struct timespec ts = {0, 200000000};
 127 for (int i = 0; i < 25; i++)
 128     {
 129     if (!isRunning)
 130         break;
 131
 132     nanosleep(&ts, NULL);
 133     }
 134
 135 users->DelNotifierUserAdd(&onAddUserNotifier);
 136 users->DelNotifierUserDel(&onDelUserNotifier);
 137
 138 std::list<USER_PTR>::iterator users_iter;
 139 users_iter = usersList.begin();
 140 while (users_iter != usersList.end())
 141     {
 142     UnSetUserNotifiers(*users_iter);
 143     ++users_iter;
 144     }
 145
 146 if (isRunning)
 147     return -1;
 148
 149 return 0;
 150 }
 151 //-----------------------------------------------------------------------------
 152 bool PING::IsRunning()
 153 {
 154 return isRunning;
 155 }
 156 //-----------------------------------------------------------------------------
 157 void * PING::Run(void * d)
 158 {
 159 sigset_t signalSet;
 160 sigfillset(&signalSet);
 161 pthread_sigmask(SIG_BLOCK, &signalSet, NULL);
 162
 163 PING * ping = static_cast<PING *>(d);
 164 ping->isRunning = true;
 165
 166 long delay = (10000000 * ping->pingSettings.GetPingDelay()) / 3 + 50000000;
 167
 168 while (ping->nonstop)
 169     {
 170     std::list<USER_PTR>::iterator iter = ping->usersList.begin();
 171         {
 172         STG_LOCKER lock(&ping->mutex);
 173         while (iter != ping->usersList.end())
 174             {
 175             if ((*iter)->GetProperty().ips.ConstData().OnlyOneIP())
 176                 {
 177                 uint32_t ip = (*iter)->GetProperty().ips.ConstData()[0].ip;
 178                 time_t t;
 179                 if (ping->pinger.GetIPTime(ip, &t) == 0)
 180                     {
 181                     if (t)
 182                         (*iter)->UpdatePingTime(t);
 183                     }
 184                 }
 185             else
 186                 {
 187                 uint32_t ip = (*iter)->GetCurrIP();
 188                 if (ip)
 189                     {
 190                     time_t t;
 191                     if (ping->pinger.GetIPTime(ip, &t) == 0)
 192                         {
 193                         if (t)
 194                             (*iter)->UpdatePingTime(t);
 195                         }
 196                     }
 197                 }
 198             ++iter;
 199             }
 200         }
 201     struct timespec ts = {delay / 1000000000, delay % 1000000000};
 202     for (int i = 0; i < 100; i++)
 203         {
 204         if (ping->nonstop)
 205             {
 206             nanosleep(&ts, NULL);
 207             }
 208         }
 209     }
 210
 211 ping->isRunning = false;
 212 return NULL;
 213 }
 214 //-----------------------------------------------------------------------------
 215 void PING::SetUserNotifiers(USER_PTR u)
 216 {
 217 CHG_CURRIP_NOTIFIER_PING ChgCurrIPNotifier(*this, u);
 218 CHG_IPS_NOTIFIER_PING ChgIPNotifier(*this, u);
 219
 220 ChgCurrIPNotifierList.push_front(ChgCurrIPNotifier);
 221 ChgIPNotifierList.push_front(ChgIPNotifier);
 222
 223 u->AddCurrIPAfterNotifier(&(*ChgCurrIPNotifierList.begin()));
 224 u->GetProperty().ips.AddAfterNotifier(&(*ChgIPNotifierList.begin()));
 225 }
 226 //-----------------------------------------------------------------------------
 227 void PING::UnSetUserNotifiers(USER_PTR u)
 228 {
 229 // ---          CurrIP              ---
 230 HAS_USER<CHG_CURRIP_NOTIFIER_PING> IsContainsUserCurrIP(u);
 231 HAS_USER<CHG_IPS_NOTIFIER_PING> IsContainsUserIP(u);
 232
 233 std::list<CHG_CURRIP_NOTIFIER_PING>::iterator currIPter;
 234 std::list<CHG_IPS_NOTIFIER_PING>::iterator IPIter;
 235
 236 currIPter = find_if(ChgCurrIPNotifierList.begin(),
 237                     ChgCurrIPNotifierList.end(),
 238                     IsContainsUserCurrIP);
 239
 240 if (currIPter != ChgCurrIPNotifierList.end())
 241     {
 242     currIPter->GetUser()->DelCurrIPAfterNotifier(&(*currIPter));
 243     ChgCurrIPNotifierList.erase(currIPter);
 244     }
 245 // ---         CurrIP end          ---
 246
 247 // ---          IP              ---
 248 IPIter = find_if(ChgIPNotifierList.begin(),
 249                  ChgIPNotifierList.end(),
 250                  IsContainsUserIP);
 251
 252 if (IPIter != ChgIPNotifierList.end())
 253     {
 254     IPIter->GetUser()->GetProperty().ips.DelAfterNotifier(&(*IPIter));
 255     ChgIPNotifierList.erase(IPIter);
 256     }
 257 // ---          IP end          ---
 258 }
 259 //-----------------------------------------------------------------------------
 260 void PING::GetUsers()
 261 {
 262 STG_LOCKER lock(&mutex);
 263
 264 USER_PTR u;
 265 int h = users->OpenSearch();
 266 assert(h && "USERS::OpenSearch is always correct");
 267
 268 while (users->SearchNext(h, &u) == 0)
 269     {
 270     usersList.push_back(u);
 271     SetUserNotifiers(u);
 272     if (u->GetProperty().ips.ConstData().OnlyOneIP())
 273         {
 274         pinger.AddIP(u->GetProperty().ips.ConstData()[0].ip);
 275         }
 276     else
 277         {
 278         uint32_t ip = u->GetCurrIP();
 279         if (ip)
 280             pinger.AddIP(ip);
 281         }
 282     }
 283
 284 users->CloseSearch(h);
 285 }
 286 //-----------------------------------------------------------------------------
 287 void PING::AddUser(USER_PTR u)
 288 {
 289 STG_LOCKER lock(&mutex);
 290
 291 SetUserNotifiers(u);
 292 usersList.push_back(u);
 293 }
 294 //-----------------------------------------------------------------------------
 295 void PING::DelUser(USER_PTR u)
 296 {
 297 STG_LOCKER lock(&mutex);
 298
 299 UnSetUserNotifiers(u);
 300
 301 std::list<USER_PTR>::iterator users_iter;
 302 users_iter = usersList.begin();
 303
 304 while (users_iter != usersList.end())
 305     {
 306     if (u == *users_iter)
 307         {
 308         usersList.erase(users_iter);
 309         break;
 310         }
 311     ++users_iter;
 312     }
 313 }
 314 //-----------------------------------------------------------------------------
 315 void CHG_CURRIP_NOTIFIER_PING::Notify(const uint32_t & oldIP, const uint32_t & newIP)
 316 {
 317 ping.pinger.DelIP(oldIP);
 318 if (newIP)
 319     ping.pinger.AddIP(newIP);
 320 }
 321 //-----------------------------------------------------------------------------
 322 void CHG_IPS_NOTIFIER_PING::Notify(const USER_IPS & oldIPS, const USER_IPS & newIPS)
 323 {
 324 if (oldIPS.OnlyOneIP())
 325     ping.pinger.DelIP(oldIPS[0].ip);
 326
 327 if (newIPS.OnlyOneIP())
 328     ping.pinger.AddIP(newIPS[0].ip);
 329 }
 330 //-----------------------------------------------------------------------------
 331 void ADD_USER_NONIFIER_PING::Notify(const USER_PTR & user)
 332 {
 333 ping.AddUser(user);
 334 }
 335 //-----------------------------------------------------------------------------
 336 void DEL_USER_NONIFIER_PING::Notify(const USER_PTR & user)
 337 {
 338 ping.DelUser(user);
 339 }
 340 //-----------------------------------------------------------------------------