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/10/07 20:04:48 $
39 #include "stg/os_int.h"
40 #include "stg/store.h"
41 #include "stg/users.h"
43 #include "stg/tariffs.h"
44 #include "stg/logger.h"
45 #include "stg/notifer.h"
46 #include "stg/noncopyable.h"
48 #include "eventloop.h"
49 #include "settings_impl.h"
50 #include "user_impl.h"
52 const int userDeleteDelayTime = 120;
54 typedef std::list<USER_IMPL>::iterator user_iter;
55 typedef std::list<USER_IMPL>::const_iterator const_user_iter;
58 //-----------------------------------------------------------------------------
65 std::list<USER_IMPL>::iterator iter;
68 //-----------------------------------------------------------------------------
69 class USERS_IMPL : private NONCOPYABLE, public USERS {
70 friend class PROPERTY_NOTIFER_IP_BEFORE;
71 friend class PROPERTY_NOTIFER_IP_AFTER;
74 USERS_IMPL(SETTINGS_IMPL * s, STORE * store,
75 TARIFFS * tariffs, SERVICES & svcs,
76 const ADMIN * sysAdmin);
77 virtual ~USERS_IMPL();
79 int FindByName(const std::string & login, USER_PTR * user);
80 int FindByName(const std::string & login, CONST_USER_PTR * user) const;
82 bool Exists(const std::string & login) const;
84 bool TariffInUse(const std::string & tariffName) const;
86 void AddNotifierUserAdd(NOTIFIER_BASE<USER_PTR> *);
87 void DelNotifierUserAdd(NOTIFIER_BASE<USER_PTR> *);
89 void AddNotifierUserDel(NOTIFIER_BASE<USER_PTR> *);
90 void DelNotifierUserDel(NOTIFIER_BASE<USER_PTR> *);
92 void AddNotifierUserAdd(NOTIFIER_BASE<USER_IMPL_PTR> *);
93 void DelNotifierUserAdd(NOTIFIER_BASE<USER_IMPL_PTR> *);
95 void AddNotifierUserDel(NOTIFIER_BASE<USER_IMPL_PTR> *);
96 void DelNotifierUserDel(NOTIFIER_BASE<USER_IMPL_PTR> *);
98 int Add(const std::string & login, const ADMIN * admin);
99 void Del(const std::string & login, const ADMIN * admin);
101 bool Authorize(const std::string & login, uint32_t ip,
102 uint32_t enabledDirs, const AUTH * auth);
103 bool Unauthorize(const std::string & login,
105 const std::string & reason = std::string());
108 size_t Count() const { return users.size(); }
110 int FindByIPIdx(uint32_t ip, USER_PTR * user) const;
111 int FindByIPIdx(uint32_t ip, USER_IMPL ** user) const;
112 bool IsIPInIndex(uint32_t ip) const;
113 bool IsIPInUse(uint32_t ip, const std::string & login, CONST_USER_PTR * user) const;
116 int SearchNext(int handler, USER_PTR * user);
117 int SearchNext(int handler, USER_IMPL ** user);
118 int CloseSearch(int handler);
124 USERS_IMPL(const USERS_IMPL & rvalue);
125 USERS_IMPL & operator=(const USERS_IMPL & rvalue);
127 void AddToIPIdx(user_iter user);
128 void DelFromIPIdx(uint32_t ip);
129 bool FindByIPIdx(uint32_t ip, user_iter & iter) const;
131 int FindByNameNonLock(const std::string & login, user_iter * user);
132 int FindByNameNonLock(const std::string & login, const_user_iter * user) const;
135 void ProcessActions();
137 void AddUserIntoIndexes(user_iter user);
138 void DelUserFromIndexes(user_iter user);
140 static void * Run(void *);
141 void NewMinute(const struct tm & t);
142 void NewDay(const struct tm & t);
143 void DayResetTraff(const struct tm & t);
145 bool TimeToWriteDetailStat(const struct tm & t);
147 std::list<USER_IMPL> users;
148 std::list<USER_TO_DEL> usersToDelete;
150 std::map<uint32_t, user_iter> ipIndex;
151 std::map<std::string, user_iter> loginIndex;
153 SETTINGS_IMPL * settings;
155 SERVICES & m_services;
157 const ADMIN * sysAdmin;
158 STG_LOGGER & WriteServLog;
163 mutable pthread_mutex_t mutex;
165 mutable unsigned int handle;
167 mutable std::map<int, user_iter> searchDescriptors;
169 std::set<NOTIFIER_BASE<USER_PTR>*> onAddNotifiers;
170 std::set<NOTIFIER_BASE<USER_PTR>*> onDelNotifiers;
171 std::set<NOTIFIER_BASE<USER_IMPL_PTR>*> onAddNotifiersImpl;
172 std::set<NOTIFIER_BASE<USER_IMPL_PTR>*> onDelNotifiersImpl;
174 //-----------------------------------------------------------------------------
176 void PROPERTY_NOTIFER_IP_BEFORE::Notify(const uint32_t & oldValue,
182 //EVENT_LOOP_SINGLETON::GetInstance().Enqueue(users, &USERS::DelFromIPIdx, oldValue);
183 // Using explicit call to assure that index is valid, because fast reconnect with delayed call can result in authorization error
184 users.DelFromIPIdx(oldValue);
186 //-----------------------------------------------------------------------------
188 void PROPERTY_NOTIFER_IP_AFTER::Notify(const uint32_t &,
189 const uint32_t & newValue)
194 //EVENT_LOOP_SINGLETON::GetInstance().Enqueue(users, &USERS::AddToIPIdx, user);
195 // Using explicit call to assure that index is valid, because fast reconnect with delayed call can result in authorization error
196 users.AddToIPIdx(user);
198 //-----------------------------------------------------------------------------