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
18 * Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
23 $Date: 2010/11/03 10:50:03 $
37 #include "stg_const.h"
38 #include "user_stat.h"
39 #include "user_conf.h"
41 #include "user_property.h"
43 #include "stg_message.h"
44 #include "noncopyable.h"
46 //-----------------------------------------------------------------------------
52 //-----------------------------------------------------------------------------
53 class USER_ID_GENERATOR {
54 friend class USER_IMPL;
56 USER_ID_GENERATOR() {}
57 int GetNextID() { static int id = 0; return id++; }
59 //-----------------------------------------------------------------------------
60 class CHG_PASSIVE_NOTIFIER : public PROPERTY_NOTIFIER_BASE<int>,
63 CHG_PASSIVE_NOTIFIER(USER_IMPL * u) : user(u) {}
64 void Notify(const int & oldPassive, const int & newPassive);
69 //-----------------------------------------------------------------------------
70 class CHG_TARIFF_NOTIFIER : public PROPERTY_NOTIFIER_BASE<std::string>,
73 CHG_TARIFF_NOTIFIER(USER_IMPL * u) : user(u) {}
74 void Notify(const std::string & oldTariff, const std::string & newTariff);
79 //-----------------------------------------------------------------------------
80 class CHG_CASH_NOTIFIER : public PROPERTY_NOTIFIER_BASE<double>,
83 CHG_CASH_NOTIFIER(USER_IMPL * u) : user(u) {}
84 void Notify(const double & oldCash, const double & newCash);
89 //-----------------------------------------------------------------------------
90 class CHG_IP_NOTIFIER : public PROPERTY_NOTIFIER_BASE<uint32_t>,
93 CHG_IP_NOTIFIER(USER_IMPL * u) : user(u) {}
94 void Notify(const uint32_t & oldCash, const uint32_t & newCash);
99 //-----------------------------------------------------------------------------
100 class USER_IMPL : public USER {
101 friend class CHG_PASSIVE_NOTIFIER;
102 friend class CHG_TARIFF_NOTIFIER;
103 friend class CHG_CASH_NOTIFIER;
104 friend class CHG_IP_NOTIFIER;
106 USER_IMPL(const SETTINGS_IMPL * settings,
108 const TARIFFS * tariffs,
109 const ADMIN * sysAdmin,
111 USER_IMPL(const USER_IMPL & u);
112 virtual ~USER_IMPL();
118 int WriteMonthStat();
120 std::string const & GetLogin() const { return login; }
121 void SetLogin(std::string const & l);
123 uint32_t GetCurrIP() const { return currIP; }
124 time_t GetCurrIPModificationTime() const { return currIP.ModificationTime(); }
126 void AddCurrIPBeforeNotifier(PROPERTY_NOTIFIER_BASE<uint32_t> *);
127 void DelCurrIPBeforeNotifier(PROPERTY_NOTIFIER_BASE<uint32_t> *);
129 void AddCurrIPAfterNotifier(PROPERTY_NOTIFIER_BASE<uint32_t> *);
130 void DelCurrIPAfterNotifier(PROPERTY_NOTIFIER_BASE<uint32_t> *);
132 int GetID() const { return id; }
134 double GetPassiveTimePart() const;
135 void ResetPassiveTime() { passiveTime = 0; }
136 void SetPassiveTimeAsNewUser();
138 int WriteDetailStat(bool hard = false);
140 const TARIFF * GetTariff() const { return tariff; }
141 void ResetNextTariff() { nextTariff = ""; }
143 #ifdef TRAFF_STAT_WITH_PORTS
144 void AddTraffStatU(int dir, uint32_t ip, uint16_t port, uint32_t len);
145 void AddTraffStatD(int dir, uint32_t ip, uint16_t port, uint32_t len);
147 void AddTraffStatU(int dir, uint32_t ip, uint32_t len);
148 void AddTraffStatD(int dir, uint32_t ip, uint32_t len);
151 const DIR_TRAFF & GetSessionUpload() const { return sessionUpload; }
152 const DIR_TRAFF & GetSessionDownload() const { return sessionDownload; }
154 bool GetConnected() const { return connected; }
155 time_t GetConnectedModificationTime() const { return connected.ModificationTime(); }
156 int GetAuthorized() const { return authorizedBy.size(); }
157 int Authorize(uint32_t ip, uint32_t enabledDirs, const AUTH * auth);
158 void Unauthorize(const AUTH * auth);
159 bool IsAuthorizedBy(const AUTH * auth) const;
161 int AddMessage(STG_MSG * msg);
163 void UpdatePingTime(time_t t = 0);
164 time_t GetPingTime() const { return pingTime; }
166 void PrintUser() const;
169 const std::string & GetStrError() const { return errorStr; }
171 USER_PROPERTIES & GetProperty() { return property; };
172 const USER_PROPERTIES & GetProperty() const { return property; };
174 void SetDeleted() { deleted = true; }
175 bool GetDeleted() const { return deleted; }
177 time_t GetLastWriteStatTime() const { return lastWriteStat; }
179 void MidnightResetSessionStat();
180 void ProcessDayFee();
181 void SetPrepaidTraff();
182 void ProcessDayFeeSpread();
183 void ProcessNewMonth();
186 std::string GetEnabledDirs();
193 USER_PROPERTIES property;
194 STG_LOGGER & WriteServLog;
196 void Connect(bool fakeConnect = false);
197 void Disconnect(bool fakeDisconnect, const std::string & reason);
198 int SaveMonthStat(int month, int year);
200 int SendMessage(STG_MSG & msg) const;
203 time_t lastScanMessages;
208 USER_PROPERTY<bool> connected;
210 bool enabledDirs[DIR_NUM];
212 USER_ID_GENERATOR userIDGenerator;
214 uint32_t __currIP; // Current user's ip
215 USER_PROPERTY<uint32_t> currIP;
217 uint32_t lastIPForDisconnect; // User's ip after unauth but before disconnect
221 const ADMIN * sysAdmin;
224 const TARIFFS * tariffs;
225 const TARIFF * tariff;
227 TRAFF_STAT traffStat;
228 std::pair<time_t, TRAFF_STAT> traffStatSaved;
230 const SETTINGS_IMPL * settings;
232 std::set<const AUTH *> authorizedBy;
234 std::list<STG_MSG> messages;
238 time_t lastWriteStat;
239 time_t lastWriteDetailedStat;
242 USER_PROPERTY<double> & cash;
243 USER_PROPERTY<DIR_TRAFF> & up;
244 USER_PROPERTY<DIR_TRAFF> & down;
245 USER_PROPERTY<double> & lastCashAdd;
246 USER_PROPERTY<time_t> & passiveTime;
247 USER_PROPERTY<time_t> & lastCashAddTime;
248 USER_PROPERTY<double> & freeMb;
249 USER_PROPERTY<time_t> & lastActivityTime;
250 USER_PROPERTY<std::string> & password;
251 USER_PROPERTY<int> & passive;
252 USER_PROPERTY<int> & disabled;
253 USER_PROPERTY<int> & disabledDetailStat;
254 USER_PROPERTY<int> & alwaysOnline;
255 USER_PROPERTY<std::string> & tariffName;
256 USER_PROPERTY<std::string> & nextTariff;
257 USER_PROPERTY<std::string> & address;
258 USER_PROPERTY<std::string> & note;
259 USER_PROPERTY<std::string> & group;
260 USER_PROPERTY<std::string> & email;
261 USER_PROPERTY<std::string> & phone;
262 USER_PROPERTY<std::string> & realName;
263 USER_PROPERTY<double> & credit;
264 USER_PROPERTY<time_t> & creditExpire;
265 USER_PROPERTY<USER_IPS> & ips;
266 USER_PROPERTY<std::string> & userdata0;
267 USER_PROPERTY<std::string> & userdata1;
268 USER_PROPERTY<std::string> & userdata2;
269 USER_PROPERTY<std::string> & userdata3;
270 USER_PROPERTY<std::string> & userdata4;
271 USER_PROPERTY<std::string> & userdata5;
272 USER_PROPERTY<std::string> & userdata6;
273 USER_PROPERTY<std::string> & userdata7;
274 USER_PROPERTY<std::string> & userdata8;
275 USER_PROPERTY<std::string> & userdata9;
279 DIR_TRAFF sessionUpload;
280 DIR_TRAFF sessionDownload;
282 CHG_PASSIVE_NOTIFIER passiveNotifier;
283 CHG_TARIFF_NOTIFIER tariffNotifier;
284 CHG_CASH_NOTIFIER cashNotifier;
285 CHG_IP_NOTIFIER ipNotifier;
287 mutable pthread_mutex_t mutex;
289 std::string errorStr;
291 //-----------------------------------------------------------------------------