absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-08 12:19:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finish transaction signing task with TX rebroadcast code

Browse Source
This commit is contained in:
Luke Parker
2024-09-06 04:15:02 -04:00
parent a353f9e2da
commit 100c80be9f
6 changed files with 109 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
processor/frost-attempt-manager/src/individual.rs
View File

@@ -278,9 +278,7 @@ impl<D: Db, M: Clone + PreprocessMachine> SigningProtocol<D, M> {
}
/// Cleanup the database entries for a specified signing protocol.
pub(crate) fn cleanup(db: &mut D, id: [u8; 32]) {
let mut txn = db.txn();
Attempted::del(&mut txn, id);
txn.commit();
pub(crate) fn cleanup(txn: &mut impl DbTxn, id: [u8; 32]) {
Attempted::del(txn, id);
}
}

15
processor/frost-attempt-manager/src/lib.rs
View File

@@ -8,7 +8,7 @@ use frost::{Participant, sign::PreprocessMachine};
use serai_validator_sets_primitives::Session;
use serai_db::Db;
use serai_db::{DbTxn, Db};
use messages::sign::{ProcessorMessage, CoordinatorMessage};
mod individual;
@@ -19,7 +19,12 @@ pub enum Response<M: PreprocessMachine> {
/// Messages to send to the coordinator.
Messages(Vec<ProcessorMessage>),
/// A produced signature.
Signature(M::Signature),
Signature {
/// The ID of the protocol this is for.
id: [u8; 32],
/// The signature.
signature: M::Signature,
},
}
/// A manager of attempts for a variety of signing protocols.
@@ -55,13 +60,13 @@ impl<D: Db, M: Clone + PreprocessMachine> AttemptManager<D, M> {
/// This does not stop the protocol from being re-registered and further worked on (with
/// undefined behavior) then. The higher-level context must never call `register` again with this
/// ID accordingly.
pub fn retire(&mut self, id: [u8; 32]) {
pub fn retire(&mut self, txn: &mut impl DbTxn, id: [u8; 32]) {
if self.active.remove(&id).is_none() {
log::info!("retiring protocol {}, which we didn't register/already retired", hex::encode(id));
} else {
log::info!("retired signing protocol {}", hex::encode(id));
}
SigningProtocol::<D, M>::cleanup(&mut self.db, id);
SigningProtocol::<D, M>::cleanup(txn, id);
}
/// Handle a message for a signing protocol.
@@ -90,7 +95,7 @@ impl<D: Db, M: Clone + PreprocessMachine> AttemptManager<D, M> {
return Response::Messages(vec![]);
};
match protocol.shares(id.attempt, shares) {
Ok(signature) => Response::Signature(signature),
Ok(signature) => Response::Signature { id: id.id, signature },
Err(messages) => Response::Messages(messages),
}
}