Split api/client/membership.rs into directory
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
File diff suppressed because it is too large
Load Diff
60
src/api/client/membership/ban.rs
Normal file
60
src/api/client/membership/ban.rs
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use ruma::{
|
||||||
|
api::client::membership::ban_user,
|
||||||
|
events::room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{Err, Result, matrix::pdu::PduBuilder};
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/ban`
|
||||||
|
///
|
||||||
|
/// Tries to send a ban event into the room.
|
||||||
|
pub(crate) async fn ban_user_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<ban_user::v3::Request>,
|
||||||
|
) -> Result<ban_user::v3::Response> {
|
||||||
|
let sender_user = body.sender_user();
|
||||||
|
|
||||||
|
if sender_user == body.user_id {
|
||||||
|
return Err!(Request(Forbidden("You cannot ban yourself.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
let state_lock = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.mutex
|
||||||
|
.lock(&body.room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let current_member_content = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(&body.room_id, &body.user_id)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|_| RoomMemberEventContent::new(MembershipState::Ban));
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(body.user_id.to_string(), &RoomMemberEventContent {
|
||||||
|
membership: MembershipState::Ban,
|
||||||
|
reason: body.reason.clone(),
|
||||||
|
displayname: None, // display name may be offensive
|
||||||
|
avatar_url: None, // avatar may be offensive
|
||||||
|
is_direct: None,
|
||||||
|
join_authorized_via_users_server: None,
|
||||||
|
third_party_invite: None,
|
||||||
|
..current_member_content
|
||||||
|
}),
|
||||||
|
sender_user,
|
||||||
|
&body.room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(ban_user::v3::Response::new())
|
||||||
|
}
|
||||||
70
src/api/client/membership/forget.rs
Normal file
70
src/api/client/membership/forget.rs
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use futures::pin_mut;
|
||||||
|
use ruma::{api::client::membership::forget_room, events::room::member::MembershipState};
|
||||||
|
use tuwunel_core::{Err, Result, is_matching, result::NotFound, utils::FutureBoolExt};
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/v3/rooms/{roomId}/forget`
|
||||||
|
///
|
||||||
|
/// Forgets about a room.
|
||||||
|
///
|
||||||
|
/// - If the sender user currently left the room: Stops sender user from
|
||||||
|
/// receiving information about the room
|
||||||
|
///
|
||||||
|
/// Note: Other devices of the user have no way of knowing the room was
|
||||||
|
/// forgotten, so this has to be called from every device
|
||||||
|
pub(crate) async fn forget_room_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<forget_room::v3::Request>,
|
||||||
|
) -> Result<forget_room::v3::Response> {
|
||||||
|
let user_id = body.sender_user();
|
||||||
|
let room_id = &body.room_id;
|
||||||
|
|
||||||
|
let joined = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_joined(user_id, room_id);
|
||||||
|
let knocked = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_knocked(user_id, room_id);
|
||||||
|
let invited = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_invited(user_id, room_id);
|
||||||
|
|
||||||
|
pin_mut!(joined, knocked, invited);
|
||||||
|
if joined.or(knocked).or(invited).await {
|
||||||
|
return Err!(Request(Unknown("You must leave the room before forgetting it")));
|
||||||
|
}
|
||||||
|
|
||||||
|
let membership = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(room_id, user_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
if membership.is_not_found() {
|
||||||
|
return Err!(Request(Unknown("No membership event was found, room was never joined")));
|
||||||
|
}
|
||||||
|
|
||||||
|
let non_membership = membership
|
||||||
|
.map(|member| member.membership)
|
||||||
|
.is_ok_and(is_matching!(MembershipState::Leave | MembershipState::Ban));
|
||||||
|
|
||||||
|
if non_membership
|
||||||
|
|| services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_left(user_id, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.forget(room_id, user_id);
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(forget_room::v3::Response::new())
|
||||||
|
}
|
||||||
245
src/api/client/membership/invite.rs
Normal file
245
src/api/client/membership/invite.rs
Normal file
@@ -0,0 +1,245 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use axum_client_ip::InsecureClientIp;
|
||||||
|
use futures::{FutureExt, join};
|
||||||
|
use ruma::{
|
||||||
|
OwnedServerName, RoomId, UserId,
|
||||||
|
api::{client::membership::invite_user, federation::membership::create_invite},
|
||||||
|
events::room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{
|
||||||
|
Err, Result, debug_error, err, info,
|
||||||
|
matrix::pdu::{PduBuilder, gen_event_id_canonical_json},
|
||||||
|
};
|
||||||
|
use tuwunel_service::Services;
|
||||||
|
|
||||||
|
use super::banned_room_check;
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/invite`
|
||||||
|
///
|
||||||
|
/// Tries to send an invite event into the room.
|
||||||
|
#[tracing::instrument(skip_all, fields(%client), name = "invite")]
|
||||||
|
pub(crate) async fn invite_user_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
InsecureClientIp(client): InsecureClientIp,
|
||||||
|
body: Ruma<invite_user::v3::Request>,
|
||||||
|
) -> Result<invite_user::v3::Response> {
|
||||||
|
let sender_user = body.sender_user();
|
||||||
|
|
||||||
|
if !services.users.is_admin(sender_user).await && services.config.block_non_admin_invites {
|
||||||
|
debug_error!(
|
||||||
|
"User {sender_user} is not an admin and attempted to send an invite to room {}",
|
||||||
|
&body.room_id
|
||||||
|
);
|
||||||
|
return Err!(Request(Forbidden("Invites are not allowed on this server.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
banned_room_check(
|
||||||
|
&services,
|
||||||
|
sender_user,
|
||||||
|
Some(&body.room_id),
|
||||||
|
body.room_id.server_name(),
|
||||||
|
client,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
match &body.recipient {
|
||||||
|
| invite_user::v3::InvitationRecipient::UserId { user_id } => {
|
||||||
|
let sender_ignored_recipient = services
|
||||||
|
.users
|
||||||
|
.user_is_ignored(sender_user, user_id);
|
||||||
|
let recipient_ignored_by_sender = services
|
||||||
|
.users
|
||||||
|
.user_is_ignored(user_id, sender_user);
|
||||||
|
|
||||||
|
let (sender_ignored_recipient, recipient_ignored_by_sender) =
|
||||||
|
join!(sender_ignored_recipient, recipient_ignored_by_sender);
|
||||||
|
|
||||||
|
if sender_ignored_recipient {
|
||||||
|
return Ok(invite_user::v3::Response {});
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Ok(target_user_membership) = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(&body.room_id, user_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
if target_user_membership.membership == MembershipState::Ban {
|
||||||
|
return Err!(Request(Forbidden("User is banned from this room.")));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if recipient_ignored_by_sender {
|
||||||
|
// silently drop the invite to the recipient if they've been ignored by the
|
||||||
|
// sender, pretend it worked
|
||||||
|
return Ok(invite_user::v3::Response {});
|
||||||
|
}
|
||||||
|
|
||||||
|
invite_helper(
|
||||||
|
&services,
|
||||||
|
sender_user,
|
||||||
|
user_id,
|
||||||
|
&body.room_id,
|
||||||
|
body.reason.clone(),
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.boxed()
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(invite_user::v3::Response {})
|
||||||
|
},
|
||||||
|
| _ => {
|
||||||
|
Err!(Request(NotFound("User not found.")))
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) async fn invite_helper(
|
||||||
|
services: &Services,
|
||||||
|
sender_user: &UserId,
|
||||||
|
user_id: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
reason: Option<String>,
|
||||||
|
is_direct: bool,
|
||||||
|
) -> Result {
|
||||||
|
if !services.users.is_admin(sender_user).await && services.config.block_non_admin_invites {
|
||||||
|
info!(
|
||||||
|
"User {sender_user} is not an admin and attempted to send an invite to room \
|
||||||
|
{room_id}"
|
||||||
|
);
|
||||||
|
return Err!(Request(Forbidden("Invites are not allowed on this server.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
if !services.globals.user_is_local(user_id) {
|
||||||
|
let (pdu, pdu_json, invite_room_state) = {
|
||||||
|
let state_lock = services.rooms.state.mutex.lock(room_id).await;
|
||||||
|
|
||||||
|
let content = RoomMemberEventContent {
|
||||||
|
avatar_url: services.users.avatar_url(user_id).await.ok(),
|
||||||
|
is_direct: Some(is_direct),
|
||||||
|
reason,
|
||||||
|
..RoomMemberEventContent::new(MembershipState::Invite)
|
||||||
|
};
|
||||||
|
|
||||||
|
let (pdu, pdu_json) = services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.create_hash_and_sign_event(
|
||||||
|
PduBuilder::state(user_id.to_string(), &content),
|
||||||
|
sender_user,
|
||||||
|
room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let invite_room_state = services.rooms.state.summary_stripped(&pdu).await;
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
(pdu, pdu_json, invite_room_state)
|
||||||
|
};
|
||||||
|
|
||||||
|
let room_version_id = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.get_room_version(room_id)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let response = services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(user_id.server_name(), create_invite::v2::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
event_id: (*pdu.event_id).to_owned(),
|
||||||
|
room_version: room_version_id.clone(),
|
||||||
|
event: services
|
||||||
|
.sending
|
||||||
|
.convert_to_outgoing_federation_event(pdu_json.clone())
|
||||||
|
.await,
|
||||||
|
invite_room_state,
|
||||||
|
via: services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.servers_route_via(room_id)
|
||||||
|
.await
|
||||||
|
.ok(),
|
||||||
|
})
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
// We do not add the event_id field to the pdu here because of signature and
|
||||||
|
// hashes checks
|
||||||
|
let (event_id, value) = gen_event_id_canonical_json(&response.event, &room_version_id)
|
||||||
|
.map_err(|e| {
|
||||||
|
err!(Request(BadJson(warn!("Could not convert event to canonical JSON: {e}"))))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
if pdu.event_id != event_id {
|
||||||
|
return Err!(Request(BadJson(warn!(
|
||||||
|
%pdu.event_id, %event_id,
|
||||||
|
"Server {} sent event with wrong event ID",
|
||||||
|
user_id.server_name()
|
||||||
|
))));
|
||||||
|
}
|
||||||
|
|
||||||
|
let origin: OwnedServerName = serde_json::from_value(serde_json::to_value(
|
||||||
|
value
|
||||||
|
.get("origin")
|
||||||
|
.ok_or_else(|| err!(Request(BadJson("Event missing origin field."))))?,
|
||||||
|
)?)
|
||||||
|
.map_err(|e| {
|
||||||
|
err!(Request(BadJson(warn!("Origin field in event is not a valid server name: {e}"))))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let pdu_id = services
|
||||||
|
.rooms
|
||||||
|
.event_handler
|
||||||
|
.handle_incoming_pdu(&origin, room_id, &event_id, value, true)
|
||||||
|
.boxed()
|
||||||
|
.await?
|
||||||
|
.ok_or_else(|| {
|
||||||
|
err!(Request(InvalidParam("Could not accept incoming PDU as timeline event.")))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
return services
|
||||||
|
.sending
|
||||||
|
.send_pdu_room(room_id, &pdu_id)
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
|
||||||
|
if !services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_joined(sender_user, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
return Err!(Request(Forbidden(
|
||||||
|
"You must be joined in the room you are trying to invite from."
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
let state_lock = services.rooms.state.mutex.lock(room_id).await;
|
||||||
|
|
||||||
|
let content = RoomMemberEventContent {
|
||||||
|
displayname: services.users.displayname(user_id).await.ok(),
|
||||||
|
avatar_url: services.users.avatar_url(user_id).await.ok(),
|
||||||
|
blurhash: services.users.blurhash(user_id).await.ok(),
|
||||||
|
is_direct: Some(is_direct),
|
||||||
|
reason,
|
||||||
|
..RoomMemberEventContent::new(MembershipState::Invite)
|
||||||
|
};
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(user_id.to_string(), &content),
|
||||||
|
sender_user,
|
||||||
|
room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
1005
src/api/client/membership/join.rs
Normal file
1005
src/api/client/membership/join.rs
Normal file
File diff suppressed because it is too large
Load Diff
66
src/api/client/membership/kick.rs
Normal file
66
src/api/client/membership/kick.rs
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use ruma::{
|
||||||
|
api::client::membership::kick_user,
|
||||||
|
events::room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{Err, Result, matrix::pdu::PduBuilder};
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/kick`
|
||||||
|
///
|
||||||
|
/// Tries to send a kick event into the room.
|
||||||
|
pub(crate) async fn kick_user_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<kick_user::v3::Request>,
|
||||||
|
) -> Result<kick_user::v3::Response> {
|
||||||
|
let state_lock = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.mutex
|
||||||
|
.lock(&body.room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let Ok(event) = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(&body.room_id, &body.user_id)
|
||||||
|
.await
|
||||||
|
else {
|
||||||
|
// copy synapse's behaviour of returning 200 without any change to the state
|
||||||
|
// instead of erroring on left users
|
||||||
|
return Ok(kick_user::v3::Response::new());
|
||||||
|
};
|
||||||
|
|
||||||
|
if !matches!(
|
||||||
|
event.membership,
|
||||||
|
MembershipState::Invite | MembershipState::Knock | MembershipState::Join,
|
||||||
|
) {
|
||||||
|
return Err!(Request(Forbidden(
|
||||||
|
"Cannot kick a user who is not apart of the room (current membership: {})",
|
||||||
|
event.membership
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(body.user_id.to_string(), &RoomMemberEventContent {
|
||||||
|
membership: MembershipState::Leave,
|
||||||
|
reason: body.reason.clone(),
|
||||||
|
is_direct: None,
|
||||||
|
join_authorized_via_users_server: None,
|
||||||
|
third_party_invite: None,
|
||||||
|
..event
|
||||||
|
}),
|
||||||
|
body.sender_user(),
|
||||||
|
&body.room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(kick_user::v3::Response::new())
|
||||||
|
}
|
||||||
677
src/api/client/membership/knock.rs
Normal file
677
src/api/client/membership/knock.rs
Normal file
@@ -0,0 +1,677 @@
|
|||||||
|
use std::{borrow::Borrow, collections::HashMap, iter::once, sync::Arc};
|
||||||
|
|
||||||
|
use axum::extract::State;
|
||||||
|
use axum_client_ip::InsecureClientIp;
|
||||||
|
use futures::{FutureExt, StreamExt};
|
||||||
|
use ruma::{
|
||||||
|
CanonicalJsonObject, CanonicalJsonValue, OwnedEventId, OwnedRoomId, OwnedServerName, RoomId,
|
||||||
|
RoomVersionId, UserId,
|
||||||
|
api::{
|
||||||
|
client::knock::knock_room,
|
||||||
|
federation::{self},
|
||||||
|
},
|
||||||
|
canonical_json::to_canonical_value,
|
||||||
|
events::{
|
||||||
|
StateEventType,
|
||||||
|
room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{
|
||||||
|
Err, Result, debug, debug_info, debug_warn, err, info,
|
||||||
|
matrix::pdu::{PduBuilder, PduEvent, gen_event_id},
|
||||||
|
result::FlatOk,
|
||||||
|
trace,
|
||||||
|
utils::{self, shuffle, stream::IterStream},
|
||||||
|
warn,
|
||||||
|
};
|
||||||
|
use tuwunel_service::{
|
||||||
|
Services,
|
||||||
|
rooms::{
|
||||||
|
state::RoomMutexGuard,
|
||||||
|
state_compressor::{CompressedState, HashSetCompressStateEvent},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::banned_room_check;
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/*/knock/{roomIdOrAlias}`
|
||||||
|
///
|
||||||
|
/// Tries to knock the room to ask permission to join for the sender user.
|
||||||
|
#[tracing::instrument(skip_all, fields(%client), name = "knock")]
|
||||||
|
pub(crate) async fn knock_room_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
InsecureClientIp(client): InsecureClientIp,
|
||||||
|
body: Ruma<knock_room::v3::Request>,
|
||||||
|
) -> Result<knock_room::v3::Response> {
|
||||||
|
let sender_user = body.sender_user();
|
||||||
|
let body = &body.body;
|
||||||
|
|
||||||
|
let (servers, room_id) = match OwnedRoomId::try_from(body.room_id_or_alias.clone()) {
|
||||||
|
| Ok(room_id) => {
|
||||||
|
banned_room_check(
|
||||||
|
&services,
|
||||||
|
sender_user,
|
||||||
|
Some(&room_id),
|
||||||
|
room_id.server_name(),
|
||||||
|
client,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let mut servers = body.via.clone();
|
||||||
|
servers.extend(
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.servers_invite_via(&room_id)
|
||||||
|
.map(ToOwned::to_owned)
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.await,
|
||||||
|
);
|
||||||
|
|
||||||
|
servers.extend(
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.invite_state(sender_user, &room_id)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
.iter()
|
||||||
|
.filter_map(|event| event.get_field("sender").ok().flatten())
|
||||||
|
.filter_map(|sender: &str| UserId::parse(sender).ok())
|
||||||
|
.map(|user| user.server_name().to_owned()),
|
||||||
|
);
|
||||||
|
|
||||||
|
if let Some(server) = room_id.server_name() {
|
||||||
|
servers.push(server.to_owned());
|
||||||
|
}
|
||||||
|
|
||||||
|
servers.sort_unstable();
|
||||||
|
servers.dedup();
|
||||||
|
shuffle(&mut servers);
|
||||||
|
|
||||||
|
(servers, room_id)
|
||||||
|
},
|
||||||
|
| Err(room_alias) => {
|
||||||
|
let (room_id, mut servers) = services
|
||||||
|
.rooms
|
||||||
|
.alias
|
||||||
|
.resolve_alias(&room_alias, Some(body.via.clone()))
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
banned_room_check(
|
||||||
|
&services,
|
||||||
|
sender_user,
|
||||||
|
Some(&room_id),
|
||||||
|
Some(room_alias.server_name()),
|
||||||
|
client,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let addl_via_servers = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.servers_invite_via(&room_id)
|
||||||
|
.map(ToOwned::to_owned);
|
||||||
|
|
||||||
|
let addl_state_servers = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.invite_state(sender_user, &room_id)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default();
|
||||||
|
|
||||||
|
let mut addl_servers: Vec<_> = addl_state_servers
|
||||||
|
.iter()
|
||||||
|
.map(|event| event.get_field("sender"))
|
||||||
|
.filter_map(FlatOk::flat_ok)
|
||||||
|
.map(|user: &UserId| user.server_name().to_owned())
|
||||||
|
.stream()
|
||||||
|
.chain(addl_via_servers)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
addl_servers.sort_unstable();
|
||||||
|
addl_servers.dedup();
|
||||||
|
shuffle(&mut addl_servers);
|
||||||
|
servers.append(&mut addl_servers);
|
||||||
|
|
||||||
|
(servers, room_id)
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
knock_room_by_id_helper(&services, sender_user, &room_id, body.reason.clone(), &servers)
|
||||||
|
.boxed()
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn knock_room_by_id_helper(
|
||||||
|
services: &Services,
|
||||||
|
sender_user: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
reason: Option<String>,
|
||||||
|
servers: &[OwnedServerName],
|
||||||
|
) -> Result<knock_room::v3::Response> {
|
||||||
|
let state_lock = services.rooms.state.mutex.lock(room_id).await;
|
||||||
|
|
||||||
|
if services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_invited(sender_user, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
debug_warn!("{sender_user} is already invited in {room_id} but attempted to knock");
|
||||||
|
return Err!(Request(Forbidden(
|
||||||
|
"You cannot knock on a room you are already invited/accepted to."
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
if services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_joined(sender_user, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
debug_warn!("{sender_user} is already joined in {room_id} but attempted to knock");
|
||||||
|
return Err!(Request(Forbidden("You cannot knock on a room you are already joined in.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
if services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_knocked(sender_user, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
debug_warn!("{sender_user} is already knocked in {room_id}");
|
||||||
|
return Ok(knock_room::v3::Response { room_id: room_id.into() });
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Ok(membership) = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(room_id, sender_user)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
if membership.membership == MembershipState::Ban {
|
||||||
|
debug_warn!("{sender_user} is banned from {room_id} but attempted to knock");
|
||||||
|
return Err!(Request(Forbidden("You cannot knock on a room you are banned from.")));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let server_in_room = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.server_in_room(services.globals.server_name(), room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let local_knock = server_in_room
|
||||||
|
|| servers.is_empty()
|
||||||
|
|| (servers.len() == 1 && services.globals.server_is_ours(&servers[0]));
|
||||||
|
|
||||||
|
if local_knock {
|
||||||
|
knock_room_helper_local(services, sender_user, room_id, reason, servers, state_lock)
|
||||||
|
.boxed()
|
||||||
|
.await?;
|
||||||
|
} else {
|
||||||
|
knock_room_helper_remote(services, sender_user, room_id, reason, servers, state_lock)
|
||||||
|
.boxed()
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(knock_room::v3::Response::new(room_id.to_owned()))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn knock_room_helper_local(
|
||||||
|
services: &Services,
|
||||||
|
sender_user: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
reason: Option<String>,
|
||||||
|
servers: &[OwnedServerName],
|
||||||
|
state_lock: RoomMutexGuard,
|
||||||
|
) -> Result {
|
||||||
|
debug_info!("We can knock locally");
|
||||||
|
|
||||||
|
let room_version_id = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.get_room_version(room_id)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
if matches!(
|
||||||
|
room_version_id,
|
||||||
|
RoomVersionId::V1
|
||||||
|
| RoomVersionId::V2
|
||||||
|
| RoomVersionId::V3
|
||||||
|
| RoomVersionId::V4
|
||||||
|
| RoomVersionId::V5
|
||||||
|
| RoomVersionId::V6
|
||||||
|
) {
|
||||||
|
return Err!(Request(Forbidden("This room does not support knocking.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
let content = RoomMemberEventContent {
|
||||||
|
displayname: services.users.displayname(sender_user).await.ok(),
|
||||||
|
avatar_url: services.users.avatar_url(sender_user).await.ok(),
|
||||||
|
blurhash: services.users.blurhash(sender_user).await.ok(),
|
||||||
|
reason: reason.clone(),
|
||||||
|
..RoomMemberEventContent::new(MembershipState::Knock)
|
||||||
|
};
|
||||||
|
|
||||||
|
// Try normal knock first
|
||||||
|
let Err(error) = services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(sender_user.to_string(), &content),
|
||||||
|
sender_user,
|
||||||
|
room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
else {
|
||||||
|
return Ok(());
|
||||||
|
};
|
||||||
|
|
||||||
|
if servers.is_empty() || (servers.len() == 1 && services.globals.server_is_ours(&servers[0]))
|
||||||
|
{
|
||||||
|
return Err(error);
|
||||||
|
}
|
||||||
|
|
||||||
|
warn!("We couldn't do the knock locally, maybe federation can help to satisfy the knock");
|
||||||
|
|
||||||
|
let (make_knock_response, remote_server) =
|
||||||
|
make_knock_request(services, sender_user, room_id, servers).await?;
|
||||||
|
|
||||||
|
info!("make_knock finished");
|
||||||
|
|
||||||
|
let room_version_id = make_knock_response.room_version;
|
||||||
|
|
||||||
|
if !services
|
||||||
|
.server
|
||||||
|
.supported_room_version(&room_version_id)
|
||||||
|
{
|
||||||
|
return Err!(BadServerResponse(
|
||||||
|
"Remote room version {room_version_id} is not supported by conduwuit"
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut knock_event_stub = serde_json::from_str::<CanonicalJsonObject>(
|
||||||
|
make_knock_response.event.get(),
|
||||||
|
)
|
||||||
|
.map_err(|e| {
|
||||||
|
err!(BadServerResponse("Invalid make_knock event json received from server: {e:?}"))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"origin".to_owned(),
|
||||||
|
CanonicalJsonValue::String(services.globals.server_name().as_str().to_owned()),
|
||||||
|
);
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"origin_server_ts".to_owned(),
|
||||||
|
CanonicalJsonValue::Integer(
|
||||||
|
utils::millis_since_unix_epoch()
|
||||||
|
.try_into()
|
||||||
|
.expect("Timestamp is valid js_int value"),
|
||||||
|
),
|
||||||
|
);
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"content".to_owned(),
|
||||||
|
to_canonical_value(RoomMemberEventContent {
|
||||||
|
displayname: services.users.displayname(sender_user).await.ok(),
|
||||||
|
avatar_url: services.users.avatar_url(sender_user).await.ok(),
|
||||||
|
blurhash: services.users.blurhash(sender_user).await.ok(),
|
||||||
|
reason,
|
||||||
|
..RoomMemberEventContent::new(MembershipState::Knock)
|
||||||
|
})
|
||||||
|
.expect("event is valid, we just created it"),
|
||||||
|
);
|
||||||
|
|
||||||
|
// In order to create a compatible ref hash (EventID) the `hashes` field needs
|
||||||
|
// to be present
|
||||||
|
services
|
||||||
|
.server_keys
|
||||||
|
.hash_and_sign_event(&mut knock_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Generate event id
|
||||||
|
let event_id = gen_event_id(&knock_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Add event_id
|
||||||
|
knock_event_stub
|
||||||
|
.insert("event_id".to_owned(), CanonicalJsonValue::String(event_id.clone().into()));
|
||||||
|
|
||||||
|
// It has enough fields to be called a proper event now
|
||||||
|
let knock_event = knock_event_stub;
|
||||||
|
|
||||||
|
info!("Asking {remote_server} for send_knock in room {room_id}");
|
||||||
|
let send_knock_request = federation::knock::send_knock::v1::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
event_id: event_id.clone(),
|
||||||
|
pdu: services
|
||||||
|
.sending
|
||||||
|
.convert_to_outgoing_federation_event(knock_event.clone())
|
||||||
|
.await,
|
||||||
|
};
|
||||||
|
|
||||||
|
let send_knock_response = services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(&remote_server, send_knock_request)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("send_knock finished");
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.short
|
||||||
|
.get_or_create_shortroomid(room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
info!("Parsing knock event");
|
||||||
|
|
||||||
|
let parsed_knock_pdu = PduEvent::from_id_val(&event_id, knock_event.clone())
|
||||||
|
.map_err(|e| err!(BadServerResponse("Invalid knock event PDU: {e:?}")))?;
|
||||||
|
|
||||||
|
info!("Updating membership locally to knock state with provided stripped state events");
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.update_membership(
|
||||||
|
room_id,
|
||||||
|
sender_user,
|
||||||
|
parsed_knock_pdu
|
||||||
|
.get_content::<RoomMemberEventContent>()
|
||||||
|
.expect("we just created this"),
|
||||||
|
sender_user,
|
||||||
|
Some(send_knock_response.knock_room_state),
|
||||||
|
None,
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("Appending room knock event locally");
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.append_pdu(
|
||||||
|
&parsed_knock_pdu,
|
||||||
|
knock_event,
|
||||||
|
once(parsed_knock_pdu.event_id.borrow()),
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn knock_room_helper_remote(
|
||||||
|
services: &Services,
|
||||||
|
sender_user: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
reason: Option<String>,
|
||||||
|
servers: &[OwnedServerName],
|
||||||
|
state_lock: RoomMutexGuard,
|
||||||
|
) -> Result {
|
||||||
|
info!("Knocking {room_id} over federation.");
|
||||||
|
|
||||||
|
let (make_knock_response, remote_server) =
|
||||||
|
make_knock_request(services, sender_user, room_id, servers).await?;
|
||||||
|
|
||||||
|
info!("make_knock finished");
|
||||||
|
|
||||||
|
let room_version_id = make_knock_response.room_version;
|
||||||
|
|
||||||
|
if !services
|
||||||
|
.server
|
||||||
|
.supported_room_version(&room_version_id)
|
||||||
|
{
|
||||||
|
return Err!(BadServerResponse(
|
||||||
|
"Remote room version {room_version_id} is not supported by conduwuit"
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut knock_event_stub: CanonicalJsonObject =
|
||||||
|
serde_json::from_str(make_knock_response.event.get()).map_err(|e| {
|
||||||
|
err!(BadServerResponse("Invalid make_knock event json received from server: {e:?}"))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"origin".to_owned(),
|
||||||
|
CanonicalJsonValue::String(services.globals.server_name().as_str().to_owned()),
|
||||||
|
);
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"origin_server_ts".to_owned(),
|
||||||
|
CanonicalJsonValue::Integer(
|
||||||
|
utils::millis_since_unix_epoch()
|
||||||
|
.try_into()
|
||||||
|
.expect("Timestamp is valid js_int value"),
|
||||||
|
),
|
||||||
|
);
|
||||||
|
knock_event_stub.insert(
|
||||||
|
"content".to_owned(),
|
||||||
|
to_canonical_value(RoomMemberEventContent {
|
||||||
|
displayname: services.users.displayname(sender_user).await.ok(),
|
||||||
|
avatar_url: services.users.avatar_url(sender_user).await.ok(),
|
||||||
|
blurhash: services.users.blurhash(sender_user).await.ok(),
|
||||||
|
reason,
|
||||||
|
..RoomMemberEventContent::new(MembershipState::Knock)
|
||||||
|
})
|
||||||
|
.expect("event is valid, we just created it"),
|
||||||
|
);
|
||||||
|
|
||||||
|
// In order to create a compatible ref hash (EventID) the `hashes` field needs
|
||||||
|
// to be present
|
||||||
|
services
|
||||||
|
.server_keys
|
||||||
|
.hash_and_sign_event(&mut knock_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Generate event id
|
||||||
|
let event_id = gen_event_id(&knock_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Add event_id
|
||||||
|
knock_event_stub
|
||||||
|
.insert("event_id".to_owned(), CanonicalJsonValue::String(event_id.clone().into()));
|
||||||
|
|
||||||
|
// It has enough fields to be called a proper event now
|
||||||
|
let knock_event = knock_event_stub;
|
||||||
|
|
||||||
|
info!("Asking {remote_server} for send_knock in room {room_id}");
|
||||||
|
let send_knock_request = federation::knock::send_knock::v1::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
event_id: event_id.clone(),
|
||||||
|
pdu: services
|
||||||
|
.sending
|
||||||
|
.convert_to_outgoing_federation_event(knock_event.clone())
|
||||||
|
.await,
|
||||||
|
};
|
||||||
|
|
||||||
|
let send_knock_response = services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(&remote_server, send_knock_request)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("send_knock finished");
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.short
|
||||||
|
.get_or_create_shortroomid(room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
info!("Parsing knock event");
|
||||||
|
let parsed_knock_pdu = PduEvent::from_id_val(&event_id, knock_event.clone())
|
||||||
|
.map_err(|e| err!(BadServerResponse("Invalid knock event PDU: {e:?}")))?;
|
||||||
|
|
||||||
|
info!("Going through send_knock response knock state events");
|
||||||
|
let state = send_knock_response
|
||||||
|
.knock_room_state
|
||||||
|
.iter()
|
||||||
|
.map(|event| serde_json::from_str::<CanonicalJsonObject>(event.clone().into_json().get()))
|
||||||
|
.filter_map(Result::ok);
|
||||||
|
|
||||||
|
let mut state_map: HashMap<u64, OwnedEventId> = HashMap::new();
|
||||||
|
|
||||||
|
for event in state {
|
||||||
|
let Some(state_key) = event.get("state_key") else {
|
||||||
|
debug_warn!("send_knock stripped state event missing state_key: {event:?}");
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
let Some(event_type) = event.get("type") else {
|
||||||
|
debug_warn!("send_knock stripped state event missing event type: {event:?}");
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
|
let Ok(state_key) = serde_json::from_value::<String>(state_key.clone().into()) else {
|
||||||
|
debug_warn!("send_knock stripped state event has invalid state_key: {event:?}");
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
let Ok(event_type) = serde_json::from_value::<StateEventType>(event_type.clone().into())
|
||||||
|
else {
|
||||||
|
debug_warn!("send_knock stripped state event has invalid event type: {event:?}");
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
|
||||||
|
let event_id = gen_event_id(&event, &room_version_id)?;
|
||||||
|
let shortstatekey = services
|
||||||
|
.rooms
|
||||||
|
.short
|
||||||
|
.get_or_create_shortstatekey(&event_type, &state_key)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.outlier
|
||||||
|
.add_pdu_outlier(&event_id, &event);
|
||||||
|
state_map.insert(shortstatekey, event_id.clone());
|
||||||
|
}
|
||||||
|
|
||||||
|
info!("Compressing state from send_knock");
|
||||||
|
let compressed: CompressedState = services
|
||||||
|
.rooms
|
||||||
|
.state_compressor
|
||||||
|
.compress_state_events(
|
||||||
|
state_map
|
||||||
|
.iter()
|
||||||
|
.map(|(ssk, eid)| (ssk, eid.borrow())),
|
||||||
|
)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
debug!("Saving compressed state");
|
||||||
|
let HashSetCompressStateEvent {
|
||||||
|
shortstatehash: statehash_before_knock,
|
||||||
|
added,
|
||||||
|
removed,
|
||||||
|
} = services
|
||||||
|
.rooms
|
||||||
|
.state_compressor
|
||||||
|
.save_state(room_id, Arc::new(compressed))
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
debug!("Forcing state for new room");
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.force_state(room_id, statehash_before_knock, added, removed, &state_lock)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
let statehash_after_knock = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.append_to_state(&parsed_knock_pdu)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("Updating membership locally to knock state with provided stripped state events");
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.update_membership(
|
||||||
|
room_id,
|
||||||
|
sender_user,
|
||||||
|
parsed_knock_pdu
|
||||||
|
.get_content::<RoomMemberEventContent>()
|
||||||
|
.expect("we just created this"),
|
||||||
|
sender_user,
|
||||||
|
Some(send_knock_response.knock_room_state),
|
||||||
|
None,
|
||||||
|
false,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("Appending room knock event locally");
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.append_pdu(
|
||||||
|
&parsed_knock_pdu,
|
||||||
|
knock_event,
|
||||||
|
once(parsed_knock_pdu.event_id.borrow()),
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
info!("Setting final room state for new room");
|
||||||
|
// We set the room state after inserting the pdu, so that we never have a moment
|
||||||
|
// in time where events in the current room state do not exist
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.set_room_state(room_id, statehash_after_knock, &state_lock);
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn make_knock_request(
|
||||||
|
services: &Services,
|
||||||
|
sender_user: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
servers: &[OwnedServerName],
|
||||||
|
) -> Result<(federation::knock::create_knock_event_template::v1::Response, OwnedServerName)> {
|
||||||
|
let mut make_knock_response_and_server =
|
||||||
|
Err!(BadServerResponse("No server available to assist in knocking."));
|
||||||
|
|
||||||
|
let mut make_knock_counter: usize = 0;
|
||||||
|
|
||||||
|
for remote_server in servers {
|
||||||
|
if services.globals.server_is_ours(remote_server) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
info!("Asking {remote_server} for make_knock ({make_knock_counter})");
|
||||||
|
|
||||||
|
let make_knock_response = services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(
|
||||||
|
remote_server,
|
||||||
|
federation::knock::create_knock_event_template::v1::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
user_id: sender_user.to_owned(),
|
||||||
|
ver: services
|
||||||
|
.server
|
||||||
|
.supported_room_versions()
|
||||||
|
.collect(),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
trace!("make_knock response: {make_knock_response:?}");
|
||||||
|
make_knock_counter = make_knock_counter.saturating_add(1);
|
||||||
|
|
||||||
|
make_knock_response_and_server = make_knock_response.map(|r| (r, remote_server.clone()));
|
||||||
|
|
||||||
|
if make_knock_response_and_server.is_ok() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
if make_knock_counter > 40 {
|
||||||
|
warn!(
|
||||||
|
"50 servers failed to provide valid make_knock response, assuming no server can \
|
||||||
|
assist in knocking."
|
||||||
|
);
|
||||||
|
make_knock_response_and_server =
|
||||||
|
Err!(BadServerResponse("No server available to assist in knocking."));
|
||||||
|
|
||||||
|
return make_knock_response_and_server;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
make_knock_response_and_server
|
||||||
|
}
|
||||||
397
src/api/client/membership/leave.rs
Normal file
397
src/api/client/membership/leave.rs
Normal file
@@ -0,0 +1,397 @@
|
|||||||
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
use axum::extract::State;
|
||||||
|
use futures::{FutureExt, StreamExt, TryFutureExt, pin_mut};
|
||||||
|
use ruma::{
|
||||||
|
CanonicalJsonObject, CanonicalJsonValue, OwnedServerName, RoomId, RoomVersionId, UserId,
|
||||||
|
api::{
|
||||||
|
client::membership::leave_room,
|
||||||
|
federation::{self},
|
||||||
|
},
|
||||||
|
events::{
|
||||||
|
StateEventType,
|
||||||
|
room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{
|
||||||
|
Err, Result, debug_info, debug_warn, err,
|
||||||
|
matrix::pdu::{PduBuilder, gen_event_id},
|
||||||
|
utils::{self, FutureBoolExt, future::ReadyEqExt},
|
||||||
|
warn,
|
||||||
|
};
|
||||||
|
use tuwunel_service::Services;
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/v3/rooms/{roomId}/leave`
|
||||||
|
///
|
||||||
|
/// Tries to leave the sender user from a room.
|
||||||
|
///
|
||||||
|
/// - This should always work if the user is currently joined.
|
||||||
|
pub(crate) async fn leave_room_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<leave_room::v3::Request>,
|
||||||
|
) -> Result<leave_room::v3::Response> {
|
||||||
|
leave_room(&services, body.sender_user(), &body.room_id, body.reason.clone())
|
||||||
|
.boxed()
|
||||||
|
.await
|
||||||
|
.map(|()| leave_room::v3::Response::new())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make a user leave all their joined rooms, rescinds knocks, forgets all rooms,
|
||||||
|
// and ignores errors
|
||||||
|
pub async fn leave_all_rooms(services: &Services, user_id: &UserId) {
|
||||||
|
let rooms_joined = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_joined(user_id)
|
||||||
|
.map(ToOwned::to_owned);
|
||||||
|
|
||||||
|
let rooms_invited = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_invited(user_id)
|
||||||
|
.map(|(r, _)| r);
|
||||||
|
|
||||||
|
let rooms_knocked = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_knocked(user_id)
|
||||||
|
.map(|(r, _)| r);
|
||||||
|
|
||||||
|
let all_rooms: Vec<_> = rooms_joined
|
||||||
|
.chain(rooms_invited)
|
||||||
|
.chain(rooms_knocked)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
for room_id in all_rooms {
|
||||||
|
// ignore errors
|
||||||
|
if let Err(e) = leave_room(services, user_id, &room_id, None)
|
||||||
|
.boxed()
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
warn!(%user_id, "Failed to leave {room_id} remotely: {e}");
|
||||||
|
}
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.forget(&room_id, user_id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn leave_room(
|
||||||
|
services: &Services,
|
||||||
|
user_id: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
reason: Option<String>,
|
||||||
|
) -> Result {
|
||||||
|
let default_member_content = RoomMemberEventContent {
|
||||||
|
membership: MembershipState::Leave,
|
||||||
|
reason: reason.clone(),
|
||||||
|
join_authorized_via_users_server: None,
|
||||||
|
is_direct: None,
|
||||||
|
avatar_url: None,
|
||||||
|
displayname: None,
|
||||||
|
third_party_invite: None,
|
||||||
|
blurhash: None,
|
||||||
|
};
|
||||||
|
|
||||||
|
let is_banned = services.rooms.metadata.is_banned(room_id);
|
||||||
|
let is_disabled = services.rooms.metadata.is_disabled(room_id);
|
||||||
|
|
||||||
|
pin_mut!(is_banned, is_disabled);
|
||||||
|
if is_banned.or(is_disabled).await {
|
||||||
|
// the room is banned/disabled, the room must be rejected locally since we
|
||||||
|
// cant/dont want to federate with this server
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.update_membership(
|
||||||
|
room_id,
|
||||||
|
user_id,
|
||||||
|
default_member_content,
|
||||||
|
user_id,
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
let dont_have_room = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.server_in_room(services.globals.server_name(), room_id)
|
||||||
|
.eq(&false);
|
||||||
|
|
||||||
|
let not_knocked = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.is_knocked(user_id, room_id)
|
||||||
|
.eq(&false);
|
||||||
|
|
||||||
|
// Ask a remote server if we don't have this room and are not knocking on it
|
||||||
|
if dont_have_room.and(not_knocked).await {
|
||||||
|
if let Err(e) = remote_leave_room(services, user_id, room_id)
|
||||||
|
.boxed()
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
warn!(%user_id, "Failed to leave room {room_id} remotely: {e}");
|
||||||
|
// Don't tell the client about this error
|
||||||
|
}
|
||||||
|
|
||||||
|
let last_state = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.invite_state(user_id, room_id)
|
||||||
|
.or_else(|_| {
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.knock_state(user_id, room_id)
|
||||||
|
})
|
||||||
|
.or_else(|_| {
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.left_state(user_id, room_id)
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.ok();
|
||||||
|
|
||||||
|
// We always drop the invite, we can't rely on other servers
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.update_membership(
|
||||||
|
room_id,
|
||||||
|
user_id,
|
||||||
|
default_member_content,
|
||||||
|
user_id,
|
||||||
|
last_state,
|
||||||
|
None,
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
} else {
|
||||||
|
let state_lock = services.rooms.state.mutex.lock(room_id).await;
|
||||||
|
|
||||||
|
let Ok(event) = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.room_state_get_content::<RoomMemberEventContent>(
|
||||||
|
room_id,
|
||||||
|
&StateEventType::RoomMember,
|
||||||
|
user_id.as_str(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
else {
|
||||||
|
debug_warn!(
|
||||||
|
"Trying to leave a room you are not a member of, marking room as left locally."
|
||||||
|
);
|
||||||
|
|
||||||
|
return services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.update_membership(
|
||||||
|
room_id,
|
||||||
|
user_id,
|
||||||
|
default_member_content,
|
||||||
|
user_id,
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
true,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
};
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(user_id.to_string(), &RoomMemberEventContent {
|
||||||
|
membership: MembershipState::Leave,
|
||||||
|
reason,
|
||||||
|
join_authorized_via_users_server: None,
|
||||||
|
is_direct: None,
|
||||||
|
..event
|
||||||
|
}),
|
||||||
|
user_id,
|
||||||
|
room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn remote_leave_room(
|
||||||
|
services: &Services,
|
||||||
|
user_id: &UserId,
|
||||||
|
room_id: &RoomId,
|
||||||
|
) -> Result<()> {
|
||||||
|
let mut make_leave_response_and_server =
|
||||||
|
Err!(BadServerResponse("No remote server available to assist in leaving {room_id}."));
|
||||||
|
|
||||||
|
let mut servers: HashSet<OwnedServerName> = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.servers_invite_via(room_id)
|
||||||
|
.map(ToOwned::to_owned)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
match services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.invite_state(user_id, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
| Ok(invite_state) => {
|
||||||
|
servers.extend(
|
||||||
|
invite_state
|
||||||
|
.iter()
|
||||||
|
.filter_map(|event| event.get_field("sender").ok().flatten())
|
||||||
|
.filter_map(|sender: &str| UserId::parse(sender).ok())
|
||||||
|
.map(|user| user.server_name().to_owned()),
|
||||||
|
);
|
||||||
|
},
|
||||||
|
| _ => {
|
||||||
|
match services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.knock_state(user_id, room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
| Ok(knock_state) => {
|
||||||
|
servers.extend(
|
||||||
|
knock_state
|
||||||
|
.iter()
|
||||||
|
.filter_map(|event| event.get_field("sender").ok().flatten())
|
||||||
|
.filter_map(|sender: &str| UserId::parse(sender).ok())
|
||||||
|
.filter_map(|sender| {
|
||||||
|
if !services.globals.user_is_local(sender) {
|
||||||
|
Some(sender.server_name().to_owned())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
},
|
||||||
|
| _ => {},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(room_id_server_name) = room_id.server_name() {
|
||||||
|
servers.insert(room_id_server_name.to_owned());
|
||||||
|
}
|
||||||
|
|
||||||
|
debug_info!("servers in remote_leave_room: {servers:?}");
|
||||||
|
|
||||||
|
for remote_server in servers {
|
||||||
|
let make_leave_response = services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(
|
||||||
|
&remote_server,
|
||||||
|
federation::membership::prepare_leave_event::v1::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
user_id: user_id.to_owned(),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
make_leave_response_and_server = make_leave_response.map(|r| (r, remote_server));
|
||||||
|
|
||||||
|
if make_leave_response_and_server.is_ok() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let (make_leave_response, remote_server) = make_leave_response_and_server?;
|
||||||
|
|
||||||
|
let Some(room_version_id) = make_leave_response.room_version else {
|
||||||
|
return Err!(BadServerResponse(warn!(
|
||||||
|
"No room version was returned by {remote_server} for {room_id}, room version is \
|
||||||
|
likely not supported by conduwuit"
|
||||||
|
)));
|
||||||
|
};
|
||||||
|
|
||||||
|
if !services
|
||||||
|
.server
|
||||||
|
.supported_room_version(&room_version_id)
|
||||||
|
{
|
||||||
|
return Err!(BadServerResponse(warn!(
|
||||||
|
"Remote room version {room_version_id} for {room_id} is not supported by conduwuit",
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut leave_event_stub = serde_json::from_str::<CanonicalJsonObject>(
|
||||||
|
make_leave_response.event.get(),
|
||||||
|
)
|
||||||
|
.map_err(|e| {
|
||||||
|
err!(BadServerResponse(warn!(
|
||||||
|
"Invalid make_leave event json received from {remote_server} for {room_id}: {e:?}"
|
||||||
|
)))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
// TODO: Is origin needed?
|
||||||
|
leave_event_stub.insert(
|
||||||
|
"origin".to_owned(),
|
||||||
|
CanonicalJsonValue::String(services.globals.server_name().as_str().to_owned()),
|
||||||
|
);
|
||||||
|
leave_event_stub.insert(
|
||||||
|
"origin_server_ts".to_owned(),
|
||||||
|
CanonicalJsonValue::Integer(
|
||||||
|
utils::millis_since_unix_epoch()
|
||||||
|
.try_into()
|
||||||
|
.expect("Timestamp is valid js_int value"),
|
||||||
|
),
|
||||||
|
);
|
||||||
|
|
||||||
|
// room v3 and above removed the "event_id" field from remote PDU format
|
||||||
|
match room_version_id {
|
||||||
|
| RoomVersionId::V1 | RoomVersionId::V2 => {},
|
||||||
|
| _ => {
|
||||||
|
leave_event_stub.remove("event_id");
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// In order to create a compatible ref hash (EventID) the `hashes` field needs
|
||||||
|
// to be present
|
||||||
|
services
|
||||||
|
.server_keys
|
||||||
|
.hash_and_sign_event(&mut leave_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Generate event id
|
||||||
|
let event_id = gen_event_id(&leave_event_stub, &room_version_id)?;
|
||||||
|
|
||||||
|
// Add event_id back
|
||||||
|
leave_event_stub
|
||||||
|
.insert("event_id".to_owned(), CanonicalJsonValue::String(event_id.clone().into()));
|
||||||
|
|
||||||
|
// It has enough fields to be called a proper event now
|
||||||
|
let leave_event = leave_event_stub;
|
||||||
|
|
||||||
|
services
|
||||||
|
.sending
|
||||||
|
.send_federation_request(
|
||||||
|
&remote_server,
|
||||||
|
federation::membership::create_leave_event::v2::Request {
|
||||||
|
room_id: room_id.to_owned(),
|
||||||
|
event_id,
|
||||||
|
pdu: services
|
||||||
|
.sending
|
||||||
|
.convert_to_outgoing_federation_event(leave_event.clone())
|
||||||
|
.await,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
150
src/api/client/membership/members.rs
Normal file
150
src/api/client/membership/members.rs
Normal file
@@ -0,0 +1,150 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use futures::{StreamExt, future::join};
|
||||||
|
use ruma::{
|
||||||
|
api::client::membership::{
|
||||||
|
get_member_events::{self, v3::MembershipEventFilter},
|
||||||
|
joined_members::{self, v3::RoomMember},
|
||||||
|
},
|
||||||
|
events::{
|
||||||
|
StateEventType,
|
||||||
|
room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{
|
||||||
|
Err, Event, Result, at,
|
||||||
|
matrix::pdu::PduEvent,
|
||||||
|
utils::{
|
||||||
|
future::TryExtExt,
|
||||||
|
stream::{BroadbandExt, ReadyExt},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/members`
|
||||||
|
///
|
||||||
|
/// Lists all joined users in a room (TODO: at a specific point in time, with a
|
||||||
|
/// specific membership).
|
||||||
|
///
|
||||||
|
/// - Only works if the user is currently joined
|
||||||
|
pub(crate) async fn get_member_events_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<get_member_events::v3::Request>,
|
||||||
|
) -> Result<get_member_events::v3::Response> {
|
||||||
|
let sender_user = body.sender_user();
|
||||||
|
let membership = body.membership.as_ref();
|
||||||
|
let not_membership = body.not_membership.as_ref();
|
||||||
|
|
||||||
|
if !services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.user_can_see_state_events(sender_user, &body.room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
return Err!(Request(Forbidden("You don't have permission to view this room.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(get_member_events::v3::Response {
|
||||||
|
chunk: services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.room_state_full(&body.room_id)
|
||||||
|
.ready_filter_map(Result::ok)
|
||||||
|
.ready_filter(|((ty, _), _)| *ty == StateEventType::RoomMember)
|
||||||
|
.map(at!(1))
|
||||||
|
.ready_filter_map(|pdu| membership_filter(pdu, membership, not_membership))
|
||||||
|
.map(Event::into_format)
|
||||||
|
.collect()
|
||||||
|
.await,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/joined_members`
|
||||||
|
///
|
||||||
|
/// Lists all members of a room.
|
||||||
|
///
|
||||||
|
/// - The sender user must be in the room
|
||||||
|
/// - TODO: An appservice just needs a puppet joined
|
||||||
|
pub(crate) async fn joined_members_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<joined_members::v3::Request>,
|
||||||
|
) -> Result<joined_members::v3::Response> {
|
||||||
|
if !services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.user_can_see_state_events(body.sender_user(), &body.room_id)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
return Err!(Request(Forbidden("You don't have permission to view this room.")));
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(joined_members::v3::Response {
|
||||||
|
joined: services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.room_members(&body.room_id)
|
||||||
|
.map(ToOwned::to_owned)
|
||||||
|
.broad_then(|user_id| async move {
|
||||||
|
let (display_name, avatar_url) = join(
|
||||||
|
services.users.displayname(&user_id).ok(),
|
||||||
|
services.users.avatar_url(&user_id).ok(),
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
(user_id, RoomMember { display_name, avatar_url })
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
.await,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn membership_filter(
|
||||||
|
pdu: PduEvent,
|
||||||
|
for_membership: Option<&MembershipEventFilter>,
|
||||||
|
not_membership: Option<&MembershipEventFilter>,
|
||||||
|
) -> Option<PduEvent> {
|
||||||
|
let membership_state_filter = match for_membership {
|
||||||
|
| Some(MembershipEventFilter::Ban) => MembershipState::Ban,
|
||||||
|
| Some(MembershipEventFilter::Invite) => MembershipState::Invite,
|
||||||
|
| Some(MembershipEventFilter::Knock) => MembershipState::Knock,
|
||||||
|
| Some(MembershipEventFilter::Leave) => MembershipState::Leave,
|
||||||
|
| Some(_) | None => MembershipState::Join,
|
||||||
|
};
|
||||||
|
|
||||||
|
let not_membership_state_filter = match not_membership {
|
||||||
|
| Some(MembershipEventFilter::Ban) => MembershipState::Ban,
|
||||||
|
| Some(MembershipEventFilter::Invite) => MembershipState::Invite,
|
||||||
|
| Some(MembershipEventFilter::Join) => MembershipState::Join,
|
||||||
|
| Some(MembershipEventFilter::Knock) => MembershipState::Knock,
|
||||||
|
| Some(_) | None => MembershipState::Leave,
|
||||||
|
};
|
||||||
|
|
||||||
|
let evt_membership = pdu
|
||||||
|
.get_content::<RoomMemberEventContent>()
|
||||||
|
.ok()?
|
||||||
|
.membership;
|
||||||
|
|
||||||
|
if for_membership.is_some() && not_membership.is_some() {
|
||||||
|
if membership_state_filter != evt_membership
|
||||||
|
|| not_membership_state_filter == evt_membership
|
||||||
|
{
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(pdu)
|
||||||
|
}
|
||||||
|
} else if for_membership.is_some() && not_membership.is_none() {
|
||||||
|
if membership_state_filter != evt_membership {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(pdu)
|
||||||
|
}
|
||||||
|
} else if not_membership.is_some() && for_membership.is_none() {
|
||||||
|
if not_membership_state_filter == evt_membership {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(pdu)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Some(pdu)
|
||||||
|
}
|
||||||
|
}
|
||||||
169
src/api/client/membership/mod.rs
Normal file
169
src/api/client/membership/mod.rs
Normal file
@@ -0,0 +1,169 @@
|
|||||||
|
mod ban;
|
||||||
|
mod forget;
|
||||||
|
mod invite;
|
||||||
|
mod join;
|
||||||
|
mod kick;
|
||||||
|
mod knock;
|
||||||
|
mod leave;
|
||||||
|
mod members;
|
||||||
|
mod unban;
|
||||||
|
|
||||||
|
use std::net::IpAddr;
|
||||||
|
|
||||||
|
use axum::extract::State;
|
||||||
|
use futures::{FutureExt, StreamExt};
|
||||||
|
use ruma::{OwnedRoomId, RoomId, ServerName, UserId, api::client::membership::joined_rooms};
|
||||||
|
use tuwunel_core::{Err, Result, warn};
|
||||||
|
use tuwunel_service::Services;
|
||||||
|
|
||||||
|
pub(crate) use self::{
|
||||||
|
ban::ban_user_route,
|
||||||
|
forget::forget_room_route,
|
||||||
|
invite::{invite_helper, invite_user_route},
|
||||||
|
join::{join_room_by_id_or_alias_route, join_room_by_id_route},
|
||||||
|
kick::kick_user_route,
|
||||||
|
knock::knock_room_route,
|
||||||
|
leave::leave_room_route,
|
||||||
|
members::{get_member_events_route, joined_members_route},
|
||||||
|
unban::unban_user_route,
|
||||||
|
};
|
||||||
|
pub use self::{
|
||||||
|
join::join_room_by_id_helper,
|
||||||
|
leave::{leave_all_rooms, leave_room},
|
||||||
|
};
|
||||||
|
use crate::{Ruma, client::full_user_deactivate};
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/joined_rooms`
|
||||||
|
///
|
||||||
|
/// Lists all rooms the user has joined.
|
||||||
|
pub(crate) async fn joined_rooms_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<joined_rooms::v3::Request>,
|
||||||
|
) -> Result<joined_rooms::v3::Response> {
|
||||||
|
Ok(joined_rooms::v3::Response {
|
||||||
|
joined_rooms: services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_joined(body.sender_user())
|
||||||
|
.map(ToOwned::to_owned)
|
||||||
|
.collect()
|
||||||
|
.await,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Checks if the room is banned in any way possible and the sender user is not
|
||||||
|
/// an admin.
|
||||||
|
///
|
||||||
|
/// Performs automatic deactivation if `auto_deactivate_banned_room_attempts` is
|
||||||
|
/// enabled
|
||||||
|
#[tracing::instrument(skip(services))]
|
||||||
|
pub(crate) async fn banned_room_check(
|
||||||
|
services: &Services,
|
||||||
|
user_id: &UserId,
|
||||||
|
room_id: Option<&RoomId>,
|
||||||
|
server_name: Option<&ServerName>,
|
||||||
|
client_ip: IpAddr,
|
||||||
|
) -> Result {
|
||||||
|
if services.users.is_admin(user_id).await {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(room_id) = room_id {
|
||||||
|
if services.rooms.metadata.is_banned(room_id).await
|
||||||
|
|| services
|
||||||
|
.config
|
||||||
|
.forbidden_remote_server_names
|
||||||
|
.is_match(
|
||||||
|
room_id
|
||||||
|
.server_name()
|
||||||
|
.expect("legacy room mxid")
|
||||||
|
.host(),
|
||||||
|
) {
|
||||||
|
warn!(
|
||||||
|
"User {user_id} who is not an admin attempted to send an invite for or \
|
||||||
|
attempted to join a banned room or banned room server name: {room_id}"
|
||||||
|
);
|
||||||
|
|
||||||
|
if services
|
||||||
|
.server
|
||||||
|
.config
|
||||||
|
.auto_deactivate_banned_room_attempts
|
||||||
|
{
|
||||||
|
warn!(
|
||||||
|
"Automatically deactivating user {user_id} due to attempted banned room join"
|
||||||
|
);
|
||||||
|
|
||||||
|
if services.server.config.admin_room_notices {
|
||||||
|
services
|
||||||
|
.admin
|
||||||
|
.send_text(&format!(
|
||||||
|
"Automatically deactivating user {user_id} due to attempted banned \
|
||||||
|
room join from IP {client_ip}"
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
|
||||||
|
let all_joined_rooms: Vec<OwnedRoomId> = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_joined(user_id)
|
||||||
|
.map(Into::into)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
full_user_deactivate(services, user_id, &all_joined_rooms)
|
||||||
|
.boxed()
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
return Err!(Request(Forbidden("This room is banned on this homeserver.")));
|
||||||
|
}
|
||||||
|
} else if let Some(server_name) = server_name {
|
||||||
|
if services
|
||||||
|
.config
|
||||||
|
.forbidden_remote_server_names
|
||||||
|
.is_match(server_name.host())
|
||||||
|
{
|
||||||
|
warn!(
|
||||||
|
"User {user_id} who is not an admin tried joining a room which has the server \
|
||||||
|
name {server_name} that is globally forbidden. Rejecting.",
|
||||||
|
);
|
||||||
|
|
||||||
|
if services
|
||||||
|
.server
|
||||||
|
.config
|
||||||
|
.auto_deactivate_banned_room_attempts
|
||||||
|
{
|
||||||
|
warn!(
|
||||||
|
"Automatically deactivating user {user_id} due to attempted banned room join"
|
||||||
|
);
|
||||||
|
|
||||||
|
if services.server.config.admin_room_notices {
|
||||||
|
services
|
||||||
|
.admin
|
||||||
|
.send_text(&format!(
|
||||||
|
"Automatically deactivating user {user_id} due to attempted banned \
|
||||||
|
room join from IP {client_ip}"
|
||||||
|
))
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
|
||||||
|
let all_joined_rooms: Vec<OwnedRoomId> = services
|
||||||
|
.rooms
|
||||||
|
.state_cache
|
||||||
|
.rooms_joined(user_id)
|
||||||
|
.map(Into::into)
|
||||||
|
.collect()
|
||||||
|
.await;
|
||||||
|
|
||||||
|
full_user_deactivate(services, user_id, &all_joined_rooms)
|
||||||
|
.boxed()
|
||||||
|
.await?;
|
||||||
|
}
|
||||||
|
|
||||||
|
return Err!(Request(Forbidden("This remote server is banned on this homeserver.")));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
59
src/api/client/membership/unban.rs
Normal file
59
src/api/client/membership/unban.rs
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
use axum::extract::State;
|
||||||
|
use ruma::{
|
||||||
|
api::client::membership::unban_user,
|
||||||
|
events::room::member::{MembershipState, RoomMemberEventContent},
|
||||||
|
};
|
||||||
|
use tuwunel_core::{Err, Result, matrix::pdu::PduBuilder};
|
||||||
|
|
||||||
|
use crate::Ruma;
|
||||||
|
|
||||||
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/unban`
|
||||||
|
///
|
||||||
|
/// Tries to send an unban event into the room.
|
||||||
|
pub(crate) async fn unban_user_route(
|
||||||
|
State(services): State<crate::State>,
|
||||||
|
body: Ruma<unban_user::v3::Request>,
|
||||||
|
) -> Result<unban_user::v3::Response> {
|
||||||
|
let state_lock = services
|
||||||
|
.rooms
|
||||||
|
.state
|
||||||
|
.mutex
|
||||||
|
.lock(&body.room_id)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
let current_member_content = services
|
||||||
|
.rooms
|
||||||
|
.state_accessor
|
||||||
|
.get_member(&body.room_id, &body.user_id)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|_| RoomMemberEventContent::new(MembershipState::Leave));
|
||||||
|
|
||||||
|
if current_member_content.membership != MembershipState::Ban {
|
||||||
|
return Err!(Request(Forbidden(
|
||||||
|
"Cannot unban a user who is not banned (current membership: {})",
|
||||||
|
current_member_content.membership
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
|
||||||
|
services
|
||||||
|
.rooms
|
||||||
|
.timeline
|
||||||
|
.build_and_append_pdu(
|
||||||
|
PduBuilder::state(body.user_id.to_string(), &RoomMemberEventContent {
|
||||||
|
membership: MembershipState::Leave,
|
||||||
|
reason: body.reason.clone(),
|
||||||
|
join_authorized_via_users_server: None,
|
||||||
|
third_party_invite: None,
|
||||||
|
is_direct: None,
|
||||||
|
..current_member_content
|
||||||
|
}),
|
||||||
|
body.sender_user(),
|
||||||
|
&body.room_id,
|
||||||
|
&state_lock,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
drop(state_lock);
|
||||||
|
|
||||||
|
Ok(unban_user::v3::Response::new())
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user