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