添加 'rest_auth_provider/rest_auth_provider.py'
This commit is contained in:
parent
3e1e6828f7
commit
35ad989c94
211
rest_auth_provider/rest_auth_provider.py
Normal file
211
rest_auth_provider/rest_auth_provider.py
Normal file
@ -0,0 +1,211 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
#
|
||||||
|
# REST endpoint Authentication module for Matrix synapse
|
||||||
|
# Copyright (C) 2017 Kamax Sarl
|
||||||
|
#
|
||||||
|
# https://www.kamax.io/
|
||||||
|
#
|
||||||
|
# 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.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU Affero General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU Affero General Public License
|
||||||
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
#
|
||||||
|
|
||||||
|
import logging
|
||||||
|
import requests
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class RestAuthProvider(object):
|
||||||
|
|
||||||
|
def __init__(self, config, account_handler):
|
||||||
|
self.account_handler = account_handler
|
||||||
|
|
||||||
|
if not config.endpoint:
|
||||||
|
raise RuntimeError('Missing endpoint config')
|
||||||
|
|
||||||
|
self.endpoint = config.endpoint
|
||||||
|
self.regLower = config.regLower
|
||||||
|
self.config = config
|
||||||
|
|
||||||
|
logger.info('Endpoint: %s', self.endpoint)
|
||||||
|
logger.info('Enforce lowercase username during registration: %s', self.regLower)
|
||||||
|
|
||||||
|
async def check_password(self, user_id, password):
|
||||||
|
logger.info("Got password check for " + user_id)
|
||||||
|
data = {'user': {'id': user_id, 'password': password}}
|
||||||
|
r = requests.post(self.endpoint + '/_matrix-internal/identity/v1/check_credentials', json=data)
|
||||||
|
r.raise_for_status()
|
||||||
|
r = r.json()
|
||||||
|
if not r["auth"]:
|
||||||
|
reason = "Invalid JSON data returned from REST endpoint"
|
||||||
|
logger.warning(reason)
|
||||||
|
raise RuntimeError(reason)
|
||||||
|
|
||||||
|
auth = r["auth"]
|
||||||
|
if not auth["success"]:
|
||||||
|
logger.info("User not authenticated")
|
||||||
|
return False
|
||||||
|
|
||||||
|
localpart = user_id.split(":", 1)[0][1:]
|
||||||
|
logger.info("User %s authenticated", user_id)
|
||||||
|
|
||||||
|
registration = False
|
||||||
|
if not (await self.account_handler.check_user_exists(user_id)):
|
||||||
|
logger.info("User %s does not exist yet, creating...", user_id)
|
||||||
|
|
||||||
|
if localpart != localpart.lower() and self.regLower:
|
||||||
|
logger.info('User %s was cannot be created due to username lowercase policy', localpart)
|
||||||
|
return False
|
||||||
|
|
||||||
|
user_id, access_token = (await self.account_handler.register(localpart=localpart))
|
||||||
|
registration = True
|
||||||
|
logger.info("Registration based on REST data was successful for %s", user_id)
|
||||||
|
else:
|
||||||
|
logger.info("User %s already exists, registration skipped", user_id)
|
||||||
|
|
||||||
|
if auth["profile"]:
|
||||||
|
logger.info("Handling profile data")
|
||||||
|
profile = auth["profile"]
|
||||||
|
|
||||||
|
store = self.account_handler._hs.get_profile_handler().store
|
||||||
|
|
||||||
|
if "display_name" in profile and ((registration and self.config.setNameOnRegister) or (self.config.setNameOnLogin)):
|
||||||
|
display_name = profile["display_name"]
|
||||||
|
logger.info("Setting display name to '%s' based on profile data", display_name)
|
||||||
|
await store.set_profile_displayname(localpart, display_name)
|
||||||
|
else:
|
||||||
|
logger.info("Display name was not set because it was not given or policy restricted it")
|
||||||
|
|
||||||
|
if (self.config.updateThreepid):
|
||||||
|
if "three_pids" in profile:
|
||||||
|
logger.info("Handling 3PIDs")
|
||||||
|
|
||||||
|
external_3pids = []
|
||||||
|
for threepid in profile["three_pids"]:
|
||||||
|
medium = threepid["medium"].lower()
|
||||||
|
address = threepid["address"].lower()
|
||||||
|
external_3pids.append({"medium": medium, "address": address})
|
||||||
|
logger.info("Looking for 3PID %s:%s in user profile", medium, address)
|
||||||
|
|
||||||
|
validated_at = time_msec()
|
||||||
|
if not (await store.get_user_id_by_threepid(medium, address)):
|
||||||
|
logger.info("3PID is not present, adding")
|
||||||
|
await store.user_add_threepid(
|
||||||
|
user_id,
|
||||||
|
medium,
|
||||||
|
address,
|
||||||
|
validated_at,
|
||||||
|
validated_at
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
logger.info("3PID is present, skipping")
|
||||||
|
|
||||||
|
if (self.config.replaceThreepid):
|
||||||
|
for threepid in (await store.user_get_threepids(user_id)):
|
||||||
|
medium = threepid["medium"].lower()
|
||||||
|
address = threepid["address"].lower()
|
||||||
|
if {"medium": medium, "address": address} not in external_3pids:
|
||||||
|
logger.info("3PID is not present in external datastore, deleting")
|
||||||
|
await store.user_delete_threepid(
|
||||||
|
user_id,
|
||||||
|
medium,
|
||||||
|
address
|
||||||
|
)
|
||||||
|
|
||||||
|
else:
|
||||||
|
logger.info("3PIDs were not updated due to policy")
|
||||||
|
else:
|
||||||
|
logger.info("No profile data")
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def parse_config(config):
|
||||||
|
# verify config sanity
|
||||||
|
_require_keys(config, ["endpoint"])
|
||||||
|
|
||||||
|
class _RestConfig(object):
|
||||||
|
endpoint = ''
|
||||||
|
regLower = True
|
||||||
|
setNameOnRegister = True
|
||||||
|
setNameOnLogin = False
|
||||||
|
updateThreepid = True
|
||||||
|
replaceThreepid = False
|
||||||
|
|
||||||
|
rest_config = _RestConfig()
|
||||||
|
rest_config.endpoint = config["endpoint"]
|
||||||
|
|
||||||
|
try:
|
||||||
|
rest_config.regLower = config['policy']['registration']['username']['enforceLowercase']
|
||||||
|
except TypeError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
rest_config.setNameOnRegister = config['policy']['registration']['profile']['name']
|
||||||
|
except TypeError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
rest_config.setNameOnLogin = config['policy']['login']['profile']['name']
|
||||||
|
except TypeError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
rest_config.updateThreepid = config['policy']['all']['threepid']['update']
|
||||||
|
except TypeError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
|
rest_config.replaceThreepid = config['policy']['all']['threepid']['replace']
|
||||||
|
except TypeError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
except KeyError:
|
||||||
|
# we don't care
|
||||||
|
pass
|
||||||
|
|
||||||
|
return rest_config
|
||||||
|
|
||||||
|
|
||||||
|
def _require_keys(config, required):
|
||||||
|
missing = [key for key in required if key not in config]
|
||||||
|
if missing:
|
||||||
|
raise Exception(
|
||||||
|
"REST Auth enabled but missing required config values: {}".format(
|
||||||
|
", ".join(missing)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def time_msec():
|
||||||
|
"""Get the current timestamp in milliseconds
|
||||||
|
"""
|
||||||
|
return int(time.time() * 1000)
|
||||||
Loading…
Reference in New Issue
Block a user