java-http/Main.java

113 lines
3.4 KiB
Java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
public class Main {
public static void main(String[] args) {
int attempts = 1;
try (ServerSocket serverSocket = new ServerSocket(8080)) {
System.out.println("listening on socket 8080...");
while (true) {
// .accept() blocks until we have a client
try (Socket client = serverSocket.accept()) {
handleClient(client);
} catch (IOException e) {
System.out.printf("Error accepting client: %s\n", e);
}
}
} catch (IOException e) {
System.out.printf("Error opening port (attempt %s) %s:\n", attempts, e);
if (attempts > 3) {
System.out.println("This isn't working. Shutting down.");
System.exit(1);
}
try {
Thread.sleep(1000);
} catch (InterruptedException ignore) {
}
}
}
public static void handleClient(Socket client) throws IOException {
System.out.println("Debug: got new client" + client.toString());
BufferedReader br = new BufferedReader(new InputStreamReader(client.getInputStream()));
StringBuilder requestBuilder = new StringBuilder();
String line;
while (!(line = br.readLine()).isBlank()) {
requestBuilder.append(line + "\r\n");
}
String request = requestBuilder.toString();
// System.out.println(request);
// PARSE THE REQUEST
String[] requestLines = request.split("\r\n");
String[] requestLine = requestLines[0].split(" ");
String method = requestLine[0];
String path = requestLine[1];
String version = requestLine[2];
String host = requestLines[1].split(" ")[1];
List<String> headers = new ArrayList<>();
// we start the loop from 2 because 0 is GET / HTTP/1.1 and 1 is the host
for (int h = 2; h < requestLines.length; h++) {
String header = requestLines[h];
headers.add(header);
}
String accessLog = String.format("Client %s, method %s, path %s, version %s, host %s, headers %s",
client.toString(),
method, path, version, host, headers.toString());
System.out.println(accessLog);
Path filepath = getFilePath(path);
if (Files.exists(filepath)) {
// file exist
String contentType = guessContentType(filepath);
sendResponse(client, "200 OK", contentType, Files.readAllBytes(filepath));
} else {
// 404
byte[] notFoundContent = "<h1>404 Something went wrong</h1>".getBytes();
sendResponse(client, "404 Not Found", "text/html", notFoundContent);
}
}
private static void sendResponse(Socket client, String status, String contentType, byte[] content)
throws IOException {
OutputStream clientOutput = client.getOutputStream();
clientOutput.write("HTTP/1.1 200 OK\r\n".getBytes());
clientOutput.write("ContentType: text/html\r\n".getBytes());
clientOutput.write("\r\n".getBytes());
clientOutput.write(content);
// These (windows) newlines indicate that the response is over
clientOutput.write("\r\n\r\n".getBytes());
clientOutput.flush();
client.close();
}
private static String guessContentType(Path filepath) throws IOException {
return Files.probeContentType(filepath);
}
private static Path getFilePath(String path) {
if ("/".equals(path)) {
path = "/index.html";
}
return Paths.get("/home/andrzej/dev/java-http", path);
}
}