make black happy

This commit is contained in:
Hubert Chathi 2019-07-22 13:04:55 -04:00
parent a1aaf3eea6
commit cfdb84422d
2 changed files with 39 additions and 27 deletions

View File

@ -56,7 +56,7 @@ class E2eKeysHandler(object):
federation_registry = hs.get_federation_registry() federation_registry = hs.get_federation_registry()
federation_registry.register_edu_handler( federation_registry.register_edu_handler(
"m.signing_key_update", self._edu_updater.incoming_signing_key_update, "m.signing_key_update", self._edu_updater.incoming_signing_key_update
) )
# doesn't really work as part of the generic query API, because the # doesn't really work as part of the generic query API, because the
# query request requires an object POST, but we abuse the # query request requires an object POST, but we abuse the
@ -1147,15 +1147,17 @@ class SigningKeyEduUpdater(object):
yield self.store.set_e2e_cross_signing_key( yield self.store.set_e2e_cross_signing_key(
user_id, "master", master_key user_id, "master", master_key
) )
device_id = \ device_id = get_verify_key_from_cross_signing_key(master_key)[
get_verify_key_from_cross_signing_key(master_key)[1].version 1
].version
device_ids.append(device_id) device_ids.append(device_id)
if self_signing_key: if self_signing_key:
yield self.store.set_e2e_cross_signing_key( yield self.store.set_e2e_cross_signing_key(
user_id, "self_signing", self_signing_key user_id, "self_signing", self_signing_key
) )
device_id = \ device_id = get_verify_key_from_cross_signing_key(self_signing_key)[
get_verify_key_from_cross_signing_key(self_signing_key)[1].version 1
].version
device_ids.append(device_id) device_ids.append(device_id)
yield device_handler.notify_device_update(user_id, device_ids) yield device_handler.notify_device_update(user_id, device_ids)

View File

@ -37,6 +37,7 @@ from synapse.storage._base import (
make_in_list_sql_clause, make_in_list_sql_clause,
) )
from synapse.storage.background_updates import BackgroundUpdateStore from synapse.storage.background_updates import BackgroundUpdateStore
from synapse.types import get_verify_key_from_cross_signing_key
from synapse.util import batch_iter from synapse.util import batch_iter
from synapse.util.caches.descriptors import cached, cachedInlineCallbacks, cachedList from synapse.util.caches.descriptors import cached, cachedInlineCallbacks, cachedList
@ -142,17 +143,19 @@ class DeviceWorkerStore(SQLBaseStore):
) )
master_key_by_user[user] = { master_key_by_user[user] = {
"key_info": cross_signing_key, "key_info": cross_signing_key,
"pubkey": verify_key.version "pubkey": verify_key.version,
} }
cross_signing_key = yield self.get_e2e_cross_signing_key(user, "self_signing") cross_signing_key = yield self.get_e2e_cross_signing_key(
user, "self_signing"
)
if cross_signing_key: if cross_signing_key:
key_id, verify_key = get_verify_key_from_cross_signing_key( key_id, verify_key = get_verify_key_from_cross_signing_key(
cross_signing_key cross_signing_key
) )
self_signing_key_by_user[user] = { self_signing_key_by_user[user] = {
"key_info": cross_signing_key, "key_info": cross_signing_key,
"pubkey": verify_key.version "pubkey": verify_key.version,
} }
# if we have exceeded the limit, we need to exclude any results with the # if we have exceeded the limit, we need to exclude any results with the
@ -185,10 +188,13 @@ class DeviceWorkerStore(SQLBaseStore):
break break
# skip over cross-signing keys # skip over cross-signing keys
if (update[0] in master_key_by_user if (
and update[1] == master_key_by_user[update[0]]["pubkey"]) \ update[0] in master_key_by_user
or (update[0] in master_key_by_user and update[1] == master_key_by_user[update[0]]["pubkey"]
and update[1] == self_signing_key_by_user[update[0]]["pubkey"]): ) or (
update[0] in master_key_by_user
and update[1] == self_signing_key_by_user[update[0]]["pubkey"]
):
continue continue
key = (update[0], update[1]) key = (update[0], update[1])
@ -209,16 +215,16 @@ class DeviceWorkerStore(SQLBaseStore):
# update list with the master/self-signing key by user maps # update list with the master/self-signing key by user maps
cross_signing_keys_by_user = {} cross_signing_keys_by_user = {}
for user_id, device_id, stream in updates: for user_id, device_id, stream in updates:
if device_id == master_key_by_user.get(user_id, {}) \ if device_id == master_key_by_user.get(user_id, {}).get("pubkey", None):
.get("pubkey", None):
result = cross_signing_keys_by_user.setdefault(user_id, {}) result = cross_signing_keys_by_user.setdefault(user_id, {})
result["master_key"] = \ result["master_key"] = master_key_by_user[user_id]["key_info"]
master_key_by_user[user_id]["key_info"] elif device_id == self_signing_key_by_user.get(user_id, {}).get(
elif device_id == self_signing_key_by_user.get(user_id, {}) \ "pubkey", None
.get("pubkey", None): ):
result = cross_signing_keys_by_user.setdefault(user_id, {}) result = cross_signing_keys_by_user.setdefault(user_id, {})
result["self_signing_key"] = \ result["self_signing_key"] = self_signing_key_by_user[user_id][
self_signing_key_by_user[user_id]["key_info"] "key_info"
]
cross_signing_results = [] cross_signing_results = []
@ -282,13 +288,17 @@ class DeviceWorkerStore(SQLBaseStore):
List[Dict]: List of objects representing an device update EDU List[Dict]: List of objects representing an device update EDU
""" """
devices = yield self.runInteraction( devices = (
"_get_e2e_device_keys_txn", yield self.runInteraction(
self._get_e2e_device_keys_txn, "_get_e2e_device_keys_txn",
query_map.keys(), self._get_e2e_device_keys_txn,
include_all_devices=True, query_map.keys(),
include_deleted_devices=True, include_all_devices=True,
) if query_map else {} include_deleted_devices=True,
)
if query_map
else {}
)
results = [] results = []
for user_id, user_devices in iteritems(devices): for user_id, user_devices in iteritems(devices):