Merge remote-tracking branch 'origin/develop' into release-v1.103

This commit is contained in:
Erik Johnston 2024-03-12 15:11:03 +00:00
commit 0d44f64c4e
2 changed files with 35 additions and 9 deletions

1
changelog.d/16968.bugfix Normal file
View file

@ -0,0 +1 @@
Prevent locking up when checking auth rules that are independent of room state for batched auth events. Contributed by @ggogel.

View file

@ -23,7 +23,20 @@
import collections.abc import collections.abc
import logging import logging
import typing import typing
from typing import Any, Dict, Iterable, List, Mapping, Optional, Set, Tuple, Union from typing import (
Any,
ChainMap,
Dict,
Iterable,
List,
Mapping,
MutableMapping,
Optional,
Set,
Tuple,
Union,
cast,
)
from canonicaljson import encode_canonical_json from canonicaljson import encode_canonical_json
from signedjson.key import decode_verify_key_bytes from signedjson.key import decode_verify_key_bytes
@ -175,12 +188,22 @@ async def check_state_independent_auth_rules(
return return
# 2. Reject if event has auth_events that: ... # 2. Reject if event has auth_events that: ...
auth_events: ChainMap[str, EventBase] = ChainMap()
if batched_auth_events: if batched_auth_events:
# Copy the batched auth events to avoid mutating them. # batched_auth_events can become very large. To avoid repeatedly copying it, which
auth_events = dict(batched_auth_events) # would significantly impact performance, we use a ChainMap.
needed_auth_event_ids = set(event.auth_event_ids()) - batched_auth_events.keys() # batched_auth_events must be cast to MutableMapping because .new_child() requires
# this type. This casting is safe as the mapping is never mutated.
auth_events = auth_events.new_child(
cast(MutableMapping[str, "EventBase"], batched_auth_events)
)
needed_auth_event_ids = [
event_id
for event_id in event.auth_event_ids()
if event_id not in batched_auth_events
]
if needed_auth_event_ids: if needed_auth_event_ids:
auth_events.update( auth_events = auth_events.new_child(
await store.get_events( await store.get_events(
needed_auth_event_ids, needed_auth_event_ids,
redact_behaviour=EventRedactBehaviour.as_is, redact_behaviour=EventRedactBehaviour.as_is,
@ -188,11 +211,13 @@ async def check_state_independent_auth_rules(
) )
) )
else: else:
auth_events = await store.get_events( auth_events = auth_events.new_child(
await store.get_events(
event.auth_event_ids(), event.auth_event_ids(),
redact_behaviour=EventRedactBehaviour.as_is, redact_behaviour=EventRedactBehaviour.as_is,
allow_rejected=True, allow_rejected=True,
) )
)
room_id = event.room_id room_id = event.room_id
auth_dict: MutableStateMap[str] = {} auth_dict: MutableStateMap[str] = {}