about summary refs log tree commit diff
path: root/tvix/nix-compat/src/wire/bytes/reader/trailer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tvix/nix-compat/src/wire/bytes/reader/trailer.rs')
-rw-r--r--tvix/nix-compat/src/wire/bytes/reader/trailer.rs17
1 files changed, 17 insertions, 0 deletions
diff --git a/tvix/nix-compat/src/wire/bytes/reader/trailer.rs b/tvix/nix-compat/src/wire/bytes/reader/trailer.rs
index 958cead42d8f..61a77678080a 100644
--- a/tvix/nix-compat/src/wire/bytes/reader/trailer.rs
+++ b/tvix/nix-compat/src/wire/bytes/reader/trailer.rs
@@ -81,6 +81,12 @@ pub(crate) fn read_trailer<R: AsyncRead + Unpin, T: Tag>(
     }
 }
 
+impl<R, T: Tag> ReadTrailer<R, T> {
+    pub fn len(&self) -> u8 {
+        self.data_len
+    }
+}
+
 impl<R: AsyncRead + Unpin, T: Tag> Future for ReadTrailer<R, T> {
     type Output = io::Result<Trailer>;
 
@@ -141,6 +147,17 @@ impl<R: AsyncRead + Unpin> TrailerReader<R> {
     pub fn new(reader: R, data_len: u8) -> Self {
         Self::Reading(read_trailer(reader, data_len))
     }
+
+    pub fn len(&self) -> u8 {
+        match self {
+            TrailerReader::Reading(fut) => fut.len(),
+            &TrailerReader::Releasing {
+                off,
+                data: Trailer { data_len, .. },
+            } => data_len - off,
+            TrailerReader::Done => 0,
+        }
+    }
 }
 
 impl<R: AsyncRead + Unpin> AsyncRead for TrailerReader<R> {