X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/4271ab433cd55bbd2612292bcf39e4dc3d7274f1..0907aa4037b12b6b88ee24495d4577a064d4f8db:/stargazer/traffcounter_impl.h diff --git a/stargazer/traffcounter_impl.h b/stargazer/traffcounter_impl.h deleted file mode 100644 index 78027cb5..00000000 --- a/stargazer/traffcounter_impl.h +++ /dev/null @@ -1,271 +0,0 @@ -/* - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ - -/* - * Author : Boris Mikhailenko - */ - -#pragma once - -#include "stg/traffcounter.h" -#include "stg/logger.h" -#include "stg/raw_ip_packet.h" -#include "stg/noncopyable.h" -#include "stg/notifer.h" -#include "actions.h" -#include "eventloop.h" -#include "user_impl.h" - -#include -#include -#include -#include -#include - -#include - -#define PROTOMAX (5) - -namespace STG -{ - -class UsersImpl; - -//----------------------------------------------------------------------------- -struct Rule { -uint32_t ip; // IP -uint32_t mask; // Network mask -uint16_t port1; // Min port -uint16_t port2; // Max port -uint8_t proto; // Protocol -uint32_t dir; // Direction -}; -//----------------------------------------------------------------------------- -struct PacketExtraData { - PacketExtraData() - : flushTime(0), - updateTime(0), - userU(NULL), - userUPresent(false), - userD(NULL), - userDPresent(false), - dirU(DIR_NUM), - dirD(DIR_NUM), - lenU(0), - lenD(0) - {} - - time_t flushTime; // Last flush time - time_t updateTime; // Last update time - UserImpl * userU; // Uploader - bool userUPresent; // Uploader is registered user - UserImpl * userD; // Downloader - bool userDPresent; // Downloader is registered user - int dirU; // Upload direction - int dirD; // Download direction - uint32_t lenU; // Upload length - uint32_t lenD; // Download length -}; -//----------------------------------------------------------------------------- -class TraffCounterImpl; -//----------------------------------------------------------------------------- -class TRF_IP_BEFORE: public PropertyNotifierBase { -public: - TRF_IP_BEFORE(TraffCounterImpl & t, UserImpl * u) - : PropertyNotifierBase(), - traffCnt(t), - user(u) - {} - TRF_IP_BEFORE(const TRF_IP_BEFORE & rvalue) - : PropertyNotifierBase(), - traffCnt(rvalue.traffCnt), - user(rvalue.user) - {} - void Notify(const uint32_t & oldValue, const uint32_t & newValue); - void SetUser(UserImpl * u) { user = u; } - UserImpl * GetUser() const { return user; } - -private: - TRF_IP_BEFORE & operator=(const TRF_IP_BEFORE & rvalue); - - TraffCounterImpl & traffCnt; - UserImpl * user; -}; -//----------------------------------------------------------------------------- -class TRF_IP_AFTER: public PropertyNotifierBase { -public: - TRF_IP_AFTER(TraffCounterImpl & t, UserImpl * u) - : PropertyNotifierBase(), - traffCnt(t), - user(u) - {} - TRF_IP_AFTER(const TRF_IP_AFTER & rvalue) - : PropertyNotifierBase(), - traffCnt(rvalue.traffCnt), - user(rvalue.user) - {} - void Notify(const uint32_t & oldValue, const uint32_t & newValue); - void SetUser(UserImpl * u) { user = u; } - UserImpl * GetUser() const { return user; } -private: - TRF_IP_AFTER & operator=(const TRF_IP_AFTER & rvalue); - - TraffCounterImpl & traffCnt; - UserImpl * user; -}; - -using UserImplPtr = UserImpl*; -//----------------------------------------------------------------------------- -class ADD_USER_NONIFIER: public NotifierBase { -public: - explicit ADD_USER_NONIFIER(TraffCounterImpl & t) : - NotifierBase(), - traffCnt(t) - {} - virtual ~ADD_USER_NONIFIER() {} - void Notify(const UserImplPtr & user); - -private: - ADD_USER_NONIFIER(const ADD_USER_NONIFIER & rvalue); - ADD_USER_NONIFIER & operator=(const ADD_USER_NONIFIER & rvalue); - - TraffCounterImpl & traffCnt; -}; -//----------------------------------------------------------------------------- -class DEL_USER_NONIFIER: public NotifierBase { -public: - explicit DEL_USER_NONIFIER(TraffCounterImpl & t) : - NotifierBase(), - traffCnt(t) - {} - virtual ~DEL_USER_NONIFIER() {} - void Notify(const UserImplPtr & user); - -private: - DEL_USER_NONIFIER(const DEL_USER_NONIFIER & rvalue); - DEL_USER_NONIFIER & operator=(const DEL_USER_NONIFIER & rvalue); - - TraffCounterImpl & traffCnt; -}; -//----------------------------------------------------------------------------- -class TraffCounterImpl : public TraffCounter { - friend class ADD_USER_NONIFIER; - friend class DEL_USER_NONIFIER; - friend class TRF_IP_BEFORE; - friend class TRF_IP_AFTER; - public: - TraffCounterImpl(UsersImpl * users, const std::string & rulesFileName); - ~TraffCounterImpl(); - - int Reload(); - int Start(); - int Stop(); - - void process(const RawPacket & rawPacket) override; - void SetMonitorDir(const std::string & monitorDir); - - size_t rulesCount() const override { return rules.size(); } - - private: - bool ParseAddress(const char * ta, Rule * rule) const; - uint32_t CalcMask(uint32_t msk) const; - void FreeRules(); - bool ReadRules(bool test = false); - - static void * Run(void * data); - - void DeterminateDir(const RawPacket & packet, - int * dirU, // Direction for upload - int * dirD) const; // Direction for download - - void FlushAndRemove(); - - void AddUser(UserImpl * user); - void DelUser(uint32_t uip); - void SetUserNotifiers(UserImpl * user); - void UnSetUserNotifiers(UserImpl * user); - - typedef std::list::iterator rule_iter; - - std::list rules; - - typedef std::map Packets; - typedef Packets::iterator pp_iter; - typedef std::multimap Index; - typedef Index::iterator ip2p_iter; - typedef Index::const_iterator ip2p_citer; - - Packets packets; // Packets tree - - Index ip2packets; // IP-to-Packet index - - std::string dirName[DIR_NUM + 1]; - - Logger & WriteServLog; - std::string rulesFileName; - - std::string monitorDir; - bool monitoring; - time_t touchTimeP; - - UsersImpl * users; - - bool running; - bool stopped; - pthread_mutex_t mutex; - pthread_t thread; - - std::list ipBeforeNotifiers; - std::list ipAfterNotifiers; - - ADD_USER_NONIFIER addUserNotifier; - DEL_USER_NONIFIER delUserNotifier; -}; -//----------------------------------------------------------------------------- -inline -void TRF_IP_BEFORE::Notify(const uint32_t & oldValue, const uint32_t &) -{ -// User changes his address. Remove old IP -if (!oldValue) - return; - -EVENT_LOOP_SINGLETON::GetInstance().Enqueue(traffCnt, &TraffCounterImpl::DelUser, oldValue); -} -//----------------------------------------------------------------------------- -inline -void TRF_IP_AFTER::Notify(const uint32_t &, const uint32_t & newValue) -{ -// User changes his address. Add new IP -if (!newValue) - return; - -EVENT_LOOP_SINGLETON::GetInstance().Enqueue(traffCnt, &TraffCounterImpl::AddUser, user); -} -//----------------------------------------------------------------------------- -inline -void ADD_USER_NONIFIER::Notify(const UserImplPtr & user) -{ -EVENT_LOOP_SINGLETON::GetInstance().Enqueue(traffCnt, &TraffCounterImpl::SetUserNotifiers, user); -} -//----------------------------------------------------------------------------- -inline -void DEL_USER_NONIFIER::Notify(const UserImplPtr & user) -{ -EVENT_LOOP_SINGLETON::GetInstance().Enqueue(traffCnt, &TraffCounterImpl::UnSetUserNotifiers, user); -EVENT_LOOP_SINGLETON::GetInstance().Enqueue(traffCnt, &TraffCounterImpl::DelUser, user->GetCurrIP()); -} -//----------------------------------------------------------------------------- -}