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 //-----------------------------------------------------------------------------