From e0ffb4d47a9fb52c8ea3b0d8b1d002606b0bc4ee Mon Sep 17 00:00:00 2001 From: Padraic-O-Mhuiris Date: Thu, 22 Feb 2024 15:33:29 +0000 Subject: feat(tvix/eval): implement `builtins.hashFile` The primary addition is the `builtin_hashFile` function which attempts value-path coercion and on success reads the file to a nix_string. Similar to the `builtins_hashString` implementation, the hash is then derived using the `hash_nix_string` function in the introduced hash module which has extracted hashing functionality initially specified in cl/11005. Change-Id: I5e3ef0317223af99488ebb79efc2fb49b4cbc1b6 Reviewed-on: https://cl.tvl.fyi/c/depot/+/11007 Tested-by: BuildkiteCI Reviewed-by: flokli --- tvix/eval/src/builtins/hash.rs | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 tvix/eval/src/builtins/hash.rs (limited to 'tvix/eval/src/builtins/hash.rs') diff --git a/tvix/eval/src/builtins/hash.rs b/tvix/eval/src/builtins/hash.rs new file mode 100644 index 0000000000..6d07fc9b2d --- /dev/null +++ b/tvix/eval/src/builtins/hash.rs @@ -0,0 +1,25 @@ +use bstr::ByteSlice; +use data_encoding::HEXLOWER; +use md5::Md5; +use sha1::Sha1; +use sha2::{digest::Output, Digest, Sha256, Sha512}; + +use crate::ErrorKind; + +fn hash(b: &[u8]) -> Output { + let mut hasher = D::new(); + hasher.update(b); + hasher.finalize() +} + +pub fn hash_nix_string(algo: impl AsRef<[u8]>, s: impl AsRef<[u8]>) -> Result { + match algo.as_ref() { + b"md5" => Ok(HEXLOWER.encode(hash::(s.as_ref()).as_bstr())), + b"sha1" => Ok(HEXLOWER.encode(hash::(s.as_ref()).as_bstr())), + b"sha256" => Ok(HEXLOWER.encode(hash::(s.as_ref()).as_bstr())), + b"sha512" => Ok(HEXLOWER.encode(hash::(s.as_ref()).as_bstr())), + _ => Err(ErrorKind::UnknownHashType( + algo.as_ref().as_bstr().to_string(), + )), + } +} -- cgit 1.4.1