DefaultVFS.java
/*
* Copyright 2010-2023 the original author or authors.
*
* 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
*
* https://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.apache.ibatis.migration.io;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLEncoder;
import java.nio.file.InvalidPathException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarInputStream;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* A default implementation of {@link VFS} that works for most application servers.
*
* @author Ben Gunter
*/
public class DefaultVFS extends VFS {
private static final Logger log = Logger.getLogger(DefaultVFS.class.getName());
/** The magic header that indicates a JAR (ZIP) file. */
private static final byte[] JAR_MAGIC = { 'P', 'K', 3, 4 };
@Override
public boolean isValid() {
return true;
}
@Override
public List<String> list(URL url, String path) throws IOException {
InputStream is = null;
try {
List<String> resources = new ArrayList<>();
// First, try to find the URL of a JAR file containing the requested resource. If a JAR
// file is found, then we'll list child resources by reading the JAR.
URL jarUrl = findJarForResource(url);
if (jarUrl != null) {
is = jarUrl.openStream();
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Listing " + url);
}
resources = listResources(new JarInputStream(is), path);
} else {
List<String> children = new ArrayList<>();
try {
if (isJar(url)) {
// Some versions of JBoss VFS might give a JAR stream even if the resource
// referenced by the URL isn't actually a JAR
is = url.openStream();
try (JarInputStream jarInput = new JarInputStream(is)) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Listing " + url);
}
for (JarEntry entry; (entry = jarInput.getNextJarEntry()) != null;) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Jar entry: " + entry.getName());
}
children.add(entry.getName());
}
}
} else {
/*
* Some servlet containers allow reading from directory resources like a text file, listing the child
* resources one per line. However, there is no way to differentiate between directory and file resources
* just by reading them. To work around that, as each line is read, try to look it up via the class loader
* as a child of the current resource. If any line fails then we assume the current resource is not a
* directory.
*/
is = url.openStream();
List<String> lines = new ArrayList<>();
try (BufferedReader reader = new BufferedReader(new InputStreamReader(is))) {
for (String line; (line = reader.readLine()) != null;) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Reader entry: " + line);
}
lines.add(line);
if (getResources(path + "/" + line).isEmpty()) {
lines.clear();
break;
}
}
} catch (InvalidPathException e) {
// #1974
lines.clear();
}
if (!lines.isEmpty()) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Listing " + url);
}
children.addAll(lines);
}
}
} catch (FileNotFoundException e) {
/*
* For file URLs the openStream() call might fail, depending on the servlet container, because directories
* can't be opened for reading. If that happens, then list the directory directly instead.
*/
if (!"file".equals(url.getProtocol())) {
// No idea where the exception came from so rethrow it
throw e;
}
File file = new File(url.getFile());
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Listing directory " + file.getAbsolutePath());
}
if (file.isDirectory()) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Listing " + url);
}
children = Arrays.asList(file.list());
}
}
// The URL prefix to use when recursively listing child resources
String prefix = url.toExternalForm();
if (!prefix.endsWith("/")) {
prefix = prefix + "/";
}
// Iterate over immediate children, adding files and recurring into directories
for (String child : children) {
String resourcePath = path + "/" + child;
resources.add(resourcePath);
URL childUrl = new URL(prefix + child);
resources.addAll(list(childUrl, resourcePath));
}
}
return resources;
} finally {
if (is != null) {
try {
is.close();
} catch (Exception e) {
// Ignore
}
}
}
}
/**
* List the names of the entries in the given {@link JarInputStream} that begin with the specified {@code path}.
* Entries will match with or without a leading slash.
*
* @param jar
* The JAR input stream
* @param path
* The leading path to match
*
* @return The names of all the matching entries
*
* @throws IOException
* If I/O errors occur
*/
protected List<String> listResources(JarInputStream jar, String path) throws IOException {
// Include the leading and trailing slash when matching names
if (!path.startsWith("/")) {
path = '/' + path;
}
if (!path.endsWith("/")) {
path = path + '/';
}
// Iterate over the entries and collect those that begin with the requested path
List<String> resources = new ArrayList<>();
for (JarEntry entry; (entry = jar.getNextJarEntry()) != null;) {
if (!entry.isDirectory()) {
// Add leading slash if it's missing
StringBuilder name = new StringBuilder(entry.getName());
if (name.charAt(0) != '/') {
name.insert(0, '/');
}
// Check file name
if (name.indexOf(path) == 0) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Found resource: " + name);
}
// Trim leading slash
resources.add(name.substring(1));
}
}
}
return resources;
}
/**
* Attempts to deconstruct the given URL to find a JAR file containing the resource referenced by the URL. That is,
* assuming the URL references a JAR entry, this method will return a URL that references the JAR file containing the
* entry. If the JAR cannot be located, then this method returns null.
*
* @param url
* The URL of the JAR entry.
*
* @return The URL of the JAR file, if one is found. Null if not.
*
* @throws MalformedURLException
* the malformed URL exception
*/
protected URL findJarForResource(URL url) throws MalformedURLException {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Find JAR URL: " + url);
}
// If the file part of the URL is itself a URL, then that URL probably points to the JAR
boolean continueLoop = true;
while (continueLoop) {
try {
url = new URL(url.getFile());
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Inner URL: " + url);
}
} catch (MalformedURLException e) {
// This will happen at some point and serves as a break in the loop
continueLoop = false;
}
}
// Look for the .jar extension and chop off everything after that
StringBuilder jarUrl = new StringBuilder(url.toExternalForm());
int index = jarUrl.lastIndexOf(".jar");
if (index < 0) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Not a JAR: " + jarUrl);
}
return null;
}
jarUrl.setLength(index + 4);
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Extracted JAR URL: " + jarUrl);
}
// Try to open and test it
try {
URL testUrl = new URL(jarUrl.toString());
if (isJar(testUrl)) {
return testUrl;
}
// WebLogic fix: check if the URL's file exists in the filesystem.
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Not a JAR: " + jarUrl);
}
jarUrl.replace(0, jarUrl.length(), testUrl.getFile());
File file = new File(jarUrl.toString());
// File name might be URL-encoded
if (!file.exists()) {
try {
file = new File(URLEncoder.encode(jarUrl.toString(), "UTF-8"));
} catch (UnsupportedEncodingException e) {
throw new RuntimeException("Unsupported encoding? UTF-8? That's impossible.");
}
}
if (file.exists()) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Trying real file: " + file.getAbsolutePath());
}
testUrl = file.toURI().toURL();
if (isJar(testUrl)) {
return testUrl;
}
}
} catch (MalformedURLException e) {
log.log(Level.WARNING, "Invalid JAR URL: " + jarUrl);
}
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Not a JAR: " + jarUrl);
}
return null;
}
/**
* Converts a Java package name to a path that can be looked up with a call to
* {@link ClassLoader#getResources(String)}.
*
* @param packageName
* The Java package name to convert to a path
*
* @return the package path
*/
protected String getPackagePath(String packageName) {
return packageName == null ? null : packageName.replace('.', '/');
}
/**
* Returns true if the resource located at the given URL is a JAR file.
*
* @param url
* The URL of the resource to test.
*
* @return true, if is jar
*/
protected boolean isJar(URL url) {
return isJar(url, new byte[JAR_MAGIC.length]);
}
/**
* Returns true if the resource located at the given URL is a JAR file.
*
* @param url
* The URL of the resource to test.
* @param buffer
* A buffer into which the first few bytes of the resource are read. The buffer must be at least the size of
* {@link #JAR_MAGIC}. (The same buffer may be reused for multiple calls as an optimization.)
*
* @return true, if is jar
*/
protected boolean isJar(URL url, byte[] buffer) {
try (InputStream is = url.openStream()) {
is.read(buffer, 0, JAR_MAGIC.length);
if (Arrays.equals(buffer, JAR_MAGIC)) {
if (log.isLoggable(Level.FINER)) {
log.log(Level.FINER, "Found JAR: " + url);
}
return true;
}
} catch (Exception e) {
// Failure to read the stream means this is not a JAR
}
return false;
}
}