261 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			Python
		
	
	
			
		
		
	
	
			261 lines
		
	
	
		
			8.9 KiB
		
	
	
	
		
			Python
		
	
	
| # -*- coding: utf-8 -*-
 | |
| # Copyright 2015, 2016 OpenMarket Ltd
 | |
| # Copyright 2017 Vector Creations Ltd
 | |
| # Copyright 2018 New Vector Ltd
 | |
| #
 | |
| # 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.
 | |
| 
 | |
| """Utilities for interacting with Identity Servers"""
 | |
| 
 | |
| import logging
 | |
| 
 | |
| from canonicaljson import json
 | |
| 
 | |
| from twisted.internet import defer
 | |
| 
 | |
| from synapse.api.errors import (
 | |
|     CodeMessageException,
 | |
|     Codes,
 | |
|     HttpResponseException,
 | |
|     SynapseError,
 | |
| )
 | |
| 
 | |
| from ._base import BaseHandler
 | |
| 
 | |
| logger = logging.getLogger(__name__)
 | |
| 
 | |
| 
 | |
| class IdentityHandler(BaseHandler):
 | |
| 
 | |
|     def __init__(self, hs):
 | |
|         super(IdentityHandler, self).__init__(hs)
 | |
| 
 | |
|         self.http_client = hs.get_simple_http_client()
 | |
|         self.federation_http_client = hs.get_http_client()
 | |
| 
 | |
|         self.trusted_id_servers = set(hs.config.trusted_third_party_id_servers)
 | |
|         self.trust_any_id_server_just_for_testing_do_not_use = (
 | |
|             hs.config.use_insecure_ssl_client_just_for_testing_do_not_use
 | |
|         )
 | |
| 
 | |
|     def _should_trust_id_server(self, id_server):
 | |
|         if id_server not in self.trusted_id_servers:
 | |
|             if self.trust_any_id_server_just_for_testing_do_not_use:
 | |
|                 logger.warn(
 | |
|                     "Trusting untrustworthy ID server %r even though it isn't"
 | |
|                     " in the trusted id list for testing because"
 | |
|                     " 'use_insecure_ssl_client_just_for_testing_do_not_use'"
 | |
|                     " is set in the config",
 | |
|                     id_server,
 | |
|                 )
 | |
|             else:
 | |
|                 return False
 | |
|         return True
 | |
| 
 | |
|     @defer.inlineCallbacks
 | |
|     def threepid_from_creds(self, creds):
 | |
|         if 'id_server' in creds:
 | |
|             id_server = creds['id_server']
 | |
|         elif 'idServer' in creds:
 | |
|             id_server = creds['idServer']
 | |
|         else:
 | |
|             raise SynapseError(400, "No id_server in creds")
 | |
| 
 | |
|         if 'client_secret' in creds:
 | |
|             client_secret = creds['client_secret']
 | |
|         elif 'clientSecret' in creds:
 | |
|             client_secret = creds['clientSecret']
 | |
|         else:
 | |
|             raise SynapseError(400, "No client_secret in creds")
 | |
| 
 | |
|         if not self._should_trust_id_server(id_server):
 | |
|             logger.warn(
 | |
|                 '%s is not a trusted ID server: rejecting 3pid ' +
 | |
|                 'credentials', id_server
 | |
|             )
 | |
|             defer.returnValue(None)
 | |
| 
 | |
|         try:
 | |
|             data = yield self.http_client.get_json(
 | |
|                 "https://%s%s" % (
 | |
|                     id_server,
 | |
|                     "/_matrix/identity/api/v1/3pid/getValidated3pid"
 | |
|                 ),
 | |
|                 {'sid': creds['sid'], 'client_secret': client_secret}
 | |
|             )
 | |
|         except HttpResponseException as e:
 | |
|             logger.info("getValidated3pid failed with Matrix error: %r", e)
 | |
|             raise e.to_synapse_error()
 | |
| 
 | |
|         if 'medium' in data:
 | |
|             defer.returnValue(data)
 | |
|         defer.returnValue(None)
 | |
| 
 | |
|     @defer.inlineCallbacks
 | |
|     def bind_threepid(self, creds, mxid):
 | |
|         logger.debug("binding threepid %r to %s", creds, mxid)
 | |
|         data = None
 | |
| 
 | |
|         if 'id_server' in creds:
 | |
|             id_server = creds['id_server']
 | |
|         elif 'idServer' in creds:
 | |
|             id_server = creds['idServer']
 | |
|         else:
 | |
|             raise SynapseError(400, "No id_server in creds")
 | |
| 
 | |
|         if 'client_secret' in creds:
 | |
|             client_secret = creds['client_secret']
 | |
|         elif 'clientSecret' in creds:
 | |
|             client_secret = creds['clientSecret']
 | |
|         else:
 | |
|             raise SynapseError(400, "No client_secret in creds")
 | |
| 
 | |
|         try:
 | |
|             data = yield self.http_client.post_urlencoded_get_json(
 | |
|                 "https://%s%s" % (
 | |
|                     id_server, "/_matrix/identity/api/v1/3pid/bind"
 | |
|                 ),
 | |
|                 {
 | |
|                     'sid': creds['sid'],
 | |
|                     'client_secret': client_secret,
 | |
|                     'mxid': mxid,
 | |
|                 }
 | |
|             )
 | |
|             logger.debug("bound threepid %r to %s", creds, mxid)
 | |
|         except CodeMessageException as e:
 | |
