chacha20blake3 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 884132d1f327e45e947d008367009753b1c014a8809a1330fb1cbc8cf014def6
-  data.tar.gz: c1ff6117acd18ead79c48d08a9bcb2f223dd4524e65b93268f06c9ca1ac16b37
+  metadata.gz: 9d875e3dca3b733cb6a6cac6f3c00539e5ea670e4128a5b2107da9c99cee63e9
+  data.tar.gz: 9c1b971c5f768b1ac59d0195e6c979f8979571eebebac242e1cd4a516b250e44
 SHA512:
-  metadata.gz: 71187d3c2e6d86858938c3fe26476c60bdc1af777d4eee842794575d1823fb95514d73fe14282c1e38828b6b691e2d71ffbf5c570e6abb2cd0efc438a3e9a547
-  data.tar.gz: affa5cb7f8123f050a9f8cb38b83f1c11110e7730ee360b53287e81c12c8017a36ffcb68dc88113be539aeab23247626864023b27b39463b6ec48f8a20b8ad84
+  metadata.gz: 9a140c81d80fb12ce9176a5fee9e2b828d9f32ff8ba27033ea7dd593d4a8584dc94373302ded86f2d1c9d20a5b5af8a5c789ed5327174fb9ad88aac46c6988d4
+  data.tar.gz: 01c1369aaea777a152ce78f1c0f14cb6b214faadf2e5dffdf38a68ece66922b92bfd087385f37b0be3cf3e4fb3055176b47602a009f440fe1f4d348f9c564fa1

data/ext/chacha20blake3/src/lib.rs

@@ -310,6 +310,39 @@ fn stream_message_index(rb_self: &Stream) -> u64 {
     *rb_self.counter.lock().unwrap()
 }
 
+fn blake3_derive_key(ruby: &Ruby, args: &[Value]) -> Result<RString, Error> {
+    let parsed = scan_args::<(RString, RString), (), (), (), RHash, ()>(args)?;
+    let (rb_context, rb_material) = parsed.required;
+    let kw = get_kwargs::<_, (), (Option<usize>,), ()>(parsed.keywords, &[], &["length"])?;
+    let (opt_length,) = kw.optional;
+    let length = opt_length.unwrap_or(32);
+
+    if length == 0 || length > 65535 {
+        return Err(Error::new(
+            ruby.exception_arg_error(),
+            format!("length must be 1..65535, got {length}"),
+        ));
+    }
+
+    // SAFETY: copy context string before any allocation
+    let context = unsafe { std::str::from_utf8(rb_context.as_slice()) }
+        .map_err(|_| Error::new(ruby.exception_arg_error(), "context must be valid UTF-8"))?
+        .to_owned();
+
+    let mut output_buf = vec![0u8; length];
+    unsafe {
+        let mut deriver = blake3::Hasher::new_derive_key(&context);
+        deriver.update(rb_material.as_slice());
+        let mut reader = deriver.finalize_xof();
+        reader.fill(&mut output_buf);
+    }
+
+    let output = ruby.str_from_slice(&output_buf);
+    output.freeze();
+    Ok(output)
+}
+
+
 fn generate_key(ruby: &Ruby) -> Result<RString, Error> {
     let mut key = [0u8; KEY_SIZE];
     getrandom::getrandom(&mut key).map_err(|e| {
@@ -365,6 +398,7 @@ fn init(ruby: &Ruby) -> Result<(), Error> {
 
     module.define_module_function("generate_key", function!(generate_key, 0))?;
     module.define_module_function("generate_nonce", function!(generate_nonce, 0))?;
+    module.define_module_function("derive_key", function!(blake3_derive_key, -1))?;
 
     Ok(())
 }

data/lib/chacha20blake3/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ChaCha20Blake3
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: chacha20blake3
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Patrik Wenger