1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* Copyright (c) 2020, 2024, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code 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
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
/*
* @test
* @bug 8246077 8300416
* @summary Make sure that digest spi and the resulting digest impl are
* consistent in the impl of Cloneable interface, and that clones do not
* share memory.
* @run testng TestCloneable
*/
import java.nio.ByteBuffer;
import java.security.*;
import java.util.Arrays;
import java.util.Random;
import java.util.Objects;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import org.testng.Assert;
public class TestCloneable {
private static final Class<CloneNotSupportedException> CNSE =
CloneNotSupportedException.class;
private static String providerName = System.getProperty("test.provider.name", "SUN");
@DataProvider
public Object[][] testData() {
return new Object[][] {
{ "MD2", providerName }, { "MD5", providerName }, { "SHA-1", providerName },
{ "SHA-224", providerName }, { "SHA-256", providerName },
{ "SHA-384", providerName }, { "SHA-512", providerName },
{ "SHA3-224", providerName }, { "SHA3-256", providerName },
{ "SHA3-384", providerName }, { "SHA3-512", providerName }
};
}
@Test(dataProvider = "testData")
public void test(String algo, String provName)
throws NoSuchProviderException, NoSuchAlgorithmException,
CloneNotSupportedException, InterruptedException {
System.out.print("Testing " + algo + " impl from " + provName);
Provider p = Security.getProvider(provName);
Provider.Service s = p.getService("MessageDigest", algo);
Objects.requireNonNull(s);
MessageDigestSpi spi = (MessageDigestSpi) s.newInstance(null);
MessageDigest md = MessageDigest.getInstance(algo, provName);
if (spi instanceof Cloneable) {
System.out.println(": Cloneable");
Assert.assertTrue(md instanceof Cloneable);
MessageDigest md2 = (MessageDigest) md.clone();
Assert.assertEquals(md2.getAlgorithm(), algo);
Assert.assertEquals(md2.getProvider().getName(), provName);
Assert.assertTrue(md2 instanceof Cloneable);
} else {
System.out.println(": NOT Cloneable");
Assert.assertThrows(CNSE, ()->md.clone());
}
System.out.print("Testing " + algo + " impl from " + provName);
final var d1 = MessageDigest.getInstance(algo, provName);
final var buffer = ByteBuffer.allocateDirect(1024);
final var r = new Random(1024);
fillBuffer(r, buffer);
d1.update(buffer); // this statement triggers tempArray allocation
final var d2 = (MessageDigest) d1.clone();
assert Arrays.equals(d1.digest(), d2.digest());
final var t1 = updateThread(d1);
final var t2 = updateThread(d2);
t1.join();
t2.join();
System.out.println(": Shared data check");
// Random is producing the same sequence of bytes for each thread,
// and thus each MessageDigest should be equal. When the memory is
// shared, they inevitably overwrite each other's tempArray and
// you get different results.
if (!Arrays.equals(d1.digest(), d2.digest())) {
throw new AssertionError("digests differ");
}
System.out.println("Test Passed");
}
private static void fillBuffer(final Random r, final ByteBuffer buffer) {
final byte[] bytes = new byte[buffer.capacity()];
r.nextBytes(bytes);
buffer.clear();
buffer.put(bytes);
buffer.flip();
}
public static Thread updateThread(final MessageDigest d) {
final var t = new Thread(() -> {
final var r = new Random(1024);
final ByteBuffer buffer = ByteBuffer.allocateDirect(1024);
for (int i = 0; i < 1024; i++) {
fillBuffer(r, buffer);
d.update(buffer);
}
});
t.start();
return t;
}
}
|
