362 lines
12 KiB
Java
362 lines
12 KiB
Java
/* wolfCryptMessageDigestMd5Test.java
|
|
*
|
|
* Copyright (C) 2006-2025 wolfSSL Inc.
|
|
*
|
|
* This file is part of wolfSSL.
|
|
*
|
|
* wolfSSL is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* wolfSSL is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335, USA
|
|
*/
|
|
|
|
package com.wolfssl.provider.jce.test;
|
|
|
|
import static org.junit.Assert.*;
|
|
import org.junit.Rule;
|
|
import org.junit.rules.TestRule;
|
|
import org.junit.rules.TestWatcher;
|
|
import org.junit.runner.Description;
|
|
import org.junit.Test;
|
|
import org.junit.Assume;
|
|
import org.junit.BeforeClass;
|
|
|
|
import java.util.Random;
|
|
import java.util.Arrays;
|
|
import java.util.Iterator;
|
|
import java.util.concurrent.Executors;
|
|
import java.util.concurrent.ExecutorService;
|
|
import java.util.concurrent.CountDownLatch;
|
|
import java.util.concurrent.LinkedBlockingQueue;
|
|
|
|
import java.security.Security;
|
|
import java.security.Provider;
|
|
import java.security.MessageDigest;
|
|
import java.security.NoSuchProviderException;
|
|
import java.security.NoSuchAlgorithmException;
|
|
|
|
import com.wolfssl.wolfcrypt.Md5;
|
|
import com.wolfssl.provider.jce.WolfCryptProvider;
|
|
import com.wolfssl.wolfcrypt.FeatureDetect;
|
|
|
|
public class WolfCryptMessageDigestMd5Test {
|
|
|
|
@Rule(order = Integer.MIN_VALUE)
|
|
public TestRule testWatcher = new TestWatcher() {
|
|
protected void starting(Description desc) {
|
|
System.out.println("\t" + desc.getMethodName());
|
|
}
|
|
};
|
|
|
|
@BeforeClass
|
|
public static void testProviderInstallationAtRuntime()
|
|
throws NoSuchProviderException {
|
|
|
|
System.out.println("JCE WolfCryptMessageDigestMd5Test");
|
|
|
|
/* install wolfJCE provider at runtime */
|
|
Security.insertProviderAt(new WolfCryptProvider(), 1);
|
|
|
|
Provider p = Security.getProvider("wolfJCE");
|
|
assertNotNull(p);
|
|
|
|
try {
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5",
|
|
"wolfJCE");
|
|
assertNotNull(md5);
|
|
|
|
} catch (NoSuchAlgorithmException e) {
|
|
/* if we also detect algo is compiled out, skip tests */
|
|
if (FeatureDetect.Md5Enabled() == false) {
|
|
System.out.println("JSSE MD5 Test skipped");
|
|
Assume.assumeTrue(false);
|
|
}
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testMd5SingleUpdate()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException {
|
|
|
|
DigestVector vectors[] = new DigestVector[] {
|
|
/* test vectors {input, expected output} */
|
|
new DigestVector(
|
|
"abc".getBytes(),
|
|
new byte[] {
|
|
(byte)0x90, (byte)0x01, (byte)0x50, (byte)0x98,
|
|
(byte)0x3c, (byte)0xd2, (byte)0x4f, (byte)0xb0,
|
|
(byte)0xd6, (byte)0x96, (byte)0x3f, (byte)0x7d,
|
|
(byte)0x28, (byte)0xe1, (byte)0x7f, (byte)0x72
|
|
}
|
|
),
|
|
new DigestVector(
|
|
"message digest".getBytes(),
|
|
new byte[] {
|
|
(byte)0xf9, (byte)0x6b, (byte)0x69, (byte)0x7d,
|
|
(byte)0x7c, (byte)0xb7, (byte)0x93, (byte)0x8d,
|
|
(byte)0x52, (byte)0x5a, (byte)0x2f, (byte)0x31,
|
|
(byte)0xaa, (byte)0xf1, (byte)0x61, (byte)0xd0
|
|
}
|
|
),
|
|
new DigestVector(
|
|
"abcdefghijklmnopqrstuvwxyz".getBytes(),
|
|
new byte[] {
|
|
(byte)0xc3, (byte)0xfc, (byte)0xd3, (byte)0xd7,
|
|
(byte)0x61, (byte)0x92, (byte)0xe4, (byte)0x00,
|
|
(byte)0x7d, (byte)0xfb, (byte)0x49, (byte)0x6c,
|
|
(byte)0xca, (byte)0x67, (byte)0xe1, (byte)0x3b
|
|
}
|
|
),
|
|
new DigestVector(
|
|
new String("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijkl" +
|
|
"mnopqrstuvwxyz0123456789").getBytes(),
|
|
new byte[] {
|
|
(byte)0xd1, (byte)0x74, (byte)0xab, (byte)0x98,
|
|
(byte)0xd2, (byte)0x77, (byte)0xd9, (byte)0xf5,
|
|
(byte)0xa5, (byte)0x61, (byte)0x1c, (byte)0x2c,
|
|
(byte)0x9f, (byte)0x41, (byte)0x9d, (byte)0x9f
|
|
}
|
|
),
|
|
new DigestVector(
|
|
new String("12345678901234567890123456789012345678" +
|
|
"90123456789012345678901234567890123456" +
|
|
"7890").getBytes(),
|
|
new byte[] {
|
|
(byte)0x57, (byte)0xed, (byte)0xf4, (byte)0xa2,
|
|
(byte)0x2b, (byte)0xe3, (byte)0xc9, (byte)0x55,
|
|
(byte)0xac, (byte)0x49, (byte)0xda, (byte)0x2e,
|
|
(byte)0x21, (byte)0x07, (byte)0xb6, (byte)0x7a
|
|
}
|
|
)
|
|
};
|
|
|
|
byte[] output;
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5", "wolfJCE");
|
|
|
|
for (int i = 0; i < vectors.length; i++) {
|
|
md5.update(vectors[i].getInput());
|
|
output = md5.digest();
|
|
|
|
assertEquals(vectors[i].getOutput().length, output.length);
|
|
assertArrayEquals(vectors[i].getOutput(), output);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testMd5SingleByteUpdate()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException {
|
|
|
|
String input = "Hello World";
|
|
byte[] inArray = input.getBytes();
|
|
final byte expected[] = new byte[] {
|
|
(byte)0xb1, (byte)0x0a, (byte)0x8d, (byte)0xb1,
|
|
(byte)0x64, (byte)0xe0, (byte)0x75, (byte)0x41,
|
|
(byte)0x05, (byte)0xb7, (byte)0xa9, (byte)0x9b,
|
|
(byte)0xe7, (byte)0x2e, (byte)0x3f, (byte)0xe5
|
|
};
|
|
|
|
byte[] output;
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5", "wolfJCE");
|
|
|
|
for (int i = 0; i < inArray.length; i++) {
|
|
md5.update(inArray[i]);
|
|
}
|
|
output = md5.digest();
|
|
assertEquals(expected.length, output.length);
|
|
assertArrayEquals(expected, output);
|
|
}
|
|
|
|
@Test
|
|
public void testMd5Reset()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException {
|
|
|
|
String input = "Hello World";
|
|
byte[] inArray = input.getBytes();
|
|
final byte expected[] = new byte[] {
|
|
(byte)0xb1, (byte)0x0a, (byte)0x8d, (byte)0xb1,
|
|
(byte)0x64, (byte)0xe0, (byte)0x75, (byte)0x41,
|
|
(byte)0x05, (byte)0xb7, (byte)0xa9, (byte)0x9b,
|
|
(byte)0xe7, (byte)0x2e, (byte)0x3f, (byte)0xe5
|
|
};
|
|
|
|
byte[] output;
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5", "wolfJCE");
|
|
|
|
for (int i = 0; i < inArray.length; i++) {
|
|
md5.update(inArray[i]);
|
|
}
|
|
|
|
md5.reset();
|
|
|
|
for (int i = 0; i < inArray.length; i++) {
|
|
md5.update(inArray[i]);
|
|
}
|
|
output = md5.digest();
|
|
assertEquals(expected.length, output.length);
|
|
assertArrayEquals(expected, output);
|
|
}
|
|
|
|
@Test
|
|
public void testMd5Clone()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException,
|
|
CloneNotSupportedException {
|
|
|
|
String input = "Hello World";
|
|
byte[] inArray = input.getBytes();
|
|
final byte expected[] = new byte[] {
|
|
(byte)0xb1, (byte)0x0a, (byte)0x8d, (byte)0xb1,
|
|
(byte)0x64, (byte)0xe0, (byte)0x75, (byte)0x41,
|
|
(byte)0x05, (byte)0xb7, (byte)0xa9, (byte)0x9b,
|
|
(byte)0xe7, (byte)0x2e, (byte)0x3f, (byte)0xe5
|
|
};
|
|
|
|
byte[] output;
|
|
byte[] output2;
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5", "wolfJCE");
|
|
|
|
for (int i = 0; i < inArray.length; i++) {
|
|
md5.update(inArray[i]);
|
|
}
|
|
|
|
/* Try to clone existing MessageDigest, should copy over same state */
|
|
MessageDigest md5Copy = (MessageDigest)md5.clone();
|
|
|
|
output = md5.digest();
|
|
output2 = md5Copy.digest();
|
|
|
|
assertEquals(expected.length, output.length);
|
|
assertEquals(expected.length, output2.length);
|
|
|
|
assertArrayEquals(expected, output);
|
|
assertArrayEquals(expected, output2);
|
|
}
|
|
|
|
@Test
|
|
public void testMd5Interop()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException {
|
|
|
|
String input = "Bozeman, MT";
|
|
String input2 = "wolfSSL is an Open Source Internet security " +
|
|
"company, focused primarily on SSL/TLS and " +
|
|
"cryptography. Main products include the wolfSSL " +
|
|
"embedded SSL/TLS library, wolfCrypt cryptography " +
|
|
"library, wolfMQTT, and wolfSSH. Products are " +
|
|
"dual licensed under both GPLv2 and a commercial" +
|
|
"license.";
|
|
|
|
byte[] wolfOutput;
|
|
byte[] interopOutput;
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5");
|
|
Provider provider = md5.getProvider();
|
|
|
|
/* if we have another MessageDigest provider, test against it */
|
|
if (!provider.equals("wolfJCE")) {
|
|
|
|
/* short message */
|
|
md5.update(input.getBytes());
|
|
interopOutput = md5.digest();
|
|
|
|
MessageDigest wolfMd5 =
|
|
MessageDigest.getInstance("MD5", "wolfJCE");
|
|
|
|
wolfMd5.update(input.getBytes());
|
|
wolfOutput = wolfMd5.digest();
|
|
|
|
assertArrayEquals(wolfOutput, interopOutput);
|
|
|
|
/* long message */
|
|
md5.update(input2.getBytes());
|
|
interopOutput = md5.digest();
|
|
|
|
wolfMd5.update(input2.getBytes());
|
|
wolfOutput = wolfMd5.digest();
|
|
|
|
assertArrayEquals(wolfOutput, interopOutput);
|
|
}
|
|
}
|
|
|
|
@Test
|
|
public void testMd5GetDigestLength()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException {
|
|
|
|
MessageDigest md5 = MessageDigest.getInstance("MD5", "wolfJCE");
|
|
assertEquals(Md5.DIGEST_SIZE, md5.getDigestLength());
|
|
}
|
|
|
|
@Test
|
|
public void testMd5Threaded()
|
|
throws NoSuchProviderException, NoSuchAlgorithmException,
|
|
InterruptedException {
|
|
|
|
int numThreads = 100;
|
|
ExecutorService service = Executors.newFixedThreadPool(numThreads);
|
|
final CountDownLatch latch = new CountDownLatch(numThreads);
|
|
final LinkedBlockingQueue<byte[]> results = new LinkedBlockingQueue<>();
|
|
final byte[] rand10kBuf = new byte[10240];
|
|
|
|
/* fill large input buffer with random bytes */
|
|
new Random().nextBytes(rand10kBuf);
|
|
|
|
/* generate hash over input data concurrently across numThreads */
|
|
for (int i = 0; i < numThreads; i++) {
|
|
service.submit(new Runnable() {
|
|
@Override public void run() {
|
|
|
|
MessageDigest md5 = null;
|
|
|
|
try {
|
|
md5 = MessageDigest.getInstance(
|
|
"MD5", "wolfJCE");
|
|
} catch (NoSuchAlgorithmException |
|
|
NoSuchProviderException e) {
|
|
/* add empty array on failure, will error out below */
|
|
results.add(new byte[] {0});
|
|
}
|
|
|
|
/* process/update in 1024-byte chunks */
|
|
for (int j = 0; j < rand10kBuf.length; j+= 1024) {
|
|
md5.update(rand10kBuf, j, 1024);
|
|
}
|
|
|
|
/* get final hash */
|
|
byte[] hash = md5.digest();
|
|
results.add(hash.clone());
|
|
|
|
latch.countDown();
|
|
}
|
|
});
|
|
}
|
|
|
|
/* wait for all threads to complete */
|
|
latch.await();
|
|
|
|
/* compare all digests, all should be the same across threads */
|
|
Iterator<byte[]> listIterator = results.iterator();
|
|
byte[] current = listIterator.next();
|
|
while (listIterator.hasNext()) {
|
|
byte[] next = listIterator.next();
|
|
if (!Arrays.equals(current, next)) {
|
|
fail("Found two non-identical digests in thread test");
|
|
}
|
|
if (listIterator.hasNext()) {
|
|
current = listIterator.next();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|