Get the end-to-end key federation working

This commit is contained in:
Mark Haines 2015-07-24 18:26:46 +01:00
parent 62c010283d
commit 2da3b1e60b
4 changed files with 17 additions and 21 deletions

View file

@ -135,7 +135,7 @@ class FederationClient(FederationBase):
)
@log_function
def query_client_keys(self, destination, content, retry_on_dns_fail=True):
def query_client_keys(self, destination, content):
"""Query device keys for a device hosted on a remote server.
Args:
@ -147,12 +147,10 @@ class FederationClient(FederationBase):
response
"""
sent_queries_counter.inc("client_device_keys")
return self.transport_layer.query_client_keys(
destination, content, retry_on_dns_fail=retry_on_dns_fail
)
return self.transport_layer.query_client_keys(destination, content)
@log_function
def claim_client_keys(self, destination, content, retry_on_dns_fail=True):
def claim_client_keys(self, destination, content):
"""Claims one-time keys for a device hosted on a remote server.
Args:
@ -164,9 +162,7 @@ class FederationClient(FederationBase):
response
"""
sent_queries_counter.inc("client_one_time_keys")
return self.transport_layer.claim_client_keys(
destination, content, retry_on_dns_fail=retry_on_dns_fail
)
return self.transport_layer.claim_client_keys(destination, content)
@defer.inlineCallbacks
@log_function

View file

@ -247,7 +247,7 @@ class TransportLayerClient(object):
Returns:
A dict containg the device keys.
"""
path = PREFIX + "/client_keys/query"
path = PREFIX + "/user/keys/query"
content = yield self.client.post_json(
destination=destination,
@ -283,7 +283,7 @@ class TransportLayerClient(object):
A dict containg the one-time keys.
"""
path = PREFIX + "/client_keys/claim"
path = PREFIX + "/user/keys/claim"
content = yield self.client.post_json(
destination=destination,

View file

@ -326,20 +326,20 @@ class FederationInviteServlet(BaseFederationServlet):
class FederationClientKeysQueryServlet(BaseFederationServlet):
PATH = "/client_keys/query"
PATH = "/user/keys/query"
@defer.inlineCallbacks
def on_POST(self, origin, content):
response = yield self.handler.on_client_key_query(origin, content)
def on_POST(self, origin, content, query):
response = yield self.handler.on_query_client_keys(origin, content)
defer.returnValue((200, response))
class FederationClientKeysClaimServlet(BaseFederationServlet):
PATH = "/client_keys/claim"
PATH = "/user/keys/claim"
@defer.inlineCallbacks
def on_POST(self, origin, content):
response = yield self.handler.on_client_key_claim(origin, content)
def on_POST(self, origin, content, query):
response = yield self.handler.on_claim_client_keys(origin, content)
defer.returnValue((200, response))