mirror of
https://mau.dev/maunium/synapse.git
synced 2024-11-15 22:42:23 +01:00
allow uploading keys for cross-signing
This commit is contained in:
parent
d1c7c2a98a
commit
c659b9f94f
11 changed files with 621 additions and 12 deletions
|
@ -61,6 +61,7 @@ class Codes(object):
|
||||||
INCOMPATIBLE_ROOM_VERSION = "M_INCOMPATIBLE_ROOM_VERSION"
|
INCOMPATIBLE_ROOM_VERSION = "M_INCOMPATIBLE_ROOM_VERSION"
|
||||||
WRONG_ROOM_KEYS_VERSION = "M_WRONG_ROOM_KEYS_VERSION"
|
WRONG_ROOM_KEYS_VERSION = "M_WRONG_ROOM_KEYS_VERSION"
|
||||||
EXPIRED_ACCOUNT = "ORG_MATRIX_EXPIRED_ACCOUNT"
|
EXPIRED_ACCOUNT = "ORG_MATRIX_EXPIRED_ACCOUNT"
|
||||||
|
INVALID_SIGNATURE = "M_INVALID_SIGNATURE"
|
||||||
|
|
||||||
|
|
||||||
class CodeMessageException(RuntimeError):
|
class CodeMessageException(RuntimeError):
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2016 OpenMarket Ltd
|
# Copyright 2016 OpenMarket Ltd
|
||||||
|
# Copyright 2019 New Vector Ltd
|
||||||
|
# Copyright 2019 The Matrix.org Foundation C.I.C.
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -408,6 +410,21 @@ class DeviceHandler(DeviceWorkerHandler):
|
||||||
for host in hosts:
|
for host in hosts:
|
||||||
self.federation_sender.send_device_messages(host)
|
self.federation_sender.send_device_messages(host)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def notify_user_signature_update(self, from_user_id, user_ids):
|
||||||
|
"""Notify a user that they have made new signatures of other users.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
from_user_id (str): the user who made the signature
|
||||||
|
user_ids (list[str]): the users IDs that have new signatures
|
||||||
|
"""
|
||||||
|
|
||||||
|
position = yield self.store.add_user_signature_change_to_streams(
|
||||||
|
from_user_id, user_ids
|
||||||
|
)
|
||||||
|
|
||||||
|
self.notifier.on_new_event("device_list_key", position, users=[from_user_id])
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def on_federation_query_user_devices(self, user_id):
|
def on_federation_query_user_devices(self, user_id):
|
||||||
stream_id, devices = yield self.store.get_devices_with_keys_by_user(user_id)
|
stream_id, devices = yield self.store.get_devices_with_keys_by_user(user_id)
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2016 OpenMarket Ltd
|
# Copyright 2016 OpenMarket Ltd
|
||||||
# Copyright 2018 New Vector Ltd
|
# Copyright 2018-2019 New Vector Ltd
|
||||||
|
# Copyright 2019 The Matrix.org Foundation C.I.C.
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -19,12 +20,17 @@ import logging
|
||||||
from six import iteritems
|
from six import iteritems
|
||||||
|
|
||||||
from canonicaljson import encode_canonical_json, json
|
from canonicaljson import encode_canonical_json, json
|
||||||
|
from signedjson.sign import SignatureVerifyException, verify_signed_json
|
||||||
|
|
||||||
from twisted.internet import defer
|
from twisted.internet import defer
|
||||||
|
|
||||||
from synapse.api.errors import CodeMessageException, SynapseError
|
from synapse.api.errors import CodeMessageException, Codes, SynapseError
|
||||||
from synapse.logging.context import make_deferred_yieldable, run_in_background
|
from synapse.logging.context import make_deferred_yieldable, run_in_background
|
||||||
from synapse.types import UserID, get_domain_from_id
|
from synapse.types import (
|
||||||
|
UserID,
|
||||||
|
get_domain_from_id,
|
||||||
|
get_verify_key_from_cross_signing_key,
|
||||||
|
)
|
||||||
from synapse.util.retryutils import NotRetryingDestination
|
from synapse.util.retryutils import NotRetryingDestination
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
@ -46,7 +52,7 @@ class E2eKeysHandler(object):
|
||||||
)
|
)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def query_devices(self, query_body, timeout):
|
def query_devices(self, query_body, timeout, from_user_id):
|
||||||
""" Handle a device key query from a client
|
""" Handle a device key query from a client
|
||||||
|
|
||||||
{
|
{
|
||||||
|
@ -64,6 +70,11 @@ class E2eKeysHandler(object):
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Args:
|
||||||
|
from_user_id (str): the user making the query. This is used when
|
||||||
|
adding cross-signing signatures to limit what signatures users
|
||||||
|
can see.
|
||||||
"""
|
"""
|
||||||
device_keys_query = query_body.get("device_keys", {})
|
device_keys_query = query_body.get("device_keys", {})
|
||||||
|
|
||||||
|
@ -118,6 +129,11 @@ class E2eKeysHandler(object):
|
||||||
r = remote_queries_not_in_cache.setdefault(domain, {})
|
r = remote_queries_not_in_cache.setdefault(domain, {})
|
||||||
r[user_id] = remote_queries[user_id]
|
r[user_id] = remote_queries[user_id]
|
||||||
|
|
||||||
|
# Get cached cross-signing keys
|
||||||
|
cross_signing_keys = yield self.query_cross_signing_keys(
|
||||||
|
device_keys_query, from_user_id
|
||||||
|
)
|
||||||
|
|
||||||
# Now fetch any devices that we don't have in our cache
|
# Now fetch any devices that we don't have in our cache
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def do_remote_query(destination):
|
def do_remote_query(destination):
|
||||||
|
@ -131,6 +147,14 @@ class E2eKeysHandler(object):
|
||||||
if user_id in destination_query:
|
if user_id in destination_query:
|
||||||
results[user_id] = keys
|
results[user_id] = keys
|
||||||
|
|
||||||
|
for user_id, key in remote_result["master_keys"].items():
|
||||||
|
if user_id in destination_query:
|
||||||
|
cross_signing_keys["master"][user_id] = key
|
||||||
|
|
||||||
|
for user_id, key in remote_result["self_signing_keys"].items():
|
||||||
|
if user_id in destination_query:
|
||||||
|
cross_signing_keys["self_signing"][user_id] = key
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
failures[destination] = _exception_to_failure(e)
|
failures[destination] = _exception_to_failure(e)
|
||||||
|
|
||||||
|
@ -144,7 +168,73 @@ class E2eKeysHandler(object):
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
defer.returnValue({"device_keys": results, "failures": failures})
|
ret = {"device_keys": results, "failures": failures}
|
||||||
|
|
||||||
|
for key, value in iteritems(cross_signing_keys):
|
||||||
|
ret[key + "_keys"] = value
|
||||||
|
|
||||||
|
defer.returnValue(ret)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def query_cross_signing_keys(self, query, from_user_id):
|
||||||
|
"""Get cross-signing keys for users
|
||||||
|
|
||||||
|
Args:
|
||||||
|
query (Iterable[string]) an iterable of user IDs. A dict whose keys
|
||||||
|
are user IDs satisfies this, so the query format used for
|
||||||
|
query_devices can be used here.
|
||||||
|
from_user_id (str): the user making the query. This is used when
|
||||||
|
adding cross-signing signatures to limit what signatures users
|
||||||
|
can see.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
defer.Deferred[dict[str, dict[str, dict]]]: map from
|
||||||
|
(master|self_signing|user_signing) -> user_id -> key
|
||||||
|
"""
|
||||||
|
master_keys = {}
|
||||||
|
self_signing_keys = {}
|
||||||
|
user_signing_keys = {}
|
||||||
|
|
||||||
|
for user_id in query:
|
||||||
|
# XXX: consider changing the store functions to allow querying
|
||||||
|
# multiple users simultaneously.
|
||||||
|
try:
|
||||||
|
key = yield self.store.get_e2e_cross_signing_key(
|
||||||
|
user_id, "master", from_user_id
|
||||||
|
)
|
||||||
|
if key:
|
||||||
|
master_keys[user_id] = key
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Error getting master key: %s", e)
|
||||||
|
|
||||||
|
try:
|
||||||
|
key = yield self.store.get_e2e_cross_signing_key(
|
||||||
|
user_id, "self_signing", from_user_id
|
||||||
|
)
|
||||||
|
if key:
|
||||||
|
self_signing_keys[user_id] = key
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Error getting self-signing key: %s", e)
|
||||||
|
|
||||||
|
# users can see other users' master and self-signing keys, but can
|
||||||
|
# only see their own user-signing keys
|
||||||
|
if from_user_id == user_id:
|
||||||
|
try:
|
||||||
|
key = yield self.store.get_e2e_cross_signing_key(
|
||||||
|
user_id, "user_signing", from_user_id
|
||||||
|
)
|
||||||
|
if key:
|
||||||
|
user_signing_keys[user_id] = key
|
||||||
|
except Exception as e:
|
||||||
|
logger.info("Error getting user-signing key: %s", e)
|
||||||
|
|
||||||
|
defer.returnValue(
|
||||||
|
{
|
||||||
|
"master": master_keys,
|
||||||
|
"self_signing": self_signing_keys,
|
||||||
|
"user_signing": user_signing_keys,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def query_local_devices(self, query):
|
def query_local_devices(self, query):
|
||||||
|
@ -342,6 +432,104 @@ class E2eKeysHandler(object):
|
||||||
|
|
||||||
yield self.store.add_e2e_one_time_keys(user_id, device_id, time_now, new_keys)
|
yield self.store.add_e2e_one_time_keys(user_id, device_id, time_now, new_keys)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def upload_signing_keys_for_user(self, user_id, keys):
|
||||||
|
"""Upload signing keys for cross-signing
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (string): the user uploading the keys
|
||||||
|
keys (dict[string, dict]): the signing keys
|
||||||
|
"""
|
||||||
|
|
||||||
|
# if a master key is uploaded, then check it. Otherwise, load the
|
||||||
|
# stored master key, to check signatures on other keys
|
||||||
|
if "master_key" in keys:
|
||||||
|
master_key = keys["master_key"]
|
||||||
|
|
||||||
|
_check_cross_signing_key(master_key, user_id, "master")
|
||||||
|
else:
|
||||||
|
master_key = yield self.store.get_e2e_cross_signing_key(user_id, "master")
|
||||||
|
|
||||||
|
# if there is no master key, then we can't do anything, because all the
|
||||||
|
# other cross-signing keys need to be signed by the master key
|
||||||
|
if not master_key:
|
||||||
|
raise SynapseError(400, "No master key available", Codes.MISSING_PARAM)
|
||||||
|
|
||||||
|
master_key_id, master_verify_key = get_verify_key_from_cross_signing_key(
|
||||||
|
master_key
|
||||||
|
)
|
||||||
|
|
||||||
|
# for the other cross-signing keys, make sure that they have valid
|
||||||
|
# signatures from the master key
|
||||||
|
if "self_signing_key" in keys:
|
||||||
|
self_signing_key = keys["self_signing_key"]
|
||||||
|
|
||||||
|
_check_cross_signing_key(
|
||||||
|
self_signing_key, user_id, "self_signing", master_verify_key
|
||||||
|
)
|
||||||
|
|
||||||
|
if "user_signing_key" in keys:
|
||||||
|
user_signing_key = keys["user_signing_key"]
|
||||||
|
|
||||||
|
_check_cross_signing_key(
|
||||||
|
user_signing_key, user_id, "user_signing", master_verify_key
|
||||||
|
)
|
||||||
|
|
||||||
|
# if everything checks out, then store the keys and send notifications
|
||||||
|
deviceids = []
|
||||||
|
if "master_key" in keys:
|
||||||
|
yield self.store.set_e2e_cross_signing_key(user_id, "master", master_key)
|
||||||
|
deviceids.append(master_verify_key.version)
|
||||||
|
if "self_signing_key" in keys:
|
||||||
|
yield self.store.set_e2e_cross_signing_key(
|
||||||
|
user_id, "self_signing", self_signing_key
|
||||||
|
)
|
||||||
|
deviceids.append(
|
||||||
|
get_verify_key_from_cross_signing_key(self_signing_key)[1].version
|
||||||
|
)
|
||||||
|
if "user_signing_key" in keys:
|
||||||
|
yield self.store.set_e2e_cross_signing_key(
|
||||||
|
user_id, "user_signing", user_signing_key
|
||||||
|
)
|
||||||
|
# the signature stream matches the semantics that we want for
|
||||||
|
# user-signing key updates: only the user themselves is notified of
|
||||||
|
# their own user-signing key updates
|
||||||
|
yield self.device_handler.notify_user_signature_update(user_id, [user_id])
|
||||||
|
|
||||||
|
# master key and self-signing key updates match the semantics of device
|
||||||
|
# list updates: all users who share an encrypted room are notified
|
||||||
|
if len(deviceids):
|
||||||
|
yield self.device_handler.notify_device_update(user_id, deviceids)
|
||||||
|
|
||||||
|
defer.returnValue({})
|
||||||
|
|
||||||
|
|
||||||
|
def _check_cross_signing_key(key, user_id, key_type, signing_key=None):
|
||||||
|
"""Check a cross-signing key uploaded by a user. Performs some basic sanity
|
||||||
|
checking, and ensures that it is signed, if a signature is required.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
key (dict): the key data to verify
|
||||||
|
user_id (str): the user whose key is being checked
|
||||||
|
key_type (str): the type of key that the key should be
|
||||||
|
signing_key (VerifyKey): (optional) the signing key that the key should
|
||||||
|
be signed with. If omitted, signatures will not be checked.
|
||||||
|
"""
|
||||||
|
if (
|
||||||
|
key.get("user_id") != user_id
|
||||||
|
or key_type not in key.get("usage", [])
|
||||||
|
or len(key.get("keys", {})) != 1
|
||||||
|
):
|
||||||
|
raise SynapseError(400, ("Invalid %s key" % (key_type,)), Codes.INVALID_PARAM)
|
||||||
|
|
||||||
|
if signing_key:
|
||||||
|
try:
|
||||||
|
verify_signed_json(key, user_id, signing_key)
|
||||||
|
except SignatureVerifyException:
|
||||||
|
raise SynapseError(
|
||||||
|
400, ("Invalid signature on %s key" % key_type), Codes.INVALID_SIGNATURE
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def _exception_to_failure(e):
|
def _exception_to_failure(e):
|
||||||
if isinstance(e, CodeMessageException):
|
if isinstance(e, CodeMessageException):
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2015, 2016 OpenMarket Ltd
|
# Copyright 2015, 2016 OpenMarket Ltd
|
||||||
# Copyright 2018 New Vector Ltd
|
# Copyright 2018, 2019 New Vector Ltd
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -1116,6 +1116,11 @@ class SyncHandler(object):
|
||||||
# weren't in the previous sync *or* they left and rejoined.
|
# weren't in the previous sync *or* they left and rejoined.
|
||||||
users_that_have_changed.update(newly_joined_or_invited_users)
|
users_that_have_changed.update(newly_joined_or_invited_users)
|
||||||
|
|
||||||
|
user_signatures_changed = yield self.store.get_users_whose_signatures_changed(
|
||||||
|
user_id, since_token.device_list_key
|
||||||
|
)
|
||||||
|
users_that_have_changed.update(user_signatures_changed)
|
||||||
|
|
||||||
# Now find users that we no longer track
|
# Now find users that we no longer track
|
||||||
for room_id in newly_left_rooms:
|
for room_id in newly_left_rooms:
|
||||||
left_users = yield self.state.get_current_users_in_room(room_id)
|
left_users = yield self.state.get_current_users_in_room(room_id)
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2015, 2016 OpenMarket Ltd
|
# Copyright 2015, 2016 OpenMarket Ltd
|
||||||
|
# Copyright 2019 New Vector Ltd
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -26,7 +27,7 @@ from synapse.http.servlet import (
|
||||||
)
|
)
|
||||||
from synapse.types import StreamToken
|
from synapse.types import StreamToken
|
||||||
|
|
||||||
from ._base import client_patterns
|
from ._base import client_patterns, interactive_auth_handler
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
@ -145,10 +146,11 @@ class KeyQueryServlet(RestServlet):
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def on_POST(self, request):
|
def on_POST(self, request):
|
||||||
yield self.auth.get_user_by_req(request, allow_guest=True)
|
requester = yield self.auth.get_user_by_req(request, allow_guest=True)
|
||||||
|
user_id = requester.user.to_string()
|
||||||
timeout = parse_integer(request, "timeout", 10 * 1000)
|
timeout = parse_integer(request, "timeout", 10 * 1000)
|
||||||
body = parse_json_object_from_request(request)
|
body = parse_json_object_from_request(request)
|
||||||
result = yield self.e2e_keys_handler.query_devices(body, timeout)
|
result = yield self.e2e_keys_handler.query_devices(body, timeout, user_id)
|
||||||
defer.returnValue((200, result))
|
defer.returnValue((200, result))
|
||||||
|
|
||||||
|
|
||||||
|
@ -227,8 +229,46 @@ class OneTimeKeyServlet(RestServlet):
|
||||||
defer.returnValue((200, result))
|
defer.returnValue((200, result))
|
||||||
|
|
||||||
|
|
||||||
|
class SigningKeyUploadServlet(RestServlet):
|
||||||
|
"""
|
||||||
|
POST /keys/device_signing/upload HTTP/1.1
|
||||||
|
Content-Type: application/json
|
||||||
|
|
||||||
|
{
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
|
||||||
|
PATTERNS = client_patterns("/keys/device_signing/upload$", releases=())
|
||||||
|
|
||||||
|
def __init__(self, hs):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
hs (synapse.server.HomeServer): server
|
||||||
|
"""
|
||||||
|
super(SigningKeyUploadServlet, self).__init__()
|
||||||
|
self.hs = hs
|
||||||
|
self.auth = hs.get_auth()
|
||||||
|
self.e2e_keys_handler = hs.get_e2e_keys_handler()
|
||||||
|
self.auth_handler = hs.get_auth_handler()
|
||||||
|
|
||||||
|
@interactive_auth_handler
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def on_POST(self, request):
|
||||||
|
requester = yield self.auth.get_user_by_req(request)
|
||||||
|
user_id = requester.user.to_string()
|
||||||
|
body = parse_json_object_from_request(request)
|
||||||
|
|
||||||
|
yield self.auth_handler.validate_user_via_ui_auth(
|
||||||
|
requester, body, self.hs.get_ip_from_request(request)
|
||||||
|
)
|
||||||
|
|
||||||
|
result = yield self.e2e_keys_handler.upload_signing_keys_for_user(user_id, body)
|
||||||
|
defer.returnValue((200, result))
|
||||||
|
|
||||||
|
|
||||||
def register_servlets(hs, http_server):
|
def register_servlets(hs, http_server):
|
||||||
KeyUploadServlet(hs).register(http_server)
|
KeyUploadServlet(hs).register(http_server)
|
||||||
KeyQueryServlet(hs).register(http_server)
|
KeyQueryServlet(hs).register(http_server)
|
||||||
KeyChangesServlet(hs).register(http_server)
|
KeyChangesServlet(hs).register(http_server)
|
||||||
OneTimeKeyServlet(hs).register(http_server)
|
OneTimeKeyServlet(hs).register(http_server)
|
||||||
|
SigningKeyUploadServlet(hs).register(http_server)
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2014-2016 OpenMarket Ltd
|
# Copyright 2014-2016 OpenMarket Ltd
|
||||||
# Copyright 2018 New Vector Ltd
|
# Copyright 2018,2019 New Vector Ltd
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -207,6 +207,9 @@ class DataStore(
|
||||||
self._device_list_stream_cache = StreamChangeCache(
|
self._device_list_stream_cache = StreamChangeCache(
|
||||||
"DeviceListStreamChangeCache", device_list_max
|
"DeviceListStreamChangeCache", device_list_max
|
||||||
)
|
)
|
||||||
|
self._user_signature_stream_cache = StreamChangeCache(
|
||||||
|
"UserSignatureStreamChangeCache", device_list_max
|
||||||
|
)
|
||||||
self._device_list_federation_stream_cache = StreamChangeCache(
|
self._device_list_federation_stream_cache = StreamChangeCache(
|
||||||
"DeviceListFederationStreamChangeCache", device_list_max
|
"DeviceListFederationStreamChangeCache", device_list_max
|
||||||
)
|
)
|
||||||
|
|
|
@ -302,6 +302,41 @@ class DeviceWorkerStore(SQLBaseStore):
|
||||||
"""
|
"""
|
||||||
txn.execute(sql, (destination, stream_id))
|
txn.execute(sql, (destination, stream_id))
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def add_user_signature_change_to_streams(self, from_user_id, user_ids):
|
||||||
|
"""Persist that a user has made new signatures
|
||||||
|
|
||||||
|
Args:
|
||||||
|
from_user_id (str): the user who made the signatures
|
||||||
|
user_ids (list[str]): the users who were signed
|
||||||
|
"""
|
||||||
|
|
||||||
|
with self._device_list_id_gen.get_next() as stream_id:
|
||||||
|
yield self.runInteraction(
|
||||||
|
"add_user_sig_change_to_streams",
|
||||||
|
self._add_user_signature_change_txn,
|
||||||
|
from_user_id,
|
||||||
|
user_ids,
|
||||||
|
stream_id,
|
||||||
|
)
|
||||||
|
defer.returnValue(stream_id)
|
||||||
|
|
||||||
|
def _add_user_signature_change_txn(self, txn, from_user_id, user_ids, stream_id):
|
||||||
|
txn.call_after(
|
||||||
|
self._user_signature_stream_cache.entity_has_changed,
|
||||||
|
from_user_id,
|
||||||
|
stream_id,
|
||||||
|
)
|
||||||
|
self._simple_insert_txn(
|
||||||
|
txn,
|
||||||
|
"user_signature_stream",
|
||||||
|
values={
|
||||||
|
"stream_id": stream_id,
|
||||||
|
"from_user_id": from_user_id,
|
||||||
|
"user_ids": json.dumps(user_ids),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
def get_device_stream_token(self):
|
def get_device_stream_token(self):
|
||||||
return self._device_list_id_gen.get_current_token()
|
return self._device_list_id_gen.get_current_token()
|
||||||
|
|
||||||
|
@ -440,6 +475,28 @@ class DeviceWorkerStore(SQLBaseStore):
|
||||||
"get_users_whose_devices_changed", _get_users_whose_devices_changed_txn
|
"get_users_whose_devices_changed", _get_users_whose_devices_changed_txn
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def get_users_whose_signatures_changed(self, user_id, from_key):
|
||||||
|
"""Get the users who have new cross-signing signatures made by `user_id` since
|
||||||
|
`from_key`.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str): the user who made the signatures
|
||||||
|
from_key (str): The device lists stream token
|
||||||
|
"""
|
||||||
|
from_key = int(from_key)
|
||||||
|
if self._user_signature_stream_cache.has_entity_changed(user_id, from_key):
|
||||||
|
sql = """
|
||||||
|
SELECT DISTINCT user_ids FROM user_signature_stream
|
||||||
|
WHERE from_user_id = ? AND stream_id > ?
|
||||||
|
"""
|
||||||
|
rows = yield self._execute(
|
||||||
|
"get_users_whose_signatures_changed", None, sql, user_id, from_key
|
||||||
|
)
|
||||||
|
defer.returnValue(set(user for row in rows for user in json.loads(row[0])))
|
||||||
|
else:
|
||||||
|
defer.returnValue(set())
|
||||||
|
|
||||||
def get_all_device_list_changes_for_remotes(self, from_key, to_key):
|
def get_all_device_list_changes_for_remotes(self, from_key, to_key):
|
||||||
"""Return a list of `(stream_id, user_id, destination)` which is the
|
"""Return a list of `(stream_id, user_id, destination)` which is the
|
||||||
combined list of changes to devices, and which destinations need to be
|
combined list of changes to devices, and which destinations need to be
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2015, 2016 OpenMarket Ltd
|
# Copyright 2015, 2016 OpenMarket Ltd
|
||||||
|
# Copyright 2019 New Vector Ltd
|
||||||
|
# Copyright 2019 The Matrix.org Foundation C.I.C.
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -12,9 +14,11 @@
|
||||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
|
import time
|
||||||
|
|
||||||
from six import iteritems
|
from six import iteritems
|
||||||
|
|
||||||
from canonicaljson import encode_canonical_json
|
from canonicaljson import encode_canonical_json, json
|
||||||
|
|
||||||
from twisted.internet import defer
|
from twisted.internet import defer
|
||||||
|
|
||||||
|
@ -85,11 +89,12 @@ class EndToEndKeyWorkerStore(SQLBaseStore):
|
||||||
" k.key_json"
|
" k.key_json"
|
||||||
" FROM devices d"
|
" FROM devices d"
|
||||||
" %s JOIN e2e_device_keys_json k USING (user_id, device_id)"
|
" %s JOIN e2e_device_keys_json k USING (user_id, device_id)"
|
||||||
" WHERE %s"
|
" WHERE (%s) AND NOT COALESCE(d.hidden, ?)"
|
||||||
) % (
|
) % (
|
||||||
"LEFT" if include_all_devices else "INNER",
|
"LEFT" if include_all_devices else "INNER",
|
||||||
" OR ".join("(" + q + ")" for q in query_clauses),
|
" OR ".join("(" + q + ")" for q in query_clauses),
|
||||||
)
|
)
|
||||||
|
query_params.append(False)
|
||||||
|
|
||||||
txn.execute(sql, query_params)
|
txn.execute(sql, query_params)
|
||||||
rows = self.cursor_to_dict(txn)
|
rows = self.cursor_to_dict(txn)
|
||||||
|
@ -281,3 +286,168 @@ class EndToEndKeyStore(EndToEndKeyWorkerStore, SQLBaseStore):
|
||||||
return self.runInteraction(
|
return self.runInteraction(
|
||||||
"delete_e2e_keys_by_device", delete_e2e_keys_by_device_txn
|
"delete_e2e_keys_by_device", delete_e2e_keys_by_device_txn
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def _set_e2e_cross_signing_key_txn(self, txn, user_id, key_type, key):
|
||||||
|
"""Set a user's cross-signing key.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
txn (twisted.enterprise.adbapi.Connection): db connection
|
||||||
|
user_id (str): the user to set the signing key for
|
||||||
|
key_type (str): the type of key that is being set: either 'master'
|
||||||
|
for a master key, 'self_signing' for a self-signing key, or
|
||||||
|
'user_signing' for a user-signing key
|
||||||
|
key (dict): the key data
|
||||||
|
"""
|
||||||
|
# the cross-signing keys need to occupy the same namespace as devices,
|
||||||
|
# since signatures are identified by device ID. So add an entry to the
|
||||||
|
# device table to make sure that we don't have a collision with device
|
||||||
|
# IDs
|
||||||
|
|
||||||
|
# the 'key' dict will look something like:
|
||||||
|
# {
|
||||||
|
# "user_id": "@alice:example.com",
|
||||||
|
# "usage": ["self_signing"],
|
||||||
|
# "keys": {
|
||||||
|
# "ed25519:base64+self+signing+public+key": "base64+self+signing+public+key",
|
||||||
|
# },
|
||||||
|
# "signatures": {
|
||||||
|
# "@alice:example.com": {
|
||||||
|
# "ed25519:base64+master+public+key": "base64+signature"
|
||||||
|
# }
|
||||||
|
# }
|
||||||
|
# }
|
||||||
|
# The "keys" property must only have one entry, which will be the public
|
||||||
|
# key, so we just grab the first value in there
|
||||||
|
pubkey = next(iter(key["keys"].values()))
|
||||||
|
self._simple_insert(
|
||||||
|
"devices",
|
||||||
|
values={
|
||||||
|
"user_id": user_id,
|
||||||
|
"device_id": pubkey,
|
||||||
|
"display_name": key_type + " signing key",
|
||||||
|
"hidden": True,
|
||||||
|
},
|
||||||
|
desc="store_master_key_device",
|
||||||
|
)
|
||||||
|
|
||||||
|
# and finally, store the key itself
|
||||||
|
self._simple_insert(
|
||||||
|
"e2e_cross_signing_keys",
|
||||||
|
values={
|
||||||
|
"user_id": user_id,
|
||||||
|
"keytype": key_type,
|
||||||
|
"keydata": json.dumps(key),
|
||||||
|
"added_ts": time.time() * 1000,
|
||||||
|
},
|
||||||
|
desc="store_master_key",
|
||||||
|
)
|
||||||
|
|
||||||
|
def set_e2e_cross_signing_key(self, user_id, key_type, key):
|
||||||
|
"""Set a user's cross-signing key.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str): the user to set the user-signing key for
|
||||||
|
key_type (str): the type of cross-signing key to set
|
||||||
|
key (dict): the key data
|
||||||
|
"""
|
||||||
|
return self.runInteraction(
|
||||||
|
"add_e2e_cross_signing_key",
|
||||||
|
self._set_e2e_cross_signing_key_txn,
|
||||||
|
user_id,
|
||||||
|
key_type,
|
||||||
|
key,
|
||||||
|
)
|
||||||
|
|
||||||
|
def _get_e2e_cross_signing_key_txn(self, txn, user_id, key_type, from_user_id=None):
|
||||||
|
"""Returns a user's cross-signing key.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
txn (twisted.enterprise.adbapi.Connection): db connection
|
||||||
|
user_id (str): the user whose key is being requested
|
||||||
|
key_type (str): the type of key that is being set: either 'master'
|
||||||
|
for a master key, 'self_signing' for a self-signing key, or
|
||||||
|
'user_signing' for a user-signing key
|
||||||
|
from_user_id (str): if specified, signatures made by this user on
|
||||||
|
the key will be included in the result
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
dict of the key data
|
||||||
|
"""
|
||||||
|
sql = (
|
||||||
|
"SELECT keydata "
|
||||||
|
" FROM e2e_cross_signing_keys "
|
||||||
|
" WHERE user_id = ? AND keytype = ? ORDER BY added_ts DESC LIMIT 1"
|
||||||
|
)
|
||||||
|
txn.execute(sql, (user_id, key_type))
|
||||||
|
row = txn.fetchone()
|
||||||
|
if not row:
|
||||||
|
return None
|
||||||
|
key = json.loads(row[0])
|
||||||
|
|
||||||
|
device_id = None
|
||||||
|
for k in key["keys"].values():
|
||||||
|
device_id = k
|
||||||
|
|
||||||
|
if from_user_id is not None:
|
||||||
|
sql = (
|
||||||
|
"SELECT key_id, signature "
|
||||||
|
" FROM e2e_cross_signing_signatures "
|
||||||
|
" WHERE user_id = ? "
|
||||||
|
" AND target_user_id = ? "
|
||||||
|
" AND target_device_id = ? "
|
||||||
|
)
|
||||||
|
txn.execute(sql, (from_user_id, user_id, device_id))
|
||||||
|
row = txn.fetchone()
|
||||||
|
if row:
|
||||||
|
key.setdefault("signatures", {}).setdefault(from_user_id, {})[
|
||||||
|
row[0]
|
||||||
|
] = row[1]
|
||||||
|
|
||||||
|
return key
|
||||||
|
|
||||||
|
def get_e2e_cross_signing_key(self, user_id, key_type, from_user_id=None):
|
||||||
|
"""Returns a user's cross-signing key.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str): the user whose self-signing key is being requested
|
||||||
|
key_type (str): the type of cross-signing key to get
|
||||||
|
from_user_id (str): if specified, signatures made by this user on
|
||||||
|
the self-signing key will be included in the result
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
dict of the key data
|
||||||
|
"""
|
||||||
|
return self.runInteraction(
|
||||||
|
"get_e2e_cross_signing_key",
|
||||||
|
self._get_e2e_cross_signing_key_txn,
|
||||||
|
user_id,
|
||||||
|
key_type,
|
||||||
|
from_user_id,
|
||||||
|
)
|
||||||
|
|
||||||
|
def store_e2e_cross_signing_signatures(self, user_id, signatures):
|
||||||
|
"""Stores cross-signing signatures.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str): the user who made the signatures
|
||||||
|
signatures (iterable[(str, str, str, str)]): signatures to add - each
|
||||||
|
a tuple of (key_id, target_user_id, target_device_id, signature),
|
||||||
|
where key_id is the ID of the key (including the signature
|
||||||
|
algorithm) that made the signature, target_user_id and
|
||||||
|
target_device_id indicate the device being signed, and signature
|
||||||
|
is the signature of the device
|
||||||
|
"""
|
||||||
|
return self._simple_insert_many(
|
||||||
|
"e2e_cross_signing_signatures",
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"user_id": user_id,
|
||||||
|
"key_id": key_id,
|
||||||
|
"target_user_id": target_user_id,
|
||||||
|
"target_device_id": target_device_id,
|
||||||
|
"signature": signature,
|
||||||
|
}
|
||||||
|
for (key_id, target_user_id, target_device_id, signature) in signatures
|
||||||
|
],
|
||||||
|
"add_e2e_signing_key",
|
||||||
|
)
|
||||||
|
|
|
@ -13,6 +13,47 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
-- cross-signing keys
|
||||||
|
CREATE TABLE IF NOT EXISTS e2e_cross_signing_keys (
|
||||||
|
user_id TEXT NOT NULL,
|
||||||
|
-- the type of cross-signing key (master, user_signing, or self_signing)
|
||||||
|
keytype TEXT NOT NULL,
|
||||||
|
-- the full key information, as a json-encoded dict
|
||||||
|
keydata TEXT NOT NULL,
|
||||||
|
-- time that the key was added
|
||||||
|
added_ts BIGINT NOT NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX e2e_cross_signing_keys_idx ON e2e_cross_signing_keys(user_id, keytype, added_ts);
|
||||||
|
|
||||||
|
-- cross-signing signatures
|
||||||
|
CREATE TABLE IF NOT EXISTS e2e_cross_signing_signatures (
|
||||||
|
-- user who did the signing
|
||||||
|
user_id TEXT NOT NULL,
|
||||||
|
-- key used to sign
|
||||||
|
key_id TEXT NOT NULL,
|
||||||
|
-- user who was signed
|
||||||
|
target_user_id TEXT NOT NULL,
|
||||||
|
-- device/key that was signed
|
||||||
|
target_device_id TEXT NOT NULL,
|
||||||
|
-- the actual signature
|
||||||
|
signature TEXT NOT NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX e2e_cross_signing_signatures_idx ON e2e_cross_signing_signatures(user_id, target_user_id, target_device_id);
|
||||||
|
|
||||||
|
-- stream of user signature updates
|
||||||
|
CREATE TABLE IF NOT EXISTS user_signature_stream (
|
||||||
|
-- uses the same stream ID as device list stream
|
||||||
|
stream_id BIGINT NOT NULL,
|
||||||
|
-- user who did the signing
|
||||||
|
from_user_id TEXT NOT NULL,
|
||||||
|
-- list of users who were signed, as a JSON array
|
||||||
|
user_ids TEXT NOT NULL
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE UNIQUE INDEX user_signature_stream_idx ON user_signature_stream(stream_id);
|
||||||
|
|
||||||
-- device list needs to know which ones are "real" devices, and which ones are
|
-- device list needs to know which ones are "real" devices, and which ones are
|
||||||
-- just used to avoid collisions
|
-- just used to avoid collisions
|
||||||
ALTER TABLE devices ADD COLUMN hidden BOOLEAN NULLABLE;
|
ALTER TABLE devices ADD COLUMN hidden BOOLEAN NULLABLE;
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2014-2016 OpenMarket Ltd
|
# Copyright 2014-2016 OpenMarket Ltd
|
||||||
|
# Copyright 2019 The Matrix.org Foundation C.I.C.
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -17,6 +18,8 @@ import string
|
||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
|
|
||||||
import attr
|
import attr
|
||||||
|
from signedjson.key import decode_verify_key_bytes
|
||||||
|
from unpaddedbase64 import decode_base64
|
||||||
|
|
||||||
from synapse.api.errors import SynapseError
|
from synapse.api.errors import SynapseError
|
||||||
|
|
||||||
|
@ -475,3 +478,24 @@ class ReadReceipt(object):
|
||||||
user_id = attr.ib()
|
user_id = attr.ib()
|
||||||
event_ids = attr.ib()
|
event_ids = attr.ib()
|
||||||
data = attr.ib()
|
data = attr.ib()
|
||||||
|
|
||||||
|
|
||||||
|
def get_verify_key_from_cross_signing_key(key_info):
|
||||||
|
"""Get the key ID and signedjson verify key from a cross-signing key dict
|
||||||
|
|
||||||
|
Args:
|
||||||
|
key_info (dict): a cross-signing key dict, which must have a "keys"
|
||||||
|
property that has exactly one item in it
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
(str, VerifyKey): the key ID and verify key for the cross-signing key
|
||||||
|
"""
|
||||||
|
# make sure that exactly one key is provided
|
||||||
|
if "keys" not in key_info:
|
||||||
|
raise SynapseError(400, "Invalid key")
|
||||||
|
keys = key_info["keys"]
|
||||||
|
if len(keys) != 1:
|
||||||
|
raise SynapseError(400, "Invalid key")
|
||||||
|
# and return that one key
|
||||||
|
for key_id, key_data in keys.items():
|
||||||
|
return (key_id, decode_verify_key_bytes(key_id, decode_base64(key_data)))
|
||||||
|
|
|
@ -1,5 +1,7 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
# Copyright 2016 OpenMarket Ltd
|
# Copyright 2016 OpenMarket Ltd
|
||||||
|
# Copyright 2019 New Vector Ltd
|
||||||
|
# Copyright 2019 The Matrix.org Foundation C.I.C.
|
||||||
#
|
#
|
||||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
# you may not use this file except in compliance with the License.
|
# you may not use this file except in compliance with the License.
|
||||||
|
@ -145,3 +147,64 @@ class E2eKeysHandlerTestCase(unittest.TestCase):
|
||||||
"one_time_keys": {local_user: {device_id: {"alg1:k1": "key1"}}},
|
"one_time_keys": {local_user: {device_id: {"alg1:k1": "key1"}}},
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def test_replace_master_key(self):
|
||||||
|
"""uploading a new signing key should make the old signing key unavailable"""
|
||||||
|
local_user = "@boris:" + self.hs.hostname
|
||||||
|
keys1 = {
|
||||||
|
"master_key": {
|
||||||
|
# private key: 2lonYOM6xYKdEsO+6KrC766xBcHnYnim1x/4LFGF8B0
|
||||||
|
"user_id": local_user,
|
||||||
|
"usage": ["master"],
|
||||||
|
"keys": {
|
||||||
|
"ed25519:nqOvzeuGWT/sRx3h7+MHoInYj3Uk2LD/unI9kDYcHwk": "nqOvzeuGWT/sRx3h7+MHoInYj3Uk2LD/unI9kDYcHwk"
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
yield self.handler.upload_signing_keys_for_user(local_user, keys1)
|
||||||
|
|
||||||
|
keys2 = {
|
||||||
|
"master_key": {
|
||||||
|
# private key: 4TL4AjRYwDVwD3pqQzcor+ez/euOB1/q78aTJ+czDNs
|
||||||
|
"user_id": local_user,
|
||||||
|
"usage": ["master"],
|
||||||
|
"keys": {
|
||||||
|
"ed25519:Hq6gL+utB4ET+UvD5ci0kgAwsX6qP/zvf8v6OInU5iw": "Hq6gL+utB4ET+UvD5ci0kgAwsX6qP/zvf8v6OInU5iw"
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
yield self.handler.upload_signing_keys_for_user(local_user, keys2)
|
||||||
|
|
||||||
|
devices = yield self.handler.query_devices({"device_keys": {local_user: []}}, 0, local_user)
|
||||||
|
self.assertDictEqual(devices["master_keys"], {local_user: keys2["master_key"]})
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def test_self_signing_key_doesnt_show_up_as_device(self):
|
||||||
|
"""signing keys should be hidden when fetching a user's devices"""
|
||||||
|
local_user = "@boris:" + self.hs.hostname
|
||||||
|
keys1 = {
|
||||||
|
"master_key": {
|
||||||
|
# private key: 2lonYOM6xYKdEsO+6KrC766xBcHnYnim1x/4LFGF8B0
|
||||||
|
"user_id": local_user,
|
||||||
|
"usage": ["master"],
|
||||||
|
"keys": {
|
||||||
|
"ed25519:nqOvzeuGWT/sRx3h7+MHoInYj3Uk2LD/unI9kDYcHwk": "nqOvzeuGWT/sRx3h7+MHoInYj3Uk2LD/unI9kDYcHwk"
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
yield self.handler.upload_signing_keys_for_user(local_user, keys1)
|
||||||
|
|
||||||
|
res = None
|
||||||
|
try:
|
||||||
|
yield self.hs.get_device_handler().check_device_registered(
|
||||||
|
user_id=local_user,
|
||||||
|
device_id="nqOvzeuGWT/sRx3h7+MHoInYj3Uk2LD/unI9kDYcHwk",
|
||||||
|
initial_device_display_name="new display name",
|
||||||
|
)
|
||||||
|
except errors.SynapseError as e:
|
||||||
|
res = e.code
|
||||||
|
self.assertEqual(res, 400)
|
||||||
|
|
||||||
|
res = yield self.handler.query_local_devices({local_user: None})
|
||||||
|
self.assertDictEqual(res, {local_user: {}})
|
||||||
|
|
Loading…
Reference in a new issue