ic_host_artifacts/artifact/
mod.rs

1//! Bounded streams and raw SHA-256 identities for artifact bytes.
2//!
3//! Consumers own trusted paths, admitted digests, and byte limits. An identity
4//! describes the bytes read; it does not freeze a path for later execution or
5//! certify an executable's version. Copying writes only to a caller-owned sink;
6//! paths, confinement, synchronization, publication and cleanup remain local.
7
8use sha2::{Digest, Sha256};
9use std::{
10    fmt,
11    io::{self, Read},
12    str::FromStr,
13};
14
15mod copy;
16#[cfg(feature = "gzip")]
17mod gzip;
18mod hashing;
19mod matching;
20#[cfg(test)]
21mod tests;
22mod writer;
23
24pub use copy::{CopyError, copy_reader};
25#[cfg(feature = "gzip")]
26pub use gzip::{GzipError, decode_gzip, encode_gzip};
27pub use hashing::HashingWriter;
28pub use matching::MatchingWriter;
29pub use writer::{BoundedWriter, WriterError};
30
31/// Raw SHA-256 identity, without a product-specific prefix or wire format.
32#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
33pub struct Sha256Digest([u8; 32]);
34
35impl Sha256Digest {
36    /// Construct an identity from exact digest bytes.
37    #[must_use]
38    pub const fn from_bytes(bytes: [u8; 32]) -> Self {
39        Self(bytes)
40    }
41
42    /// Borrow the digest bytes.
43    #[must_use]
44    pub const fn as_bytes(&self) -> &[u8; 32] {
45        &self.0
46    }
47
48    /// Hash bytes already held by the caller.
49    #[must_use]
50    pub fn compute(bytes: &[u8]) -> Self {
51        Self(Sha256::digest(bytes).into())
52    }
53}
54
55impl fmt::Display for Sha256Digest {
56    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
57        for byte in self.0 {
58            write!(f, "{byte:02x}")?;
59        }
60        Ok(())
61    }
62}
63
64/// Invalid lowercase hexadecimal digest authority.
65#[derive(Clone, Copy, Debug, Eq, PartialEq)]
66pub enum DigestParseError {
67    /// The digest does not contain exactly 64 ASCII bytes.
68    Length {
69        /// Observed byte length.
70        actual: usize,
71    },
72    /// A byte is not a lowercase hexadecimal digit.
73    Digit {
74        /// Offset of the invalid byte.
75        offset: usize,
76    },
77}
78
79impl fmt::Display for DigestParseError {
80    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
81        match self {
82            Self::Length { actual } => {
83                write!(f, "SHA-256 requires 64 hex bytes, received {actual}")
84            }
85            Self::Digit { offset } => write!(f, "invalid lowercase SHA-256 digit at byte {offset}"),
86        }
87    }
88}
89impl std::error::Error for DigestParseError {}
90
91impl FromStr for Sha256Digest {
92    type Err = DigestParseError;
93
94    fn from_str(text: &str) -> Result<Self, Self::Err> {
95        if text.len() != 64 {
96            return Err(DigestParseError::Length { actual: text.len() });
97        }
98        let mut bytes = [0; 32];
99        for (offset, digit) in text.bytes().enumerate() {
100            let nibble = match digit {
101                b'0'..=b'9' => digit - b'0',
102                b'a'..=b'f' => digit - b'a' + 10,
103                _ => return Err(DigestParseError::Digit { offset }),
104            };
105            bytes[offset / 2] |= nibble << if offset % 2 == 0 { 4 } else { 0 };
106        }
107        Ok(Self(bytes))
108    }
109}
110
111/// Exact size and digest of one bounded byte stream.
112#[derive(Clone, Copy, Debug, Eq, PartialEq)]
113pub struct ArtifactIdentity {
114    /// Number of bytes successfully read.
115    pub bytes: u64,
116    /// SHA-256 of those bytes.
117    pub sha256: Sha256Digest,
118}
119
120/// A bounded read or digest verification failed.
121#[derive(Debug)]
122pub enum ArtifactError {
123    /// The reader or filesystem failed. No partial identity is returned.
124    Io(io::Error),
125    /// The selected path is not a regular file.
126    NotRegularFile,
127    /// A stream exceeded the caller's byte allowance.
128    LimitExceeded {
129        /// Maximum permitted bytes.
130        limit: u64,
131    },
132    /// The complete bounded stream does not match the admitted digest.
133    DigestMismatch {
134        /// Caller-selected authority.
135        expected: Sha256Digest,
136        /// Observed size and digest.
137        actual: ArtifactIdentity,
138    },
139    /// Storage could not be allocated for a bounded file read.
140    Allocation(std::collections::TryReserveError),
141}
142
143impl fmt::Display for ArtifactError {
144    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
145        match self {
146            Self::Io(source) => write!(f, "artifact read failed: {source}"),
147            Self::NotRegularFile => f.write_str("artifact path is not a regular file"),
148            Self::LimitExceeded { limit } => write!(f, "artifact exceeds {limit} bytes"),
149            Self::DigestMismatch { expected, actual } => write!(
150                f,
151                "artifact SHA-256 is {}, expected {expected}",
152                actual.sha256
153            ),
154            Self::Allocation(source) => write!(f, "artifact allocation failed: {source}"),
155        }
156    }
157}
158impl std::error::Error for ArtifactError {
159    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
160        match self {
161            Self::Io(source) => Some(source),
162            Self::Allocation(source) => Some(source),
163            _ => None,
164        }
165    }
166}
167impl From<io::Error> for ArtifactError {
168    fn from(source: io::Error) -> Self {
169        Self::Io(source)
170    }
171}
172
173/// Preserve native I/O identity or retain the typed artifact cause at an I/O boundary.
174impl From<ArtifactError> for io::Error {
175    fn from(error: ArtifactError) -> Self {
176        match error {
177            ArtifactError::Io(source) => source,
178            other => Self::other(other),
179        }
180    }
181}
182
183/// Hash one stream with constant working memory and no reliance on metadata.
184///
185/// Reads at most `max_bytes + 1` bytes, using the extra byte to detect overflow.
186/// Retries interrupted reads only; no subprocess or network retry is performed.
187/// A blocking reader's timeouts remain the caller's responsibility.
188///
189/// # Errors
190/// Returns [`ArtifactError::LimitExceeded`] or the underlying read error.
191/// An impossible reader byte count returns IO [`io::ErrorKind::InvalidData`].
192pub fn hash_reader(
193    mut reader: impl Read,
194    max_bytes: u64,
195) -> Result<ArtifactIdentity, ArtifactError> {
196    let mut hasher = Sha256::new();
197    let bytes = visit_reader::<ArtifactError>(&mut reader, max_bytes, |chunk| {
198        hasher.update(chunk);
199        Ok(())
200    })?;
201    Ok(ArtifactIdentity {
202        bytes,
203        sha256: Sha256Digest(hasher.finalize().into()),
204    })
205}
206
207/// Verify a complete bounded stream before the caller extracts or uses its bytes.
208///
209/// # Errors
210/// Returns read/limit failures or a typed mismatch with the observed identity.
211pub fn verify_reader(
212    reader: impl Read,
213    max_bytes: u64,
214    expected: Sha256Digest,
215) -> Result<ArtifactIdentity, ArtifactError> {
216    let actual = hash_reader(reader, max_bytes)?;
217    if actual.sha256 != expected {
218        return Err(ArtifactError::DigestMismatch { expected, actual });
219    }
220    Ok(actual)
221}
222
223/// Read one stream into bounded, fallibly allocated storage.
224///
225/// Observes at most `max_bytes + 1` bytes to detect overflow. A blocking reader's
226/// deadline remains caller-owned. Only interrupted reads are retried.
227///
228/// # Errors
229/// Returns read, allocation, or byte-limit failures. An impossible reader byte
230/// count returns IO [`io::ErrorKind::InvalidData`].
231pub fn read_reader(mut reader: impl Read, max_bytes: usize) -> Result<Vec<u8>, ArtifactError> {
232    let mut bytes = Vec::new();
233    visit_reader::<ArtifactError>(&mut reader, max_bytes as u64, |chunk| {
234        bytes
235            .try_reserve_exact(chunk.len())
236            .map_err(ArtifactError::Allocation)?;
237        bytes.extend_from_slice(chunk);
238        Ok(())
239    })?;
240    Ok(bytes)
241}
242
243fn visit_reader<E: From<ArtifactError>>(
244    reader: &mut impl Read,
245    limit: u64,
246    mut visit: impl FnMut(&[u8]) -> Result<(), E>,
247) -> Result<u64, E> {
248    let mut bytes = 0_u64;
249    let mut buffer = [0_u8; 16 * 1024];
250    loop {
251        let remaining = usize::try_from(limit - bytes).unwrap_or(usize::MAX);
252        let allowance = remaining.saturating_add(1).min(buffer.len());
253        let count = match reader.read(&mut buffer[..allowance]) {
254            Ok(count) => count,
255            Err(source) if source.kind() == io::ErrorKind::Interrupted => continue,
256            Err(source) => return Err(ArtifactError::Io(source).into()),
257        };
258        if count > allowance {
259            return Err(ArtifactError::Io(io::Error::new(
260                io::ErrorKind::InvalidData,
261                "reader returned more bytes than its buffer can hold",
262            ))
263            .into());
264        }
265        if count == 0 {
266            return Ok(bytes);
267        }
268        if count as u64 > limit - bytes {
269            return Err(ArtifactError::LimitExceeded { limit }.into());
270        }
271        visit(&buffer[..count])?;
272        bytes += count as u64;
273    }
274}
ic_host_artifacts/artifact/mod.rs

ic_host_artifacts/artifact/
mod.rs
