use log::trace;
use openmls::prelude::{GroupEpoch, GroupId, JoinProposal, LeafNodeIndex, MlsMessageOut, Proposal};
use std::collections::HashSet;
use crate::{
mls::{credential::typ::MlsCredentialType, ClientId, ConversationId},
prelude::{CryptoError, CryptoResult, MlsCiphersuite, MlsConversation, MlsError},
};
use crate::context::CentralContext;
impl MlsConversation {
pub fn members_in_next_epoch(&self) -> Vec<ClientId> {
let pending_removals = self.pending_removals();
let existing_clients = self
.group
.members()
.filter_map(|kp| {
if !pending_removals.contains(&kp.index) {
Some(kp.credential.identity().into())
} else {
trace!(client_index:% = kp.index; "Client is pending removal");
None
}
})
.collect::<HashSet<_>>();
existing_clients.into_iter().collect()
}
fn pending_removals(&self) -> Vec<LeafNodeIndex> {
self.group
.pending_proposals()
.filter_map(|proposal| match proposal.proposal() {
Proposal::Remove(ref remove) => Some(remove.removed()),
_ => None,
})
.collect::<Vec<_>>()
}
}
impl CentralContext {
#[cfg_attr(test, crate::dispotent)]
pub async fn new_external_add_proposal(
&self,
conversation_id: ConversationId,
epoch: GroupEpoch,
ciphersuite: MlsCiphersuite,
credential_type: MlsCredentialType,
) -> CryptoResult<MlsMessageOut> {
let group_id = GroupId::from_slice(&conversation_id[..]);
let mls_provider = self.mls_provider().await?;
let client = self.mls_client().await?;
let cb = client
.find_most_recent_credential_bundle(ciphersuite.signature_algorithm(), credential_type)
.await;
let cb = match (cb, credential_type) {
(Ok(cb), _) => cb,
(Err(CryptoError::CredentialNotFound(_)), MlsCredentialType::Basic) => {
client
.init_basic_credential_bundle_if_missing(&mls_provider, ciphersuite.signature_algorithm())
.await?;
client
.find_most_recent_credential_bundle(ciphersuite.signature_algorithm(), credential_type)
.await?
}
(Err(CryptoError::CredentialNotFound(_)), MlsCredentialType::X509) => {
return Err(CryptoError::E2eiEnrollmentNotDone)
}
(Err(e), _) => return Err(e),
};
let kp = client
.generate_one_keypackage_from_credential_bundle(&mls_provider, ciphersuite, &cb)
.await?;
Ok(JoinProposal::new(kp, group_id, epoch, &cb.signature_key).map_err(MlsError::from)?)
}
}
#[cfg(test)]
mod tests {
use wasm_bindgen_test::*;
use crate::{prelude::MlsCommitBundle, test_utils::*};
wasm_bindgen_test_configure!(run_in_browser);
mod add {
use super::*;
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn guest_should_externally_propose_adding_itself_to_owner_group(case: TestCase) {
run_test_with_client_ids(
case.clone(),
["owner", "guest"],
move |[owner_central, guest_central]| {
Box::pin(async move {
let id = conversation_id();
owner_central
.context
.new_conversation(&id, case.credential_type, case.cfg.clone())
.await
.unwrap();
let epoch = owner_central.get_conversation_unchecked(&id).await.group.epoch();
let external_add = guest_central
.context
.new_external_add_proposal(id.clone(), epoch, case.ciphersuite(), case.credential_type)
.await
.unwrap();
let decrypted = owner_central
.context
.decrypt_message(&id, external_add.to_bytes().unwrap())
.await
.unwrap();
assert_eq!(owner_central.get_conversation_unchecked(&id).await.members().len(), 1);
guest_central.verify_sender_identity(&case, &decrypted).await;
let MlsCommitBundle { welcome, .. } = owner_central
.context
.commit_pending_proposals(&id)
.await
.unwrap()
.unwrap();
owner_central.context.commit_accepted(&id).await.unwrap();
assert_eq!(owner_central.get_conversation_unchecked(&id).await.members().len(), 2);
guest_central
.context
.process_welcome_message(welcome.unwrap().into(), case.custom_cfg())
.await
.unwrap();
assert_eq!(guest_central.get_conversation_unchecked(&id).await.members().len(), 2);
assert!(guest_central.try_talk_to(&id, &owner_central).await.is_ok());
})
},
)
.await
}
}
mod remove {
use super::*;
use crate::prelude::{CryptoError, MlsConversationCreationMessage, MlsConversationInitBundle, MlsError};
use openmls::prelude::{
ExternalProposal, GroupId, MlsMessageIn, ProcessMessageError, SenderExtensionIndex, ValidationError,
};
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn ds_should_remove_guest_from_conversation(case: TestCase) {
run_test_with_client_ids(case.clone(), ["owner", "guest", "ds"], move |[owner, guest, ds]| {
Box::pin(async move {
let owner_central = &owner.context;
let guest_central = &guest.context;
let id = conversation_id();
let ds_signature_key = ds.client_signature_key(&case).await.as_slice().to_vec();
let mut cfg = case.cfg.clone();
owner_central
.set_raw_external_senders(&mut cfg, vec![ds_signature_key])
.await
.unwrap();
owner_central
.new_conversation(&id, case.credential_type, cfg)
.await
.unwrap();
owner.invite_all(&case, &id, [&guest]).await.unwrap();
assert_eq!(owner.get_conversation_unchecked(&id).await.members().len(), 2);
let to_remove = owner.index_of(&id, guest.get_client_id().await).await;
let sender_index = SenderExtensionIndex::new(0);
let (sc, ct) = (case.signature_scheme(), case.credential_type);
let cb = ds.find_most_recent_credential_bundle(sc, ct).await.unwrap();
let group_id = GroupId::from_slice(&id[..]);
let epoch = owner.get_conversation_unchecked(&id).await.group.epoch();
let proposal =
ExternalProposal::new_remove(to_remove, group_id, epoch, &cb.signature_key, sender_index)
.unwrap();
owner_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
guest_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
let MlsCommitBundle { commit, .. } =
owner_central.commit_pending_proposals(&id).await.unwrap().unwrap();
assert_eq!(owner.get_conversation_unchecked(&id).await.members().len(), 2);
owner_central.commit_accepted(&id).await.unwrap();
assert_eq!(owner.get_conversation_unchecked(&id).await.members().len(), 1);
guest_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert!(guest_central.get_conversation(&id).await.is_err());
assert!(guest.try_talk_to(&id, &owner).await.is_err());
})
})
.await
}
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn should_fail_when_invalid_external_sender(case: TestCase) {
run_test_with_client_ids(
case.clone(),
["owner", "guest", "ds", "attacker"],
move |[owner, guest, ds, attacker]| {
Box::pin(async move {
let id = conversation_id();
let ds_signature_key = ds.client_signature_key(&case).await.as_slice().to_vec();
let mut cfg = case.cfg.clone();
owner
.context
.set_raw_external_senders(&mut cfg, vec![ds_signature_key])
.await
.unwrap();
owner
.context
.new_conversation(&id, case.credential_type, cfg)
.await
.unwrap();
owner.invite_all(&case, &id, [&guest]).await.unwrap();
assert_eq!(owner.get_conversation_unchecked(&id).await.members().len(), 2);
let to_remove = owner.index_of(&id, guest.get_client_id().await).await;
let sender_index = SenderExtensionIndex::new(1);
let (sc, ct) = (case.signature_scheme(), case.credential_type);
let cb = attacker.find_most_recent_credential_bundle(sc, ct).await.unwrap();
let group_id = GroupId::from_slice(&id[..]);
let epoch = owner.get_conversation_unchecked(&id).await.group.epoch();
let proposal =
ExternalProposal::new_remove(to_remove, group_id, epoch, &cb.signature_key, sender_index)
.unwrap();
let owner_decrypt = owner.context.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(matches!(
owner_decrypt.unwrap_err(),
CryptoError::MlsError(MlsError::MlsMessageError(ProcessMessageError::ValidationError(
ValidationError::UnauthorizedExternalSender
)))
));
let guest_decrypt = owner.context.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(matches!(
guest_decrypt.unwrap_err(),
CryptoError::MlsError(MlsError::MlsMessageError(ProcessMessageError::ValidationError(
ValidationError::UnauthorizedExternalSender
)))
));
})
},
)
.await
}
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn should_fail_when_wrong_signature_key(case: TestCase) {
run_test_with_client_ids(case.clone(), ["owner", "guest", "ds"], move |[owner, guest, ds]| {
Box::pin(async move {
let id = conversation_id();
let key = ds.client_signature_key(&case).await.as_slice().to_vec();
let mut cfg = case.cfg.clone();
owner
.context
.set_raw_external_senders(&mut cfg, vec![key.as_slice().to_vec()])
.await
.unwrap();
owner
.context
.new_conversation(&id, case.credential_type, cfg)
.await
.unwrap();
owner.invite_all(&case, &id, [&guest]).await.unwrap();
assert_eq!(owner.get_conversation_unchecked(&id).await.members().len(), 2);
let to_remove = owner.index_of(&id, guest.get_client_id().await).await;
let sender_index = SenderExtensionIndex::new(0);
let (sc, ct) = (case.signature_scheme(), case.credential_type);
let cb = guest.find_most_recent_credential_bundle(sc, ct).await.unwrap();
let group_id = GroupId::from_slice(&id[..]);
let epoch = owner.get_conversation_unchecked(&id).await.group.epoch();
let proposal =
ExternalProposal::new_remove(to_remove, group_id, epoch, &cb.signature_key, sender_index)
.unwrap();
let owner_decrypt = owner.context.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(matches!(
owner_decrypt.unwrap_err(),
CryptoError::MlsError(MlsError::MlsMessageError(ProcessMessageError::InvalidSignature))
));
let guest_decrypt = owner.context.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(matches!(
guest_decrypt.unwrap_err(),
CryptoError::MlsError(MlsError::MlsMessageError(ProcessMessageError::InvalidSignature))
));
})
})
.await
}
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn joiners_from_welcome_can_accept_external_remove_proposals(case: TestCase) {
run_test_with_client_ids(
case.clone(),
["alice", "bob", "charlie", "ds"],
move |[alice, bob, charlie, ds]| {
Box::pin(async move {
let alice_central = &alice.context;
let bob_central = &bob.context;
let charlie_central = &charlie.context;
let id = conversation_id();
let ds_signature_key = ds.client_signature_key(&case).await.as_slice().to_vec();
let mut cfg = case.cfg.clone();
alice_central
.set_raw_external_senders(&mut cfg, vec![ds_signature_key])
.await
.unwrap();
alice_central
.new_conversation(&id, case.credential_type, cfg)
.await
.unwrap();
alice.invite_all(&case, &id, [&bob]).await.unwrap();
assert_eq!(alice.get_conversation_unchecked(&id).await.members().len(), 2);
let charlie_kp = charlie.rand_key_package(&case).await;
let MlsConversationCreationMessage { welcome, commit, .. } = alice_central
.add_members_to_conversation(&id, vec![charlie_kp])
.await
.unwrap();
alice_central.commit_accepted(&id).await.unwrap();
bob_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
charlie_central
.process_welcome_message(MlsMessageIn::from(welcome), case.custom_cfg())
.await
.unwrap();
assert_eq!(charlie.get_conversation_unchecked(&id).await.members().len(), 3);
assert!(charlie.try_talk_to(&id, &alice).await.is_ok());
assert!(charlie.try_talk_to(&id, &bob).await.is_ok());
let to_remove = alice.index_of(&id, bob.get_client_id().await).await;
let sender_index = SenderExtensionIndex::new(0);
let (sc, ct) = (case.signature_scheme(), case.credential_type);
let cb = ds.find_most_recent_credential_bundle(sc, ct).await.unwrap();
let group_id = GroupId::from_slice(&id[..]);
let epoch = alice.get_conversation_unchecked(&id).await.group.epoch();
let proposal =
ExternalProposal::new_remove(to_remove, group_id, epoch, &cb.signature_key, sender_index)
.unwrap();
let charlie_can_verify_ext_proposal =
charlie_central.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(charlie_can_verify_ext_proposal.is_ok());
alice_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
bob_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
let commit = charlie_central
.commit_pending_proposals(&id)
.await
.unwrap()
.unwrap()
.commit;
charlie_central.commit_accepted(&id).await.unwrap();
assert_eq!(charlie.get_conversation_unchecked(&id).await.members().len(), 2);
alice_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert_eq!(alice.get_conversation_unchecked(&id).await.members().len(), 2);
bob_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert!(alice.try_talk_to(&id, &charlie).await.is_ok());
assert!(alice.try_talk_to(&id, &bob).await.is_err());
})
},
)
.await
}
#[apply(all_cred_cipher)]
#[wasm_bindgen_test]
async fn joiners_from_external_commit_can_accept_external_remove_proposals(case: TestCase) {
run_test_with_client_ids(
case.clone(),
["alice", "bob", "charlie", "ds"],
move |[alice, bob, charlie, ds]| {
Box::pin(async move {
let alice_central = &alice.context;
let bob_central = &bob.context;
let charlie_central = &charlie.context;
let id = conversation_id();
let ds_signature_key = ds.client_signature_key(&case).await.as_slice().to_vec();
let mut cfg = case.cfg.clone();
alice_central
.set_raw_external_senders(&mut cfg, vec![ds_signature_key])
.await
.unwrap();
alice_central
.new_conversation(&id, case.credential_type, cfg)
.await
.unwrap();
alice.invite_all(&case, &id, [&bob]).await.unwrap();
assert_eq!(alice.get_conversation_unchecked(&id).await.members().len(), 2);
let public_group_state = alice.get_group_info(&id).await;
let MlsConversationInitBundle { commit, .. } = charlie_central
.join_by_external_commit(public_group_state, case.custom_cfg(), case.credential_type)
.await
.unwrap();
charlie_central
.merge_pending_group_from_external_commit(&id)
.await
.unwrap();
alice_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
bob_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert_eq!(charlie.get_conversation_unchecked(&id).await.members().len(), 3);
assert!(charlie.try_talk_to(&id, &alice).await.is_ok());
assert!(charlie.try_talk_to(&id, &bob).await.is_ok());
let to_remove = alice.index_of(&id, bob.get_client_id().await).await;
let sender_index = SenderExtensionIndex::new(0);
let (sc, ct) = (case.signature_scheme(), case.credential_type);
let cb = ds.find_most_recent_credential_bundle(sc, ct).await.unwrap();
let group_id = GroupId::from_slice(&id[..]);
let epoch = alice.get_conversation_unchecked(&id).await.group.epoch();
let proposal =
ExternalProposal::new_remove(to_remove, group_id, epoch, &cb.signature_key, sender_index)
.unwrap();
let charlie_can_verify_ext_proposal =
charlie_central.decrypt_message(&id, proposal.to_bytes().unwrap()).await;
assert!(charlie_can_verify_ext_proposal.is_ok());
alice_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
bob_central
.decrypt_message(&id, proposal.to_bytes().unwrap())
.await
.unwrap();
let commit = charlie_central
.commit_pending_proposals(&id)
.await
.unwrap()
.unwrap()
.commit;
charlie_central.commit_accepted(&id).await.unwrap();
assert_eq!(charlie.get_conversation_unchecked(&id).await.members().len(), 2);
alice_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert_eq!(alice.get_conversation_unchecked(&id).await.members().len(), 2);
bob_central
.decrypt_message(&id, commit.to_bytes().unwrap())
.await
.unwrap();
assert!(alice.try_talk_to(&id, &charlie).await.is_ok());
assert!(alice.try_talk_to(&id, &bob).await.is_err());
})
},
)
.await
}
}
}