MirrorHub/construct
0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-09-28 19:58:53 +02:00
Code Issues Releases Wiki Activity

modules/federation/send: Limit evals per node w/ conf item.

Browse source
This commit is contained in:
Jason Volk 2020-01-07 15:14:47 -08:00
parent e0810afd87
commit a90b4e1344
1 changed files with 25 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
modules/federation/send.cc
View file

@ -33,6 +33,13 @@ allow_self
{ "default", "false" },
};
conf::item<size_t>
eval_max_per_node
{
{ "name", "ircd.federation.send.eval.max_per_node" },
{ "default", 1L },
};
void
handle_edu(client &client,
const m::resource::request::object<m::txn> &request,
@ -160,19 +167,27 @@ handle_put(client &client,
txn_id
};
size_t evals{0};
const bool txn_in_progress
{
!m::vm::eval::for_each([&txn_id, &request]
!m::vm::eval::for_each([&txn_id, &request, &evals]
(const auto &eval)
{
assert(eval.opts);
const bool match
const bool match_node
{
eval.opts->node_id == request.origin &&
eval.opts->node_id == request.origin
};
const bool match_txn
{
match_node &&
eval.opts->txn_id == txn_id
};
return !match; // false to break; for_each() returns false
evals += match_node;
return !match_txn; // false to break; for_each() returns false
})
};
@ -182,6 +197,12 @@ handle_put(client &client,
client, http::ACCEPTED
};
if(evals >= size_t(eval_max_per_node))
return m::resource::response
{
client, http::TOO_MANY_REQUESTS
};
for(const auto &pdu_failure : pdu_failures)
handle_pdu_failure(client, request, txn_id, pdu_failure);