| /* | |
| * Copyright (c) 2021, 2025, 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 Tests for SimpleFileServer | |
| * @library /test/lib | |
| * @build jdk.test.lib.Platform jdk.test.lib.net.URIBuilder | |
| * @run testng/othervm SimpleFileServerTest | |
| */ | |
| import java.io.IOException; | |
| import java.io.UncheckedIOException; | |
| import java.net.InetAddress; | |
| import java.net.InetSocketAddress; | |
| import java.net.URI; | |
| import java.net.http.HttpClient; | |
| import java.net.http.HttpRequest; | |
| import java.net.http.HttpRequest.BodyPublishers; | |
| import java.net.http.HttpResponse.BodyHandlers; | |
| import java.nio.file.Files; | |
| import java.nio.file.LinkOption; | |
| import java.nio.file.Path; | |
| import java.time.ZoneId; | |
| import java.time.format.DateTimeFormatter; | |
| import java.util.logging.ConsoleHandler; | |
| import java.util.logging.Level; | |
| import java.util.logging.Logger; | |
| import com.sun.net.httpserver.HttpServer; | |
| import com.sun.net.httpserver.SimpleFileServer; | |
| import com.sun.net.httpserver.SimpleFileServer.OutputLevel; | |
| import jdk.test.lib.Platform; | |
| import jdk.test.lib.net.URIBuilder; | |
| import jdk.test.lib.util.FileUtils; | |
| import org.testng.annotations.AfterTest; | |
| import org.testng.annotations.BeforeTest; | |
| import org.testng.annotations.DataProvider; | |
| import org.testng.annotations.Test; | |
| import org.testng.SkipException; | |
| import static java.net.http.HttpClient.Builder.NO_PROXY; | |
| import static java.nio.charset.StandardCharsets.UTF_8; | |
| import static java.nio.file.StandardOpenOption.CREATE; | |
| import static org.testng.Assert.*; | |
| public class SimpleFileServerTest { | |
| static final Class<NullPointerException> NPE = NullPointerException.class; | |
| static final Class<IllegalArgumentException> IAE = IllegalArgumentException.class; | |
| static final Class<UncheckedIOException> UIOE = UncheckedIOException.class; | |
| static final Path CWD = Path.of(".").toAbsolutePath(); | |
| static final Path TEST_DIR = CWD.resolve("SimpleFileServerTest"); | |
| static final InetSocketAddress LOOPBACK_ADDR = new InetSocketAddress(InetAddress.getLoopbackAddress(), 0); | |
| static final boolean ENABLE_LOGGING = true; | |
| static final Logger LOGGER = Logger.getLogger("com.sun.net.httpserver"); | |
| static final String EXPECTED_LAST_MODIFIED_OF_FAVICON = "Mon, 23 May 1995 11:11:11 GMT"; | |
| @BeforeTest | |
| public void setup() throws IOException { | |
| if (ENABLE_LOGGING) { | |
| ConsoleHandler ch = new ConsoleHandler(); | |
| LOGGER.setLevel(Level.ALL); | |
| ch.setLevel(Level.ALL); | |
| LOGGER.addHandler(ch); | |
| } | |
| if (Files.exists(TEST_DIR)) { | |
| FileUtils.deleteFileTreeWithRetry(TEST_DIR); | |
| } | |
| Files.createDirectories(TEST_DIR); | |
| } | |
| @Test | |
| public void testFileGET() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testFileGET")); | |
| var file = Files.writeString(root.resolve("aFile.txt"), "some text", CREATE); | |
| var lastModified = getLastModified(file); | |
| var expectedLength = Long.toString(Files.size(file)); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "aFile.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.body(), "some text"); | |
| assertEquals(response.headers().firstValue("content-type").get(), "text/plain"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testDirectoryGET() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>Directory listing for /</h1> | |
| <ul> | |
| <li><a href="aFile.txt">aFile.txt</a></li> | |
| </ul> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testDirectoryGET")); | |
| var file = Files.writeString(root.resolve("aFile.txt"), "some text", CREATE); | |
| var lastModified = getLastModified(root); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "text/html; charset=UTF-8"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testFavIconGET() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testFavIconGET")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| // expect built-in icon | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "favicon.ico")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "image/x-icon"); | |
| assertEquals(response.headers().firstValue("last-modified").get(), EXPECTED_LAST_MODIFIED_OF_FAVICON); | |
| // expect custom (and broken) icon | |
| var file = Files.writeString(root.resolve("favicon.ico"), "broken icon", CREATE); | |
| try { | |
| var lastModified = getLastModified(file); | |
| var expectedLength = Long.toString(Files.size(file)); | |
| response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "application/octet-stream"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| } finally { | |
| Files.delete(file); | |
| } | |
| // expect built-in icon | |
| response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "image/x-icon"); | |
| assertEquals(response.headers().firstValue("last-modified").get(), EXPECTED_LAST_MODIFIED_OF_FAVICON); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testFavIconHEAD() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testFavIconHEAD")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "favicon.ico")) | |
| .method("HEAD", BodyPublishers.noBody()).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "image/x-icon"); | |
| assertEquals(response.headers().firstValue("last-modified").get(), EXPECTED_LAST_MODIFIED_OF_FAVICON); | |
| assertEquals(response.body(), ""); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testFileHEAD() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testFileHEAD")); | |
| var file = Files.writeString(root.resolve("aFile.txt"), "some text", CREATE); | |
| var lastModified = getLastModified(file); | |
| var expectedLength = Long.toString(Files.size(file)); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "aFile.txt")) | |
| .method("HEAD", BodyPublishers.noBody()).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "text/plain"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| assertEquals(response.body(), ""); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testDirectoryHEAD() throws Exception { | |
| var expectedLength = Integer.toString( | |
| (openHTML + """ | |
| <h1>Directory listing for /</h1> | |
| <ul> | |
| <li><a href="aFile.txt">aFile.txt</a></li> | |
| </ul> | |
| """ + closeHTML).getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testDirectoryHEAD")); | |
| var file = Files.writeString(root.resolve("aFile.txt"), "some text", CREATE); | |
| var lastModified = getLastModified(root); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "")) | |
| .method("HEAD", BodyPublishers.noBody()).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), "text/html; charset=UTF-8"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| assertEquals(response.body(), ""); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @DataProvider | |
| public Object[][] indexFiles() { | |
| var fileContent = openHTML + """ | |
| <h1>This is an index file</h1> | |
| """ + closeHTML; | |
| var dirListing = openHTML + """ | |
| <h1>Directory listing for /</h1> | |
| <ul> | |
| </ul> | |
| """ + closeHTML; | |
| return new Object[][] { | |
| {"1", "index.html", "text/html", "116", fileContent, true}, | |
| {"2", "index.htm", "text/html", "116", fileContent, true}, | |
| {"3", "index.txt", "text/html; charset=UTF-8", "134", dirListing, false} | |
| }; | |
| } | |
| @Test(dataProvider = "indexFiles") | |
| public void testDirectoryWithIndexGET(String id, | |
| String filename, | |
| String contentType, | |
| String contentLength, | |
| String expectedBody, | |
| boolean serveIndexFile) throws Exception { | |
| var root = Files.createDirectories(TEST_DIR.resolve("testDirectoryWithIndexGET"+id)); | |
| var lastModified = getLastModified(root); | |
| if (serveIndexFile) { | |
| var file = Files.writeString(root.resolve(filename), expectedBody, CREATE); | |
| lastModified = getLastModified(file); | |
| } | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.headers().firstValue("content-type").get(), contentType); | |
| assertEquals(response.headers().firstValue("content-length").get(), contentLength); | |
| assertEquals(response.headers().firstValue("last-modified").get(), lastModified); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| if (serveIndexFile) { | |
| Files.delete(root.resolve(filename)); | |
| } | |
| } | |
| } | |
| @Test | |
| public void testNotReadableFileGET() throws Exception { | |
| if (!Platform.isWindows()) { // not applicable on Windows | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/aFile.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testNotReadableFileGET")); | |
| var file = Files.writeString(root.resolve("aFile.txt"), "some text", CREATE); | |
| file.toFile().setReadable(false, false); | |
| assert !Files.isReadable(file); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "aFile.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| file.toFile().setReadable(true, false); | |
| } | |
| } | |
| } | |
| @Test | |
| public void testNotReadableSegmentGET() throws Exception { | |
| if (!Platform.isWindows()) { // not applicable on Windows | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/dir/aFile.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testNotReadableSegmentGET")); | |
| var dir = Files.createDirectory(root.resolve("dir")); | |
| var file = Files.writeString(dir.resolve("aFile.txt"), "some text", CREATE); | |
| dir.toFile().setReadable(false, false); | |
| assert !Files.isReadable(dir); | |
| assert Files.isReadable(file); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "dir/aFile.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| dir.toFile().setReadable(true, false); | |
| } | |
| } | |
| } | |
| @Test | |
| public void testInvalidRequestURIGET() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/aFile?#.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testInvalidRequestURIGET")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "aFile?#.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testNotFoundGET() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/doesNotExist.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testNotFoundGET")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "doesNotExist.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testNotFoundHEAD() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/doesNotExist.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testNotFoundHEAD")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "doesNotExist.txt")) | |
| .method("HEAD", BodyPublishers.noBody()).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), ""); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testSymlinkGET() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/symlink</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testSymlinkGET")); | |
| var symlink = root.resolve("symlink"); | |
| var target = Files.writeString(root.resolve("target.txt"), "some text", CREATE); | |
| createSymLink(symlink, target); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "symlink")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testSymlinkSegmentGET() throws Exception { | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/symlink/aFile.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var root = Files.createDirectory(TEST_DIR.resolve("testSymlinkSegmentGET")); | |
| var symlink = root.resolve("symlink"); | |
| var target = Files.createDirectory(root.resolve("target")); | |
| Files.writeString(target.resolve("aFile.txt"), "some text", CREATE); | |
| createSymLink(symlink, target); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "symlink/aFile.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| private void createSymLink(Path symlink, Path target) { | |
| try { | |
| Files.createSymbolicLink(symlink, target); | |
| } catch (UnsupportedOperationException uoe) { | |
| throw new SkipException("sym link creation not supported", uoe); | |
| } catch (IOException ioe) { | |
| throw new SkipException("probably insufficient privileges to create sym links (Windows)", ioe); | |
| } | |
| } | |
| @Test | |
| public void testHiddenFileGET() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testHiddenFileGET")); | |
| var file = createHiddenFile(root); | |
| var fileName = file.getFileName().toString(); | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/""" + fileName + | |
| """ | |
| </p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, fileName)).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testHiddenSegmentGET() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testHiddenSegmentGET")); | |
| var file = createFileInHiddenDirectory(root); | |
| var expectedBody = openHTML + """ | |
| <h1>File not found</h1> | |
| <p>/.hiddenDirectory/aFile.txt</p> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, ".hiddenDirectory/aFile.txt")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| assertEquals(response.body(), expectedBody); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| private Path createHiddenFile(Path root) throws IOException { | |
| Path file; | |
| if (Platform.isWindows()) { | |
| file = Files.createFile(root.resolve("aFile.txt")); | |
| Files.setAttribute(file, "dos:hidden", true, LinkOption.NOFOLLOW_LINKS); | |
| } else { | |
| file = Files.writeString(root.resolve(".aFile.txt"), "some text", CREATE); | |
| } | |
| assertTrue(Files.isHidden(file)); | |
| return file; | |
| } | |
| private Path createFileInHiddenDirectory(Path root) throws IOException { | |
| Path dir; | |
| Path file; | |
| if (Platform.isWindows()) { | |
| dir = Files.createDirectory(root.resolve("hiddenDirectory")); | |
| Files.setAttribute(dir, "dos:hidden", true, LinkOption.NOFOLLOW_LINKS); | |
| } else { | |
| dir = Files.createDirectory(root.resolve(".hiddenDirectory")); | |
| } | |
| file = Files.writeString(dir.resolve("aFile.txt"), "some text", CREATE); | |
| assertTrue(Files.isHidden(dir)); | |
| assertFalse(Files.isHidden(file)); | |
| return file; | |
| } | |
| @Test | |
| public void testMovedPermanently() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testMovedPermanently")); | |
| Files.createDirectory(root.resolve("aDirectory")); | |
| var expectedBody = openHTML + """ | |
| <h1>Directory listing for /aDirectory/</h1> | |
| <ul> | |
| </ul> | |
| """ + closeHTML; | |
| var expectedLength = Integer.toString(expectedBody.getBytes(UTF_8).length); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY) | |
| .followRedirects(HttpClient.Redirect.NEVER).build(); | |
| var uri = uri(server, "aDirectory"); | |
| var request = HttpRequest.newBuilder(uri).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 301); | |
| assertEquals(response.headers().firstValue("content-length").get(), "0"); | |
| assertEquals(response.headers().firstValue("location").get(), "/aDirectory/"); | |
| // tests that query component is preserved during redirect | |
| var uri2 = uri(server, "aDirectory", "query"); | |
| var req2 = HttpRequest.newBuilder(uri2).build(); | |
| var res2 = client.send(req2, BodyHandlers.ofString()); | |
| assertEquals(res2.statusCode(), 301); | |
| assertEquals(res2.headers().firstValue("content-length").get(), "0"); | |
| assertEquals(res2.headers().firstValue("location").get(), "/aDirectory/?query"); | |
| } | |
| { // tests that redirect to returned relative URI works | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY) | |
| .followRedirects(HttpClient.Redirect.ALWAYS).build(); | |
| var uri = uri(server, "aDirectory"); | |
| var request = HttpRequest.newBuilder(uri).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 200); | |
| assertEquals(response.body(), expectedBody); | |
| assertEquals(response.headers().firstValue("content-type").get(), "text/html; charset=UTF-8"); | |
| assertEquals(response.headers().firstValue("content-length").get(), expectedLength); | |
| } | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @Test | |
| public void testNull() { | |
| final var addr = InetSocketAddress.createUnresolved("foo", 8080); | |
| final var path = Path.of("/tmp"); | |
| final var levl = OutputLevel.INFO; | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(null, null, null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(null, null, levl)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(null, path, null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(null, path, levl)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(addr, null, null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(addr, null, levl)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileServer(addr, path, null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createFileHandler(null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createOutputFilter(null, null)); | |
| assertThrows(NPE, () -> SimpleFileServer.createOutputFilter(null, levl)); | |
| assertThrows(NPE, () -> SimpleFileServer.createOutputFilter(System.out, null)); | |
| } | |
| @Test | |
| public void testInitialSlashContext() { | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, TEST_DIR, OutputLevel.INFO); | |
| server.removeContext("/"); // throws if no context | |
| server.stop(0); | |
| } | |
| @Test | |
| public void testBound() { | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, TEST_DIR, OutputLevel.INFO); | |
| var boundAddr = server.getAddress(); | |
| server.stop(0); | |
| assertTrue(boundAddr.getAddress() != null); | |
| assertTrue(boundAddr.getPort() > 0); | |
| } | |
| @Test | |
| public void testIllegalPath() throws Exception { | |
| var addr = LOOPBACK_ADDR; | |
| { // not a directory | |
| Path p = Files.createFile(TEST_DIR.resolve("aFile")); | |
| assert !Files.isDirectory(p); | |
| var iae = expectThrows(IAE, () -> SimpleFileServer.createFileServer(addr, p, OutputLevel.INFO)); | |
| assertTrue(iae.getMessage().contains("not a directory")); | |
| } | |
| { // does not exist | |
| Path p = TEST_DIR.resolve("doesNotExist"); | |
| assert !Files.exists(p); | |
| var iae = expectThrows(IAE, () -> SimpleFileServer.createFileServer(addr, p, OutputLevel.INFO)); | |
| assertTrue(iae.getMessage().contains("does not exist")); | |
| } | |
| { // not readable | |
| if (!Platform.isWindows()) { // not applicable on Windows | |
| Path p = Files.createDirectory(TEST_DIR.resolve("aDir")); | |
| p.toFile().setReadable(false, false); | |
| assert !Files.isReadable(p); | |
| try { | |
| var iae = expectThrows(IAE, () -> SimpleFileServer.createFileServer(addr, p, OutputLevel.INFO)); | |
| assertTrue(iae.getMessage().contains("not readable")); | |
| } finally { | |
| p.toFile().setReadable(true, false); | |
| } | |
| } | |
| } | |
| } | |
| @Test | |
| public void testUncheckedIOException() { | |
| var addr = InetSocketAddress.createUnresolved("foo", 8080); | |
| assertThrows(UIOE, () -> SimpleFileServer.createFileServer(addr, TEST_DIR, OutputLevel.INFO)); | |
| assertThrows(UIOE, () -> SimpleFileServer.createFileServer(addr, TEST_DIR, OutputLevel.VERBOSE)); | |
| } | |
| @Test | |
| public void testXss() throws Exception { | |
| var root = Files.createDirectory(TEST_DIR.resolve("testXss")); | |
| var server = SimpleFileServer.createFileServer(LOOPBACK_ADDR, root, OutputLevel.VERBOSE); | |
| server.start(); | |
| try { | |
| var client = HttpClient.newBuilder().proxy(NO_PROXY).build(); | |
| var request = HttpRequest.newBuilder(uri(server, "beginDelim%3C%3EEndDelim")).build(); | |
| var response = client.send(request, BodyHandlers.ofString()); | |
| assertEquals(response.statusCode(), 404); | |
| assertTrue(response.body().contains("beginDelim%3C%3EEndDelim")); | |
| assertTrue(response.body().contains("File not found")); | |
| } finally { | |
| server.stop(0); | |
| } | |
| } | |
| @AfterTest | |
| public void teardown() throws IOException { | |
| if (Files.exists(TEST_DIR)) { | |
| FileUtils.deleteFileTreeWithRetry(TEST_DIR); | |
| } | |
| } | |
| static final String openHTML = """ | |
| <!DOCTYPE html> | |
| <html> | |
| <head> | |
| <meta charset="utf-8"/> | |
| </head> | |
| <body> | |
| """; | |
| static final String closeHTML = """ | |
| </body> | |
| </html> | |
| """; | |
| static URI uri(HttpServer server, String path) { | |
| return URIBuilder.newBuilder() | |
| .host("localhost") | |
| .port(server.getAddress().getPort()) | |
| .scheme("http") | |
| .path("/" + path) | |
| .buildUnchecked(); | |
| } | |
| static URI uri(HttpServer server, String path, String query) { | |
| return URIBuilder.newBuilder() | |
| .host("localhost") | |
| .port(server.getAddress().getPort()) | |
| .scheme("http") | |
| .path("/" + path) | |
| .query(query) | |
| .buildUnchecked(); | |
| } | |
| static String getLastModified(Path path) throws IOException { | |
| return Files.getLastModifiedTime(path).toInstant().atZone(ZoneId.of("GMT")) | |
| .format(DateTimeFormatter.RFC_1123_DATE_TIME); | |
| } | |
| } |