lokinet/llarp/path/pathbuilder.hpp

89 lines
2.0 KiB
C++
Raw Normal View History

2018-07-09 17:32:11 +00:00
#ifndef LLARP_PATHBUILDER_HPP_
#define LLARP_PATHBUILDER_HPP_
2018-08-30 18:48:43 +00:00
2019-01-11 01:19:36 +00:00
#include <path/pathset.hpp>
#include <atomic>
2018-06-18 22:03:50 +00:00
2019-01-16 00:24:16 +00:00
struct llarp_dht_context;
2018-08-30 18:48:43 +00:00
namespace llarp
2018-06-18 22:03:50 +00:00
{
2018-08-30 18:48:43 +00:00
namespace path
{
// milliseconds waiting between builds on a path
2019-01-07 17:28:59 +00:00
constexpr llarp_time_t MIN_PATH_BUILD_INTERVAL = 5 * 1000;
2018-08-30 18:48:43 +00:00
struct Builder : public PathSet
{
protected:
/// flag for PathSet::Stop()
std::atomic< bool > _run;
public:
llarp::Router* router;
2019-01-16 00:24:16 +00:00
llarp_dht_context* dht;
2018-08-30 18:48:43 +00:00
llarp::SecretKey enckey;
size_t numHops;
llarp_time_t lastBuild = 0;
llarp_time_t buildIntervalLimit = MIN_PATH_BUILD_INTERVAL;
// how many keygens are currently happening
std::atomic< uint8_t > keygens;
2018-08-30 18:48:43 +00:00
/// construct
2019-01-16 00:24:16 +00:00
Builder(llarp::Router* p_router, llarp_dht_context* p_dht,
2018-08-30 18:48:43 +00:00
size_t numPaths, size_t numHops);
virtual ~Builder();
virtual bool
SelectHop(llarp_nodedb* db, const RouterContact& prev, RouterContact& cur,
2018-12-24 21:10:35 +00:00
size_t hop, PathRole roles) override;
2018-08-30 18:48:43 +00:00
virtual bool
2018-12-24 21:10:35 +00:00
ShouldBuildMore(llarp_time_t now) const override;
2018-10-29 16:48:36 +00:00
2018-12-27 12:00:28 +00:00
/// return true if we hit our soft limit for building paths too fast
bool
2018-12-27 12:00:28 +00:00
BuildCooldownHit(llarp_time_t now) const;
virtual bool
2018-12-24 21:10:35 +00:00
Stop() override;
2019-02-05 14:50:33 +00:00
bool
IsStopped() const override;
bool
ShouldRemove() const override;
2018-10-29 16:48:36 +00:00
llarp_time_t
2018-12-24 21:10:35 +00:00
Now() const override;
2018-08-30 18:48:43 +00:00
void
2018-11-14 18:02:27 +00:00
BuildOne(PathRole roles = ePathRoleAny);
2018-08-30 18:48:43 +00:00
void
2018-11-14 18:02:27 +00:00
Build(const std::vector< RouterContact >& hops,
PathRole roles = ePathRoleAny);
bool
2018-11-14 18:02:27 +00:00
SelectHops(llarp_nodedb* db, std::vector< RouterContact >& hops,
PathRole roles = ePathRoleAny);
2018-08-30 18:48:43 +00:00
void
2018-11-14 18:02:27 +00:00
ManualRebuild(size_t N, PathRole roles = ePathRoleAny);
2018-08-30 18:48:43 +00:00
virtual const SecretKey&
2018-08-30 18:48:43 +00:00
GetTunnelEncryptionSecretKey() const;
virtual void
2018-12-24 21:10:35 +00:00
HandlePathBuilt(Path* p) override;
virtual void
2018-12-24 21:10:35 +00:00
HandlePathBuildTimeout(Path* p) override;
2018-08-30 18:48:43 +00:00
};
} // namespace path
} // namespace llarp
2018-06-18 22:03:50 +00:00
#endif