lib/noise/functions/hash.rb in noise-ruby-0.6.3 vs lib/noise/functions/hash.rb in noise-ruby-0.7.0

@@ -7,14 +7,15 @@
       autoload :Blake2s, 'noise/functions/hash/blake2s'
       autoload :Sha256, 'noise/functions/hash/sha256'
       autoload :Sha512, 'noise/functions/hash/sha512'
 
       def self.hmac_hash(key, data, digest)
-        # TODO: support for blake2b, blake2s
         if digest.include?('SHA')
           OpenSSL::HMAC.digest(OpenSSL::Digest.new(digest), key, data)
         elsif digest.include?('BLAKE2b')
           Noise::Functions::Hash::Blake2bHMAC.new(key).update(data).digest
+        elsif digest.include?('BLAKE2s')
+          Noise::Functions::Hash::Blake2sHMAC.new(key).update(data).digest
         end
       end
 
       def self.create_hkdf_fn(digest)
         lambda do |chaining_key, input_key_material, num_output|