fix(membership): perform stricter checks when choosing an authorized user

Signed-off-by: strawberry <strawberry@puppygock.gay>
This commit is contained in:
Matthias Ahouansou 2024-04-03 18:48:51 -04:00 committed by strawberry
parent 3692b2790c
commit 290506a970

View file

@ -20,7 +20,6 @@ use ruma::{
room::{ room::{
join_rules::{AllowRule, JoinRule, RoomJoinRulesEventContent}, join_rules::{AllowRule, JoinRule, RoomJoinRulesEventContent},
member::{MembershipState, RoomMemberEventContent}, member::{MembershipState, RoomMemberEventContent},
power_levels::RoomPowerLevelsEventContent,
}, },
StateEventType, TimelineEventType, StateEventType, TimelineEventType,
}, },
@ -880,11 +879,6 @@ pub(crate) async fn join_room_by_id_helper(
.rooms .rooms
.state_accessor .state_accessor
.room_state_get(room_id, &StateEventType::RoomJoinRules, "")?; .room_state_get(room_id, &StateEventType::RoomJoinRules, "")?;
let power_levels_event =
services()
.rooms
.state_accessor
.room_state_get(room_id, &StateEventType::RoomPowerLevels, "")?;
let join_rules_event_content: Option<RoomJoinRulesEventContent> = join_rules_event let join_rules_event_content: Option<RoomJoinRulesEventContent> = join_rules_event
.as_ref() .as_ref()
@ -895,15 +889,6 @@ pub(crate) async fn join_room_by_id_helper(
}) })
}) })
.transpose()?; .transpose()?;
let power_levels_event_content: Option<RoomPowerLevelsEventContent> = power_levels_event
.as_ref()
.map(|power_levels_event| {
serde_json::from_str(power_levels_event.content.get()).map_err(|e| {
warn!("Invalid power levels event: {}", e);
Error::bad_database("Invalid power levels event in db.")
})
})
.transpose()?;
let restriction_rooms = match join_rules_event_content { let restriction_rooms = match join_rules_event_content {
Some(RoomJoinRulesEventContent { Some(RoomJoinRulesEventContent {
@ -919,51 +904,30 @@ pub(crate) async fn join_room_by_id_helper(
_ => Vec::new(), _ => Vec::new(),
}; };
let authorized_user = restriction_rooms let authorized_user = restriction_rooms.iter().find_map(|restriction_room_id| {
.iter() if !services()
.find_map(|restriction_room_id| { .rooms
if !services() .state_cache
.is_invited(sender_user, restriction_room_id)
.unwrap_or(true)
{
services()
.rooms .rooms
.state_cache .state_cache
.is_joined(sender_user, restriction_room_id) .room_members(restriction_room_id)
.ok()? .filter_map(Result::ok)
{ .find(|user| {
return None; user.server_name() == services().globals.server_name()
} && services()
let authorized_user = power_levels_event_content .rooms
.as_ref() .state_accessor
.and_then(|c| { .user_can_invite(user, restriction_room_id)
c.users .unwrap_or(false)
.iter()
.filter(|(uid, i)| {
uid.server_name() == services().globals.server_name()
&& **i > ruma::int!(0) && services()
.rooms
.state_cache
.is_joined(uid, restriction_room_id)
.unwrap_or(false)
})
.max_by_key(|(_, i)| *i)
.map(|(u, _)| u.to_owned())
}) })
.or_else(|| { } else {
services() None
.rooms }
.state_cache });
.room_members(restriction_room_id)
.filter_map(Result::ok)
.find(|uid| {
uid.server_name() == services().globals.server_name()
&& services()
.rooms
.state_accessor
.user_can_invite(uid, restriction_room_id)
.unwrap_or(false)
})
});
Some(authorized_user)
})
.flatten();
let event = RoomMemberEventContent { let event = RoomMemberEventContent {
membership: MembershipState::Join, membership: MembershipState::Join,