Something went wrong. Try again.
A lexicon-driven AppView for ATProto. forked from gamesgamesgamesgames.games/happyview
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286use hkdf::Hkdf;use hmac::{Hmac, KeyInit, Mac};use rand::Rng;use sha2::Sha256;
use crate::error::AppError;
pub struct SignedCommit { pub ver: u32, pub hash: [u8; 32], pub ikm: [u8; 32], pub mac: [u8; 32], pub rev: String,}
pub fn build_context(space_uri: &str, author_did: &str, rev: &str, ikm: &[u8; 32]) -> Vec<u8> { let tag = b"atproto-space-v1"; let space_bytes = space_uri.as_bytes(); let author_bytes = author_did.as_bytes(); let rev_bytes = rev.as_bytes();
let mut ctx = Vec::with_capacity( tag.len() + 2 + space_bytes.len() + 2 + author_bytes.len() + 2 + rev_bytes.len() + 2 + 32, );
ctx.extend_from_slice(tag);
// TLS 1.3 variable-length encoding: big-endian uint16 length prefix ctx.extend_from_slice(&(space_bytes.len() as u16).to_be_bytes()); ctx.extend_from_slice(space_bytes);
ctx.extend_from_slice(&(author_bytes.len() as u16).to_be_bytes()); ctx.extend_from_slice(author_bytes);
ctx.extend_from_slice(&(rev_bytes.len() as u16).to_be_bytes()); ctx.extend_from_slice(rev_bytes);
ctx.extend_from_slice(&(ikm.len() as u16).to_be_bytes()); ctx.extend_from_slice(ikm);
ctx}
pub fn sign_commit( hash: &[u8; 32], space_uri: &str, author_did: &str, rev: &str,) -> Result<SignedCommit, AppError> { let mut ikm = [0u8; 32]; rand::rng().fill_bytes(&mut ikm);
let ctx = build_context(space_uri, author_did, rev, &ikm);
// mac = HMAC-SHA256(HKDF-SHA256(ikm, ctx), hash) let hk = Hkdf::<Sha256>::new(None, &ikm); let mut derived_key = [0u8; 32]; hk.expand(&ctx, &mut derived_key) .map_err(|e| AppError::Internal(format!("HKDF expand failed: {e}")))?;
let mut mac_hasher = <Hmac<Sha256> as KeyInit>::new_from_slice(&derived_key) .map_err(|e| AppError::Internal(format!("HMAC init failed: {e}")))?; mac_hasher.update(hash); let mac: [u8; 32] = mac_hasher.finalize().into_bytes().into();
Ok(SignedCommit { ver: 1, hash: *hash, ikm, mac, rev: rev.to_string(), })}
pub fn verify_commit( commit: &SignedCommit, space_uri: &str, author_did: &str,) -> Result<(), AppError> { if commit.ver != 1 { return Err(AppError::BadRequest(format!( "unsupported commit version: {}", commit.ver ))); }
let ctx = build_context(space_uri, author_did, &commit.rev, &commit.ikm);
// Recompute and verify MAC let hk = Hkdf::<Sha256>::new(None, &commit.ikm); let mut derived_key = [0u8; 32]; hk.expand(&ctx, &mut derived_key) .map_err(|e| AppError::Internal(format!("HKDF expand failed: {e}")))?;
let mut mac_hasher = <Hmac<Sha256> as KeyInit>::new_from_slice(&derived_key) .map_err(|e| AppError::Internal(format!("HMAC init failed: {e}")))?; mac_hasher.update(&commit.hash);
mac_hasher.verify_slice(&commit.mac).map_err(|_| { AppError::Auth("commit MAC verification failed — repo hash mismatch".into()) })?;
Ok(())}
#[cfg(test)]mod tests { use super::*;
#[test] fn context_string_format() { let ctx = build_context( "at://did:plc:abc/space/com.example.forum/main", "did:plc:testuser", "3k2abc", &[0xAA; 32], ); // Starts with protocol tag assert!(ctx.starts_with(b"atproto-space-v1")); }
#[test] fn context_includes_all_fields() { let space = "at://did:plc:abc/space/com.example.forum/main"; let author = "did:plc:testuser"; let rev = "3k2abc"; let ikm = [0xBB; 32]; let ctx = build_context(space, author, rev, &ikm);
// Context must contain the space URI, author, rev, and ikm assert!(ctx.windows(space.len()).any(|w| w == space.as_bytes())); assert!(ctx.windows(author.len()).any(|w| w == author.as_bytes())); assert!(ctx.windows(rev.len()).any(|w| w == rev.as_bytes())); assert!(ctx.windows(32).any(|w| w == ikm)); }
#[test] fn context_includes_author_did() { let space = "at://did:plc:abc/space/com.example.forum/main"; let author = "did:plc:user1"; let rev = "3k2abc"; let ikm = [0xBB; 32]; let ctx = build_context(space, author, rev, &ikm);
assert!(ctx.starts_with(b"atproto-space-v1")); assert!(ctx.windows(author.len()).any(|w| w == author.as_bytes()));
// Author must appear after space and before rev in the byte stream let space_pos = ctx .windows(space.len()) .position(|w| w == space.as_bytes()) .unwrap(); let author_pos = ctx .windows(author.len()) .position(|w| w == author.as_bytes()) .unwrap(); let rev_pos = ctx .windows(rev.len()) .position(|w| w == rev.as_bytes()) .unwrap(); assert!(space_pos < author_pos); assert!(author_pos < rev_pos); }
#[test] fn sign_and_verify_roundtrip() { let hash = [0xCC; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let commit = sign_commit(&hash, space, "did:plc:testuser", "3k2rev1").unwrap();
assert_eq!(commit.hash, hash); assert_eq!(commit.rev, "3k2rev1"); assert_eq!(commit.mac.len(), 32); assert_eq!(commit.ver, 1);
assert!(verify_commit(&commit, space, "did:plc:testuser").is_ok()); }
#[test] fn commit_has_version() { let hash = [0xCC; 32]; let space = "at://did:plc:abc/space/com.example.forum/main"; let commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); assert_eq!(commit.ver, 1); }
#[test] fn verify_rejects_wrong_author() { let hash = [0xAA; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let commit = sign_commit(&hash, space, "did:plc:user1", "rev1").unwrap(); assert!(verify_commit(&commit, space, "did:plc:user1").is_ok()); assert!(verify_commit(&commit, space, "did:plc:user2").is_err()); }
#[test] fn verify_rejects_tampered_hash() { let hash = [0xEE; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let mut commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); commit.hash[0] ^= 0xFF; // tamper assert!(verify_commit(&commit, space, "did:plc:testuser").is_err()); }
#[test] fn verify_rejects_tampered_ikm() { let hash = [0xEE; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let mut commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); commit.ikm[0] ^= 0xFF; // tamper — changes both the HKDF salt and the context assert!(verify_commit(&commit, space, "did:plc:testuser").is_err()); }
#[test] fn verify_rejects_tampered_rev() { let hash = [0xEE; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let mut commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); commit.rev = "rev2".into(); assert!(verify_commit(&commit, space, "did:plc:testuser").is_err()); }
#[test] fn verify_rejects_wrong_space() { let hash = [0xFF; 32];
let commit = sign_commit( &hash, "at://did:plc:abc/space/com.example.forum/main", "did:plc:user", "rev1", ) .unwrap(); assert!( verify_commit( &commit, "at://did:plc:xyz/space/com.example.forum/other", "did:plc:user", ) .is_err() ); }
#[test] fn different_ikm_per_commit() { let hash = [0xAA; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let c1 = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); let c2 = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap();
// Each call generates fresh ikm assert_ne!(c1.ikm, c2.ikm); assert_ne!(c1.mac, c2.mac); // But both verify assert!(verify_commit(&c1, space, "did:plc:testuser").is_ok()); assert!(verify_commit(&c2, space, "did:plc:testuser").is_ok()); }
#[test] fn verify_rejects_unknown_version() { let hash = [0xCC; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let mut commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); commit.ver = 2; assert!(verify_commit(&commit, space, "did:plc:testuser").is_err()); }
#[test] fn verify_rejects_tampered_mac() { let hash = [0xCC; 32]; let space = "at://did:plc:abc/space/com.example.forum/main";
let mut commit = sign_commit(&hash, space, "did:plc:testuser", "rev1").unwrap(); assert!(verify_commit(&commit, space, "did:plc:testuser").is_ok()); commit.mac[0] ^= 0xFF; assert!(verify_commit(&commit, space, "did:plc:testuser").is_err()); }}