MirrorMirror
/
Signal-Server
mirror of https://github.com/signalapp/Signal-Server.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Throw an exception instead of using Optional<Resposne>

This commit is contained in:
Chris Eager 2022-02-15 17:46:27 -08:00 committed by Chris Eager
parent 73fa3c3fe4
commit ba5e5a780f
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
service/src/main/java/org/whispersystems/textsecuregcm/controllers/MessageController.java
View File

@ -202,12 +202,7 @@ public class MessageController {
contentLength += message.getBody().length();
}
Optional<Response> maybeResponse = validateContentLength(contentLength, userAgent);
if (maybeResponse.isPresent()) {
return maybeResponse.get();
}
validateContentLength(contentLength, userAgent);
validateEnvelopeType(message.getType(), userAgent);
}
@ -778,17 +773,16 @@ public class MessageController {
}
}
private Optional<Response> validateContentLength(final int contentLength, final String userAgent) {
private void validateContentLength(final int contentLength, final String userAgent) {
Metrics.summary(CONTENT_SIZE_DISTRIBUTION_NAME, Tags.of(UserAgentTagUtil.getPlatformTag(userAgent)))
.record(contentLength);
if (contentLength > MAX_MESSAGE_SIZE) {
Metrics.counter(REJECT_OVERSIZE_MESSAGE_COUNTER, Tags.of(UserAgentTagUtil.getPlatformTag(userAgent)))
.increment();
return Optional.of(Response.status(Status.REQUEST_ENTITY_TOO_LARGE).build());
throw new WebApplicationException(Status.REQUEST_ENTITY_TOO_LARGE);
}
return Optional.empty();
}
private void validateEnvelopeType(final int type, final String userAgent) {