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 $
31 #include "stg/user_stat.h"
32 #include "stg/user_conf.h"
33 #include "stg/user_ips.h"
34 #include "stg/user_property.h"
36 #include "stg/message.h"
37 #include "stg/noncopyable.h"
38 #include "stg/os_int.h"
39 #include "stg/const.h"
48 //-----------------------------------------------------------------------------
53 #ifdef USE_ABSTRACT_SETTINGS
58 //-----------------------------------------------------------------------------
59 class USER_ID_GENERATOR {
60 friend class USER_IMPL;
62 USER_ID_GENERATOR() {}
63 int GetNextID() { static int id = 0; return id++; }
65 //-----------------------------------------------------------------------------
66 class CHG_PASSIVE_NOTIFIER : public PROPERTY_NOTIFIER_BASE<int>,
69 CHG_PASSIVE_NOTIFIER(USER_IMPL * u) : user(u) {}
70 void Notify(const int & oldPassive, const int & newPassive);
75 //-----------------------------------------------------------------------------
76 class CHG_DISABLED_NOTIFIER : public PROPERTY_NOTIFIER_BASE<int>,
79 CHG_DISABLED_NOTIFIER(USER_IMPL * u) : user(u) {}
80 void Notify(const int & oldValue, const int & newValue);
85 //-----------------------------------------------------------------------------
86 class CHG_TARIFF_NOTIFIER : public PROPERTY_NOTIFIER_BASE<std::string>,
89 CHG_TARIFF_NOTIFIER(USER_IMPL * u) : user(u) {}
90 void Notify(const std::string & oldTariff, const std::string & newTariff);
95 //-----------------------------------------------------------------------------
96 class CHG_CASH_NOTIFIER : public PROPERTY_NOTIFIER_BASE<double>,
99 CHG_CASH_NOTIFIER(USER_IMPL * u) : user(u) {}
100 void Notify(const double & oldCash, const double & newCash);
105 //-----------------------------------------------------------------------------
106 class CHG_IPS_NOTIFIER : public PROPERTY_NOTIFIER_BASE<USER_IPS>,
107 private NONCOPYABLE {
109 CHG_IPS_NOTIFIER(USER_IMPL * u) : user(u) {}
110 void Notify(const USER_IPS & oldIPs, const USER_IPS & newIPs);
115 //-----------------------------------------------------------------------------
116 class USER_IMPL : public USER {
117 friend class CHG_PASSIVE_NOTIFIER;
118 friend class CHG_DISABLED_NOTIFIER;
119 friend class CHG_TARIFF_NOTIFIER;
120 friend class CHG_CASH_NOTIFIER;
121 friend class CHG_IPS_NOTIFIER;
123 #ifdef USE_ABSTRACT_SETTINGS
124 USER_IMPL(const SETTINGS * settings,
126 const TARIFFS * tariffs,
127 const ADMIN * sysAdmin,
130 USER_IMPL(const SETTINGS_IMPL * settings,
132 const TARIFFS * tariffs,
133 const ADMIN * sysAdmin,
136 USER_IMPL(const USER_IMPL & u);
137 virtual ~USER_IMPL();
143 int WriteMonthStat();
145 const std::string & GetLogin() const { return login; }
146 void SetLogin(std::string const & l);
148 uint32_t GetCurrIP() const { return currIP; }
149 time_t GetCurrIPModificationTime() const { return currIP.ModificationTime(); }
151 void AddCurrIPBeforeNotifier(CURR_IP_NOTIFIER * notifier);
152 void DelCurrIPBeforeNotifier(const CURR_IP_NOTIFIER * notifier);
154 void AddCurrIPAfterNotifier(CURR_IP_NOTIFIER * notifier);
155 void DelCurrIPAfterNotifier(const CURR_IP_NOTIFIER * notifier);
157 void AddConnectedBeforeNotifier(CONNECTED_NOTIFIER * notifier);
158 void DelConnectedBeforeNotifier(const CONNECTED_NOTIFIER * notifier);
160 void AddConnectedAfterNotifier(CONNECTED_NOTIFIER * notifier);
161 void DelConnectedAfterNotifier(const CONNECTED_NOTIFIER * notifier);
163 int GetID() const { return id; }
165 double GetPassiveTimePart() const;
166 void ResetPassiveTime() { passiveTime = 0; }
167 void SetPassiveTimeAsNewUser();
169 int WriteDetailStat(bool hard = false);
171 const TARIFF * GetTariff() const { return tariff; }
172 void ResetNextTariff() { nextTariff = ""; }
174 #ifdef TRAFF_STAT_WITH_PORTS
175 void AddTraffStatU(int dir, uint32_t ip, uint16_t port, uint32_t len);
176 void AddTraffStatD(int dir, uint32_t ip, uint16_t port, uint32_t len);
178 void AddTraffStatU(int dir, uint32_t ip, uint32_t len);
179 void AddTraffStatD(int dir, uint32_t ip, uint32_t len);
182 const DIR_TRAFF & GetSessionUpload() const { return sessionUpload; }
183 const DIR_TRAFF & GetSessionDownload() const { return sessionDownload; }
184 time_t GetSessionUploadModificationTime() const { return sessionUploadModTime; }
185 time_t GetSessionDownloadModificationTime() const { return sessionDownloadModTime; }
187 bool GetConnected() const { return connected; }
188 time_t GetConnectedModificationTime() const { return connected.ModificationTime(); }
189 const std::string & GetLastDisconnectReason() const { return lastDisconnectReason; }
190 int GetAuthorized() const { return static_cast<int>(authorizedBy.size()); }
191 time_t GetAuthorizedModificationTime() const { return authorizedModificationTime; }
192 int Authorize(uint32_t ip, uint32_t enabledDirs, const AUTH * auth);
193 void Unauthorize(const AUTH * auth,
194 const std::string & reason = std::string());
195 bool IsAuthorizedBy(const AUTH * auth) const;
196 std::vector<std::string> GetAuthorizers() const;
198 int AddMessage(STG_MSG * msg);
200 void UpdatePingTime(time_t t = 0);
201 time_t GetPingTime() const { return pingTime; }
203 void PrintUser() const;
206 const std::string & GetStrError() const { return errorStr; }
208 USER_PROPERTIES & GetProperty() { return property; }
209 const USER_PROPERTIES & GetProperty() const { return property; }
211 void SetDeleted() { deleted = true; }
212 bool GetDeleted() const { return deleted; }
214 time_t GetLastWriteStatTime() const { return lastWriteStat; }
216 void MidnightResetSessionStat();
217 void ProcessDayFee();
218 void ProcessDayFeeSpread();
219 void ProcessNewMonth();
220 void ProcessDailyFee();
223 std::string GetEnabledDirs() const;
228 virtual std::string GetParamValue(const std::string & name) const;
231 USER_IMPL & operator=(const USER_IMPL & rvalue);
234 USER_PROPERTIES property;
235 STG_LOGGER & WriteServLog;
237 void Connect(bool fakeConnect = false);
238 void Disconnect(bool fakeDisconnect, const std::string & reason);
239 int SaveMonthStat(int month, int year);
241 void SetPrepaidTraff();
243 int SendMessage(STG_MSG & msg) const;
246 time_t lastScanMessages;
251 USER_PROPERTY<bool> connected;
253 bool enabledDirs[DIR_NUM];
255 USER_ID_GENERATOR userIDGenerator;
257 uint32_t __currIP; // Current user's ip
258 USER_PROPERTY<uint32_t> currIP;
260 uint32_t lastIPForDisconnect; // User's ip after unauth but before disconnect
261 std::string lastDisconnectReason;
265 const ADMIN * sysAdmin;
268 const TARIFFS * tariffs;
269 const TARIFF * tariff;
271 TRAFF_STAT traffStat;
272 std::pair<time_t, TRAFF_STAT> traffStatSaved;
274 #ifdef USE_ABSTRACT_SETTINGS
275 const SETTINGS * settings;
277 const SETTINGS_IMPL * settings;
280 std::set<const AUTH *> authorizedBy;
281 time_t authorizedModificationTime;
283 std::list<STG_MSG> messages;
287 time_t lastWriteStat;
288 time_t lastWriteDetailedStat;
291 USER_PROPERTY<double> & cash;
292 USER_PROPERTY<DIR_TRAFF> & up;
293 USER_PROPERTY<DIR_TRAFF> & down;
294 USER_PROPERTY<double> & lastCashAdd;
295 USER_PROPERTY<time_t> & passiveTime;
296 USER_PROPERTY<time_t> & lastCashAddTime;
297 USER_PROPERTY<double> & freeMb;
298 USER_PROPERTY<time_t> & lastActivityTime;
299 USER_PROPERTY<std::string> & password;
300 USER_PROPERTY<int> & passive;
301 USER_PROPERTY<int> & disabled;
302 USER_PROPERTY<int> & disabledDetailStat;
303 USER_PROPERTY<int> & alwaysOnline;
304 USER_PROPERTY<std::string> & tariffName;
305 USER_PROPERTY<std::string> & nextTariff;
306 USER_PROPERTY<std::string> & address;
307 USER_PROPERTY<std::string> & note;
308 USER_PROPERTY<std::string> & group;
309 USER_PROPERTY<std::string> & email;
310 USER_PROPERTY<std::string> & phone;
311 USER_PROPERTY<std::string> & realName;
312 USER_PROPERTY<double> & credit;
313 USER_PROPERTY<time_t> & creditExpire;
314 USER_PROPERTY<USER_IPS> & ips;
315 USER_PROPERTY<std::string> & userdata0;
316 USER_PROPERTY<std::string> & userdata1;
317 USER_PROPERTY<std::string> & userdata2;
318 USER_PROPERTY<std::string> & userdata3;
319 USER_PROPERTY<std::string> & userdata4;
320 USER_PROPERTY<std::string> & userdata5;
321 USER_PROPERTY<std::string> & userdata6;
322 USER_PROPERTY<std::string> & userdata7;
323 USER_PROPERTY<std::string> & userdata8;
324 USER_PROPERTY<std::string> & userdata9;
328 DIR_TRAFF sessionUpload;
329 DIR_TRAFF sessionDownload;
330 time_t sessionUploadModTime;
331 time_t sessionDownloadModTime;
333 CHG_PASSIVE_NOTIFIER passiveNotifier;
334 CHG_DISABLED_NOTIFIER disabledNotifier;
335 CHG_TARIFF_NOTIFIER tariffNotifier;
336 CHG_CASH_NOTIFIER cashNotifier;
337 CHG_IPS_NOTIFIER ipNotifier;
339 mutable pthread_mutex_t mutex;
341 std::string errorStr;
343 //-----------------------------------------------------------------------------
345 typedef USER_IMPL * USER_IMPL_PTR;