projects/stargazer/plugins/other/ping/ping.cpp

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