Merge pull request #87 from theseus-rs/add-sha512-hash

feat: add sha2-512 support

Author: brianheineman

Cargo.lock

@@ -1,2 +1,3 @@
 pub mod registry;
 pub mod sha2_256;
+pub mod sha2_512;

postgresql_archive/src/hasher/mod.rs

@@ -1,4 +1,4 @@
-use crate::hasher::sha2_256;
+use crate::hasher::{sha2_256, sha2_512};
 use crate::Result;
 use lazy_static::lazy_static;
 use std::collections::HashMap;
@@ -60,6 +60,7 @@ impl Default for HasherRegistry {
     fn default() -> Self {
         let mut registry = Self::new();
         registry.register("sha256", sha2_256::hash);
+        registry.register("sha512", sha2_512::hash);
         registry
     }
 }
@@ -131,4 +132,17 @@ mod tests {
         );
         Ok(())
     }
+
+    #[test]
+    fn test_sha2_512() -> Result<()> {
+        let hasher = get("sha512").unwrap();
+        let data = vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 0];
+        let hash = hasher(&data)?;
+
+        assert_eq!(
+            "3ad3f36979450d4f53366244ecf1010f4f9121d6888285ff14104fd5aded85d48aa171bf1e33a112602f92b7a7088b298789012fb87b9056321241a19fb74e0b",
+            hash
+        );
+        Ok(())
+    }
 }

postgresql_archive/src/hasher/registry.rs

@@ -0,0 +1,35 @@
+use crate::Result;
+use sha2::{Digest, Sha512};
+
+/// Hashes the data using SHA2-512.
+///
+/// # Arguments
+/// * `data` - The data to hash.
+///
+/// # Returns
+/// * The hash of the data.
+///
+/// # Errors
+/// * If the data cannot be hashed.
+pub fn hash(data: &Vec<u8>) -> Result<String> {
+    let mut hasher = Sha512::new();
+    hasher.update(data);
+    let hash = hex::encode(hasher.finalize());
+    Ok(hash)
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_hash() -> Result<()> {
+        let data = vec![4, 2];
+        let hash = hash(&data)?;
+        assert_eq!(
+            "7df6418d1791a6fe80e726319f16f107534a663346f99e0d155e359a54f6c74391e2f3be19c995c3c903926d348bd86c339bd982e10f09aa776e4ff85d36387a",
+            hash
+        );
+        Ok(())
+    }
+}

postgresql_archive/src/hasher/sha2_512.rs

@@ -276,14 +276,20 @@ impl Repository for GitHub {
         );
 
         if let Some(asset_hash) = asset_hash {
+            let archive_hash = match asset_hasher_fn {
+                Some(hasher_fn) => hasher_fn(&bytes)?,
+                None => return Err(AssetHashNotFound(asset.name))?,
+            };
+            let hash_len = archive_hash.len();
+
             debug!(
                 "Downloading archive hash {}",
                 asset_hash.browser_download_url
             );
             let request = client.get(&asset_hash.browser_download_url);
             let response = request.send().await?.error_for_status()?;
             let text = response.text().await?;
-            let re = Regex::new(r"[0-9a-f]{64}")?;
+            let re = Regex::new(&format!(r"[0-9a-f]{{{hash_len}}}"))?;
             let hash = match re.find(&text) {
                 Some(hash) => hash.as_str().to_string(),
                 None => return Err(AssetHashNotFound(asset.name)),
@@ -294,11 +300,6 @@ impl Repository for GitHub {
                 human_bytes(text.len() as f64)
             );
 
-            let archive_hash = match asset_hasher_fn {
-                Some(hasher_fn) => hasher_fn(&bytes)?,
-                None => String::new(),
-            };
-
             if archive_hash != hash {
                 return Err(ArchiveHashMismatch { archive_hash, hash });
             }