mirror of https://github.com/PurpleI2P/i2pd.git
I2P: End-to-End encrypted and anonymous Internet
https://i2pd.website/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
2.4 KiB
127 lines
2.4 KiB
#ifndef UTIL_H |
|
#define UTIL_H |
|
|
|
#include <string> |
|
#include <functional> |
|
#include <memory> |
|
#include <mutex> |
|
#include <boost/asio.hpp> |
|
|
|
#ifdef ANDROID |
|
#include <boost/lexical_cast.hpp> |
|
namespace std |
|
{ |
|
template <typename T> |
|
std::string to_string(T value) |
|
{ |
|
return boost::lexical_cast<std::string>(value); |
|
} |
|
|
|
inline int stoi(const std::string& str) |
|
{ |
|
return boost::lexical_cast<int>(str); |
|
} |
|
} |
|
#endif |
|
|
|
namespace i2p |
|
{ |
|
namespace util |
|
{ |
|
|
|
template<class T> |
|
class MemoryPool |
|
{ |
|
public: |
|
|
|
MemoryPool (): m_Head (nullptr) {}; |
|
~MemoryPool () |
|
{ |
|
while (m_Head) |
|
{ |
|
auto tmp = m_Head; |
|
m_Head = static_cast<T*>(*(void * *)m_Head); // next |
|
delete tmp; |
|
} |
|
} |
|
|
|
template<typename... TArgs> |
|
T * Acquire (TArgs&&... args) |
|
{ |
|
if (!m_Head) return new T(args...); |
|
else |
|
{ |
|
auto tmp = m_Head; |
|
m_Head = static_cast<T*>(*(void * *)m_Head); // next |
|
return new (tmp)T(args...); |
|
} |
|
} |
|
|
|
void Release (T * t) |
|
{ |
|
if (!t) return; |
|
t->~T (); |
|
*(void * *)t = m_Head; // next |
|
m_Head = t; |
|
} |
|
|
|
template<typename... TArgs> |
|
std::unique_ptr<T, std::function<void(T*)> > AcquireUnique (TArgs&&... args) |
|
{ |
|
return std::unique_ptr<T, std::function<void(T*)> >(Acquire (args...), |
|
std::bind (&MemoryPool<T>::Release, this, std::placeholders::_1)); |
|
} |
|
|
|
template<typename... TArgs> |
|
std::shared_ptr<T> AcquireShared (TArgs&&... args) |
|
{ |
|
return std::shared_ptr<T>(Acquire (args...), std::bind (&MemoryPool<T>::Release, this, std::placeholders::_1)); |
|
} |
|
|
|
protected: |
|
|
|
T * m_Head; |
|
}; |
|
|
|
template<class T> |
|
class MemoryPoolMt: public MemoryPool<T> |
|
{ |
|
public: |
|
|
|
MemoryPoolMt () {}; |
|
template<typename... TArgs> |
|
T * AcquireMt (TArgs&&... args) |
|
{ |
|
if (!this->m_Head) return new T(args...); |
|
std::lock_guard<std::mutex> l(m_Mutex); |
|
return this->Acquire (args...); |
|
} |
|
|
|
void ReleaseMt (T * t) |
|
{ |
|
std::lock_guard<std::mutex> l(m_Mutex); |
|
this->Release (t); |
|
} |
|
|
|
template<template<typename, typename...>class C, typename... R> |
|
void ReleaseMt(const C<T *, R...>& c) |
|
{ |
|
std::lock_guard<std::mutex> l(m_Mutex); |
|
for (auto& it: c) |
|
this->Release (it); |
|
} |
|
|
|
private: |
|
|
|
std::mutex m_Mutex; |
|
}; |
|
|
|
namespace net |
|
{ |
|
int GetMTU (const boost::asio::ip::address& localAddress); |
|
const boost::asio::ip::address GetInterfaceAddress(const std::string & ifname, bool ipv6=false); |
|
} |
|
} |
|
} |
|
|
|
#endif
|
|
|