209 lines
7.4 KiB
Java
209 lines
7.4 KiB
Java
/**
|
|
* Copyright (c) 2015, 2016, 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
|
|
* @summary Test the recording and checking of module hashes
|
|
* @author Andrei Eremeev
|
|
* @library /lib/testlibrary
|
|
* @modules java.base/jdk.internal.misc
|
|
* java.base/jdk.internal.module
|
|
* jdk.jlink
|
|
* jdk.compiler
|
|
* @build CompilerUtils
|
|
* @run testng HashesTest
|
|
*/
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.lang.module.ModuleDescriptor;
|
|
import java.lang.module.ModuleFinder;
|
|
import java.lang.module.ModuleReader;
|
|
import java.lang.module.ModuleReference;
|
|
import java.nio.file.FileVisitResult;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.nio.file.SimpleFileVisitor;
|
|
import java.nio.file.attribute.BasicFileAttributes;
|
|
import java.util.ArrayList;
|
|
import java.util.Arrays;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.Optional;
|
|
import java.util.Set;
|
|
import java.util.spi.ToolProvider;
|
|
import java.util.stream.Collectors;
|
|
|
|
import jdk.internal.misc.SharedSecrets;
|
|
import jdk.internal.misc.JavaLangModuleAccess;
|
|
import jdk.internal.module.ModuleHashes;
|
|
|
|
import org.testng.annotations.BeforeTest;
|
|
import org.testng.annotations.Test;
|
|
|
|
import static org.testng.Assert.*;
|
|
|
|
public class HashesTest {
|
|
static final ToolProvider JMOD_TOOL = ToolProvider.findFirst("jmod")
|
|
.orElseThrow(() ->
|
|
new RuntimeException("jmod tool not found")
|
|
);
|
|
|
|
private final Path testSrc = Paths.get(System.getProperty("test.src"));
|
|
private final Path modSrc = testSrc.resolve("src");
|
|
private final Path mods = Paths.get("mods");
|
|
private final Path jmods = Paths.get("jmods");
|
|
private final String[] modules = new String[] { "m1", "m2", "m3"};
|
|
|
|
@BeforeTest
|
|
private void setup() throws Exception {
|
|
if (Files.exists(jmods)) {
|
|
deleteDirectory(jmods);
|
|
}
|
|
Files.createDirectories(jmods);
|
|
|
|
// build m2, m3 required by m1
|
|
compileModule("m2", modSrc);
|
|
jmod("m2");
|
|
|
|
compileModule("m3", modSrc);
|
|
jmod("m3");
|
|
|
|
// build m1
|
|
compileModule("m1", modSrc);
|
|
// no hash is recorded since m1 has outgoing edges
|
|
jmod("m1", "--module-path", jmods.toString(), "--hash-modules", ".*");
|
|
|
|
// compile org.bar and org.foo
|
|
compileModule("org.bar", modSrc);
|
|
compileModule("org.foo", modSrc);
|
|
}
|
|
|
|
@Test
|
|
public void test() throws Exception {
|
|
for (String mn : modules) {
|
|
assertFalse(hashes(mn).isPresent());
|
|
}
|
|
|
|
// hash m1 in m2
|
|
jmod("m2", "--module-path", jmods.toString(), "--hash-modules", "m1");
|
|
checkHashes(hashes("m2").get(), "m1");
|
|
|
|
// hash m1 in m2
|
|
jmod("m2", "--module-path", jmods.toString(), "--hash-modules", ".*");
|
|
checkHashes(hashes("m2").get(), "m1");
|
|
|
|
// create m2.jmod with no hash
|
|
jmod("m2");
|
|
// run jmod hash command to hash m1 in m2 and m3
|
|
runJmod(Arrays.asList("hash", "--module-path", jmods.toString(),
|
|
"--hash-modules", ".*"));
|
|
checkHashes(hashes("m2").get(), "m1");
|
|
checkHashes(hashes("m3").get(), "m1");
|
|
|
|
jmod("org.bar");
|
|
jmod("org.foo");
|
|
|
|
jmod("org.bar", "--module-path", jmods.toString(), "--hash-modules", "org.*");
|
|
checkHashes(hashes("org.bar").get(), "org.foo");
|
|
|
|
jmod("m3", "--module-path", jmods.toString(), "--hash-modules", ".*");
|
|
checkHashes(hashes("m3").get(), "org.foo", "org.bar", "m1");
|
|
}
|
|
|
|
private void checkHashes(ModuleHashes hashes, String... hashModules) {
|
|
assertTrue(hashes.names().equals(Set.of(hashModules)));
|
|
}
|
|
|
|
private Optional<ModuleHashes> hashes(String name) throws Exception {
|
|
ModuleFinder finder = SharedSecrets.getJavaLangModuleAccess()
|
|
.newModulePath(Runtime.version(), true, jmods.resolve(name + ".jmod"));
|
|
ModuleReference mref = finder.find(name).orElseThrow(RuntimeException::new);
|
|
ModuleReader reader = mref.open();
|
|
try (InputStream in = reader.open("module-info.class").get()) {
|
|
ModuleDescriptor md = ModuleDescriptor.read(in);
|
|
JavaLangModuleAccess jmla = SharedSecrets.getJavaLangModuleAccess();
|
|
Optional<ModuleHashes> hashes = jmla.hashes(md);
|
|
System.out.format("hashes in module %s %s%n", name,
|
|
hashes.isPresent() ? "present" : "absent");
|
|
if (hashes.isPresent()) {
|
|
hashes.get().names().stream()
|
|
.sorted()
|
|
.forEach(n -> System.out.format(" %s %s%n", n, hashes.get().hashFor(n)));
|
|
}
|
|
return hashes;
|
|
} finally {
|
|
reader.close();
|
|
}
|
|
}
|
|
|
|
private void deleteDirectory(Path dir) throws IOException {
|
|
Files.walkFileTree(dir, new SimpleFileVisitor<Path>() {
|
|
@Override
|
|
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)
|
|
throws IOException
|
|
{
|
|
Files.delete(file);
|
|
return FileVisitResult.CONTINUE;
|
|
}
|
|
|
|
@Override
|
|
public FileVisitResult postVisitDirectory(Path dir, IOException exc)
|
|
throws IOException
|
|
{
|
|
Files.delete(dir);
|
|
return FileVisitResult.CONTINUE;
|
|
}
|
|
});
|
|
}
|
|
|
|
private void compileModule(String moduleName, Path src) throws IOException {
|
|
Path msrc = src.resolve(moduleName);
|
|
assertTrue(CompilerUtils.compile(msrc, mods, "--module-source-path", src.toString()));
|
|
}
|
|
|
|
private void jmod(String moduleName, String... options) throws IOException {
|
|
Path mclasses = mods.resolve(moduleName);
|
|
Path outfile = jmods.resolve(moduleName + ".jmod");
|
|
List<String> args = new ArrayList<>();
|
|
args.add("create");
|
|
Collections.addAll(args, options);
|
|
Collections.addAll(args, "--class-path", mclasses.toString(),
|
|
outfile.toString());
|
|
|
|
if (Files.exists(outfile))
|
|
Files.delete(outfile);
|
|
|
|
runJmod(args);
|
|
}
|
|
|
|
private void runJmod(List<String> args) {
|
|
int rc = JMOD_TOOL.run(System.out, System.out, args.toArray(new String[args.size()]));
|
|
System.out.println("jmod options: " + args.stream().collect(Collectors.joining(" ")));
|
|
if (rc != 0) {
|
|
throw new AssertionError("Jmod failed: rc = " + rc);
|
|
}
|
|
}
|
|
}
|