2  *    This program is free software; you can redistribute it and/or modify
 
   3  *    it under the terms of the GNU General Public License as published by
 
   4  *    the Free Software Foundation; either version 2 of the License, or
 
   5  *    (at your option) any later version.
 
   7  *    This program is distributed in the hope that it will be useful,
 
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
  10  *    GNU General Public License for more details.
 
  12  *    You should have received a copy of the GNU General Public License
 
  13  *    along with this program; if not, write to the Free Software
 
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
  22  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
 
  27  $Date: 2010/09/13 05:56:42 $
 
  47 #include "stg_timer.h"
 
  51 extern const volatile time_t stgTime;
 
  53 //#define USERS_DEBUG 1
 
  55 //-----------------------------------------------------------------------------
 
  56 USERS::USERS(SETTINGS * s, BASE_STORE * st, TARIFFS * t, const ADMIN & sa)
 
  59       userIPNotifiersBefore(),
 
  60       userIPNotifiersAfter(),
 
  67       WriteServLog(GetStgLogger()),
 
  77 pthread_mutexattr_t attr;
 
  78 pthread_mutexattr_init(&attr);
 
  79 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
 
  80 pthread_mutex_init(&mutex, &attr);
 
  82 //-----------------------------------------------------------------------------
 
  85 pthread_mutex_destroy(&mutex);
 
  87 //-----------------------------------------------------------------------------
 
  88 int USERS::FindByNameNonLock(const string & login, user_iter * user) const
 
  90 map<string, user_iter>::const_iterator iter;
 
  91 iter = loginIndex.find(login);
 
  92 if (iter != loginIndex.end())
 
 100 //-----------------------------------------------------------------------------
 
 101 int USERS::FindByName(const string & login, user_iter * user) const
 
 103 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 104 return FindByNameNonLock(login, user);
 
 106 //-----------------------------------------------------------------------------
 
 107 bool USERS::TariffInUse(const string & tariffName)
 
 109 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 110 list<USER>::iterator iter;
 
 111 iter = users.begin();
 
 112 while (iter != users.end())
 
 114     if (iter->property.tariffName.Get() == tariffName)
 
 120 //-----------------------------------------------------------------------------
 
 121 int USERS::Add(const string & login, const ADMIN & admin)
 
 123 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 124 const PRIV * priv = admin.GetPriv();
 
 126 if (!priv->userAddDel)
 
 128     WriteServLog("%s tried to add user \'%s\'. Access denied.",
 
 129          admin.GetLogStr().c_str(), login.c_str());
 
 130     /*errorStr = "Admin \'" + admin.GetLogin() +
 
 131                "\': tried to add user \'" + ud->login + "\'. Access denied.";*/
 
 136 if (store->AddUser(login))
 
 139     //WriteServLog("Admin \'%s\': tried to add user \'%s\'. Access denied.",
 
 140     //     admin.GetLogin().c_str(), ud->login.c_str());
 
 145 USER u(settings, store, tariffs, sysAdmin, &ipIndex);
 
 156 if (settings->GetDayResetTraff() > tms->tm_mday)
 
 159 tms->tm_mday = settings->GetDayResetTraff();
 
 163 u.SetPassiveTimeAsNewUser();
 
 168 WriteServLog("%s User \'%s\' added.",
 
 169          admin.GetLogStr().c_str(), login.c_str());
 
 175 AddUserIntoIndexes(users.begin());
 
 176 SetUserNotifiers(users.begin());
 
 178 // õ×ÅÄÏÍÌÑÅÍ ×ÓÅÈ ÖÅÌÁÀÝÉÈ, ÞÔÏ ÄÏÂÁ×ÌÅΠÎÏ×ÙÊ ÐÏÌØÚÏ×ÁÔÅÌØ
 
 179 set<NOTIFIER_BASE<user_iter> *>::iterator ni = onAddNotifiers.begin();
 
 180 while (ni != onAddNotifiers.end())
 
 182     (*ni)->Notify(users.begin());
 
 188 //-----------------------------------------------------------------------------
 
 189 void USERS::Del(const string & login, const ADMIN & admin)
 
 191 const PRIV * priv = admin.GetPriv();
 
 194 if (!priv->userAddDel)
 
 196     WriteServLog("%s tried to remove user \'%s\'. Access denied.",
 
 197          admin.GetLogStr().c_str(), login.c_str());
 
 203     STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 205     if (FindByNameNonLock(login, &u))
 
 207         WriteServLog("%s tried to delete user \'%s\': not found.",
 
 208                      admin.GetLogStr().c_str(),
 
 214 set<NOTIFIER_BASE<user_iter> *>::iterator ni = onDelNotifiers.begin();
 
 215 while (ni != onDelNotifiers.end())
 
 222     STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 229     utd.delTime = stgTime;
 
 230     usersToDelete.push_back(utd);
 
 232     UnSetUserNotifiers(u);
 
 233     DelUserFromIndexes(u);
 
 235     WriteServLog("%s User \'%s\' deleted.",
 
 236              admin.GetLogStr().c_str(), login.c_str());
 
 240 //-----------------------------------------------------------------------------
 
 241 int USERS::ReadUsers()
 
 243 vector<string> usersList;
 
 245 if (store->GetUsersList(&usersList) < 0)
 
 247     WriteServLog(store->GetStrError().c_str());
 
 253 for (unsigned int i = 0; i < usersList.size(); i++)
 
 255     USER u(settings, store, tariffs, sysAdmin, &ipIndex);
 
 257     u.SetLogin(usersList[i]);
 
 261     AddUserIntoIndexes(ui);
 
 262     SetUserNotifiers(ui);
 
 264     if (ui->ReadConf() < 0)
 
 267     if (ui->ReadStat() < 0)
 
 273 //-----------------------------------------------------------------------------
 
 274 void * USERS::Run(void * d)
 
 276 printfd(__FILE__, "=====================| pid: %d |===================== \n", getpid());
 
 277 USERS * us = (USERS*) d;
 
 284 int day = t->tm_mday;
 
 286 printfd(__FILE__,"Day = %d Min = %d\n", day, min);
 
 288 time_t touchTime = stgTime - MONITOR_TIME_DELAY_SEC;
 
 289 string monFile = us->settings->GetMonitorDir() + "/users_r";
 
 290 printfd(__FILE__, "Monitor=%d file USERS %s\n", us->settings->GetMonitoring(), monFile.c_str());
 
 292 us->isRunning = true;
 
 295     //printfd(__FILE__,"New Minute. old = %02d current = %02d\n", min, t->tm_min);
 
 296     //printfd(__FILE__,"New Day.    old = %2d current = %2d\n", day, t->tm_mday);
 
 298     for_each(us->users.begin(), us->users.end(), mem_fun_ref(&USER::Run));
 
 303     if (min != t->tm_min)
 
 305         printfd(__FILE__,"Sec = %d\n", stgTime);
 
 306         printfd(__FILE__,"New Minute. old = %d current = %d\n", min, t->tm_min);
 
 313     if (day != t->tm_mday)
 
 315         printfd(__FILE__,"Sec = %d\n", stgTime);
 
 316         printfd(__FILE__,"New Day. old = %d current = %d\n", day, t->tm_mday);
 
 321     if (us->settings->GetMonitoring() && (touchTime + MONITOR_TIME_DELAY_SEC <= stgTime))
 
 323         //printfd(__FILE__, "Monitor=%d file TRAFFCOUNTER %s\n", tc->monitoring, monFile.c_str());
 
 325         TouchFile(monFile.c_str());
 
 329     } //while (us->nonstop)
 
 331 user_iter ui = us->users.begin();
 
 332 while (ui != us->users.end())
 
 334     us->UnSetUserNotifiers(ui);
 
 335     us->DelUserFromIndexes(ui);
 
 339 list<USER_TO_DEL>::iterator iter;
 
 340 iter = us->usersToDelete.begin();
 
 341 while (iter != us->usersToDelete.end())
 
 343     iter->delTime -= 2 * userDeleteDelayTime;
 
 348 us->isRunning = false;
 
 352 //-----------------------------------------------------------------------------
 
 353 void USERS::NewMinute(const struct tm * t)
 
 356 list<USER>::iterator usr;
 
 358 //Write traff, reset session traff. Fake disconnect-connect
 
 359 if (t->tm_hour == 23 && t->tm_min == 59)
 
 361     printfd(__FILE__,"MidnightResetSessionStat\n");
 
 362     for_each(users.begin(), users.end(), mem_fun_ref(&USER::MidnightResetSessionStat));
 
 365 if (TimeToWriteDetailStat(t))
 
 367     //printfd(__FILE__, "USER::WriteInetStat\n");
 
 368     //for_each(users.begin(), users.end(), mem_fun_ref(&USER::SwapDetailStat));
 
 371     // ðÉÛÅÍ ÀÚÅÒÏ× ÞÁÓÔÑÍÉ. ÷ ÐÅÒÅÒÙ×ÁÈ ×ÙÚÙ×ÁÅÍ USER::Run
 
 373     while (usr != users.end())
 
 376         usr->WriteDetailStat();
 
 378         if (usersCnt % 10 == 0)
 
 379             for_each(users.begin(), users.end(), mem_fun_ref(&USER::Run));
 
 382     //for_each(users.begin(), users.end(), mem_fun_ref(&USER::ResetDetailStat));
 
 387 //-----------------------------------------------------------------------------
 
 388 void USERS::NewDay(const struct tm * t)
 
 393 int dayFee = settings->GetDayFee();
 
 396     dayFee = DaysInCurrentMonth();
 
 398 printfd(__FILE__, "DayFee = %d\n", dayFee);
 
 399 printfd(__FILE__, "Today = %d DayResetTraff = %d\n", t1->tm_mday, settings->GetDayResetTraff());
 
 400 printfd(__FILE__, "DayFeeIsLastDay = %d\n", settings->GetDayFeeIsLastDay());
 
 402 if (!settings->GetDayFeeIsLastDay())
 
 404     printfd(__FILE__, "DayResetTraff - 1 -\n");
 
 406     //printfd(__FILE__, "DayResetTraff - 1 - 1 -\n");
 
 409 if (settings->GetSpreadFee())
 
 411     printfd(__FILE__, "Spread DayFee\n");
 
 412     for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessDayFeeSpread));
 
 416     if (t->tm_mday == dayFee)
 
 418         printfd(__FILE__, "DayFee\n");
 
 419         for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessDayFee));
 
 423 if (settings->GetDayFeeIsLastDay())
 
 425     printfd(__FILE__, "DayResetTraff - 2 -\n");
 
 429 //-----------------------------------------------------------------------------
 
 430 void USERS::DayResetTraff(const struct tm * t1)
 
 432 int dayResetTraff = settings->GetDayResetTraff();
 
 433 if (dayResetTraff == 0)
 
 434     dayResetTraff = DaysInCurrentMonth();
 
 435 if (t1->tm_mday == dayResetTraff)
 
 437     printfd(__FILE__, "ResetTraff\n");
 
 438     for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessNewMonth));
 
 439     for_each(users.begin(), users.end(), mem_fun_ref(&USER::SetPrepaidTraff));
 
 442 //-----------------------------------------------------------------------------
 
 447     WriteServLog("USERS: Error: Cannot read users!");
 
 452 if (pthread_create(&thread, NULL, Run, this))
 
 454     WriteServLog("USERS: Error: Cannot start thread!");
 
 459 //-----------------------------------------------------------------------------
 
 462 printfd(__FILE__, "USERS::Stop()\n");
 
 466     //printfd(__FILE__, "Alredy stopped\n");
 
 472 //5 seconds to thread stops itself
 
 474 for (i = 0; i < 25 * (users.size() / 50 + 1); i++)
 
 482 //after 5 seconds waiting thread still running. now kill it
 
 485     printfd(__FILE__, "kill USERS thread.\n");
 
 486     //TODO pthread_cancel()
 
 487     if (pthread_kill(thread, SIGINT))
 
 489         //errorStr = "Cannot kill USERS thread.";
 
 490         //printfd(__FILE__, "Cannot kill USERS thread.\n");
 
 493     printfd(__FILE__, "USERS killed\n");
 
 496 printfd(__FILE__, "Before USERS::Run()\n");
 
 497 for_each(users.begin(), users.end(), mem_fun_ref(&USER::Run));
 
 498 //for_each(users.begin(), users.end(), mem_fun_ref(&USER::SwapDetailStat));
 
 499 for_each(users.begin(), users.end(), bind2nd(mem_fun_ref(&USER::WriteDetailStat), true));
 
 500 for_each(users.begin(), users.end(), mem_fun_ref(&USER::WriteStat));
 
 501 for_each(users.begin(), users.end(), mem_fun_ref(&USER::WriteConf));
 
 503 printfd(__FILE__, "USERS::Stop()\n");
 
 506 //-----------------------------------------------------------------------------
 
 507 void USERS::RealDelUser()
 
 509 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 511 printfd(__FILE__, "RealDelUser() users to del: %d\n", usersToDelete.size());
 
 513 list<USER_TO_DEL>::iterator iter;
 
 514 iter = usersToDelete.begin();
 
 515 while (iter != usersToDelete.end())
 
 517     printfd(__FILE__, "RealDelUser() user=%s\n", iter->iter->GetLogin().c_str());
 
 518     if (iter->delTime + userDeleteDelayTime < stgTime)
 
 520         printfd(__FILE__, "RealDelUser() user=%s removed from DB\n", iter->iter->GetLogin().c_str());
 
 521         if (store->DelUser(iter->iter->GetLogin()))
 
 523             WriteServLog("Error removing user \'%s\' from database.", iter->iter->GetLogin().c_str());
 
 525         users.erase(iter->iter);
 
 526         usersToDelete.erase(iter++);
 
 535 //-----------------------------------------------------------------------------
 
 536 int USERS::GetUserNum()
 
 538 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 541 //-----------------------------------------------------------------------------
 
 542 void USERS::AddToIPIdx(user_iter user)
 
 544 printfd(__FILE__, "USERS: Add IP Idx\n");
 
 545 uint32_t ip = user->GetCurrIP();
 
 546 //assert(ip && "User has non-null ip");
 
 548     return; // User has disconnected
 
 550 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 552 const map<uint32_t, user_iter>::iterator it(
 
 553         ipIndex.lower_bound(ip)
 
 556 assert((it == ipIndex.end() || it->first != ip) && "User is not in index");
 
 558 ipIndex.insert(it, std::make_pair(ip, user));
 
 560 //-----------------------------------------------------------------------------
 
 561 void USERS::DelFromIPIdx(uint32_t ip)
 
 563 printfd(__FILE__, "USERS: Del IP Idx\n");
 
 564 assert(ip && "User has non-null ip");
 
 566 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 568 const map<uint32_t, user_iter>::iterator it(
 
 572 //assert(it != ipIndex.end() && "User is in index");
 
 573 if (it == ipIndex.end())
 
 574     return; // User has not been added
 
 578 //-----------------------------------------------------------------------------
 
 579 int USERS::FindByIPIdx(uint32_t ip, user_iter * usr)
 
 581 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 583 map<uint32_t, user_iter>::iterator it;
 
 584 it = ipIndex.find(ip);
 
 586 if (it == ipIndex.end())
 
 588     //printfd(__FILE__, "User NOT found in IP_Index!!!\n");
 
 592 //printfd(__FILE__, "User found in IP_Index\n");
 
 595 //-----------------------------------------------------------------------------
 
 596 void USERS::AddNotifierUserAdd(NOTIFIER_BASE<user_iter> * n)
 
 598 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 599 onAddNotifiers.insert(n);
 
 601 //-----------------------------------------------------------------------------
 
 602 void USERS::DelNotifierUserAdd(NOTIFIER_BASE<user_iter> * n)
 
 604 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 605 //printfd(__FILE__, "DelNotifierUserAdd\n");
 
 606 onAddNotifiers.erase(n);
 
 608 //-----------------------------------------------------------------------------
 
 609 void USERS::AddNotifierUserDel(NOTIFIER_BASE<user_iter> * n)
 
 611 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 612 onDelNotifiers.insert(n);
 
 614 //-----------------------------------------------------------------------------
 
 615 void USERS::DelNotifierUserDel(NOTIFIER_BASE<user_iter> * n)
 
 617 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 618 onDelNotifiers.erase(n);
 
 620 //-----------------------------------------------------------------------------
 
 621 int USERS::OpenSearch()
 
 623 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 625 searchDescriptors[handle] = users.begin();
 
 628 //-----------------------------------------------------------------------------
 
 629 int USERS::SearchNext(int h, user_iter * u)
 
 631 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 633 if (searchDescriptors.find(h) == searchDescriptors.end())
 
 635     WriteServLog("USERS. Incorrect search handle.");
 
 639 if (searchDescriptors[h] == users.end())
 
 642 while (searchDescriptors[h]->GetDeleted())
 
 644     ++searchDescriptors[h];
 
 645     if (searchDescriptors[h] == users.end())
 
 651 *u = searchDescriptors[h];
 
 653 ++searchDescriptors[h];
 
 657 //-----------------------------------------------------------------------------
 
 658 int USERS::CloseSearch(int h)
 
 660 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 661 if (searchDescriptors.find(h) != searchDescriptors.end())
 
 663     searchDescriptors.erase(searchDescriptors.find(h));
 
 667 WriteServLog("USERS. Incorrect search handle.");
 
 670 //-----------------------------------------------------------------------------
 
 671 void USERS::SetUserNotifiers(user_iter user)
 
 673 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 675 PROPERTY_NOTIFER_IP_BEFORE nb(*this, user);
 
 676 PROPERTY_NOTIFER_IP_AFTER  na(*this, user);
 
 678 userIPNotifiersBefore.push_front(nb);
 
 679 userIPNotifiersAfter.push_front(na);
 
 681 user->AddCurrIPBeforeNotifier(&(*userIPNotifiersBefore.begin()));
 
 682 user->AddCurrIPAfterNotifier(&(*userIPNotifiersAfter.begin()));
 
 684 //-----------------------------------------------------------------------------
 
 685 void USERS::UnSetUserNotifiers(user_iter user)
 
 687 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 689 list<PROPERTY_NOTIFER_IP_BEFORE>::iterator  bi;
 
 690 list<PROPERTY_NOTIFER_IP_AFTER>::iterator   ai;
 
 692 bi = userIPNotifiersBefore.begin();
 
 693 while (bi != userIPNotifiersBefore.end())
 
 695     if (bi->GetUser() == user)
 
 697         bi->GetUser()->DelCurrIPBeforeNotifier(&(*bi));
 
 698         userIPNotifiersBefore.erase(bi);
 
 699         //printfd(__FILE__, "Notifier Before removed. User %s\n", bi->GetUser()->GetLogin().c_str());
 
 705 ai = userIPNotifiersAfter.begin();
 
 706 while (ai != userIPNotifiersAfter.end())
 
 708     if (ai->GetUser() == user)
 
 710         ai->GetUser()->DelCurrIPAfterNotifier(&(*ai));
 
 711         userIPNotifiersAfter.erase(ai);
 
 712         //printfd(__FILE__, "Notifier After removed. User %s\n", ai->GetUser()->GetLogin().c_str());
 
 718 //-----------------------------------------------------------------------------
 
 719 void USERS::AddUserIntoIndexes(user_iter user)
 
 721 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 722 loginIndex.insert(pair<string, user_iter>(user->GetLogin(), user));
 
 724 //-----------------------------------------------------------------------------
 
 725 void USERS::DelUserFromIndexes(user_iter user)
 
 727 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
 
 728 loginIndex.erase(user->GetLogin());
 
 730 //-----------------------------------------------------------------------------
 
 731 bool USERS::TimeToWriteDetailStat(const struct tm * t)
 
 733 int statTime = settings->GetDetailStatWritePeriod();
 
 742         if (t->tm_min % 30 == 0)
 
 746         if (t->tm_min % 15 == 0)
 
 750         if (t->tm_min % 10 == 0)
 
 756 //-----------------------------------------------------------------------------
 
 757 /*int USERS::SendMessage(const string & login,
 
 761                        const string & text) const
 
 765 //-----------------------------------------------------------------------------