attest.rs 93.9%

200 of 213 lines run, 23 of 28 functions · run #357 on main · run missed

//! Attestations (spec §14.4): signed statements about references and batches
//! of changes, and runs (receipts, §14.6).

use crate::cbor::{self, Value};
use crate::hash::{hash_object, ChangeHash, KeyId, ObjectRef, ObjectType};
use crate::objects::{
    array, bytes, check_version, field, map, oref, text, uint, Hlc, ObjectError, FORMAT_VERSION,
};

pub const MAX_SEAL_HASHES: usize = 4096;

#[derive(Clone, Debug, PartialEq, Eq)]
pub enum AttestBody {
    /// A bookmark moved from `old_state` to `new_state` (spec §10.2).
    Ref {
        name: String,
        old_state: ObjectRef,
        new_state: ObjectRef,
        serial: u64,
    },
    /// One signature covering a batch of changes (spec §14.4).
    Seal { hashes: Vec<ChangeHash> },
    /// An immutable named state (spec §10.1).
    Tag {
        name: String,
        state: ObjectRef,
        message: String,
    },
    /// The signing key's identity was replaced by `new` (spec §14.2).
    Rotate { new: KeyId },
    /// `key` must no longer be trusted (spec §14.2).
    Revoke { key: KeyId, reason: String },
    /// A runner ran the job of `receipt` on `state` (spec §14.6); `run`
    /// names the run, usually by URL.
    Receipt {
        receipt: ObjectRef,
        state: ObjectRef,
        run: String,
    },
    /// A person approved (or rejected) one job of one run (OxygenFlow
    /// approval gates; oxygenhub#235): the job as hashed with its sandbox,
    /// on that state, taking exactly those input blobs (a reviewed plan),
    /// for that environment. It approves nothing else.
    Approval {
        run: String,
        job: String,
        job_hash: ObjectRef,
        state: ObjectRef,
        inputs: Vec<ObjectRef>,
        environment: String,
        approve: bool,
    },
    /// A host accepted the reference attestation `attest`, at this
    /// attestation's own time, the host's clock (§14.7): the time a
    /// certificate of a short-lived key is checked at on a fetch.
    Accepted { attest: ObjectRef },
}

#[derive(Clone, Debug, PartialEq, Eq)]
pub struct Attest {
    /// Repository ID chosen at init.
    pub repo: [u8; 16],
    pub hlc: Hlc,
    pub body: AttestBody,
}

