Add more tests

patrickfav · Sep 30, 2017 · 10c104b · 10c104b
1 parent 9dd5925
commit 10c104b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/main/java/at/favre/crypto/HKDF.java b/src/main/java/at/favre/crypto/HKDF.java
@@ -173,7 +173,7 @@ public static byte[] expand(HkdfMacFactory macFactory, byte[] pseudoRandomKey, b
      * @return new byte array of output keying material (OKM)
      */
     public static byte[] hkdfSha256(byte[] inputKeyingMaterial, byte[] saltExtract, byte[] infoExpand, int outLengthByte) {
-        return hkdf(HkdfMacFactory.Default.hmacSha256(), infoExpand, saltExtract, infoExpand, outLengthByte);
+        return hkdf(HkdfMacFactory.Default.hmacSha256(), inputKeyingMaterial, saltExtract, infoExpand, outLengthByte);
     }
 
     /**

diff --git a/src/test/java/at/favre/crypto/HKDFTest.java b/src/test/java/at/favre/crypto/HKDFTest.java
@@ -100,6 +100,8 @@ public void testExpandFailures() throws Exception {
 
     @Test
     public void hkdf() throws Exception {
+        checkLength(HKDF.hkdf(HkdfMacFactory.Default.hmacSha1(), RandomUtils.nextBytes(16), RandomUtils.nextBytes(16), null, 80), 80);
+        checkLength(HKDF.hkdfSha256(RandomUtils.nextBytes(16), RandomUtils.nextBytes(16), null, 80), 80);
     }
 
     @Test