Skip to content

Commit

Permalink
Merge pull request #252 from spyrkob/MODULES-378-1.8
Browse files Browse the repository at this point in the history
[MODULES-378] Module resource loader follows symbolic links (1.8.x)
  • Loading branch information
ropalka committed Feb 10, 2020
2 parents 81308ea + 8a2592a commit f6e375b
Show file tree
Hide file tree
Showing 2 changed files with 80 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/main/java/org/jboss/modules/PathResourceLoader.java
Expand Up @@ -24,6 +24,7 @@
import java.net.URI;
import java.net.URL;
import java.nio.file.FileSystems;
import java.nio.file.FileVisitOption;
import java.nio.file.Files;
import java.nio.file.InvalidPathException;
import java.nio.file.Path;
Expand Down Expand Up @@ -168,7 +169,7 @@ public Iterator<Resource> iterateResources(final String startPath, final boolean
} catch (InvalidPathException ignored) {
return Collections.emptyIterator();
}
return Files.walk(path, recursive ? Integer.MAX_VALUE : 1)
return Files.walk(path, recursive ? Integer.MAX_VALUE : 1, FileVisitOption.FOLLOW_LINKS)
.filter(it -> !Files.isDirectory(it))
.<Resource>map(resourcePath -> new PathResource(resourcePath, PathUtils.toGenericSeparators(root.relativize(resourcePath).toString()), context))
.iterator();
Expand All @@ -180,7 +181,7 @@ public Iterator<Resource> iterateResources(final String startPath, final boolean
@Override
public Collection<String> getPaths() {
try {
return doPrivilegedIfNeeded(context, IOException.class, () -> Files.walk(root)
return doPrivilegedIfNeeded(context, IOException.class, () -> Files.walk(root, FileVisitOption.FOLLOW_LINKS)
.filter(Files::isDirectory)
.map(dir -> {
final String result = root.relativize(dir).toString();
Expand Down
77 changes: 77 additions & 0 deletions src/test/java/org/jboss/modules/SymlinkResourceLoaderTest.java
@@ -0,0 +1,77 @@
/*
* JBoss, Home of Professional Open Source.
* Copyright 2014 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package org.jboss.modules;

import java.io.File;
import java.nio.file.FileSystemException;
import java.nio.file.Files;
import java.security.AccessController;

import org.jboss.modules.filter.PathFilter;
import org.junit.After;
import org.junit.Assert;
import org.junit.Assume;

/**
* Test the functionality of the PathResourceLoader with resources containing symbolic links
*
* @author Bartosz Spyrko-Smietanko
*/
public class SymlinkResourceLoaderTest extends AbstractResourceLoaderTestCase {

private File resourceRoot;

@After
public void tearDown() throws Exception {
File base = getResource("test");
File symlink = new File(base, "symlink");
if (symlink.exists()) {
symlink.delete();
}
}

protected ResourceLoader createLoader(final PathFilter exportFilter) throws Exception {
File base = getResource("test");
File realRoot = getResource("test/fileresourceloader");
try {
resourceRoot = Files.createSymbolicLink(new File(base, "symlink").toPath(), realRoot.toPath()).toFile();
} catch (UnsupportedOperationException | FileSystemException e) {
Assume.assumeNoException(e);
}

// Copy the classfile over
copyResource("org/jboss/modules/test/TestClass.class", "test/fileresourceloader", "org/jboss/modules/test");
return new PathResourceLoader("test-root", resourceRoot.toPath(), AccessController.getContext());
}

@Override
protected void assertResource(Resource resource, String fileName) {
final File resourceFile = getExpectedFile(fileName);

Assert.assertEquals(resourceFile.length(), resource.getSize());
}

public void testGetClassSpec() throws Exception {
super.testGetClassSpec();
}

protected File getExpectedFile(String fileName) {
return new File(resourceRoot, fileName);
}
}

0 comments on commit f6e375b

Please sign in to comment.