mirror of
https://github.com/logos-messaging/libchat.git
synced 2026-02-10 00:43:09 +00:00
10940321ff
* feat: costom encode for double ratchet * chore: correct capacity * chore: refactor reference * chore: reader for parse bytes * chore: extract reader * chore: example with persist state. * chore: update example * chore: implement serde compatibility. * chore: as_bytes * chore: zerorize the secrec material * chore: use as_types to return reference for static key. * chore: extract example from basic demo
76 lines
2.6 KiB
Rust
76 lines
2.6 KiB
Rust
use double_ratchets::{InstallationKeyPair, RatchetState, hkdf::PrivateV1Domain};
|
|
|
|
fn main() {
|
|
// === Initial shared secret (X3DH / prekey result in real systems) ===
|
|
let shared_secret = [42u8; 32];
|
|
|
|
let bob_dh = InstallationKeyPair::generate();
|
|
|
|
let mut alice: RatchetState<PrivateV1Domain> =
|
|
RatchetState::init_sender(shared_secret, bob_dh.public().clone());
|
|
let mut bob: RatchetState<PrivateV1Domain> = RatchetState::init_receiver(shared_secret, bob_dh);
|
|
|
|
let (ciphertext, header) = alice.encrypt_message(b"Hello Bob!");
|
|
|
|
// === Bob receives ===
|
|
let plaintext = bob.decrypt_message(&ciphertext, header);
|
|
println!(
|
|
"Bob received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
|
|
// === Bob replies (triggers DH ratchet) ===
|
|
let (ciphertext, header) = bob.encrypt_message(b"Hi Alice!");
|
|
|
|
let plaintext = alice.decrypt_message(&ciphertext, header);
|
|
println!(
|
|
"Alice received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
|
|
// === Serialize the state of alice and bob ===
|
|
println!("Before restart, persist the state");
|
|
let alice_state = alice.as_bytes();
|
|
let bob_state = bob.as_bytes();
|
|
|
|
// === Deserialize alice and bob state from bytes ===
|
|
println!("Restart alice and bob");
|
|
let mut alice_new: RatchetState<PrivateV1Domain> =
|
|
RatchetState::from_bytes(&alice_state).unwrap();
|
|
let mut bob_new: RatchetState<PrivateV1Domain> = RatchetState::from_bytes(&bob_state).unwrap();
|
|
|
|
// === Alice sends a message ===
|
|
let (ciphertext, header) = alice_new.encrypt_message(b"Hello Bob!");
|
|
|
|
// === Bob receives ===
|
|
let plaintext = bob_new.decrypt_message(&ciphertext, header);
|
|
println!(
|
|
"New Bob received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
|
|
// === Bob replies (triggers DH ratchet) ===
|
|
let (ciphertext, header) = bob_new.encrypt_message(b"Hi Alice!");
|
|
|
|
let plaintext = alice_new.decrypt_message(&ciphertext, header);
|
|
println!(
|
|
"New Alice received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
|
|
let (skipped_ciphertext, skipped_header) = bob_new.encrypt_message(b"Hi Alice skipped!");
|
|
let (resumed_ciphertext, resumed_header) = bob_new.encrypt_message(b"Hi Alice resumed!");
|
|
|
|
let plaintext = alice_new.decrypt_message(&resumed_ciphertext, resumed_header);
|
|
println!(
|
|
"New Alice received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
|
|
let plaintext = alice_new.decrypt_message(&skipped_ciphertext, skipped_header);
|
|
println!(
|
|
"New Alice received: {}",
|
|
String::from_utf8_lossy(&plaintext.unwrap())
|
|
);
|
|
}
|