mirror of
https://github.com/element-hq/synapse.git
synced 2024-11-22 09:35:45 +03:00
Faster joins: use servers list approximation in assert_host_in_room
(#14515)
Signed-off-by: Mathieu Velten <mathieuv@matrix.org>
This commit is contained in:
parent
f38d7d79c8
commit
3b4e150868
2 changed files with 18 additions and 11 deletions
1
changelog.d/14515.misc
Normal file
1
changelog.d/14515.misc
Normal file
|
@ -0,0 +1 @@
|
||||||
|
Faster joins: use servers list approximation received during `send_join` (potentially updated with received membership events) in `assert_host_in_room`.
|
|
@ -45,6 +45,7 @@ class EventAuthHandler:
|
||||||
def __init__(self, hs: "HomeServer"):
|
def __init__(self, hs: "HomeServer"):
|
||||||
self._clock = hs.get_clock()
|
self._clock = hs.get_clock()
|
||||||
self._store = hs.get_datastores().main
|
self._store = hs.get_datastores().main
|
||||||
|
self._state_storage_controller = hs.get_storage_controllers().state
|
||||||
self._server_name = hs.hostname
|
self._server_name = hs.hostname
|
||||||
|
|
||||||
async def check_auth_rules_from_context(
|
async def check_auth_rules_from_context(
|
||||||
|
@ -179,15 +180,20 @@ class EventAuthHandler:
|
||||||
this function may return an incorrect result as we are not able to fully
|
this function may return an incorrect result as we are not able to fully
|
||||||
track server membership in a room without full state.
|
track server membership in a room without full state.
|
||||||
"""
|
"""
|
||||||
if not allow_partial_state_rooms and await self._store.is_partial_state_room(
|
if await self._store.is_partial_state_room(room_id):
|
||||||
|
if allow_partial_state_rooms:
|
||||||
|
current_hosts = await self._state_storage_controller.get_current_hosts_in_room_or_partial_state_approximation(
|
||||||
room_id
|
room_id
|
||||||
):
|
)
|
||||||
|
if host not in current_hosts:
|
||||||
|
raise AuthError(403, "Host not in room (partial-state approx).")
|
||||||
|
else:
|
||||||
raise AuthError(
|
raise AuthError(
|
||||||
403,
|
403,
|
||||||
"Unable to authorise you right now; room is partial-stated here.",
|
"Unable to authorise you right now; room is partial-stated here.",
|
||||||
errcode=Codes.UNABLE_DUE_TO_PARTIAL_STATE,
|
errcode=Codes.UNABLE_DUE_TO_PARTIAL_STATE,
|
||||||
)
|
)
|
||||||
|
else:
|
||||||
if not await self.is_host_in_room(room_id, host):
|
if not await self.is_host_in_room(room_id, host):
|
||||||
raise AuthError(403, "Host not in room.")
|
raise AuthError(403, "Host not in room.")
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue