use hkdf::Hkdf;
use nostr_sdk::nips::nip44::v2::ConversationKey;
use nostr_sdk::prelude::{Keys, PublicKey, SecretKey};
use sha2::{Digest, Sha256};
use super::super::{ChannelId, CommunityId, Epoch};
const LABEL_CHANNEL: &str = "concord/channel";
const LABEL_CONTROL: &str = "concord/control";
const LABEL_REKEY_PSEUDONYM: &str = "concord/rekey-pseudonym";
const LABEL_BASE_REKEY_PSEUDONYM: &str = "concord/base-rekey-pseudonym";
const LABEL_RECIPIENT_PSEUDONYM: &str = "concord/recipient-pseudonym";
const LABEL_GUESTBOOK: &str = "concord/guestbook";
const LABEL_VOICE_SIGNER: &str = "concord/voice-signer";
const LABEL_VOICE_MEDIA: &str = "concord/voice-media";
const LABEL_VOICE_SENDER: &str = "concord/voice-sender";
const LABEL_DISSOLVED: &str = "concord/dissolved";
const LABEL_GRANT: &str = "concord/grant";
const LABEL_BANLIST: &str = "concord/banlist";
const LABEL_INVITE_LINKS: &str = "concord/invite-links";
const LABEL_INVITE_KEY: &str = "concord/invite-key";
const LABEL_COMMUNITY: &str = "concord/community";
const LABEL_EPOCH_COMMITMENT: &str = "concord/epoch-key-commitment";
const ZERO32: [u8; 32] = [0u8; 32];
pub const TOKEN_LEN: usize = 16;
fn build_info(label: &str, id32: &[u8; 32], epoch: Option<u64>) -> Vec<u8> {
let mut info = Vec::with_capacity(label.len() + 1 + 32 + 8);
info.extend_from_slice(label.as_bytes());
info.push(0x00);
info.extend_from_slice(id32);
if let Some(e) = epoch {
info.extend_from_slice(&e.to_be_bytes());
}
info
}
fn hkdf32(ikm: &[u8], info: &[u8]) -> [u8; 32] {
let hk = Hkdf::<Sha256>::new(None, ikm);
let mut okm = [0u8; 32];
hk.expand(info, &mut okm)
.expect("HKDF expand of 32 bytes is infallible");
okm
}
fn hkdf_to_secret_key(ikm: &[u8], base_info: &[u8]) -> SecretKey {
if let Ok(sk) = SecretKey::from_slice(&hkdf32(ikm, base_info)) {
return sk;
}
for counter in 0u8..=255 {
let mut info = base_info.to_vec();
info.push(counter);
if let Ok(sk) = SecretKey::from_slice(&hkdf32(ikm, &info)) {
return sk;
}
}
unreachable!("secp256k1 scalar rejection 257 times running is impossible")
}
#[derive(Clone)]
pub struct GroupKey {
keys: Keys,
conv_key: ConversationKey,
}
impl GroupKey {
fn derive(label: &str, secret: &[u8], id32: &[u8; 32], epoch: Option<u64>) -> Self {
let info = build_info(label, id32, epoch);
let sk = hkdf_to_secret_key(secret, &info);
let keys = Keys::new(sk);
let conv_key = ConversationKey::derive(keys.secret_key(), &keys.public_key())
.expect("self-ECDH of a valid keypair cannot fail");
GroupKey { keys, conv_key }
}
pub fn pk(&self) -> PublicKey {
self.keys.public_key()
}
pub fn pk_hex(&self) -> String {
self.keys.public_key().to_hex()
}
pub fn keys(&self) -> &Keys {
&self.keys
}
pub fn conv_key(&self) -> &ConversationKey {
&self.conv_key
}
}
impl std::fmt::Debug for GroupKey {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
f.debug_struct("GroupKey").field("pk", &self.pk_hex()).finish()
}
}
pub fn channel_group_key(secret: &[u8; 32], channel_id: &ChannelId, epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_CHANNEL, secret, &channel_id.0, Some(epoch.0))
}
pub fn control_group_key(community_root: &[u8; 32], community_id: &CommunityId, epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_CONTROL, community_root, &community_id.0, Some(epoch.0))
}
pub fn guestbook_group_key(community_root: &[u8; 32], community_id: &CommunityId, epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_GUESTBOOK, community_root, &community_id.0, Some(epoch.0))
}
pub fn channel_rekey_group_key(root: &[u8; 32], channel_id: &ChannelId, new_epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_REKEY_PSEUDONYM, root, &channel_id.0, Some(new_epoch.0))
}
pub fn base_rekey_group_key(prior_root: &[u8; 32], community_id: &CommunityId, new_epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_BASE_REKEY_PSEUDONYM, prior_root, &community_id.0, Some(new_epoch.0))
}
pub fn dissolved_group_key(community_id: &CommunityId) -> GroupKey {
GroupKey::derive(LABEL_DISSOLVED, &community_id.0, &ZERO32, None)
}
pub fn voice_group_key(secret: &[u8; 32], channel_id: &ChannelId, epoch: Epoch) -> GroupKey {
GroupKey::derive(LABEL_VOICE_SIGNER, secret, &channel_id.0, Some(epoch.0))
}
pub fn voice_media_key(secret: &[u8; 32], channel_id: &ChannelId, epoch: Epoch) -> [u8; 32] {
hkdf32(secret, &build_info(LABEL_VOICE_MEDIA, &channel_id.0, Some(epoch.0)))
}
pub fn voice_sender_key(media_key: &[u8; 32], identity: &str) -> [u8; 32] {
let id: [u8; 32] = Sha256::digest(identity.as_bytes()).into();
hkdf32(media_key, &build_info(LABEL_VOICE_SENDER, &id, None))
}
pub fn grant_locator(community_id: &CommunityId, member_xonly: &[u8; 32]) -> [u8; 32] {
hkdf32(&community_id.0, &build_info(LABEL_GRANT, member_xonly, None))
}
pub fn banlist_locator(community_id: &CommunityId) -> [u8; 32] {
hkdf32(&community_id.0, &build_info(LABEL_BANLIST, &ZERO32, None))
}
pub fn invite_links_locator(community_id: &CommunityId, creator_xonly: &[u8; 32]) -> [u8; 32] {
hkdf32(&community_id.0, &build_info(LABEL_INVITE_LINKS, creator_xonly, None))
}
pub fn recipient_locator(
rotator_xonly: &[u8; 32],
recipient_xonly: &[u8; 32],
scope_id: &[u8; 32],
new_epoch: Epoch,
) -> [u8; 32] {
let mut ikm = [0u8; 64];
ikm[..32].copy_from_slice(rotator_xonly);
ikm[32..].copy_from_slice(recipient_xonly);
hkdf32(&ikm, &build_info(LABEL_RECIPIENT_PSEUDONYM, scope_id, Some(new_epoch.0)))
}
pub fn invite_bundle_key(token: &[u8; TOKEN_LEN]) -> [u8; 32] {
hkdf32(token, &build_info(LABEL_INVITE_KEY, &ZERO32, None))
}
pub fn community_id_of(owner_xonly: &[u8; 32], owner_salt: &[u8; 32]) -> CommunityId {
let mut h = Sha256::new();
h.update(LABEL_COMMUNITY.as_bytes());
h.update(owner_xonly);
h.update(owner_salt);
CommunityId(h.finalize().into())
}
pub fn verify_community_id(community_id: &CommunityId, owner_xonly: &[u8; 32], owner_salt: &[u8; 32]) -> bool {
community_id_of(owner_xonly, owner_salt) == *community_id
}
pub fn epoch_key_commitment(prev_epoch: Epoch, prev_key: &[u8; 32]) -> [u8; 32] {
let mut h = Sha256::new();
h.update(LABEL_EPOCH_COMMITMENT.as_bytes());
h.update(prev_epoch.0.to_be_bytes());
h.update(prev_key);
h.finalize().into()
}
#[cfg(test)]
mod tests {
use super::*;
fn secret() -> [u8; 32] {
let mut k = [0u8; 32];
for (i, b) in k.iter_mut().enumerate() {
*b = i as u8;
}
k
}
fn id32() -> [u8; 32] {
let mut id = [0u8; 32];
for (i, b) in id.iter_mut().enumerate() {
*b = (255 - i) as u8;
}
id
}
fn alt() -> [u8; 32] {
[0x11u8; 32]
}
fn cid() -> CommunityId {
CommunityId(id32())
}
fn chan() -> ChannelId {
ChannelId(id32())
}
const EPOCH_MULTI: u64 = 0x0102030405060708;
const GOLDEN_CHANNEL_E0_SEED: &str = "1a99a5958bf9fcc5336e6e19db42aabf36ffbfa12f38a1d5fbde2ae383ed751b";
const GOLDEN_CHANNEL_E0_PK: &str = "7a5c5dff759a63f1fc2779864487432bae3d1ea72c4ffabd39f4c1fdaf62097a";
const GOLDEN_CHANNEL_EMULTI_PK: &str = "f20c7d192cc87615d7341e86f38f85303f4708b40232d4fea521ab8217767391";
const GOLDEN_CONTROL_E0_PK: &str = "c43df20bf4d6eeaea5149619662ffe9b211f31e11bb4a59f56b6e906f702d46f";
const GOLDEN_GUESTBOOK_E0_PK: &str = "ad09de582026fa7a052db18bb5827fa24c15e929d59aadcc91efb8508f5368ad";
const GOLDEN_CHANNEL_REKEY_E1_PK: &str = "7c55cdb957e9db2b4800d687b2a07d3f7066b1a35824a1e86ba871f55e87e8b5";
const GOLDEN_BASE_REKEY_E1_PK: &str = "fb2fa44fba66ba15595f784255a1cb569531db8784432ac0e4fe838498dd9dea";
const GOLDEN_DISSOLVED_PK: &str = "4d3d55d88fdf9d9c2089651e5cbb0dfa93b6b9b10cdcb2319b0dce1a1398096a";
const GOLDEN_GRANT_LOCATOR: &str = "fd2f88cc7f1eb8d7d862c91dc22afe700c358d1845158b3f353b769ce4898e35";
const GOLDEN_BANLIST_LOCATOR: &str = "88089214afae6d3c412fd817ada44d6df4d485a53565646471e74476397693c9";
const GOLDEN_INVITE_LINKS_LOCATOR: &str = "f4ae29994165767bac23e8dce630f81b926d2c8aa150e5cbf0bdf75865e8379a";
const GOLDEN_RECIPIENT_LOCATOR: &str = "342deb400e191f0f52c81f27600934552550beb85aa9bf169f02d0e7f826cf74";
const GOLDEN_INVITE_KEY: &str = "94bf8b0d89e579ddaeccf8d9db3f5de5c86a1259c597f2560ff0120173bc5e1f";
const GOLDEN_VOICE_MEDIA_E0: &str = "8ab5b935c5e17f156563860ae6263f3700bfd836c326f8b3d7082be2fbaef6a0";
const GOLDEN_VOICE_SIGNER_E0_PK: &str = "7591f1306c265ee1dce6a07b72c76fadb3af13bf9ccce0d284cb3af6134211a1";
const GOLDEN_VOICE_SENDER: &str = "9ce1c11a39ce16a84b72c2697724a39e5c41ec07f4d703dcb01241271837599e";
const GOLDEN_COMMUNITY_ID: &str = "2b790bd59df98bdc52092b74ebd6933a89ef8eaeecc9030861cbdeae7c814c46";
const GOLDEN_EPOCH_COMMITMENT: &str = "3e6d6a3c9973c16d1ca7c5602d36979927c55c21a7e2c840f883af3f047e80a4";
fn hex(bytes: &[u8]) -> String {
crate::simd::hex::bytes_to_hex_32(bytes.try_into().expect("32 bytes"))
}
#[test]
fn channel_group_key_golden_vector() {
let gk = channel_group_key(&secret(), &chan(), Epoch(0));
assert_eq!(hex(gk.keys().secret_key().as_secret_bytes()), GOLDEN_CHANNEL_E0_SEED);
assert_eq!(gk.pk_hex(), GOLDEN_CHANNEL_E0_PK);
}
#[test]
fn channel_group_key_golden_multibyte_epoch_is_big_endian() {
let gk = channel_group_key(&secret(), &chan(), Epoch(EPOCH_MULTI));
assert_eq!(gk.pk_hex(), GOLDEN_CHANNEL_EMULTI_PK);
}
#[test]
fn control_group_key_golden_vector() {
assert_eq!(control_group_key(&secret(), &cid(), Epoch(0)).pk_hex(), GOLDEN_CONTROL_E0_PK);
}
#[test]
fn guestbook_group_key_golden_vector() {
assert_eq!(guestbook_group_key(&secret(), &cid(), Epoch(0)).pk_hex(), GOLDEN_GUESTBOOK_E0_PK);
}
#[test]
fn rekey_group_keys_golden_vectors() {
assert_eq!(
channel_rekey_group_key(&secret(), &chan(), Epoch(1)).pk_hex(),
GOLDEN_CHANNEL_REKEY_E1_PK
);
assert_eq!(
base_rekey_group_key(&secret(), &cid(), Epoch(1)).pk_hex(),
GOLDEN_BASE_REKEY_E1_PK
);
}
#[test]
fn dissolved_group_key_golden_and_is_epoch_free() {
assert_eq!(dissolved_group_key(&cid()).pk_hex(), GOLDEN_DISSOLVED_PK);
let with_epoch = GroupKey::derive(LABEL_DISSOLVED, &cid().0, &ZERO32, Some(0));
assert_ne!(with_epoch.pk_hex(), GOLDEN_DISSOLVED_PK);
}
#[test]
fn locator_golden_vectors() {
assert_eq!(hex(&grant_locator(&cid(), &alt())), GOLDEN_GRANT_LOCATOR);
assert_eq!(hex(&banlist_locator(&cid())), GOLDEN_BANLIST_LOCATOR);
assert_eq!(hex(&invite_links_locator(&cid(), &alt())), GOLDEN_INVITE_LINKS_LOCATOR);
assert_eq!(
hex(&recipient_locator(&secret(), &alt(), &id32(), Epoch(3))),
GOLDEN_RECIPIENT_LOCATOR
);
assert_eq!(hex(&invite_bundle_key(&[0x07u8; TOKEN_LEN])), GOLDEN_INVITE_KEY);
}
#[test]
fn voice_golden_vectors() {
let media = voice_media_key(&secret(), &chan(), Epoch(0));
assert_eq!(hex(&media), GOLDEN_VOICE_MEDIA_E0);
assert_eq!(voice_group_key(&secret(), &chan(), Epoch(0)).pk_hex(), GOLDEN_VOICE_SIGNER_E0_PK);
assert_eq!(
hex(&voice_sender_key(&media, "00112233445566778899aabbccddeeff")),
GOLDEN_VOICE_SENDER
);
}
#[test]
fn community_id_golden_and_verifies() {
let id = community_id_of(&secret(), &alt());
assert_eq!(hex(&id.0), GOLDEN_COMMUNITY_ID);
assert!(verify_community_id(&id, &secret(), &alt()));
assert!(!verify_community_id(&id, &alt(), &alt()));
assert!(!verify_community_id(&id, &secret(), &id32()));
}
#[test]
fn epoch_key_commitment_golden_and_binds_both_inputs() {
assert_eq!(hex(&epoch_key_commitment(Epoch(2), &secret())), GOLDEN_EPOCH_COMMITMENT);
assert_ne!(hex(&epoch_key_commitment(Epoch(3), &secret())), GOLDEN_EPOCH_COMMITMENT);
assert_ne!(hex(&epoch_key_commitment(Epoch(2), &alt())), GOLDEN_EPOCH_COMMITMENT);
}
#[test]
fn labels_domain_separate_every_plane() {
let pks = [
channel_group_key(&secret(), &chan(), Epoch(0)).pk_hex(),
control_group_key(&secret(), &cid(), Epoch(0)).pk_hex(),
guestbook_group_key(&secret(), &cid(), Epoch(0)).pk_hex(),
channel_rekey_group_key(&secret(), &chan(), Epoch(0)).pk_hex(),
base_rekey_group_key(&secret(), &cid(), Epoch(0)).pk_hex(),
voice_group_key(&secret(), &chan(), Epoch(0)).pk_hex(),
];
let unique: std::collections::HashSet<_> = pks.iter().collect();
assert_eq!(unique.len(), pks.len(), "two labels collided on one address");
}
#[test]
fn epoch_rotates_every_keyed_address() {
assert_ne!(
channel_group_key(&secret(), &chan(), Epoch(0)).pk_hex(),
channel_group_key(&secret(), &chan(), Epoch(1)).pk_hex()
);
assert_ne!(
control_group_key(&secret(), &cid(), Epoch(0)).pk_hex(),
control_group_key(&secret(), &cid(), Epoch(1)).pk_hex()
);
assert_ne!(
guestbook_group_key(&secret(), &cid(), Epoch(0)).pk_hex(),
guestbook_group_key(&secret(), &cid(), Epoch(1)).pk_hex()
);
}
#[test]
fn recipient_locator_binds_direction_scope_and_epoch() {
let base = recipient_locator(&secret(), &alt(), &id32(), Epoch(1));
assert_ne!(recipient_locator(&alt(), &secret(), &id32(), Epoch(1)), base);
assert_ne!(recipient_locator(&secret(), &alt(), &id32(), Epoch(2)), base);
assert_ne!(recipient_locator(&secret(), &alt(), &ZERO32, Epoch(1)), base);
}
#[test]
fn conv_key_is_deterministic_self_ecdh() {
let a = channel_group_key(&secret(), &chan(), Epoch(0));
let b = channel_group_key(&secret(), &chan(), Epoch(0));
assert_eq!(a.conv_key().as_bytes(), b.conv_key().as_bytes());
}
}