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)
355 //Write traff, reset session traff. Fake disconnect-connect
356 if (t->tm_hour == 23 && t->tm_min == 59)
358 printfd(__FILE__,"MidnightResetSessionStat\n");
359 for_each(users.begin(), users.end(), mem_fun_ref(&USER::MidnightResetSessionStat));
362 if (TimeToWriteDetailStat(t))
364 //printfd(__FILE__, "USER::WriteInetStat\n");
367 // ðÉÛÅÍ ÀÚÅÒÏ× ÞÁÓÔÑÍÉ. ÷ ÐÅÒÅÒÙ×ÁÈ ×ÙÚÙ×ÁÅÍ USER::Run
368 list<USER>::iterator usr = users.begin();
369 while (usr != users.end())
372 usr->WriteDetailStat();
374 if (usersCnt % 10 == 0)
375 for_each(users.begin(), users.end(), mem_fun_ref(&USER::Run));
381 //-----------------------------------------------------------------------------
382 void USERS::NewDay(const struct tm * t)
387 int dayFee = settings->GetDayFee();
390 dayFee = DaysInCurrentMonth();
392 printfd(__FILE__, "DayFee = %d\n", dayFee);
393 printfd(__FILE__, "Today = %d DayResetTraff = %d\n", t1->tm_mday, settings->GetDayResetTraff());
394 printfd(__FILE__, "DayFeeIsLastDay = %d\n", settings->GetDayFeeIsLastDay());
396 if (!settings->GetDayFeeIsLastDay())
398 printfd(__FILE__, "DayResetTraff - 1 -\n");
400 //printfd(__FILE__, "DayResetTraff - 1 - 1 -\n");
403 if (settings->GetSpreadFee())
405 printfd(__FILE__, "Spread DayFee\n");
406 for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessDayFeeSpread));
410 if (t->tm_mday == dayFee)
412 printfd(__FILE__, "DayFee\n");
413 for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessDayFee));
417 if (settings->GetDayFeeIsLastDay())
419 printfd(__FILE__, "DayResetTraff - 2 -\n");
423 //-----------------------------------------------------------------------------
424 void USERS::DayResetTraff(const struct tm * t1)
426 int dayResetTraff = settings->GetDayResetTraff();
427 if (dayResetTraff == 0)
428 dayResetTraff = DaysInCurrentMonth();
429 if (t1->tm_mday == dayResetTraff)
431 printfd(__FILE__, "ResetTraff\n");
432 for_each(users.begin(), users.end(), mem_fun_ref(&USER::ProcessNewMonth));
433 for_each(users.begin(), users.end(), mem_fun_ref(&USER::SetPrepaidTraff));
436 //-----------------------------------------------------------------------------
441 WriteServLog("USERS: Error: Cannot read users!");
446 if (pthread_create(&thread, NULL, Run, this))
448 WriteServLog("USERS: Error: Cannot start thread!");
453 //-----------------------------------------------------------------------------
456 printfd(__FILE__, "USERS::Stop()\n");
460 //printfd(__FILE__, "Alredy stopped\n");
466 //5 seconds to thread stops itself
468 for (i = 0; i < 25 * (users.size() / 50 + 1); i++)
476 //after 5 seconds waiting thread still running. now kill it
479 printfd(__FILE__, "kill USERS thread.\n");
480 //TODO pthread_cancel()
481 if (pthread_kill(thread, SIGINT))
483 //errorStr = "Cannot kill USERS thread.";
484 //printfd(__FILE__, "Cannot kill USERS thread.\n");
487 printfd(__FILE__, "USERS killed\n");
490 printfd(__FILE__, "Before USERS::Run()\n");
491 for_each(users.begin(), users.end(), mem_fun_ref(&USER::Run));
493 // 'cause bind2st accepts only constant first param
494 for (list<USER>::iterator it = users.begin();
497 it->WriteDetailStat(true);
499 for_each(users.begin(), users.end(), mem_fun_ref(&USER::WriteStat));
500 for_each(users.begin(), users.end(), mem_fun_ref(&USER::WriteConf));
502 printfd(__FILE__, "USERS::Stop()\n");
505 //-----------------------------------------------------------------------------
506 void USERS::RealDelUser()
508 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
510 printfd(__FILE__, "RealDelUser() users to del: %d\n", usersToDelete.size());
512 list<USER_TO_DEL>::iterator iter;
513 iter = usersToDelete.begin();
514 while (iter != usersToDelete.end())
516 printfd(__FILE__, "RealDelUser() user=%s\n", iter->iter->GetLogin().c_str());
517 if (iter->delTime + userDeleteDelayTime < stgTime)
519 printfd(__FILE__, "RealDelUser() user=%s removed from DB\n", iter->iter->GetLogin().c_str());
520 if (store->DelUser(iter->iter->GetLogin()))
522 WriteServLog("Error removing user \'%s\' from database.", iter->iter->GetLogin().c_str());
524 users.erase(iter->iter);
525 usersToDelete.erase(iter++);
534 //-----------------------------------------------------------------------------
535 int USERS::GetUserNum()
537 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
540 //-----------------------------------------------------------------------------
541 void USERS::AddToIPIdx(user_iter user)
543 printfd(__FILE__, "USERS: Add IP Idx\n");
544 uint32_t ip = user->GetCurrIP();
545 //assert(ip && "User has non-null ip");
547 return; // User has disconnected
549 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
551 const map<uint32_t, user_iter>::iterator it(
552 ipIndex.lower_bound(ip)
555 assert((it == ipIndex.end() || it->first != ip) && "User is not in index");
557 ipIndex.insert(it, std::make_pair(ip, user));
559 //-----------------------------------------------------------------------------
560 void USERS::DelFromIPIdx(uint32_t ip)
562 printfd(__FILE__, "USERS: Del IP Idx\n");
563 assert(ip && "User has non-null ip");
565 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
567 const map<uint32_t, user_iter>::iterator it(
571 //assert(it != ipIndex.end() && "User is in index");
572 if (it == ipIndex.end())
573 return; // User has not been added
577 //-----------------------------------------------------------------------------
578 int USERS::FindByIPIdx(uint32_t ip, user_iter * usr)
580 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
582 map<uint32_t, user_iter>::iterator it;
583 it = ipIndex.find(ip);
585 if (it == ipIndex.end())
587 //printfd(__FILE__, "User NOT found in IP_Index!!!\n");
591 //printfd(__FILE__, "User found in IP_Index\n");
594 //-----------------------------------------------------------------------------
595 void USERS::AddNotifierUserAdd(NOTIFIER_BASE<user_iter> * n)
597 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
598 onAddNotifiers.insert(n);
600 //-----------------------------------------------------------------------------
601 void USERS::DelNotifierUserAdd(NOTIFIER_BASE<user_iter> * n)
603 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
604 //printfd(__FILE__, "DelNotifierUserAdd\n");
605 onAddNotifiers.erase(n);
607 //-----------------------------------------------------------------------------
608 void USERS::AddNotifierUserDel(NOTIFIER_BASE<user_iter> * n)
610 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
611 onDelNotifiers.insert(n);
613 //-----------------------------------------------------------------------------
614 void USERS::DelNotifierUserDel(NOTIFIER_BASE<user_iter> * n)
616 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
617 onDelNotifiers.erase(n);
619 //-----------------------------------------------------------------------------
620 int USERS::OpenSearch()
622 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
624 searchDescriptors[handle] = users.begin();
627 //-----------------------------------------------------------------------------
628 int USERS::SearchNext(int h, user_iter * u)
630 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
632 if (searchDescriptors.find(h) == searchDescriptors.end())
634 WriteServLog("USERS. Incorrect search handle.");
638 if (searchDescriptors[h] == users.end())
641 while (searchDescriptors[h]->GetDeleted())
643 ++searchDescriptors[h];
644 if (searchDescriptors[h] == users.end())
650 *u = searchDescriptors[h];
652 ++searchDescriptors[h];
656 //-----------------------------------------------------------------------------
657 int USERS::CloseSearch(int h)
659 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
660 if (searchDescriptors.find(h) != searchDescriptors.end())
662 searchDescriptors.erase(searchDescriptors.find(h));
666 WriteServLog("USERS. Incorrect search handle.");
669 //-----------------------------------------------------------------------------
670 void USERS::SetUserNotifiers(user_iter user)
672 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
674 PROPERTY_NOTIFER_IP_BEFORE nb(*this, user);
675 PROPERTY_NOTIFER_IP_AFTER na(*this, user);
677 userIPNotifiersBefore.push_front(nb);
678 userIPNotifiersAfter.push_front(na);
680 user->AddCurrIPBeforeNotifier(&(*userIPNotifiersBefore.begin()));
681 user->AddCurrIPAfterNotifier(&(*userIPNotifiersAfter.begin()));
683 //-----------------------------------------------------------------------------
684 void USERS::UnSetUserNotifiers(user_iter user)
686 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
688 list<PROPERTY_NOTIFER_IP_BEFORE>::iterator bi;
689 list<PROPERTY_NOTIFER_IP_AFTER>::iterator ai;
691 bi = userIPNotifiersBefore.begin();
692 while (bi != userIPNotifiersBefore.end())
694 if (bi->GetUser() == user)
696 bi->GetUser()->DelCurrIPBeforeNotifier(&(*bi));
697 userIPNotifiersBefore.erase(bi);
698 //printfd(__FILE__, "Notifier Before removed. User %s\n", bi->GetUser()->GetLogin().c_str());
704 ai = userIPNotifiersAfter.begin();
705 while (ai != userIPNotifiersAfter.end())
707 if (ai->GetUser() == user)
709 ai->GetUser()->DelCurrIPAfterNotifier(&(*ai));
710 userIPNotifiersAfter.erase(ai);
711 //printfd(__FILE__, "Notifier After removed. User %s\n", ai->GetUser()->GetLogin().c_str());
717 //-----------------------------------------------------------------------------
718 void USERS::AddUserIntoIndexes(user_iter user)
720 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
721 loginIndex.insert(pair<string, user_iter>(user->GetLogin(), user));
723 //-----------------------------------------------------------------------------
724 void USERS::DelUserFromIndexes(user_iter user)
726 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
727 loginIndex.erase(user->GetLogin());
729 //-----------------------------------------------------------------------------
730 bool USERS::TimeToWriteDetailStat(const struct tm * t)
732 int statTime = settings->GetDetailStatWritePeriod();
741 if (t->tm_min % 30 == 0)
745 if (t->tm_min % 15 == 0)
749 if (t->tm_min % 10 == 0)