|             data = json.loads(e.msg)  # XXX WAT?
 | |
|         defer.returnValue(data)
 | |
| 
 | |
|     @defer.inlineCallbacks
 | |
|     def try_unbind_threepid(self, mxid, threepid):
 | |
|         """Removes a binding from an identity server
 | |
| 
 | |
|         Args:
 | |
|             mxid (str): Matrix user ID of binding to be removed
 | |
|             threepid (dict): Dict with medium & address of binding to be removed
 | |
| 
 | |
|         Raises:
 | |
|             SynapseError: If we failed to contact the identity server
 | |
| 
 | |
|         Returns:
 | |
|             Deferred[bool]: True on success, otherwise False if the identity
 | |
|             server doesn't support unbinding
 | |
|         """
 | |
|         logger.debug("unbinding threepid %r from %s", threepid, mxid)
 | |
|         if not self.trusted_id_servers:
 | |
|             logger.warn("Can't unbind threepid: no trusted ID servers set in config")
 | |
|             defer.returnValue(False)
 | |
| 
 | |
|         # We don't track what ID server we added 3pids on (perhaps we ought to)
 | |
|         # but we assume that any of the servers in the trusted list are in the
 | |
|         # same ID server federation, so we can pick any one of them to send the
 | |
|         # deletion request to.
 | |
|         id_server = next(iter(self.trusted_id_servers))
 | |
| 
 | |
|         url = "https://%s/_matrix/identity/api/v1/3pid/unbind" % (id_server,)
 | |
|         content = {
 | |
|             "mxid": mxid,
 | |
|             "threepid": threepid,
 | |
|         }
 | |
| 
 | |
|         # we abuse the federation http client to sign the request, but we have to send it
 | |
|         # using the normal http client since we don't want the SRV lookup and want normal
 | |
|         # 'browser-like' HTTPS.
 | |
|         auth_headers = self.federation_http_client.build_auth_headers(
 | |
|             destination=None,
 | |
|             method='POST',
 | |
|             url_bytes='/_matrix/identity/api/v1/3pid/unbind'.encode('ascii'),
 | |
|             content=content,
 | |
|             destination_is=id_server,
 | |
|         )
 | |
|         headers = {
 | |
|             b"Authorization": auth_headers,
 | |
|         }
 | |
| 
 | |
|         try:
 | |
|             yield self.http_client.post_json_get_json(
 | |
|                 url,
 | |
|                 content,
 | |
|                 headers,
 | |
|             )
 | |
|         except HttpResponseException as e:
 | |
|             if e.code in (400, 404, 501,):
 | |
|                 # The remote server probably doesn't support unbinding (yet)
 | |
|                 logger.warn("Received %d response while unbinding threepid", e.code)
 | |
|                 defer.returnValue(False)
 | |
|             else:
 | |
|                 logger.error("Failed to unbind threepid on identity server: %s", e)
 | |
|                 raise SynapseError(502, "Failed to contact identity server")
 | |
| 
 | |
|         defer.returnValue(True)
 | |
| 
 | |
|     @defer.inlineCallbacks
 | |
|     def requestEmailToken(self, id_server, email, client_secret, send_attempt, **kwargs):
 | |
|         if not self._should_trust_id_server(id_server):
 | |
|             raise SynapseError(
 | |
|                 400, "Untrusted ID server '%s'" % id_server,
 | |
|                 Codes.SERVER_NOT_TRUSTED
 | |
|             )
 | |
| 
 | |
|         params = {
 | |
|             'email': email,
 | |
|             'client_secret': client_secret,
 | |
|             'send_attempt': send_attempt,
 | |
|         }
 | |
|         params.update(kwargs)
 | |
| 
 | |
|         try:
 | |
|             data = yield self.http_client.post_json_get_json(
 | |
|                 "https://%s%s" % (
 | |
|                     id_server,
 | |
|                     "/_matrix/identity/api/v1/validate/email/requestToken"
 | |
|                 ),
 | |
|                 params
 | |
|             )
 | |
|             defer.returnValue(data)
 | |
|         except HttpResponseException as e:
 | |
|             logger.info("Proxied requestToken failed: %r", e)
 | |
|             raise e.to_synapse_error()
 | |
| 
 | |
|     @defer.inlineCallbacks
 | |
|     def requestMsisdnToken(
 | |
|             self, id_server, country, phone_number,
 | |
|             client_secret, send_attempt, **kwargs
 | |
|     ):
 | |
|         if not self._should_trust_id_server(id_server):
 | |
|             raise SynapseError(
 | |
|                 400, "Untrusted ID server '%s'" % id_server,
 | |
|                 Codes.SERVER_NOT_TRUSTED
 | |
|             )
 | |
| 
 | |
|         params = {
 | |
|             'country': country,
 | |
|             'phone_number': phone_number,
 | |
|             'client_secret': client_secret,
 | |
|             'send_attempt': send_attempt,
 | |
|         }
 | |
|         params.update(kwargs)
 | |
| 
 | |
|         try:
 | |
|             data = yield self.http_client.post_json_get_json(
 | |
|                 "https://%s%s" % (
 | |
|                     id_server,
 | |
|                     "/_matrix/identity/api/v1/validate/msisdn/requestToken"
 | |
|                 ),
 | |
|                 params
 | |
|             )
 | |
|             defer.returnValue(data)
 | |
|         except HttpResponseException as e:
 | |
|             logger.info("Proxied requestToken failed: %r", e)
 | |
|             raise e.to_synapse_error()
 |