Added support functions for filesha512 and BLAKE-3 hashing. (#1697)

* Add functions for SHA-512 and BLAKE-3 hashing * Fix filesha512 * Fix filesha512 formatting Signed-off-by: slashexx <dhruvpuri.35@gmail.com> --------- Signed-off-by: slashexx <dhruvpuri.35@gmail.com>
kcl-lang · Oct 17, 2024 · f328749 · f328749
1 parent 965b5e1
commit f328749
Showing 1 changed file with 71 additions and 0 deletions.
diff --git a/kclvm/runtime/src/crypto/mod.rs b/kclvm/runtime/src/crypto/mod.rs
@@ -269,3 +269,74 @@ pub extern "C" fn kclvm_crypto_filesha256(
     }
     panic!("filesha256() missing 1 required positional argument: 'filepath'");
 }
+
+#[no_mangle]
+#[runtime_fn]
+pub extern "C" fn kclvm_crypto_filesha512(
+    ctx: *mut kclvm_context_t,
+    args: *const kclvm_value_ref_t,
+    kwargs: *const kclvm_value_ref_t,
+) -> *const kclvm_value_ref_t {
+    let args = ptr_as_ref(args);
+    let kwargs = ptr_as_ref(kwargs);
+    let ctx = mut_ptr_as_ref(ctx);
+
+    if let Some(filepath) = get_call_arg_str(args, kwargs, 0, Some("filepath")) {
+        let mut file = File::open(&filepath)
+            .unwrap_or_else(|e| panic!("failed to access file '{}': {}", filepath, e));
+
+        let mut hasher = Sha512::new();
+
+        let mut buffer = [0; 4096];
+        while let Ok(bytes_read) = file.read(&mut buffer) {
+            if bytes_read == 0 {
+                break; // End of file
+            }
+            hasher.update(&buffer[..bytes_read]);
+        }
+
+        let hash_result = hasher.finalize();
+
+        let hex = hash_result
+            .iter()
+            .map(|byte| format!("{byte:02x}"))
+            .collect::<String>();
+
+        return ValueRef::str(&hex).into_raw(ctx);
+    }
+    panic!("filesha512() missing 1 required positional argument: 'filepath'");
+}
+
+// fileblake3
+#[no_mangle]
+#[runtime_fn]
+pub extern "C" fn kclvm_crypto_fileblake3(
+    ctx: *mut kclvm_context_t,
+    args: *const kclvm_value_ref_t,
+    kwargs: *const kclvm_value_ref_t,
+) -> *const kclvm_value_ref_t {
+    let args = ptr_as_ref(args);
+    let kwargs = ptr_as_ref(kwargs);
+    let ctx = mut_ptr_as_ref(ctx);
+
+    if let Some(filepath) = get_call_arg_str(args, kwargs, 0, Some("filepath")) {
+        let mut file = File::open(&filepath)
+            .unwrap_or_else(|e| panic!("failed to access file '{}': {}", filepath, e));
+
+        let mut buffer = Vec::new();
+        file.read_to_end(&mut buffer)
+            .unwrap_or_else(|e| panic!("failed to read file '{}': {}", filepath, e));
+
+        let hasher = blake3::hash(&buffer);
+
+        let mut hex = String::with_capacity(2 * blake3::OUT_LEN);
+        use std::fmt::Write;
+
+        for byte in hasher.as_bytes() {
+            let _ = write!(&mut hex, "{byte:02x}");
+        }
+
+        return ValueRef::str(hex.as_str()).into_raw(ctx);
+    }
+    panic!("fileblake3() missing 1 required positional argument: 'filepath'");
+}