fix(tvix/eval): allow reading non-UTF8 files
With our values using bstr now, we're not restricted to only reading files that contain valid UTF-8. Update our `read_to_string` function to `read_to_end` (named like `std::io::Read::read_to_end`), and have it return a Vec<u8>. Change-Id: I87f0291dc855a132689576559c891d66c30ddf2b Reviewed-on: https://cl.tvl.fyi/c/depot/+/11003 Tested-by: BuildkiteCI Autosubmit: flokli <flokli@flokli.de> Reviewed-by: Pádraic Ó Mhuiris <patrick.morris.310@gmail.com> Reviewed-by: flokli <flokli@flokli.de>
This commit is contained in:
parent
3e93efdc8c
commit
771200df7c
4 changed files with 14 additions and 14 deletions
|
@ -48,13 +48,13 @@ pub trait EvalIO {
|
||||||
/// * `builtins.pathExists :: path -> bool`
|
/// * `builtins.pathExists :: path -> bool`
|
||||||
fn path_exists(&self, path: &Path) -> io::Result<bool>;
|
fn path_exists(&self, path: &Path) -> io::Result<bool>;
|
||||||
|
|
||||||
/// Read the file at the specified path to a string.
|
/// Read the file at the specified path to a Vec<u8>.
|
||||||
///
|
///
|
||||||
/// This is used for the following language evaluation cases:
|
/// This is used for the following language evaluation cases:
|
||||||
///
|
///
|
||||||
/// * `builtins.readFile :: path -> string`
|
/// * `builtins.readFile :: path -> string`
|
||||||
/// * `builtins.import :: path -> any`
|
/// * `builtins.import :: path -> any`
|
||||||
fn read_to_string(&self, path: &Path) -> io::Result<String>;
|
fn read_to_end(&self, path: &Path) -> io::Result<Vec<u8>>;
|
||||||
|
|
||||||
/// Read the directory at the specified path and return the names
|
/// Read the directory at the specified path and return the names
|
||||||
/// of its entries associated with their [`FileType`].
|
/// of its entries associated with their [`FileType`].
|
||||||
|
@ -99,8 +99,8 @@ impl EvalIO for StdIO {
|
||||||
path.try_exists()
|
path.try_exists()
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_to_string(&self, path: &Path) -> io::Result<String> {
|
fn read_to_end(&self, path: &Path) -> io::Result<Vec<u8>> {
|
||||||
std::fs::read_to_string(path)
|
std::fs::read(path)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_dir(&self, path: &Path) -> io::Result<Vec<(bytes::Bytes, FileType)>> {
|
fn read_dir(&self, path: &Path) -> io::Result<Vec<(bytes::Bytes, FileType)>> {
|
||||||
|
@ -145,7 +145,7 @@ impl EvalIO for DummyIO {
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_to_string(&self, _: &Path) -> io::Result<String> {
|
fn read_to_end(&self, _: &Path) -> io::Result<Vec<u8>> {
|
||||||
Err(io::Error::new(
|
Err(io::Error::new(
|
||||||
io::ErrorKind::Unsupported,
|
io::ErrorKind::Unsupported,
|
||||||
"I/O methods are not implemented in DummyIO",
|
"I/O methods are not implemented in DummyIO",
|
||||||
|
|
|
@ -429,7 +429,7 @@ where
|
||||||
let content = self
|
let content = self
|
||||||
.io_handle
|
.io_handle
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.read_to_string(&path)
|
.read_to_end(&path)
|
||||||
.map_err(|e| ErrorKind::IO {
|
.map_err(|e| ErrorKind::IO {
|
||||||
path: Some(path),
|
path: Some(path),
|
||||||
error: e.into(),
|
error: e.into(),
|
||||||
|
|
|
@ -44,7 +44,7 @@ where
|
||||||
self.actual.as_ref().path_exists(path)
|
self.actual.as_ref().path_exists(path)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_to_string(&self, path: &Path) -> io::Result<String> {
|
fn read_to_end(&self, path: &Path) -> io::Result<Vec<u8>> {
|
||||||
// Bundled version of corepkgs/fetchurl.nix. The counterpart
|
// Bundled version of corepkgs/fetchurl.nix. The counterpart
|
||||||
// of this happens in [crate::configure_nix_path], where the `nix_path`
|
// of this happens in [crate::configure_nix_path], where the `nix_path`
|
||||||
// of the evaluation has `nix=/__corepkgs__` added to it.
|
// of the evaluation has `nix=/__corepkgs__` added to it.
|
||||||
|
@ -55,10 +55,10 @@ where
|
||||||
// TODO: this comparison is bad and allocates, we should use
|
// TODO: this comparison is bad and allocates, we should use
|
||||||
// the sane path library.
|
// the sane path library.
|
||||||
if path.starts_with("/__corepkgs__/fetchurl.nix") {
|
if path.starts_with("/__corepkgs__/fetchurl.nix") {
|
||||||
return Ok(include_str!("fetchurl.nix").to_string());
|
return Ok(include_bytes!("fetchurl.nix").to_vec());
|
||||||
}
|
}
|
||||||
|
|
||||||
self.actual.as_ref().read_to_string(path)
|
self.actual.as_ref().read_to_end(path)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn read_dir(&self, path: &Path) -> io::Result<Vec<(bytes::Bytes, FileType)>> {
|
fn read_dir(&self, path: &Path) -> io::Result<Vec<(bytes::Bytes, FileType)>> {
|
||||||
|
|
|
@ -371,7 +371,7 @@ impl EvalIO for TvixStoreIO {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[instrument(skip(self), err)]
|
#[instrument(skip(self), err)]
|
||||||
fn read_to_string(&self, path: &Path) -> io::Result<String> {
|
fn read_to_end(&self, path: &Path) -> io::Result<Vec<u8>> {
|
||||||
if let Ok((store_path, sub_path)) =
|
if let Ok((store_path, sub_path)) =
|
||||||
StorePath::from_absolute_path_full(&path.to_string_lossy())
|
StorePath::from_absolute_path_full(&path.to_string_lossy())
|
||||||
{
|
{
|
||||||
|
@ -416,9 +416,9 @@ impl EvalIO for TvixStoreIO {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut buf = String::new();
|
let mut buf = Vec::new();
|
||||||
|
|
||||||
reader.read_to_string(&mut buf).await?;
|
reader.read_to_end(&mut buf).await?;
|
||||||
Ok(buf)
|
Ok(buf)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
@ -430,11 +430,11 @@ impl EvalIO for TvixStoreIO {
|
||||||
} else {
|
} else {
|
||||||
// As tvix-store doesn't manage /nix/store on the filesystem,
|
// As tvix-store doesn't manage /nix/store on the filesystem,
|
||||||
// we still need to also ask self.std_io here.
|
// we still need to also ask self.std_io here.
|
||||||
self.std_io.read_to_string(path)
|
self.std_io.read_to_end(path)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// The store path is no store path, so do regular StdIO.
|
// The store path is no store path, so do regular StdIO.
|
||||||
self.std_io.read_to_string(path)
|
self.std_io.read_to_end(path)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue