Update JsonMappingExceptionMapper “Early EOF” logic

signalapp · May 1, 2024 · cf307db · cf307db
1 parent dcfca4d
commit cf307db
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...ce/src/main/java/org/whispersystems/textsecuregcm/mappers/JsonMappingExceptionMapper.java b/...ce/src/main/java/org/whispersystems/textsecuregcm/mappers/JsonMappingExceptionMapper.java
@@ -3,15 +3,15 @@
 import com.fasterxml.jackson.databind.JsonMappingException;
 import javax.ws.rs.core.Response;
 import javax.ws.rs.ext.ExceptionMapper;
-import java.util.concurrent.TimeoutException;
 
 public class JsonMappingExceptionMapper implements ExceptionMapper<JsonMappingException> {
   @Override
   public Response toResponse(final JsonMappingException exception) {
-    if (exception.getCause() instanceof TimeoutException) {
+    if (exception.getCause() instanceof java.util.concurrent.TimeoutException) {
       return Response.status(Response.Status.REQUEST_TIMEOUT).build();
     }
-    if ("Early EOF".equals(exception.getMessage())) {
+    if (exception.getCause() instanceof org.eclipse.jetty.io.EofException
+        || exception.getMessage() != null && exception.getMessage().startsWith("Early EOF")) {
       // Some sort of timeout or broken connection
       return Response.status(Response.Status.BAD_REQUEST).build();
     }