/*
* Copyright (c) 2005, 2017, 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.
*/
import java.util.*;
import java.util.concurrent.*;
import java.util.logging.*;
import java.io.*;
import java.net.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.security.*;
import javax.net.ssl.*;
import com.sun.net.httpserver.*;
/**
* Implements a basic static content HTTP file server handler
* which understands text/html, text/plain content types
*
* Must be given an abs pathname to the document root.
* Directory listings together with text + html files
* can be served.
*
*/
public class FileServerHandler implements HttpHandler {
String docroot;
public FileServerHandler (String docroot) {
this.docroot = docroot;
}
int invocation = 1;
public void handle (HttpExchange t)
throws IOException
{
InputStream is = t.getRequestBody();
Headers map = t.getRequestHeaders();
Headers rmap = t.getResponseHeaders();
URI uri = t.getRequestURI();
String path = uri.getPath();
is.readAllBytes();
is.close();
File f = new File (docroot, path);
Path filepath = Files.isSymbolicLink(f.toPath()) ? Files.readSymbolicLink(f.toPath()) : f.toPath();
f = filepath.toFile();
if (!f.exists()) {
notfound (t, path);
return;
}
String fixedrequest = map.getFirst ("XFixed");
String method = t.getRequestMethod();
if (method.equals ("HEAD")) {
rmap.set ("Content-length", Long.toString (f.length()));
t.sendResponseHeaders (200, -1);
t.close();
} else if (!method.equals("GET")) {
t.sendResponseHeaders (405, -1);
t.close();
return;
}
if (path.endsWith (".html") || path.endsWith (".htm")) {
rmap.set ("Content-type", "text/html");
} else {
rmap.set ("Content-type", "text/plain");
}
if (f.isDirectory()) {
if (!path.endsWith ("/")) {
moved (t);
return;
}
rmap.set ("Content-Type", "text/html");
t.sendResponseHeaders (200, 0);
String[] list = f.list();
OutputStream os = t.getResponseBody();
PrintStream p = new PrintStream (os);
p.println ("Directory listing for: " + path+ "
");
p.println ("
");
p.flush();
p.close();
} else {
int clen;
if (fixedrequest != null) {
clen = (int) f.length();
} else {
clen = 0;
}
t.sendResponseHeaders (200, clen);
try (OutputStream os = t.getResponseBody(); FileInputStream fis = new FileInputStream (f)) {
fis.transferTo(os);
}
}
}
void moved (HttpExchange t) throws IOException {
Headers req = t.getRequestHeaders();
Headers map = t.getResponseHeaders();
URI uri = t.getRequestURI();
String host = req.getFirst ("Host");
String location = "http://"+host+uri.getPath() + "/";
map.set ("Content-Type", "text/html");
map.set ("Location", location);
t.sendResponseHeaders (301, -1);
t.close();
}
void notfound (HttpExchange t, String p) throws IOException {
t.getResponseHeaders().set ("Content-Type", "text/html");
t.sendResponseHeaders (404, 0);
OutputStream os = t.getResponseBody();
String s = "File not found
";
s = s + p + "";
os.write (s.getBytes());
os.close();
t.close();
}
}