forked from premiere/premiere-libtorrent
attempt to fix #212. Store 4 bytes and 16 bytes arrays instead of in_addr in ip-filter
This commit is contained in:
parent
d5748419c1
commit
81cf247cda
|
@ -76,9 +76,9 @@ namespace detail
|
|||
template<class Addr>
|
||||
Addr zero()
|
||||
{
|
||||
typename Addr::bytes_type zero;
|
||||
Addr zero;
|
||||
std::fill(zero.begin(), zero.end(), 0);
|
||||
return Addr(zero);
|
||||
return zero;
|
||||
}
|
||||
|
||||
template<>
|
||||
|
@ -87,8 +87,8 @@ namespace detail
|
|||
template<class Addr>
|
||||
Addr plus_one(Addr const& a)
|
||||
{
|
||||
typename Addr::bytes_type tmp(a.to_bytes());
|
||||
typedef typename Addr::bytes_type::reverse_iterator iter;
|
||||
Addr tmp(a);
|
||||
typedef typename Addr::reverse_iterator iter;
|
||||
for (iter i = tmp.rbegin()
|
||||
, end(tmp.rend()); i != end; ++i)
|
||||
{
|
||||
|
@ -99,7 +99,7 @@ namespace detail
|
|||
}
|
||||
*i = 0;
|
||||
}
|
||||
return Addr(tmp);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
inline boost::uint16_t plus_one(boost::uint16_t val) { return val + 1; }
|
||||
|
@ -107,8 +107,8 @@ namespace detail
|
|||
template<class Addr>
|
||||
Addr minus_one(Addr const& a)
|
||||
{
|
||||
typename Addr::bytes_type tmp(a.to_bytes());
|
||||
typedef typename Addr::bytes_type::reverse_iterator iter;
|
||||
Addr tmp(a);
|
||||
typedef typename Addr::reverse_iterator iter;
|
||||
for (iter i = tmp.rbegin()
|
||||
, end(tmp.rend()); i != end; ++i)
|
||||
{
|
||||
|
@ -119,7 +119,7 @@ namespace detail
|
|||
}
|
||||
*i = (std::numeric_limits<typename iter::value_type>::max)();
|
||||
}
|
||||
return Addr(tmp);
|
||||
return tmp;
|
||||
}
|
||||
|
||||
inline boost::uint16_t minus_one(boost::uint16_t val) { return val - 1; }
|
||||
|
@ -127,9 +127,9 @@ namespace detail
|
|||
template<class Addr>
|
||||
Addr max_addr()
|
||||
{
|
||||
typename Addr::bytes_type tmp;
|
||||
Addr tmp;
|
||||
std::fill(tmp.begin(), tmp.end()
|
||||
, (std::numeric_limits<typename Addr::bytes_type::value_type>::max)());
|
||||
, (std::numeric_limits<typename Addr::value_type>::max)());
|
||||
return Addr(tmp);
|
||||
}
|
||||
|
||||
|
@ -220,23 +220,24 @@ namespace detail
|
|||
return i->access;
|
||||
}
|
||||
|
||||
std::vector<ip_range<Addr> > export_filter() const
|
||||
template <class ExternalAddressType>
|
||||
std::vector<ip_range<ExternalAddressType> > export_filter() const
|
||||
{
|
||||
std::vector<ip_range<Addr> > ret;
|
||||
std::vector<ip_range<ExternalAddressType> > ret;
|
||||
ret.reserve(m_access_list.size());
|
||||
|
||||
for (typename range_t::const_iterator i = m_access_list.begin()
|
||||
, end(m_access_list.end()); i != end;)
|
||||
{
|
||||
ip_range<Addr> r;
|
||||
r.first = i->start;
|
||||
ip_range<ExternalAddressType> r;
|
||||
r.first = ExternalAddressType(i->start);
|
||||
r.flags = i->access;
|
||||
|
||||
++i;
|
||||
if (i == end)
|
||||
r.last = max_addr<Addr>();
|
||||
r.last = ExternalAddressType(max_addr<Addr>());
|
||||
else
|
||||
r.last = minus_one(i->start);
|
||||
r.last = ExternalAddressType(minus_one(i->start));
|
||||
|
||||
ret.push_back(r);
|
||||
}
|
||||
|
@ -288,8 +289,8 @@ public:
|
|||
|
||||
private:
|
||||
|
||||
detail::filter_impl<address_v4> m_filter4;
|
||||
detail::filter_impl<address_v6> m_filter6;
|
||||
detail::filter_impl<address_v4::bytes_type> m_filter4;
|
||||
detail::filter_impl<address_v6::bytes_type> m_filter6;
|
||||
};
|
||||
|
||||
class TORRENT_EXPORT port_filter
|
||||
|
|
|
@ -44,12 +44,12 @@ namespace libtorrent
|
|||
if (first.is_v4())
|
||||
{
|
||||
TORRENT_ASSERT(last.is_v4());
|
||||
m_filter4.add_rule(first.to_v4(), last.to_v4(), flags);
|
||||
m_filter4.add_rule(first.to_v4().to_bytes(), last.to_v4().to_bytes(), flags);
|
||||
}
|
||||
else if (first.is_v6())
|
||||
{
|
||||
TORRENT_ASSERT(last.is_v6());
|
||||
m_filter6.add_rule(first.to_v6(), last.to_v6(), flags);
|
||||
m_filter6.add_rule(first.to_v6().to_bytes(), last.to_v6().to_bytes(), flags);
|
||||
}
|
||||
else
|
||||
TORRENT_ASSERT(false);
|
||||
|
@ -58,15 +58,15 @@ namespace libtorrent
|
|||
int ip_filter::access(address const& addr) const
|
||||
{
|
||||
if (addr.is_v4())
|
||||
return m_filter4.access(addr.to_v4());
|
||||
return m_filter4.access(addr.to_v4().to_bytes());
|
||||
TORRENT_ASSERT(addr.is_v6());
|
||||
return m_filter6.access(addr.to_v6());
|
||||
return m_filter6.access(addr.to_v6().to_bytes());
|
||||
}
|
||||
|
||||
ip_filter::filter_tuple_t ip_filter::export_filter() const
|
||||
{
|
||||
return boost::make_tuple(m_filter4.export_filter()
|
||||
, m_filter6.export_filter());
|
||||
return boost::make_tuple(m_filter4.export_filter<address_v4>()
|
||||
, m_filter6.export_filter<address_v6>());
|
||||
}
|
||||
|
||||
void port_filter::add_rule(boost::uint16_t first, boost::uint16_t last, int flags)
|
||||
|
|
Loading…
Reference in New Issue