// Copyright 2024-2026 WritersLogic Contributors
// SPDX-License-Identifier: AGPL-3.0-or-later
use anyhow::{anyhow, Result};
use coset::iana::Algorithm;
use coset::{CborSerializable, CoseSign1, CoseSign1Builder, HeaderBuilder};
use ed25519_dalek::{Signer, SigningKey, Verifier, VerifyingKey};
/// Create a COSE Sign1 envelope over arbitrary payload bytes.
pub fn sign_payload(signing_key: &SigningKey, payload: &[u8]) -> Result<Vec<u8>> {
let protected = HeaderBuilder::new()
.algorithm(Algorithm::EdDSA)
.key_id(signing_key.verifying_key().to_bytes().to_vec())
.build();
let sign1 = CoseSign1Builder::new()
.protected(protected)
.payload(payload.to_vec())
.create_signature(b"", |data| signing_key.sign(data).to_bytes().to_vec())
.build();
sign1
.to_vec()
.map_err(|e| anyhow!("COSE serialization failed: {e}"))
}
/// Verify a COSE Sign1 envelope and return the payload if valid.
pub fn verify_and_extract(verifying_key: &VerifyingKey, cose_bytes: &[u8]) -> Result<Vec<u8>> {
let sign1 = CoseSign1::from_slice(cose_bytes)
.map_err(|e| anyhow!("COSE deserialization failed: {e}"))?;
let payload = sign1
.payload
.as_ref()
.ok_or_else(|| anyhow!("COSE envelope has no payload"))?;
sign1
.verify_signature(b"", |sig, data| {
let signature = ed25519_dalek::Signature::from_slice(sig)
.map_err(|e| anyhow!("invalid signature bytes: {e}"))?;
verifying_key
.verify(data, &signature)
.map_err(|e| anyhow!("signature verification failed: {e}"))
})
.map_err(|e| anyhow!("COSE verification failed: {e}"))?;
Ok(payload.clone())
}
/// Extract the key ID (verifying key bytes) from a COSE Sign1 envelope without verifying.
pub fn extract_key_id(cose_bytes: &[u8]) -> Result<Vec<u8>> {
let sign1 = CoseSign1::from_slice(cose_bytes)
.map_err(|e| anyhow!("COSE deserialization failed: {e}"))?;
let kid = sign1.protected.header.key_id.clone();
if kid.is_empty() {
return Err(anyhow!("no key ID in COSE envelope"));
}
Ok(kid)
}
/// Build a COSE Sign1 envelope for a provenance claim about a fact.
/// The payload is CBOR-encoded with fields: id, content_hash, source_uri, timestamp.
pub fn sign_fact_claim(
signing_key: &SigningKey,
fact_id: &str,
content_hash: &[u8; 32],
source_uri: Option<&str>,
timestamp_ms: u64,
) -> Result<Vec<u8>> {
let mut claim = ciborium::Value::Map(vec![
(
ciborium::Value::Text("id".to_string()),
ciborium::Value::Text(fact_id.to_string()),
),
(
ciborium::Value::Text("content_hash".to_string()),
ciborium::Value::Bytes(content_hash.to_vec()),
),
(
ciborium::Value::Text("timestamp_ms".to_string()),
ciborium::Value::Integer(timestamp_ms.into()),
),
]);
if let Some(uri) = source_uri {
if let ciborium::Value::Map(ref mut map) = claim {
map.push((
ciborium::Value::Text("source_uri".to_string()),
ciborium::Value::Text(uri.to_string()),
));
}
}
let mut payload = Vec::new();
ciborium::into_writer(&claim, &mut payload)
.map_err(|e| anyhow!("CBOR encoding failed: {e}"))?;
sign_payload(signing_key, &payload)
}
#[cfg(test)]
mod tests {
use super::*;
fn test_keypair() -> SigningKey {
SigningKey::from_bytes(&rand::random())
}
#[test]
fn sign_verify_roundtrip() {
let key = test_keypair();
let payload = b"test payload data";
let envelope = sign_payload(&key, payload).unwrap();
let recovered = verify_and_extract(&key.verifying_key(), &envelope).unwrap();
assert_eq!(recovered, payload);
}
#[test]
fn tampered_payload_rejected() {
let key = test_keypair();
let envelope = sign_payload(&key, b"original").unwrap();
let wrong_key = test_keypair();
assert!(verify_and_extract(&wrong_key.verifying_key(), &envelope).is_err());
}
#[test]
fn key_id_extraction() {
let key = test_keypair();
let envelope = sign_payload(&key, b"data").unwrap();
let kid = extract_key_id(&envelope).unwrap();
assert_eq!(kid, key.verifying_key().to_bytes().to_vec());
}
#[test]
fn fact_claim_roundtrip() {
let key = test_keypair();
let hash = [0xABu8; 32];
let envelope = sign_fact_claim(
&key,
"fact-001",
&hash,
Some("https://example.com"),
1719878400000,
)
.unwrap();
let payload = verify_and_extract(&key.verifying_key(), &envelope).unwrap();
let decoded: ciborium::Value = ciborium::from_reader(payload.as_slice()).unwrap();
if let ciborium::Value::Map(entries) = decoded {
let id_val = entries
.iter()
.find(|(k, _)| k == &ciborium::Value::Text("id".to_string()))
.unwrap();
assert_eq!(id_val.1, ciborium::Value::Text("fact-001".to_string()));
} else {
panic!("expected CBOR map");
}
}
fn unhex(s: &str) -> Vec<u8> {
(0..s.len())
.step_by(2)
.map(|i| u8::from_str_radix(&s[i..i + 2], 16).unwrap())
.collect()
}
#[test]
fn verifies_cogmem_cross_conformance_vector() {
// Cross-implementation conformance: a COSE_Sign1 EdDSA signed statement produced
// by cogmem (Python / cbor2). HMS must verify it and recover the payload, proving
// the two projects emit byte-compatible signed statements. See tests/vectors/.
let pk = unhex("4b5f52db17ebdeb555101922e89beac9b43e864086b02e4529951d7f491f0cfa");
let cose = unhex("845838a3012703706170706c69636174696f6e2f63626f720458204b5f52db17ebdeb555101922e89beac9b43e864086b02e4529951d7f491f0cfaa05887a6636973736c6469643a6b65793a74657374686d656d6f7279496468766563746f722d316a6d656d6f7279547970656472756c65656576656e7467637265617465646d73746174656d656e74486173685820000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f6b74696d657374616d704d731b0000019070bdb80058402fc837df236fa9d4d3a8bad98148ea020b4ea7bbb717d57819675eaee65264b280f5ec8d22f21dc9531f91fe4571305f84f05396cfde33ac897eac52a4937e09");
let expected = unhex("a6636973736c6469643a6b65793a74657374686d656d6f7279496468766563746f722d316a6d656d6f7279547970656472756c65656576656e7467637265617465646d73746174656d656e74486173685820000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f6b74696d657374616d704d731b0000019070bdb800");
let vk = VerifyingKey::from_bytes(&pk.try_into().unwrap()).unwrap();
let payload = verify_and_extract(&vk, &cose).unwrap();
assert_eq!(payload, expected);
}
}