123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202 |
- # -*- coding: utf-8 -*-
- # Copyright 2014 OpenMarket 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.
- from __future__ import absolute_import
- import collections
- import logging
- import math
- import signedjson.sign
- from sydent.db.invite_tokens import JoinTokenStore
- from sydent.db.threepid_associations import LocalAssociationStore
- from sydent.util import time_msec
- from sydent.util.hash import sha256_and_url_safe_base64
- from sydent.db.hashing_metadata import HashingMetadataStore
- from sydent.threepid.signer import Signer
- from sydent.http.httpclient import FederationHttpClient
- from sydent.threepid import ThreepidAssociation
- from twisted.internet import defer
- logger = logging.getLogger(__name__)
- class ThreepidBinder:
- # the lifetime of a 3pid association
- THREEPID_ASSOCIATION_LIFETIME_MS = 100 * 365 * 24 * 60 * 60 * 1000
- def __init__(self, sydent):
- self.sydent = sydent
- self.hashing_store = HashingMetadataStore(sydent)
- def addBinding(self, medium, address, mxid):
- """
- Binds the given 3pid to the given mxid.
- It's assumed that we have somehow validated that the given user owns
- the given 3pid
- :param medium: The medium of the 3PID to bind.
- :type medium: unicode
- :param address: The address of the 3PID to bind.
- :type address: unicode
- :param mxid: The MXID to bind the 3PID to.
- :type mxid: unicode
- :return: The signed association.
- :rtype: dict[str, any]
- """
- localAssocStore = LocalAssociationStore(self.sydent)
- # Fill out the association details
- createdAt = time_msec()
- expires = createdAt + ThreepidBinder.THREEPID_ASSOCIATION_LIFETIME_MS
- # Hash the medium + address and store that hash for the purposes of
- # later lookups
- str_to_hash = u' '.join(
- [address, medium, self.hashing_store.get_lookup_pepper()],
- )
- lookup_hash = sha256_and_url_safe_base64(str_to_hash)
- assoc = ThreepidAssociation(
- medium, address, lookup_hash, mxid, createdAt, createdAt, expires,
- )
- localAssocStore.addOrUpdateAssociation(assoc)
- self.sydent.pusher.doLocalPush()
- joinTokenStore = JoinTokenStore(self.sydent)
- pendingJoinTokens = joinTokenStore.getTokens(medium, address)
- invites = []
- for token in pendingJoinTokens:
- token["mxid"] = mxid
- token["signed"] = {
- "mxid": mxid,
- "token": token["token"],
- }
- token["signed"] = signedjson.sign.sign_json(token["signed"], self.sydent.server_name, self.sydent.keyring.ed25519)
- invites.append(token)
- if invites:
- assoc.extra_fields["invites"] = invites
- joinTokenStore.markTokensAsSent(medium, address)
- signer = Signer(self.sydent)
- sgassoc = signer.signedThreePidAssociation(assoc)
- self._notify(sgassoc, 0)
- return sgassoc
- def removeBinding(self, threepid, mxid):
- """
- Removes the binding between a given 3PID and a given MXID.
- :param threepid: The 3PID of the binding to remove.
- :type threepid: dict[unicode, unicode]
- :param mxid: The MXID of the binding to remove.
- :type mxid: unicode
- """
- localAssocStore = LocalAssociationStore(self.sydent)
- localAssocStore.removeAssociation(threepid, mxid)
- self.sydent.pusher.doLocalPush()
- @defer.inlineCallbacks
- def _notify(self, assoc, attempt):
- """
- Sends data about a new association (and, if necessary, the associated invites)
- to the associated MXID's homeserver.
- :param assoc: The association to send down to the homeserver.
- :type assoc: dict[str, any]
- :param attempt: The number of previous attempts to send this association.
- :type attempt: int
- """
- mxid = assoc["mxid"]
- mxid_parts = mxid.split(":", 1)
- if len(mxid_parts) != 2:
- logger.error(
- "Can't notify on bind for unparseable mxid %s. Not retrying.",
- assoc["mxid"],
- )
- return
- post_url = "matrix://%s/_matrix/federation/v1/3pid/onbind" % (
- mxid_parts[1],
- )
- logger.info("Making bind callback to: %s", post_url)
- # Make a POST to the chosen Synapse server
- http_client = FederationHttpClient(self.sydent)
- try:
- response = yield http_client.post_json_get_nothing(post_url, assoc, {})
- except Exception as e:
- self._notifyErrback(assoc, attempt, e)
- return
- # If the request failed, try again with exponential backoff
- if response.code != 200:
- self._notifyErrback(
- assoc, attempt, "Non-OK error code received (%d)" % response.code
- )
- else:
- logger.info("Successfully notified on bind for %s" % (mxid,))
- # Skip the deletion step if instructed so by the config.
- if not self.sydent.delete_tokens_on_bind:
- return
- # Only remove sent tokens when they've been successfully sent.
- try:
- joinTokenStore = JoinTokenStore(self.sydent)
- joinTokenStore.deleteTokens(assoc["medium"], assoc["address"])
- logger.info(
- "Successfully deleted invite for %s from the store",
- assoc["address"],
- )
- except Exception as e:
- logger.exception(
- "Couldn't remove invite for %s from the store",
- assoc["address"],
- )
- def _notifyErrback(self, assoc, attempt, error):
- """
- Handles errors when trying to send an association down to a homeserver by
- logging the error and scheduling a new attempt.
- :param assoc: The association to send down to the homeserver.
- :type assoc: dict[str, any]
- :param attempt: The number of previous attempts to send this association.
- :type attempt: int
- :param error: The error that was raised when trying to send the association.
- :type error: Exception
- """
- logger.warning(
- "Error notifying on bind for %s: %s - rescheduling", assoc["mxid"], error
- )
- self.sydent.reactor.callLater(
- math.pow(2, attempt), self._notify, assoc, attempt + 1
- )
- # The below is lovingly ripped off of synapse/http/endpoint.py
- _Server = collections.namedtuple("_Server", "priority weight host port")
|