mirror of https://github.com/oxen-io/lokinet
Merge branch 'master' of https://github.com/majestrate/llarp
Ryan Tharp
6 years ago
commit
3241655ed9
@ -0,0 +1,63 @@
|
|||||||
|
#ifndef LLARP_PATHSET_HPP
|
||||||
|
#define LLARP_PATHSET_HPP
|
||||||
|
|
||||||
|
#include <llarp/path_types.hpp>
|
||||||
|
#include <map>
|
||||||
|
#include <tuple>
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace path
|
||||||
|
{
|
||||||
|
enum PathStatus
|
||||||
|
{
|
||||||
|
ePathBuilding,
|
||||||
|
ePathEstablished,
|
||||||
|
ePathTimeout,
|
||||||
|
ePathExpired
|
||||||
|
};
|
||||||
|
// forward declare
|
||||||
|
struct Path;
|
||||||
|
|
||||||
|
/// a set of paths owned by an entity
|
||||||
|
struct PathSet
|
||||||
|
{
|
||||||
|
/// construct
|
||||||
|
/// @params numPaths the number of paths to maintain
|
||||||
|
PathSet(size_t numPaths);
|
||||||
|
|
||||||
|
void
|
||||||
|
RemovePath(Path* path);
|
||||||
|
|
||||||
|
void
|
||||||
|
HandlePathBuilt(Path* path);
|
||||||
|
|
||||||
|
void
|
||||||
|
AddPath(Path* path);
|
||||||
|
|
||||||
|
Path*
|
||||||
|
GetByUpstream(const RouterID& remote, const PathID_t& rxid);
|
||||||
|
|
||||||
|
void
|
||||||
|
ExpirePaths(llarp_time_t now);
|
||||||
|
|
||||||
|
size_t
|
||||||
|
NumInStatus(PathStatus st) const;
|
||||||
|
|
||||||
|
/// return true if we should build another path
|
||||||
|
bool
|
||||||
|
ShouldBuildMore() const;
|
||||||
|
|
||||||
|
private:
|
||||||
|
typedef std::map< PathID_t, Path* > PathMap_t;
|
||||||
|
// (tx,rx)
|
||||||
|
typedef std::tuple< PathMap_t, PathMap_t > PathContainer_t;
|
||||||
|
|
||||||
|
size_t m_NumPaths;
|
||||||
|
PathContainer_t m_Paths;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace path
|
||||||
|
} // namespace llarp
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,74 @@
|
|||||||
|
#ifndef LLARP_CODEL_QUEUE_HPP
|
||||||
|
#define LLARP_CODEL_QUEUE_HPP
|
||||||
|
#include <llarp/time.h>
|
||||||
|
#include <cmath>
|
||||||
|
#include <functional>
|
||||||
|
#include <mutex>
|
||||||
|
#include <queue>
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace util
|
||||||
|
{
|
||||||
|
template < typename T, typename GetTime, llarp_time_t dropMs = 20,
|
||||||
|
llarp_time_t initialIntervalMs = 100 >
|
||||||
|
struct CoDelQueue
|
||||||
|
{
|
||||||
|
struct CoDelCompare
|
||||||
|
{
|
||||||
|
GetTime getTime = GetTime();
|
||||||
|
bool
|
||||||
|
operator()(const T& left, const T& right) const
|
||||||
|
{
|
||||||
|
return getTime(left) < getTime(right);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void
|
||||||
|
Put(T* item)
|
||||||
|
{
|
||||||
|
std::unique_lock< std::mutex > lock(m_QueueMutex);
|
||||||
|
m_Queue.push(*item);
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
Process(std::queue< T >& result)
|
||||||
|
{
|
||||||
|
llarp_time_t lowest = 0xFFFFFFFFFFFFFFFFUL;
|
||||||
|
auto start = llarp_time_now_ms();
|
||||||
|
std::unique_lock< std::mutex > lock(m_QueueMutex);
|
||||||
|
while(m_Queue.size())
|
||||||
|
{
|
||||||
|
const auto& item = m_Queue.top();
|
||||||
|
auto dlt = start - getTime(item);
|
||||||
|
lowest = std::min(dlt, lowest);
|
||||||
|
if(m_Queue.size() == 1)
|
||||||
|
{
|
||||||
|
if(lowest > dropMs)
|
||||||
|
{
|
||||||
|
// drop
|
||||||
|
nextTickInterval = initialIntervalMs / std::sqrt(++dropNum);
|
||||||
|
m_Queue.pop();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
nextTickInterval = initialIntervalMs;
|
||||||
|
dropNum = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
result.push(item);
|
||||||
|
m_Queue.pop();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
GetTime getTime = GetTime();
|
||||||
|
size_t dropNum = 0;
|
||||||
|
llarp_time_t nextTickInterval = initialIntervalMs;
|
||||||
|
std::mutex m_QueueMutex;
|
||||||
|
std::priority_queue< T, std::vector< T >, CoDelCompare > m_Queue;
|
||||||
|
};
|
||||||
|
} // namespace util
|
||||||
|
} // namespace llarp
|
||||||
|
|
||||||
|
#endif
|
||||||
@ -0,0 +1,97 @@
|
|||||||
|
#include <llarp/path.hpp>
|
||||||
|
#include <llarp/pathset.hpp>
|
||||||
|
|
||||||
|
namespace llarp
|
||||||
|
{
|
||||||
|
namespace path
|
||||||
|
{
|
||||||
|
PathSet::PathSet(size_t num) : m_NumPaths(num)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
bool
|
||||||
|
PathSet::ShouldBuildMore() const
|
||||||
|
{
|
||||||
|
return std::get< 0 >(m_Paths).size() < m_NumPaths;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
PathSet::ExpirePaths(llarp_time_t now)
|
||||||
|
{
|
||||||
|
{
|
||||||
|
auto& map = std::get< 0 >(m_Paths);
|
||||||
|
auto itr = map.begin();
|
||||||
|
while(itr != map.end())
|
||||||
|
{
|
||||||
|
if(itr->second->Expired(now))
|
||||||
|
{
|
||||||
|
itr = map.erase(itr);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto& map = std::get< 1 >(m_Paths);
|
||||||
|
auto itr = map.begin();
|
||||||
|
while(itr != map.end())
|
||||||
|
{
|
||||||
|
if(itr->second->Expired(now))
|
||||||
|
{
|
||||||
|
// delete path on second iteration
|
||||||
|
delete itr->second;
|
||||||
|
itr = map.erase(itr);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t
|
||||||
|
PathSet::NumInStatus(PathStatus st) const
|
||||||
|
{
|
||||||
|
size_t count = 0;
|
||||||
|
auto& map = std::get< 0 >(m_Paths);
|
||||||
|
auto itr = map.begin();
|
||||||
|
while(itr != map.end())
|
||||||
|
{
|
||||||
|
if(itr->second->status == st)
|
||||||
|
++count;
|
||||||
|
++itr;
|
||||||
|
}
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
PathSet::AddPath(Path* path)
|
||||||
|
{
|
||||||
|
std::get< 0 >(m_Paths).emplace(path->TXID(), path);
|
||||||
|
std::get< 1 >(m_Paths).emplace(path->RXID(), path);
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
PathSet::RemovePath(Path* path)
|
||||||
|
{
|
||||||
|
std::get< 0 >(m_Paths).erase(path->TXID());
|
||||||
|
std::get< 1 >(m_Paths).erase(path->RXID());
|
||||||
|
}
|
||||||
|
|
||||||
|
Path*
|
||||||
|
PathSet::GetByUpstream(const RouterID& remote, const PathID_t& rxid)
|
||||||
|
{
|
||||||
|
auto& set = std::get< 1 >(m_Paths);
|
||||||
|
auto itr = set.begin();
|
||||||
|
while(itr != set.end())
|
||||||
|
{
|
||||||
|
if(itr->second->Upstream() == remote)
|
||||||
|
return itr->second;
|
||||||
|
++itr;
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
void
|
||||||
|
PathSet::HandlePathBuilt(Path* path)
|
||||||
|
{
|
||||||
|
// TODO: implement me
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace path
|
||||||
|
} // namespace llarp
|
||||||
Loading…
Reference in New Issue