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/11/03 10:50:03 $
35 #include "user_impl.h"
36 #include "settings_impl.h"
37 #include "stg_timer.h"
39 #include "stg/users.h"
40 #include "stg/common.h"
41 #include "stg/scriptexecuter.h"
42 #include "stg/tariff.h"
43 #include "stg/tariffs.h"
44 #include "stg/admin.h"
54 #include <unistd.h> // access
56 #ifdef USE_ABSTRACT_SETTINGS
57 USER_IMPL::USER_IMPL(const SETTINGS * s,
64 property(s->GetScriptsDir()),
65 WriteServLog(GetStgLogger()),
69 connected(__connected),
72 lastIPForDisconnect(0),
79 authorizedModificationTime(0),
82 lastWriteDetailedStat(0),
86 lastCashAdd(property.lastCashAdd),
87 passiveTime(property.passiveTime),
88 lastCashAddTime(property.lastCashAddTime),
89 freeMb(property.freeMb),
90 lastActivityTime(property.lastActivityTime),
91 password(property.password),
92 passive(property.passive),
93 disabled(property.disabled),
94 disabledDetailStat(property.disabledDetailStat),
95 alwaysOnline(property.alwaysOnline),
96 tariffName(property.tariffName),
97 nextTariff(property.nextTariff),
98 address(property.address),
100 group(property.group),
101 email(property.email),
102 phone(property.phone),
103 realName(property.realName),
104 credit(property.credit),
105 creditExpire(property.creditExpire),
107 userdata0(property.userdata0),
108 userdata1(property.userdata1),
109 userdata2(property.userdata2),
110 userdata3(property.userdata3),
111 userdata4(property.userdata4),
112 userdata5(property.userdata5),
113 userdata6(property.userdata6),
114 userdata7(property.userdata7),
115 userdata8(property.userdata8),
116 userdata9(property.userdata9),
117 passiveNotifier(this),
118 tariffNotifier(this),
122 password = "*_EMPTY_PASSWORD_*";
123 tariffName = NO_TARIFF_NAME;
125 lastWriteStat = stgTime + random() % settings->GetStatWritePeriod();
126 lastWriteDetailedStat = stgTime;
128 property.tariffName.AddBeforeNotifier(&tariffNotifier);
129 property.passive.AddBeforeNotifier(&passiveNotifier);
130 property.disabled.AddAfterNotifier(&disabledNotifier);
131 property.cash.AddBeforeNotifier(&cashNotifier);
132 ips.AddAfterNotifier(&ipNotifier);
134 pthread_mutexattr_t attr;
135 pthread_mutexattr_init(&attr);
136 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
137 pthread_mutex_init(&mutex, &attr);
140 USER_IMPL::USER_IMPL(const SETTINGS_IMPL * s,
147 property(s->GetScriptsDir()),
148 WriteServLog(GetStgLogger()),
152 connected(__connected),
155 lastIPForDisconnect(0),
162 authorizedModificationTime(0),
165 lastWriteDetailedStat(0),
169 lastCashAdd(property.lastCashAdd),
170 passiveTime(property.passiveTime),
171 lastCashAddTime(property.lastCashAddTime),
172 freeMb(property.freeMb),
173 lastActivityTime(property.lastActivityTime),
174 password(property.password),
175 passive(property.passive),
176 disabled(property.disabled),
177 disabledDetailStat(property.disabledDetailStat),
178 alwaysOnline(property.alwaysOnline),
179 tariffName(property.tariffName),
180 nextTariff(property.nextTariff),
181 address(property.address),
183 group(property.group),
184 email(property.email),
185 phone(property.phone),
186 realName(property.realName),
187 credit(property.credit),
188 creditExpire(property.creditExpire),
190 userdata0(property.userdata0),
191 userdata1(property.userdata1),
192 userdata2(property.userdata2),
193 userdata3(property.userdata3),
194 userdata4(property.userdata4),
195 userdata5(property.userdata5),
196 userdata6(property.userdata6),
197 userdata7(property.userdata7),
198 userdata8(property.userdata8),
199 userdata9(property.userdata9),
200 passiveNotifier(this),
201 disabledNotifier(this),
202 tariffNotifier(this),
206 password = "*_EMPTY_PASSWORD_*";
207 tariffName = NO_TARIFF_NAME;
209 lastWriteStat = stgTime + random() % settings->GetStatWritePeriod();
210 lastWriteDetailedStat = stgTime;
212 property.tariffName.AddBeforeNotifier(&tariffNotifier);
213 property.passive.AddBeforeNotifier(&passiveNotifier);
214 property.disabled.AddAfterNotifier(&disabledNotifier);
215 property.cash.AddBeforeNotifier(&cashNotifier);
216 ips.AddAfterNotifier(&ipNotifier);
218 pthread_mutexattr_t attr;
219 pthread_mutexattr_init(&attr);
220 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
221 pthread_mutex_init(&mutex, &attr);
224 //-----------------------------------------------------------------------------
225 USER_IMPL::USER_IMPL(const USER_IMPL & u)
228 property(u.settings->GetScriptsDir()),
229 WriteServLog(GetStgLogger()),
234 connected(__connected),
235 userIDGenerator(u.userIDGenerator),
236 __currIP(u.__currIP),
238 lastIPForDisconnect(0),
239 pingTime(u.pingTime),
240 sysAdmin(u.sysAdmin),
244 traffStat(u.traffStat),
245 traffStatSaved(u.traffStatSaved),
246 settings(u.settings),
247 authorizedModificationTime(u.authorizedModificationTime),
248 messages(u.messages),
250 lastWriteStat(u.lastWriteStat),
251 lastWriteDetailedStat(u.lastWriteDetailedStat),
255 lastCashAdd(property.lastCashAdd),
256 passiveTime(property.passiveTime),
257 lastCashAddTime(property.lastCashAddTime),
258 freeMb(property.freeMb),
259 lastActivityTime(property.lastActivityTime),
260 password(property.password),
261 passive(property.passive),
262 disabled(property.disabled),
263 disabledDetailStat(property.disabledDetailStat),
264 alwaysOnline(property.alwaysOnline),
265 tariffName(property.tariffName),
266 nextTariff(property.nextTariff),
267 address(property.address),
269 group(property.group),
270 email(property.email),
271 phone(property.phone),
272 realName(property.realName),
273 credit(property.credit),
274 creditExpire(property.creditExpire),
276 userdata0(property.userdata0),
277 userdata1(property.userdata1),
278 userdata2(property.userdata2),
279 userdata3(property.userdata3),
280 userdata4(property.userdata4),
281 userdata5(property.userdata5),
282 userdata6(property.userdata6),
283 userdata7(property.userdata7),
284 userdata8(property.userdata8),
285 userdata9(property.userdata9),
288 passiveNotifier(this),
289 disabledNotifier(this),
290 tariffNotifier(this),
297 property.tariffName.AddBeforeNotifier(&tariffNotifier);
298 property.passive.AddBeforeNotifier(&passiveNotifier);
299 property.disabled.AddAfterNotifier(&disabledNotifier);
300 property.cash.AddBeforeNotifier(&cashNotifier);
301 ips.AddAfterNotifier(&ipNotifier);
303 property.SetProperties(u.property);
305 pthread_mutexattr_t attr;
306 pthread_mutexattr_init(&attr);
307 pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
308 pthread_mutex_init(&mutex, &attr);
310 //-----------------------------------------------------------------------------
311 USER_IMPL::~USER_IMPL()
313 property.tariffName.DelBeforeNotifier(&tariffNotifier);
314 property.passive.DelBeforeNotifier(&passiveNotifier);
315 property.disabled.DelAfterNotifier(&disabledNotifier);
316 property.cash.DelBeforeNotifier(&cashNotifier);
317 pthread_mutex_destroy(&mutex);
319 //-----------------------------------------------------------------------------
320 void USER_IMPL::SetLogin(const std::string & l)
322 STG_LOCKER lock(&mutex);
323 assert(login.empty() && "Login is already set");
325 id = userIDGenerator.GetNextID();
327 //-----------------------------------------------------------------------------
328 int USER_IMPL::ReadConf()
330 STG_LOCKER lock(&mutex);
333 if (store->RestoreUserConf(&conf, login))
335 WriteServLog("Cannot read conf for user %s.", login.c_str());
336 WriteServLog("%s", store->GetStrError().c_str());
337 printfd(__FILE__, "Cannot read conf for user %s.\n", login.c_str());
338 printfd(__FILE__, "%s\n", store->GetStrError().c_str());
342 property.SetConf(conf);
344 tariff = tariffs->FindByName(tariffName);
347 WriteServLog("Cannot read user %s. Tariff %s not exist.",
348 login.c_str(), property.tariffName.Get().c_str());
352 std::vector<STG_MSG_HDR> hdrsList;
354 if (store->GetMessageHdrs(&hdrsList, login))
356 printfd(__FILE__, "Error GetMessageHdrs %s\n", store->GetStrError().c_str());
357 WriteServLog("Cannot read user %s. Error reading message headers: %s.",
359 store->GetStrError().c_str());
363 std::vector<STG_MSG_HDR>::const_iterator it;
364 for (it = hdrsList.begin(); it != hdrsList.end(); ++it)
367 if (store->GetMessage(it->id, &msg, login) == 0)
369 messages.push_back(msg);
375 //-----------------------------------------------------------------------------
376 int USER_IMPL::ReadStat()
378 STG_LOCKER lock(&mutex);
381 if (store->RestoreUserStat(&stat, login))
383 WriteServLog("Cannot read stat for user %s.", login.c_str());
384 WriteServLog("%s", store->GetStrError().c_str());
385 printfd(__FILE__, "Cannot read stat for user %s.\n", login.c_str());
386 printfd(__FILE__, "%s\n", store->GetStrError().c_str());
390 property.SetStat(stat);
394 //-----------------------------------------------------------------------------
395 int USER_IMPL::WriteConf()
397 STG_LOCKER lock(&mutex);
398 USER_CONF conf(property.GetConf());
400 printfd(__FILE__, "USER::WriteConf()\n");
402 if (store->SaveUserConf(conf, login))
404 WriteServLog("Cannot write conf for user %s.", login.c_str());
405 WriteServLog("%s", store->GetStrError().c_str());
406 printfd(__FILE__, "Cannot write conf for user %s.\n", login.c_str());
407 printfd(__FILE__, "%s\n", store->GetStrError().c_str());
413 //-----------------------------------------------------------------------------
414 int USER_IMPL::WriteStat()
416 STG_LOCKER lock(&mutex);
417 USER_STAT stat(property.GetStat());
419 if (store->SaveUserStat(stat, login))
421 WriteServLog("Cannot write stat for user %s.", login.c_str());
422 WriteServLog("%s", store->GetStrError().c_str());
423 printfd(__FILE__, "Cannot write stat for user %s.\n", login.c_str());
424 printfd(__FILE__, "%s\n", store->GetStrError().c_str());
428 lastWriteStat = stgTime;
432 //-----------------------------------------------------------------------------
433 int USER_IMPL::WriteMonthStat()
435 STG_LOCKER lock(&mutex);
436 time_t tt = stgTime - 3600;
438 localtime_r(&tt, &t1);
440 USER_STAT stat(property.GetStat());
441 if (store->SaveMonthStat(stat, t1.tm_mon, t1.tm_year, login))
443 WriteServLog("Cannot write month stat for user %s.", login.c_str());
444 WriteServLog("%s", store->GetStrError().c_str());
445 printfd(__FILE__, "Cannot write month stat for user %s.\n", login.c_str());
446 printfd(__FILE__, "%s\n", store->GetStrError().c_str());
452 //-----------------------------------------------------------------------------
453 int USER_IMPL::Authorize(uint32_t ip, uint32_t dirs, const AUTH * auth)
455 STG_LOCKER lock(&mutex);
457 * Authorize user. It only means that user will be authorized. Nothing more.
458 * User can be connected or disconnected while authorized.
459 * Example: user is authorized but disconnected due to 0 money or blocking
463 * Prevent double authorization by identical authorizers
465 if (authorizedBy.find(auth) != authorizedBy.end())
471 for (int i = 0; i < DIR_NUM; i++)
473 enabledDirs[i] = dirs & (1 << i);
476 if (!authorizedBy.empty())
480 // We are already authorized, but with different IP address
481 errorStr = "User " + login + " already authorized with IP address " + inet_ntostring(ip);
486 if (!users->FindByIPIdx(ip, &u))
488 // Address is already present in IP-index
489 // If it's not our IP - throw an error
492 errorStr = "IP address " + inet_ntostring(ip) + " already in use";
499 if (users->IsIPInIndex(ip))
501 // Address is already present in IP-index
502 errorStr = "IP address " + inet_ntostring(ip) + " already in use";
506 if (ips.ConstData().IsIPInIPS(ip))
509 lastIPForDisconnect = currIP;
513 printfd(__FILE__, " user %s: ips = %s\n", login.c_str(), ips.ConstData().GetIpStr().c_str());
514 errorStr = "IP address " + inet_ntostring(ip) + " not belong user " + login;
519 if (authorizedBy.empty())
520 authorizedModificationTime = stgTime;
521 authorizedBy.insert(auth);
527 //-----------------------------------------------------------------------------
528 void USER_IMPL::Unauthorize(const AUTH * auth, const std::string & reason)
530 STG_LOCKER lock(&mutex);
532 * Authorizer tries to unauthorize user, that was not authorized by it
534 if (!authorizedBy.erase(auth))
537 if (authorizedBy.empty())
539 authorizedModificationTime = stgTime;
540 lastDisconnectReason = reason;
541 lastIPForDisconnect = currIP;
542 currIP = 0; // DelUser in traffcounter
546 //-----------------------------------------------------------------------------
547 bool USER_IMPL::IsAuthorizedBy(const AUTH * auth) const
549 STG_LOCKER lock(&mutex);
550 // Is this user authorized by specified authorizer?
551 return authorizedBy.find(auth) != authorizedBy.end();
553 //-----------------------------------------------------------------------------
554 std::vector<std::string> USER_IMPL::GetAuthorizers() const
556 std::vector<std::string> list;
557 std::transform(authorizedBy.begin(), authorizedBy.end(), std::back_inserter(list), std::mem_fun(&AUTH::GetVersion));
560 //-----------------------------------------------------------------------------
561 void USER_IMPL::Connect(bool fakeConnect)
564 * Connect user to Internet. This function is differ from Authorize() !!!
567 STG_LOCKER lock(&mutex);
571 std::string scriptOnConnect = settings->GetScriptsDir() + "/OnConnect";
573 if (access(scriptOnConnect.c_str(), X_OK) == 0)
575 char dirsStr[DIR_NUM + 1];
576 dirsStr[DIR_NUM] = 0;
577 for (int i = 0; i < DIR_NUM; i++)
579 dirsStr[i] = enabledDirs[i] ? '1' : '0';
582 std::string scriptOnConnectParams;
584 strprintf(&scriptOnConnectParams,
585 "%s \"%s\" \"%s\" \"%f\" \"%d\" \"%s\"",
586 scriptOnConnect.c_str(),
588 inet_ntostring(currIP).c_str(),
593 std::vector<std::string>::const_iterator it(settings->GetScriptParams().begin());
594 while (it != settings->GetScriptParams().end())
596 scriptOnConnectParams += " \"" + GetParamValue(it->c_str()) + "\"";
600 ScriptExec(scriptOnConnectParams.c_str());
604 WriteServLog("Script %s cannot be executed. File not found.", scriptOnConnect.c_str());
610 if (store->WriteUserConnect(login, currIP))
612 WriteServLog("Cannot write connect for user %s.", login.c_str());
613 WriteServLog("%s", store->GetStrError().c_str());
617 lastIPForDisconnect = currIP;
619 //-----------------------------------------------------------------------------
620 void USER_IMPL::Disconnect(bool fakeDisconnect, const std::string & reason)
623 * Disconnect user from Internet. This function is differ from UnAuthorize() !!!
626 STG_LOCKER lock(&mutex);
628 if (!lastIPForDisconnect)
630 printfd(__FILE__, "lastIPForDisconnect\n");
636 lastDisconnectReason = reason;
637 std::string scriptOnDisonnect = settings->GetScriptsDir() + "/OnDisconnect";
639 if (access(scriptOnDisonnect.c_str(), X_OK) == 0)
641 char dirsStr[DIR_NUM + 1];
642 dirsStr[DIR_NUM] = 0;
643 for (int i = 0; i < DIR_NUM; i++)
645 dirsStr[i] = enabledDirs[i] ? '1' : '0';
648 std::string scriptOnDisonnectParams;
649 strprintf(&scriptOnDisonnectParams,
650 "%s \"%s\" \"%s\" \"%f\" \"%d\" \"%s\"",
651 scriptOnDisonnect.c_str(),
653 inet_ntostring(lastIPForDisconnect).c_str(),
658 std::vector<std::string>::const_iterator it(settings->GetScriptParams().begin());
659 while (it != settings->GetScriptParams().end())
661 scriptOnDisonnectParams += " \"" + GetParamValue(it->c_str()) + "\"";
665 ScriptExec(scriptOnDisonnectParams.c_str());
669 WriteServLog("Script OnDisconnect cannot be executed. File not found.");
675 std::string reasonMessage(reason);
676 if (!lastDisconnectReason.empty())
677 reasonMessage += ": " + lastDisconnectReason;
679 if (store->WriteUserDisconnect(login, up, down, sessionUpload, sessionDownload,
680 cash, freeMb, reasonMessage))
682 WriteServLog("Cannot write disconnect for user %s.", login.c_str());
683 WriteServLog("%s", store->GetStrError().c_str());
687 lastIPForDisconnect = 0;
689 DIR_TRAFF zeroSesssion;
691 sessionUpload = zeroSesssion;
692 sessionDownload = zeroSesssion;
693 sessionUploadModTime = stgTime;
694 sessionDownloadModTime = stgTime;
696 //-----------------------------------------------------------------------------
697 void USER_IMPL::PrintUser() const
700 STG_LOCKER lock(&mutex);
701 std::cout << "============================================================" << std::endl;
702 std::cout << "id=" << id << std::endl;
703 std::cout << "login=" << login << std::endl;
704 std::cout << "password=" << password << std::endl;
705 std::cout << "passive=" << passive << std::endl;
706 std::cout << "disabled=" << disabled << std::endl;
707 std::cout << "disabledDetailStat=" << disabledDetailStat << std::endl;
708 std::cout << "alwaysOnline=" << alwaysOnline << std::endl;
709 std::cout << "tariffName=" << tariffName << std::endl;
710 std::cout << "address=" << address << std::endl;
711 std::cout << "phone=" << phone << std::endl;
712 std::cout << "email=" << email << std::endl;
713 std::cout << "note=" << note << std::endl;
714 std::cout << "realName=" <<realName << std::endl;
715 std::cout << "group=" << group << std::endl;
716 std::cout << "credit=" << credit << std::endl;
717 std::cout << "nextTariff=" << nextTariff << std::endl;
718 std::cout << "userdata0" << userdata0 << std::endl;
719 std::cout << "userdata1" << userdata1 << std::endl;
720 std::cout << "creditExpire=" << creditExpire << std::endl;
721 std::cout << "ips=" << ips << std::endl;
722 std::cout << "------------------------" << std::endl;
723 std::cout << "up=" << up << std::endl;
724 std::cout << "down=" << down << std::endl;
725 std::cout << "cash=" << cash << std::endl;
726 std::cout << "freeMb=" << freeMb << std::endl;
727 std::cout << "lastCashAdd=" << lastCashAdd << std::endl;
728 std::cout << "lastCashAddTime=" << lastCashAddTime << std::endl;
729 std::cout << "passiveTime=" << passiveTime << std::endl;
730 std::cout << "lastActivityTime=" << lastActivityTime << std::endl;
731 std::cout << "============================================================" << std::endl;
733 //-----------------------------------------------------------------------------
734 void USER_IMPL::Run()
736 STG_LOCKER lock(&mutex);
738 if (stgTime > static_cast<time_t>(lastWriteStat + settings->GetStatWritePeriod()))
740 printfd(__FILE__, "USER::WriteStat user=%s\n", GetLogin().c_str());
743 if (creditExpire.ConstData() && creditExpire.ConstData() < stgTime)
745 WriteServLog("User: %s. Credit expired.", login.c_str());
751 if (passive.ConstData()
752 && (stgTime % 30 == 0)
753 && (passiveTime.ModificationTime() != stgTime))
755 passiveTime = passiveTime + (stgTime - passiveTime.ModificationTime());
756 printfd(__FILE__, "===== %s: passiveTime=%d =====\n", login.c_str(), passiveTime.ConstData());
759 if (!authorizedBy.empty())
763 property.Stat().lastActivityTime = stgTime;
765 if (!connected && IsInetable())
769 if (connected && !IsInetable())
772 Disconnect(false, "disabled");
774 Disconnect(false, "passive");
776 Disconnect(false, "no cash");
779 if (stgTime - lastScanMessages > 10)
782 lastScanMessages = stgTime;
789 Disconnect(false, "not authorized");
794 //-----------------------------------------------------------------------------
795 void USER_IMPL::UpdatePingTime(time_t t)
797 STG_LOCKER lock(&mutex);
798 //printfd(__FILE__, "UpdatePingTime(%d) %s\n", t, login.c_str());
804 //-----------------------------------------------------------------------------
805 bool USER_IMPL::IsInetable()
807 //STG_LOCKER lock(&mutex);
809 if (disabled || passive)
812 if (settings->GetFreeMbAllowInet())
818 if (settings->GetShowFeeInCash() || tariff == NULL)
820 return (cash >= -credit);
823 return (cash - tariff->GetFee() >= -credit);
825 //-----------------------------------------------------------------------------
826 std::string USER_IMPL::GetEnabledDirs() const
828 //STG_LOCKER lock(&mutex);
830 std::string dirs = "";
831 for(int i = 0; i < DIR_NUM; i++)
832 dirs += enabledDirs[i] ? "1" : "0";
835 //-----------------------------------------------------------------------------
836 #ifdef TRAFF_STAT_WITH_PORTS
837 void USER_IMPL::AddTraffStatU(int dir, uint32_t ip, uint16_t port, uint32_t len)
839 void USER_IMPL::AddTraffStatU(int dir, uint32_t ip, uint32_t len)
842 STG_LOCKER lock(&mutex);
844 if (!connected || tariff == NULL)
850 int64_t traff = tariff->GetTraffByType(up.ConstData()[dir], down.ConstData()[dir]);
851 int64_t threshold = tariff->GetThreshold(dir) * 1024 * 1024;
855 int tt = tariff->GetTraffType();
856 if (tt == TRAFF_UP ||
857 tt == TRAFF_UP_DOWN ||
858 // Check NEW traff data
859 (tt == TRAFF_MAX && dt[dir] > down.ConstData()[dir]))
862 if (traff < threshold &&
863 traff + len >= threshold)
865 // cash = partBeforeThreshold * priceBeforeThreshold +
866 // partAfterThreshold * priceAfterThreshold
867 int64_t before = threshold - traff; // Chunk part before threshold
868 int64_t after = len - before; // Chunk part after threshold
869 dc = tariff->GetPriceWithTraffType(up.ConstData()[dir], // Traff before chunk
870 down.ConstData()[dir],
873 tariff->GetPriceWithTraffType(dt[dir], // Traff after chunk
874 down.ConstData()[dir],
880 dc = tariff->GetPriceWithTraffType(up.ConstData()[dir],
881 down.ConstData()[dir],
886 if (freeMb.ConstData() <= 0) // FreeMb is exhausted
888 else if (freeMb.ConstData() < dc) // FreeMb is partially exhausted
889 cost = dc - freeMb.ConstData();
891 // Direct access to internal data structures via friend-specifier
892 property.Stat().freeMb -= dc;
893 property.Stat().cash -= cost;
899 sessionUpload[dir] += len;
900 sessionUploadModTime = stgTime;
904 if (!settings->GetWriteFreeMbTraffCost() &&
905 freeMb.ConstData() >= 0)
908 #ifdef TRAFF_STAT_WITH_PORTS
909 IP_DIR_PAIR idp(ip, dir, port);
911 IP_DIR_PAIR idp(ip, dir);
914 std::map<IP_DIR_PAIR, STAT_NODE>::iterator lb;
915 lb = traffStat.lower_bound(idp);
916 if (lb == traffStat.end() || lb->first != idp)
920 STAT_NODE(len, 0, cost)));
924 lb->second.cash += cost;
925 lb->second.up += len;
928 //-----------------------------------------------------------------------------
929 #ifdef TRAFF_STAT_WITH_PORTS
930 void USER_IMPL::AddTraffStatD(int dir, uint32_t ip, uint16_t port, uint32_t len)
932 void USER_IMPL::AddTraffStatD(int dir, uint32_t ip, uint32_t len)
935 STG_LOCKER lock(&mutex);
937 if (!connected || tariff == NULL)
943 int64_t traff = tariff->GetTraffByType(up.ConstData()[dir], down.ConstData()[dir]);
944 int64_t threshold = tariff->GetThreshold(dir) * 1024 * 1024;
948 int tt = tariff->GetTraffType();
949 if (tt == TRAFF_DOWN ||
950 tt == TRAFF_UP_DOWN ||
951 // Check NEW traff data
952 (tt == TRAFF_MAX && up.ConstData()[dir] <= dt[dir]))
955 if (traff < threshold &&
956 traff + len >= threshold)
958 // cash = partBeforeThreshold * priceBeforeThreshold +
959 // partAfterThreshold * priceAfterThreshold
960 int64_t before = threshold - traff; // Chunk part before threshold
961 int64_t after = len - before; // Chunk part after threshold
962 dc = tariff->GetPriceWithTraffType(up.ConstData()[dir],
963 down.ConstData()[dir], // Traff before chunk
966 tariff->GetPriceWithTraffType(up.ConstData()[dir],
967 dt[dir], // Traff after chunk
973 dc = tariff->GetPriceWithTraffType(up.ConstData()[dir],
974 down.ConstData()[dir],
979 if (freeMb.ConstData() <= 0) // FreeMb is exhausted
981 else if (freeMb.ConstData() < dc) // FreeMb is partially exhausted
982 cost = dc - freeMb.ConstData();
984 property.Stat().freeMb -= dc;
985 property.Stat().cash -= cost;
991 sessionDownload[dir] += len;
992 sessionDownloadModTime = stgTime;
996 if (!settings->GetWriteFreeMbTraffCost() &&
997 freeMb.ConstData() >= 0)
1000 #ifdef TRAFF_STAT_WITH_PORTS
1001 IP_DIR_PAIR idp(ip, dir, port);
1003 IP_DIR_PAIR idp(ip, dir);
1006 std::map<IP_DIR_PAIR, STAT_NODE>::iterator lb;
1007 lb = traffStat.lower_bound(idp);
1008 if (lb == traffStat.end() || lb->first != idp)
1010 traffStat.insert(lb,
1012 STAT_NODE(0, len, cost)));
1016 lb->second.cash += cost;
1017 lb->second.down += len;
1020 //-----------------------------------------------------------------------------
1021 void USER_IMPL::AddCurrIPBeforeNotifier(CURR_IP_NOTIFIER * notifier)
1023 STG_LOCKER lock(&mutex);
1024 currIP.AddBeforeNotifier(notifier);
1026 //-----------------------------------------------------------------------------
1027 void USER_IMPL::DelCurrIPBeforeNotifier(const CURR_IP_NOTIFIER * notifier)
1029 STG_LOCKER lock(&mutex);
1030 currIP.DelBeforeNotifier(notifier);
1032 //-----------------------------------------------------------------------------
1033 void USER_IMPL::AddCurrIPAfterNotifier(CURR_IP_NOTIFIER * notifier)
1035 STG_LOCKER lock(&mutex);
1036 currIP.AddAfterNotifier(notifier);
1038 //-----------------------------------------------------------------------------
1039 void USER_IMPL::DelCurrIPAfterNotifier(const CURR_IP_NOTIFIER * notifier)
1041 STG_LOCKER lock(&mutex);
1042 currIP.DelAfterNotifier(notifier);
1044 //-----------------------------------------------------------------------------
1045 void USER_IMPL::AddConnectedBeforeNotifier(CONNECTED_NOTIFIER * notifier)
1047 STG_LOCKER lock(&mutex);
1048 connected.AddBeforeNotifier(notifier);
1050 //-----------------------------------------------------------------------------
1051 void USER_IMPL::DelConnectedBeforeNotifier(const CONNECTED_NOTIFIER * notifier)
1053 STG_LOCKER lock(&mutex);
1054 connected.DelBeforeNotifier(notifier);
1056 //-----------------------------------------------------------------------------
1057 void USER_IMPL::AddConnectedAfterNotifier(CONNECTED_NOTIFIER * notifier)
1059 STG_LOCKER lock(&mutex);
1060 connected.AddAfterNotifier(notifier);
1062 //-----------------------------------------------------------------------------
1063 void USER_IMPL::DelConnectedAfterNotifier(const CONNECTED_NOTIFIER * notifier)
1065 STG_LOCKER lock(&mutex);
1066 connected.DelAfterNotifier(notifier);
1068 //-----------------------------------------------------------------------------
1069 void USER_IMPL::OnAdd()
1071 STG_LOCKER lock(&mutex);
1073 std::string scriptOnAdd = settings->GetScriptsDir() + "/OnUserAdd";
1075 if (access(scriptOnAdd.c_str(), X_OK) == 0)
1077 std::string scriptOnAddParams;
1078 strprintf(&scriptOnAddParams,
1080 scriptOnAdd.c_str(),
1083 ScriptExec(scriptOnAddParams.c_str());
1087 WriteServLog("Script %s cannot be executed. File not found.", scriptOnAdd.c_str());
1090 //-----------------------------------------------------------------------------
1091 void USER_IMPL::OnDelete()
1093 STG_LOCKER lock(&mutex);
1095 std::string scriptOnDel = settings->GetScriptsDir() + "/OnUserDel";
1097 if (access(scriptOnDel.c_str(), X_OK) == 0)
1099 std::string scriptOnDelParams;
1100 strprintf(&scriptOnDelParams,
1102 scriptOnDel.c_str(),
1105 ScriptExec(scriptOnDelParams.c_str());
1109 WriteServLog("Script %s cannot be executed. File not found.", scriptOnDel.c_str());
1114 //-----------------------------------------------------------------------------
1115 int USER_IMPL::WriteDetailStat(bool hard)
1117 printfd(__FILE__, "USER::WriteDetailedStat() - saved size = %d\n", traffStatSaved.second.size());
1119 if (!traffStatSaved.second.empty())
1121 if (store->WriteDetailedStat(traffStatSaved.second, traffStatSaved.first, login))
1123 printfd(__FILE__, "USER::WriteDetailStat() - failed to write detail stat from queue\n");
1124 WriteServLog("Cannot write detail stat from queue (of size %d recs) for user %s.", traffStatSaved.second.size(), login.c_str());
1125 WriteServLog("%s", store->GetStrError().c_str());
1128 traffStatSaved.second.erase(traffStatSaved.second.begin(), traffStatSaved.second.end());
1134 STG_LOCKER lock(&mutex);
1138 printfd(__FILE__, "USER::WriteDetailedStat() - size = %d\n", ts.size());
1140 if (ts.size() && !disabledDetailStat)
1142 if (store->WriteDetailedStat(ts, lastWriteDetailedStat, login))
1144 printfd(__FILE__, "USER::WriteDetailStat() - failed to write current detail stat\n");
1145 WriteServLog("Cannot write detail stat for user %s.", login.c_str());
1146 WriteServLog("%s", store->GetStrError().c_str());
1149 printfd(__FILE__, "USER::WriteDetailStat() - pushing detail stat to queue\n");
1150 STG_LOCKER lock(&mutex);
1151 traffStatSaved.second.swap(ts);
1152 traffStatSaved.first = lastWriteDetailedStat;
1157 lastWriteDetailedStat = stgTime;
1160 //-----------------------------------------------------------------------------
1161 double USER_IMPL::GetPassiveTimePart() const
1163 STG_LOCKER lock(&mutex);
1165 static int daysInMonth[12] =
1166 {31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
1170 localtime_r(&t, &tms);
1172 time_t secMonth = daysInMonth[(tms.tm_mon + 11) % 12] * 24 * 3600; // Previous month
1174 if (tms.tm_year % 4 == 0 && tms.tm_mon == 1)
1177 secMonth += 24 * 3600;
1180 time_t dt = secMonth - passiveTime;
1185 return static_cast<double>(dt) / secMonth;
1187 //-----------------------------------------------------------------------------
1188 void USER_IMPL::SetPassiveTimeAsNewUser()
1190 STG_LOCKER lock(&mutex);
1194 localtime_r(&t, &tm);
1195 int daysCurrMon = DaysInCurrentMonth();
1196 double pt = tm.tm_mday - 1;
1199 passiveTime = static_cast<time_t>(pt * 24 * 3600 * daysCurrMon);
1201 //-----------------------------------------------------------------------------
1202 void USER_IMPL::MidnightResetSessionStat()
1204 STG_LOCKER lock(&mutex);
1208 Disconnect(true, "fake");
1212 //-----------------------------------------------------------------------------
1213 void USER_IMPL::ProcessNewMonth()
1215 STG_LOCKER lock(&mutex);
1219 Disconnect(true, "fake");
1221 DIR_TRAFF zeroTarff;
1234 if (nextTariff.ConstData() != "")
1237 nt = tariffs->FindByName(nextTariff);
1240 WriteServLog("Cannot change tariff for user %s. Tariff %s not exist.",
1241 login.c_str(), property.tariffName.Get().c_str());
1245 property.tariffName.Set(nextTariff, sysAdmin, login, store);
1252 //-----------------------------------------------------------------------------
1253 void USER_IMPL::ProcessDayFeeSpread()
1255 STG_LOCKER lock(&mutex);
1257 if (passive.ConstData() || tariff == NULL)
1260 if (tariff->GetPeriod() != TARIFF::MONTH)
1263 double fee = tariff->GetFee() / DaysInCurrentMonth();
1265 if (std::fabs(fee) < 1.0e-3)
1269 switch (settings->GetFeeChargeType())
1272 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1275 if (c + credit >= 0)
1276 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1279 if (c + credit >= fee)
1280 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1284 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1289 //-----------------------------------------------------------------------------
1290 void USER_IMPL::ProcessDayFee()
1292 STG_LOCKER lock(&mutex);
1297 if (tariff->GetPeriod() != TARIFF::MONTH)
1300 double passiveTimePart = 1.0;
1301 if (!settings->GetFullFee())
1303 passiveTimePart = GetPassiveTimePart();
1307 if (passive.ConstData())
1309 printfd(__FILE__, "Don't charge fee `cause we are passive\n");
1313 double fee = tariff->GetFee() * passiveTimePart;
1317 if (std::fabs(fee) < 1.0e-3)
1324 printfd(__FILE__, "login: %8s Cash=%f Credit=%f Fee=%f PassiveTimePart=%f fee=%f\n",
1331 switch (settings->GetFeeChargeType())
1334 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1338 if (c + credit >= 0)
1340 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1345 if (c + credit >= fee)
1347 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1354 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1360 //-----------------------------------------------------------------------------
1361 void USER_IMPL::ProcessDailyFee()
1363 STG_LOCKER lock(&mutex);
1365 if (passive.ConstData() || tariff == NULL)
1368 if (tariff->GetPeriod() != TARIFF::DAY)
1371 double fee = tariff->GetFee();
1377 switch (settings->GetFeeChargeType())
1380 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1383 if (c + credit >= 0)
1384 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1387 if (c + credit >= fee)
1388 property.cash.Set(c - fee, sysAdmin, login, store, "Subscriber fee charge");
1393 //-----------------------------------------------------------------------------
1394 void USER_IMPL::SetPrepaidTraff()
1397 property.freeMb.Set(tariff->GetFree(), sysAdmin, login, store, "Prepaid traffic");
1399 //-----------------------------------------------------------------------------
1400 int USER_IMPL::AddMessage(STG_MSG * msg)
1402 STG_LOCKER lock(&mutex);
1404 if (SendMessage(*msg))
1406 if (store->AddMessage(msg, login))
1408 errorStr = store->GetStrError();
1409 WriteServLog("Error adding message: '%s'", errorStr.c_str());
1410 printfd(__FILE__, "Error adding message: '%s'\n", errorStr.c_str());
1413 messages.push_back(*msg);
1417 if (msg->header.repeat > 0)
1419 msg->header.repeat--;
1421 //TODO: gcc v. 4.x generate ICE on x86_64
1422 msg->header.lastSendTime = static_cast<int>(time(NULL));
1424 msg->header.lastSendTime = static_cast<int>(stgTime);
1426 if (store->AddMessage(msg, login))
1428 errorStr = store->GetStrError();
1429 WriteServLog("Error adding repeatable message: '%s'", errorStr.c_str());
1430 printfd(__FILE__, "Error adding repeatable message: '%s'\n", errorStr.c_str());
1433 messages.push_back(*msg);
1438 //-----------------------------------------------------------------------------
1439 int USER_IMPL::SendMessage(STG_MSG & msg) const
1441 // No lock `cause we are already locked from caller
1443 std::set<const AUTH*>::iterator it(authorizedBy.begin());
1444 while (it != authorizedBy.end())
1446 if (!(*it++)->SendMessage(msg, currIP))
1452 //TODO: gcc v. 4.x generate ICE on x86_64
1453 msg.header.lastSendTime = static_cast<int>(time(NULL));
1455 msg.header.lastSendTime = static_cast<int>(stgTime);
1457 msg.header.repeat--;
1461 //-----------------------------------------------------------------------------
1462 void USER_IMPL::ScanMessage()
1464 // No lock `cause we are already locked from caller
1465 // We need not check for the authorizedBy `cause it has already checked by caller
1467 std::list<STG_MSG>::iterator it(messages.begin());
1468 while (it != messages.end())
1470 if (settings->GetMessageTimeout() > 0 &&
1471 difftime(stgTime, it->header.creationTime) > settings->GetMessageTimeout())
1474 if (store->DelMessage(it->header.id, login))
1476 WriteServLog("Error deleting message: '%s'", store->GetStrError().c_str());
1477 printfd(__FILE__, "Error deleting message: '%s'\n", store->GetStrError().c_str());
1479 messages.erase(it++);
1482 if (it->GetNextSendTime() <= stgTime)
1484 if (SendMessage(*it))
1486 // We need to check all messages in queue for timeout
1490 if (it->header.repeat < 0)
1492 if (store->DelMessage(it->header.id, login))
1494 WriteServLog("Error deleting message: '%s'", store->GetStrError().c_str());
1495 printfd(__FILE__, "Error deleting message: '%s'\n", store->GetStrError().c_str());
1497 messages.erase(it++);
1501 if (store->EditMessage(*it, login))
1503 WriteServLog("Error modifying message: '%s'", store->GetStrError().c_str());
1504 printfd(__FILE__, "Error modifying message: '%s'\n", store->GetStrError().c_str());
1515 //-----------------------------------------------------------------------------
1516 std::string USER_IMPL::GetParamValue(const std::string & name) const
1518 std::string lowerName = ToLower(name);
1519 if (lowerName == "id")
1521 std::ostringstream stream;
1523 return stream.str();
1525 if (lowerName == "login") return login;
1526 if (lowerName == "currip") return currIP.ToString();
1527 if (lowerName == "enableddirs") return GetEnabledDirs();
1528 if (lowerName == "tariff") return property.tariffName;
1529 if (property.Exists(lowerName))
1530 return property.GetPropertyValue(lowerName);
1533 WriteServLog("User’s parameter '%s' does not exist.", name.c_str());
1537 //-----------------------------------------------------------------------------
1538 //-----------------------------------------------------------------------------
1539 //-----------------------------------------------------------------------------
1540 void CHG_PASSIVE_NOTIFIER::Notify(const int & oldPassive, const int & newPassive)
1542 if (newPassive && !oldPassive && user->tariff != NULL)
1543 user->property.cash.Set(user->cash - user->tariff->GetPassiveCost(),
1549 //-----------------------------------------------------------------------------
1550 void CHG_DISABLED_NOTIFIER::Notify(const int & oldValue, const int & newValue)
1552 if (oldValue && !newValue && user->GetConnected())
1554 user->Disconnect(false, "disabled");
1556 else if (!oldValue && newValue && user->IsInetable())
1558 user->Connect(false);
1562 //-----------------------------------------------------------------------------
1563 void CHG_TARIFF_NOTIFIER::Notify(const std::string &, const std::string & newTariff)
1565 if (user->settings->GetReconnectOnTariffChange() && user->connected)
1566 user->Disconnect(false, "Change tariff");
1567 user->tariff = user->tariffs->FindByName(newTariff);
1568 if (user->settings->GetReconnectOnTariffChange() &&
1569 !user->authorizedBy.empty() &&
1571 user->Connect(false);
1573 //-----------------------------------------------------------------------------
1574 void CHG_CASH_NOTIFIER::Notify(const double & oldCash, const double & newCash)
1576 user->lastCashAddTime = *const_cast<time_t *>(&stgTime);
1577 user->lastCashAdd = newCash - oldCash;
1579 //-----------------------------------------------------------------------------
1580 void CHG_IPS_NOTIFIER::Notify(const USER_IPS & from, const USER_IPS & to)
1582 printfd(__FILE__, "Change IP from '%s' to '%s'\n", from.GetIpStr().c_str(), to.GetIpStr().c_str());
1583 if (user->connected)
1584 user->Disconnect(false, "Change IP");
1585 if (!user->authorizedBy.empty() && user->IsInetable())
1586 user->Connect(false);
1588 //-----------------------------------------------------------------------------