mirror of
https://github.com/element-hq/synapse
synced 2024-08-19 22:00:22 +00:00
During the migration the automated script to update the copyright headers accidentally got rid of some of the existing copyright lines. Reinstate them.
919 lines
32 KiB
Python
919 lines
32 KiB
Python
#
|
||
# This file is licensed under the Affero General Public License (AGPL) version 3.
|
||
#
|
||
# Copyright 2018-2022 The Matrix.org Foundation C.I.C.
|
||
# Copyright (C) 2023 New Vector, Ltd
|
||
#
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU Affero General Public License as
|
||
# published by the Free Software Foundation, either version 3 of the
|
||
# License, or (at your option) any later version.
|
||
#
|
||
# See the GNU Affero General Public License for more details:
|
||
# <https://www.gnu.org/licenses/agpl-3.0.html>.
|
||
#
|
||
# Originally licensed under the Apache License, Version 2.0:
|
||
# <http://www.apache.org/licenses/LICENSE-2.0>.
|
||
#
|
||
# [This file includes modifications made by New Vector Limited]
|
||
#
|
||
#
|
||
|
||
import unittest
|
||
from typing import Any, Collection, Dict, Iterable, List, Optional
|
||
|
||
from parameterized import parameterized
|
||
|
||
from synapse import event_auth
|
||
from synapse.api.constants import EventContentFields
|
||
from synapse.api.errors import AuthError, SynapseError
|
||
from synapse.api.room_versions import EventFormatVersions, RoomVersion, RoomVersions
|
||
from synapse.events import EventBase, make_event_from_dict
|
||
from synapse.storage.databases.main.events_worker import EventRedactBehaviour
|
||
from synapse.types import JsonDict, get_domain_from_id
|
||
|
||
from tests.test_utils import get_awaitable_result
|
||
|
||
|
||
class _StubEventSourceStore:
|
||
"""A stub implementation of the EventSourceStore"""
|
||
|
||
def __init__(self) -> None:
|
||
self._store: Dict[str, EventBase] = {}
|
||
|
||
def add_event(self, event: EventBase) -> None:
|
||
self._store[event.event_id] = event
|
||
|
||
def add_events(self, events: Iterable[EventBase]) -> None:
|
||
for event in events:
|
||
self._store[event.event_id] = event
|
||
|
||
async def get_events(
|
||
self,
|
||
event_ids: Collection[str],
|
||
redact_behaviour: EventRedactBehaviour,
|
||
get_prev_content: bool = False,
|
||
allow_rejected: bool = False,
|
||
) -> Dict[str, EventBase]:
|
||
assert allow_rejected
|
||
assert not get_prev_content
|
||
assert redact_behaviour == EventRedactBehaviour.as_is
|
||
results = {}
|
||
for e in event_ids:
|
||
if e in self._store:
|
||
results[e] = self._store[e]
|
||
return results
|
||
|
||
|
||
class EventAuthTestCase(unittest.TestCase):
|
||
def test_rejected_auth_events(self) -> None:
|
||
"""
|
||
Events that refer to rejected events in their auth events are rejected
|
||
"""
|
||
creator = "@creator:example.com"
|
||
auth_events = [
|
||
_create_event(RoomVersions.V9, creator),
|
||
_join_event(RoomVersions.V9, creator),
|
||
]
|
||
|
||
event_store = _StubEventSourceStore()
|
||
event_store.add_events(auth_events)
|
||
|
||
# creator should be able to send state
|
||
event = _random_state_event(RoomVersions.V9, creator, auth_events)
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, event)
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(event, auth_events)
|
||
|
||
# ... but a rejected join_rules event should cause it to be rejected
|
||
rejected_join_rules = _join_rules_event(
|
||
RoomVersions.V9,
|
||
creator,
|
||
"public",
|
||
)
|
||
rejected_join_rules.rejected_reason = "stinky"
|
||
auth_events.append(rejected_join_rules)
|
||
event_store.add_event(rejected_join_rules)
|
||
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(
|
||
event_store,
|
||
_random_state_event(RoomVersions.V9, creator),
|
||
)
|
||
)
|
||
|
||
# ... even if there is *also* a good join rules
|
||
auth_events.append(_join_rules_event(RoomVersions.V9, creator, "public"))
|
||
event_store.add_event(rejected_join_rules)
|
||
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(
|
||
event_store,
|
||
_random_state_event(RoomVersions.V9, creator),
|
||
)
|
||
)
|
||
|
||
def test_create_event_with_prev_events(self) -> None:
|
||
"""A create event with prev_events should be rejected
|
||
|
||
https://spec.matrix.org/v1.3/rooms/v9/#authorization-rules
|
||
1: If type is m.room.create:
|
||
1. If it has any previous events, reject.
|
||
"""
|
||
creator = f"@creator:{TEST_DOMAIN}"
|
||
|
||
# we make both a good event and a bad event, to check that we are rejecting
|
||
# the bad event for the reason we think we are.
|
||
good_event = make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
"type": "m.room.create",
|
||
"state_key": "",
|
||
"sender": creator,
|
||
"content": {
|
||
"creator": creator,
|
||
"room_version": RoomVersions.V9.identifier,
|
||
},
|
||
"auth_events": [],
|
||
"prev_events": [],
|
||
},
|
||
room_version=RoomVersions.V9,
|
||
)
|
||
bad_event = make_event_from_dict(
|
||
{**good_event.get_dict(), "prev_events": ["$fakeevent"]},
|
||
room_version=RoomVersions.V9,
|
||
)
|
||
|
||
event_store = _StubEventSourceStore()
|
||
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, good_event)
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, bad_event)
|
||
)
|
||
|
||
def test_duplicate_auth_events(self) -> None:
|
||
"""Events with duplicate auth_events should be rejected
|
||
|
||
https://spec.matrix.org/v1.3/rooms/v9/#authorization-rules
|
||
2. Reject if event has auth_events that:
|
||
1. have duplicate entries for a given type and state_key pair
|
||
"""
|
||
creator = "@creator:example.com"
|
||
|
||
create_event = _create_event(RoomVersions.V9, creator)
|
||
join_event1 = _join_event(RoomVersions.V9, creator)
|
||
pl_event = _power_levels_event(
|
||
RoomVersions.V9,
|
||
creator,
|
||
{"state_default": 30, "users": {"creator": 100}},
|
||
)
|
||
|
||
# create a second join event, so that we can make a duplicate
|
||
join_event2 = _join_event(RoomVersions.V9, creator)
|
||
|
||
event_store = _StubEventSourceStore()
|
||
event_store.add_events([create_event, join_event1, join_event2, pl_event])
|
||
|
||
good_event = _random_state_event(
|
||
RoomVersions.V9, creator, [create_event, join_event2, pl_event]
|
||
)
|
||
bad_event = _random_state_event(
|
||
RoomVersions.V9, creator, [create_event, join_event1, join_event2, pl_event]
|
||
)
|
||
# a variation: two instances of the *same* event
|
||
bad_event2 = _random_state_event(
|
||
RoomVersions.V9, creator, [create_event, join_event2, join_event2, pl_event]
|
||
)
|
||
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, good_event)
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, bad_event)
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, bad_event2)
|
||
)
|
||
|
||
def test_unexpected_auth_events(self) -> None:
|
||
"""Events with excess auth_events should be rejected
|
||
|
||
https://spec.matrix.org/v1.3/rooms/v9/#authorization-rules
|
||
2. Reject if event has auth_events that:
|
||
2. have entries whose type and state_key don’t match those specified by the
|
||
auth events selection algorithm described in the server specification.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
|
||
create_event = _create_event(RoomVersions.V9, creator)
|
||
join_event = _join_event(RoomVersions.V9, creator)
|
||
pl_event = _power_levels_event(
|
||
RoomVersions.V9,
|
||
creator,
|
||
{"state_default": 30, "users": {"creator": 100}},
|
||
)
|
||
join_rules_event = _join_rules_event(RoomVersions.V9, creator, "public")
|
||
|
||
event_store = _StubEventSourceStore()
|
||
event_store.add_events([create_event, join_event, pl_event, join_rules_event])
|
||
|
||
good_event = _random_state_event(
|
||
RoomVersions.V9, creator, [create_event, join_event, pl_event]
|
||
)
|
||
# join rules should *not* be included in the auth events.
|
||
bad_event = _random_state_event(
|
||
RoomVersions.V9,
|
||
creator,
|
||
[create_event, join_event, pl_event, join_rules_event],
|
||
)
|
||
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, good_event)
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
get_awaitable_result(
|
||
event_auth.check_state_independent_auth_rules(event_store, bad_event)
|
||
)
|
||
|
||
def test_random_users_cannot_send_state_before_first_pl(self) -> None:
|
||
"""
|
||
Check that, before the first PL lands, the creator is the only user
|
||
that can send a state event.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
joiner = "@joiner:example.com"
|
||
auth_events = [
|
||
_create_event(RoomVersions.V1, creator),
|
||
_join_event(RoomVersions.V1, creator),
|
||
_join_event(RoomVersions.V1, joiner),
|
||
]
|
||
|
||
# creator should be able to send state
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_random_state_event(RoomVersions.V1, creator),
|
||
auth_events,
|
||
)
|
||
|
||
# joiner should not be able to send state
|
||
self.assertRaises(
|
||
AuthError,
|
||
event_auth.check_state_dependent_auth_rules,
|
||
_random_state_event(RoomVersions.V1, joiner),
|
||
auth_events,
|
||
)
|
||
|
||
def test_state_default_level(self) -> None:
|
||
"""
|
||
Check that users above the state_default level can send state and
|
||
those below cannot
|
||
"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
king = "@joiner2:example.com"
|
||
|
||
auth_events = [
|
||
_create_event(RoomVersions.V1, creator),
|
||
_join_event(RoomVersions.V1, creator),
|
||
_power_levels_event(
|
||
RoomVersions.V1,
|
||
creator,
|
||
{"state_default": "30", "users": {pleb: "29", king: "30"}},
|
||
),
|
||
_join_event(RoomVersions.V1, pleb),
|
||
_join_event(RoomVersions.V1, king),
|
||
]
|
||
|
||
# pleb should not be able to send state
|
||
self.assertRaises(
|
||
AuthError,
|
||
event_auth.check_state_dependent_auth_rules,
|
||
_random_state_event(RoomVersions.V1, pleb),
|
||
auth_events,
|
||
),
|
||
|
||
# king should be able to send state
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_random_state_event(RoomVersions.V1, king),
|
||
auth_events,
|
||
)
|
||
|
||
def test_alias_event(self) -> None:
|
||
"""Alias events have special behavior up through room version 6."""
|
||
creator = "@creator:example.com"
|
||
other = "@other:example.com"
|
||
auth_events = [
|
||
_create_event(RoomVersions.V1, creator),
|
||
_join_event(RoomVersions.V1, creator),
|
||
]
|
||
|
||
# creator should be able to send aliases
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V1, creator),
|
||
auth_events,
|
||
)
|
||
|
||
# Reject an event with no state key.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V1, creator, state_key=""),
|
||
auth_events,
|
||
)
|
||
|
||
# If the domain of the sender does not match the state key, reject.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V1, creator, state_key="test.com"),
|
||
auth_events,
|
||
)
|
||
|
||
# Note that the member does *not* need to be in the room.
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V1, other),
|
||
auth_events,
|
||
)
|
||
|
||
def test_msc2432_alias_event(self) -> None:
|
||
"""After MSC2432, alias events have no special behavior."""
|
||
creator = "@creator:example.com"
|
||
other = "@other:example.com"
|
||
auth_events = [
|
||
_create_event(RoomVersions.V6, creator),
|
||
_join_event(RoomVersions.V6, creator),
|
||
]
|
||
|
||
# creator should be able to send aliases
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V6, creator),
|
||
auth_events,
|
||
)
|
||
|
||
# No particular checks are done on the state key.
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V6, creator, state_key=""),
|
||
auth_events,
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V6, creator, state_key="test.com"),
|
||
auth_events,
|
||
)
|
||
|
||
# Per standard auth rules, the member must be in the room.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_alias_event(RoomVersions.V6, other),
|
||
auth_events,
|
||
)
|
||
|
||
@parameterized.expand([(RoomVersions.V1, True), (RoomVersions.V6, False)])
|
||
def test_notifications(
|
||
self, room_version: RoomVersion, allow_modification: bool
|
||
) -> None:
|
||
"""
|
||
Notifications power levels get checked due to MSC2209.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
|
||
auth_events = [
|
||
_create_event(room_version, creator),
|
||
_join_event(room_version, creator),
|
||
_power_levels_event(
|
||
room_version, creator, {"state_default": "30", "users": {pleb: "30"}}
|
||
),
|
||
_join_event(room_version, pleb),
|
||
]
|
||
|
||
pl_event = _power_levels_event(
|
||
room_version, pleb, {"notifications": {"room": 100}}
|
||
)
|
||
|
||
# on room V1, pleb should be able to modify the notifications power level.
|
||
if allow_modification:
|
||
event_auth.check_state_dependent_auth_rules(pl_event, auth_events)
|
||
|
||
else:
|
||
# But an MSC2209 room rejects this change.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(pl_event, auth_events)
|
||
|
||
def test_join_rules_public(self) -> None:
|
||
"""
|
||
Test joining a public room.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
|
||
auth_events = {
|
||
("m.room.create", ""): _create_event(RoomVersions.V6, creator),
|
||
("m.room.member", creator): _join_event(RoomVersions.V6, creator),
|
||
("m.room.join_rules", ""): _join_rules_event(
|
||
RoomVersions.V6, creator, "public"
|
||
),
|
||
}
|
||
|
||
# Check join.
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user cannot be force-joined to a room.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_member_event(RoomVersions.V6, pleb, "join", sender=creator),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# Banned should be rejected.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "ban"
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user who left can re-join.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "leave"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can send a join if they're in the room.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "join"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can accept an invite.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "invite", sender=creator
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
def test_join_rules_invite(self) -> None:
|
||
"""
|
||
Test joining an invite only room.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
|
||
auth_events = {
|
||
("m.room.create", ""): _create_event(RoomVersions.V6, creator),
|
||
("m.room.member", creator): _join_event(RoomVersions.V6, creator),
|
||
("m.room.join_rules", ""): _join_rules_event(
|
||
RoomVersions.V6, creator, "invite"
|
||
),
|
||
}
|
||
|
||
# A join without an invite is rejected.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user cannot be force-joined to a room.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_member_event(RoomVersions.V6, pleb, "join", sender=creator),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# Banned should be rejected.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "ban"
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user who left cannot re-join.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "leave"
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can send a join if they're in the room.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "join"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can accept an invite.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V6, pleb, "invite", sender=creator
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
def test_join_rules_restricted_old_room(self) -> None:
|
||
"""Old room versions should reject joins to restricted rooms"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
|
||
auth_events = {
|
||
("m.room.create", ""): _create_event(RoomVersions.V6, creator),
|
||
("m.room.member", creator): _join_event(RoomVersions.V6, creator),
|
||
("m.room.power_levels", ""): _power_levels_event(
|
||
RoomVersions.V6, creator, {"invite": 0}
|
||
),
|
||
("m.room.join_rules", ""): _join_rules_event(
|
||
RoomVersions.V6, creator, "restricted"
|
||
),
|
||
}
|
||
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V6, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
def test_join_rules_msc3083_restricted(self) -> None:
|
||
"""
|
||
Test joining a restricted room from MSC3083.
|
||
|
||
This is similar to the public test, but has some additional checks on
|
||
signatures.
|
||
|
||
The checks which care about signatures fake them by simply adding an
|
||
object of the proper form, not generating valid signatures.
|
||
"""
|
||
creator = "@creator:example.com"
|
||
pleb = "@joiner:example.com"
|
||
|
||
auth_events = {
|
||
("m.room.create", ""): _create_event(RoomVersions.V8, creator),
|
||
("m.room.member", creator): _join_event(RoomVersions.V8, creator),
|
||
("m.room.power_levels", ""): _power_levels_event(
|
||
RoomVersions.V8, creator, {"invite": 0}
|
||
),
|
||
("m.room.join_rules", ""): _join_rules_event(
|
||
RoomVersions.V8, creator, "restricted"
|
||
),
|
||
}
|
||
|
||
# A properly formatted join event should work.
|
||
authorised_join_event = _join_event(
|
||
RoomVersions.V8,
|
||
pleb,
|
||
additional_content={
|
||
EventContentFields.AUTHORISING_USER: "@creator:example.com"
|
||
},
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
authorised_join_event,
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A join issued by a specific user works (i.e. the power level checks
|
||
# are done properly).
|
||
pl_auth_events = auth_events.copy()
|
||
pl_auth_events[("m.room.power_levels", "")] = _power_levels_event(
|
||
RoomVersions.V8,
|
||
creator,
|
||
{"invite": 100, "users": {"@inviter:foo.test": 150}},
|
||
)
|
||
pl_auth_events[("m.room.member", "@inviter:foo.test")] = _join_event(
|
||
RoomVersions.V8, "@inviter:foo.test"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(
|
||
RoomVersions.V8,
|
||
pleb,
|
||
additional_content={
|
||
EventContentFields.AUTHORISING_USER: "@inviter:foo.test"
|
||
},
|
||
),
|
||
pl_auth_events.values(),
|
||
)
|
||
|
||
# A join which is missing an authorised server is rejected.
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V8, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# An join authorised by a user who is not in the room is rejected.
|
||
pl_auth_events = auth_events.copy()
|
||
pl_auth_events[("m.room.power_levels", "")] = _power_levels_event(
|
||
RoomVersions.V8,
|
||
creator,
|
||
{"invite": 100, "users": {"@other:example.com": 150}},
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(
|
||
RoomVersions.V8,
|
||
pleb,
|
||
additional_content={
|
||
EventContentFields.AUTHORISING_USER: "@other:example.com"
|
||
},
|
||
),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user cannot be force-joined to a room. (This uses an event which
|
||
# *would* be valid, but is sent be a different user.)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_member_event(
|
||
RoomVersions.V8,
|
||
pleb,
|
||
"join",
|
||
sender=creator,
|
||
additional_content={
|
||
EventContentFields.AUTHORISING_USER: "@inviter:foo.test"
|
||
},
|
||
),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# Banned should be rejected.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V8, pleb, "ban"
|
||
)
|
||
with self.assertRaises(AuthError):
|
||
event_auth.check_state_dependent_auth_rules(
|
||
authorised_join_event,
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user who left can re-join.
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V8, pleb, "leave"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
authorised_join_event,
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can send a join if they're in the room. (This doesn't need to
|
||
# be authorised since the user is already joined.)
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V8, pleb, "join"
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V8, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
# A user can accept an invite. (This doesn't need to be authorised since
|
||
# the user was invited.)
|
||
auth_events[("m.room.member", pleb)] = _member_event(
|
||
RoomVersions.V8, pleb, "invite", sender=creator
|
||
)
|
||
event_auth.check_state_dependent_auth_rules(
|
||
_join_event(RoomVersions.V8, pleb),
|
||
auth_events.values(),
|
||
)
|
||
|
||
def test_room_v10_rejects_string_power_levels(self) -> None:
|
||
pl_event_content = {"users_default": "42"}
|
||
pl_event = make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(RoomVersions.V10),
|
||
"type": "m.room.power_levels",
|
||
"sender": "@test:test.com",
|
||
"state_key": "",
|
||
"content": pl_event_content,
|
||
"signatures": {"test.com": {"ed25519:0": "some9signature"}},
|
||
},
|
||
room_version=RoomVersions.V10,
|
||
)
|
||
|
||
pl_event2_content = {"events": {"m.room.name": "42", "m.room.power_levels": 42}}
|
||
pl_event2 = make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(RoomVersions.V10),
|
||
"type": "m.room.power_levels",
|
||
"sender": "@test:test.com",
|
||
"state_key": "",
|
||
"content": pl_event2_content,
|
||
"signatures": {"test.com": {"ed25519:0": "some9signature"}},
|
||
},
|
||
room_version=RoomVersions.V10,
|
||
)
|
||
|
||
with self.assertRaises(SynapseError):
|
||
event_auth._check_power_levels(
|
||
pl_event.room_version, pl_event, {("fake_type", "fake_key"): pl_event2}
|
||
)
|
||
|
||
with self.assertRaises(SynapseError):
|
||
event_auth._check_power_levels(
|
||
pl_event.room_version, pl_event2, {("fake_type", "fake_key"): pl_event}
|
||
)
|
||
|
||
def test_room_v10_rejects_other_non_integer_power_levels(self) -> None:
|
||
"""We should reject PLs that are non-integer, non-string JSON values.
|
||
|
||
test_room_v10_rejects_string_power_levels above handles the string case.
|
||
"""
|
||
|
||
def create_event(pl_event_content: Dict[str, Any]) -> EventBase:
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(RoomVersions.V10),
|
||
"type": "m.room.power_levels",
|
||
"sender": "@test:test.com",
|
||
"state_key": "",
|
||
"content": pl_event_content,
|
||
"signatures": {"test.com": {"ed25519:0": "some9signature"}},
|
||
},
|
||
room_version=RoomVersions.V10,
|
||
)
|
||
|
||
contents: Iterable[Dict[str, Any]] = [
|
||
{"notifications": {"room": None}},
|
||
{"users": {"@alice:wonderland": []}},
|
||
{"users_default": {}},
|
||
]
|
||
for content in contents:
|
||
event = create_event(content)
|
||
with self.assertRaises(SynapseError):
|
||
event_auth._check_power_levels(event.room_version, event, {})
|
||
|
||
|
||
# helpers for making events
|
||
TEST_DOMAIN = "example.com"
|
||
TEST_ROOM_ID = f"!test_room:{TEST_DOMAIN}"
|
||
|
||
|
||
def _create_event(
|
||
room_version: RoomVersion,
|
||
user_id: str,
|
||
) -> EventBase:
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "m.room.create",
|
||
"state_key": "",
|
||
"sender": user_id,
|
||
"content": {"creator": user_id},
|
||
"auth_events": [],
|
||
},
|
||
room_version=room_version,
|
||
)
|
||
|
||
|
||
def _member_event(
|
||
room_version: RoomVersion,
|
||
user_id: str,
|
||
membership: str,
|
||
sender: Optional[str] = None,
|
||
additional_content: Optional[dict] = None,
|
||
) -> EventBase:
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "m.room.member",
|
||
"sender": sender or user_id,
|
||
"state_key": user_id,
|
||
"content": {"membership": membership, **(additional_content or {})},
|
||
"auth_events": [],
|
||
"prev_events": [],
|
||
},
|
||
room_version=room_version,
|
||
)
|
||
|
||
|
||
def _join_event(
|
||
room_version: RoomVersion,
|
||
user_id: str,
|
||
additional_content: Optional[dict] = None,
|
||
) -> EventBase:
|
||
return _member_event(
|
||
room_version,
|
||
user_id,
|
||
"join",
|
||
additional_content=additional_content,
|
||
)
|
||
|
||
|
||
def _power_levels_event(
|
||
room_version: RoomVersion,
|
||
sender: str,
|
||
content: JsonDict,
|
||
) -> EventBase:
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "m.room.power_levels",
|
||
"sender": sender,
|
||
"state_key": "",
|
||
"content": content,
|
||
},
|
||
room_version=room_version,
|
||
)
|
||
|
||
|
||
def _alias_event(room_version: RoomVersion, sender: str, **kwargs: Any) -> EventBase:
|
||
data = {
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "m.room.aliases",
|
||
"sender": sender,
|
||
"state_key": get_domain_from_id(sender),
|
||
"content": {"aliases": []},
|
||
}
|
||
data.update(**kwargs)
|
||
return make_event_from_dict(data, room_version=room_version)
|
||
|
||
|
||
def _build_auth_dict_for_room_version(
|
||
room_version: RoomVersion, auth_events: Iterable[EventBase]
|
||
) -> List:
|
||
if room_version.event_format == EventFormatVersions.ROOM_V1_V2:
|
||
return [(e.event_id, "not_used") for e in auth_events]
|
||
else:
|
||
return [e.event_id for e in auth_events]
|
||
|
||
|
||
def _random_state_event(
|
||
room_version: RoomVersion,
|
||
sender: str,
|
||
auth_events: Optional[Iterable[EventBase]] = None,
|
||
) -> EventBase:
|
||
if auth_events is None:
|
||
auth_events = []
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "test.state",
|
||
"sender": sender,
|
||
"state_key": "",
|
||
"content": {"membership": "join"},
|
||
"auth_events": _build_auth_dict_for_room_version(room_version, auth_events),
|
||
},
|
||
room_version=room_version,
|
||
)
|
||
|
||
|
||
def _join_rules_event(
|
||
room_version: RoomVersion, sender: str, join_rule: str
|
||
) -> EventBase:
|
||
return make_event_from_dict(
|
||
{
|
||
"room_id": TEST_ROOM_ID,
|
||
**_maybe_get_event_id_dict_for_room_version(room_version),
|
||
"type": "m.room.join_rules",
|
||
"sender": sender,
|
||
"state_key": "",
|
||
"content": {
|
||
"join_rule": join_rule,
|
||
},
|
||
},
|
||
room_version=room_version,
|
||
)
|
||
|
||
|
||
event_count = 0
|
||
|
||
|
||
def _maybe_get_event_id_dict_for_room_version(room_version: RoomVersion) -> dict:
|
||
"""If this room version needs it, generate an event id"""
|
||
if room_version.event_format != EventFormatVersions.ROOM_V1_V2:
|
||
return {}
|
||
|
||
global event_count
|
||
c = event_count
|
||
event_count += 1
|
||
return {"event_id": "!%i:example.com" % (c,)}
|