oss/conduit
oss/conduit
1
0
Fork 0
mirror of https://gitlab.com/famedly/conduit.git synced 2025-07-02 16:38:36 +00:00
Code Wiki Activity

add rate limiting for registration token validity

Browse source
This commit is contained in:
Matthias Ahouansou 2024-07-11 13:14:10 +01:00
parent 499548321f
commit 613107e7cf
No known key found for this signature in database
2 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/config/mod.rs
View file

@ -105,13 +105,13 @@ pub struct WellKnownConfig {
} }
#[derive(Clone, Copy, Debug, Default, Deserialize, Eq, Hash, Ord, PartialEq, PartialOrd)] #[derive(Clone, Copy, Debug, Default, Deserialize, Eq, Hash, Ord, PartialEq, PartialOrd)]
#[serde(rename_all = "lowercase")] #[serde(rename_all = "snake_case")]
pub enum Restriction { pub enum Restriction {
Registration, Registration,
Login, Login,
RegistrationTokenValidity,
#[default] #[default]
#[serde(rename = "")]
CatchAll, CatchAll,
} }

9
src/service/rate_limiting/mod.rs
View file

@ -7,7 +7,10 @@ use std::{
use dashmap::{mapref::entry::Entry, DashMap}; use dashmap::{mapref::entry::Entry, DashMap};
use ruma::{ use ruma::{
api::{ api::{
client::{account::register, session::login}, client::{
account::{check_registration_token_validity, register},
session::login,
},
IncomingRequest, Metadata, IncomingRequest, Metadata,
}, },
OwnedServerName, OwnedUserId, OwnedServerName, OwnedUserId,
@ -20,6 +23,10 @@ impl From<Metadata> for Restriction {
[ [
(register::v3::Request::METADATA, Restriction::Registration), (register::v3::Request::METADATA, Restriction::Registration),
(login::v3::Request::METADATA, Restriction::Login), (login::v3::Request::METADATA, Restriction::Login),
(
check_registration_token_validity::v1::Request::METADATA,
Restriction::RegistrationTokenValidity,
),
] ]
.into_iter() .into_iter()
.find(|(other, _)| { .find(|(other, _)| {