支持 Keep-Alive 的 HTTP 服务器

Question

我正在尝试用 Java 创建一个能够提供保持连接的 http 服务器。我正在使用 com.sun.net.httpserver.HttpServer 类。

import java.io.IOException;
import java.io.OutputStream;
import java.net.InetSocketAddress;

import com.sun.net.httpserver.Headers;
import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.HttpServer;

public class httpHandler implements HttpHandler {

private String resp = "<?xml version='1.0'?><root-node></root-node>";

private OutputStream os = null;

public void handle(HttpExchange t) throws IOException {
    System.out.println("Handling message...");
    java.io.InputStream is = t.getRequestBody();

    System.out.println("Got request body. Reading request body...");
    byte[] b = new byte[500];
    is.read(b);
    System.out.println("This is the request: " + new String(b));

    String response = resp;
    Headers header = t.getResponseHeaders();
    header.add("Connection", "Keep-Alive");
    header.add("Keep-Alive", "timeout=14 max=100");
    header.add("Content-Type", "application/soap+xml");
    t.sendResponseHeaders(200, response.length());

    if(os == null) {
        os = t.getResponseBody();
    }

    os.write(response.getBytes());

    System.out.println("Done with exchange. Closing connection");
    os.close();
}

public static void main(String[] args) {
    HttpServer server = null;
    try {
        server = HttpServer.create(new InetSocketAddress(8080), 5);
    } catch (IOException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
    }
    server.createContext("/", new httpHandler());
    server.setExecutor(null); // creates a default executor
    System.out.println("Starting server...");
    server.start();
}
}

客户端没有关闭连接。服务器似乎在交换发生后直接关闭它。我尝试删除 os.close 行，但服务器不会回复第二个请求。但它也不会关闭它。我有一种感觉，它涉及在主代码中使用服务器对象做一些事情，但我不知道是什么。谷歌也没有出现太多。

这里有人有什么想法吗？任何帮助将非常感激。

Answer 1

看起来问题是您没有从请求中耗尽所有数据。你应该一直做is.read()直到它返回-1，然后关闭它。

由于您没有耗尽请求，因此仍有字节剩余。服务器不能简单地“跳转”到下一个请求；它不像磁盘，更像磁带。服务器必须读取（并丢弃）当前请求中的所有数据，然后才能到达下一个请求。

无限制可以用来攻击服务器；所以服务器只会尝试消耗到一个限制，默认情况下是 64K。您可能收到大于 64K 的请求。

通常处理程序应该首先读取整个请求。否则它怎么知道如何处理请求？

更严重的是，如果请求没有先被耗尽，可能会发生死锁。客户端通常很简单：他们编写请求，然后读取响应。如果服务器在读取所有请求之前写入响应，则客户端可能仍在写入请求。两者都在互相写信，但都没有阅读。如果缓冲区已满，则我们处于死锁状态，两者都在写入时被阻塞。注意 write() 没有超时！

Answer 2

为什么只有在前一个为空的情况下才能获得 ResponseBody OutputStream？

if(os == null) {
    os = t.getResponseBody();
}

每次获取 OutputStream 可能会更好，因为您无法确定前一个是否与当前请求相同。

Answer 3

尝试拨打t.close()，服务器会回复

Answer 4

client 发送 keep-alive 标头。如果这样做，HttpServer 不会关闭连接。您无需为此做任何事情。