backups/grapevine
1
0
Fork 0
mirror of https://gitlab.computer.surgery/matrix/grapevine.git synced 2025-12-17 07:41:23 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

log failed remote device key requests

Browse source
This commit is contained in:
Benjamin Lee 2024-08-10 12:46:50 -07:00
parent 827f670c54
commit 1551833501
No known key found for this signature in database
GPG key ID: FB9624E2885D55A4
1 changed files with 40 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

73
src/api/client_server/keys.rs
View file

@ -440,48 +440,55 @@ pub(crate) async fn get_keys_helper<F: Fn(&UserId) -> bool>(
), ),
) )
.await .await
.map_err(|_e| Error::BadServerResponse("Query took too long")), .map_err(|_e| Error::BadServerResponse("Query took too long"))
// TODO: switch to .flatten() when stable
// <https://github.com/rust-lang/rust/issues/70142>
.and_then(|result| result),
) )
}) })
.collect(); .collect();
while let Some((server, response)) = futures.next().await { while let Some((server, response)) = futures.next().await {
if let Ok(Ok(response)) = response { let response = match response {
for (user, masterkey) in response.master_keys { Ok(response) => response,
let (master_key_id, mut master_key) = Err(error) => {
services().users.parse_master_key(&user, &masterkey)?; back_off(server.to_owned()).await;
debug!(%server, %error, "remote device key query failed");
if let Some(our_master_key) = services().users.get_key( failures.insert(server.to_string(), json!({}));
&master_key_id, continue;
sender_user,
&user,
&allowed_signatures,
)? {
let (_, our_master_key) = services()
.users
.parse_master_key(&user, &our_master_key)?;
master_key.signatures.extend(our_master_key.signatures);
}
let json = serde_json::to_value(master_key)
.expect("to_value always works");
let raw = serde_json::from_value(json)
.expect("Raw::from_value always works");
services().users.add_cross_signing_keys(
&user, &raw, &None, &None,
// Dont notify. A notification would trigger another key
// request resulting in an endless loop
false,
)?;
master_keys.insert(user, raw);
} }
};
self_signing_keys.extend(response.self_signing_keys); for (user, masterkey) in response.master_keys {
device_keys.extend(response.device_keys); let (master_key_id, mut master_key) =
} else { services().users.parse_master_key(&user, &masterkey)?;
back_off(server.to_owned()).await;
failures.insert(server.to_string(), json!({})); if let Some(our_master_key) = services().users.get_key(
&master_key_id,
sender_user,
&user,
&allowed_signatures,
)? {
let (_, our_master_key) = services()
.users
.parse_master_key(&user, &our_master_key)?;
master_key.signatures.extend(our_master_key.signatures);
}
let json = serde_json::to_value(master_key)
.expect("to_value always works");
let raw = serde_json::from_value(json)
.expect("Raw::from_value always works");
services().users.add_cross_signing_keys(
&user, &raw, &None, &None,
// Dont notify. A notification would trigger another key
// request resulting in an endless loop
false,
)?;
master_keys.insert(user, raw);
} }
self_signing_keys.extend(response.self_signing_keys);
device_keys.extend(response.device_keys);
} }
Ok(get_keys::v3::Response { Ok(get_keys::v3::Response {