2014-08-12 08:10:52 -06:00
|
|
|
# -*- coding: utf-8 -*-
|
2016-01-05 11:01:18 -07:00
|
|
|
# Copyright 2014 - 2016 OpenMarket Ltd
|
2014-08-12 08:10:52 -06:00
|
|
|
#
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
# you may not use this file except in compliance with the License.
|
|
|
|
# You may obtain a copy of the License at
|
|
|
|
#
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
#
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
# See the License for the specific language governing permissions and
|
|
|
|
# limitations under the License.
|
2014-08-12 20:14:34 -06:00
|
|
|
|
2016-07-26 09:46:53 -06:00
|
|
|
import logging
|
|
|
|
|
2018-05-31 03:03:47 -06:00
|
|
|
from six import itervalues
|
|
|
|
|
2016-07-26 09:46:53 -06:00
|
|
|
import pymacaroons
|
2018-06-28 13:31:53 -06:00
|
|
|
from netaddr import IPAddress
|
2014-08-12 08:10:52 -06:00
|
|
|
|
2018-07-09 00:09:20 -06:00
|
|
|
from twisted.internet import defer
|
|
|
|
|
2016-07-26 09:46:53 -06:00
|
|
|
import synapse.types
|
2017-01-13 08:07:32 -07:00
|
|
|
from synapse import event_auth
|
2018-07-09 00:09:20 -06:00
|
|
|
from synapse.api.constants import EventTypes, JoinRules, Membership
|
2018-08-16 11:02:02 -06:00
|
|
|
from synapse.api.errors import AuthError, Codes, ResourceLimitError
|
2018-08-31 10:11:11 -06:00
|
|
|
from synapse.config.server import is_threepid_reserved
|
2017-01-13 08:07:32 -07:00
|
|
|
from synapse.types import UserID
|
2018-07-09 00:09:20 -06:00
|
|
|
from synapse.util.caches import CACHE_SIZE_FACTOR, register_cache
|
2017-06-29 07:50:18 -06:00
|
|
|
from synapse.util.caches.lrucache import LruCache
|
2016-04-13 04:15:59 -06:00
|
|
|
from synapse.util.metrics import Measure
|
2014-08-12 08:10:52 -06:00
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
2015-03-15 18:18:08 -06:00
|
|
|
AuthEventTypes = (
|
|
|
|
EventTypes.Create, EventTypes.Member, EventTypes.PowerLevels,
|
2015-07-03 03:31:17 -06:00
|
|
|
EventTypes.JoinRules, EventTypes.RoomHistoryVisibility,
|
2015-10-01 10:49:52 -06:00
|
|
|
EventTypes.ThirdPartyInvite,
|
2015-03-15 18:18:08 -06:00
|
|
|
)
|
|
|
|
|
2016-11-25 08:25:30 -07:00
|
|
|
# guests always get this device id.
|
|
|
|
GUEST_DEVICE_ID = "guest_device"
|
|
|
|
|
2015-03-15 18:18:08 -06:00
|
|
|
|
2017-06-29 07:50:18 -06:00
|
|
|
class _InvalidMacaroonException(Exception):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
class Auth(object):
|
|
|
|
"""
|
|
|
|
FIXME: This class contains a mix of functions for authenticating users
|
|
|
|
of our client-server API and authenticating events added to room graphs.
|
|
|
|
"""
|
|
|
|
def __init__(self, hs):
|
|
|
|
self.hs = hs
|
|
|
|
self.clock = hs.get_clock()
|
|
|
|
self.store = hs.get_datastore()
|
|
|
|
self.state = hs.get_state_handler()
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS = 401
|
|
|
|
|
2017-06-29 07:50:18 -06:00
|
|
|
self.token_cache = LruCache(CACHE_SIZE_FACTOR * 10000)
|
2018-05-21 18:47:37 -06:00
|
|
|
register_cache("cache", "token_cache", self.token_cache)
|
2017-06-29 07:50:18 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_from_context(self, event, context, do_sig_check=True):
|
2018-07-23 06:00:22 -06:00
|
|
|
prev_state_ids = yield context.get_prev_state_ids(self.store)
|
2017-01-10 11:16:54 -07:00
|
|
|
auth_events_ids = yield self.compute_auth_events(
|
2018-07-23 06:00:22 -06:00
|
|
|
event, prev_state_ids, for_verification=True,
|
2017-01-10 11:16:54 -07:00
|
|
|
)
|
|
|
|
auth_events = yield self.store.get_events(auth_events_ids)
|
|
|
|
auth_events = {
|
2018-05-31 03:03:47 -06:00
|
|
|
(e.type, e.state_key): e for e in itervalues(auth_events)
|
2017-01-10 11:16:54 -07:00
|
|
|
}
|
|
|
|
self.check(event, auth_events=auth_events, do_sig_check=do_sig_check)
|
|
|
|
|
|
|
|
def check(self, event, auth_events, do_sig_check=True):
|
|
|
|
""" Checks if this event is correctly authed.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
event: the event being checked.
|
|
|
|
auth_events (dict: event-key -> event): the existing room state.
|
|
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
True if the auth checks pass.
|
|
|
|
"""
|
|
|
|
with Measure(self.clock, "auth.check"):
|
2017-01-13 08:07:32 -07:00
|
|
|
event_auth.check(event, auth_events, do_sig_check=do_sig_check)
|
2017-01-10 11:16:54 -07:00
|
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_joined_room(self, room_id, user_id, current_state=None):
|
|
|
|
"""Check if the user is currently joined in the room
|
|
|
|
Args:
|
|
|
|
room_id(str): The room to check.
|
|
|
|
user_id(str): The user to check.
|
|
|
|
current_state(dict): Optional map of the current state of the room.
|
|
|
|
If provided then that map is used to check whether they are a
|
|
|
|
member of the room. Otherwise the current membership is
|
|
|
|
loaded from the database.
|
|
|
|
Raises:
|
|
|
|
AuthError if the user is not in the room.
|
|
|
|
Returns:
|
|
|
|
A deferred membership event for the user if the user is in
|
|
|
|
the room.
|
|
|
|
"""
|
|
|
|
if current_state:
|
|
|
|
member = current_state.get(
|
|
|
|
(EventTypes.Member, user_id),
|
|
|
|
None
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
member = yield self.state.get_current_state(
|
|
|
|
room_id=room_id,
|
|
|
|
event_type=EventTypes.Member,
|
|
|
|
state_key=user_id
|
|
|
|
)
|
|
|
|
|
|
|
|
self._check_joined_room(member, user_id, room_id)
|
|
|
|
defer.returnValue(member)
|
2015-11-05 09:43:19 -07:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_user_was_in_room(self, room_id, user_id):
|
|
|
|
"""Check if the user was in the room at some point.
|
|
|
|
Args:
|
|
|
|
room_id(str): The room to check.
|
|
|
|
user_id(str): The user to check.
|
|
|
|
Raises:
|
|
|
|
AuthError if the user was never in the room.
|
|
|
|
Returns:
|
|
|
|
A deferred membership event for the user if the user was in the
|
|
|
|
room. This will be the join event if they are currently joined to
|
|
|
|
the room. This will be the leave event if they have left the room.
|
|
|
|
"""
|
|
|
|
member = yield self.state.get_current_state(
|
|
|
|
room_id=room_id,
|
|
|
|
event_type=EventTypes.Member,
|
|
|
|
state_key=user_id
|
|
|
|
)
|
|
|
|
membership = member.membership if member else None
|
2015-10-01 10:49:52 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
if membership not in (Membership.JOIN, Membership.LEAVE):
|
|
|
|
raise AuthError(403, "User %s not in room %s" % (
|
|
|
|
user_id, room_id
|
|
|
|
))
|
2016-02-23 08:11:25 -07:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
if membership == Membership.LEAVE:
|
|
|
|
forgot = yield self.store.did_forget(user_id, room_id)
|
|
|
|
if forgot:
|
|
|
|
raise AuthError(403, "User %s not in room %s" % (
|
|
|
|
user_id, room_id
|
|
|
|
))
|
2016-02-23 08:11:25 -07:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
defer.returnValue(member)
|
2016-02-23 08:11:25 -07:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_host_in_room(self, room_id, host):
|
|
|
|
with Measure(self.clock, "check_host_in_room"):
|
2017-06-09 03:52:26 -06:00
|
|
|
latest_event_ids = yield self.store.is_host_joined(room_id, host)
|
|
|
|
defer.returnValue(latest_event_ids)
|
2015-04-22 07:20:04 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
def _check_joined_room(self, member, user_id, room_id):
|
|
|
|
if not member or member.membership != Membership.JOIN:
|
|
|
|
raise AuthError(403, "User %s not in room %s (%s)" % (
|
|
|
|
user_id, room_id, repr(member)
|
|
|
|
))
|
2014-10-15 09:06:59 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
def can_federate(self, event, auth_events):
|
|
|
|
creation_event = auth_events.get((EventTypes.Create, ""))
|
2015-04-21 13:53:23 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
return creation_event.content.get("m.federate", True) is True
|
2015-04-21 13:53:23 -06:00
|
|
|
|
2017-01-10 11:16:54 -07:00
|
|
|
def get_public_keys(self, invite_event):
|
2017-01-13 08:07:32 -07:00
|
|
|
return event_auth.get_public_keys(invite_event)
|
2014-10-15 09:06:59 -06:00
|
|
|
|
2014-09-26 09:36:24 -06:00
|
|
|
@defer.inlineCallbacks
|
2016-06-01 10:40:52 -06:00
|
|
|
def get_user_by_req(self, request, allow_guest=False, rights="access"):
|
2014-08-12 08:10:52 -06:00
|
|
|
""" Get a registered user's ID.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
request - An HTTP request with an access_token query parameter.
|
|
|
|
Returns:
|
2016-07-26 09:46:53 -06:00
|
|
|
defer.Deferred: resolves to a ``synapse.types.Requester`` object
|
2014-08-12 08:10:52 -06:00
|
|
|
Raises:
|
|
|
|
AuthError if no user by that token exists or the token is invalid.
|
|
|
|
"""
|
|
|
|
# Can optionally look elsewhere in the request (e.g. headers)
|
|
|
|
try:
|
2018-12-04 04:44:41 -07:00
|
|
|
ip_addr = self.hs.get_ip_from_request(request)
|
|
|
|
user_agent = request.requestHeaders.getRawHeaders(
|
|
|
|
b"User-Agent",
|
|
|
|
default=[b""]
|
|
|
|
)[0].decode('ascii', 'surrogateescape')
|
|
|
|
|
|
|
|
access_token = self.get_access_token_from_request(
|
|
|
|
request, self.TOKEN_NOT_FOUND_HTTP_STATUS
|
|
|
|
)
|
|
|
|
|
2016-10-20 05:04:54 -06:00
|
|
|
user_id, app_service = yield self._get_appservice_user_id(request)
|
2016-01-18 09:32:33 -07:00
|
|
|
if user_id:
|
2015-08-18 08:16:28 -06:00
|
|
|
request.authenticated_entity = user_id
|
2018-12-04 04:44:41 -07:00
|
|
|
|
|
|
|
if ip_addr and self.hs.config.track_appservice_user_ips:
|
|
|
|
yield self.store.insert_client_ip(
|
|
|
|
user_id=user_id,
|
|
|
|
access_token=access_token,
|
|
|
|
ip=ip_addr,
|
|
|
|
user_agent=user_agent,
|
|
|
|
device_id="dummy-device", # stubbed
|
|
|
|
)
|
|
|
|
|
2016-10-20 04:52:46 -06:00
|
|
|
defer.returnValue(
|
2016-10-20 05:04:54 -06:00
|
|
|
synapse.types.create_requester(user_id, app_service=app_service)
|
2016-10-20 04:52:46 -06:00
|
|
|
)
|
2015-02-05 08:00:33 -07:00
|
|
|
|
2016-06-01 10:40:52 -06:00
|
|
|
user_info = yield self.get_user_by_access_token(access_token, rights)
|
2014-09-29 07:59:52 -06:00
|
|
|
user = user_info["user"]
|
2015-01-28 09:58:23 -07:00
|
|
|
token_id = user_info["token_id"]
|
2015-11-04 10:29:07 -07:00
|
|
|
is_guest = user_info["is_guest"]
|
2014-09-26 09:36:24 -06:00
|
|
|
|
2016-07-20 08:25:40 -06:00
|
|
|
# device_id may not be present if get_user_by_access_token has been
|
|
|
|
# stubbed out.
|
|
|
|
device_id = user_info.get("device_id")
|
|
|
|
|
2014-09-26 09:36:24 -06:00
|
|
|
if user and access_token and ip_addr:
|
2018-08-02 06:47:19 -06:00
|
|
|
yield self.store.insert_client_ip(
|
2017-06-27 08:53:45 -06:00
|
|
|
user_id=user.to_string(),
|
2014-09-29 07:59:52 -06:00
|
|
|
access_token=access_token,
|
|
|
|
ip=ip_addr,
|
2016-07-20 08:25:40 -06:00
|
|
|
user_agent=user_agent,
|
|
|
|
device_id=device_id,
|
2014-09-29 06:35:15 -06:00
|
|
|
)
|
2014-09-26 09:36:24 -06:00
|
|
|
|
2015-11-04 10:29:07 -07:00
|
|
|
if is_guest and not allow_guest:
|
|
|
|
raise AuthError(
|
|
|
|
403, "Guest access not allowed", errcode=Codes.GUEST_ACCESS_FORBIDDEN
|
|
|
|
)
|
|
|
|
|
2015-06-15 10:11:44 -06:00
|
|
|
request.authenticated_entity = user.to_string()
|
|
|
|
|
2016-07-26 09:46:53 -06:00
|
|
|
defer.returnValue(synapse.types.create_requester(
|
2016-10-20 05:07:16 -06:00
|
|
|
user, token_id, is_guest, device_id, app_service=app_service)
|
|
|
|
)
|
2014-08-12 08:10:52 -06:00
|
|
|
except KeyError:
|
2015-03-24 11:24:15 -06:00
|
|
|
raise AuthError(
|
2015-04-23 06:23:44 -06:00
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "Missing access token.",
|
|
|
|
errcode=Codes.MISSING_TOKEN
|
2015-03-24 11:24:15 -06:00
|
|
|
)
|
2014-08-12 08:10:52 -06:00
|
|
|
|
2016-01-18 09:32:33 -07:00
|
|
|
@defer.inlineCallbacks
|
2016-09-09 09:29:10 -06:00
|
|
|
def _get_appservice_user_id(self, request):
|
2016-10-06 02:43:32 -06:00
|
|
|
app_service = self.store.get_app_service_by_token(
|
2018-07-13 15:34:49 -06:00
|
|
|
self.get_access_token_from_request(
|
2016-09-09 09:29:10 -06:00
|
|
|
request, self.TOKEN_NOT_FOUND_HTTP_STATUS
|
|
|
|
)
|
2016-01-18 09:32:33 -07:00
|
|
|
)
|
|
|
|
if app_service is None:
|
2016-10-20 04:43:05 -06:00
|
|
|
defer.returnValue((None, None))
|
2016-01-18 09:32:33 -07:00
|
|
|
|
2018-06-28 13:31:53 -06:00
|
|
|
if app_service.ip_range_whitelist:
|
|
|
|
ip_address = IPAddress(self.hs.get_ip_from_request(request))
|
|
|
|
if ip_address not in app_service.ip_range_whitelist:
|
|
|
|
defer.returnValue((None, None))
|
|
|
|
|
2018-08-01 08:54:06 -06:00
|
|
|
if b"user_id" not in request.args:
|
2016-10-20 04:43:05 -06:00
|
|
|
defer.returnValue((app_service.sender, app_service))
|
2016-01-18 09:32:33 -07:00
|
|
|
|
2018-08-01 08:54:06 -06:00
|
|
|
user_id = request.args[b"user_id"][0].decode('utf8')
|
2016-01-18 09:33:05 -07:00
|
|
|
if app_service.sender == user_id:
|
2016-10-20 04:43:05 -06:00
|
|
|
defer.returnValue((app_service.sender, app_service))
|
2016-01-18 09:32:33 -07:00
|
|
|
|
|
|
|
if not app_service.is_interested_in_user(user_id):
|
|
|
|
raise AuthError(
|
|
|
|
403,
|
|
|
|
"Application service cannot masquerade as this user."
|
|
|
|
)
|
|
|
|
if not (yield self.store.get_user_by_id(user_id)):
|
|
|
|
raise AuthError(
|
|
|
|
403,
|
|
|
|
"Application service has not registered this user"
|
2016-02-02 10:18:50 -07:00
|
|
|
)
|
2016-10-20 04:43:05 -06:00
|
|
|
defer.returnValue((user_id, app_service))
|
2016-01-18 09:32:33 -07:00
|
|
|
|
2014-08-12 08:10:52 -06:00
|
|
|
@defer.inlineCallbacks
|
2016-06-01 10:40:52 -06:00
|
|
|
def get_user_by_access_token(self, token, rights="access"):
|
2016-12-06 08:31:37 -07:00
|
|
|
""" Validate access token and get user_id from it
|
2014-08-12 08:10:52 -06:00
|
|
|
|
|
|
|
Args:
|
2014-09-29 08:35:54 -06:00
|
|
|
token (str): The access token to get the user by.
|
2016-12-06 08:31:37 -07:00
|
|
|
rights (str): The operation being performed; the access token must
|
|
|
|
allow this.
|
2014-08-12 08:10:52 -06:00
|
|
|
Returns:
|
2017-11-29 08:41:20 -07:00
|
|
|
Deferred[dict]: dict that includes:
|
|
|
|
`user` (UserID)
|
|
|
|
`is_guest` (bool)
|
|
|
|
`token_id` (int|None): access token id. May be None if guest
|
|
|
|
`device_id` (str|None): device corresponding to access token
|
2014-08-12 08:10:52 -06:00
|
|
|
Raises:
|
|
|
|
AuthError if no user by that token exists or the token is invalid.
|
|
|
|
"""
|
2015-08-26 06:22:23 -06:00
|
|
|
try:
|
2017-06-29 07:50:18 -06:00
|
|
|
user_id, guest = self._parse_and_validate_macaroon(token, rights)
|
|
|
|
except _InvalidMacaroonException:
|
2016-12-06 08:31:37 -07:00
|
|
|
# doesn't look like a macaroon: treat it as an opaque token which
|
|
|
|
# must be in the database.
|
|
|
|
# TODO: it would be nice to get rid of this, but apparently some
|
|
|
|
# people use access tokens which aren't macaroons
|
|
|
|
r = yield self._look_up_user_by_access_token(token)
|
|
|
|
defer.returnValue(r)
|
2015-08-26 06:22:23 -06:00
|
|
|
|
|
|
|
try:
|
2016-08-08 09:34:07 -06:00
|
|
|
user = UserID.from_string(user_id)
|
2015-11-04 10:29:07 -07:00
|
|
|
|
|
|
|
if guest:
|
2016-12-06 08:31:37 -07:00
|
|
|
# Guest access tokens are not stored in the database (there can
|
|
|
|
# only be one access token per guest, anyway).
|
|
|
|
#
|
|
|
|
# In order to prevent guest access tokens being used as regular
|
|
|
|
# user access tokens (and hence getting around the invalidation
|
|
|
|
# process), we look up the user id and check that it is indeed
|
|
|
|
# a guest user.
|
|
|
|
#
|
|
|
|
# It would of course be much easier to store guest access
|
|
|
|
# tokens in the database as well, but that would break existing
|
|
|
|
# guest tokens.
|
|
|
|
stored_user = yield self.store.get_user_by_id(user_id)
|
|
|
|
if not stored_user:
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS,
|
|
|
|
"Unknown user_id %s" % user_id,
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
|
|
|
if not stored_user["is_guest"]:
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS,
|
|
|
|
"Guest access token used for regular user",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
2015-11-04 10:29:07 -07:00
|
|
|
ret = {
|
|
|
|
"user": user,
|
|
|
|
"is_guest": True,
|
|
|
|
"token_id": None,
|
2016-11-25 08:25:30 -07:00
|
|
|
# all guests get the same device id
|
|
|
|
"device_id": GUEST_DEVICE_ID,
|
2015-11-04 10:29:07 -07:00
|
|
|
}
|
2016-06-02 10:21:31 -06:00
|
|
|
elif rights == "delete_pusher":
|
|
|
|
# We don't store these tokens in the database
|
|
|
|
ret = {
|
|
|
|
"user": user,
|
|
|
|
"is_guest": False,
|
|
|
|
"token_id": None,
|
2016-07-20 08:25:40 -06:00
|
|
|
"device_id": None,
|
2016-06-02 10:21:31 -06:00
|
|
|
}
|
2015-11-04 10:29:07 -07:00
|
|
|
else:
|
2016-07-20 08:25:40 -06:00
|
|
|
# This codepath exists for several reasons:
|
|
|
|
# * so that we can actually return a token ID, which is used
|
|
|
|
# in some parts of the schema (where we probably ought to
|
|
|
|
# use device IDs instead)
|
|
|
|
# * the only way we currently have to invalidate an
|
|
|
|
# access_token is by removing it from the database, so we
|
|
|
|
# have to check here that it is still in the db
|
|
|
|
# * some attributes (notably device_id) aren't stored in the
|
|
|
|
# macaroon. They probably should be.
|
|
|
|
# TODO: build the dictionary from the macaroon once the
|
|
|
|
# above are fixed
|
2016-12-06 08:31:37 -07:00
|
|
|
ret = yield self._look_up_user_by_access_token(token)
|
2015-11-04 10:29:07 -07:00
|
|
|
if ret["user"] != user:
|
|
|
|
logger.error(
|
|
|
|
"Macaroon user (%s) != DB user (%s)",
|
|
|
|
user,
|
|
|
|
ret["user"]
|
|
|
|
)
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS,
|
|
|
|
"User mismatch in macaroon",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
|
|
|
defer.returnValue(ret)
|
2015-08-26 06:22:23 -06:00
|
|
|
except (pymacaroons.exceptions.MacaroonException, TypeError, ValueError):
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "Invalid macaroon passed.",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
|
|
|
|
2017-06-29 07:50:18 -06:00
|
|
|
def _parse_and_validate_macaroon(self, token, rights="access"):
|
|
|
|
"""Takes a macaroon and tries to parse and validate it. This is cached
|
|
|
|
if and only if rights == access and there isn't an expiry.
|
|
|
|
|
|
|
|
On invalid macaroon raises _InvalidMacaroonException
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
(user_id, is_guest)
|
|
|
|
"""
|
|
|
|
if rights == "access":
|
|
|
|
cached = self.token_cache.get(token, None)
|
|
|
|
if cached:
|
|
|
|
return cached
|
|
|
|
|
|
|
|
try:
|
|
|
|
macaroon = pymacaroons.Macaroon.deserialize(token)
|
|
|
|
except Exception: # deserialize can throw more-or-less anything
|
|
|
|
# doesn't look like a macaroon: treat it as an opaque token which
|
|
|
|
# must be in the database.
|
|
|
|
# TODO: it would be nice to get rid of this, but apparently some
|
|
|
|
# people use access tokens which aren't macaroons
|
|
|
|
raise _InvalidMacaroonException()
|
|
|
|
|
|
|
|
try:
|
|
|
|
user_id = self.get_user_id_from_macaroon(macaroon)
|
|
|
|
|
|
|
|
has_expiry = False
|
|
|
|
guest = False
|
|
|
|
for caveat in macaroon.caveats:
|
|
|
|
if caveat.caveat_id.startswith("time "):
|
|
|
|
has_expiry = True
|
|
|
|
elif caveat.caveat_id == "guest = true":
|
|
|
|
guest = True
|
|
|
|
|
|
|
|
self.validate_macaroon(
|
|
|
|
macaroon, rights, self.hs.config.expire_access_token,
|
|
|
|
user_id=user_id,
|
|
|
|
)
|
|
|
|
except (pymacaroons.exceptions.MacaroonException, TypeError, ValueError):
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "Invalid macaroon passed.",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
|
|
|
|
|
|
|
if not has_expiry and rights == "access":
|
|
|
|
self.token_cache[token] = (user_id, guest)
|
|
|
|
|
|
|
|
return user_id, guest
|
|
|
|
|
2016-08-08 09:34:07 -06:00
|
|
|
def get_user_id_from_macaroon(self, macaroon):
|
|
|
|
"""Retrieve the user_id given by the caveats on the macaroon.
|
|
|
|
|
|
|
|
Does *not* validate the macaroon.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
macaroon (pymacaroons.Macaroon): The macaroon to validate
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
(str) user id
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
AuthError if there is no user_id caveat in the macaroon
|
|
|
|
"""
|
|
|
|
user_prefix = "user_id = "
|
|
|
|
for caveat in macaroon.caveats:
|
|
|
|
if caveat.caveat_id.startswith(user_prefix):
|
|
|
|
return caveat.caveat_id[len(user_prefix):]
|
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "No user caveat in macaroon",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
|
|
|
|
2016-07-07 09:11:37 -06:00
|
|
|
def validate_macaroon(self, macaroon, type_string, verify_expiry, user_id):
|
2015-11-19 08:16:25 -07:00
|
|
|
"""
|
|
|
|
validate that a Macaroon is understood by and was signed by this server.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
macaroon(pymacaroons.Macaroon): The macaroon to validate
|
2016-11-30 10:40:18 -07:00
|
|
|
type_string(str): The kind of token required (e.g. "access",
|
2016-06-01 10:40:52 -06:00
|
|
|
"delete_pusher")
|
2015-11-19 08:16:25 -07:00
|
|
|
verify_expiry(bool): Whether to verify whether the macaroon has expired.
|
2016-08-08 09:34:07 -06:00
|
|
|
user_id (str): The user_id required
|
2015-11-19 08:16:25 -07:00
|
|
|
"""
|
2015-08-26 06:22:23 -06:00
|
|
|
v = pymacaroons.Verifier()
|
2016-11-24 05:38:17 -07:00
|
|
|
|
|
|
|
# the verifier runs a test for every caveat on the macaroon, to check
|
|
|
|
# that it is met for the current request. Each caveat must match at
|
|
|
|
# least one of the predicates specified by satisfy_exact or
|
|
|
|
# specify_general.
|
2015-08-26 06:22:23 -06:00
|
|
|
v.satisfy_exact("gen = 1")
|
2015-11-11 04:12:35 -07:00
|
|
|
v.satisfy_exact("type = " + type_string)
|
2016-07-07 09:11:37 -06:00
|
|
|
v.satisfy_exact("user_id = %s" % user_id)
|
2015-11-17 03:58:05 -07:00
|
|
|
v.satisfy_exact("guest = true")
|
2016-11-30 00:36:32 -07:00
|
|
|
|
|
|
|
# verify_expiry should really always be True, but there exist access
|
|
|
|
# tokens in the wild which expire when they should not, so we can't
|
|
|
|
# enforce expiry yet (so we have to allow any caveat starting with
|
|
|
|
# 'time < ' in access tokens).
|
|
|
|
#
|
|
|
|
# On the other hand, short-term login tokens (as used by CAS login, for
|
|
|
|
# example) have an expiry time which we do want to enforce.
|
|
|
|
|
2015-11-19 08:16:25 -07:00
|
|
|
if verify_expiry:
|
|
|
|
v.satisfy_general(self._verify_expiry)
|
|
|
|
else:
|
|
|
|
v.satisfy_general(lambda c: c.startswith("time < "))
|
2015-11-11 04:12:35 -07:00
|
|
|
|
2016-11-30 10:40:18 -07:00
|
|
|
# access_tokens include a nonce for uniqueness: any value is acceptable
|
2016-11-28 02:55:21 -07:00
|
|
|
v.satisfy_general(lambda c: c.startswith("nonce = "))
|
|
|
|
|
2015-08-26 06:22:23 -06:00
|
|
|
v.verify(macaroon, self.hs.config.macaroon_secret_key)
|
|
|
|
|
2015-11-19 08:16:25 -07:00
|
|
|
def _verify_expiry(self, caveat):
|
2015-08-26 06:22:23 -06:00
|
|
|
prefix = "time < "
|
|
|
|
if not caveat.startswith(prefix):
|
|
|
|
return False
|
|
|
|
expiry = int(caveat[len(prefix):])
|
|
|
|
now = self.hs.get_clock().time_msec()
|
|
|
|
return now < expiry
|
|
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
|
|
def _look_up_user_by_access_token(self, token):
|
2015-08-20 09:01:29 -06:00
|
|
|
ret = yield self.store.get_user_by_access_token(token)
|
2015-03-24 11:24:15 -06:00
|
|
|
if not ret:
|
2018-06-24 12:17:01 -06:00
|
|
|
logger.warn("Unrecognised access token - not in store.")
|
2015-03-24 11:24:15 -06:00
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "Unrecognised access token.",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
2016-07-20 08:25:40 -06:00
|
|
|
# we use ret.get() below because *lots* of unit tests stub out
|
|
|
|
# get_user_by_access_token in a way where it only returns a couple of
|
|
|
|
# the fields.
|
2015-03-24 11:24:15 -06:00
|
|
|
user_info = {
|
|
|
|
"user": UserID.from_string(ret.get("name")),
|
|
|
|
"token_id": ret.get("token_id", None),
|
2015-11-04 10:29:07 -07:00
|
|
|
"is_guest": False,
|
2016-07-20 08:25:40 -06:00
|
|
|
"device_id": ret.get("device_id"),
|
2015-03-24 11:24:15 -06:00
|
|
|
}
|
|
|
|
defer.returnValue(user_info)
|
2014-09-01 06:44:19 -06:00
|
|
|
|
2015-02-06 03:57:14 -07:00
|
|
|
def get_appservice_by_req(self, request):
|
|
|
|
try:
|
2018-07-13 15:34:49 -06:00
|
|
|
token = self.get_access_token_from_request(
|
2016-09-09 09:29:10 -06:00
|
|
|
request, self.TOKEN_NOT_FOUND_HTTP_STATUS
|
|
|
|
)
|
2016-10-06 02:43:32 -06:00
|
|
|
service = self.store.get_app_service_by_token(token)
|
2015-02-06 03:57:14 -07:00
|
|
|
if not service:
|
2018-06-24 12:17:01 -06:00
|
|
|
logger.warn("Unrecognised appservice access token.")
|
2015-03-24 11:24:15 -06:00
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS,
|
|
|
|
"Unrecognised access token.",
|
|
|
|
errcode=Codes.UNKNOWN_TOKEN
|
|
|
|
)
|
2015-08-18 08:16:28 -06:00
|
|
|
request.authenticated_entity = service.sender
|
2016-10-06 02:43:32 -06:00
|
|
|
return defer.succeed(service)
|
2015-02-06 03:57:14 -07:00
|
|
|
except KeyError:
|
2015-03-24 11:24:15 -06:00
|
|
|
raise AuthError(
|
|
|
|
self.TOKEN_NOT_FOUND_HTTP_STATUS, "Missing access token."
|
|
|
|
)
|
2015-02-06 03:57:14 -07:00
|
|
|
|
2014-09-29 06:35:38 -06:00
|
|
|
def is_server_admin(self, user):
|
2017-09-19 09:08:14 -06:00
|
|
|
""" Check if the given user is a local server admin.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
user (str): mxid of user to check
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
bool: True if the user is an admin
|
|
|
|
"""
|
2014-09-29 06:35:38 -06:00
|
|
|
return self.store.is_server_admin(user)
|
|
|
|
|
2014-11-07 03:42:44 -07:00
|
|
|
@defer.inlineCallbacks
|
2014-12-05 09:20:48 -07:00
|
|
|
def add_auth_events(self, builder, context):
|
2018-07-23 06:00:22 -06:00
|
|
|
prev_state_ids = yield context.get_prev_state_ids(self.store)
|
|
|
|
auth_ids = yield self.compute_auth_events(builder, prev_state_ids)
|
2015-01-28 09:16:53 -07:00
|
|
|
|
|
|
|
auth_events_entries = yield self.store.add_event_hashes(
|
|
|
|
auth_ids
|
|
|
|
)
|
|
|
|
|
|
|
|
builder.auth_events = auth_events_entries
|
|
|
|
|
2016-08-25 10:32:22 -06:00
|
|
|
@defer.inlineCallbacks
|
|
|
|
def compute_auth_events(self, event, current_state_ids, for_verification=False):
|
2015-01-28 09:16:53 -07:00
|
|
|
if event.type == EventTypes.Create:
|
2016-08-25 10:32:22 -06:00
|
|
|
defer.returnValue([])
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2014-12-08 02:08:26 -07:00
|
|
|
auth_ids = []
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2014-12-12 09:31:50 -07:00
|
|
|
key = (EventTypes.PowerLevels, "", )
|
2016-08-25 10:32:22 -06:00
|
|
|
power_level_event_id = current_state_ids.get(key)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2016-08-25 10:32:22 -06:00
|
|
|
if power_level_event_id:
|
|
|
|
auth_ids.append(power_level_event_id)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2014-12-12 09:31:50 -07:00
|
|
|
key = (EventTypes.JoinRules, "", )
|
2016-08-25 10:32:22 -06:00
|
|
|
join_rule_event_id = current_state_ids.get(key)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2015-01-28 09:16:53 -07:00
|
|
|
key = (EventTypes.Member, event.user_id, )
|
2016-08-25 10:32:22 -06:00
|
|
|
member_event_id = current_state_ids.get(key)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2014-12-12 09:31:50 -07:00
|
|
|
key = (EventTypes.Create, "", )
|
2016-08-25 10:32:22 -06:00
|
|
|
create_event_id = current_state_ids.get(key)
|
|
|
|
if create_event_id:
|
|
|
|
auth_ids.append(create_event_id)
|
2014-11-25 04:31:18 -07:00
|
|
|
|
2016-08-25 10:32:22 -06:00
|
|
|
if join_rule_event_id:
|
|
|
|
join_rule_event = yield self.store.get_event(join_rule_event_id)
|
2014-11-07 03:42:44 -07:00
|
|
|
join_rule = join_rule_event.content.get("join_rule")
|
|
|
|
is_public = join_rule == JoinRules.PUBLIC if join_rule else False
|
2014-11-10 04:15:02 -07:00
|
|
|
else:
|
|
|
|
is_public = False
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2015-01-28 09:16:53 -07:00
|
|
|
if event.type == EventTypes.Member:
|
|
|
|
e_type = event.content["membership"]
|
2014-11-10 04:15:02 -07:00
|
|
|
if e_type in [Membership.JOIN, Membership.INVITE]:
|
2016-08-25 10:32:22 -06:00
|
|
|
if join_rule_event_id:
|
|
|
|
auth_ids.append(join_rule_event_id)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2014-12-12 03:42:27 -07:00
|
|
|
if e_type == Membership.JOIN:
|
2016-08-25 10:32:22 -06:00
|
|
|
if member_event_id and not is_public:
|
|
|
|
auth_ids.append(member_event_id)
|
2015-11-05 09:43:19 -07:00
|
|
|
else:
|
2016-08-25 10:32:22 -06:00
|
|
|
if member_event_id:
|
|
|
|
auth_ids.append(member_event_id)
|
|
|
|
|
|
|
|
if for_verification:
|
|
|
|
key = (EventTypes.Member, event.state_key, )
|
|
|
|
existing_event_id = current_state_ids.get(key)
|
|
|
|
if existing_event_id:
|
|
|
|
auth_ids.append(existing_event_id)
|
2015-11-05 09:43:19 -07:00
|
|
|
|
|
|
|
if e_type == Membership.INVITE:
|
|
|
|
if "third_party_invite" in event.content:
|
2015-10-13 08:48:12 -06:00
|
|
|
key = (
|
|
|
|
EventTypes.ThirdPartyInvite,
|
2015-12-17 10:09:51 -07:00
|
|
|
event.content["third_party_invite"]["signed"]["token"]
|
2015-10-13 08:48:12 -06:00
|
|
|
)
|
2016-08-25 10:32:22 -06:00
|
|
|
third_party_invite_id = current_state_ids.get(key)
|
|
|
|
if third_party_invite_id:
|
|
|
|
auth_ids.append(third_party_invite_id)
|
|
|
|
elif member_event_id:
|
|
|
|
member_event = yield self.store.get_event(member_event_id)
|
2014-11-07 03:42:44 -07:00
|
|
|
if member_event.content["membership"] == Membership.JOIN:
|
2014-12-08 02:08:26 -07:00
|
|
|
auth_ids.append(member_event.event_id)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2016-08-25 10:32:22 -06:00
|
|
|
defer.returnValue(auth_ids)
|
2014-11-07 03:42:44 -07:00
|
|
|
|
2015-08-28 08:31:49 -06:00
|
|
|
def check_redaction(self, event, auth_events):
|
|
|
|
"""Check whether the event sender is allowed to redact the target event.
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
True if the the sender is allowed to redact the target event if the
|
|
|
|
target event was created by them.
|
|
|
|
False if the sender is allowed to redact the target event with no
|
|
|
|
further checks.
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
AuthError if the event sender is definitely not allowed to redact
|
|
|
|
the target event.
|
|
|
|
"""
|
2017-01-13 08:07:32 -07:00
|
|
|
return event_auth.check_redaction(event, auth_events)
|
2016-03-21 08:03:20 -06:00
|
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_can_change_room_list(self, room_id, user):
|
|
|
|
"""Check if the user is allowed to edit the room's entry in the
|
|
|
|
published room list.
|
|
|
|
|
|
|
|
Args:
|
|
|
|
room_id (str)
|
|
|
|
user (UserID)
|
|
|
|
"""
|
|
|
|
|
|
|
|
is_admin = yield self.is_server_admin(user)
|
|
|
|
if is_admin:
|
|
|
|
defer.returnValue(True)
|
|
|
|
|
|
|
|
user_id = user.to_string()
|
|
|
|
yield self.check_joined_room(room_id, user_id)
|
|
|
|
|
|
|
|
# We currently require the user is a "moderator" in the room. We do this
|
|
|
|
# by checking if they would (theoretically) be able to change the
|
|
|
|
# m.room.aliases events
|
|
|
|
power_level_event = yield self.state.get_current_state(
|
|
|
|
room_id, EventTypes.PowerLevels, ""
|
|
|
|
)
|
|
|
|
|
|
|
|
auth_events = {}
|
|
|
|
if power_level_event:
|
|
|
|
auth_events[(EventTypes.PowerLevels, "")] = power_level_event
|
|
|
|
|
2017-01-13 08:07:32 -07:00
|
|
|
send_level = event_auth.get_send_level(
|
2018-06-14 04:26:27 -06:00
|
|
|
EventTypes.Aliases, "", power_level_event,
|
2016-03-21 08:03:20 -06:00
|
|
|
)
|
2017-01-13 08:07:32 -07:00
|
|
|
user_level = event_auth.get_user_power_level(user_id, auth_events)
|
2016-03-21 08:03:20 -06:00
|
|
|
|
|
|
|
if user_level < send_level:
|
|
|
|
raise AuthError(
|
|
|
|
403,
|
|
|
|
"This server requires you to be a moderator in the room to"
|
|
|
|
" edit its room list entry"
|
|
|
|
)
|
2016-09-09 09:29:10 -06:00
|
|
|
|
2018-07-13 15:34:49 -06:00
|
|
|
@staticmethod
|
|
|
|
def has_access_token(request):
|
|
|
|
"""Checks if the request has an access_token.
|
2016-09-09 09:29:10 -06:00
|
|
|
|
2018-07-13 15:34:49 -06:00
|
|
|
Returns:
|
|
|
|
bool: False if no access_token was given, True otherwise.
|
|
|
|
"""
|
2018-08-20 07:54:49 -06:00
|
|
|
query_params = request.args.get(b"access_token")
|
2018-07-13 15:34:49 -06:00
|
|
|
auth_headers = request.requestHeaders.getRawHeaders(b"Authorization")
|
|
|
|
return bool(query_params) or bool(auth_headers)
|
2016-09-09 09:29:10 -06:00
|
|
|
|
2018-07-13 15:34:49 -06:00
|
|
|
@staticmethod
|
|
|
|
def get_access_token_from_request(request, token_not_found_http_status=401):
|
|
|
|
"""Extracts the access_token from the request.
|
2016-09-09 11:17:42 -06:00
|
|
|
|
2018-07-13 15:34:49 -06:00
|
|
|
Args:
|
|
|
|
request: The http request.
|
|
|
|
token_not_found_http_status(int): The HTTP status code to set in the
|
|
|
|
AuthError if the token isn't found. This is used in some of the
|
|
|
|
legacy APIs to change the status code to 403 from the default of
|
|
|
|
401 since some of the old clients depended on auth errors returning
|
|
|
|
403.
|
|
|
|
Returns:
|
2018-08-20 07:54:49 -06:00
|
|
|
unicode: The access_token
|
2018-07-13 15:34:49 -06:00
|
|
|
Raises:
|
|
|
|
AuthError: If there isn't an access_token in the request.
|
|
|
|
"""
|
|
|
|
|
|
|
|
auth_headers = request.requestHeaders.getRawHeaders(b"Authorization")
|
|
|
|
query_params = request.args.get(b"access_token")
|
|
|
|
if auth_headers:
|
|
|
|
# Try the get the access_token from a "Authorization: Bearer"
|
|
|
|
# header
|
|
|
|
if query_params is not None:
|
|
|
|
raise AuthError(
|
|
|
|
token_not_found_http_status,
|
|
|
|
"Mixing Authorization headers and access_token query parameters.",
|
|
|
|
errcode=Codes.MISSING_TOKEN,
|
|
|
|
)
|
|
|
|
if len(auth_headers) > 1:
|
|
|
|
raise AuthError(
|
|
|
|
token_not_found_http_status,
|
|
|
|
"Too many Authorization headers.",
|
|
|
|
errcode=Codes.MISSING_TOKEN,
|
|
|
|
)
|
2018-08-20 07:54:49 -06:00
|
|
|
parts = auth_headers[0].split(b" ")
|
|
|
|
if parts[0] == b"Bearer" and len(parts) == 2:
|
|
|
|
return parts[1].decode('ascii')
|
2018-07-13 15:34:49 -06:00
|
|
|
else:
|
|
|
|
raise AuthError(
|
|
|
|
token_not_found_http_status,
|
|
|
|
"Invalid Authorization header.",
|
|
|
|
errcode=Codes.MISSING_TOKEN,
|
|
|
|
)
|
2016-09-09 11:17:42 -06:00
|
|
|
else:
|
2018-07-13 15:34:49 -06:00
|
|
|
# Try to get the access_token from the query params.
|
|
|
|
if not query_params:
|
|
|
|
raise AuthError(
|
|
|
|
token_not_found_http_status,
|
|
|
|
"Missing access token.",
|
|
|
|
errcode=Codes.MISSING_TOKEN
|
|
|
|
)
|
2016-09-09 09:29:10 -06:00
|
|
|
|
2018-08-20 07:54:49 -06:00
|
|
|
return query_params[0].decode('ascii')
|
2018-07-20 08:30:59 -06:00
|
|
|
|
|
|
|
@defer.inlineCallbacks
|
|
|
|
def check_in_room_or_world_readable(self, room_id, user_id):
|
|
|
|
"""Checks that the user is or was in the room or the room is world
|
|
|
|
readable. If it isn't then an exception is raised.
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
Deferred[tuple[str, str|None]]: Resolves to the current membership of
|
|
|
|
the user in the room and the membership event ID of the user. If
|
|
|
|
the user is not in the room and never has been, then
|
|
|
|
`(Membership.JOIN, None)` is returned.
|
|
|
|
"""
|
|
|
|
|
|
|
|
try:
|
|
|
|
# check_user_was_in_room will return the most recent membership
|
|
|
|
# event for the user if:
|
|
|
|
# * The user is a non-guest user, and was ever in the room
|
|
|
|
# * The user is a guest user, and has joined the room
|
|
|
|
# else it will throw.
|
|
|
|
member_event = yield self.check_user_was_in_room(room_id, user_id)
|
|
|
|
defer.returnValue((member_event.membership, member_event.event_id))
|
|
|
|
except AuthError:
|
|
|
|
visibility = yield self.state.get_current_state(
|
|
|
|
room_id, EventTypes.RoomHistoryVisibility, ""
|
|
|
|
)
|
|
|
|
if (
|
|
|
|
visibility and
|
|
|
|
visibility.content["history_visibility"] == "world_readable"
|
|
|
|
):
|
|
|
|
defer.returnValue((Membership.JOIN, None))
|
|
|
|
return
|
|
|
|
raise AuthError(
|
|
|
|
403, "Guest access not allowed", errcode=Codes.GUEST_ACCESS_FORBIDDEN
|
|
|
|
)
|
2018-08-02 09:57:35 -06:00
|
|
|
|
|
|
|
@defer.inlineCallbacks
|
2018-08-31 03:49:14 -06:00
|
|
|
def check_auth_blocking(self, user_id=None, threepid=None):
|
2018-08-02 09:57:35 -06:00
|
|
|
"""Checks if the user should be rejected for some external reason,
|
|
|
|
such as monthly active user limiting or global disable flag
|
2018-08-09 10:39:12 -06:00
|
|
|
|
|
|
|
Args:
|
2018-08-14 08:48:12 -06:00
|
|
|
user_id(str|None): If present, checks for presence against existing
|
|
|
|
MAU cohort
|
2018-08-31 10:29:35 -06:00
|
|
|
|
2018-08-31 10:11:11 -06:00
|
|
|
threepid(dict|None): If present, checks for presence against configured
|
|
|
|
reserved threepid. Used in cases where the user is trying register
|
|
|
|
with a MAU blocked server, normally they would be rejected but their
|
|
|
|
threepid is on the reserved list. user_id and
|
|
|
|
threepid should never be set at the same time.
|
2018-08-06 11:01:46 -06:00
|
|
|
"""
|
2018-08-18 05:33:07 -06:00
|
|
|
|
|
|
|
# Never fail an auth check for the server notices users
|
|
|
|
# This can be a problem where event creation is prohibited due to blocking
|
|
|
|
if user_id == self.hs.config.server_notices_mxid:
|
|
|
|
return
|
|
|
|
|
2018-08-04 15:07:04 -06:00
|
|
|
if self.hs.config.hs_disabled:
|
2018-08-16 11:02:02 -06:00
|
|
|
raise ResourceLimitError(
|
2018-08-13 11:00:23 -06:00
|
|
|
403, self.hs.config.hs_disabled_message,
|
2018-08-18 07:39:45 -06:00
|
|
|
errcode=Codes.RESOURCE_LIMIT_EXCEEDED,
|
2018-08-24 09:51:27 -06:00
|
|
|
admin_contact=self.hs.config.admin_contact,
|
2018-08-16 11:02:02 -06:00
|
|
|
limit_type=self.hs.config.hs_disabled_limit_type
|
2018-08-04 15:07:04 -06:00
|
|
|
)
|
2018-08-02 09:57:35 -06:00
|
|
|
if self.hs.config.limit_usage_by_mau is True:
|
2018-08-31 10:29:35 -06:00
|
|
|
assert not (user_id and threepid)
|
2018-08-31 10:11:11 -06:00
|
|
|
|
2018-08-23 12:17:08 -06:00
|
|
|
# If the user is already part of the MAU cohort or a trial user
|
2018-08-09 10:39:12 -06:00
|
|
|
if user_id:
|
2018-08-09 11:02:12 -06:00
|
|
|
timestamp = yield self.store.user_last_seen_monthly_active(user_id)
|
2018-08-09 10:39:12 -06:00
|
|
|
if timestamp:
|
|
|
|
return
|
2018-08-23 12:17:08 -06:00
|
|
|
|
|
|
|
is_trial = yield self.store.is_trial_user(user_id)
|
|
|
|
if is_trial:
|
|
|
|
return
|
2018-08-31 03:49:14 -06:00
|
|
|
elif threepid:
|
|
|
|
# If the user does not exist yet, but is signing up with a
|
|
|
|
# reserved threepid then pass auth check
|
2018-08-31 10:11:11 -06:00
|
|
|
if is_threepid_reserved(self.hs.config, threepid):
|
2018-08-31 08:42:51 -06:00
|
|
|
return
|
2018-08-09 10:39:12 -06:00
|
|
|
# Else if there is no room in the MAU bucket, bail
|
2018-08-02 09:57:35 -06:00
|
|
|
current_mau = yield self.store.get_monthly_active_count()
|
|
|
|
if current_mau >= self.hs.config.max_mau_value:
|
2018-08-16 11:02:02 -06:00
|
|
|
raise ResourceLimitError(
|
2018-08-16 15:55:32 -06:00
|
|
|
403, "Monthly Active User Limit Exceeded",
|
2018-08-24 09:51:27 -06:00
|
|
|
admin_contact=self.hs.config.admin_contact,
|
2018-08-18 07:39:45 -06:00
|
|
|
errcode=Codes.RESOURCE_LIMIT_EXCEEDED,
|
2018-08-16 11:02:02 -06:00
|
|
|
limit_type="monthly_active_user"
|
2018-08-03 10:55:50 -06:00
|
|
|
)
|