impl Attest {
    pub fn kind(&self) -> &'static str {
        match self.body {
            AttestBody::Ref { .. } => "ref",
            AttestBody::Seal { .. } => "seal",
            AttestBody::Tag { .. } => "tag",
            AttestBody::Rotate { .. } => "rotate",
            AttestBody::Revoke { .. } => "revoke",
            AttestBody::Receipt { .. } => "receipt",
            AttestBody::Approval { .. } => "approval",
            AttestBody::Accepted { .. } => "accepted",
        }
    }

    pub fn seal(repo: [u8; 16], mut hashes: Vec<ChangeHash>) -> Attest {
        hashes.sort();
        hashes.dedup();
        Attest {
            repo,
            hlc: Hlc::now(),
            body: AttestBody::Seal { hashes },
        }
    }

    pub fn to_value(&self) -> Value {
        let body = match &self.body {
            AttestBody::Ref {
                name,
                old_state,
                new_state,
                serial,
            } => Value::map([
                (0, Value::Text(name.clone())),
                (1, Value::Bytes(old_state.0.to_vec())),
                (2, Value::Bytes(new_state.0.to_vec())),
                (3, Value::Uint(*serial)),
            ]),
            AttestBody::Seal { hashes } => Value::map([(
                0,
                Value::Array(hashes.iter().map(|h| Value::Bytes(h.0.to_vec())).collect()),
            )]),
            AttestBody::Tag {
                name,
                state,
                message,
            } => Value::map([
                (0, Value::Text(name.clone())),
                (1, Value::Bytes(state.0.to_vec())),
                (2, Value::Text(message.clone())),
            ]),
            AttestBody::Rotate { new } => Value::map([(0, Value::Bytes(new.0.to_vec()))]),
            AttestBody::Revoke { key, reason } => Value::map([
                (0, Value::Bytes(key.0.to_vec())),
                (1, Value::Text(reason.clone())),
            ]),
            AttestBody::Receipt {
                receipt,
                state,
                run,
            } => Value::map([
                (0, Value::Bytes(receipt.0.to_vec())),
                (1, Value::Bytes(state.0.to_vec())),
                (2, Value::Text(run.clone())),
            ]),
            AttestBody::Accepted { attest } => Value::map([(0, Value::Bytes(attest.0.to_vec()))]),
            AttestBody::Approval {
                run,
                job,
                job_hash,
                state,
                inputs,
                environment,
                approve,
            } => Value::map([
                (0, Value::Text(run.clone())),
                (1, Value::Text(job.clone())),
                (2, Value::Bytes(job_hash.0.to_vec())),
                (3, Value::Bytes(state.0.to_vec())),
                (
                    4,
                    Value::Array(inputs.iter().map(|i| Value::Bytes(i.0.to_vec())).collect()),
                ),
                (5, Value::Text(environment.clone())),
                (6, Value::Uint(u64::from(*approve))),
            ]),
        };
        Value::map([
            (0, Value::Uint(FORMAT_VERSION)),
            (1, Value::Text(self.kind().into())),
            (2, Value::Bytes(self.repo.to_vec())),
            (3, self.hlc.to_value()),
            (4, body),
        ])
    }

    pub fn encode(&self) -> Vec<u8> {
        cbor::encode(&self.to_value())
    }

    pub fn hash(&self) -> ObjectRef {
        hash_object(ObjectType::Attest, &self.encode())
    }

    pub fn decode(b: &[u8]) -> Result<Attest, ObjectError> {
        let v = cbor::decode(b)?;
        let m = map(&v, "attest")?;
        check_version(m)?;
        let kind = text(field(m, 1, "kind")?, "kind")?;
        let repo: [u8; 16] = bytes(field(m, 2, "repo")?, "repo")?
            .try_into()
            .map_err(|_| ObjectError::Schema("repo: expected 16 bytes".into()))?;
        let hlc = Hlc::from_value(field(m, 3, "hlc")?)?;
        let b = map(field(m, 4, "body")?, "body")?;
        let body = match kind {
            "ref" => AttestBody::Ref {
                name: text(field(b, 0, "name")?, "name")?.to_owned(),
                old_state: oref(field(b, 1, "old_state")?, "old_state")?,
                new_state: oref(field(b, 2, "new_state")?, "new_state")?,
                serial: uint(field(b, 3, "serial")?, "serial")?,
            },
            "seal" => {
                let hashes = array(field(b, 0, "hashes")?, "hashes")?
                    .iter()
                    .map(|h| oref(h, "hash"))
                    .collect::<Result<Vec<_>, _>>()?;
                if hashes.is_empty()
                    || hashes.len() > MAX_SEAL_HASHES
                    || !hashes.windows(2).all(|w| w[0] < w[1])
                {
                    return Err(ObjectError::Schema(
                        "seal: hashes must be 1..4096, sorted and unique".into(),
                    ));
                }
                AttestBody::Seal { hashes }
            }
            "tag" => AttestBody::Tag {
                name: text(field(b, 0, "name")?, "name")?.to_owned(),
                state: oref(field(b, 1, "state")?, "state")?,
                message: text(field(b, 2, "message")?, "message")?.to_owned(),
            },
            "rotate" => AttestBody::Rotate {
                new: oref(field(b, 0, "new")?, "new")?,
            },
            "revoke" => AttestBody::Revoke {
                key: oref(field(b, 0, "key")?, "key")?,
                reason: text(field(b, 1, "reason")?, "reason")?.to_owned(),
            },
            "receipt" => AttestBody::Receipt {
                receipt: oref(field(b, 0, "receipt")?, "receipt")?,
                state: oref(field(b, 1, "state")?, "state")?,
                run: text(field(b, 2, "run")?, "run")?.to_owned(),
            },
            "approval" => {
                let inputs = array(field(b, 4, "inputs")?, "inputs")?
                    .iter()
                    .map(|h| oref(h, "input"))
                    .collect::<Result<Vec<_>, _>>()?;
                if !inputs.windows(2).all(|w| w[0] < w[1]) {
                    return Err(ObjectError::Schema(
                        "approval: inputs must be sorted and unique".into(),
                    ));
                }
                AttestBody::Approval {
                    run: text(field(b, 0, "run")?, "run")?.to_owned(),
                    job: text(field(b, 1, "job")?, "job")?.to_owned(),
                    job_hash: oref(field(b, 2, "job_hash")?, "job_hash")?,
                    state: oref(field(b, 3, "state")?, "state")?,
                    inputs,
                    environment: text(field(b, 5, "environment")?, "environment")?.to_owned(),
                    approve: match uint(field(b, 6, "approve")?, "approve")? {
                        0 => false,
                        1 => true,
                        _ => return Err(ObjectError::Schema("approval: approve is 0 or 1".into())),
                    },
                }
            }
            "accepted" => AttestBody::Accepted {
                attest: oref(field(b, 0, "attest")?, "attest")?,
            },
            other => {
                return Err(ObjectError::Schema(format!(
                    "unknown attestation kind {other:?}"
                )))
            }
        };
        Ok(Attest { repo, hlc, body })
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn round_trips() {
        let r = Attest {
            repo: [1; 16],
            hlc: Hlc {
                wall_ms: 3,
                logical: 0,
            },
            body: AttestBody::Ref {
                name: "main".into(),
                old_state: ObjectRef([1; 32]),
                new_state: ObjectRef([2; 32]),
                serial: 7,
            },
        };
        assert_eq!(Attest::decode(&r.encode()).unwrap(), r);
        assert_eq!(r.kind(), "ref");
        let s = Attest::seal(
            [1; 16],
            vec![ObjectRef([9; 32]), ObjectRef([3; 32]), ObjectRef([9; 32])],
        );
        let d = Attest::decode(&s.encode()).unwrap();
        assert!(
            matches!(&d.body, AttestBody::Seal { hashes } if hashes == &vec![ObjectRef([3; 32]), ObjectRef([9; 32])])
        );
        assert_ne!(s.hash(), r.hash());
        let t = Attest {
            repo: [1; 16],
            hlc: Hlc::default(),
            body: AttestBody::Tag {
                name: "v1".into(),
                state: ObjectRef([5; 32]),
                message: "release".into(),
            },
        };
        assert_eq!(Attest::decode(&t.encode()).unwrap(), t);
        assert_eq!(t.kind(), "tag");
    }

    #[test]
    fn rejects_bad_seals() {
        let empty = Attest {
            repo: [0; 16],
            hlc: Hlc::default(),
            body: AttestBody::Seal { hashes: vec![] },
        };
        assert!(Attest::decode(&empty.encode()).is_err());
        let unsorted = Attest {
            repo: [0; 16],
            hlc: Hlc::default(),
            body: AttestBody::Seal {
                hashes: vec![ObjectRef([2; 32]), ObjectRef([1; 32])],
            },
        };
        assert!(Attest::decode(&unsorted.encode()).is_err());
    }
}

