improvement: use transaction ids for federation requests

next
Timo Kösters 2021-03-02 14:36:48 +01:00
parent dd68031b69
commit 278751eb23
No known key found for this signature in database
GPG Key ID: 24DA7517711A2BA4
1 changed files with 16 additions and 2 deletions

View File

@ -9,6 +9,7 @@ use std::{
use crate::{appservice_server, server_server, utils, Error, PduEvent, Result}; use crate::{appservice_server, server_server, utils, Error, PduEvent, Result};
use federation::transactions::send_transaction_message; use federation::transactions::send_transaction_message;
use log::info; use log::info;
use ring::digest;
use rocket::futures::stream::{FuturesUnordered, StreamExt}; use rocket::futures::stream::{FuturesUnordered, StreamExt};
use ruma::{ use ruma::{
api::{appservice, federation, OutgoingRequest}, api::{appservice, federation, OutgoingRequest},
@ -229,6 +230,13 @@ impl Sending {
Ok(()) Ok(())
} }
fn calculate_hash(keys: &[IVec]) -> Vec<u8> {
// We only hash the pdu's event ids, not the whole pdu
let bytes = keys.join(&0xff);
let hash = digest::digest(&digest::SHA256, &bytes);
hash.as_ref().to_owned()
}
async fn handle_event( async fn handle_event(
server: Box<ServerName>, server: Box<ServerName>,
is_appservice: bool, is_appservice: bool,
@ -266,7 +274,10 @@ impl Sending {
.unwrap(), // TODO: handle error .unwrap(), // TODO: handle error
appservice::event::push_events::v1::Request { appservice::event::push_events::v1::Request {
events: &pdu_jsons, events: &pdu_jsons,
txn_id: &utils::random_string(16), txn_id: &base64::encode_config(
Self::calculate_hash(&pdu_ids),
base64::URL_SAFE_NO_PAD,
),
}, },
) )
.await .await
@ -309,7 +320,10 @@ impl Sending {
pdus: &pdu_jsons, pdus: &pdu_jsons,
edus: &[], edus: &[],
origin_server_ts: SystemTime::now(), origin_server_ts: SystemTime::now(),
transaction_id: &utils::random_string(16), transaction_id: &base64::encode_config(
Self::calculate_hash(&pdu_ids),
base64::URL_SAFE_NO_PAD,
),
}, },
) )
.await .await