Skip to content

Commit 0dfa0c5

Browse files
committed
fix: assert that fsspec returned the expect number of bytes
1 parent 6f9bcd6 commit 0dfa0c5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/fsspec_backend.rs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,12 @@ impl OmFileReaderBackend for FsSpecBackend {
130130
.extract::<Vec<u8>>()
131131
})
132132
.map_err(|e| OmFilesError::DecoderError(format!("Python I/O error {}", e)))?;
133+
134+
if bytes.len() != count as usize {
135+
return Err(OmFilesError::DecoderError(format!(
136+
"Obtained unexpected number of bytes from fsspec"
137+
)));
138+
}
133139
Ok(bytes)
134140
}
135141
}

0 commit comments

Comments
 (0)