`/`	Search
`g` `c`	Code
`g` `h`	History
`g` `b`	Bookmarks
`g` `t`	Trust
`y`	Permalink to this state (files and trees)
`?`	This sheet

1		//! Attestations (spec §14.4): signed statements about references and batches
2		//! of changes, and runs (receipts, §14.6).
3
4		use crate::cbor::{self, Value};
5		use crate::hash::{hash_object, ChangeHash, KeyId, ObjectRef, ObjectType};
6		use crate::objects::{
7		array, bytes, check_version, field, map, oref, text, uint, Hlc, ObjectError, FORMAT_VERSION,
8		};
9
10		pub const MAX_SEAL_HASHES: usize = 4096;
11
12		#[derive(Clone, Debug, PartialEq, Eq)]
13		pub enum AttestBody {
14		/// A bookmark moved from `old_state` to `new_state` (spec §10.2).
15		Ref {
16		name: String,
17		old_state: ObjectRef,
18		new_state: ObjectRef,
19		serial: u64,
20		},
21		/// One signature covering a batch of changes (spec §14.4).
22		Seal { hashes: Vec<ChangeHash> },
23		/// An immutable named state (spec §10.1).
24		Tag {
25		name: String,
26		state: ObjectRef,
27		message: String,
28		},
29		/// The signing key's identity was replaced by `new` (spec §14.2).
30		Rotate { new: KeyId },
31		/// `key` must no longer be trusted (spec §14.2).
32		Revoke { key: KeyId, reason: String },
33		/// A runner ran the job of `receipt` on `state` (spec §14.6); `run`
34		/// names the run, usually by URL.
35		Receipt {
36		receipt: ObjectRef,
37		state: ObjectRef,
38		run: String,
39		},
40		/// A person approved (or rejected) one job of one run (OxygenFlow
41		/// approval gates; oxygenhub#235): the job as hashed with its sandbox,
42		/// on that state, taking exactly those input blobs (a reviewed plan),
43		/// for that environment. It approves nothing else.
44		Approval {
45		run: String,
46		job: String,
47		job_hash: ObjectRef,
48		state: ObjectRef,
49		inputs: Vec<ObjectRef>,
50		environment: String,
51		approve: bool,
52		},
53		/// A host accepted the reference attestation `attest`, at this
54		/// attestation's own time, the host's clock (§14.7): the time a
55		/// certificate of a short-lived key is checked at on a fetch.
56		Accepted { attest: ObjectRef },
57		}
58
59		#[derive(Clone, Debug, PartialEq, Eq)]
60		pub struct Attest {
61		/// Repository ID chosen at init.
62		pub repo: [u8; 16],
63		pub hlc: Hlc,
64		pub body: AttestBody,
65		}
66
67		impl Attest {
68	902×	pub fn kind(&self) -> &'static str {
69	902×	match self.body {
70	612×	AttestBody::Ref { .. } => "ref",
71	157×	AttestBody::Seal { .. } => "seal",
72	50×	AttestBody::Tag { .. } => "tag",
73	17×	AttestBody::Rotate { .. } => "rotate",
74	4×	AttestBody::Revoke { .. } => "revoke",
75	41×	AttestBody::Receipt { .. } => "receipt",
76	12×	AttestBody::Approval { .. } => "approval",
77	9×	AttestBody::Accepted { .. } => "accepted",
78		}
79	902×	}
80
81	49×	pub fn seal(repo: [u8; 16], mut hashes: Vec<ChangeHash>) -> Attest {
82	49×	hashes.sort();
83	49×	hashes.dedup();
84	49×	Attest {
85	49×	repo,
86	49×	hlc: Hlc::now(),
87	49×	body: AttestBody::Seal { hashes },
88	49×	}
89	49×	}
90
91	900×	pub fn to_value(&self) -> Value {
92	900×	let body = match &self.body {
93		AttestBody::Ref {
94	611×	name,
95	611×	old_state,
96	611×	new_state,
97	611×	serial,
98	611×	} => Value::map([
99	611×	(0, Value::Text(name.clone())),
100	611×	(1, Value::Bytes(old_state.0.to_vec())),
101	611×	(2, Value::Bytes(new_state.0.to_vec())),
102	611×	(3, Value::Uint(*serial)),
103	611×	]),
104	157×	AttestBody::Seal { hashes } => Value::map([(
105		0,
106	168×	Value::Array(hashes.iter().map(\|h\| Value::Bytes(h.0.to_vec())).collect()),
107		)]),
108		AttestBody::Tag {
109	49×	name,
110	49×	state,
111	49×	message,
112	49×	} => Value::map([
113	49×	(0, Value::Text(name.clone())),
114	49×	(1, Value::Bytes(state.0.to_vec())),
115	49×	(2, Value::Text(message.clone())),
116	49×	]),
117	17×	AttestBody::Rotate { new } => Value::map([(0, Value::Bytes(new.0.to_vec()))]),
118	4×	AttestBody::Revoke { key, reason } => Value::map([
119	4×	(0, Value::Bytes(key.0.to_vec())),
120	4×	(1, Value::Text(reason.clone())),
121	4×	]),
122		AttestBody::Receipt {
123	41×	receipt,
124	41×	state,
125	41×	run,
126	41×	} => Value::map([
127	41×	(0, Value::Bytes(receipt.0.to_vec())),
128	41×	(1, Value::Bytes(state.0.to_vec())),
129	41×	(2, Value::Text(run.clone())),
130	41×	]),
131	9×	AttestBody::Accepted { attest } => Value::map([(0, Value::Bytes(attest.0.to_vec()))]),
132		AttestBody::Approval {
133	12×	run,
134	12×	job,
135	12×	job_hash,
136	12×	state,
137	12×	inputs,
138	12×	environment,
139	12×	approve,
140	12×	} => Value::map([
141	12×	(0, Value::Text(run.clone())),
142	12×	(1, Value::Text(job.clone())),
143	12×	(2, Value::Bytes(job_hash.0.to_vec())),
144	12×	(3, Value::Bytes(state.0.to_vec())),
145		(
146		4,
147	18×	Value::Array(inputs.iter().map(\|i\| Value::Bytes(i.0.to_vec())).collect()),
148		),
149	12×	(5, Value::Text(environment.clone())),
150	12×	(6, Value::Uint(u64::from(*approve))),
151		]),
152		};
153	900×	Value::map([
154	900×	(0, Value::Uint(FORMAT_VERSION)),
155	900×	(1, Value::Text(self.kind().into())),
156	900×	(2, Value::Bytes(self.repo.to_vec())),
157	900×	(3, self.hlc.to_value()),
158	900×	(4, body),
159	900×	])
160	900×	}
161
162	900×	pub fn encode(&self) -> Vec<u8> {
163	900×	cbor::encode(&self.to_value())
164	900×	}
165
166	425×	pub fn hash(&self) -> ObjectRef {
167	425×	hash_object(ObjectType::Attest, &self.encode())
168	425×	}
169
170	967×	pub fn decode(b: &[u8]) -> Result<Attest, ObjectError> {
171	967×	let v = cbor::decode(b)?;
172	967×	let m = map(&v, "attest")?;
173	967×	check_version(m)?;
174	967×	let kind = text(field(m, 1, "kind")?, "kind")?;
175	967×	let repo: [u8; 16] = bytes(field(m, 2, "repo")?, "repo")?
176	967×	.try_into()
177	967×	.map_err(\|_\| ObjectError::Schema("repo: expected 16 bytes".into()))?;
178	967×	let hlc = Hlc::from_value(field(m, 3, "hlc")?)?;
179	967×	let b = map(field(m, 4, "body")?, "body")?;
180	967×	let body = match kind {
181	967×	"ref" => AttestBody::Ref {
182	422×	name: text(field(b, 0, "name")?, "name")?.to_owned(),
183	422×	old_state: oref(field(b, 1, "old_state")?, "old_state")?,
184	422×	new_state: oref(field(b, 2, "new_state")?, "new_state")?,
185	422×	serial: uint(field(b, 3, "serial")?, "serial")?,
186		},
187	545×	"seal" => {
188	368×	let hashes = array(field(b, 0, "hashes")?, "hashes")?
189	368×	.iter()
190	404×	.map(\|h\| oref(h, "hash"))
191	368×	.collect::<Result<Vec<_>, _>>()?;
192	368×	if hashes.is_empty()
193	367×	\|\| hashes.len() > MAX_SEAL_HASHES
194	367×	\|\| !hashes.windows(2).all(\|w\| w[0] < w[1])
195		{
196	2×	return Err(ObjectError::Schema(
197	2×	"seal: hashes must be 1..4096, sorted and unique".into(),
198	2×	));
199	366×	}
200	366×	AttestBody::Seal { hashes }
201		}
202	177×	"tag" => AttestBody::Tag {
203	143×	name: text(field(b, 0, "name")?, "name")?.to_owned(),
204	143×	state: oref(field(b, 1, "state")?, "state")?,
205	143×	message: text(field(b, 2, "message")?, "message")?.to_owned(),
206		},
207	34×	"rotate" => AttestBody::Rotate {
208	0×	new: oref(field(b, 0, "new")?, "new")?,
209		},
210	34×	"revoke" => AttestBody::Revoke {
211	0×	key: oref(field(b, 0, "key")?, "key")?,
212	0×	reason: text(field(b, 1, "reason")?, "reason")?.to_owned(),
213		},
214	34×	"receipt" => AttestBody::Receipt {
215	21×	receipt: oref(field(b, 0, "receipt")?, "receipt")?,
216	21×	state: oref(field(b, 1, "state")?, "state")?,
217	21×	run: text(field(b, 2, "run")?, "run")?.to_owned(),
218		},
219	13×	"approval" => {
220	3×	let inputs = array(field(b, 4, "inputs")?, "inputs")?
221	3×	.iter()
222	4×	.map(\|h\| oref(h, "input"))
223	3×	.collect::<Result<Vec<_>, _>>()?;
224	3×	if !inputs.windows(2).all(\|w\| w[0] < w[1]) {
225	0×	return Err(ObjectError::Schema(
226	0×	"approval: inputs must be sorted and unique".into(),
227	0×	));
228	3×	}
229		AttestBody::Approval {
230	3×	run: text(field(b, 0, "run")?, "run")?.to_owned(),
231	3×	job: text(field(b, 1, "job")?, "job")?.to_owned(),
232	3×	job_hash: oref(field(b, 2, "job_hash")?, "job_hash")?,
233	3×	state: oref(field(b, 3, "state")?, "state")?,
234	3×	inputs,
235	3×	environment: text(field(b, 5, "environment")?, "environment")?.to_owned(),
236	3×	approve: match uint(field(b, 6, "approve")?, "approve")? {
237	0×	0 => false,
238	3×	1 => true,
239	0×	_ => return Err(ObjectError::Schema("approval: approve is 0 or 1".into())),
240		},
241		}
242		}
243	10×	"accepted" => AttestBody::Accepted {
244	10×	attest: oref(field(b, 0, "attest")?, "attest")?,
245		},
246	0×	other => {
247	0×	return Err(ObjectError::Schema(format!(
248	0×	"unknown attestation kind {other:?}"
249	0×	)))
250		}
251		};
252	965×	Ok(Attest { repo, hlc, body })
253	967×	}
254		}
255
256		#[cfg(test)]
257		mod tests {
258		use super::*;
259
260		#[test]
261	1×	fn round_trips() {
262	1×	let r = Attest {
263	1×	repo: [1; 16],
264	1×	hlc: Hlc {
265	1×	wall_ms: 3,
266	1×	logical: 0,
267	1×	},
268	1×	body: AttestBody::Ref {
269	1×	name: "main".into(),
270	1×	old_state: ObjectRef([1; 32]),
271	1×	new_state: ObjectRef([2; 32]),
272	1×	serial: 7,
273	1×	},
274	1×	};
275	1×	assert_eq!(Attest::decode(&r.encode()).unwrap(), r);
276	1×	assert_eq!(r.kind(), "ref");
277	1×	let s = Attest::seal(
278	1×	[1; 16],
279	1×	vec![ObjectRef([9; 32]), ObjectRef([3; 32]), ObjectRef([9; 32])],
280		);
281	1×	let d = Attest::decode(&s.encode()).unwrap();
282	1×	assert!(
283	1×	matches!(&d.body, AttestBody::Seal { hashes } if hashes == &vec![ObjectRef([3; 32]), ObjectRef([9; 32])])
284		);
285	1×	assert_ne!(s.hash(), r.hash());
286	1×	let t = Attest {
287	1×	repo: [1; 16],
288	1×	hlc: Hlc::default(),
289	1×	body: AttestBody::Tag {
290	1×	name: "v1".into(),
291	1×	state: ObjectRef([5; 32]),
292	1×	message: "release".into(),
293	1×	},
294	1×	};
295	1×	assert_eq!(Attest::decode(&t.encode()).unwrap(), t);
296	1×	assert_eq!(t.kind(), "tag");
297	1×	}
298
299		#[test]
300	1×	fn rejects_bad_seals() {
301	1×	let empty = Attest {
302	1×	repo: [0; 16],
303	1×	hlc: Hlc::default(),
304	1×	body: AttestBody::Seal { hashes: vec![] },
305	1×	};
306	1×	assert!(Attest::decode(&empty.encode()).is_err());
307	1×	let unsorted = Attest {
308	1×	repo: [0; 16],
309	1×	hlc: Hlc::default(),
310	1×	body: AttestBody::Seal {
311	1×	hashes: vec![ObjectRef([2; 32]), ObjectRef([1; 32])],
312	1×	},
313	1×	};
314	1×	assert!(Attest::decode(&unsorted.encode()).is_err());
315	1×	}
316		}

Coverage / crates/oxy-core/src/attest.rs 93.9%

Keyboard