X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/1347f3d1e04bedd1508589173f577673ee2c5554..6142244f4146342445f7a9f5f6272763f689e89d:/include/stg/user_ips.h diff --git a/include/stg/user_ips.h b/include/stg/user_ips.h index e68190c5..35c20925 100644 --- a/include/stg/user_ips.h +++ b/include/stg/user_ips.h @@ -18,14 +18,15 @@ * Author : Boris Mikhailenko */ - /* - $Revision: 1.22 $ - $Date: 2010/03/04 11:49:53 $ - $Author: faust $ - */ +#pragma once + +#include "stg/common.h" -#ifndef USER_IPS_H -#define USER_IPS_H +#include +#include +#include +#include +#include #ifdef FREE_BSD #include @@ -35,241 +36,180 @@ #include #include -#include -#include -#include -#include -#include - -#include "stg/common.h" -#include "os_int.h" +namespace STG +{ //------------------------------------------------------------------------- -struct IP_MASK +struct IPMask { -IP_MASK() : ip(0), mask(0) {} -IP_MASK(const IP_MASK & ipm) : ip(ipm.ip), mask(ipm.mask) {} -uint32_t ip; -uint32_t mask; + IPMask() noexcept : ip(0), mask(0) {} + IPMask(uint32_t i, uint32_t m) noexcept : ip(i), mask(m) {} + + IPMask(const IPMask&) = default; + IPMask& operator=(const IPMask&) = default; + IPMask(IPMask&&) = default; + IPMask& operator=(IPMask&&) = default; + + uint32_t ip; + uint32_t mask; }; //------------------------------------------------------------------------- -class USER_IPS +class UserIPs { - friend std::ostream & operator<< (std::ostream & o, const USER_IPS & i); - //friend stringstream & operator<< (stringstream & s, const USER_IPS & i); - friend const USER_IPS StrToIPS(const std::string & ipsStr); - -public: - typedef std::vector ContainerType; - typedef ContainerType::size_type IndexType; - - USER_IPS(); - USER_IPS(const USER_IPS &); - USER_IPS & operator=(const USER_IPS &); - const IP_MASK & operator[](IndexType idx) const; - std::string GetIpStr() const; - bool IsIPInIPS(uint32_t ip) const; - bool OnlyOneIP() const; - int Count() const; - void Add(const IP_MASK &im); - void Erase(); - -private: - uint32_t CalcMask(unsigned int msk) const; - ContainerType ips; + friend std::ostream & operator<< (std::ostream & o, const UserIPs & i); + + public: + using ContainerType = std::vector; + using IndexType = ContainerType::size_type; + + UserIPs() = default; + + UserIPs(const UserIPs&) = default; + UserIPs& operator=(const UserIPs&) = default; + UserIPs(UserIPs&&) = default; + UserIPs& operator=(UserIPs&&) = default; + + static UserIPs parse(const std::string& source); + + const IPMask& operator[](IndexType idx) const noexcept { return ips[idx]; } + std::string toString() const noexcept; + bool find(uint32_t ip) const noexcept; + bool onlyOneIP() const noexcept; + bool isAnyIP() const noexcept; + size_t count() const noexcept { return ips.size(); } + void add(const IPMask& im) noexcept{ ips.push_back(im); } + + private: + uint32_t calcMask(unsigned int msk) const noexcept; + ContainerType ips; }; //------------------------------------------------------------------------- -//----------------------------------------------------------------------------- -inline -USER_IPS::USER_IPS() - : ips() -{} -//----------------------------------------------------------------------------- -inline -USER_IPS::USER_IPS(const USER_IPS & i) - : ips(i.ips) -{} -//----------------------------------------------------------------------------- inline -USER_IPS & USER_IPS::operator=(const USER_IPS & i) +std::string UserIPs::toString() const noexcept { -ips = i.ips; -return *this; + if (ips.empty()) + return ""; + + if (ips[0].ip == 0) + return "*"; + + auto it = ips.begin(); + std::string res = inet_ntostring(it->ip); + ++it; + for (; it != ips.end(); ++it) + res += "," + inet_ntostring(it->ip); + return res; } //----------------------------------------------------------------------------- inline -const IP_MASK & USER_IPS::operator[](IndexType idx) const +uint32_t UserIPs::calcMask(unsigned int msk) const noexcept { -return ips[idx]; + if (msk > 32) + return 0; + return htonl(0xFFffFFff << (32 - msk)); } //----------------------------------------------------------------------------- inline -std::string USER_IPS::GetIpStr() const +bool UserIPs::find(uint32_t ip) const noexcept { -if (ips.empty()) - { - return ""; - } + if (ips.empty()) + return false; -if (ips[0].ip == 0) - { - return "*"; - } + if (ips.front().ip == 0) + return true; -ContainerType::const_iterator it(ips.begin()); -std::ostringstream s; -s << inet_ntostring(it->ip); -++it; -for (; it != ips.end(); ++it) + for (auto it = ips.begin(); it != ips.end(); ++it) { - s << "," << inet_ntostring(it->ip); + const auto mask = calcMask(it->mask); + if ((ip & mask) == (it->ip & mask)) + return true; } -return s.str(); -} -//----------------------------------------------------------------------------- -inline -int USER_IPS::Count() const -{ -return static_cast(ips.size()); -} -//----------------------------------------------------------------------------- -inline -uint32_t USER_IPS::CalcMask(unsigned int msk) const -{ -if (msk > 32) - return 0; -return htonl(0xFFffFFff << (32 - msk)); -} -//----------------------------------------------------------------------------- -inline -bool USER_IPS::IsIPInIPS(uint32_t ip) const -{ -if (ips.empty()) - { return false; - } - -if (ips.front().ip == 0) - return true; - -for (ContainerType::const_iterator it(ips.begin()); it != ips.end(); ++it) - { - uint32_t mask(CalcMask(it->mask)); - if ((ip & mask) == (it->ip & mask)) - return true; - } -return false; } //----------------------------------------------------------------------------- inline -bool USER_IPS::OnlyOneIP() const +bool UserIPs::onlyOneIP() const noexcept { -if (ips.size() == 1 && ips.front().mask == 32) - return true; + if (ips.size() == 1 && ips.front().mask == 32 && ips.front().ip != 0) + return true; -return false; -} -//----------------------------------------------------------------------------- -inline -void USER_IPS::Add(const IP_MASK &im) -{ -ips.push_back(im); + return false; } //----------------------------------------------------------------------------- inline -void USER_IPS::Erase() +bool UserIPs::isAnyIP() const noexcept { -ips.erase(ips.begin(), ips.end()); + return !ips.empty() && ips.front().ip == 0; } //----------------------------------------------------------------------------- inline -std::ostream & operator<<(std::ostream & o, const USER_IPS & i) +std::ostream & operator<<(std::ostream& o, const UserIPs& i) { -return o << i.GetIpStr(); + return o << i.toString(); } //----------------------------------------------------------------------------- -/*inline -stringstream & operator<<(std::stringstream & s, const USER_IPS & i) -{ -s << i.GetIpStr(); -return s; -}*/ -//----------------------------------------------------------------------------- inline -const USER_IPS StrToIPS(const std::string & ipsStr) +UserIPs UserIPs::parse(const std::string& source) { -USER_IPS ips; -char * paddr; -IP_MASK im; -std::vector ipMask; -if (ipsStr.empty()) - { - return ips; - } + if (source.empty()) + return {}; -if (ipsStr[0] == '*' && ipsStr.size() == 1) + UserIPs ips; + if (source[0] == '*' && source.size() == 1) { - im.ip = 0; - im.mask = 0; - ips.ips.push_back(im); - return ips; + ips.ips.push_back(IPMask()); + return ips; } -char * tmp = new char[ipsStr.size() + 1]; -strcpy(tmp, ipsStr.c_str()); -char * pstr = tmp; -while ((paddr = strtok(pstr, ","))) + std::vector ipMask; + char * tmp = new char[source.size() + 1]; + strcpy(tmp, source.c_str()); + char * pstr = tmp; + char * paddr = NULL; + while ((paddr = strtok(pstr, ","))) { - pstr = NULL; - ipMask.push_back(paddr); + pstr = NULL; + ipMask.push_back(paddr); } -delete[] tmp; + delete[] tmp; -for (USER_IPS::IndexType i = 0; i < ipMask.size(); i++) + for (UserIPs::IndexType i = 0; i < ipMask.size(); i++) { - char str[128]; - char * strIp; - char * strMask; - strcpy(str, ipMask[i].c_str()); - strIp = strtok(str, "/"); - if (strIp == NULL) - { - return ips; - } - strMask = strtok(NULL, "/"); + char str[128]; + char * strIp; + char * strMask; + strcpy(str, ipMask[i].c_str()); + strIp = strtok(str, "/"); + if (strIp == NULL) + return ips; + strMask = strtok(NULL, "/"); - im.ip = inet_addr(strIp); - if (im.ip == INADDR_NONE) - { - return ips; - } + IPMask im; - im.mask = 32; - if (strMask != NULL) - { - int m = 0; - if (str2x(strMask, m) != 0) - { + im.ip = inet_addr(strIp); + if (im.ip == INADDR_NONE) return ips; - } - im.mask = m; - if (im.mask > 32) - { - return ips; - } + im.mask = 32; + if (strMask != NULL) + { + int m = 0; + if (str2x(strMask, m) != 0) + return ips; + im.mask = m; - if ((im.ip & ips.CalcMask(im.mask)) != im.ip) - { - return ips; - } + if (im.mask > 32) + return ips; + + if ((im.ip & ips.calcMask(im.mask)) != im.ip) + return ips; } - ips.ips.push_back(im); + ips.ips.push_back(im); } -return ips; + return ips; } //------------------------------------------------------------------------- -#endif //USER_IPS_H +}