Merge branch '2.4.x'

Closes gh-25509
pull/25532/head
Andy Wilkinson 4 years ago
commit 2b7f23b29d

@ -29,7 +29,6 @@ import java.util.zip.ZipInputStream;
import org.springframework.util.Assert; import org.springframework.util.Assert;
import org.springframework.util.StreamUtils; import org.springframework.util.StreamUtils;
import org.springframework.util.StringUtils;
/** /**
* The {@code 'extract'} tools command. * The {@code 'extract'} tools command.
@ -86,15 +85,18 @@ class ExtractCommand extends Command {
} }
private void write(ZipInputStream zip, ZipEntry entry, File destination) throws IOException { private void write(ZipInputStream zip, ZipEntry entry, File destination) throws IOException {
String path = StringUtils.cleanPath(entry.getName()); String canonicalOutputPath = destination.getCanonicalPath() + File.separator;
File file = new File(destination, path); File file = new File(destination, entry.getName());
if (file.getAbsolutePath().startsWith(destination.getAbsolutePath())) { String canonicalEntryPath = file.getCanonicalPath();
mkParentDirs(file); Assert.state(canonicalEntryPath.startsWith(canonicalOutputPath),
try (OutputStream out = new FileOutputStream(file)) { () -> "Entry '" + entry.getName() + "' would be written to '" + canonicalEntryPath
StreamUtils.copy(zip, out); + "'. This is outside the output location of '" + canonicalOutputPath
} + "'. Verify the contents of your archive.");
Files.setAttribute(file.toPath(), "creationTime", entry.getCreationTime()); mkParentDirs(file);
try (OutputStream out = new FileOutputStream(file)) {
StreamUtils.copy(zip, out);
} }
Files.setAttribute(file.toPath(), "creationTime", entry.getCreationTime());
} }
private void mkParentDirs(File file) throws IOException { private void mkParentDirs(File file) throws IOException {

@ -24,6 +24,7 @@ import java.io.InputStreamReader;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
import java.util.Iterator; import java.util.Iterator;
import java.util.function.Consumer;
import java.util.jar.JarEntry; import java.util.jar.JarEntry;
import java.util.zip.ZipEntry; import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream; import java.util.zip.ZipOutputStream;
@ -46,6 +47,7 @@ import static org.mockito.BDDMockito.given;
* Tests for {@link ExtractCommand}. * Tests for {@link ExtractCommand}.
* *
* @author Phillip Webb * @author Phillip Webb
* @author Andy Wilkinson
*/ */
@ExtendWith(MockitoExtension.class) @ExtendWith(MockitoExtension.class)
class ExtractCommandTests { class ExtractCommandTests {
@ -82,6 +84,7 @@ class ExtractCommandTests {
assertThat(new File(this.extract, "b/b/b.jar")).exists(); assertThat(new File(this.extract, "b/b/b.jar")).exists();
assertThat(new File(this.extract, "c/c/c.jar")).exists(); assertThat(new File(this.extract, "c/c/c.jar")).exists();
assertThat(new File(this.extract, "d")).isDirectory(); assertThat(new File(this.extract, "d")).isDirectory();
assertThat(new File(this.extract.getParentFile(), "e.jar")).doesNotExist();
} }
@Test @Test
@ -104,6 +107,7 @@ class ExtractCommandTests {
assertThat(this.extract.list()).containsOnly("a", "c"); assertThat(this.extract.list()).containsOnly("a", "c");
assertThat(new File(this.extract, "a/a/a.jar")).exists(); assertThat(new File(this.extract, "a/a/a.jar")).exists();
assertThat(new File(this.extract, "c/c/c.jar")).exists(); assertThat(new File(this.extract, "c/c/c.jar")).exists();
assertThat(new File(this.extract.getParentFile(), "e.jar")).doesNotExist();
} }
@Test @Test
@ -119,7 +123,29 @@ class ExtractCommandTests {
.withMessageContaining("not compatible with layertools"); .withMessageContaining("not compatible with layertools");
} }
@Test
void runWithJarFileThatWouldWriteEntriesOutsideDestinationFails() throws Exception {
this.jarFile = createJarFile("test.jar", (out) -> {
try {
out.putNextEntry(new ZipEntry("e/../../e.jar"));
out.closeEntry();
}
catch (IOException ex) {
throw new IllegalStateException(ex);
}
});
given(this.context.getArchiveFile()).willReturn(this.jarFile);
assertThatIllegalStateException()
.isThrownBy(() -> this.command.run(Collections.emptyMap(), Collections.emptyList()))
.withMessageContaining("Entry 'e/../../e.jar' would be written");
}
private File createJarFile(String name) throws Exception { private File createJarFile(String name) throws Exception {
return createJarFile(name, (out) -> {
});
}
private File createJarFile(String name, Consumer<ZipOutputStream> streamHandler) throws Exception {
File file = new File(this.temp, name); File file = new File(this.temp, name);
try (ZipOutputStream out = new ZipOutputStream(new FileOutputStream(file))) { try (ZipOutputStream out = new ZipOutputStream(new FileOutputStream(file))) {
out.putNextEntry(new ZipEntry("a/")); out.putNextEntry(new ZipEntry("a/"));
@ -139,6 +165,7 @@ class ExtractCommandTests {
out.putNextEntry(new JarEntry("META-INF/MANIFEST.MF")); out.putNextEntry(new JarEntry("META-INF/MANIFEST.MF"));
out.write(getFile("test-manifest.MF").getBytes()); out.write(getFile("test-manifest.MF").getBytes());
out.closeEntry(); out.closeEntry();
streamHandler.accept(out);
} }
return file; return file;
} }

Loading…
Cancel
Save