ic_backup/model/artifacts/
mod.rs1use ic_host_artifacts::artifact::Sha256Digest;
4use serde::{Deserialize, Serialize, Serializer, ser::SerializeStruct};
5use thiserror::Error;
6
7#[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
9#[serde(try_from = "ChecksumFields")]
10pub struct ArtifactChecksumRecord {
11 hash: String,
12}
13
14#[derive(Deserialize)]
15#[serde(deny_unknown_fields)]
16struct ChecksumFields {
17 algorithm: String,
18 hash: String,
19}
20
21impl TryFrom<ChecksumFields> for ArtifactChecksumRecord {
22 type Error = ChecksumError;
23
24 fn try_from(value: ChecksumFields) -> Result<Self, Self::Error> {
25 if value.algorithm != "sha256" {
26 return Err(ChecksumError::UnsupportedAlgorithm(value.algorithm));
27 }
28 Self::from_hash(&value.hash)
29 }
30}
31
32impl Serialize for ArtifactChecksumRecord {
33 fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
34 let mut record = serializer.serialize_struct("ArtifactChecksumRecord", 2)?;
35 record.serialize_field("algorithm", self.algorithm())?;
36 record.serialize_field("hash", self.hash())?;
37 record.end()
38 }
39}
40
41impl ArtifactChecksumRecord {
42 #[must_use]
44 pub fn from_bytes(bytes: &[u8]) -> Self {
45 Self {
46 hash: Sha256Digest::compute(bytes).to_string(),
47 }
48 }
49
50 pub fn from_hash(hash: &str) -> Result<Self, ChecksumError> {
55 Ok(Self {
56 hash: canonical_hash(hash)?,
57 })
58 }
59
60 #[must_use]
64 pub fn from_digest(digest: [u8; 32]) -> Self {
65 Self {
66 hash: Sha256Digest::from_bytes(digest).to_string(),
67 }
68 }
69
70 #[must_use]
72 #[expect(
73 clippy::unnecessary_literal_bound,
74 reason = "preserve the public accessor signature while removing stored constant data"
75 )]
76 pub fn algorithm(&self) -> &str {
77 "sha256"
78 }
79
80 #[must_use]
82 pub fn hash(&self) -> &str {
83 &self.hash
84 }
85
86 pub fn verify(&self, expected_hash: &str) -> Result<(), ChecksumError> {
91 let expected = canonical_hash(expected_hash)?;
92 if self.hash == expected {
93 Ok(())
94 } else {
95 Err(ChecksumError::ChecksumMismatch {
96 expected,
97 actual: self.hash.clone(),
98 })
99 }
100 }
101}
102
103pub(crate) fn canonical_hash(hash: &str) -> Result<String, ChecksumError> {
104 let normalized = hash.to_ascii_lowercase();
105 normalized
106 .parse::<Sha256Digest>()
107 .map_err(|_| ChecksumError::InvalidHash(hash.to_owned()))?;
108 Ok(normalized)
109}
110
111#[derive(Debug, Error)]
113pub enum ChecksumError {
114 #[error("checksum mismatch: expected {expected}, actual {actual}")]
116 ChecksumMismatch {
117 expected: String,
119 actual: String,
121 },
122 #[error("invalid SHA-256 checksum: {0}")]
124 InvalidHash(String),
125 #[error("unsupported checksum algorithm {0}")]
127 UnsupportedAlgorithm(String),
128}
129
130#[cfg(test)]
131mod tests;