mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-29 11:05:43 +00:00
136 lines
3.2 KiB
C++
136 lines
3.2 KiB
C++
#ifndef LLARP_ROUTER_OUTBOUND_MESSAGE_HANDLER_HPP
|
|
#define LLARP_ROUTER_OUTBOUND_MESSAGE_HANDLER_HPP
|
|
|
|
#include <router/i_outbound_message_handler.hpp>
|
|
|
|
#include <util/thread/logic.hpp>
|
|
#include <util/thread/queue.hpp>
|
|
#include <util/thread/threading.hpp>
|
|
#include <path/path_types.hpp>
|
|
#include <router_id.hpp>
|
|
|
|
#include <list>
|
|
#include <unordered_map>
|
|
#include <utility>
|
|
|
|
struct llarp_buffer_t;
|
|
|
|
namespace llarp
|
|
{
|
|
struct ILinkManager;
|
|
class Logic;
|
|
enum class SessionResult;
|
|
|
|
struct OutboundMessageHandler final : public IOutboundMessageHandler
|
|
{
|
|
public:
|
|
~OutboundMessageHandler() override = default;
|
|
|
|
OutboundMessageHandler(size_t maxQueueSize = MAX_OUTBOUND_QUEUE_SIZE);
|
|
|
|
bool
|
|
QueueMessage(const RouterID &remote, const ILinkMessage *msg,
|
|
SendStatusHandler callback) override LOCKS_EXCLUDED(_mutex);
|
|
|
|
void
|
|
Tick() override;
|
|
|
|
void
|
|
QueueRemoveEmptyPath(const PathID_t &pathid) override;
|
|
|
|
util::StatusObject
|
|
ExtractStatus() const override;
|
|
|
|
void
|
|
Init(ILinkManager *linkManager, std::shared_ptr< Logic > logic);
|
|
|
|
private:
|
|
using Message = std::pair< std::vector< byte_t >, SendStatusHandler >;
|
|
|
|
struct MessageQueueEntry
|
|
{
|
|
Message message;
|
|
PathID_t pathid;
|
|
RouterID router;
|
|
};
|
|
|
|
using MessageQueue = std::queue< MessageQueueEntry >;
|
|
|
|
void
|
|
OnSessionEstablished(const RouterID &router);
|
|
|
|
void
|
|
OnConnectTimeout(const RouterID &router);
|
|
|
|
void
|
|
OnRouterNotFound(const RouterID &router);
|
|
|
|
void
|
|
OnInvalidRouter(const RouterID &router);
|
|
|
|
void
|
|
OnNoLink(const RouterID &router);
|
|
|
|
void
|
|
OnSessionResult(const RouterID &router, const SessionResult result);
|
|
|
|
void
|
|
DoCallback(SendStatusHandler callback, SendStatus status);
|
|
|
|
void
|
|
QueueSessionCreation(const RouterID &remote);
|
|
|
|
bool
|
|
EncodeBuffer(const ILinkMessage *msg, llarp_buffer_t &buf);
|
|
|
|
bool
|
|
Send(const RouterID &remote, const Message &msg);
|
|
|
|
bool
|
|
SendIfSession(const RouterID &remote, const Message &msg);
|
|
|
|
bool
|
|
QueueOutboundMessage(const RouterID &remote, Message &&msg,
|
|
const PathID_t &pathid);
|
|
|
|
void
|
|
ProcessOutboundQueue();
|
|
|
|
void
|
|
RemoveEmptyPathQueues();
|
|
|
|
void
|
|
SendRoundRobin();
|
|
|
|
void
|
|
FinalizeSessionRequest(const RouterID &router, SendStatus status)
|
|
LOCKS_EXCLUDED(_mutex);
|
|
|
|
llarp::thread::Queue< MessageQueueEntry > outboundQueue;
|
|
llarp::thread::Queue< PathID_t > removedPaths;
|
|
bool removedSomePaths;
|
|
|
|
mutable util::Mutex _mutex; // protects pendingSessionMessageQueues
|
|
|
|
std::unordered_map< RouterID, MessageQueue, RouterID::Hash >
|
|
pendingSessionMessageQueues GUARDED_BY(_mutex);
|
|
|
|
std::unordered_map< PathID_t, MessageQueue, PathID_t::Hash >
|
|
outboundMessageQueues;
|
|
|
|
std::queue< PathID_t > roundRobinOrder;
|
|
|
|
ILinkManager *_linkManager;
|
|
std::shared_ptr< Logic > _logic;
|
|
|
|
util::ContentionKiller m_Killer;
|
|
|
|
// paths cannot have pathid "0", so it can be used as the "pathid"
|
|
// for non-traffic (control) messages, so they can be prioritized.
|
|
static const PathID_t zeroID;
|
|
};
|
|
|
|
} // namespace llarp
|
|
|
|
#endif // LLARP_ROUTER_OUTBOUND_MESSAGE_HANDLER_HPP
|