]> git.stg.codes - stg.git/blob - projects/stargazer/users_impl.cpp
Merge branch 'master' into full-month-stats
[stg.git] / projects / stargazer / users_impl.cpp
1 /*
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.
6  *
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.
11  *
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
15  */
16
17 /*
18  *    Date: 27.10.2002
19  */
20
21 /*
22  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
23  */
24
25 /*
26  $Revision: 1.61 $
27  $Date: 2010/09/13 05:56:42 $
28  $Author: faust $
29  */
30
31 #ifndef _GNU_SOURCE
32 #define _GNU_SOURCE
33 #endif
34
35 #include <pthread.h>
36 #include <csignal>
37 #include <cassert>
38 #include <algorithm>
39 #include <utility>
40 #include <string>
41 #include <vector>
42
43 #include "stg/settings.h"
44 #include "stg/common.h"
45 #include "users_impl.h"
46 #include "stg_timer.h"
47
48 using namespace std;
49
50 extern const volatile time_t stgTime;
51
52 //#define USERS_DEBUG 1
53
54 //-----------------------------------------------------------------------------
55 USERS_IMPL::USERS_IMPL(SETTINGS_IMPL * s, STORE * st, TARIFFS * t, const ADMIN * sa)
56     : USERS(),
57       users(),
58       usersToDelete(),
59       userIPNotifiersBefore(),
60       userIPNotifiersAfter(),
61       ipIndex(),
62       loginIndex(),
63       settings(s),
64       tariffs(t),
65       store(st),
66       sysAdmin(sa),
67       WriteServLog(GetStgLogger()),
68       nonstop(false),
69       isRunning(false),
70       mutex(),
71       thread(),
72       handle(0),
73       searchDescriptors(),
74       onAddNotifiers(),
75       onDelNotifiers(),
76       onAddNotifiersImpl(),
77       onDelNotifiersImpl()
78 {
79 pthread_mutexattr_t attr;
80 pthread_mutexattr_init(&attr);
81 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
82 pthread_mutex_init(&mutex, &attr);
83 }
84 //-----------------------------------------------------------------------------
85 USERS_IMPL::~USERS_IMPL()
86 {
87 pthread_mutex_destroy(&mutex);
88 }
89 //-----------------------------------------------------------------------------
90 int USERS_IMPL::FindByNameNonLock(const string & login, user_iter * user)
91 {
92 map<string, user_iter>::iterator iter;
93 iter = loginIndex.find(login);
94 if (iter != loginIndex.end())
95     {
96     if (user)
97         *user = iter->second;
98     return 0;
99     }
100 return -1;
101 }
102 //-----------------------------------------------------------------------------
103 int USERS_IMPL::FindByName(const string & login, USER_PTR * user)
104 {
105 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
106 user_iter u;
107 int res = FindByNameNonLock(login, &u);
108 if (res)
109     return -1;
110 *user = &(*u);
111 return 0;
112 }
113 //-----------------------------------------------------------------------------
114 bool USERS_IMPL::TariffInUse(const string & tariffName) const
115 {
116 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
117 list<USER_IMPL>::const_iterator iter;
118 iter = users.begin();
119 while (iter != users.end())
120     {
121     if (iter->GetProperty().tariffName.Get() == tariffName)
122         return true;
123     ++iter;
124     }
125 return false;
126 }
127 //-----------------------------------------------------------------------------
128 int USERS_IMPL::Add(const string & login, const ADMIN * admin)
129 {
130 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
131 const PRIV * priv = admin->GetPriv();
132
133 if (!priv->userAddDel)
134     {
135     WriteServLog("%s tried to add user \'%s\'. Access denied.",
136          admin->GetLogStr().c_str(), login.c_str());
137     /*errorStr = "Admin \'" + admin->GetLogin() +
138                "\': tried to add user \'" + ud->login + "\'. Access denied.";*/
139     return -1;
140     }
141
142 //////
143 if (store->AddUser(login))
144     {
145     //TODO
146     //WriteServLog("Admin \'%s\': tried to add user \'%s\'. Access denied.",
147     //     admin->GetLogin().c_str(), ud->login.c_str());
148     return -1;
149     }
150 //////
151
152 USER_IMPL u(settings, store, tariffs, sysAdmin, this);
153
154 /*struct tm * tms;
155 time_t t = stgTime;
156
157 tms = localtime(&t);
158
159 tms->tm_hour = 0;
160 tms->tm_min = 0;
161 tms->tm_sec = 0;
162
163 if (settings->GetDayResetTraff() > tms->tm_mday)
164     tms->tm_mon -= 1;
165
166 tms->tm_mday = settings->GetDayResetTraff();*/
167
168 u.SetLogin(login);
169
170 u.SetPassiveTimeAsNewUser();
171
172 u.WriteConf();
173 u.WriteStat();
174
175 WriteServLog("%s User \'%s\' added.",
176          admin->GetLogStr().c_str(), login.c_str());
177
178 u.OnAdd();
179
180 users.push_front(u);
181
182 AddUserIntoIndexes(users.begin());
183 SetUserNotifiers(users.begin());
184
185     {
186     // Fire all "on add" notifiers
187     set<NOTIFIER_BASE<USER_PTR> *>::iterator ni = onAddNotifiers.begin();
188     while (ni != onAddNotifiers.end())
189         {
190         (*ni)->Notify(&users.front());
191         ++ni;
192         }
193     }
194
195     {
196     // Fire all "on add" implementation notifiers
197     set<NOTIFIER_BASE<USER_IMPL_PTR> *>::iterator ni = onAddNotifiersImpl.begin();
198     while (ni != onAddNotifiersImpl.end())
199         {
200         (*ni)->Notify(&users.front());
201         ++ni;
202         }
203     }
204
205 return 0;
206 }
207 //-----------------------------------------------------------------------------
208 void USERS_IMPL::Del(const string & login, const ADMIN * admin)
209 {
210 const PRIV * priv = admin->GetPriv();
211 user_iter u;
212
213 if (!priv->userAddDel)
214     {
215     WriteServLog("%s tried to remove user \'%s\'. Access denied.",
216          admin->GetLogStr().c_str(), login.c_str());
217     return;
218     }
219
220
221     {
222     STG_LOCKER lock(&mutex, __FILE__, __LINE__);
223
224     if (FindByNameNonLock(login, &u))
225         {
226         WriteServLog("%s tried to delete user \'%s\': not found.",
227                      admin->GetLogStr().c_str(),
228                      login.c_str());
229         return;
230         }
231     }
232
233     {
234     set<NOTIFIER_BASE<USER_PTR> *>::iterator ni = onDelNotifiers.begin();
235     while (ni != onDelNotifiers.end())
236         {
237         (*ni)->Notify(&(*u));
238         ++ni;
239         }
240     }
241
242     {
243     set<NOTIFIER_BASE<USER_IMPL_PTR> *>::iterator ni = onDelNotifiersImpl.begin();
244     while (ni != onDelNotifiersImpl.end())
245         {
246         (*ni)->Notify(&(*u));
247         ++ni;
248         }
249     }
250
251     {
252     STG_LOCKER lock(&mutex, __FILE__, __LINE__);
253
254     u->OnDelete();
255     u->SetDeleted();
256
257     USER_TO_DEL utd;
258     utd.iter = u;
259     utd.delTime = stgTime;
260     usersToDelete.push_back(utd);
261
262     UnSetUserNotifiers(u);
263     DelUserFromIndexes(u);
264
265     WriteServLog("%s User \'%s\' deleted.",
266              admin->GetLogStr().c_str(), login.c_str());
267
268     }
269 }
270 //-----------------------------------------------------------------------------
271 int USERS_IMPL::ReadUsers()
272 {
273 vector<string> usersList;
274 usersList.clear();
275 if (store->GetUsersList(&usersList) < 0)
276     {
277     WriteServLog(store->GetStrError().c_str());
278     return -1;
279     }
280
281 user_iter ui;
282
283 for (unsigned int i = 0; i < usersList.size(); i++)
284     {
285     USER_IMPL u(settings, store, tariffs, sysAdmin, this);
286
287     u.SetLogin(usersList[i]);
288     users.push_front(u);
289     ui = users.begin();
290
291     AddUserIntoIndexes(ui);
292     SetUserNotifiers(ui);
293
294     if (ui->ReadConf() < 0)
295         return -1;
296
297     if (ui->ReadStat() < 0)
298         return -1;
299     }
300
301 return 0;
302 }
303 //-----------------------------------------------------------------------------
304 void * USERS_IMPL::Run(void * d)
305 {
306 printfd(__FILE__, "=====================| pid: %d |===================== \n", getpid());
307 USERS_IMPL * us = (USERS_IMPL*) d;
308
309 struct tm t;
310 time_t tt = stgTime;
311 localtime_r(&tt, &t);
312
313 int min = t.tm_min;
314 int day = t.tm_mday;
315
316 printfd(__FILE__,"Day = %d Min = %d\n", day, min);
317
318 time_t touchTime = stgTime - MONITOR_TIME_DELAY_SEC;
319 string monFile = us->settings->GetMonitorDir() + "/users_r";
320 printfd(__FILE__, "Monitor=%d file USERS %s\n", us->settings->GetMonitoring(), monFile.c_str());
321
322 us->isRunning = true;
323 while (us->nonstop)
324     {
325     //printfd(__FILE__,"New Minute. old = %02d current = %02d\n", min, t->tm_min);
326     //printfd(__FILE__,"New Day.    old = %2d current = %2d\n", day, t->tm_mday);
327
328     for_each(us->users.begin(), us->users.end(), mem_fun_ref(&USER_IMPL::Run));
329
330     tt = stgTime;
331     localtime_r(&tt, &t);
332
333     if (min != t.tm_min)
334         {
335         printfd(__FILE__,"Sec = %d\n", stgTime);
336         printfd(__FILE__,"New Minute. old = %d current = %d\n", min, t.tm_min);
337         min = t.tm_min;
338
339         us->NewMinute(t);
340         }
341
342     if (day != t.tm_mday)
343         {
344         printfd(__FILE__,"Sec = %d\n", stgTime);
345         printfd(__FILE__,"New Day. old = %d current = %d\n", day, t.tm_mday);
346         day = t.tm_mday;
347         us->NewDay(t);
348         }
349
350     if (us->settings->GetMonitoring() && (touchTime + MONITOR_TIME_DELAY_SEC <= stgTime))
351         {
352         //printfd(__FILE__, "Monitor=%d file TRAFFCOUNTER %s\n", tc->monitoring, monFile.c_str());
353         touchTime = stgTime;
354         TouchFile(monFile.c_str());
355         }
356
357     stgUsleep(100000);
358     } //while (us->nonstop)
359
360 user_iter ui = us->users.begin();
361 while (ui != us->users.end())
362     {
363     us->UnSetUserNotifiers(ui);
364     us->DelUserFromIndexes(ui);
365     ++ui;
366     }
367
368 list<USER_TO_DEL>::iterator iter;
369 iter = us->usersToDelete.begin();
370 while (iter != us->usersToDelete.end())
371     {
372     iter->delTime -= 2 * userDeleteDelayTime;
373     ++iter;
374     }
375 us->RealDelUser();
376
377 us->isRunning = false;
378
379 return NULL;
380 }
381 //-----------------------------------------------------------------------------
382 void USERS_IMPL::NewMinute(const struct tm & t)
383 {
384 //Write traff, reset session traff. Fake disconnect-connect
385 if (t.tm_hour == 23 && t.tm_min == 59)
386     {
387     printfd(__FILE__,"MidnightResetSessionStat\n");
388     for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::MidnightResetSessionStat));
389     }
390
391 if (TimeToWriteDetailStat(t))
392     {
393     //printfd(__FILE__, "USER::WriteInetStat\n");
394     int usersCnt = 0;
395
396     // ðÉÛÅÍ ÀÚÅÒÏ× ÞÁÓÔÑÍÉ. ÷ ÐÅÒÅÒÙ×ÁÈ ×ÙÚÙ×ÁÅÍ USER::Run
397     list<USER_IMPL>::iterator usr = users.begin();
398     while (usr != users.end())
399         {
400         usersCnt++;
401         usr->WriteDetailStat();
402         ++usr;
403         if (usersCnt % 10 == 0)
404             for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::Run));
405         }
406     }
407
408 RealDelUser();
409 }
410 //-----------------------------------------------------------------------------
411 void USERS_IMPL::NewDay(const struct tm & t)
412 {
413 struct tm t1;
414 time_t tt = stgTime;
415 localtime_r(&tt, &t1);
416 int dayFee = settings->GetDayFee();
417
418 if (dayFee == 0)
419     dayFee = DaysInCurrentMonth();
420
421 printfd(__FILE__, "DayFee = %d\n", dayFee);
422 printfd(__FILE__, "Today = %d DayResetTraff = %d\n", t1.tm_mday, settings->GetDayResetTraff());
423 printfd(__FILE__, "DayFeeIsLastDay = %d\n", settings->GetDayFeeIsLastDay());
424
425 if (!settings->GetDayFeeIsLastDay())
426     {
427     printfd(__FILE__, "DayResetTraff - 1 -\n");
428     DayResetTraff(t1);
429     //printfd(__FILE__, "DayResetTraff - 1 - 1 -\n");
430     }
431
432 if (settings->GetSpreadFee())
433     {
434     printfd(__FILE__, "Spread DayFee\n");
435     for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::ProcessDayFeeSpread));
436     }
437 else
438     {
439     if (t.tm_mday == dayFee)
440         {
441         printfd(__FILE__, "DayFee\n");
442         for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::ProcessDayFee));
443         }
444     }
445
446 if (settings->GetDayFeeIsLastDay())
447     {
448     printfd(__FILE__, "DayResetTraff - 2 -\n");
449     DayResetTraff(t1);
450     }
451 }
452 //-----------------------------------------------------------------------------
453 void USERS_IMPL::DayResetTraff(const struct tm & t1)
454 {
455 int dayResetTraff = settings->GetDayResetTraff();
456 if (dayResetTraff == 0)
457     dayResetTraff = DaysInCurrentMonth();
458 if (t1.tm_mday == dayResetTraff)
459     {
460     printfd(__FILE__, "ResetTraff\n");
461     for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::ProcessNewMonth));
462     //for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::SetPrepaidTraff));
463     }
464 }
465 //-----------------------------------------------------------------------------
466 int USERS_IMPL::Start()
467 {
468 if (ReadUsers())
469     {
470     WriteServLog("USERS: Error: Cannot read users!");
471     return -1;
472     }
473
474 nonstop = true;
475 if (pthread_create(&thread, NULL, Run, this))
476     {
477     WriteServLog("USERS: Error: Cannot start thread!");
478     return -1;
479     }
480 return 0;
481 }
482 //-----------------------------------------------------------------------------
483 int USERS_IMPL::Stop()
484 {
485 printfd(__FILE__, "USERS::Stop()\n");
486
487 if (!isRunning)
488     {
489     //printfd(__FILE__, "Alredy stopped\n");
490     return 0;
491     }
492
493 nonstop = false;
494
495 //5 seconds to thread stops itself
496 struct timespec ts = {0, 200000000};
497 for (size_t i = 0; i < 25 * (users.size() / 50 + 1); i++)
498     {
499     if (!isRunning)
500         break;
501
502     nanosleep(&ts, NULL);
503     }
504
505 //after 5 seconds waiting thread still running. now kill it
506 if (isRunning)
507     {
508     printfd(__FILE__, "kill USERS thread.\n");
509     //TODO pthread_cancel()
510     if (pthread_kill(thread, SIGINT))
511         {
512         //errorStr = "Cannot kill USERS thread.";
513         //printfd(__FILE__, "Cannot kill USERS thread.\n");
514         //return 0;
515         }
516     printfd(__FILE__, "USERS killed\n");
517     }
518
519 printfd(__FILE__, "Before USERS::Run()\n");
520 for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::Run));
521
522 // 'cause bind2st accepts only constant first param
523 for (list<USER_IMPL>::iterator it = users.begin();
524      it != users.end();
525      ++it)
526     it->WriteDetailStat(true);
527
528 for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::WriteStat));
529 //for_each(users.begin(), users.end(), mem_fun_ref(&USER_IMPL::WriteConf));
530
531 printfd(__FILE__, "USERS::Stop()\n");
532 return 0;
533 }
534 //-----------------------------------------------------------------------------
535 void USERS_IMPL::RealDelUser()
536 {
537 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
538
539 printfd(__FILE__, "RealDelUser() users to del: %d\n", usersToDelete.size());
540
541 list<USER_TO_DEL>::iterator iter;
542 iter = usersToDelete.begin();
543 while (iter != usersToDelete.end())
544     {
545     printfd(__FILE__, "RealDelUser() user=%s\n", iter->iter->GetLogin().c_str());
546     if (iter->delTime + userDeleteDelayTime < stgTime)
547         {
548         printfd(__FILE__, "RealDelUser() user=%s removed from DB\n", iter->iter->GetLogin().c_str());
549         if (store->DelUser(iter->iter->GetLogin()))
550             {
551             WriteServLog("Error removing user \'%s\' from database.", iter->iter->GetLogin().c_str());
552             }
553         users.erase(iter->iter);
554         usersToDelete.erase(iter++);
555         }
556     else
557         {
558         ++iter;
559         }
560     }
561 return;
562 }
563 //-----------------------------------------------------------------------------
564 int USERS_IMPL::GetUserNum() const
565 {
566 return users.size();
567 }
568 //-----------------------------------------------------------------------------
569 void USERS_IMPL::AddToIPIdx(user_iter user)
570 {
571 printfd(__FILE__, "USERS: Add IP Idx\n");
572 uint32_t ip = user->GetCurrIP();
573 //assert(ip && "User has non-null ip");
574 if (!ip)
575     return; // User has disconnected
576
577 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
578
579 const map<uint32_t, user_iter>::iterator it(
580         ipIndex.lower_bound(ip)
581 );
582
583 assert((it == ipIndex.end() || it->first != ip) && "User is not in index");
584
585 ipIndex.insert(it, std::make_pair(ip, user));
586 }
587 //-----------------------------------------------------------------------------
588 void USERS_IMPL::DelFromIPIdx(uint32_t ip)
589 {
590 printfd(__FILE__, "USERS: Del IP Idx\n");
591 assert(ip && "User has non-null ip");
592
593 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
594
595 const map<uint32_t, user_iter>::iterator it(
596         ipIndex.find(ip)
597 );
598
599 //assert(it != ipIndex.end() && "User is in index");
600 if (it == ipIndex.end())
601     return; // User has not been added
602
603 ipIndex.erase(it);
604 }
605 //-----------------------------------------------------------------------------
606 int USERS_IMPL::FindByIPIdx(uint32_t ip, USER_PTR * usr) const
607 {
608     USER_IMPL * ptr = NULL;
609     if (FindByIPIdx(ip, &ptr))
610         return -1;
611     *usr = ptr;
612     return 0;
613 }
614 //-----------------------------------------------------------------------------
615 int USERS_IMPL::FindByIPIdx(uint32_t ip, USER_IMPL ** usr) const
616 {
617 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
618
619 map<uint32_t, user_iter>::const_iterator it;
620 it = ipIndex.find(ip);
621
622 if (it == ipIndex.end())
623     {
624     //printfd(__FILE__, "User NOT found in IP_Index!!!\n");
625     return -1;
626     }
627 *usr = &(*it->second);
628 //printfd(__FILE__, "User found in IP_Index\n");
629 return 0;
630 }
631 //-----------------------------------------------------------------------------
632 bool USERS_IMPL::IsIPInIndex(uint32_t ip) const
633 {
634 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
635
636 map<uint32_t, user_iter>::const_iterator it;
637 it = ipIndex.find(ip);
638
639 return it != ipIndex.end();
640 }
641 //-----------------------------------------------------------------------------
642 void USERS_IMPL::AddNotifierUserAdd(NOTIFIER_BASE<USER_PTR> * n)
643 {
644 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
645 onAddNotifiers.insert(n);
646 }
647 //-----------------------------------------------------------------------------
648 void USERS_IMPL::DelNotifierUserAdd(NOTIFIER_BASE<USER_PTR> * n)
649 {
650 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
651 onAddNotifiers.erase(n);
652 }
653 //-----------------------------------------------------------------------------
654 void USERS_IMPL::AddNotifierUserDel(NOTIFIER_BASE<USER_PTR> * n)
655 {
656 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
657 onDelNotifiers.insert(n);
658 }
659 //-----------------------------------------------------------------------------
660 void USERS_IMPL::DelNotifierUserDel(NOTIFIER_BASE<USER_PTR> * n)
661 {
662 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
663 onDelNotifiers.erase(n);
664 }
665 //-----------------------------------------------------------------------------
666 void USERS_IMPL::AddNotifierUserAdd(NOTIFIER_BASE<USER_IMPL_PTR> * n)
667 {
668 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
669 onAddNotifiersImpl.insert(n);
670 }
671 //-----------------------------------------------------------------------------
672 void USERS_IMPL::DelNotifierUserAdd(NOTIFIER_BASE<USER_IMPL_PTR> * n)
673 {
674 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
675 onAddNotifiersImpl.erase(n);
676 }
677 //-----------------------------------------------------------------------------
678 void USERS_IMPL::AddNotifierUserDel(NOTIFIER_BASE<USER_IMPL_PTR> * n)
679 {
680 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
681 onDelNotifiersImpl.insert(n);
682 }
683 //-----------------------------------------------------------------------------
684 void USERS_IMPL::DelNotifierUserDel(NOTIFIER_BASE<USER_IMPL_PTR> * n)
685 {
686 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
687 onDelNotifiersImpl.erase(n);
688 }
689 //-----------------------------------------------------------------------------
690 int USERS_IMPL::OpenSearch()
691 {
692 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
693 handle++;
694 searchDescriptors[handle] = users.begin();
695 return handle;
696 }
697 //-----------------------------------------------------------------------------
698 int USERS_IMPL::SearchNext(int h, USER_PTR * user)
699 {
700     USER_IMPL * ptr = NULL;
701     if (SearchNext(h, &ptr))
702         return -1;
703     *user = ptr;
704     return 0;
705 }
706 //-----------------------------------------------------------------------------
707 int USERS_IMPL::SearchNext(int h, USER_IMPL ** user)
708 {
709 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
710
711 if (searchDescriptors.find(h) == searchDescriptors.end())
712     {
713     WriteServLog("USERS. Incorrect search handle.");
714     return -1;
715     }
716
717 if (searchDescriptors[h] == users.end())
718     return -1;
719
720 while (searchDescriptors[h]->GetDeleted())
721     {
722     ++searchDescriptors[h];
723     if (searchDescriptors[h] == users.end())
724         {
725         return -1;
726         }
727     }
728
729 *user = &(*searchDescriptors[h]);
730
731 ++searchDescriptors[h];
732
733 return 0;
734 }
735 //-----------------------------------------------------------------------------
736 int USERS_IMPL::CloseSearch(int h)
737 {
738 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
739 if (searchDescriptors.find(h) != searchDescriptors.end())
740     {
741     searchDescriptors.erase(searchDescriptors.find(h));
742     return 0;
743     }
744
745 WriteServLog("USERS. Incorrect search handle.");
746 return -1;
747 }
748 //-----------------------------------------------------------------------------
749 void USERS_IMPL::SetUserNotifiers(user_iter user)
750 {
751 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
752
753 PROPERTY_NOTIFER_IP_BEFORE nb(*this, user);
754 PROPERTY_NOTIFER_IP_AFTER  na(*this, user);
755
756 userIPNotifiersBefore.push_front(nb);
757 userIPNotifiersAfter.push_front(na);
758
759 user->AddCurrIPBeforeNotifier(&(*userIPNotifiersBefore.begin()));
760 user->AddCurrIPAfterNotifier(&(*userIPNotifiersAfter.begin()));
761 }
762 //-----------------------------------------------------------------------------
763 void USERS_IMPL::UnSetUserNotifiers(user_iter user)
764 {
765 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
766
767 list<PROPERTY_NOTIFER_IP_BEFORE>::iterator  bi;
768 list<PROPERTY_NOTIFER_IP_AFTER>::iterator   ai;
769
770 bi = userIPNotifiersBefore.begin();
771 while (bi != userIPNotifiersBefore.end())
772     {
773     if (bi->GetUser() == user)
774         {
775         bi->GetUser()->DelCurrIPBeforeNotifier(&(*bi));
776         userIPNotifiersBefore.erase(bi);
777         //printfd(__FILE__, "Notifier Before removed. User %s\n", bi->GetUser()->GetLogin().c_str());
778         break;
779         }
780     ++bi;
781     }
782
783 ai = userIPNotifiersAfter.begin();
784 while (ai != userIPNotifiersAfter.end())
785     {
786     if (ai->GetUser() == user)
787         {
788         ai->GetUser()->DelCurrIPAfterNotifier(&(*ai));
789         userIPNotifiersAfter.erase(ai);
790         //printfd(__FILE__, "Notifier After removed. User %s\n", ai->GetUser()->GetLogin().c_str());
791         break;
792         }
793     ++ai;
794     }
795 }
796 //-----------------------------------------------------------------------------
797 void USERS_IMPL::AddUserIntoIndexes(user_iter user)
798 {
799 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
800 loginIndex.insert(make_pair(user->GetLogin(), user));
801 }
802 //-----------------------------------------------------------------------------
803 void USERS_IMPL::DelUserFromIndexes(user_iter user)
804 {
805 STG_LOCKER lock(&mutex, __FILE__, __LINE__);
806 loginIndex.erase(user->GetLogin());
807 }
808 //-----------------------------------------------------------------------------
809 bool USERS_IMPL::TimeToWriteDetailStat(const struct tm & t)
810 {
811 int statTime = settings->GetDetailStatWritePeriod();
812
813 switch (statTime)
814     {
815     case dsPeriod_1:
816         if (t.tm_min == 0)
817             return true;
818         break;
819     case dsPeriod_1_2:
820         if (t.tm_min % 30 == 0)
821             return true;
822         break;
823     case dsPeriod_1_4:
824         if (t.tm_min % 15 == 0)
825             return true;
826         break;
827     case dsPeriod_1_6:
828         if (t.tm_min % 10 == 0)
829             return true;
830         break;
831     }
832 return false;
833 }