private boolean handleFileRequest(Path path, final HttpResponse response) {
WebServerLog.log(this, "Request handler checking file: " + path.toString());
final File file = new File(path.toString());
HttpEntity entity = null;
// bad request
if (!path.startsWith(baseDir) || !file.exists() || !file.canRead()) {
WebServerLog.log(this, "File " + path.toString() + " is invalid");
response.setStatusCode(HttpStatus.SC_NOT_FOUND);
response.setReasonPhrase(ReasonPhrases.NOT_FOUND);
entity = createNotFoundEntity();
response.setEntity(entity);
return false;
}
assert response.getStatusLine().getStatusCode() == HttpStatus.SC_OK;
WebServerLog.log(this, "File " + path.toString() + " is locked and loaded!! THIS IS SPARTA!");
if (file.isDirectory()) {
// directory request
String s = "Directory " + path.toString() + ":\n";
for (final String fileName : file.list()) {
s += fileName + "\n";
}
try {
entity = new StringEntity(s);
response.setEntity(entity);
} catch (final UnsupportedEncodingException e) {
System.err.println("Couldn't set response entity to directory content");
e.printStackTrace();
}
} else {
// file request
entity = new FileEntity(file);
response.setEntity(entity);
}
return file.isFile();
}
private HttpEntity createNotFoundEntity() {
HttpEntity entity = null;
try {
entity =
new FileEntity(
new File(getClass().getResource("not_found.html").toURI()), ContentType.TEXT_HTML);
} catch (final URISyntaxException e) {
WebServerLog.log(this, "Request handler failed to load not found file");
e.printStackTrace();
}
return entity;
}
private void handleRequest(RequestObject reqObj) {
final Path path = reqObj.getPath();
final HttpServerConnection conn = reqObj.getConn();
final HttpRequest req = reqObj.getRequest();
WebServerLog.log(this, "Request handler got a request:\n" + req.toString());
if (!req.getRequestLine().getMethod().toUpperCase().equals("GET")) {
return;
}
final HttpResponse response =
new BasicHttpResponse(HttpVersion.HTTP_1_1, HttpStatus.SC_OK, ReasonPhrases.OK);
boolean applyFilters;
applyFilters = handleFileRequest(path, response);
if (applyFilters) {
filterChain.reset(path.toString());
filterChain.doFilter(reqObj.getRequest(), response);
}
try {
proc.process(response, new BasicHttpContext());
addContentType(path, response);
} catch (HttpException | IOException e1) {
WebServerLog.log(this, "Request handler failed to 'process' response after filtering");
e1.printStackTrace();
}
WebServerLog.log(this, "Request handler is sending a response:\n" + response.toString());
try {
conn.sendResponseHeader(response);
conn.sendResponseEntity(response);
} catch (HttpException | IOException e) {
WebServerLog.log(this, "Request handler has encountered error while sending response");
e.printStackTrace();
}
}
@Override
public void run() {
buildProcessor();
WebServerLog.log(this, "Request handler has started running");
RequestObject reqObj = null;
while (true) {
try {
reqObj = requestsQueue.take();
// timer.schedule(new MyTimerTask(Thread.currentThread()),
// // 5 * 60 * 1000);
// 5 * 1000);
WebServerLog.log(this, "Request handler got a new request");
handleRequest(reqObj);
reqObj.getConn().shutdown();
} catch (final InterruptedException e) {
WebServerLog.log(this, "Request handler was interrupted");
System.err.println("Request handler was interrupted");
try {
reqObj.getConn().shutdown();
} catch (final IOException e1) {
System.err.println("Request handler failed to shutdown connection");
e1.printStackTrace();
}
e.printStackTrace();
} catch (final IOException e) {
WebServerLog.log(this, "Request handler failed to close connection");
System.err.println("Request handler failed to close connection");
e.printStackTrace();
}
}
}
private void addContentType(Path path, final HttpResponse response) {
if (response.getEntity() != null && response.getEntity().getContentType() != null) {
WebServerLog.log(
this,
"Response entity already contains content type: "
+ response.getEntity().getContentType().getValue());
return;
}
if (response.containsHeader(HttpHeaders.CONTENT_TYPE)) {
response.removeHeaders(HttpHeaders.CONTENT_TYPE);
}
String contentType = null;
final int i = path.toString().lastIndexOf('.');
if (i >= 0) {
String extension = path.toString().substring(i + 1);
contentType = extension2contentType.get(extension);
}
if (contentType == null) {
return;
}
response.addHeader(HttpHeaders.CONTENT_TYPE, contentType);
}
