Skip to content
This repository was archived by the owner on Apr 26, 2024. It is now read-only.

Commit c5251c6

Browse files
authored
Do not assume that account data is of the correct form. (#8454)
This fixes a bug where `m.ignored_user_list` was assumed to be a dict, leading to odd behavior for users who set it to something else.
1 parent e3debf9 commit c5251c6

File tree

6 files changed

+34
-21
lines changed

6 files changed

+34
-21
lines changed

changelog.d/8454.bugfix

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Fix a longstanding bug where invalid ignored users in account data could break clients.

synapse/api/constants.py

+5
Original file line numberDiff line numberDiff line change
@@ -155,3 +155,8 @@ class EventContentFields:
155155
class RoomEncryptionAlgorithms:
156156
MEGOLM_V1_AES_SHA2 = "m.megolm.v1.aes-sha2"
157157
DEFAULT = MEGOLM_V1_AES_SHA2
158+
159+
160+
class AccountDataTypes:
161+
DIRECT = "m.direct"
162+
IGNORED_USER_LIST = "m.ignored_user_list"

synapse/handlers/room_member.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from unpaddedbase64 import encode_base64
2323

2424
from synapse import types
25-
from synapse.api.constants import MAX_DEPTH, EventTypes, Membership
25+
from synapse.api.constants import MAX_DEPTH, AccountDataTypes, EventTypes, Membership
2626
from synapse.api.errors import (
2727
AuthError,
2828
Codes,
@@ -247,7 +247,7 @@ async def copy_room_tags_and_direct_to_room(
247247
user_account_data, _ = await self.store.get_account_data_for_user(user_id)
248248

249249
# Copy direct message state if applicable
250-
direct_rooms = user_account_data.get("m.direct", {})
250+
direct_rooms = user_account_data.get(AccountDataTypes.DIRECT, {})
251251

252252
# Check which key this room is under
253253
if isinstance(direct_rooms, dict):
@@ -258,7 +258,7 @@ async def copy_room_tags_and_direct_to_room(
258258

259259
# Save back to user's m.direct account data
260260
await self.store.add_account_data_for_user(
261-
user_id, "m.direct", direct_rooms
261+
user_id, AccountDataTypes.DIRECT, direct_rooms
262262
)
263263
break
264264

synapse/handlers/sync.py

+11-8
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import attr
2222
from prometheus_client import Counter
2323

24-
from synapse.api.constants import EventTypes, Membership
24+
from synapse.api.constants import AccountDataTypes, EventTypes, Membership
2525
from synapse.api.filtering import FilterCollection
2626
from synapse.events import EventBase
2727
from synapse.logging.context import current_context
@@ -1378,13 +1378,16 @@ async def _generate_sync_entry_for_rooms(
13781378
return set(), set(), set(), set()
13791379

13801380
ignored_account_data = await self.store.get_global_account_data_by_type_for_user(
1381-
"m.ignored_user_list", user_id=user_id
1381+
AccountDataTypes.IGNORED_USER_LIST, user_id=user_id
13821382
)
13831383

1384+
# If there is ignored users account data and it matches the proper type,
1385+
# then use it.
1386+
ignored_users = frozenset() # type: FrozenSet[str]
13841387
if ignored_account_data:
1385-
ignored_users = ignored_account_data.get("ignored_users", {}).keys()
1386-
else:
1387-
ignored_users = frozenset()
1388+
ignored_users_data = ignored_account_data.get("ignored_users", {})
1389+
if isinstance(ignored_users_data, dict):
1390+
ignored_users = frozenset(ignored_users_data.keys())
13881391

13891392
if since_token:
13901393
room_changes = await self._get_rooms_changed(
@@ -1478,7 +1481,7 @@ async def _have_rooms_changed(
14781481
return False
14791482

14801483
async def _get_rooms_changed(
1481-
self, sync_result_builder: "SyncResultBuilder", ignored_users: Set[str]
1484+
self, sync_result_builder: "SyncResultBuilder", ignored_users: FrozenSet[str]
14821485
) -> _RoomChanges:
14831486
"""Gets the the changes that have happened since the last sync.
14841487
"""
@@ -1690,7 +1693,7 @@ async def _get_rooms_changed(
16901693
return _RoomChanges(room_entries, invited, newly_joined_rooms, newly_left_rooms)
16911694

16921695
async def _get_all_rooms(
1693-
self, sync_result_builder: "SyncResultBuilder", ignored_users: Set[str]
1696+
self, sync_result_builder: "SyncResultBuilder", ignored_users: FrozenSet[str]
16941697
) -> _RoomChanges:
16951698
"""Returns entries for all rooms for the user.
16961699
@@ -1764,7 +1767,7 @@ async def _get_all_rooms(
17641767
async def _generate_room_entry(
17651768
self,
17661769
sync_result_builder: "SyncResultBuilder",
1767-
ignored_users: Set[str],
1770+
ignored_users: FrozenSet[str],
17681771
room_builder: "RoomSyncResultBuilder",
17691772
ephemeral: List[JsonDict],
17701773
tags: Optional[Dict[str, Dict[str, Any]]],

synapse/storage/databases/main/account_data.py

+7-2
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import logging
1919
from typing import Dict, List, Optional, Tuple
2020

21+
from synapse.api.constants import AccountDataTypes
2122
from synapse.storage._base import SQLBaseStore, db_to_json
2223
from synapse.storage.database import DatabasePool
2324
from synapse.storage.util.id_generators import StreamIdGenerator
@@ -291,14 +292,18 @@ async def is_ignored_by(
291292
self, ignored_user_id: str, ignorer_user_id: str, cache_context: _CacheContext
292293
) -> bool:
293294
ignored_account_data = await self.get_global_account_data_by_type_for_user(
294-
"m.ignored_user_list",
295+
AccountDataTypes.IGNORED_USER_LIST,
295296
ignorer_user_id,
296297
on_invalidate=cache_context.invalidate,
297298
)
298299
if not ignored_account_data:
299300
return False
300301

301-
return ignored_user_id in ignored_account_data.get("ignored_users", {})
302+
try:
303+
return ignored_user_id in ignored_account_data.get("ignored_users", {})
304+
except TypeError:
305+
# The type of the ignored_users field is invalid.
306+
return False
302307

303308

304309
class AccountDataStore(AccountDataWorkerStore):

synapse/visibility.py

+7-8
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import logging
1717
import operator
1818

19-
from synapse.api.constants import EventTypes, Membership
19+
from synapse.api.constants import AccountDataTypes, EventTypes, Membership
2020
from synapse.events.utils import prune_event
2121
from synapse.storage import Storage
2222
from synapse.storage.state import StateFilter
@@ -77,15 +77,14 @@ async def filter_events_for_client(
7777
)
7878

7979
ignore_dict_content = await storage.main.get_global_account_data_by_type_for_user(
80-
"m.ignored_user_list", user_id
80+
AccountDataTypes.IGNORED_USER_LIST, user_id
8181
)
8282

83-
# FIXME: This will explode if people upload something incorrect.
84-
ignore_list = frozenset(
85-
ignore_dict_content.get("ignored_users", {}).keys()
86-
if ignore_dict_content
87-
else []
88-
)
83+
ignore_list = frozenset()
84+
if ignore_dict_content:
85+
ignored_users_dict = ignore_dict_content.get("ignored_users", {})
86+
if isinstance(ignored_users_dict, dict):
87+
ignore_list = frozenset(ignored_users_dict.keys())
8988

9089
erased_senders = await storage.main.are_users_erased((e.sender for e in events))
9190

0 commit comments

Comments
 (0)