/*
* Copyright (c) 2005, 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.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintStream;
import java.net.InetSocketAddress;
import java.net.URI;
import java.util.concurrent.Executors;
import org.jboss.com.sun.net.httpserver.Headers;
import org.jboss.com.sun.net.httpserver.HttpContext;
import org.jboss.com.sun.net.httpserver.HttpExchange;
import org.jboss.com.sun.net.httpserver.HttpHandler;
import org.jboss.com.sun.net.httpserver.HttpServer;
/**
* Implements a basic static content HTTP server
* 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 {
public static void main (String[] args) throws Exception {
if (args.length != 3) {
System.out.println ("usage: java FileServerHandler rootDir port logfilename");
System.exit(1);
}
String rootDir = args[0];
int port = Integer.parseInt (args[1]);
String logfile = args[2];
HttpServer server = HttpServer.create (new InetSocketAddress (8000), 0);
HttpHandler h = new FileServerHandler (rootDir);
HttpContext c = server.createContext ("/", h);
c.getFilters().add (new LogFilter (new File (logfile)));
server.setExecutor (Executors.newCachedThreadPool());
server.start ();
}
String docroot;
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();
while (is.read () != -1) ;
is.close();
File f = new File (docroot, path);
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 ("<h2>Directory listing for: " + path+ "</h2>");
p.println ("<ul>");
for (int i=0; i<list.length; i++) {
p.println ("<li><a href=\""+list[i]+"\">"+list[i]+"</a></li>");
}
p.println ("</ul><p><hr>");
p.flush();
p.close();
} else {
int clen;
if (fixedrequest != null) {
clen = (int) f.length();
} else {
clen = 0;
}
t.sendResponseHeaders (200, clen);
OutputStream os = t.getResponseBody();
FileInputStream fis = new FileInputStream (f);
int count = 0;
try {
byte[] buf = new byte [16 * 1024];
int len;
while ((len=fis.read (buf)) != -1) {
os.write (buf, 0, len);
count += len;
}
} catch (IOException e) {
e.printStackTrace();
}
fis.close();
os.close();
}
}
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 = "<h2>File not found</h2>";
s = s + p + "<p>";
os.write (s.getBytes());
os.close();
t.close();
}
}