standardize blake2b implementation

jshawl · Feb 4, 2024 · 18f1e81 · 18f1e81
1 parent e5af882
commit 18f1e81
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/lib/minisign/utils.rb b/lib/minisign/utils.rb
@@ -3,8 +3,12 @@
 module Minisign
   # Helpers used in multiple classes
   module Utils
+    def blake2b256(message)
+      RbNaCl::Hash::Blake2b.digest(message, { digest_size: 32 })
+    end
+
     def blake2b512(message)
-      OpenSSL::Digest.new('BLAKE2b512').digest(message)
+      RbNaCl::Hash::Blake2b.digest(message, { digest_size: 64 })
     end
   end
 end
diff --git a/spec/minisign/private_key_spec.rb b/spec/minisign/private_key_spec.rb
@@ -68,8 +68,7 @@
         @private_key.public_key
       ].inject(&:+).pack('C*')
 
-      computed_checksum = RbNaCl::Hash::Blake2b.digest(key_data, { digest_size: 32 }).bytes
-
+      computed_checksum = blake2b256(key_data).bytes
       expect(@private_key.checksum).to eq(computed_checksum)
     end
   end

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
@@ -8,6 +8,7 @@
 require 'minisign'
 
 RSpec.configure do |config|
+  config.include Minisign::Utils
   config.expect_with :rspec do |expectations|
     expectations.include_chain_clauses_in_custom_matcher_descriptions = true
   end