4nk/sdk_common
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Take an optional pairingid as sender instead of Member

Browse Source
This commit is contained in:
Sosthene 2025-09-03 15:13:50 +02:00
parent dab1a4dd2c
commit e205229e92
1 changed files with 36 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/prd.rs
View File

@ -11,7 +11,8 @@ use sp_client::silentpayments::SilentPaymentAddress;
use sp_client::SpClient;
use tsify::Tsify;
use crate::pcd::{Member, Pcd, PcdCommitments, Roles};
use crate::pcd::{Pcd, PcdCommitments, Roles};
use crate::serialization::OutPointMemberMap;
use crate::signature::{AnkHash, AnkMessageHash, Proof};
#[derive(Debug, Default, Clone, PartialEq, Serialize, Deserialize, Tsify)]
@ -58,7 +59,7 @@ impl AnkPrdHash {
pub struct Prd {
pub prd_type: PrdType,
pub process_id: OutPoint,
pub sender: Member,
pub sender: Option<OutPoint>,
pub keys: BTreeMap<String, [u8; 32]>, // key is a key in pcd, value is the key to decrypt it
pub pcd_commitments: PcdCommitments,
pub validation_tokens: Vec<Proof>,
@ -74,7 +75,7 @@ impl Prd {
/// If validation_tokens contains a valid proof signed by ourselves of empty prd,
/// we confirm the secret if necessary and don't return anything
pub fn new_connect(
sender: Member,
sender: Option<OutPoint>,
secret_hash: AnkMessageHash,
previous_proof: Option<Proof>,
) -> Self {
@ -98,7 +99,7 @@ impl Prd {
pub fn new_update(
process_id: OutPoint,
sender: Member,
sender: OutPoint,
roles: Roles,
public_data: Pcd,
keys: BTreeMap<String, [u8; 32]>,
@ -107,7 +108,7 @@ impl Prd {
Self {
prd_type: PrdType::Update,
process_id,
sender,
sender: Some(sender),
validation_tokens: vec![],
keys,
pcd_commitments,
@ -120,14 +121,14 @@ impl Prd {
pub fn new_response(
process_id: OutPoint,
sender: Member,
sender: OutPoint,
validation_tokens: Vec<Proof>,
pcd_commitments: PcdCommitments,
) -> Self {
Self {
prd_type: PrdType::Response,
process_id,
sender,
sender: Some(sender),
validation_tokens,
pcd_commitments,
..Default::default()
@ -136,29 +137,29 @@ impl Prd {
pub fn new_confirm(
process_id: OutPoint,
sender: Member,
sender: OutPoint,
pcd_commitments: PcdCommitments,
) -> Self {
Self {
prd_type: PrdType::Confirm,
process_id,
pcd_commitments,
sender,
sender: Some(sender),
..Default::default()
}
}
pub fn new_request(process_id: OutPoint, sender: Member, state_ids: Vec<[u8; 32]>) -> Self {
pub fn new_request(process_id: OutPoint, sender: OutPoint, state_ids: Vec<[u8; 32]>) -> Self {
Self {
prd_type: PrdType::Request,
process_id,
sender,
sender: Some(sender),
payload: serde_json::to_string(&state_ids).unwrap(),
..Default::default()
}
}
pub fn extract_from_message(plain: &[u8], local_address: SilentPaymentAddress) -> Result<Self> {
pub fn extract_from_message(plain: &[u8], local_address: SilentPaymentAddress, members: &OutPointMemberMap) -> Result<Self> {
let prd: Prd = serde_json::from_slice(plain)?;
// check that the proof is consistent
@ -171,8 +172,12 @@ impl Prd {
"Proof signed by ourselves, we are parsing our own message",
));
}
proof.verify()?;
if let Some(sender) = prd.sender {
// take the spending keys in sender
let addresses = prd.sender.get_addresses();
let addresses = members.0.get(&sender).ok_or(anyhow::Error::msg("Unknown sender"))?.get_addresses();
let mut spend_keys: Vec<PublicKey> = vec![];
for address in addresses {
spend_keys.push(<SilentPaymentAddress>::try_from(address)?.get_spend_key());
@ -186,9 +191,11 @@ impl Prd {
}
}
if !known_key {
return Err(anyhow::Error::msg("Proof signed with an unknown key"));
log::warn!("Proof signed with a key that doesn't match declared sender");
}
} else {
log::warn!("No declared sender");
}
proof.verify()?;
} else {
log::warn!("No proof for prd with process_id {}", prd.process_id);
}