https://bz.apache.org/bugzilla/show_bug.cgi?id=68037
Bug ID: 68037
Summary: Async Servlet implementation doesn't allow setting a
response status
Product: Tomcat 9
Version: 9.0.82
Hardware: PC
OS: Mac OS X 10.1
Status: NEW
Severity: blocker
Priority: P2
Component: Catalina
Assignee: dev@tomcat.apache.org
Reporter: adwsi...@gmail.com
Target Milestone: -----
Please find the below reproducible test case to demonstrate the issue.
It is expected both testTimeoutGets408Async and testTimeoutGets408Sync to pass,
but testTimeoutGets408Async fails.
Test Case:
import org.apache.catalina.Context;
import org.apache.catalina.LifecycleException;
import org.apache.catalina.Wrapper;
import org.apache.catalina.connector.ClientAbortException;
import org.apache.catalina.connector.Connector;
import org.apache.catalina.core.StandardHost;
import org.apache.catalina.startup.Tomcat;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import javax.servlet.ReadListener;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.Socket;
import java.net.SocketTimeoutException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.Assertions.fail;
public class ReproducibleTest {
static Tomcat tomcat;
@BeforeAll
static void setup() throws LifecycleException {
tomcat = new Tomcat();
ExecutorService executorService = Executors.newFixedThreadPool(5);
Context ctx = tomcat.addContext("", new File(".").getAbsolutePath());
Wrapper asyncWrapper = Tomcat.addServlet(ctx, "TestAsync", new
TestAsyncServlet());
Wrapper syncWrapper = Tomcat.addServlet(ctx, "TestSync", new
SyncServlet());
asyncWrapper.setAsyncSupported(true);
StandardHost host = (StandardHost) tomcat.getHost();
host.setErrorReportValveClass(null);
Connector connector = new Connector();
connector.setProperty("address", "http://localhost";);
connector.setPort(8000);
connector.setProperty("connectionTimeout", String.valueOf(100));
connector.getProtocolHandler().setExecutor(executorService);
tomcat.getService().addConnector(connector);
ctx.addServletMappingDecoded("/async/*", "TestAsync");
ctx.addServletMappingDecoded("/sync/*", "TestSync");
tomcat.start();
}
@AfterAll
static void destroy() throws LifecycleException {
tomcat.stop();
tomcat.destroy();
}
@Test
void testTimeoutGets408Async() throws LifecycleException, IOException {
try (Socket s = new Socket("localhost", 8000)) {
String request = "GET /async HTTP/1.1\r\nHost:
localhost\r\ncontent-length: 101\r\n\r\n";
sendBadRequest(s, request, 408);
}
}
@Test
void testTimeoutGets408Sync() throws LifecycleException, IOException {
try (Socket s = new Socket("localhost", 8000)) {
String request = "GET /sync HTTP/1.1\r\nHost:
localhost\r\ncontent-length: 100\r\n\r\n";
sendBadRequest(s, request, 408);
}
}
private static void sendBadRequest(Socket socket, String request, int
expectedStatusCode) throws IOException {
OutputStream os = socket.getOutputStream();
os.write(request.getBytes(UTF_8));
InputStream is = socket.getInputStream();
BufferedReader reader = new BufferedReader(new InputStreamReader(is,
UTF_8));
String opening = reader.readLine();
assertNotNull(opening, "Didn't get back a response");
StringBuilder sb = new StringBuilder(opening);
try {
assertTrue(opening.startsWith("HTTP/1.1 " + expectedStatusCode),
"expected status code " + expectedStatusCode + " but got " + opening);
boolean connectionClose = false;
while (reader.ready()) {
String line = reader.readLine();
if (line == null) {
break;
}
sb.append("\n").append(line);
if ("connection: close".equalsIgnoreCase(line)) {
connectionClose = true;
}
assertFalse(line.contains("Exception Report"));
assertFalse(line.contains("Status Report"));
}
assertTrue(connectionClose, "No 'Connection: close' header seen");
} catch (Throwable t) {
fail("Response:\n" + sb, t);
}
}
static final class SyncServlet extends HttpServlet {
@Override
protected void service(HttpServletRequest req, HttpServletResponse
resp) throws ServletException, IOException {
try {
while (req.getInputStream().read() != -1) ;
resp.setStatus(200);
resp.flushBuffer();
} catch (ClientAbortException e) {
resp.sendError(408);
}
}
}
static final class TestAsyncServlet extends HttpServlet {
@Override
protected void service(HttpServletRequest req, HttpServletResponse
resp) throws ServletException, IOException {
req.startAsync();
req.getInputStream().setReadListener(new ReadListener() {
@Override
public void onDataAvailable() throws IOException {
while (req.getInputStream().isReady()) {
req.getInputStream().read();
}
}
@Override
public void onAllDataRead() throws IOException {
resp.setStatus(200);
req.getAsyncContext().complete();
}
@Override
public void onError(Throwable t) {
if (t instanceof SocketTimeoutException) {
try {
resp.sendError(408);
} catch (IOException e) {
e.printStackTrace();
}
}
req.getAsyncContext().complete();
}
});
}
}
}
--
You are receiving this mail because:
You are the assignee for the bug.
---------------------------------------------------------------------
To unsubscribe, e-mail: dev-unsubscr...@tomcat.apache.org
For additional commands, e-mail: dev-h...@tomcat.apache.org
